planLogicCreater.c 57.7 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/>.
 */

X
Xiaoyu Wang 已提交
16
#include "planInt.h"
17

X
Xiaoyu Wang 已提交
18 19 20
#include "functionMgt.h"

typedef struct SLogicPlanContext {
21
  SPlanContext* pPlanCxt;
22
  SLogicNode*   pCurrRoot;
23
  bool          hasScan;
X
Xiaoyu Wang 已提交
24 25
} SLogicPlanContext;

X
Xiaoyu Wang 已提交
26 27
typedef int32_t (*FCreateLogicNode)(SLogicPlanContext*, void*, SLogicNode**);
typedef int32_t (*FCreateSelectLogicNode)(SLogicPlanContext*, SSelectStmt*, SLogicNode**);
X
Xiaoyu Wang 已提交
28
typedef int32_t (*FCreateSetOpLogicNode)(SLogicPlanContext*, SSetOperator*, SLogicNode**);
X
Xiaoyu Wang 已提交
29
typedef int32_t (*FCreateDeleteLogicNode)(SLogicPlanContext*, SDeleteStmt*, SLogicNode**);
30
typedef int32_t (*FCreateInsertLogicNode)(SLogicPlanContext*, SInsertStmt*, SLogicNode**);
X
Xiaoyu Wang 已提交
31

X
Xiaoyu Wang 已提交
32 33
static int32_t doCreateLogicNodeByTable(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SNode* pTable,
                                        SLogicNode** pLogicNode);
34
static int32_t createQueryLogicNode(SLogicPlanContext* pCxt, SNode* pStmt, SLogicNode** pLogicNode);
X
Xiaoyu Wang 已提交
35 36

typedef struct SRewriteExprCxt {
X
Xiaoyu Wang 已提交
37
  int32_t    errCode;
X
Xiaoyu Wang 已提交
38
  SNodeList* pExprs;
X
Xiaoyu Wang 已提交
39
  bool*      pOutputs;
X
Xiaoyu Wang 已提交
40 41
} SRewriteExprCxt;

42 43 44 45 46 47
static void setColumnInfo(SFunctionNode* pFunc, SColumnNode* pCol) {
  switch (pFunc->funcType) {
    case FUNCTION_TYPE_TBNAME:
      pCol->colType = COLUMN_TYPE_TBNAME;
      break;
    case FUNCTION_TYPE_WSTART:
48 49 50
      pCol->colId = PRIMARYKEY_TIMESTAMP_COL_ID;
      pCol->colType = COLUMN_TYPE_WINDOW_START;
      break;
51 52
    case FUNCTION_TYPE_WEND:
      pCol->colId = PRIMARYKEY_TIMESTAMP_COL_ID;
53
      pCol->colType = COLUMN_TYPE_WINDOW_END;
54 55
      break;
    case FUNCTION_TYPE_WDURATION:
56
      pCol->colType = COLUMN_TYPE_WINDOW_DURATION;
57 58 59 60 61 62 63 64 65
      break;
    case FUNCTION_TYPE_GROUP_KEY:
      pCol->colType = COLUMN_TYPE_GROUP_KEY;
      break;
    default:
      break;
  }
}

X
Xiaoyu Wang 已提交
66
static EDealRes doRewriteExpr(SNode** pNode, void* pContext) {
X
Xiaoyu Wang 已提交
67
  SRewriteExprCxt* pCxt = (SRewriteExprCxt*)pContext;
X
Xiaoyu Wang 已提交
68
  switch (nodeType(*pNode)) {
X
Xiaoyu Wang 已提交
69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84
    case QUERY_NODE_COLUMN: {
      if (NULL != pCxt->pOutputs) {
        SNode*  pExpr;
        int32_t index = 0;
        FOREACH(pExpr, pCxt->pExprs) {
          if (QUERY_NODE_GROUPING_SET == nodeType(pExpr)) {
            pExpr = nodesListGetNode(((SGroupingSetNode*)pExpr)->pParameterList, 0);
          }
          if (nodesEqualNode(pExpr, *pNode)) {
            pCxt->pOutputs[index] = true;
            break;
          }
        }
      }
      break;
    }
X
Xiaoyu Wang 已提交
85 86
    case QUERY_NODE_OPERATOR:
    case QUERY_NODE_LOGIC_CONDITION:
D
dapan1121 已提交
87 88
    case QUERY_NODE_FUNCTION:
    case QUERY_NODE_CASE_WHEN: {
X
Xiaoyu Wang 已提交
89 90
      SNode*  pExpr;
      int32_t index = 0;
X
Xiaoyu Wang 已提交
91 92 93 94 95 96
      FOREACH(pExpr, pCxt->pExprs) {
        if (QUERY_NODE_GROUPING_SET == nodeType(pExpr)) {
          pExpr = nodesListGetNode(((SGroupingSetNode*)pExpr)->pParameterList, 0);
        }
        if (nodesEqualNode(pExpr, *pNode)) {
          SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
X
Xiaoyu Wang 已提交
97 98 99
          if (NULL == pCol) {
            return DEAL_RES_ERROR;
          }
X
Xiaoyu Wang 已提交
100 101 102 103
          SExprNode* pToBeRewrittenExpr = (SExprNode*)(*pNode);
          pCol->node.resType = pToBeRewrittenExpr->resType;
          strcpy(pCol->node.aliasName, pToBeRewrittenExpr->aliasName);
          strcpy(pCol->colName, ((SExprNode*)pExpr)->aliasName);
X
Xiaoyu Wang 已提交
104
          if (QUERY_NODE_FUNCTION == nodeType(pExpr)) {
105
            setColumnInfo((SFunctionNode*)pExpr, pCol);
X
Xiaoyu Wang 已提交
106
          }
X
Xiaoyu Wang 已提交
107 108
          nodesDestroyNode(*pNode);
          *pNode = (SNode*)pCol;
X
Xiaoyu Wang 已提交
109 110 111
          if (NULL != pCxt->pOutputs) {
            pCxt->pOutputs[index] = true;
          }
X
Xiaoyu Wang 已提交
112 113 114 115 116 117 118 119
          return DEAL_RES_IGNORE_CHILD;
        }
        ++index;
      }
      break;
    }
    default:
      break;
H
Haojun Liao 已提交
120 121
  }

X
Xiaoyu Wang 已提交
122
  return DEAL_RES_CONTINUE;
X
Xiaoyu Wang 已提交
123 124
}

X
Xiaoyu Wang 已提交
125 126 127 128 129
static EDealRes doNameExpr(SNode* pNode, void* pContext) {
  switch (nodeType(pNode)) {
    case QUERY_NODE_OPERATOR:
    case QUERY_NODE_LOGIC_CONDITION:
    case QUERY_NODE_FUNCTION: {
130 131 132
      if ('\0' == ((SExprNode*)pNode)->aliasName[0]) {
        sprintf(((SExprNode*)pNode)->aliasName, "#expr_%p", pNode);
      }
X
Xiaoyu Wang 已提交
133
      return DEAL_RES_IGNORE_CHILD;
X
Xiaoyu Wang 已提交
134 135
    }
    default:
X
Xiaoyu Wang 已提交
136
      break;
X
Xiaoyu Wang 已提交
137
  }
138

X
Xiaoyu Wang 已提交
139
  return DEAL_RES_CONTINUE;
140 141
}

D
dapan1121 已提交
142 143
static int32_t rewriteExprForSelect(SNode* pExpr, SSelectStmt* pSelect, ESqlClause clause) {
  nodesWalkExpr(pExpr, doNameExpr, NULL);
X
Xiaoyu Wang 已提交
144
  SRewriteExprCxt cxt = {.errCode = TSDB_CODE_SUCCESS, .pExprs = NULL, .pOutputs = NULL};
D
dapan1121 已提交
145 146 147 148 149 150 151 152
  cxt.errCode = nodesListMakeAppend(&cxt.pExprs, pExpr);
  if (TSDB_CODE_SUCCESS == cxt.errCode) {
    nodesRewriteSelectStmt(pSelect, clause, doRewriteExpr, &cxt);
    nodesClearList(cxt.pExprs);
  }
  return cxt.errCode;
}

X
Xiaoyu Wang 已提交
153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169
static int32_t cloneRewriteExprs(SNodeList* pExprs, bool* pOutputs, SNodeList** pRewriteExpr) {
  int32_t code = TSDB_CODE_SUCCESS;
  int32_t index = 0;
  SNode*  pExpr = NULL;
  FOREACH(pExpr, pExprs) {
    if (pOutputs[index]) {
      code = nodesListMakeStrictAppend(pRewriteExpr, nodesCloneNode(pExpr));
      if (TSDB_CODE_SUCCESS != code) {
        NODES_DESTORY_LIST(*pRewriteExpr);
        break;
      }
    }
  }
  return code;
}

static int32_t rewriteExprsForSelect(SNodeList* pExprs, SSelectStmt* pSelect, ESqlClause clause,
X
Xiaoyu Wang 已提交
170
                                     SNodeList** pRewriteExprs) {
X
Xiaoyu Wang 已提交
171
  nodesWalkExprs(pExprs, doNameExpr, NULL);
X
Xiaoyu Wang 已提交
172
  SRewriteExprCxt cxt = {.errCode = TSDB_CODE_SUCCESS, .pExprs = pExprs, .pOutputs = NULL};
X
Xiaoyu Wang 已提交
173
  if (NULL != pRewriteExprs) {
X
Xiaoyu Wang 已提交
174 175 176 177 178
    cxt.pOutputs = taosMemoryCalloc(LIST_LENGTH(pExprs), sizeof(bool));
    if (NULL == cxt.pOutputs) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  }
X
Xiaoyu Wang 已提交
179
  nodesRewriteSelectStmt(pSelect, clause, doRewriteExpr, &cxt);
X
Xiaoyu Wang 已提交
180 181
  if (TSDB_CODE_SUCCESS == cxt.errCode && NULL != pRewriteExprs) {
    cxt.errCode = cloneRewriteExprs(pExprs, cxt.pOutputs, pRewriteExprs);
X
Xiaoyu Wang 已提交
182 183
  }
  taosMemoryFree(cxt.pOutputs);
X
Xiaoyu Wang 已提交
184
  return cxt.errCode;
185 186
}

X
Xiaoyu Wang 已提交
187 188
static int32_t rewriteExpr(SNodeList* pExprs, SNode** pTarget) {
  nodesWalkExprs(pExprs, doNameExpr, NULL);
X
Xiaoyu Wang 已提交
189
  SRewriteExprCxt cxt = {.errCode = TSDB_CODE_SUCCESS, .pExprs = pExprs, .pOutputs = NULL};
X
Xiaoyu Wang 已提交
190 191 192 193
  nodesRewriteExpr(pTarget, doRewriteExpr, &cxt);
  return cxt.errCode;
}

X
Xiaoyu Wang 已提交
194
static int32_t rewriteExprs(SNodeList* pExprs, SNodeList* pTarget) {
X
Xiaoyu Wang 已提交
195
  nodesWalkExprs(pExprs, doNameExpr, NULL);
X
Xiaoyu Wang 已提交
196
  SRewriteExprCxt cxt = {.errCode = TSDB_CODE_SUCCESS, .pExprs = pExprs, .pOutputs = NULL};
X
Xiaoyu Wang 已提交
197 198 199 200
  nodesRewriteExprs(pTarget, doRewriteExpr, &cxt);
  return cxt.errCode;
}

