使用 PoolingHttpClientConnectionManager

问题描述

如何从 PoolingHttpClientConnectionManager 获取重用连接数

它包含返回 PoolStats 的 getTotalStats() 方法,但不存在有关重用连接数量的信息。

PoolingHttpClientConnectionManager 有一个私有的 CPool 实例,它从 httpcomponents-core 扩展了 AbstractConnPool。当连接被重用时,方法 getPoolEntryBlocking 调用可用于此目的的 onReuse(entry) 方法

我没有找到获取此信息以进行监控的方法。 有人看到另一种选择吗?

解决方法

如果您扩展标准池连接管理器或对其进行修饰,它应该为您提供一个入口点来收集所需的连接重用详细信息

public class MyPoolingHttpClientConnectionManager extends PoolingHttpClientConnectionManager {

    final AtomicLong reusedCount = new AtomicLong();

    @Override
    protected HttpClientConnection leaseConnection(
            Future<CPoolEntry> future,long timeout,TimeUnit timeUnit) throws InterruptedException,ExecutionException,ConnectionPoolTimeoutException {
        HttpClientConnection conn = super.leaseConnection(future,timeout,timeUnit);
        if (conn.isOpen()) {
            reusedCount.incrementAndGet();
        }
        return conn;
    }

}
,

解决方案是将 ConnectionRequest 子类化并覆盖 get 方法,以将执行委托给在构造函数上接收到的 ConnectionRequest 实例。>

然后我覆盖 PoolingHttpClientConnectionManager 中的 requestConnection 并用创建的子类包装返回的 ConnectionRequest

如果连接是打开的,我会增加重用连接的统计信息。否则,我会增加已创建连接的统计信息。

public class MonitoredPoolingHttpClientConnectionManager extends PoolingHttpClientConnectionManager

    private final AtomicInteger createdConnections = new AtomicInteger();
    private final AtomicInteger reusedConnections = new AtomicInteger();

    @Override
    public ConnectionRequest requestConnection(HttpRoute route,Object state) {
        ConnectionRequest requestConnection = super.requestConnection(route,state);
        return new MonitoredConnectionRequest(requestConnection);
    }
    
    private class MonitoredConnectionRequest implements ConnectionRequest {

        private ConnectionRequest wrapped;

        public MonitoredConnectionRequest(ConnectionRequest requestConnection) {
            this.wrapped = requestConnection;
        }

        @Override
        public boolean cancel() {
            return wrapped.cancel();
        }

        @Override
        public HttpClientConnection get(long timeout,ConnectionPoolTimeoutException {
            HttpClientConnection conn = wrapped.get(timeout,timeUnit);
            if (conn.isOpen()) {
                reusedConnections.incrementAndGet();
            } else {
                createdConnections.incrementAndGet();
            }
            return conn;
        }
    }