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

Added check to see if the driver is invalid

Browse Source
This commit is contained in:
workdone0 2024-04-24 19:37:51 +05:30
parent e09e0f5e99
commit b2651885e3
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
engine-core/src/main/java/io/nosqlbench/engine/core/lifecycle/session/CmdParser.java
View File

@ -16,8 +16,10 @@
package io.nosqlbench.engine.core.lifecycle.session;
import io.nosqlbench.adapter.diag.DriverAdapterLoader;
import io.nosqlbench.engine.cmdstream.Cmd;
import io.nosqlbench.engine.cmdstream.CmdArg;
import io.nosqlbench.nb.annotations.ServiceSelector;
import io.nosqlbench.nb.api.errors.BasicError;
import java.util.*;
@ -72,6 +74,13 @@ public class CmdParser {
cmdstructs.removeFirst();
Map<String,CmdArg> params = new LinkedHashMap<>();
while (cmdstructs.peekFirst() instanceof parameter param) {
if (Objects.equals(param.name(), "driver")) {
Optional<? extends DriverAdapterLoader> driverAdapter =
ServiceSelector.of(param.value(), ServiceLoader.load(DriverAdapterLoader.class)).get();
if (driverAdapter.isEmpty()) {
throw new BasicError("Unable to load default driver adapter '" + param.value() + '\'');
}
}
cmdstructs.removeFirst();
params.put(param.name(),CmdArg.of(cmd.name(),param.name(),param.op(),param.value()));
}