From 24f7f046f8854d63839ad52d0cff34ea45a0f449 Mon Sep 17 00:00:00 2001
From: gongzuming <gongzuming>
Date: 星期二, 12 十一月 2024 13:10:17 +0800
Subject: [PATCH] 优化

---
 src/main/java/com/mzl/flower/service/customer/CustomerService.java |   72 +++++++++++++++++++++++++++++++++--
 1 files changed, 67 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/mzl/flower/service/customer/CustomerService.java b/src/main/java/com/mzl/flower/service/customer/CustomerService.java
index 4fec567..1933dc4 100644
--- a/src/main/java/com/mzl/flower/service/customer/CustomerService.java
+++ b/src/main/java/com/mzl/flower/service/customer/CustomerService.java
@@ -10,27 +10,40 @@
 import com.mzl.flower.dto.request.customer.ChangePartnerDTO;
 import com.mzl.flower.dto.request.customer.QueryCustomerDTO;
 import com.mzl.flower.dto.request.customer.UpdateCustomerDTO;
+import com.mzl.flower.dto.request.menber.UserGrowthRecordDTO;
 import com.mzl.flower.dto.response.customer.CustomerDTO;
 import com.mzl.flower.dto.response.partner.PartnerDTO;
 import com.mzl.flower.entity.customer.Customer;
 import com.mzl.flower.entity.partner.Partner;
+import com.mzl.flower.enums.TrueOrFalseEnum;
 import com.mzl.flower.mapper.customer.CustomerMapper;
 import com.mzl.flower.mapper.partner.PartnerMapper;
+import com.mzl.flower.service.menber.MemberGrowthRecordService;
+import lombok.extern.slf4j.Slf4j;
+import com.mzl.flower.service.BaseService;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.BeanUtils;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
+import org.springframework.util.ObjectUtils;
 
+import javax.annotation.Resource;
 import java.util.List;
 
 @Service
 @Transactional
+@Slf4j
 public class CustomerService {
 
 
     private final CustomerMapper customerMapper;
 
     private final PartnerMapper partnerMapper;
+    @Resource
+    private MemberGrowthRecordService memberGrowthRecordService;
+
+    @Resource
+    private BaseService baseService;
 
     public CustomerService(CustomerMapper customerMapper, PartnerMapper partnerMapper) {
         this.customerMapper = customerMapper;
@@ -38,7 +51,12 @@
     }
 
     public CustomerDTO getCurrentCustomer() {
-        return customerMapper.getCurrentCustomer(SecurityUtils.getUserId());
+        CustomerDTO currentCustomer = customerMapper.getCurrentCustomer(SecurityUtils.getUserId());
+        UserGrowthRecordDTO userGrowthRecordDTO = memberGrowthRecordService.getInfoByUserId(SecurityUtils.getUserId());
+        if(!ObjectUtils.isEmpty(userGrowthRecordDTO)){
+            currentCustomer.setUserGrowthRecord(userGrowthRecordDTO);
+        }
+        return currentCustomer;
     }
 
     public PartnerDTO getCurrentBindPartner() {
@@ -65,6 +83,8 @@
             }
 
             customer.create(SecurityUtils.getUserId());
+            customer.setIsEnabled(true);
+            customer.setLevelId(Long.valueOf(Constants.DEFAULT_MEMBER_ID));
             customerMapper.insert(customer);
         } else {//重新修改
             customer = customerMapper.selectById(dto.getId());
@@ -167,9 +187,13 @@
         if (StringUtils.isNotBlank(c.getPartnerId()) || StringUtils.isNotBlank(c.getPartnerUserId())) {
             throw new ValidationException("商户已绑定合伙人，请联系客服人员进行解绑后再进行绑定");
         }
-//        Partner partner = partnerMapper.selectOne(new QueryWrapper<Partner>()
-//                .eq("user_id", dto.getPartnerUserId()));
-        Partner partner = partnerMapper.getByIdOrUserId(dto.getPartnerUserId()); //适配id和userId
+        Partner partner;
+        try {
+            long id = Long.parseLong(dto.getPartnerUserId());
+            partner = partnerMapper.selectById(id);
+        } catch (Exception e) {
+            partner =partnerMapper.selectOne(new QueryWrapper<Partner>().eq("user_id", dto.getPartnerUserId()));
+        }
         if (partner == null) {
             throw new ValidationException("合伙人不存在");
         }
@@ -182,7 +206,13 @@
 
     public String getPartnerName(String partnerUserId) {
         if (StringUtils.isNotBlank(partnerUserId)) {
-            Partner partner = partnerMapper.getByIdOrUserId(partnerUserId);
+            Partner partner;
+            try {
+                long id = Long.parseLong(partnerUserId);
+                partner = partnerMapper.selectById(id);
+            } catch (Exception e) {
+                partner =partnerMapper.selectOne(new QueryWrapper<Partner>().eq("user_id", partnerUserId));
+            }
             if (partner == null) {
                 throw new ValidationException("合伙人不存在");
             }
@@ -190,4 +220,36 @@
         }
         return null;
     }
+    public void isEnable(Long id) {
+        Customer customer = customerMapper.selectById(id);
+        if (customer == null) {
+            throw new ValidationException("商户信息不存在");
+        }
+        if (customer.getIsEnabled()) {
+            customer.setIsEnabled(false);
+            //强制下线
+            baseService.removeToken(customer.getUserId());
+        } else {
+            customer.setIsEnabled(true);
+        }
+        customer.update(SecurityUtils.getUserId());
+        customerMapper.updateById(customer);
+    }
+
+    /**
+     * 根据会员等级获取等级下的customer信息
+     * @param levelId
+     * @return
+     */
+    public List<Customer> getCustomerListByLevelId(Integer levelId){
+        if(null != levelId){
+            QueryWrapper<Customer> customerQueryWrapper=new QueryWrapper<>();
+            customerQueryWrapper.lambda()
+                    .eq(Customer::getDeleted, TrueOrFalseEnum.FALSE.isFlag())
+                    .eq(Customer::getLevelId,levelId);
+            return customerMapper.selectList(customerQueryWrapper);
+        }
+        return null;
+    }
+
 }

--
Gitblit v1.9.3