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

taosdata / TDengine / #4931

16 Jan 2026 02:32AM UTC coverage: 66.749% (+0.03%) from 66.716%
#4931

push

travis-ci

web-flow
enh: interp supports using non-null prev/next values to fill (#34236)

281 of 327 new or added lines in 11 files covered. (85.93%)

1890 existing lines in 121 files now uncovered.

203303 of 304580 relevant lines covered (66.75%)

129941648.02 hits per line

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

59.65
/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 "tname.h"
19

20
#include "tglobal.h"
21

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

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

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

48
  return fpSet;
819,153,908✔
49
}
50

51
void setOperatorStreamStateFn(SOperatorInfo* pOperator, __optr_state_fn_t relaseFn, __optr_state_fn_t reloadFn) {
136,820,442✔
52
  pOperator->fpSet.releaseStreamStateFn = relaseFn;
136,820,442✔
53
  pOperator->fpSet.reloadStreamStateFn = reloadFn;
136,918,258✔
54
}
136,847,604✔
55

56
void setOperatorResetStateFn(SOperatorInfo* pOperator, __optr_reset_state_fn_t resetFn) {
818,053,675✔
57
  pOperator->fpSet.resetStateFn = resetFn;
818,053,675✔
58
}
818,252,427✔
59

60
int32_t optrDummyOpenFn(SOperatorInfo* pOperator) {
66,620,772✔
61
  OPTR_SET_OPENED(pOperator);
66,620,772✔
62
  pOperator->cost.openCost = 0;
66,625,646✔
63
  return TSDB_CODE_SUCCESS;
66,617,919✔
64
}
65

66
int32_t appendDownstream(SOperatorInfo* p, SOperatorInfo** pDownstream, int32_t num) {
390,718,948✔
67
  p->pDownstream = taosMemoryCalloc(1, num * POINTER_BYTES);
390,718,948✔
68
  if (p->pDownstream == NULL) {
390,815,077✔
69
    return terrno;
×
70
  }
71

72
  memcpy(p->pDownstream, pDownstream, num * POINTER_BYTES);
390,653,860✔
73
  p->numOfDownstream = num;
390,828,624✔
74
  p->numOfRealDownstream = num;
390,777,311✔
75
  return TSDB_CODE_SUCCESS;
390,839,654✔
76
}
77

78
void setOperatorCompleted(SOperatorInfo* pOperator) {
950,461,933✔
79
  pOperator->status = OP_EXEC_DONE;
950,461,933✔
80
  pOperator->cost.totalCost = (taosGetTimestampUs() - pOperator->pTaskInfo->cost.start) / 1000.0;
950,686,581✔
81
  setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
950,558,490✔
82
}
950,525,247✔
83

84
void setOperatorInfo(SOperatorInfo* pOperator, const char* name, int32_t type, bool blocking, int32_t status,
819,050,143✔
85
                     void* pInfo, SExecTaskInfo* pTaskInfo) {
86
  pOperator->name = (char*)name;
819,050,143✔
87
  pOperator->operatorType = type;
819,289,468✔
88
  pOperator->blocking = blocking;
818,840,963✔
89
  pOperator->status = status;
819,042,397✔
90
  pOperator->info = pInfo;
819,197,262✔
91
  pOperator->pTaskInfo = pTaskInfo;
819,378,097✔
92
}
819,202,396✔
93

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

103
static int64_t getQuerySupportBufSize(size_t numOfTables) {
×
104
  size_t s1 = sizeof(STableQueryInfo);
×
105
  //  size_t s3 = sizeof(STableCheckInfo);  buffer consumption in tsdb
106
  return (int64_t)(s1 * 1.5 * numOfTables);
×
107
}
108

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

127
  // disable query processing if the value of tsQueryBufferSize is zero.
128
  return TSDB_CODE_QRY_NOT_ENOUGH_BUFFER;
×
129
}
130

131
void releaseQueryBuf(size_t numOfTables) {
×
132
  if (tsQueryBufferSizeBytes < 0) {
×
133
    return;
×
134
  }
135

136
  int64_t t = getQuerySupportBufSize(numOfTables);
×
137

138
  // restore value is not enough buffer available
139
  (void)atomic_add_fetch_64(&tsQueryBufferSizeBytes, t);
×
140
}
141

142
typedef enum {
143
  OPTR_FN_RET_CONTINUE = 0x1,
144
  OPTR_FN_RET_ABORT = 0x2,
145
} ERetType;
146

147
typedef struct STraverParam {
148
  void*   pRet;
149
  int32_t code;
150
  void*   pParam;
151
} STraverParam;
152

153
// iterate the operator tree helper
154
typedef ERetType (*optr_fn_t)(SOperatorInfo* pOperator, STraverParam* pParam, const char* pIdstr);
155

156
void traverseOperatorTree(SOperatorInfo* pOperator, optr_fn_t fn, STraverParam* pParam, const char* id) {
41,176,895✔
157
  if (pOperator == NULL) {
41,176,895✔
158
    return;
×
159
  }
160

161
  ERetType ret = fn(pOperator, pParam, id);
41,176,895✔
162
  if (ret == OPTR_FN_RET_ABORT || pParam->code != TSDB_CODE_SUCCESS) {
41,177,812✔
163
    return;
21,391,661✔
164
  }
165

166
  for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
39,565,796✔
167
    traverseOperatorTree(pOperator->pDownstream[i], fn, pParam, id);
19,778,063✔
168
    if (pParam->code != 0) {
19,777,031✔
169
      break;
×
170
    }
171
  }
172
}
173

174
ERetType extractOperatorInfo(SOperatorInfo* pOperator, STraverParam* pParam, const char* pIdStr) {
41,055,036✔
175
  STraverParam* p = pParam;
41,055,036✔
176
  if (pOperator->operatorType == *(int32_t*)p->pParam) {
41,055,036✔
177
    p->pRet = pOperator;
21,329,561✔
178
    return OPTR_FN_RET_ABORT;
21,329,527✔
179
  } else {
180
    return OPTR_FN_RET_CONTINUE;
19,727,479✔
181
  }
182
}
183

184
// QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN
185
int32_t extractOperatorInTree(SOperatorInfo* pOperator, int32_t type, const char* id, SOperatorInfo** pOptrInfo) {
21,330,974✔
186
  QRY_PARAM_CHECK(pOptrInfo);
21,330,974✔
187

188
  if (pOperator == NULL) {
21,331,423✔
189
    qError("invalid operator, failed to find tableScanOperator %s", id);
×
190
    return TSDB_CODE_PAR_INTERNAL_ERROR;
×
191
  }
192

193
  STraverParam p = {.pParam = &type, .pRet = NULL};
21,331,423✔
194
  traverseOperatorTree(pOperator, extractOperatorInfo, &p, id);
21,330,940✔
195
  if (p.code == 0) {
21,329,067✔
196
    *pOptrInfo = p.pRet;
21,328,834✔
197
  }
198
  return p.code;
21,329,165✔
199
}
200

