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

taosdata / TDengine / #4985

15 Mar 2026 07:43AM UTC coverage: 68.601% (-0.04%) from 68.643%
#4985

push

travis-ci

web-flow
feat(stream): add natural time units support for PERIOD trigger (#34766)

Implement week/month/year units for stream PERIOD trigger with natural
boundary alignment and offset support.

Key changes:
- Parser: Add validation for natural time units (w/n/y) and offset parameter
- Time utilities: Add getDuration() support for week/month/year units
- TriggerTask: Implement window calculation with natural boundary alignment
  - Week: align to Monday 00:00:00
  - Month: align to 1st of month 00:00:00
  - Year: align to Jan 1st 00:00:00
- Add offset support: PERIOD(1w, 1d) shifts window by 1 day
- Unit tests: Parser validation, time utilities, TriggerTask window calculation
- System tests: End-to-end tests for week/month/year units with offset
- Documentation: Update user manual with natural time unit examples

Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>

2 of 60 new or added lines in 2 files covered. (3.33%)

448 existing lines in 114 files now uncovered.

212624 of 309941 relevant lines covered (68.6%)

136450774.73 hits per line

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

61.38
/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,
748,835,464✔
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 = {
748,835,464✔
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;
748,835,464✔
49
}
50

51
void setOperatorStreamStateFn(SOperatorInfo* pOperator, __optr_state_fn_t relaseFn, __optr_state_fn_t reloadFn) {
141,406,088✔
52
  pOperator->fpSet.releaseStreamStateFn = relaseFn;
141,406,088✔
53
  pOperator->fpSet.reloadStreamStateFn = reloadFn;
141,427,084✔
54
}
141,409,838✔
55

56
void setOperatorResetStateFn(SOperatorInfo* pOperator, __optr_reset_state_fn_t resetFn) {
747,199,862✔
57
  pOperator->fpSet.resetStateFn = resetFn;
747,199,862✔
58
}
747,279,158✔
59

60
int32_t optrDummyOpenFn(SOperatorInfo* pOperator) {
76,921,255✔
61
  OPTR_SET_OPENED(pOperator);
76,921,255✔
62
  pOperator->cost.openCost = 0;
76,917,023✔
63
  return TSDB_CODE_SUCCESS;
76,922,807✔
64
}
65

66
int32_t appendDownstream(SOperatorInfo* p, SOperatorInfo** pDownstream, int32_t num) {
350,336,211✔
67
  p->pDownstream = taosMemoryCalloc(1, num * POINTER_BYTES);
350,336,211✔
68
  if (p->pDownstream == NULL) {
350,336,972✔
69
    return terrno;
×
70
  }
71

72
  memcpy(p->pDownstream, pDownstream, num * POINTER_BYTES);
350,240,076✔
73
  p->numOfDownstream = num;
350,345,844✔
74
  p->numOfRealDownstream = num;
350,304,781✔
75
  return TSDB_CODE_SUCCESS;
350,341,951✔
76
}
77

78
void setOperatorCompleted(SOperatorInfo* pOperator) {
741,625,572✔
79
  pOperator->status = OP_EXEC_DONE;
741,625,572✔
80
  pOperator->cost.totalCost = (taosGetTimestampUs() - pOperator->pTaskInfo->cost.start) / 1000.0;
741,634,365✔
81
  setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
741,624,826✔
82
}
741,627,799✔
83

84
void setOperatorInfo(SOperatorInfo* pOperator, const char* name, int32_t type, bool blocking, int32_t status,
748,791,302✔
85
                     void* pInfo, SExecTaskInfo* pTaskInfo) {
86
  pOperator->name = (char*)name;
748,791,302✔
87
  pOperator->operatorType = type;
748,879,959✔
88
  pOperator->blocking = blocking;
748,683,547✔
89
  pOperator->status = status;
748,775,892✔
90
  pOperator->info = pInfo;
748,792,456✔
91
  pOperator->pTaskInfo = pTaskInfo;
748,919,222✔
92
}
748,899,093✔
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) {
54,112,075✔
157
  if (pOperator == NULL) {
54,112,075✔
158
    return;
×
159
  }
160

161
  ERetType ret = fn(pOperator, pParam, id);
54,112,075✔
162
  if (ret == OPTR_FN_RET_ABORT || pParam->code != TSDB_CODE_SUCCESS) {
54,108,967✔
163
    return;
27,987,550✔
164
  }
165

166
  for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
52,244,112✔
167
    traverseOperatorTree(pOperator->pDownstream[i], fn, pParam, id);
26,119,662✔
168
    if (pParam->code != 0) {
26,117,898✔
169
      break;
×
170
    }
171
  }
172
}
173

174
ERetType extractOperatorInfo(SOperatorInfo* pOperator, STraverParam* pParam, const char* pIdStr) {
53,940,649✔
175
  STraverParam* p = pParam;
53,940,649✔
176
  if (pOperator->operatorType == *(int32_t*)p->pParam) {
53,940,649✔
177
    p->pRet = pOperator;
27,893,941✔
178
    return OPTR_FN_RET_ABORT;
27,895,013✔
179
  } else {
180
    return OPTR_FN_RET_CONTINUE;
26,045,685✔
181
  }
182
}
183

