PostgresQL FDW 源码分析之 postgresBeginForeignScan()

最新推荐文章于 2023-01-19 19:09:58 发布

davidzwb

最新推荐文章于 2023-01-19 19:09:58 发布

阅读量746

点赞数

分类专栏： PosgresQL 扩展开发文章标签：数据库源码 postgresql fdw

本文链接：https://blog.csdn.net/davidzwb/article/details/51810306

版权

PosgresQL 扩展开发专栏收录该内容

10 篇文章 0 订阅

订阅专栏

在 postgresBeginForeignScan() 中主要进行 postgresIterateForeignScan() 中需要数据的准备工作。

获取建立的外部表信息：

	/* Get info about foreign table. */
	fsstate->rel = node->ss.ss_currentRelation;
	table = GetForeignTable(RelationGetRelid(fsstate->rel));
	server = GetForeignServer(table->serverid);
	user = GetUserMapping(userid, server->serverid);

与 remote 端建立连接，并获取将要在 remote 端创建的游标 cursor 的 unique ID。

	/*
	 * Get connection to the foreign server.  Connection manager will
	 * establish new connection if necessary.
	 */
	fsstate->conn = GetConnection(server, user, false);

	/* Assign a unique ID for my cursor */
	fsstate->cursor_number = GetCursorNumber(fsstate->conn);
	fsstate->cursor_exists = false;

从 fsplan->fdw_private 中取得保存的、要发送到 remote 端的 SQL 语句，存放到 fsstate->query 中，之后在 postgresIterateForeignScan() 中使用。

	/* Get private info created by planner functions. */
	fsstate->query = strVal(list_nth(fsplan->fdw_private,
									 FdwScanPrivateSelectSql));
	fsstate->retrieved_attrs = (List *) list_nth(fsplan->fdw_private,
											   FdwScanPrivateRetrievedAttrs);

创建处理 tuple 的 memory context:

	/* Create contexts for batches of tuples and per-tuple temp workspace. */
	fsstate->batch_cxt = AllocSetContextCreate(estate->es_query_cxt,
											   "postgres_fdw tuple data",
											   ALLOCSET_DEFAULT_MINSIZE,
											   ALLOCSET_DEFAULT_INITSIZE,
											   ALLOCSET_DEFAULT_MAXSIZE);
	fsstate->temp_cxt = AllocSetContextCreate(estate->es_query_cxt,
											  "postgres_fdw temporary data",
											  ALLOCSET_SMALL_MINSIZE,
											  ALLOCSET_SMALL_INITSIZE,
											  ALLOCSET_SMALL_MAXSIZE);