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