201 202 203 204 205
static int32_t pushLogicNode(SLogicPlanContext* pCxt, SLogicNode** pOldRoot, SLogicNode* pNewRoot) {
  if (NULL == pNewRoot->pChildren) {
    pNewRoot->pChildren = nodesMakeList();
    if (NULL == pNewRoot->pChildren) {
      return TSDB_CODE_OUT_OF_MEMORY;
H
Haojun Liao 已提交
206
    }
207
  }
208 209
  if (TSDB_CODE_SUCCESS != nodesListAppend(pNewRoot->pChildren, (SNode*)*pOldRoot)) {
    return TSDB_CODE_OUT_OF_MEMORY;
210 211
  }

212 213
  (*pOldRoot)->pParent = pNewRoot;
  *pOldRoot = pNewRoot;
214

215 216
  return TSDB_CODE_SUCCESS;
}
217

X
Xiaoyu Wang 已提交
218 219
static int32_t createRootLogicNode(SLogicPlanContext* pCxt, void* pStmt, uint8_t precision, FCreateLogicNode func,
                                   SLogicNode** pRoot) {
X
Xiaoyu Wang 已提交
220
  SLogicNode* pNode = NULL;
X
Xiaoyu Wang 已提交
221
  int32_t     code = func(pCxt, pStmt, &pNode);
X
Xiaoyu Wang 已提交
222
  if (TSDB_CODE_SUCCESS == code && NULL != pNode) {
X
Xiaoyu Wang 已提交
223
    pNode->precision = precision;
X
Xiaoyu Wang 已提交
224
    code = pushLogicNode(pCxt, pRoot, pNode);
225
    pCxt->pCurrRoot = pNode;
226
  }
X
Xiaoyu Wang 已提交
227
  if (TSDB_CODE_SUCCESS != code) {
228
    nodesDestroyNode((SNode*)pNode);
X
Xiaoyu Wang 已提交
229
  }
X
Xiaoyu Wang 已提交
230
  return code;
231 232
}

X
Xiaoyu Wang 已提交
233 234 235 236
static int32_t createSelectRootLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, FCreateSelectLogicNode func,
                                         SLogicNode** pRoot) {
  return createRootLogicNode(pCxt, pSelect, pSelect->precision, (FCreateLogicNode)func, pRoot);
}
237

238
static EScanType getScanType(SLogicPlanContext* pCxt, SNodeList* pScanPseudoCols, SNodeList* pScanCols,
239
                             int8_t tableType, bool tagScan) {
X
Xiaoyu Wang 已提交
240 241 242 243
  if (pCxt->pPlanCxt->topicQuery || pCxt->pPlanCxt->streamQuery) {
    return SCAN_TYPE_STREAM;
  }

244 245 246 247
  if (TSDB_SYSTEM_TABLE == tableType) {
    return SCAN_TYPE_SYSTEM_TABLE;
  }

248 249 250 251
  if (tagScan) {
    return SCAN_TYPE_TAG;
  }

X
Xiaoyu Wang 已提交
252
  if (NULL == pScanCols) {
253 254 255
    if (NULL == pScanPseudoCols) {
      return SCAN_TYPE_TABLE;
    }
X
Xiaoyu Wang 已提交
256
    return FUNCTION_TYPE_BLOCK_DIST_INFO == ((SFunctionNode*)nodesListGetNode(pScanPseudoCols, 0))->funcType
257
               ? SCAN_TYPE_BLOCK_INFO
X
Xiaoyu Wang 已提交
258
               : SCAN_TYPE_TABLE;
X
Xiaoyu Wang 已提交
259 260
  }

261
  return SCAN_TYPE_TABLE;
X
Xiaoyu Wang 已提交
262 263
}

264
static SNode* createFirstCol(uint64_t tableId, const SSchema* pSchema) {
265
  SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
266 267 268
  if (NULL == pCol) {
    return NULL;
  }
269 270
  pCol->node.resType.type = pSchema->type;
  pCol->node.resType.bytes = pSchema->bytes;
271
  pCol->tableId = tableId;
272
  pCol->colId = pSchema->colId;
273
  pCol->colType = COLUMN_TYPE_COLUMN;
274
  strcpy(pCol->colName, pSchema->name);
275
  return (SNode*)pCol;
276 277
}

278
static int32_t addPrimaryKeyCol(uint64_t tableId, const SSchema* pSchema, SNodeList** pCols) {
X
Xiaoyu Wang 已提交
279
  bool   found = false;
280 281 282 283 284 285 286 287 288
  SNode* pCol = NULL;
  FOREACH(pCol, *pCols) {
    if (PRIMARYKEY_TIMESTAMP_COL_ID == ((SColumnNode*)pCol)->colId) {
      found = true;
      break;
    }
  }

  if (!found) {
289
    return nodesListMakeStrictAppend(pCols, createFirstCol(tableId, pSchema));
290 291 292 293
  }
  return TSDB_CODE_SUCCESS;
}

294 295 296 297 298 299 300 301 302 303 304 305 306 307
static int32_t addSystableFirstCol(uint64_t tableId, const SSchema* pSchema, SNodeList** pCols) {
  if (LIST_LENGTH(*pCols) > 0) {
    return TSDB_CODE_SUCCESS;
  }
  return nodesListMakeStrictAppend(pCols, createFirstCol(tableId, pSchema));
}

static int32_t addDefaultScanCol(const STableMeta* pMeta, SNodeList** pCols) {
  if (TSDB_SYSTEM_TABLE == pMeta->tableType) {
    return addSystableFirstCol(pMeta->uid, pMeta->schema, pCols);
  }
  return addPrimaryKeyCol(pMeta->uid, pMeta->schema, pCols);
}

X
Xiaoyu Wang 已提交
308 309
static int32_t makeScanLogicNode(SLogicPlanContext* pCxt, SRealTableNode* pRealTable, bool hasRepeatScanFuncs,
                                 SLogicNode** pLogicNode) {
X
Xiaoyu Wang 已提交
310
  SScanLogicNode* pScan = (SScanLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_SCAN);
X
Xiaoyu Wang 已提交
311
  if (NULL == pScan) {
312
    return TSDB_CODE_OUT_OF_MEMORY;
313 314
  }

wafwerar's avatar
wafwerar 已提交
315
  TSWAP(pScan->pVgroupList, pRealTable->pVgroupList);
X
Xiaoyu Wang 已提交
316
  TSWAP(pScan->pSmaIndexes, pRealTable->pSmaIndexes);
X
Xiaoyu Wang 已提交
317
  pScan->tableId = pRealTable->pMeta->uid;
X
Xiaoyu Wang 已提交
318
  pScan->stableId = pRealTable->pMeta->suid;
X
Xiaoyu Wang 已提交
319 320
  pScan->tableType = pRealTable->pMeta->tableType;
  pScan->scanSeq[0] = hasRepeatScanFuncs ? 2 : 1;
X
Xiaoyu Wang 已提交
321
  pScan->scanSeq[1] = 0;
X
Xiaoyu Wang 已提交
322
  pScan->scanRange = TSWINDOW_INITIALIZER;
X
Xiaoyu Wang 已提交
323 324 325 326
  pScan->tableName.type = TSDB_TABLE_NAME_T;
  pScan->tableName.acctId = pCxt->pPlanCxt->acctId;
  strcpy(pScan->tableName.dbname, pRealTable->table.dbName);
  strcpy(pScan->tableName.tname, pRealTable->table.tableName);
D
dapan1121 已提交
327
  pScan->showRewrite = pCxt->pPlanCxt->showRewrite;
328
  pScan->ratio = pRealTable->ratio;
329
  pScan->dataRequired = FUNC_DATA_REQUIRED_DATA_LOAD;
330
  pScan->cacheLastMode = pRealTable->cacheLastMode;
331

X
Xiaoyu Wang 已提交
332 333 334 335 336
  *pLogicNode = (SLogicNode*)pScan;

  return TSDB_CODE_SUCCESS;
}

337 338
static bool needScanDefaultCol(EScanType scanType) { return SCAN_TYPE_TABLE_COUNT != scanType; }

X
Xiaoyu Wang 已提交
339 340 341 342 343
static int32_t createScanLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SRealTableNode* pRealTable,
                                   SLogicNode** pLogicNode) {
  SScanLogicNode* pScan = NULL;
  int32_t         code = makeScanLogicNode(pCxt, pRealTable, pSelect->hasRepeatScanFuncs, (SLogicNode**)&pScan);

344 345 346
  pScan->node.groupAction = GROUP_ACTION_NONE;
  pScan->node.resultDataOrder = DATA_ORDER_LEVEL_IN_BLOCK;

X
Xiaoyu Wang 已提交
347
  // set columns to scan
X
Xiaoyu Wang 已提交
348 349 350 351
  if (TSDB_CODE_SUCCESS == code) {
    code = nodesCollectColumns(pSelect, SQL_CLAUSE_FROM, pRealTable->table.tableAlias, COLLECT_COL_TYPE_COL,
                               &pScan->pScanCols);
  }
352

353
  if (TSDB_CODE_SUCCESS == code) {
354 355
    code = nodesCollectColumns(pSelect, SQL_CLAUSE_FROM, pRealTable->table.tableAlias, COLLECT_COL_TYPE_TAG,
                               &pScan->pScanPseudoCols);
356 357 358
  }

  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
359
    code = nodesCollectFuncs(pSelect, SQL_CLAUSE_FROM, fmIsScanPseudoColumnFunc, &pScan->pScanPseudoCols);
360 361
  }

362 363
  // rewrite the expression in subsequent clauses
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
364
    code = rewriteExprsForSelect(pScan->pScanPseudoCols, pSelect, SQL_CLAUSE_FROM, NULL);
365 366
  }

367
  pScan->scanType = getScanType(pCxt, pScan->pScanPseudoCols, pScan->pScanCols, pScan->tableType, pSelect->tagScan);
X
Xiaoyu Wang 已提交
368

369 370 371 372
  if (NULL != pScan->pScanCols) {
    pScan->hasNormalCols = true;
  }

373
  if (TSDB_CODE_SUCCESS == code && needScanDefaultCol(pScan->scanType)) {
374
    code = addDefaultScanCol(pRealTable->pMeta, &pScan->pScanCols);
375 376
  }

377 378 379 380 381 382 383 384 385 386 387 388 389 390
  if (TSDB_CODE_SUCCESS == code && NULL != pSelect->pTags && NULL == pSelect->pPartitionByList) {
    pScan->pTags = nodesCloneList(pSelect->pTags);
    if (NULL == pScan->pTags) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
  }

  if (TSDB_CODE_SUCCESS == code && NULL != pSelect->pSubtable && NULL == pSelect->pPartitionByList) {
    pScan->pSubtable = nodesCloneNode(pSelect->pSubtable);
    if (NULL == pScan->pSubtable) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
  }

391 392
  // set output
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
393
    code = createColumnByRewriteExprs(pScan->pScanCols, &pScan->node.pTargets);
394 395
  }
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
396
    code = createColumnByRewriteExprs(pScan->pScanPseudoCols, &pScan->node.pTargets);
397
  }
398

X
Xiaoyu Wang 已提交
399 400 401
  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pScan;
  } else {
402
    nodesDestroyNode((SNode*)pScan);
X
Xiaoyu Wang 已提交
403
  }
404

405 406
  pCxt->hasScan = true;

X
Xiaoyu Wang 已提交
407
  return code;
X
Xiaoyu Wang 已提交
408
}
409

X
Xiaoyu Wang 已提交
410 411
static int32_t createSubqueryLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, STempTableNode* pTable,
                                       SLogicNode** pLogicNode) {
412
  return createQueryLogicNode(pCxt, pTable->pSubquery, pLogicNode);
413 414
}

