HAMA-559, caliper benchmark

HAMA-559_bench_1.java

package de.jungblut.benchmark;

import java.util.Random;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.io.IntWritable;
import org.apache.hama.bsp.TaskAttemptID;
import org.apache.hama.bsp.TaskID;
import org.apache.hama.bsp.message.DiskQueue;
import org.apache.hama.bsp.message.MessageQueue;
import org.apache.hama.bsp.message.SpillingQueue;

import com.google.caliper.Param;
import com.google.caliper.Runner;
import com.google.caliper.SimpleBenchmark;

public class SpillBenchmark extends SimpleBenchmark {

  @Param({ "10", "100", "1000", "10000", "100000", "1000000" })
  private int size;

  @Param
  Type type;

  public enum Type {
    SPILLING_QUEUE, DISK_QUEUE
  };

  int[] array;
  MessageQueue<IntWritable> queue;

  @Override
  protected void setUp() throws Exception {
    array = new int[size];
    Random r = new Random();
    for (int i = 0; i < size; i++) {
      array[i] = r.nextInt();
    }
    if (type == Type.DISK_QUEUE) {
      queue = new DiskQueue<>();
    } else {
      queue = new SpillingQueue<>();
    }
    Configuration configuration = new Configuration();
    queue.init(configuration,
        new TaskAttemptID(new TaskID("bla", 1, 1337), r.nextInt(100000)));
  }

  public void timeSpill(int reps) {
    for (int rep = 0; rep < reps; rep++) {
      int sum = 0;
      for (int i = 0; i < size; i++) {
        queue.add(new IntWritable(array[i]));
        sum += i;
      }
      System.out.println(sum);
    }

  }

  public static void main(String[] args) {
    Runner.main(SpillBenchmark.class, args);
  }
}