201
typedef struct SExtScanInfo {
202
  int32_t order;
203
  int32_t scanFlag;
204
  int32_t inheritUsOrder;
205
} SExtScanInfo;
206

207
static ERetType extractScanInfo(SOperatorInfo* pOperator, STraverParam* pParam, const char* pIdStr) {
×
208
  int32_t       type = pOperator->operatorType;
×
209
  SExtScanInfo* pInfo = pParam->pParam;
×
210

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

238
int32_t getTableScanInfo(SOperatorInfo* pOperator, int32_t* order, int32_t* scanFlag, bool inheritUsOrder) {
×
239
  SExtScanInfo info = {.inheritUsOrder = inheritUsOrder, .order = *order};
×
240
  STraverParam p = {.pParam = &info};
×
241

242
  traverseOperatorTree(pOperator, extractScanInfo, &p, NULL);
×
243
  *order = info.order;
×
244
  *scanFlag = info.scanFlag;
×
245

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

255
static ERetType doStopDataReader(SOperatorInfo* pOperator, STraverParam* pParam, const char* pIdStr) {
123,901✔
256
  SStorageAPI* pAPI = pParam->pParam;
123,901✔
257
  if (pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
123,901✔
258
    STableScanInfo* pInfo = pOperator->info;
62,721✔
259

260
    if (pInfo->base.dataReader != NULL) {
62,721✔
261
      pAPI->tsdReader.tsdReaderNotifyClosing(pInfo->base.dataReader);
61,605✔
262
    }
263
    return OPTR_FN_RET_ABORT;
62,721✔
264
  } else if (pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN) {
61,180✔
265
    SStreamScanInfo* pInfo = pOperator->info;
×
266

267
    if (pInfo->pTableScanOp != NULL) {
×
268
      STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
×
269
      if (pTableScanInfo != NULL && pTableScanInfo->base.dataReader != NULL) {
×
270
        pAPI->tsdReader.tsdReaderNotifyClosing(pTableScanInfo->base.dataReader);
×
271
      }
272
    }
273

274
    return OPTR_FN_RET_ABORT;
×
275
  }
276

277
  return OPTR_FN_RET_CONTINUE;
61,180✔
278
}
279

280
int32_t stopTableScanOperator(SOperatorInfo* pOperator, const char* pIdStr, SStorageAPI* pAPI) {
72,011✔
281
  STraverParam p = {.pParam = pAPI};
72,011✔
282
  traverseOperatorTree(pOperator, doStopDataReader, &p, pIdStr);
72,011✔
283
  return p.code;
72,011✔
284
}
285

286
int32_t createOperator(SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo, SReadHandle* pHandle, SNode* pTagCond,
827,158,454✔
287
                       SNode* pTagIndexCond, const char* pUser, const char* dbname, SOperatorInfo** pOptrInfo,
288
                       EOPTR_EXEC_MODEL model) {
289
  QRY_PARAM_CHECK(pOptrInfo);
827,158,454✔
290

291
  int32_t     code = 0;
827,302,524✔
292
  int32_t     type = nodeType(pPhyNode);
827,302,524✔
293
  const char* idstr = GET_TASKID(pTaskInfo);
827,336,116✔
294

295
  if (pPhyNode->pChildren == NULL || LIST_LENGTH(pPhyNode->pChildren) == 0) {
827,364,265✔
296
    SOperatorInfo* pOperator = NULL;
428,296,915✔
297
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == type) {
428,095,204✔
298
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pPhyNode;
258,240,373✔
299
      // NOTE: this is an patch to fix the physical plan
300
      // TODO remove it later
301
      if (pTableScanNode->scan.node.pLimit != NULL) {
258,240,373✔
302
        pTableScanNode->groupSort = true;
4,417,734✔
303
      }
304

305
      STableListInfo* pTableListInfo = tableListCreate();
258,248,524✔
306
      if (!pTableListInfo) {
258,362,895✔
307
        pTaskInfo->code = terrno;
×
308
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
309
        return terrno;
×
310
      }
311

312
      // Since virtual stable scan use virtual super table's uid in scan operator, the origin table might be stored on
313
      // different vnode, so we should not get table schema for virtual stable scan.
314
      if (!pTableScanNode->scan.virtualStableScan) {
258,362,895✔
315
        code = initQueriedTableSchemaInfo(pHandle, &pTableScanNode->scan, dbname, pTaskInfo);
251,628,758✔
316
        if (code) {
251,333,421✔
317
          pTaskInfo->code = code;
×
318
          tableListDestroy(pTableListInfo);
×
319
          return code;
×
320
        }
321
      }
322

323
      if (pTableScanNode->scan.node.dynamicOp) {
258,061,868✔
324
        pTaskInfo->dynamicTask = true;
11,735,177✔
325
        pTableListInfo->idInfo.suid = pTableScanNode->scan.suid;
11,736,019✔
326
        pTableListInfo->idInfo.tableType = pTableScanNode->scan.tableType;
11,734,629✔
327
      } else {
328
        code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
246,326,043✔
329
                                       pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo, NULL);
330
        if (code) {
246,528,482✔
331
          pTaskInfo->code = code;
914✔
332
          tableListDestroy(pTableListInfo);
914✔
333
          qError("failed to createScanTableListInfo, code:%s, %s", tstrerror(code), idstr);
914✔
334
          return code;
914✔
335
        }
336
      }
337

338
      code = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
258,260,225✔
339
      if (NULL == pOperator || code != 0) {
258,107,111✔
340
        pTaskInfo->code = code;
5,932✔
341
        tableListDestroy(pTableListInfo);
22,776✔
342
        return code;
22,776✔
343
      }
344

345
      STableScanInfo* pScanInfo = pOperator->info;
258,104,736✔
346
      pTaskInfo->cost.pRecoder = &pScanInfo->base.readRecorder;
258,203,017✔
347
    } else if (QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN == type) {
169,854,831✔
348
      STableMergeScanPhysiNode* pTableScanNode = (STableMergeScanPhysiNode*)pPhyNode;
24,505,474✔
349
      STableListInfo*           pTableListInfo = tableListCreate();
24,505,474✔
350
      if (!pTableListInfo) {
24,525,716✔
351
        pTaskInfo->code = terrno;
×
352
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
353
        return terrno;
×
354
      }
355

356
      code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, true, pHandle, pTableListInfo,
24,525,716✔
357
                                     pTagCond, pTagIndexCond, pTaskInfo, NULL);
358
      if (code) {
24,522,960✔
359
        pTaskInfo->code = code;
×
360
        tableListDestroy(pTableListInfo);
×
361
        qError("failed to createScanTableListInfo, code:%s", tstrerror(code));
×
362
        return code;
×
363
      }
364

365
      code = initQueriedTableSchemaInfo(pHandle, &pTableScanNode->scan, dbname, pTaskInfo);
24,522,960✔
366
      if (code) {
24,481,104✔
367
        pTaskInfo->code = code;
×
368
        tableListDestroy(pTableListInfo);
×
369
        return code;
×
370
      }
371

372
      code = createTableMergeScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
24,481,104✔
373
      if (NULL == pOperator || code != 0) {
24,470,553✔
UNCOV
374
        pTaskInfo->code = code;
×
375
        tableListDestroy(pTableListInfo);
×
376
        return code;
×
377
      }
378

379
      STableMergeScanInfo* pScanInfo = pOperator->info;
24,486,463✔
380
      pTaskInfo->cost.pRecoder = &pScanInfo->base.readRecorder;
24,496,424✔
381
    } else if (QUERY_NODE_PHYSICAL_PLAN_EXCHANGE == type) {
145,349,357✔
382
      code = createExchangeOperatorInfo(pHandle ? pHandle->pMsgCb->clientRpc : NULL, (SExchangePhysiNode*)pPhyNode,
109,400,143✔
383
                                        pTaskInfo, &pOperator);
384
    } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN == type) {
35,949,214✔
385
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pPhyNode;
325,755✔
386
      STableListInfo*      pTableListInfo = tableListCreate();
325,755✔
387
      if (!pTableListInfo) {
325,821✔
388
        pTaskInfo->code = terrno;
×
389
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
390
        return terrno;
×
391
      }
392

393
      if (pHandle->vnode && (pTaskInfo->pSubplan->pVTables == NULL)) {
325,821✔
394
        code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
325,748✔
395
                                       pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo, NULL);
396
        if (code) {
326,074✔
397
          pTaskInfo->code = code;
×
398
          tableListDestroy(pTableListInfo);
×
399
          qError("failed to createScanTableListInfo, code:%s", tstrerror(code));
×
400
          return code;
×
401
        }
402
      }
