Java：MultiDataSource 新增后台 Headless 自动化测试接口，并支持内部请求突破默认查询数量限制

Author: TommyLemon

APIJSON-Java-Server/APIJSONBoot-MultiDataSource/pom.xml

@@ -142,11 +142,11 @@
     </dependency>
 
     <!-- 需要用的 HikariCP 数据库连接池库，3.1.0 以上 -->
-<!--    <dependency>-->
-<!--      <groupId>com.zaxxer</groupId>-->
-<!--      <artifactId>HikariCP</artifactId>-->
-<!--      <version>5.0.1</version>-->
-<!--    </dependency>-->
+    <dependency>
+      <groupId>com.zaxxer</groupId>
+      <artifactId>HikariCP</artifactId>
+      <version>5.0.1</version>
+    </dependency>
 
     <!-- 需要用的 Druid 数据库连接池库，1.0.29 以上 -->
     <dependency>

APIJSON-Java-Server/APIJSONBoot-MultiDataSource/src/main/java/apijson/boot/DemoController.java

@@ -48,6 +48,7 @@
 import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.client.RestTemplate;
 
+import java.io.IOException;
 import java.net.URLDecoder;
 import java.rmi.ServerException;
 import java.sql.PreparedStatement;
@@ -75,7 +76,6 @@
 import apijson.Log;
 import apijson.RequestMethod;
 import apijson.StringUtil;
-import apijson.demo.CompareUtil;
 import apijson.demo.DemoFunctionParser;
 import apijson.demo.DemoParser;
 import apijson.demo.DemoSQLConfig;
@@ -2030,6 +2030,20 @@ public void invokeMethod(@RequestBody String request, HttpServletRequest servlet
     super.invokeMethod(request, servletRequest);
   }
 
+
+  @GetMapping("/api/test/start")
+  public String startApiTest(HttpServletResponse response, HttpSession session) throws Exception {
+    //response.sendRedirect("http://localhost:3000/test/start");
+    long id = 100000 + Math.round(899999*Math.random());
+    DemoParser.KEY_MAP.put(id, session);  // 调这个接口一般是前端/CI/CD，调查询接口的是 Node，Session 不同 session.setAttribute("key", id);
+    return delegate("http://localhost:3000/test/start?key=" + id, null, null, null, null, HttpMethod.GET, session);
+  }
+  @GetMapping("/api/test/status")
+  public String getApiTestStatus(HttpServletResponse response, HttpSession session) throws Exception {
+    //response.sendRedirect("http://localhost:3000/test/status");
+    return delegate("http://localhost:3000/test/status", null, null, null, null, HttpMethod.GET, session);
+  }
+
   // 为 UnitAuto 提供的单元测试接口  https://github.com/TommyLemon/UnitAuto  >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
 
 

APIJSON-Java-Server/APIJSONBoot-MultiDataSource/src/main/java/apijson/demo/DemoParser.java

@@ -16,6 +16,9 @@
 
 import com.alibaba.fastjson.JSONObject;
 
+import java.util.HashMap;
+import java.util.Map;
+
 import javax.servlet.http.HttpSession;
 
 import apijson.RequestMethod;
@@ -35,6 +38,11 @@
  */
 public class DemoParser extends APIJSONParser<Long> {
 
+    public static final Map<Long, HttpSession> KEY_MAP;
+    static {
+      KEY_MAP = new HashMap<>();
+    }
+
     public DemoParser() {
         super();
     }
@@ -45,12 +53,26 @@ public DemoParser(RequestMethod method, boolean needVerify) {
         super(method, needVerify);
     }
 
+    private int maxQueryCount = MAX_QUERY_COUNT;
     //	可重写来设置最大查询数量
-    //	@Override
-    //	public int getMaxQueryCount() {
-    //		return 50;
-    //	}
+    @Override
+    public int getMaxQueryCount() {
+      return maxQueryCount;
+    }
 
+    @Override
+    public JSONObject parseResponse(JSONObject request) {
+        try {  // 内部使用，可通过这种方式来突破限制，获得更大的自由度
+          HttpSession session = KEY_MAP.get(request.getLong("key"));
+          //DemoVerifier.verifyLogin(session);
+          if (session != null) {
+            request.remove("key");
+            maxQueryCount = 1000;
+          }
+        } catch (Exception e) {}
+
+        return super.parseResponse(request);
+    }
 
     @Override
     public APIJSONObjectParser createObjectParser(JSONObject request, String parentPath, SQLConfig arrayConfig