From feac6e0c2db4e60d55417fde1971a0bc6d703566 Mon Sep 17 00:00:00 2001
From: zhanzhiqin <895896009@qq.com>
Date: 星期五, 27 五月 2022 15:58:28 +0800
Subject: [PATCH] fix

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/WaterQualityDataController.java |   17 +++++++++++++++--
 1 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/WaterQualityDataController.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/WaterQualityDataController.java
index ff26e09..4458695 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/WaterQualityDataController.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/WaterQualityDataController.java
@@ -4,8 +4,10 @@
 import com.sandu.common.object.BaseConditionVO;
 import com.sandu.common.util.ResponseUtil;
 import com.sandu.ximon.admin.dto.WaterQualityDataDto;
+import com.sandu.ximon.admin.security.PermissionConfig;
 import com.sandu.ximon.admin.service.WaterQualityDataService;
 import com.sandu.ximon.dao.bo.WaterQualityDataBo;
+import com.sandu.ximon.dao.enums.MenuEnum;
 import lombok.AllArgsConstructor;
 import org.springframework.web.bind.annotation.*;
 
@@ -19,6 +21,7 @@
 @RequestMapping("/v1/admin/waterqualitydata")
 public class WaterQualityDataController {
     private final WaterQualityDataService waterQualityDataService;
+    private PermissionConfig permissionConfig;
 
     /**
      * 姘磋川鏁版嵁鏌ヨ
@@ -26,15 +29,25 @@
      * @return
      */
     @GetMapping("/listWaterQualityData")
-    public ResponseVO<Object> listWaterQualityDataByKeyword(BaseConditionVO baseConditionVO, @RequestParam(required = false) String keyword) {
+    public ResponseVO<Object> listWaterQualityDataByKeyword(BaseConditionVO baseConditionVO, @RequestParam(value = "keyword", required = false) String keyword) {
+        if (!permissionConfig.check(MenuEnum.WATER_DATA_LIST.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         List<WaterQualityDataBo> list = waterQualityDataService.listWaterQualityDataByKeyword(baseConditionVO, keyword);
         return ResponseUtil.successPage(list);
     }
 
     @PostMapping("/getWaterQualityDataInfo/{waterQualityEquipmentCode}")
     public ResponseVO<Object> getWaterQualityDataInfo(@PathVariable String waterQualityEquipmentCode) {
+        if (!permissionConfig.check(MenuEnum.WATER_DATA_DETAIL.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         WaterQualityDataDto waterQualityDataDto = waterQualityDataService.getWaterQualityDataInfo(waterQualityEquipmentCode);
-        return ResponseUtil.success(waterQualityDataDto);
+        if(waterQualityDataDto != null){
+            return ResponseUtil.success(waterQualityDataDto);
+        }else{
+            return ResponseUtil.fail("姘磋川璁惧鍙蜂笉姝g‘鎴栨按璐ㄦ暟鎹В鏋愬嚭閿�");
+        }
     }
 
 }

--
Gitblit v1.9.3