1/*
2 * Copyright (C) 2012 The Guava Authors
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.google.common.cache;
18
19import com.google.caliper.BeforeExperiment;
20import com.google.caliper.Benchmark;
21import com.google.common.base.Function;
22import com.google.common.base.Functions;
23import com.google.common.collect.MapMaker;
24
25import java.util.Map;
26
27/**
28 * Compare CacheBuilder and MapMaker performance, ensuring that they remain on par with each other.
29 *
30 * @author Nikita Sidorov
31 */
32public class MapMakerComparisonBenchmark {
33  private static final String TEST_KEY = "test key";
34  private static final String TEST_VALUE = "test value";
35
36  private static final Function<Object, Object> IDENTITY = Functions.identity();
37
38  // Loading/computing versions:
39  private final Map<Object, Object> computingMap = new MapMaker().makeComputingMap(IDENTITY);
40  private final LoadingCache<Object, Object> loadingCache =
41      CacheBuilder.newBuilder().recordStats().build(CacheLoader.from(IDENTITY));
42  private final LoadingCache<Object, Object> loadingCacheNoStats =
43      CacheBuilder.newBuilder().build(CacheLoader.from(IDENTITY));
44
45  // Non-loading versions:
46  private final Map<Object, Object> map = new MapMaker().makeMap(); // Returns ConcurrentHashMap
47  private final Cache<Object, Object> cache = CacheBuilder.newBuilder().recordStats().build();
48  private final Cache<Object, Object> cacheNoStats = CacheBuilder.newBuilder().build();
49
50  @BeforeExperiment
51  void setUp() {
52    map.put(TEST_KEY, TEST_VALUE);
53    cache.put(TEST_KEY, TEST_VALUE);
54    cacheNoStats.put(TEST_KEY, TEST_VALUE);
55  }
56
57  @Benchmark void computingMapMaker(int rep) {
58    for (int i = 0; i < rep; i++) {
59      computingMap.get(TEST_KEY);
60    }
61  }
62
63  @Benchmark void loadingCacheBuilder_stats(int rep) {
64    for (int i = 0; i < rep; i++) {
65      loadingCache.getUnchecked(TEST_KEY);
66    }
67  }
68
69  @Benchmark void loadingCacheBuilder(int rep) {
70    for (int i = 0; i < rep; i++) {
71      loadingCacheNoStats.getUnchecked(TEST_KEY);
72    }
73  }
74
75  @Benchmark void concurrentHashMap(int rep) {
76    for (int i = 0; i < rep; i++) {
77      map.get(TEST_KEY);
78    }
79  }
80
81  @Benchmark void cacheBuilder_stats(int rep) {
82    for (int i = 0; i < rep; i++) {
83      cache.getIfPresent(TEST_KEY);
84    }
85  }
86
87  @Benchmark void cacheBuilder(int rep) {
88    for (int i = 0; i < rep; i++) {
89      cacheNoStats.getIfPresent(TEST_KEY);
90    }
91  }
92}
93