Runner for JMH with a modified output

JMHRunner.java

import org.junit.runners.model.InitializationError;
import org.openjdk.jmh.results.IterationResult;
import org.openjdk.jmh.results.RunResult;
import org.openjdk.jmh.runner.Runner;
import org.openjdk.jmh.runner.RunnerException;
import org.openjdk.jmh.runner.options.Options;

import java.util.Collection;
import java.util.Comparator;

/**
 * Created by José
 */
public class JMHRunner {

    private final Options options;
    public static final Comparator<IterationResult> ITERATION_RESULT_COMPARATOR =
            Comparator.comparing(iterationResult -> iterationResult.getPrimaryResult().getScore());

    public JMHRunner(Options options) throws InitializationError {
        this.options = options;
    }

    public Collection<RunResult> run() throws RunnerException {
        Collection<RunResult> runResults = new Runner(options).run();

        runResults.forEach(
                runResult -> runResult.getBenchmarkResults().forEach(
                        benchmarkResult -> {
                            Collection<IterationResult> iterationResults = benchmarkResult.getIterationResults();
                            IterationResult maxIterationResult = iterationResults.stream()
                                    .sorted(ITERATION_RESULT_COMPARATOR)
                                    .limit(10)
                                    .max(ITERATION_RESULT_COMPARATOR)
                                    .get();
                            iterationResults.removeIf(iterationResult -> ITERATION_RESULT_COMPARATOR.compare(iterationResult, maxIterationResult) > 0);
                        }
                )
        );

        TextResultFormat resultFormat = new TextResultFormat(System.out);
        resultFormat.writeOut(runResults);

        return runResults;
    }
}