From 0ebdfd97b5400c864584147bd72a0de768172d6e Mon Sep 17 00:00:00 2001
From: dd <gitluke@outlook.com>
Date: Tue, 27 Jan 2026 10:19:48 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/UserAssetsServices.java |  235 +++++++++++++++++++++++++++++++++++++++++++++-------------
 1 files changed, 180 insertions(+), 55 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/UserAssetsServices.java b/src/main/java/com/nq/service/impl/UserAssetsServices.java
index 1584494..6f22fd7 100644
--- a/src/main/java/com/nq/service/impl/UserAssetsServices.java
+++ b/src/main/java/com/nq/service/impl/UserAssetsServices.java
@@ -2,7 +2,7 @@
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
-import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
+import com.nq.Repository.ExchangeRateRepository;
 import com.nq.common.ServerResponse;
 import com.nq.dao.*;
 import com.nq.enums.EStockType;
@@ -15,8 +15,9 @@
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
-import javax.validation.constraints.Email;
 import java.math.BigDecimal;
+import java.math.RoundingMode;
+import java.util.Arrays;
 import java.util.Date;
 import java.util.List;
 import java.util.Objects;
@@ -37,9 +38,6 @@
     @Autowired
     UserRechargeServiceImpl userRechargeService;
 
-    @Autowired
-    UserWithdrawServiceImpl withdrawService;
-
     @Resource
     UserRechargeMapper userRechargeMapper;
 
@@ -49,14 +47,15 @@
     @Autowired
     UserPositionMapper userPositionMapper;
 
+    @Autowired
+    ExchangeRateRepository exchangeRateRepository;
+
+
     @Override
     public UserAssets assetsByTypeAndUserId(String accetType, Integer userId) {
-        /*if(accetType.equals("SZHB")){
-            accetType = "US";
+        /*if(!accetType.equals(EStockType.US.getCode())){
+            accetType = EStockType.US.getCode();
         }*/
-        if(accetType.equals(EStockType.US.getCode()) || accetType.equals("IN")){
-            accetType = EStockType.MX.getCode();
-        }
         QueryWrapper<UserAssets>  queryWrapper = new QueryWrapper<>();
         queryWrapper.eq("accect_type",accetType);
         queryWrapper.eq("user_id",userId);
@@ -72,9 +71,37 @@
     }
 
     @Override
-    public ServerResponse updateUserAssets(Integer id, String amt,String type) {
+    public void addUserAssetsListByUserId(Integer userId) {
+        QueryWrapper<UserAssets>  queryWrapper = new QueryWrapper<>();
+        queryWrapper.eq("user_id",userId);
+        List<UserAssets> userAssetsList =  userAssetsMapper.selectList(queryWrapper);
+        List<EStockType> stockTypes = Arrays.asList(
+                EStockType.US,
+                EStockType.JP
+        );
+        for (EStockType stockType : stockTypes) {
+            //是否存在记录
+            boolean isExist = false;
+            if (!userAssetsList.isEmpty()) {
+                boolean hasType = userAssetsList.stream()
+                        .anyMatch(assets -> stockType.getCode().equals(assets.getAccectType()));
+                if (hasType) {
+                    isExist = true;
+                }
+            }
+            if (!isExist) {     //不存在账户新增
+                UserAssets userAssets = new UserAssets();
+                userAssets.setUserId(userId);
+                userAssets.setAccectType(stockType.getCode());
+                userAssetsMapper.insert(userAssets);
+            }
+        }
+    }
+
+    @Override
+    public ServerResponse updateUserAssets(Integer id, String amt,String type, String accectType) {
         UserAssets userAssets = userAssetsMapper.selectById(id);
-        UserPosition userPosition = userPositionMapper.selectOne(new LambdaQueryWrapper<UserPosition>().gt(UserPosition::getAmountToBeCovered, BigDecimal.ZERO).eq(UserPosition::getUserId,userAssets.getUserId()));
+        //UserPosition userPosition = userPositionMapper.selectOne(new LambdaQueryWrapper<UserPosition>().gt(UserPosition::getAmountToBeCovered, BigDecimal.ZERO).eq(UserPosition::getUserId,userAssets.getUserId()));
         //  0 入款  1是扣钱  2 是充值  3 是提币
         BigDecimal bigAmt = new BigDecimal(amt);
         if(type.equals("0") || type.equals("1")){
@@ -90,20 +117,25 @@
                         userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(userAssets.getAmountToBeCovered()));
                         userAssets.setAmountToBeCovered(BigDecimal.ZERO);
                         userAssets.setHandlingChargeWritten(BigDecimal.ZERO);
-                        if(null != userPosition){
-                            userPosition.setAmountToBeCovered(userAssets.getAmountToBeCovered());
-                        }
                     }else{
-                        BigDecimal amountToBeCovered = userAssets.getAmountToBeCovered().subtract(bigAmt);
-                        userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(bigAmt));
-                        userAssets.setAmountToBeCovered(amountToBeCovered);
-                        if(null != userPosition){
-                            userPosition.setAmountToBeCovered(userAssets.getAmountToBeCovered());
+                        BigDecimal subtract = bigAmt.subtract(userAssets.getHandlingChargeWritten());
+                        if(subtract.compareTo(BigDecimal.ZERO) >= 0) {
+                            BigDecimal decimal = bigAmt.subtract(userAssets.getAmountToBeCovered());
+                            decimal = decimal.subtract(userAssets.getHandlingChargeWritten());
+                            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(bigAmt.subtract(userAssets.getHandlingChargeWritten())));
+                            userAssets.setHandlingChargeWritten(BigDecimal.ZERO);
+                            userAssets.setAmountToBeCovered(decimal.negate());
+                        }else{
+                            userAssets.setHandlingChargeWritten(userAssets.getHandlingChargeWritten().subtract(bigAmt));
                         }
                     }
                 }else{
                     userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(bigAmt));
                 }
