如何将ANTLR 4生成的AST转换为源代码

时间:2013-08-20 16:03:32

标签: java sql antlr antlr4

我正在编写一个简单的解析器来将查询转换为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;

谢谢。

1 个答案:

答案 0 :(得分:1)

我不知道代码生成在v4中是如何工作的,但是根据我的小经验,我认为你可以通过使用语法中的规则吐出生成的输出来完成语法定向翻译。

  boolOp returns[String output] : OR {$output=$OR.text}
                                 | XOR {$output=$XOR.text}
                                 | AND{$output=$AND.text}
                                 ;

然后在你的select语句中,你也可以做同样的事情来返回字符串“select”“where”等。 您可以通过将walker的对象传递给sysout来遍历树并打印输出。

您只需要处理格式化(换行符,制表符等)以获得良好的漂亮打印输出。