d

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

@@ -43,9 +43,15 @@ public class AlarmController {
         EnterpriseDTO enterprise = this.enterpriseService.getEnterpriseById(SessionContextUtils.getLoginUserEnterpriseId());
         Integer alarmWay = enterprise.getAlarmWay();
         FreeQuerySource freeQuerySource = this.freeQueryService.getFreeQuerySource(SessionContextUtils.getLoginUserEnterpriseId());
-        Date expireTime = freeQuerySource.getDbExpiration();
-        Map<String, Object> alarmResult = this.getAlarmResult(expireTime, alarmWay);
-        alarmResult.put("isPrivate", freeQuerySource.getIsPrivate());
+        Map<String, Object> alarmResult = new HashMap<>();
+        if(freeQuerySource == null){
+            alarmResult.put("type", 0);
+            alarmResult.put("remain", 0);
+        }else {
+            Date expireTime = freeQuerySource.getDbExpiration();
+            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());
         if(org.apache.commons.collections.CollectionUtils.isNotEmpty(list)){

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

@@ -177,6 +177,10 @@ public class FreeQueryController {
 	public FreeQuerySource getCustomDatabase(){
 		FreeQuerySource freeQuerySource = this.freeQueryService.getFreeQuerySource(SessionContextUtils.getLoginUserEnterpriseId());
 		System.out.println(JSON.toJSONString(freeQuerySource));
+		if(freeQuerySource == null){
+			freeQuerySource = new FreeQuerySource();
+			freeQuerySource.setDatabase(this.enterpriseService.getEnterpriseById(SessionContextUtils.getLoginUserEnterpriseId()).getFactoryCode());
+		}
 		return freeQuerySource;
 	}