403

404
      code = createTmqScanOperatorInfo(pHandle, pTableScanNode, pTagCond, pTableListInfo, pTaskInfo, &pOperator);
326,147✔
405
      if (code) {
326,074✔
406
        pTaskInfo->code = code;
×
407
        tableListDestroy(pTableListInfo);
×
408
        return code;
×
409
      }
410
    } else if (QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN == type) {
35,623,459✔
411
      SSystemTableScanPhysiNode* pSysScanPhyNode = (SSystemTableScanPhysiNode*)pPhyNode;
19,026,731✔
412
      if (pSysScanPhyNode->scan.virtualStableScan) {
19,026,731✔
413
        STableListInfo*           pTableListInfo = tableListCreate();
8,735,686✔
414
        if (!pTableListInfo) {
8,737,198✔
415
          pTaskInfo->code = terrno;
×
416
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
417
          return terrno;
×
418
        }
419

420
        code = createScanTableListInfo((SScanPhysiNode*)pSysScanPhyNode, NULL, false, pHandle, pTableListInfo, pTagCond,
8,737,198✔
421
                                       pTagIndexCond, pTaskInfo, NULL);
422
        if (code != TSDB_CODE_SUCCESS) {
8,733,698✔
423
          pTaskInfo->code = code;
×
424
          tableListDestroy(pTableListInfo);
×
425
          return code;
×
426
        }
427

428
        code = createSysTableScanOperatorInfo(pHandle, pSysScanPhyNode, pTableListInfo, pUser, pTaskInfo, &pOperator);
8,733,698✔
429
      } else {
430
        code = createSysTableScanOperatorInfo(pHandle, pSysScanPhyNode, NULL, pUser, pTaskInfo, &pOperator);
10,291,555✔
431
      }
432
    } else if (QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN == type) {
16,596,728✔
433
      STableCountScanPhysiNode* pTblCountScanNode = (STableCountScanPhysiNode*)pPhyNode;
25,567✔
434
      code = createTableCountScanOperatorInfo(pHandle, pTblCountScanNode, pTaskInfo, &pOperator);
25,567✔
435
    } else if (QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN == type) {
16,571,161✔
436
      STagScanPhysiNode* pTagScanPhyNode = (STagScanPhysiNode*)pPhyNode;
11,947,467✔
437
      STableListInfo*    pTableListInfo = tableListCreate();
11,947,467✔
438
      if (!pTableListInfo) {
11,950,036✔
439
        pTaskInfo->code = terrno;
×
440
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
441
        return terrno;
×
442
      }
443

444
      code = initQueriedTableSchemaInfo(pHandle, &pTagScanPhyNode->scan, dbname, pTaskInfo);
11,950,036✔
445
      if (code != TSDB_CODE_SUCCESS) {
11,945,206✔
446
        pTaskInfo->code = code;
×
447
        tableListDestroy(pTableListInfo);
×
448
        return code;
×
449
      }
450

451
      if (!pTagScanPhyNode->onlyMetaCtbIdx) {
11,945,206✔
452
        code = createScanTableListInfo((SScanPhysiNode*)pTagScanPhyNode, NULL, false, pHandle, pTableListInfo, pTagCond,
7,247,267✔
453
                                       pTagIndexCond, pTaskInfo, NULL);
454
        if (code != TSDB_CODE_SUCCESS) {
7,250,378✔
455
          pTaskInfo->code = code;
×
456
          qError("failed to getTableList, code:%s", tstrerror(code));
×
457
          tableListDestroy(pTableListInfo);
×
458
          return code;
×
459
        }
460
      }
461

462
      if (pTagScanPhyNode->scan.node.dynamicOp) {
11,948,317✔
463
        pTaskInfo->dynamicTask = true;
2,015,137✔
464
        pTableListInfo->idInfo.suid = pTagScanPhyNode->scan.suid;
2,015,137✔
465
        pTableListInfo->idInfo.tableType = pTagScanPhyNode->scan.tableType;
2,015,137✔
466
      }
467

468
      code = createTagScanOperatorInfo(pHandle, pTagScanPhyNode, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo,
11,947,289✔
469
                                       &pOperator);
470
      if (code) {
11,945,388✔
471
        pTaskInfo->code = code;
×
472
        tableListDestroy(pTableListInfo);
×
473
        return code;
×
474
      }
475
    } else if (QUERY_NODE_PHYSICAL_PLAN_BLOCK_DIST_SCAN == type) {
4,623,694✔
476
      SBlockDistScanPhysiNode* pBlockNode = (SBlockDistScanPhysiNode*)pPhyNode;
5,218✔
477
      STableListInfo*          pTableListInfo = tableListCreate();
5,218✔
478
      if (!pTableListInfo) {
5,218✔
479
        pTaskInfo->code = terrno;
×
480
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
481
        return terrno;
×
482
      }
483

484
      if (pBlockNode->tableType == TSDB_SUPER_TABLE) {
5,218✔
485
        SArray* pList = taosArrayInit(4, sizeof(uint64_t));
4,435✔
486
        code = pTaskInfo->storageAPI.metaFn.getChildTableList(pHandle->vnode, pBlockNode->uid, pList);
4,435✔
487
        if (code != TSDB_CODE_SUCCESS) {
4,435✔
488
          pTaskInfo->code = code;
×
489
          taosArrayDestroy(pList);
×
490
          tableListDestroy(pTableListInfo);
×
491
          return code;
×
492
        }
493

494
        size_t num = taosArrayGetSize(pList);
4,435✔
495
        for (int32_t i = 0; i < num; ++i) {
9,640✔
496
          uint64_t* id = taosArrayGet(pList, i);
5,205✔
497
          if (id == NULL) {
5,205✔
498
            continue;
×
499
          }
500

501
          code = tableListAddTableInfo(pTableListInfo, *id, 0);
5,205✔
502
          if (code) {
5,205✔
503
            pTaskInfo->code = code;
×
504
            tableListDestroy(pTableListInfo);
×
505
            taosArrayDestroy(pList);
×
506
            return code;
×
507
          }
508
        }
509

510
        taosArrayDestroy(pList);
4,435✔
511
      } else {  // Create group with only one table
512
        code = tableListAddTableInfo(pTableListInfo, pBlockNode->uid, 0);
783✔
513
        if (code) {
783✔
514
          pTaskInfo->code = code;
×
515
          tableListDestroy(pTableListInfo);
×
516
          return code;
×
517
        }
518
      }
519

520
      code = createDataBlockInfoScanOperator(pHandle, pBlockNode, pTableListInfo, pTaskInfo, &pOperator);
5,218✔
521
      if (code) {
5,218✔
522
        pTaskInfo->code = code;
×
523
        tableListDestroy(pTableListInfo);
×
524
        return code;
×
525
      }
526
    } else if (QUERY_NODE_PHYSICAL_PLAN_LAST_ROW_SCAN == type) {
4,618,476✔
527
      SLastRowScanPhysiNode* pScanNode = (SLastRowScanPhysiNode*)pPhyNode;
485,934✔
528
      STableListInfo*        pTableListInfo = tableListCreate();
485,934✔
529
      if (!pTableListInfo) {
486,338✔
530
        pTaskInfo->code = terrno;
×
531
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
532
        return terrno;
×
533
      }
534

535
      code = createScanTableListInfo(&pScanNode->scan, pScanNode->pGroupTags, true, pHandle, pTableListInfo, pTagCond,
486,338✔
536
                                     pTagIndexCond, pTaskInfo, NULL);
537
      if (code != TSDB_CODE_SUCCESS) {
486,338✔
538
        pTaskInfo->code = code;
×
539
        tableListDestroy(pTableListInfo);
×
540
        return code;
×
541
      }
542

543
      code = initQueriedTableSchemaInfo(pHandle, &pScanNode->scan, dbname, pTaskInfo);
486,338✔
544
      if (code != TSDB_CODE_SUCCESS) {
485,535✔
545
        pTaskInfo->code = code;
×
546
        tableListDestroy(pTableListInfo);
×
547
        return code;
×
548
      }
549

550
      code = createCacherowsScanOperator(pScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
485,535✔
551
      if (code) {
485,535✔
552
        tableListDestroy(pTableListInfo);
×
553
        pTaskInfo->code = code;
×
554
        return code;
×
555
      }
556
    } else if (QUERY_NODE_PHYSICAL_PLAN_PROJECT == type) {
4,132,542✔
557
      code = createProjectOperatorInfo(NULL, (SProjectPhysiNode*)pPhyNode, pTaskInfo, &pOperator);
4,119,042✔
558
    } else if (QUERY_NODE_PHYSICAL_PLAN_VIRTUAL_TABLE_SCAN == type) {
13,500✔
559
      // NOTE: this is an patch to fix the physical plan
560
      code = createVirtualTableMergeOperatorInfo(NULL, 0, (SVirtualScanPhysiNode*)pPhyNode, pTaskInfo, &pOperator);
14,737✔
561
    } else {
562
      code = TSDB_CODE_INVALID_PARA;
17✔
563
      pTaskInfo->code = code;
17✔
564
      return code;
×
565
    }
566

567
    if (pOperator != NULL) {  // todo moved away
427,561,229✔
568
      pOperator->resultDataBlockId = pPhyNode->pOutputDataBlockDesc->dataBlockId;
427,890,281✔
569
    }
570

571
    *pOptrInfo = pOperator;
427,612,435✔
572
    return code;
427,779,776✔
573
  }
