zipkin2.reporter.brave.ZipkinSpanHandlerBenchmarks Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of benchmarks Show documentation
Benchmarks (JMH)
The newest version!
/*
 * Copyright 2016-2020 The OpenZipkin Authors
 *
 * Licensed 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.
 */
package zipkin2.reporter.brave;

import brave.handler.MutableSpan;
import brave.handler.SpanHandler;
import brave.propagation.TraceContext;
import java.util.concurrent.TimeUnit;
import org.openjdk.jmh.annotations.Benchmark;
import org.openjdk.jmh.annotations.BenchmarkMode;
import org.openjdk.jmh.annotations.Fork;
import org.openjdk.jmh.annotations.Measurement;
import org.openjdk.jmh.annotations.Mode;
import org.openjdk.jmh.annotations.OutputTimeUnit;
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.runner.Runner;
import org.openjdk.jmh.runner.RunnerException;
import org.openjdk.jmh.runner.options.Options;
import org.openjdk.jmh.runner.options.OptionsBuilder;
import zipkin2.reporter.Reporter;

import static zipkin2.reporter.brave.MutableSpanBenchmarks.newBigClientSpan;
import static zipkin2.reporter.brave.MutableSpanBenchmarks.newServerSpan;

@Measurement(iterations = 5, time = 1)
@Warmup(iterations = 10, time = 1)
@Fork(3)
@BenchmarkMode(Mode.SampleTime)
@OutputTimeUnit(TimeUnit.MICROSECONDS)
@State(Scope.Thread)
@Threads(1)
public class ZipkinSpanHandlerBenchmarks {
  final SpanHandler handler =
      ZipkinSpanHandler.newBuilder(Reporter.NOOP).alwaysReportSpans(true).build();
  final TraceContext context = TraceContext.newBuilder().traceId(1).spanId(2).sampled(true).build();
  final MutableSpan serverSpan = newServerSpan();
  final MutableSpan bigClientSpan = newBigClientSpan();

  @Benchmark public boolean handleServerSpan() {
    return handler.end(context, serverSpan, SpanHandler.Cause.FINISHED);
  }

  @Benchmark public boolean handleBigClientSpan() {
    return handler.end(context, bigClientSpan, SpanHandler.Cause.FINISHED);
  }

  // Convenience main entry-point
  public static void main(String[] args) throws RunnerException {
    Options opt = new OptionsBuilder()
        .addProfiler("gc")
        .include(".*" + ZipkinSpanHandlerBenchmarks.class.getSimpleName() + ".*")
        .build();

    new Runner(opt).run();
  }
}