SQL 查询执行总体流程（3）

引言

上一篇文章中，我们学习了内存上下文的只是，并且对PortalStart函数做了简要的了解。今天这篇文章，我将学习PortalRun函数的有关内容。

执行过程(续前文)

PortalRun

1. //代码清单1
2. bool PortalRun(
3. Portal portal, long count, bool isTopLevel, DestReceiver* dest, DestReceiver* altdest, char* completionTag)
4. {
5. ············
6. /*
7. * Check for improper portal use, and mark portal active.
8. */
9. MarkPortalActive(portal);
11. QueryDesc* queryDesc = portal->queryDesc;
13. PotalSetIoState(portal);
14. ············
15. PG_TRY();
16. {
17. switch (portal->strategy) {
18. case PORTAL_ONE_SELECT:
19. case PORTAL_ONE_RETURNING:
20. case PORTAL_ONE_MOD_WITH:
21. case PORTAL_UTIL_SELECT:
23. /*
24. * If we have not yet run the command, do so, storing its
25. * results in the portal's tuplestore. But we don't do that
26. * for the PORTAL_ONE_SELECT case.
27. */
28. if (portal->strategy != PORTAL_ONE_SELECT && !portal->holdStore) {
29. /* DestRemoteExecute can not send T message automatically */
30. if (strcmp(portal->commandTag, "EXPLAIN") == 0 && dest->mydest != DestRemote)
31. t_thrd.explain_cxt.explain_perf_mode = EXPLAIN_NORMAL;
32. FillPortalStore(portal, isTopLevel);
33. }
35. /*
36. * Now fetch desired portion of results.
37. */
38. nprocessed = PortalRunSelect(portal, true, count, dest);
40. /*
41. * If the portal result contains a command tag and the caller
42. * gave us a pointer to store it, copy it. Patch the "SELECT"
43. * tag to also provide the rowcount.
44. */
45. if (completionTag != NULL) {
46. if (strcmp(portal->commandTag, "SELECT") == 0) {
47. errorno = snprintf_s(completionTag,
48. COMPLETION_TAG_BUFSIZE,
49. COMPLETION_TAG_BUFSIZE - 1,
50. "SELECT %lu",
51. nprocessed);
52. securec_check_ss(errorno, "\0", "\0");
53. } else {
54. errorno = strcpy_s(completionTag, COMPLETION_TAG_BUFSIZE,
55. portal->commandTag);
56. securec_check(errorno, "\0", "\0");
57. }
58. }
60. /* Mark portal not active */
61. portal->status = PORTAL_READY;
63. /*
64. * Since it's a forward fetch, say DONE iff atEnd is now true.
65. */
66. result = portal->atEnd;
67. break;
69. case PORTAL_MULTI_QUERY:
70. PortalRunMulti(portal, isTopLevel, dest, altdest, completionTag);
72. /* Prevent portal's commands from being re-executed */
73. MarkPortalDone(portal);
74. /* Always complete at end of RunMulti */
75. result = true;
76. break;
78. default:
79. ereport(ERROR,
80. (errcode(ERRCODE_UNRECOGNIZED_NODE_TYPE),
81. errmodule(MOD_EXECUTOR),
82. errmsg("Unrecognized portal strategy: %d", (int)portal->strategy)));
83. result = false; /* keep compiler quiet */
84. break;
85. }
86. }
87. PG_CATCH();
88. {
90. }
91. PG_END_TRY();
93. u_sess->plsql_cxt.portal_depth = savePortalDepth;
94. stp_reset_xact_state_and_err_msg(savedisAllowCommitRollback, needResetErrMsg);
96. ············
98. /* update unique sql stat */
99. if (is_instr_top_portal() && is_unique_sql_enabled() && is_local_unique_sql()) {
100. /* Instrumentation: update unique sql returned rows(SELECT) */
101. // only CN can update this counter
102. if (portal->queryDesc != NULL && portal->queryDesc->estate && portal->queryDesc->estate->es_plannedstmt &&
103. portal->queryDesc->estate->es_plannedstmt->commandType == CMD_SELECT) {
104. ereport(DEBUG1,
105. (errmodule(MOD_INSTR),
106. errmsg("[UniqueSQL]"
107. "unique id: %lu , select returned rows: %lu",
108. u_sess->unique_sql_cxt.unique_sql_id,
109. portal->queryDesc->estate->es_processed)));
110. UniqueSQLStatCountReturnedRows(portal->queryDesc->estate->es_processed);
111. }
113. /* PortalRun using unique_sql_start_time as unique sql elapse start time */
114. if (IsNeedUpdateUniqueSQLStat(portal) && IS_UNIQUE_SQL_TRACK_TOP && IsTopUniqueSQL()) {
115. UpdateUniqueSQLStat(NULL, NULL, u_sess->unique_sql_cxt.unique_sql_start_time);
116. }
118. if (u_sess->unique_sql_cxt.unique_sql_start_time != 0) {
119. int64 duration = GetCurrentTimestamp() - u_sess->unique_sql_cxt.unique_sql_start_time;
120. if (IS_SINGLE_NODE) {
121. pgstat_update_responstime_singlenode(
122. u_sess->unique_sql_cxt.unique_sql_id, u_sess->unique_sql_cxt.unique_sql_start_time, duration);
123. } else {
124. pgstat_report_sql_rt(
125. u_sess->unique_sql_cxt.unique_sql_id, u_sess->unique_sql_cxt.unique_sql_start_time, duration);
126. }
127. }
128. }
129. decrease_instr_portal_nesting_level();
130. gstrace_exit(GS_TRC_ID_PortalRun);
131. u_sess->pcache_cxt.cur_stmt_name = old_stmt_name;
132. return result;
133. }

