From dece32aeb6b9053e345fda5991a1671463e35e4d Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期二, 31 八月 2021 00:33:32 +0800
Subject: [PATCH] 优化代码

---
 service-api/src/main/java/com/java110/api/bmo/allocationUserStorehouse/impl/AllocationUserStorehouseBMOImpl.java |  206 ++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 145 insertions(+), 61 deletions(-)

diff --git a/service-api/src/main/java/com/java110/api/bmo/allocationUserStorehouse/impl/AllocationUserStorehouseBMOImpl.java b/service-api/src/main/java/com/java110/api/bmo/allocationUserStorehouse/impl/AllocationUserStorehouseBMOImpl.java
index b8dcab2..75bfa5c 100644
--- a/service-api/src/main/java/com/java110/api/bmo/allocationUserStorehouse/impl/AllocationUserStorehouseBMOImpl.java
+++ b/service-api/src/main/java/com/java110/api/bmo/allocationUserStorehouse/impl/AllocationUserStorehouseBMOImpl.java
@@ -1,28 +1,27 @@
 package com.java110.api.bmo.allocationUserStorehouse.impl;
 
+import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.java110.api.bmo.ApiBaseBMO;
 import com.java110.api.bmo.allocationUserStorehouse.IAllocationUserStorehouseBMO;
 import com.java110.core.context.DataFlowContext;
 import com.java110.dto.userStorehouse.UserStorehouseDto;
-import com.java110.intf.store.IAllocationUserStorehouseInnerServiceSMO;
 import com.java110.intf.store.IUserStorehouseInnerServiceSMO;
 import com.java110.po.allocationUserStorehouse.AllocationUserStorehousePo;
 import com.java110.po.userStorehouse.UserStorehousePo;
 import com.java110.utils.constant.BusinessTypeConstant;
 import com.java110.utils.util.Assert;
 import com.java110.utils.util.BeanConvertUtil;