X
Xiaoyu Wang 已提交
415 416
static int32_t createJoinLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SJoinTableNode* pJoinTable,
                                   SLogicNode** pLogicNode) {
X
Xiaoyu Wang 已提交
417
  SJoinLogicNode* pJoin = (SJoinLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_JOIN);
X
Xiaoyu Wang 已提交
418
  if (NULL == pJoin) {
419 420
    return TSDB_CODE_OUT_OF_MEMORY;
  }
X
Xiaoyu Wang 已提交
421 422

  pJoin->joinType = pJoinTable->joinType;
X
Xiaoyu Wang 已提交
423
  pJoin->isSingleTableJoin = pJoinTable->table.singleTable;
424
  pJoin->inputTsOrder = ORDER_ASC;
425 426
  pJoin->node.groupAction = GROUP_ACTION_CLEAR;
  pJoin->node.requireDataOrder = DATA_ORDER_LEVEL_GLOBAL;
427
  pJoin->node.resultDataOrder = DATA_ORDER_LEVEL_GLOBAL;
X
Xiaoyu Wang 已提交
428

429
  int32_t code = TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
430 431 432

  // set left and right node
  pJoin->node.pChildren = nodesMakeList();
X
Xiaoyu Wang 已提交
433
  if (NULL == pJoin->node.pChildren) {
434
    code = TSDB_CODE_OUT_OF_MEMORY;
435 436
  }

X
Xiaoyu Wang 已提交
437
  SLogicNode* pLeft = NULL;
438
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
439 440 441 442
    code = doCreateLogicNodeByTable(pCxt, pSelect, pJoinTable->pLeft, &pLeft);
    if (TSDB_CODE_SUCCESS == code) {
      code = nodesListStrictAppend(pJoin->node.pChildren, (SNode*)pLeft);
    }
443
  }
444

X
Xiaoyu Wang 已提交
445
  SLogicNode* pRight = NULL;
446
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
447 448 449 450 451
    code = doCreateLogicNodeByTable(pCxt, pSelect, pJoinTable->pRight, &pRight);
    if (TSDB_CODE_SUCCESS == code) {
      code = nodesListStrictAppend(pJoin->node.pChildren, (SNode*)pRight);
    }
  }
X
Xiaoyu Wang 已提交
452 453

  // set on conditions
X
Xiaoyu Wang 已提交
454
  if (TSDB_CODE_SUCCESS == code && NULL != pJoinTable->pOnCond) {
X
Xiaoyu Wang 已提交
455
    pJoin->pOnConditions = nodesCloneNode(pJoinTable->pOnCond);
X
Xiaoyu Wang 已提交
456 457 458
    if (NULL == pJoin->pOnConditions) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
459 460
  }

X
Xiaoyu Wang 已提交
461
  // set the output
X
Xiaoyu Wang 已提交
462 463
  if (TSDB_CODE_SUCCESS == code) {
    pJoin->node.pTargets = nodesCloneList(pLeft->pTargets);
D
dapan1121 已提交
464
    if (NULL == pJoin->node.pTargets) {
X
Xiaoyu Wang 已提交
465 466 467 468 469 470 471 472 473
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
    if (TSDB_CODE_SUCCESS == code) {
      code = nodesListStrictAppendList(pJoin->node.pTargets, nodesCloneList(pRight->pTargets));
    }
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pJoin;
474
  } else {
475
    nodesDestroyNode((SNode*)pJoin);
476 477
  }

478
  return code;
X
Xiaoyu Wang 已提交
479
}
480

X
Xiaoyu Wang 已提交
481 482
static int32_t doCreateLogicNodeByTable(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SNode* pTable,
                                        SLogicNode** pLogicNode) {
X
Xiaoyu Wang 已提交
483 484
  switch (nodeType(pTable)) {
    case QUERY_NODE_REAL_TABLE:
485
      return createScanLogicNode(pCxt, pSelect, (SRealTableNode*)pTable, pLogicNode);
X
Xiaoyu Wang 已提交
486
    case QUERY_NODE_TEMP_TABLE:
X
Xiaoyu Wang 已提交
487
      return createSubqueryLogicNode(pCxt, pSelect, (STempTableNode*)pTable, pLogicNode);
X
Xiaoyu Wang 已提交
488
    case QUERY_NODE_JOIN_TABLE:
X
Xiaoyu Wang 已提交
489
      return createJoinLogicNode(pCxt, pSelect, (SJoinTableNode*)pTable, pLogicNode);
X
Xiaoyu Wang 已提交
490 491
    default:
      break;
492
  }
X
Xiaoyu Wang 已提交
493 494 495
  return TSDB_CODE_FAILED;
}

X
Xiaoyu Wang 已提交
496 497
static int32_t createLogicNodeByTable(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SNode* pTable,
                                      SLogicNode** pLogicNode) {
X
Xiaoyu Wang 已提交
498
  SLogicNode* pNode = NULL;
X
Xiaoyu Wang 已提交
499
  int32_t     code = doCreateLogicNodeByTable(pCxt, pSelect, pTable, &pNode);
X
Xiaoyu Wang 已提交
500 501 502
  if (TSDB_CODE_SUCCESS == code) {
    pNode->pConditions = nodesCloneNode(pSelect->pWhere);
    if (NULL != pSelect->pWhere && NULL == pNode->pConditions) {
503
      nodesDestroyNode((SNode*)pNode);
X
Xiaoyu Wang 已提交
504 505
      return TSDB_CODE_OUT_OF_MEMORY;
    }
506
    pNode->precision = pSelect->precision;
X
Xiaoyu Wang 已提交
507
    *pLogicNode = pNode;
508
    pCxt->pCurrRoot = pNode;
X
Xiaoyu Wang 已提交
509 510 511 512
  }
  return code;
}

X
Xiaoyu Wang 已提交
513
static SColumnNode* createColumnByExpr(const char* pStmtName, SExprNode* pExpr) {
514
  SColumnNode* pCol = (SColumnNode*)nodesMakeNode(QUERY_NODE_COLUMN);
X
Xiaoyu Wang 已提交
515 516 517 518
  if (NULL == pCol) {
    return NULL;
  }
  pCol->node.resType = pExpr->resType;
X
Xiaoyu Wang 已提交
519
  snprintf(pCol->colName, sizeof(pCol->colName), "%s", pExpr->aliasName);
X
Xiaoyu Wang 已提交
520
  if (NULL != pStmtName) {
X
Xiaoyu Wang 已提交
521
    snprintf(pCol->tableAlias, sizeof(pCol->tableAlias), "%s", pStmtName);
X
Xiaoyu Wang 已提交
522
  }
X
Xiaoyu Wang 已提交
523
  return pCol;
524 525
}

526 527 528 529 530 531 532 533 534 535 536 537 538
static SNode* createGroupingSetNode(SNode* pExpr) {
  SGroupingSetNode* pGroupingSet = (SGroupingSetNode*)nodesMakeNode(QUERY_NODE_GROUPING_SET);
  if (NULL == pGroupingSet) {
    return NULL;
  }
  pGroupingSet->groupingSetType = GP_TYPE_NORMAL;
  if (TSDB_CODE_SUCCESS != nodesListMakeStrictAppend(&pGroupingSet->pParameterList, nodesCloneNode(pExpr))) {
    nodesDestroyNode((SNode*)pGroupingSet);
    return NULL;
  }
  return (SNode*)pGroupingSet;
}

539
static EGroupAction getGroupAction(SLogicPlanContext* pCxt, SSelectStmt* pSelect) {
X
Xiaoyu Wang 已提交
540 541
  return (pCxt->pPlanCxt->streamQuery || NULL != pSelect->pLimit || NULL != pSelect->pSlimit) ? GROUP_ACTION_KEEP
                                                                                              : GROUP_ACTION_NONE;
542 543
}

544 545 546 547 548
static EDataOrderLevel getRequireDataOrder(bool needTimeline, SSelectStmt* pSelect) {
  return needTimeline ? (NULL != pSelect->pPartitionByList ? DATA_ORDER_LEVEL_IN_GROUP : DATA_ORDER_LEVEL_GLOBAL)
                      : DATA_ORDER_LEVEL_NONE;
}

549
static int32_t createAggLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SLogicNode** pLogicNode) {
X
Xiaoyu Wang 已提交
550
  if (!pSelect->hasAggFuncs && NULL == pSelect->pGroupByList) {
551
    return TSDB_CODE_SUCCESS;
552 553
  }

X
Xiaoyu Wang 已提交
554
  SAggLogicNode* pAgg = (SAggLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_AGG);
555 556 557
  if (NULL == pAgg) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
558

559
  pAgg->hasLastRow = pSelect->hasLastRowFunc;
X
Xiaoyu Wang 已提交
560
  pAgg->hasLast = pSelect->hasLastFunc;
561
  pAgg->hasTimeLineFunc = pSelect->hasTimeLineFunc;
562
  pAgg->hasGroupKeyOptimized = false;
563
  pAgg->onlyHasKeepOrderFunc = pSelect->onlyHasKeepOrderFunc;
564
  pAgg->node.groupAction = getGroupAction(pCxt, pSelect);
565 566
  pAgg->node.requireDataOrder = getRequireDataOrder(pAgg->hasTimeLineFunc, pSelect);
  pAgg->node.resultDataOrder = pAgg->onlyHasKeepOrderFunc ? pAgg->node.requireDataOrder : DATA_ORDER_LEVEL_NONE;
567

X
Xiaoyu Wang 已提交
568 569
  int32_t code = TSDB_CODE_SUCCESS;

X
Xiaoyu Wang 已提交
570
  // set grouyp keys, agg funcs and having conditions
571
  if (TSDB_CODE_SUCCESS == code) {
572 573 574
    code = nodesCollectFuncs(pSelect, SQL_CLAUSE_GROUP_BY, fmIsAggFunc, &pAgg->pAggFuncs);
  }

X
Xiaoyu Wang 已提交
575
  // rewrite the expression in subsequent clauses
576
  if (TSDB_CODE_SUCCESS == code && NULL != pAgg->pAggFuncs) {
X
Xiaoyu Wang 已提交
577
    code = rewriteExprsForSelect(pAgg->pAggFuncs, pSelect, SQL_CLAUSE_GROUP_BY, NULL);
578 579 580
  }

  if (NULL != pSelect->pGroupByList) {
X
Xiaoyu Wang 已提交
581 582 583
    pAgg->pGroupKeys = nodesCloneList(pSelect->pGroupByList);
    if (NULL == pAgg->pGroupKeys) {
      code = TSDB_CODE_OUT_OF_MEMORY;
584
    }
585
  }
X
Xiaoyu Wang 已提交
586 587

  // rewrite the expression in subsequent clauses
X
Xiaoyu Wang 已提交
588
  SNodeList* pOutputGroupKeys = NULL;
589
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
590
    code = rewriteExprsForSelect(pAgg->pGroupKeys, pSelect, SQL_CLAUSE_GROUP_BY, &pOutputGroupKeys);
591
  }
592

593
  if (TSDB_CODE_SUCCESS == code && NULL != pSelect->pHaving) {
X
Xiaoyu Wang 已提交
594
    pAgg->node.pConditions = nodesCloneNode(pSelect->pHaving);
595 596 597
    if (NULL == pAgg->node.pConditions) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
X
Xiaoyu Wang 已提交
598
  }
599

X
Xiaoyu Wang 已提交
600
  // set the output
601
  if (TSDB_CODE_SUCCESS == code && NULL != pAgg->pAggFuncs) {
X
Xiaoyu Wang 已提交
602
    code = createColumnByRewriteExprs(pAgg->pAggFuncs, &pAgg->node.pTargets);
603 604
  }

