From 03fe4dca7008ea95b60058740e880f31fe5be6dd Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Mon, 08 Jun 2026 10:43:24 +0800
Subject: [PATCH] 1

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

diff --git a/src/main/java/com/nq/service/impl/UserServiceImpl.java b/src/main/java/com/nq/service/impl/UserServiceImpl.java
index 59bb0d5..609ffcd 100644
--- a/src/main/java/com/nq/service/impl/UserServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/UserServiceImpl.java
@@ -14,6 +14,7 @@
 import com.nq.utils.KeyUtils;
 import com.nq.utils.PropertiesUtil;
 import com.nq.utils.SymmetricCryptoUtil;
+import com.nq.utils.UserFundUtil;
 import com.nq.utils.ip.IpUtils;
 import com.nq.utils.ip.JuheIpApi;
 import com.nq.utils.redis.CookieUtils;
@@ -106,6 +107,8 @@
     IUserService iUserService;
     @Autowired
     UserPositionMapper userPositionMapper;
+    @Autowired
+    UserWithdrawMapper userWithdrawMapper;
 
     @Autowired
     IUserRechargeService userRechargeService;
@@ -388,7 +391,7 @@
         User user = (User)JsonUtil.string2Obj(userJson, User.class);
         User dbuser = this.userMapper.selectByPrimaryKey(user.getId());
         RedisShardedPoolUtils.setEx(loginToken, JsonUtil.obj2String(dbuser), 9999);
-        UserInfoVO userInfoVO = assembleUserInfoVO(dbuser);
+        UserInfoVO userInfoVO = assembleUserInfoVO(dbuser, true);
 
         return ServerResponse.createBySuccess(userInfoVO);
     }
@@ -674,7 +677,7 @@
             BigDecimal enableIndexAmt = user.getEnableIndexAmt().add(new BigDecimal(amt.intValue()));
 
             user.setUserAmt(userAmt);
-            user.setEnableAmt(enableAmt);
+            UserFundUtil.debitEnableAndWithdraw(user, new BigDecimal(amt.intValue()));
             user.setUserIndexAmt(userIndexAmt);
             user.setEnableIndexAmt(enableIndexAmt);
             int updateCount = this.userMapper.updateByPrimaryKeySelective(user);
@@ -693,12 +696,11 @@
             }
 
             BigDecimal userAmt = user.getUserAmt().add(new BigDecimal(amt.intValue()));
-            BigDecimal enableAmt = user.getEnableAmt().add(new BigDecimal(amt.intValue()));
             BigDecimal userIndexAmt = user.getUserIndexAmt().subtract(new BigDecimal(amt.intValue()));
             BigDecimal enableIndexAmt = user.getEnableIndexAmt().subtract(new BigDecimal(amt.intValue()));
 
             user.setUserAmt(userAmt);
-            user.setEnableAmt(enableAmt);
+            UserFundUtil.creditEnableAndWithdraw(user, new BigDecimal(amt.intValue()));
             user.setUserIndexAmt(userIndexAmt);
             user.setEnableIndexAmt(enableIndexAmt);
             int updateCount = this.userMapper.updateByPrimaryKeySelective(user);
@@ -717,12 +719,11 @@
             }
 
             BigDecimal userAmt = user.getUserAmt().subtract(new BigDecimal(amt.intValue()));
-            BigDecimal enableAmt = user.getEnableAmt().subtract(new BigDecimal(amt.intValue()));
             BigDecimal userFutAmt = user.getUserFutAmt().add(new BigDecimal(amt.intValue()));
             BigDecimal enableFutAmt = user.getEnableFutAmt().add(new BigDecimal(amt.intValue()));
 
             user.setUserAmt(userAmt);
-            user.setEnableAmt(enableAmt);
+            UserFundUtil.debitEnableAndWithdraw(user, new BigDecimal(amt.intValue()));
             user.setUserFutAmt(userFutAmt);
             user.setEnableFutAmt(enableFutAmt);
             int updateCount = this.userMapper.updateByPrimaryKeySelective(user);
