Examples with NBConfigModel io.nosqlbench.nb.api.config.standard.NBConfigModel used on opensource projects

Example 1 with NBConfigModel

use of io.nosqlbench.nb.api.config.standard.NBConfigModel in project nosqlbench by nosqlbench.

the class StmtsDocList method getConfigModel.

public NBConfigModel getConfigModel() {
    ConfigModel cfgmodel = ConfigModel.of(StmtsDocList.class);
    getTemplateVariables().forEach((k, v) -> {
        cfgmodel.add(Param.defaultTo(k, v, "template parameter found in the yaml workload"));
    });
    return cfgmodel.asReadOnly();
}

Example 2 with NBConfigModel

use of io.nosqlbench.nb.api.config.standard.NBConfigModel in project nosqlbench by nosqlbench.

the class ActivityTypeLoader method getDriverAdapter.

private Optional<ActivityType> getDriverAdapter(String activityTypeName, ActivityDef activityDef) {
    Optional<DriverAdapter> oda = DRIVERADAPTER_SPI_FINDER.getOptionally(activityTypeName);
    if (oda.isPresent()) {
        DriverAdapter<?, ?> driverAdapter = oda.get();
        activityDef.getParams().remove("driver");
        if (driverAdapter instanceof NBConfigurable) {
            NBConfigModel cfgModel = ((NBConfigurable) driverAdapter).getConfigModel();
            Optional<String> op_yaml_loc = activityDef.getParams().getOptionalString("yaml", "workload");
            if (op_yaml_loc.isPresent()) {
                Map<String, Object> disposable = new LinkedHashMap<>(activityDef.getParams());
                StmtsDocList workload = StatementsLoader.loadPath(logger, op_yaml_loc.get(), disposable, "activities");
                cfgModel = cfgModel.add(workload.getConfigModel());
            }
            NBConfiguration cfg = cfgModel.apply(activityDef.getParams());
            ((NBConfigurable) driverAdapter).applyConfig(cfg);
        }
        ActivityType activityType = new StandardActivityType<>(driverAdapter, activityDef);
        return Optional.of(activityType);
    } else {
        return Optional.empty();
    }
}