From b9da8289051937fbbbd47bbda97414ad6c769a90 Mon Sep 17 00:00:00 2001
From: hjl <1657978663@qq.com>
Date: 星期四, 18 七月 2024 14:37:00 +0800
Subject: [PATCH] feat: 代码重构

---
 ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/service/impl/UserServiceImpl.java |   50 +++++++++++++++++++++-----------------------------
 1 files changed, 21 insertions(+), 29 deletions(-)

diff --git a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/service/impl/UserServiceImpl.java b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/service/impl/UserServiceImpl.java
index 074d63e..320b592 100644
--- a/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/service/impl/UserServiceImpl.java
+++ b/ruoyi-service/ruoyi-admin/src/main/java/com/ruoyi/admin/service/impl/UserServiceImpl.java
@@ -1,15 +1,17 @@
 package com.ruoyi.admin.service.impl;
 
+import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ruoyi.admin.entity.User;
 import com.ruoyi.admin.mapper.UserMapper;
 import com.ruoyi.admin.request.UserDataCountRequest;
-import com.ruoyi.admin.service.OrderService;
 import com.ruoyi.admin.service.UserService;
-import com.ruoyi.admin.service.WithdrawService;
 import com.ruoyi.admin.vo.UserDataCountVO;
 import com.ruoyi.admin.vo.UserTrendsVO;
 import com.ruoyi.common.core.constant.OrderConstants;
+import com.ruoyi.order.api.entity.MoneyQueryRequest;
+import com.ruoyi.order.api.feignClient.OrderClient;
+import com.ruoyi.order.api.feignClient.WithdrawClient;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
@@ -33,22 +35,20 @@
 public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService {
 
     @Resource
-    private OrderService orderService;
+    private OrderClient orderClient;
     @Resource
-    private WithdrawService withdrawService;
+    private WithdrawClient withdrawClient;
 
     @Override
     public List<UserTrendsVO> userTrends(List<String> city) {
-        if (null == city || city.isEmpty()) {
-            return new ArrayList<>();
-        }
         List<UserTrendsVO> userTrendsList = baseMapper.userTrends(city);
         return null == userTrendsList || userTrendsList.isEmpty() ? new ArrayList<>() : userTrendsList;
     }
 
     @Override
-    public Object userDataCount(UserDataCountRequest userDataCountRequest) {
+    public UserDataCountVO userDataCount(UserDataCountRequest userDataCountRequest) {
         List<String> cityList = userDataCountRequest.getCityList();
+        String type = userDataCountRequest.getCountType();
         List<User> userList = lambdaQuery().eq(User::getIsDelete, 0).list();
         // 用户总量
         Integer userNumber = null == userList ? 0 : userList.size();
@@ -63,35 +63,27 @@
         String endDateStr = String.valueOf(endDate);
         // 结果数据封装
         Long increaseNumber;
-        BigDecimal totalMoney;
-        BigDecimal withdrawalTotalMoney;
-        if (OrderConstants.QUARTER.equals(userDataCountRequest.getCountType())) {
+        // 交易额
+        BigDecimal totalMoney = orderClient.tradeMoney(cityList, userDataCountRequest.getCountType()).getData();
+        BigDecimal withdrawalTotalMoney = withdrawClient.withdrawalTotalMoney(new MoneyQueryRequest(cityList, type))
+                .getData();
+        if (OrderConstants.QUARTER.equals(type)) {
             // 用户增长数
-            increaseNumber = lambdaQuery().ge(User::getCreateTime, startDateStr).le(User::getCreateTime, endDateStr)
-                    .in(User::getCity, cityList).eq(User::getIsDelete, 0).count();
-            // 用户总交易额
-            totalMoney = orderService.totalMoneyByQuarter(startDateStr, endDateStr, cityList);
-            // 用户提现总额
-            withdrawalTotalMoney = withdrawService.withdrawalTotalMoney(cityList);
-        } else if (OrderConstants.YEAR.equals(userDataCountRequest.getCountType())) {
+            LambdaQueryChainWrapper<User> wrapper = lambdaQuery().ge(User::getCreateTime, startDateStr)
+                    .le(User::getCreateTime, endDateStr);
+            if (!cityList.isEmpty()) {
+                wrapper.in(User::getCity, cityList);
+            }
+            increaseNumber = wrapper.eq(User::getIsDelete, 0).count();
+        } else if (OrderConstants.YEAR.equals(type)) {
             // 用户增长数
             increaseNumber = baseMapper.increaseNumberByYear(cityList);
-            // 用户总交易额
-            totalMoney = orderService.totalMoneyByYear(cityList);
-            // 用户提现总额
-            withdrawalTotalMoney = withdrawService.withdrawalTotalMoneyByYear(cityList);
-        } else if (OrderConstants.MONTH.equals(userDataCountRequest.getCountType())) {
+        } else if (OrderConstants.MONTH.equals(type)) {
             // 用户增长数
             increaseNumber = baseMapper.increaseNumberByMonth(cityList);
-            // 用户总交易额
-            totalMoney = orderService.totalMoneyByMonth(cityList);
-            // 用户提现总额
-            withdrawalTotalMoney = withdrawService.withdrawalTotalMoneyByMonth(cityList);
         } else {
             // 数量初始化
             increaseNumber = 0L;
-            totalMoney = BigDecimal.ZERO;
-            withdrawalTotalMoney = BigDecimal.ZERO;
         }
         // 查询结果判断
         increaseNumber = null == increaseNumber ? 0L : increaseNumber;

--
Gitblit v1.7.1