优化代码

service-store/src/main/java/com/java110/store/api/ContractApi.java

@@ -141,7 +141,7 @@ public class ContractApi {
 
 
         ContractPo contractPo = BeanConvertUtil.covertBean(reqJson, ContractPo.class);
-        return updateContractBMOImpl.update(contractPo);
+        return updateContractBMOImpl.update(contractPo,reqJson);
     }
 
     /**

service-store/src/main/java/com/java110/store/bmo/contract/IUpdateContractBMO.java

@@ -1,4 +1,5 @@
 package com.java110.store.bmo.contract;
+import com.alibaba.fastjson.JSONObject;
 import com.java110.po.contract.ContractPo;
 import org.springframework.http.ResponseEntity;
 
@@ -11,7 +12,7 @@ public interface IUpdateContractBMO {
      * @param contractPo
      * @return
      */
-    ResponseEntity<String> update(ContractPo contractPo);
+    ResponseEntity<String> update(ContractPo contractPo, JSONObject reqJson);
 
 
 }

service-store/src/main/java/com/java110/store/bmo/contract/impl/UpdateContractBMOImpl.java

@@ -1,34 +1,82 @@
 package com.java110.store.bmo.contract.impl;
 
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
 import com.java110.core.annotation.Java110Transactional;
+import com.java110.dto.contractAttr.ContractAttrDto;
+import com.java110.intf.store.IContractAttrInnerServiceSMO;
 import com.java110.intf.store.IContractInnerServiceSMO;
 import com.java110.po.contract.ContractPo;
+import com.java110.po.contractAttr.ContractAttrPo;
 import com.java110.store.bmo.contract.IUpdateContractBMO;
+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("updateContractBMOImpl")
 public class UpdateContractBMOImpl implements IUpdateContractBMO {
 
     @Autowired
     private IContractInnerServiceSMO contractInnerServiceSMOImpl;
 
+    @Autowired
+    private IContractAttrInnerServiceSMO contractAttrInnerServiceSMOImpl;
+
     /**
      * @param contractPo
      * @return 订单服务能够接受的报文
      */
     @Java110Transactional
-    public ResponseEntity<String> update(ContractPo contractPo) {
+    public ResponseEntity<String> update(ContractPo contractPo, JSONObject reqJson) {
 
         int flag = contractInnerServiceSMOImpl.updateContract(contractPo);
 
-        if (flag > 0) {
+        if (flag < 1) {
+            return ResultVo.createResponseEntity(ResultVo.CODE_ERROR, "保存失败");
+
+        }
+
+        if (!reqJson.containsKey("contractTypeSpecs")) {
             return ResultVo.createResponseEntity(ResultVo.CODE_OK, "保存成功");
         }
 
-        return ResultVo.createResponseEntity(ResultVo.CODE_ERROR, "保存失败");
+
+        JSONArray contractTypeSpecs = reqJson.getJSONArray("contractTypeSpecs");
+
+        if (contractTypeSpecs.size() < 1) {
+            return ResultVo.createResponseEntity(ResultVo.CODE_OK, "保存成功");
+        }
+
+        for (int typeSpecIndex = 0; typeSpecIndex < contractTypeSpecs.size(); typeSpecIndex++) {
+            updateContractAttr(contractTypeSpecs.getJSONObject(typeSpecIndex), contractPo);
+        }
+
+        return ResultVo.createResponseEntity(ResultVo.CODE_OK, "保存成功");
+
+    }
+
+    private void updateContractAttr(JSONObject jsonObject, ContractPo contractPo) {
+        ContractAttrDto contractAttrDto = new ContractAttrDto();
+        contractAttrDto.setContractId(contractPo.getContractId());
+        contractAttrDto.setSpecCd(jsonObject.getString("specCd"));
+        List<ContractAttrDto> contractAttrDtos = contractAttrInnerServiceSMOImpl.queryContractAttrs(contractAttrDto);
+
+        Assert.listOnlyOne(contractAttrDtos, "未找到需要修改的合同属性");
+        ContractAttrPo contractAttrPo = new ContractAttrPo();
+        contractAttrPo.setAttrId(contractAttrDtos.get(0).getAttrId());
+        contractAttrPo.setContractId(contractPo.getContractId());
+        contractAttrPo.setSpecCd(jsonObject.getString("specCd"));
+        contractAttrPo.setValue(jsonObject.getString("value"));
+        contractAttrPo.setStoreId(contractPo.getStoreId());
+        int count = contractAttrInnerServiceSMOImpl.updateContractAttr(contractAttrPo);
+
+        if (count < 1) {
+            throw new IllegalArgumentException("保存属性失败");
+        }
     }
 
 }