From 8a3f4a7f9df7d215bfa6048f01fe1d1d28e7d0dc Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Wed, 04 Sep 2024 16:15:30 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/service/impl/UserAssetsServices.java |   24 ++++++++++++++++++++++--
 1 files changed, 22 insertions(+), 2 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..bc730c5 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")){
@@ -57,7 +63,7 @@
 
     @Override
     public ServerResponse updateUserAssets(Integer id, String amt,String type) {
-        UserAssets userAssets = userAssetsMapper.selectById(id);
+        UserAssets userAssets = userAssetsMapper.selectOne(new LambdaQueryWrapper<>(UserAssets.class).eq(UserAssets::getUserId,id));
         //  0 入款  1是扣钱  2 是充值  3 是提币
         BigDecimal bigAmt = new BigDecimal(amt);
         if(type.equals("0") || type.equals("1")){
@@ -77,15 +83,29 @@
                 }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");
             }else{
 
 

--
Gitblit v1.9.3