forked from senseidb/sensei
-
Notifications
You must be signed in to change notification settings - Fork 0
/
BQLCompiler.java
116 lines (102 loc) · 4 KB
/
BQLCompiler.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
package com.senseidb.bql.parsers;
import java.util.HashMap;
import java.util.Map;
import org.antlr.v4.runtime.ANTLRInputStream;
import org.antlr.v4.runtime.BailErrorStrategy;
import org.antlr.v4.runtime.CommonTokenStream;
import org.antlr.v4.runtime.FailedPredicateException;
import org.antlr.v4.runtime.InputMismatchException;
import org.antlr.v4.runtime.NoViableAltException;
import org.antlr.v4.runtime.Token;
import org.antlr.v4.runtime.TokenStream;
import org.antlr.v4.runtime.misc.ParseCancellationException;
import org.antlr.v4.runtime.tree.ParseTreeWalker;
import org.json.JSONObject;
public class BQLCompiler extends AbstractCompiler {
// A map containing facet type and data type info for a facet
private Map<String, String[]> _facetInfoMap = new HashMap<String, String[]>();
private final ThreadLocal<BQLParser> _parser = new ThreadLocal<BQLParser>();
public BQLCompiler(Map<String, String[]> facetInfoMap) {
_facetInfoMap = facetInfoMap;
}
@Override
public JSONObject compile(String bqlStmt) {
// Lexer splits input into tokens
ANTLRInputStream input = new ANTLRInputStream(bqlStmt);
TokenStream tokens = new CommonTokenStream(new BQLLexer(input));
// Parser generates abstract syntax tree
BQLParser parser = new BQLParser(tokens);
_parser.set(parser);
parser.removeErrorListeners();
parser.addErrorListener(BQLErrorListener.INSTANCE);
parser.setErrorHandler(new BailErrorStrategy());
BQLParser.StatementContext ret = parser.statement();
BQLCompilerAnalyzer analyzer = new BQLCompilerAnalyzer(parser, _facetInfoMap);
ParseTreeWalker.DEFAULT.walk(analyzer, ret);
JSONObject json = (JSONObject) analyzer.getJsonProperty(ret);
return json;
}
@Override
public String getErrorMessage(IllegalStateException ex) {
String errMsg = null;
if (ex instanceof ParseCancellationException) {
BQLParser parser = _parser.get();
Throwable err = ex.getCause();
if (err instanceof NoViableAltException) {
NoViableAltException nvae = (NoViableAltException) err;
int line = nvae.getOffendingToken().getLine();
TokenStream tokens = parser.getInputStream();
String input;
if (tokens != null) {
if (nvae.getStartToken().getType() == Token.EOF) {
input = "<EOF>";
} else {
input = tokens.getText(nvae.getStartToken(), nvae.getOffendingToken());
}
} else {
input = "<unknown input>";
}
errMsg = "[line " + line + "] no viable alternative at input " + escapeWSAndQuote(input);
} else if (err instanceof InputMismatchException) {
InputMismatchException ime = (InputMismatchException) err;
int line = ime.getOffendingToken().getLine();
errMsg = "[line " + line + "] mismatched input "
+ getTokenErrorDisplay(ime.getOffendingToken()) + " expecting "
+ ime.getExpectedTokens().toString(parser.getTokenNames());
} else if (err instanceof FailedPredicateException) {
FailedPredicateException fpe = (FailedPredicateException) err;
int line = fpe.getOffendingToken().getLine();
String ruleName = parser.getRuleNames()[parser.getContext().getRuleIndex()];
errMsg = "[line " + line + "] rule " + ruleName + " " + err.getMessage();
}
}
if (errMsg == null) {
errMsg = ex.getMessage();
}
if (errMsg == null) {
errMsg = "Unknown parsing error.";
}
return errMsg;
}
private String getTokenErrorDisplay(Token t) {
if (t == null) return "<no token>";
String s = t.getText();
if (s == null) {
if (t.getType() == Token.EOF) {
s = "<EOF>";
} else {
s = "<" + t.getType() + ">";
}
}
return escapeWSAndQuote(s);
}
private String escapeWSAndQuote(String s) {
s = s.replace("\n", "\\n");
s = s.replace("\r", "\\r");
s = s.replace("\t", "\\t");
return "'" + s + "'";
}
public void setFacetInfoMap(Map<String, String[]> facetInfoMap) {
_facetInfoMap = facetInfoMap;
}
}