From e9be8d15ab3ee20685793e16071c56485b32e33a Mon Sep 17 00:00:00 2001
From: wuxw <928255095@qq.com>
Date: 星期五, 23 四月 2021 10:10:24 +0800
Subject: [PATCH] 优化代码

---
 service-common/src/main/java/com/java110/common/bmo/attendanceClassesTask/impl/SaveAttendanceClassesTaskBMOImpl.java |   31 ++++++++++++++++++++++++++++---
 1 files changed, 28 insertions(+), 3 deletions(-)

diff --git a/service-common/src/main/java/com/java110/common/bmo/attendanceClassesTask/impl/SaveAttendanceClassesTaskBMOImpl.java b/service-common/src/main/java/com/java110/common/bmo/attendanceClassesTask/impl/SaveAttendanceClassesTaskBMOImpl.java
old mode 100644
new mode 100755
index a2ddfed..2f080c5
--- a/service-common/src/main/java/com/java110/common/bmo/attendanceClassesTask/impl/SaveAttendanceClassesTaskBMOImpl.java
+++ b/service-common/src/main/java/com/java110/common/bmo/attendanceClassesTask/impl/SaveAttendanceClassesTaskBMOImpl.java
@@ -2,19 +2,30 @@
 
 import com.java110.common.bmo.attendanceClassesTask.ISaveAttendanceClassesTaskBMO;
 import com.java110.core.annotation.Java110Transactional;
-import com.java110.core.factory.GenerateCodeFactory;
+import com.java110.dto.attendanceClasses.AttendanceClassesDto;
+import com.java110.intf.common.IAttendanceClassesInnerServiceSMO;
+import com.java110.intf.common.IAttendanceClassesTaskDetailInnerServiceSMO;
 import com.java110.intf.common.IAttendanceClassesTaskInnerServiceSMO;
 import com.java110.po.attendanceClassesTask.AttendanceClassesTaskPo;
+import com.java110.po.attendanceClassesTaskDetail.AttendanceClassesTaskDetailPo;
+import com.java110.utils.util.Assert;
 import com.java110.vo.ResultVo;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.ResponseEntity;
 import org.springframework.stereotype.Service;
+
+import java.util.List;
 
 @Service("saveAttendanceClassesTaskBMOImpl")
 public class SaveAttendanceClassesTaskBMOImpl implements ISaveAttendanceClassesTaskBMO {
 
     @Autowired
     private IAttendanceClassesTaskInnerServiceSMO attendanceClassesTaskInnerServiceSMOImpl;
+
+    @Autowired
+    private IAttendanceClassesTaskDetailInnerServiceSMO attendanceClassesTaskDetailInnerServiceSMOImpl;
+    @Autowired
+    private IAttendanceClassesInnerServiceSMO attendanceClassesInnerServiceSMOImpl;
 
     /**
      * 娣诲姞灏忓尯淇℃伅
@@ -23,11 +34,25 @@
      * @return 璁㈠崟鏈嶅姟鑳藉鎺ュ彈鐨勬姤鏂�
      */
     @Java110Transactional
-    public ResponseEntity<String> save(AttendanceClassesTaskPo attendanceClassesTaskPo) {
+    public ResponseEntity<String> save(AttendanceClassesTaskPo attendanceClassesTaskPo,
+                                       List<AttendanceClassesTaskDetailPo> attendanceClassesTaskDetailPos) {
 
-        attendanceClassesTaskPo.setTaskId(GenerateCodeFactory.getGeneratorId(GenerateCodeFactory.CODE_PREFIX_taskId));
+        //attendanceClassesTaskPo.setTaskId(GenerateCodeFactory.getGeneratorId(GenerateCodeFactory.CODE_PREFIX_taskId));
+        //鏌ヨ鐝粍鏄惁瀛樺湪
+        AttendanceClassesDto attendanceClassesDto = new AttendanceClassesDto();
+        attendanceClassesDto.setClassesId(attendanceClassesTaskPo.getClassId());
+        List<AttendanceClassesDto> attendanceClassesDtos = attendanceClassesInnerServiceSMOImpl.queryAttendanceClassess(attendanceClassesDto);
+
+        Assert.listOnlyOne(attendanceClassesDtos, "鐝粍涓嶅瓨鍦�");
+
+        attendanceClassesTaskPo.setStoreId(attendanceClassesDtos.get(0).getStoreId());
         int flag = attendanceClassesTaskInnerServiceSMOImpl.saveAttendanceClassesTask(attendanceClassesTaskPo);
 
+        for (AttendanceClassesTaskDetailPo attendanceClassesTaskDetailPo : attendanceClassesTaskDetailPos) {
+            attendanceClassesTaskDetailPo.setStoreId(attendanceClassesDtos.get(0).getStoreId());
+            attendanceClassesTaskDetailInnerServiceSMOImpl.saveAttendanceClassesTaskDetail(attendanceClassesTaskDetailPo);
+        }
+
         if (flag > 0) {
             return ResultVo.createResponseEntity(ResultVo.CODE_OK, "淇濆瓨鎴愬姛");
         }

--
Gitblit v1.8.0