@@ -741,12 +742,11 @@
             }
 
             BigDecimal userAmt = user.getUserAmt().add(new BigDecimal(amt.intValue()));
-            BigDecimal enableAmt = user.getEnableAmt().add(new BigDecimal(amt.intValue()));
             BigDecimal userFutAmt = user.getUserFutAmt().subtract(new BigDecimal(amt.intValue()));
             BigDecimal enableFutAmt = user.getEnableFutAmt().subtract(new BigDecimal(amt.intValue()));
 
             user.setUserAmt(userAmt);
-            user.setEnableAmt(enableAmt);
+            UserFundUtil.creditEnableAndWithdraw(user, new BigDecimal(amt.intValue()));
             user.setUserFutAmt(userFutAmt);
             user.setEnableFutAmt(enableFutAmt);
 
@@ -978,49 +978,82 @@
             BigDecimal enable_user_amt = user.getEnableAmt();
             BigDecimal all_freez_amt = new BigDecimal("0");
             for (UserPosition position : userPositions) {
-//                PositionProfitVO positionProfitVO = iUserPositionService.getPositionProfitVO(position);
-//                if (positionProfitVO == null) {
-//                    continue;
-//                }
-
-                StockListVO stockListVO = new StockListVO();
-                stockListVO=SinaStockApi.assembleLideStockListVO(LiDeDataUtils.getStock(position.getStockCode()));
-                if(org.springframework.util.ObjectUtils.isEmpty(stockListVO)){
-                    stockListVO = SinaStockApi.assembleStockListVO(SinaStockApi.getSinaStock(position.getStockGid()));
-                }
-                if (stockListVO == null) {
+                if (position.getStatus() != null && position.getStatus().intValue() == 0) {
                     continue;
                 }
 
-                if(position.getProfitTargetPrice()!=null && position.getProfitTargetPrice().compareTo(new BigDecimal(stockListVO.getNowPrice())) <= 0 ||position.getStopTargetPrice() != null && position.getStopTargetPrice().compareTo(new BigDecimal(stockListVO.getNowPrice())) >= 0) {
+                StockListVO stockListVO = SinaStockApi.assembleLideStockListVO(LiDeDataUtils.getStock(position.getStockCode()));
+                if(org.springframework.util.ObjectUtils.isEmpty(stockListVO)){
+                    stockListVO = SinaStockApi.assembleStockListVO(SinaStockApi.getSinaStock(position.getStockGid()));
+                }
+                if (stockListVO == null || stockListVO.getNowPrice() == null) {
+                    continue;
+                }
+                BigDecimal nowPrice;
+                try {
+                    nowPrice = new BigDecimal(stockListVO.getNowPrice());
+                } catch (Exception e) {
+                    continue;
+                }
 
-                        try {
-                            this.iUserPositionService.sell(position.getPositionSn(), 0);
-                            SiteTaskLog siteTaskLog = new SiteTaskLog();
-                            siteTaskLog.setTaskType("单股止盈止损强平任务-股票持仓");
-                            String accountType = (user.getAccountType().intValue() == 0) ? "正式用户" : "模拟用户";
-                            String taskcnt = accountType + "-" + user.getRealName() + "被强平[达到目标盈亏] 用户id = " + user.getId() + ", 可用资金 = " + enable_user_amt + "冻结保证金 = " + all_freez_amt + ", 强平比例 = " + force_stop_percent + ",现价"+stockListVO.getNowPrice()+ ", 目标止盈价格:" + position.getProfitTargetPrice()+ ", 目标止损价格:" + position.getStopTargetPrice();
-                            siteTaskLog.setTaskCnt(taskcnt);
-                            String tasktarget = "此次强平订单号为:" + position.getPositionSn();
-                            siteTaskLog.setTaskTarget(tasktarget);
-                            siteTaskLog.setAddTime(new Date());
-                            siteTaskLog.setIsSuccess(Integer.valueOf(0));
-                            siteTaskLog.setErrorMsg("");
-                            int insertTaskCount = this.siteTaskLogMapper.insert(siteTaskLog);
-                            if (insertTaskCount > 0) {
-                                log.info("[盈利达到目标盈利]保存强制平仓task任务成功");
-                            } else {
-                                log.info("[盈利达到目标盈利]保存强制平仓task任务失败");
-                            }
-                        } catch (Exception e) {
-                            log.error("[盈利达到目标盈利]强制平仓失败...");
-                        }
+                if (!shouldForceSellByProfitStop(position, nowPrice)) {
+                    continue;
+                }
 
+                try {
+                    ServerResponse sellResult = this.iUserPositionService.sell(position.getPositionSn(), 0);
+                    if (sellResult != null && !sellResult.isSuccess()) {
+                        log.warn("[止盈止损强平] positionSn={} 失败: {}", position.getPositionSn(), sellResult.getMsg());
+                        continue;
+                    }
+                    SiteTaskLog siteTaskLog = new SiteTaskLog();
+                    siteTaskLog.setTaskType("单股止盈止损强平任务-股票持仓");
+                    String accountType = (user.getAccountType().intValue() == 0) ? "正式用户" : "模拟用户";
+                    String taskcnt = accountType + "-" + user.getRealName() + "被强平[达到目标盈亏] 用户id = " + user.getId() + ", 可用资金 = " + enable_user_amt + "冻结保证金 = " + all_freez_amt + ", 强平比例 = " + force_stop_percent + ",现价"+stockListVO.getNowPrice()+ ", 目标止盈价格:" + position.getProfitTargetPrice()+ ", 目标止损价格:" + position.getStopTargetPrice();
+                    siteTaskLog.setTaskCnt(taskcnt);
+                    String tasktarget = "此次强平订单号为:" + position.getPositionSn();
+                    siteTaskLog.setTaskTarget(tasktarget);
+                    siteTaskLog.setAddTime(new Date());
+                    siteTaskLog.setIsSuccess(Integer.valueOf(0));
+                    siteTaskLog.setErrorMsg("");
+                    int insertTaskCount = this.siteTaskLogMapper.insert(siteTaskLog);
+                    if (insertTaskCount > 0) {
+                        log.info("[盈利达到目标盈利]保存强制平仓task任务成功");
+                    } else {
+                        log.info("[盈利达到目标盈利]保存强制平仓task任务失败");
+                    }
+                } catch (Exception e) {
+                    log.error("[盈利达到目标盈利]强制平仓失败 positionSn={}", position.getPositionSn(), e);
                 }
 
             }
             log.info("=========止盈止损定时任务============");
         }
+    }
+
+    /** 买涨:现价>=止盈 或 现价<=止损;买跌:现价<=止盈 或 现价>=止损 */
+    private boolean shouldForceSellByProfitStop(UserPosition position, BigDecimal nowPrice) {
+        if (position == null || nowPrice == null || nowPrice.compareTo(BigDecimal.ZERO) <= 0) {
+            return false;
+        }
+        BigDecimal profitTarget = position.getProfitTargetPrice();
+        BigDecimal stopTarget = position.getStopTargetPrice();
+        boolean hasProfit = profitTarget != null && profitTarget.compareTo(BigDecimal.ZERO) > 0;
+        boolean hasStop = stopTarget != null && stopTarget.compareTo(BigDecimal.ZERO) > 0;
+        if (!hasProfit && !hasStop) {
+            return false;
+        }
+        boolean isLong = !"买跌".equals(position.getOrderDirection());
+        if (isLong) {
+            if (hasProfit && nowPrice.compareTo(profitTarget) >= 0) {
+                return true;
+            }
+            return hasStop && nowPrice.compareTo(stopTarget) <= 0;
+        }
+        if (hasProfit && nowPrice.compareTo(profitTarget) <= 0) {
+            return true;
+        }
+        return hasStop && nowPrice.compareTo(stopTarget) >= 0;
     }
     /*用户股票持仓单-强平提醒推送消息定时*/
     public void ForceSellMessageTask() {
@@ -1536,18 +1569,38 @@
 
         List<User> users = this.userMapper.listByAdmin(realName, phone, agentId, accountType);
 
-        for(User user:users){
+        List<UserInfoVO> voList = Lists.newArrayListWithCapacity(users.size());
+        for (User user : users) {
             user.setUserPwd("");
+            voList.add(assembleUserInfoVO(user, false));
         }
 
         PageInfo pageInfo = new PageInfo(users);
+        pageInfo.setList(voList);
 
         return ServerResponse.createBySuccess(pageInfo);
     }
 
 
 
-    public ServerResponse findByUserId(Integer userId) { return ServerResponse.createBySuccess(this.userMapper.selectByPrimaryKey(userId)); }
+    public ServerResponse findByUserId(Integer userId) {
+        if (userId == null) {
+            return ServerResponse.createByErrorMsg("用户ID不能为空");
+        }
+        User user = this.userMapper.selectByPrimaryKey(userId);
+        if (user == null) {
+            return ServerResponse.createByErrorMsg("用户不存在");
+        }
+        return ServerResponse.createBySuccess(assembleUserInfoVO(user, false));
+    }
+
+    @Override
+    public User getUserEntityById(Integer userId) {
+        if (userId == null) {
+            return null;
+        }
+        return this.userMapper.selectByPrimaryKey(userId);
+    }
 
 
 
@@ -1598,7 +1651,17 @@
         } else {
             return ServerResponse.createByErrorMsg("不存在此操作");
         }
