-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add benchmark to measure performance of CustomBinaryDocValuesField
Benchmarks show that ArrayList performs better than TreeSet. Added a comment on where to find the results and benchmark. Signed-off-by: Kiran Reddy <kkreddy@amazon.com>
- Loading branch information
Showing
2 changed files
with
85 additions
and
0 deletions.
There are no files selected for viewing
81 changes: 81 additions & 0 deletions
81
.../main/java/org/opensearch/benchmark/index/mapper/CustomBinaryDocValuesFieldBenchmark.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,81 @@ | ||
/* | ||
* SPDX-License-Identifier: Apache-2.0 | ||
* | ||
* The OpenSearch Contributors require contributions made to | ||
* this file be licensed under the Apache-2.0 license or a | ||
* compatible open source license. | ||
*/ | ||
|
||
package org.opensearch.benchmark.index.mapper; | ||
|
||
import org.apache.lucene.util.BytesRef; | ||
import org.openjdk.jmh.annotations.Benchmark; | ||
import org.openjdk.jmh.annotations.BenchmarkMode; | ||
import org.openjdk.jmh.annotations.Fork; | ||
import org.openjdk.jmh.annotations.Measurement; | ||
import org.openjdk.jmh.annotations.Mode; | ||
import org.openjdk.jmh.annotations.OutputTimeUnit; | ||
import org.openjdk.jmh.annotations.Param; | ||
import org.openjdk.jmh.annotations.Scope; | ||
import org.openjdk.jmh.annotations.Setup; | ||
import org.openjdk.jmh.annotations.State; | ||
import org.openjdk.jmh.annotations.Warmup; | ||
import org.openjdk.jmh.infra.Blackhole; | ||
import org.opensearch.index.mapper.BinaryFieldMapper; | ||
|
||
import java.util.concurrent.ThreadLocalRandom; | ||
import java.util.concurrent.TimeUnit; | ||
|
||
|
||
@Warmup(iterations = 1) | ||
@Measurement(iterations = 1) | ||
@Fork(1) | ||
@BenchmarkMode(Mode.Throughput) | ||
@OutputTimeUnit(TimeUnit.MILLISECONDS) | ||
@State(Scope.Thread) | ||
@SuppressWarnings("unused") // invoked by benchmarking framework | ||
public class CustomBinaryDocValuesFieldBenchmark { | ||
|
||
static final String FIELD_NAME = "dummy"; | ||
static final String SEED_VALUE = "seed"; | ||
|
||
@Benchmark | ||
public void add(CustomBinaryDocValuesFieldBenchmark.BenchmarkParameters parameters, Blackhole blackhole) { | ||
// Don't use the parameter binary doc values object. | ||
// Start with a fresh object every call and add maximum number of entries | ||
BinaryFieldMapper.CustomBinaryDocValuesField customBinaryDocValuesField = | ||
new BinaryFieldMapper.CustomBinaryDocValuesField(FIELD_NAME, new BytesRef(SEED_VALUE).bytes); | ||
for (int i = 0; i < parameters.maximumNumberOfEntries; ++i) { | ||
ThreadLocalRandom.current().nextBytes(parameters.bytes); | ||
customBinaryDocValuesField.add(parameters.bytes); | ||
} | ||
} | ||
|
||
@Benchmark | ||
public void binaryValue(CustomBinaryDocValuesFieldBenchmark.BenchmarkParameters parameters, Blackhole blackhole) { | ||
blackhole.consume(parameters.customBinaryDocValuesField.binaryValue()); | ||
} | ||
|
||
@State(Scope.Benchmark) | ||
public static class BenchmarkParameters { | ||
@Param({"8", "32", "128", "512"}) | ||
int maximumNumberOfEntries; | ||
|
||
@Param({"8", "32", "128", "512"}) | ||
int entrySize; | ||
|
||
BinaryFieldMapper.CustomBinaryDocValuesField customBinaryDocValuesField; | ||
byte[] bytes; | ||
|
||
@Setup | ||
public void setup() { | ||
customBinaryDocValuesField = new BinaryFieldMapper.CustomBinaryDocValuesField(FIELD_NAME, | ||
new BytesRef(SEED_VALUE).bytes); | ||
bytes = new byte[entrySize]; | ||
for (int i = 0; i < maximumNumberOfEntries; ++i) { | ||
ThreadLocalRandom.current().nextBytes(bytes); | ||
customBinaryDocValuesField.add(bytes); | ||
} | ||
} | ||
} | ||
} |
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