From b52ecd08ce2c34f500ec414703dfea89a4f5cf3f Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Fri, 26 Jul 2024 10:36:18 +0800
Subject: [PATCH] 1

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

diff --git a/src/main/java/com/nq/service/impl/UserAssetsServices.java b/src/main/java/com/nq/service/impl/UserAssetsServices.java
index 9506c9e..8c448c4 100644
--- a/src/main/java/com/nq/service/impl/UserAssetsServices.java
+++ b/src/main/java/com/nq/service/impl/UserAssetsServices.java
@@ -4,10 +4,12 @@
 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;
@@ -35,6 +37,9 @@
 
     @Autowired
     UserWithdrawServiceImpl withdrawService;
+
+    @Resource
+    UserOptionLogMapper userOptionLogMapper;
 
     @Override
     public UserAssets assetsByTypeAndUserId(String accetType, Integer userId) {
@@ -77,8 +82,22 @@
                 }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("修改金额失败");

--
Gitblit v1.9.3