Java tutorial
/* * ==================================================================== * * Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements. See the NOTICE file distributed with * this work for additional information regarding copyright ownership. * The ASF licenses this file to You under the Apache License, Version 2.0 * (the "License"); you may not use this file except in compliance with * the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. * ==================================================================== * * This software consists of voluntary contributions made by many * individuals on behalf of the Apache Software Foundation. For more * information on the Apache Software Foundation, please see * <http://www.apache.org/>. * */ package com.jrodeo.remote; import java.io.IOException; import java.io.InputStream; import java.net.URI; import java.util.concurrent.ExecutionException; import java.util.concurrent.Future; import org.apache.http.ConnectionReuseStrategy; import org.apache.http.HeaderIterator; import org.apache.http.HttpClientConnection; import org.apache.http.HttpEntity; import org.apache.http.HttpException; import org.apache.http.HttpHost; import org.apache.http.HttpRequest; import org.apache.http.HttpRequestInterceptor; import org.apache.http.HttpResponse; import org.apache.http.HttpVersion; import org.apache.http.entity.ByteArrayEntity; import org.apache.http.impl.DefaultConnectionReuseStrategy; import org.apache.http.impl.pool.BasicConnPool; import org.apache.http.impl.pool.BasicPoolEntry; import org.apache.http.message.BasicHttpEntityEnclosingRequest; import org.apache.http.message.BasicHttpRequest; import org.apache.http.params.HttpConnectionParams; import org.apache.http.params.HttpParams; import org.apache.http.params.HttpProtocolParams; import org.apache.http.params.SyncBasicHttpParams; import org.apache.http.protocol.BasicHttpContext; import org.apache.http.protocol.ExecutionContext; import org.apache.http.protocol.HttpContext; import org.apache.http.protocol.HttpProcessor; import org.apache.http.protocol.HttpRequestExecutor; import org.apache.http.protocol.ImmutableHttpProcessor; import org.apache.http.protocol.RequestConnControl; import org.apache.http.protocol.RequestContent; import org.apache.http.protocol.RequestTargetHost; import org.apache.http.protocol.RequestUserAgent; import org.apache.http.util.VersionInfo; public class TestHttpCore implements TestHttpAgent { private final HttpParams params; private final HttpProcessor httpproc; private final HttpRequestExecutor httpexecutor; private final ConnectionReuseStrategy connStrategy; private final BasicConnPool pool; public TestHttpCore() { super(); this.params = new SyncBasicHttpParams(); this.params.setParameter(HttpProtocolParams.PROTOCOL_VERSION, HttpVersion.HTTP_1_1); this.params.setBooleanParameter(HttpProtocolParams.USE_EXPECT_CONTINUE, false); this.params.setBooleanParameter(HttpConnectionParams.STALE_CONNECTION_CHECK, false); this.params.setIntParameter(HttpConnectionParams.SOCKET_BUFFER_SIZE, 8 * 1024); this.params.setIntParameter(HttpConnectionParams.SO_TIMEOUT, 15000); this.httpproc = new ImmutableHttpProcessor(new HttpRequestInterceptor[] { new RequestContent(), new RequestTargetHost(), new RequestConnControl(), new RequestUserAgent() }, null); this.httpexecutor = new HttpRequestExecutor(); this.connStrategy = new DefaultConnectionReuseStrategy(); this.pool = new BasicConnPool(this.params); } public void init() { } public void shutdown() { } Stats execute(final URI target, final byte[] content, int n, int c) throws Exception { this.pool.setMaxTotal(2000); this.pool.setDefaultMaxPerRoute(c); HttpHost targetHost = new HttpHost(target.getHost(), target.getPort()); StringBuilder buffer = new StringBuilder(); buffer.append(target.getPath()); if (target.getQuery() != null) { buffer.append("?"); buffer.append(target.getQuery()); } String requestUri = buffer.toString(); Stats stats = new Stats(n, c); WorkerThread[] workers = new WorkerThread[c]; for (int i = 0; i < workers.length; i++) { workers[i] = new WorkerThread(stats, targetHost, requestUri, content); } for (int i = 0; i < workers.length; i++) { workers[i].start(); } for (int i = 0; i < workers.length; i++) { workers[i].join(); } return stats; } class WorkerThread extends Thread { private final Stats stats; private final HttpHost targetHost; private final String requestUri; private final byte[] content; WorkerThread(final Stats stats, final HttpHost targetHost, final String requestUri, final byte[] content) { super(); this.stats = stats; this.targetHost = targetHost; this.requestUri = requestUri; this.content = content; } @Override public void run() { byte[] buffer = new byte[4096]; HttpContext context = new BasicHttpContext(); while (!this.stats.isComplete()) { HttpRequest request; if (this.content == null) { BasicHttpRequest httpget = new BasicHttpRequest("GET", this.requestUri); request = httpget; } else { BasicHttpEntityEnclosingRequest httppost = new BasicHttpEntityEnclosingRequest("POST", this.requestUri); httppost.setEntity(new ByteArrayEntity(this.content)); request = httppost; } long contentLen = 0; boolean reusable = false; Future<BasicPoolEntry> future = pool.lease(targetHost, null); try { BasicPoolEntry entry = future.get(); try { HttpClientConnection conn = entry.getConnection(); context.setAttribute(ExecutionContext.HTTP_CONNECTION, conn); context.setAttribute(ExecutionContext.HTTP_TARGET_HOST, targetHost); httpexecutor.preProcess(request, httpproc, context); HttpResponse response = httpexecutor.execute(request, conn, context); httpexecutor.postProcess(response, httpproc, context); HttpEntity entity = response.getEntity(); if (entity != null) { // TODO can this be null? InputStream instream = entity.getContent(); contentLen = 0; if (instream != null) { try { int l = 0; while ((l = instream.read(buffer)) != -1) { contentLen += l; } } finally { instream.close(); } } } if (connStrategy.keepAlive(response, context)) { reusable = true; } for (HeaderIterator it = request.headerIterator(); it.hasNext();) { it.next(); it.remove(); } if (response.getStatusLine().getStatusCode() == 200) { this.stats.success(contentLen); } else { this.stats.failure(contentLen); } } finally { pool.release(entry, reusable); } } catch (InterruptedException ex) { this.stats.failure(contentLen); } catch (ExecutionException ex) { this.stats.failure(contentLen); } catch (IOException ex) { this.stats.failure(contentLen); } catch (HttpException ex) { this.stats.failure(contentLen); } } } } public Stats get(final URI target, int n, int c) throws Exception { return execute(target, null, n, c); } public Stats post(final URI target, byte[] content, int n, int c) throws Exception { return execute(target, content, n, c); } public String getClientName() { VersionInfo vinfo = VersionInfo.loadVersionInfo("org.apache.http", Thread.currentThread().getContextClassLoader()); return "Apache HttpCore 4 (ver: " + ((vinfo != null) ? vinfo.getRelease() : VersionInfo.UNAVAILABLE) + ")"; } public static void main(String[] args) throws Exception { if (args.length < 2) { System.out.println("Usage: <target URI> <no of requests> <concurrent connections>"); System.exit(-1); } URI targetURI = new URI(args[0]); int n = Integer.parseInt(args[1]); int c = 1; if (args.length > 2) { c = Integer.parseInt(args[2]); } TestHttpCore test = new TestHttpCore(); test.init(); try { long startTime = System.currentTimeMillis(); Stats stats = test.get(targetURI, n, c); long finishTime = System.currentTimeMillis(); Stats.printStats(targetURI, startTime, finishTime, stats); } finally { test.shutdown(); } } }