-        user.setEnableAmt(sh_amt_back);
+        BigDecimal amtDecimal = new BigDecimal(amt);
+        if (direction.intValue() == 0) {
+            UserFundUtil.creditUserBalance(user, amtDecimal);
+        } else if (direction.intValue() == 1) {
+            user.setEnableAmt(sh_amt_back);
+            BigDecimal withdrawAmt = user.getEnaleWithdrawAmt() == null ? BigDecimal.ZERO : user.getEnaleWithdrawAmt();
+            BigDecimal newWithdraw = withdrawAmt.subtract(amtDecimal);
+            user.setEnaleWithdrawAmt(newWithdraw.compareTo(BigDecimal.ZERO) < 0 ? BigDecimal.ZERO : newWithdraw);
+        } else {
+            user.setEnableAmt(sh_amt_back);
+        }
 
 //        BigDecimal hk_amt = user.getHkAmt();
 //
@@ -1806,6 +1869,11 @@
         agentUserListVO.setUserAmt(user.getUserAmt());
         agentUserListVO.setEnableAmt(user.getEnableAmt());
 
+        BigDecimal pendingWithdrawAmt = userWithdrawMapper.sumPendingAmtByUserId(user.getId());
+        BigDecimal resolvedWithdrawAmt = UserFundUtil.resolveWithdrawAmt(
+                user.getEnableAmt(), user.getEnaleWithdrawAmt(), pendingWithdrawAmt);
+        agentUserListVO.setEnaleWithdrawAmt(resolvedWithdrawAmt);
+
         agentUserListVO.setUserIndexAmt(user.getUserIndexAmt());
         agentUserListVO.setEnableIndexAmt(user.getEnableIndexAmt());
 
