From 1fe626c90ab7b98b6b9a67d728de96772b5add51 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Sat, 05 Oct 2024 10:57:14 +0800
Subject: [PATCH] 1

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

diff --git a/src/main/java/com/nq/service/impl/UserServiceImpl.java b/src/main/java/com/nq/service/impl/UserServiceImpl.java
index e46c8b5..24daae5 100644
--- a/src/main/java/com/nq/service/impl/UserServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/UserServiceImpl.java
@@ -1,6 +1,7 @@
 package com.nq.service.impl;
 
 import cn.hutool.core.convert.Convert;
+import cn.hutool.core.date.DateUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.github.pagehelper.PageHelper;
@@ -105,6 +106,8 @@
     @Autowired
     StockPoll stockPoll;
 
+    @Autowired
+    StockSubscribeMapper stockSubscribeMapper;
 
     @Resource
     MoneyLogMapper mapper;
@@ -615,7 +618,21 @@
         if (userPositions.size() > 0) {
             for (UserPosition position : userPositions) {
                 UserPositionVO userPositionVO = UserPointUtil.assembleUserPositionVO(position, priceServices.getNowPrice(position.getStockCode()));
+
+                StockSubscribe stockSubscribe = stockSubscribeMapper.selectOne(new LambdaQueryWrapper<StockSubscribe>()
+                        .eq(StockSubscribe::getCode, userPositionVO.getStockCode()));
+                if(position.getSellOrderId() == null){
+                    if (null != stockSubscribe && DateUtil.date().before(stockSubscribe.getListDate())) {
+                        userPositionVO.setProfitAndLose(BigDecimal.ZERO);
+                    }else{
+                        userPositionVO.setProfitAndLose(userPositionVO.getProfitAndLose().multiply(new BigDecimal(userPositionVO.getOrderLever())));
+                    }
+                }else{
+                    userPositionVO.setProfitAndLose(userPositionVO.getProfitAndLose().multiply(new BigDecimal(userPositionVO.getOrderLever())));
+                }
+
                 userPositionVOS.add(userPositionVO);
+
             }
         }
         BigDecimal profitAndLose = BigDecimal.ZERO;
@@ -746,10 +763,24 @@
         List<User> users = this.userMapper.listByAgent(realName, phone, searchId, accountType);
 
         List<AgentUserListVO> agentUserListVOS = Lists.newArrayList();
+
         for (User user : users) {
+
+            ServerResponse money = iUserService.getMoney(user.getId());
+            List<RUserAssets> rUserAssetsList = (List<RUserAssets>) money.getData();
+            RUserAssets rUserAssets = rUserAssetsList.stream()
+                    .filter(stock -> "IN".equals(stock.getAccectType()))
+                    .findFirst()
+                    .orElse(null);
+
             AgentUserListVO agentUserListVO = assembleAgentUserListVO(user, siteSetting
                     .getForceStopPercent(), siteIndexSetting
                     .getForceSellPercent(), siteFuturesSetting.getForceSellPercent());
+            if (rUserAssets != null) {
+                agentUserListVO.setUserAmt(rUserAssets.getTotalMoney().equals("0E-8") ? new BigDecimal("0") : new BigDecimal(rUserAssets.getTotalMoney()));
+                agentUserListVO.setFreezeMoney(rUserAssets.getFreezeMoney().equals("0E-8") ? "0" : rUserAssets.getFreezeMoney());
+                agentUserListVO.setAvailableBalance(rUserAssets.getAvailableBalance().equals("0E-8") ? "0" : rUserAssets.getAvailableBalance());
+            }
             agentUserListVOS.add(agentUserListVO);
         }
 

--
Gitblit v1.9.3