mirror of
https://github.com/nosqlbench/nosqlbench.git
synced 2025-02-25 18:55:28 -06:00
Workloads can use a protocol
This commit is contained in:
parent
92bd0b4df9
commit
3ed2af6d1a
@ -20,6 +20,7 @@ import com.amazonaws.util.StringInputStream;
|
|||||||
import com.google.gson.GsonBuilder;
|
import com.google.gson.GsonBuilder;
|
||||||
import io.nosqlbench.nb.api.nbio.Content;
|
import io.nosqlbench.nb.api.nbio.Content;
|
||||||
import io.nosqlbench.nb.api.nbio.NBIO;
|
import io.nosqlbench.nb.api.nbio.NBIO;
|
||||||
|
import io.nosqlbench.nb.api.nbio.ParseProtocol;
|
||||||
import io.nosqlbench.nb.api.advisor.NBAdvisorException;
|
import io.nosqlbench.nb.api.advisor.NBAdvisorException;
|
||||||
import io.nosqlbench.nb.api.errors.BasicError;
|
import io.nosqlbench.nb.api.errors.BasicError;
|
||||||
import io.nosqlbench.adapters.api.activityconfig.rawyaml.RawOpsDocList;
|
import io.nosqlbench.adapters.api.activityconfig.rawyaml.RawOpsDocList;
|
||||||
@ -58,8 +59,8 @@ public class OpsLoader {
|
|||||||
|
|
||||||
public static OpsDocList loadPath(String path, Map<String, ?> params, String... searchPaths) {
|
public static OpsDocList loadPath(String path, Map<String, ?> params, String... searchPaths) {
|
||||||
String[] extensions = path.indexOf('.')>-1 ? new String[]{} : YAML_EXTENSIONS;
|
String[] extensions = path.indexOf('.')>-1 ? new String[]{} : YAML_EXTENSIONS;
|
||||||
|
ParseProtocol proto = new ParseProtocol(path);
|
||||||
Content<?> foundPath = NBIO.all().searchPrefixes(searchPaths).pathname(path).extensionSet(extensions).first()
|
Content<?> foundPath = NBIO.protocol(proto.getProtocols()).searchPrefixes(searchPaths).pathname(proto.getPath()).extensionSet(extensions).first()
|
||||||
.orElseThrow(() -> new RuntimeException("Unable to load path '" + path + "'"));
|
.orElseThrow(() -> new RuntimeException("Unable to load path '" + path + "'"));
|
||||||
OpTemplateFormat fmt = OpTemplateFormat.valueOfURI(foundPath.getURI());
|
OpTemplateFormat fmt = OpTemplateFormat.valueOfURI(foundPath.getURI());
|
||||||
return loadString(foundPath.asString(), fmt, params, foundPath.getURI());
|
return loadString(foundPath.asString(), fmt, params, foundPath.getURI());
|
||||||
|
Loading…
Reference in New Issue
Block a user