优化班次功能

service-store/src/main/java/com/java110/store/cmd/classes/ListClassesCmd.java

@@ -30,8 +30,10 @@ import com.java110.utils.util.BeanConvertUtil;
 import com.java110.vo.ResultVo;
 import org.springframework.beans.factory.annotation.Autowired;
 import com.java110.dto.classes.ClassesDto;
+
 import java.util.List;
 import java.util.ArrayList;
+
 import org.springframework.http.ResponseEntity;
 import org.springframework.http.HttpStatus;
 import org.slf4j.Logger;
@@ -51,7 +53,7 @@ import org.slf4j.LoggerFactory;
 @Java110Cmd(serviceCode = "classes.listClasses")
 public class ListClassesCmd extends Cmd {
 
-  private static Logger logger = LoggerFactory.getLogger(ListClassesCmd.class);
+    private static Logger logger = LoggerFactory.getLogger(ListClassesCmd.class);
     @Autowired
     private IClassesV1InnerServiceSMO classesV1InnerServiceSMOImpl;
 
@@ -63,22 +65,25 @@ public class ListClassesCmd extends Cmd {
     @Override
     public void doCmd(CmdEvent event, ICmdDataFlowContext cmdDataFlowContext, JSONObject reqJson) throws CmdException {
 
-           ClassesDto classesDto = BeanConvertUtil.covertBean(reqJson, ClassesDto.class);
+        String storeId = cmdDataFlowContext.getReqHeaders().get("store-id");
+
+        ClassesDto classesDto = BeanConvertUtil.covertBean(reqJson, ClassesDto.class);
+        classesDto.setStoreId(storeId);
 
-           int count = classesV1InnerServiceSMOImpl.queryClassessCount(classesDto);
+        int count = classesV1InnerServiceSMOImpl.queryClassessCount(classesDto);
 
-           List<ClassesDto> classesDtos = null;
+        List<ClassesDto> classesDtos = null;
 
-           if (count > 0) {
-               classesDtos = classesV1InnerServiceSMOImpl.queryClassess(classesDto);
-           } else {
-               classesDtos = new ArrayList<>();
-           }
+        if (count > 0) {
+            classesDtos = classesV1InnerServiceSMOImpl.queryClassess(classesDto);
+        } else {
+            classesDtos = new ArrayList<>();
+        }
 
-           ResultVo resultVo = new ResultVo((int) Math.ceil((double) count / (double) reqJson.getInteger("row")), count, classesDtos);
+        ResultVo resultVo = new ResultVo((int) Math.ceil((double) count / (double) reqJson.getInteger("row")), count, classesDtos);
 
-           ResponseEntity<String> responseEntity = new ResponseEntity<String>(resultVo.toString(), HttpStatus.OK);
+        ResponseEntity<String> responseEntity = new ResponseEntity<String>(resultVo.toString(), HttpStatus.OK);
 
-           cmdDataFlowContext.setResponseEntity(responseEntity);
+        cmdDataFlowContext.setResponseEntity(responseEntity);
     }
 }

service-store/src/main/java/com/java110/store/cmd/classes/SaveClassesCmd.java

@@ -62,10 +62,12 @@ public class SaveClassesCmd extends Cmd {
     @Override
     @Java110Transactional
     public void doCmd(CmdEvent event, ICmdDataFlowContext cmdDataFlowContext, JSONObject reqJson) throws CmdException {
+        String storeId = cmdDataFlowContext.getReqHeaders().get("store-id");
 
         ClassesPo classesPo = BeanConvertUtil.covertBean(reqJson, ClassesPo.class);
         classesPo.setClassesId(GenerateCodeFactory.getGeneratorId(CODE_PREFIX_ID));
         classesPo.setState(ClassesDto.STATE_START);
+        classesPo.setStoreId(storeId);
         int flag = classesV1InnerServiceSMOImpl.saveClasses(classesPo);
 
         if (flag < 1) {