抽取项目数据到screen_project_*

epmet-module/data-statistical/data-statistical-client/src/main/java/com/epmet/dto/screen/form/ScreenCentralZoneDataFormDTO.java

@@ -1,9 +1,6 @@
 package com.epmet.dto.screen.form;
 
-import lombok.AllArgsConstructor;
 import lombok.Data;
-import lombok.NoArgsConstructor;
-import lombok.RequiredArgsConstructor;
 
 import java.io.Serializable;
 
@@ -26,4 +23,8 @@ public class ScreenCentralZoneDataFormDTO implements Serializable {
      * 时间维度 不一定是dateId 需要根据其格式进行判断维度的类型
      * */
     private String dateId;
+
+    private String startDate;
+
+    private String endDate;
 }

epmet-module/data-statistical/data-statistical-server/src/main/java/com/epmet/controller/DemoController.java

@@ -881,10 +881,17 @@ public class DemoController {
      */
     @PostMapping("project-test")
     public Result project(@RequestBody ScreenCentralZoneDataFormDTO param) {
-        if(StringUtils.isBlank(param.getDateId())){
-            //默认截止到昨天。
-            param.setDateId(DateUtils.getBeforeNDay(1));
+        List<String> dateIds = new ArrayList<>();
+        if ((StringUtils.isBlank(param.getStartDate()) && StringUtils.isBlank(param.getEndDate()))){
+            if (StringUtils.isNotBlank(param.getDateId())){
+                dateIds.add(param.getDateId());
+            }else {
+                dateIds.add(DimIdGenerator.getDateDimId(DateUtils.addDateDays(new Date(), -1)));
+            }
+        }else {
+            dateIds = DateUtils.getDaysBetween(param.getStartDate(), param.getEndDate());
         }
+
         List<String> customerIds = new ArrayList();
         if (StringUtils.isNotBlank(param.getCustomerId())) {
             customerIds.add(param.getCustomerId());
@@ -892,10 +899,12 @@ public class DemoController {
             customerIds = dimCustomerService.selectCustomerIdPage(1, 100);
         }
         for (String customerId : customerIds) {
-            ScreenCentralZoneDataFormDTO formDTO = new ScreenCentralZoneDataFormDTO();
-            formDTO.setCustomerId(customerId);
-            formDTO.setDateId(param.getDateId());
-            screenProjectSettleService.extractScreenData(param);
+            dateIds.forEach(dateId -> {
+                ScreenCentralZoneDataFormDTO formDTO = new ScreenCentralZoneDataFormDTO();
+                formDTO.setCustomerId(customerId);
+                formDTO.setDateId(dateId);
+                screenProjectSettleService.extractScreenData(param);
+            });
         }
         return new Result();
     }