From 077c4e365617db3793a212888c9a8bb6f66a19cc Mon Sep 17 00:00:00 2001
From: KKSU <15274802129@163.com>
Date: Tue, 31 Dec 2024 13:31:20 +0800
Subject: [PATCH] refactor(mall): 优化会员积分计算逻辑
---
src/test/java/cc/mrbird/febs/AgentTest.java | 14 ++++---
src/main/java/cc/mrbird/febs/mall/service/impl/MemberProfitServiceImpl.java | 56 ++++++++++++++++-----------
2 files changed, 41 insertions(+), 29 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/mall/service/impl/MemberProfitServiceImpl.java b/src/main/java/cc/mrbird/febs/mall/service/impl/MemberProfitServiceImpl.java
index 2c79b71..5e4d4e8 100644
--- a/src/main/java/cc/mrbird/febs/mall/service/impl/MemberProfitServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/mall/service/impl/MemberProfitServiceImpl.java
@@ -654,24 +654,33 @@
public void updateMemberCoin() {
try {
// 操作时间
- DateTime operationDate = DateUtil.date();
+ DateTime operationDate = DateUtil.parseDate(DateUtil.today());
+
+ // 获取会员等级
+ List<RunVip> runVips = runVipMapper.selectList(null);
+ if (CollUtil.isEmpty(runVips)) {
+ log.error("未找到任何会员等级");
+ return;
+ }
+ Map<String, RunVip> runVipMap = runVips.stream()
+ .collect(Collectors.toMap(RunVip::getVipCode, runVip -> runVip));
// 获取最低级别的会员等级
- RunVip minRunVip = runVipMapper.selectOne(new LambdaQueryWrapper<RunVip>()
- .orderByAsc(RunVip::getOrderNumber)
- .last("LIMIT 1")
- );
+ Optional<RunVip> optionalRunVip = runVips.stream().min(Comparator.comparingInt(RunVip::getOrderNumber));
+ RunVip minRunVip = optionalRunVip.orElse(null);
if (minRunVip == null) {
log.error("最低级别的会员等级未找到");
return;
}
- // 获取全部referrerId不为空的用户，并获取每个用户的所有上级id，返回一个set集合
+ // 获取全部等级不是游客的用户，返回一个set集合
List<MallMember> mallMembers = mallMemberMapper.selectList(
new LambdaQueryWrapper<MallMember>()
- .isNotNull(MallMember::getReferrerId)
+ .select(MallMember::getId, MallMember::getLevel, MallMember::getInviteId, MallMember::getReferrerId)
+ .ne(MallMember::getLevel,minRunVip.getVipCode())
);
if (CollUtil.isEmpty(mallMembers)) {
+ log.info("没有符合条件的会员");
return;
}
@@ -679,13 +688,18 @@
.map(MallMember::getId)
.collect(Collectors.toSet());
if (CollUtil.isEmpty(memberIds)) {
+ log.info("没有有效的会员ID");
return;
}
// 构建成员ID到MallMember对象的映射
- Map<Long, MallMember> memberMap = mallMemberMapper.selectBatchIds(memberIds).stream()
+ Map<Long, MallMember> memberMap = mallMembers.stream()
.collect(Collectors.toMap(MallMember::getId, member -> member));
+ // 缓存直推成员
+ Map<String, Set<MallMember>> directMembersCache = mallMembers.stream()
+ .collect(Collectors.groupingBy(MallMember::getReferrerId, Collectors.toSet()));
+
for (Long memberId : memberIds) {
MallMember mallMember = memberMap.get(memberId);
@@ -694,17 +708,18 @@
}
// 获取所有购买了会员等级的直推成员
- List<MallMember> directMembers = mallMemberMapper.selectList(
- new LambdaQueryWrapper<MallMember>()
- .eq(MallMember::getReferrerId, mallMember.getInviteId())
- .ne(MallMember::getLevel, minRunVip.getVipCode())
- );
+ Set<MallMember> directMembers = directMembersCache.get(mallMember.getInviteId());
if (CollUtil.isEmpty(directMembers)) {
continue;
}
+ RunVip memberRunVip = runVipMap.get(mallMember.getLevel());
+ if (memberRunVip == null) {
+ log.warn("会员等级 {} 不存在", mallMember.getLevel());
+ continue;
+ }
for (MallMember item : directMembers) {
- BigDecimal realScore = calculateDirectScore(mallMember, item, operationDate);
+ BigDecimal realScore = calculateDirectScore(memberRunVip, item, operationDate, runVipMap);
if(BigDecimal.ZERO.compareTo(realScore) >= 0){
continue;
}
@@ -729,30 +744,25 @@
}
- private BigDecimal calculateDirectScore(MallMember mallMember, MallMember directMember, Date operationDate) {
+ private BigDecimal calculateDirectScore(RunVip memberRunVip, MallMember directMember, Date operationDate, Map<String, RunVip> runVipMap) {
// 实际助力
BigDecimal realScore = BigDecimal.ZERO;
try {
- String directLevel = directMember.getLevel();
- RunVip directRunVip = runVipMapper.selectOne(new LambdaQueryWrapper<RunVip>().eq(RunVip::getVipCode, directLevel));
+ RunVip directRunVip = runVipMap.get(directMember.getLevel());
if (directRunVip == null) {
return realScore;
}
// 获取每一个会员的前一日碳积分总和
LambdaQueryWrapper<MallMoneyFlow> mallMoneyFlowLambdaQueryWrapper = new LambdaQueryWrapper<>();
mallMoneyFlowLambdaQueryWrapper.ge(MallMoneyFlow::getCreatedTime, DateUtil.offsetDay(operationDate, -1));
+ mallMoneyFlowLambdaQueryWrapper.lt(MallMoneyFlow::getCreatedTime, operationDate);
mallMoneyFlowLambdaQueryWrapper.eq(MallMoneyFlow::getFlowType, FlowTypeEnum.SCORE.getValue());
mallMoneyFlowLambdaQueryWrapper.eq(MallMoneyFlow::getType, RunVipMoneyFlowTypeEnum.GET_SCORE.getValue());
+ mallMoneyFlowLambdaQueryWrapper.eq(MallMoneyFlow::getMemberId, directMember.getId());
List<MallMoneyFlow> mallMoneyFlows = mallMoneyFlowMapper.selectList(mallMoneyFlowLambdaQueryWrapper);
if (CollUtil.isEmpty(mallMoneyFlows)) {
- return realScore;
- }
-
- String memberLevel = mallMember.getLevel();
- RunVip memberRunVip = runVipMapper.selectOne(new LambdaQueryWrapper<RunVip>().eq(RunVip::getVipCode, memberLevel));
- if (memberRunVip == null) {
return realScore;
}
diff --git a/src/test/java/cc/mrbird/febs/AgentTest.java b/src/test/java/cc/mrbird/febs/AgentTest.java
index 4358b15..5fba906 100644
--- a/src/test/java/cc/mrbird/febs/AgentTest.java
+++ b/src/test/java/cc/mrbird/febs/AgentTest.java
@@ -2,6 +2,7 @@
import cc.mrbird.febs.mall.service.IMemberProfitService;
import lombok.extern.slf4j.Slf4j;
+import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
@@ -11,10 +12,11 @@
@Autowired
private IMemberProfitService memberProfitService;
-// @Test
-// public void getCouponAmountMap(){
-//
-// memberProfitService.updateRunScore();
-//
-// }
+ @Test
+ public void getCouponAmountMap(){
+
+
+ memberProfitService.updateMemberCoin();
+
+ }
}
--
Gitblit v1.9.1