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.UserService;
|
import com.ruoyi.admin.vo.UserDataCountVO;
|
import com.ruoyi.admin.vo.UserTrendsVO;
|
import com.ruoyi.common.core.constant.OrderConstants;
|
import com.ruoyi.common.core.domain.R;
|
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;
|
import java.math.BigDecimal;
|
import java.time.LocalDate;
|
import java.time.Month;
|
import java.time.Year;
|
import java.time.temporal.TemporalAdjusters;
|
import java.util.ArrayList;
|
import java.util.Iterator;
|
import java.util.List;
|
|
/**
|
* <p>
|
* 用户列表 服务实现类
|
* </p>
|
*
|
* @author hjl
|
* @since 2024-05-29
|
*/
|
@Service
|
public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService {
|
|
@Resource
|
private OrderClient orderClient;
|
@Resource
|
private WithdrawClient withdrawClient;
|
|
@Override
|
public List<UserTrendsVO> userTrends(List<String> city) {
|
List<UserTrendsVO> userTrendsList = baseMapper.userTrends(city);
|
return null == userTrendsList || userTrendsList.isEmpty() ? new ArrayList<>() : userTrendsList;
|
}
|
|
@Override
|
public UserDataCountVO userDataCount(UserDataCountRequest userDataCountRequest) {
|
List<String> cityList = userDataCountRequest.getCityList();
|
String type = userDataCountRequest.getCountType();
|
LambdaQueryChainWrapper<User> userWrapper = lambdaQuery().eq(User::getIsDelete, 0);
|
if (null != cityList && !cityList.isEmpty()) {
|
userWrapper.in(User::getCityCode, cityList);
|
}
|
List<User> userList = userWrapper.list();
|
|
// 判断用户是否在管辖城市内下过订单
|
Iterator<User> iterator = userList.iterator();
|
while (iterator.hasNext()) {
|
User user = iterator.next();
|
Integer userId = user.getId();
|
R<Boolean> r = orderClient.orderByUserId(userId);
|
if (!r.getData()) {
|
iterator.remove(); // 安全地移除元素
|
}
|
}
|
|
Integer userNumber = userList.size();
|
// 用户总量
|
// 计算当前季度时间所包含时间
|
int currentYear = Year.now().getValue();
|
// 获取当前月份
|
Month currentMonth = LocalDate.now().getMonth();
|
// 计算当前季度的开始时间和结束时间
|
LocalDate startDate = LocalDate.of(currentYear, getStartMonthOfQuarter(currentMonth), 1);
|
LocalDate endDate = startDate.plusMonths(2).with(TemporalAdjusters.lastDayOfMonth());
|
String startDateStr = String.valueOf(startDate);
|
String endDateStr = String.valueOf(endDate);
|
// 结果数据封装
|
Long increaseNumber;
|
// 交易额
|
BigDecimal totalMoney = orderClient.tradeMoney(cityList, userDataCountRequest.getCountType()).getData();
|
BigDecimal withdrawalTotalMoney = withdrawClient.withdrawalTotalMoney(new MoneyQueryRequest(cityList, type))
|
.getData();
|
if (OrderConstants.QUARTER.equals(type)) {
|
// 用户增长数
|
LambdaQueryChainWrapper<User> wrapper = lambdaQuery().ge(User::getCreateTime, startDateStr)
|
.le(User::getCreateTime, endDateStr);
|
if (null != cityList && !cityList.isEmpty()) {
|
wrapper.in(User::getCityCode, cityList);
|
}
|
increaseNumber = wrapper.eq(User::getIsDelete, 0).count();
|
} else if (OrderConstants.YEAR.equals(type)) {
|
// 用户增长数
|
increaseNumber = baseMapper.increaseNumberByYear(cityList);
|
} else if (OrderConstants.MONTH.equals(type)) {
|
// 用户增长数
|
increaseNumber = baseMapper.increaseNumberByMonth(cityList);
|
} else {
|
// 数量初始化
|
increaseNumber = 0L;
|
}
|
// 查询结果判断
|
increaseNumber = null == increaseNumber ? 0L : increaseNumber;
|
totalMoney = null == totalMoney ? BigDecimal.ZERO : totalMoney;
|
withdrawalTotalMoney = null == withdrawalTotalMoney ? BigDecimal.ZERO : withdrawalTotalMoney;
|
return new UserDataCountVO(userNumber, increaseNumber, totalMoney, withdrawalTotalMoney);
|
}
|
|
/**
|
* 根据当前月份获取当前季度的开始月份
|
*/
|
private static Month getStartMonthOfQuarter(Month currentMonth) {
|
if (currentMonth.compareTo(Month.JANUARY) >= 0 && currentMonth.compareTo(Month.MARCH) <= 0) {
|
return Month.JANUARY;
|
} else if (currentMonth.compareTo(Month.APRIL) >= 0 && currentMonth.compareTo(Month.JUNE) <= 0) {
|
return Month.APRIL;
|
} else if (currentMonth.compareTo(Month.JULY) >= 0 && currentMonth.compareTo(Month.SEPTEMBER) <= 0) {
|
return Month.JULY;
|
} else {
|
return Month.OCTOBER;
|
}
|
}
|
|
}
|