+                /*if(null != userPosition){
+                    userPosition.setAmountToBeCovered(userAssets.getAmountToBeCovered().add(userAssets.getHandlingChargeWritten()));
+                    userPositionMapper.updateById(userPosition);
+                }*/
             }else{
                 if(userAssets.getAvailableBalance().compareTo(BigDecimal.ZERO) <= 0){
                     return ServerResponse.createByErrorMsg("客户账户余额为0");
@@ -112,9 +144,9 @@
                 }
                 userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(bigAmt));
             }
-            if(null != userPosition){
+            /*if(null != userPosition){
                 userPositionMapper.updateById(userPosition);
-            }
+            }*/
             if( userAssetsMapper.updateById(userAssets)>0){
 //                saveUserRecharge(userAssets.getUserId(),bigAmt);
                 return ServerResponse.createBySuccess();
@@ -123,7 +155,7 @@
             }
         }else{
             if(type.equals("2")){
-                return  userRechargeService.createOrder(userAssets.getUserId(),1,bigAmt.intValue(),"1");
+                return  userRechargeService.createOrder(userAssets.getUserId(),1,bigAmt.intValue(),"1", accectType);
             }else{
                 return ServerResponse.createBySuccess();
             }
@@ -170,20 +202,26 @@
 
 
     @Override
-    public Boolean availablebalanceChange(String accetType, Integer userId, EUserAssets eUserAssets, BigDecimal amount, String desc, String descType) {
-
-        UserAssets userAssets = assetsByTypeAndUserId(accetType,userId);
-        UserPosition userPosition = userPositionMapper.selectOne(new LambdaQueryWrapper<UserPosition>().gt(UserPosition::getAmountToBeCovered, BigDecimal.ZERO).eq(UserPosition::getUserId,userAssets.getUserId()));
+    public Boolean availablebalanceChange(String accetType, Integer userId, EUserAssets eUserAssets, BigDecimal amount, String desc, String descType) throws Exception {
+        //查询账户
+        /*UserAssets userAssets = assetsByTypeAndUserId(EStockType.getDefault().getCode(), userId);
+        //是否需要转换金额
+        if (!accetType.equals(EStockType.getDefault().getCode())) {
+            amount = exchangeAmountByRate(accetType, amount);
+        }*/
+        UserAssets userAssets = assetsByTypeAndUserId(accetType, userId);
+        //UserPosition userPosition = userPositionMapper.selectOne(new LambdaQueryWrapper<UserPosition>().gt(UserPosition::getAmountToBeCovered, BigDecimal.ZERO).eq(UserPosition::getUserId,userAssets.getUserId()));
         String type = eUserAssets.getDesc();
         String before = userAssets.getAvailableBalance().toString();
-        String accectType = userAssets.getAccectType();
-       if(Objects.equals(eUserAssets.getCode(), EUserAssets.BUY.getCode())){
+        accetType = userAssets.getAccectType();
+        BigDecimal amountLog = amount;
+        if(Objects.equals(eUserAssets.getCode(), EUserAssets.BUY.getCode())){
             userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
             userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(amount.negate()));
         }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.CLOSE_POSITION.getCode())){
             userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
            extracted(amount, userAssets);
-       }else  if(Objects.equals(eUserAssets.getCode(), EUserAssets.CLOSE_POSITION_RETURN_SECURITY_DEPOSIT.getCode())){
+        }else  if(Objects.equals(eUserAssets.getCode(), EUserAssets.CLOSE_POSITION_RETURN_SECURITY_DEPOSIT.getCode())){
            if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
                BigDecimal availableBalance = amount.subtract(userAssets.getAmountToBeCovered());
                availableBalance = availableBalance.subtract(userAssets.getHandlingChargeWritten());
@@ -192,18 +230,51 @@
                    userAssets.setAmountToBeCovered(BigDecimal.ZERO);
                    userAssets.setHandlingChargeWritten(BigDecimal.ZERO);
                }else{
-                   BigDecimal amountToBeCovered = userAssets.getAmountToBeCovered().subtract(amount);
-                   userAssets.setAmountToBeCovered(amountToBeCovered);
+                   BigDecimal subtract = amount.subtract(userAssets.getHandlingChargeWritten());
+                   if(subtract.compareTo(BigDecimal.ZERO) >= 0) {
+                       BigDecimal decimal = amount.subtract(userAssets.getAmountToBeCovered());
+                       decimal = decimal.subtract(userAssets.getHandlingChargeWritten());
+                       userAssets.setHandlingChargeWritten(BigDecimal.ZERO);
+                       userAssets.setAmountToBeCovered(decimal.negate());
+                   }else{
+                       userAssets.setHandlingChargeWritten(userAssets.getHandlingChargeWritten().subtract(amount));
+                   }
                }
            }else{
                userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
            }
