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

subops

Browse Source
This commit is contained in:
Jonathan Shook
2024-02-08 16:18:14 -06:00
parent 99a1b287e2
commit 2028293d8f
1 changed files with 71 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
adapters-api/src/main/java/io/nosqlbench/adapters/api/templating/ParsedOp.java
View File

@@ -844,6 +844,77 @@ public class ParsedOp extends NBBaseComponent implements LongFunction<Map<String
return lfa;
}
private ParsedOp makeSubOp(String fromOpField, String elemName, String stringOpField) {
return makeSubOp(fromOpField, elemName, new LinkedHashMap(Map.of("stmt", stringOpField)));
}
private ParsedOp makeSubOp(String fromOpField, String elemName, Map<String, Object> opfields) {
String subopName = opfields.containsKey("name") ? opfields.get("name").toString() : fromOpField + "_" + elemName;
return new ParsedOp(
new OpData(
"sub-op of '" + this.getName() + "' field '" + fromOpField + "', element '" + elemName + "' name '" + subopName + "'",
subopName,
new LinkedHashMap<String, String>(_opTemplate.getTags()) {{
put("subop", subopName);
}},
_opTemplate.getBindings(),
_opTemplate.getParams(),
opfields
),
this.activityCfg,
List.of(),
this
);
}
public Map<String, ParsedOp> getAsSubOps(String fromOpField) {
Map<String, ParsedOp> subOpMap = new LinkedHashMap<>();
Object o = getStaticValue(fromOpField);
if (o instanceof List list) {
String format = "subop%0" + ((int) Math.log10(list.size()) + 1) + "d";
for (int i = 0; i < list.size(); i++) {
Object listElem = list.get(i);
if (listElem instanceof Map lmap) {
ParsedOp parsedOp = makeSubOp(fromOpField, String.format(format, i), lmap);
subOpMap.put(parsedOp.getName(), parsedOp);
} else if (listElem instanceof CharSequence stmt) {
ParsedOp parsedOp = makeSubOp(fromOpField, String.format(format, i), stmt.toString());
subOpMap.put(parsedOp.getName(), parsedOp);
} else {
throw new OpConfigError("For sub-ops field " + fromOpField + " of op '" + this.getName() + "', element " +
"types must be of Map or String, not '" + o.getClass().getCanonicalName() + "'");
}
}
} else if (o instanceof Map map) {
for (Object nameKey : map.keySet()) {
Object opref = map.get(nameKey);
if (opref instanceof Map fmap) {
ParsedOp subOp = makeSubOp(fromOpField, nameKey.toString(), fmap);
subOpMap.put(subOp.getName(), subOp);
} else if (opref instanceof CharSequence stmt) {
ParsedOp subOp = makeSubOp(fromOpField, nameKey.toString(), stmt.toString());
subOpMap.put(subOp.getName(), subOp);
} else {
throw new OpConfigError("For sub-ops field " + fromOpField + " of op '" + this.getName() + "', element " +
"types must be of Map or String, not '" + o.getClass().getCanonicalName() + "'");
}
}
} else {
throw new OpConfigError("invalid subtype for mapping sub operations in op " + fromOpField + ":" + o.getClass().getCanonicalName());
}
return subOpMap;
}
public ParsedOp getAsSubOp(String name) {
Object o = getStaticValue(name);
if (o instanceof Map map) {
return makeSubOp(name, name, map);
} else {
throw new RuntimeException("Not allowed: op field named '" + name + "' as sub-op");
}
}
/**
* <p>Enhance an {@link Optional} {@link Function} with an optional named field or value combiner,