如何在PoolingHttpClientConnectionManager每次创建连接时调用指定方法？

其实你要的需求完全可以实现，核心是自定义连接工厂——Apache HttpClient的连接池是通过ConnectionFactory来创建新连接的，我们只需要覆写这个工厂的创建方法，插入自己的逻辑就行。

步骤1：自定义连接工厂类 #

继承ManagedHttpClientConnectionFactory（这是连接池默认使用的连接工厂），覆写create方法，在创建连接前后调用你的指定方法：

import org.apache.http.config.ConnectionConfig;
import org.apache.http.impl.conn.ManagedHttpClientConnectionFactory;
import org.apache.http.protocol.HttpContext;

public class CustomConnectionFactory extends ManagedHttpClientConnectionFactory {

    // 这里就是你要在创建连接时调用的指定方法
    private void onConnectionCreated() {
        System.out.println("新的连接已创建！可以在这里执行你的自定义逻辑");
        // 比如记录日志、初始化参数、发送通知等操作都可以放在这里
    }

    @Override
    public org.apache.http.conn.ManagedHttpClientConnection create(
            HttpContext context,
            ConnectionConfig config) {
        // 先执行你的自定义方法
        onConnectionCreated();
        // 再调用父类方法创建实际的连接
        return super.create(context, config);
    }
}

步骤2：将自定义工厂注入连接池管理器 #

修改你的PoolingConnectionFactory类，在初始化PoolingHttpClientConnectionManager时传入自定义的连接工厂：

import org.apache.http.*;
import org.apache.http.conn.ConnectionKeepAliveStrategy;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.impl.client.TargetAuthenticationStrategy;
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
import org.apache.http.protocol.HttpContext;
import org.apache.log4j.LogManager;
import org.apache.log4j.Logger;
import java.util.concurrent.TimeUnit;

public class PoolingConnectionFactory {
    // 初始化连接池时传入自定义连接工厂
    public PoolingHttpClientConnectionManager connManager = new PoolingHttpClientConnectionManager(
            new CustomConnectionFactory()
    );

    // 补全你原来未定义的KeepAliveStrategy示例
    private final ConnectionKeepAliveStrategy strategy = new ConnectionKeepAliveStrategy() {
        @Override
        public long getKeepAliveDuration(HttpResponse response, HttpContext context) {
            // 示例：让连接保持30秒
            return 30 * 1000;
        }
    };

    public CloseableHttpClient CreateHttpClient(PoolingHttpClientConnectionManager connManager) {
        CloseableHttpClient client = HttpClients.custom()
                .setTargetAuthenticationStrategy(new AuthendictionSessionStrategy()) // 假设该类你已实现
                .setKeepAliveStrategy(strategy)
                .setConnectionManager(connManager)
                .build();
        return client;
    }

    public CloseableHttpClient getConnection(PoolingHttpClientConnectionManager connMan) {
        connMan.closeExpiredConnections();
        connMan.closeIdleConnections(50, TimeUnit.SECONDS);
        return CreateHttpClient(connMan);
    }
}

关键点说明 #

• ManagedHttpClientConnectionFactory是连接池创建ManagedHttpClientConnection的核心类，所有新连接的创建都会经过它的create方法。
• 覆写create方法时，先执行自定义逻辑再调用父类方法，就能保证每次新连接创建时都触发你的指定方法。
• 如果你的自定义方法需要上下文信息（比如请求URL、HttpContext等），可以直接从create方法的参数中获取并传入。

这样修改后，每次连接池需要创建新连接时，都会自动调用你指定的方法，完美满足需求！

内容的提问来源于stack exchange，提问作者Lahiru Lakshitha Samaranayaka