From 52358cd76aee8d5f7edc54e177b6eab0d1f58533 Mon Sep 17 00:00:00 2001 From: zhujie <leon.zhu@cloudroam.com.cn> Date: 星期五, 11 四月 2025 17:31:06 +0800 Subject: [PATCH] 邀请功能 --- src/main/java/com/mzl/flower/service/customer/CustomerCenterService.java | 21 ++++++++++++++++++++- 1 files changed, 20 insertions(+), 1 deletions(-) diff --git a/src/main/java/com/mzl/flower/service/customer/CustomerCenterService.java b/src/main/java/com/mzl/flower/service/customer/CustomerCenterService.java index e161096..d53a2f0 100644 --- a/src/main/java/com/mzl/flower/service/customer/CustomerCenterService.java +++ b/src/main/java/com/mzl/flower/service/customer/CustomerCenterService.java @@ -1,16 +1,23 @@ package com.mzl.flower.service.customer; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.mzl.flower.config.security.SecurityUtils; import com.mzl.flower.dto.response.customer.CustomerTjDTO; +import com.mzl.flower.dto.response.customer.InviteDTO; +import com.mzl.flower.dto.response.customer.InviteRewardDTO; import com.mzl.flower.entity.customer.BrowseHistory; import com.mzl.flower.entity.customer.Collect; import com.mzl.flower.entity.customer.Follow; import com.mzl.flower.mapper.customer.BrowseHistoryMapper; import com.mzl.flower.mapper.customer.CollectMapper; +import com.mzl.flower.mapper.customer.CustomerMapper; import com.mzl.flower.mapper.customer.FollowMapper; import lombok.extern.slf4j.Slf4j; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; + +import java.util.List; +import java.util.stream.Collectors; @Service @Transactional @@ -20,11 +27,13 @@ private final FollowMapper followMapper; private final CollectMapper collectMapper; private final BrowseHistoryMapper browseHistoryMapper; + private final CustomerMapper customerMapper; - public CustomerCenterService(FollowMapper followMapper, CollectMapper collectMapper, BrowseHistoryMapper browseHistoryMapper) { + public CustomerCenterService(FollowMapper followMapper, CollectMapper collectMapper, BrowseHistoryMapper browseHistoryMapper, CustomerMapper customerMapper) { this.followMapper = followMapper; this.collectMapper = collectMapper; this.browseHistoryMapper = browseHistoryMapper; + this.customerMapper = customerMapper; } @@ -35,4 +44,14 @@ dto.setBrowseNum(browseHistoryMapper.selectCount(new LambdaQueryWrapper<BrowseHistory>().eq(BrowseHistory::getUserId, userId))); return dto; } + + //获取所有的邀请记录和当前登陆人的邀请记录 + public InviteRewardDTO customerReward() { + InviteRewardDTO rewardto = new InviteRewardDTO(); + List<InviteDTO> reward = customerMapper.getReward(); + rewardto.setSuccessInvite(reward); + List<InviteDTO> myreward = reward.stream().filter(s->s!=null&&s.getUserId()== SecurityUtils.getUserId()).collect(Collectors.toList());; + rewardto.setMyInvite(myreward); + return rewardto; + } } -- Gitblit v1.9.3