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