X
Xiaoyu Wang 已提交
605 606 607 608 609 610 611 612 613
  if (TSDB_CODE_SUCCESS == code) {
    if (NULL != pOutputGroupKeys) {
      code = createColumnByRewriteExprs(pOutputGroupKeys, &pAgg->node.pTargets);
    } else if (NULL == pAgg->node.pTargets && NULL != pAgg->pGroupKeys) {
      code = createColumnByRewriteExprs(pAgg->pGroupKeys, &pAgg->node.pTargets);
    }
  }
  nodesDestroyList(pOutputGroupKeys);

614 615 616
  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pAgg;
  } else {
617
    nodesDestroyNode((SNode*)pAgg);
618 619 620
  }

  return code;
X
Xiaoyu Wang 已提交
621
}
622

623 624
static int32_t createIndefRowsFuncLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SLogicNode** pLogicNode) {
  // top/bottom are both an aggregate function and a indefinite rows function
625
  if (!pSelect->hasIndefiniteRowsFunc || pSelect->hasAggFuncs || NULL != pSelect->pWindow) {
626 627 628 629 630 631 632 633 634
    return TSDB_CODE_SUCCESS;
  }

  SIndefRowsFuncLogicNode* pIdfRowsFunc =
      (SIndefRowsFuncLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_INDEF_ROWS_FUNC);
  if (NULL == pIdfRowsFunc) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

635
  pIdfRowsFunc->isTailFunc = pSelect->hasTailFunc;
636
  pIdfRowsFunc->isUniqueFunc = pSelect->hasUniqueFunc;
637
  pIdfRowsFunc->isTimeLineFunc = pSelect->hasTimeLineFunc;
638
  pIdfRowsFunc->node.groupAction = getGroupAction(pCxt, pSelect);
639
  pIdfRowsFunc->node.requireDataOrder = getRequireDataOrder(pIdfRowsFunc->isTimeLineFunc, pSelect);
640
  pIdfRowsFunc->node.resultDataOrder = pIdfRowsFunc->node.requireDataOrder;
641

X
Xiaoyu Wang 已提交
642 643
  // indefinite rows functions and _select_values functions
  int32_t code = nodesCollectFuncs(pSelect, SQL_CLAUSE_SELECT, fmIsVectorFunc, &pIdfRowsFunc->pFuncs);
644
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
645
    code = rewriteExprsForSelect(pIdfRowsFunc->pFuncs, pSelect, SQL_CLAUSE_SELECT, NULL);
646 647 648 649
  }

  // set the output
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
650
    code = createColumnByRewriteExprs(pIdfRowsFunc->pFuncs, &pIdfRowsFunc->node.pTargets);
651 652 653 654 655
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pIdfRowsFunc;
  } else {
656
    nodesDestroyNode((SNode*)pIdfRowsFunc);
657 658 659 660 661
  }

  return code;
}

X
Xiaoyu Wang 已提交
662 663 664
static bool isInterpFunc(int32_t funcId) {
  return fmIsInterpFunc(funcId) || fmIsInterpPseudoColumnFunc(funcId) || fmIsGroupKeyFunc(funcId);
}
X
Xiaoyu Wang 已提交
665

X
Xiaoyu Wang 已提交
666 667 668 669 670 671 672 673 674 675
static int32_t createInterpFuncLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SLogicNode** pLogicNode) {
  if (!pSelect->hasInterpFunc) {
    return TSDB_CODE_SUCCESS;
  }

  SInterpFuncLogicNode* pInterpFunc = (SInterpFuncLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_INTERP_FUNC);
  if (NULL == pInterpFunc) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

676
  pInterpFunc->node.groupAction = getGroupAction(pCxt, pSelect);
677
  pInterpFunc->node.requireDataOrder = getRequireDataOrder(true, pSelect);
678 679
  pInterpFunc->node.resultDataOrder = pInterpFunc->node.requireDataOrder;

X
Xiaoyu Wang 已提交
680
  // interp functions and _group_key functions
X
Xiaoyu Wang 已提交
681
  int32_t code = nodesCollectFuncs(pSelect, SQL_CLAUSE_SELECT, isInterpFunc, &pInterpFunc->pFuncs);
X
Xiaoyu Wang 已提交
682
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
683
    code = rewriteExprsForSelect(pInterpFunc->pFuncs, pSelect, SQL_CLAUSE_SELECT, NULL);
X
Xiaoyu Wang 已提交
684 685
  }

X
Xiaoyu Wang 已提交
686 687 688 689 690 691 692 693 694
  if (TSDB_CODE_SUCCESS == code && NULL != pSelect->pFill) {
    SFillNode* pFill = (SFillNode*)pSelect->pFill;
    pInterpFunc->timeRange = pFill->timeRange;
    pInterpFunc->fillMode = pFill->mode;
    pInterpFunc->pTimeSeries = nodesCloneNode(pFill->pWStartTs);
    pInterpFunc->pFillValues = nodesCloneNode(pFill->pValues);
    if (NULL == pInterpFunc->pTimeSeries || (NULL != pFill->pValues && NULL == pInterpFunc->pFillValues)) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
X
Xiaoyu Wang 已提交
695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714
  }

  if (TSDB_CODE_SUCCESS == code && NULL != pSelect->pEvery) {
    pInterpFunc->interval = ((SValueNode*)pSelect->pEvery)->datum.i;
  }

  // set the output
  if (TSDB_CODE_SUCCESS == code) {
    code = createColumnByRewriteExprs(pInterpFunc->pFuncs, &pInterpFunc->node.pTargets);
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pInterpFunc;
  } else {
    nodesDestroyNode((SNode*)pInterpFunc);
  }

  return code;
}

X
Xiaoyu Wang 已提交
715 716
static int32_t createWindowLogicNodeFinalize(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SWindowLogicNode* pWindow,
                                             SLogicNode** pLogicNode) {
X
Xiaoyu Wang 已提交
717 718 719
  if (pCxt->pPlanCxt->streamQuery) {
    pWindow->triggerType = pCxt->pPlanCxt->triggerType;
    pWindow->watermark = pCxt->pPlanCxt->watermark;
720
    pWindow->deleteMark = pCxt->pPlanCxt->deleteMark;
721
    pWindow->igExpired = pCxt->pPlanCxt->igExpired;
5
54liuyao 已提交
722
  }
723
  pWindow->inputTsOrder = ORDER_ASC;
X
Xiaoyu Wang 已提交
724
  pWindow->outputTsOrder = ORDER_ASC;
5
54liuyao 已提交
725

726
  int32_t code = nodesCollectFuncs(pSelect, SQL_CLAUSE_WINDOW, fmIsWindowClauseFunc, &pWindow->pFuncs);
X
Xiaoyu Wang 已提交
727
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
728
    code = rewriteExprsForSelect(pWindow->pFuncs, pSelect, SQL_CLAUSE_WINDOW, NULL);
X
Xiaoyu Wang 已提交
729 730 731
  }

  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
732
    code = createColumnByRewriteExprs(pWindow->pFuncs, &pWindow->node.pTargets);
X
Xiaoyu Wang 已提交
733 734
  }

X
Xiaoyu Wang 已提交
735 736
  pSelect->hasAggFuncs = false;

X
Xiaoyu Wang 已提交
737 738 739
  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pWindow;
  } else {
740
    nodesDestroyNode((SNode*)pWindow);
X
Xiaoyu Wang 已提交
741 742 743 744 745
  }

  return code;
}

X
Xiaoyu Wang 已提交
746 747
static int32_t createWindowLogicNodeByState(SLogicPlanContext* pCxt, SStateWindowNode* pState, SSelectStmt* pSelect,
                                            SLogicNode** pLogicNode) {
748
  SWindowLogicNode* pWindow = (SWindowLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_WINDOW);
749 750 751 752 753
  if (NULL == pWindow) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  pWindow->winType = WINDOW_TYPE_STATE;
754
  pWindow->node.groupAction = getGroupAction(pCxt, pSelect);
755 756 757 758
  pWindow->node.requireDataOrder =
      pCxt->pPlanCxt->streamQuery ? DATA_ORDER_LEVEL_IN_BLOCK : getRequireDataOrder(true, pSelect);
  pWindow->node.resultDataOrder =
      pCxt->pPlanCxt->streamQuery ? DATA_ORDER_LEVEL_GLOBAL : pWindow->node.requireDataOrder;
759
  pWindow->pStateExpr = nodesCloneNode(pState->pExpr);
760
  pWindow->pTspk = nodesCloneNode(pState->pCol);
761
  if (NULL == pWindow->pStateExpr || NULL == pWindow->pTspk) {
762
    nodesDestroyNode((SNode*)pWindow);
763 764
    return TSDB_CODE_OUT_OF_MEMORY;
  }
D
dapan1121 已提交
765 766 767 768 769
  // rewrite the expression in subsequent clauses
  int32_t code = rewriteExprForSelect(pWindow->pStateExpr, pSelect, SQL_CLAUSE_WINDOW);
  if (TSDB_CODE_SUCCESS == code) {
    code = createWindowLogicNodeFinalize(pCxt, pSelect, pWindow, pLogicNode);
  }
X
Xiaoyu Wang 已提交
770

D
dapan1121 已提交
771
  return code;
772 773
}

X
Xiaoyu Wang 已提交
774 775
static int32_t createWindowLogicNodeBySession(SLogicPlanContext* pCxt, SSessionWindowNode* pSession,
                                              SSelectStmt* pSelect, SLogicNode** pLogicNode) {
776
  SWindowLogicNode* pWindow = (SWindowLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_WINDOW);
X
Xiaoyu Wang 已提交
777 778 779 780 781 782
  if (NULL == pWindow) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  pWindow->winType = WINDOW_TYPE_SESSION;
  pWindow->sessionGap = ((SValueNode*)pSession->pGap)->datum.i;
783
  pWindow->windowAlgo = pCxt->pPlanCxt->streamQuery ? SESSION_ALGO_STREAM_SINGLE : SESSION_ALGO_MERGE;
784
  pWindow->node.groupAction = getGroupAction(pCxt, pSelect);
785 786 787 788
  pWindow->node.requireDataOrder =
      pCxt->pPlanCxt->streamQuery ? DATA_ORDER_LEVEL_IN_BLOCK : getRequireDataOrder(true, pSelect);
  pWindow->node.resultDataOrder =
      pCxt->pPlanCxt->streamQuery ? DATA_ORDER_LEVEL_GLOBAL : pWindow->node.requireDataOrder;
X
Xiaoyu Wang 已提交
789

790
  pWindow->pTspk = nodesCloneNode((SNode*)pSession->pCol);
791
  if (NULL == pWindow->pTspk) {
792
    nodesDestroyNode((SNode*)pWindow);
793 794
    return TSDB_CODE_OUT_OF_MEMORY;
  }
5
54liuyao 已提交
795
  pWindow->pTsEnd = nodesCloneNode((SNode*)pSession->pCol);
796

X
Xiaoyu Wang 已提交
797 798 799
  return createWindowLogicNodeFinalize(pCxt, pSelect, pWindow, pLogicNode);
}

