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

updated splitCommand method

Browse Source
This commit is contained in:
sahankj2000
2024-05-03 14:34:01 +05:30
parent 571b0f34f2
commit 1521887e89
1 changed files with 18 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
nb-engine/nb-engine-core/src/main/java/io/nosqlbench/engine/api/scenarios/NBCLIScenarioPreprocessor.java
View File

@@ -252,66 +252,31 @@ public class NBCLIScenarioPreprocessor {
private static final Pattern WordAndMaybeAssignment = Pattern.compile("(?<name>\\w[-_\\d\\w.]+)((?<oper>=+)(?<val>.+))?"); private static final Pattern WordAndMaybeAssignment = Pattern.compile("(?<name>\\w[-_\\d\\w.]+)((?<oper>=+)(?<val>.+))?");
public static List<String> splitCommand(String cmd) { public static String[] splitCommand(String cmd) {
List<String> cmdSplit = new ArrayList<>(); // split command by honoring single quotes, double quotes and escape characters
Stack<Character> stack = new Stack<>(); String[] namedStepPieces = cmd.split(" +(?=([^\"]*\"[^\"]*\")*[^\"]*$)(?=([^']*'[^']*')*[^']*$)");
StringBuilder builder = new StringBuilder(); Pattern pattern1 = Pattern.compile("(?<!\\\\)\"");
for (char c : cmd.toCharArray()) { Pattern pattern2 = Pattern.compile("(?<!\\\\)'");
switch (c) { for (int i = 0; i < namedStepPieces.length; i++) {
case ' ': // check if the quotes are balanced
if (!stack.isEmpty() && (stack.peek() == '\'' || stack.peek() == '\"')) { String stepPiece = namedStepPieces[i];
builder.append(c); boolean balanced = pattern1.matcher(stepPiece).results().count() % 2 == 0;
} else if (!builder.isEmpty()) { balanced = balanced && (pattern2.matcher(stepPiece).results().count() % 2 == 0);
cmdSplit.add(builder.toString()); if (!balanced) {
builder.delete(0, builder.length()); throw new BasicError("Unclosed quote found in scenario cmd '" + cmd + "'");
}
break;
case '\"':
case '\'':
if (!stack.isEmpty()) {
if (stack.peek() == c) {
stack.pop();
} else if (stack.peek() == '\\') {
stack.pop();
builder.append(c);
} else {
stack.push(c);
}
} else {
stack.push(c);
}
break;
case '\\':
if (!stack.isEmpty() && stack.peek() == c) {
stack.pop();
builder.append(c);
} else {
stack.push(c);
}
break;
default:
builder.append(c);
} }
// remove single quotes, double quotes and escape character
stepPiece = pattern1.matcher(stepPiece).replaceAll("");
stepPiece = pattern2.matcher(stepPiece).replaceAll("");
namedStepPieces[i] = stepPiece.replaceAll(Matcher.quoteReplacement("\\(?!\\)"), "");
} }
if (!stack.isEmpty()) { return namedStepPieces;
if (stack.peek() == '\'') {
throw new BasicError("Unclosed single quote found in scenario cmd '" + cmd + "'");
} else if (stack.peek() == '\"') {
throw new BasicError("Unclosed double quote found in scenario cmd '" + cmd + "'");
} else {
throw new BasicError("Unused escape character \"\\\" found in scenario cmd '" + cmd + "'");
}
}
if (!builder.isEmpty()) {
cmdSplit.add(builder.toString());
}
return cmdSplit;
} }
private static LinkedHashMap<String, SCNamedParam> parseStep(String cmd, String stepName, String scenarioName) { private static LinkedHashMap<String, SCNamedParam> parseStep(String cmd, String stepName, String scenarioName) {
LinkedHashMap<String, SCNamedParam> parsedStep = new LinkedHashMap<>(); LinkedHashMap<String, SCNamedParam> parsedStep = new LinkedHashMap<>();
List<String> namedStepPieces = splitCommand(cmd); String[] namedStepPieces = splitCommand(cmd);
for (String commandFragment : namedStepPieces) { for (String commandFragment : namedStepPieces) {
Matcher matcher = WordAndMaybeAssignment.matcher(commandFragment); Matcher matcher = WordAndMaybeAssignment.matcher(commandFragment);