修正分页查询错误

sqlrest-common/src/main/java/com/gitee/sqlrest/common/dto/ProductContext.java

@@ -28,4 +28,5 @@ public class ProductContext {
   private boolean hasCatalogAndSchema;
   private Function<String, Pair<String, String>> adapter;
   private String pageSql;
+  private ThreeConsumer<Integer, Integer, List<Object>> pageConsumer;
 }

sqlrest-common/src/main/java/com/gitee/sqlrest/common/dto/ThreeConsumer.java

+package com.gitee.sqlrest.common.dto;
+
+@FunctionalInterface
+public interface ThreeConsumer<X, Y, Z> {
+
+  void accept(X arg1, Y arg2, Z arg3);
+}

sqlrest-core/src/main/java/com/gitee/sqlrest/core/exec/engine/impl/SqlExecutorService.java

 package com.gitee.sqlrest.core.exec.engine.impl;
 
-import cn.hutool.core.util.NumberUtil;
-import com.gitee.sqlrest.common.consts.Constants;
 import com.gitee.sqlrest.common.enums.NamingStrategyEnum;
 import com.gitee.sqlrest.common.enums.ProductTypeEnum;
 import com.gitee.sqlrest.core.exec.engine.AbstractExecutorEngine;
+import com.gitee.sqlrest.core.util.PageSizeUtils;
 import com.gitee.sqlrest.core.util.SqlJdbcUtils;
 import com.gitee.sqlrest.persistence.entity.ApiContextEntity;
 import com.gitee.sqlrest.template.Configuration;
