nodesUtilFuncs.c 57.5 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

16
#include "cmdnodes.h"
X
Xiaoyu Wang 已提交
17
#include "nodesUtil.h"
X
Xiaoyu Wang 已提交
18
#include "plannodes.h"
19
#include "querynodes.h"
X
Xiaoyu Wang 已提交
20
#include "taos.h"
21
#include "taoserror.h"
X
Xiaoyu Wang 已提交
22
#include "thash.h"
23

X
Xiaoyu Wang 已提交
24 25 26 27 28 29 30 31 32
static SNode* makeNode(ENodeType type, size_t size) {
  SNode* p = taosMemoryCalloc(1, size);
  if (NULL == p) {
    return NULL;
  }
  setNodeType(p, type);
  return p;
}

33
SNode* nodesMakeNode(ENodeType type) {
34 35
  switch (type) {
    case QUERY_NODE_COLUMN:
X
Xiaoyu Wang 已提交
36
      return makeNode(type, sizeof(SColumnNode));
37
    case QUERY_NODE_VALUE:
X
Xiaoyu Wang 已提交
38
      return makeNode(type, sizeof(SValueNode));
39
    case QUERY_NODE_OPERATOR:
X
Xiaoyu Wang 已提交
40
      return makeNode(type, sizeof(SOperatorNode));
41
    case QUERY_NODE_LOGIC_CONDITION:
X
Xiaoyu Wang 已提交
42
      return makeNode(type, sizeof(SLogicConditionNode));
43
    case QUERY_NODE_FUNCTION:
X
Xiaoyu Wang 已提交
44
      return makeNode(type, sizeof(SFunctionNode));
45
    case QUERY_NODE_REAL_TABLE:
X
Xiaoyu Wang 已提交
46
      return makeNode(type, sizeof(SRealTableNode));
47
    case QUERY_NODE_TEMP_TABLE:
X
Xiaoyu Wang 已提交
48
      return makeNode(type, sizeof(STempTableNode));
49
    case QUERY_NODE_JOIN_TABLE:
X
Xiaoyu Wang 已提交
50
      return makeNode(type, sizeof(SJoinTableNode));
51
    case QUERY_NODE_GROUPING_SET:
X
Xiaoyu Wang 已提交
52
      return makeNode(type, sizeof(SGroupingSetNode));
53
    case QUERY_NODE_ORDER_BY_EXPR:
X
Xiaoyu Wang 已提交
54
      return makeNode(type, sizeof(SOrderByExprNode));
55
    case QUERY_NODE_LIMIT:
X
Xiaoyu Wang 已提交
56
      return makeNode(type, sizeof(SLimitNode));
57
    case QUERY_NODE_STATE_WINDOW:
X
Xiaoyu Wang 已提交
58
      return makeNode(type, sizeof(SStateWindowNode));
59
    case QUERY_NODE_SESSION_WINDOW:
X
Xiaoyu Wang 已提交
60
      return makeNode(type, sizeof(SSessionWindowNode));
61
    case QUERY_NODE_INTERVAL_WINDOW:
X
Xiaoyu Wang 已提交
62
      return makeNode(type, sizeof(SIntervalWindowNode));
63
    case QUERY_NODE_NODE_LIST:
X
Xiaoyu Wang 已提交
64
      return makeNode(type, sizeof(SNodeListNode));
65
    case QUERY_NODE_FILL:
X
Xiaoyu Wang 已提交
66
      return makeNode(type, sizeof(SFillNode));
67
    case QUERY_NODE_RAW_EXPR:
X
Xiaoyu Wang 已提交
68
      return makeNode(type, sizeof(SRawExprNode));
69
    case QUERY_NODE_TARGET:
X
Xiaoyu Wang 已提交
70
      return makeNode(type, sizeof(STargetNode));
71
    case QUERY_NODE_DATABLOCK_DESC:
X
Xiaoyu Wang 已提交
72
      return makeNode(type, sizeof(SDataBlockDescNode));
73
    case QUERY_NODE_SLOT_DESC:
X
Xiaoyu Wang 已提交
74
      return makeNode(type, sizeof(SSlotDescNode));
75
    case QUERY_NODE_COLUMN_DEF:
X
Xiaoyu Wang 已提交
76
      return makeNode(type, sizeof(SColumnDefNode));
X
Xiaoyu Wang 已提交
77
    case QUERY_NODE_DOWNSTREAM_SOURCE:
X
Xiaoyu Wang 已提交
78
      return makeNode(type, sizeof(SDownstreamSourceNode));
X
Xiaoyu Wang 已提交
79
    case QUERY_NODE_DATABASE_OPTIONS:
X
Xiaoyu Wang 已提交
80
      return makeNode(type, sizeof(SDatabaseOptions));
X
Xiaoyu Wang 已提交
81
    case QUERY_NODE_TABLE_OPTIONS:
X
Xiaoyu Wang 已提交
82
      return makeNode(type, sizeof(STableOptions));
X
Xiaoyu Wang 已提交
83
    case QUERY_NODE_INDEX_OPTIONS:
X
Xiaoyu Wang 已提交
84
      return makeNode(type, sizeof(SIndexOptions));
85
    case QUERY_NODE_EXPLAIN_OPTIONS:
X
Xiaoyu Wang 已提交
86
      return makeNode(type, sizeof(SExplainOptions));
87
    case QUERY_NODE_STREAM_OPTIONS:
X
Xiaoyu Wang 已提交
88
      return makeNode(type, sizeof(SStreamOptions));
89
    case QUERY_NODE_LEFT_VALUE:
X
Xiaoyu Wang 已提交
90
      return makeNode(type, sizeof(SLeftValueNode));
91 92
    case QUERY_NODE_COLUMN_REF:
      return makeNode(type, sizeof(SColumnDefNode));
93
    case QUERY_NODE_SET_OPERATOR:
X
Xiaoyu Wang 已提交
94
      return makeNode(type, sizeof(SSetOperator));
95
    case QUERY_NODE_SELECT_STMT:
X
Xiaoyu Wang 已提交
96
      return makeNode(type, sizeof(SSelectStmt));
97
    case QUERY_NODE_VNODE_MODIF_STMT:
X
Xiaoyu Wang 已提交
98
      return makeNode(type, sizeof(SVnodeModifOpStmt));
99
    case QUERY_NODE_CREATE_DATABASE_STMT:
X
Xiaoyu Wang 已提交
100
      return makeNode(type, sizeof(SCreateDatabaseStmt));
101
    case QUERY_NODE_DROP_DATABASE_STMT:
X
Xiaoyu Wang 已提交
102
      return makeNode(type, sizeof(SDropDatabaseStmt));
103
    case QUERY_NODE_ALTER_DATABASE_STMT:
X
Xiaoyu Wang 已提交
104
      return makeNode(type, sizeof(SAlterDatabaseStmt));
105
    case QUERY_NODE_CREATE_TABLE_STMT:
X
Xiaoyu Wang 已提交
106
      return makeNode(type, sizeof(SCreateTableStmt));
107
    case QUERY_NODE_CREATE_SUBTABLE_CLAUSE:
X
Xiaoyu Wang 已提交
108
      return makeNode(type, sizeof(SCreateSubTableClause));
109
    case QUERY_NODE_CREATE_MULTI_TABLE_STMT:
X
Xiaoyu Wang 已提交
110
      return makeNode(type, sizeof(SCreateMultiTableStmt));
111
    case QUERY_NODE_DROP_TABLE_CLAUSE:
X
Xiaoyu Wang 已提交
112
      return makeNode(type, sizeof(SDropTableClause));
113
    case QUERY_NODE_DROP_TABLE_STMT:
X
Xiaoyu Wang 已提交
114
      return makeNode(type, sizeof(SDropTableStmt));
115
    case QUERY_NODE_DROP_SUPER_TABLE_STMT:
X
Xiaoyu Wang 已提交
116
      return makeNode(type, sizeof(SDropSuperTableStmt));
117
    case QUERY_NODE_ALTER_TABLE_STMT:
X
Xiaoyu Wang 已提交
118
      return makeNode(type, sizeof(SAlterTableStmt));
119
    case QUERY_NODE_CREATE_USER_STMT:
X
Xiaoyu Wang 已提交
120
      return makeNode(type, sizeof(SCreateUserStmt));
121
    case QUERY_NODE_ALTER_USER_STMT:
X
Xiaoyu Wang 已提交
122
      return makeNode(type, sizeof(SAlterUserStmt));
123
    case QUERY_NODE_DROP_USER_STMT:
X
Xiaoyu Wang 已提交
124
      return makeNode(type, sizeof(SDropUserStmt));
125
    case QUERY_NODE_USE_DATABASE_STMT:
X
Xiaoyu Wang 已提交
126
      return makeNode(type, sizeof(SUseDatabaseStmt));
127
    case QUERY_NODE_CREATE_DNODE_STMT:
X
Xiaoyu Wang 已提交
128
      return makeNode(type, sizeof(SCreateDnodeStmt));
129
    case QUERY_NODE_DROP_DNODE_STMT:
X
Xiaoyu Wang 已提交
130
      return makeNode(type, sizeof(SDropDnodeStmt));
131
    case QUERY_NODE_ALTER_DNODE_STMT:
X
Xiaoyu Wang 已提交
132
      return makeNode(type, sizeof(SAlterDnodeStmt));
X
Xiaoyu Wang 已提交
133
    case QUERY_NODE_CREATE_INDEX_STMT:
X
Xiaoyu Wang 已提交
134
      return makeNode(type, sizeof(SCreateIndexStmt));
135
    case QUERY_NODE_DROP_INDEX_STMT:
X
Xiaoyu Wang 已提交
136
      return makeNode(type, sizeof(SDropIndexStmt));
X
Xiaoyu Wang 已提交
137
    case QUERY_NODE_CREATE_QNODE_STMT:
138 139 140
    case QUERY_NODE_CREATE_BNODE_STMT:
    case QUERY_NODE_CREATE_SNODE_STMT:
    case QUERY_NODE_CREATE_MNODE_STMT:
X
Xiaoyu Wang 已提交
141
      return makeNode(type, sizeof(SCreateComponentNodeStmt));
142
    case QUERY_NODE_DROP_QNODE_STMT:
143 144 145
    case QUERY_NODE_DROP_BNODE_STMT:
    case QUERY_NODE_DROP_SNODE_STMT:
    case QUERY_NODE_DROP_MNODE_STMT:
X
Xiaoyu Wang 已提交
146
      return makeNode(type, sizeof(SDropComponentNodeStmt));
147
    case QUERY_NODE_CREATE_TOPIC_STMT:
X
Xiaoyu Wang 已提交
148
      return makeNode(type, sizeof(SCreateTopicStmt));
149
    case QUERY_NODE_DROP_TOPIC_STMT:
X
Xiaoyu Wang 已提交
150 151 152
      return makeNode(type, sizeof(SDropTopicStmt));
    case QUERY_NODE_DROP_CGROUP_STMT:
      return makeNode(type, sizeof(SDropCGroupStmt));
153 154
    case QUERY_NODE_ALTER_LOCAL_STMT:
      return makeNode(type, sizeof(SAlterLocalStmt));
155
    case QUERY_NODE_EXPLAIN_STMT:
X
Xiaoyu Wang 已提交
156
      return makeNode(type, sizeof(SExplainStmt));
157
    case QUERY_NODE_DESCRIBE_STMT:
X
Xiaoyu Wang 已提交
158
      return makeNode(type, sizeof(SDescribeStmt));
159
    case QUERY_NODE_RESET_QUERY_CACHE_STMT:
X
Xiaoyu Wang 已提交
160
      return makeNode(type, sizeof(SNode));
161
    case QUERY_NODE_COMPACT_STMT:
162
      break;
163
    case QUERY_NODE_CREATE_FUNCTION_STMT:
X
Xiaoyu Wang 已提交
164
      return makeNode(type, sizeof(SCreateFunctionStmt));
165
    case QUERY_NODE_DROP_FUNCTION_STMT:
X
Xiaoyu Wang 已提交
166
      return makeNode(type, sizeof(SDropFunctionStmt));
167
    case QUERY_NODE_CREATE_STREAM_STMT:
X
Xiaoyu Wang 已提交
168
      return makeNode(type, sizeof(SCreateStreamStmt));
169
    case QUERY_NODE_DROP_STREAM_STMT:
X
Xiaoyu Wang 已提交
170
      return makeNode(type, sizeof(SDropStreamStmt));
X
Xiaoyu Wang 已提交
171 172
    case QUERY_NODE_BALANCE_VGROUP_STMT:
      return makeNode(type, sizeof(SBalanceVgroupStmt));
173
    case QUERY_NODE_MERGE_VGROUP_STMT:
X
Xiaoyu Wang 已提交
174
      return makeNode(type, sizeof(SMergeVgroupStmt));
175
    case QUERY_NODE_REDISTRIBUTE_VGROUP_STMT:
X
Xiaoyu Wang 已提交
176
      return makeNode(type, sizeof(SRedistributeVgroupStmt));
177
    case QUERY_NODE_SPLIT_VGROUP_STMT:
X
Xiaoyu Wang 已提交
178
      return makeNode(type, sizeof(SSplitVgroupStmt));
179 180
    case QUERY_NODE_SYNCDB_STMT:
      break;
181
    case QUERY_NODE_GRANT_STMT:
X
Xiaoyu Wang 已提交
182
      return makeNode(type, sizeof(SGrantStmt));
183
    case QUERY_NODE_REVOKE_STMT:
X
Xiaoyu Wang 已提交
184
      return makeNode(type, sizeof(SRevokeStmt));
185
    case QUERY_NODE_SHOW_DNODES_STMT:
X
Xiaoyu Wang 已提交
186
    case QUERY_NODE_SHOW_MNODES_STMT:
X
Xiaoyu Wang 已提交
187 188
    case QUERY_NODE_SHOW_MODULES_STMT:
    case QUERY_NODE_SHOW_QNODES_STMT:
189 190
    case QUERY_NODE_SHOW_SNODES_STMT:
    case QUERY_NODE_SHOW_BNODES_STMT:
191
    case QUERY_NODE_SHOW_CLUSTER_STMT:
192
    case QUERY_NODE_SHOW_DATABASES_STMT:
X
Xiaoyu Wang 已提交
193 194
    case QUERY_NODE_SHOW_FUNCTIONS_STMT:
    case QUERY_NODE_SHOW_INDEXES_STMT:
195
    case QUERY_NODE_SHOW_STABLES_STMT:
X
Xiaoyu Wang 已提交
196
    case QUERY_NODE_SHOW_STREAMS_STMT:
197 198
    case QUERY_NODE_SHOW_TABLES_STMT:
    case QUERY_NODE_SHOW_USERS_STMT:
S
Shengliang Guan 已提交
199
    case QUERY_NODE_SHOW_LICENCE_STMT:
200
    case QUERY_NODE_SHOW_VGROUPS_STMT:
S
Shengliang Guan 已提交
201
    case QUERY_NODE_SHOW_TOPICS_STMT:
202 203 204
    case QUERY_NODE_SHOW_CONSUMERS_STMT:
    case QUERY_NODE_SHOW_SUBSCRIBES_STMT:
    case QUERY_NODE_SHOW_SMAS_STMT:
205
    case QUERY_NODE_SHOW_CONNECTIONS_STMT:
206 207
    case QUERY_NODE_SHOW_QUERIES_STMT:
    case QUERY_NODE_SHOW_VNODES_STMT:
208 209 210
    case QUERY_NODE_SHOW_APPS_STMT:
    case QUERY_NODE_SHOW_SCORES_STMT:
    case QUERY_NODE_SHOW_VARIABLE_STMT:
211 212
    case QUERY_NODE_SHOW_TRANSACTIONS_STMT:
      return makeNode(type, sizeof(SShowStmt));
213
    case QUERY_NODE_SHOW_CREATE_DATABASE_STMT:
214
      return makeNode(type, sizeof(SShowCreateDatabaseStmt));
215 216
    case QUERY_NODE_SHOW_CREATE_TABLE_STMT:
    case QUERY_NODE_SHOW_CREATE_STABLE_STMT:
217
      return makeNode(type, sizeof(SShowCreateTableStmt));
218 219
    case QUERY_NODE_SHOW_TABLE_DISTRIBUTED_STMT:
      return makeNode(type, sizeof(SShowTableDistributedStmt));
220
    case QUERY_NODE_KILL_QUERY_STMT:
X
Xiaoyu Wang 已提交
221
      return makeNode(type, sizeof(SKillQueryStmt));
222
    case QUERY_NODE_KILL_TRANSACTION_STMT:
X
Xiaoyu Wang 已提交
223
    case QUERY_NODE_KILL_CONNECTION_STMT:
X
Xiaoyu Wang 已提交
224
      return makeNode(type, sizeof(SKillStmt));
X
Xiaoyu Wang 已提交
225 226
    case QUERY_NODE_DELETE_STMT:
      return makeNode(type, sizeof(SDeleteStmt));
227 228
    case QUERY_NODE_QUERY:
      return makeNode(type, sizeof(SQuery));
X
Xiaoyu Wang 已提交
229
    case QUERY_NODE_LOGIC_PLAN_SCAN:
X
Xiaoyu Wang 已提交
230
      return makeNode(type, sizeof(SScanLogicNode));
X
Xiaoyu Wang 已提交
231
    case QUERY_NODE_LOGIC_PLAN_JOIN:
X
Xiaoyu Wang 已提交
232
      return makeNode(type, sizeof(SJoinLogicNode));
X
Xiaoyu Wang 已提交
233
    case QUERY_NODE_LOGIC_PLAN_AGG:
X
Xiaoyu Wang 已提交
234
      return makeNode(type, sizeof(SAggLogicNode));
X
Xiaoyu Wang 已提交
235
    case QUERY_NODE_LOGIC_PLAN_PROJECT:
X
Xiaoyu Wang 已提交
236
      return makeNode(type, sizeof(SProjectLogicNode));
X
Xiaoyu Wang 已提交
237 238
    case QUERY_NODE_LOGIC_PLAN_VNODE_MODIFY:
      return makeNode(type, sizeof(SVnodeModifyLogicNode));
X
Xiaoyu Wang 已提交
239
    case QUERY_NODE_LOGIC_PLAN_EXCHANGE:
X
Xiaoyu Wang 已提交
240
      return makeNode(type, sizeof(SExchangeLogicNode));
X
Xiaoyu Wang 已提交
241 242
    case QUERY_NODE_LOGIC_PLAN_MERGE:
      return makeNode(type, sizeof(SMergeLogicNode));
X
Xiaoyu Wang 已提交
243
    case QUERY_NODE_LOGIC_PLAN_WINDOW:
X
Xiaoyu Wang 已提交
244
      return makeNode(type, sizeof(SWindowLogicNode));
X
Xiaoyu Wang 已提交
245
    case QUERY_NODE_LOGIC_PLAN_FILL:
X
Xiaoyu Wang 已提交
246
      return makeNode(type, sizeof(SFillLogicNode));
X
Xiaoyu Wang 已提交
247
    case QUERY_NODE_LOGIC_PLAN_SORT:
X
Xiaoyu Wang 已提交
248
      return makeNode(type, sizeof(SSortLogicNode));
249
    case QUERY_NODE_LOGIC_PLAN_PARTITION:
X
Xiaoyu Wang 已提交
250
      return makeNode(type, sizeof(SPartitionLogicNode));
251 252
    case QUERY_NODE_LOGIC_PLAN_INDEF_ROWS_FUNC:
      return makeNode(type, sizeof(SIndefRowsFuncLogicNode));
X
Xiaoyu Wang 已提交
253
    case QUERY_NODE_LOGIC_SUBPLAN:
X
Xiaoyu Wang 已提交
254
      return makeNode(type, sizeof(SLogicSubplan));
X
Xiaoyu Wang 已提交
255
    case QUERY_NODE_LOGIC_PLAN:
X
Xiaoyu Wang 已提交
256
      return makeNode(type, sizeof(SQueryLogicPlan));
X
Xiaoyu Wang 已提交
257
    case QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN:
X
Xiaoyu Wang 已提交
258
      return makeNode(type, sizeof(STagScanPhysiNode));
X
Xiaoyu Wang 已提交
259
    case QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN:
X
Xiaoyu Wang 已提交
260
      return makeNode(type, sizeof(STableScanPhysiNode));
261
    case QUERY_NODE_PHYSICAL_PLAN_TABLE_SEQ_SCAN:
X
Xiaoyu Wang 已提交
262
      return makeNode(type, sizeof(STableSeqScanPhysiNode));
X
Xiaoyu Wang 已提交
263 264
    case QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN:
      return makeNode(type, sizeof(STableMergeScanPhysiNode));
265
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
X
Xiaoyu Wang 已提交
266
      return makeNode(type, sizeof(SStreamScanPhysiNode));
X
Xiaoyu Wang 已提交
267
    case QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN:
X
Xiaoyu Wang 已提交
268
      return makeNode(type, sizeof(SSystemTableScanPhysiNode));
269 270
    case QUERY_NODE_PHYSICAL_PLAN_BLOCK_DIST_SCAN:
      return makeNode(type, sizeof(SBlockDistScanPhysiNode));
X
Xiaoyu Wang 已提交
271
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
X
Xiaoyu Wang 已提交
272
      return makeNode(type, sizeof(SProjectPhysiNode));
273
    case QUERY_NODE_PHYSICAL_PLAN_MERGE_JOIN:
X
Xiaoyu Wang 已提交
274
      return makeNode(type, sizeof(SJoinPhysiNode));
275
    case QUERY_NODE_PHYSICAL_PLAN_HASH_AGG:
X
Xiaoyu Wang 已提交
276
      return makeNode(type, sizeof(SAggPhysiNode));
277
    case QUERY_NODE_PHYSICAL_PLAN_EXCHANGE:
X
Xiaoyu Wang 已提交
278
      return makeNode(type, sizeof(SExchangePhysiNode));
X
Xiaoyu Wang 已提交
279 280
    case QUERY_NODE_PHYSICAL_PLAN_MERGE:
      return makeNode(type, sizeof(SMergePhysiNode));
281
    case QUERY_NODE_PHYSICAL_PLAN_SORT:
X
Xiaoyu Wang 已提交
282
      return makeNode(type, sizeof(SSortPhysiNode));
X
Xiaoyu Wang 已提交
283
    case QUERY_NODE_PHYSICAL_PLAN_HASH_INTERVAL:
X
Xiaoyu Wang 已提交
284
      return makeNode(type, sizeof(SIntervalPhysiNode));
X
Xiaoyu Wang 已提交
285 286
    case QUERY_NODE_PHYSICAL_PLAN_MERGE_INTERVAL:
      return makeNode(type, sizeof(SMergeIntervalPhysiNode));
X
Xiaoyu Wang 已提交
287
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL:
X
Xiaoyu Wang 已提交
288
      return makeNode(type, sizeof(SStreamIntervalPhysiNode));
289 290 291 292
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_INTERVAL:
      return makeNode(type, sizeof(SStreamFinalIntervalPhysiNode));
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_INTERVAL:
      return makeNode(type, sizeof(SStreamSemiIntervalPhysiNode));
X
Xiaoyu Wang 已提交
293
    case QUERY_NODE_PHYSICAL_PLAN_FILL:
X
Xiaoyu Wang 已提交
294
      return makeNode(type, sizeof(SFillPhysiNode));
295
    case QUERY_NODE_PHYSICAL_PLAN_MERGE_SESSION:
X
Xiaoyu Wang 已提交
296
      return makeNode(type, sizeof(SSessionWinodwPhysiNode));
297
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SESSION:
X
Xiaoyu Wang 已提交
298
      return makeNode(type, sizeof(SStreamSessionWinodwPhysiNode));
299 300 301 302
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_SESSION:
      return makeNode(type, sizeof(SStreamSemiSessionWinodwPhysiNode));
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_SESSION:
      return makeNode(type, sizeof(SStreamFinalSessionWinodwPhysiNode));
303
    case QUERY_NODE_PHYSICAL_PLAN_MERGE_STATE:
X
Xiaoyu Wang 已提交
304
      return makeNode(type, sizeof(SStateWinodwPhysiNode));
305
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_STATE:
5
54liuyao 已提交
306
      return makeNode(type, sizeof(SStreamStateWinodwPhysiNode));
307
    case QUERY_NODE_PHYSICAL_PLAN_PARTITION:
X
Xiaoyu Wang 已提交
308
      return makeNode(type, sizeof(SPartitionPhysiNode));
309 310
    case QUERY_NODE_PHYSICAL_PLAN_INDEF_ROWS_FUNC:
      return makeNode(type, sizeof(SIndefRowsFuncPhysiNode));
311
    case QUERY_NODE_PHYSICAL_PLAN_DISPATCH:
X
Xiaoyu Wang 已提交
312
      return makeNode(type, sizeof(SDataDispatcherNode));
313
    case QUERY_NODE_PHYSICAL_PLAN_INSERT:
X
Xiaoyu Wang 已提交
314
      return makeNode(type, sizeof(SDataInserterNode));
X
Xiaoyu Wang 已提交
315 316
    case QUERY_NODE_PHYSICAL_PLAN_DELETE:
      return makeNode(type, sizeof(SDataDeleterNode));
X
Xiaoyu Wang 已提交
317
    case QUERY_NODE_PHYSICAL_SUBPLAN:
X
Xiaoyu Wang 已提交
318
      return makeNode(type, sizeof(SSubplan));
X
Xiaoyu Wang 已提交
319
    case QUERY_NODE_PHYSICAL_PLAN:
X
Xiaoyu Wang 已提交
320
      return makeNode(type, sizeof(SQueryPlan));
321 322 323
    default:
      break;
  }
324
  nodesError("nodesMakeNode unknown node = %s", nodesNodeName(type));
X
Xiaoyu Wang 已提交
325
  return NULL;
326 327
}