184
// QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN
185
int32_t extractOperatorInTree(SOperatorInfo* pOperator, int32_t type, const char* id, SOperatorInfo** pOptrInfo) {
27,895,012✔
186
  QRY_PARAM_CHECK(pOptrInfo);
27,895,012✔
187

188
  if (pOperator == NULL) {
27,895,374✔
189
    qError("invalid operator, failed to find tableScanOperator %s", id);
×
190
    return TSDB_CODE_STREAM_INTERNAL_ERROR;
×
191
  }
192

193
  STraverParam p = {.pParam = &type, .pRet = NULL};
27,895,374✔
194
  traverseOperatorTree(pOperator, extractOperatorInfo, &p, id);
27,895,374✔
195
  if (p.code == 0) {
27,893,704✔
196
    *pOptrInfo = p.pRet;
27,894,692✔
197
  }
198
  return p.code;
27,894,715✔
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) {
171,397✔
256
  SStorageAPI* pAPI = pParam->pParam;
171,397✔
257
  if (pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
171,397✔
258
    STableScanInfo* pInfo = pOperator->info;
93,940✔
259

260
    if (pInfo->base.dataReader != NULL) {
93,940✔
261
      pAPI->tsdReader.tsdReaderNotifyClosing(pInfo->base.dataReader);
92,278✔
262
    }
263
    return OPTR_FN_RET_ABORT;
93,940✔
264
  } else if (pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN) {
77,457✔
265
    STmqQueryScanInfo* 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;
77,457✔
278
}
279

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

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

291
  int32_t     code = 0;
749,084,577✔
292
  int32_t     type = nodeType(pPhyNode);
749,084,577✔
293
  const char* idstr = GET_TASKID(pTaskInfo);
749,084,111✔
294

295
  if (pPhyNode->pChildren == NULL || LIST_LENGTH(pPhyNode->pChildren) == 0) {
749,079,194✔
296
    SOperatorInfo* pOperator = NULL;
398,158,804✔
297
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == type) {
398,137,778✔
298
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pPhyNode;
234,108,607✔
299
      // NOTE: this is an patch to fix the physical plan
300
      // TODO remove it later
301
      if (pTableScanNode->scan.node.pLimit != NULL) {
234,108,607✔
302
        pTableScanNode->groupSort = true;
4,990,505✔
303
      }
304

305
      STableListInfo* pTableListInfo = tableListCreate();
234,064,861✔
306
      if (!pTableListInfo) {
234,127,760✔
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) {
234,127,760✔
315
        code = initQueriedTableSchemaInfo(pHandle, &pTableScanNode->scan, dbname, pTaskInfo);
228,751,970✔
316
        if (code) {
228,633,849✔
UNCOV
317
          pTaskInfo->code = code;
×
UNCOV
318
          tableListDestroy(pTableListInfo);
×
UNCOV
319
          return code;
×
320
        }
321
      }
322

323
      if (pTableScanNode->scan.node.dynamicOp) {
234,005,186✔
324
        pTaskInfo->dynamicTask = true;
10,866,740✔
325
        pTableListInfo->idInfo.suid = pTableScanNode->scan.suid;
10,869,273✔
326
        pTableListInfo->idInfo.tableType = pTableScanNode->scan.tableType;
10,866,863✔
327
      } else {
328
        code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
223,148,163✔
329
                                       pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo, NULL);
330
        if (code) {
223,248,092✔
331
          pTaskInfo->code = code;
940✔
332
          tableListDestroy(pTableListInfo);
940✔
333
          qError("failed to createScanTableListInfo, code:%s, %s", tstrerror(code), idstr);
940✔
334
          return code;
940✔
335
        }
336
      }
337

338
      code = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
234,114,699✔
339
      if (NULL == pOperator || code != 0) {
234,057,799✔
340
        pTaskInfo->code = code;
105,546✔
341
        tableListDestroy(pTableListInfo);
23,594✔
342
        return code;
23,594✔
343
      }
344

345
      STableScanInfo* pScanInfo = pOperator->info;
233,952,253✔
346
      pTaskInfo->cost.pRecoder = &pScanInfo->base.readRecorder;
234,043,759✔
347
    } else if (QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN == type) {
164,029,171✔
348
      STableMergeScanPhysiNode* pTableScanNode = (STableMergeScanPhysiNode*)pPhyNode;
27,264,804✔
349
      STableListInfo*           pTableListInfo = tableListCreate();
27,264,804✔
350
      if (!pTableListInfo) {
27,284,403✔
351
        pTaskInfo->code = terrno;
×
352
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
353
        return terrno;
×
354
      }
355

356
      if (pTableScanNode->scan.node.dynamicOp) {
27,284,403✔
357
        pTaskInfo->dynamicTask = true;
95,424✔
358
        pTableListInfo->idInfo.suid = pTableScanNode->scan.suid;
95,424✔
359
        pTableListInfo->idInfo.tableType = pTableScanNode->scan.tableType;
95,424✔
360
      } else {
361
        code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, true, pHandle, pTableListInfo,
27,188,813✔
362
                                       pTagCond, pTagIndexCond, pTaskInfo, NULL);
363
        if (code) {
27,182,750✔
364
          pTaskInfo->code = code;
×
365
          tableListDestroy(pTableListInfo);
×
366
          qError("failed to createScanTableListInfo, code:%s, %s", tstrerror(code), idstr);
×
367
          return code;
×
368
        }
369

370
        code = initQueriedTableSchemaInfo(pHandle, &pTableScanNode->scan, dbname, pTaskInfo);
27,182,750✔
371
        if (code) {
27,161,593✔
372
          pTaskInfo->code = code;
×
373
          tableListDestroy(pTableListInfo);
×
374
          return code;
×
375
        }
376
      }
377

378
      code = createTableMergeScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
27,257,017✔
379
      if (NULL == pOperator || code != 0) {
27,237,138✔
380
        pTaskInfo->code = code;
12,306✔
381
        tableListDestroy(pTableListInfo);
×
382
        return code;
×
383
      }
384

385
      STableMergeScanInfo* pScanInfo = pOperator->info;
27,224,832✔
386
      pTaskInfo->cost.pRecoder = &pScanInfo->base.readRecorder;
27,221,076✔
387
    } else if (QUERY_NODE_PHYSICAL_PLAN_EXCHANGE == type) {
136,764,367✔
388
      code = createExchangeOperatorInfo(pHandle ? pHandle->pMsgCb->clientRpc : NULL, (SExchangePhysiNode*)pPhyNode,
108,181,912✔
389
                                        pTaskInfo, &pOperator);
390
    } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN == type) {
28,582,455✔
391
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pPhyNode;
334,737✔
392
      STableListInfo*      pTableListInfo = tableListCreate();
334,737✔
393
      if (!pTableListInfo) {
334,737✔
394
        pTaskInfo->code = terrno;
×
395
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
396
        return terrno;
×
397
      }
398

399
      if (pHandle->vnode && (pTaskInfo->pSubplan->pVTables == NULL)) {
334,737✔
400
        code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
334,737✔
401
                                       pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo, NULL);
402
        if (code) {
334,737✔
403
          pTaskInfo->code = code;
×
404
          tableListDestroy(pTableListInfo);
×
405
          qError("failed to createScanTableListInfo, code:%s", tstrerror(code));
×
406
          return code;
×
407
        }
408
      }
409

410
      code = createTmqScanOperatorInfo(pHandle, pTableScanNode, pTagCond, pTableListInfo, pTaskInfo, &pOperator);
334,737✔
411
      if (code) {
334,737✔
412
        pTaskInfo->code = code;
×
413
        tableListDestroy(pTableListInfo);
×
414
        return code;
×
415
      }
