IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-04 11:50:34 -06:00
Code Issues Packages Projects Releases Wiki Activity

make script params work with graaljs

Browse Source
This commit is contained in:
Jonathan Shook 2020-08-06 09:50:19 -05:00
parent a533901349
commit e2505f717a
1 changed files with 97 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

120
engine-core/src/main/java/io/nosqlbench/engine/core/script/ScriptParams.java
View File

@ -21,70 +21,98 @@ import ch.qos.logback.classic.Logger;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import io.nosqlbench.nb.api.errors.BasicError;
import org.graalvm.polyglot.Value;
import org.graalvm.polyglot.proxy.Proxy;
import org.graalvm.polyglot.proxy.ProxyObject;
import org.slf4j.LoggerFactory;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.Flow;
import java.util.function.Function;
import java.util.stream.Collectors;
/**
* Wrap the script parameters in a type which allows for easy manipulation
*/
public class ScriptParams extends HashMap<String, String> {
public class ScriptParams extends HashMap<String, String> implements ProxyObject {
private static final Logger logger = (Logger) LoggerFactory.getLogger(ScriptParams.class);
private static final Gson gson = new GsonBuilder().setPrettyPrinting().create();
public Map<String,String> withOverrides(Map<String, String> overrides) {
public ScriptParams withOverrides(Object overrides) {
Map<String, String> map;
if (overrides instanceof Map) {
map = (Map) overrides;
} else if (overrides instanceof Value) {
Value v = (Value) overrides;
map = v.as(Map.class);
} else {
throw new RuntimeException("Unrecognized overrides type: " + overrides.getClass().getCanonicalName());
}
if (overrides == null) {
logger.warn("A null map was provided to withOverrides. This could be a bug in your script.");
overrides=Map.of();
overrides = Map.of();
}
checkForNulls("params", "calling withOverrides", this);
checkForNulls("overrides", "calling withOverrides", overrides);
checkForNulls("overrides", "calling withOverrides", map);
ScriptParams result = new ScriptParams();
result.putAll(this);
for (Entry<String, String> overrideEntry : overrides.entrySet()) {
for (Entry<String, String> overrideEntry : map.entrySet()) {
String oKey = overrideEntry.getKey();
String oVal = overrideEntry.getValue();
if (oVal.toUpperCase().endsWith("UNDEF")) {
String removed = result.remove(oKey);
logger.trace("Removed key '" + oKey + "': '" + removed + "' from script params because it was "+ oVal + " in overrides");
logger.trace("Removed key '" + oKey + "': '" + removed + "' from script params because it was " + oVal + " in overrides");
} else {
String was = result.get(oKey);
was = (was==null ? "NULL" : was);
result.put(oKey,oVal);
was = (was == null ? "NULL" : was);
result.put(oKey, oVal);
logger.trace("Overrode key '" + oKey + "': from '" + was + " to " + oVal);
}
}
return result;
ScriptParams p = new ScriptParams();
p.putAll(result);
return p;
}
public Map<String,String> withDefaults(Map<String, String> defaults) {
if (defaults == null) {
public ScriptParams withDefaults(Object defaults) {
Map<String, String> map;
if (defaults instanceof Map) {
map = (Map) defaults;
} else if (defaults instanceof Value) {
map = ((Value) defaults).as(Map.class);
} else {
throw new RuntimeException("Unrecognized type for defaults: " + defaults.getClass().getCanonicalName());
}
if (map == null) {
logger.warn("A null map was provided to withDefaults. This could be a bug in your script.");
defaults=Map.of();
map = Map.of();
}
ScriptParams result = new ScriptParams();
checkForNulls("params", "calling withDefaults", this);
checkForNulls("defaults", "calling withDefaults", defaults);
result.putAll(defaults);
checkForNulls("defaults", "calling withDefaults", map);
result.putAll(map);
result.putAll(this);
return result;
ScriptParams p = new ScriptParams();
p.putAll(result);
return p;
}
private static void checkForNulls(String name, String action, Map<String, String> map) {
for (String s : map.keySet()) {
if (s == null) {
printMapToLog(name,map);
printMapToLog(name, map);
throw new BasicError("Found a null key in " + name + " while " + action + ". Please ensure that you " +
"only provide non-null keys and values in parameter maps.");
"only provide non-null keys and values in parameter maps.");
} else if (map.get(s) == null) {
printMapToLog(name,map);
printMapToLog(name, map);
throw new BasicError("Found a null value for key '" + s + "' in " + name + " while " + action + ". " +
"Please ensure you provide non-null keys and values in parameter maps.");
"Please ensure you provide non-null keys and values in parameter maps.");
}
}
}
@ -92,15 +120,15 @@ public class ScriptParams extends HashMap<String, String> {
private static void printMapToLog(String name, Map<String, String> map) {
logger.info("contents of map '" + name + "':");
String mapdetail = map.entrySet()
.stream()
.map(e -> valueOf(e.getKey()) + ":" + valueOf(e.getValue()))
.collect(Collectors.joining(","));
.stream()
.map(e -> valueOf(e.getKey()) + ":" + valueOf(e.getValue()))
.collect(Collectors.joining(","));
logger.info(mapdetail);
}
@Override
public String toString() {
return gson.toJson(this,Map.class).toString();
return gson.toJson(this, Map.class).toString();
}
private static String valueOf(Object o) {
@ -110,4 +138,50 @@ public class ScriptParams extends HashMap<String, String> {
return String.valueOf(o);
}
//region ProxyObject methods
@Override
public Object getMember(String key) {
if (key.equals("withDefaults")) {
return (Function<Object, Object>) this::withDefaults;
}
if (key.equals("withOverrides")) {
return (Function<Object, Object>) this::withOverrides;
}
return super.get(key);
}
@Override
public boolean containsKey(Object key) {
return
key.equals("withOverrides")
|| key.equals("withDefaults")
|| super.containsKey(key);
}
@Override
public Object getMemberKeys() {
ArrayList<String> memberKeys = new ArrayList<>(super.keySet());
memberKeys.add("withDefaults");
memberKeys.add("withOverride");
return memberKeys;
}
@Override
public boolean hasMember(String key) {
if (super.containsKey(key)) {
return true;
}
if (key.equals("withOverrides") || key.equals("withDefaults")) {
return true;
}
return false;
}
@Override
public void putMember(String key, Value value) {
super.put(key, value.asString());
}
//endregion
}