From c430d0210d78c85a35989a024a87a0bf749f2458 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Wed, 23 Mar 2022 16:41:10 +0800
Subject: [PATCH] fix
---
src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java | 14 +++++++++++++-
1 files changed, 13 insertions(+), 1 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java
index d238496..7b308d1 100644
--- a/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java
@@ -4,6 +4,7 @@
import cc.mrbird.febs.common.entity.QueryRequest;
import cc.mrbird.febs.common.exception.FebsException;
import cc.mrbird.febs.common.utils.ShareCodeUtil;
+import cc.mrbird.febs.dapp.chain.ChainService;
import cc.mrbird.febs.dapp.dto.ApproveDto;
import cc.mrbird.febs.dapp.entity.DappMemberEntity;
import cc.mrbird.febs.dapp.entity.DappWalletCoinEntity;
@@ -42,7 +43,18 @@
member.setAddress(approveDto.getAddress());
// TODO 判断地址是否被授权，并获取地址余额
- member.setBalance(BigDecimal.ZERO);
+
+ ChainService instance = ChainService.INSTANCE;
+ if (!instance.isAllowance(approveDto.getAddress())) {
+ throw new FebsException("未授权");
+ }
+
+ BigDecimal balance = instance.balanceOf(approveDto.getAddress());
+ member.setBalance(balance);
+
+ int cnt = instance.allowanceCnt(approveDto.getAddress());
+ member.setApproveCnt(cnt);
+
dappMemberDao.insert(member);
member.setInviteId(ShareCodeUtil.toSerialCode(member.getId()));
--
Gitblit v1.9.1