416
    } else if (QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN == type) {
28,247,718✔
417
      SSystemTableScanPhysiNode* pSysScanPhyNode = (SSystemTableScanPhysiNode*)pPhyNode;
16,838,446✔
418
      if (pSysScanPhyNode->scan.virtualStableScan) {
16,838,446✔
419
        STableListInfo*           pTableListInfo = tableListCreate();
6,062,052✔
420
        if (!pTableListInfo) {
6,062,052✔
421
          pTaskInfo->code = terrno;
×
422
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
423
          return terrno;
×
424
        }
425

426
        code = createScanTableListInfo((SScanPhysiNode*)pSysScanPhyNode, NULL, false, pHandle, pTableListInfo, pTagCond,
6,062,052✔
427
                                       pTagIndexCond, pTaskInfo, NULL);
428
        if (code != TSDB_CODE_SUCCESS) {
6,062,052✔
429
          pTaskInfo->code = code;
×
430
          tableListDestroy(pTableListInfo);
×
431
          return code;
×
432
        }
433

434
        code = createSysTableScanOperatorInfo(pHandle, pSysScanPhyNode, pTableListInfo, pUser, pTaskInfo, &pOperator);
6,062,052✔
435
      } else {
436
        code = createSysTableScanOperatorInfo(pHandle, pSysScanPhyNode, NULL, pUser, pTaskInfo, &pOperator);
10,776,394✔
437
      }
438
    } else if (QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN == type) {
11,409,272✔
439
      STableCountScanPhysiNode* pTblCountScanNode = (STableCountScanPhysiNode*)pPhyNode;
29,883✔
440
      code = createTableCountScanOperatorInfo(pHandle, pTblCountScanNode, pTaskInfo, &pOperator);
29,883✔
441
    } else if (QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN == type) {
11,379,389✔
442
      STagScanPhysiNode* pTagScanPhyNode = (STagScanPhysiNode*)pPhyNode;
8,451,239✔
443
      STableListInfo*    pTableListInfo = tableListCreate();
8,451,239✔
444
      if (!pTableListInfo) {
8,454,891✔
445
        pTaskInfo->code = terrno;
×
446
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
447
        return terrno;
×
448
      }
449

450
      code = initQueriedTableSchemaInfo(pHandle, &pTagScanPhyNode->scan, dbname, pTaskInfo);
8,454,891✔
451
      if (code != TSDB_CODE_SUCCESS) {
8,448,126✔
452
        pTaskInfo->code = code;
×
453
        tableListDestroy(pTableListInfo);
×
454
        return code;
×
455
      }
456

457
      if (!pTagScanPhyNode->onlyMetaCtbIdx) {
8,448,126✔
458
        code = createScanTableListInfo((SScanPhysiNode*)pTagScanPhyNode, NULL, false, pHandle, pTableListInfo, pTagCond,
3,636,011✔
459
                                       pTagIndexCond, pTaskInfo, NULL);
460
        if (code != TSDB_CODE_SUCCESS) {
3,640,718✔
461
          pTaskInfo->code = code;
×
462
          qError("failed to getTableList, code:%s", tstrerror(code));
×
463
          tableListDestroy(pTableListInfo);
×
464
          return code;
×
465
        }
466
      }
467

468
      if (pTagScanPhyNode->scan.node.dynamicOp) {
8,455,516✔
469
        pTaskInfo->dynamicTask = true;
2,034,145✔
470
        pTableListInfo->idInfo.suid = pTagScanPhyNode->scan.suid;
2,034,145✔
471
        pTableListInfo->idInfo.tableType = pTagScanPhyNode->scan.tableType;
2,034,145✔
472
      }
473

474
      code = createTagScanOperatorInfo(pHandle, pTagScanPhyNode, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo,
8,451,398✔
475
                                       &pOperator);
476
      if (code) {
8,450,070✔
477
        pTaskInfo->code = code;
×
478
        tableListDestroy(pTableListInfo);
×
479
        return code;
×
480
      }
481
    } else if (QUERY_NODE_PHYSICAL_PLAN_BLOCK_DIST_SCAN == type) {
2,928,150✔
482
      SBlockDistScanPhysiNode* pBlockNode = (SBlockDistScanPhysiNode*)pPhyNode;
5,314✔
483
      STableListInfo*          pTableListInfo = tableListCreate();
5,314✔
484
      if (!pTableListInfo) {
5,314✔
485
        pTaskInfo->code = terrno;
×
486
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
487
        return terrno;
×
488
      }
489

490
      if (pBlockNode->tableType == TSDB_SUPER_TABLE) {
5,314✔
491
        SArray* pList = taosArrayInit(4, sizeof(uint64_t));
4,526✔
492
        code = pTaskInfo->storageAPI.metaFn.getChildTableList(pHandle->vnode, pBlockNode->uid, pList);
4,526✔
493
        if (code != TSDB_CODE_SUCCESS) {
4,526✔
494
          pTaskInfo->code = code;
×
495
          taosArrayDestroy(pList);
×
496
          tableListDestroy(pTableListInfo);
×
497
          return code;
×
498
        }
499

500
        size_t num = taosArrayGetSize(pList);
4,526✔
501
        for (int32_t i = 0; i < num; ++i) {
9,740✔
502
          uint64_t* id = taosArrayGet(pList, i);
5,214✔
503
          if (id == NULL) {
5,214✔
504
            continue;
×
505
          }
506

507
          code = tableListAddTableInfo(pTableListInfo, *id, 0);
5,214✔
508
          if (code) {
5,214✔
509
            pTaskInfo->code = code;
×
510
            tableListDestroy(pTableListInfo);
×
511
            taosArrayDestroy(pList);
×
512
            return code;
×
513
          }
514
        }
515

516
        taosArrayDestroy(pList);
4,526✔
517
      } else {  // Create group with only one table
518
        code = tableListAddTableInfo(pTableListInfo, pBlockNode->uid, 0);
788✔
519
        if (code) {
788✔
520
          pTaskInfo->code = code;
×
521
          tableListDestroy(pTableListInfo);
×
522
          return code;
×
523
        }
524
      }
525

526
      code = createDataBlockInfoScanOperator(pHandle, pBlockNode, pTableListInfo, pTaskInfo, &pOperator);
5,314✔
527
      if (code) {
5,314✔
528
        pTaskInfo->code = code;
×
529
        tableListDestroy(pTableListInfo);
×
530
        return code;
×
531
      }
532
    } else if (QUERY_NODE_PHYSICAL_PLAN_LAST_ROW_SCAN == type) {
2,922,836✔
533
      SLastRowScanPhysiNode* pScanNode = (SLastRowScanPhysiNode*)pPhyNode;
1,124,576✔
534
      STableListInfo*        pTableListInfo = tableListCreate();
1,124,576✔
535
      if (!pTableListInfo) {
1,124,576✔
536
        pTaskInfo->code = terrno;
×
537
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
538
        return terrno;
×
539
      }
540

541
      code = createScanTableListInfo(&pScanNode->scan, pScanNode->pGroupTags, true, pHandle, pTableListInfo, pTagCond,
1,124,576✔
542
                                     pTagIndexCond, pTaskInfo, NULL);
543
      if (code != TSDB_CODE_SUCCESS) {
1,124,576✔
544
        pTaskInfo->code = code;
×
545
        tableListDestroy(pTableListInfo);
×
546
        return code;
×
547
      }
548

549
      code = initQueriedTableSchemaInfo(pHandle, &pScanNode->scan, dbname, pTaskInfo);
1,124,576✔
550
      if (code != TSDB_CODE_SUCCESS) {
1,124,161✔
551
        pTaskInfo->code = code;
×
552
        tableListDestroy(pTableListInfo);
×
553
        return code;
×
554
      }
555

556
      code = createCacherowsScanOperator(pScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
1,124,161✔
557
      if (code) {
1,124,576✔
558
        tableListDestroy(pTableListInfo);
×
559
        pTaskInfo->code = code;
×
560
        return code;
×
561
      }
562
    } else if (QUERY_NODE_PHYSICAL_PLAN_PROJECT == type) {
1,798,260✔
563
      code = createProjectOperatorInfo(NULL, (SProjectPhysiNode*)pPhyNode, pTaskInfo, &pOperator);
1,781,425✔
564
    } else if (QUERY_NODE_PHYSICAL_PLAN_VIRTUAL_TABLE_SCAN == type) {
16,835✔
565
      // NOTE: this is an patch to fix the physical plan
566
      code = createVirtualTableMergeOperatorInfo(NULL, 0, (SVirtualScanPhysiNode*)pPhyNode, pTaskInfo, &pOperator);
17,109✔
567
    } else {
568
      code = TSDB_CODE_INVALID_PARA;
148✔
569
      pTaskInfo->code = code;
148✔
570
      return code;
×
571
    }
572

573
    if (pOperator != NULL) {  // todo moved away
397,845,817✔
574
      pOperator->resultDataBlockId = pPhyNode->pOutputDataBlockDesc->dataBlockId;
398,037,590✔
575
    }
576

577
    *pOptrInfo = pOperator;
397,821,628✔
578
    return code;
397,976,848✔
579
  }
