From ef49bcedc5f787cbf47fca657b751a6727189f5e Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Fri, 15 Jul 2022 10:25:54 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java |    5 +++++
 src/main/java/cc/mrbird/febs/dapp/vo/WalletInfoVo.java                    |    3 +++
 src/main/resources/mapper/dapp/DappIdoAssetsDao.xml                       |    4 ++++
 src/main/java/cc/mrbird/febs/dapp/mapper/DappIdoAssetsDao.java            |    4 ++++
 4 files changed, 16 insertions(+), 0 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/dapp/mapper/DappIdoAssetsDao.java b/src/main/java/cc/mrbird/febs/dapp/mapper/DappIdoAssetsDao.java
index 55f8155..61d3b71 100644
--- a/src/main/java/cc/mrbird/febs/dapp/mapper/DappIdoAssetsDao.java
+++ b/src/main/java/cc/mrbird/febs/dapp/mapper/DappIdoAssetsDao.java
@@ -4,7 +4,11 @@
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import org.apache.ibatis.annotations.Param;
 
+import java.util.List;
+
 public interface DappIdoAssetsDao extends BaseMapper<DappIdoAssetsEntity> {
 
     DappIdoAssetsEntity selectByMemberId(@Param("memberId") Long memberId);
+
+    List<DappIdoAssetsEntity> selectListHasAssets();
 }
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 c9ae7d1..da5dc95 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
@@ -61,6 +61,11 @@
         walletInfo.setCoinAmount(idoAssets.getCoinAmount());
         walletInfo.setUsdtAmount(idoAssets.getUsdtAmount());
         walletInfo.setInviteId(member.getInviteId());
+
+        List<DappIdoAssetsEntity> hasAssets = dappIdoAssetsDao.selectListHasAssets();
+        if (CollUtil.isNotEmpty(hasAssets)) {
+            walletInfo.setBuyCnt(hasAssets.size());
+        }
         return walletInfo;
     }
 
diff --git a/src/main/java/cc/mrbird/febs/dapp/vo/WalletInfoVo.java b/src/main/java/cc/mrbird/febs/dapp/vo/WalletInfoVo.java
index 979b352..5f15bd5 100644
--- a/src/main/java/cc/mrbird/febs/dapp/vo/WalletInfoVo.java
+++ b/src/main/java/cc/mrbird/febs/dapp/vo/WalletInfoVo.java
@@ -49,4 +49,7 @@
 
     @ApiModelProperty(value = "IDO USDT数量")
     private BigDecimal usdtAmount;
+
+    @ApiModelProperty(value = "IDO 参与数量")
+    private int buyCnt;
 }
diff --git a/src/main/resources/mapper/dapp/DappIdoAssetsDao.xml b/src/main/resources/mapper/dapp/DappIdoAssetsDao.xml
index 7ff0bc1..f86e08a 100644
--- a/src/main/resources/mapper/dapp/DappIdoAssetsDao.xml
+++ b/src/main/resources/mapper/dapp/DappIdoAssetsDao.xml
@@ -5,4 +5,8 @@
     <select id="selectByMemberId" resultType="cc.mrbird.febs.dapp.entity.DappIdoAssetsEntity">
         select * from dapp_ido_assets where member_id=#{memberId}
     </select>
+
+    <select id="selectListHasAssets" resultType="cc.mrbird.febs.dapp.entity.DappIdoAssetsEntity">
+        select * from dapp_ido_assets where coin_amount > 0
+    </select>
 </mapper>
\ No newline at end of file

--
Gitblit v1.9.1