-            userAssets.setFreezeMoney(userAssets.getFreezeMoney().subtract(amount));
+           userAssets.setFreezeMoney(userAssets.getFreezeMoney().subtract(amount));
         }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.CALCULATE_PROFIT_AND_LOSS.getCode())){
-           extracted(amount, userAssets);
-       }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.TRANSFER.getCode())){
-            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
-           eUserAssets.setDesc(desc);
+            extracted(amount, userAssets);
+        }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.TRANSFER.getCode())){
+            if (amount.compareTo(BigDecimal.ZERO) > 0) {  //转换加钱,先补待补资金
+                if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
+                    BigDecimal availableBalance = amount.subtract(userAssets.getAmountToBeCovered());
+                    availableBalance = availableBalance.subtract(userAssets.getHandlingChargeWritten());
+                    if(availableBalance.compareTo(BigDecimal.ZERO) >= 0){
+                        userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(availableBalance));
+                        userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(userAssets.getAmountToBeCovered()));
+                        userAssets.setAmountToBeCovered(BigDecimal.ZERO);
+                        userAssets.setHandlingChargeWritten(BigDecimal.ZERO);
+                    }else{
+                        BigDecimal subtract = amount.subtract(userAssets.getHandlingChargeWritten());
+                        if(subtract.compareTo(BigDecimal.ZERO) >= 0) {
+                            BigDecimal decimal = amount.subtract(userAssets.getAmountToBeCovered());
+                            decimal = decimal.subtract(userAssets.getHandlingChargeWritten());
+                            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(amount.subtract(userAssets.getHandlingChargeWritten())));
+                            userAssets.setHandlingChargeWritten(BigDecimal.ZERO);
+                            userAssets.setAmountToBeCovered(decimal.negate());
+                        }else{
+                            userAssets.setHandlingChargeWritten(userAssets.getHandlingChargeWritten().subtract(amount));
+                        }
+                    }
+                }else{
+                    userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
+                }
+            } else { //扣钱
+                userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
+            }
+            eUserAssets.setDesc(desc);
         }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.TOP_UP.getCode())){
            if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
                BigDecimal availableBalance = amount.subtract(userAssets.getAmountToBeCovered());
@@ -214,9 +285,16 @@
                    userAssets.setAmountToBeCovered(BigDecimal.ZERO);
                    userAssets.setHandlingChargeWritten(BigDecimal.ZERO);
                }else{
-                   userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(amount));
-                   userAssets.setHandlingChargeWritten(BigDecimal.ZERO);
-                   userAssets.setAmountToBeCovered(availableBalance.negate());
+                   BigDecimal subtract = amount.subtract(userAssets.getHandlingChargeWritten());
+                   if(subtract.compareTo(BigDecimal.ZERO) >= 0) {
+                       BigDecimal decimal = amount.subtract(userAssets.getAmountToBeCovered());
+                       decimal = decimal.subtract(userAssets.getHandlingChargeWritten());
+                       userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(amount.subtract(userAssets.getHandlingChargeWritten())));
+                       userAssets.setHandlingChargeWritten(BigDecimal.ZERO);
+                       userAssets.setAmountToBeCovered(decimal.negate());
+                   }else{
+                       userAssets.setHandlingChargeWritten(userAssets.getHandlingChargeWritten().subtract(amount));
+                   }
                }
            }else{
                userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
@@ -227,13 +305,14 @@
         }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.HANDLING_CHARGE.getCode())){
            if(userAssets.getAvailableBalance().compareTo(amount.abs()) >= 0){
                userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount.negate()));
