postgres创建表的过程以及部分源码分析

Posted 2020-08-22

背景：修改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]

注：未经同意，不得转载！