Bläddra i källkod

Merge branch 'lyh-sn-export' into test

lyhzzz 2 år sedan
förälder
incheckning
51e8e67c82

+ 18 - 3
src/main/java/com/fdkankan/ucenter/controller/CameraController.java

@@ -185,10 +185,12 @@ public class CameraController extends BaseController {
         List<CameraExcelVo> responseCameraList = cameraDetailService.getListByUserAndTypeEx(user.getId());
         List<Long> cameraIdList = responseCameraList.stream().map(CameraExcelVo::getCameraId).collect(Collectors.toList());
 
+        List<ScenePro> sceneEntityList = new ArrayList<>();
         LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
-        wrapper.in(ScenePro::getCameraId,cameraIdList);
-        wrapper.orderByDesc(ScenePro::getCreateTime);
-        List<ScenePro> sceneEntityList = sceneProService.list(wrapper);
+        if(cameraIdList.size() >0){
+            wrapper.in(ScenePro::getCameraId,cameraIdList);
+            sceneEntityList = sceneProService.list(wrapper);
+        }
 
         HashMap<Long,List<ScenePro>> sceneProMap = new HashMap<>();
         for (ScenePro scenePro : sceneEntityList) {
@@ -282,6 +284,19 @@ public class CameraController extends BaseController {
 
         }
 
+        if(sceneEntityList.size() >0){
+            if(StringUtils.isNotBlank(param.getStartTime())){
+                wrapper.ge(ScenePro::getCreateTime,param.getStartTime());
+            }
+            if(StringUtils.isNotBlank(param.getEndTime())){
+                String date = DateUserUtil.getLastZeroTime(param.getEndTime());
+                wrapper.le(ScenePro::getCreateTime,date);
+            }
+            if(StringUtils.isNotBlank(param.getStartTime()) || StringUtils.isNotBlank(param.getEndTime())){
+                sceneEntityList = sceneProService.list(wrapper);
+            }
+        }
+
         HashMap<Long, Camera> cameraMap = cameraService.getByIds(cameraIdList);
         HashMap<Long, Integer> cameraSnCodeCountMap = new HashMap<>();
         for (ScenePro sceneProEntity : sceneEntityList) {

+ 20 - 0
src/main/java/com/fdkankan/ucenter/util/DateUserUtil.java

@@ -2,7 +2,9 @@ package com.fdkankan.ucenter.util;
 
 import cn.hutool.core.date.DateUnit;
 import cn.hutool.core.date.DateUtil;
+import com.fdkankan.ucenter.common.MailUtil;
 import com.fdkankan.ucenter.entity.IncrementType;
+import com.fdkankan.ucenter.entity.ScenePro;
 import org.apache.commons.lang3.StringUtils;
 import org.joda.time.DateTime;
 
@@ -33,6 +35,9 @@ public class DateUserUtil {
     public static String getDate(Date time){
         return new SimpleDateFormat(DEF_FMT).format(time);
     }
+    public static String getDate(Date time,String fmt){
+        return new SimpleDateFormat(fmt).format(time);
+    }
     public static String getZeroDate(){
         return new SimpleDateFormat(DAY_FMT).format(new Date()) +" 00:00:00";
     }
@@ -117,4 +122,19 @@ public class DateUserUtil {
         }
         return date;
     }
+
+    public static String getLastZeroTime(String endTime) {
+        try {
+            Date date = DateUserUtil.getDate(endTime);
+            String date1 = DateUserUtil.getDate(date, DateUserUtil.DAY_FMT);
+            return date1 + " 23:59:59";
+        }catch ( Exception e){
+            e.printStackTrace();
+        }
+        return endTime;
+    }
+
+    public static void main(String[] args) {
+        System.out.println(getLastZeroTime("2023-03-01 00:00:00"));
+    }
 }