forked from apache/flink-benchmarks
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
愚鲤
committed
Aug 2, 2022
1 parent
5d2f5a9
commit 5c4fab9
Showing
3 changed files
with
151 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
85 changes: 85 additions & 0 deletions
85
src/main/java/org/apache/flink/benchmark/GuavaCacheBenchmark.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,85 @@ | ||
package org.apache.flink.benchmark; | ||
|
||
|
||
import org.apache.flink.state.benchmark.ListStateBenchmark; | ||
import org.apache.flink.table.shaded.com.jayway.jsonpath.JsonPath; | ||
import org.apache.flink.table.shaded.com.jayway.jsonpath.spi.cache.Cache; | ||
import org.apache.flink.table.shaded.com.jayway.jsonpath.spi.cache.CacheProvider; | ||
import org.apache.flink.table.shaded.com.jayway.jsonpath.spi.cache.LRUCache; | ||
|
||
import org.apache.flink.shaded.guava30.com.google.common.cache.CacheBuilder; | ||
|
||
import org.openjdk.jmh.annotations.Benchmark; | ||
import org.openjdk.jmh.annotations.Setup; | ||
import org.openjdk.jmh.annotations.Threads; | ||
import org.openjdk.jmh.runner.Runner; | ||
import org.openjdk.jmh.runner.options.Options; | ||
import org.openjdk.jmh.runner.options.OptionsBuilder; | ||
import org.openjdk.jmh.runner.options.VerboseMode; | ||
|
||
import java.util.ArrayList; | ||
import java.util.List; | ||
import java.util.concurrent.ThreadLocalRandom; | ||
|
||
public class GuavaCacheBenchmark extends BenchmarkBase { | ||
|
||
final ThreadLocalRandom random = ThreadLocalRandom.current(); | ||
Cache cache; | ||
Cache readCache = new JsonPathCache(); | ||
JsonPath jsonPath; | ||
List<String> randomKey = new ArrayList<>(); | ||
int cacheSize = 400; | ||
|
||
public static void main(String[] args) throws Exception { | ||
Options opt = | ||
new OptionsBuilder() | ||
.verbosity(VerboseMode.NORMAL) | ||
.include(".*" + GuavaCacheBenchmark.class.getCanonicalName() + ".*") | ||
.build(); | ||
|
||
new Runner(opt).run(); | ||
} | ||
|
||
@Setup | ||
public void setUp() { | ||
cache = new JsonPathCache(); | ||
this.jsonPath = JsonPath.compile("$.store.book[1]"); | ||
for (int i = 0; i < cacheSize; i++) { | ||
String key = random.nextLong() + ""; | ||
randomKey.add(key); | ||
readCache.put(key, jsonPath); | ||
} | ||
} | ||
|
||
@Benchmark | ||
@Threads(4) | ||
public void get() { | ||
readCache.get(randomKey.get(random.nextInt(cacheSize))); | ||
} | ||
|
||
|
||
@Benchmark | ||
@Threads(4) | ||
public void put() { | ||
cache.put(randomKey.get(random.nextInt(cacheSize)), jsonPath); | ||
} | ||
|
||
public class JsonPathCache implements Cache { | ||
|
||
private static final long DEFAULT_CACHE_MAXIMUM_SIZE = 400; | ||
|
||
private final org.apache.flink.shaded.guava30.com.google.common.cache.Cache<String, JsonPath> | ||
jsonPathCache = | ||
CacheBuilder.newBuilder().maximumSize(DEFAULT_CACHE_MAXIMUM_SIZE).build(); | ||
|
||
@Override | ||
public JsonPath get(String s) { | ||
return jsonPathCache.getIfPresent(s); | ||
} | ||
|
||
@Override | ||
public void put(String s, JsonPath jsonPath) { | ||
jsonPathCache.put(s, jsonPath); | ||
} | ||
} | ||
} |
65 changes: 65 additions & 0 deletions
65
src/main/java/org/apache/flink/benchmark/LRUCacheBenchmark.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,65 @@ | ||
package org.apache.flink.benchmark; | ||
|
||
import org.apache.flink.state.benchmark.ListStateBenchmark; | ||
import org.apache.flink.table.shaded.com.jayway.jsonpath.JsonPath; | ||
import org.apache.flink.table.shaded.com.jayway.jsonpath.spi.cache.Cache; | ||
import org.apache.flink.table.shaded.com.jayway.jsonpath.spi.cache.CacheProvider; | ||
import org.apache.flink.table.shaded.com.jayway.jsonpath.spi.cache.LRUCache; | ||
|
||
import org.openjdk.jmh.annotations.Benchmark; | ||
import org.openjdk.jmh.annotations.Fork; | ||
import org.openjdk.jmh.annotations.Setup; | ||
import org.openjdk.jmh.annotations.Threads; | ||
import org.openjdk.jmh.runner.Runner; | ||
import org.openjdk.jmh.runner.options.Options; | ||
import org.openjdk.jmh.runner.options.OptionsBuilder; | ||
import org.openjdk.jmh.runner.options.VerboseMode; | ||
|
||
import java.util.ArrayList; | ||
import java.util.List; | ||
import java.util.concurrent.ThreadLocalRandom; | ||
|
||
public class LRUCacheBenchmark extends BenchmarkBase { | ||
|
||
final ThreadLocalRandom random = ThreadLocalRandom.current(); | ||
Cache cache; | ||
JsonPath jsonPath; | ||
List<String> randomKey = new ArrayList<>(); | ||
int cacheSize = 400; | ||
Cache readCache = new LRUCache(cacheSize); | ||
|
||
public static void main(String[] args) throws Exception { | ||
Options opt = | ||
new OptionsBuilder() | ||
.verbosity(VerboseMode.NORMAL) | ||
.include(".*" + LRUCacheBenchmark.class.getCanonicalName() + ".*") | ||
.build(); | ||
|
||
new Runner(opt).run(); | ||
} | ||
|
||
@Setup | ||
public void setUp() { | ||
cache = new LRUCache(cacheSize); | ||
this.jsonPath = JsonPath.compile("$.store.book[1]"); | ||
for (int i = 0; i < cacheSize; i++) { | ||
String key = random.nextLong() + ""; | ||
randomKey.add(key); | ||
readCache.put(key, jsonPath); | ||
} | ||
} | ||
|
||
@Benchmark | ||
@Threads(4) | ||
public void get() { | ||
readCache.get(randomKey.get(random.nextInt(cacheSize))); | ||
} | ||
|
||
|
||
@Benchmark | ||
@Threads(4) | ||
public void put() { | ||
cache.put(randomKey.get(random.nextInt(cacheSize)), jsonPath); | ||
} | ||
|
||
} |