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

taosdata / TDengine / #3819

01 Apr 2025 09:27AM UTC coverage: 34.076% (+0.01%) from 34.065%
#3819

push

travis-ci

happyguoxy
test:alter gcda dir

148544 of 599532 branches covered (24.78%)

Branch coverage included in aggregate %.

222541 of 489451 relevant lines covered (45.47%)

763329.1 hits per line

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

42.92
/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,
4,201✔
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 = {
4,201✔
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;
4,201✔
49
}
50

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

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

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

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

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

80
void setOperatorInfo(SOperatorInfo* pOperator, const char* name, int32_t type, bool blocking, int32_t status,
4,189✔
81
                     void* pInfo, SExecTaskInfo* pTaskInfo) {
82
  pOperator->name = (char*)name;
4,189✔
83
  pOperator->operatorType = type;
4,189✔
84
  pOperator->blocking = blocking;
4,189✔
85
  pOperator->status = status;
4,189✔
86
  pOperator->info = pInfo;
4,189✔
87
  pOperator->pTaskInfo = pTaskInfo;
4,189✔
88
}
4,189✔
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) {
19,081✔
153
  if (pOperator == NULL) {
19,081!
154
    return;
×
155
  }
156

157
  ERetType ret = fn(pOperator, pParam, id);
19,081✔
158
  if (ret == OPTR_FN_RET_ABORT || pParam->code != TSDB_CODE_SUCCESS) {
19,081!
159
    return;
11,149✔
160
  }
161

162
  for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
15,864✔
163
    traverseOperatorTree(pOperator->pDownstream[i], fn, pParam, id);
7,932✔
164
    if (pParam->code != 0) {
7,932!
165
      break;
×
166
    }
167
  }
168
}
169

170
ERetType extractOperatorInfo(SOperatorInfo* pOperator, STraverParam* pParam, const char* pIdStr) {
19,040✔
171
  STraverParam* p = pParam;
19,040✔
172
  if (pOperator->operatorType == *(int32_t*)p->pParam) {
19,040✔
173
    p->pRet = pOperator;
11,127✔
174
    return OPTR_FN_RET_ABORT;
11,127✔
175
  } else {
176
    return OPTR_FN_RET_CONTINUE;
7,913✔
177
  }
178
}
179

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

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

189
  STraverParam p = {.pParam = &type, .pRet = NULL};
11,127✔
190
  traverseOperatorTree(pOperator, extractOperatorInfo, &p, id);
11,127✔
191
  if (p.code == 0) {
11,127!
192
    *pOptrInfo = p.pRet;
11,127✔
193
  }
194
  return p.code;
11,127✔
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) {
41✔
252
  SStorageAPI* pAPI = pParam->pParam;
41✔
253
  if (pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
41✔
254
    STableScanInfo* pInfo = pOperator->info;
5✔
255

256
    if (pInfo->base.dataReader != NULL) {
5!
257
      pAPI->tsdReader.tsdReaderNotifyClosing(pInfo->base.dataReader);
5✔
258
    }
259
    return OPTR_FN_RET_ABORT;
5✔
260
  } else if (pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN) {
36✔
261
    SStreamScanInfo* pInfo = pOperator->info;
17✔
262

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

270
    return OPTR_FN_RET_ABORT;
17✔
271
  }
272

273
  return OPTR_FN_RET_CONTINUE;
19✔
274
}
275

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

282
int32_t createOperator(SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo, SReadHandle* pHandle, SNode* pTagCond,
4,128✔
283
                       SNode* pTagIndexCond, const char* pUser, const char* dbname, SOperatorInfo** pOptrInfo,
284
                       EOPTR_EXEC_MODEL model) {
285
  QRY_PARAM_CHECK(pOptrInfo);
4,128!
286

287
  int32_t     code = 0;
4,128✔
288
  int32_t     type = nodeType(pPhyNode);
4,128✔
289
  const char* idstr = GET_TASKID(pTaskInfo);
4,128✔
290

291
  if (pPhyNode->pChildren == NULL || LIST_LENGTH(pPhyNode->pChildren) == 0) {
4,128!
292
    SOperatorInfo* pOperator = NULL;
2,590✔
293
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == type) {
2,590✔
294
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pPhyNode;
1,450✔
295
      // NOTE: this is an patch to fix the physical plan
296
      // TODO remove it later
297
      if (pTableScanNode->scan.node.pLimit != NULL) {
1,450✔
298
        pTableScanNode->groupSort = true;
106✔
299
      }
300

301
      STableListInfo* pTableListInfo = tableListCreate();
1,450✔
302
      if (!pTableListInfo) {
1,451!
303
        pTaskInfo->code = terrno;
×
304
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
305
        return terrno;
×
306
      }
307

308
      // Since virtual stable scan use virtual super table's uid in scan operator, the origin table might be stored on
309
      // different vnode, so we should not get table schema for virtual stable scan.
310
      if (!pTableScanNode->scan.virtualStableScan) {
1,451!
311
        code = initQueriedTableSchemaInfo(pHandle, &pTableScanNode->scan, dbname, pTaskInfo);
1,451✔
312
        if (code) {
1,451!
313
          pTaskInfo->code = code;
×
314
          tableListDestroy(pTableListInfo);
×
315
          return code;
×
316
        }
317
      }
318

319
      if (pTableScanNode->scan.node.dynamicOp) {
1,451✔
320
        pTaskInfo->dynamicTask = true;
4✔
321
        pTableListInfo->idInfo.suid = pTableScanNode->scan.suid;
4✔
322
        pTableListInfo->idInfo.tableType = pTableScanNode->scan.tableType;
4✔
323
      } else {
324
        code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
1,447✔
325
                                       pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo);
326
        if (code) {
1,447!
327
          pTaskInfo->code = code;
×
328
          tableListDestroy(pTableListInfo);
×
329
          qError("failed to createScanTableListInfo, code:%s, %s", tstrerror(code), idstr);
×
330
          return code;
×
331
        }
332
      }
333

334
      code = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
1,451✔
335
      if (NULL == pOperator || code != 0) {
1,451!
336
        pTaskInfo->code = code;
×
337
        tableListDestroy(pTableListInfo);
×
338
        return code;
×
339
      }
340

341
      STableScanInfo* pScanInfo = pOperator->info;
1,451✔
342
      pTaskInfo->cost.pRecoder = &pScanInfo->base.readRecorder;
1,451✔
343
    } else if (QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN == type) {
1,140✔
344
      STableMergeScanPhysiNode* pTableScanNode = (STableMergeScanPhysiNode*)pPhyNode;
64✔
345
      STableListInfo*           pTableListInfo = tableListCreate();
64✔
346
      if (!pTableListInfo) {
64!
347
        pTaskInfo->code = terrno;
×
348
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
349
        return terrno;
×
350
      }
351

352
      code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, true, pHandle, pTableListInfo,
64✔
353
                                     pTagCond, pTagIndexCond, pTaskInfo);
354
      if (code) {
64!
355
        pTaskInfo->code = code;
×
356
        tableListDestroy(pTableListInfo);
×
357
        qError("failed to createScanTableListInfo, code:%s", tstrerror(code));
×
358
        return code;
×
359
      }
360

361
      code = initQueriedTableSchemaInfo(pHandle, &pTableScanNode->scan, dbname, pTaskInfo);
64✔
362
      if (code) {
64!
363
        pTaskInfo->code = code;
×
364
        tableListDestroy(pTableListInfo);
×
365
        return code;
×
366
      }
367

368
      code = createTableMergeScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
64✔
369
      if (NULL == pOperator || code != 0) {
64!
370
        pTaskInfo->code = code;
×
371
        tableListDestroy(pTableListInfo);
×
372
        return code;
×
373
      }
374

375
      STableScanInfo* pScanInfo = pOperator->info;
64✔
376
      pTaskInfo->cost.pRecoder = &pScanInfo->base.readRecorder;
64✔
377
    } else if (QUERY_NODE_PHYSICAL_PLAN_EXCHANGE == type) {
1,076✔
378
      code = createExchangeOperatorInfo(pHandle ? pHandle->pMsgCb->clientRpc : NULL, (SExchangePhysiNode*)pPhyNode,
636!
379
                                        pTaskInfo, &pOperator);
380
    } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN == type) {
440✔
381
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pPhyNode;
76✔
382
      STableListInfo*      pTableListInfo = tableListCreate();
76✔
383
      if (!pTableListInfo) {
76!
384
        pTaskInfo->code = terrno;
×
385
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
386
        return terrno;
×
387
      }
388

389
      if (pHandle->vnode && (pTaskInfo->pSubplan->pVTables == NULL)) {
76!
390
        code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
72✔
391
                                       pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo);
392
        if (code) {
72!
393
          pTaskInfo->code = code;
×
394
          tableListDestroy(pTableListInfo);
×
395
          qError("failed to createScanTableListInfo, code:%s", tstrerror(code));
×
396
          return code;
×
397
        }
398
      }
399

400
      code = createStreamScanOperatorInfo(pHandle, pTableScanNode, pTagCond, pTableListInfo, pTaskInfo, &pOperator);
76✔
401
      if (code) {
76!
402
        pTaskInfo->code = code;
×
403
        tableListDestroy(pTableListInfo);
×
404
        return code;
×
405
      }
