• Home
  • Features
  • Pricing
  • Docs
  • Announcements
  • Sign In

taosdata / TDengine / #4069

12 May 2025 05:35AM UTC coverage: 63.048% (+0.5%) from 62.547%
#4069

push

travis-ci

web-flow
Merge pull request #31053 from taosdata/merge/mainto3.0

merge: from main to 3.0 branch

157521 of 317858 branches covered (49.56%)

Branch coverage included in aggregate %.

374 of 573 new or added lines in 31 files covered. (65.27%)

4949 existing lines in 87 files now uncovered.

242707 of 316936 relevant lines covered (76.58%)

18229906.31 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

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

16
#include "filter.h"
17
#include "function.h"
18
#include "os.h"
19
#include "tname.h"
20

21
#include "tglobal.h"
22

23
#include "executorInt.h"
24
#include "index.h"
25
#include "operator.h"
26
#include "query.h"
27
#include "querytask.h"
28

29
#include "storageapi.h"
30
#include "tdatablock.h"
31

32
SOperatorFpSet createOperatorFpSet(__optr_open_fn_t openFn, __optr_fn_t nextFn, __optr_fn_t cleanup,
20,253,478✔
33
                                   __optr_close_fn_t closeFn, __optr_reqBuf_fn_t reqBufFn, __optr_explain_fn_t explain,
34
                                   __optr_get_ext_fn_t nextExtFn, __optr_notify_fn_t notifyFn) {
35
  SOperatorFpSet fpSet = {
20,253,478✔
36
      ._openFn = openFn,
37
      .getNextFn = nextFn,
38
      .cleanupFn = cleanup,
39
      .closeFn = closeFn,
40
      .reqBufFn = reqBufFn,
41
      .getExplainFn = explain,
42
      .getNextExtFn = nextExtFn,
43
      .notifyFn = notifyFn,
44
      .releaseStreamStateFn = NULL,
45
      .reloadStreamStateFn = NULL,
46
  };
47

48
  return fpSet;
20,253,478✔
49
}
50

51
void setOperatorStreamStateFn(SOperatorInfo* pOperator, __optr_state_fn_t relaseFn, __optr_state_fn_t reloadFn) {
1,933,590✔
52
  pOperator->fpSet.releaseStreamStateFn = relaseFn;
1,933,590✔
53
  pOperator->fpSet.reloadStreamStateFn = reloadFn;
1,933,590✔
54
}
1,933,590✔
55

56
int32_t optrDummyOpenFn(SOperatorInfo* pOperator) {
8,804,735✔
57
  OPTR_SET_OPENED(pOperator);
8,804,735✔
58
  pOperator->cost.openCost = 0;
8,804,735✔
59
  return TSDB_CODE_SUCCESS;
8,804,735✔
60
}
61

62
int32_t appendDownstream(SOperatorInfo* p, SOperatorInfo** pDownstream, int32_t num) {
10,079,723✔
63
  p->pDownstream = taosMemoryCalloc(1, num * POINTER_BYTES);
10,079,723!
64
  if (p->pDownstream == NULL) {
10,088,103!
65
    return terrno;
×
66
  }
67

68
  memcpy(p->pDownstream, pDownstream, num * POINTER_BYTES);
10,088,103✔
69
  p->numOfDownstream = num;
10,088,103✔
70
  p->numOfRealDownstream = num;
10,088,103✔
71
  return TSDB_CODE_SUCCESS;
10,088,103✔
72
}
73

74
void setOperatorCompleted(SOperatorInfo* pOperator) {
18,142,209✔
75
  pOperator->status = OP_EXEC_DONE;
18,142,209✔
76
  pOperator->cost.totalCost = (taosGetTimestampUs() - pOperator->pTaskInfo->cost.start) / 1000.0;
18,143,779✔
77
  setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
18,143,779✔
78
}
18,143,716✔
79

80
void setOperatorInfo(SOperatorInfo* pOperator, const char* name, int32_t type, bool blocking, int32_t status,
20,249,891✔
81
                     void* pInfo, SExecTaskInfo* pTaskInfo) {
82
  pOperator->name = (char*)name;
20,249,891✔
83
  pOperator->operatorType = type;
20,249,891✔
84
  pOperator->blocking = blocking;
20,249,891✔
85
  pOperator->status = status;
20,249,891✔
86
  pOperator->info = pInfo;
20,249,891✔
87
  pOperator->pTaskInfo = pTaskInfo;
20,249,891✔
88
}
20,249,891✔
89

90
// each operator should be set their own function to return total cost buffer
91
int32_t optrDefaultBufFn(SOperatorInfo* pOperator) {
×
92
  if (pOperator->blocking) {
×
93
    return -1;
×
94
  } else {
95
    return 0;
×
96
  }
97
}
98

99
static int64_t getQuerySupportBufSize(size_t numOfTables) {
×
100
  size_t s1 = sizeof(STableQueryInfo);
×
101
  //  size_t s3 = sizeof(STableCheckInfo);  buffer consumption in tsdb
102
  return (int64_t)(s1 * 1.5 * numOfTables);
×
103
}
104

105
int32_t checkForQueryBuf(size_t numOfTables) {
×
106
  int64_t t = getQuerySupportBufSize(numOfTables);
×
107
  if (tsQueryBufferSizeBytes < 0) {
×
108
    return TSDB_CODE_SUCCESS;
×
109
  } else if (tsQueryBufferSizeBytes > 0) {
×
110
    while (1) {
×
111
      int64_t s = tsQueryBufferSizeBytes;
×
112
      int64_t remain = s - t;
×
113
      if (remain >= 0) {
×
114
        if (atomic_val_compare_exchange_64(&tsQueryBufferSizeBytes, s, remain) == s) {
×
115
          return TSDB_CODE_SUCCESS;
×
116
        }
117
      } else {
118
        return TSDB_CODE_QRY_NOT_ENOUGH_BUFFER;
×
119
      }
120
    }
121
  }
122

123
  // disable query processing if the value of tsQueryBufferSize is zero.
124
  return TSDB_CODE_QRY_NOT_ENOUGH_BUFFER;
×
125
}
126

127
void releaseQueryBuf(size_t numOfTables) {
×
128
  if (tsQueryBufferSizeBytes < 0) {
×
129
    return;
×
130
  }
131

132
  int64_t t = getQuerySupportBufSize(numOfTables);
×
133

134
  // restore value is not enough buffer available
135
  (void)atomic_add_fetch_64(&tsQueryBufferSizeBytes, t);
×
136
}
137

138
typedef enum {
139
  OPTR_FN_RET_CONTINUE = 0x1,
140
  OPTR_FN_RET_ABORT = 0x2,
141
} ERetType;
142

143
typedef struct STraverParam {
144
  void*   pRet;
145
  int32_t code;
146
  void*   pParam;
147
} STraverParam;
148

149
// iterate the operator tree helper
150
typedef ERetType (*optr_fn_t)(SOperatorInfo* pOperator, STraverParam* pParam, const char* pIdstr);
151

