From 47267bb39808dc4b371df933c6aa1255c32a8840 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Fri, 26 Apr 2024 11:45:46 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/UserAssetsServices.java |   49 +++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 45 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/nq/service/impl/UserAssetsServices.java b/src/main/java/com/nq/service/impl/UserAssetsServices.java
index 29f2e2f..7ab3733 100644
--- a/src/main/java/com/nq/service/impl/UserAssetsServices.java
+++ b/src/main/java/com/nq/service/impl/UserAssetsServices.java
@@ -64,8 +64,27 @@
             if(type.equals("1")){
                 bigAmt = bigAmt.negate();
             }
-
-            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(bigAmt));
+            if(type.equals("0")){
+                if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
+                    BigDecimal availableBalance = bigAmt.subtract(userAssets.getAmountToBeCovered());
+                    if(availableBalance.compareTo(BigDecimal.ZERO) > 0){
+                        userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(availableBalance));
+                        userAssets.setAmountToBeCovered(BigDecimal.ZERO);
+                    }else{
+                        BigDecimal amountToBeCovered = userAssets.getAmountToBeCovered().subtract(bigAmt);
+                        userAssets.setAmountToBeCovered(amountToBeCovered);
+                    }
+                }else{
+                    userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(bigAmt));
+                }
+            }else{
+                if(userAssets.getAvailableBalance().compareTo(BigDecimal.ZERO) <= 0){
+                    return ServerResponse.createByErrorMsg("客户账户余额为0");
+                }else if(userAssets.getAvailableBalance().subtract(bigAmt.abs()).compareTo(BigDecimal.ZERO) < 0){
+                    return ServerResponse.createByErrorMsg("客户账户余额不足扣款");
+                }
+                userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(bigAmt));
+            }
             if( userAssetsMapper.updateById(userAssets)>0){
                 return ServerResponse.createBySuccess();
             }else{
@@ -114,7 +133,18 @@
             userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
            extracted(amount, userAssets);
        }else  if(Objects.equals(eUserAssets.getCode(), EUserAssets.CLOSE_POSITION_RETURN_SECURITY_DEPOSIT.getCode())){
-            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
+           if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
+               BigDecimal availableBalance = amount.subtract(userAssets.getAmountToBeCovered());
+               if(availableBalance.compareTo(BigDecimal.ZERO) > 0){
+                   userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(availableBalance));
+                   userAssets.setAmountToBeCovered(BigDecimal.ZERO);
+               }else{
+                   BigDecimal amountToBeCovered = userAssets.getAmountToBeCovered().subtract(amount);
+                   userAssets.setAmountToBeCovered(amountToBeCovered);
+               }
+           }else{
+               userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
+           }
             userAssets.setFreezeMoney(userAssets.getFreezeMoney().subtract(amount));
         }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.CALCULATE_PROFIT_AND_LOSS.getCode())){
            extracted(amount, userAssets);
@@ -122,7 +152,18 @@
             userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
            eUserAssets.setDesc(desc);
         }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.TOP_UP.getCode())){
-            userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
+           if(userAssets.getAmountToBeCovered().compareTo(BigDecimal.ZERO) > 0){
+               BigDecimal availableBalance = amount.subtract(userAssets.getAmountToBeCovered());
+               if(availableBalance.compareTo(BigDecimal.ZERO) > 0){
+                   userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(availableBalance));
+                   userAssets.setAmountToBeCovered(BigDecimal.ZERO);
+               }else{
+                   BigDecimal amountToBeCovered = userAssets.getAmountToBeCovered().subtract(amount);
+                   userAssets.setAmountToBeCovered(amountToBeCovered);
+               }
+           }else{
+               userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
+           }
         }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.WITHDRAW.getCode())){
             userAssets.setAvailableBalance(userAssets.getAvailableBalance().add(amount));
         }else if(Objects.equals(eUserAssets.getCode(), EUserAssets.HANDLING_CHARGE.getCode())){

--
Gitblit v1.9.3