added benchmark
Signed-off-by: Greg Wilkins <gregw@webtide.com>
This commit is contained in:
Greg Wilkins
parent
09a60cd4f3
commit
8e4756b89c
|
|
@ -0,0 +1,196 @@
|
|||
//
|
||||
// ========================================================================
|
||||
// Copyright (c) 1995-2019 Mort Bay Consulting Pty. Ltd.
|
||||
// ------------------------------------------------------------------------
|
||||
// All rights reserved. This program and the accompanying materials
|
||||
// are made available under the terms of the Eclipse Public License v1.0
|
||||
// and Apache License v2.0 which accompanies this distribution.
|
||||
//
|
||||
// The Eclipse Public License is available at
|
||||
// http://www.eclipse.org/legal/epl-v10.html
|
||||
//
|
||||
// The Apache License v2.0 is available at
|
||||
// http://www.opensource.org/licenses/apache2.0.php
|
||||
//
|
||||
// You may elect to redistribute this code under either of these licenses.
|
||||
// ========================================================================
|
||||
//
|
||||
|
||||
package org.eclipse.jetty.server.jmh;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.lang.invoke.MethodHandle;
|
||||
import java.lang.invoke.MethodHandles;
|
||||
import java.lang.invoke.MethodType;
|
||||
import java.util.Arrays;
|
||||
import java.util.concurrent.ThreadLocalRandom;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
import javax.servlet.ServletException;
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
import javax.servlet.http.HttpServletResponse;
|
||||
|
||||
import org.eclipse.jetty.http.HttpField;
|
||||
import org.eclipse.jetty.http.HttpFields;
|
||||
import org.eclipse.jetty.http.HttpHeader;
|
||||
import org.eclipse.jetty.server.ForwardedRequestCustomizer;
|
||||
import org.eclipse.jetty.server.HttpConfiguration;
|
||||
import org.eclipse.jetty.server.HttpConnectionFactory;
|
||||
import org.eclipse.jetty.server.LocalConnector;
|
||||
import org.eclipse.jetty.server.Request;
|
||||
import org.eclipse.jetty.server.Server;
|
||||
import org.eclipse.jetty.server.handler.AbstractHandler;
|
||||
import org.eclipse.jetty.util.ArrayTrie;
|
||||
import org.eclipse.jetty.util.Trie;
|
||||
import org.openjdk.jmh.annotations.Benchmark;
|
||||
import org.openjdk.jmh.annotations.BenchmarkMode;
|
||||
import org.openjdk.jmh.annotations.Measurement;
|
||||
import org.openjdk.jmh.annotations.Mode;
|
||||
import org.openjdk.jmh.annotations.Scope;
|
||||
import org.openjdk.jmh.annotations.State;
|
||||
import org.openjdk.jmh.annotations.Threads;
|
||||
import org.openjdk.jmh.annotations.Warmup;
|
||||
import org.openjdk.jmh.profile.GCProfiler;
|
||||
import org.openjdk.jmh.runner.Runner;
|
||||
import org.openjdk.jmh.runner.RunnerException;
|
||||
import org.openjdk.jmh.runner.options.Options;
|
||||
import org.openjdk.jmh.runner.options.OptionsBuilder;
|
||||
|
||||
import static java.lang.invoke.MethodHandles.dropArguments;
|
||||
import static java.lang.invoke.MethodHandles.foldArguments;
|
||||
import static java.lang.invoke.MethodType.methodType;
|
||||
|
||||
|
||||
@State(Scope.Benchmark)
|
||||
@Threads(4)
|
||||
@Warmup(iterations = 7, time = 500, timeUnit = TimeUnit.MILLISECONDS)
|
||||
@Measurement(iterations = 7, time = 500, timeUnit = TimeUnit.MILLISECONDS)
|
||||
public class ForwardBenchmark
|
||||
{
|
||||
static HttpFields __fields = new HttpFields();
|
||||
static
|
||||
{
|
||||
Arrays.stream(new String[][]
|
||||
{
|
||||
{"accept", "*/*"},
|
||||
{"accept-encoding", "gzip, deflate, br"},
|
||||
{"accept-language", "en,en-AU;q=0.9,it;q=0.8"},
|
||||
{"content-length", "0"},
|
||||
{"content-type", "text/plain;charset=UTF-8"},
|
||||
{"cookie", "S=maestro=cH6W-eZ0JweknIgCwBxWD4FxQk647Uru:sso=HEiTd0qT5KdU7X6eL1m8snK1jNHwVJ9d"},
|
||||
{"origin", "https://www.google.com"},
|
||||
{"referer", "https://www.google.com/"},
|
||||
{"user-agent", "Mozilla/5.0"},
|
||||
{"x-client-data", "CLK1yQEIkLbJAQiltskBCKmdygEIoZ7KAQioo8oBCL+nygEI4qjKAQ=="},
|
||||
{"Forwarded", "for=192.0.2.43,for=198.51.100.17;by=203.0.113.60;proto=http;host=example.com"}
|
||||
}).map(a->new HttpField(a[0],a[1])).forEach(__fields::add);
|
||||
}
|
||||
|
||||
public ForwardBenchmark()
|
||||
{
|
||||
}
|
||||
|
||||
|
||||
|
||||
@Benchmark
|
||||
@BenchmarkMode({ Mode.Throughput})
|
||||
public String testStringCompare()
|
||||
{
|
||||
String forwardedFor = null;
|
||||
String forwardedHost = null;
|
||||
String forwardedServer = null;
|
||||
String forwarded = null;
|
||||
|
||||
for (HttpField field : __fields)
|
||||
{
|
||||
String name = field.getName();
|
||||
if (HttpHeader.X_FORWARDED_FOR.asString().equalsIgnoreCase(name))
|
||||
forwardedFor = field.getValue();
|
||||
else if (HttpHeader.X_FORWARDED_HOST.asString().equalsIgnoreCase(name))
|
||||
forwardedHost = field.getValue();
|
||||
else if (HttpHeader.X_FORWARDED_SERVER.asString().equalsIgnoreCase(name))
|
||||
forwardedServer = field.getValue();
|
||||
else if (HttpHeader.FORWARDED.asString().equalsIgnoreCase(name))
|
||||
forwardedServer = field.getValue();
|
||||
}
|
||||
|
||||
if (forwardedFor!=null)
|
||||
return forwardedFor;
|
||||
if (forwardedHost!=null)
|
||||
return forwardedHost;
|
||||
if (forwardedServer!=null)
|
||||
return forwardedServer;
|
||||
return forwarded;
|
||||
}
|
||||
|
||||
|
||||
|
||||
static class Forwarded
|
||||
{
|
||||
String host;
|
||||
|
||||
public void getHost(HttpField field)
|
||||
{
|
||||
if (host==null)
|
||||
host = field.getValue();
|
||||
}
|
||||
}
|
||||
|
||||
static Trie<MethodHandle> __handles = new ArrayTrie<>(1024);
|
||||
static
|
||||
{
|
||||
MethodHandles.Lookup lookup = MethodHandles.lookup();
|
||||
MethodType type = methodType(Void.TYPE, HttpField.class);
|
||||
|
||||
try
|
||||
{
|
||||
__handles.put(HttpHeader.X_FORWARDED_FOR.asString(), lookup.findVirtual(Forwarded.class, "getHost", type));
|
||||
__handles.put(HttpHeader.X_FORWARDED_HOST.asString(), lookup.findVirtual(Forwarded.class, "getHost", type));
|
||||
__handles.put(HttpHeader.X_FORWARDED_SERVER.asString(), lookup.findVirtual(Forwarded.class, "getHost", type));
|
||||
__handles.put(HttpHeader.FORWARDED.asString(), lookup.findVirtual(Forwarded.class, "getHost", type));
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@Benchmark
|
||||
@BenchmarkMode({ Mode.Throughput})
|
||||
public String testTrieMethodHandle()
|
||||
{
|
||||
Forwarded forwarded = new Forwarded();
|
||||
|
||||
try
|
||||
{
|
||||
for (HttpField field : __fields)
|
||||
{
|
||||
MethodHandle handle = __handles.get(field.getName());
|
||||
if (handle != null)
|
||||
handle.invoke(forwarded, field);
|
||||
}
|
||||
}
|
||||
catch (Throwable e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
return forwarded.host;
|
||||
}
|
||||
|
||||
|
||||
public static void main(String[] args) throws RunnerException
|
||||
{
|
||||
Options opt = new OptionsBuilder()
|
||||
.include(ForwardBenchmark.class.getSimpleName())
|
||||
.warmupIterations(20)
|
||||
.measurementIterations(10)
|
||||
.addProfiler(GCProfiler.class)
|
||||
.forks(1)
|
||||
.threads(100)
|
||||
.build();
|
||||
|
||||
new Runner(opt).run();
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue