From 7df9727dfebb36744607062c4e8cfc56fb52851b Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Sun, 28 Aug 2022 01:57:16 +0800
Subject: [PATCH] fix
---
src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java | 5 +++++
src/main/java/cc/mrbird/febs/dapp/dto/WithdrawDto.java | 3 +++
2 files changed, 8 insertions(+), 0 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/dapp/dto/WithdrawDto.java b/src/main/java/cc/mrbird/febs/dapp/dto/WithdrawDto.java
index ec17b38..2a0a642 100644
--- a/src/main/java/cc/mrbird/febs/dapp/dto/WithdrawDto.java
+++ b/src/main/java/cc/mrbird/febs/dapp/dto/WithdrawDto.java
@@ -4,6 +4,7 @@
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
+import javax.validation.constraints.NotNull;
import java.math.BigDecimal;
/**
@@ -14,9 +15,11 @@
@ApiModel(value = "WithdrawDto", description = "提现")
public class WithdrawDto {
+ @NotNull(message = "Param Error")
@ApiModelProperty(value = "提现金额")
private BigDecimal amount;
+ @NotNull(message = "Param Error")
@ApiModelProperty(value = "手续费")
private BigDecimal fee;
}
diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
index 76aa551..8e56a86 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java
@@ -178,6 +178,11 @@
public void withdraw(WithdrawDto withdrawDto) {
DappMemberEntity member = LoginUserUtil.getAppUser();
+ DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId());
+ if (walletCoin.getAvailableAmount().compareTo(withdrawDto.getAmount()) < 0) {
+ throw new FebsException("Balance Not Enough");
+ }
+
updateWalletCoinWithLock(withdrawDto.getAmount().negate(), member.getId());
updateWalletMineWithLock(withdrawDto.getFee().negate(), member.getId());
--
Gitblit v1.9.1