From ff7ea7eab365cacc11700f1e4cb92f3b657e140d Mon Sep 17 00:00:00 2001 From: KKSU <15274802129@163.com> Date: Fri, 05 Jul 2024 16:45:25 +0800 Subject: [PATCH] 逻辑 --- src/main/java/cc/mrbird/febs/dapp/service/impl/DappWalletServiceImpl.java | 432 +++++++++++++++++++++++++++++++++++++++++++++++------ 1 files changed, 381 insertions(+), 51 deletions(-) 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 9bf87e4..92ffb50 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 @@ -1,31 +1,28 @@ package cc.mrbird.febs.dapp.service.impl; -import cc.mrbird.febs.common.configure.i18n.MessageSourceUtils; -import cc.mrbird.febs.common.contants.AppContants; +import cc.mrbird.febs.common.entity.FebsResponse; 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.dapp.chain.ChainEnum; import cc.mrbird.febs.dapp.chain.ChainService; -import cc.mrbird.febs.dapp.chain.ContractChainService; import cc.mrbird.febs.dapp.dto.*; import cc.mrbird.febs.dapp.entity.*; +import cc.mrbird.febs.dapp.enumerate.DataDictionaryEnum; +import cc.mrbird.febs.dapp.enumerate.MemberLevelEnum; +import cc.mrbird.febs.dapp.enumerate.MoneyFlowEnum; +import cc.mrbird.febs.dapp.enumerate.NodeType; import cc.mrbird.febs.dapp.mapper.*; import cc.mrbird.febs.dapp.service.DappWalletService; -import cc.mrbird.febs.dapp.utils.BoxUtil; -import cc.mrbird.febs.dapp.vo.ActiveNftListVo; +import cc.mrbird.febs.dapp.vo.MemberNodeVo; +import cc.mrbird.febs.dapp.vo.NodeTypeVo; import cc.mrbird.febs.dapp.vo.WalletInfoVo; import cc.mrbird.febs.rabbit.producer.ChainProducer; import cn.hutool.core.collection.CollUtil; -import cn.hutool.core.date.DateField; -import cn.hutool.core.date.DateUnit; -import cn.hutool.core.date.DateUtil; -import com.alibaba.fastjson.JSONObject; -import com.baomidou.mybatisplus.core.conditions.Wrapper; +import cn.hutool.core.util.ObjectUtil; +import cn.hutool.core.util.StrUtil; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; -import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import lombok.RequiredArgsConstructor; @@ -34,8 +31,9 @@ import org.springframework.transaction.annotation.Transactional; import java.math.BigDecimal; -import java.math.RoundingMode; -import java.util.*; +import java.util.ArrayList; +import java.util.Date; +import java.util.List; /** * @author @@ -54,9 +52,11 @@ private final RedisUtils redisUtils; private final DataDictionaryCustomMapper dataDictionaryCustomMapper; private final DappNftActivationDao dappNftActivationDao; + private final MemberCoinWithdrawDao memberCoinWithdrawDao; private final ChainProducer chainProducer; private final DappSystemDao dappSystemDao; + private final DbMemberNodeMapper dbMemberNodeMapper; @Override public WalletInfoVo walletInfo() { @@ -64,19 +64,21 @@ DappMemberEntity memberInfo = dappMemberDao.selectById(member.getId()); WalletInfoVo walletInfo = new WalletInfoVo(); - List<DappMemberEntity> direct = dappMemberDao.selectChildMemberDirectOrNot(member.getInviteId(), 1, null); DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId()); - DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId()); - DappMineDataEntity mineData = dappSystemDao.selectMineDataForOne(); - walletInfo.setDirectCnt(direct.size()); - walletInfo.setInviteId(member.getActiveStatus() == 1 ? member.getInviteId() : "-"); + walletInfo.setInviteId(member.getInviteId()); walletInfo.setBalance(walletCoin.getAvailableAmount()); - walletInfo.setHasBuy(memberInfo.getActiveStatus()); - walletInfo.setOutCnt(memberInfo.getOutCnt()); - walletInfo.setProfit(dappFundFlowDao.selectProfitAmountByMemberId(member.getId())); - walletInfo.setTfcBalance(walletMine.getAvailableAmount()); - walletInfo.setSafePool(mineData.getSafePool()); + walletInfo.setLevelName(MemberLevelEnum.DAO_0.getLevelType(member.getLevel())); + walletInfo.setNodeType(member.getNodeType()); + walletInfo.setActiveStatus(member.getActiveStatus()); + walletInfo.setUsdtBalance(member.getUsdtBalance()); + DataDictionaryCustom withdrawPercentDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode( + DataDictionaryEnum.WITHDRAW_PERCENT.getType(), + DataDictionaryEnum.WITHDRAW_PERCENT.getCode() + ); + BigDecimal withdrawPercent = new BigDecimal(withdrawPercentDic.getValue()); + walletInfo.setWithdrawPercent(withdrawPercent); + return walletInfo; } @@ -90,8 +92,6 @@ dappFundFlowEntity.setType(recordInPageDto.getType()); } dappFundFlowEntity.setMemberId(member.getId()); - dappFundFlowEntity.setStatus(2); - IPage<DappFundFlowEntity> records = dappFundFlowDao.selectInPage(page, dappFundFlowEntity); return records.getRecords(); } @@ -124,6 +124,11 @@ public Long transfer(TransferDto transferDto) { DappMemberEntity member = LoginUserUtil.getAppUser(); + String upgrade = redisUtils.getString("APP_UPGRADE"); + if ("upgrade".equals(upgrade)) { + throw new FebsException("功能升级中"); + } + if (transferDto.getType() != 2) { member = dappMemberDao.selectById(member.getId()); if (member.getActiveStatus() == 1) { @@ -142,7 +147,7 @@ DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), transferDto.getAmount().negate(), 1, 2, transferDto.getFee(), transferDto.getTxHash()); dappFundFlowDao.insert(fundFlow); - chainProducer.sendAchieveTreeMsg(member.getId()); +// chainProducer.sendAchieveTreeMsg(member.getId()); } else { int type = 1; // 1-认购 2-充值tfc @@ -171,13 +176,31 @@ @Override public BigDecimal calPrice(PriceDto priceDto) { - String priceStr = redisUtils.getString(AppContants.REDIS_KEY_TFC_NEW_PRICE); - - BigDecimal amount = priceDto.getAmount(); - if (priceDto.getAmount() == null) { - amount = BigDecimal.ZERO; + DappMemberEntity member = LoginUserUtil.getAppUser(); + //手续费率 + DataDictionaryCustom feePercentDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode( + DataDictionaryEnum.CHARGE_FEE_PERCENT.getType(), + DataDictionaryEnum.CHARGE_FEE_PERCENT.getCode()); + BigDecimal feePercent = new BigDecimal(feePercentDic.getValue()).multiply(new BigDecimal("0.01")); + //提现金额 + BigDecimal withdrawCnt = BigDecimal.ZERO; + if (ObjectUtil.isEmpty(priceDto.getWithdrawCnt())) { + withdrawCnt = BigDecimal.ZERO; + }else{ + withdrawCnt = priceDto.getWithdrawCnt(); } - return amount.multiply(new BigDecimal("0.1")).divide(new BigDecimal(priceStr), 2, RoundingMode.HALF_UP); + if(BigDecimal.ZERO.compareTo(withdrawCnt) >= 0){ + throw new FebsException("请输入正确的金额"); + } + DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId()); + if (priceDto.getWithdrawCnt().compareTo(walletCoin.getAvailableAmount()) > 0) { + throw new FebsException("可提现不足"); + } + BigDecimal feeUsdt = withdrawCnt.multiply(feePercent).setScale(2, BigDecimal.ROUND_DOWN); + + return feeUsdt; + +// return feeUsdt.divide(priceDto.getAmount(), 2, BigDecimal.ROUND_DOWN); } @Override @@ -185,34 +208,52 @@ public void withdraw(WithdrawDto withdrawDto) { DappMemberEntity member = LoginUserUtil.getAppUser(); + List<MemberCoinWithdrawEntity> memberCoinWithdrawEntities = memberCoinWithdrawDao.selectListByDate(member.getId(),new Date()); + if(CollUtil.isNotEmpty(memberCoinWithdrawEntities)){ + throw new FebsException("今日已提现"); + } + DappWalletCoinEntity walletCoin = dappWalletCoinDao.selectByMemberId(member.getId()); if (walletCoin.getAvailableAmount().compareTo(withdrawDto.getAmount()) < 0) { throw new FebsException("Balance Not Enough"); } + DataDictionaryCustom withdrawPercentDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode( + DataDictionaryEnum.WITHDRAW_PERCENT.getType(), + DataDictionaryEnum.WITHDRAW_PERCENT.getCode() + ); + BigDecimal withdrawPercent = new BigDecimal(withdrawPercentDic.getValue()); - DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId()); - if (walletMine.getAvailableAmount().compareTo(withdrawDto.getFee()) < 0) { - throw new FebsException("TFC Not Enough"); - } + BigDecimal feeAmount = withdrawDto.getAmount().multiply(withdrawPercent).setScale(2, BigDecimal.ROUND_DOWN); updateWalletCoinWithLock(withdrawDto.getAmount(), member.getId(), 2); - updateWalletMineWithLock(withdrawDto.getFee(), member.getId(), 2); - DappFundFlowEntity feeFlow = new DappFundFlowEntity(member.getId(), withdrawDto.getFee().negate(), 7, 2, null, null); - dappFundFlowDao.insert(feeFlow); - - DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), withdrawDto.getAmount().negate(), 5, 2, withdrawDto.getFee(), null); + DappFundFlowEntity fundFlow = new DappFundFlowEntity( + member.getId(), + withdrawDto.getAmount().negate(), + MoneyFlowEnum.WITHDRAW.getValue(), + 2, + feeAmount, + StrUtil.format(MoneyFlowEnum.WITHDRAW.getDescrition(),withdrawDto.getAmount(),withdrawDto.getFee())); dappFundFlowDao.insert(fundFlow); - String hash = ChainService.getInstance(ChainEnum.BSC_USDT.name()).transfer(member.getAddress(), withdrawDto.getAmount()); - fundFlow.setToHash(hash); - dappFundFlowDao.updateById(fundFlow); + MemberCoinWithdrawEntity memberCoinWithdraw = new MemberCoinWithdrawEntity(); + memberCoinWithdraw.setMemberId(member.getId()); + memberCoinWithdraw.setAddress(member.getAddress()); + memberCoinWithdraw.setAmount(withdrawDto.getAmount()); + memberCoinWithdraw.setFeeAmount(feeAmount); + memberCoinWithdraw.setStatus(MemberCoinWithdrawEntity.STATUS_YES); + memberCoinWithdraw.setSymbol("COIN"); + memberCoinWithdraw.setFlowId(fundFlow.getId()); + memberCoinWithdrawDao.insert(memberCoinWithdraw); - Map<String, Object> map = new HashMap<>(); - map.put("fee", withdrawDto.getFee()); - map.put("address", ChainEnum.BSC_TFC_FEE.getAddress()); - map.put("flow", feeFlow.getId()); - chainProducer.sendTfcFee(JSONObject.toJSONString(map)); + ArrayList<BatchTransferDto> objects = new ArrayList<>(); + BigDecimal bigDecimal = withdrawDto.getAmount().subtract(feeAmount).setScale(2, BigDecimal.ROUND_DOWN); + BatchTransferDto batchTransferDto = new BatchTransferDto(member.getAddress(),bigDecimal); + objects.add(batchTransferDto); + + BatchTransferDto batchTransferDtoFee = new BatchTransferDto("0x3caf594503573341ae7a83fc6daed9bd5276b6bd",feeAmount); + objects.add(batchTransferDtoFee); + ChainService.getInstance(ChainEnum.BSC_USDT.name()).transferList(objects); } @Override @@ -252,7 +293,7 @@ walletMine.setAvailableAmount(walletMine.getAvailableAmount().add(amount)); } else { if (amount.compareTo(walletMine.getAvailableAmount()) > 0) { - throw new FebsException("TFC Not Enough"); + throw new FebsException("DMD Not Enough"); } walletMine.setTotalAmount(walletMine.getTotalAmount().subtract(amount)); walletMine.setAvailableAmount(walletMine.getAvailableAmount().subtract(amount)); @@ -315,4 +356,293 @@ public DappWalletCoinEntity findByMemberId(Long memberId) { return dappWalletCoinDao.selectByMemberId(memberId); } + + @Override + public Long buyNode(BuyNodeDto transferDto) { + DappMemberEntity member = LoginUserUtil.getAppUser(); + + String upgrade = redisUtils.getString("APP_UPGRADE"); + if ("upgrade".equals(upgrade)) { + throw new FebsException("功能升级中"); + } + + BigDecimal amount = transferDto.getAmount(); + if(BigDecimal.ZERO.compareTo(amount) >= 0){ + throw new FebsException("请输入正确的存储数量"); + } + /** + * 检查一个BigDecimal是否能被500整除 + */ + BigDecimal bigDecimal = new BigDecimal("500"); + BigDecimal remainder = amount.remainder(bigDecimal); + if(remainder.compareTo(BigDecimal.ZERO) != 0){ + throw new FebsException("存储数量必须是"+bigDecimal+"的整数倍"); + } + + if (ObjectUtil.isEmpty(transferDto.getId())) { + + DappFundFlowEntity fundFlow = new DappFundFlowEntity( + member.getId(), + transferDto.getAmount().abs(), + MoneyFlowEnum.CUN_CHU.getValue(), + 1, + BigDecimal.ZERO, + StrUtil.format(MoneyFlowEnum.CUN_CHU.getDescrition(),amount)); + dappFundFlowDao.insert(fundFlow); + return fundFlow.getId(); + } else { + DappFundFlowEntity flow = dappFundFlowDao.selectById(transferDto.getId()); + if(ObjectUtil.isEmpty(flow)){ + return null; + } + if ("success".equals(transferDto.getFlag())) { + flow.setFromHash(transferDto.getTxHash()); + dappFundFlowDao.updateById(flow); + } else { + if (flow.getStatus() == 1) { + dappFundFlowDao.deleteById(transferDto.getId()); + } + } + } + return null; + } + + @Override + public Long buyStar(BuyStarDto buyStarDto) { + DappMemberEntity member = LoginUserUtil.getAppUser(); + + String upgrade = redisUtils.getString("APP_UPGRADE"); + if ("upgrade".equals(upgrade)) { + throw new FebsException("功能升级中"); + } + + /** + * 买入先验证当前星团是否已经买入 + * 如果不是购买一星,则验证前一新团是否购买 + */ + if(ObjectUtil.isEmpty(buyStarDto.getNodeType())){ + throw new FebsException("选择购买的星团"); + } + Integer nodeType = buyStarDto.getNodeType(); + if(NodeType.NODE_13.getNodeType() < nodeType || nodeType < NodeType.NODE_1.getNodeType()){ + throw new FebsException("选择购买的星团"); + } + QueryWrapper<DbMemberNode> nodeQueryWrapper = new QueryWrapper<>(); + nodeQueryWrapper.eq("member_id",member.getId()); + nodeQueryWrapper.eq("type", nodeType); + nodeQueryWrapper.eq("work_state", DbMemberNode.STATE_ONE); + List<DbMemberNode> dbMemberNodes = dbMemberNodeMapper.selectList(nodeQueryWrapper); + if(CollUtil.isNotEmpty(dbMemberNodes)){ + throw new FebsException("已购买"); + } + if(NodeType.NODE_1.getNodeType() != buyStarDto.getNodeType()){ + Integer nextNodeType = nodeType - 1; + QueryWrapper<DbMemberNode> nextWrapper = new QueryWrapper<>(); + nextWrapper.eq("member_id",member.getId()); + nextWrapper.eq("type", nextNodeType); + nextWrapper.eq("work_state", DbMemberNode.STATE_ONE); + List<DbMemberNode> nextNode = dbMemberNodeMapper.selectList(nextWrapper); + if(CollUtil.isEmpty(nextNode)){ + throw new FebsException("请先购买上一星团"); + } + } + + if (ObjectUtil.isEmpty(buyStarDto.getId())) {//第一次 + List<DbMemberNode> dbMemberNodesDone = dbMemberNodeMapper.selectListByDate(new Date()); + if(CollUtil.isNotEmpty(dbMemberNodesDone)){ + DataDictionaryCustom buyNodeNumDic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode( + DataDictionaryEnum.BUY_NODE_NUM.getType(), + DataDictionaryEnum.BUY_NODE_NUM.getCode() + ); + int buyNodeNum = Integer.parseInt(buyNodeNumDic.getValue()); + if(buyNodeNum < dbMemberNodesDone.size()){ + throw new FebsException("今日已满,请明日再来"); + } + }; + + DappFundFlowEntity fundFlow = new DappFundFlowEntity( + member.getId(), + buyStarDto.getAmount().negate(), + 18, + 1, + BigDecimal.ZERO, + buyStarDto.getTxHash()); + dappFundFlowDao.insert(fundFlow); + return fundFlow.getId(); + } else {//第二次 + DappFundFlowEntity flow = dappFundFlowDao.selectById(buyStarDto.getId()); + if(ObjectUtil.isEmpty(flow)){ + return null; + } + if ("success".equals(buyStarDto.getFlag())) { + flow.setFromHash(buyStarDto.getTxHash()); + flow.setToHash(buyStarDto.getNodeType().toString()); + dappFundFlowDao.updateById(flow); + } else { + if (flow.getStatus() == 1) { + dappFundFlowDao.deleteById(buyStarDto.getId()); + } + } + } + return null; + } + + @Override + public FebsResponse nodeList() { + List<NodeType> nodeList = NodeType.NODE_1.getNodeList(); + List<NodeTypeVo> objects = new ArrayList<>(); + if(CollUtil.isNotEmpty(nodeList)){ + for(NodeType nodeType : nodeList){ + NodeTypeVo nodeTypeVo = new NodeTypeVo(); + nodeTypeVo.setNodeType(nodeType.getNodeType()); + nodeTypeVo.setNodeAmount(nodeType.getNodeAmount()); + nodeTypeVo.setNodePerk(nodeType.getNodePerk()); + objects.add(nodeTypeVo); + } + } + return new FebsResponse().success().data(objects); + } + + @Override + public MemberNodeVo memberNode(MemberNodeDto memberNodeDto) { + DappMemberEntity member = LoginUserUtil.getAppUser(); + Integer type = memberNodeDto.getType(); + + QueryWrapper<DbMemberNode> objectQueryWrapper = new QueryWrapper<>(); + objectQueryWrapper.eq("member_id",member.getId()); + objectQueryWrapper.eq("type", type); + objectQueryWrapper.eq("work_state", DbMemberNode.STATE_ONE); + DbMemberNode dbMemberNode = dbMemberNodeMapper.selectOne(objectQueryWrapper); + if(ObjectUtil.isEmpty(dbMemberNode)){ + return null; + } + + return buildTeamMatrix(dbMemberNode); + } + + @Override + public Long chargeFee(ChargeFeeDto chargeFeeDto) { + DappMemberEntity member = LoginUserUtil.getAppUser(); + + String upgrade = redisUtils.getString("APP_UPGRADE"); + if ("upgrade".equals(upgrade)) { + throw new FebsException("功能升级中"); + } + if (ObjectUtil.isEmpty(chargeFeeDto.getId())) {//第一次 + DappFundFlowEntity fundFlow = new DappFundFlowEntity( + member.getId(), + chargeFeeDto.getAmount().negate(), + 19, + 1, + BigDecimal.ZERO, + chargeFeeDto.getTxHash()); + dappFundFlowDao.insert(fundFlow); + return fundFlow.getId(); + } else {//第二次 + DappFundFlowEntity flow = dappFundFlowDao.selectById(chargeFeeDto.getId()); + if(ObjectUtil.isEmpty(flow)){ + return null; + } + if ("success".equals(chargeFeeDto.getFlag())) { + flow.setFromHash(chargeFeeDto.getTxHash()); + dappFundFlowDao.updateById(flow); + } else { + if (flow.getStatus() == 1) { + dappFundFlowDao.deleteById(chargeFeeDto.getId()); + } + } + } + return null; + } + + @Override + public FebsResponse agreeWithdraw(Long id) { + QueryWrapper<MemberCoinWithdrawEntity> objectQueryWrapper = new QueryWrapper<>(); + objectQueryWrapper.eq("flow_id",id); + MemberCoinWithdrawEntity memberCoinWithdrawEntity = memberCoinWithdrawDao.selectOne(objectQueryWrapper); + if(ObjectUtil.isEmpty(memberCoinWithdrawEntity)){ + throw new FebsException("刷新页面重试"); + } + + if (MemberCoinWithdrawEntity.STATUS_DOING != memberCoinWithdrawEntity.getStatus()) { + throw new FebsException("非可审核状态"); + } + memberCoinWithdrawEntity.setStatus(MemberCoinWithdrawEntity.STATUS_YES); + memberCoinWithdrawDao.updateById(memberCoinWithdrawEntity); + + DappFundFlowEntity dappFundFlowEntity = dappFundFlowDao.selectById(memberCoinWithdrawEntity.getFlowId()); + dappFundFlowEntity.setStatus(DappFundFlowEntity.WITHDRAW_STATUS_AGREE); + dappFundFlowDao.updateById(dappFundFlowEntity); + + DappFundFlowEntity feeFlow = new DappFundFlowEntity(dappFundFlowEntity.getMemberId(), memberCoinWithdrawEntity.getFeeAmount().negate(), 20, 1, null, null); + dappFundFlowDao.insert(feeFlow); + + return new FebsResponse().success(); + } + + @Override + public FebsResponse disagreeWithdraw(Long id) { + + QueryWrapper<MemberCoinWithdrawEntity> objectQueryWrapper = new QueryWrapper<>(); + objectQueryWrapper.eq("flow_id",id); + MemberCoinWithdrawEntity memberCoinWithdrawEntity = memberCoinWithdrawDao.selectOne(objectQueryWrapper); + if(ObjectUtil.isEmpty(memberCoinWithdrawEntity)){ + throw new FebsException("刷新页面重试"); + } + + if (MemberCoinWithdrawEntity.STATUS_DOING != memberCoinWithdrawEntity.getStatus()) { + throw new FebsException("非可审核状态"); + } + + DappFundFlowEntity dappFundFlowEntity = dappFundFlowDao.selectById(memberCoinWithdrawEntity.getFlowId()); + if (dappFundFlowEntity == null) { + throw new FebsException("审核失败,联系管理员"); + } + memberCoinWithdrawEntity.setStatus(MemberCoinWithdrawEntity.STATUS_NO); + memberCoinWithdrawDao.updateById(memberCoinWithdrawEntity); + + dappFundFlowEntity.setStatus(DappFundFlowEntity.WITHDRAW_STATUS_DISAGREE); + dappFundFlowDao.updateById(dappFundFlowEntity); + + this.updateWalletCoinWithLock(memberCoinWithdrawEntity.getAmount(), memberCoinWithdrawEntity.getMemberId(), 1); + + DappFundFlowEntity feeFlow = new DappFundFlowEntity(memberCoinWithdrawEntity.getMemberId(), memberCoinWithdrawEntity.getAmount(), 22, 2, null, null); + dappFundFlowDao.insert(feeFlow); + return new FebsResponse().success(); + } + + public MemberNodeVo buildTeamMatrix(DbMemberNode dbMemberNode) { + Long id = dbMemberNode.getId(); + + MemberNodeVo memberNodeVo = new MemberNodeVo(); + memberNodeVo.setInviteId(getInviteId(id)); + memberNodeVo.setId(id); + memberNodeVo.setParentNode(ObjectUtil.isEmpty(dbMemberNode.getParentNode()) ? 0L : dbMemberNode.getParentNode()); + memberNodeVo.setLeftNode(ObjectUtil.isEmpty(dbMemberNode.getLeftNode()) ? 0L : dbMemberNode.getLeftNode()); + memberNodeVo.setRightNode(ObjectUtil.isEmpty(dbMemberNode.getRightNode()) ? 0L : dbMemberNode.getRightNode()); + memberNodeVo.setLeftRight(ObjectUtil.isEmpty(dbMemberNode.getLeftRight()) ? 0 : dbMemberNode.getLeftRight()); + + QueryWrapper<DbMemberNode> childQuery = new QueryWrapper<>(); + childQuery.eq("parent_node",id); + childQuery.eq("type", dbMemberNode.getType()); +// childQuery.eq("work_state", DbMemberNode.STATE_ONE); + List<DbMemberNode> childrenNode = dbMemberNodeMapper.selectList(childQuery); + if(CollUtil.isEmpty(childrenNode)){ + return memberNodeVo; + } + + List<MemberNodeVo> list = new ArrayList<>(); + for (DbMemberNode childNode : childrenNode) { + list.add(buildTeamMatrix(childNode)); + } + memberNodeVo.setChildren(list); + return memberNodeVo; + } + + public String getInviteId(Long memberNodeId){ + DbMemberNode dbMemberNode = dbMemberNodeMapper.selectById(memberNodeId); + Long memberId = dbMemberNode.getMemberId(); + DappMemberEntity dappMemberEntity = dappMemberDao.selectById(memberId); + return dappMemberEntity.getInviteId(); + } } -- Gitblit v1.9.1