+               amountLog = amount.negate();
            }else {
                userAssets.setHandlingChargeWritten(userAssets.getHandlingChargeWritten().add(amount.abs()));
            }
            userAssets.setHandlingCharge(userAssets.getHandlingCharge().add(amount.abs()));
 
            extracted(amount.negate(),userAssets);
-       }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.CONSTRAINT_CLOSE_POSITION.getCode())){
+        }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.CONSTRAINT_CLOSE_POSITION.getCode())){
            userAssets.setFreezeMoney(userAssets.getFreezeMoney().subtract(amount));
            if(userAssets.getIsZf() == 0){
                userAssets.setCumulativeProfitAndLoss(userAssets.getCumulativeProfitAndLoss().subtract(amount));
@@ -241,19 +320,29 @@
                userAssets.setCumulativeProfitAndLoss(userAssets.getCumulativeProfitAndLoss().add(amount));
            }
            extracted(userAssets);
-       }
+        }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.PENDING_ORDER_FREEZE.getCode())){
+            // 挂单冻结资金:从可用余额转到冻结金额
+            // amount 是负数,表示扣除
+            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
+            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(amount.negate()));
+        }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.PENDING_ORDER_UNFREEZE.getCode())){
+            // 取消挂单解冻资金:从冻结金额转回可用余额
+            // amount 是正数,表示增加
+            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
+            userAssets.setFreezeMoney(userAssets.getFreezeMoney().subtract(amount));
+        }
 
-        if(null != userPosition){
+        /*if(null != userPosition){
             userPosition.setAmountToBeCovered(userAssets.getAmountToBeCovered().add(userAssets.getHandlingChargeWritten()));
             userPositionMapper.updateById(userPosition);
-        }
+        }*/
         String  after =  userAssets.getAvailableBalance().toString();
         MoneyLog moneyLog = new MoneyLog();
         moneyLog.setDescs(eUserAssets.getDesc());
         moneyLog.setBeFore(before);
         moneyLog.setAfter(after);
-        moneyLog.setAmount(amount.toString());
-        moneyLog.setAccectType(accectType);
+        moneyLog.setAmount(amountLog.toString());
+        moneyLog.setAccectType(accetType);
         moneyLog.setType(eUserAssets.getCode());
         moneyLog.setUserId(userId+"");
         moneyLog.setSymbol(EStockType.getEStockTypeByCode(accetType).getSymbol());
@@ -293,25 +382,49 @@
     }
 
 
