From e36c1e2363e36a69a3cc8ccbc00d28b16f926abd Mon Sep 17 00:00:00 2001
From: eight <641137800@qq.com>
Date: 星期四, 07 十一月 2024 14:38:57 +0800
Subject: [PATCH] 序号表操作

---
 jh-module-ecg/jh-module-ecg-biz/src/main/java/cn/lihu/jh/module/ecg/service/devrent/DevRentServiceImpl.java |  307 +++++++++++++++++++++++++++++++++-----------------
 1 files changed, 201 insertions(+), 106 deletions(-)

diff --git a/jh-module-ecg/jh-module-ecg-biz/src/main/java/cn/lihu/jh/module/ecg/service/devrent/DevRentServiceImpl.java b/jh-module-ecg/jh-module-ecg-biz/src/main/java/cn/lihu/jh/module/ecg/service/devrent/DevRentServiceImpl.java
index c0fcb55..4d11472 100644
--- a/jh-module-ecg/jh-module-ecg-biz/src/main/java/cn/lihu/jh/module/ecg/service/devrent/DevRentServiceImpl.java
+++ b/jh-module-ecg/jh-module-ecg-biz/src/main/java/cn/lihu/jh/module/ecg/service/devrent/DevRentServiceImpl.java
@@ -60,18 +60,26 @@
     }
 
     @Override
