会员统计当日数据问题处理

framework/src/main/java/cn/lili/modules/statistics/serviceimpl/MemberStatisticsDataServiceImpl.java

@@ -4,14 +4,15 @@ import cn.hutool.core.date.DateUtil;
 import cn.lili.modules.statistics.mapper.MemberStatisticsDataMapper;
 import cn.lili.modules.statistics.model.dos.MemberStatisticsData;
 import cn.lili.modules.statistics.model.dto.StatisticsQueryParam;
+import cn.lili.modules.statistics.model.enums.SearchTypeEnum;
 import cn.lili.modules.statistics.service.MemberStatisticsDataService;
 import cn.lili.modules.statistics.util.StatisticsDateUtil;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
 
@@ -24,31 +25,26 @@ import java.util.List;
 @Service
 public class MemberStatisticsDataServiceImpl extends ServiceImpl<MemberStatisticsDataMapper, MemberStatisticsData> implements MemberStatisticsDataService {
 
-    /**
-     * 会员统计
-     */
-    @Autowired
-    private MemberStatisticsDataMapper memberStatisticsDataMapper;
 
     @Override
     public Integer getMemberCount() {
         QueryWrapper queryWrapper = new QueryWrapper();
         queryWrapper.eq("disabled", true);
-        return memberStatisticsDataMapper.customSqlQuery(queryWrapper);
+        return this.baseMapper.customSqlQuery(queryWrapper);
     }
 
     @Override
     public Integer todayMemberNum() {
         QueryWrapper queryWrapper = Wrappers.query();
         queryWrapper.gt("create_time", DateUtil.beginOfDay(new Date()));
-        return memberStatisticsDataMapper.customSqlQuery(queryWrapper);
+        return this.baseMapper.customSqlQuery(queryWrapper);
     }
 
     @Override
     public Integer memberCount(Date endTime) {
         QueryWrapper queryWrapper = Wrappers.query();
         queryWrapper.lt("create_time", endTime);
-        return memberStatisticsDataMapper.customSqlQuery(queryWrapper);
+        return this.baseMapper.customSqlQuery(queryWrapper);
     }
 
     @Override
@@ -56,20 +52,35 @@ public class MemberStatisticsDataServiceImpl extends ServiceImpl<MemberStatistic
 
         QueryWrapper queryWrapper = Wrappers.query();
         queryWrapper.ge("last_login_date", startTime);
-        return memberStatisticsDataMapper.customSqlQuery(queryWrapper);
+        return this.baseMapper.customSqlQuery(queryWrapper);
     }
 
     @Override
     public Integer newlyAdded(Date startTime, Date endTime) {
         QueryWrapper queryWrapper = Wrappers.query();
         queryWrapper.between("create_time", startTime, endTime);
-        return memberStatisticsDataMapper.customSqlQuery(queryWrapper);
+        return this.baseMapper.customSqlQuery(queryWrapper);
     }
 
     @Override
     public List<MemberStatisticsData> statistics(StatisticsQueryParam statisticsQueryParam) {
+
         Date[] dates = StatisticsDateUtil.getDateArray(statisticsQueryParam);
         Date startTime = dates[0], endTime = dates[1];
+
+        //如果统计今天，则自行构造数据
+        if(statisticsQueryParam.getSearchType().equals(SearchTypeEnum.TODAY.name())){
+            //构建数据，然后返回集合，提供给前端展示
+            MemberStatisticsData memberStatisticsData = new MemberStatisticsData();
+            memberStatisticsData.setMemberCount(this.memberCount(endTime));
+            memberStatisticsData.setCreateDate(startTime);
+            memberStatisticsData.setActiveQuantity(this.activeQuantity(startTime));
+            memberStatisticsData.setNewlyAdded(this.newlyAdded(startTime, endTime));
+            List result = new ArrayList<MemberStatisticsData>();
+            result.add(memberStatisticsData);
+            return result;
+        }
+
         QueryWrapper queryWrapper = Wrappers.query();
         queryWrapper.between("create_date", startTime, endTime);