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

时间:2021-01-22 12:06:55

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

注:未经同意,不得转载!