aboutsummaryrefslogtreecommitdiffstats
path: root/guava-tests/benchmark/com/google/common/hash/HashBenchmark.java
blob: 20272184f0f48eaeb06d0b54420ebfe6517cbbb3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
/*
 * Copyright (C) 2011 The Guava 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 com.google.common.hash;

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

import java.util.Random;

/**
 * Benchmarks for hashing functions. This class benchmarks various hasing functions for a range of
 * sizes of byte array. The input data is generated by a call to {@link Random#nextBytes}.
 *
 * <p>Parameters for the benchmark are:
 * <ul>
 * <li>size: The length of the byte array to hash.
 * <li>function: The name of the function(s) to test (eg, "goodFastHash32" or "murmur3_32")
 * </ul>
 *
 * @author David Beaumont
 */
public class HashBenchmark extends SimpleBenchmark {

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

  @Param private HashType function;

  private enum HashType {
    goodFastHash32() {
      @Override public long hash(byte[] data) {
        return Hashing.goodFastHash(32).hashBytes(data).asInt();
      }
    },
    goodFastHash64() {
      @Override public long hash(byte[] data) {
        return Hashing.goodFastHash(64).hashBytes(data).asLong();
      }
    },
    murmur32() {
      @Override public long hash(byte[] data) {
        return Hashing.murmur3_32().hashBytes(data).asInt();
      }
    },
    murmur128() {
      @Override public long hash(byte[] data) {
        return Hashing.murmur3_128().hashBytes(data).asLong();
      }
    },
    md5() {
      @Override public long hash(byte[] data) {
        return Hashing.md5().hashBytes(data).asLong();
      }
    };
    public abstract long hash(byte[] data);
  }

  private byte[] testData;

  @Override
  protected void setUp() {
    testData = new byte[size];
    new Random().nextBytes(testData);
  }

  public int timeHashFunction(int reps) {
    long dummy = 0;
    for (int i = 0; i < reps; i++) {
      dummy ^= function.hash(testData);
    }
    return (int) dummy;
  }

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