Java Examples for org.apache.http.pool.PoolStats

private String formatStats(final SessionEndpoint endpoint) {
    final StringBuilder buf = new StringBuilder();
    final PoolStats totals = this.pool.getTotalStats();
    final PoolStats stats = this.pool.getStats(endpoint);
    buf.append("[total kept alive: ").append(totals.getAvailable()).append("; ");
    buf.append("address allocated: ").append(stats.getLeased() + stats.getAvailable());
    buf.append(" of ").append(stats.getMax()).append("; ");
    buf.append("total allocated: ").append(totals.getLeased() + totals.getAvailable());
    buf.append(" of ").append(totals.getMax()).append("]");
    return buf.toString();
}

@ApiOperation(value = "Get proxy repository pool stats")
@ApiResponses(value = { @ApiResponse(code = 200, message = "Proxy repository pool stats where retrieved."), @ApiResponse(code = 500, message = "An error occurred.") })
@RequestMapping(value = "{storageId}/{repositoryId}", method = RequestMethod.GET, produces = MediaType.TEXT_PLAIN_VALUE)
public ResponseEntity getPoolStatsForProxyRepository(@PathVariable(value = "storageId") String storageId, @PathVariable(value = "repositoryId") String repositoryId) {
    Storage storage = getConfiguration().getStorage(storageId);
    if (storage == null) {
        return ResponseEntity.status(HttpStatus.NOT_FOUND).body("The storage does not exist!");
    }
    Repository repository = storage.getRepository(repositoryId);
    if (repository == null) {
        return ResponseEntity.status(HttpStatus.NOT_FOUND).body("The repository does not exist!");
    }
    if (storage.getRepository(repositoryId).getRemoteRepository() == null) {
        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("Repository doesn't have remote repository!");
    }
    PoolStats poolStats = proxyRepositoryConnectionPoolConfigurationService.getPoolStats(repository.getRemoteRepository().getUrl());
    return ResponseEntity.ok(poolStats.toString());
}

@Override
public void doCollect() {
    for (Map.Entry<String, ConnPoolControl<HttpRoute>> each : pool.entrySet()) {
        Map<String, String> tags = new LinkedHashMap<String, String>();
        Map<String, Object> props = new LinkedHashMap<String, Object>();
        tags.put("name", each.getKey());
        ConnPoolControl<HttpRoute> connPoolControl = each.getValue();
        PoolStats stats = connPoolControl.getTotalStats();
        props.put("available", stats.getAvailable());
        props.put("leased", stats.getLeased());
        props.put("max", stats.getMax());
        props.put("pending", stats.getPending());
        handler.handle(MetricQuota.HTTPCLIENT, tags, props);
        //资源已满打印堆栈
        if (stats.getMax() == stats.getLeased()) {
            Collection<StackTraceElement[]> stackTraces = ((HttpClientStackTraceRepository) connPoolControl).getBlockingStackTrace();
            blockingStackTraceCollector.collect(MetricQuota.HTTPCLIENT, each.getKey(), stackTraces);
        }
    }
}

public void testPoolSize() throws SolrServerException, IOException {
    PoolingHttpClientConnectionManager pool = HttpClientUtil.createPoolingConnectionManager();
    final HttpSolrClient client1;
    final String fooUrl;
    {
        fooUrl = jetty.getBaseUrl().toString() + "/" + "collection1";
        CloseableHttpClient httpClient = HttpClientUtil.createClient(new ModifiableSolrParams(), pool, false);
        client1 = getHttpSolrClient(fooUrl, httpClient);
        client1.setConnectionTimeout(DEFAULT_CONNECTION_TIMEOUT);
    }
    final String barUrl = yetty.getBaseUrl().toString() + "/" + "collection1";
    {
        client1.setBaseURL(fooUrl);
        client1.deleteByQuery("*:*");
        client1.setBaseURL(barUrl);
        client1.deleteByQuery("*:*");
    }
    List<String> urls = new ArrayList<>();
    for (int i = 0; i < 17; i++) {
        urls.add(fooUrl);
    }
    for (int i = 0; i < 31; i++) {
        urls.add(barUrl);
    }
    Collections.shuffle(urls, random());
    try {
        int i = 0;
        for (String url : urls) {
            if (!client1.getBaseURL().equals(url)) {
                client1.setBaseURL(url);
            }
            client1.add(new SolrInputDocument("id", "" + (i++)));
        }
        client1.setBaseURL(fooUrl);
        client1.commit();
        assertEquals(17, client1.query(new SolrQuery("*:*")).getResults().getNumFound());
        client1.setBaseURL(barUrl);
        client1.commit();
        assertEquals(31, client1.query(new SolrQuery("*:*")).getResults().getNumFound());
        PoolStats stats = pool.getTotalStats();
        assertEquals("oh " + stats, 2, stats.getAvailable());
    } finally {
        for (HttpSolrClient c : new HttpSolrClient[] { client1 }) {
            HttpClientUtil.close(c.getHttpClient());
            c.close();
        }
    }
}

/**
     * Forgive me these sins.  This is the only way I can think of to determine the *actual* size of the connection pool
     * without wrapping large quantities of the underlying client.
     * <p>
     * This whole method is cheating and full of bad examples.  Don't copy this.  You've been warned.
     */
