From fc4b1e66b241df686dbd7bebb6df6793060a8107 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Sun, 10 Nov 2024 21:36:53 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/UserAssetsServices.java |   68 ++++++++++++++++++++++++++++++---
 1 files changed, 61 insertions(+), 7 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..a7f3319 100644
--- a/src/main/java/com/nq/service/impl/UserAssetsServices.java
+++ b/src/main/java/com/nq/service/impl/UserAssetsServices.java
@@ -1,13 +1,16 @@
 package com.nq.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.nq.common.ServerResponse;
 import com.nq.dao.MoneyLogMapper;
 import com.nq.dao.UserAssetsMapper;
+import com.nq.dao.UserOptionLogMapper;
 import com.nq.enums.EStockType;
 import com.nq.enums.EUserAssets;
 import com.nq.pojo.MoneyLog;
 import com.nq.pojo.UserAssets;
+import com.nq.pojo.UserOptionLog;
 import com.nq.service.IUserAssetsServices;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -36,6 +39,9 @@
     @Autowired
     UserWithdrawServiceImpl withdrawService;
 
+    @Resource
+    UserOptionLogMapper userOptionLogMapper;
+
     @Override
     public UserAssets assetsByTypeAndUserId(String accetType, Integer userId) {
         if(accetType.equals("SZHB")){
@@ -56,24 +62,50 @@
     }
 
     @Override
-    public ServerResponse updateUserAssets(Integer id, String amt,String type) {
-        UserAssets userAssets = userAssetsMapper.selectById(id);
+    public ServerResponse updateUserAssets(Integer id, String amt,String type,String accectType) {
+        UserAssets userAssets = userAssetsMapper.selectOne(new LambdaQueryWrapper<>(UserAssets.class).eq(UserAssets::getUserId,id).eq(UserAssets::getAccectType,accectType));
         //  0 入款  1是扣钱  2 是充值  3 是提币
         BigDecimal bigAmt = new BigDecimal(amt);
         if(type.equals("0") || type.equals("1")){
             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){
+                userOptionLogMapper.insert(UserOptionLog.builder()
+                        .userId(userAssets.getUserId())
+                        .money(bigAmt)
+                        .userAccectId(userAssets.getId())
+                        .createTime(new Date())
+                        .type(amt.contains("-")?0:1)
+                        .build());
                 return ServerResponse.createBySuccess();
             }else{
                 return ServerResponse.createByErrorMsg("修改金额失败");
             }
         }else{
             if(type.equals("2")){
-                return  userRechargeService.createOrder(userAssets.getUserId(),1,bigAmt.intValue(),"1");
+                return  userRechargeService.createOrder(id,1,bigAmt.intValue(),"1",accectType);
             }else{
 
 
@@ -114,7 +146,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 +165,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