feat: add configurable OpenCode HTTP client pool

Author: loong10k

src/main/java/io/github/easy4j/opencode/OpenCodeClient.java

@@ -212,6 +212,13 @@ private void copyHttpConfig(OpenCodeHttpClientConfig src) {
         this.config.getHttp().setPassword(src.getPassword());
         this.config.getHttp().setConnectTimeoutMillis(src.getConnectTimeoutMillis());
         this.config.getHttp().setReadTimeoutMillis(src.getReadTimeoutMillis());
+        this.config.getHttp().setWriteTimeoutMillis(src.getWriteTimeoutMillis());
+        this.config.getHttp().setCallTimeoutMillis(src.getCallTimeoutMillis());
+        this.config.getHttp().setMaxIdleConnections(src.getMaxIdleConnections());
+        this.config.getHttp().setKeepAliveDurationMillis(src.getKeepAliveDurationMillis());
+        this.config.getHttp().setMaxRequests(src.getMaxRequests());
+        this.config.getHttp().setMaxRequestsPerHost(src.getMaxRequestsPerHost());
+        this.config.getHttp().setRetryOnConnectionFailure(src.isRetryOnConnectionFailure());
         this.config.getHttp().setVerifySsl(src.isVerifySsl());
         this.config.getHttp().setDefaultModel(src.getDefaultModel());
         this.config.getHttp().setDefaultAgent(src.getDefaultAgent());

src/main/java/io/github/easy4j/opencode/OpenCodeHttpClientConfig.java

@@ -55,6 +55,41 @@ public class OpenCodeHttpClientConfig {
      */
     private int readTimeoutMillis = 300_000;
 
+    /**
+     * 写入超时（毫秒）。
+     */
+    private int writeTimeoutMillis = 120_000;
+
+    /**
+     * 整个调用超时（毫秒）；0 表示不额外限制。
+     */
+    private int callTimeoutMillis;
+
+    /**
+     * 连接池最大空闲连接数。
+     */
+    private int maxIdleConnections = 32;
+
+    /**
+     * 空闲连接保活时间（毫秒）。
+     */
+    private long keepAliveDurationMillis = 300_000L;
+
+    /**
+     * 异步请求最大并发数。
+     */
+    private int maxRequests = 128;
+
+    /**
+     * 单主机异步请求最大并发数。
+     */
+    private int maxRequestsPerHost = 128;
+
+    /**
+     * 遇到失效连接等传输故障时是否允许 OkHttp 自动恢复。
+     */
+    private boolean retryOnConnectionFailure = true;
+
     /**
      * 是否校验 HTTPS 证书；为 false 时关闭校验（仅建议开发环境）。
      */

src/main/java/io/github/easy4j/opencode/OpenCodeOkHttpClientFactory.java

@@ -0,0 +1,62 @@
+package io.github.easy4j.opencode;
+
+import okhttp3.ConnectionPool;
+import okhttp3.Dispatcher;
+import okhttp3.OkHttpClient;
+
+import java.util.Objects;
+import java.util.concurrent.TimeUnit;
+
+/**
+ * OpenCode 独立运行时使用的 OkHttpClient 工厂。
+ *
+ * <p>Spring 容器已经提供外部 {@link OkHttpClient} 时应使用注入构造器，本工厂不会参与。</p>
+ */
+public final class OpenCodeOkHttpClientFactory {
+
+    private OpenCodeOkHttpClientFactory() {
+    }
+
+    /**
+     * 根据 OpenCode HTTP 配置创建客户端。
+     *
+     * @param config HTTP 配置
+     * @return SDK 自主管理的 OkHttpClient
+     */
+    public static OkHttpClient create(OpenCodeHttpClientConfig config) {
+        Objects.requireNonNull(config, "config");
+        Dispatcher dispatcher = new Dispatcher();
+        dispatcher.setMaxRequests(Math.max(1, config.getMaxRequests()));
+        dispatcher.setMaxRequestsPerHost(Math.max(1, config.getMaxRequestsPerHost()));
+        ConnectionPool connectionPool = new ConnectionPool(
+                Math.max(1, config.getMaxIdleConnections()),
+                Math.max(1L, config.getKeepAliveDurationMillis()),
+                TimeUnit.MILLISECONDS);
+        OkHttpClient.Builder builder = new OkHttpClient.Builder()
+                .dispatcher(dispatcher)
+                .connectionPool(connectionPool)
+                .connectTimeout(Math.max(1, config.getConnectTimeoutMillis()), TimeUnit.MILLISECONDS)
+                .readTimeout(Math.max(0, config.getReadTimeoutMillis()), TimeUnit.MILLISECONDS)
+                .writeTimeout(Math.max(1, config.getWriteTimeoutMillis()), TimeUnit.MILLISECONDS)
+                .callTimeout(Math.max(0, config.getCallTimeoutMillis()), TimeUnit.MILLISECONDS)
+                .retryOnConnectionFailure(config.isRetryOnConnectionFailure());
+        if (!config.isVerifySsl()) {
+            builder.hostnameVerifier((hostname, session) -> true);
+        }
+        return builder.build();
+    }
+
+    /**
+     * 释放 SDK 自建客户端资源。
+     *
+     * @param client SDK 自建客户端
+     */
+    public static void shutdown(OkHttpClient client) {
+        if (Objects.isNull(client)) {
+            return;
+        }
+        client.dispatcher().cancelAll();
+        client.connectionPool().evictAll();
+        client.dispatcher().executorService().shutdown();
+    }
+}

src/main/java/io/github/easy4j/opencode/api/OpenCodeHttpClient.java

@@ -4,6 +4,7 @@
 import com.fasterxml.jackson.databind.DeserializationFeature;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import io.github.easy4j.opencode.OpenCodeHttpClientConfig;
+import io.github.easy4j.opencode.OpenCodeOkHttpClientFactory;
 import io.github.easy4j.opencode.api.model.*;
 import io.github.easy4j.opencode.exception.OpenCodeHttpException;
 import lombok.extern.slf4j.Slf4j;
@@ -18,7 +19,6 @@
 import java.util.Map;
 import java.util.Objects;
 import java.util.Optional;
-import java.util.concurrent.TimeUnit;
 
 /**
  * OpenCode Server HTTP 客户端，封装 REST API。
@@ -44,14 +44,7 @@ public OpenCodeHttpClient(OpenCodeHttpClientConfig config, ObjectMapper objectMa
     }
 
     private static OkHttpClient buildOkHttpClient(OpenCodeHttpClientConfig config) {
-        // 兜底创建
-        OkHttpClient.Builder builder = new OkHttpClient.Builder()
-                .connectTimeout(config.getConnectTimeoutMillis(), TimeUnit.MILLISECONDS)
-                .readTimeout(config.getReadTimeoutMillis(), TimeUnit.MILLISECONDS);
-        if (!config.isVerifySsl()) {
-            builder.hostnameVerifier((hostname, session) -> true);
-        }
-        return builder.build();
+        return OpenCodeOkHttpClientFactory.create(config);
     }
 
     // ============================================================

src/test/java/io/github/easy4j/opencode/OpenCodeOkHttpClientFactoryTest.java

@@ -0,0 +1,40 @@
+package io.github.easy4j.opencode;
+
+import okhttp3.OkHttpClient;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+
+/**
+ * {@link OpenCodeOkHttpClientFactory} 配置测试。
+ */
+class OpenCodeOkHttpClientFactoryTest {
+
+    @Test
+    void shouldApplyIndependentHttpConfiguration() {
+        OpenCodeHttpClientConfig config = new OpenCodeHttpClientConfig();
+        config.setConnectTimeoutMillis(11001);
+        config.setReadTimeoutMillis(22002);
+        config.setWriteTimeoutMillis(33003);
+        config.setCallTimeoutMillis(44004);
+        config.setMaxIdleConnections(13);
+        config.setKeepAliveDurationMillis(55005L);
+        config.setMaxRequests(66);
+        config.setMaxRequestsPerHost(33);
+        config.setRetryOnConnectionFailure(false);
+
+        OkHttpClient client = OpenCodeOkHttpClientFactory.create(config);
+        try {
+            assertEquals(11001, client.connectTimeoutMillis());
+            assertEquals(22002, client.readTimeoutMillis());
+            assertEquals(33003, client.writeTimeoutMillis());
+            assertEquals(44004, client.callTimeoutMillis());
+            assertEquals(66, client.dispatcher().getMaxRequests());
+            assertEquals(33, client.dispatcher().getMaxRequestsPerHost());
+            assertFalse(client.retryOnConnectionFailure());
+        } finally {
+            OpenCodeOkHttpClientFactory.shutdown(client);
+        }
+    }
+}