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

adding changes

Browse Source
This commit is contained in:
sahankj2000 2024-05-08 16:51:17 +05:30
parent dead305986
commit 2dad5d0d3d
2 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nb-virtdata/virtdata-lang/src/main/java/io/nosqlbench/virtdata/lang/grammars/VirtData.g4
View File

@ -3,7 +3,7 @@ grammar VirtData;
virtdataRecipe : virtdataFlow (specend virtdataFlow?)* EOF ; virtdataRecipe : virtdataFlow (specend virtdataFlow?)* EOF ;
virtdataFlow : (COMPOSE)? expression (';' expression?)* ; virtdataFlow : (COMPOSE)? expression (';' expression? | WS? expression)* ;
expression : (lvalue ASSIGN)? virtdataCall ; expression : (lvalue ASSIGN)? virtdataCall ;

25
nb-virtdata/virtdata-lang/src/main/java/io/nosqlbench/virtdata/lang/parser/VirtDataDSL.java
View File

@ -20,9 +20,7 @@ import io.nosqlbench.virtdata.lang.ast.VirtDataAST;
import io.nosqlbench.virtdata.lang.ast.VirtDataFlow; import io.nosqlbench.virtdata.lang.ast.VirtDataFlow;
import io.nosqlbench.virtdata.lang.generated.VirtDataLexer; import io.nosqlbench.virtdata.lang.generated.VirtDataLexer;
import io.nosqlbench.virtdata.lang.generated.VirtDataParser; import io.nosqlbench.virtdata.lang.generated.VirtDataParser;
import org.antlr.v4.runtime.CharStreams; import org.antlr.v4.runtime.*;
import org.antlr.v4.runtime.CodePointCharStream;
import org.antlr.v4.runtime.CommonTokenStream;
import org.apache.logging.log4j.LogManager; import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger; import org.apache.logging.log4j.Logger;
@ -36,14 +34,11 @@ public class VirtDataDSL {
try { try {
CodePointCharStream cstream = CharStreams.fromString(input); CodePointCharStream cstream = CharStreams.fromString(input);
VirtDataLexer lexer = new VirtDataLexer(cstream); VirtDataParser parser = getVirtDataParser(cstream);
CommonTokenStream tokens = new CommonTokenStream(lexer);
VirtDataParser parser = new VirtDataParser(tokens);
VirtDataBuilder astListener = new VirtDataBuilder(); VirtDataBuilder astListener = new VirtDataBuilder();
parser.addParseListener(astListener); parser.addParseListener(astListener);
VirtDataParser.VirtdataRecipeContext virtdataRecipeContext = parser.virtdataRecipe();
VirtDataParser.VirtdataFlowContext virtdataFlowContext = parser.virtdataFlow(); logger.trace(() -> "parse tree: " + virtdataRecipeContext.toStringTree(parser));
logger.trace(() -> "parse tree: " + virtdataFlowContext.toStringTree(parser));
if (astListener.hasErrors()) { if (astListener.hasErrors()) {
System.out.println(astListener.getErrorNodes()); System.out.println(astListener.getErrorNodes());
@ -67,6 +62,18 @@ public class VirtDataDSL {
} }
} }
private static VirtDataParser getVirtDataParser(CodePointCharStream cstream) {
VirtDataLexer lexer = new VirtDataLexer(cstream);
lexer.addErrorListener(new BaseErrorListener() {
@Override
public void syntaxError(Recognizer<?, ?> recognizer, Object offendingSymbol, int line, int charPositionInLine, String msg, RecognitionException e) {
throw new RuntimeException(e);
}
});
CommonTokenStream tokens = new CommonTokenStream(lexer);
return new VirtDataParser(tokens);
}
public static class ParseResult { public static class ParseResult {
public Throwable throwable; public Throwable throwable;
public VirtDataFlow flow; public VirtDataFlow flow;