@@ -1819,6 +1887,21 @@
         BigDecimal allFreezAmt = positionVO.getAllFreezAmt();
         agentUserListVO.setAllProfitAndLose(allProfitAndLose);
         agentUserListVO.setAllFreezAmt(allFreezAmt);
+
+        BigDecimal openBuyFees = BigDecimal.ZERO;
+        List<UserPosition> openPositions = this.iUserPositionService.findPositionByUserIdAndSellIdIsNull(user.getId());
+        if (openPositions != null) {
+            for (UserPosition position : openPositions) {
+                if (position.getOrderFee() != null) {
+                    openBuyFees = openBuyFees.add(position.getOrderFee());
+                }
+            }
+        }
+        BigDecimal enableAmt = user.getEnableAmt() == null ? BigDecimal.ZERO : user.getEnableAmt();
+        BigDecimal freezAmt = allFreezAmt == null ? BigDecimal.ZERO : allFreezAmt;
+        BigDecimal profitAndLose = allProfitAndLose == null ? BigDecimal.ZERO : allProfitAndLose;
+        BigDecimal userAllAmt = enableAmt.add(freezAmt).add(profitAndLose).add(openBuyFees);
+        agentUserListVO.setUserAmt(userAllAmt.setScale(2, RoundingMode.HALF_UP));
 
         BigDecimal forceLine = forcePercent.multiply(allFreezAmt);
         forceLine = forceLine.add(user.getEnableAmt());