406
    } else if (QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN == type) {
364✔
407
      SSystemTableScanPhysiNode* pSysScanPhyNode = (SSystemTableScanPhysiNode*)pPhyNode;
341✔
408
      code = createSysTableScanOperatorInfo(pHandle, pSysScanPhyNode, pUser, pTaskInfo, &pOperator);
341✔
409
    } else if (QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN == type) {
23!
410
      STableCountScanPhysiNode* pTblCountScanNode = (STableCountScanPhysiNode*)pPhyNode;
×
411
      code = createTableCountScanOperatorInfo(pHandle, pTblCountScanNode, pTaskInfo, &pOperator);
×
412
    } else if (QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN == type) {
23✔
413
      STagScanPhysiNode* pTagScanPhyNode = (STagScanPhysiNode*)pPhyNode;
6✔
414
      STableListInfo*    pTableListInfo = tableListCreate();
6✔
415
      if (!pTableListInfo) {
6!
416
        pTaskInfo->code = terrno;
×
417
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
418
        return terrno;
×
419
      }
420
      if (!pTagScanPhyNode->onlyMetaCtbIdx) {
6✔
421
        code = createScanTableListInfo((SScanPhysiNode*)pTagScanPhyNode, NULL, false, pHandle, pTableListInfo, pTagCond,
2✔
422
                                       pTagIndexCond, pTaskInfo);
423
        if (code != TSDB_CODE_SUCCESS) {
2!
424
          pTaskInfo->code = code;
×
425
          qError("failed to getTableList, code:%s", tstrerror(code));
×
426
          tableListDestroy(pTableListInfo);
×
427
          return code;
×
428
        }
429
      }
430
      code = createTagScanOperatorInfo(pHandle, pTagScanPhyNode, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo,
6✔
431
                                       &pOperator);
432
      if (code) {
6!
433
        pTaskInfo->code = code;
×
434
        tableListDestroy(pTableListInfo);
×
435
        return code;
×
436
      }
437
    } else if (QUERY_NODE_PHYSICAL_PLAN_BLOCK_DIST_SCAN == type) {
17✔
438
      SBlockDistScanPhysiNode* pBlockNode = (SBlockDistScanPhysiNode*)pPhyNode;
5✔
439
      STableListInfo*          pTableListInfo = tableListCreate();
5✔
440
      if (!pTableListInfo) {
5!
441
        pTaskInfo->code = terrno;
×
442
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
443
        return terrno;
×
444
      }
445

446
      if (pBlockNode->tableType == TSDB_SUPER_TABLE) {
5✔
447
        SArray* pList = taosArrayInit(4, sizeof(uint64_t));
4✔
448
        code = pTaskInfo->storageAPI.metaFn.getChildTableList(pHandle->vnode, pBlockNode->uid, pList);
4✔
449
        if (code != TSDB_CODE_SUCCESS) {
4!
450
          pTaskInfo->code = code;
×
451
          taosArrayDestroy(pList);
×
452
          tableListDestroy(pTableListInfo);
×
453
          return code;
×
454
        }
455

456
        size_t num = taosArrayGetSize(pList);
4✔
457
        for (int32_t i = 0; i < num; ++i) {
14✔
458
          uint64_t* id = taosArrayGet(pList, i);
10✔
459
          if (id == NULL) {
10!
460
            continue;
×
461
          }
462

463
          code = tableListAddTableInfo(pTableListInfo, *id, 0);
10✔
464
          if (code) {
10!
465
            pTaskInfo->code = code;
×
466
            tableListDestroy(pTableListInfo);
×
467
            taosArrayDestroy(pList);
×
468
            return code;
×
469
          }
470
        }
471

472
        taosArrayDestroy(pList);
4✔
473
      } else {  // Create group with only one table
474
        code = tableListAddTableInfo(pTableListInfo, pBlockNode->uid, 0);
1✔
475
        if (code) {
1!
476
          pTaskInfo->code = code;
×
477
          tableListDestroy(pTableListInfo);
×
478
          return code;
×
479
        }
480
      }
481

482
      code = createDataBlockInfoScanOperator(pHandle, pBlockNode, pTableListInfo, pTaskInfo, &pOperator);
5✔
483
      if (code) {
5!
484
        pTaskInfo->code = code;
×
485
        tableListDestroy(pTableListInfo);
×
486
        return code;
×
487
      }
488
    } else if (QUERY_NODE_PHYSICAL_PLAN_LAST_ROW_SCAN == type) {
12!
489
      SLastRowScanPhysiNode* pScanNode = (SLastRowScanPhysiNode*)pPhyNode;
12✔
490
      STableListInfo*        pTableListInfo = tableListCreate();
12✔
491
      if (!pTableListInfo) {
12!
492
        pTaskInfo->code = terrno;
×
493
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
494
        return terrno;
×
495
      }
496

497
      code = createScanTableListInfo(&pScanNode->scan, pScanNode->pGroupTags, true, pHandle, pTableListInfo, pTagCond,
12✔
498
                                     pTagIndexCond, pTaskInfo);
499
      if (code != TSDB_CODE_SUCCESS) {
12!
500
        pTaskInfo->code = code;
×
501
        tableListDestroy(pTableListInfo);
×
502
        return code;
×
503
      }
504

505
      code = initQueriedTableSchemaInfo(pHandle, &pScanNode->scan, dbname, pTaskInfo);
12✔
506
      if (code != TSDB_CODE_SUCCESS) {
12!
507
        pTaskInfo->code = code;
×
508
        tableListDestroy(pTableListInfo);
×
509
        return code;
×
510
      }
511

512
      code = createCacherowsScanOperator(pScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
12✔
513
      if (code) {
12!
514
        tableListDestroy(pTableListInfo);
×
515
        pTaskInfo->code = code;
×
516
        return code;
×
517
      }
518
    } else if (QUERY_NODE_PHYSICAL_PLAN_PROJECT == type) {
×
519
      code = createProjectOperatorInfo(NULL, (SProjectPhysiNode*)pPhyNode, pTaskInfo, &pOperator);
×
520
    } else if (QUERY_NODE_PHYSICAL_PLAN_VIRTUAL_TABLE_SCAN == type && model != OPTR_EXEC_MODEL_STREAM) {
×
521
      code = createVirtualTableMergeOperatorInfo(NULL, pHandle, NULL, 0, (SVirtualScanPhysiNode*)pPhyNode, pTaskInfo, &pOperator);
×
522
    } else {
523
      code = TSDB_CODE_INVALID_PARA;
×
524
      pTaskInfo->code = code;
×
525
      return code;
×
526
    }
527

528
    if (pOperator != NULL) {  // todo moved away
2,590!
529
      pOperator->resultDataBlockId = pPhyNode->pOutputDataBlockDesc->dataBlockId;
2,591✔
530
    }
531

532
    *pOptrInfo = pOperator;
2,590✔
533
    return code;
2,590✔
534
  }