X
Xiaoyu Wang 已提交
800 801
static int32_t createWindowLogicNodeByInterval(SLogicPlanContext* pCxt, SIntervalWindowNode* pInterval,
                                               SSelectStmt* pSelect, SLogicNode** pLogicNode) {
802
  SWindowLogicNode* pWindow = (SWindowLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_WINDOW);
803 804 805
  if (NULL == pWindow) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
X
Xiaoyu Wang 已提交
806 807 808

  pWindow->winType = WINDOW_TYPE_INTERVAL;
  pWindow->interval = ((SValueNode*)pInterval->pInterval)->datum.i;
X
Xiaoyu Wang 已提交
809
  pWindow->intervalUnit = ((SValueNode*)pInterval->pInterval)->unit;
X
Xiaoyu Wang 已提交
810
  pWindow->offset = (NULL != pInterval->pOffset ? ((SValueNode*)pInterval->pOffset)->datum.i : 0);
811
  pWindow->sliding = (NULL != pInterval->pSliding ? ((SValueNode*)pInterval->pSliding)->datum.i : pWindow->interval);
X
Xiaoyu Wang 已提交
812 813
  pWindow->slidingUnit =
      (NULL != pInterval->pSliding ? ((SValueNode*)pInterval->pSliding)->unit : pWindow->intervalUnit);
814
  pWindow->windowAlgo = pCxt->pPlanCxt->streamQuery ? INTERVAL_ALGO_STREAM_SINGLE : INTERVAL_ALGO_HASH;
815
  pWindow->node.groupAction = (NULL != pInterval->pFill ? GROUP_ACTION_KEEP : getGroupAction(pCxt, pSelect));
816 817 818 819 820 821
  pWindow->node.requireDataOrder =
      pCxt->pPlanCxt->streamQuery
          ? DATA_ORDER_LEVEL_IN_BLOCK
          : (pSelect->hasTimeLineFunc ? getRequireDataOrder(true, pSelect) : DATA_ORDER_LEVEL_IN_BLOCK);
  pWindow->node.resultDataOrder =
      pCxt->pPlanCxt->streamQuery ? DATA_ORDER_LEVEL_GLOBAL : getRequireDataOrder(true, pSelect);
822

X
bugfix  
Xiaoyu Wang 已提交
823 824
  pWindow->pTspk = nodesCloneNode(pInterval->pCol);
  if (NULL == pWindow->pTspk) {
825
    nodesDestroyNode((SNode*)pWindow);
X
bugfix  
Xiaoyu Wang 已提交
826 827 828
    return TSDB_CODE_OUT_OF_MEMORY;
  }

X
Xiaoyu Wang 已提交
829
  return createWindowLogicNodeFinalize(pCxt, pSelect, pWindow, pLogicNode);
X
Xiaoyu Wang 已提交
830 831
}

X
Xiaoyu Wang 已提交
832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847
static int32_t createWindowLogicNodeByEvent(SLogicPlanContext* pCxt, SEventWindowNode* pEvent, SSelectStmt* pSelect,
                                            SLogicNode** pLogicNode) {
  SWindowLogicNode* pWindow = (SWindowLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_WINDOW);
  if (NULL == pWindow) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  pWindow->winType = WINDOW_TYPE_EVENT;
  pWindow->node.groupAction = getGroupAction(pCxt, pSelect);
  pWindow->node.requireDataOrder =
      pCxt->pPlanCxt->streamQuery ? DATA_ORDER_LEVEL_IN_BLOCK : getRequireDataOrder(true, pSelect);
  pWindow->node.resultDataOrder =
      pCxt->pPlanCxt->streamQuery ? DATA_ORDER_LEVEL_GLOBAL : pWindow->node.requireDataOrder;
  pWindow->pStartCond = nodesCloneNode(pEvent->pStartCond);
  pWindow->pEndCond = nodesCloneNode(pEvent->pEndCond);
  pWindow->pTspk = nodesCloneNode(pEvent->pCol);
X
Xiaoyu Wang 已提交
848
  if (NULL == pWindow->pStartCond || NULL == pWindow->pEndCond || NULL == pWindow->pTspk) {
X
Xiaoyu Wang 已提交
849 850 851
    nodesDestroyNode((SNode*)pWindow);
    return TSDB_CODE_OUT_OF_MEMORY;
  }
X
Xiaoyu Wang 已提交
852
  return createWindowLogicNodeFinalize(pCxt, pSelect, pWindow, pLogicNode);
X
Xiaoyu Wang 已提交
853 854
}

855
static int32_t createWindowLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SLogicNode** pLogicNode) {
X
Xiaoyu Wang 已提交
856
  if (NULL == pSelect->pWindow) {
857
    return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
858 859 860
  }

  switch (nodeType(pSelect->pWindow)) {
861 862
    case QUERY_NODE_STATE_WINDOW:
      return createWindowLogicNodeByState(pCxt, (SStateWindowNode*)pSelect->pWindow, pSelect, pLogicNode);
X
Xiaoyu Wang 已提交
863 864
    case QUERY_NODE_SESSION_WINDOW:
      return createWindowLogicNodeBySession(pCxt, (SSessionWindowNode*)pSelect->pWindow, pSelect, pLogicNode);
X
Xiaoyu Wang 已提交
865
    case QUERY_NODE_INTERVAL_WINDOW:
X
Xiaoyu Wang 已提交
866
      return createWindowLogicNodeByInterval(pCxt, (SIntervalWindowNode*)pSelect->pWindow, pSelect, pLogicNode);
X
Xiaoyu Wang 已提交
867 868
    case QUERY_NODE_EVENT_WINDOW:
      return createWindowLogicNodeByEvent(pCxt, (SEventWindowNode*)pSelect->pWindow, pSelect, pLogicNode);
X
Xiaoyu Wang 已提交
869 870 871 872
    default:
      break;
  }

873
  return TSDB_CODE_FAILED;
X
Xiaoyu Wang 已提交
874 875
}

876 877 878
static EDealRes needFillValueImpl(SNode* pNode, void* pContext) {
  if (QUERY_NODE_COLUMN == nodeType(pNode)) {
    SColumnNode* pCol = (SColumnNode*)pNode;
879 880
    if (COLUMN_TYPE_WINDOW_START != pCol->colType && COLUMN_TYPE_WINDOW_END != pCol->colType &&
        COLUMN_TYPE_WINDOW_DURATION != pCol->colType && COLUMN_TYPE_GROUP_KEY != pCol->colType) {
881 882 883 884 885 886 887 888 889 890 891 892 893
      *(bool*)pContext = true;
      return DEAL_RES_END;
    }
  }
  return DEAL_RES_CONTINUE;
}

static bool needFillValue(SNode* pNode) {
  bool hasFillCol = false;
  nodesWalkExpr(pNode, needFillValueImpl, &hasFillCol);
  return hasFillCol;
}

X
Xiaoyu Wang 已提交
894
static int32_t partFillExprs(SSelectStmt* pSelect, SNodeList** pFillExprs, SNodeList** pNotFillExprs) {
895 896
  int32_t code = TSDB_CODE_SUCCESS;
  SNode*  pProject = NULL;
X
Xiaoyu Wang 已提交
897
  FOREACH(pProject, pSelect->pProjectionList) {
898 899
    if (needFillValue(pProject)) {
      code = nodesListMakeStrictAppend(pFillExprs, nodesCloneNode(pProject));
X
Xiaoyu Wang 已提交
900
    } else if (QUERY_NODE_VALUE != nodeType(pProject)) {
901 902 903 904 905 906 907 908
      code = nodesListMakeStrictAppend(pNotFillExprs, nodesCloneNode(pProject));
    }
    if (TSDB_CODE_SUCCESS != code) {
      NODES_DESTORY_LIST(*pFillExprs);
      NODES_DESTORY_LIST(*pNotFillExprs);
      break;
    }
  }
X
Xiaoyu Wang 已提交
909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924
  if (!pSelect->isDistinct) {
    SNode* pOrderExpr = NULL;
    FOREACH(pOrderExpr, pSelect->pOrderByList) {
      SNode* pExpr = ((SOrderByExprNode*)pOrderExpr)->pExpr;
      if (needFillValue(pExpr)) {
        code = nodesListMakeStrictAppend(pFillExprs, nodesCloneNode(pExpr));
      } else if (QUERY_NODE_VALUE != nodeType(pExpr)) {
        code = nodesListMakeStrictAppend(pNotFillExprs, nodesCloneNode(pExpr));
      }
      if (TSDB_CODE_SUCCESS != code) {
        NODES_DESTORY_LIST(*pFillExprs);
        NODES_DESTORY_LIST(*pNotFillExprs);
        break;
      }
    }
  }
925 926 927
  return code;
}

X
Xiaoyu Wang 已提交
928 929 930 931 932 933 934
static int32_t createFillLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SLogicNode** pLogicNode) {
  if (NULL == pSelect->pWindow || QUERY_NODE_INTERVAL_WINDOW != nodeType(pSelect->pWindow) ||
      NULL == ((SIntervalWindowNode*)pSelect->pWindow)->pFill) {
    return TSDB_CODE_SUCCESS;
  }

  SFillNode* pFillNode = (SFillNode*)(((SIntervalWindowNode*)pSelect->pWindow)->pFill);
935 936 937
  if (FILL_MODE_NONE == pFillNode->mode) {
    return TSDB_CODE_SUCCESS;
  }
X
Xiaoyu Wang 已提交
938

939
  SFillLogicNode* pFill = (SFillLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_FILL);
X
Xiaoyu Wang 已提交
940 941 942 943
  if (NULL == pFill) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

944
  pFill->node.groupAction = getGroupAction(pCxt, pSelect);
945 946
  pFill->node.requireDataOrder = getRequireDataOrder(true, pSelect);
  pFill->node.resultDataOrder = pFill->node.requireDataOrder;
X
Xiaoyu Wang 已提交
947
  pFill->inputTsOrder = ORDER_ASC;
948

X
Xiaoyu Wang 已提交
949
  int32_t code = partFillExprs(pSelect, &pFill->pFillExprs, &pFill->pNotFillExprs);
950
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
951
    code = rewriteExprsForSelect(pFill->pFillExprs, pSelect, SQL_CLAUSE_FILL, NULL);
952 953
  }
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
954
    code = rewriteExprsForSelect(pFill->pNotFillExprs, pSelect, SQL_CLAUSE_FILL, NULL);
955 956 957 958 959 960
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = createColumnByRewriteExprs(pFill->pFillExprs, &pFill->node.pTargets);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = createColumnByRewriteExprs(pFill->pNotFillExprs, &pFill->node.pTargets);
961
  }
X
Xiaoyu Wang 已提交
962 963

  pFill->mode = pFillNode->mode;
X
Xiaoyu Wang 已提交
964
  pFill->timeRange = pFillNode->timeRange;
X
Xiaoyu Wang 已提交
965 966 967 968 969 970
  pFill->pValues = nodesCloneNode(pFillNode->pValues);
  pFill->pWStartTs = nodesCloneNode(pFillNode->pWStartTs);
  if ((NULL != pFillNode->pValues && NULL == pFill->pValues) || NULL == pFill->pWStartTs) {
    code = TSDB_CODE_OUT_OF_MEMORY;
  }

X
Xiaoyu Wang 已提交
971 972 973 974
  if (TSDB_CODE_SUCCESS == code && 0 == LIST_LENGTH(pFill->node.pTargets)) {
    code = createColumnByRewriteExpr(pFill->pWStartTs, &pFill->node.pTargets);
  }

X
Xiaoyu Wang 已提交
975 976 977
  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pFill;
  } else {
978
    nodesDestroyNode((SNode*)pFill);
X
Xiaoyu Wang 已提交
979 980 981 982 983
  }

  return code;
}

984 985 986 987 988 989 990 991
static bool isPrimaryKeySort(SNodeList* pOrderByList) {
  SNode* pExpr = ((SOrderByExprNode*)nodesListGetNode(pOrderByList, 0))->pExpr;
  if (QUERY_NODE_COLUMN != nodeType(pExpr)) {
    return false;
  }
  return PRIMARYKEY_TIMESTAMP_COL_ID == ((SColumnNode*)pExpr)->colId;
}