此函数的主要功能为：

1. 调用MarkPortalActive函数，使portal从Ready状态转为Active状态

2. 执行Switch-Case语句，根据不同语句，选择不同的执行方法。

   • PORTAL_ONE_SELECT 、 PORTAL_ONE_RETURNING 、 PORTAL_ONE_MOD_WITH 和 PORTAL_UTIL_SELECT：

     如果portal对应的语句还没有被执行过（portal->holdStore字段为空），并且不是PORTAL_ONE_SELECT语句，则调用FillPortalStore函数执行查询并将结果加载到portal的存储元组中。

     然后调用PortalRunSelect函数获取结果。

     把portal更新为Ready状态，准备下一次执行。

   • PORTAL_MULTI_QUERY：

     调用PortalRunMulti函数执行此语句（一般为多个查询或者非查询语句，例如对表的创建和删除等）。并把portal更新为Done状态。

3. 执行后后续的一些恢复及其他操作，例如恢复原来的内存上下文，记录和更新一些时间和资源的统计信息等。

在学习PortalRun函数的过程中，我发现了两个陌生的点，并对其深入拓展学习了一番。

1.Portal对象的持久化连接和非持久化连接

在代码清单1中，第28行中出现了holdStore（持久化存储）字段，通过引申学习，我发现Portal对象有两种连接方式。

• 非持久化连接

  如果查询是一个简单的SELECT语句，那么查询结果不会存储在任何地方，而是直接返回给客户端。此时Portal对象是非持久化连接，即当事务结束时就释放内存。

• 持久化连接

  如果查询是一个带有游标(CURSOR)的SELECT语句，那么查询中间结果会储存在Portal对象中。此时Portal对象是持久化连接，可以让Portal对象在多个请求——响应事务中保持有效，而不是每次都重新创建和销毁。

2.UNIQUE SQL

在代码清单1中的第98行~128行代码中，我发现unique_sql字符串频繁出现，我查找了相关资料，发现opengauss数据库执行过程的一个精妙之处。

在opengauss代码架构中，拓扑结构相同的SQL语句会赋给一个相同ID。

例如，有如下两条SQL语句：

1. select* from t where id=1
2. select* from t where id=2

这两条SQL语句生成的查询树的拓扑结构是相同的，只有选择算子的输入不同，因此属于同一个UNIQUE SQL。

引入UNIQUE SQL后SQL语句的执行流程如下图所示。

可以看出，当UNIQUE ID存在时，SQL可以省去创建UNIQUE SQL，也即创建查询树的过程。这样对代码的执行性能又有了一个提升。

小结

本篇文章对PortalRun函数做了简要的分析，学习了Portal对象的持久性和非持久性连接，并对UNIQUE SQL有了一定了解。下一篇文章中，我将学习Portal相关的最后一个函数PortalDrop。