535

536
  size_t          size = LIST_LENGTH(pPhyNode->pChildren);
1,538!
537
  SOperatorInfo** ops = taosMemoryCalloc(size, POINTER_BYTES);
1,538!
538
  if (ops == NULL) {
1,538!
539
    code = terrno;
×
540
    pTaskInfo->code = code;
×
541
    return code;
×
542
  }
543

544
  for (int32_t i = 0; i < size; ++i) {
3,222✔
545
    SPhysiNode* pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, i);
1,685✔
546
    code = createOperator(pChildNode, pTaskInfo, pHandle, pTagCond, pTagIndexCond, pUser, dbname, &ops[i], model);
1,684✔
547
    if (ops[i] == NULL || code != 0) {
1,684!
548
      for (int32_t j = 0; j < i; ++j) {
×
549
        destroyOperator(ops[j]);
×
550
      }
551
      taosMemoryFree(ops);
×
552
      return code;
×
553
    }
554
  }
555

556
  SOperatorInfo* pOptr = NULL;
1,537✔
557
  if (QUERY_NODE_PHYSICAL_PLAN_PROJECT == type) {
1,537✔
558
    code = createProjectOperatorInfo(ops[0], (SProjectPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
610✔
559
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_AGG == type) {
927✔
560
    SAggPhysiNode* pAggNode = (SAggPhysiNode*)pPhyNode;
565✔
561
    if (pAggNode->pGroupKeys != NULL) {
565✔
562
      code = createGroupOperatorInfo(ops[0], pAggNode, pTaskInfo, &pOptr);
183✔
563
    } else {
564
      code = createAggregateOperatorInfo(ops[0], pAggNode, pTaskInfo, &pOptr);
382✔
565
    }
566
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_INTERVAL == type) {
362✔
567
    SIntervalPhysiNode* pIntervalPhyNode = (SIntervalPhysiNode*)pPhyNode;
133✔
568
    code = createIntervalOperatorInfo(ops[0], pIntervalPhyNode, pTaskInfo, &pOptr);
133✔
569
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL == type) {
229✔
570
    code = createStreamSingleIntervalOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
3✔
571
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_INTERVAL == type) {
226!
572
    code = createStreamIntervalSliceOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
573
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_ALIGNED_INTERVAL == type) {
226✔
574
    SMergeAlignedIntervalPhysiNode* pIntervalPhyNode = (SMergeAlignedIntervalPhysiNode*)pPhyNode;
41✔
575
    code = createMergeAlignedIntervalOperatorInfo(ops[0], pIntervalPhyNode, pTaskInfo, &pOptr);
41✔
576
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_INTERVAL == type) {
185!
577
    SMergeIntervalPhysiNode* pIntervalPhyNode = (SMergeIntervalPhysiNode*)pPhyNode;
×
578
    code = createMergeIntervalOperatorInfo(ops[0], pIntervalPhyNode, pTaskInfo, &pOptr);
×
579
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_INTERVAL == type) {
185✔
580
    int32_t children = 0;
9✔
581
    code = createStreamFinalIntervalOperatorInfo(ops[0], pPhyNode, pTaskInfo, children, pHandle, &pOptr);
9✔
582
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_SEMI_INTERVAL == type) {
176!
583
    code = createSemiIntervalSliceOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
584
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_MID_INTERVAL == type) {
176!
585
    int32_t children = pHandle->numOfVgroups;
×
586
    code = createStreamFinalIntervalOperatorInfo(ops[0], pPhyNode, pTaskInfo, children, pHandle, &pOptr);
×
587
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_INTERVAL == type) {
176✔
588
    int32_t children = pHandle->numOfVgroups;
4✔
589
    code = createStreamFinalIntervalOperatorInfo(ops[0], pPhyNode, pTaskInfo, children, pHandle, &pOptr);
4✔
590
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_FINAL_INTERVAL == type) {
172!
591
    code = createFinalIntervalSliceOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
592
  } else if (QUERY_NODE_PHYSICAL_PLAN_SORT == type) {
172✔
593
    code = createSortOperatorInfo(ops[0], (SSortPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
50✔
594
  } else if (QUERY_NODE_PHYSICAL_PLAN_GROUP_SORT == type) {
122!
595
    code = createGroupSortOperatorInfo(ops[0], (SGroupSortPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
×
596
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE == type) {
122✔
597
    SMergePhysiNode* pMergePhyNode = (SMergePhysiNode*)pPhyNode;
77✔
598
    code = createMultiwayMergeOperatorInfo(ops, size, pMergePhyNode, pTaskInfo, &pOptr);
77✔
599
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_SESSION == type) {
45✔
600
    SSessionWinodwPhysiNode* pSessionNode = (SSessionWinodwPhysiNode*)pPhyNode;
1✔
601
    code = createSessionAggOperatorInfo(ops[0], pSessionNode, pTaskInfo, &pOptr);
1✔
602
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SESSION == type) {
44!
603
    code = createStreamSessionAggOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
604
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_SESSION == type) {
44!
605
    int32_t children = 0;
×
606
    code = createStreamFinalSessionAggOperatorInfo(ops[0], pPhyNode, pTaskInfo, children, pHandle, &pOptr);
×
607
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_SESSION == type) {
44!
608
    int32_t children = pHandle->numOfVgroups;
×
609
    code = createStreamFinalSessionAggOperatorInfo(ops[0], pPhyNode, pTaskInfo, children, pHandle, &pOptr);
×
610
  } else if (QUERY_NODE_PHYSICAL_PLAN_PARTITION == type) {
44!
611
    code = createPartitionOperatorInfo(ops[0], (SPartitionPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
×
612
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_PARTITION == type) {
44!
613
    code = createStreamPartitionOperatorInfo(ops[0], (SStreamPartitionPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
×
614
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_STATE == type) {
44!
615
    SStateWinodwPhysiNode* pStateNode = (SStateWinodwPhysiNode*)pPhyNode;
×
616
    code = createStatewindowOperatorInfo(ops[0], pStateNode, pTaskInfo, &pOptr);
×
617
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_STATE == type) {
44!
618
    code = createStreamStateAggOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
619
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_EVENT == type) {
44!
620
    code = createStreamEventAggOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
621
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_JOIN == type) {
44✔
622
    code = createMergeJoinOperatorInfo(ops, size, (SSortMergeJoinPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
32✔
623
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_JOIN == type) {
12✔
624
    code = createHashJoinOperatorInfo(ops, size, (SHashJoinPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
1✔
625
  } else if (QUERY_NODE_PHYSICAL_PLAN_FILL == type) {
11✔
626
    code = createFillOperatorInfo(ops[0], (SFillPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
8✔
627
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_FILL == type) {
3!
628
    code = createStreamFillOperatorInfo(ops[0], (SStreamFillPhysiNode*)pPhyNode, pTaskInfo, pHandle, &pOptr);
×
629
  } else if (QUERY_NODE_PHYSICAL_PLAN_INDEF_ROWS_FUNC == type) {
3✔
630
    code = createIndefinitOutputOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
2✔
631
  } else if (QUERY_NODE_PHYSICAL_PLAN_INTERP_FUNC == type) {
1!
632
    code = createTimeSliceOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
×
633
  } else if (QUERY_NODE_PHYSICAL_PLAN_FORECAST_FUNC == type) {
1!
634
    code = createForecastOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
×
635
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_EVENT == type) {
1!
636
    code = createEventwindowOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
×
637
  } else if (QUERY_NODE_PHYSICAL_PLAN_GROUP_CACHE == type) {
1!
638
    code = createGroupCacheOperatorInfo(ops, size, (SGroupCachePhysiNode*)pPhyNode, pTaskInfo, &pOptr);
1✔
639
  } else if (QUERY_NODE_PHYSICAL_PLAN_DYN_QUERY_CTRL == type) {
×
640
    code = createDynQueryCtrlOperatorInfo(ops, size, (SDynQueryCtrlPhysiNode*)pPhyNode, pTaskInfo, pHandle, &pOptr);
1✔
641
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_COUNT == type) {
×
642
    code = createStreamCountAggOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
643
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_COUNT == type) {
×
644
    code = createCountwindowOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
×
645
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERP_FUNC == type) {
×
646
    code = createStreamTimeSliceOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
647
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_ANOMALY == type) {
×
648
    code = createAnomalywindowOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
×
649
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_SESSION == type) {
×
650
    code = createSessionNonblockOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
651
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_SEMI_SESSION == type) {
×
652
    code = createSemiSessionNonblockOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
653
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_FINAL_SESSION == type) {
×
654
    code = createFinalSessionNonblockOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
655
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_STATE == type) {
×
656
    code = createStateNonblockOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
657
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_EVENT == type) {
×
658
    code = createEventNonblockOperatorInfo(ops[0], pPhyNode, pTaskInfo, pHandle, &pOptr);
×
659
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_COUNT == type) {
×
660
    //todo (liuyao) add
661
  } else if (QUERY_NODE_PHYSICAL_PLAN_VIRTUAL_TABLE_SCAN == type && model != OPTR_EXEC_MODEL_STREAM) {
×
662
    SVirtualScanPhysiNode* pVirtualTableScanNode = (SVirtualScanPhysiNode*)pPhyNode;
×
663
    // NOTE: this is an patch to fix the physical plan
664

665
    if (pVirtualTableScanNode->scan.node.pLimit != NULL) {
×
666
      pVirtualTableScanNode->groupSort = true;
×
667
    }
668

669
    STableListInfo* pTableListInfo = tableListCreate();
×
670
    if (!pTableListInfo) {
×
671
      pTaskInfo->code = terrno;
×
672
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
673
      return terrno;
×
674
    }
675

676
    code = createScanTableListInfo(&pVirtualTableScanNode->scan, pVirtualTableScanNode->pGroupTags, pVirtualTableScanNode->groupSort,
×
677
                                   pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo);
678
    if (code) {
×
679
      pTaskInfo->code = code;
×
680
      tableListDestroy(pTableListInfo);
×
681
      qError("failed to createScanTableListInfo, code:%s, %s", tstrerror(code), idstr);
×
682
      return code;
×
683
    }
684

685
    code = createVirtualTableMergeOperatorInfo(ops, pHandle, pTableListInfo, size, (SVirtualScanPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
×
686
  } else if (QUERY_NODE_PHYSICAL_PLAN_VIRTUAL_TABLE_SCAN == type && model == OPTR_EXEC_MODEL_STREAM) {
×
687
    SVirtualScanPhysiNode* pVirtualTableScanNode = (SVirtualScanPhysiNode*)pPhyNode;
×
688
    STableListInfo*        pTableListInfo = tableListCreate();
×
689
    if (!pTableListInfo) {
×
690
      pTaskInfo->code = terrno;
×
691
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
692
      return terrno;
×
693
    }
694

695
    code = createScanTableListInfo(&pVirtualTableScanNode->scan, pVirtualTableScanNode->pGroupTags,
×
696
                                   pVirtualTableScanNode->groupSort, pHandle, pTableListInfo, pTagCond, pTagIndexCond,
×
697
                                   pTaskInfo);
698
    if (code) {
×
699
      pTaskInfo->code = code;
×
700
      tableListDestroy(pTableListInfo);
×
701
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
702
      return code;
×
703
    }
704

705
    code = createStreamVtableMergeOperatorInfo(ops[0], pHandle, pVirtualTableScanNode, pTagCond, pTableListInfo, pTaskInfo, &pOptr);
×
706
  } else {
707
    code = TSDB_CODE_INVALID_PARA;
×
708
    pTaskInfo->code = code;
×
709
    for (int32_t i = 0; i < size; ++i) {
×
710
      destroyOperator(ops[i]);
×
711
    }
712
    taosMemoryFree(ops);
×
713
    qError("invalid operator type %d", type);
×
714
    return code;
×
715
  }
716

717
  taosMemoryFree(ops);
1,537!
718
  if (pOptr) {
1,538!
719
    pOptr->resultDataBlockId = pPhyNode->pOutputDataBlockDesc->dataBlockId;
1,538✔
720
  }
721

722
  *pOptrInfo = pOptr;
1,538✔
723
  return code;
1,538✔
724
}
725

726
void destroyOperator(SOperatorInfo* pOperator) {
4,202✔
727
  if (pOperator == NULL) {
4,202!
728
    return;
×
729
  }
730

731
  freeResetOperatorParams(pOperator, OP_GET_PARAM, true);
4,202✔
732
  freeResetOperatorParams(pOperator, OP_NOTIFY_PARAM, true);
4,201✔
733

734
  if (pOperator->pDownstream != NULL) {
4,199✔
735
    for (int32_t i = 0; i < pOperator->numOfRealDownstream; ++i) {
3,223✔
736
      destroyOperator(pOperator->pDownstream[i]);
1,685✔
737
    }
738

739
    taosMemoryFreeClear(pOperator->pDownstream);
1,538!
740
    pOperator->numOfDownstream = 0;
1,538✔
741
  }
742

743
  if (pOperator->fpSet.closeFn != NULL && pOperator->info != NULL) {
4,199!
744
    pOperator->fpSet.closeFn(pOperator->info);
4,202✔
745
  }
746

747
  cleanupExprSupp(&pOperator->exprSupp);
4,199✔
748
  taosMemoryFreeClear(pOperator);
4,202!
749
}
750

751
void destroyOperatorAndDownstreams(SOperatorInfo* pOperator, SOperatorInfo** downstreams, int32_t num) {
×
752
  if (downstreams != NULL) {
×
753
    for (int i = 0; i < num; i++) {
×
754
      destroyOperator(downstreams[i]);
×
755
    }
756
  }
757

758
  if (pOperator != NULL) {
×
759
    pOperator->info = NULL;
×
760
    if (pOperator->pDownstream != NULL) {
×
761
      taosMemoryFreeClear(pOperator->pDownstream);
×
762
      pOperator->pDownstream = NULL;
×
763
    }
764
    destroyOperator(pOperator);
×
765
  }
766
}
×
767

768
int32_t getOperatorExplainExecInfo(SOperatorInfo* operatorInfo, SArray* pExecInfoList) {
39✔
769
  SExplainExecInfo  execInfo = {0};
39✔
770
  SExplainExecInfo* pExplainInfo = taosArrayPush(pExecInfoList, &execInfo);
39✔
771
  if (pExplainInfo == NULL) {
39!
772
    return terrno;
×
773
  }
774

775
  pExplainInfo->numOfRows = operatorInfo->resultInfo.totalRows;
39✔
776
  pExplainInfo->startupCost = operatorInfo->cost.openCost;
39✔
777
  pExplainInfo->totalCost = operatorInfo->cost.totalCost;
39✔
778
  pExplainInfo->verboseLen = 0;
39✔
779
  pExplainInfo->verboseInfo = NULL;
39✔
780

781
  if (operatorInfo->fpSet.getExplainFn) {
39✔
782
    int32_t code =
783
        operatorInfo->fpSet.getExplainFn(operatorInfo, &pExplainInfo->verboseInfo, &pExplainInfo->verboseLen);
14✔
784
    if (code) {
14!
785
      qError("%s operator getExplainFn failed, code:%s", GET_TASKID(operatorInfo->pTaskInfo), tstrerror(code));
×
786
      return code;
×
787
    }
788
  }
789

790
  int32_t code = 0;
39✔
791
  for (int32_t i = 0; i < operatorInfo->numOfDownstream; ++i) {
58✔
792
    code = getOperatorExplainExecInfo(operatorInfo->pDownstream[i], pExecInfoList);
19✔
793
    if (code != TSDB_CODE_SUCCESS) {
19!
794
      //      taosMemoryFreeClear(*pRes);
795
      return code;
×
796
    }
797
  }
798

799
  return TSDB_CODE_SUCCESS;
39✔
800
}
801

802
int32_t mergeOperatorParams(SOperatorParam* pDst, SOperatorParam* pSrc) {
×
803
  if (pDst->opType != pSrc->opType) {
×
804
    qError("different optype %d:%d for merge operator params", pDst->opType, pSrc->opType);
×
805
    return TSDB_CODE_INVALID_PARA;
×
806
  }
807

808
  switch (pDst->opType) {
×
809
    case QUERY_NODE_PHYSICAL_PLAN_EXCHANGE: {
×
810
      SExchangeOperatorParam* pDExc = pDst->value;
×
811
      SExchangeOperatorParam* pSExc = pSrc->value;
×
812
      if (!pDExc->multiParams) {
×
813
        if (pSExc->basic.vgId != pDExc->basic.vgId) {
×
814
          SExchangeOperatorBatchParam* pBatch = taosMemoryMalloc(sizeof(SExchangeOperatorBatchParam));
×
815
          if (NULL == pBatch) {
×
816
            return terrno;
×
817
          }
818

819
          pBatch->multiParams = true;
×
820
          pBatch->pBatchs = tSimpleHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
×
821
          if (NULL == pBatch->pBatchs) {
×
822
            taosMemoryFree(pBatch);
×
823
            return terrno;
×
824
          }
825

826
          tSimpleHashSetFreeFp(pBatch->pBatchs, freeExchangeGetBasicOperatorParam);
×
827

828
          int32_t code = tSimpleHashPut(pBatch->pBatchs, &pDExc->basic.vgId, sizeof(pDExc->basic.vgId), &pDExc->basic,
×
829
                                        sizeof(pDExc->basic));
830
          if (code) {
×
831
            return code;
×
832
          }
833

834
          code = tSimpleHashPut(pBatch->pBatchs, &pSExc->basic.vgId, sizeof(pSExc->basic.vgId), &pSExc->basic,
×
835
                                sizeof(pSExc->basic));
836
          if (code) {
×
837
            return code;
×
838
          }
839

840
          taosMemoryFree(pDst->value);
×
841
          pDst->value = pBatch;
×
842
        } else {
843
          void* p = taosArrayAddAll(pDExc->basic.uidList, pSExc->basic.uidList);
×
844
          if (p == NULL) {
×
845
            return terrno;
×
846
          }
847
        }
848
      } else {
849
        SExchangeOperatorBatchParam* pBatch = pDst->value;
×
850
        SExchangeOperatorBasicParam* pBasic =
851
            tSimpleHashGet(pBatch->pBatchs, &pSExc->basic.vgId, sizeof(pSExc->basic.vgId));
×
852
        if (pBasic) {
×
853
          void* p = taosArrayAddAll(pBasic->uidList, pSExc->basic.uidList);
×
854
          if (p == NULL) {
×
855
            return terrno;
×
856
          }
857
        } else {
858
          int32_t code = tSimpleHashPut(pBatch->pBatchs, &pSExc->basic.vgId, sizeof(pSExc->basic.vgId), &pSExc->basic,
×
859
                                        sizeof(pSExc->basic));
860
          if (code) {
×
861
            return code;
×
862
          }
863
        }
864
      }
865
      break;
×
866
    }
867
    default:
×
868
      qError("invalid optype %d for merge operator params", pDst->opType);
×
869
      return TSDB_CODE_INVALID_PARA;
×
870
  }
871

872
  return TSDB_CODE_SUCCESS;
×
873
}
874

875
int32_t setOperatorParams(struct SOperatorInfo* pOperator, SOperatorParam* pInput, SOperatorParamType type) {
7✔
876
  SOperatorParam**  ppParam = NULL;
7✔
877
  SOperatorParam*** pppDownstramParam = NULL;
7✔
878
  switch (type) {
7!
879
    case OP_GET_PARAM:
7✔
880
      ppParam = &pOperator->pOperatorGetParam;
7✔
881
      pppDownstramParam = &pOperator->pDownstreamGetParams;
7✔
882
      break;
7✔
883
    case OP_NOTIFY_PARAM:
×
884
      ppParam = &pOperator->pOperatorNotifyParam;
×
885
      pppDownstramParam = &pOperator->pDownstreamNotifyParams;
×
886
      break;
×
887
    default:
×
888
      return TSDB_CODE_INVALID_PARA;
×
889
  }
890

891
  freeResetOperatorParams(pOperator, type, false);
7✔
892

893
  if (NULL == pInput) {
7!
894
    return TSDB_CODE_SUCCESS;
×
895
  }
896

897
  *ppParam = (pInput->opType == pOperator->operatorType) ? pInput : NULL;
7!
898

899
  if (NULL == *pppDownstramParam) {
7✔
900
    *pppDownstramParam = taosMemoryCalloc(pOperator->numOfDownstream, POINTER_BYTES);
6!
901
    if (NULL == *pppDownstramParam) {
6!
902
      return terrno;
×
903
    }
904
  }
905

906
  if (NULL == *ppParam) {
7!
907
    for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
×
908
      (*pppDownstramParam)[i] = pInput;
×
909
    }
910
    return TSDB_CODE_SUCCESS;
×
911
  }
912

913
  memset(*pppDownstramParam, 0, pOperator->numOfDownstream * POINTER_BYTES);
7✔
914

915
  int32_t childrenNum = taosArrayGetSize((*ppParam)->pChildren);
7✔
916
  if (childrenNum <= 0) {
7✔
917
    return TSDB_CODE_SUCCESS;
5✔
918
  }
919

920
  for (int32_t i = 0; i < childrenNum; ++i) {
6✔
921
    SOperatorParam* pChild = *(SOperatorParam**)taosArrayGet((*ppParam)->pChildren, i);
4✔
922
    if (pChild == NULL) {
4!
923
      return terrno;
×
924
    }
925

926
    if ((*pppDownstramParam)[pChild->downstreamIdx]) {
4!
927
      int32_t code = mergeOperatorParams((*pppDownstramParam)[pChild->downstreamIdx], pChild);
×
928
      if (code) {
×
929
        return code;
×
930
      }
931
    } else {
932
      (*pppDownstramParam)[pChild->downstreamIdx] = pChild;
4✔
933
    }
934
  }
935

936
  taosArrayDestroy((*ppParam)->pChildren);
2✔
937
  (*ppParam)->pChildren = NULL;
2✔
938

939
  return TSDB_CODE_SUCCESS;
2✔
940
}
941

942
SSDataBlock* getNextBlockFromDownstream(struct SOperatorInfo* pOperator, int32_t idx) {
247,906✔
943
  SSDataBlock* p = NULL;
247,906✔
944
  int32_t      code = getNextBlockFromDownstreamImpl(pOperator, idx, true, &p);
247,906✔
945
  if (code == TSDB_CODE_SUCCESS) {
247,866!
946
    code = blockDataCheck(p);
247,872✔
947
    if (code != TSDB_CODE_SUCCESS) {
247,879!
948
      qError("blockDataCheck failed, code:%s", tstrerror(code));
×
949
    }
950
  }
951
  return (code == 0) ? p : NULL;
247,876!
952
}
953

954
SSDataBlock* getNextBlockFromDownstreamRemain(struct SOperatorInfo* pOperator, int32_t idx) {
3,061✔
955
  SSDataBlock* p = NULL;
3,061✔
956
  int32_t      code = getNextBlockFromDownstreamImpl(pOperator, idx, false, &p);
3,061✔
957
  if (code == TSDB_CODE_SUCCESS) {
3,061!
958
    code = blockDataCheck(p);
3,061✔
959
    if (code != TSDB_CODE_SUCCESS) {
3,061!
960
      qError("blockDataCheck failed, code:%s", tstrerror(code));
×
961
    }
962
  }
963
  return (code == 0)? p:NULL;
3,061!
964
}
965

966
int32_t optrDefaultGetNextExtFn(struct SOperatorInfo* pOperator, SOperatorParam* pParam, SSDataBlock** pRes) {
7✔
967
  QRY_PARAM_CHECK(pRes);
7!
968

969
  int32_t lino = 0;
7✔
970
  int32_t code = setOperatorParams(pOperator, pParam, OP_GET_PARAM);
7✔
971
  QUERY_CHECK_CODE(code, lino, _end);
7!
972
  code = pOperator->fpSet.getNextFn(pOperator, pRes);
7✔
973
  QUERY_CHECK_CODE(code, lino, _end);
7!
974

975
_end:
7✔
976
  if (code != TSDB_CODE_SUCCESS) {
7!
977
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
978
    pOperator->pTaskInfo->code = code;
×
979
    T_LONG_JMP(pOperator->pTaskInfo->env, code);
×
980
  }
981

982
  return code;
7✔
983
}
984

985
int32_t optrDefaultNotifyFn(struct SOperatorInfo* pOperator, SOperatorParam* pParam) {
×
986
  int32_t code = setOperatorParams(pOperator, pParam, OP_NOTIFY_PARAM);
×
987
  if (TSDB_CODE_SUCCESS == code && pOperator->fpSet.notifyFn && pOperator->pOperatorNotifyParam) {
×
988
    code = pOperator->fpSet.notifyFn(pOperator, pOperator->pOperatorNotifyParam);
×
989
  }
990
  if (TSDB_CODE_SUCCESS == code) {
×
991
    for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
×
992
      if (pOperator->pDownstreamNotifyParams[i]) {
×
993
        code = optrDefaultNotifyFn(pOperator->pDownstream[i], pOperator->pDownstreamNotifyParams[i]);
×
994
        if (TSDB_CODE_SUCCESS != code) {
×
995
          break;
×
996
        }
997
        pOperator->pDownstreamNotifyParams[i] = NULL;
×
998
      }
999
    }
1000
  }
1001
  if (TSDB_CODE_SUCCESS != code) {
×
1002
    pOperator->pTaskInfo->code = code;
×
1003
    T_LONG_JMP(pOperator->pTaskInfo->env, pOperator->pTaskInfo->code);
×
1004
  }
1005

1006
  return code;
×
1007
}
1008

1009
int16_t getOperatorResultBlockId(struct SOperatorInfo* pOperator, int32_t idx) {
66✔
1010
  if (pOperator->transparent) {
66✔
1011
    return getOperatorResultBlockId(pOperator->pDownstream[idx], 0);
2✔
1012
  }
1013
  return pOperator->resultDataBlockId;
64✔
1014
}
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