X
Xiaoyu Wang 已提交
328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346
static void destroyVgDataBlockArray(SArray* pArray) {
  size_t size = taosArrayGetSize(pArray);
  for (size_t i = 0; i < size; ++i) {
    SVgDataBlocks* pVg = taosArrayGetP(pArray, i);
    taosMemoryFreeClear(pVg->pData);
    taosMemoryFreeClear(pVg);
  }
  taosArrayDestroy(pArray);
}

static void destroyLogicNode(SLogicNode* pNode) {
  nodesDestroyList(pNode->pChildren);
  nodesDestroyNode(pNode->pConditions);
  nodesDestroyList(pNode->pTargets);
}

static void destroyPhysiNode(SPhysiNode* pNode) {
  nodesDestroyList(pNode->pChildren);
  nodesDestroyNode(pNode->pConditions);
347
  nodesDestroyNode((SNode*)pNode->pOutputDataBlockDesc);
X
Xiaoyu Wang 已提交
348 349 350 351 352 353
}

static void destroyWinodwPhysiNode(SWinodwPhysiNode* pNode) {
  destroyPhysiNode((SPhysiNode*)pNode);
  nodesDestroyList(pNode->pExprs);
  nodesDestroyList(pNode->pFuncs);
354
  nodesDestroyNode(pNode->pTspk);
X
Xiaoyu Wang 已提交
355 356 357 358 359 360 361
}

