任务添加代码

tall/src/main/java/com/ccsens/tall/service/InputDocService.java

@@ -254,19 +254,6 @@ public class InputDocService implements IInputDocService {
      */
     @Override
     public void addDoc(InputDocDto.AddDoc param) {
-        //判断是否有重复名称
-        /*ProTaskInputDocExample proTaskInputDocExample = new ProTaskInputDocExample();
-        proTaskInputDocExample.createCriteria().andTaskDetailIdEqualTo(param.getTaskId()).andRecStatusEqualTo((byte)0);
-        List<ProTaskInputDoc> proTaskInputDocs = proTaskInputDocMapper.selectByExample(proTaskInputDocExample);
-        if (null != proTaskInputDocs && 0 < proTaskInputDocs.size()){
-            for (ProTaskInputDoc proTaskInputDoc : proTaskInputDocs){
-                for (int i = 0; i < param.getFileName().length; i++) {
-                    if (proTaskInputDoc.getName().equals(param.getFileName()[i])){
-                        throw new BaseException(CodeEnum.DOC_REPEAT);
-                    }
-                }
-            }
-        }*/
 
         //查找任务详情id
         Long taskDetailId = null;
@@ -277,8 +264,8 @@ public class InputDocService implements IInputDocService {
             ProTaskSubTime proTaskSubTime = taskSubTimeDao.selectByPrimaryKey(param.getTaskId());
             if(ObjectUtil.isNotNull(proTaskSubTime)){
                 ProTaskDetail task = taskDetailDao.selectByPrimaryKey(proTaskSubTime.getTaskDetailId());
-                if(ObjectUtil.isNotNull(taskDetail)){
-                    taskDetailId = taskDetail.getId();
+                if(ObjectUtil.isNotNull(task)){
+                    taskDetailId = task.getId();
                 }
             }
         }
@@ -294,31 +281,14 @@ public class InputDocService implements IInputDocService {
             if(l > 0){
                 throw new BaseException(CodeEnum.DOC_REPEAT);
             }
-//            List<ProTaskInputDoc> proTaskInputDocs = proTaskInputDocMapper.selectByExample(proTaskInputDocExample);
-//            for (ProTaskInputDoc proTaskInputDoc : proTaskInputDocs){
-//                if(ObjectUtil.isNotNull(proTaskInputDoc)){
-//                    throw new BaseException(CodeEnum.DOC_REPEAT);
-//                }
-//            }
+            //添加输入文档
             ProTaskInputDoc proTaskInputDoc = new ProTaskInputDoc();
             proTaskInputDoc.setId(snowflake.nextId());
-            proTaskInputDoc.setTaskDetailId(param.getTaskId());
+            proTaskInputDoc.setTaskDetailId(taskDetailId);
             proTaskInputDoc.setName(param.getFileName()[i]);
-//            proTaskInputDoc.setIsUpload(0);
             proTaskInputDocMapper.insertSelective(proTaskInputDoc);
         }
 
-
-//        //添加输入文档
-//        for (int i = 0; i < param.getFileName().length; i++) {
-//            ProTaskInputDoc proTaskInputDoc = new ProTaskInputDoc();
-//            proTaskInputDoc.setId(snowflake.nextId());
-//            proTaskInputDoc.setTaskDetailId(param.getTaskId());
-//            proTaskInputDoc.setName(param.getFileName()[i]);
-////            proTaskInputDoc.setIsUpload(0);
-//            proTaskInputDocMapper.insertSelective(proTaskInputDoc);
-//        }
-
     }
 
     /**

tall/src/main/java/com/ccsens/tall/service/TaskDeliverService.java

@@ -1248,23 +1248,36 @@ public class TaskDeliverService implements ITaskDeliverService {
      */
     @Override
     public void addTaskDeliver(InputDocDto.addTaskDeliver param) {
+        //查找任务详情id
+        Long taskDetailId = null;
+        ProTaskDetail taskDetail = taskDetailDao.selectByPrimaryKey(param.getTaskId());
+        if(ObjectUtil.isNotNull(taskDetail)){
+            taskDetailId = taskDetail.getId();
+        }else {
+            ProTaskSubTime proTaskSubTime = taskSubTimeDao.selectByPrimaryKey(param.getTaskId());
+            if(ObjectUtil.isNotNull(proTaskSubTime)){
+                ProTaskDetail task = taskDetailDao.selectByPrimaryKey(proTaskSubTime.getTaskDetailId());
+                if(ObjectUtil.isNotNull(task)){
+                    taskDetailId = task.getId();
+                }
+            }
+        }
+        if(taskDetailId == null) {
+            throw new BaseException(CodeEnum.NOT_TASK);
+        }
+
         //判断是否有重复名称
         for(int i=0; i<param.getFileName().length; i++){
-            //String s = param.getFileName()[i];
             ProTaskDeliverExample proTaskDeliverExample = new ProTaskDeliverExample();
-            proTaskDeliverExample.createCriteria().andTaskDetailIdEqualTo(param.getTaskId()).andNameEqualTo(param.getFileName()[i]);
-            List<ProTaskDeliver> proTaskDelivers = taskDeliverDao.selectByExample(proTaskDeliverExample);
-            for (ProTaskDeliver proTaskDeliver : proTaskDelivers){
-                if(ObjectUtil.isNotNull(proTaskDeliver)){
+            proTaskDeliverExample.createCriteria().andTaskDetailIdEqualTo(taskDetailId).andNameEqualTo(param.getFileName()[i]);
+            long l = taskDeliverDao.countByExample(proTaskDeliverExample);
+            if(l>0){
                 throw new BaseException(CodeEnum.Deliver_REPEAT);
             }
-            }
-        }
             //添加交付物
-        for (int i = 0; i < param.getFileName().length; i++) {
             ProTaskDeliver proTaskDeliver = new ProTaskDeliver();
             proTaskDeliver.setId(snowflake.nextId());
-            proTaskDeliver.setTaskDetailId(param.getTaskId());
+            proTaskDeliver.setTaskDetailId(taskDetailId);
             proTaskDeliver.setName(param.getFileName()[i]);
             proTaskDeliver.setIsUpload(0);
             proTaskDeliver.setIsInput(0);