From 68b9175323a2f9b40ffcc48bb01af1d8279e5d01 Mon Sep 17 00:00:00 2001
From: zyy <zyy@email.com>
Date: Fri, 22 Aug 2025 10:21:51 +0800
Subject: [PATCH] 平仓返回保证金修改

---
 src/main/java/com/nq/service/impl/UserAssetsServices.java |  172 +++++++++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 143 insertions(+), 29 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/UserAssetsServices.java b/src/main/java/com/nq/service/impl/UserAssetsServices.java
index db2dc98..f424c5d 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;
@@ -10,11 +10,13 @@
 import com.nq.pojo.*;
 import com.nq.service.IUserAssetsServices;
 import com.nq.utils.KeyUtils;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
 import java.math.BigDecimal;
+import java.math.RoundingMode;
 import java.util.Date;
 import java.util.List;
 import java.util.Objects;
@@ -22,6 +24,7 @@
 /**
  * 用户资产
  */
+@Slf4j
 @Service
 public class UserAssetsServices implements IUserAssetsServices {
 
@@ -34,9 +37,6 @@
     @Autowired
     UserRechargeServiceImpl userRechargeService;
 
-    @Autowired
-    UserWithdrawServiceImpl withdrawService;
-
     @Resource
     UserRechargeMapper userRechargeMapper;
 
@@ -46,10 +46,17 @@
     @Autowired
     UserPositionMapper userPositionMapper;
 
+    @Autowired
+    ExchangeRateRepository exchangeRateRepository;
+
+
     @Override
     public UserAssets assetsByTypeAndUserId(String accetType, Integer userId) {
-        if(accetType.equals("SZHB")){
+        /*if(accetType.equals("SZHB")){
             accetType = "US";
+        }*/
+        if(accetType.equals(EStockType.US.getCode()) || accetType.equals("IN")){
+            accetType = EStockType.MX.getCode();
         }
         QueryWrapper<UserAssets>  queryWrapper = new QueryWrapper<>();
         queryWrapper.eq("accect_type",accetType);
@@ -78,10 +85,12 @@
             if(type.equals("0")){
                 if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
                     BigDecimal availableBalance = bigAmt.subtract(userAssets.getAmountToBeCovered());
-                    if(availableBalance.compareTo(BigDecimal.ZERO) > 0){
+                    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);
                         if(null != userPosition){
                             userPosition.setAmountToBeCovered(userAssets.getAmountToBeCovered());
                         }
@@ -162,60 +171,92 @@
 
 
     @Override
-    public Boolean availablebalanceChange(String accetType, Integer userId, EUserAssets eUserAssets, BigDecimal amount, String desc, String descType) {
-
-        UserAssets userAssets =   assetsByTypeAndUserId(accetType,userId);
+    public Boolean availablebalanceChange(String accetType, Integer userId, EUserAssets eUserAssets, BigDecimal amount, String desc, String descType) throws Exception {
+        //UserAssets userAssets = assetsByTypeAndUserId(accetType,userId);
+        //查询墨西哥账户
+        UserAssets userAssets = assetsByTypeAndUserId(EStockType.MX.getCode(), userId);
+        //如果不是墨西哥币需要转换金额
+        if (!accetType.equals(EStockType.MX.getCode())) {
+            amount = exchangeAmountByRate(accetType, amount);
+        }
         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());
-               if(availableBalance.compareTo(BigDecimal.ZERO) > 0){
+               availableBalance = availableBalance.subtract(userAssets.getHandlingChargeWritten());
+               if(availableBalance.compareTo(BigDecimal.ZERO) >= 0){
                    userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(availableBalance));
                    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.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));
            }
-            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())){
+            extracted(amount, userAssets);
+        }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.TRANSFER.getCode())){
             userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
-           eUserAssets.setDesc(desc);
+            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());
-               if(availableBalance.compareTo(BigDecimal.ZERO) > 0){
+               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 amountToBeCovered = userAssets.getAmountToBeCovered().subtract(amount);
-                   userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(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.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 if(Objects.equals(eUserAssets.getCode(), EUserAssets.WITHDRAW.getCode())){
             userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
+            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(amount.abs()));
         }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()));
-           userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount.negate()));
+
            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));
@@ -223,10 +264,10 @@
                userAssets.setCumulativeProfitAndLoss(userAssets.getCumulativeProfitAndLoss().add(amount));
            }
            extracted(userAssets);
-       }
+        }
 
         if(null != userPosition){
-            userPosition.setAmountToBeCovered(userAssets.getAmountToBeCovered());
+            userPosition.setAmountToBeCovered(userAssets.getAmountToBeCovered().add(userAssets.getHandlingChargeWritten()));
             userPositionMapper.updateById(userPosition);
         }
         String  after =  userAssets.getAvailableBalance().toString();
@@ -234,8 +275,8 @@
         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());
@@ -273,4 +314,77 @@
         }
         extracted(userAssets);
     }
+
+
+    public Boolean aiAvailableBalanceChange(String accetType, Integer userId, EUserAssets eUserAssets, BigDecimal amount) throws Exception {
+        //查询墨西哥账户
+        UserAssets userAssets = assetsByTypeAndUserId(EStockType.MX.getCode(), userId);
+        //如果不是墨西哥币需要转换金额
+        if (!accetType.equals(EStockType.MX.getCode())) {
+            amount = exchangeAmountByRate(accetType, amount);
+        }
+        String before = userAssets.getAvailableBalance().toString();
+        if (eUserAssets.getCode().equals(EUserAssets.BUY_AI.getCode())) {
+            //冻结金额
+            userAssets.setFreezeMoney(userAssets.getFreezeMoney().add(amount));
+            //扣除可用金额
+            userAssets.setAvailableBalance(userAssets.getAvailableBalance().subtract(amount));
+            amount = amount.negate();
+        } else if (eUserAssets.getCode().equals(EUserAssets.BUY_AI_REJECT.getCode())) {
+            //解除冻结金额
+            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().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);
+        moneyLog.setAfter(after);
+        moneyLog.setAmount(amount.toString());
+        moneyLog.setAccectType(userAssets.getAccectType());
+        moneyLog.setType(eUserAssets.getCode());
+        moneyLog.setUserId(userAssets.getUserId()+"");
+        moneyLog.setSymbol(EStockType.getEStockTypeByCode(userAssets.getAccectType()).getSymbol());
+        moneyLog.setCreateTime(new Date());
+        moneyLogMapper.insert(moneyLog);
+        return userAssetsMapper.updateById(userAssets)>1;
+    }
+
+    @Override
+    public BigDecimal exchangeAmountByRate(String accetType, BigDecimal amount) throws Exception {
+        EStockType stockType = EStockType.getEStockTypeByCode(accetType);
+        ExchangeRate exchangeRate = exchangeRateRepository.findExchangeRateByCurrencyAndConversionCurrency(
+                stockType.getSymbol(), EStockType.MX.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(), EStockType.MX.getSymbol());
+            throw new Exception("货币转换汇率未设置");
+        }
+
+
+    }
 }

--
Gitblit v1.9.3