前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Flink源码

Flink源码

原创
作者头像
ruochen
修改2021-11-22 09:24:04
2970
修改2021-11-22 09:24:04
举报

下面我们以一个具体例子来说怎么做SQL语法拓展,比如我们需要支持语法

代码语言:txt
复制
USE FUNCTION func_name [ WITH (name = '', version = '') ]

1. 创建maven项目

2. 复制模板文件

从Flink源码flink-table/flink-sql-parser/src/main/codegen文件夹copy到自己项目的src/main目录下

3. 拓展SQL语法

主要分为下面三步

3.1 在 codegen/includes/parserImpls.ftl 定义解析use function 的规则,包含了方法名及对应的规则。

代码语言:txt
复制
/**
代码语言:txt
复制
*  USE FUNCTION func_name [ WITH (name = '', version = '') ]
*/
代码语言:txt
复制
SqlCall SqlUseFunction() :
代码语言:txt
复制
{
代码语言:txt
复制
     SqlParserPos pos;
代码语言:txt
复制
     SqlIdentifier funcName;
代码语言:txt
复制
     SqlNodeList funcProps = null;
代码语言:txt
复制
}
代码语言:txt
复制
{
代码语言:txt
复制
     <USE>
代码语言:txt
复制
     {
代码语言:txt
复制
         // Token位置
代码语言:txt
复制
         pos = getPos();
代码语言:txt
复制
     }
代码语言:txt
复制
     <FUNCTION>
代码语言:txt
复制
         funcName = CompoundIdentifier()
代码语言:txt
复制
     [ <WITH> funcProps = TableProperties() ]
代码语言:txt
复制
     {
代码语言:txt
复制
         return new SqlUseFunction(pos, funcName, funcProps);
代码语言:txt
复制
     }
代码语言:txt
复制
}

3.2 拓展 SqlCall

规则匹配成功返回一个SqlUseFunction节点,作为解析树中的SqlNode。

代码语言:txt
复制
public class SqlUseFunction extends SqlCall {
代码语言:txt
复制
    private static final SqlSpecialOperator OPERATOR = new SqlSpecialOperator("USE FUNCTION",
代码语言:txt
复制
            SqlKind.OTHER_FUNCTION);
代码语言:txt
复制
    private final SqlIdentifier funcName;
代码语言:txt
复制
    private final SqlNodeList funcProps;
代码语言:txt
复制
    /**
代码语言:txt
复制
     * SqlUseFunction constructor.
     *
     * @param pos sql define location
     * @param funcName function name
     * @param funcProps function property
     * */
    public SqlUseFunction(SqlParserPos pos, SqlIdentifier funcName, SqlNodeList funcProps) {
        super(pos);
        this.funcName = funcName;
        this.funcProps = funcProps;
    }
代码语言:txt
复制
    @Override
代码语言:txt
复制
    public void unparse(SqlWriter writer, int leftPrec, int rightPrec) {
代码语言:txt
复制
        writer.keyword("USE FUNCTION");
代码语言:txt
复制
        funcName.unparse(writer, leftPrec, rightPrec);
代码语言:txt
复制
        if (funcProps != null) {
代码语言:txt
复制
            writer.keyword("WITH");
代码语言:txt
复制
            SqlWriter.Frame frame = writer.startList("(", ")");
代码语言:txt
复制
            for (SqlNode c : funcProps) {
代码语言:txt
复制
                writer.sep(",");
代码语言:txt
复制
                c.unparse(writer, 0, 0);
代码语言:txt
复制
            }
代码语言:txt
复制
            writer.endList(frame);
代码语言:txt
复制
        }
代码语言:txt
复制
    }
代码语言:txt
复制
    @Override
代码语言:txt
复制
    public SqlOperator getOperator() {
代码语言:txt
复制
        return OPERATOR;
代码语言:txt
复制
    }
代码语言:txt
复制
    @Override
代码语言:txt
复制
    public List<SqlNode> getOperandList() {
代码语言:txt
复制
        return ImmutableNullableList.of(funcName, funcProps);
代码语言:txt
复制
    }
代码语言:txt
复制
}

3.3 将添加的规则,加入配置文件 codegen/data/Parser.tdd

  • imports增加SqlUseFunction类。
  • statementParserMethods增加定义的规则方法。在Parser中使用新规则。
代码语言:txt
复制
 imports: [
代码语言:txt
复制
    "org.apache.flink.sql.extended.SqlUseFunction"
代码语言:txt
复制
 ]
代码语言:txt
复制
  # List of methods for parsing custom SQL statements.
代码语言:txt
复制
  # Return type of method implementation should be 'SqlNode'.
代码语言:txt
复制
  # Example: SqlShowDatabases(), SqlShowTables().
代码语言:txt
复制
  statementParserMethods: [
代码语言:txt
复制
    "SqlUseFunction()"
代码语言:txt
复制
  ]
  1. 编译?developer/article/1903453/undefined执行mvn clean compile生成定义的解析类SqlParserImpl.java。重点看生成的SqlUseFunction(),编译成功就可以在sql中使用该语法了。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
作者已关闭评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 创建maven项目
  • 2. 复制模板文件
  • 3. 拓展SQL语法
相关产品与服务
大数据
全栈大数据产品,面向海量数据场景,帮助您 “智理无数,心中有数”!
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档
http://www.vxiaotou.com