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

allow ops to layer labels within activity labels

Browse Source
This commit is contained in:
Jonathan Shook
2023-08-28 16:15:21 -05:00
parent df5076e8e9
commit def7e99ed8
1 changed files with 17 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -17,6 +17,7 @@
package io.nosqlbench.adapters.api.templating; package io.nosqlbench.adapters.api.templating;
import io.nosqlbench.adapters.api.activityconfig.yaml.OpTemplate; import io.nosqlbench.adapters.api.activityconfig.yaml.OpTemplate;
import io.nosqlbench.api.config.NBLabelSpec;
import io.nosqlbench.api.config.NBLabeledElement; import io.nosqlbench.api.config.NBLabeledElement;
import io.nosqlbench.api.config.NBLabels; import io.nosqlbench.api.config.NBLabels;
import io.nosqlbench.api.config.fieldreaders.DynamicFieldReader; import io.nosqlbench.api.config.fieldreaders.DynamicFieldReader;
@@ -328,19 +329,19 @@ public class ParsedOp implements LongFunction<Map<String, ?>>, NBLabeledElement,
* The activity configuration, used to resolve nested config parameters * The activity configuration, used to resolve nested config parameters
* @param preprocessors * @param preprocessors
* Map->Map transformers. * Map->Map transformers.
* @param labels
*/ */
public ParsedOp( public ParsedOp(
OpTemplate opTemplate, OpTemplate opTemplate,
NBConfiguration activityCfg, NBConfiguration activityCfg,
List<Function<Map<String, Object>, Map<String, Object>>> preprocessors, List<Function<Map<String, Object>, Map<String, Object>>> preprocessors,
NBLabeledElement parent) { NBLabeledElement parent
) {
this._opTemplate = opTemplate; this._opTemplate = opTemplate;
this.activityCfg = activityCfg; this.activityCfg = activityCfg;
labels=parent.getLabels().and("op", this.getName());
Map<String, Object> map = opTemplate.getOp().orElseThrow(() -> Map<String, Object> map = opTemplate.getOp().orElseThrow(() ->
new OpConfigError("ParsedOp constructor requires a non-null value for the op field, but it was missing.")); new OpConfigError("ParsedOp constructor requires a non-null value for the op field, but it was missing."));
for (Function<Map<String, Object>, Map<String, Object>> preprocessor : preprocessors) { for (Function<Map<String, Object>, Map<String, Object>> preprocessor : preprocessors) {
map = preprocessor.apply(map); map = preprocessor.apply(map);
} }
@@ -353,6 +354,19 @@ public class ParsedOp implements LongFunction<Map<String, ?>>, NBLabeledElement,
activityCfg.getMap()) activityCfg.getMap())
); );
NBLabels opLabels = parent.getLabels().andTypes("op", this.getName());
if (tmap.isStatic("labels")) {
Object labelSpecObject = tmap.takeStaticValue("labels", Object.class);
if (labelSpecObject instanceof String labelsSpec) {
NBLabels op_specific_labels = NBLabelSpec.parseLabels(labelsSpec);
opLabels=opLabels.and(op_specific_labels);
} else {
throw new OpConfigError("parsing labels as type '" + labelSpecObject.getClass().getSimpleName() +"' is not supported.");
}
} else if (tmap.isDynamic("labels")) {
throw new OpConfigError("Labels may not be dynamic.");
}
this.labels=opLabels;
} }