X
Xiaoyu Wang 已提交
992 993 994 995 996
static int32_t createSortLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SLogicNode** pLogicNode) {
  if (NULL == pSelect->pOrderByList) {
    return TSDB_CODE_SUCCESS;
  }

997
  SSortLogicNode* pSort = (SSortLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_SORT);
X
Xiaoyu Wang 已提交
998 999 1000 1001
  if (NULL == pSort) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

1002
  pSort->groupSort = pSelect->groupSort;
1003 1004
  pSort->node.groupAction = pSort->groupSort ? GROUP_ACTION_KEEP : GROUP_ACTION_CLEAR;
  pSort->node.requireDataOrder = DATA_ORDER_LEVEL_NONE;
1005 1006 1007
  pSort->node.resultDataOrder = isPrimaryKeySort(pSelect->pOrderByList)
                                    ? (pSort->groupSort ? DATA_ORDER_LEVEL_IN_GROUP : DATA_ORDER_LEVEL_GLOBAL)
                                    : DATA_ORDER_LEVEL_NONE;
1008

1009
  int32_t code = nodesCollectColumns(pSelect, SQL_CLAUSE_ORDER_BY, NULL, COLLECT_COL_TYPE_ALL, &pSort->node.pTargets);
1010 1011 1012 1013
  if (TSDB_CODE_SUCCESS == code && NULL == pSort->node.pTargets) {
    code = nodesListMakeStrictAppend(&pSort->node.pTargets,
                                     nodesCloneNode(nodesListGetNode(pCxt->pCurrRoot->pTargets, 0)));
  }
X
Xiaoyu Wang 已提交
1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024

  if (TSDB_CODE_SUCCESS == code) {
    pSort->pSortKeys = nodesCloneList(pSelect->pOrderByList);
    if (NULL == pSort->pSortKeys) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pSort;
  } else {
1025
    nodesDestroyNode((SNode*)pSort);
X
Xiaoyu Wang 已提交
1026 1027 1028 1029 1030
  }

  return code;
}

X
Xiaoyu Wang 已提交
1031 1032
static int32_t createColumnByProjections(SLogicPlanContext* pCxt, const char* pStmtName, SNodeList* pExprs,
                                         SNodeList** pCols) {
X
Xiaoyu Wang 已提交
1033
  SNodeList* pList = nodesMakeList();
X
Xiaoyu Wang 已提交
1034
  if (NULL == pList) {
1035 1036 1037
    return TSDB_CODE_OUT_OF_MEMORY;
  }

X
Xiaoyu Wang 已提交
1038
  SNode* pNode;
X
Xiaoyu Wang 已提交
1039
  FOREACH(pNode, pExprs) {
1040
    if (TSDB_CODE_SUCCESS != nodesListAppend(pList, (SNode*)createColumnByExpr(pStmtName, (SExprNode*)pNode))) {
X
Xiaoyu Wang 已提交
1041 1042
      nodesDestroyList(pList);
      return TSDB_CODE_OUT_OF_MEMORY;
1043 1044
    }
  }
1045

X
Xiaoyu Wang 已提交
1046 1047
  *pCols = pList;
  return TSDB_CODE_SUCCESS;
1048 1049
}

X
Xiaoyu Wang 已提交
1050
static int32_t createProjectLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SLogicNode** pLogicNode) {
X
Xiaoyu Wang 已提交
1051
  SProjectLogicNode* pProject = (SProjectLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_PROJECT);
X
Xiaoyu Wang 已提交
1052
  if (NULL == pProject) {
1053 1054 1055
    return TSDB_CODE_OUT_OF_MEMORY;
  }

X
Xiaoyu Wang 已提交
1056 1057
  TSWAP(pProject->node.pLimit, pSelect->pLimit);
  TSWAP(pProject->node.pSlimit, pSelect->pSlimit);
1058
  pProject->ignoreGroupId = (NULL == pSelect->pPartitionByList);
1059 1060
  pProject->node.groupAction =
      (!pSelect->isSubquery && pCxt->pPlanCxt->streamQuery) ? GROUP_ACTION_KEEP : GROUP_ACTION_CLEAR;
1061 1062
  pProject->node.requireDataOrder = DATA_ORDER_LEVEL_NONE;
  pProject->node.resultDataOrder = DATA_ORDER_LEVEL_NONE;
1063

X
Xiaoyu Wang 已提交
1064
  int32_t code = TSDB_CODE_SUCCESS;
1065

X
Xiaoyu Wang 已提交
1066
  pProject->pProjections = nodesCloneList(pSelect->pProjectionList);
X
Xiaoyu Wang 已提交
1067 1068
  if (NULL == pProject->pProjections) {
    code = TSDB_CODE_OUT_OF_MEMORY;
1069
  }
X
Xiaoyu Wang 已提交
1070
  strcpy(pProject->stmtName, pSelect->stmtName);
1071 1072

  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1073
    code = createColumnByProjections(pCxt, pSelect->stmtName, pSelect->pProjectionList, &pProject->node.pTargets);
1074
  }
1075

1076
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1077 1078
    *pLogicNode = (SLogicNode*)pProject;
  } else {
1079
    nodesDestroyNode((SNode*)pProject);
1080
  }
1081

1082
  return code;
1083 1084
}

1085
static int32_t createPartitionLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SLogicNode** pLogicNode) {
1086
  if (NULL == pSelect->pPartitionByList) {
1087 1088 1089
    return TSDB_CODE_SUCCESS;
  }

1090
  SPartitionLogicNode* pPartition = (SPartitionLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_PARTITION);
1091 1092 1093 1094
  if (NULL == pPartition) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

1095 1096 1097 1098
  pPartition->node.groupAction = GROUP_ACTION_SET;
  pPartition->node.requireDataOrder = DATA_ORDER_LEVEL_NONE;
  pPartition->node.resultDataOrder = DATA_ORDER_LEVEL_NONE;

1099 1100
  int32_t code =
      nodesCollectColumns(pSelect, SQL_CLAUSE_PARTITION_BY, NULL, COLLECT_COL_TYPE_ALL, &pPartition->node.pTargets);
1101
  if (TSDB_CODE_SUCCESS == code && NULL == pPartition->node.pTargets) {
1102 1103
    code = nodesListMakeStrictAppend(&pPartition->node.pTargets,
                                     nodesCloneNode(nodesListGetNode(pCxt->pCurrRoot->pTargets, 0)));
1104
  }
1105 1106 1107 1108 1109 1110 1111 1112

  if (TSDB_CODE_SUCCESS == code) {
    pPartition->pPartitionKeys = nodesCloneList(pSelect->pPartitionByList);
    if (NULL == pPartition->pPartitionKeys) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
  }

1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126
  if (TSDB_CODE_SUCCESS == code && NULL != pSelect->pTags) {
    pPartition->pTags = nodesCloneList(pSelect->pTags);
    if (NULL == pPartition->pTags) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
  }

  if (TSDB_CODE_SUCCESS == code && NULL != pSelect->pSubtable) {
    pPartition->pSubtable = nodesCloneNode(pSelect->pSubtable);
    if (NULL == pPartition->pSubtable) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
  }

1127 1128 1129
  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pPartition;
  } else {
1130
    nodesDestroyNode((SNode*)pPartition);
1131 1132 1133
  }

  return code;
1134 1135 1136 1137 1138 1139 1140
}

static int32_t createDistinctLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SLogicNode** pLogicNode) {
  if (!pSelect->isDistinct) {
    return TSDB_CODE_SUCCESS;
  }

1141 1142 1143 1144 1145
  SAggLogicNode* pAgg = (SAggLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_AGG);
  if (NULL == pAgg) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

1146
  pAgg->node.groupAction = GROUP_ACTION_CLEAR;
1147 1148 1149
  pAgg->node.requireDataOrder = DATA_ORDER_LEVEL_NONE;
  pAgg->node.resultDataOrder = DATA_ORDER_LEVEL_NONE;

1150 1151
  int32_t code = TSDB_CODE_SUCCESS;
  // set grouyp keys, agg funcs and having conditions
1152
  SNodeList* pGroupKeys = NULL;
1153
  SNode*     pProjection = NULL;
1154 1155 1156 1157 1158 1159 1160 1161 1162
  FOREACH(pProjection, pSelect->pProjectionList) {
    code = nodesListMakeStrictAppend(&pGroupKeys, createGroupingSetNode(pProjection));
    if (TSDB_CODE_SUCCESS != code) {
      nodesDestroyList(pGroupKeys);
      break;
    }
  }
  if (TSDB_CODE_SUCCESS == code) {
    pAgg->pGroupKeys = pGroupKeys;
1163 1164 1165 1166
  }

  // rewrite the expression in subsequent clauses
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1167
    code = rewriteExprsForSelect(pAgg->pGroupKeys, pSelect, SQL_CLAUSE_DISTINCT, NULL);
1168 1169 1170 1171
  }

  // set the output
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1172
    code = createColumnByRewriteExprs(pAgg->pGroupKeys, &pAgg->node.pTargets);
1173 1174 1175 1176 1177
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pAgg;
  } else {
1178
    nodesDestroyNode((SNode*)pAgg);
1179 1180 1181
  }

  return code;
1182 1183
}

1184 1185 1186 1187 1188 1189
static int32_t createSelectWithoutFromLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect,
                                                SLogicNode** pLogicNode) {
  return createProjectLogicNode(pCxt, pSelect, pLogicNode);
}

static int32_t createSelectFromLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SLogicNode** pLogicNode) {
1190
  SLogicNode* pRoot = NULL;
X
Xiaoyu Wang 已提交
1191
  int32_t     code = createLogicNodeByTable(pCxt, pSelect, pSelect->pFromTable, &pRoot);
1192
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1193
    code = createSelectRootLogicNode(pCxt, pSelect, createPartitionLogicNode, &pRoot);
1194
  }
1195
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1196
    code = createSelectRootLogicNode(pCxt, pSelect, createWindowLogicNode, &pRoot);
1197
  }
X
Xiaoyu Wang 已提交
1198
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1199
    code = createSelectRootLogicNode(pCxt, pSelect, createFillLogicNode, &pRoot);
X
Xiaoyu Wang 已提交
1200
  }
1201
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1202
    code = createSelectRootLogicNode(pCxt, pSelect, createAggLogicNode, &pRoot);
X
Xiaoyu Wang 已提交
1203
  }
1204
  if (TSDB_CODE_SUCCESS == code) {
1205
    code = createSelectRootLogicNode(pCxt, pSelect, createIndefRowsFuncLogicNode, &pRoot);
1206
  }
X
Xiaoyu Wang 已提交
1207 1208 1209
  if (TSDB_CODE_SUCCESS == code) {
    code = createSelectRootLogicNode(pCxt, pSelect, createInterpFuncLogicNode, &pRoot);
  }
1210
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1211
    code = createSelectRootLogicNode(pCxt, pSelect, createDistinctLogicNode, &pRoot);
1212
  }
X
Xiaoyu Wang 已提交
1213
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1214
    code = createSelectRootLogicNode(pCxt, pSelect, createSortLogicNode, &pRoot);
X
Xiaoyu Wang 已提交
1215
  }
1216
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1217
    code = createSelectRootLogicNode(pCxt, pSelect, createProjectLogicNode, &pRoot);
X
Xiaoyu Wang 已提交
1218
  }
1219 1220 1221 1222

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = pRoot;
  } else {
1223
    nodesDestroyNode((SNode*)pRoot);
X
Xiaoyu Wang 已提交
1224
  }
1225 1226

  return code;
1227 1228
}

1229 1230 1231 1232 1233 1234 1235 1236
static int32_t createSelectLogicNode(SLogicPlanContext* pCxt, SSelectStmt* pSelect, SLogicNode** pLogicNode) {
  if (NULL == pSelect->pFromTable) {
    return createSelectWithoutFromLogicNode(pCxt, pSelect, pLogicNode);
  } else {
    return createSelectFromLogicNode(pCxt, pSelect, pLogicNode);
  }
}

