test/microbench/org/apache/cassandra/test/microbench/StreamingTombstoneHistogramBuilderBench.java

/*
 * 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.
 */

package org.apache.cassandra.test.microbench;


import java.util.concurrent.TimeUnit;
import java.util.Random;

import org.apache.cassandra.utils.streamhist.StreamingTombstoneHistogramBuilder;
import org.openjdk.jmh.annotations.*;
import org.openjdk.jmh.profile.*;
import org.openjdk.jmh.runner.*;
import org.openjdk.jmh.runner.options.*;

@BenchmarkMode(Mode.AverageTime)
@OutputTimeUnit(TimeUnit.MILLISECONDS)
@Warmup(iterations = 3, time = 1, timeUnit = TimeUnit.SECONDS)
@Measurement(iterations = 5, time = 2, timeUnit = TimeUnit.SECONDS)
@Fork(value = 1)
@Threads(1)
@State(Scope.Benchmark)
public class StreamingTombstoneHistogramBuilderBench
{

    static int[] secondInMonth = new int[10000000];
    static int[] secondInDay = new int[10000000];
    static int[] secondIn3Hour = new int[10000000];
    static int[] secondInMin = new int[10000000];

    static
    {
        final int now = (int) (System.currentTimeMillis() / 1000L);
        Random random = new Random();
        for(int i = 0 ; i < 10000000; i++)
        {
            // Seconds in a month
            secondInMonth[i] = now + random.nextInt(3600 * 24 * 30);
            // Seconds in a day
            secondInDay[i] = now + random.nextInt(3600 * 24);
            // Seconds in 3 hours
            secondIn3Hour[i] = now + random.nextInt(3600 * 3);
            // Seconds in a minute
            secondInMin[i] = now + random.nextInt(60);
        }
    }

    @Param({ "secondInMonth", "secondInDay", "secondIn3Hour", "secondInMin" })
    String a_workLoad;

    @Param({ "0", "1000", "10000", "100000" })
    int b_spoolSize;

    @Benchmark
    public void test()
    {
        StreamingTombstoneHistogramBuilder histogram = new StreamingTombstoneHistogramBuilder(100, b_spoolSize, 1);
        int[] data = selectWorkload(a_workLoad);

        for (int time : data)
        {
            histogram.update(time);
        }

        histogram.flushHistogram();
    }

    private int[] selectWorkload(String workLoad)
    {
        switch (workLoad)
        {
            case "secondInMonth":
                return secondInMonth;
            case "secondInDay":
                return secondInDay;
            case "secondIn3Hour":
                return secondIn3Hour;
            case "secondInMin":
                return secondInMin;
            default:
                throw new IllegalArgumentException("Invalid workload type: " + workLoad);
        }
    }


    public static void main(String[] args) throws Exception
    {
        Options opt = new OptionsBuilder()
                      .include(StreamingTombstoneHistogramBuilderBench.class.getSimpleName())
                      .warmupIterations(3)
                      .measurementIterations(10)
                      .addProfiler(GCProfiler.class)
                      .threads(1)
                      .forks(1)
                      .build();
        new Runner(opt).run();
    }
}