-    public CommonResult<Long> routineFinishOperation(DevRentSaveReqVO createReqVO) {
+    @Transactional(rollbackFor = Exception.class)
+    public CommonResult<Long> routineFinishOperation(RoutineFinishReqVO routineFinishReqVO) {
+        DevRentSaveReqVO createReqVO = BeanUtils.toBean(routineFinishReqVO, DevRentSaveReqVO.class);
+
         Long userId = SecurityFrameworkUtils.getLoginUserId();
         String userNickname = SecurityFrameworkUtils.getLoginUserNickname();
 
         // 鍚屾牱 鍦ㄩ绾︾‘瀹炴椂 灏辫繘鍏� [dev_rent] 琛ㄤ簡
         Long rent_id = createReqVO.getId();
-        if (null == rent_id) {
+
+        List<Integer> jobTypeList = new ArrayList<Integer>();
+        jobTypeList.add( DevRentStateEnum.ROUTINE_FINISH.getState());
+        jobTypeList.add( DevRentStateEnum.ROUTINE_CANCELLED.getState());
+        JobRecordDO jobRecordDO = jobRecordMapper.selectByRentIdDocId(rent_id, userId, jobTypeList);
+        if (null == jobRecordDO) {
             List<DevRentDO>  devRentDOList = devRentMapper.selectByPatIdAndState(createReqVO.getPatId(), DevRentStateEnum.FREE.getState());
             if (devRentDOList.size() == 0)
-                return CommonResult.error(DEV_INSTALL_NOT_APPOINTMENT);
+                throw  exception(DEV_INSTALL_NOT_APPOINTMENT);
             else if (devRentDOList.size() != 1) {
-                return CommonResult.error(DEV_INSTALL_EXCEPTION);
+                throw  exception(DEV_INSTALL_EXCEPTION);
             }
 
             rent_id = devRentDOList.getFirst().getId();
@@ -99,19 +107,13 @@
 
             updateDevRent( createReqVO );
 
-            List<Integer> jobTypeList = new ArrayList<Integer>();
-            jobTypeList.add( DevRentStateEnum.ROUTINE_FINISH.getState());
-            jobTypeList.add( DevRentStateEnum.READY_CANCELLED.getState());
-            JobRecordDO jobRecordDO = jobRecordMapper.selectByRentIdDocId(rent_id, userId, jobTypeList);
-            if (null != jobRecordDO) {
-                jobRecordDO.setDevId(createReqVO.getDevId());
-                jobRecordDO.setJobType( DevRentStateEnum.ROUTINE_FINISH.getState() );
-                jobRecordDO.setJobTime(createReqVO.getRentTime());
-                jobRecordDO.setRemark(createReqVO.getRemark());
-                jobRecordDO.setUpdater(String.valueOf(userId));
-                jobRecordDO.setUpdateTime(LocalDateTime.now());
-                jobRecordMapper.updateById(jobRecordDO);
-            }
+            jobRecordDO.setDevId(createReqVO.getDevId());
+            jobRecordDO.setJobType( DevRentStateEnum.ROUTINE_FINISH.getState() );
+            jobRecordDO.setJobTime(createReqVO.getRentTime());
+            jobRecordDO.setRemark(createReqVO.getRemark());
+            jobRecordDO.setUpdater(String.valueOf(userId));
+            jobRecordDO.setUpdateTime(LocalDateTime.now());
+            jobRecordMapper.updateById(jobRecordDO);
         }
 
         // 鏇存柊..闃熷垪鐘舵��
@@ -122,24 +124,28 @@
     }
 
     @Override
-    @Transactional
+    @Transactional(rollbackFor = Exception.class)
     public CommonResult<Long> readyOperation(DevRentSaveReqVO createReqVO) {
         Long userId = SecurityFrameworkUtils.getLoginUserId();
         String userNickname = SecurityFrameworkUtils.getLoginUserNickname();
+        Long rent_id = createReqVO.getId();
 
         // 鏍囨敞璁惧 宸查鍙�
-        Integer ret = markDevRecieved(createReqVO.getDevId());
+        Integer ret = markDevRecieved(createReqVO.getDevId(), rent_id);
         if (null == ret || 0 == ret) {
-            return CommonResult.error(DEVICE_NOT_FREE);
+            throw exception(DEVICE_NOT_FREE);
         }
 
-        Long rent_id = createReqVO.getId();
-        if (null == rent_id) {
+        List<Integer> jobTypeList = new ArrayList<Integer>();
+        jobTypeList.add( DevRentStateEnum.READY.getState());
+        jobTypeList.add( DevRentStateEnum.READY_CANCELLED.getState());
+        JobRecordDO jobRecordDO = jobRecordMapper.selectByRentIdDocId(rent_id, userId, jobTypeList);
+        if (null == jobRecordDO) {
             List<DevRentDO>  devRentDOList = devRentMapper.selectByPatIdAndState(createReqVO.getPatId(), DevRentStateEnum.FREE.getState());
             if (devRentDOList.size() == 0)
-                return CommonResult.error(DEV_INSTALL_NOT_APPOINTMENT);
+                throw exception(DEV_INSTALL_NOT_APPOINTMENT);
             else if (devRentDOList.size() != 1) {
-                return CommonResult.error(DEV_INSTALL_EXCEPTION);
+                throw exception(DEV_INSTALL_EXCEPTION);
             }
 
             rent_id = devRentDOList.getFirst().getId();
@@ -167,19 +173,13 @@
 
             updateDevRent( createReqVO );
 
-            List<Integer> jobTypeList = new ArrayList<Integer>();
-            jobTypeList.add( DevRentStateEnum.READY.getState());
-            jobTypeList.add( DevRentStateEnum.READY_CANCELLED.getState());
-            JobRecordDO jobRecordDO = jobRecordMapper.selectByRentIdDocId(rent_id, userId, jobTypeList);
-            if (null != jobRecordDO) {
-                jobRecordDO.setDevId(createReqVO.getDevId());
-                jobRecordDO.setJobType( DevRentStateEnum.READY.getState() );
-                jobRecordDO.setJobTime(createReqVO.getRentTime());
-                jobRecordDO.setRemark(createReqVO.getRemark());
-                jobRecordDO.setUpdater(String.valueOf(userId));
-                jobRecordDO.setUpdateTime(LocalDateTime.now());
-                jobRecordMapper.updateById(jobRecordDO);
-            }
+            jobRecordDO.setDevId(createReqVO.getDevId());
+            jobRecordDO.setJobType( DevRentStateEnum.READY.getState() );
+            jobRecordDO.setJobTime(createReqVO.getRentTime());
+            jobRecordDO.setRemark(createReqVO.getRemark());
+            jobRecordDO.setUpdater(String.valueOf(userId));
+            jobRecordDO.setUpdateTime(LocalDateTime.now());
+            jobRecordMapper.updateById(jobRecordDO);
         }
 
         // 鏇存柊..闃熷垪鐘舵��
@@ -190,18 +190,23 @@
     }
 
     @Override
-    @Transactional
+    @Transactional(rollbackFor = Exception.class)
     public CommonResult<Long> installOperation(DevRentSaveReqVO createReqVO) {
         Long userId = SecurityFrameworkUtils.getLoginUserId();
         String userNickname = SecurityFrameworkUtils.getLoginUserNickname();
+        Long rent_id = createReqVO.getId();
 
-        // 璁剧疆璁惧 浣跨敤涓�
-        Integer ret = markDevInUse(createReqVO.getDevId());
-        if (null == ret || 0 == ret) {
-            return CommonResult.error(DEVICE_NOT_RECEIVED);
+        // 瀵逛簬宸叉湁棰嗙敤锛屼絾鏄鏈烘椂浣跨敤鐨勪笉鏄凡棰嗙敤璁惧锛岄渶閲婃斁 婧愰鐢� 璁惧
+        DevRentDO devRentDO = getDevRent(rent_id);
+        if ( null != devRentDO.getDevId() && !createReqVO.getDevId().equals(devRentDO.getDevId()) ) {
+            markDevFree( devRentDO.getDevId(), rent_id);
         }
 
-        Long rent_id = createReqVO.getId();
+        // 璁剧疆璁惧 浣跨敤涓�
+        Integer ret = markDevInUse(createReqVO.getDevId(), rent_id);
+        if (null == ret || 0 == ret) {
+            throw exception(DEVICE_NOT_RECEIVED);
+        }
 
         // 妫�鏌� job_record 鐢ㄤ簬纭鏄惁鏄涓�娆″畨瑁呮搷浣�
         List<Integer> jobTypeList = new ArrayList<Integer>();
@@ -213,9 +218,9 @@
         if (null == jobRecordDO) {
             List<DevRentDO>  devRentDOList = devRentMapper.selectByPatIdAndState(createReqVO.getPatId(), DevRentStateEnum.READY.getState());
             if (devRentDOList.size() == 0)
-                return CommonResult.error(DEV_INSTALL_NOT_RECEIVED);
+                throw exception(DEV_INSTALL_NOT_RECEIVED);
             else if (devRentDOList.size() != 1) {
-                return CommonResult.error(DEV_INSTALL_EXCEPTION);
+                throw exception(DEV_INSTALL_EXCEPTION);
             }
 
             rent_id = devRentDOList.getFirst().getId();
@@ -259,21 +264,22 @@
     }
 
     @Override
-    @Transactional
+    @Transactional(rollbackFor = Exception.class)
     public Long dismantleOperation(DevRentSaveReqVO updateReqVO) {
         updateReqVO.setState( DevRentStateEnum.DISMANTLED.getState() );
         updateDevRent( updateReqVO );
+        Long rentId = updateReqVO.getId();
 
         Long userId = SecurityFrameworkUtils.getLoginUserId();
         String userNickname = SecurityFrameworkUtils.getLoginUserNickname();
 
         // 璁惧鎭㈠鍒扮┖闂�
-        resumeDevFree( updateReqVO.getDevId() );
+        markDevFree( updateReqVO.getDevId(), rentId );
 
         List<Integer> jobTypeList = new ArrayList<Integer>();
         jobTypeList.add( DevRentStateEnum.DISMANTLED.getState());
         JobRecordDO jobRecordDO = jobRecordMapper.selectByRentIdDocId(updateReqVO.getId(), userId, jobTypeList);
-        if (jobRecordDO == null) {
+        if (null == jobRecordDO) {
             JobRecordSaveReqVO jobRecordSaveReqVO = new JobRecordSaveReqVO();
             jobRecordSaveReqVO.setRentId( updateReqVO.getId() );
             jobRecordSaveReqVO.setDevId( updateReqVO.getDevId() );
@@ -300,7 +306,7 @@
     }
 
     @Override
-    @Transactional
+    @Transactional(rollbackFor = Exception.class)
     public Long dataEntryOperation(DevRentSaveReqVO updateReqVO) {
         updateReqVO.setState( DevRentStateEnum.DATAENTERED.getState() );
         updateDevRent( updateReqVO );
@@ -311,7 +317,7 @@
         List<Integer> jobTypeList = new ArrayList<Integer>();
         jobTypeList.add( DevRentStateEnum.DATAENTERED.getState());
         JobRecordDO jobRecordDO = jobRecordMapper.selectByRentIdDocId(updateReqVO.getId(), userId, jobTypeList);
-        if (jobRecordDO == null) {
+        if (null == jobRecordDO) {
             JobRecordSaveReqVO jobRecordSaveReqVO = new JobRecordSaveReqVO();
             jobRecordSaveReqVO.setRentId( updateReqVO.getId() );
             jobRecordSaveReqVO.setDevId( updateReqVO.getDevId() );
@@ -336,15 +342,71 @@
         return updateReqVO.getId();
     }
 
+    // 甯歌妫�鏌� 鍙栨秷
     @Override
-    @Transactional
-    public CommonResult<Long> cancelReadyOperation(DevCancelReqVO cancelReqVO) {
+    @Transactional(rollbackFor = Exception.class)
+    public CommonResult<Long> cancelRoutineOperation(DevCancelReqVO cancelReqVO) {
         Long userId = SecurityFrameworkUtils.getLoginUserId();
         String userNickname = SecurityFrameworkUtils.getLoginUserNickname();
 
+        DevRentSaveReqVO createReqVO = BeanUtils.toBean(cancelReqVO, DevRentSaveReqVO.class);
+        createReqVO.setDevId(null);
+        createReqVO.setState( DevRentStateEnum.ROUTINE_CANCELLED.getState());
+
+        Long rent_id = createReqVO.getId();
+
+        List<Integer> jobTypeList = new ArrayList<Integer>();
+        jobTypeList.add( DevRentStateEnum.ROUTINE_FINISH.getState());
+        jobTypeList.add( DevRentStateEnum.ROUTINE_CANCELLED.getState());
+        JobRecordDO jobRecordDO = jobRecordMapper.selectByRentIdDocId(rent_id, userId, jobTypeList);
+        if (null == jobRecordDO) {
+            List<DevRentDO>  devRentDOList = devRentMapper.selectByPatIdAndState(createReqVO.getPatId(), DevRentStateEnum.FREE.getState());
+            if (devRentDOList.size() != 1)
+                throw exception(ROUTINE_CHECK_EXIST);
+
+            rent_id = devRentDOList.getFirst().getId();
+            createReqVO.setId( rent_id );
+            createReqVO.setState( DevRentStateEnum.ROUTINE_CANCELLED.getState() );
+            updateDevRent( createReqVO );
+
+            JobRecordSaveReqVO jobRecordSaveReqVO = new JobRecordSaveReqVO();
+            jobRecordSaveReqVO.setRentId(rent_id);
+            jobRecordSaveReqVO.setDevId(null);
+            jobRecordSaveReqVO.setDocId(userId);
+            jobRecordSaveReqVO.setDocName(userNickname);
+            jobRecordSaveReqVO.setPatId(createReqVO.getPatId());
+            jobRecordSaveReqVO.setPatName(createReqVO.getPatName());
+            jobRecordSaveReqVO.setJobType( DevRentStateEnum.ROUTINE_CANCELLED.getState() );
+            jobRecordSaveReqVO.setJobTime(createReqVO.getRentTime());
+            jobRecordSaveReqVO.setRemark(createReqVO.getRemark());
+            jobRecordSaveReqVO.setSummary("");
+
+            JobRecordDO jobRecord = BeanUtils.toBean(jobRecordSaveReqVO, JobRecordDO.class);
+            jobRecordMapper.insert(jobRecord);
+        } else {
+            updateDevRent( createReqVO );
+
+            jobRecordDO.setDevId(null);
+            jobRecordDO.setJobType(DevRentStateEnum.ROUTINE_CANCELLED.getState());
+            jobRecordDO.setJobTime(createReqVO.getRentTime());
+            jobRecordDO.setRemark(createReqVO.getRemark());
+            jobRecordDO.setUpdater(String.valueOf(userId));
+            jobRecordDO.setUpdateTime(LocalDateTime.now());
+            jobRecordMapper.updateById(jobRecordDO);
+        }
+        return CommonResult.success(rent_id);
+    }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public CommonResult<Long> cancelReadyOperation(DevCancelReqVO cancelReqVO) {
+        Long userId = SecurityFrameworkUtils.getLoginUserId();
+        String userNickname = SecurityFrameworkUtils.getLoginUserNickname();
+        Long rentId = cancelReqVO.getId();
+
         // 鑻ヤ粠 瑁呮満 鏀逛负 鍙栨秷瑁呮満锛屾仮澶嶈澶囩姸鎬�
         if (null != cancelReqVO.getDevId()) {
-            resumeDevFree( cancelReqVO.getDevId() );
+            markDevFree( cancelReqVO.getDevId(),  rentId );
         }
 
         DevRentSaveReqVO createReqVO = BeanUtils.toBean(cancelReqVO, DevRentSaveReqVO.class);
@@ -352,10 +414,15 @@
         createReqVO.setState( DevRentStateEnum.READY_CANCELLED.getState());
 
         Long rent_id = createReqVO.getId();
-        if (null == rent_id) {
+
+        List<Integer> jobTypeList = new ArrayList<Integer>();
+        jobTypeList.add( DevRentStateEnum.READY.getState());
+        jobTypeList.add( DevRentStateEnum.READY_CANCELLED.getState());
+        JobRecordDO jobRecordDO = jobRecordMapper.selectByRentIdDocId(rent_id, userId, jobTypeList);
+        if (null == jobRecordDO) {
             List<DevRentDO>  devRentDOList = devRentMapper.selectByPatIdAndState(createReqVO.getPatId(), DevRentStateEnum.FREE.getState());
             if (devRentDOList.size() != 1)
-                return CommonResult.error(DEV_INSTALL_EXIST);
+                throw exception(DEV_INSTALL_EXIST);
 
             rent_id = devRentDOList.getFirst().getId();
             createReqVO.setId( rent_id );
@@ -379,32 +446,27 @@
         } else {
             updateDevRent( createReqVO );
 
-            List<Integer> jobTypeList = new ArrayList<Integer>();
-            jobTypeList.add( DevRentStateEnum.READY.getState());
-            jobTypeList.add( DevRentStateEnum.READY_CANCELLED.getState());
-            JobRecordDO jobRecordDO = jobRecordMapper.selectByRentIdDocId(rent_id, userId, jobTypeList);
-            if (null != jobRecordDO) {
-                jobRecordDO.setDevId(null);
-                jobRecordDO.setJobType(DevRentStateEnum.READY_CANCELLED.getState());
-                jobRecordDO.setJobTime(createReqVO.getRentTime());
-                jobRecordDO.setRemark(createReqVO.getRemark());
-                jobRecordDO.setUpdater(String.valueOf(userId));
-                jobRecordDO.setUpdateTime(LocalDateTime.now());
-                jobRecordMapper.updateById(jobRecordDO);
-            }
+            jobRecordDO.setDevId(null);
+            jobRecordDO.setJobType(DevRentStateEnum.READY_CANCELLED.getState());
+            jobRecordDO.setJobTime(createReqVO.getRentTime());
+            jobRecordDO.setRemark(createReqVO.getRemark());
+            jobRecordDO.setUpdater(String.valueOf(userId));
+            jobRecordDO.setUpdateTime(LocalDateTime.now());
+            jobRecordMapper.updateById(jobRecordDO);
         }
         return CommonResult.success(rent_id);
     }
 
     @Override
-    @Transactional
+    @Transactional(rollbackFor = Exception.class)
     public CommonResult<Long> cancelInstallOperation(DevCancelReqVO cancelReqVO) {
         Long userId = SecurityFrameworkUtils.getLoginUserId();
         String userNickname = SecurityFrameworkUtils.getLoginUserNickname();
+        Long rentId = cancelReqVO.getId();
 
         // 鑻ヤ粠 瑁呮満 鏀逛负 鍙栨秷瑁呮満锛屾仮澶嶈澶囩姸鎬�
         if (null != cancelReqVO.getDevId()) {
-            resumeDevFree( cancelReqVO.getDevId() );
+            markDevFree( cancelReqVO.getDevId(), rentId );
         }
 
         DevRentSaveReqVO createReqVO = BeanUtils.toBean(cancelReqVO, DevRentSaveReqVO.class);
@@ -420,7 +482,7 @@
         if (null == jobRecordDO) {
             List<DevRentDO>  devRentDOList = devRentMapper.selectByPatIdAndState(createReqVO.getPatId(), DevRentStateEnum.READY.getState());
             if (devRentDOList.size() != 1)
-                return CommonResult.error(DEV_INSTALL_EXIST);
+                throw exception(DEV_INSTALL_EXIST);
 
             rent_id = devRentDOList.getFirst().getId();
             createReqVO.setId( rent_id );
@@ -456,21 +518,35 @@
     }
 
     @Override
-    @Transactional
-    public CommonResult<Long> loseOperation(DevRentSaveReqVO updateReqVO) {
-        // 璁惧閬楀け鏃�, 鏍囨敞閬楀け
-        markDevLost( updateReqVO.getDevId(), updateReqVO.getReturnTime().toLocalDate() );
+    @Transactional(rollbackFor = Exception.class)
+    public CommonResult<Long> dismantleExceptionOperation(DevRentSaveReqVO updateReqVO) {
+        DevStateEnum devStateEnum = null;
+        DevRentStateEnum devRentStateEnum = null;
+        if (updateReqVO.getState() == DevStateEnum.DAMAGED.getState()) {
+            devStateEnum = DevStateEnum.DAMAGED;
+            devRentStateEnum = DevRentStateEnum.DAMAGED;
+        }
+        else if (updateReqVO.getState() == DevStateEnum.LOST.getState()) {
+            devStateEnum = DevStateEnum.LOST;
+            devRentStateEnum = DevRentStateEnum.LOST;
+        }
+        else
+            throw exception(DEV_DISMANTLE_EXCEPTION);
 
-        updateReqVO.setState( DevRentStateEnum.LOST.getState() );
+        // 璁惧閬楀け鏃�, 鏍囨敞閬楀け
+        markDevDismantleException( updateReqVO.getDevId(), updateReqVO.getId(), updateReqVO.getReturnTime().toLocalDate(), devStateEnum );
+
+        updateReqVO.setState( devRentStateEnum.getState() );
         updateDevRent( updateReqVO );
 
         Long userId = SecurityFrameworkUtils.getLoginUserId();
         String userNickname = SecurityFrameworkUtils.getLoginUserNickname();
 
         List<Integer> jobTypeList = new ArrayList<Integer>();
+        jobTypeList.add( DevRentStateEnum.DAMAGED.getState());
         jobTypeList.add( DevRentStateEnum.LOST.getState());
         JobRecordDO jobRecordDO = jobRecordMapper.selectByRentIdDocId(updateReqVO.getId(), userId, jobTypeList);
-        if (jobRecordDO == null) {
+        if (null == jobRecordDO) {
             JobRecordSaveReqVO jobRecordSaveReqVO = new JobRecordSaveReqVO();
             jobRecordSaveReqVO.setRentId( updateReqVO.getId() );
             jobRecordSaveReqVO.setDevId( updateReqVO.getDevId() );
@@ -479,13 +555,13 @@
             jobRecordSaveReqVO.setPatId( updateReqVO.getPatId() );
             jobRecordSaveReqVO.setPatName( updateReqVO.getPatName() );
             jobRecordSaveReqVO.setJobTime( updateReqVO.getReturnTime() );
-            jobRecordSaveReqVO.setJobType( DevRentStateEnum.LOST.getState() );
+            jobRecordSaveReqVO.setJobType( devRentStateEnum.getState() );
             jobRecordSaveReqVO.setRemark( updateReqVO.getRemark() );
             jobRecordSaveReqVO.setSummary( updateReqVO.getInterference() + " " + updateReqVO.getBaseline() + " " + updateReqVO.getDetachment() );
             JobRecordDO jobRecord = BeanUtils.toBean(jobRecordSaveReqVO, JobRecordDO.class);
             jobRecordMapper.insert(jobRecord);
         } else {
-            jobRecordDO.setJobType( DevRentStateEnum.LOST.getState() );
+            jobRecordDO.setJobType( devRentStateEnum.getState() );
             jobRecordDO.setJobTime( updateReqVO.getReturnTime() );
             jobRecordDO.setRemark( updateReqVO.getRemark() );
             jobRecordDO.setSummary( updateReqVO.getInterference() + " " + updateReqVO.getBaseline() + " " + updateReqVO.getDetachment() );
@@ -499,8 +575,6 @@
 
     @Override
     public Long createDevRent(DevRentSaveReqVO createReqVO) {
-        //
-
         // 鎻掑叆
         DevRentDO devRent = BeanUtils.toBean(createReqVO, DevRentDO.class);
         devRentMapper.insert(devRent);
@@ -551,6 +625,14 @@
     }
 
     @Override
+    public DevRentDO getFreeRent(DevRentSearchReqVO searchReqVO) {
+        List<Integer> stateList = new ArrayList<>();
+        stateList.add( DevRentStateEnum.FREE.getState() );
+        searchReqVO.setStateList( stateList );
+        return devRentMapper.getRentByState(searchReqVO);
+    }
+
+    @Override
     public DevRentDO getReadyOrCancelledRent(DevRentSearchReqVO searchReqVO) {
         List<Integer> stateList = new ArrayList<>();
         stateList.add( DevRentStateEnum.READY.getState() );
@@ -567,49 +649,62 @@
         return devRentMapper.getRentByState(searchReqVO);
     }
 
-    private Integer markDevRecieved(String devId) {
+    // 鏍囪 璁惧宸茶棰嗗彇
+    private Integer markDevRecieved(String devId, Long rentId) {
         // 璁惧鏍囨敞 宸查鍙�
-        DeviceDO deviveDO = deviceMapper.getDeviceByDevId(devId);
-        if (null != deviveDO && deviveDO.getState() == DevStateEnum.FREE.getState()) {
-            deviveDO.setState( DevStateEnum.RECEIVED.getState() );
-            deviveDO.setStateDate( LocalDate.now() );
-            return deviceMapper.updateById(deviveDO);
+        DeviceDO deviceDO = deviceMapper.getDeviceByDevId(devId);
+        if (null != deviceDO && DevStateEnum.FREE.isEqual(deviceDO.getState())) {
+            deviceDO.setState( DevStateEnum.RECEIVED.getState() );
+            deviceDO.setRentId( rentId );
+            deviceDO.setStateDate( LocalDate.now() );
+            return deviceMapper.updateById(deviceDO);
         }
 
         return 0;
     }
 
-    private Integer markDevInUse(String devId) {
-        // 璁惧鏍囨敞 鍦ㄤ娇鐢�
-        DeviceDO deviveDO = deviceMapper.getDeviceByDevId(devId);
-        if (null != deviveDO && (deviveDO.getState() == DevStateEnum.RECEIVED.getState() || deviveDO.getState() == DevStateEnum.FREE.getState())) {
-            deviveDO.setState( DevStateEnum.INUSE.getState() );
-            deviveDO.setStateDate( LocalDate.now() );
-            return deviceMapper.updateById(deviveDO);
+    // 鏍囪 璁惧宸茶浣跨敤
+    private Integer markDevInUse(String devId, Long rentId) {
+        DeviceDO deviceDO = deviceMapper.getDeviceByDevId(devId);
+        if (null == deviceDO)
+            return 0;
+
+        if (DevStateEnum.RECEIVED.isEqual(deviceDO.getState()) && rentId.equals(deviceDO.getRentId())) {
+            deviceDO.setState( DevStateEnum.INUSE.getState() );
+            deviceDO.setStateDate( LocalDate.now() );
+            return deviceMapper.updateById(deviceDO);
+        }
+
+        if (DevStateEnum.FREE.isEqual(deviceDO.getState())) {
+            deviceDO.setState( DevStateEnum.INUSE.getState() );
+            deviceDO.setStateDate( LocalDate.now() );
+            deviceDO.setRentId( rentId );
+            return deviceMapper.updateById(deviceDO);
         }
 
         return 0;
     }
 
-    private Integer markDevLost(String devId, LocalDate localDate) {
+    private Integer markDevDismantleException(String devId, Long rentId, LocalDate localDate, DevStateEnum devStateEnum) {
         // 璁惧鏍囨敞閬楀け
-        DeviceDO deviveDO = deviceMapper.getDeviceByDevId(devId);
-        if (null != deviveDO) {
-            deviveDO.setState( DevStateEnum.LOST.getState() );
-            deviveDO.setStateDate( localDate );
-            return deviceMapper.updateById(deviveDO);
+        DeviceDO deviceDO = deviceMapper.getDeviceByDevId(devId);
+        if (null != deviceDO && DevStateEnum.INUSE.isEqual(deviceDO.getState()) && rentId.equals(deviceDO.getRentId())) {
+            deviceDO.setState( devStateEnum.getState() );
+            deviceDO.setStateDate( localDate );
+            return deviceMapper.updateById(deviceDO);
         }
 
         return 0;
     }
 
-    private Integer resumeDevFree(String devId) {
+    private Integer markDevFree(String devId, Long rentId) {
         // 璁惧鏍囨敞閬楀け 鎭㈠涓虹┖闂�
-        DeviceDO deviveDO = deviceMapper.getDeviceByDevId( devId );
-        if (null != deviveDO) {
-            deviveDO.setState( DevStateEnum.FREE.getState() );
-            deviveDO.setStateDate( LocalDate.now() );
-            return deviceMapper.updateById(deviveDO);
+        DeviceDO deviceDO = deviceMapper.getDeviceByDevId( devId );
+        if (null != deviceDO && rentId.equals(deviceDO.getRentId())) {
+            deviceDO.setState( DevStateEnum.FREE.getState() );
+            deviceDO.setStateDate( LocalDate.now() );
+            deviceDO.setRentId(null);
+            return deviceMapper.updateById(deviceDO);
         }
         
         return 0;

--
Gitblit v1.9.3