X
Xiaoyu Wang 已提交
1237 1238 1239
static int32_t createSetOpRootLogicNode(SLogicPlanContext* pCxt, SSetOperator* pSetOperator, FCreateSetOpLogicNode func,
                                        SLogicNode** pRoot) {
  return createRootLogicNode(pCxt, pSetOperator, pSetOperator->precision, (FCreateLogicNode)func, pRoot);
X
Xiaoyu Wang 已提交
1240 1241 1242 1243 1244 1245 1246
}

static int32_t createSetOpSortLogicNode(SLogicPlanContext* pCxt, SSetOperator* pSetOperator, SLogicNode** pLogicNode) {
  if (NULL == pSetOperator->pOrderByList) {
    return TSDB_CODE_SUCCESS;
  }

1247
  SSortLogicNode* pSort = (SSortLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_SORT);
X
Xiaoyu Wang 已提交
1248 1249 1250 1251
  if (NULL == pSort) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

X
Xiaoyu Wang 已提交
1252
  TSWAP(pSort->node.pLimit, pSetOperator->pLimit);
1253

X
Xiaoyu Wang 已提交
1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270
  int32_t code = TSDB_CODE_SUCCESS;

  pSort->node.pTargets = nodesCloneList(pSetOperator->pProjectionList);
  if (NULL == pSort->node.pTargets) {
    code = TSDB_CODE_OUT_OF_MEMORY;
  }

  if (TSDB_CODE_SUCCESS == code) {
    pSort->pSortKeys = nodesCloneList(pSetOperator->pOrderByList);
    if (NULL == pSort->pSortKeys) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pSort;
  } else {
1271
    nodesDestroyNode((SNode*)pSort);
X
Xiaoyu Wang 已提交
1272 1273 1274 1275 1276
  }

  return code;
}

X
Xiaoyu Wang 已提交
1277 1278
static int32_t createSetOpProjectLogicNode(SLogicPlanContext* pCxt, SSetOperator* pSetOperator,
                                           SLogicNode** pLogicNode) {
X
Xiaoyu Wang 已提交
1279 1280 1281 1282 1283
  SProjectLogicNode* pProject = (SProjectLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_PROJECT);
  if (NULL == pProject) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

1284
  if (NULL == pSetOperator->pOrderByList) {
X
Xiaoyu Wang 已提交
1285
    TSWAP(pProject->node.pLimit, pSetOperator->pLimit);
X
Xiaoyu Wang 已提交
1286
  }
1287
  pProject->ignoreGroupId = true;
X
Xiaoyu Wang 已提交
1288 1289 1290 1291 1292 1293 1294 1295 1296

  int32_t code = TSDB_CODE_SUCCESS;

  pProject->pProjections = nodesCloneList(pSetOperator->pProjectionList);
  if (NULL == pProject->pProjections) {
    code = TSDB_CODE_OUT_OF_MEMORY;
  }

  if (TSDB_CODE_SUCCESS == code) {
1297 1298
    code = createColumnByProjections(pCxt, pSetOperator->stmtName, pSetOperator->pProjectionList,
                                     &pProject->node.pTargets);
X
Xiaoyu Wang 已提交
1299 1300 1301 1302 1303
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pProject;
  } else {
1304
    nodesDestroyNode((SNode*)pProject);
X
Xiaoyu Wang 已提交
1305 1306 1307 1308 1309
  }

  return code;
}

X
Xiaoyu Wang 已提交
1310 1311 1312 1313 1314 1315
static int32_t createSetOpAggLogicNode(SLogicPlanContext* pCxt, SSetOperator* pSetOperator, SLogicNode** pLogicNode) {
  SAggLogicNode* pAgg = (SAggLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_AGG);
  if (NULL == pAgg) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

1316
  if (NULL == pSetOperator->pOrderByList) {
X
Xiaoyu Wang 已提交
1317
    TSWAP(pAgg->node.pSlimit, pSetOperator->pLimit);
1318 1319
  }

X
Xiaoyu Wang 已提交
1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332
  int32_t code = TSDB_CODE_SUCCESS;
  pAgg->pGroupKeys = nodesCloneList(pSetOperator->pProjectionList);
  if (NULL == pAgg->pGroupKeys) {
    code = TSDB_CODE_OUT_OF_MEMORY;
  }

  // rewrite the expression in subsequent clauses
  if (TSDB_CODE_SUCCESS == code) {
    code = rewriteExprs(pAgg->pGroupKeys, pSetOperator->pOrderByList);
  }

  // set the output
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1333
    code = createColumnByRewriteExprs(pAgg->pGroupKeys, &pAgg->node.pTargets);
X
Xiaoyu Wang 已提交
1334 1335 1336 1337 1338
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pAgg;
  } else {
1339
    nodesDestroyNode((SNode*)pAgg);
X
Xiaoyu Wang 已提交
1340 1341 1342 1343 1344
  }

  return code;
}

X
Xiaoyu Wang 已提交
1345 1346
static int32_t createSetOpLogicNode(SLogicPlanContext* pCxt, SSetOperator* pSetOperator, SLogicNode** pLogicNode) {
  SLogicNode* pSetOp = NULL;
X
Xiaoyu Wang 已提交
1347
  int32_t     code = TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
1348 1349 1350 1351
  switch (pSetOperator->opType) {
    case SET_OP_TYPE_UNION_ALL:
      code = createSetOpProjectLogicNode(pCxt, pSetOperator, &pSetOp);
      break;
X
Xiaoyu Wang 已提交
1352 1353 1354
    case SET_OP_TYPE_UNION:
      code = createSetOpAggLogicNode(pCxt, pSetOperator, &pSetOp);
      break;
X
Xiaoyu Wang 已提交
1355
    default:
1356
      code = TSDB_CODE_FAILED;
X
Xiaoyu Wang 已提交
1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375
      break;
  }

  SLogicNode* pLeft = NULL;
  if (TSDB_CODE_SUCCESS == code) {
    code = createQueryLogicNode(pCxt, pSetOperator->pLeft, &pLeft);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = nodesListMakeStrictAppend(&pSetOp->pChildren, (SNode*)pLeft);
  }
  SLogicNode* pRight = NULL;
  if (TSDB_CODE_SUCCESS == code) {
    code = createQueryLogicNode(pCxt, pSetOperator->pRight, &pRight);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = nodesListStrictAppend(pSetOp->pChildren, (SNode*)pRight);
  }

  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1376
    pSetOp->precision = pSetOperator->precision;
X
Xiaoyu Wang 已提交
1377 1378
    *pLogicNode = (SLogicNode*)pSetOp;
  } else {
1379
    nodesDestroyNode((SNode*)pSetOp);
X
Xiaoyu Wang 已提交
1380 1381 1382 1383 1384
  }

  return code;
}

X
Xiaoyu Wang 已提交
1385 1386
static int32_t createSetOperatorLogicNode(SLogicPlanContext* pCxt, SSetOperator* pSetOperator,
                                          SLogicNode** pLogicNode) {
1387
  SLogicNode* pRoot = NULL;
X
Xiaoyu Wang 已提交
1388
  int32_t     code = createSetOpLogicNode(pCxt, pSetOperator, &pRoot);
1389
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1390
    code = createSetOpRootLogicNode(pCxt, pSetOperator, createSetOpSortLogicNode, &pRoot);
1391 1392 1393 1394 1395
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = pRoot;
  } else {
1396
    nodesDestroyNode((SNode*)pRoot);
1397 1398 1399 1400 1401
  }

  return code;
}

1402
static int32_t getMsgType(ENodeType sqlType) {
X
Xiaoyu Wang 已提交
1403 1404
  switch (sqlType) {
    case QUERY_NODE_CREATE_TABLE_STMT:
X
Xiaoyu Wang 已提交
1405
    case QUERY_NODE_CREATE_MULTI_TABLES_STMT:
X
Xiaoyu Wang 已提交
1406 1407 1408
      return TDMT_VND_CREATE_TABLE;
    case QUERY_NODE_DROP_TABLE_STMT:
      return TDMT_VND_DROP_TABLE;
X
Xiaoyu Wang 已提交
1409 1410
    case QUERY_NODE_ALTER_TABLE_STMT:
      return TDMT_VND_ALTER_TABLE;
X
Xiaoyu Wang 已提交
1411 1412
    case QUERY_NODE_FLUSH_DATABASE_STMT:
      return TDMT_VND_COMMIT;
X
Xiaoyu Wang 已提交
1413 1414 1415 1416
    default:
      break;
  }
  return TDMT_VND_SUBMIT;
1417 1418
}

X
Xiaoyu Wang 已提交
1419
static int32_t createVnodeModifLogicNode(SLogicPlanContext* pCxt, SVnodeModifyOpStmt* pStmt, SLogicNode** pLogicNode) {
1420
  SVnodeModifyLogicNode* pModif = (SVnodeModifyLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_VNODE_MODIFY);
1421 1422 1423
  if (NULL == pModif) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }
X
Xiaoyu Wang 已提交
1424
  pModif->modifyType = MODIFY_TABLE_TYPE_INSERT;
wafwerar's avatar
wafwerar 已提交
1425
  TSWAP(pModif->pDataBlocks, pStmt->pDataBlocks);
1426
  pModif->msgType = getMsgType(pStmt->sqlNodeType);
1427 1428
  *pLogicNode = (SLogicNode*)pModif;
  return TSDB_CODE_SUCCESS;
1429 1430
}

X
Xiaoyu Wang 已提交
1431 1432 1433 1434 1435 1436
static int32_t createDeleteRootLogicNode(SLogicPlanContext* pCxt, SDeleteStmt* pDelete, FCreateDeleteLogicNode func,
                                         SLogicNode** pRoot) {
  return createRootLogicNode(pCxt, pDelete, pDelete->precision, (FCreateLogicNode)func, pRoot);
}

static int32_t createDeleteScanLogicNode(SLogicPlanContext* pCxt, SDeleteStmt* pDelete, SLogicNode** pLogicNode) {
X
Xiaoyu Wang 已提交
1437 1438 1439 1440 1441 1442
  SScanLogicNode* pScan = NULL;
  int32_t         code = makeScanLogicNode(pCxt, (SRealTableNode*)pDelete->pFromTable, false, (SLogicNode**)&pScan);

  // set columns to scan
  if (TSDB_CODE_SUCCESS == code) {
    pScan->scanType = SCAN_TYPE_TABLE;
D
dapan1121 已提交
1443
    pScan->scanRange = pDelete->timeRange;
X
Xiaoyu Wang 已提交
1444 1445 1446 1447 1448 1449
    pScan->pScanCols = nodesCloneList(((SFunctionNode*)pDelete->pCountFunc)->pParameterList);
    if (NULL == pScan->pScanCols) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
  }

X
Xiaoyu Wang 已提交
1450 1451
  if (TSDB_CODE_SUCCESS == code && NULL != pDelete->pTagCond) {
    pScan->pTagCond = nodesCloneNode(pDelete->pTagCond);
X
Xiaoyu Wang 已提交
1452 1453 1454 1455 1456 1457 1458
    if (NULL == pScan->pTagCond) {
      code = TSDB_CODE_OUT_OF_MEMORY;
    }
  }

  // set output
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1459
    code = createColumnByRewriteExprs(pScan->pScanCols, &pScan->node.pTargets);
X
Xiaoyu Wang 已提交
1460 1461 1462 1463 1464
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pScan;
  } else {
1465
    nodesDestroyNode((SNode*)pScan);
X
Xiaoyu Wang 已提交
1466 1467 1468
  }

  return code;
