IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-25 18:55:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

Add support for using a sizing function in AlphaNumericString()

Browse Source
This commit is contained in:
yabinmeng 2024-05-10 14:58:06 -05:00
parent b1a8e1117e
commit f4455558cf
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
nb-virtdata/virtdata-lib-basics/src/main/java/io/nosqlbench/virtdata/library/basics/shared/from_long/to_string/AlphaNumericString.java
View File

@ -19,9 +19,11 @@ package io.nosqlbench.virtdata.library.basics.shared.from_long.to_string;
import io.nosqlbench.virtdata.api.annotations.Categories;
import io.nosqlbench.virtdata.api.annotations.Category;
import io.nosqlbench.virtdata.api.annotations.ThreadSafeMapper;
import io.nosqlbench.virtdata.api.bindings.VirtDataConversions;
import io.nosqlbench.virtdata.library.basics.shared.from_long.to_long.Hash;
import java.util.function.LongFunction;
import java.util.function.LongToIntFunction;
/**
* Create an alpha-numeric string of the specified length, character-by-character.
@ -33,6 +35,8 @@ public class AlphaNumericString implements LongFunction<String> {
private final ThreadLocal<StringBuilder> threadStringBuilder = ThreadLocal.withInitial(StringBuilder::new);
private final Hash hash = new Hash();
private final int length;
private final LongToIntFunction sizeFunc;
public AlphaNumericString(int length)
{
@ -41,6 +45,17 @@ public class AlphaNumericString implements LongFunction<String> {
throw new RuntimeException("AlphaNumericString must have length >= 0");
}
this.length = length;
this.sizeFunc = (l) -> length;
}
public AlphaNumericString(Object sizefunc)
{
this.sizeFunc = VirtDataConversions.adaptFunction(sizefunc, LongToIntFunction.class);
this.length = this.sizeFunc.applyAsInt(0);
if (length < 0)
{
throw new RuntimeException("AlphaNumericString must have length >= 0");
}
}
@Override