@@ -1857,6 +1940,11 @@
     }
 
     private UserInfoVO assembleUserInfoVO(User user) {
+        return assembleUserInfoVO(user, true);
+    }
+
+    /** 与用户端 getUserInfo 一致的账户资产展示 */
+    private UserInfoVO assembleUserInfoVO(User user, boolean persistWithdrawHeal) {
         UserInfoVO userInfoVO = new UserInfoVO();
 
         userInfoVO.setId(user.getId());
@@ -1869,6 +1957,7 @@
         userInfoVO.setAccountType(user.getAccountType());
         userInfoVO.setRecomPhone(user.getRecomPhone());
         userInfoVO.setIsLock(user.getIsLock());
+        userInfoVO.setIsLogin(user.getIsLogin());
         userInfoVO.setRegTime(user.getRegTime());
         userInfoVO.setRegIp(user.getRegIp());
         userInfoVO.setRegAddress(user.getRegAddress());
@@ -1881,6 +1970,15 @@
         userInfoVO.setEnableAmt(user.getEnableAmt());
         userInfoVO.setTradingAmount(user.getTradingAmount());
 
+        BigDecimal pendingWithdrawAmt = userWithdrawMapper.sumPendingAmtByUserId(user.getId());
+        BigDecimal resolvedWithdrawAmt = UserFundUtil.resolveWithdrawAmt(
+                user.getEnableAmt(), user.getEnaleWithdrawAmt(), pendingWithdrawAmt);
+        if (persistWithdrawHeal
+                && (user.getEnaleWithdrawAmt() == null || resolvedWithdrawAmt.compareTo(user.getEnaleWithdrawAmt()) != 0)) {
+            user.setEnaleWithdrawAmt(resolvedWithdrawAmt);
+            this.userMapper.updateByPrimaryKeySelective(user);
+        }
+        userInfoVO.setEnaleWithdrawAmt(resolvedWithdrawAmt);
 
         PositionVO positionVO = this.iUserPositionService.findUserPositionAllProfitAndLose(user.getId());
         userInfoVO.setAllFreezAmt(positionVO.getAllFreezAmt());
@@ -1903,7 +2001,6 @@
         BigDecimal userAllAmt = enableAmt.add(allFreezAmt).add(allProfitAndLose).add(openBuyFees);
 
         userInfoVO.setEnableIndexAmt(user.getEnableIndexAmt());
-        userInfoVO.setEnaleWithdrawAmt(user.getEnaleWithdrawAmt());
         userInfoVO.setIsSignedContract(user.getIsSignedContract());
         userInfoVO.setIsSignedAgreement(user.getIsSignedAgreement());
         userInfoVO.setHkAmt(user.getHkAmt());
@@ -2080,7 +2177,18 @@
 
     @Override
     public void updateUserAmt(Double amt, Integer user_id) {
-        userMapper.updateUserAmt(amt, user_id);
+        if (amt == null || user_id == null || amt <= 0) {
+            return;
+        }
+        User user = this.userMapper.selectByPrimaryKey(user_id);
+        if (user == null) {
+            log.error("updateUserAmt 用户不存在 userId={}", user_id);
+            return;
+        }
+        UserFundUtil.creditUserBalance(user, BigDecimal.valueOf(amt));
+        this.userMapper.updateByPrimaryKeySelective(user);
+        log.info("updateUserAmt 充值入账 userId={} amt={} enable={} withdraw={}",
+                user_id, amt, user.getEnableAmt(), user.getEnaleWithdrawAmt());
     }
 
 

--
Gitblit v1.9.3