private int getPoolSize(JestHttpClient client) throws Exception {
    try {
        Field fieldHttpClient = client.getClass().getDeclaredField("httpClient");
        fieldHttpClient.setAccessible(true);
        Object objInternalHttpClient = fieldHttpClient.get(client);
        Field fieldConnectionManager = objInternalHttpClient.getClass().getDeclaredField("connManager");
        fieldConnectionManager.setAccessible(true);
        PoolingHttpClientConnectionManager poolingHttpClientConnectionManager = (PoolingHttpClientConnectionManager) fieldConnectionManager.get(objInternalHttpClient);
        PoolStats poolStats = poolingHttpClientConnectionManager.getTotalStats();
        return poolStats.getAvailable() + poolStats.getLeased();
    } catch (Exception e) {
        e.printStackTrace();
    }
    return -1;
}

public static Map<String, Integer> getConnectionPoolStats() {
    Map<String, Integer> stats = new HashMap<>();
    PoolStats poolStats = pool.getTotalStats();
    stats.put("availableConnections", poolStats.getAvailable());
    stats.put("maxConnections", poolStats.getMax());
    stats.put("leasedConnections", poolStats.getLeased());
    stats.put("pendingConnections", poolStats.getPending());
    stats.put("defaultMaxPerRoute", pool.getDefaultMaxPerRoute());
    return stats;
}

private String formatStats(final HttpRoute route) {
    final StringBuilder buf = new StringBuilder();
    final PoolStats totals = this.pool.getTotalStats();
    final PoolStats stats = this.pool.getStats(route);
    buf.append("[total kept alive: ").append(totals.getAvailable()).append("; ");
    buf.append("route allocated: ").append(stats.getLeased() + stats.getAvailable());
    buf.append(" of ").append(stats.getMax()).append("; ");
    buf.append("total allocated: ").append(totals.getLeased() + totals.getAvailable());
    buf.append(" of ").append(totals.getMax()).append("]");
    return buf.toString();
}

private String formatStats(final HttpRoute route) {
    StringBuilder buf = new StringBuilder();
    PoolStats totals = this.pool.getTotalStats();
    PoolStats stats = this.pool.getStats(route);
    buf.append("[total kept alive: ").append(totals.getAvailable()).append("; ");
    buf.append("route allocated: ").append(stats.getLeased() + stats.getAvailable());
    buf.append(" of ").append(stats.getMax()).append("; ");
    buf.append("total allocated: ").append(totals.getLeased() + totals.getAvailable());
    buf.append(" of ").append(totals.getMax()).append("]");
    return buf.toString();
}

private String formatStats(final HttpRoute route) {
    final StringBuilder buf = new StringBuilder();
    final PoolStats totals = this.pool.getTotalStats();
    final PoolStats stats = this.pool.getStats(route);
    buf.append("[total kept alive: ").append(totals.getAvailable()).append("; ");
    buf.append("route allocated: ").append(stats.getLeased() + stats.getAvailable());
    buf.append(" of ").append(stats.getMax()).append("; ");
    buf.append("total allocated: ").append(totals.getLeased() + totals.getAvailable());
    buf.append(" of ").append(totals.getMax()).append("]");
    return buf.toString();
}

@Override
public void release(final BasicNIOPoolEntry entry, boolean reusable) {
    System.out.println("[proxy->origin] connection released " + entry.getConnection());
    super.release(entry, reusable);
    StringBuilder buf = new StringBuilder();
    PoolStats totals = getTotalStats();
    buf.append("[total kept alive: ").append(totals.getAvailable()).append("; ");
    buf.append("total allocated: ").append(totals.getLeased() + totals.getAvailable());
    buf.append(" of ").append(totals.getMax()).append("]");
    System.out.println("[proxy->origin] " + buf.toString());
}

@Test
public void testConnectionReuse() throws Exception {
    httpServer.addSslConnector();
    session.setCache(new DefaultRepositoryCache());
    for (int i = 0; i < 3; i++) {
        newTransporter(httpServer.getHttpsUrl());
        GetTask task = new GetTask(URI.create("repo/file.txt"));
        transporter.get(task);
        assertEquals("test", task.getDataString());
    }
    PoolStats stats = ((ConnPoolControl<?>) ((HttpTransporter) transporter).getState().getConnectionManager()).getTotalStats();
    assertEquals(stats.toString(), 1, stats.getAvailable());
}

/**
         * Captures the connection pool metrics.
         */
private void captureConnectionPoolMetrics() {
    if (awsRequestMetrics.isEnabled() && httpClient.getHttpClientConnectionManager() instanceof ConnPoolControl<?>) {
        final PoolStats stats = ((ConnPoolControl<?>) httpClient.getHttpClientConnectionManager()).getTotalStats();
        awsRequestMetrics.withCounter(HttpClientPoolAvailableCount, stats.getAvailable()).withCounter(HttpClientPoolLeasedCount, stats.getLeased()).withCounter(HttpClientPoolPendingCount, stats.getPending());
    }
}

/**
         * Captures the connection pool metrics.
         */
private void captureConnectionPoolMetrics() {
    if (awsRequestMetrics.isEnabled() && httpClient.getHttpClientConnectionManager() instanceof ConnPoolControl<?>) {
        final PoolStats stats = ((ConnPoolControl<?>) httpClient.getHttpClientConnectionManager()).getTotalStats();
        awsRequestMetrics.withCounter(HttpClientPoolAvailableCount, stats.getAvailable()).withCounter(HttpClientPoolLeasedCount, stats.getLeased()).withCounter(HttpClientPoolPendingCount, stats.getPending());
    }
}

public PoolStats getPoolStats() {
    return getConnectionManager().getTotalStats();
}

private PoolStats getClientConnectionPoolStats(final DefaultDockerClient client) {
    return ((PoolingHttpClientConnectionManager) client.getClient().getConfiguration().getProperty(ApacheClientProperties.CONNECTION_MANAGER)).getTotalStats();
}