From cd969de1f50903ca87deda67bff2d6fcc35b4107 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Sun, 07 Jun 2026 02:42:32 +0800
Subject: [PATCH] 1
---
src/main/java/com/nq/service/impl/UserServiceImpl.java | 93 ++++++++++++++++++++++++++++++++++++++++------
1 files changed, 80 insertions(+), 13 deletions(-)
diff --git a/src/main/java/com/nq/service/impl/UserServiceImpl.java b/src/main/java/com/nq/service/impl/UserServiceImpl.java
index 59ee1a7..cd96cf3 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);
@@ -1569,18 +1569,30 @@
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));
+ }
@@ -1631,7 +1643,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();
//
@@ -1839,6 +1861,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());
@@ -1852,6 +1879,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());
@@ -1890,6 +1932,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());
@@ -1902,6 +1949,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());
@@ -1914,6 +1962,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());
@@ -1936,7 +1993,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());
@@ -2113,7 +2169,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