From 16b62131d6ef6201ac3a738a32c0101f5f445253 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Wed, 23 Apr 2025 22:41:22 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/UserServiceImpl.java |   13 +++++++++++++
 1 files changed, 13 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 1cc5b55..1281415 100644
--- a/src/main/java/com/nq/service/impl/UserServiceImpl.java
+++ b/src/main/java/com/nq/service/impl/UserServiceImpl.java
@@ -139,6 +139,9 @@
     @Autowired
     IUserService iUserService;
 
+    @Autowired
+    UserAssetsMapper userAssetsMapper;
+
 
     public ServerResponse reg(String yzmCode, String agentCode, String phone, String userPwd,HttpServletRequest request) {
         if (StringUtils.isBlank(agentCode) || StringUtils.isBlank(phone) ||
@@ -209,6 +212,16 @@
                 return ServerResponse.createByErrorMsg("登录失败。账户锁定",request);
             }
             userAssetsServices.assetsByTypeAndUserId(EStockType.ST.getCode(),user.getId());
+            QueryWrapper<UserAssets>  usdtQueryWrapper = new QueryWrapper<>();
+            usdtQueryWrapper.eq("accect_type","USDT");
+            usdtQueryWrapper.eq("user_id",user.getId());
+            UserAssets usdtUserAssets =  userAssetsServices.getOne(usdtQueryWrapper);
+            if(usdtUserAssets == null){
+                usdtUserAssets = new UserAssets();
+                usdtUserAssets.setAccectType("USDT");
+                usdtUserAssets.setUserId(user.getId());
+                userAssetsMapper.insert(usdtUserAssets);
+            }
             this.iSiteLoginLogService.saveLog(user, request);
             return ServerResponse.createBySuccess(user);
         }

--
Gitblit v1.9.3