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 |   17 +++++++++++++++--
 1 files changed, 15 insertions(+), 2 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 2d35ff7..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,11 +2,13 @@
 
 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;
@@ -22,6 +24,8 @@
 
     @Autowired
     private IAttendanceClassesTaskDetailInnerServiceSMO attendanceClassesTaskDetailInnerServiceSMOImpl;
+    @Autowired
+    private IAttendanceClassesInnerServiceSMO attendanceClassesInnerServiceSMOImpl;
 
     /**
      * 娣诲姞灏忓尯淇℃伅
@@ -34,9 +38,18 @@
                                        List<AttendanceClassesTaskDetailPo> attendanceClassesTaskDetailPos) {
 
         //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){
+        for (AttendanceClassesTaskDetailPo attendanceClassesTaskDetailPo : attendanceClassesTaskDetailPos) {
+            attendanceClassesTaskDetailPo.setStoreId(attendanceClassesDtos.get(0).getStoreId());
             attendanceClassesTaskDetailInnerServiceSMOImpl.saveAttendanceClassesTaskDetail(attendanceClassesTaskDetailPo);
         }
 

--
Gitblit v1.8.0