-    public Boolean aiAvailableBalanceChange(UserAssets userAssets, EUserAssets eUserAssets, BigDecimal amount) {
+    public Boolean aiAvailableBalanceChange(String accetType, Integer userId, EUserAssets eUserAssets, BigDecimal amount) throws Exception {
+        //查询账户
+        /*UserAssets userAssets = assetsByTypeAndUserId(EStockType.getDefault().getCode(), userId);
+        //是否需要转换金额
+        if (!accetType.equals(EStockType.getDefault().getCode())) {
+            amount = exchangeAmountByRate(accetType, amount);
+        }*/
+        UserAssets userAssets = assetsByTypeAndUserId(accetType, userId);
         String before = userAssets.getAvailableBalance().toString();
         if (eUserAssets.getCode().equals(EUserAssets.BUY_AI.getCode())) {
             //冻结金额
             userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(amount));
             //扣除可用金额
-            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount.negate()));
+            userAssets.setAvailableBalance(userAssets.getAvailableBalance().subtract(amount));
+            amount = amount.negate();
         } else if (eUserAssets.getCode().equals(EUserAssets.BUY_AI_REJECT.getCode())) {
             //解除冻结金额
-            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(amount.negate()));
+            userAssets.setFreezeMoney(userAssets.getFreezeMoney().subtract(amount));
             //归还可用金额
             userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
         } else if (eUserAssets.getCode().equals(EUserAssets.AI_SETTLEMENT.getCode())) {
-            //解除冻结金额
-            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(amount.negate()));
-            //归还可用金额 + 收益
+            //解除购买冻结金额
+            userAssets.setFreezeMoney(userAssets.getFreezeMoney().subtract(amount));
+            //归还购买金额
             userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
+        } else if (eUserAssets.getCode().equals(EUserAssets.AI_SETTLEMENT_INT.getCode())) {
+            //收益
+            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
+        } else if(eUserAssets.getCode().equals(EUserAssets.DK.getCode())) {
+            //发放贷款金额
+            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
+        } else if(eUserAssets.getCode().equals(EUserAssets.RT_DK.getCode())) {
+            //归还贷款
+            userAssets.setAvailableBalance(userAssets.getAvailableBalance().subtract(amount));
+            amount = amount.negate();
+        } else if(eUserAssets.getCode().equals(EUserAssets.RT_DK_INT.getCode())) {
+            //贷款利息
+            userAssets.setAvailableBalance(userAssets.getAvailableBalance().subtract(amount));
+            amount = amount.negate();
         }
+
         String after = userAssets.getAvailableBalance().toString();
+        //资金明细
         MoneyLog moneyLog = new MoneyLog();
         moneyLog.setDescs(eUserAssets.getDesc());
         moneyLog.setBeFore(before);
@@ -319,7 +432,7 @@
         moneyLog.setAmount(amount.toString());
         moneyLog.setAccectType(userAssets.getAccectType());
         moneyLog.setType(eUserAssets.getCode());
-        moneyLog.setUserId(userAssets.getId()+"");
+        moneyLog.setUserId(userAssets.getUserId()+"");
         moneyLog.setSymbol(EStockType.getEStockTypeByCode(userAssets.getAccectType()).getSymbol());
         moneyLog.setCreateTime(new Date());
         moneyLogMapper.insert(moneyLog);
@@ -327,7 +440,19 @@
     }
 
     @Override
-    public BigDecimal exchangeAmountByRate(BigDecimal amount, BigDecimal rate) {
-        return amount.multiply(rate);
+    public BigDecimal exchangeAmountByRate(String fromType, String toType, BigDecimal amount) throws Exception {
+        EStockType stockType = EStockType.getEStockTypeByCode(fromType);
+        EStockType toStockType = EStockType.getEStockTypeByCode(toType);
+        ExchangeRate exchangeRate = exchangeRateRepository.findExchangeRateByCurrencyAndConversionCurrency(
+                stockType.getSymbol(), toStockType.getSymbol()).orElse(null);
+        if (exchangeRate != null) {
+            //转换为默认货币 保留5位小数
+            return amount.multiply(exchangeRate.getRata()).setScale(5, RoundingMode.HALF_UP);
+        } else {
+            log.error("exchangeAmountByRate is null:{}>>{}", stockType.getSymbol(), toStockType.getSymbol());
+            throw new Exception("货币转换汇率未设置");
+        }
+
+
     }
 }

--
Gitblit v1.9.3