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