X
Xiaoyu Wang 已提交
1469 1470 1471
}

static int32_t createDeleteAggLogicNode(SLogicPlanContext* pCxt, SDeleteStmt* pDelete, SLogicNode** pLogicNode) {
X
Xiaoyu Wang 已提交
1472 1473 1474 1475 1476 1477
  SAggLogicNode* pAgg = (SAggLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_AGG);
  if (NULL == pAgg) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  int32_t code = nodesListMakeStrictAppend(&pAgg->pAggFuncs, nodesCloneNode(pDelete->pCountFunc));
1478 1479 1480 1481 1482 1483
  if (TSDB_CODE_SUCCESS == code) {
    code = nodesListStrictAppend(pAgg->pAggFuncs, nodesCloneNode(pDelete->pFirstFunc));
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = nodesListStrictAppend(pAgg->pAggFuncs, nodesCloneNode(pDelete->pLastFunc));
  }
X
Xiaoyu Wang 已提交
1484 1485 1486
  if (TSDB_CODE_SUCCESS == code) {
    code = rewriteExpr(pAgg->pAggFuncs, &pDelete->pCountFunc);
  }
1487 1488 1489 1490 1491 1492
  if (TSDB_CODE_SUCCESS == code) {
    code = rewriteExpr(pAgg->pAggFuncs, &pDelete->pFirstFunc);
  }
  if (TSDB_CODE_SUCCESS == code) {
    code = rewriteExpr(pAgg->pAggFuncs, &pDelete->pLastFunc);
  }
X
Xiaoyu Wang 已提交
1493 1494
  // set the output
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1495
    code = createColumnByRewriteExprs(pAgg->pAggFuncs, &pAgg->node.pTargets);
X
Xiaoyu Wang 已提交
1496 1497 1498 1499 1500
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = (SLogicNode*)pAgg;
  } else {
1501
    nodesDestroyNode((SNode*)pAgg);
X
Xiaoyu Wang 已提交
1502 1503 1504
  }

  return code;
X
Xiaoyu Wang 已提交
1505 1506
}

X
Xiaoyu Wang 已提交
1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518
static int32_t createVnodeModifLogicNodeByDelete(SLogicPlanContext* pCxt, SDeleteStmt* pDelete,
                                                 SLogicNode** pLogicNode) {
  SVnodeModifyLogicNode* pModify = (SVnodeModifyLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_VNODE_MODIFY);
  if (NULL == pModify) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  SRealTableNode* pRealTable = (SRealTableNode*)pDelete->pFromTable;

  pModify->modifyType = MODIFY_TABLE_TYPE_DELETE;
  pModify->tableId = pRealTable->pMeta->uid;
  pModify->tableType = pRealTable->pMeta->tableType;
1519
  snprintf(pModify->tableName, sizeof(pModify->tableName), "%s", pRealTable->table.tableName);
1520
  strcpy(pModify->tsColName, pRealTable->pMeta->schema->name);
X
Xiaoyu Wang 已提交
1521
  pModify->deleteTimeRange = pDelete->timeRange;
X
Xiaoyu Wang 已提交
1522
  pModify->pAffectedRows = nodesCloneNode(pDelete->pCountFunc);
1523 1524 1525
  pModify->pStartTs = nodesCloneNode(pDelete->pFirstFunc);
  pModify->pEndTs = nodesCloneNode(pDelete->pLastFunc);
  if (NULL == pModify->pAffectedRows || NULL == pModify->pStartTs || NULL == pModify->pEndTs) {
1526
    nodesDestroyNode((SNode*)pModify);
X
Xiaoyu Wang 已提交
1527 1528 1529 1530 1531
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  *pLogicNode = (SLogicNode*)pModify;
  return TSDB_CODE_SUCCESS;
X
Xiaoyu Wang 已提交
1532 1533 1534 1535
}

static int32_t createDeleteLogicNode(SLogicPlanContext* pCxt, SDeleteStmt* pDelete, SLogicNode** pLogicNode) {
  SLogicNode* pRoot = NULL;
X
Xiaoyu Wang 已提交
1536
  int32_t     code = createDeleteScanLogicNode(pCxt, pDelete, &pRoot);
X
Xiaoyu Wang 已提交
1537 1538 1539 1540
  if (TSDB_CODE_SUCCESS == code) {
    code = createDeleteRootLogicNode(pCxt, pDelete, createDeleteAggLogicNode, &pRoot);
  }
  if (TSDB_CODE_SUCCESS == code) {
X
Xiaoyu Wang 已提交
1541
    code = createDeleteRootLogicNode(pCxt, pDelete, createVnodeModifLogicNodeByDelete, &pRoot);
X
Xiaoyu Wang 已提交
1542 1543 1544 1545 1546
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = pRoot;
  } else {
1547
    nodesDestroyNode((SNode*)pRoot);
X
Xiaoyu Wang 已提交
1548 1549 1550 1551 1552
  }

  return code;
}

1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568
static int32_t creatInsertRootLogicNode(SLogicPlanContext* pCxt, SInsertStmt* pInsert, FCreateInsertLogicNode func,
                                        SLogicNode** pRoot) {
  return createRootLogicNode(pCxt, pInsert, pInsert->precision, (FCreateLogicNode)func, pRoot);
}

static int32_t createVnodeModifLogicNodeByInsert(SLogicPlanContext* pCxt, SInsertStmt* pInsert,
                                                 SLogicNode** pLogicNode) {
  SVnodeModifyLogicNode* pModify = (SVnodeModifyLogicNode*)nodesMakeNode(QUERY_NODE_LOGIC_PLAN_VNODE_MODIFY);
  if (NULL == pModify) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  SRealTableNode* pRealTable = (SRealTableNode*)pInsert->pTable;

  pModify->modifyType = MODIFY_TABLE_TYPE_INSERT;
  pModify->tableId = pRealTable->pMeta->uid;
1569
  pModify->stableId = pRealTable->pMeta->suid;
1570
  pModify->tableType = pRealTable->pMeta->tableType;
1571
  snprintf(pModify->tableName, sizeof(pModify->tableName), "%s", pRealTable->table.tableName);
1572
  TSWAP(pModify->pVgroupList, pRealTable->pVgroupList);
1573 1574 1575 1576 1577
  pModify->pInsertCols = nodesCloneList(pInsert->pCols);
  if (NULL == pModify->pInsertCols) {
    nodesDestroyNode((SNode*)pModify);
    return TSDB_CODE_OUT_OF_MEMORY;
  }
1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598

  *pLogicNode = (SLogicNode*)pModify;
  return TSDB_CODE_SUCCESS;
}

static int32_t createInsertLogicNode(SLogicPlanContext* pCxt, SInsertStmt* pInsert, SLogicNode** pLogicNode) {
  SLogicNode* pRoot = NULL;
  int32_t     code = createQueryLogicNode(pCxt, pInsert->pQuery, &pRoot);
  if (TSDB_CODE_SUCCESS == code) {
    code = creatInsertRootLogicNode(pCxt, pInsert, createVnodeModifLogicNodeByInsert, &pRoot);
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicNode = pRoot;
  } else {
    nodesDestroyNode((SNode*)pRoot);
  }

  return code;
}

1599
static int32_t createQueryLogicNode(SLogicPlanContext* pCxt, SNode* pStmt, SLogicNode** pLogicNode) {
X
Xiaoyu Wang 已提交
1600 1601
  switch (nodeType(pStmt)) {
    case QUERY_NODE_SELECT_STMT:
1602
      return createSelectLogicNode(pCxt, (SSelectStmt*)pStmt, pLogicNode);
X
Xiaoyu Wang 已提交
1603 1604
    case QUERY_NODE_VNODE_MODIFY_STMT:
      return createVnodeModifLogicNode(pCxt, (SVnodeModifyOpStmt*)pStmt, pLogicNode);
1605 1606
    case QUERY_NODE_EXPLAIN_STMT:
      return createQueryLogicNode(pCxt, ((SExplainStmt*)pStmt)->pQuery, pLogicNode);
1607 1608
    case QUERY_NODE_SET_OPERATOR:
      return createSetOperatorLogicNode(pCxt, (SSetOperator*)pStmt, pLogicNode);
X
Xiaoyu Wang 已提交
1609 1610
    case QUERY_NODE_DELETE_STMT:
      return createDeleteLogicNode(pCxt, (SDeleteStmt*)pStmt, pLogicNode);
1611 1612
    case QUERY_NODE_INSERT_STMT:
      return createInsertLogicNode(pCxt, (SInsertStmt*)pStmt, pLogicNode);
X
Xiaoyu Wang 已提交
1613 1614 1615
    default:
      break;
  }
1616
  return TSDB_CODE_FAILED;
1617 1618
}

X
Xiaoyu Wang 已提交
1619 1620 1621 1622 1623 1624 1625 1626
static void doSetLogicNodeParent(SLogicNode* pNode, SLogicNode* pParent) {
  pNode->pParent = pParent;
  SNode* pChild;
  FOREACH(pChild, pNode->pChildren) { doSetLogicNodeParent((SLogicNode*)pChild, pNode); }
}

static void setLogicNodeParent(SLogicNode* pNode) { doSetLogicNodeParent(pNode, NULL); }

1627
static void setLogicSubplanType(bool hasScan, SLogicSubplan* pSubplan) {
1628
  if (QUERY_NODE_LOGIC_PLAN_VNODE_MODIFY != nodeType(pSubplan->pNode)) {
1629
    pSubplan->subplanType = hasScan ? SUBPLAN_TYPE_SCAN : SUBPLAN_TYPE_MERGE;
1630 1631 1632 1633 1634 1635 1636 1637
  } else {
    SVnodeModifyLogicNode* pModify = (SVnodeModifyLogicNode*)pSubplan->pNode;
    pSubplan->subplanType = (MODIFY_TABLE_TYPE_INSERT == pModify->modifyType && NULL != pModify->node.pChildren)
                                ? SUBPLAN_TYPE_SCAN
                                : SUBPLAN_TYPE_MODIFY;
  }
}

X
Xiaoyu Wang 已提交
1638
int32_t createLogicPlan(SPlanContext* pCxt, SLogicSubplan** pLogicSubplan) {
1639
  SLogicPlanContext cxt = {.pPlanCxt = pCxt, .pCurrRoot = NULL, .hasScan = false};
X
Xiaoyu Wang 已提交
1640 1641 1642 1643

  SLogicSubplan* pSubplan = (SLogicSubplan*)nodesMakeNode(QUERY_NODE_LOGIC_SUBPLAN);
  if (NULL == pSubplan) {
    return TSDB_CODE_OUT_OF_MEMORY;
X
Xiaoyu Wang 已提交
1644
  }
X
Xiaoyu Wang 已提交
1645 1646 1647 1648 1649 1650 1651
  pSubplan->id.queryId = pCxt->queryId;
  pSubplan->id.groupId = 1;
  pSubplan->id.subplanId = 1;

  int32_t code = createQueryLogicNode(&cxt, pCxt->pAstRoot, &pSubplan->pNode);
  if (TSDB_CODE_SUCCESS == code) {
    setLogicNodeParent(pSubplan->pNode);
1652
    setLogicSubplanType(cxt.hasScan, pSubplan);
1653
    code = adjustLogicNodeDataRequirement(pSubplan->pNode, DATA_ORDER_LEVEL_NONE);
X
Xiaoyu Wang 已提交
1654 1655 1656 1657 1658
  }

  if (TSDB_CODE_SUCCESS == code) {
    *pLogicSubplan = pSubplan;
  } else {
1659
    nodesDestroyNode((SNode*)pSubplan);
X
Xiaoyu Wang 已提交
1660 1661 1662
  }

  return code;
1663
}