152
void traverseOperatorTree(SOperatorInfo* pOperator, optr_fn_t fn, STraverParam* pParam, const char* id) {
2,508,986✔
153
  if (pOperator == NULL) {
2,508,986!
154
    return;
×
155
  }
156

157
  ERetType ret = fn(pOperator, pParam, id);
2,508,986✔
158
  if (ret == OPTR_FN_RET_ABORT || pParam->code != TSDB_CODE_SUCCESS) {
2,509,038✔
159
    return;
339,538✔
160
  }
161

162
  for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
3,425,519✔
163
    traverseOperatorTree(pOperator->pDownstream[i], fn, pParam, id);
1,255,970✔
164
    if (pParam->code != 0) {
1,256,019!
165
      break;
×
166
    }
167
  }
168
}
169

170
ERetType extractOperatorInfo(SOperatorInfo* pOperator, STraverParam* pParam, const char* pIdStr) {
670,077✔
171
  STraverParam* p = pParam;
670,077✔
172
  if (pOperator->operatorType == *(int32_t*)p->pParam) {
670,077✔
173
    p->pRet = pOperator;
330,905✔
174
    return OPTR_FN_RET_ABORT;
330,905✔
175
  } else {
176
    return OPTR_FN_RET_CONTINUE;
339,172✔
177
  }
178
}
179

180
// QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN
181
int32_t extractOperatorInTree(SOperatorInfo* pOperator, int32_t type, const char* id, SOperatorInfo** pOptrInfo) {
330,912✔
182
  QRY_PARAM_CHECK(pOptrInfo);
330,912!
183

184
  if (pOperator == NULL) {
330,912✔
185
    qError("invalid operator, failed to find tableScanOperator %s", id);
4!
186
    return TSDB_CODE_PAR_INTERNAL_ERROR;
4✔
187
  }
188

189
  STraverParam p = {.pParam = &type, .pRet = NULL};
330,908✔
190
  traverseOperatorTree(pOperator, extractOperatorInfo, &p, id);
330,908✔
191
  if (p.code == 0) {
330,902!
192
    *pOptrInfo = p.pRet;
330,902✔
193
  }
194
  return p.code;
330,902✔
195
}
196

197
typedef struct SExtScanInfo {
198
  int32_t order;
199
  int32_t scanFlag;
200
  int32_t inheritUsOrder;
201
} SExtScanInfo;
202

203
static ERetType extractScanInfo(SOperatorInfo* pOperator, STraverParam* pParam, const char* pIdStr) {
×
204
  int32_t       type = pOperator->operatorType;
×
205
  SExtScanInfo* pInfo = pParam->pParam;
×
206

207
  if (type == QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN || type == QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN ||
×
208
      type == QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN || type == QUERY_NODE_PHYSICAL_PLAN_BLOCK_DIST_SCAN ||
×
209
      type == QUERY_NODE_PHYSICAL_PLAN_LAST_ROW_SCAN || type == QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN) {
×
210
    pInfo->order = TSDB_ORDER_ASC;
×
211
    pInfo->scanFlag = MAIN_SCAN;
×
212
    return OPTR_FN_RET_ABORT;
×
213
  } else if (type == QUERY_NODE_PHYSICAL_PLAN_EXCHANGE) {
×
214
    if (!pInfo->inheritUsOrder) {
×
215
      pInfo->order = TSDB_ORDER_ASC;
×
216
    }
217
    pInfo->scanFlag = MAIN_SCAN;
×
218
    return OPTR_FN_RET_ABORT;
×
219
  } else if (type == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
×
220
    STableScanInfo* pTableScanInfo = pOperator->info;
×
221
    pInfo->order = pTableScanInfo->base.cond.order;
×
222
    pInfo->scanFlag = pTableScanInfo->base.scanFlag;
×
223
    return OPTR_FN_RET_ABORT;
×
224
  } else if (type == QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN) {
×
225
    STableMergeScanInfo* pTableScanInfo = pOperator->info;
×
226
    pInfo->order = pTableScanInfo->base.cond.order;
×
227
    pInfo->scanFlag = pTableScanInfo->base.scanFlag;
×
228
    return OPTR_FN_RET_ABORT;
×
229
  } else {
230
    return OPTR_FN_RET_CONTINUE;
×
231
  }
232
}
233

234
int32_t getTableScanInfo(SOperatorInfo* pOperator, int32_t* order, int32_t* scanFlag, bool inheritUsOrder) {
×
235
  SExtScanInfo info = {.inheritUsOrder = inheritUsOrder, .order = *order};
×
236
  STraverParam p = {.pParam = &info};
×
237

238
  traverseOperatorTree(pOperator, extractScanInfo, &p, NULL);
×
239
  *order = info.order;
×
240
  *scanFlag = info.scanFlag;
×
241

242
  if (p.code == TSDB_CODE_SUCCESS) {
×
243
    if (!(*order == TSDB_ORDER_ASC || *order == TSDB_ORDER_DESC)) {
×
244
      qError("operator failed at: %s:%d", __func__, __LINE__);
×
245
      p.code = TSDB_CODE_INVALID_PARA;
×
246
    }
247
  }
248
  return p.code;
×
249
}
250

251
static ERetType doStopDataReader(SOperatorInfo* pOperator, STraverParam* pParam, const char* pIdStr) {
80,728✔
252
  SStorageAPI* pAPI = pParam->pParam;
80,728✔
253
  if (pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
80,728✔
254
    STableScanInfo* pInfo = pOperator->info;
1,237✔
255

256
    if (pInfo->base.dataReader != NULL) {
1,237✔
257
      pAPI->tsdReader.tsdReaderNotifyClosing(pInfo->base.dataReader);
1,233✔
258
    }
259
    return OPTR_FN_RET_ABORT;
1,237✔
260
  } else if (pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN) {
79,491✔
261
    SStreamScanInfo* pInfo = pOperator->info;
6,346✔
262

263
    if (pInfo->pTableScanOp != NULL) {
6,346✔
264
      STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
6,061✔
265
      if (pTableScanInfo != NULL && pTableScanInfo->base.dataReader != NULL) {
6,061✔
266
        pAPI->tsdReader.tsdReaderNotifyClosing(pTableScanInfo->base.dataReader);
5✔
267
      }
268
    }
269

270
    return OPTR_FN_RET_ABORT;
6,347✔
271
  }
272

273
  return OPTR_FN_RET_CONTINUE;
73,145✔
274
}
275

276
int32_t stopTableScanOperator(SOperatorInfo* pOperator, const char* pIdStr, SStorageAPI* pAPI) {
38,507✔
277
  STraverParam p = {.pParam = pAPI};
38,507✔
278
  traverseOperatorTree(pOperator, doStopDataReader, &p, pIdStr);
38,507✔
279
  return p.code;
38,521✔
280
}
281

282
static ERetType doClearTbNameHashPtr(SOperatorInfo* pOperator, STraverParam* pParam, const char* pIdStr) {
1,758,189✔
283
  SStorageAPI* pAPI = pParam->pParam;
1,758,189✔
284
  if (pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_STREAM_FILL) {
1,758,189✔
285
    SStreamFillOperatorInfo* pInfo = pOperator->info;
1,065✔
286
    if (pInfo->pState != NULL) {
1,065✔
287
      pInfo->pState->parNameMap = NULL;
120✔
288
    }
289

290
    qDebug("%s clear the parNameMap for fill operator", pIdStr);
1,065!
291
    return OPTR_FN_RET_ABORT;
1,065✔
292
  }
293

294
  return OPTR_FN_RET_CONTINUE;
1,757,124✔
295
}
296