+import com.java110.utils.util.StringUtil;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import java.math.BigDecimal;
 import java.util.Date;
 import java.util.List;
 
 @Service("allocationUserStorehouseBMOImpl")
 public class AllocationUserStorehouseBMOImpl extends ApiBaseBMO implements IAllocationUserStorehouseBMO {
-
-    @Autowired
-    private IAllocationUserStorehouseInnerServiceSMO allocationUserStorehouseInnerServiceSMOImpl;
 
     @Autowired
     private IUserStorehouseInnerServiceSMO userStorehouseInnerServiceSMOImpl;
@@ -35,63 +34,148 @@
      * @return 璁㈠崟鏈嶅姟鑳藉鎺ュ彈鐨勬姤鏂�
      */
     public void addAllocationUserStorehouse(JSONObject paramInJson, DataFlowContext dataFlowContext) {
-        //鑾峰彇搴撳瓨鏁伴噺
-        String stock = paramInJson.getString("stock");
-        //鑾峰彇杞鏁伴噺
-        String giveQuantity = paramInJson.getString("giveQuantity");
-        //鑾峰彇褰撳墠鐢ㄦ埛id
-        String userId = paramInJson.getString("userId");
-        //鑾峰彇鎺ュ彈杞鐢ㄦ埛id
-        String acceptUserId = paramInJson.getString("acceptUserId");
-        //鑾峰彇鐗╁搧id
-        String resId = paramInJson.getString("resId");
-        paramInJson.put("ausId", "-1");
-        paramInJson.put("createTime", new Date());
-        paramInJson.put("startUserId", userId);
-        paramInJson.put("startUserName", paramInJson.getString("userName"));
-        paramInJson.put("remark", paramInJson.getString("description"));
-        AllocationUserStorehousePo allocationUserStorehousePo = BeanConvertUtil.covertBean(paramInJson, AllocationUserStorehousePo.class);
-        super.insert(dataFlowContext, allocationUserStorehousePo, BusinessTypeConstant.BUSINESS_TYPE_SAVE_ALLOCATION_USER_STOREHOUSE);
-        UserStorehouseDto userStorehouseDto = new UserStorehouseDto();
-        userStorehouseDto.setUserId(userId);
-        userStorehouseDto.setResId(resId);
-        List<UserStorehouseDto> userStorehouseDtos = userStorehouseInnerServiceSMOImpl.queryUserStorehouses(userStorehouseDto);
-        Assert.listOnlyOne(userStorehouseDtos, "鏌ヨ涓汉鐗╁搧淇℃伅閿欒锛�");
-        //鑾峰彇涓汉鐗╁搧淇℃伅id
-        String usId = userStorehouseDtos.get(0).getUsId();
-        UserStorehousePo userStorehousePo = new UserStorehousePo();
-        userStorehousePo.setUsId(usId);
-        //鑾峰彇杞鍚庣殑搴撳瓨鏁伴噺
-        int newStock = Integer.parseInt(stock) - Integer.parseInt(giveQuantity);
-        userStorehousePo.setStock(String.valueOf(newStock));
-        //鏇存柊褰撳墠鐢ㄦ埛搴撳瓨鏁�
-        super.update(dataFlowContext, userStorehousePo, BusinessTypeConstant.BUSINESS_TYPE_UPDATE_USER_STOREHOUSE);
-        UserStorehouseDto userStorehouse = new UserStorehouseDto();
-        userStorehouse.setUserId(acceptUserId);
-        userStorehouse.setResId(resId);
-        List<UserStorehouseDto> userStorehouses = userStorehouseInnerServiceSMOImpl.queryUserStorehouses(userStorehouse);
-        if (userStorehouses != null && userStorehouses.size() == 1) {
-            //鑾峰彇鎺ュ彈鐢ㄦ埛搴撳瓨鏁�
-            String userStock = userStorehouses.get(0).getStock();
-            //鑾峰彇杞鍚庣殑搴撳瓨鏁伴噺
-            int myStock = Integer.parseInt(userStock) + Integer.parseInt(giveQuantity);
-            UserStorehousePo userStorePo = new UserStorehousePo();
-            userStorePo.setUsId(userStorehouses.get(0).getUsId());
-            userStorePo.setStock(String.valueOf(myStock));
-            //鏇存柊褰撳墠鐢ㄦ埛鐨勫簱瀛樻暟閲�
-            super.update(dataFlowContext, userStorePo, BusinessTypeConstant.BUSINESS_TYPE_UPDATE_USER_STOREHOUSE);
-        } else if (userStorehouses != null && userStorehouses.size() > 1) {
-            throw new IllegalArgumentException("鏌ヨ涓汉鐗╁搧淇℃伅閿欒锛�");
-        } else {
-            UserStorehousePo userStorePo = new UserStorehousePo();
-            userStorePo.setUsId("-1");
-            userStorePo.setResId(resId);
-            userStorePo.setResName(paramInJson.getString("resName"));
-            userStorePo.setStoreId(paramInJson.getString("storeId"));
-            userStorePo.setStock(giveQuantity);
-            userStorePo.setUserId(acceptUserId);
-            //淇濆瓨鎺ュ彈杞鐢ㄦ埛涓汉鐗╁搧淇℃伅
-            super.insert(dataFlowContext, userStorePo, BusinessTypeConstant.BUSINESS_TYPE_SAVE_USER_STOREHOUSE);
+        String resourceStores = paramInJson.getString("resourceStores");
+        JSONArray json = JSONArray.parseArray(resourceStores);
+        if (json.size() > 0) {
+            Object[] objects = json.toArray();
+            for (int i = 0; i < objects.length; i++) {
+                Object object = objects[i];
+                JSONObject paramIn = JSONObject.parseObject(String.valueOf(object));
+                String stock = paramIn.getString("stock");
+                //鑾峰彇鏈�灏忚閲忔�绘暟
+                String miniStock = paramIn.getString("miniStock");
+                //鑾峰彇鏈�灏忚閲忓崟浣嶆暟閲�
+                String miniUnitStock = paramIn.getString("miniUnitStock");
+                //鑾峰彇杞禒鏁伴噺
+                String giveQuantity = paramIn.getString("giveQuantity");
+                //鑾峰彇鐗╁搧id
+                String resId = paramIn.getString("resId");
+                //鑾峰彇鐗╁搧鍚嶇О
+                String resName = paramIn.getString("resName");
+                //鑾峰彇褰撳墠鐢ㄦ埛id
+                String userId = paramInJson.getString("userId");
+                //鑾峰彇鎺ュ彈杞禒鐢ㄦ埛id
+                String acceptUserId = paramInJson.getString("acceptUserId");
+                //鑾峰彇鎺ュ彈杞禒鐢ㄦ埛鍚嶇О
+                String acceptUserName = paramInJson.getString("acceptUserName");
+                //鑾峰彇鍟嗘埛id
+                String storeId = paramInJson.getString("storeId");
+                JSONObject allocationUserStorehouseJson = new JSONObject();
+                allocationUserStorehouseJson.put("ausId", "-1");
+                allocationUserStorehouseJson.put("resId", resId);
+                allocationUserStorehouseJson.put("resName", resName);
+                allocationUserStorehouseJson.put("storeId", storeId);
+                allocationUserStorehouseJson.put("stock", stock);
+                allocationUserStorehouseJson.put("giveQuantity", giveQuantity);
+                allocationUserStorehouseJson.put("startUserId", userId);
+                allocationUserStorehouseJson.put("startUserName", paramInJson.getString("userName"));
+                allocationUserStorehouseJson.put("acceptUserId", acceptUserId);
+                allocationUserStorehouseJson.put("acceptUserName", acceptUserName);
+                allocationUserStorehouseJson.put("createTime", new Date());
+                allocationUserStorehouseJson.put("remark", paramInJson.getString("description"));
+                AllocationUserStorehousePo allocationUserStorehousePo = BeanConvertUtil.covertBean(allocationUserStorehouseJson, AllocationUserStorehousePo.class);
+                super.insert(dataFlowContext, allocationUserStorehousePo, BusinessTypeConstant.BUSINESS_TYPE_SAVE_ALLOCATION_USER_STOREHOUSE);
+                UserStorehouseDto userStorehouseDto = new UserStorehouseDto();
+                userStorehouseDto.setUserId(userId);
+                userStorehouseDto.setResId(resId);
+                List<UserStorehouseDto> userStorehouseDtos = userStorehouseInnerServiceSMOImpl.queryUserStorehouses(userStorehouseDto);
+                Assert.listOnlyOne(userStorehouseDtos, "鏌ヨ涓汉鐗╁搧淇℃伅閿欒锛�");
+                //鑾峰彇涓汉鐗╁搧淇℃伅id
+                String usId = userStorehouseDtos.get(0).getUsId();
+                //鑾峰彇鐗╁搧鍗曚綅
+                if (StringUtil.isEmpty(userStorehouseDtos.get(0).getUnitCode())) {
+                    throw new IllegalArgumentException("鐗╁搧鍗曚綅涓嶈兘涓虹┖");
+                }
+                String unitCode = userStorehouseDtos.get(0).getUnitCode();
+                //鑾峰彇鐗╁搧鏈�灏忚閲忓崟浣�
+                if (StringUtil.isEmpty(userStorehouseDtos.get(0).getMiniUnitCode())) {
+                    throw new IllegalArgumentException("鐗╁搧鏈�灏忚閲忓崟浣嶄笉鑳戒负绌�");
+                }
+                String miniUnitCode = userStorehouseDtos.get(0).getMiniUnitCode();
+                UserStorehousePo userStorehousePo = new UserStorehousePo();
+                userStorehousePo.setUsId(usId);
+                //杞禒鍚庝釜浜虹墿鍝佹渶灏忚閲忔�绘暟
+                BigDecimal num1 = new BigDecimal(miniStock);
+                BigDecimal num2 = new BigDecimal(giveQuantity);
+                BigDecimal quantity = num1.subtract(num2);
+                if (quantity.doubleValue() == 0.0) {
+                    userStorehousePo.setMiniStock("0");
+                    userStorehousePo.setStock("0");
+                } else {
+                    userStorehousePo.setMiniStock(String.valueOf(quantity));
+                    BigDecimal reduceNum = num1.subtract(num2);
+                    if (unitCode.equals(miniUnitCode)) { //濡傛灉鐗╁搧鍗曚綅涓庢渶灏忚閲忓崟浣嶇浉鍚岋紝灏变笉鍚戜笂鍙栨暣
+                        userStorehousePo.setStock(String.valueOf(reduceNum));
+                    } else { //濡傛灉鐗╁搧鏈�灏忚閲忓崟浣嶄笌鐗╁搧鍗曚綅涓嶅悓锛屽氨鍚戜笂鍙栨暣
+                        //鐢ㄨ浆璧犲悗鏈�灏忚閲忔�绘暟闄や互鏈�灏忚閲忓崟浣嶆暟閲忥紝骞跺悜涓婂彇鏁达紝鑾峰彇杞禒鍚庣殑搴撳瓨鏁�
+                        BigDecimal num3 = new BigDecimal(miniUnitStock);
+                        BigDecimal unitStock = reduceNum.divide(num3, 2, BigDecimal.ROUND_HALF_UP);
+                        Integer stockNumber = (int) Math.ceil(unitStock.doubleValue());
+                        userStorehousePo.setStock(String.valueOf(stockNumber));
+                    }
+                }
+                //鏇存柊褰撳墠鐢ㄦ埛搴撳瓨鏁�
+                super.update(dataFlowContext, userStorehousePo, BusinessTypeConstant.BUSINESS_TYPE_UPDATE_USER_STOREHOUSE);
+                UserStorehouseDto userStorehouse = new UserStorehouseDto();
+                userStorehouse.setUserId(acceptUserId);
+                userStorehouse.setResId(resId);
+                //鏌ヨ鎺ュ彈杞禒浜虹殑涓汉鐗╁搧淇℃伅
+                List<UserStorehouseDto> userStorehouses = userStorehouseInnerServiceSMOImpl.queryUserStorehouses(userStorehouse);
+                if (userStorehouses != null && userStorehouses.size() == 1) {
+                    UserStorehousePo userStorePo = new UserStorehousePo();
+                    //璁$畻鎺ュ彈鐢ㄦ埛鐨勬渶灏忚閲忔�绘暟
+                    BigDecimal num4 = new BigDecimal(userStorehouses.get(0).getMiniStock());
+                    BigDecimal num5 = new BigDecimal(giveQuantity);
+                    BigDecimal addNum = num4.add(num5);
+                    BigDecimal acceptMiniStock = num4.add(num5);
+                    userStorePo.setMiniStock(String.valueOf(acceptMiniStock));
+                    //鑾峰彇鐗╁搧鍗曚綅
+                    if (StringUtil.isEmpty(userStorehouses.get(0).getUnitCode())) {
+                        throw new IllegalArgumentException("鐗╁搧鍗曚綅涓嶈兘涓虹┖");
+                    }
+                    String unitCode1 = userStorehouses.get(0).getUnitCode();
+                    //鑾峰彇鐗╁搧鏈�灏忚閲忓崟浣�
+                    if (StringUtil.isEmpty(userStorehouses.get(0).getMiniUnitCode())) {
+                        throw new IllegalArgumentException("鐗╁搧鏈�灏忚閲忓崟浣嶄笉鑳戒负绌�");
+                    }
+                    String miniUnitCode1 = userStorehouses.get(0).getMiniUnitCode();
+                    //璁$畻鎺ュ彈鐢ㄦ埛鐨勫簱瀛樻暟閲�
+                    BigDecimal num6 = new BigDecimal(miniUnitStock);
+                    BigDecimal unitStock = addNum.divide(num6, 2, BigDecimal.ROUND_HALF_UP);
+                    if (unitCode1.equals(miniUnitCode1)) { //濡傛灉鐗╁搧鍗曚綅涓庣墿鍝佹渶灏忚閲忓崟浣嶇浉鍚岋紝灏变笉鍚戜笂鍙栨暣
+                        //濡傛灉鐗╁搧鍗曚綅涓庢渶灏忚閲忓崟浣嶇浉鍚岋紝鐗╁搧搴撳瓨灏辩瓑浜庢渶灏忚閲忔�绘暟
+                        userStorePo.setStock(String.valueOf(acceptMiniStock));
+                    } else { //濡傛灉鐗╁搧鍗曚綅涓庣墿鍝佹渶灏忚閲忓崟浣嶄笉鍚岋紝灏卞悜涓婂彇鏁�
+                        Integer stockNumber = (int) Math.ceil(unitStock.doubleValue());
+                        userStorePo.setStock(String.valueOf(stockNumber));
+                    }
+                    userStorePo.setUsId(userStorehouses.get(0).getUsId());
+                    //鏇存柊褰撳墠鐢ㄦ埛鐨勫簱瀛樻暟閲�
+                    super.update(dataFlowContext, userStorePo, BusinessTypeConstant.BUSINESS_TYPE_UPDATE_USER_STOREHOUSE);
+                } else if (userStorehouses != null && userStorehouses.size() > 1) {
+                    throw new IllegalArgumentException("鏌ヨ涓汉鐗╁搧淇℃伅閿欒锛�");
+                } else {
+                    //璁$畻杞禒鍚庡簱瀛樻暟閲�
+                    BigDecimal num7 = new BigDecimal(giveQuantity);
+                    BigDecimal num8 = new BigDecimal(miniUnitStock);
+                    BigDecimal unitStock = num7.divide(num8, 2, BigDecimal.ROUND_HALF_UP);
+                    UserStorehousePo userStorePo = new UserStorehousePo();
+                    userStorePo.setUsId("-1");
+                    userStorePo.setResId(resId);
+                    userStorePo.setResName(resName);
+                    userStorePo.setStoreId(storeId);
+                    if (unitCode.equals(miniUnitCode)) { //濡傛灉鐗╁搧鍗曚綅涓庣墿鍝佹渶灏忚閲忓崟浣嶇浉鍚岋紝灏变笉鍚戜笂鍙栨暣
+                        userStorePo.setStock(String.valueOf(num7));
+                    } else { //濡傛灉鐗╁搧鍗曚綅涓庣墿鍝佹渶灏忚閲忓崟浣嶄笉鍚岋紝灏卞悜涓婂彇鏁�
+                        Integer stockNumber = (int) Math.ceil(unitStock.doubleValue());
+                        userStorePo.setStock(String.valueOf(stockNumber));
+                    }
+                    userStorePo.setMiniStock(giveQuantity);
+                    userStorePo.setUserId(acceptUserId);
+                    //淇濆瓨鎺ュ彈杞禒鐢ㄦ埛涓汉鐗╁搧淇℃伅
+                    super.insert(dataFlowContext, userStorePo, BusinessTypeConstant.BUSINESS_TYPE_SAVE_USER_STOREHOUSE);
+                }
+            }
         }
     }
 

--
Gitblit v1.8.0