static void destroyScanPhysiNode(SScanPhysiNode* pNode) {
  destroyPhysiNode((SPhysiNode*)pNode);
  nodesDestroyList(pNode->pScanCols);
}

362
static void destroyDataSinkNode(SDataSinkNode* pNode) { nodesDestroyNode((SNode*)pNode->pInputDataBlockDesc); }
X
Xiaoyu Wang 已提交
363

364
void nodesDestroyNode(SNode* pNode) {
X
Xiaoyu Wang 已提交
365 366 367 368 369
  if (NULL == pNode) {
    return;
  }

  switch (nodeType(pNode)) {
X
Xiaoyu Wang 已提交
370
    case QUERY_NODE_COLUMN:  // pProjectRef is weak reference, no need to release
X
Xiaoyu Wang 已提交
371
      break;
D
dapan1121 已提交
372
    case QUERY_NODE_VALUE: {
X
Xiaoyu Wang 已提交
373
      SValueNode* pValue = (SValueNode*)pNode;
wafwerar's avatar
wafwerar 已提交
374
      taosMemoryFreeClear(pValue->literal);
D
dapan1121 已提交
375
      if (IS_VAR_DATA_TYPE(pValue->node.resType.type)) {
wafwerar's avatar
wafwerar 已提交
376
        taosMemoryFreeClear(pValue->datum.p);
D
dapan1121 已提交
377
      }
X
Xiaoyu Wang 已提交
378 379 380 381 382 383
      break;
    }
    case QUERY_NODE_OPERATOR: {
      SOperatorNode* pOp = (SOperatorNode*)pNode;
      nodesDestroyNode(pOp->pLeft);
      nodesDestroyNode(pOp->pRight);
D
dapan1121 已提交
384 385 386
      break;
    }
    case QUERY_NODE_LOGIC_CONDITION:
X
Xiaoyu Wang 已提交
387
      nodesDestroyList(((SLogicConditionNode*)pNode)->pParameterList);
D
dapan1121 已提交
388 389
      break;
    case QUERY_NODE_FUNCTION:
X
Xiaoyu Wang 已提交
390
      nodesDestroyList(((SFunctionNode*)pNode)->pParameterList);
X
Xiaoyu Wang 已提交
391 392
      break;
    case QUERY_NODE_REAL_TABLE: {
X
Xiaoyu Wang 已提交
393
      SRealTableNode* pReal = (SRealTableNode*)pNode;
wafwerar's avatar
wafwerar 已提交
394 395
      taosMemoryFreeClear(pReal->pMeta);
      taosMemoryFreeClear(pReal->pVgroupList);
X
Xiaoyu Wang 已提交
396 397 398
      break;
    }
    case QUERY_NODE_TEMP_TABLE:
X
Xiaoyu Wang 已提交
399
      nodesDestroyNode(((STempTableNode*)pNode)->pSubquery);
D
dapan1121 已提交
400
      break;
X
Xiaoyu Wang 已提交
401 402 403 404 405
    case QUERY_NODE_JOIN_TABLE: {
      SJoinTableNode* pJoin = (SJoinTableNode*)pNode;
      nodesDestroyNode(pJoin->pLeft);
      nodesDestroyNode(pJoin->pRight);
      nodesDestroyNode(pJoin->pOnCond);
D
dapan1121 已提交
406
      break;
X
Xiaoyu Wang 已提交
407
    }
D
dapan1121 已提交
408
    case QUERY_NODE_GROUPING_SET:
X
Xiaoyu Wang 已提交
409 410 411 412 413
      nodesDestroyList(((SGroupingSetNode*)pNode)->pParameterList);
      break;
    case QUERY_NODE_ORDER_BY_EXPR:
      nodesDestroyNode(((SOrderByExprNode*)pNode)->pExpr);
      break;
X
Xiaoyu Wang 已提交
414
    case QUERY_NODE_LIMIT:  // no pointer field
D
dapan1121 已提交
415
      break;
X
Xiaoyu Wang 已提交
416
    case QUERY_NODE_STATE_WINDOW:
417
      nodesDestroyNode(((SStateWindowNode*)pNode)->pExpr);
X
Xiaoyu Wang 已提交
418 419 420
      break;
    case QUERY_NODE_SESSION_WINDOW: {
      SSessionWindowNode* pSession = (SSessionWindowNode*)pNode;
421 422
      nodesDestroyNode((SNode*)pSession->pCol);
      nodesDestroyNode((SNode*)pSession->pGap);
D
dapan1121 已提交
423
      break;
X
Xiaoyu Wang 已提交
424 425 426 427 428 429 430 431 432 433
    }
    case QUERY_NODE_INTERVAL_WINDOW: {
      SIntervalWindowNode* pJoin = (SIntervalWindowNode*)pNode;
      nodesDestroyNode(pJoin->pCol);
      nodesDestroyNode(pJoin->pInterval);
      nodesDestroyNode(pJoin->pOffset);
      nodesDestroyNode(pJoin->pSliding);
      nodesDestroyNode(pJoin->pFill);
      break;
    }
D
dapan1121 已提交
434
    case QUERY_NODE_NODE_LIST:
X
Xiaoyu Wang 已提交
435 436
      nodesDestroyList(((SNodeListNode*)pNode)->pNodeList);
      break;
X
Xiaoyu Wang 已提交
437 438 439 440
    case QUERY_NODE_FILL: {
      SFillNode* pFill = (SFillNode*)pNode;
      nodesDestroyNode(pFill->pValues);
      nodesDestroyNode(pFill->pWStartTs);
X
Xiaoyu Wang 已提交
441
      break;
X
Xiaoyu Wang 已提交
442
    }
X
Xiaoyu Wang 已提交
443 444
    case QUERY_NODE_RAW_EXPR:
      nodesDestroyNode(((SRawExprNode*)pNode)->pNode);
X
Xiaoyu Wang 已提交
445
      break;
X
Xiaoyu Wang 已提交
446 447 448 449 450 451
    case QUERY_NODE_TARGET:
      nodesDestroyNode(((STargetNode*)pNode)->pExpr);
      break;
    case QUERY_NODE_DATABLOCK_DESC:
      nodesDestroyList(((SDataBlockDescNode*)pNode)->pSlots);
      break;
X
Xiaoyu Wang 已提交
452 453 454
    case QUERY_NODE_SLOT_DESC:          // no pointer field
    case QUERY_NODE_COLUMN_DEF:         // no pointer field
    case QUERY_NODE_DOWNSTREAM_SOURCE:  // no pointer field
X
Xiaoyu Wang 已提交
455
      break;
X
Xiaoyu Wang 已提交
456 457
    case QUERY_NODE_DATABASE_OPTIONS: {
      SDatabaseOptions* pOptions = (SDatabaseOptions*)pNode;
458
      nodesDestroyNode((SNode*)pOptions->pDaysPerFile);
X
Xiaoyu Wang 已提交
459 460
      nodesDestroyList(pOptions->pKeep);
      nodesDestroyList(pOptions->pRetentions);
X
Xiaoyu Wang 已提交
461
      break;
X
Xiaoyu Wang 已提交
462
    }
X
Xiaoyu Wang 已提交
463
    case QUERY_NODE_TABLE_OPTIONS: {
X
Xiaoyu Wang 已提交
464 465 466
      STableOptions* pOptions = (STableOptions*)pNode;
      nodesDestroyList(pOptions->pSma);
      nodesDestroyList(pOptions->pRollupFuncs);
X
Xiaoyu Wang 已提交
467
      break;
X
Xiaoyu Wang 已提交
468
    }
469
    case QUERY_NODE_INDEX_OPTIONS: {
X
Xiaoyu Wang 已提交
470 471 472 473 474
      SIndexOptions* pOptions = (SIndexOptions*)pNode;
      nodesDestroyList(pOptions->pFuncs);
      nodesDestroyNode(pOptions->pInterval);
      nodesDestroyNode(pOptions->pOffset);
      nodesDestroyNode(pOptions->pSliding);
475 476
      break;
    }
477 478 479 480 481 482 483
    case QUERY_NODE_EXPLAIN_OPTIONS:  // no pointer field
      break;
    case QUERY_NODE_STREAM_OPTIONS:
      nodesDestroyNode(((SStreamOptions*)pNode)->pWatermark);
      break;
    case QUERY_NODE_LEFT_VALUE:  // no pointer field
      break;
X
Xiaoyu Wang 已提交
484 485 486 487 488 489 490 491
    case QUERY_NODE_SET_OPERATOR: {
      SSetOperator* pStmt = (SSetOperator*)pNode;
      nodesDestroyNode(pStmt->pLeft);
      nodesDestroyNode(pStmt->pRight);
      nodesDestroyList(pStmt->pOrderByList);
      nodesDestroyNode(pStmt->pLimit);
      break;
    }
X
Xiaoyu Wang 已提交
492
    case QUERY_NODE_SELECT_STMT: {
X
Xiaoyu Wang 已提交
493
      SSelectStmt* pStmt = (SSelectStmt*)pNode;
X
Xiaoyu Wang 已提交
494 495 496 497 498 499 500 501
      nodesDestroyList(pStmt->pProjectionList);
      nodesDestroyNode(pStmt->pFromTable);
      nodesDestroyNode(pStmt->pWhere);
      nodesDestroyList(pStmt->pPartitionByList);
      nodesDestroyNode(pStmt->pWindow);
      nodesDestroyList(pStmt->pGroupByList);
      nodesDestroyNode(pStmt->pHaving);
      nodesDestroyList(pStmt->pOrderByList);
502 503
      nodesDestroyNode((SNode*)pStmt->pLimit);
      nodesDestroyNode((SNode*)pStmt->pSlimit);
X
Xiaoyu Wang 已提交
504 505
      break;
    }
X
Xiaoyu Wang 已提交
506 507 508 509
    case QUERY_NODE_VNODE_MODIF_STMT:
      destroyVgDataBlockArray(((SVnodeModifOpStmt*)pNode)->pDataBlocks);
      break;
    case QUERY_NODE_CREATE_DATABASE_STMT:
510
      nodesDestroyNode((SNode*)((SCreateDatabaseStmt*)pNode)->pOptions);
X
Xiaoyu Wang 已提交
511
      break;
X
Xiaoyu Wang 已提交
512
    case QUERY_NODE_DROP_DATABASE_STMT:  // no pointer field
X
Xiaoyu Wang 已提交
513 514
      break;
    case QUERY_NODE_ALTER_DATABASE_STMT:
515
      nodesDestroyNode((SNode*)((SAlterDatabaseStmt*)pNode)->pOptions);
X
Xiaoyu Wang 已提交
516
      break;
X
Xiaoyu Wang 已提交
517
    case QUERY_NODE_CREATE_TABLE_STMT: {
X
Xiaoyu Wang 已提交
518
      SCreateTableStmt* pStmt = (SCreateTableStmt*)pNode;
X
Xiaoyu Wang 已提交
519 520
      nodesDestroyList(pStmt->pCols);
      nodesDestroyList(pStmt->pTags);
521
      nodesDestroyNode((SNode*)pStmt->pOptions);
X
Xiaoyu Wang 已提交
522 523 524
      break;
    }
    case QUERY_NODE_CREATE_SUBTABLE_CLAUSE: {
X
Xiaoyu Wang 已提交
525
      SCreateSubTableClause* pStmt = (SCreateSubTableClause*)pNode;
X
Xiaoyu Wang 已提交
526 527 528 529 530
      nodesDestroyList(pStmt->pSpecificTags);
      nodesDestroyList(pStmt->pValsOfTags);
      break;
    }
    case QUERY_NODE_CREATE_MULTI_TABLE_STMT:
X
Xiaoyu Wang 已提交
531 532
      nodesDestroyList(((SCreateMultiTableStmt*)pNode)->pSubTables);
      break;
X
Xiaoyu Wang 已提交
533
    case QUERY_NODE_DROP_TABLE_CLAUSE:  // no pointer field
X
Xiaoyu Wang 已提交
534 535
      break;
    case QUERY_NODE_DROP_TABLE_STMT:
536
      nodesDestroyList(((SDropTableStmt*)pNode)->pTables);
X
Xiaoyu Wang 已提交
537
      break;
X
Xiaoyu Wang 已提交
538
    case QUERY_NODE_DROP_SUPER_TABLE_STMT:  // no pointer field
X
Xiaoyu Wang 已提交
539 540 541
      break;
    case QUERY_NODE_ALTER_TABLE_STMT: {
      SAlterTableStmt* pStmt = (SAlterTableStmt*)pNode;
542 543
      nodesDestroyNode((SNode*)pStmt->pOptions);
      nodesDestroyNode((SNode*)pStmt->pVal);
X
Xiaoyu Wang 已提交
544 545
      break;
    }
X
Xiaoyu Wang 已提交
546 547 548 549 550 551 552
    case QUERY_NODE_CREATE_USER_STMT:   // no pointer field
    case QUERY_NODE_ALTER_USER_STMT:    // no pointer field
    case QUERY_NODE_DROP_USER_STMT:     // no pointer field
    case QUERY_NODE_USE_DATABASE_STMT:  // no pointer field
    case QUERY_NODE_CREATE_DNODE_STMT:  // no pointer field
    case QUERY_NODE_DROP_DNODE_STMT:    // no pointer field
    case QUERY_NODE_ALTER_DNODE_STMT:   // no pointer field
553
      break;
554
    case QUERY_NODE_CREATE_INDEX_STMT: {
X
Xiaoyu Wang 已提交
555
      SCreateIndexStmt* pStmt = (SCreateIndexStmt*)pNode;
556
      nodesDestroyNode((SNode*)pStmt->pOptions);
557 558 559
      nodesDestroyList(pStmt->pCols);
      break;
    }
X
Xiaoyu Wang 已提交
560 561 562
    case QUERY_NODE_DROP_INDEX_STMT:    // no pointer field
    case QUERY_NODE_CREATE_QNODE_STMT:  // no pointer field
    case QUERY_NODE_DROP_QNODE_STMT:    // no pointer field
563 564 565 566 567 568
    case QUERY_NODE_CREATE_BNODE_STMT:  // no pointer field
    case QUERY_NODE_DROP_BNODE_STMT:    // no pointer field
    case QUERY_NODE_CREATE_SNODE_STMT:  // no pointer field
    case QUERY_NODE_DROP_SNODE_STMT:    // no pointer field
    case QUERY_NODE_CREATE_MNODE_STMT:  // no pointer field
    case QUERY_NODE_DROP_MNODE_STMT:    // no pointer field
X
Xiaoyu Wang 已提交
569 570 571 572
      break;
    case QUERY_NODE_CREATE_TOPIC_STMT:
      nodesDestroyNode(((SCreateTopicStmt*)pNode)->pQuery);
      break;
X
Xiaoyu Wang 已提交
573
    case QUERY_NODE_DROP_TOPIC_STMT:   // no pointer field
574
    case QUERY_NODE_DROP_CGROUP_STMT:  // no pointer field
X
Xiaoyu Wang 已提交
575
    case QUERY_NODE_ALTER_LOCAL_STMT:  // no pointer field
X
Xiaoyu Wang 已提交
576
      break;
577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608
    case QUERY_NODE_EXPLAIN_STMT: {
      SExplainStmt* pStmt = (SExplainStmt*)pNode;
      nodesDestroyNode((SNode*)pStmt->pOptions);
      nodesDestroyNode(pStmt->pQuery);
      break;
    }
    case QUERY_NODE_DESCRIBE_STMT:
      taosMemoryFree(((SDescribeStmt*)pNode)->pMeta);
      break;
    case QUERY_NODE_RESET_QUERY_CACHE_STMT:  // no pointer field
    case QUERY_NODE_COMPACT_STMT:            // no pointer field
    case QUERY_NODE_CREATE_FUNCTION_STMT:    // no pointer field
    case QUERY_NODE_DROP_FUNCTION_STMT:      // no pointer field
      break;
    case QUERY_NODE_CREATE_STREAM_STMT: {
      SCreateStreamStmt* pStmt = (SCreateStreamStmt*)pNode;
      nodesDestroyNode((SNode*)pStmt->pOptions);
      nodesDestroyNode(pStmt->pQuery);
      break;
    }
    case QUERY_NODE_DROP_STREAM_STMT:     // no pointer field
    case QUERY_NODE_BALANCE_VGROUP_STMT:  // no pointer field
    case QUERY_NODE_MERGE_VGROUP_STMT:    // no pointer field
      break;
    case QUERY_NODE_REDISTRIBUTE_VGROUP_STMT:
      nodesDestroyList(((SRedistributeVgroupStmt*)pNode)->pDnodes);
      break;
    case QUERY_NODE_SPLIT_VGROUP_STMT:  // no pointer field
    case QUERY_NODE_SYNCDB_STMT:        // no pointer field
    case QUERY_NODE_GRANT_STMT:         // no pointer field
    case QUERY_NODE_REVOKE_STMT:        // no pointer field
      break;
X
Xiaoyu Wang 已提交
609 610 611 612
    case QUERY_NODE_SHOW_DNODES_STMT:
    case QUERY_NODE_SHOW_MNODES_STMT:
    case QUERY_NODE_SHOW_MODULES_STMT:
    case QUERY_NODE_SHOW_QNODES_STMT:
613 614 615 616
    case QUERY_NODE_SHOW_SNODES_STMT:
    case QUERY_NODE_SHOW_BNODES_STMT:
    case QUERY_NODE_SHOW_CLUSTER_STMT:
    case QUERY_NODE_SHOW_DATABASES_STMT:
X
Xiaoyu Wang 已提交
617 618
    case QUERY_NODE_SHOW_FUNCTIONS_STMT:
    case QUERY_NODE_SHOW_INDEXES_STMT:
619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635
    case QUERY_NODE_SHOW_STABLES_STMT:
    case QUERY_NODE_SHOW_STREAMS_STMT:
    case QUERY_NODE_SHOW_TABLES_STMT:
    case QUERY_NODE_SHOW_USERS_STMT:
    case QUERY_NODE_SHOW_LICENCE_STMT:
    case QUERY_NODE_SHOW_VGROUPS_STMT:
    case QUERY_NODE_SHOW_TOPICS_STMT:
    case QUERY_NODE_SHOW_CONSUMERS_STMT:
    case QUERY_NODE_SHOW_SUBSCRIBES_STMT:
    case QUERY_NODE_SHOW_SMAS_STMT:
    case QUERY_NODE_SHOW_CONNECTIONS_STMT:
    case QUERY_NODE_SHOW_QUERIES_STMT:
    case QUERY_NODE_SHOW_VNODES_STMT:
    case QUERY_NODE_SHOW_APPS_STMT:
    case QUERY_NODE_SHOW_SCORES_STMT:
    case QUERY_NODE_SHOW_VARIABLE_STMT:
    case QUERY_NODE_SHOW_TRANSACTIONS_STMT: {
X
Xiaoyu Wang 已提交
636 637
      SShowStmt* pStmt = (SShowStmt*)pNode;
      nodesDestroyNode(pStmt->pDbName);
638
      nodesDestroyNode(pStmt->pTbName);
X
Xiaoyu Wang 已提交
639 640
      break;
    }
641 642 643 644 645 646 647 648 649 650 651
    case QUERY_NODE_SHOW_CREATE_DATABASE_STMT:
      taosMemoryFreeClear(((SShowCreateDatabaseStmt*)pNode)->pCfg);
      break;
    case QUERY_NODE_SHOW_CREATE_TABLE_STMT:
    case QUERY_NODE_SHOW_CREATE_STABLE_STMT:
      taosMemoryFreeClear(((SShowCreateTableStmt*)pNode)->pMeta);
      break;
    case QUERY_NODE_SHOW_TABLE_DISTRIBUTED_STMT:  // no pointer field
    case QUERY_NODE_KILL_CONNECTION_STMT:         // no pointer field
    case QUERY_NODE_KILL_QUERY_STMT:              // no pointer field
    case QUERY_NODE_KILL_TRANSACTION_STMT:        // no pointer field
652 653 654 655 656 657 658 659 660
      break;
    case QUERY_NODE_DELETE_STMT: {
      SDeleteStmt* pStmt = (SDeleteStmt*)pNode;
      nodesDestroyNode(pStmt->pFromTable);
      nodesDestroyNode(pStmt->pWhere);
      nodesDestroyNode(pStmt->pCountFunc);
      nodesDestroyNode(pStmt->pTagIndexCond);
      break;
    }
661 662 663 664 665 666 667 668 669 670 671 672
    case QUERY_NODE_QUERY: {
      SQuery* pQuery = (SQuery*)pNode;
      nodesDestroyNode(pQuery->pRoot);
      taosMemoryFreeClear(pQuery->pResSchema);
      if (NULL != pQuery->pCmdMsg) {
        taosMemoryFreeClear(pQuery->pCmdMsg->pMsg);
        taosMemoryFreeClear(pQuery->pCmdMsg);
      }
      taosArrayDestroy(pQuery->pDbList);
      taosArrayDestroy(pQuery->pTableList);
      break;
    }
X
Xiaoyu Wang 已提交
673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698
    case QUERY_NODE_LOGIC_PLAN_SCAN: {
      SScanLogicNode* pLogicNode = (SScanLogicNode*)pNode;
      destroyLogicNode((SLogicNode*)pLogicNode);
      nodesDestroyList(pLogicNode->pScanCols);
      taosMemoryFreeClear(pLogicNode->pVgroupList);
      break;
    }
    case QUERY_NODE_LOGIC_PLAN_JOIN: {
      SJoinLogicNode* pLogicNode = (SJoinLogicNode*)pNode;
      destroyLogicNode((SLogicNode*)pLogicNode);
      nodesDestroyNode(pLogicNode->pOnConditions);
      break;
    }
    case QUERY_NODE_LOGIC_PLAN_AGG: {
      SAggLogicNode* pLogicNode = (SAggLogicNode*)pNode;
      destroyLogicNode((SLogicNode*)pLogicNode);
      nodesDestroyList(pLogicNode->pAggFuncs);
      nodesDestroyList(pLogicNode->pGroupKeys);
      break;
    }
    case QUERY_NODE_LOGIC_PLAN_PROJECT: {
      SProjectLogicNode* pLogicNode = (SProjectLogicNode*)pNode;
      destroyLogicNode((SLogicNode*)pLogicNode);
      nodesDestroyList(pLogicNode->pProjections);
      break;
    }
X
Xiaoyu Wang 已提交
699 700
    case QUERY_NODE_LOGIC_PLAN_VNODE_MODIFY: {
      SVnodeModifyLogicNode* pLogicNode = (SVnodeModifyLogicNode*)pNode;
X
Xiaoyu Wang 已提交
701 702
      destroyLogicNode((SLogicNode*)pLogicNode);
      destroyVgDataBlockArray(pLogicNode->pDataBlocks);
X
bugfix  
Xiaoyu Wang 已提交
703
      // pVgDataBlocks is weak reference
X
Xiaoyu Wang 已提交
704 705 706 707 708
      break;
    }
    case QUERY_NODE_LOGIC_PLAN_EXCHANGE:
      destroyLogicNode((SLogicNode*)pNode);
      break;
709 710 711 712 713 714 715
    case QUERY_NODE_LOGIC_PLAN_MERGE: {
      SMergeLogicNode* pLogicNode = (SMergeLogicNode*)pNode;
      destroyLogicNode((SLogicNode*)pLogicNode);
      nodesDestroyList(pLogicNode->pMergeKeys);
      nodesDestroyList(pLogicNode->pInputs);
      break;
    }
X
Xiaoyu Wang 已提交
716 717 718 719 720 721 722
    case QUERY_NODE_LOGIC_PLAN_WINDOW: {
      SWindowLogicNode* pLogicNode = (SWindowLogicNode*)pNode;
      destroyLogicNode((SLogicNode*)pLogicNode);
      nodesDestroyList(pLogicNode->pFuncs);
      nodesDestroyNode(pLogicNode->pTspk);
      break;
    }
723 724 725 726 727 728 729
    case QUERY_NODE_LOGIC_PLAN_FILL: {
      SFillLogicNode* pLogicNode = (SFillLogicNode*)pNode;
      destroyLogicNode((SLogicNode*)pLogicNode);
      nodesDestroyNode(pLogicNode->pWStartTs);
      nodesDestroyNode(pLogicNode->pValues);
      break;
    }
X
Xiaoyu Wang 已提交
730 731 732 733 734 735 736 737 738 739 740 741
    case QUERY_NODE_LOGIC_PLAN_SORT: {
      SSortLogicNode* pLogicNode = (SSortLogicNode*)pNode;
      destroyLogicNode((SLogicNode*)pLogicNode);
      nodesDestroyList(pLogicNode->pSortKeys);
      break;
    }
    case QUERY_NODE_LOGIC_PLAN_PARTITION: {
      SPartitionLogicNode* pLogicNode = (SPartitionLogicNode*)pNode;
      destroyLogicNode((SLogicNode*)pLogicNode);
      nodesDestroyList(pLogicNode->pPartitionKeys);
      break;
    }
742 743 744 745 746 747
    case QUERY_NODE_LOGIC_PLAN_INDEF_ROWS_FUNC: {
      SIndefRowsFuncLogicNode* pLogicNode = (SIndefRowsFuncLogicNode*)pNode;
      destroyLogicNode((SLogicNode*)pLogicNode);
      nodesDestroyList(pLogicNode->pVectorFuncs);
      break;
    }
X
Xiaoyu Wang 已提交
748 749 750
    case QUERY_NODE_LOGIC_SUBPLAN: {
      SLogicSubplan* pSubplan = (SLogicSubplan*)pNode;
      nodesDestroyList(pSubplan->pChildren);
751
      nodesDestroyNode((SNode*)pSubplan->pNode);
X
bugfix  
Xiaoyu Wang 已提交
752
      nodesClearList(pSubplan->pParents);
X
Xiaoyu Wang 已提交
753 754 755 756 757 758 759 760 761 762 763
      taosMemoryFreeClear(pSubplan->pVgroupList);
      break;
    }
    case QUERY_NODE_LOGIC_PLAN:
      nodesDestroyList(((SQueryLogicPlan*)pNode)->pTopSubplans);
      break;
    case QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN:
    case QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN:
    case QUERY_NODE_PHYSICAL_PLAN_TABLE_SEQ_SCAN:
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
    case QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN:
764
    case QUERY_NODE_PHYSICAL_PLAN_BLOCK_DIST_SCAN:
X
Xiaoyu Wang 已提交
765 766 767 768 769 770 771 772
      destroyScanPhysiNode((SScanPhysiNode*)pNode);
      break;
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT: {
      SProjectPhysiNode* pPhyNode = (SProjectPhysiNode*)pNode;
      destroyPhysiNode((SPhysiNode*)pPhyNode);
      nodesDestroyList(pPhyNode->pProjections);
      break;
    }
773
    case QUERY_NODE_PHYSICAL_PLAN_MERGE_JOIN: {
X
Xiaoyu Wang 已提交
774 775 776 777 778 779
      SJoinPhysiNode* pPhyNode = (SJoinPhysiNode*)pNode;
      destroyPhysiNode((SPhysiNode*)pPhyNode);
      nodesDestroyNode(pPhyNode->pOnConditions);
      nodesDestroyList(pPhyNode->pTargets);
      break;
    }
780
    case QUERY_NODE_PHYSICAL_PLAN_HASH_AGG: {
X
Xiaoyu Wang 已提交
781 782 783 784 785 786 787 788 789 790 791 792 793
      SAggPhysiNode* pPhyNode = (SAggPhysiNode*)pNode;
      destroyPhysiNode((SPhysiNode*)pPhyNode);
      nodesDestroyList(pPhyNode->pExprs);
      nodesDestroyList(pPhyNode->pAggFuncs);
      nodesDestroyList(pPhyNode->pGroupKeys);
      break;
    }
    case QUERY_NODE_PHYSICAL_PLAN_EXCHANGE: {
      SExchangePhysiNode* pPhyNode = (SExchangePhysiNode*)pNode;
      destroyPhysiNode((SPhysiNode*)pPhyNode);
      nodesDestroyList(pPhyNode->pSrcEndPoints);
      break;
    }
794 795 796 797 798 799 800
    case QUERY_NODE_PHYSICAL_PLAN_MERGE: {
      SMergePhysiNode* pPhyNode = (SMergePhysiNode*)pNode;
      destroyPhysiNode((SPhysiNode*)pPhyNode);
      nodesDestroyList(pPhyNode->pMergeKeys);
      nodesDestroyList(pPhyNode->pTargets);
      break;
    }
X
Xiaoyu Wang 已提交
801 802 803
    case QUERY_NODE_PHYSICAL_PLAN_SORT: {
      SSortPhysiNode* pPhyNode = (SSortPhysiNode*)pNode;
      destroyPhysiNode((SPhysiNode*)pPhyNode);
804 805
      nodesDestroyList(pPhyNode->pExprs);
      nodesDestroyList(pPhyNode->pSortKeys);
X
Xiaoyu Wang 已提交
806 807
      break;
    }
X
Xiaoyu Wang 已提交
808
    case QUERY_NODE_PHYSICAL_PLAN_HASH_INTERVAL:
809
    case QUERY_NODE_PHYSICAL_PLAN_MERGE_INTERVAL:
X
Xiaoyu Wang 已提交
810
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL:
811 812
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_INTERVAL:
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_INTERVAL:
X
Xiaoyu Wang 已提交
813
      destroyWinodwPhysiNode((SWinodwPhysiNode*)pNode);
X
Xiaoyu Wang 已提交
814
      break;
815 816 817 818 819 820 821 822
    case QUERY_NODE_PHYSICAL_PLAN_FILL: {
      SFillPhysiNode* pPhyNode = (SFillPhysiNode*)pNode;
      destroyPhysiNode((SPhysiNode*)pPhyNode);
      nodesDestroyNode(pPhyNode->pWStartTs);
      nodesDestroyNode(pPhyNode->pValues);
      nodesDestroyList(pPhyNode->pTargets);
      break;
    }
823 824
    case QUERY_NODE_PHYSICAL_PLAN_MERGE_SESSION:
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SESSION:
825
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_SESSION:
826
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_SESSION:
X
Xiaoyu Wang 已提交
827 828
      destroyWinodwPhysiNode((SWinodwPhysiNode*)pNode);
      break;
829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850
    case QUERY_NODE_PHYSICAL_PLAN_MERGE_STATE:
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_STATE: {
      SStateWinodwPhysiNode* pPhyNode = (SStateWinodwPhysiNode*)pNode;
      destroyWinodwPhysiNode((SWinodwPhysiNode*)pPhyNode);
      nodesDestroyNode(pPhyNode->pStateKey);
      break;
    }
    case QUERY_NODE_PHYSICAL_PLAN_PARTITION: {
      SPartitionPhysiNode* pPhyNode = (SPartitionPhysiNode*)pNode;
      destroyPhysiNode((SPhysiNode*)pPhyNode);
      nodesDestroyList(pPhyNode->pExprs);
      nodesDestroyList(pPhyNode->pPartitionKeys);
      nodesDestroyList(pPhyNode->pTargets);
      break;
    }
    case QUERY_NODE_PHYSICAL_PLAN_INDEF_ROWS_FUNC: {
      SIndefRowsFuncPhysiNode* pPhyNode = (SIndefRowsFuncPhysiNode*)pNode;
      destroyPhysiNode((SPhysiNode*)pPhyNode);
      nodesDestroyList(pPhyNode->pExprs);
      nodesDestroyList(pPhyNode->pVectorFuncs);
      break;
    }
X
Xiaoyu Wang 已提交
851 852 853 854 855 856 857 858 859
    case QUERY_NODE_PHYSICAL_PLAN_DISPATCH:
      destroyDataSinkNode((SDataSinkNode*)pNode);
      break;
    case QUERY_NODE_PHYSICAL_PLAN_INSERT: {
      SDataInserterNode* pSink = (SDataInserterNode*)pNode;
      destroyDataSinkNode((SDataSinkNode*)pSink);
      taosMemoryFreeClear(pSink->pData);
      break;
    }
860 861 862 863 864 865
    case QUERY_NODE_PHYSICAL_PLAN_DELETE: {
      SDataDeleterNode* pSink = (SDataDeleterNode*)pNode;
      destroyDataSinkNode((SDataSinkNode*)pSink);
      nodesDestroyNode(pSink->pAffectedRows);
      break;
    }
X
Xiaoyu Wang 已提交
866 867 868
    case QUERY_NODE_PHYSICAL_SUBPLAN: {
      SSubplan* pSubplan = (SSubplan*)pNode;
      nodesDestroyList(pSubplan->pChildren);
869 870
      nodesDestroyNode((SNode*)pSubplan->pNode);
      nodesDestroyNode((SNode*)pSubplan->pDataSink);
X
bugfix  
Xiaoyu Wang 已提交
871 872 873 874 875 876 877
      nodesClearList(pSubplan->pParents);
      break;
    }
    case QUERY_NODE_PHYSICAL_PLAN: {
      SQueryPlan* pPlan = (SQueryPlan*)pNode;
      if (NULL != pPlan->pSubplans) {
        // only need to destroy the top-level subplans, because they will recurse to all the subplans below
X
Xiaoyu Wang 已提交
878
        bool   first = true;
X
bugfix  
Xiaoyu Wang 已提交
879 880 881 882 883 884 885 886 887 888 889 890
        SNode* pElement = NULL;
        FOREACH(pElement, pPlan->pSubplans) {
          if (first) {
            first = false;
            nodesDestroyNode(pElement);
          } else {
            nodesClearList(((SNodeListNode*)pElement)->pNodeList);
            taosMemoryFreeClear(pElement);
          }
        }
        nodesClearList(pPlan->pSubplans);
      }
X
Xiaoyu Wang 已提交
891 892
      break;
    }
893 894 895
    default:
      break;
  }
X
Xiaoyu Wang 已提交
896 897
  taosMemoryFreeClear(pNode);
  return;
898 899
}