297
int32_t clearParTbNameHashPtr(SOperatorInfo* pOperator, const char* pIdStr, SStorageAPI* pAPI) {
883,671✔
298
  STraverParam p = {.pParam = pAPI};
883,671✔
299
  traverseOperatorTree(pOperator, doClearTbNameHashPtr, &p, pIdStr);
883,671✔
300
  return p.code;
883,762✔
301
}
302

303
int32_t createOperator(SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo, SReadHandle* pHandle, SNode* pTagCond,
20,245,678✔
304
                       SNode* pTagIndexCond, const char* pUser, const char* dbname, SOperatorInfo** pOptrInfo,
305
                       EOPTR_EXEC_MODEL model) {
306
  QRY_PARAM_CHECK(pOptrInfo);
20,245,678!
307

308
  int32_t     code = 0;
20,245,678✔
309
  int32_t     type = nodeType(pPhyNode);
20,245,678✔
310
  const char* idstr = GET_TASKID(pTaskInfo);
20,245,678✔
311

312
  if (pPhyNode->pChildren == NULL || LIST_LENGTH(pPhyNode->pChildren) == 0) {
20,245,678!
313
    SOperatorInfo* pOperator = NULL;
10,160,372✔
314
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == type) {
10,160,372✔
315
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pPhyNode;
3,087,967✔
316
      // NOTE: this is an patch to fix the physical plan
317
      // TODO remove it later
318
      if (pTableScanNode->scan.node.pLimit != NULL) {
3,087,967✔
319
        pTableScanNode->groupSort = true;
128,305✔
320
      }
321

322
      STableListInfo* pTableListInfo = tableListCreate();
3,087,967✔
323
      if (!pTableListInfo) {
3,088,673!
324
        pTaskInfo->code = terrno;
×
325
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
326
        return terrno;
×
327
      }
328

329
      // Since virtual stable scan use virtual super table's uid in scan operator, the origin table might be stored on
330
      // different vnode, so we should not get table schema for virtual stable scan.
331
      if (!pTableScanNode->scan.virtualStableScan) {
3,088,673✔
332
        code = initQueriedTableSchemaInfo(pHandle, &pTableScanNode->scan, dbname, pTaskInfo);
3,088,399✔
333
        if (code) {
3,084,318✔
334
          pTaskInfo->code = code;
1,447✔
335
          tableListDestroy(pTableListInfo);
1,447✔
336
          return code;
120✔
337
        }
338
      }
339

340
      if (pTableScanNode->scan.node.dynamicOp) {
3,083,145✔
341
        pTaskInfo->dynamicTask = true;
44,566✔
342
        pTableListInfo->idInfo.suid = pTableScanNode->scan.suid;
44,566✔
343
        pTableListInfo->idInfo.tableType = pTableScanNode->scan.tableType;
44,566✔
344
      } else {
345
        code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
3,038,579✔
346
                                       pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo);
347
        if (code) {
3,045,241✔
348
          pTaskInfo->code = code;
24✔
349
          tableListDestroy(pTableListInfo);
24✔
350
          qError("failed to createScanTableListInfo, code:%s, %s", tstrerror(code), idstr);
24!
351
          return code;
24✔
352
        }
353
      }
354

355
      code = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
3,089,783✔
356
      if (NULL == pOperator || code != 0) {
3,085,704!
357
        pTaskInfo->code = code;
×
358
        tableListDestroy(pTableListInfo);
×
359
        return code;
222✔
360
      }
361

362
      STableScanInfo* pScanInfo = pOperator->info;
3,087,540✔
363
      pTaskInfo->cost.pRecoder = &pScanInfo->base.readRecorder;
3,087,540✔
364
    } else if (QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN == type) {
7,072,405✔
365
      STableMergeScanPhysiNode* pTableScanNode = (STableMergeScanPhysiNode*)pPhyNode;
2,013,477✔
366
      STableListInfo*           pTableListInfo = tableListCreate();
2,013,477✔
367
      if (!pTableListInfo) {
2,016,410!
368
        pTaskInfo->code = terrno;
×
369
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
370
        return terrno;
×
371
      }
372

373
      code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, true, pHandle, pTableListInfo,
2,016,410✔
374
                                     pTagCond, pTagIndexCond, pTaskInfo);
375
      if (code) {
2,016,459!
376
        pTaskInfo->code = code;
×
377
        tableListDestroy(pTableListInfo);
×
378
        qError("failed to createScanTableListInfo, code:%s", tstrerror(code));
×
379
        return code;
×
380
      }
381

382
      code = initQueriedTableSchemaInfo(pHandle, &pTableScanNode->scan, dbname, pTaskInfo);
2,016,459✔
383
      if (code) {
2,012,676!
384
        pTaskInfo->code = code;
×
385
        tableListDestroy(pTableListInfo);
×
386
        return code;
×
387
      }
388

389
      code = createTableMergeScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
2,012,676✔
390
      if (NULL == pOperator || code != 0) {
2,013,262!
391
        pTaskInfo->code = code;
×
392
        tableListDestroy(pTableListInfo);
×
393
        return code;
×
394
      }
395

396
      STableScanInfo* pScanInfo = pOperator->info;
2,015,928✔
397
      pTaskInfo->cost.pRecoder = &pScanInfo->base.readRecorder;
2,015,928✔
398
    } else if (QUERY_NODE_PHYSICAL_PLAN_EXCHANGE == type) {
5,058,928✔
399
      code = createExchangeOperatorInfo(pHandle ? pHandle->pMsgCb->clientRpc : NULL, (SExchangePhysiNode*)pPhyNode,
3,743,574!
400
                                        pTaskInfo, &pOperator);
401
    } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN == type) {
1,315,354✔
402
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pPhyNode;
7,623✔
403
      STableListInfo*      pTableListInfo = tableListCreate();
7,623✔
404
      if (!pTableListInfo) {
7,629!
405
        pTaskInfo->code = terrno;
×
406
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
407
        return terrno;
×
408
      }
409

410
      if (pHandle->vnode && (pTaskInfo->pSubplan->pVTables == NULL)) {
7,629!
411
        code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
7,343✔
412
                                       pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo);
413
        if (code) {
7,345!
414
          pTaskInfo->code = code;
×
415
          tableListDestroy(pTableListInfo);
×
416
          qError("failed to createScanTableListInfo, code:%s", tstrerror(code));
×
417
          return code;
×
418
        }
419
      }
420

421
      code = createStreamScanOperatorInfo(pHandle, pTableScanNode, pTagCond, pTableListInfo, pTaskInfo, &pOperator);
7,631✔
422
      if (code) {
7,623!
423
        pTaskInfo->code = code;
×
424
        tableListDestroy(pTableListInfo);
×
425
        return code;
×
426
      }
