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

时间：2023-03-08 18:31:54

背景：修改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内部函数的调用过程:

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

在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]

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

　　

相关文章