From 6c29204c6a4afed0eb476c92bb4f464c88cde3a9 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Fri, 27 May 2022 18:43:38 +0800
Subject: [PATCH] fix
---
src/main/java/cc/mrbird/febs/dapp/service/impl/DappMemberServiceImpl.java | 62 +++++++++++++++++++++++++++++++
1 files changed, 62 insertions(+), 0 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 6acec79..7e41717 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,11 +4,13 @@
import cc.mrbird.febs.common.entity.QueryRequest;
import cc.mrbird.febs.common.exception.FebsException;
import cc.mrbird.febs.common.utils.FebsUtil;
+import cc.mrbird.febs.common.utils.LoginUserUtil;
import cc.mrbird.febs.common.utils.RedisUtils;
import cc.mrbird.febs.common.utils.ShareCodeUtil;
import cc.mrbird.febs.dapp.chain.ChainService;
import cc.mrbird.febs.dapp.chain.ContractChainService;
import cc.mrbird.febs.dapp.dto.ApproveDto;
+import cc.mrbird.febs.dapp.dto.ConnectDto;
import cc.mrbird.febs.dapp.dto.PriceSettingDto;
import cc.mrbird.febs.dapp.entity.DappMemberEntity;
import cc.mrbird.febs.dapp.entity.DappTransferRecordEntity;
@@ -31,6 +33,7 @@
import org.springframework.transaction.annotation.Transactional;
import java.math.BigDecimal;
+import java.util.HashMap;
import java.util.Map;
/**
@@ -114,6 +117,26 @@
walletCoin.setAvailableAmount(BigDecimal.ZERO);
walletCoin.setFrozenAmount(BigDecimal.ZERO);
dappWalletCoinDao.insert(walletCoin);
+ }
+
+ @Override
+ public void connect(ConnectDto connectDto) {
+ DappMemberEntity member = dappMemberDao.selectByAddress(connectDto.getAddress(), null);
+
+ if (member == null) {
+ member = insertMember(connectDto.getAddress(), null);
+ }
+
+ String key = LoginUserUtil.getLoginKey(connectDto.getAddress(), connectDto.getNonce(), connectDto.getSign());
+ redisUtils.set(connectDto.getAddress(), member);
+
+ Map<Object, Object> keys = redisUtils.hmget(AppContants.REDIS_KEY_SIGN);
+ if (keys == null) {
+ keys = new HashMap<>();
+ }
+
+ keys.put(connectDto.getAddress(), key);
+ redisUtils.hmset(AppContants.REDIS_KEY_SIGN, keys);
}
@Override
@@ -232,4 +255,43 @@
redisUtils.set(AppContants.REDIS_KEY_ETH_NEW_PRICE, priceSettingDto.getNewestPrice());
}
}
+
+ @Override
+ public DappMemberEntity insertMember(String address, String refererId) {
+ DappMemberEntity member = new DappMemberEntity();
+ member.setAddress(address);
+
+ dappMemberDao.insert(member);
+
+ member.setInviteId(ShareCodeUtil.toSerialCode(member.getId()));
+ member.setRefererId(refererId);
+ if (StrUtil.isNotBlank(refererId)){
+ boolean flag = false;
+ String parentId = refererId;
+ String ids = "";
+ while (!flag && StringUtils.isNotBlank(parentId)) {
+ if (StrUtil.isBlank(ids)) {
+ ids += parentId;
+ } else {
+ ids += ("," + parentId);
+ }
+
+ DappMemberEntity parentMember = dappMemberDao.selectMemberInfoByInviteId(parentId);
+ if (parentMember == null) {
+ break;
+ }
+ parentId = parentMember.getRefererId();
+ if(StringUtils.isBlank(parentId)){
+ break;
+ }
+ if (parentMember.getRefererId().equals(parentMember.getInviteId())) {
+ flag = true;
+ }
+ }
+ member.setRefererIds(ids);
+ }
+ dappMemberDao.updateById(member);
+
+ return member;
+ }
}
--
Gitblit v1.9.1