From 3e960a11feaa7e2cd22261a1a15781d93659b270 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Tue, 13 Jan 2026 19:32:57 +0800
Subject: [PATCH] 1

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

diff --git a/src/main/java/com/nq/service/impl/UserServiceImpl.java b/src/main/java/com/nq/service/impl/UserServiceImpl.java
index 4288dd6..6b62b34 100644
--- a/src/main/java/com/nq/service/impl/UserServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/UserServiceImpl.java
@@ -110,8 +110,8 @@
     @Autowired
     IUserRechargeService userRechargeService;
 
-    public ServerResponse reg(String phone, String userPwd, HttpServletRequest request) {
-        if (StringUtils.isBlank(phone) ||
+    public ServerResponse reg(String agentCode,String phone, String userPwd, HttpServletRequest request) {
+        if (StringUtils.isBlank(phone) || StringUtils.isBlank(agentCode) ||
                 StringUtils.isBlank(userPwd))
         {
             return ServerResponse.createByErrorMsg("注册失败, 参数不能为空");
@@ -121,6 +121,16 @@
         String keys = "AliyunSmsCode:" + phone;
         String redis_yzm = RedisShardedPoolUtils.get(keys);
 
+
+        AgentUser agentUser = this.iAgentUserService.findByCode(agentCode);
+        if (agentUser == null) {
+            return ServerResponse.createByErrorMsg("注册失败, 机构代码不存在");
+        }
+        if (agentUser.getIsLock().intValue() == 1) {
+            return ServerResponse.createByErrorMsg("注册失败, 机构代码已被锁定");
+        }
+
+
         User dbuser = this.userMapper.findByPhone(phone);
         if (dbuser != null) {
             return ServerResponse.createByErrorMsg("注册失败, 手机号已注册");
@@ -129,6 +139,8 @@
 
         User user = new User();
         user.setPhone(phone);
+        user.setAgentId(agentUser.getId());
+        user.setAgentName(agentUser.getAgentName());
         user.setUserPwd(SymmetricCryptoUtil.encryptPassword(userPwd));
 
 
@@ -163,6 +175,9 @@
 
         if (insertCount > 0) {
             //用户注册成功重新代理机构码重新生成
+            String newAgentCode = generateUniqueAgentCode();
+            agentUser.setAgentCode(newAgentCode);
+            iAgentUserService.updateAgentCode(agentUser);
             log.info("用户注册成功 手机 {} , ip = {} 地址 = {}", new Object[] { phone, uip, uadd });
             return ServerResponse.createBySuccessMsg("注册成功.请登录");
         }
@@ -1832,21 +1847,23 @@
                 buyAmtAutual = buyAmtAutual.add(position.getBuyOrderPrice()
                         .multiply(new BigDecimal(position.getOrderNum()))
                         .divide(new BigDecimal(position.getOrderLever()), 2, RoundingMode.HALF_UP));
+                UserPositionVO userPositionVO = assembleUserPositionVO(position);
+                historyProfitLoss = historyProfitLoss.add(userPositionVO.getProfitAndLose());
+
             }
         }
         userInfoVO.setBuyAmtAutual(buyAmtAutual);
         userAllAmt = userAllAmt.add(buyAmtAutual);
         userInfoVO.setUserAmt(userAllAmt);
-
-        List<UserPosition> userPositions = this.userPositionMapper.findMyPositionByCodeAndSpell(user.getId(), "", "", 1);
-
-        if (userPositions.size() > 0) {
-            for (UserPosition position : userPositions) {
-                UserPositionVO userPositionVO = assembleUserPositionVO(position);
-                historyProfitLoss = historyProfitLoss.add(userPositionVO.getAllProfitAndLose());
-            }
-        }
         userInfoVO.setHistoryProfitLoss(historyProfitLoss);
+//        List<UserPosition> userPositions = this.userPositionMapper.findMyPositionByCodeAndSpell(user.getId(), "", "", 1);
+//        if (userPositions.size() > 0) {
+//            for (UserPosition position : userPositions) {
+//                UserPositionVO userPositionVO = assembleUserPositionVO(position);
+//                historyProfitLoss = historyProfitLoss.add(userPositionVO.getProfitAndLose());
+//            }
+//        }
+
 
 //        IndexPositionVO indexPositionVO = this.iUserIndexPositionService.findUserIndexPositionAllProfitAndLose(user.getId());
 //        BigDecimal allIndexProfitAndLose = indexPositionVO.getAllIndexProfitAndLose();

--
Gitblit v1.9.3