900
SNodeList* nodesMakeList() {
wafwerar's avatar
wafwerar 已提交
901
  SNodeList* p = taosMemoryCalloc(1, sizeof(SNodeList));
902 903 904 905 906 907
  if (NULL == p) {
    return NULL;
  }
  return p;
}

908
int32_t nodesListAppend(SNodeList* pList, SNode* pNode) {
909
  if (NULL == pList || NULL == pNode) {
X
Xiaoyu Wang 已提交
910
    return TSDB_CODE_FAILED;
911
  }
wafwerar's avatar
wafwerar 已提交
912
  SListCell* p = taosMemoryCalloc(1, sizeof(SListCell));
913
  if (NULL == p) {
X
Xiaoyu Wang 已提交
914 915
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return TSDB_CODE_OUT_OF_MEMORY;
916 917
  }
  p->pNode = pNode;
918 919 920 921 922 923
  if (NULL == pList->pHead) {
    pList->pHead = p;
  }
  if (NULL != pList->pTail) {
    pList->pTail->pNext = p;
  }
924
  p->pPrev = pList->pTail;
925
  pList->pTail = p;
926
  ++(pList->length);
X
Xiaoyu Wang 已提交
927
  return TSDB_CODE_SUCCESS;
928 929
}

930
int32_t nodesListStrictAppend(SNodeList* pList, SNode* pNode) {
931
  if (NULL == pNode) {
X
Xiaoyu Wang 已提交
932
    terrno = TSDB_CODE_OUT_OF_MEMORY;
933 934 935 936 937 938 939 940 941
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  int32_t code = nodesListAppend(pList, pNode);
  if (TSDB_CODE_SUCCESS != code) {
    nodesDestroyNode(pNode);
  }
  return code;
}

942
int32_t nodesListMakeAppend(SNodeList** pList, SNode* pNode) {
X
Xiaoyu Wang 已提交
943 944 945 946 947 948 949 950 951 952
  if (NULL == *pList) {
    *pList = nodesMakeList();
    if (NULL == *pList) {
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  }
  return nodesListAppend(*pList, pNode);
}

953
int32_t nodesListMakeStrictAppend(SNodeList** pList, SNode* pNode) {
S
Shengliang Guan 已提交
954 955 956 957 958 959 960 961 962 963
  if (NULL == *pList) {
    *pList = nodesMakeList();
    if (NULL == *pList) {
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  }
  return nodesListStrictAppend(*pList, pNode);
}

X
Xiaoyu Wang 已提交
964
int32_t nodesListAppendList(SNodeList* pTarget, SNodeList* pSrc) {
X
Xiaoyu Wang 已提交
965
  if (NULL == pTarget || NULL == pSrc) {
X
Xiaoyu Wang 已提交
966
    return TSDB_CODE_FAILED;
X
Xiaoyu Wang 已提交
967 968 969 970 971 972 973 974 975 976
  }

  if (NULL == pTarget->pHead) {
    pTarget->pHead = pSrc->pHead;
  } else {
    pTarget->pTail->pNext = pSrc->pHead;
    if (NULL != pSrc->pHead) {
      pSrc->pHead->pPrev = pTarget->pTail;
    }
  }
X
Xiaoyu Wang 已提交
977 978
  pTarget->pTail = pSrc->pTail;
  pTarget->length += pSrc->length;
wafwerar's avatar
wafwerar 已提交
979
  taosMemoryFreeClear(pSrc);
X
Xiaoyu Wang 已提交
980

X
Xiaoyu Wang 已提交
981 982 983
  return TSDB_CODE_SUCCESS;
}

984 985 986 987 988 989 990 991 992 993 994
int32_t nodesListStrictAppendList(SNodeList* pTarget, SNodeList* pSrc) {
  if (NULL == pSrc) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  int32_t code = nodesListAppendList(pTarget, pSrc);
  if (TSDB_CODE_SUCCESS != code) {
    nodesDestroyList(pSrc);
  }
  return code;
}

995
int32_t nodesListPushFront(SNodeList* pList, SNode* pNode) {
X
Xiaoyu Wang 已提交
996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013
  if (NULL == pList || NULL == pNode) {
    return TSDB_CODE_FAILED;
  }
  SListCell* p = taosMemoryCalloc(1, sizeof(SListCell));
  if (NULL == p) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  p->pNode = pNode;
  if (NULL != pList->pHead) {
    pList->pHead->pPrev = p;
    p->pNext = pList->pHead;
  }
  pList->pHead = p;
  ++(pList->length);
  return TSDB_CODE_SUCCESS;
}

1014 1015 1016 1017 1018
SListCell* nodesListErase(SNodeList* pList, SListCell* pCell) {
  if (NULL == pCell->pPrev) {
    pList->pHead = pCell->pNext;
  } else {
    pCell->pPrev->pNext = pCell->pNext;
X
Xiaoyu Wang 已提交
1019 1020 1021 1022
  }
  if (NULL == pCell->pNext) {
    pList->pTail = pCell->pPrev;
  } else {
1023 1024 1025
    pCell->pNext->pPrev = pCell->pPrev;
  }
  SListCell* pNext = pCell->pNext;
X
Xiaoyu Wang 已提交
1026
  nodesDestroyNode(pCell->pNode);
wafwerar's avatar
wafwerar 已提交
1027
  taosMemoryFreeClear(pCell);
1028 1029 1030 1031
  --(pList->length);
  return pNext;
}

X
Xiaoyu Wang 已提交
1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049
void nodesListInsertList(SNodeList* pTarget, SListCell* pPos, SNodeList* pSrc) {
  if (NULL == pTarget || NULL == pPos || NULL == pSrc) {
    return;
  }

  if (NULL == pPos->pPrev) {
    pTarget->pHead = pSrc->pHead;
  } else {
    pPos->pPrev->pNext = pSrc->pHead;
  }
  pSrc->pHead->pPrev = pPos->pPrev;
  pSrc->pTail->pNext = pPos;
  pPos->pPrev = pSrc->pTail;

  pTarget->length += pSrc->length;
  taosMemoryFreeClear(pSrc);
}

1050
SNode* nodesListGetNode(SNodeList* pList, int32_t index) {
1051 1052 1053 1054 1055 1056 1057 1058 1059
  SNode* node;
  FOREACH(node, pList) {
    if (0 == index--) {
      return node;
    }
  }
  return NULL;
}

1060
void nodesDestroyList(SNodeList* pList) {
1061 1062 1063
  if (NULL == pList) {
    return;
  }
X
Xiaoyu Wang 已提交
1064

D
dapan1121 已提交
1065 1066 1067
  SListCell* pNext = pList->pHead;
  while (NULL != pNext) {
    pNext = nodesListErase(pList, pNext);
1068
  }
wafwerar's avatar
wafwerar 已提交
1069
  taosMemoryFreeClear(pList);
1070 1071
}

X
Xiaoyu Wang 已提交
1072 1073 1074 1075 1076 1077 1078 1079 1080
void nodesClearList(SNodeList* pList) {
  if (NULL == pList) {
    return;
  }

  SListCell* pNext = pList->pHead;
  while (NULL != pNext) {
    SListCell* tmp = pNext;
    pNext = pNext->pNext;
wafwerar's avatar
wafwerar 已提交
1081
    taosMemoryFreeClear(tmp);
X
Xiaoyu Wang 已提交
1082
  }
wafwerar's avatar
wafwerar 已提交
1083
  taosMemoryFreeClear(pList);
X
Xiaoyu Wang 已提交
1084 1085
}

X
Xiaoyu Wang 已提交
1086
void* nodesGetValueFromNode(SValueNode* pNode) {
D
dapan1121 已提交
1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098
  switch (pNode->node.resType.type) {
    case TSDB_DATA_TYPE_BOOL:
    case TSDB_DATA_TYPE_TINYINT:
    case TSDB_DATA_TYPE_SMALLINT:
    case TSDB_DATA_TYPE_INT:
    case TSDB_DATA_TYPE_BIGINT:
    case TSDB_DATA_TYPE_TIMESTAMP:
    case TSDB_DATA_TYPE_UTINYINT:
    case TSDB_DATA_TYPE_USMALLINT:
    case TSDB_DATA_TYPE_UINT:
    case TSDB_DATA_TYPE_UBIGINT:
    case TSDB_DATA_TYPE_FLOAT:
1099
    case TSDB_DATA_TYPE_DOUBLE:
D
dapan1121 已提交
1100
      return (void*)&pNode->typeData;
D
dapan1121 已提交
1101 1102
    case TSDB_DATA_TYPE_NCHAR:
    case TSDB_DATA_TYPE_VARCHAR:
1103
    case TSDB_DATA_TYPE_VARBINARY:
X
Xiaoyu Wang 已提交
1104
      return (void*)pNode->datum.p;
D
dapan1121 已提交
1105 1106 1107 1108 1109 1110 1111
    default:
      break;
  }

  return NULL;
}

X
Xiaoyu Wang 已提交
1112
int32_t nodesSetValueNodeValue(SValueNode* pNode, void* value) {
D
dapan1121 已提交
1113 1114 1115
  switch (pNode->node.resType.type) {
    case TSDB_DATA_TYPE_BOOL:
      pNode->datum.b = *(bool*)value;
D
dapan1121 已提交
1116
      *(bool*)&pNode->typeData = pNode->datum.b;
D
dapan1121 已提交
1117 1118 1119
      break;
    case TSDB_DATA_TYPE_TINYINT:
      pNode->datum.i = *(int8_t*)value;
D
dapan1121 已提交
1120
      *(int8_t*)&pNode->typeData = pNode->datum.i;
D
dapan1121 已提交
1121 1122 1123
      break;
    case TSDB_DATA_TYPE_SMALLINT:
      pNode->datum.i = *(int16_t*)value;
D
dapan1121 已提交
1124
      *(int16_t*)&pNode->typeData = pNode->datum.i;
D
dapan1121 已提交
1125 1126 1127
      break;
    case TSDB_DATA_TYPE_INT:
      pNode->datum.i = *(int32_t*)value;
D
dapan1121 已提交
1128
      *(int32_t*)&pNode->typeData = pNode->datum.i;
D
dapan1121 已提交
1129 1130 1131
      break;
    case TSDB_DATA_TYPE_BIGINT:
      pNode->datum.i = *(int64_t*)value;
D
dapan1121 已提交
1132
      *(int64_t*)&pNode->typeData = pNode->datum.i;
D
dapan1121 已提交
1133 1134 1135
      break;
    case TSDB_DATA_TYPE_TIMESTAMP:
      pNode->datum.i = *(int64_t*)value;
D
dapan1121 已提交
1136
      *(int64_t*)&pNode->typeData = pNode->datum.i;
D
dapan1121 已提交
1137 1138 1139
      break;
    case TSDB_DATA_TYPE_UTINYINT:
      pNode->datum.u = *(int8_t*)value;
D
dapan1121 已提交
1140
      *(int8_t*)&pNode->typeData = pNode->datum.u;
D
dapan1121 已提交
1141 1142 1143
      break;
    case TSDB_DATA_TYPE_USMALLINT:
      pNode->datum.u = *(int16_t*)value;
D
dapan1121 已提交
1144
      *(int16_t*)&pNode->typeData = pNode->datum.u;
D
dapan1121 已提交
1145 1146 1147
      break;
    case TSDB_DATA_TYPE_UINT:
      pNode->datum.u = *(int32_t*)value;
D
dapan1121 已提交
1148
      *(int32_t*)&pNode->typeData = pNode->datum.u;
D
dapan1121 已提交
1149 1150 1151
      break;
    case TSDB_DATA_TYPE_UBIGINT:
      pNode->datum.u = *(uint64_t*)value;
D
dapan1121 已提交
1152
      *(uint64_t*)&pNode->typeData = pNode->datum.u;
D
dapan1121 已提交
1153 1154 1155
      break;
    case TSDB_DATA_TYPE_FLOAT:
      pNode->datum.d = *(float*)value;
D
dapan1121 已提交
1156
      *(float*)&pNode->typeData = pNode->datum.d;
D
dapan1121 已提交
1157 1158 1159
      break;
    case TSDB_DATA_TYPE_DOUBLE:
      pNode->datum.d = *(double*)value;
D
dapan1121 已提交
1160
      *(double*)&pNode->typeData = pNode->datum.d;
D
dapan1121 已提交
1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173
      break;
    case TSDB_DATA_TYPE_NCHAR:
    case TSDB_DATA_TYPE_VARCHAR:
    case TSDB_DATA_TYPE_VARBINARY:
      pNode->datum.p = (char*)value;
      break;
    default:
      return TSDB_CODE_QRY_APP_ERROR;
  }

  return TSDB_CODE_SUCCESS;
}

X
Xiaoyu Wang 已提交
1174
char* nodesGetStrValueFromNode(SValueNode* pNode) {
D
dapan1121 已提交
1175
  switch (pNode->node.resType.type) {
D
dapan1121 已提交
1176
    case TSDB_DATA_TYPE_BOOL: {
X
Xiaoyu Wang 已提交
1177
      void* buf = taosMemoryMalloc(MAX_NUM_STR_SIZE);
D
dapan1121 已提交
1178 1179 1180
      if (NULL == buf) {
        return NULL;
      }
1181

D
dapan1121 已提交
1182 1183
      sprintf(buf, "%s", pNode->datum.b ? "true" : "false");
      return buf;
D
dapan1121 已提交
1184
    }
D
dapan1121 已提交
1185 1186 1187 1188 1189
    case TSDB_DATA_TYPE_TINYINT:
    case TSDB_DATA_TYPE_SMALLINT:
    case TSDB_DATA_TYPE_INT:
    case TSDB_DATA_TYPE_BIGINT:
    case TSDB_DATA_TYPE_TIMESTAMP: {
X
Xiaoyu Wang 已提交
1190
      void* buf = taosMemoryMalloc(MAX_NUM_STR_SIZE);
D
dapan1121 已提交
1191 1192 1193
      if (NULL == buf) {
        return NULL;
      }
1194

D
dapan1121 已提交
1195 1196 1197 1198 1199 1200 1201
      sprintf(buf, "%" PRId64, pNode->datum.i);
      return buf;
    }
    case TSDB_DATA_TYPE_UTINYINT:
    case TSDB_DATA_TYPE_USMALLINT:
    case TSDB_DATA_TYPE_UINT:
    case TSDB_DATA_TYPE_UBIGINT: {
X
Xiaoyu Wang 已提交
1202
      void* buf = taosMemoryMalloc(MAX_NUM_STR_SIZE);
D
dapan1121 已提交
1203 1204 1205
      if (NULL == buf) {
        return NULL;
      }
1206

D
dapan1121 已提交
1207 1208 1209 1210 1211
      sprintf(buf, "%" PRIu64, pNode->datum.u);
      return buf;
    }
    case TSDB_DATA_TYPE_FLOAT:
    case TSDB_DATA_TYPE_DOUBLE: {
X
Xiaoyu Wang 已提交
1212
      void* buf = taosMemoryMalloc(MAX_NUM_STR_SIZE);
D
dapan1121 已提交
1213 1214 1215
      if (NULL == buf) {
        return NULL;
      }
1216

D
dapan1121 已提交
1217 1218 1219 1220 1221 1222
      sprintf(buf, "%e", pNode->datum.d);
      return buf;
    }
    case TSDB_DATA_TYPE_NCHAR:
    case TSDB_DATA_TYPE_VARCHAR:
    case TSDB_DATA_TYPE_VARBINARY: {
D
dapan1121 已提交
1223
      int32_t bufSize = varDataLen(pNode->datum.p) + 2 + 1;
X
Xiaoyu Wang 已提交
1224
      void*   buf = taosMemoryMalloc(bufSize);
D
dapan1121 已提交
1225 1226 1227
      if (NULL == buf) {
        return NULL;
      }
1228

D
dapan1121 已提交
1229
      snprintf(buf, bufSize, "'%s'", varDataVal(pNode->datum.p));
D
dapan1121 已提交
1230 1231 1232 1233 1234 1235 1236 1237 1238
      return buf;
    }
    default:
      break;
  }

  return NULL;
}

1239 1240
bool nodesIsExprNode(const SNode* pNode) {
  ENodeType type = nodeType(pNode);
X
Xiaoyu Wang 已提交
1241
  return (QUERY_NODE_COLUMN == type || QUERY_NODE_VALUE == type || QUERY_NODE_OPERATOR == type ||
X
Xiaoyu Wang 已提交
1242
          QUERY_NODE_FUNCTION == type || QUERY_NODE_LOGIC_CONDITION == type);
1243 1244
}

X
Xiaoyu Wang 已提交
1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262
bool nodesIsUnaryOp(const SOperatorNode* pOp) {
  switch (pOp->opType) {
    case OP_TYPE_MINUS:
    case OP_TYPE_IS_NULL:
    case OP_TYPE_IS_NOT_NULL:
    case OP_TYPE_IS_TRUE:
    case OP_TYPE_IS_FALSE:
    case OP_TYPE_IS_UNKNOWN:
    case OP_TYPE_IS_NOT_TRUE:
    case OP_TYPE_IS_NOT_FALSE:
    case OP_TYPE_IS_NOT_UNKNOWN:
      return true;
    default:
      break;
  }
  return false;
}

1263 1264 1265 1266 1267 1268
bool nodesIsArithmeticOp(const SOperatorNode* pOp) {
  switch (pOp->opType) {
    case OP_TYPE_ADD:
    case OP_TYPE_SUB:
    case OP_TYPE_MULTI:
    case OP_TYPE_DIV:
1269
    case OP_TYPE_REM:
1270 1271 1272 1273 1274 1275
      return true;
    default:
      break;
  }
  return false;
}
1276

1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290
bool nodesIsComparisonOp(const SOperatorNode* pOp) {
  switch (pOp->opType) {
    case OP_TYPE_GREATER_THAN:
    case OP_TYPE_GREATER_EQUAL:
    case OP_TYPE_LOWER_THAN:
    case OP_TYPE_LOWER_EQUAL:
    case OP_TYPE_EQUAL:
    case OP_TYPE_NOT_EQUAL:
    case OP_TYPE_IN:
    case OP_TYPE_NOT_IN:
    case OP_TYPE_LIKE:
    case OP_TYPE_NOT_LIKE:
    case OP_TYPE_MATCH:
    case OP_TYPE_NMATCH:
1291
    case OP_TYPE_JSON_CONTAINS:
X
Xiaoyu Wang 已提交
1292 1293 1294 1295 1296 1297 1298 1299
    case OP_TYPE_IS_NULL:
    case OP_TYPE_IS_NOT_NULL:
    case OP_TYPE_IS_TRUE:
    case OP_TYPE_IS_FALSE:
    case OP_TYPE_IS_UNKNOWN:
    case OP_TYPE_IS_NOT_TRUE:
    case OP_TYPE_IS_NOT_FALSE:
    case OP_TYPE_IS_NOT_UNKNOWN:
1300 1301 1302 1303 1304
      return true;
    default:
      break;
  }
  return false;
1305 1306
}

1307 1308 1309
bool nodesIsJsonOp(const SOperatorNode* pOp) {
  switch (pOp->opType) {
    case OP_TYPE_JSON_GET_VALUE:
wmmhello's avatar
wmmhello 已提交
1310
    case OP_TYPE_JSON_CONTAINS:
1311 1312 1313 1314 1315 1316
      return true;
    default:
      break;
  }
  return false;
}
1317

X
Xiaoyu Wang 已提交
1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330
bool nodesIsRegularOp(const SOperatorNode* pOp) {
  switch (pOp->opType) {
    case OP_TYPE_LIKE:
    case OP_TYPE_NOT_LIKE:
    case OP_TYPE_MATCH:
    case OP_TYPE_NMATCH:
      return true;
    default:
      break;
  }
  return false;
}

X
Xiaoyu Wang 已提交
1331
typedef struct SCollectColumnsCxt {
1332 1333 1334 1335 1336
  int32_t         errCode;
  const char*     pTableAlias;
  ECollectColType collectType;
  SNodeList*      pCols;
  SHashObj*       pColHash;
X
Xiaoyu Wang 已提交
1337 1338
} SCollectColumnsCxt;

1339
static EDealRes doCollect(SCollectColumnsCxt* pCxt, SColumnNode* pCol, SNode* pNode) {
X
Xiaoyu Wang 已提交
1340
  char    name[TSDB_TABLE_NAME_LEN + TSDB_COL_NAME_LEN];
1341 1342 1343 1344 1345 1346 1347
  int32_t len = 0;
  if ('\0' == pCol->tableAlias[0]) {
    len = sprintf(name, "%s", pCol->colName);
  }
  len = sprintf(name, "%s.%s", pCol->tableAlias, pCol->colName);
  if (NULL == taosHashGet(pCxt->pColHash, name, len)) {
    pCxt->errCode = taosHashPut(pCxt->pColHash, name, len, NULL, 0);
X
Xiaoyu Wang 已提交
1348
    if (TSDB_CODE_SUCCESS == pCxt->errCode) {
1349
      pCxt->errCode = nodesListStrictAppend(pCxt->pCols, nodesCloneNode(pNode));
X
Xiaoyu Wang 已提交
1350 1351 1352 1353 1354 1355
    }
    return (TSDB_CODE_SUCCESS == pCxt->errCode ? DEAL_RES_IGNORE_CHILD : DEAL_RES_ERROR);
  }
  return DEAL_RES_CONTINUE;
}

1356 1357 1358 1359 1360 1361
static bool isCollectType(ECollectColType collectType, EColumnType colType) {
  return COLLECT_COL_TYPE_ALL == collectType
             ? true
             : (COLLECT_COL_TYPE_TAG == collectType ? COLUMN_TYPE_TAG == colType : COLUMN_TYPE_TAG != colType);
}

X
Xiaoyu Wang 已提交
1362 1363
static EDealRes collectColumns(SNode* pNode, void* pContext) {
  SCollectColumnsCxt* pCxt = (SCollectColumnsCxt*)pContext;
X
Xiaoyu Wang 已提交
1364
  if (QUERY_NODE_COLUMN == nodeType(pNode)) {
X
Xiaoyu Wang 已提交
1365
    SColumnNode* pCol = (SColumnNode*)pNode;
1366 1367
    if (isCollectType(pCxt->collectType, pCol->colType) &&
        (NULL == pCxt->pTableAlias || 0 == strcmp(pCxt->pTableAlias, pCol->tableAlias))) {
1368
      return doCollect(pCxt, pCol, pNode);
X
Xiaoyu Wang 已提交
1369 1370 1371 1372 1373
    }
  }
  return DEAL_RES_CONTINUE;
}

1374 1375
int32_t nodesCollectColumns(SSelectStmt* pSelect, ESqlClause clause, const char* pTableAlias, ECollectColType type,
                            SNodeList** pCols) {
X
Xiaoyu Wang 已提交
1376
  if (NULL == pSelect || NULL == pCols) {
1377
    return TSDB_CODE_FAILED;
X
Xiaoyu Wang 已提交
1378 1379 1380
  }

  SCollectColumnsCxt cxt = {
X
Xiaoyu Wang 已提交
1381 1382
      .errCode = TSDB_CODE_SUCCESS,
      .pTableAlias = pTableAlias,
1383
      .collectType = type,
1384
      .pCols = (NULL == *pCols ? nodesMakeList() : *pCols),
X
Xiaoyu Wang 已提交
1385
      .pColHash = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK)};
1386
  if (NULL == cxt.pCols || NULL == cxt.pColHash) {
X
Xiaoyu Wang 已提交
1387 1388
    return TSDB_CODE_OUT_OF_MEMORY;
  }
1389
  *pCols = NULL;
X
Xiaoyu Wang 已提交
1390
  nodesWalkSelectStmt(pSelect, clause, collectColumns, &cxt);
1391
  taosHashCleanup(cxt.pColHash);
X
Xiaoyu Wang 已提交
1392
  if (TSDB_CODE_SUCCESS != cxt.errCode) {
1393
    nodesDestroyList(cxt.pCols);
X
Xiaoyu Wang 已提交
1394 1395
    return cxt.errCode;
  }
1396 1397 1398
  if (LIST_LENGTH(cxt.pCols) > 0) {
    *pCols = cxt.pCols;
  } else {
1399
    nodesDestroyList(cxt.pCols);
X
Xiaoyu Wang 已提交
1400
  }
1401

X
Xiaoyu Wang 已提交
1402 1403 1404 1405
  return TSDB_CODE_SUCCESS;
}

typedef struct SCollectFuncsCxt {
X
Xiaoyu Wang 已提交
1406
  int32_t         errCode;
X
Xiaoyu Wang 已提交
1407
  FFuncClassifier classifier;
X
Xiaoyu Wang 已提交
1408
  SNodeList*      pFuncs;
X
Xiaoyu Wang 已提交
1409 1410 1411 1412 1413
} SCollectFuncsCxt;

static EDealRes collectFuncs(SNode* pNode, void* pContext) {
  SCollectFuncsCxt* pCxt = (SCollectFuncsCxt*)pContext;
  if (QUERY_NODE_FUNCTION == nodeType(pNode) && pCxt->classifier(((SFunctionNode*)pNode)->funcId)) {
1414
    pCxt->errCode = nodesListStrictAppend(pCxt->pFuncs, nodesCloneNode(pNode));
X
Xiaoyu Wang 已提交
1415 1416 1417 1418 1419
    return (TSDB_CODE_SUCCESS == pCxt->errCode ? DEAL_RES_IGNORE_CHILD : DEAL_RES_ERROR);
  }
  return DEAL_RES_CONTINUE;
}

X
Xiaoyu Wang 已提交
1420
int32_t nodesCollectFuncs(SSelectStmt* pSelect, ESqlClause clause, FFuncClassifier classifier, SNodeList** pFuncs) {
X
Xiaoyu Wang 已提交
1421
  if (NULL == pSelect || NULL == pFuncs) {
1422
    return TSDB_CODE_FAILED;
X
Xiaoyu Wang 已提交
1423 1424
  }

1425 1426
  SCollectFuncsCxt cxt = {
      .errCode = TSDB_CODE_SUCCESS, .classifier = classifier, .pFuncs = (NULL == *pFuncs ? nodesMakeList() : *pFuncs)};
X
Xiaoyu Wang 已提交
1427 1428 1429
  if (NULL == cxt.pFuncs) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
1430
  *pFuncs = NULL;
X
Xiaoyu Wang 已提交
1431
  nodesWalkSelectStmt(pSelect, clause, collectFuncs, &cxt);
X
Xiaoyu Wang 已提交
1432 1433 1434 1435
  if (TSDB_CODE_SUCCESS != cxt.errCode) {
    nodesDestroyList(cxt.pFuncs);
    return cxt.errCode;
  }
X
Xiaoyu Wang 已提交
1436 1437 1438 1439 1440
  if (LIST_LENGTH(cxt.pFuncs) > 0) {
    *pFuncs = cxt.pFuncs;
  } else {
    nodesDestroyList(cxt.pFuncs);
  }
X
Xiaoyu Wang 已提交
1441

1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481
  return TSDB_CODE_SUCCESS;
}

typedef struct SCollectSpecialNodesCxt {
  int32_t    errCode;
  ENodeType  type;
  SNodeList* pNodes;
} SCollectSpecialNodesCxt;

static EDealRes collectSpecialNodes(SNode* pNode, void* pContext) {
  SCollectSpecialNodesCxt* pCxt = (SCollectSpecialNodesCxt*)pContext;
  if (pCxt->type == nodeType(pNode)) {
    pCxt->errCode = nodesListStrictAppend(pCxt->pNodes, nodesCloneNode(pNode));
    return (TSDB_CODE_SUCCESS == pCxt->errCode ? DEAL_RES_IGNORE_CHILD : DEAL_RES_ERROR);
  }
  return DEAL_RES_CONTINUE;
}

int32_t nodesCollectSpecialNodes(SSelectStmt* pSelect, ESqlClause clause, ENodeType type, SNodeList** pNodes) {
  if (NULL == pSelect || NULL == pNodes) {
    return TSDB_CODE_FAILED;
  }

  SCollectSpecialNodesCxt cxt = {
      .errCode = TSDB_CODE_SUCCESS, .type = type, .pNodes = (NULL == *pNodes ? nodesMakeList() : *pNodes)};
  if (NULL == cxt.pNodes) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
  *pNodes = NULL;
  nodesWalkSelectStmt(pSelect, SQL_CLAUSE_GROUP_BY, collectSpecialNodes, &cxt);
  if (TSDB_CODE_SUCCESS != cxt.errCode) {
    nodesDestroyList(cxt.pNodes);
    return cxt.errCode;
  }
  if (LIST_LENGTH(cxt.pNodes) > 0) {
    *pNodes = cxt.pNodes;
  } else {
    nodesDestroyList(cxt.pNodes);
  }

X
Xiaoyu Wang 已提交
1482 1483
  return TSDB_CODE_SUCCESS;
}
D
dapan1121 已提交
1484

1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504
static EDealRes hasColumn(SNode* pNode, void* pContext) {
  if (QUERY_NODE_COLUMN == nodeType(pNode)) {
    *(bool*)pContext = true;
    return DEAL_RES_END;
  }
  return DEAL_RES_CONTINUE;
}

bool nodesExprHasColumn(SNode* pNode) {
  bool hasCol = false;
  nodesWalkExprPostOrder(pNode, hasColumn, &hasCol);
  return hasCol;
}

bool nodesExprsHasColumn(SNodeList* pList) {
  bool hasCol = false;
  nodesWalkExprsPostOrder(pList, hasColumn, &hasCol);
  return hasCol;
}

X
Xiaoyu Wang 已提交
1505
char* nodesGetFillModeString(EFillMode mode) {
D
dapan1121 已提交
1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523
  switch (mode) {
    case FILL_MODE_NONE:
      return "none";
    case FILL_MODE_VALUE:
      return "value";
    case FILL_MODE_PREV:
      return "prev";
    case FILL_MODE_NULL:
      return "null";
    case FILL_MODE_LINEAR:
      return "linear";
    case FILL_MODE_NEXT:
      return "next";
    default:
      return "unknown";
  }
}

X
Xiaoyu Wang 已提交
1524
char* nodesGetNameFromColumnNode(SNode* pNode) {
D
dapan1121 已提交
1525 1526 1527
  if (NULL == pNode || QUERY_NODE_COLUMN != pNode->type) {
    return "NULL";
  }
X
Xiaoyu Wang 已提交
1528 1529

  return ((SColumnNode*)pNode)->colName;
D
dapan1121 已提交
1530 1531
}

D
dapan1121 已提交
1532 1533 1534 1535 1536
int32_t nodesGetOutputNumFromSlotList(SNodeList* pSlots) {
  if (NULL == pSlots || pSlots->length <= 0) {
    return 0;
  }

X
Xiaoyu Wang 已提交
1537
  SNode*  pNode = NULL;
D
dapan1121 已提交
1538 1539 1540 1541 1542 1543
  int32_t num = 0;
  FOREACH(pNode, pSlots) {
    if (QUERY_NODE_SLOT_DESC != pNode->type) {
      continue;
    }

X
Xiaoyu Wang 已提交
1544
    SSlotDescNode* descNode = (SSlotDescNode*)pNode;
D
dapan1121 已提交
1545 1546 1547 1548 1549 1550 1551 1552
    if (descNode->output) {
      ++num;
    }
  }

  return num;
}

X
Xiaoyu Wang 已提交
1553
void nodesValueNodeToVariant(const SValueNode* pNode, SVariant* pVal) {
D
dapan1121 已提交
1554 1555 1556 1557
  pVal->nType = pNode->node.resType.type;
  pVal->nLen = pNode->node.resType.bytes;
  switch (pNode->node.resType.type) {
    case TSDB_DATA_TYPE_NULL:
X
Xiaoyu Wang 已提交
1558
      break;
D
dapan1121 已提交
1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581
    case TSDB_DATA_TYPE_BOOL:
      pVal->i = pNode->datum.b;
      break;
    case TSDB_DATA_TYPE_TINYINT:
    case TSDB_DATA_TYPE_SMALLINT:
    case TSDB_DATA_TYPE_INT:
    case TSDB_DATA_TYPE_BIGINT:
    case TSDB_DATA_TYPE_TIMESTAMP:
      pVal->i = pNode->datum.i;
      break;
    case TSDB_DATA_TYPE_UTINYINT:
    case TSDB_DATA_TYPE_USMALLINT:
    case TSDB_DATA_TYPE_UINT:
    case TSDB_DATA_TYPE_UBIGINT:
      pVal->u = pNode->datum.u;
      break;
    case TSDB_DATA_TYPE_FLOAT:
    case TSDB_DATA_TYPE_DOUBLE:
      pVal->d = pNode->datum.d;
      break;
    case TSDB_DATA_TYPE_NCHAR:
    case TSDB_DATA_TYPE_VARCHAR:
    case TSDB_DATA_TYPE_VARBINARY:
D
dapan1121 已提交
1582 1583 1584
      pVal->pz = taosMemoryMalloc(pVal->nLen + VARSTR_HEADER_SIZE + 1);
      memcpy(pVal->pz, pNode->datum.p, pVal->nLen + VARSTR_HEADER_SIZE);
      pVal->pz[pVal->nLen + VARSTR_HEADER_SIZE] = 0;
D
dapan1121 已提交
1585 1586 1587 1588 1589 1590 1591 1592 1593
      break;
    case TSDB_DATA_TYPE_JSON:
    case TSDB_DATA_TYPE_DECIMAL:
    case TSDB_DATA_TYPE_BLOB:
      // todo
    default:
      break;
  }
}
X
Xiaoyu Wang 已提交
1594 1595 1596 1597 1598 1599 1600 1601 1602 1603

int32_t nodesMergeConds(SNode** pDst, SNodeList** pSrc) {
  if (NULL == *pSrc) {
    return TSDB_CODE_SUCCESS;
  }

  if (1 == LIST_LENGTH(*pSrc)) {
    *pDst = nodesListGetNode(*pSrc, 0);
    nodesClearList(*pSrc);
  } else {
1604
    SLogicConditionNode* pLogicCond = (SLogicConditionNode*)nodesMakeNode(QUERY_NODE_LOGIC_CONDITION);
X
Xiaoyu Wang 已提交
1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749
    if (NULL == pLogicCond) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
    pLogicCond->node.resType.type = TSDB_DATA_TYPE_BOOL;
    pLogicCond->node.resType.bytes = tDataTypes[TSDB_DATA_TYPE_BOOL].bytes;
    pLogicCond->condType = LOGIC_COND_TYPE_AND;
    pLogicCond->pParameterList = *pSrc;
    *pDst = (SNode*)pLogicCond;
  }
  *pSrc = NULL;

  return TSDB_CODE_SUCCESS;
}

typedef struct SClassifyConditionCxt {
  bool hasPrimaryKey;
  bool hasTagIndexCol;
  bool hasOtherCol;
} SClassifyConditionCxt;

static EDealRes classifyConditionImpl(SNode* pNode, void* pContext) {
  SClassifyConditionCxt* pCxt = (SClassifyConditionCxt*)pContext;
  if (QUERY_NODE_COLUMN == nodeType(pNode)) {
    SColumnNode* pCol = (SColumnNode*)pNode;
    if (PRIMARYKEY_TIMESTAMP_COL_ID == pCol->colId) {
      pCxt->hasPrimaryKey = true;
    } else if (pCol->hasIndex) {
      pCxt->hasTagIndexCol = true;
    } else {
      pCxt->hasOtherCol = true;
    }
    return *((bool*)pContext) ? DEAL_RES_CONTINUE : DEAL_RES_END;
  }
  return DEAL_RES_CONTINUE;
}

typedef enum EConditionType { COND_TYPE_PRIMARY_KEY = 1, COND_TYPE_TAG_INDEX, COND_TYPE_NORMAL } EConditionType;

static EConditionType classifyCondition(SNode* pNode) {
  SClassifyConditionCxt cxt = {.hasPrimaryKey = false, .hasTagIndexCol = false, .hasOtherCol = false};
  nodesWalkExpr(pNode, classifyConditionImpl, &cxt);
  return cxt.hasOtherCol ? COND_TYPE_NORMAL
                         : (cxt.hasPrimaryKey && cxt.hasTagIndexCol
                                ? COND_TYPE_NORMAL
                                : (cxt.hasPrimaryKey ? COND_TYPE_PRIMARY_KEY : COND_TYPE_TAG_INDEX));
}

static int32_t partitionLogicCond(SNode** pCondition, SNode** pPrimaryKeyCond, SNode** pTagCond, SNode** pOtherCond) {
  SLogicConditionNode* pLogicCond = (SLogicConditionNode*)(*pCondition);

  int32_t code = TSDB_CODE_SUCCESS;

  SNodeList* pPrimaryKeyConds = NULL;
  SNodeList* pTagConds = NULL;
  SNodeList* pOtherConds = NULL;
  SNode*     pCond = NULL;
  FOREACH(pCond, pLogicCond->pParameterList) {
    switch (classifyCondition(pCond)) {
      case COND_TYPE_PRIMARY_KEY:
        if (NULL != pPrimaryKeyCond) {
          code = nodesListMakeAppend(&pPrimaryKeyConds, nodesCloneNode(pCond));
        }
        break;
      case COND_TYPE_TAG_INDEX:
        if (NULL != pTagCond) {
          code = nodesListMakeAppend(&pTagConds, nodesCloneNode(pCond));
        }
        break;
      case COND_TYPE_NORMAL:
      default:
        if (NULL != pOtherCond) {
          code = nodesListMakeAppend(&pOtherConds, nodesCloneNode(pCond));
        }
        break;
    }
    if (TSDB_CODE_SUCCESS != code) {
      break;
    }
  }

  SNode* pTempPrimaryKeyCond = NULL;
  SNode* pTempTagCond = NULL;
  SNode* pTempOtherCond = NULL;
  if (TSDB_CODE_SUCCESS == code) {
    code = nodesMergeConds(&pTempPrimaryKeyCond, &pPrimaryKeyConds);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = nodesMergeConds(&pTempTagCond, &pTagConds);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = nodesMergeConds(&pTempOtherCond, &pOtherConds);
  }

  if (TSDB_CODE_SUCCESS == code) {
    if (NULL != pPrimaryKeyCond) {
      *pPrimaryKeyCond = pTempPrimaryKeyCond;
    }
    if (NULL != pTagCond) {
      *pTagCond = pTempTagCond;
    }
    if (NULL != pOtherCond) {
      *pOtherCond = pTempOtherCond;
    }
    nodesDestroyNode(*pCondition);
    *pCondition = NULL;
  } else {
    nodesDestroyList(pPrimaryKeyConds);
    nodesDestroyList(pTagConds);
    nodesDestroyList(pOtherConds);
    nodesDestroyNode(pTempPrimaryKeyCond);
    nodesDestroyNode(pTempTagCond);
    nodesDestroyNode(pTempOtherCond);
  }

  return code;
}

int32_t nodesPartitionCond(SNode** pCondition, SNode** pPrimaryKeyCond, SNode** pTagCond, SNode** pOtherCond) {
  if (QUERY_NODE_LOGIC_CONDITION == nodeType(*pCondition) &&
      LOGIC_COND_TYPE_AND == ((SLogicConditionNode*)*pCondition)->condType) {
    return partitionLogicCond(pCondition, pPrimaryKeyCond, pTagCond, pOtherCond);
  }

  switch (classifyCondition(*pCondition)) {
    case COND_TYPE_PRIMARY_KEY:
      if (NULL != pPrimaryKeyCond) {
        *pPrimaryKeyCond = *pCondition;
      }
      break;
    case COND_TYPE_TAG_INDEX:
      if (NULL != pTagCond) {
        *pTagCond = *pCondition;
      }
      break;
    case COND_TYPE_NORMAL:
    default:
      if (NULL != pOtherCond) {
        *pOtherCond = *pCondition;
      }
      break;
  }
  *pCondition = NULL;

  return TSDB_CODE_SUCCESS;
}