楚江ERP：锁规接口

@@ -455,6 +455,7 @@ public class ContractDistributorStandardController extends DefaultBaseController
         }
         }
 
 
         // todo ymk 锁规接口待完善
         // todo ymk 锁规接口待完善
+        contractDistributorStandardService.specificationLock(vo);
 
 
         contractDistributorStandardService.cleanCacheByKey(vo.getId());
         contractDistributorStandardService.cleanCacheByKey(vo.getId());
 
 

@@ -41,6 +41,7 @@ import org.springframework.transaction.annotation.Transactional;
 
 
 import javax.annotation.Resource;
 import javax.annotation.Resource;
 import java.io.Serializable;
 import java.io.Serializable;
+import java.math.BigDecimal;
 import java.time.LocalDate;
 import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.time.LocalDateTime;
 import java.util.ArrayList;
 import java.util.ArrayList;
@@ -558,6 +559,69 @@ public class ContractDistributorStandardServiceImpl extends
         OpLogUtil.setExtra(vo);
         OpLogUtil.setExtra(vo);
     }
     }
 
 
+    @Override
+    @OpLog(type = OtherOpLogType.class, name = "合同锁规，ID：{}", params = {"#vo.id"})
+    @Transactional(rollbackFor = Exception.class)
+    public void specificationLock(UpdateContractDistributorStandardVo vo) {
+        ContractDistributorStandard data = getBaseMapper().selectById(vo.getId());
+        if (ObjectUtil.isNull(data)) {
+            throw new DefaultClientException("合同不存在！");
+        }
+
+        List<UpdateContractDistributorLineVo> lineVoList = vo.getContractDistributorLineList();
+        if (CollectionUtils.isEmpty(lineVoList)) {
+            throw new DefaultClientException("合同行数据为空！");
+        }
+
+        Wrapper<ContractDistributorStandard> childWrapper = Wrappers.lambdaQuery(ContractDistributorStandard.class)
+                .eq(ContractDistributorStandard::getParentId, vo.getId());
+
+        List<ContractDistributorStandard> childContractList = getBaseMapper().selectList(childWrapper);
+        BigDecimal allChildTotalQuantity = vo.getTotalQuantity().add(CollectionUtils.isEmpty(childContractList) ? new BigDecimal(0) : childContractList.get(0).getTotalQuantity());
+        if (allChildTotalQuantity.compareTo(data.getTotalQuantity()) > 0) {
+            throw new DefaultClientException("锁规总数量大于合同总数量，锁规失败！");
+        }
+
+        // 判断锁规是否完成（锁规两次或锁规数量等于合同数量）
+        boolean checkPriceSpecLocked = CollectionUtils.isNotEmpty(childContractList) || (allChildTotalQuantity.compareTo(data.getTotalQuantity()) == 0);
+
+        data.setId(IdUtil.getUUID());
+        data.setTotalQuantity(vo.getTotalQuantity());
+        data.setTotalAmountCapital(vo.getTotalAmountCapital());
+        data.setTotalAmountExcludingTax(vo.getTotalAmountExcludingTax());
+        data.setTotalAmountIncludingTax(vo.getTotalAmountIncludingTax());
+        data.setStatus("STANDARD");
+        data.setParentId(vo.getId());
+        getBaseMapper().insert(data);
+
+        Wrapper<ContractDistributorLine> lineWrapper = Wrappers.lambdaQuery(ContractDistributorLine.class)
+                .eq(ContractDistributorLine::getContractId, vo.getId())
+                .orderByAsc(ContractDistributorLine::getShowOrder);
+        List<ContractDistributorLine> oldContractDistributorLineList = contractDistributorLineService.list(lineWrapper);
+        Map<String, ContractDistributorLine> contractDistributorLineMap = CollectionUtils.emptyIfNull(oldContractDistributorLineList)
+                .stream().collect(Collectors.toMap(ContractDistributorLine::getId, Function.identity()));
+
+        for (UpdateContractDistributorLineVo lineVo : lineVoList) {
+            ContractDistributorLine contractDistributorLine = contractDistributorLineMap.remove(lineVo.getId());
+            if (contractDistributorLine == null) {
+                throw new DefaultClientException("合同行数据不存在！");
+            }
+
+            contractDistributorLine.setId(IdUtil.getUUID());
+            contractDistributorLine.setContractId(data.getId());
+            contractDistributorLine.setQuantity(lineVo.getQuantity());
+            contractDistributorLine.setUnitPrice(lineVo.getUnitPrice());
+            contractDistributorLine.setAmountExcludingTax(lineVo.getAmountExcludingTax());
+            contractDistributorLine.setTotalAmount(lineVo.getTotalAmount());
+            contractDistributorLineService.getBaseMapper().insert(contractDistributorLine);
+        }
+
+        Wrapper<ContractDistributorStandard> changeLocked = Wrappers.lambdaUpdate(ContractDistributorStandard.class)
+                .set(ContractDistributorStandard::getPriceSpecLocked, checkPriceSpecLocked)
+                .eq(ContractDistributorStandard::getId, vo.getId());
+        getBaseMapper().update(changeLocked);
+    }
+
     @CacheEvict(value = ContractDistributorStandard.CACHE_NAME, key = "@cacheVariables.tenantId() + #key")
     @CacheEvict(value = ContractDistributorStandard.CACHE_NAME, key = "@cacheVariables.tenantId() + #key")
     @Override
     @Override
     public void cleanCacheByKey(Serializable key) {
     public void cleanCacheByKey(Serializable key) {

@@ -88,4 +88,11 @@ public interface ContractDistributorStandardService extends BaseMpService<Contra
      */
      */
     void stockContractPriceLock(UpdateContractDistributorStandardVo vo);
     void stockContractPriceLock(UpdateContractDistributorStandardVo vo);
 
 
+    /**
+     * 锁规
+     *
+     * @param vo
+     */
+    void specificationLock(UpdateContractDistributorStandardVo vo);
+
 }
 }