背景:修改pg内核,在创建表时,表名不能和当前的用户名同名。
首先我们知道DefineRelation此函数是最终创建表结构的函数,最主要的参数是CreateStmt这个结构,该结构如下
typedef struct CreateStmt { NodeTag type; RangeVar *relation; /* relation to create */ List *tableElts; /* column definitions (list of ColumnDef) */ List *inhRelations; /* relations to inherit from (list of * inhRelation) */ TypeName *ofTypename; /* OF typename */ List *constraints; /* constraints (list of Constraint nodes) */ List *options; /* options from WITH clause */ OnCommitAction oncommit; /* what do we do at COMMIT? */ char *tablespacename; /* table space to use, or NULL */ bool if_not_exists; /* just do nothing if it already exists? */ } CreateStmt;
结构中relation中包含了catalogname,schemaname,relname此时的relname就能够顺利的拿到。
tableElts 这个list定义了表结构中的所有列名,如若想增加个非隐藏列,可以append进去。
下图是创建一个简单表,PG内部函数的调用过程:
在PG backend上敲入的sql,入口函数都是exec_simple_query,把这串sql解析,重写后生产执行计划。
附带一个PG内核删除的函数,将query结构反解析出sql的函数
char * deparse_query_def(Query *query) { StringInfoData buf; initStringInfo(&buf); get_query_def(query, &buf, NIL, NULL, PRETTYFLAG_INDENT, WRAP_COLUMN_DEFAULT, 0); return buf.data; }
如果想很好的看清query以及subquery重写以后的任务是什么,可以将此函数编译进pg内核。
在ProcessUtility这个过程中有钩子函数可以挂接,可以根据nodeTag(parseTree)的类型来分别进行处理,例如cstore_fdw中T_DropStmt这种操作的时候,将物理文件删除
这个钩子的用处还可以控制某些操作加入你想的内容。
我们可以在ProcessUtilitySlow这个函数进行内核修改,加入对当前用户名获取,并从CreateStmt结构中的relname进行对比,然后控制是否创建表或者进行报错信息。
获取当前用户名如下
Datum current_user(PG_FUNCTION_ARGS) { PG_RETURN_DATUM(DirectFunctionCall1(namein, CStringGetDatum(GetUserNameFromId(GetUserId())))); }
此函数是PG的内部函数,使用效果是:
postgres=# select CURRENT_USER; current_user -------------- postgres (1 row)
按照这样的做法就能够完成背景下的内容了。
PS:通常还有几个搭配函数使用->DefineRelation->CommandCounterIncrement->transformRelOptions
->heap_reloptions->NewRelationCreateToastTable[AlterTableCreateToastTable]
注:未经同意,不得转载!