574

575
  size_t          size = LIST_LENGTH(pPhyNode->pChildren);
398,994,013✔
576
  SOperatorInfo** ops = taosMemoryCalloc(size, POINTER_BYTES);
399,193,209✔
577
  if (ops == NULL) {
399,033,962✔
578
    code = terrno;
×
579
    pTaskInfo->code = code;
×
580
    return code;
×
581
  }
582

583
  for (int32_t i = 0; i < size; ++i) {
839,883,138✔
584
    SPhysiNode* pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, i);
442,947,532✔
585
    code = createOperator(pChildNode, pTaskInfo, pHandle, pTagCond, pTagIndexCond, pUser, dbname, &ops[i], model);
442,813,692✔
586
    if (ops[i] == NULL || code != 0) {
442,790,990✔
587
      for (int32_t j = 0; j < i; ++j) {
1,700,625✔
588
        destroyOperator(ops[j]);
×
589
      }
590
      taosMemoryFree(ops);
1,700,625✔
591
      return code;
1,700,625✔
592
    }
593
  }
594

595
  SOperatorInfo* pOptr = NULL;
396,935,606✔
596
  if (QUERY_NODE_PHYSICAL_PLAN_PROJECT == type) {
397,047,811✔
597
    code = createProjectOperatorInfo(ops[0], (SProjectPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
136,475,974✔
598
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_AGG == type) {
260,571,837✔
599
    SAggPhysiNode* pAggNode = (SAggPhysiNode*)pPhyNode;
132,202,729✔
600
    if (pAggNode->pGroupKeys != NULL) {
132,202,729✔
601
      code = createGroupOperatorInfo(ops[0], pAggNode, pTaskInfo, &pOptr);
47,348,898✔
602
    } else {
603
      code = createAggregateOperatorInfo(ops[0], pAggNode, pTaskInfo, &pOptr);
84,938,102✔
604
    }
605
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_INTERVAL == type) {
128,369,108✔
606
    SIntervalPhysiNode* pIntervalPhyNode = (SIntervalPhysiNode*)pPhyNode;
12,096,412✔
607
    code = createIntervalOperatorInfo(ops[0], pIntervalPhyNode, pTaskInfo, &pOptr);
12,096,412✔
608
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_ALIGNED_INTERVAL == type) {
116,272,696✔
609
    SMergeAlignedIntervalPhysiNode* pIntervalPhyNode = (SMergeAlignedIntervalPhysiNode*)pPhyNode;
661,042✔
610
    code = createMergeAlignedIntervalOperatorInfo(ops[0], pIntervalPhyNode, pTaskInfo, &pOptr);
661,042✔
611
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_INTERVAL == type) {
115,611,654✔
612
    SMergeIntervalPhysiNode* pIntervalPhyNode = (SMergeIntervalPhysiNode*)pPhyNode;
×
613
    code = createMergeIntervalOperatorInfo(ops[0], pIntervalPhyNode, pTaskInfo, &pOptr);
×
614
  } else if (QUERY_NODE_PHYSICAL_PLAN_SORT == type) {
115,611,654✔
615
    code = createSortOperatorInfo(ops[0], (SSortPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
50,911,879✔
616
  } else if (QUERY_NODE_PHYSICAL_PLAN_GROUP_SORT == type) {
64,699,775✔
617
    code = createGroupSortOperatorInfo(ops[0], (SGroupSortPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
43,199✔
618
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE == type) {
64,656,576✔
619
    SMergePhysiNode* pMergePhyNode = (SMergePhysiNode*)pPhyNode;
12,165,930✔
620
    code = createMultiwayMergeOperatorInfo(ops, size, pMergePhyNode, pTaskInfo, &pOptr);
12,165,930✔
621
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_SESSION == type) {
52,490,646✔
622
    SSessionWinodwPhysiNode* pSessionNode = (SSessionWinodwPhysiNode*)pPhyNode;
2,700,086✔
623
    code = createSessionAggOperatorInfo(ops[0], pSessionNode, pTaskInfo, &pOptr);
2,700,086✔
624
  } else if (QUERY_NODE_PHYSICAL_PLAN_PARTITION == type) {
49,790,560✔
625
    code = createPartitionOperatorInfo(ops[0], (SPartitionPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
10,090,075✔
626
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_STATE == type) {
39,700,485✔
627
    SStateWindowPhysiNode* pStateNode = (SStateWindowPhysiNode*)pPhyNode;
3,022,963✔
628
    code = createStatewindowOperatorInfo(ops[0], pStateNode, pTaskInfo, &pOptr);
3,022,963✔
629
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_JOIN == type) {
36,677,522✔
630
    code = createMergeJoinOperatorInfo(ops, size, (SSortMergeJoinPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
15,286,947✔
631
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_JOIN == type) {
21,390,575✔
632
    code = createHashJoinOperatorInfo(ops, size, (SHashJoinPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
1,151,930✔
633
  } else if (QUERY_NODE_PHYSICAL_PLAN_FILL == type) {
20,238,645✔
634
    code = createFillOperatorInfo(ops[0], (SFillPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
452,985✔
635
  } else if (QUERY_NODE_PHYSICAL_PLAN_INDEF_ROWS_FUNC == type) {
19,785,660✔
636
    code = createIndefinitOutputOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
2,902,891✔
637
  } else if (QUERY_NODE_PHYSICAL_PLAN_INTERP_FUNC == type) {
16,882,769✔
638
    code = createTimeSliceOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
3,256,992✔
639
  } else if (QUERY_NODE_PHYSICAL_PLAN_FORECAST_FUNC == type) {
13,625,777✔
640
    code = createForecastOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
×
641
  } else if (QUERY_NODE_PHYSICAL_PLAN_ANALYSIS_FUNC == type) {
13,625,777✔
642
    code = createGenericAnalysisOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
×
643
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_EVENT == type) {
13,625,777✔
644
    code = createEventwindowOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
1,904,854✔
645
  } else if (QUERY_NODE_PHYSICAL_PLAN_GROUP_CACHE == type) {
11,720,923✔
646
    code = createGroupCacheOperatorInfo(ops, size, (SGroupCachePhysiNode*)pPhyNode, pTaskInfo, &pOptr);
1,151,182✔
647
  } else if (QUERY_NODE_PHYSICAL_PLAN_DYN_QUERY_CTRL == type) {
10,569,741✔
648
    code = createDynQueryCtrlOperatorInfo(ops, size, (SDynQueryCtrlPhysiNode*)pPhyNode, pTaskInfo, pHandle->pMsgCb, &pOptr);
3,895,153✔
649
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_COUNT == type) {
6,674,588✔
650
    code = createCountwindowOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
1,892,186✔
651
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_ANOMALY == type) {
4,782,402✔
652
    code = createAnomalywindowOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
×
653
  } else if (QUERY_NODE_PHYSICAL_PLAN_VIRTUAL_TABLE_SCAN == type) {
4,782,402✔
654
    SVirtualScanPhysiNode* pVirtualTableScanNode = (SVirtualScanPhysiNode*)pPhyNode;
4,265,888✔
655
    // NOTE: this is an patch to fix the physical plan
656

657
    if (pVirtualTableScanNode->scan.node.pLimit != NULL) {
4,265,888✔
658
      pVirtualTableScanNode->groupSort = true;
×
659
    }
660
    code = createVirtualTableMergeOperatorInfo(ops, size, (SVirtualScanPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
4,265,888✔
661
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_EXTERNAL == type) {
516,514✔
662
    code = createExternalWindowOperator(ops[0], pPhyNode, pTaskInfo, &pOptr);
540,026✔
663
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_ALIGNED_EXTERNAL == type) {
×
664
    code = createMergeAlignedExternalWindowOperator(ops[0], pPhyNode, pTaskInfo, &pOptr);
2,181✔
665
  } else {
666
    code = TSDB_CODE_INVALID_PARA;
×
667
    pTaskInfo->code = code;
×
668
    for (int32_t i = 0; i < size; ++i) {
×
669
      destroyOperator(ops[i]);
×
670
    }
671
    taosMemoryFree(ops);
×
672
    qError("invalid operator type %d", type);
×
673
    return code;
×
674
  }
675

676
  taosMemoryFree(ops);
397,345,655✔
677
  if (pOptr) {
397,149,447✔
678
    pOptr->resultDataBlockId = pPhyNode->pOutputDataBlockDesc->dataBlockId;
390,625,017✔
679
  }
680

681
  *pOptrInfo = pOptr;
397,288,784✔
682
  return code;
397,365,186✔
683
}
684

685
void destroyOperator(SOperatorInfo* pOperator) {
832,574,722✔
686
  if (pOperator == NULL) {
832,574,722✔
687
    return;
6,552,666✔
688
  }
689

690
  freeResetOperatorParams(pOperator, OP_GET_PARAM, true);
826,022,056✔
691
  freeResetOperatorParams(pOperator, OP_NOTIFY_PARAM, true);
825,923,348✔
692

693
  if (pOperator->pDownstream != NULL) {
825,943,495✔
694
    for (int32_t i = 0; i < pOperator->numOfRealDownstream; ++i) {
825,598,643✔
695
      destroyOperator(pOperator->pDownstream[i]);
434,827,559✔
696
    }
697

698
    taosMemoryFreeClear(pOperator->pDownstream);
390,948,719✔
699
    pOperator->numOfDownstream = 0;
390,778,791✔
700
  }
701

702
  if (pOperator->fpSet.closeFn != NULL && pOperator->info != NULL) {
825,888,471✔
703
    pOperator->fpSet.closeFn(pOperator->info);
819,542,641✔
704
    pOperator->info = NULL;
819,220,839✔
705
  }
706

707
  cleanupExprSupp(&pOperator->exprSupp);
825,783,660✔
708
  taosMemoryFreeClear(pOperator);
825,924,004✔
709
}
710

711
void destroyOperatorAndDownstreams(SOperatorInfo* pOperator, SOperatorInfo** downstreams, int32_t num) {
6,528,976✔
712
  if (downstreams != NULL) {
6,528,976✔
713
    for (int i = 0; i < num; i++) {
13,057,404✔
714
      destroyOperator(downstreams[i]);
6,528,976✔
715
    }
716
  }
717

718
  if (pOperator != NULL) {
6,528,428✔
719
    pOperator->info = NULL;
6,528,428✔
720
    if (pOperator->pDownstream != NULL) {
6,528,428✔
721
      taosMemoryFreeClear(pOperator->pDownstream);
×
722
      pOperator->pDownstream = NULL;
×
723
    }
724
    destroyOperator(pOperator);
6,528,702✔
725
  }
726
}
6,527,668✔
727

728
int32_t getOperatorExplainExecInfo(SOperatorInfo* operatorInfo, SArray* pExecInfoList) {
31,199,732✔
729
  SExplainExecInfo  execInfo = {0};
31,199,732✔
730
  SExplainExecInfo* pExplainInfo = taosArrayPush(pExecInfoList, &execInfo);
31,280,039✔
731
  if (pExplainInfo == NULL) {
31,280,039✔
732
    return terrno;
×
733
  }
734

735
  pExplainInfo->numOfRows = operatorInfo->resultInfo.totalRows;
31,280,039✔
736
  pExplainInfo->startupCost = operatorInfo->cost.openCost;
31,288,234✔
737
  pExplainInfo->totalCost = operatorInfo->cost.totalCost;
31,302,980✔
738
  pExplainInfo->verboseLen = 0;
31,262,895✔
739
  pExplainInfo->verboseInfo = NULL;
31,228,446✔
740

741
  if (operatorInfo->fpSet.getExplainFn) {
31,296,622✔
742
    int32_t code =
743
        operatorInfo->fpSet.getExplainFn(operatorInfo, &pExplainInfo->verboseInfo, &pExplainInfo->verboseLen);
14,219,418✔
744
    if (code) {
14,224,487✔
745
      qError("%s operator getExplainFn failed, code:%s", GET_TASKID(operatorInfo->pTaskInfo), tstrerror(code));
×
746
      return code;
×
747
    }
748
  }
749

750
  int32_t code = 0;
31,260,996✔
751
  for (int32_t i = 0; i < operatorInfo->numOfDownstream; ++i) {
47,783,024✔
752
    code = getOperatorExplainExecInfo(operatorInfo->pDownstream[i], pExecInfoList);
16,535,099✔
753
    if (code != TSDB_CODE_SUCCESS) {
16,522,028✔
754
      //      taosMemoryFreeClear(*pRes);
755
      return code;
×
756
    }
757
  }
758

759
  return TSDB_CODE_SUCCESS;
31,210,543✔
760
}
761

762
int32_t mergeOperatorParams(SOperatorParam* pDst, SOperatorParam* pSrc) {
×
763
  if (pDst->opType != pSrc->opType) {
×
764
    qError("different optype %d:%d for merge operator params", pDst->opType, pSrc->opType);
×
765
    return TSDB_CODE_INVALID_PARA;
×
766
  }
767

768
  switch (pDst->opType) {
×
769
    case QUERY_NODE_PHYSICAL_PLAN_EXCHANGE: {
×
770
      SExchangeOperatorParam* pDExc = pDst->value;
×
771
      SExchangeOperatorParam* pSExc = pSrc->value;
×
NEW
772
      if (pSExc->basic.paramType != DYN_TYPE_EXCHANGE_PARAM) {
×
NEW
773
        qError("%s, invalid exchange operator param type %d for "
×
774
          "source operator", __func__, pSExc->basic.paramType);
NEW
775
        return TSDB_CODE_INVALID_PARA;
×
776
      }
777
      if (!pDExc->multiParams) {
×
NEW
778
        if (pDExc->basic.paramType != DYN_TYPE_EXCHANGE_PARAM) {
×
NEW
779
          qError("%s, invalid exchange operator param type %d for "
×
780
            "destination operator", __func__, pDExc->basic.paramType);
NEW
781
          return TSDB_CODE_INVALID_PARA;
×
782
        }
UNCOV
783
        if (pSExc->basic.vgId != pDExc->basic.vgId) {
×
UNCOV
784
          SExchangeOperatorBatchParam* pBatch = taosMemoryMalloc(sizeof(SExchangeOperatorBatchParam));
×
785
          if (NULL == pBatch) {
×
UNCOV
786
            return terrno;
×
787
          }
788

789
          pBatch->multiParams = true;
×
UNCOV
790
          pBatch->pBatchs = tSimpleHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
×
791
          if (NULL == pBatch->pBatchs) {
×
UNCOV
792
            taosMemoryFree(pBatch);
×
793
            return terrno;
×
794
          }
795

796
          tSimpleHashSetFreeFp(pBatch->pBatchs, freeExchangeGetBasicOperatorParam);
×
797

UNCOV
798
          int32_t code = tSimpleHashPut(pBatch->pBatchs, &pDExc->basic.vgId, sizeof(pDExc->basic.vgId), &pDExc->basic,
×
799
                                        sizeof(pDExc->basic));
800
          if (code) {
×
801
            return code;
×
802
          }
803

UNCOV
804
          code = tSimpleHashPut(pBatch->pBatchs, &pSExc->basic.vgId, sizeof(pSExc->basic.vgId), &pSExc->basic,
×
805
                                sizeof(pSExc->basic));
806
          if (code) {
×
UNCOV
807
            return code;
×
808
          }
809

810
          taosMemoryFree(pDst->value);
×
811
          pDst->value = pBatch;
×
812
        } else {
UNCOV
813
          void* p = taosArrayAddAll(pDExc->basic.uidList, pSExc->basic.uidList);
×
814
          if (p == NULL) {
×
UNCOV
815
            return terrno;
×
816
          }
817
        }
818
      } else {
UNCOV
819
        SExchangeOperatorBatchParam* pBatch = pDst->value;
×
820
        SExchangeOperatorBasicParam* pBasic =
821
            tSimpleHashGet(pBatch->pBatchs, &pSExc->basic.vgId, sizeof(pSExc->basic.vgId));
×
UNCOV
822
        if (pBasic) {
×
823
          void* p = taosArrayAddAll(pBasic->uidList, pSExc->basic.uidList);
×
824
          if (p == NULL) {
×
825
            return terrno;
×
826
          }
827
        } else {
UNCOV
828
          int32_t code = tSimpleHashPut(pBatch->pBatchs, &pSExc->basic.vgId, sizeof(pSExc->basic.vgId), &pSExc->basic,
×
829
                                        sizeof(pSExc->basic));
UNCOV
830
          if (code) {
×
831
            return code;
×
832
          }
833
        }
834
      }
835
      break;
×
836
    }
UNCOV
837
    default:
×
838
      qError("invalid optype %d for merge operator params", pDst->opType);
×
UNCOV
839
      return TSDB_CODE_INVALID_PARA;
×
840
  }
841

UNCOV
842
  return TSDB_CODE_SUCCESS;
×
843
}
844

845
int32_t setOperatorParams(struct SOperatorInfo* pOperator, SOperatorParam* pInput, SOperatorParamType type) {
254,025,838✔
846
  SOperatorParam**  ppParam = NULL;
254,025,838✔
847
  SOperatorParam*** pppDownstramParam = NULL;
254,025,838✔
848
  switch (type) {
254,025,838✔
849
    case OP_GET_PARAM:
254,026,320✔
850
      ppParam = &pOperator->pOperatorGetParam;
254,026,320✔
851
      pppDownstramParam = &pOperator->pDownstreamGetParams;
254,028,318✔
852
      break;
254,028,828✔
UNCOV
853
    case OP_NOTIFY_PARAM:
×
UNCOV
854
      ppParam = &pOperator->pOperatorNotifyParam;
×
UNCOV
855
      pppDownstramParam = &pOperator->pDownstreamNotifyParams;
×
UNCOV
856
      break;
×
857
    default:
14✔
858
      return TSDB_CODE_INVALID_PARA;
14✔
859
  }
860

861
  freeResetOperatorParams(pOperator, type, false);
254,028,828✔
862

863
  if (NULL == pInput) {
254,056,654✔
864
    return TSDB_CODE_SUCCESS;
×
865
  }
866

867
  *ppParam = (pInput->opType == pOperator->operatorType) ? pInput : NULL;
254,056,654✔
868

869
  if (NULL == *pppDownstramParam) {
254,055,662✔
870
    *pppDownstramParam = taosMemoryCalloc(pOperator->numOfDownstream, POINTER_BYTES);
19,726,887✔
871
    if (NULL == *pppDownstramParam) {
19,726,887✔
UNCOV
872
      return terrno;
×
873
    }
874
  }
875

876
  if (NULL == *ppParam) {
254,056,668✔
UNCOV
877
    for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
×
UNCOV
878
      (*pppDownstramParam)[i] = pInput;
×
879
    }
UNCOV
880
    return TSDB_CODE_SUCCESS;
×
881
  }
882

883
  memset(*pppDownstramParam, 0, pOperator->numOfDownstream * POINTER_BYTES);
254,057,660✔
884

885
  int32_t childrenNum = taosArrayGetSize((*ppParam)->pChildren);
254,054,160✔
886
  if (childrenNum <= 0) {
254,050,632✔
887
    return TSDB_CODE_SUCCESS;
184,463,471✔
888
  }
889

890
  for (int32_t i = 0; i < childrenNum; ++i) {
169,933,748✔
891
    SOperatorParam* pChild = *(SOperatorParam**)taosArrayGet((*ppParam)->pChildren, i);
100,349,081✔
892
    if (pChild == NULL) {
100,346,077✔
UNCOV
893
      return terrno;
×
894
    }
895

896
    if ((*pppDownstramParam)[pChild->downstreamIdx]) {
100,346,077✔
UNCOV
897
      int32_t code = mergeOperatorParams((*pppDownstramParam)[pChild->downstreamIdx], pChild);
×
UNCOV
898
      if (code) {
×
UNCOV
899
        return code;
×
900
      }
901
    } else {
902
      (*pppDownstramParam)[pChild->downstreamIdx] = pChild;
100,344,051✔
903
    }
904
  }
905

906
  taosArrayDestroy((*ppParam)->pChildren);
69,584,667✔
907
  (*ppParam)->pChildren = NULL;
69,582,585✔
908

909
  return TSDB_CODE_SUCCESS;
69,584,115✔
910
}
911

912
SSDataBlock* getNextBlockFromDownstream(struct SOperatorInfo* pOperator, int32_t idx) {
1,288,125,439✔
913
  SSDataBlock* p = NULL;
1,288,125,439✔
914
  int32_t      code = getNextBlockFromDownstreamImpl(pOperator, idx, true, &p);
1,288,251,073✔
915
  if (code == TSDB_CODE_SUCCESS) {
1,284,915,170✔
916
    code = blockDataCheck(p);
1,285,059,198✔
917
    if (code != TSDB_CODE_SUCCESS) {
1,285,264,778✔
UNCOV
918
      qError("blockDataCheck failed, code:%s", tstrerror(code));
×
919
    }
920
  }
921
  return (code == 0) ? p : NULL;
1,285,059,957✔
922
}
923

924
SSDataBlock* getNextBlockFromDownstreamRemain(struct SOperatorInfo* pOperator, int32_t idx) {
733,953,616✔
925
  SSDataBlock* p = NULL;
733,953,616✔
926
  int32_t      code = getNextBlockFromDownstreamImpl(pOperator, idx, false, &p);
733,972,083✔
927
  if (code == TSDB_CODE_SUCCESS) {
731,715,512✔
928
    code = blockDataCheck(p);
731,742,726✔
929
    if (code != TSDB_CODE_SUCCESS) {
731,739,401✔
UNCOV
930
      qError("blockDataCheck failed, code:%s", tstrerror(code));
×
931
    }
932
  }
933
  return (code == 0)? p:NULL;
731,757,182✔
934
}
935

936
int32_t optrDefaultGetNextExtFn(struct SOperatorInfo* pOperator, SOperatorParam* pParam, SSDataBlock** pRes) {
254,056,640✔
937
  QRY_PARAM_CHECK(pRes);
254,056,640✔
938

939
  int32_t lino = 0;
254,058,666✔
940
  int32_t code = setOperatorParams(pOperator, pParam, OP_GET_PARAM);
254,058,666✔
941
  QUERY_CHECK_CODE(code, lino, _end);
254,043,080✔
942
  code = pOperator->fpSet.getNextFn(pOperator, pRes);
254,043,080✔
943
  QUERY_CHECK_CODE(code, lino, _end);
254,053,955✔
944

945
_end:
254,053,955✔
946
  if (code != TSDB_CODE_SUCCESS) {
254,053,955✔
UNCOV
947
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
UNCOV
948
    pOperator->pTaskInfo->code = code;
×
UNCOV
949
    T_LONG_JMP(pOperator->pTaskInfo->env, code);
×
950
  }
951

952
  return code;
254,053,955✔
953
}
954

UNCOV
955
int32_t optrDefaultNotifyFn(struct SOperatorInfo* pOperator, SOperatorParam* pParam) {
×
UNCOV
956
  int32_t code = setOperatorParams(pOperator, pParam, OP_NOTIFY_PARAM);
×
957
  if (TSDB_CODE_SUCCESS == code && pOperator->fpSet.notifyFn && pOperator->pOperatorNotifyParam) {
×
958
    code = pOperator->fpSet.notifyFn(pOperator, pOperator->pOperatorNotifyParam);
×
959
  }
UNCOV
960
  if (TSDB_CODE_SUCCESS == code) {
×
UNCOV
961
    for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
×
UNCOV
962
      if (pOperator->pDownstreamNotifyParams[i]) {
×
UNCOV
963
        code = optrDefaultNotifyFn(pOperator->pDownstream[i], pOperator->pDownstreamNotifyParams[i]);
×
UNCOV
964
        if (TSDB_CODE_SUCCESS != code) {
×
965
          break;
×
966
        }
967
        pOperator->pDownstreamNotifyParams[i] = NULL;
×
968
      }
969
    }
970
  }
971
  if (TSDB_CODE_SUCCESS != code) {
×
972
    pOperator->pTaskInfo->code = code;
×
973
    T_LONG_JMP(pOperator->pTaskInfo->env, pOperator->pTaskInfo->code);
×
974
  }
975

UNCOV
976
  return code;
×
977
}
978

979
int16_t getOperatorResultBlockId(struct SOperatorInfo* pOperator, int32_t idx) {
33,273,430✔
980
  if (pOperator->transparent) {
33,273,430✔
981
    return getOperatorResultBlockId(pOperator->pDownstream[idx], 0);
2,302,364✔
982
  }
983
  return pOperator->resultDataBlockId;
30,972,035✔
984
}
985

986
void resetOperatorState(SOperatorInfo *pOper) {
×
UNCOV
987
  pOper->status = OP_NOT_OPENED;
×
UNCOV
988
}
×
989

990
void resetBasicOperatorState(SOptrBasicInfo *pBasicInfo) {
15,857,017✔
991
  if (pBasicInfo->pRes) blockDataCleanup(pBasicInfo->pRes);
15,857,017✔
992
  initResultRowInfo(&pBasicInfo->resultRowInfo);
15,858,531✔
993
}
15,857,949✔
994

995
int32_t resetAggSup(SExprSupp* pExprSupp, SAggSupporter* pSup, SExecTaskInfo* pTaskInfo,
13,436,685✔
996
                    SNodeList* pNodeList, SNodeList* pGroupKeys, size_t keyBufSize, const char* pKey, void* pState,
997
                    SFunctionStateStore* pStore) {
998
  int32_t    code = 0, lino = 0, num = 0;
13,436,685✔
999
  SExprInfo* pExprInfo = NULL;
13,436,685✔
1000
  cleanupAggSup(pSup);
13,436,901✔
1001
  cleanupExprSuppWithoutFilter(pExprSupp);
13,435,133✔
1002
  code = createExprInfo(pNodeList, pGroupKeys, &pExprInfo, &num);
13,435,941✔
1003
  QUERY_CHECK_CODE(code, lino, _error);
13,435,996✔
1004
  code = initAggSup(pExprSupp, pSup, pExprInfo, num, keyBufSize, pKey, pState, pStore);
13,435,996✔
1005
  QUERY_CHECK_CODE(code, lino, _error);
13,434,304✔
1006
  return code;
13,434,304✔
UNCOV
1007
_error:
×
UNCOV
1008
  if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
1009
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
UNCOV
1010
    pTaskInfo->code = code;
×
1011
  }
UNCOV
1012
  return code;
×
1013
}
1014

1015
int32_t resetExprSupp(SExprSupp* pExprSupp, SExecTaskInfo* pTaskInfo, SNodeList* pNodeList,
5,591,752✔
1016
                      SNodeList* pGroupKeys, SFunctionStateStore* pStore) {
1017
  int32_t code = 0, lino = 0, num = 0;
5,591,752✔
1018
  SExprInfo* pExprInfo = NULL;
5,591,752✔
1019
  cleanupExprSuppWithoutFilter(pExprSupp);
5,591,009✔
1020
  code = createExprInfo(pNodeList, pGroupKeys, &pExprInfo, &num);
5,591,534✔
1021
  QUERY_CHECK_CODE(code, lino, _error);
5,590,505✔
1022
  code = initExprSupp(pExprSupp, pExprInfo, num, pStore);
5,590,505✔
1023
  QUERY_CHECK_CODE(code, lino, _error);
5,589,978✔
1024
  return code;
5,589,978✔
UNCOV
1025
_error:
×
UNCOV
1026
  if (code != TSDB_CODE_SUCCESS) {
×
UNCOV
1027
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
UNCOV
1028
    pTaskInfo->code = code;
×
1029
  }
UNCOV
1030
  return code;
×
1031
}
1032

1033
int32_t copyColumnsValue(SNodeList* pNodeList, uint64_t targetBlkId, SSDataBlock* pDst, SSDataBlock* pSrc, int32_t totalRows) {
58,646,487✔
1034
  bool    isNull = (NULL == pSrc || pSrc->info.rows <= 0);
58,646,487✔
1035
  size_t  numOfCols = LIST_LENGTH(pNodeList);
58,646,487✔
1036
  int64_t numOfRows = isNull ? 0 : pSrc->info.rows;
58,646,487✔
1037
  int32_t code = 0;
58,646,487✔
1038
  int32_t lino = 0;
58,646,487✔
1039

1040
  for (int32_t i = 0; i < numOfCols; ++i) {
492,921,095✔
1041
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
434,274,608✔
1042
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pExpr)->dataBlockId == targetBlkId) {
434,274,608✔
1043
      SColumnInfoData* pDstCol = taosArrayGet(pDst->pDataBlock, pNode->slotId);
203,444,503✔
1044
      QUERY_CHECK_NULL(pDstCol, code, lino, _return, terrno)
203,444,503✔
1045

1046
      if (isNull) {
203,444,503✔
1047
        colDataSetNNULL(pDstCol, 0, totalRows);
9,604,626✔
1048
      } else {
1049
        SColumnInfoData* pSrcCol = taosArrayGet(pSrc->pDataBlock, ((SColumnNode*)pNode->pExpr)->slotId);
193,839,877✔
1050
        QUERY_CHECK_NULL(pSrcCol, code, lino, _return, terrno)
193,839,877✔
1051

1052
        code = colDataAssign(pDstCol, pSrcCol, pSrc->info.rows, &pDst->info);
193,839,877✔
1053

1054
        QUERY_CHECK_CODE(code, lino, _return);
193,839,877✔
1055
        if (pSrc->info.rows < totalRows) {
193,839,877✔
1056
          colDataSetNNULL(pDstCol, pSrc->info.rows, totalRows - pSrc->info.rows);
3,438✔
1057
        }
1058
      }
1059
    }
1060
  }
1061

1062
  return code;
58,646,487✔
UNCOV
1063
_return:
×
UNCOV
1064
  qError("failed to copy columns value, line:%d code:%s", lino, tstrerror(code));
×
UNCOV
1065
  return code;
×
1066
}
1067

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