取数平台权限

gic-cloud-data-hook/src/main/java/com/gic/cloud/data/hook/web/AlarmController.java

 package com.gic.cloud.data.hook.web;
 
+import com.gic.clerk.api.constant.ThirdProjectEnum;
+import com.gic.clerk.api.dto.right.RightMenuDTO;
+import com.gic.clerk.api.service.RightService;
 import com.gic.cloud.data.hook.api.dto.FreeQuerySource;
 import com.gic.cloud.data.hook.api.service.IFreeQueryService;
 import com.gic.enterprise.api.dto.EnterpriseDTO;
@@ -14,10 +17,7 @@ import org.springframework.web.bind.annotation.RequestMapping;
 
 import java.text.ParseException;
 import java.text.SimpleDateFormat;
-import java.util.Calendar;
-import java.util.Date;
-import java.util.HashMap;
-import java.util.Map;
+import java.util.*;
 
 @Controller
 public class AlarmController {
@@ -27,6 +27,8 @@ public class AlarmController {
     private EnterpriseService enterpriseService;
     @Autowired
     private IFreeQueryService freeQueryService;
+    @Autowired
+    private RightService rightService;
 
     @RequestMapping("service-alarm")
     public Map<String, Object> serviceAlarm(){
@@ -44,6 +46,16 @@ public class AlarmController {
         Date expireTime = freeQuerySource.getDbExpiration();
         Map<String, Object> alarmResult = this.getAlarmResult(expireTime, alarmWay);
         alarmResult.put("isPrivate", freeQuerySource.getIsPrivate());
+        alarmResult.put("customSearch", 0);
+        List<RightMenuDTO> list = this.rightService.listSuperAdminAllGicWebMenu(SessionContextUtils.getLoginUserEnterpriseId(), ThirdProjectEnum.DATA_HOOK.getCode());
+        List<String> tables = new ArrayList<>();
+        if(org.apache.commons.collections.CollectionUtils.isNotEmpty(list)){
+            for(RightMenuDTO menuDTO : list){
+                if("custom_search".equals(menuDTO.getMenuCode())){
+                    alarmResult.put("customSearch", 1);
+                }
+            }
+        }
         return alarmResult;
     }