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 15:34:18 -05:00
parent f4455558cf
commit f22c7f84db
1 changed files with 11 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -34,33 +34,28 @@ public class AlphaNumericString implements LongFunction<String> {
private static final String AVAILABLE_CHARS = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";
private final ThreadLocal<StringBuilder> threadStringBuilder = ThreadLocal.withInitial(StringBuilder::new);
private final Hash hash = new Hash();
private final int length;
private final LongToIntFunction sizeFunc;
private final LongToIntFunction lengthFunc;
public AlphaNumericString(int length)
{
if (length < 0)
{
throw new RuntimeException("AlphaNumericString must have length >= 0");
}
this.length = length;
this.sizeFunc = (l) -> length;
this.lengthFunc = (l) -> length;
}
public AlphaNumericString(Object sizefunc)
public AlphaNumericString(Object lengthfunc)
{
this.sizeFunc = VirtDataConversions.adaptFunction(sizefunc, LongToIntFunction.class);
this.length = this.sizeFunc.applyAsInt(0);
if (length < 0)
{
throw new RuntimeException("AlphaNumericString must have length >= 0");
}
this.lengthFunc = VirtDataConversions.adaptFunction(lengthfunc, LongToIntFunction.class);
}
@Override
public String apply(long operand)
{
int length = lengthFunc.applyAsInt(operand);
if (length < 0)
{
throw new RuntimeException("AlphaNumericString must have length >= 0");
}
long hashValue = operand;
StringBuilder sb = threadStringBuilder.get();
sb.setLength(0);
@ -76,6 +71,6 @@ public class AlphaNumericString implements LongFunction<String> {
@Override
public String toString()
{
return "AlphaNumericString(length=" + length + ")";
return "AlphaNumericString(lengthFunc.class=" + lengthFunc.getClass().getSimpleName() + ")";
}
}