接口

gather-app/src/main/java/com/ruoyi/database/controller/TransitUserRecordController.java

@@ -10,9 +10,11 @@ import com.ruoyi.common.core.page.TableDataInfo;
 import com.ruoyi.common.enums.BusinessType;
 import com.ruoyi.common.utils.PageUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
+import com.ruoyi.database.domain.DevopsDeviceInfo;
 import com.ruoyi.database.domain.TransitImpUserRecord;
 import com.ruoyi.database.domain.TransitUserRecord;
 import com.ruoyi.database.domain.vo.TransitUserRecordVO;
+import com.ruoyi.database.service.DevopsDeviceInfoService;
 import com.ruoyi.database.service.TransitImpUserRecordService;
 import com.ruoyi.database.service.TransitUserRecordService;
 import io.swagger.annotations.Api;
@@ -30,6 +32,7 @@ import java.time.OffsetDateTime;
 import java.time.format.DateTimeFormatter;
 import java.util.Date;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * 人脸记录表(TransitUserRecord)Controller
@@ -45,16 +48,28 @@ public class TransitUserRecordController extends BaseController {
 
     private final TransitUserRecordService transitUserRecordService;
     private final TransitImpUserRecordService transitImpUserRecordService;
+    private final DevopsDeviceInfoService devopsDeviceInfoService;
 
     @GetMapping
     @ApiOperation("查询人脸记录表")
     public TableDataInfo<TransitUserRecord> list(TransitUserRecord transitUserRecord) throws Exception {
         startPage();
         DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
+        DateTimeFormatter formatter1 = DateTimeFormatter.ofPattern("yyyy-MM-dd");
+        List<DevopsDeviceInfo> deviceInfos = devopsDeviceInfoService.lambdaQuery().eq(DevopsDeviceInfo::getParentMonitoringType, 1).list();
+        List<String> collect = deviceInfos.stream().map(devopsDeviceInfo -> devopsDeviceInfo.getGbsChannelNo()).collect(Collectors.toList());
+        QueryWrapper<TransitUserRecord> queryWrapper = new QueryWrapper<>(transitUserRecord);
+        queryWrapper.in("device_code",collect);
+        if (transitUserRecord.getFirstTime() == null) {
             LocalDateTime now = LocalDateTime.now();
             LocalDateTime minusDay = now.minusDays(1);
-        QueryWrapper<TransitUserRecord> queryWrapper = new QueryWrapper<>(transitUserRecord);
             queryWrapper.between("partition_field", formatter.format(minusDay), formatter.format(now));
+        }else {
+            LocalDate date = LocalDate.parse(transitUserRecord.getFirstTime(), formatter1);
+            LocalDate nextDay = date.plusDays(1);
+            String nextDayString = nextDay.format(formatter1);
+            queryWrapper.between("partition_field", transitUserRecord.getFirstTime(), nextDayString);
+        }
         queryWrapper.orderByDesc("partition_field");
         List<TransitUserRecord> list = transitUserRecordService.list(queryWrapper);
         return getDataTable(list);

gather-app/src/main/java/com/ruoyi/database/controller/TransitVehicleRecordController.java

@@ -12,11 +12,9 @@ import com.ruoyi.common.enums.BusinessType;
 import com.ruoyi.common.utils.PageUtils;
 import com.ruoyi.common.utils.bean.BeanUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
-import com.ruoyi.database.domain.BoneVehicle;
+import com.ruoyi.database.domain.*;
-import com.ruoyi.database.domain.TransitImpVehicleRecord;
-import com.ruoyi.database.domain.TransitUserRecord;
-import com.ruoyi.database.domain.TransitVehicleRecord;
 import com.ruoyi.database.service.BoneVehicleService;
+import com.ruoyi.database.service.DevopsDeviceInfoService;
 import com.ruoyi.database.service.TransitImpVehicleRecordService;
 import com.ruoyi.database.service.TransitVehicleRecordService;
 import io.swagger.annotations.Api;
@@ -26,11 +24,13 @@ import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
 import javax.servlet.http.HttpServletResponse;
+import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.time.format.DateTimeFormatter;
 import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * 车辆记录表(TransitVehicleRecord)Controller
@@ -46,16 +46,28 @@ public class TransitVehicleRecordController extends BaseController {
 
     private final TransitVehicleRecordService transitVehicleRecordService;
     private final TransitImpVehicleRecordService transitImpVehicleRecordService;
+    private final DevopsDeviceInfoService devopsDeviceInfoService;
 
     @GetMapping
     @ApiOperation("查询车辆记录表")
     public TableDataInfo<TransitVehicleRecord> list(TransitVehicleRecord transitVehicleRecord) {
         startPage();
         DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
+        DateTimeFormatter formatter1 = DateTimeFormatter.ofPattern("yyyy-MM-dd");
+        List<DevopsDeviceInfo> deviceInfos = devopsDeviceInfoService.lambdaQuery().eq(DevopsDeviceInfo::getParentMonitoringType, 1).list();
+        List<String> collect = deviceInfos.stream().map(devopsDeviceInfo -> devopsDeviceInfo.getGbsChannelNo()).collect(Collectors.toList());
+        QueryWrapper<TransitVehicleRecord> queryWrapper = new QueryWrapper<>(transitVehicleRecord);
+        queryWrapper.in("doorway_code",collect);
+        if (transitVehicleRecord.getFirstTime() == null) {
             LocalDateTime now = LocalDateTime.now();
             LocalDateTime minusDay = now.minusDays(1);
-        QueryWrapper<TransitVehicleRecord> queryWrapper = new QueryWrapper<>(transitVehicleRecord);
             queryWrapper.between("partition_field", formatter.format(minusDay), formatter.format(now));
+        }else {
+            LocalDate date = LocalDate.parse(transitVehicleRecord.getFirstTime(), formatter1);
+            LocalDate nextDay = date.plusDays(1);
+            String nextDayString = nextDay.format(formatter1);
+            queryWrapper.between("partition_field", transitVehicleRecord.getFirstTime(), nextDayString);
+        }
         queryWrapper.orderByDesc("partition_field");
         List<TransitVehicleRecord> list = transitVehicleRecordService.list(queryWrapper);
         return getDataTable(list);