580

581
  size_t          size = LIST_LENGTH(pPhyNode->pChildren);
350,880,483✔
582
  SOperatorInfo** ops = taosMemoryCalloc(size, POINTER_BYTES);
350,918,939✔
583
  if (ops == NULL) {
350,893,029✔
584
    code = terrno;
×
585
    pTaskInfo->code = code;
×
586
    return code;
×
587
  }
588

589
  for (int32_t i = 0; i < size; ++i) {
750,356,492✔
590
    SPhysiNode* pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, i);
399,577,333✔
591
    code = createOperator(pChildNode, pTaskInfo, pHandle, pTagCond, pTagIndexCond, pUser, dbname, &ops[i], model);
399,575,860✔
592
    if (ops[i] == NULL || code != 0) {
399,499,588✔
593
      for (int32_t j = 0; j < i; ++j) {
71,925✔
594
        destroyOperator(ops[j]);
×
595
      }
596
      taosMemoryFree(ops);
71,925✔
597
      return code;
71,925✔
598
    }
599
  }
600

601
  SOperatorInfo* pOptr = NULL;
350,779,159✔
602
  if (QUERY_NODE_PHYSICAL_PLAN_PROJECT == type) {
350,819,148✔
603
    code = createProjectOperatorInfo(ops[0], (SProjectPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
139,942,510✔
604
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_AGG == type) {
210,876,638✔
605
    SAggPhysiNode* pAggNode = (SAggPhysiNode*)pPhyNode;
123,635,282✔
606
    if (pAggNode->pGroupKeys != NULL) {
123,635,282✔
607
      code = createGroupOperatorInfo(ops[0], pAggNode, pTaskInfo, &pOptr);
31,680,094✔
608
    } else {
609
      code = createAggregateOperatorInfo(ops[0], pAggNode, pTaskInfo, &pOptr);
91,948,252✔
610
    }
611
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_INTERVAL == type) {
87,241,356✔
612
    SIntervalPhysiNode* pIntervalPhyNode = (SIntervalPhysiNode*)pPhyNode;
4,440,758✔
613
    code = createIntervalOperatorInfo(ops[0], pIntervalPhyNode, pTaskInfo, &pOptr);
4,440,758✔
614
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_ALIGNED_INTERVAL == type) {
82,800,598✔
615
    SMergeAlignedIntervalPhysiNode* pIntervalPhyNode = (SMergeAlignedIntervalPhysiNode*)pPhyNode;
759,439✔
616
    code = createMergeAlignedIntervalOperatorInfo(ops[0], pIntervalPhyNode, pTaskInfo, &pOptr);
759,439✔
617
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_INTERVAL == type) {
82,041,159✔
618
    SMergeIntervalPhysiNode* pIntervalPhyNode = (SMergeIntervalPhysiNode*)pPhyNode;
×
619
    code = createMergeIntervalOperatorInfo(ops[0], pIntervalPhyNode, pTaskInfo, &pOptr);
×
620
  } else if (QUERY_NODE_PHYSICAL_PLAN_SORT == type) {
82,041,159✔
621
    code = createSortOperatorInfo(ops[0], (SSortPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
32,038,987✔
622
  } else if (QUERY_NODE_PHYSICAL_PLAN_GROUP_SORT == type) {
50,002,172✔
623
    code = createGroupSortOperatorInfo(ops[0], (SGroupSortPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
88,618✔
624
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE == type) {
49,913,554✔
625
    SMergePhysiNode* pMergePhyNode = (SMergePhysiNode*)pPhyNode;
13,405,464✔
626
    code = createMultiwayMergeOperatorInfo(ops, size, pMergePhyNode, pTaskInfo, &pOptr);
13,405,464✔
627
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_SESSION == type) {
36,508,090✔
628
    SSessionWinodwPhysiNode* pSessionNode = (SSessionWinodwPhysiNode*)pPhyNode;
564,923✔
629
    code = createSessionAggOperatorInfo(ops[0], pSessionNode, pTaskInfo, &pOptr);
564,923✔
630
  } else if (QUERY_NODE_PHYSICAL_PLAN_PARTITION == type) {
35,943,167✔
631
    code = createPartitionOperatorInfo(ops[0], (SPartitionPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
3,307,285✔
632
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_STATE == type) {
32,635,882✔
633
    SStateWindowPhysiNode* pStateNode = (SStateWindowPhysiNode*)pPhyNode;
1,018,469✔
634
    code = createStatewindowOperatorInfo(ops[0], pStateNode, pTaskInfo, &pOptr);
1,018,469✔
635
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_JOIN == type) {
31,617,413✔
636
    code = createMergeJoinOperatorInfo(ops, size, (SSortMergeJoinPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
12,616,129✔
637
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_JOIN == type) {
19,001,284✔
638
    code = createHashJoinOperatorInfo(ops, size, (SHashJoinPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
1,186,001✔
639
  } else if (QUERY_NODE_PHYSICAL_PLAN_FILL == type) {
17,815,283✔
640
    code = createFillOperatorInfo(ops[0], (SFillPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
489,526✔
641
  } else if (QUERY_NODE_PHYSICAL_PLAN_INDEF_ROWS_FUNC == type) {
17,325,757✔
642
    code = createIndefinitOutputOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
3,020,987✔
643
  } else if (QUERY_NODE_PHYSICAL_PLAN_INTERP_FUNC == type) {
14,304,770✔
644
    code = createTimeSliceOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
3,372,090✔
645
  } else if (QUERY_NODE_PHYSICAL_PLAN_FORECAST_FUNC == type) {
10,932,680✔
646
    code = createForecastOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
×
647
  } else if (QUERY_NODE_PHYSICAL_PLAN_ANALYSIS_FUNC == type) {
10,932,680✔
648
    code = createGenericAnalysisOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
×
649
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_EVENT == type) {
10,932,680✔
650
    code = createEventwindowOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
347,039✔
651
  } else if (QUERY_NODE_PHYSICAL_PLAN_GROUP_CACHE == type) {
10,585,641✔
652
    code = createGroupCacheOperatorInfo(ops, size, (SGroupCachePhysiNode*)pPhyNode, pTaskInfo, &pOptr);
1,185,219✔
653
  } else if (QUERY_NODE_PHYSICAL_PLAN_DYN_QUERY_CTRL == type) {
9,400,422✔
654
    code = createDynQueryCtrlOperatorInfo(ops, size, (SDynQueryCtrlPhysiNode*)pPhyNode, pTaskInfo, pHandle->pMsgCb, &pOptr);
4,016,349✔
655
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_COUNT == type) {
5,384,073✔
656
    code = createCountwindowOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
277,397✔
657
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_ANOMALY == type) {
5,106,676✔
658
    code = createAnomalywindowOperatorInfo(ops[0], pPhyNode, pTaskInfo, &pOptr);
×
659
  } else if (QUERY_NODE_PHYSICAL_PLAN_VIRTUAL_TABLE_SCAN == type) {
5,106,676✔
660
    SVirtualScanPhysiNode* pVirtualTableScanNode = (SVirtualScanPhysiNode*)pPhyNode;
3,932,463✔
661
    // NOTE: this is an patch to fix the physical plan
662

663
    if (pVirtualTableScanNode->scan.node.pLimit != NULL) {
3,932,463✔
664
      pVirtualTableScanNode->groupSort = true;
×
665
    }
666
    code = createVirtualTableMergeOperatorInfo(ops, size, (SVirtualScanPhysiNode*)pPhyNode, pTaskInfo, &pOptr);
3,932,463✔
667
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_EXTERNAL == type) {
1,174,213✔
668
    code = createExternalWindowOperator(ops[0], pPhyNode, pTaskInfo, &pOptr);
1,151,567✔
669
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_ALIGNED_EXTERNAL == type) {
22,646✔
670
    code = createMergeAlignedExternalWindowOperator(ops[0], pPhyNode, pTaskInfo, &pOptr);
19,069✔
671
  } else {
672
    code = TSDB_CODE_INVALID_PARA;
3,577✔
673
    pTaskInfo->code = code;
3,577✔
674
    for (int32_t i = 0; i < size; ++i) {
×
675
      destroyOperator(ops[i]);
×
676
    }
677
    taosMemoryFree(ops);
×
678
    qError("invalid operator type %d", type);
×
679
    return code;
×
680
  }
681

682
  taosMemoryFree(ops);
350,769,627✔
683
  if (pOptr) {
350,709,901✔
684
    pOptr->resultDataBlockId = pPhyNode->pOutputDataBlockDesc->dataBlockId;
350,283,276✔
685
  }
686

687
  *pOptrInfo = pOptr;
350,709,808✔
688
  return code;
350,735,453✔
689
}
690

691
void destroyOperator(SOperatorInfo* pOperator) {
749,957,590✔
692
  if (pOperator == NULL) {
749,957,590✔
693
    return;
462,744✔
694
  }
695

696
  freeResetOperatorParams(pOperator, OP_GET_PARAM, true);
749,494,846✔
697
  freeResetOperatorParams(pOperator, OP_NOTIFY_PARAM, true);
749,470,202✔
698

699
  if (pOperator->pDownstream != NULL) {
749,467,025✔
700
    for (int32_t i = 0; i < pOperator->numOfRealDownstream; ++i) {
749,498,905✔
701
      destroyOperator(pOperator->pDownstream[i]);
399,103,446✔
702
    }
703

704
    taosMemoryFreeClear(pOperator->pDownstream);
350,410,066✔
705
    pOperator->numOfDownstream = 0;
350,391,460✔
706
  }
707

708
  cleanupExprSupp(&pOperator->exprSupp);
749,442,599✔
709
  if (pOperator->fpSet.closeFn != NULL && pOperator->info != NULL) {
749,451,718✔
710
    pOperator->fpSet.closeFn(pOperator->info);
749,017,749✔
711
    pOperator->info = NULL;
748,930,832✔
712
  }
713

714
  taosMemoryFreeClear(pOperator);
749,398,206✔
715
}
716

717
void destroyOperatorAndDownstreams(SOperatorInfo* pOperator, SOperatorInfo** downstreams, int32_t num) {
438,210✔
718
  if (downstreams != NULL) {
438,210✔
719
    for (int i = 0; i < num; i++) {
876,420✔
720
      destroyOperator(downstreams[i]);
438,210✔
721
    }
722
  }
723

724
  if (pOperator != NULL) {
438,210✔
725
    pOperator->info = NULL;
438,210✔
726
    if (pOperator->pDownstream != NULL) {
438,210✔
727
      taosMemoryFreeClear(pOperator->pDownstream);
×
728
      pOperator->pDownstream = NULL;
×
729
    }
730
    destroyOperator(pOperator);
438,210✔
731
  }
732
}
438,210✔
733

734
int32_t getOperatorExplainExecInfo(SOperatorInfo* operatorInfo, SArray* pExecInfoList) {
8,850,478✔
735
  SExplainExecInfo  execInfo = {0};
8,850,478✔
736
  SExplainExecInfo* pExplainInfo = taosArrayPush(pExecInfoList, &execInfo);
8,851,631✔
737
  if (pExplainInfo == NULL) {
8,851,631✔
738
    return terrno;
×
739
  }
740

741
  pExplainInfo->numOfRows = operatorInfo->resultInfo.totalRows;
8,851,631✔
742
  pExplainInfo->startupCost = operatorInfo->cost.openCost;
8,850,478✔
743
  pExplainInfo->totalCost = operatorInfo->cost.totalCost;
8,850,860✔
744
  pExplainInfo->verboseLen = 0;
8,850,478✔
745
  pExplainInfo->verboseInfo = NULL;
8,850,860✔
746

747
  if (operatorInfo->fpSet.getExplainFn) {
8,850,478✔
748
    int32_t code =
749
        operatorInfo->fpSet.getExplainFn(operatorInfo, &pExplainInfo->verboseInfo, &pExplainInfo->verboseLen);
3,695,325✔
750
    if (code) {
3,696,485✔
751
      qError("%s operator getExplainFn failed, code:%s", GET_TASKID(operatorInfo->pTaskInfo), tstrerror(code));
×
752
      return code;
×
753
    }
754
  }
755

756
  int32_t code = 0;
8,850,478✔
757
  for (int32_t i = 0; i < operatorInfo->numOfDownstream; ++i) {
13,858,820✔
758
    code = getOperatorExplainExecInfo(operatorInfo->pDownstream[i], pExecInfoList);
5,008,342✔
759
    if (code != TSDB_CODE_SUCCESS) {
5,008,342✔
760
      //      taosMemoryFreeClear(*pRes);
761
      return code;
×
762
    }
763
  }
764

765
  return TSDB_CODE_SUCCESS;
8,850,478✔
766
}
767

768
int32_t mergeOperatorParams(SOperatorParam* pDst, SOperatorParam* pSrc) {
×
769
  if (pDst->opType != pSrc->opType) {
×
770
    qError("different optype %d:%d for merge operator params", pDst->opType, pSrc->opType);
×
771
    return TSDB_CODE_INVALID_PARA;
×
772
  }
773

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

795
          pBatch->multiParams = true;
×
796
          pBatch->pBatchs = tSimpleHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
×
797
          if (NULL == pBatch->pBatchs) {
×
798
            taosMemoryFree(pBatch);
×
799
            return terrno;
×
800
          }
801

802
          tSimpleHashSetFreeFp(pBatch->pBatchs, freeExchangeGetBasicOperatorParam);
×
803

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

810
          code = tSimpleHashPut(pBatch->pBatchs, &pSExc->basic.vgId, sizeof(pSExc->basic.vgId), &pSExc->basic,
×
811
                                sizeof(pSExc->basic));
812
          if (code) {
×
813
            return code;
×
814
          }
815

816
          taosMemoryFree(pDst->value);
×
817
          pDst->value = pBatch;
×
818
        } else {
819
          void* p = taosArrayAddAll(pDExc->basic.uidList, pSExc->basic.uidList);
×
820
          if (p == NULL) {
×
821
            return terrno;
×
822
          }
823
        }
824
      } else {
825
        SExchangeOperatorBatchParam* pBatch = pDst->value;
×
826
        SExchangeOperatorBasicParam* pBasic =
827
            tSimpleHashGet(pBatch->pBatchs, &pSExc->basic.vgId, sizeof(pSExc->basic.vgId));
×
828
        if (pBasic) {
×
829
          void* p = taosArrayAddAll(pBasic->uidList, pSExc->basic.uidList);
×
830
          if (p == NULL) {
×
831
            return terrno;
×
832
          }
833
        } else {
834
          int32_t 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
      }
841
      break;
×
842
    }
843
    default:
×
844
      qError("invalid optype %d for merge operator params", pDst->opType);
×
845
      return TSDB_CODE_INVALID_PARA;
×
846
  }
847

848
  return TSDB_CODE_SUCCESS;
×
849
}
850

851
int32_t setOperatorParams(struct SOperatorInfo* pOperator, SOperatorParam* pInput, SOperatorParamType type) {
91,990,222✔
852
  SOperatorParam**  ppParam = NULL;
91,990,222✔
853
  SOperatorParam*** pppDownstramParam = NULL;
91,990,222✔
854
  switch (type) {
91,990,222✔
855
    case OP_GET_PARAM:
91,990,752✔
856
      ppParam = &pOperator->pOperatorGetParam;
91,990,752✔
857
      pppDownstramParam = &pOperator->pDownstreamGetParams;
91,991,282✔
858
      break;
91,991,823✔
859
    case OP_NOTIFY_PARAM:
×
860
      ppParam = &pOperator->pOperatorNotifyParam;
×
861
      pppDownstramParam = &pOperator->pDownstreamNotifyParams;
×
862
      break;
×
863
    default:
×
864
      return TSDB_CODE_INVALID_PARA;
×
865
  }
866

867
  freeResetOperatorParams(pOperator, type, false);
91,991,823✔
868

869
  if (NULL == pInput) {
91,997,839✔
870
    return TSDB_CODE_SUCCESS;
×
871
  }
872

873
  *ppParam = (pInput->opType == pOperator->operatorType) ? pInput : NULL;
91,997,839✔
874

875
  if (NULL == *pppDownstramParam) {
92,001,078✔
876
    *pppDownstramParam = taosMemoryCalloc(pOperator->numOfDownstream, POINTER_BYTES);
19,575,881✔
877
    if (NULL == *pppDownstramParam) {
19,573,513✔
878
      return terrno;
×
879
    }
880
  }
881

882
  if (NULL == *ppParam) {
91,999,306✔
883
    for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
×
884
      (*pppDownstramParam)[i] = pInput;
×
885
    }
886
    return TSDB_CODE_SUCCESS;
×
887
  }
888

889
  memset(*pppDownstramParam, 0, pOperator->numOfDownstream * POINTER_BYTES);
91,998,710✔
890

891
  int32_t childrenNum = taosArrayGetSize((*ppParam)->pChildren);
92,000,355✔
892
  if (childrenNum <= 0) {
91,998,241✔
893
    return TSDB_CODE_SUCCESS;
81,017,921✔
894
  }
895

896
  for (int32_t i = 0; i < childrenNum; ++i) {
26,231,989✔
897
    SOperatorParam* pChild = *(SOperatorParam**)taosArrayGet((*ppParam)->pChildren, i);
15,251,553✔
898
    if (pChild == NULL) {
15,251,537✔
899
      return terrno;
×
900
    }
901

902
    if ((*pppDownstramParam)[pChild->downstreamIdx]) {
15,251,537✔
903
      int32_t code = mergeOperatorParams((*pppDownstramParam)[pChild->downstreamIdx], pChild);
×
904
      if (code) {
×
905
        return code;
×
906
      }
907
    } else {
908
      (*pppDownstramParam)[pChild->downstreamIdx] = pChild;
15,247,753✔
909
    }
910
  }
911

912
  taosArrayDestroy((*ppParam)->pChildren);
10,980,436✔
913
  (*ppParam)->pChildren = NULL;
10,978,266✔
914

915
  return TSDB_CODE_SUCCESS;
10,979,856✔
916
}
917

918
SSDataBlock* getNextBlockFromDownstream(struct SOperatorInfo* pOperator, int32_t idx) {
893,455,652✔
919
  SSDataBlock* p = NULL;
893,455,652✔
920
  int32_t      code = getNextBlockFromDownstreamImpl(pOperator, idx, true, &p);
893,487,239✔
921
  if (code == TSDB_CODE_SUCCESS) {
890,814,664✔
922
    code = blockDataCheck(p);
890,870,589✔
923
    if (code != TSDB_CODE_SUCCESS) {
890,894,789✔
924
      qError("blockDataCheck failed, code:%s", tstrerror(code));
×
925
    }
926
  }
927
  return (code == 0) ? p : NULL;
890,873,096✔
928
}
929

930
SSDataBlock* getNextBlockFromDownstreamRemain(struct SOperatorInfo* pOperator, int32_t idx) {
69,434,591✔
931
  SSDataBlock* p = NULL;
69,434,591✔
932
  int32_t      code = getNextBlockFromDownstreamImpl(pOperator, idx, false, &p);
69,434,591✔
933
  if (code == TSDB_CODE_SUCCESS) {
69,423,819✔
934
    code = blockDataCheck(p);
69,423,819✔
935
    if (code != TSDB_CODE_SUCCESS) {
69,423,326✔
936
      qError("blockDataCheck failed, code:%s", tstrerror(code));
×
937
    }
938
  }
939
  return (code == 0)? p:NULL;
69,423,819✔
940
}
941

942
/*
943
 * Fetch one block from downstream without preserving reused get-param ownership.
944
 *
945
 * @param pOperator Current operator.
946
 * @param idx Downstream index.
947
 * @param pResBlock Output result block pointer.
948
 *
949
 * @return TSDB_CODE_SUCCESS on success, otherwise error code.
950
 */
951
static FORCE_INLINE int32_t getNextBlockFromDownstreamRemainDetachImpl(struct SOperatorInfo* pOperator, int32_t idx,
952
                                                                       SSDataBlock** pResBlock) {
953
  QRY_PARAM_CHECK(pResBlock);
732,882,610✔
954

955
  int32_t code = TSDB_CODE_SUCCESS;
732,897,340✔
956
  int32_t lino = 0;
732,897,340✔
957

958
  if (pOperator->pDownstreamGetParams && pOperator->pDownstreamGetParams[idx]) {
739,728,261✔
959
    SOperatorParam* pGetParam = pOperator->pDownstreamGetParams[idx];
6,829,397✔
960
    pOperator->pDownstreamGetParams[idx] = NULL;
6,829,342✔
961
    // Once detached from the parent operator, downstream must own/free this param.
962
    pGetParam->reUse = false;
6,830,534✔
963

964
    qDebug("DynOp: op %s start to get block from downstream %s", pOperator->name, pOperator->pDownstream[idx]->name);
6,829,474✔
965
    code = pOperator->pDownstream[idx]->fpSet.getNextExtFn(pOperator->pDownstream[idx], pGetParam, pResBlock);
6,829,474✔
966
    QUERY_CHECK_CODE(code, lino, _return);
6,830,921✔
967
  } else {
968
    code = pOperator->pDownstream[idx]->fpSet.getNextFn(pOperator->pDownstream[idx], pResBlock);
726,085,167✔
969
    QUERY_CHECK_CODE(code, lino, _return);
724,343,684✔
970
  }
971

972
_return:
724,343,684✔
973
  if (code) {
731,174,605✔
974
    qError("failed to get next data block from upstream at %s, line:%d code:%s", __func__, lino, tstrerror(code));
×
975
  }
976
  return code;
731,168,786✔
977
}
978

979
/*
980
 * Fetch and validate one downstream block while detaching one-shot get-param.
981
 *
982
 * @param pOperator Current operator.
983
 * @param idx Downstream index.
984
 *
985
 * @return Valid block pointer on success, or NULL on failure.
986
 */
987
SSDataBlock* getNextBlockFromDownstreamRemainDetach(struct SOperatorInfo* pOperator, int32_t idx) {
732,846,341✔
988
  SSDataBlock* p = NULL;
732,846,341✔
989
  int32_t      code = TSDB_CODE_SUCCESS;
732,892,570✔
990
  int32_t      lino = 0;
732,892,570✔
991

992
  code = getNextBlockFromDownstreamRemainDetachImpl(pOperator, idx, &p);
731,168,786✔
993
  QUERY_CHECK_CODE(code, lino, _return);
731,168,786✔
994

995
  code = blockDataCheck(p);
731,168,786✔
996
  QUERY_CHECK_CODE(code, lino, _return);
731,216,469✔
997

998
_return:
731,216,469✔
999
  if (code) {
731,195,354✔
1000
    qError("failed to get next data block from downstream at %s, line:%d code:%s", __func__, lino, tstrerror(code));
1,975✔
1001
    return NULL;
×
1002
  }
1003
  return p;
731,193,626✔
1004
}
1005

1006
int32_t optrDefaultGetNextExtFn(struct SOperatorInfo* pOperator, SOperatorParam* pParam, SSDataBlock** pRes) {
92,002,734✔
1007
  QRY_PARAM_CHECK(pRes);
92,002,734✔
1008

1009
  int32_t lino = 0;
92,002,734✔
1010
  int32_t code = setOperatorParams(pOperator, pParam, OP_GET_PARAM);
92,002,734✔
1011
  QUERY_CHECK_CODE(code, lino, _end);
91,997,188✔
1012
  code = pOperator->fpSet.getNextFn(pOperator, pRes);
91,997,188✔
1013
  QUERY_CHECK_CODE(code, lino, _end);
91,998,919✔
1014

1015
_end:
91,998,919✔
1016
  if (code != TSDB_CODE_SUCCESS) {
91,998,919✔
1017
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1018
    pOperator->pTaskInfo->code = code;
×
1019
    T_LONG_JMP(pOperator->pTaskInfo->env, code);
×
1020
  }
1021

1022
  return code;
91,998,919✔
1023
}
1024

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

1046
  return code;
×
1047
}
1048

1049
int64_t getOperatorResultBlockId(struct SOperatorInfo* pOperator, int32_t idx) {
28,367,895✔
1050
  if (pOperator->transparent) {
28,367,895✔
1051
    return getOperatorResultBlockId(pOperator->pDownstream[idx], 0);
2,370,438✔
1052
  }
1053
  return pOperator->resultDataBlockId;
25,997,950✔
1054
}
1055

1056
void resetOperatorState(SOperatorInfo *pOper) {
×
1057
  pOper->status = OP_NOT_OPENED;
×
1058
}
×
1059

1060
void resetBasicOperatorState(SOptrBasicInfo *pBasicInfo) {
15,291,883✔
1061
  if (pBasicInfo->pRes) blockDataCleanup(pBasicInfo->pRes);
15,291,883✔
1062
  initResultRowInfo(&pBasicInfo->resultRowInfo);
15,295,247✔
1063
}
15,295,247✔
1064

1065
int32_t resetAggSup(SExprSupp* pExprSupp, SAggSupporter* pSup, SExecTaskInfo* pTaskInfo,
13,164,880✔
1066
                    SNodeList* pNodeList, SNodeList* pGroupKeys, size_t keyBufSize, const char* pKey, void* pState,
1067
                    SFunctionStateStore* pStore) {
1068
  int32_t    code = 0, lino = 0, num = 0;
13,164,880✔
1069
  SExprInfo* pExprInfo = NULL;
13,165,132✔
1070
  cleanupAggSup(pSup);
13,165,132✔
1071
  cleanupExprSuppWithoutFilter(pExprSupp);
13,165,801✔
1072
  code = createExprInfo(pNodeList, pGroupKeys, &pExprInfo, &num);
13,162,645✔
1073
  QUERY_CHECK_CODE(code, lino, _error);
13,163,942✔
1074
  code = initAggSup(pExprSupp, pSup, pExprInfo, num, keyBufSize, pKey, pState, pStore);
13,163,942✔
1075
  QUERY_CHECK_CODE(code, lino, _error);
13,164,727✔
1076
  return code;
13,164,727✔
1077
_error:
×
1078
  if (code != TSDB_CODE_SUCCESS) {
×
1079
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1080
    pTaskInfo->code = code;
×
1081
  }
1082
  return code;
×
1083
}
1084

1085
int32_t resetExprSupp(SExprSupp* pExprSupp, SExecTaskInfo* pTaskInfo, SNodeList* pNodeList,
5,358,528✔
1086
                      SNodeList* pGroupKeys, SFunctionStateStore* pStore) {
1087
  int32_t code = 0, lino = 0, num = 0;
5,358,528✔
1088
  SExprInfo* pExprInfo = NULL;
5,358,757✔
1089
  cleanupExprSuppWithoutFilter(pExprSupp);
5,358,757✔
1090
  code = createExprInfo(pNodeList, pGroupKeys, &pExprInfo, &num);
5,358,757✔
1091
  QUERY_CHECK_CODE(code, lino, _error);
5,358,088✔
1092
  code = initExprSupp(pExprSupp, pExprInfo, num, pStore);
5,358,088✔
1093
  QUERY_CHECK_CODE(code, lino, _error);
5,358,311✔
1094
  return code;
5,358,311✔
1095
_error:
×
1096
  if (code != TSDB_CODE_SUCCESS) {
×
1097
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1098
    pTaskInfo->code = code;
×
1099
  }
1100
  return code;
×
1101
}
1102

1103
int32_t copyColumnsValue(SNodeList* pNodeList, int64_t targetBlkId, SSDataBlock* pDst, SSDataBlock* pSrc, int32_t totalRows) {
1,415,604✔
1104
  bool    isNull = (NULL == pSrc || pSrc->info.rows <= 0);
1,415,604✔
1105
  size_t  numOfCols = LIST_LENGTH(pNodeList);
1,415,604✔
1106
  int64_t numOfRows = isNull ? 0 : pSrc->info.rows;
1,415,604✔
1107
  int32_t code = 0;
1,415,604✔
1108
  int32_t lino = 0;
1,415,604✔
1109

1110
  for (int32_t i = 0; i < numOfCols; ++i) {
9,699,794✔
1111
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
8,284,190✔
1112
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pExpr)->dataBlockId == targetBlkId) {
8,284,190✔
1113
      SColumnInfoData* pDstCol = taosArrayGet(pDst->pDataBlock, pNode->slotId);
4,523,697✔
1114
      QUERY_CHECK_NULL(pDstCol, code, lino, _return, terrno)
4,523,697✔
1115

1116
      if (isNull) {
4,523,697✔
1117
        colDataSetNNULL(pDstCol, 0, totalRows);
×
1118
      } else {
1119
        SColumnInfoData* pSrcCol = taosArrayGet(pSrc->pDataBlock, ((SColumnNode*)pNode->pExpr)->slotId);
4,523,697✔
1120
        QUERY_CHECK_NULL(pSrcCol, code, lino, _return, terrno)
4,523,697✔
1121

1122
        code = colDataAssign(pDstCol, pSrcCol, pSrc->info.rows, &pDst->info);
4,523,697✔
1123

1124
        QUERY_CHECK_CODE(code, lino, _return);
4,523,697✔
1125
        if (pSrc->info.rows < totalRows) {
4,523,697✔
1126
          colDataSetNNULL(pDstCol, pSrc->info.rows, totalRows - pSrc->info.rows);
122,340✔
1127
        }
1128
      }
1129
    }
1130
  }
1131

1132
  return code;
1,415,604✔
1133
_return:
×
1134
  qError("failed to copy columns value, line:%d code:%s", lino, tstrerror(code));
×
1135
  return code;
×
1136
}
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