经过这么多源码的洗礼,下面让我们动手把玩PostgreSQL一把.
第一个任务是希望添加一个系统字段,该字段输出的为当前日期(与PG的CURRENT_DATE日期函数输出一致).
我们假定该字段名称为zz_sysdate(之所以zz打头是为了方便起见,把该字段放在PG其他字段之后),客户端输出的名称为sysdate.
下面对这个任务进行分解,总的来说有几个工作需要完成:
1.输入:让PG可以识别zz_sysdate是系统列;
2.输出:结果投影列名称命名为sysdate;
3.实现:参照current_date实现

本节介绍第一部分:输入.

一、输入

标准PG对于zz_sysdate的反应是:


testdb=# select zz_sysdate from t1;
ERROR:  column "zz_sysdate" does not exist
LINE 1: select zz_sysdate from t1;

报错,提示列不存在.

为了让PG可识别zz_sysdate,需改造词法和语法分析模块.
词法
涉及的文件包括scan.l,kwlist.h

scan.l:
词法文件,会把zz_sysdate识别为IDENT,该文件无需修改;

kwlist.h:
关键字列表,在最后添加一句:


PG_KEYWORD("zz_sysdate", ZZ_SYSDATE, RESERVED_KEYWORD)

RESERVED_KEYWORD表示保留关键字

语法
语法文件是gram.y,需添加ZZ_SYSDATE


//添加token,与词法分析中的ZZ_SYSDATE对应
%token <keyword> ...
... ZZ_SYSDATE
//产生式reserved_keyword添加ZZ_SYSDATE
//注:reserved_keyword是非终结符,ZZ_SYSDATE是终结符
reserved_keyword:
            ...
            | ZZ_SYSDATE
            ;
//参照current_date,在func_expr_common_subexpr添加
        ...
        | ZZ_SYSDATE
                {
                    $$ = makeSQLValueFunction(SVFOP_ZZ_SYSDATE, -1, @1);
                }

另外,为了识别SVFOP_ZZ_SYSDATE,在preproc.h的yytokentype枚举变量中添加


ZZ_SYSDATE = 793

经上述修改,现在PG可以识别zz_sysdate了,但执行会出错.

二、参考资料

Flex&Bison

PostgreSQL 源码解读（172）- 内核研发#1(如何添加系统字段)#1

一、输入

二、参考资料