427
    } else if (QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN == type) {
1,307,731✔
428
      SSystemTableScanPhysiNode* pSysScanPhyNode = (SSystemTableScanPhysiNode*)pPhyNode;
1,252,540✔
429
      code = createSysTableScanOperatorInfo(pHandle, pSysScanPhyNode, pUser, pTaskInfo, &pOperator);
1,252,540✔
430
    } else if (QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN == type) {
55,191!
431
      STableCountScanPhysiNode* pTblCountScanNode = (STableCountScanPhysiNode*)pPhyNode;
×
432
      code = createTableCountScanOperatorInfo(pHandle, pTblCountScanNode, pTaskInfo, &pOperator);
×
433
    } else if (QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN == type) {
55,191✔
434
      STagScanPhysiNode* pTagScanPhyNode = (STagScanPhysiNode*)pPhyNode;
47,955✔
435
      STableListInfo*    pTableListInfo = tableListCreate();
47,955✔
436
      if (!pTableListInfo) {
48,021!
437
        pTaskInfo->code = terrno;
×
438
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
439
        return terrno;
×
440
      }
441
      if (!pTagScanPhyNode->onlyMetaCtbIdx) {
48,021✔
442
        code = createScanTableListInfo((SScanPhysiNode*)pTagScanPhyNode, NULL, false, pHandle, pTableListInfo, pTagCond,
3,569✔
443
                                       pTagIndexCond, pTaskInfo);
444
        if (code != TSDB_CODE_SUCCESS) {
3,570!
445
          pTaskInfo->code = code;
×
446
          qError("failed to getTableList, code:%s", tstrerror(code));
×
447
          tableListDestroy(pTableListInfo);
×
448
          return code;
×
449
        }
450
      }
451
      code = createTagScanOperatorInfo(pHandle, pTagScanPhyNode, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo,
48,022✔
452
                                       &pOperator);
453
      if (code) {
48,021!
454
        pTaskInfo->code = code;
×
455
        tableListDestroy(pTableListInfo);
×
456
        return code;
×
457
      }
458
    } else if (QUERY_NODE_PHYSICAL_PLAN_BLOCK_DIST_SCAN == type) {
7,236✔
459
      SBlockDistScanPhysiNode* pBlockNode = (SBlockDistScanPhysiNode*)pPhyNode;
3,250✔
460
      STableListInfo*          pTableListInfo = tableListCreate();
3,250✔
461
      if (!pTableListInfo) {
3,255✔
462
        pTaskInfo->code = terrno;
2✔
463
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
464
        return terrno;
×
465
      }
466

467
      if (pBlockNode->tableType == TSDB_SUPER_TABLE) {
3,253✔
468
        SArray* pList = taosArrayInit(4, sizeof(uint64_t));
3,249✔
469
        code = pTaskInfo->storageAPI.metaFn.getChildTableList(pHandle->vnode, pBlockNode->uid, pList);
3,248✔
470
        if (code != TSDB_CODE_SUCCESS) {
3,250!
471
          pTaskInfo->code = code;
×
472
          taosArrayDestroy(pList);
×
473
          tableListDestroy(pTableListInfo);
×
474
          return code;
×
475
        }
476

477
        size_t num = taosArrayGetSize(pList);
3,250✔
478
        for (int32_t i = 0; i < num; ++i) {
10,276✔
479
          uint64_t* id = taosArrayGet(pList, i);
7,026✔
480
          if (id == NULL) {
7,011!
481
            continue;
×
482
          }
483

484
          code = tableListAddTableInfo(pTableListInfo, *id, 0);
7,011✔
485
          if (code) {
7,031!
486
            pTaskInfo->code = code;
×
487
            tableListDestroy(pTableListInfo);
×
488
            taosArrayDestroy(pList);
×
489
            return code;
11,878✔
490
          }
491
        }
492

493
        taosArrayDestroy(pList);
3,250✔
494
      } else {  // Create group with only one table
495
        code = tableListAddTableInfo(pTableListInfo, pBlockNode->uid, 0);
4✔
496
        if (code) {
4!
497
          pTaskInfo->code = code;
×
498
          tableListDestroy(pTableListInfo);
×
499
          return code;
×
500
        }
501
      }
502

503
      code = createDataBlockInfoScanOperator(pHandle, pBlockNode, pTableListInfo, pTaskInfo, &pOperator);
3,255✔
504
      if (code) {
3,244!
505
        pTaskInfo->code = code;
×
506
        tableListDestroy(pTableListInfo);
×
507
        return code;
×
508
      }
509
    } else if (QUERY_NODE_PHYSICAL_PLAN_LAST_ROW_SCAN == type) {
3,986!
510
      SLastRowScanPhysiNode* pScanNode = (SLastRowScanPhysiNode*)pPhyNode;
11,102✔
511
      STableListInfo*        pTableListInfo = tableListCreate();
11,102✔
512
      if (!pTableListInfo) {
11,119!
513
        pTaskInfo->code = terrno;
×
514
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
515
        return terrno;
×
516
      }
517

518
      code = createScanTableListInfo(&pScanNode->scan, pScanNode->pGroupTags, true, pHandle, pTableListInfo, pTagCond,
11,119✔
519
                                     pTagIndexCond, pTaskInfo);
520
      if (code != TSDB_CODE_SUCCESS) {
11,115!
521
        pTaskInfo->code = code;
×
522
        tableListDestroy(pTableListInfo);
×
523
        return code;
×
524
      }
525

526
      code = initQueriedTableSchemaInfo(pHandle, &pScanNode->scan, dbname, pTaskInfo);
11,115✔
527
      if (code != TSDB_CODE_SUCCESS) {
11,108!
528
        pTaskInfo->code = code;
×
529
        tableListDestroy(pTableListInfo);
×
530
        return code;
×
531
      }
532

533
      code = createCacherowsScanOperator(pScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
11,108✔
534
      if (code) {
11,109!
535
        tableListDestroy(pTableListInfo);
×
536
        pTaskInfo->code = code;
×
537
        return code;
×
538
      }
539
    } else if (QUERY_NODE_PHYSICAL_PLAN_PROJECT == type) {
×
540
      code = createProjectOperatorInfo(NULL, (SProjectPhysiNode*)pPhyNode, pTaskInfo, &pOperator);
1,677✔
541
    } else if (QUERY_NODE_PHYSICAL_PLAN_VIRTUAL_TABLE_SCAN == type && model != OPTR_EXEC_MODEL_STREAM) {
×
NEW
542
      SVirtualScanPhysiNode* pVirtualTableScanNode = (SVirtualScanPhysiNode*)pPhyNode;
×
543
      // NOTE: this is an patch to fix the physical plan
544

NEW
545
      code = initQueriedTableSchemaInfo(pHandle, &pVirtualTableScanNode->scan, dbname, pTaskInfo);
×
NEW
546
      if (code) {
×
NEW
547
        pTaskInfo->code = code;
×
NEW
548
        return code;
×
549
      }
550

UNCOV
551
      code = createVirtualTableMergeOperatorInfo(NULL, pHandle, NULL, 0, (SVirtualScanPhysiNode*)pPhyNode, pTaskInfo, &pOperator);
×
552
    } else {
UNCOV
553
      code = TSDB_CODE_INVALID_PARA;
×
554
      pTaskInfo->code = code;
×
555
      return code;
×
556
    }
557

558
    if (pOperator != NULL) {  // todo moved away
10,172,330✔
559
      pOperator->resultDataBlockId = pPhyNode->pOutputDataBlockDesc->dataBlockId;
10,167,336✔
560
    }
561

562
    *pOptrInfo = pOperator;
10,172,330✔
563
    return code;
10,172,330✔
564
  }
565

