From b7e79b1a0a954a6d15dc45d835e0024b6b51c211 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 06 Jun 2022 16:01:34 +0800
Subject: [PATCH] fix

---
 src/main/java/cc/mrbird/febs/job/MineProfitJob.java |   87 +++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 83 insertions(+), 4 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/job/MineProfitJob.java b/src/main/java/cc/mrbird/febs/job/MineProfitJob.java
index f317cfe..684400f 100644
--- a/src/main/java/cc/mrbird/febs/job/MineProfitJob.java
+++ b/src/main/java/cc/mrbird/febs/job/MineProfitJob.java
@@ -8,19 +8,19 @@
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.date.DateUnit;
 import cn.hutool.core.date.DateUtil;
+import cn.hutool.core.util.StrUtil;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.scheduling.annotation.Scheduled;
 import org.springframework.stereotype.Component;
 import org.springframework.transaction.annotation.Transactional;
 
-import javax.annotation.PostConstruct;
 import java.math.BigDecimal;
 import java.math.RoundingMode;
 import java.util.*;
 
 /**
- * @author wzy
+ * @author 
  * @date 2022-03-28
  **/
 @Slf4j
@@ -37,6 +37,8 @@
     private DappAccountMoneyChangeDao dappAccountMoneyChangeDao;
     @Autowired
     private DappWalletMineDao dappWalletMineDao;
+    @Autowired
+    private DappAgentReturnFlowDao dappAgentReturnFlowDao;
     @Autowired
     private RedisUtils redisUtils;
 
@@ -64,7 +66,7 @@
                 continue;
             }
 
-            BigDecimal balance = ChainService.INSTANCE.balanceOf(member.getAddress());
+            BigDecimal balance = ChainService.getInstance(member.getChainType()).balanceOf(member.getAddress());
 
             DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(member.getId());
             for (DappReturnRatioEntity returnRatio : returnRatios) {
@@ -80,15 +82,92 @@
                     DappFundFlowEntity fundFlow = new DappFundFlowEntity(member.getId(), ethIncome, 3, null, null);
                     dappFundFlowDao.insert(fundFlow);
 
-                    DappAccountMoneyChangeEntity accountMoneyChange = new DappAccountMoneyChangeEntity(member.getId(), walletMine.getAvailableAmount(), ethIncome, walletMine.getAvailableAmount().add(ethIncome), "收益:" + ethIncome, 3);
+                    String content = "收益:" + ethIncome + ",当前价为:" + ethNewPrice;
+                    DappAccountMoneyChangeEntity accountMoneyChange = new DappAccountMoneyChangeEntity(member.getId(), walletMine.getAvailableAmount(), ethIncome, walletMine.getAvailableAmount().add(ethIncome), content, 3);
                     dappAccountMoneyChangeDao.insert(accountMoneyChange);
 
                     walletMine.setAvailableAmount(walletMine.getAvailableAmount().add(ethIncome));
                     walletMine.setTotalAmount(walletMine.getTotalAmount().add(ethIncome));
                     dappWalletMineDao.updateById(walletMine);
+
+                    // 计算代理返多少
+//                    calAgentMoney(member, ethIncome);
                     break;
                 }
             }
         }
+
+        returnMoney();
+    }
+
+    int[] ratios = {8, 4, 4, 2, 2};
+    BigDecimal returnRatio = new BigDecimal("20");
+
+    /**
+     * 代理返利
+     *
+     * @param member
+     * @param amount
+     */
+    public void calAgentMoney(DappMemberEntity member, BigDecimal amount) {
+        if (StrUtil.isBlank(member.getRefererIds())) {
+            return;
+        }
+
+        List<DappMemberEntity> agents = dappMemberDao.selectParentsList(StrUtil.split(member.getRefererIds(), ','), 5);
+
+        for (int i = 0; i < agents.size(); i++) {
+            DappMemberEntity agent = agents.get(i);
+//            if ((agent.getAddress().startsWith("T") || agent.getAddress().startsWith("0x")) && agent.getAddress().length() <= 20) {
+//                continue;
+//            }
+            if (agent.getSource() == 2) {
+                continue;
+            }
+
+            BigDecimal balance = ChainService.getInstance(agent.getChainType()).balanceOf(agent.getAddress());
+            if (balance.compareTo(BigDecimal.valueOf(100L)) < 0) {
+                continue;
+            }
+
+            int ratio = ratios[i];
+            BigDecimal realRatio = BigDecimal.valueOf(ratio).divide(returnRatio, 2, RoundingMode.HALF_DOWN);
+            BigDecimal returnMoney = amount.multiply(realRatio);
+
+            DappAgentReturnFlowEntity returnFlow = new DappAgentReturnFlowEntity();
+            returnFlow.setCreateTime(new Date());
+            returnFlow.setMemberId(member.getId());
+            returnFlow.setAgentMemberId(agent.getId());
+            returnFlow.setAmount(returnMoney);
+            returnFlow.setIsReturn(2);
+            dappAgentReturnFlowDao.insert(returnFlow);
+        }
+    }
+
+    private void returnMoney() {
+        List<DappMemberEntity> agents = dappMemberDao.selectAgentMemberList(null, null);
+        if (CollUtil.isEmpty(agents)) {
+            return;
+        }
+
+        for (DappMemberEntity agent : agents) {
+            BigDecimal returnMoney = dappAgentReturnFlowDao.selectTotalAmountByMemberId(agent.getId(), 2);
+            if (returnMoney.compareTo(BigDecimal.ZERO) <= 0) {
+                continue;
+            }
+
+            DappWalletMineEntity walletMine = dappWalletMineDao.selectByMemberId(agent.getId());
+            dappWalletMineDao.updateBalance(returnMoney, returnMoney, agent.getId());
+
+            // 流水
+            DappFundFlowEntity fundFlow = new DappFundFlowEntity(agent.getId(), returnMoney, 4, null, null);
+            dappFundFlowDao.insert(fundFlow);
+
+            String content = "邀请返利:" + returnMoney.toPlainString();
+            DappAccountMoneyChangeEntity accountMoneyChange = new DappAccountMoneyChangeEntity(agent.getId(), walletMine.getAvailableAmount(), returnMoney, walletMine.getAvailableAmount().add(returnMoney), content, 4);
+            dappAccountMoneyChangeDao.insert(accountMoneyChange);
+
+            dappAgentReturnFlowDao.updateIsReturnByMemberId(1, agent.getId());
+        }
     }
 }

--
Gitblit v1.9.1