From eab93d7bef852c1b1f6356c7cde3dc45c2ed9327 Mon Sep 17 00:00:00 2001
From: zss <zss@example.com>
Date: 星期五, 26 七月 2024 17:12:51 +0800
Subject: [PATCH] 班次排班优化

---
 cnas-server/src/main/java/com/yuanchu/mom/controller/DeviceController.java |   24 ++++++++++++++++++------
 1 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/cnas-server/src/main/java/com/yuanchu/mom/controller/DeviceController.java b/cnas-server/src/main/java/com/yuanchu/mom/controller/DeviceController.java
index 6a73257..024130c 100644
--- a/cnas-server/src/main/java/com/yuanchu/mom/controller/DeviceController.java
+++ b/cnas-server/src/main/java/com/yuanchu/mom/controller/DeviceController.java
@@ -12,6 +12,7 @@
 import com.yuanchu.mom.pojo.Device;
 import com.yuanchu.mom.service.DataConfigService;
 import com.yuanchu.mom.service.DeviceService;
+import com.yuanchu.mom.utils.DataAcquisition;
 import com.yuanchu.mom.utils.JackSonUtil;
 import com.yuanchu.mom.vo.Result;
 import io.swagger.annotations.Api;
@@ -176,11 +177,13 @@
     @ValueAuth
     @ApiOperation(value = "鍒ゆ柇璇ヨ澶囨槸鍚﹀彲浠ユ暟閲�")
     @GetMapping("/determineWhetherToCollectData")
-    public Result<?> determineWhetherToCollectData(@RequestParam("managementNumber") String managementNumber) {
+    public Result<?> determineWhetherToCollectData(@RequestParam("managementNumber") String managementNumber, HttpServletRequest request) {
+        String ip = DataAcquisition.getIp(request);
         Device device = deviceService.getOne(Wrappers.<Device>lambdaQuery()
-                .eq(Device::getManagementNumber, managementNumber));
+                .eq(Device::getManagementNumber, managementNumber)
+                .eq(Device::getIp, ip));
         if (ObjectUtils.isEmpty(device)) {
-            return Result.fail("璁惧缂栧彿閿欒锛�");
+            return Result.success(false);
         }
         if (ObjectUtils.isEmpty(device.getFileType()) || ObjectUtils.isEmpty(device.getCollectUrl())) {
             return Result.success(false);
@@ -199,7 +202,7 @@
             device.setCollectUrl(dataConfigList.getCollectUrl());
             device.setStorageUrl(dataConfigList.getStorageUrl());
             device.setIp(dataConfigList.getIp());
-            device.setFileType(device.getFileType());
+            device.setFileType(dataConfigList.getFileType());
             deviceService.updateById(device);
         } else {
             dataConfigService.saveOrUpdateBatch(dataConfigList.getDataConfigList());
@@ -222,11 +225,13 @@
     @GetMapping("/queryDataAcquisitionConfiguration")
     public Result<?> queryDataAcquisitionConfiguration(@RequestParam("deviceId") Integer deviceId,
                                                        @RequestParam("isDevice") Boolean isDevice,
-                                                       @RequestParam(value = "insProductItem", required = false) String insProductItem) {
+                                                       @RequestParam(value = "inspectionItem", required = false) String inspectionItem,
+                                                       @RequestParam(value = "inspectionItemSubclass", required = false) String inspectionItemSubclass) {
         if (isDevice) {
             List<DataConfig> list = dataConfigService.list(Wrappers.<DataConfig>lambdaQuery()
                 .eq(DataConfig::getDeviceId, deviceId)
-                .eq(DataConfig::getInsProductItem, insProductItem));
+                .eq(DataConfig::getInspectionItem, inspectionItem)
+                .eq(DataConfig::getInspectionItemSubclass, inspectionItemSubclass));
             return Result.success(list);
         } else {
             return Result.success(dataConfigService.selectDataConfigList(deviceId));
@@ -242,5 +247,12 @@
         dataConfigService.removeBatchByIds(collect);
         return Result.success();
     }
+
+    @ValueAuth
+    @ApiOperation(value = "鑿滃崟")
+    @PostMapping("/menu")
+    public Result menu() {
+        return Result.success(deviceService.menu());
+    }
 }
 

--
Gitblit v1.9.3