取数平台权限

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

@@ -121,8 +121,10 @@ public class FreeQueryController {
 	@RequestMapping("/get-free-query-count")
 	public GeneralResult getFreeQueryCount(String sql, String enterpriseId, HttpServletRequest request, HttpServletResponse response) {
 		GeneralResult result = new GeneralResult();
-		EnterpriseDTO enterprise = this.enterpriseService.getEnterpriseById(enterpriseId);
-		if(enterprise.getDatahookEndDate() != null && enterprise.getDatahookEndDate().before(new Date())){
+		FreeQuerySource freeQuerySource = this.freeQueryService.getFreeQuerySource(SessionContextUtils.getLoginUserEnterpriseId());
+		Date expireTime = freeQuerySource.getDbExpiration();
+		Integer isPrivate = freeQuerySource.getIsPrivate();
+		if(isPrivate != null && isPrivate == 1 && expireTime != null && expireTime.before(new Date())){
 			result.setInfo("0");
 		}else {
 			result.setInfo(String.valueOf(this.freeQueryService.getFreeQueryCount(sql, enterpriseId)));