From 38e2dae2721ef4b2f458f88974fbdaef1afa6db6 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Wed, 15 May 2024 00:18:16 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/UserServiceImpl.java |   58 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 56 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/UserServiceImpl.java b/src/main/java/com/nq/service/impl/UserServiceImpl.java
index afdaf50..0e639bf 100644
--- a/src/main/java/com/nq/service/impl/UserServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/UserServiceImpl.java
@@ -13,6 +13,7 @@
 import com.nq.pojo.*;
 import com.nq.pojo.reponse.RUserAssets;
 import com.nq.service.*;
+import com.nq.utils.UserPointUtil;
 import com.nq.utils.timeutil.DateTimeUtil;
 import com.nq.utils.PropertiesUtil;
 import com.nq.utils.SymmetricCryptoUtil;
@@ -27,6 +28,7 @@
 import com.nq.vo.indexposition.IndexPositionVO;
 import com.nq.vo.position.PositionProfitVO;
 import com.nq.vo.position.PositionVO;
+import com.nq.vo.position.UserPositionVO;
 import com.nq.vo.stock.StockListVO;
 import com.nq.vo.user.UserInfoVO;
 
@@ -117,8 +119,14 @@
     @Autowired
     ISiteMessageService iSiteMessageService;
 
+    @Autowired
+    private ApplyLeverMapper applyLeverMapper;
 
+    @Autowired
+    private UserPositionMapper userPositionMapper;
 
+    @Autowired
+    IPriceServices priceServices;
 
 
     public ServerResponse reg(String yzmCode, String agentCode, String phone, String userPwd, HttpServletRequest request) {
@@ -284,9 +292,30 @@
         String userJson = RedisShardedPoolUtils.get(loginToken);
         User user = (User) JsonUtil.string2Obj(userJson, User.class);
         User dbuser = this.userMapper.selectById(user.getId());
+
         SiteSetting siteSetting = iSiteSettingService.getSiteSetting();
         UserInfoVO userInfoVO = assembleUserInfoVO(dbuser, siteSetting);
+        ApplyLever applyLever = applyLeverMapper.selectOne(new LambdaQueryWrapper<ApplyLever>()
+                .eq(ApplyLever::getUserId, user.getId())
+                .eq(ApplyLever::getState, 1)
+                .orderByDesc(ApplyLever::getCreateTime)
+                .last(" limit 1"));
+        if(null == applyLever || applyLever.getLever().equals("1")){
+            userInfoVO.setSiteLever("1");
+        }else{
+            userInfoVO.setSiteLever(leverSplit(applyLever.getLever()));
+        }
         return ServerResponse.createBySuccess(userInfoVO);
+    }
+
+    public String leverSplit(String lever){
+        String levers = "1/2/5/10";
+        String[] parts = levers.split("/");
+        int index = Arrays.asList(parts).indexOf(lever);
+        if (index != -1) {
+            return String.join("/", Arrays.copyOfRange(parts, 0, index + 1));
+        }
+        return null;
     }
 
 
@@ -482,7 +511,8 @@
             rUserAssets.setCumulativeProfitAndLoss(hProfitAndLose.toString());
             rUserAssets.setHandlingCharge(hMoney.toString());
             rUserAssets.setProfitAndLoss(profitAndLose.toString());
-
+            rUserAssets.setIsZf(userAssets.getIsZf());
+            rUserAssets.setAmountToBeCovered(userAssets.getAmountToBeCovered().toString());
             BigDecimal rate = rateServices.currencyRate(
                     EStockType.getEStockTypeByCode(userAssets.getAccectType()),EStockType.US);
 
@@ -525,7 +555,9 @@
             rUserAssets.setCumulativeProfitAndLossUSD(cumulativeProfitAndLossUSD.setScale(s,BigDecimal.ROUND_UP).toString());
             rUserAssets.setHandlingChargeUSD(handlingChargeUSD.setScale(s,BigDecimal.ROUND_UP).toString());
             rUserAssets.setProfitAndLossUSD(profitAndLossUSD.setScale(s,BigDecimal.ROUND_UP).toString());
-
+            rUserAssets.setProfitAndLoss(getProfitAndLose(userId).toString());
+            BigDecimal decimal = new BigDecimal(rUserAssets.getTotalMoney()).add(new BigDecimal(rUserAssets.getProfitAndLoss()));
+            rUserAssets.setTotalMoney(decimal.toString());
             AllProfitAndLose = AllProfitAndLose.add(profitAndLossUSD);
             allTotalAssets = allTotalAssets.add(totleMoneyUSD);
             allAmt = allAmt.add(availableBalanceUSD);
@@ -564,6 +596,28 @@
         return ServerResponse.createBySuccess(rUserAssetsList);
     }
 
+    public BigDecimal getProfitAndLose(Integer userId){
+        List<UserPosition> userPositions;
+        userPositions = userPositionMapper.
+                findMyPositionByCodeAndSpell(userId,
+                        "","",
+                        0, "IN");
+
+
+        List<UserPositionVO> userPositionVOS = Lists.newArrayList();
+        if (userPositions.size() > 0) {
+            for (UserPosition position : userPositions) {
+                UserPositionVO userPositionVO = UserPointUtil.assembleUserPositionVO(position, priceServices.getNowPrice(position.getStockCode()));
+                userPositionVOS.add(userPositionVO);
+            }
+        }
+        BigDecimal profitAndLose = BigDecimal.ZERO;
+        for (UserPositionVO f : userPositionVOS) {
+            profitAndLose = profitAndLose.add(f.getProfitAndLose());
+        }
+        return profitAndLose;
+    }
+
     @Override
     public ServerResponse transfer(String fromType, String toType, String amt,HttpServletRequest request) {
         User user =  userService.getCurrentUser(request);

--
Gitblit v1.9.3