@@ -35,12 +34,8 @@ public class SqlExecutorService extends AbstractExecutorEngine {
         for (ApiContextEntity sql : scripts) {
           SqlTemplate template = cfg.getTemplate(sql.getSqlText());
           SqlMeta sqlMeta = template.process(params);
-          int page = (null == params.get(Constants.PARAM_PAGE_NUMBER))
-              ? 1
-              : NumberUtil.parseInt(params.get(Constants.PARAM_PAGE_NUMBER).toString());
-          int size = (null == params.get(Constants.PARAM_PAGE_SIZE))
-              ? 10
-              : NumberUtil.parseInt(params.get(Constants.PARAM_PAGE_SIZE).toString());
+          int page = PageSizeUtils.getPageFromParams(params);
+          int size = PageSizeUtils.getSizeFromParams(params);
           dataList.add(SqlJdbcUtils.execute(productType, connection, sqlMeta, strategy, page, size));
         }
         connection.commit();

sqlrest-core/src/main/java/com/gitee/sqlrest/core/exec/module/DbVarModule.java

@@ -7,6 +7,7 @@ import com.gitee.sqlrest.common.enums.ProductTypeEnum;
 import com.gitee.sqlrest.core.exec.annotation.Comment;
 import com.gitee.sqlrest.core.exec.annotation.Module;
 import com.gitee.sqlrest.core.util.ConvertUtils;
+import com.gitee.sqlrest.core.util.PageSizeUtils;
 import com.gitee.sqlrest.template.Configuration;
 import com.gitee.sqlrest.template.SqlMeta;
 import com.gitee.sqlrest.template.SqlTemplate;
@@ -109,14 +110,9 @@ public class DbVarModule {
     SqlMeta sqlMeta = template.process(params);
     String pageSql = productType.getPageSql(sqlMeta.getSql());
     List<Object> parameters = sqlMeta.getParameter();
-    int page = (null == params.get(Constants.PARAM_PAGE_NUMBER))
-        ? 1
-        : NumberUtil.parseInt(params.get(Constants.PARAM_PAGE_NUMBER).toString());
-    int size = (null == params.get(Constants.PARAM_PAGE_SIZE))
-        ? 10
-        : NumberUtil.parseInt(params.get(Constants.PARAM_PAGE_SIZE).toString());
-    parameters.add(((page - 1) * size) < 0 ? 0 : (page - 1) * size);
-    parameters.add(size);
+    int page = PageSizeUtils.getPageFromParams(params);
+    int size = PageSizeUtils.getSizeFromParams(params);
+    this.productType.getPageConsumer().accept(page, size, parameters);
     return build(jdbcTemplate.queryForList(pageSql, parameters.toArray()));
   }
 

sqlrest-core/src/main/java/com/gitee/sqlrest/core/gateway/FirewallFilterService.java

 package com.gitee.sqlrest.core.gateway;
 
-import static com.gitee.sqlrest.common.consts.Constants.GATEWAY_APPLICATION_NAME;
-
-import com.gitee.sqlrest.common.dto.ResultEntity;
 import com.gitee.sqlrest.common.enums.OnOffEnum;
 import com.gitee.sqlrest.common.enums.WhiteBlackEnum;
 import com.gitee.sqlrest.core.dto.UpdateFirewallRulesRequest;
 import com.gitee.sqlrest.persistence.dao.FirewallRulesDao;
 import com.gitee.sqlrest.persistence.entity.FirewallRulesEntity;
 import java.util.Arrays;
-import java.util.List;
 import java.util.Optional;
 import java.util.Set;
 import java.util.stream.Collectors;
@@ -18,10 +14,7 @@ import javax.annotation.Resource;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
 import org.apache.logging.log4j.util.Strings;
-import org.springframework.cloud.client.ServiceInstance;
-import org.springframework.cloud.client.discovery.DiscoveryClient;
 import org.springframework.stereotype.Service;
-import org.springframework.web.client.RestTemplate;
 
 @Slf4j
 @Service
@@ -31,8 +24,6 @@ public class FirewallFilterService {
 
   @Resource
   private FirewallRulesDao firewallRulesDao;
-  @Resource
-  private DiscoveryClient discoveryClient;
 
   @PostConstruct
   public void refresh() {
@@ -68,18 +59,7 @@ public class FirewallFilterService {
 
   public void updateFirewallRules(UpdateFirewallRulesRequest request) {
     firewallRulesDao.update(request.getStatus(), request.getMode(), request.getAddresses());
-    refresh();
-
-    // 通过HTTP接口同步到所有的gateway节点(如何保证强一致性呢?)
-    RestTemplate restTemplate = new RestTemplate();
-    List<ServiceInstance> instances = discoveryClient.getInstances(GATEWAY_APPLICATION_NAME);
-    for (ServiceInstance instance : instances) {
-      String url = String.format("%s://%s:%s/cache/refresh",
-          instance.isSecure() ? "https" : "http",
-          instance.getHost(), instance.getPort());
-      restTemplate.getForEntity(url, ResultEntity.class);
-      log.info("sync ip rule cache to gateway node: {}", instance.getHost());
-    }
+    this.refresh();
   }
 
 }

sqlrest-core/src/main/java/com/gitee/sqlrest/core/util/PageSizeUtils.java

+package com.gitee.sqlrest.core.util;
+
+import cn.hutool.core.util.NumberUtil;
+import com.gitee.sqlrest.common.consts.Constants;
+import java.util.Map;
+import lombok.experimental.UtilityClass;
+
+@UtilityClass
+public class PageSizeUtils {
+
+  public static int getPageFromParams(Map<String, Object> params) {
+    int page = (null == params.get(Constants.PARAM_PAGE_NUMBER))
+        ? 1
+        : NumberUtil.parseInt(params.get(Constants.PARAM_PAGE_NUMBER).toString());
+    if (page <= 0) {
+      page = 1;
+    }
+    return page;
+  }
+
+  public static int getSizeFromParams(Map<String, Object> params) {
+    int size = (null == params.get(Constants.PARAM_PAGE_SIZE))
+        ? 10
+        : NumberUtil.parseInt(params.get(Constants.PARAM_PAGE_SIZE).toString());
+    if (size <= 0) {
+      size = 10;
+    }
+    return size;
+  }
+  
+}

sqlrest-core/src/main/java/com/gitee/sqlrest/core/util/SqlJdbcUtils.java

@@ -19,6 +19,8 @@ import lombok.extern.slf4j.Slf4j;
 @UtilityClass
 public class SqlJdbcUtils {
 
+  private static final int QUERY_TIMEOUT = 300;
+
   public static boolean isQuerySQL(String sql) {
     String upperSql = sql.toUpperCase().trim();
     return upperSql.startsWith("SELECT") || upperSql.startsWith("WITH");
@@ -29,23 +31,15 @@ public class SqlJdbcUtils {
   }
 
   public static Object execute(ProductTypeEnum productType, Connection connection, SqlMeta sqlMeta,
-      NamingStrategyEnum strategy, int page, int size)
-      throws SQLException {
+      NamingStrategyEnum strategy, int page, int size) throws SQLException {
     List<Object> paramValues = sqlMeta.getParameter();
     boolean isQuerySql = isQuerySQL(sqlMeta.getSql());
     String sql = isQuerySql ? productType.getPageSql(sqlMeta.getSql()) : sqlMeta.getSql();
     PreparedStatement statement = connection.prepareStatement(sql);
-    statement.setQueryTimeout(300);
-    statement.setFetchSize(isMySqlConnection(connection) ? Integer.MIN_VALUE : 100);
+    statement.setQueryTimeout(QUERY_TIMEOUT);
+    statement.setFetchSize(isMySqlConnection(connection) ? Integer.MIN_VALUE : size);
     if (isQuerySql) {
-      if (page <= 0) {
-        page = 1;
-      }
-      if (size <= 0) {
-        size = 10;
-      }
-      paramValues.add(size);
-      paramValues.add((page - 1) * size);
+      productType.getPageConsumer().accept(page, size, paramValues);
     }
     for (int i = 1; i <= paramValues.size(); i++) {
       statement.setObject(i, paramValues.get(i - 1));

sqlrest-gateway/src/main/java/com/gitee/sqlrest/gateway/controller/RefreshCacheController.java

-package com.gitee.sqlrest.gateway.controller;
-
-import com.gitee.sqlrest.common.dto.ResultEntity;
-import com.gitee.sqlrest.core.gateway.FirewallFilterService;
-import javax.annotation.Resource;
-import org.springframework.http.MediaType;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
-
-@RestController
-@RequestMapping("/cache")
-public class RefreshCacheController {
-
-  @Resource
-  private FirewallFilterService firewallFilterService;
-
-  @PostMapping(value = "/refresh", produces = MediaType.APPLICATION_JSON_VALUE)
-  public ResultEntity refresh() {
-    firewallFilterService.refresh();
-    return ResultEntity.success();
-  }
-
-}

sqlrest-gateway/src/main/java/com/gitee/sqlrest/gateway/filter/FirewallRefreshService.java

+package com.gitee.sqlrest.gateway.filter;
+
+import com.gitee.sqlrest.core.gateway.FirewallFilterService;
+import javax.annotation.Resource;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.boot.context.event.ApplicationReadyEvent;
+import org.springframework.context.event.EventListener;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+
+@Slf4j
+@Component
+public class FirewallRefreshService {
+
+  @Resource
+  private FirewallFilterService firewallFilterService;
+
+  /*每30秒执行一次*/
+  @EventListener(ApplicationReadyEvent.class)
+  @Scheduled(cron = "${cron.firewall.expression:0/30 0 * * * ?}")
+  public void loadFlowRules() {
+    try {
+      firewallFilterService.refresh();
+    } catch (Exception e) {
+      log.error("load firewall rules failed:{}", e.getMessage(), e);
+    }
+  }
+}