566
  size_t          size = LIST_LENGTH(pPhyNode->pChildren);
10,085,306✔
567
  SOperatorInfo** ops = taosMemoryCalloc(size, POINTER_BYTES);
10,085,306!
568
  if (ops == NULL) {
10,087,894!
UNCOV
569
    code = terrno;
×
UNCOV
570
    pTaskInfo->code = code;
×
UNCOV
571
    return code;
×
572
  }
573

574
  for (int32_t i = 0; i < size; ++i) {
21,750,438✔
575
    SPhysiNode* pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, i);
11,657,963✔
576
    code = createOperator(pChildNode, pTaskInfo, pHandle, pTagCond, pTagIndexCond, pUser, dbname, &ops[i], model);
11,659,456✔
577
    if (ops[i] == NULL || code != 0) {
11,662,666!
578
      for (int32_t j = 0; j < i; ++j) {
122!
579
        destroyOperator(ops[j]);
×
580
      }
581
      taosMemoryFree(ops);
122!
582
      return code;
122✔
583
    }
584
  }
585

586
  SOperatorInfo* pOptr = NULL;
10,092,475✔
587
  if (QUERY_NODE_PHYSICAL_PLAN_PROJECT == type) {
10,092,475✔
588
    code = createProjectOperatorInfo(ops[0], (SProjectPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
1,918,696✔
589
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_AGG == type) {
8,173,779✔
590
    SAggPhysiNode* pAggNode = (SAggPhysiNode*)pPhyNode;
2,564,142✔
591
    if (pAggNode->pGroupKeys != NULL) {
2,564,142✔
592
      code = createGroupOperatorInfo(ops[0], pAggNode, pTaskInfo, &pOptr);
417,166✔
593
    } else {
594
      code = createAggregateOperatorInfo(ops[0], pAggNode, pTaskInfo, &pOptr);
2,146,976✔
595
    }
596
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_INTERVAL == type) {
5,609,637✔
597
    SIntervalPhysiNode* pIntervalPhyNode = (SIntervalPhysiNode*)pPhyNode;
1,458,088✔
598
    code = createIntervalOperatorInfo(ops[0], pIntervalPhyNode, pTaskInfo, &pOptr);
1,458,088✔
599
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL == type) {
4,151,549✔
600
    code = createStreamSingleIntervalOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
1,957✔
601
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_INTERVAL == type) {
4,149,592✔
602
    code = createStreamIntervalSliceOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
79✔
603
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_ALIGNED_INTERVAL == type) {
4,149,513✔
604
    SMergeAlignedIntervalPhysiNode* pIntervalPhyNode = (SMergeAlignedIntervalPhysiNode*)pPhyNode;
537,772✔
605
    code = createMergeAlignedIntervalOperatorInfo(ops[0], pIntervalPhyNode, pTaskInfo, &pOptr);
537,772✔
606
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_INTERVAL == type) {
3,611,741!
UNCOV
607
    SMergeIntervalPhysiNode* pIntervalPhyNode = (SMergeIntervalPhysiNode*)pPhyNode;
×
UNCOV
608
    code = createMergeIntervalOperatorInfo(ops[0], pIntervalPhyNode, pTaskInfo, &pOptr);
×
609
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_INTERVAL == type) {
3,611,741✔
610
    int32_t children = 0;
676✔
611
    code = createStreamFinalIntervalOperatorInfo(ops[0], pPhyNode, pTaskInfo, children, pHandle, &pOptr);
676✔
612
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_SEMI_INTERVAL == type) {
3,611,065!
UNCOV
613
    code = createSemiIntervalSliceOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
614
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_MID_INTERVAL == type) {
3,611,065✔
615
    int32_t children = pHandle->numOfVgroups;
22✔
616
    code = createStreamFinalIntervalOperatorInfo(ops[0], pPhyNode, pTaskInfo, children, pHandle, &pOptr);
22✔
617
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_INTERVAL == type) {
3,611,043✔
618
    int32_t children = pHandle->numOfVgroups;
226✔
619
    code = createStreamFinalIntervalOperatorInfo(ops[0], pPhyNode, pTaskInfo, children, pHandle, &pOptr);
226✔
620
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_FINAL_INTERVAL == type) {
3,610,817!
UNCOV
621
    code = createFinalIntervalSliceOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
622
  } else if (QUERY_NODE_PHYSICAL_PLAN_SORT == type) {
3,610,817✔
623
    code = createSortOperatorInfo(ops[0], (SSortPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
921,763✔
624
  } else if (QUERY_NODE_PHYSICAL_PLAN_GROUP_SORT == type) {
2,689,054✔
625
    code = createGroupSortOperatorInfo(ops[0], (SGroupSortPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
205✔
626
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE == type) {
2,688,849✔
627
    SMergePhysiNode* pMergePhyNode = (SMergePhysiNode*)pPhyNode;
1,378,123✔
628
    code = createMultiwayMergeOperatorInfo(ops, size, pMergePhyNode, pTaskInfo, &pOptr);
1,378,123✔
629
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_SESSION == type) {
1,310,726✔
630
    SSessionWinodwPhysiNode* pSessionNode = (SSessionWinodwPhysiNode*)pPhyNode;
100,634✔
631
    code = createSessionAggOperatorInfo(ops[0], pSessionNode, pTaskInfo, &pOptr);
100,634✔
632
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SESSION == type) {
1,210,092✔
633
    code = createStreamSessionAggOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
618✔
634
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_SESSION == type) {
1,209,474✔
635
    int32_t children = 0;
125✔
636
    code = createStreamFinalSessionAggOperatorInfo(ops[0], pPhyNode, pTaskInfo, children, pHandle, &pOptr);
125✔
637
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_SESSION == type) {
1,209,349✔
638
    int32_t children = pHandle->numOfVgroups;
38✔
639
    code = createStreamFinalSessionAggOperatorInfo(ops[0], pPhyNode, pTaskInfo, children, pHandle, &pOptr);
38✔
640
  } else if (QUERY_NODE_PHYSICAL_PLAN_PARTITION == type) {
1,209,311✔
641
    code = createPartitionOperatorInfo(ops[0], (SPartitionPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
29,284✔
642
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_PARTITION == type) {
1,180,027✔
643
    code = createStreamPartitionOperatorInfo(ops[0], (SStreamPartitionPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
1,208✔
644
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_STATE == type) {
1,178,819✔
645
    SStateWinodwPhysiNode* pStateNode = (SStateWinodwPhysiNode*)pPhyNode;
72,848✔
646
    code = createStatewindowOperatorInfo(ops[0], pStateNode, pTaskInfo, &pOptr);
72,848✔
647
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_STATE == type) {
1,105,971✔
648
    code = createStreamStateAggOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
352✔
649
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_EVENT == type) {
1,105,619✔
650
    code = createStreamEventAggOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
217✔
651
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_JOIN == type) {
1,105,402✔
652
    code = createMergeJoinOperatorInfo(ops, size, (SSortMergeJoinPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
103,058✔
653
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_JOIN == type) {
1,002,344✔
654
    code = createHashJoinOperatorInfo(ops, size, (SHashJoinPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
11,126✔
655
  } else if (QUERY_NODE_PHYSICAL_PLAN_FILL == type) {
991,218✔
656
    code = createFillOperatorInfo(ops[0], (SFillPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
483,468✔
657
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_FILL == type) {
507,750✔
658
    code = createStreamFillOperatorInfo(ops[0], (SStreamFillPhysiNode*)pPhyNode, pTaskInfo, pHandle, &pOptr);
451✔
659
  } else if (QUERY_NODE_PHYSICAL_PLAN_INDEF_ROWS_FUNC == type) {
507,299✔
660
    code = createIndefinitOutputOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
122,496✔
661
  } else if (QUERY_NODE_PHYSICAL_PLAN_INTERP_FUNC == type) {
384,803✔
662
    code = createTimeSliceOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
76,808✔
663
  } else if (QUERY_NODE_PHYSICAL_PLAN_FORECAST_FUNC == type) {
307,995✔
664
    code = createForecastOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
23✔
665
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_EVENT == type) {
307,972✔
666
    code = createEventwindowOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
271,364✔
667
  } else if (QUERY_NODE_PHYSICAL_PLAN_GROUP_CACHE == type) {
36,608✔
668
    code = createGroupCacheOperatorInfo(ops, size, (SGroupCachePhysiNode*)pPhyNode, pTaskInfo, &pOptr);
11,126✔
669
  } else if (QUERY_NODE_PHYSICAL_PLAN_DYN_QUERY_CTRL == type) {
25,482✔
670
    code = createDynQueryCtrlOperatorInfo(ops, size, (SDynQueryCtrlPhysiNode*)pPhyNode, pTaskInfo, pHandle, &pOptr);
11,126✔
671
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_COUNT == type) {
14,356✔
672
    code = createStreamCountAggOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
315✔
673
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_COUNT == type) {
14,041✔
674
    code = createCountwindowOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
7,068✔
675
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERP_FUNC == type) {
6,973✔
676
    code = createStreamTimeSliceOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
242✔
677
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_ANOMALY == type) {
6,731✔
678
    code = createAnomalywindowOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
12✔
679
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_SESSION == type) {
6,719!
UNCOV
680
    code = createSessionNonblockOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
681
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_SEMI_SESSION == type) {
6,719!
UNCOV
682
    code = createSemiSessionNonblockOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
683
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_FINAL_SESSION == type) {
6,719!
UNCOV
684
    code = createFinalSessionNonblockOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
685
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_STATE == type) {
6,719!
UNCOV
686
    code = createStateNonblockOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
687
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_EVENT == type) {
6,719!
UNCOV
688
    code = createEventNonblockOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
689
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_COUNT == type) {
6,719!
690
    //todo (liuyao) add
691
  } else if (QUERY_NODE_PHYSICAL_PLAN_VIRTUAL_TABLE_SCAN == type && model != OPTR_EXEC_MODEL_STREAM) {
×
UNCOV
692
    SVirtualScanPhysiNode* pVirtualTableScanNode = (SVirtualScanPhysiNode*)pPhyNode;
×
693
    // NOTE: this is an patch to fix the physical plan
694

695
    if (pVirtualTableScanNode->scan.node.pLimit != NULL) {
×
UNCOV
696
      pVirtualTableScanNode->groupSort = true;
×
697
    }
698

NEW
699
    code = initQueriedTableSchemaInfo(pHandle, &pVirtualTableScanNode->scan, dbname, pTaskInfo);
×
NEW
700
    if (code) {
×
NEW
701
      pTaskInfo->code = code;
×
NEW
702
      return code;
×
703
    }
704

705
    STableListInfo* pTableListInfo = tableListCreate();
×
706
    if (!pTableListInfo) {
×
707
      pTaskInfo->code = terrno;
×
UNCOV
708
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
UNCOV
709
      return terrno;
×
710
    }
711

UNCOV
712
    code = createScanTableListInfo(&pVirtualTableScanNode->scan, pVirtualTableScanNode->pGroupTags, pVirtualTableScanNode->groupSort,
×
713
                                   pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo);
714
    if (code) {
×
715
      pTaskInfo->code = code;
×
716
      tableListDestroy(pTableListInfo);
×
717
      qError("failed to createScanTableListInfo, code:%s, %s", tstrerror(code), idstr);
×
UNCOV
718
      return code;
×
719
    }
720

721
    code = createVirtualTableMergeOperatorInfo(ops, pHandle, pTableListInfo, size, (SVirtualScanPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
×
722
  } else if (QUERY_NODE_PHYSICAL_PLAN_VIRTUAL_TABLE_SCAN == type && model == OPTR_EXEC_MODEL_STREAM) {
×
723
    SVirtualScanPhysiNode* pVirtualTableScanNode = (SVirtualScanPhysiNode*)pPhyNode;
×
724
    STableListInfo*        pTableListInfo = tableListCreate();
×
UNCOV
725
    if (!pTableListInfo) {
×
UNCOV
726
      pTaskInfo->code = terrno;
×
727
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
UNCOV
728
      return terrno;
×
729
    }
730

731
    code = createScanTableListInfo(&pVirtualTableScanNode->scan, pVirtualTableScanNode->pGroupTags,
×
732
                                   pVirtualTableScanNode->groupSort, pHandle, pTableListInfo, pTagCond, pTagIndexCond,
×
733
                                   pTaskInfo);
UNCOV
734
    if (code) {
×
UNCOV
735
      pTaskInfo->code = code;
×
736
      tableListDestroy(pTableListInfo);
×
737
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
738
      return code;
×
739
    }
740

741
    code = createStreamVtableMergeOperatorInfo(ops[0], pHandle, pVirtualTableScanNode, pTagCond, pTableListInfo, pTaskInfo, &pOptr);
×
742
  } else {
743
    code = TSDB_CODE_INVALID_PARA;
×
UNCOV
744
    pTaskInfo->code = code;
×
UNCOV
745
    for (int32_t i = 0; i < size; ++i) {
×
746
      destroyOperator(ops[i]);
×
747
    }
UNCOV
748
    taosMemoryFree(ops);
×
749
    qError("invalid operator type %d", type);
×
750
    return code;
×
751
  }
752

753
  taosMemoryFree(ops);
10,088,876✔
754
  if (pOptr) {
10,084,522✔
755
    pOptr->resultDataBlockId = pPhyNode->pOutputDataBlockDesc->dataBlockId;
10,083,495✔
756
  }
757

758
  *pOptrInfo = pOptr;
10,084,522✔
759
  return code;
10,084,522✔
760
}
761

762
void destroyOperator(SOperatorInfo* pOperator) {
20,275,360✔
763
  if (pOperator == NULL) {
20,275,360✔
764
    return;
368✔
765
  }
766

767
  freeResetOperatorParams(pOperator, OP_GET_PARAM, true);
20,274,992✔
768
  freeResetOperatorParams(pOperator, OP_NOTIFY_PARAM, true);
20,275,818✔
769

770
  if (pOperator->pDownstream != NULL) {
20,277,229✔
771
    for (int32_t i = 0; i < pOperator->numOfRealDownstream; ++i) {
21,756,153✔
772
      destroyOperator(pOperator->pDownstream[i]);
11,665,396✔
773
    }
774

775
    taosMemoryFreeClear(pOperator->pDownstream);
10,090,757!
776
    pOperator->numOfDownstream = 0;
10,090,596✔
777
  }
778

779
  if (pOperator->fpSet.closeFn != NULL && pOperator->info != NULL) {
20,278,162!
780
    pOperator->fpSet.closeFn(pOperator->info);
20,276,652✔
781
  }
782

783
  cleanupExprSupp(&pOperator->exprSupp);
20,279,208✔
784
  taosMemoryFreeClear(pOperator);
20,276,225!
785
}
786

787
void destroyOperatorAndDownstreams(SOperatorInfo* pOperator, SOperatorInfo** downstreams, int32_t num) {
2✔
788
  if (downstreams != NULL) {
2!
789
    for (int i = 0; i < num; i++) {
4✔
790
      destroyOperator(downstreams[i]);
2✔
791
    }
792
  }
793

794
  if (pOperator != NULL) {
2!
795
    pOperator->info = NULL;
2✔
796
    if (pOperator->pDownstream != NULL) {
2!
UNCOV
797
      taosMemoryFreeClear(pOperator->pDownstream);
×
UNCOV
798
      pOperator->pDownstream = NULL;
×
799
    }
800
    destroyOperator(pOperator);
2✔
801
  }
802
}
2✔
803

804
int32_t getOperatorExplainExecInfo(SOperatorInfo* operatorInfo, SArray* pExecInfoList) {
1,354,895✔
805
  SExplainExecInfo  execInfo = {0};
1,354,895✔
806
  SExplainExecInfo* pExplainInfo = taosArrayPush(pExecInfoList, &execInfo);
1,354,819✔
807
  if (pExplainInfo == NULL) {
1,354,819!
UNCOV
808
    return terrno;
×
809
  }
810

811
  pExplainInfo->numOfRows = operatorInfo->resultInfo.totalRows;
1,354,819✔
812
  pExplainInfo->startupCost = operatorInfo->cost.openCost;
1,354,819✔
813
  pExplainInfo->totalCost = operatorInfo->cost.totalCost;
1,354,819✔
814
  pExplainInfo->verboseLen = 0;
1,354,819✔
815
  pExplainInfo->verboseInfo = NULL;
1,354,819✔
816

817
  if (operatorInfo->fpSet.getExplainFn) {
1,354,819✔
818
    int32_t code =
819
        operatorInfo->fpSet.getExplainFn(operatorInfo, &pExplainInfo->verboseInfo, &pExplainInfo->verboseLen);
594,477✔
820
    if (code) {
594,535✔
821
      qError("%s operator getExplainFn failed, code:%s", GET_TASKID(operatorInfo->pTaskInfo), tstrerror(code));
24!
UNCOV
822
      return code;
×
823
    }
824
  }
825

826
  int32_t code = 0;
1,354,853✔
827
  for (int32_t i = 0; i < operatorInfo->numOfDownstream; ++i) {
2,133,931✔
828
    code = getOperatorExplainExecInfo(operatorInfo->pDownstream[i], pExecInfoList);
779,059✔
829
    if (code != TSDB_CODE_SUCCESS) {
779,078!
830
      //      taosMemoryFreeClear(*pRes);
UNCOV
831
      return code;
×
832
    }
833
  }
834

835
  return TSDB_CODE_SUCCESS;
1,354,872✔
836
}
837

UNCOV
838
int32_t mergeOperatorParams(SOperatorParam* pDst, SOperatorParam* pSrc) {
×
UNCOV
839
  if (pDst->opType != pSrc->opType) {
×
UNCOV
840
    qError("different optype %d:%d for merge operator params", pDst->opType, pSrc->opType);
×
UNCOV
841
    return TSDB_CODE_INVALID_PARA;
×
842
  }
843

UNCOV
844
  switch (pDst->opType) {
×
UNCOV
845
    case QUERY_NODE_PHYSICAL_PLAN_EXCHANGE: {
×
846
      SExchangeOperatorParam* pDExc = pDst->value;
×
UNCOV
847
      SExchangeOperatorParam* pSExc = pSrc->value;
×
UNCOV
848
      if (!pDExc->multiParams) {
×
UNCOV
849
        if (pSExc->basic.vgId != pDExc->basic.vgId) {
×
UNCOV
850
          SExchangeOperatorBatchParam* pBatch = taosMemoryMalloc(sizeof(SExchangeOperatorBatchParam));
×
UNCOV
851
          if (NULL == pBatch) {
×
UNCOV
852
            return terrno;
×
853
          }
854

855
          pBatch->multiParams = true;
×
856
          pBatch->pBatchs = tSimpleHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
×
UNCOV
857
          if (NULL == pBatch->pBatchs) {
×
UNCOV
858
            taosMemoryFree(pBatch);
×
859
            return terrno;
×
860
          }
861

862
          tSimpleHashSetFreeFp(pBatch->pBatchs, freeExchangeGetBasicOperatorParam);
×
863

864
          int32_t code = tSimpleHashPut(pBatch->pBatchs, &pDExc->basic.vgId, sizeof(pDExc->basic.vgId), &pDExc->basic,
×
865
                                        sizeof(pDExc->basic));
866
          if (code) {
×
867
            return code;
×
868
          }
869

870
          code = tSimpleHashPut(pBatch->pBatchs, &pSExc->basic.vgId, sizeof(pSExc->basic.vgId), &pSExc->basic,
×
871
                                sizeof(pSExc->basic));
872
          if (code) {
×
873
            return code;
×
874
          }
875

UNCOV
876
          taosMemoryFree(pDst->value);
×
877
          pDst->value = pBatch;
×
878
        } else {
879
          void* p = taosArrayAddAll(pDExc->basic.uidList, pSExc->basic.uidList);
×
UNCOV
880
          if (p == NULL) {
×
881
            return terrno;
×
882
          }
883
        }
884
      } else {
885
        SExchangeOperatorBatchParam* pBatch = pDst->value;
×
886
        SExchangeOperatorBasicParam* pBasic =
887
            tSimpleHashGet(pBatch->pBatchs, &pSExc->basic.vgId, sizeof(pSExc->basic.vgId));
×
888
        if (pBasic) {
×
UNCOV
889
          void* p = taosArrayAddAll(pBasic->uidList, pSExc->basic.uidList);
×
UNCOV
890
          if (p == NULL) {
×
891
            return terrno;
×
892
          }
893
        } else {
894
          int32_t code = tSimpleHashPut(pBatch->pBatchs, &pSExc->basic.vgId, sizeof(pSExc->basic.vgId), &pSExc->basic,
×
895
                                        sizeof(pSExc->basic));
896
          if (code) {
×
UNCOV
897
            return code;
×
898
          }
899
        }
900
      }
UNCOV
901
      break;
×
902
    }
903
    default:
×
904
      qError("invalid optype %d for merge operator params", pDst->opType);
×
905
      return TSDB_CODE_INVALID_PARA;
×
906
  }
907

UNCOV
908
  return TSDB_CODE_SUCCESS;
×
909
}
910

911
int32_t setOperatorParams(struct SOperatorInfo* pOperator, SOperatorParam* pInput, SOperatorParamType type) {
36,700✔
912
  SOperatorParam**  ppParam = NULL;
36,700✔
913
  SOperatorParam*** pppDownstramParam = NULL;
36,700✔
914
  switch (type) {
36,700!
915
    case OP_GET_PARAM:
36,700✔
916
      ppParam = &pOperator->pOperatorGetParam;
36,700✔
917
      pppDownstramParam = &pOperator->pDownstreamGetParams;
36,700✔
918
      break;
36,700✔
919
    case OP_NOTIFY_PARAM:
×
920
      ppParam = &pOperator->pOperatorNotifyParam;
×
UNCOV
921
      pppDownstramParam = &pOperator->pDownstreamNotifyParams;
×
UNCOV
922
      break;
×
923
    default:
×
UNCOV
924
      return TSDB_CODE_INVALID_PARA;
×
925
  }
926

927
  freeResetOperatorParams(pOperator, type, false);
36,700✔
928

929
  if (NULL == pInput) {
36,701!
UNCOV
930
    return TSDB_CODE_SUCCESS;
×
931
  }
932

933
  *ppParam = (pInput->opType == pOperator->operatorType) ? pInput : NULL;
36,701!
934

935
  if (NULL == *pppDownstramParam) {
36,701✔
936
    *pppDownstramParam = taosMemoryCalloc(pOperator->numOfDownstream, POINTER_BYTES);
2,337!
937
    if (NULL == *pppDownstramParam) {
2,335!
938
      return terrno;
×
939
    }
940
  }
941

942
  if (NULL == *ppParam) {
36,699!
UNCOV
943
    for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
×
UNCOV
944
      (*pppDownstramParam)[i] = pInput;
×
945
    }
UNCOV
946
    return TSDB_CODE_SUCCESS;
×
947
  }
948

949
  memset(*pppDownstramParam, 0, pOperator->numOfDownstream * POINTER_BYTES);
36,699✔
950

951
  int32_t childrenNum = taosArrayGetSize((*ppParam)->pChildren);
36,699✔
952
  if (childrenNum <= 0) {
36,699✔
953
    return TSDB_CODE_SUCCESS;
1,997✔
954
  }
955

956
  for (int32_t i = 0; i < childrenNum; ++i) {
104,106✔
957
    SOperatorParam* pChild = *(SOperatorParam**)taosArrayGet((*ppParam)->pChildren, i);
69,404✔
958
    if (pChild == NULL) {
69,404!
959
      return terrno;
×
960
    }
961

962
    if ((*pppDownstramParam)[pChild->downstreamIdx]) {
69,404!
UNCOV
963
      int32_t code = mergeOperatorParams((*pppDownstramParam)[pChild->downstreamIdx], pChild);
×
UNCOV
964
      if (code) {
×
UNCOV
965
        return code;
×
966
      }
967
    } else {
968
      (*pppDownstramParam)[pChild->downstreamIdx] = pChild;
69,404✔
969
    }
970
  }
971

972
  taosArrayDestroy((*ppParam)->pChildren);
34,702✔
973
  (*ppParam)->pChildren = NULL;
34,702✔
974

975
  return TSDB_CODE_SUCCESS;
34,702✔
976
}
977

978
SSDataBlock* getNextBlockFromDownstream(struct SOperatorInfo* pOperator, int32_t idx) {
38,356,491✔
979
  SSDataBlock* p = NULL;
38,356,491✔
980
  int32_t      code = getNextBlockFromDownstreamImpl(pOperator, idx, true, &p);
38,356,491✔
981
  if (code == TSDB_CODE_SUCCESS) {
38,338,828!
982
    code = blockDataCheck(p);
38,340,354✔
983
    if (code != TSDB_CODE_SUCCESS) {
38,332,753✔
984
      qError("blockDataCheck failed, code:%s", tstrerror(code));
167!
985
    }
986
  }
987
  return (code == 0) ? p : NULL;
38,332,269✔
988
}
989

990
SSDataBlock* getNextBlockFromDownstreamRemain(struct SOperatorInfo* pOperator, int32_t idx) {
565,218✔
991
  SSDataBlock* p = NULL;
565,218✔
992
  int32_t      code = getNextBlockFromDownstreamImpl(pOperator, idx, false, &p);
565,218✔
993
  if (code == TSDB_CODE_SUCCESS) {
565,194!
994
    code = blockDataCheck(p);
565,221✔
995
    if (code != TSDB_CODE_SUCCESS) {
565,238!
UNCOV
996
      qError("blockDataCheck failed, code:%s", tstrerror(code));
×
997
    }
998
  }
999
  return (code == 0)? p:NULL;
565,236!
1000
}
1001

1002
int32_t optrDefaultGetNextExtFn(struct SOperatorInfo* pOperator, SOperatorParam* pParam, SSDataBlock** pRes) {
36,700✔
1003
  QRY_PARAM_CHECK(pRes);
36,700!
1004

1005
  int32_t lino = 0;
36,700✔
1006
  int32_t code = setOperatorParams(pOperator, pParam, OP_GET_PARAM);
36,700✔
1007
  QUERY_CHECK_CODE(code, lino, _end);
36,699!
1008
  code = pOperator->fpSet.getNextFn(pOperator, pRes);
36,699✔
1009
  QUERY_CHECK_CODE(code, lino, _end);
36,696!
1010

1011
_end:
36,696✔
1012
  if (code != TSDB_CODE_SUCCESS) {
36,696!
UNCOV
1013
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
UNCOV
1014
    pOperator->pTaskInfo->code = code;
×
UNCOV
1015
    T_LONG_JMP(pOperator->pTaskInfo->env, code);
×
1016
  }
1017

1018
  return code;
36,696✔
1019
}
1020

UNCOV
1021
int32_t optrDefaultNotifyFn(struct SOperatorInfo* pOperator, SOperatorParam* pParam) {
×
UNCOV
1022
  int32_t code = setOperatorParams(pOperator, pParam, OP_NOTIFY_PARAM);
×
UNCOV
1023
  if (TSDB_CODE_SUCCESS == code && pOperator->fpSet.notifyFn && pOperator->pOperatorNotifyParam) {
×
UNCOV
1024
    code = pOperator->fpSet.notifyFn(pOperator, pOperator->pOperatorNotifyParam);
×
1025
  }
UNCOV
1026
  if (TSDB_CODE_SUCCESS == code) {
×
UNCOV
1027
    for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
×
1028
      if (pOperator->pDownstreamNotifyParams[i]) {
×
1029
        code = optrDefaultNotifyFn(pOperator->pDownstream[i], pOperator->pDownstreamNotifyParams[i]);
×
1030
        if (TSDB_CODE_SUCCESS != code) {
×
UNCOV
1031
          break;
×
1032
        }
UNCOV
1033
        pOperator->pDownstreamNotifyParams[i] = NULL;
×
1034
      }
1035
    }
1036
  }
1037
  if (TSDB_CODE_SUCCESS != code) {
×
1038
    pOperator->pTaskInfo->code = code;
×
1039
    T_LONG_JMP(pOperator->pTaskInfo->env, pOperator->pTaskInfo->code);
×
1040
  }
1041

1042
  return code;
×
1043
}
1044

1045
int16_t getOperatorResultBlockId(struct SOperatorInfo* pOperator, int32_t idx) {
228,865✔
1046
  if (pOperator->transparent) {
228,865✔
1047
    return getOperatorResultBlockId(pOperator->pDownstream[idx], 0);
22,252✔
1048
  }
1049
  return pOperator->resultDataBlockId;
206,613✔
1050
}
STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2026 Coveralls, Inc