我正在编写一个简单的解析器来将查询转换为SQL。
我已经完成了解析器并得到了AST,但现在我需要使用ANTLR 4将这些语法翻译成SQL。
有人知道如何在ANTLR 4中打印或翻译查询"SELECT" as "SELECT", "FROM" to "FROM"
等吗?
我遵循了权威的ANTLR 4参考书。因为我们必须创建2个java文件。
在第一个java文件中我想按照这些步骤而不是
public void enterInit(ArrayInitParser.InitContext ctx)
{
System.out.print('"');
}
我已改为
public void enterSelect(ArrayInitParser.SelectContext ctx)
{
System.out.print('SELECT');
}
ShortToUnicodeString.java
public class ShortToUnicodeString extends ArrayInitBaseListener
{
/** Translate { to " */
@Override
public void enterInit(ArrayInitParser.InitContext ctx)
{
System.out.print('"');
}
/** Translate } to " */
@Override
public void exitInit(ArrayInitParser.InitContext ctx)
{
System.out.print('"');
}
/** Translate integers to 4-digit hexadecimal strings prefixed with \\u */
@Override
public void enterValue(ArrayInitParser.ValueContext ctx)
{
// Assumes no nested array initializers
int value = Integer.valueOf(ctx.INT().getText());
System.out.printf("\\u%04x", value);
}
}
在第二个java文件中,我按照以下步骤操作:
Translate.java
import org.antlr.v4.runtime.*;
import org.antlr.v4.runtime.tree.*;
public class Translate
{
public static void main(String[] args) throws Exception
{
// create a CharStream that reads from standard input
ANTLRInputStream input = new ANTLRInputStream(System.in);
// create a lexer that feeds off of input CharStream
ArrayInitLexer lexer = new ArrayInitLexer(input);
// create a buffer of tokens pulled from the lexer
CommonTokenStream tokens = new CommonTokenStream(lexer);
// create a parser that feeds off the tokens buffer
ArrayInitParser parser = new ArrayInitParser(tokens);
ParseTree tree = parser.init(); // begin parsing at init rule
// Create a generic parse tree walker that can trigger callbacks
ParseTreeWalker walker = new ParseTreeWalker();
// Walk the tree created during the parse, trigger callbacks
walker.walk(new ShortToUnicodeString(), tree);
System.out.println(); // print a \n after translation
}
}
TRING WITH STRINGTEMPLATE
我在antlr 4中尝试过使用StringTemplate。我可以翻译查询select, from, where
,但是当我翻译查询and
时,我无法翻译。我的代码如下:
import org.stringtemplate.v4.*;
public class ToSql extends BaseListener
{
/** Translate select to select */
@Override
public void enterSelect( Parser.SelectContext ctx)
{
ST select = new ST("select");
System.out.print(select.render()+"\t");
}
/** Translate from to from */
@Override
public void enterFrom( Parser.FromContext ctx)
{
ST from = new ST("from");
System.out.print(from.render()+"\t");
}
/** Translate where to where */
@Override
public void enterWhere( Parser.WhereContext ctx)
{
ST where = new ST("where");
System.out.print(where.render()+"\t");
}
/** Translate boolOp to AND|OR|XOR */
@Override
public void enterBoolOp( Parser.BoolOpContext ctx)
{
ST boolOp = new ST("<and>");
if (boolOp == and)
{
boolOp.add("and", "AND");
System.out.print(boolOp.render()+"\t");
}
}
}
我的格拉玛的一部分:
boolOp:OR | XOR | AND;
谢谢。
答案 0 :(得分:1)
我不知道代码生成在v4中是如何工作的,但是根据我的小经验,我认为你可以通过使用语法中的规则吐出生成的输出来完成语法定向翻译。
boolOp returns[String output] : OR {$output=$OR.text}
| XOR {$output=$XOR.text}
| AND{$output=$AND.text}
;
然后在你的select语句中,你也可以做同样的事情来返回字符串“select”“where”等。 您可以通过将walker的对象传递给sysout来遍历树并打印输出。
您只需要处理格式化(换行符,制表符等)以获得良好的漂亮打印输出。