From 2c74d176be9d103bf47f69868ca7975a3a73256d Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 27 Sep 2021 12:34:44 +0800
Subject: [PATCH] Merge branch 'master' of http://120.27.238.55:7000/r/xc-mall

---
 src/main/java/cc/mrbird/febs/mall/service/impl/AgentServiceImpl.java |  122 ++++++++++++++++++++++++++++++++++++++++
 1 files changed, 122 insertions(+), 0 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/mall/service/impl/AgentServiceImpl.java b/src/main/java/cc/mrbird/febs/mall/service/impl/AgentServiceImpl.java
index cfc8201..47820ca 100644
--- a/src/main/java/cc/mrbird/febs/mall/service/impl/AgentServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/mall/service/impl/AgentServiceImpl.java
@@ -1,12 +1,19 @@
 package cc.mrbird.febs.mall.service.impl;
 
+import cc.mrbird.febs.common.enumerates.AgentLevelEnum;
+import cc.mrbird.febs.common.enumerates.MoneyFlowTypeEnum;
+import cc.mrbird.febs.common.utils.AppContants;
 import cc.mrbird.febs.mall.entity.AgentInfo;
 import cc.mrbird.febs.mall.entity.DataDictionaryCustom;
 import cc.mrbird.febs.mall.entity.MallMember;
+import cc.mrbird.febs.mall.entity.MallOrderInfo;
 import cc.mrbird.febs.mall.mapper.DataDictionaryCustomMapper;
 import cc.mrbird.febs.mall.mapper.MallMemberMapper;
+import cc.mrbird.febs.mall.mapper.MallMemberWalletMapper;
 import cc.mrbird.febs.mall.mapper.MallOrderInfoMapper;
 import cc.mrbird.febs.mall.service.IAgentService;
+import cc.mrbird.febs.mall.service.IApiMallMemberService;
+import cc.mrbird.febs.mall.service.IApiMallMemberWalletService;
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.util.StrUtil;
 import com.alibaba.fastjson.JSONObject;
@@ -14,8 +21,14 @@
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
+import java.math.BigDecimal;
+import java.math.RoundingMode;
+import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 /**
  * @author wzy
@@ -29,8 +42,11 @@
     private final DataDictionaryCustomMapper dataDictionaryCustomMapper;
     private final MallMemberMapper memberMapper;
     private final MallOrderInfoMapper orderInfoMapper;
+    private final IApiMallMemberWalletService memberWalletService;
+    private final IApiMallMemberService memberService;
 
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public void autoUpAgentLevel(Long memberId) {
         MallMember member = memberMapper.selectById(memberId);
         if(StrUtil.isBlank(member.getReferrerIds())) {
@@ -109,7 +125,113 @@
     }
 
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public void returnMoneyToAgent(Long orderId) {
+        MallOrderInfo orderInfo = orderInfoMapper.selectById(orderId);
+
+        MallMember member = memberMapper.selectById(orderInfo.getMemberId());
+        if (StrUtil.isBlank(member.getReferrerIds())) {
+            return;
+        }
+
+        // 直推奖励
+        directReturnMoney(member, orderInfo);
+        // 团队奖励
+        teamReturnMoney(member, orderInfo);
+    }
+
+    /**
+     * 直推奖励
+     * @param member
+     */
+    private void directReturnMoney(MallMember member, MallOrderInfo orderInfo) {
+        MallMember parent = memberMapper.selectInfoByInviteId(member.getReferrerId());
+        if (AgentLevelEnum.FIRST_LEVEL.name().equals(parent.getLevel())) {
+            return;
+        }
+
+        AgentInfo agentInfo = getAgentInfo(parent.getLevel());
+        if (agentInfo == null) {
+            return;
+        }
+
+        BigDecimal returnMoney = agentInfo.getDirectIncome().divide(BigDecimal.valueOf(100), 2,RoundingMode.DOWN).multiply(orderInfo.getAmount());
+        memberWalletService.addBalance(returnMoney, parent.getId());
+
+        memberService.addMoneyFlow(parent.getId(), returnMoney, MoneyFlowTypeEnum.BONUS.getValue(), orderInfo.getOrderNo(), null, null, orderInfo.getMemberId(), null);
+    }
+
+    /**
+     * 团队奖励
+     *
+     */
+    public void teamReturnMoney(MallMember member, MallOrderInfo orderInfo) {
+        List<String> inviteIds = StrUtil.split(member.getReferrerIds(), ',');
+        List<MallMember> agentList = memberMapper.selectMemberParentAgentList(inviteIds);
+        if (CollUtil.isEmpty(agentList)) {
+            return;
+        }
+
+        Map<String, Map<Long, BigDecimal>> needReturn = new HashMap<>();
+        for (MallMember mallMember : agentList) {
+            String level = mallMember.getLevel();
+
+            Map<Long, BigDecimal> amount = needReturn.get(level);
+
+            // 按照代理关系,返到第二级,平级奖
+            if (amount == null) {
+                amount = new HashMap<>();
+                AgentInfo agentInfo = getAgentInfo(level);
+                if (agentInfo == null) {
+                    continue;
+                }
+
+                BigDecimal returnMoney;
+                if (AgentInfo.TEAM_INCOME_TYPE_AMOUNT.equals(agentInfo.getTeamIncomeType())) {
+                    returnMoney = agentInfo.getTeamIncome();
+                } else {
+                    returnMoney = agentInfo.getTeamIncome().divide(BigDecimal.valueOf(100), 2, RoundingMode.DOWN).multiply(orderInfo.getAmount());
+                }
+                amount.put(mallMember.getId(), returnMoney);
+            } else {
+                if (!AgentLevelEnum.SECOND_LEVEL.name().equals(mallMember.getLevel())) {
+                    amount.put(mallMember.getId(), BigDecimal.ONE);
+                }
+            }
+
+            // 代理两级
+            if (amount.size() > 2) {
+                continue;
+            }
+
+            needReturn.put(level, amount);
+        }
+
+        if(needReturn.isEmpty()) {
+           return;
+        }
+
+        for (Map.Entry<String, Map<Long, BigDecimal>> entry : needReturn.entrySet()) {
+            for (Map.Entry<Long, BigDecimal> amount : entry.getValue().entrySet()) {
+                memberWalletService.addBalance(amount.getValue(), amount.getKey());
+
+                memberService.addMoneyFlow(amount.getKey(), amount.getValue(), MoneyFlowTypeEnum.ACHIEVE.getValue(), orderInfo.getOrderNo(), null, null, orderInfo.getMemberId(), null);
+            }
+        }
+    }
+
+    private AgentInfo getAgentInfo(String level) {
+        DataDictionaryCustom dic = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(AppContants.AGENT_LEVEL_REQUIRE, level);
+        if (dic == null) {
+            log.info("当前用户父级等级:{}", level);
+            return null;
+        }
+
+        return JSONObject.parseObject(dic.getValue(), AgentInfo.class);
+    }
+
+    @Override
+    public void rankReturnMoney(Long orderId) {
 
     }
 }

--
Gitblit v1.9.1