org.apache.hc.core5.benchmark.HttpBenchmark Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of httpcore5-testing Show documentation
Show all versions of httpcore5-testing Show documentation
Apache HttpComponents HTTP/2 and HTTP/1.1 core component integration tests
/*
* ====================================================================
* 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
* () {
@Override
public IOSession decorate(final IOSession ioSession) {
return new IOSession() {
public String getId() {
return ioSession.getId();
}
public Lock lock() {
return ioSession.lock();
}
public void enqueue(final Command command, final Command.Priority priority) {
ioSession.enqueue(command, priority);
}
public boolean hasCommands() {
return ioSession.hasCommands();
}
public Command poll() {
return ioSession.poll();
}
public ByteChannel channel() {
return new ByteChannel() {
@Override
public int read(final ByteBuffer dst) throws IOException {
final int bytesRead = ioSession.channel().read(dst);
if (bytesRead > 0) {
stats.incTotalBytesRecv(bytesRead);
}
return bytesRead;
}
@Override
public int write(final ByteBuffer src) throws IOException {
final int bytesWritten = ioSession.channel().write(src);
if (bytesWritten > 0) {
stats.incTotalBytesSent(bytesWritten);
}
return bytesWritten;
}
@Override
public boolean isOpen() {
return ioSession.channel().isOpen();
}
@Override
public void close() throws IOException {
ioSession.channel().close();
}
};
}
public SocketAddress getRemoteAddress() {
return ioSession.getRemoteAddress();
}
public SocketAddress getLocalAddress() {
return ioSession.getLocalAddress();
}
public int getEventMask() {
return ioSession.getEventMask();
}
public void setEventMask(final int ops) {
ioSession.setEventMask(ops);
}
public void setEvent(final int op) {
ioSession.setEvent(op);
}
public void clearEvent(final int op) {
ioSession.clearEvent(op);
}
public void close() {
ioSession.close();
}
public int getStatus() {
return ioSession.getStatus();
}
public boolean isClosed() {
return ioSession.isClosed();
}
public Timeout getSocketTimeout() {
return ioSession.getSocketTimeout();
}
public void setSocketTimeout(final Timeout timeout) {
ioSession.setSocketTimeout(timeout);
}
public long getLastReadTime() {
return ioSession.getLastReadTime();
}
public long getLastWriteTime() {
return ioSession.getLastWriteTime();
}
public void updateReadTime() {
ioSession.updateReadTime();
}
public void updateWriteTime() {
ioSession.updateWriteTime();
}
public void close(final CloseMode closeMode) {
ioSession.close(closeMode);
}
};
}
})
.setStreamListener(new Http1StreamListener() {
@Override
public void onRequestHead(final HttpConnection connection, final HttpRequest request) {
if (config.getVerbosity() >= 3) {
System.out.println(">> " + request.getMethod() + " " + request.getRequestUri());
final Header[] headers = request.getHeaders();
for (final Header header : headers) {
System.out.println(">> " + header.toString());
}
System.out.println();
}
}
@Override
public void onResponseHead(final HttpConnection connection, final HttpResponse response) {
if (config.getVerbosity() >= 3) {
System.out.println("<< " + response.getCode() + " " + response.getReasonPhrase());
final Header[] headers = response.getHeaders();
for (final Header header : headers) {
System.out.println("<< " + header.toString());
}
System.out.println();
}
}
@Override
public void onExchangeComplete(final HttpConnection connection, final boolean keepAlive) {
if (keepAlive) {
stats.incKeepAliveCount();
}
}
})
.setStreamListener(new Http2StreamListener() {
private final FramePrinter framePrinter = new FramePrinter();
@Override
public void onHeaderInput(
final HttpConnection connection,
final int streamId,
final List headers) {
if (config.getVerbosity() >= 3) {
for (final Header header : headers) {
System.out.println(">> " + header.toString());
}
System.out.println();
}
}
@Override
public void onHeaderOutput(
final HttpConnection connection,
final int streamId,
final List headers) {
if (config.getVerbosity() >= 3) {
for (final Header header : headers) {
System.out.println("<< " + header.toString());
}
System.out.println();
}
}
@Override
public void onFrameInput(
final HttpConnection connection,
final int streamId,
final RawFrame frame) {
if (config.getVerbosity() >= 4) {
System.out.print(">> ");
try {
framePrinter.printFrameInfo(frame, System.out);
} catch (final IOException ignore) {
}
System.out.println();
}
}
@Override
public void onFrameOutput(
final HttpConnection connection,
final int streamId,
final RawFrame frame) {
if (config.getVerbosity() >= 4) {
System.out.print("<< ");
try {
framePrinter.printFrameInfo(frame, System.out);
} catch (final IOException ignore) {
}
System.out.println();
}
}
@Override
public void onInputFlowControl(
final HttpConnection connection,
final int streamId,
final int delta,
final int actualSize) {
if (config.getVerbosity() >= 5) {
System.out.println(">> stream " + streamId + ": " + actualSize + " " + delta);
}
}
@Override
public void onOutputFlowControl(
final HttpConnection connection,
final int streamId,
final int delta,
final int actualSize) {
if (config.getVerbosity() >= 5) {
System.out.println("<< stream " + streamId + ": " + actualSize + " " + delta);
}
}
})
.setIOReactorConfig(IOReactorConfig.custom()
.setSoTimeout(config.getSocketTimeout())
.build())
.create()) {
requester.setDefaultMaxPerRoute(config.getConcurrencyLevel());
requester.setMaxTotal(config.getConcurrencyLevel() * 2);
requester.start();
return doExecute(requester, stats);
}
}
private Results doExecute(final HttpAsyncRequester requester, final Stats stats) throws Exception {
final URI requestUri = config.getUri();
final HttpHost host = new HttpHost(requestUri.getScheme(), requestUri.getHost(), requestUri.getPort());
final AtomicLong requestCount = new AtomicLong(config.getRequests());
final HttpVersion version = HttpVersion.HTTP_1_1;
final CountDownLatch completionLatch = new CountDownLatch(config.getConcurrencyLevel());
final BenchmarkWorker[] workers = new BenchmarkWorker[config.getConcurrencyLevel()];
for (int i = 0; i < workers.length; i++) {
final HttpCoreContext context = HttpCoreContext.create();
context.setProtocolVersion(version);
final BenchmarkWorker worker = new BenchmarkWorker(
requester,
host,
context,
requestCount,
completionLatch,
stats,
config);
workers[i] = worker;
}
final long deadline = config.getTimeLimit() != null ? config.getTimeLimit().toMillis() : Long.MAX_VALUE;
final long startTime = System.currentTimeMillis();
for (int i = 0; i < workers.length; i++) {
workers[i].execute();
}
completionLatch.await(deadline, TimeUnit.MILLISECONDS);
if (config.getVerbosity() >= 3) {
System.out.println("...done");
}
final long endTime = System.currentTimeMillis();
for (int i = 0; i < workers.length; i++) {
workers[i].releaseResources();
}
return new Results(
stats.getServerName(),
stats.getVersion(),
host.getHostName(),
host.getPort() > 0 ? host.getPort() : host.getSchemeName().equalsIgnoreCase("https") ? 443 : 80,
requestUri.toASCIIString(),
stats.getContentLength(),
config.getConcurrencyLevel(),
endTime - startTime,
stats.getSuccessCount(),
stats.getFailureCount(),
stats.getKeepAliveCount(),
stats.getTotalBytesRecv(),
stats.getTotalBytesSent(),
stats.getTotalContentLength());
}
}