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

taosdata / TDengine / #3796

31 Mar 2025 10:39AM UTC coverage: 30.372% (-7.1%) from 37.443%
#3796

push

travis-ci

happyguoxy
test:add test cases

69287 of 309062 branches covered (22.42%)

Branch coverage included in aggregate %.

118044 of 307720 relevant lines covered (38.36%)

278592.15 hits per line

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

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

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

21
#include "tglobal.h"
22

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

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

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

48
  return fpSet;
84✔
49
}
50

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

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

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

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

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

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

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

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

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

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

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

132
  int64_t t = getQuerySupportBufSize(numOfTables);
×
133

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

270
    return OPTR_FN_RET_ABORT;
×
271
  }
272

273
  return OPTR_FN_RET_CONTINUE;
×
274
}
275

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

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

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

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

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

308
      code = initQueriedTableSchemaInfo(pHandle, &pTableScanNode->scan, dbname, pTaskInfo);
40✔
309
      if (code) {
40!
310
        pTaskInfo->code = code;
×
311
        tableListDestroy(pTableListInfo);
×
312
        return code;
×
313
      }
314

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

330
      code = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
40✔
331
      if (NULL == pOperator || code != 0) {
40!
332
        pTaskInfo->code = code;
×
333
        tableListDestroy(pTableListInfo);
×
334
        return code;
×
335
      }
336

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

348
      code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, true, pHandle, pTableListInfo,
×
349
                                     pTagCond, pTagIndexCond, pTaskInfo);
350
      if (code) {
×
351
        pTaskInfo->code = code;
×
352
        tableListDestroy(pTableListInfo);
×
353
        qError("failed to createScanTableListInfo, code:%s", tstrerror(code));
×
354
        return code;
×
355
      }
356

357
      code = initQueriedTableSchemaInfo(pHandle, &pTableScanNode->scan, dbname, pTaskInfo);
×
358
      if (code) {
×
359
        pTaskInfo->code = code;
×
360
        tableListDestroy(pTableListInfo);
×
361
        return code;
×
362
      }
363

364
      code = createTableMergeScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pOperator);
×
365
      if (NULL == pOperator || code != 0) {
×
366
        pTaskInfo->code = code;
×
367
        tableListDestroy(pTableListInfo);
×
368
        return code;
×
369
      }
370

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

385
      if (pHandle->vnode && (pTaskInfo->pSubplan->pVTables == NULL)) {
×
386
        code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
×
387
                                       pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo);
388
        if (code) {
×
389
          pTaskInfo->code = code;
×
390
          tableListDestroy(pTableListInfo);
×
391
          qError("failed to createScanTableListInfo, code:%s", tstrerror(code));
×
392
          return code;
×
393
        }
394
      }
395

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

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

452
        size_t num = taosArrayGetSize(pList);
×
453
        for (int32_t i = 0; i < num; ++i) {
×
454
          uint64_t* id = taosArrayGet(pList, i);
×
455
          if (id == NULL) {
×
456
            continue;
×
457
          }
458

459
          code = tableListAddTableInfo(pTableListInfo, *id, 0);
×
460
          if (code) {
×
461
            pTaskInfo->code = code;
×
462
            tableListDestroy(pTableListInfo);
×
463
            taosArrayDestroy(pList);
×
464
            return code;
×
465
          }
466
        }
467

468
        taosArrayDestroy(pList);
×
469
      } else {  // Create group with only one table
470
        code = tableListAddTableInfo(pTableListInfo, pBlockNode->uid, 0);
×
471
        if (code) {
×
472
          pTaskInfo->code = code;
×
473
          tableListDestroy(pTableListInfo);
×
474
          return code;
×
475
        }
476
      }
477

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

493
      code = createScanTableListInfo(&pScanNode->scan, pScanNode->pGroupTags, true, pHandle, pTableListInfo, pTagCond,
×
494
                                     pTagIndexCond, pTaskInfo);
495
      if (code != TSDB_CODE_SUCCESS) {
×
496
        pTaskInfo->code = code;
×
497
        tableListDestroy(pTableListInfo);
×
498
        return code;
×
499
      }
500

501
      code = initQueriedTableSchemaInfo(pHandle, &pScanNode->scan, dbname, pTaskInfo);
×
502
      if (code != TSDB_CODE_SUCCESS) {
×
503
        pTaskInfo->code = code;
×
504
        tableListDestroy(pTableListInfo);
×
505
        return code;
×
506
      }
507

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

524
    if (pOperator != NULL) {  // todo moved away
46!
525
      pOperator->resultDataBlockId = pPhyNode->pOutputDataBlockDesc->dataBlockId;
46✔
526
    }
527

528
    *pOptrInfo = pOperator;
46✔
529
    return code;
46✔
530
  }
531

532
  size_t          size = LIST_LENGTH(pPhyNode->pChildren);
41!
533
  SOperatorInfo** ops = taosMemoryCalloc(size, POINTER_BYTES);
41!
534
  if (ops == NULL) {
41!
535
    code = terrno;
×
536
    pTaskInfo->code = code;
×
537
    return code;
×
538
  }
539

540
  for (int32_t i = 0; i < size; ++i) {
82✔
541
    SPhysiNode* pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, i);
41✔
542
    code = createOperator(pChildNode, pTaskInfo, pHandle, pTagCond, pTagIndexCond, pUser, dbname, &ops[i], model);
41✔
543
    if (ops[i] == NULL || code != 0) {
41!
544
      for (int32_t j = 0; j < i; ++j) {
×
545
        destroyOperator(ops[j]);
×
546
      }
547
      taosMemoryFree(ops);
×
548
      return code;
×
549
    }
550
  }
551

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

661
    if (pVirtualTableScanNode->scan.node.pLimit != NULL) {
×
662
      pVirtualTableScanNode->groupSort = true;
×
663
    }
664

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

672
    code = initQueriedTableSchemaInfo(pHandle, &pVirtualTableScanNode->scan, dbname, pTaskInfo);
×
673
    if (code) {
×
674
      pTaskInfo->code = code;
×
675
      tableListDestroy(pTableListInfo);
×
676
      return code;
×
677
    }
678

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

688

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

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

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

721
  taosMemoryFree(ops);
41!
722
  if (pOptr) {
41✔
723
    pOptr->resultDataBlockId = pPhyNode->pOutputDataBlockDesc->dataBlockId;
38✔
724
  }
725

726
  *pOptrInfo = pOptr;
41✔
727
  return code;
41✔
728
}
729

730
void destroyOperator(SOperatorInfo* pOperator) {
90✔
731
  if (pOperator == NULL) {
90✔
732
    return;
3✔
733
  }
734

735
  freeResetOperatorParams(pOperator, OP_GET_PARAM, true);
87✔
736
  freeResetOperatorParams(pOperator, OP_NOTIFY_PARAM, true);
87✔
737

738
  if (pOperator->pDownstream != NULL) {
87✔
739
    for (int32_t i = 0; i < pOperator->numOfRealDownstream; ++i) {
76✔
740
      destroyOperator(pOperator->pDownstream[i]);
38✔
741
    }
742

743
    taosMemoryFreeClear(pOperator->pDownstream);
38!
744
    pOperator->numOfDownstream = 0;
38✔
745
  }
746

747
  if (pOperator->fpSet.closeFn != NULL && pOperator->info != NULL) {
87!
748
    pOperator->fpSet.closeFn(pOperator->info);
84✔
749
  }
750

751
  cleanupExprSupp(&pOperator->exprSupp);
87✔
752
  taosMemoryFreeClear(pOperator);
87!
753
}
754

755
void destroyOperatorAndDownstreams(SOperatorInfo* pOperator, SOperatorInfo** downstreams, int32_t num) {
3✔
756
  if (downstreams != NULL) {
3!
757
    for (int i = 0; i < num; i++) {
6✔
758
      destroyOperator(downstreams[i]);
3✔
759
    }
760
  }
761

762
  if (pOperator != NULL) {
3!
763
    pOperator->info = NULL;
3✔
764
    if (pOperator->pDownstream != NULL) {
3!
765
      taosMemoryFreeClear(pOperator->pDownstream);
×
766
      pOperator->pDownstream = NULL;
×
767
    }
768
    destroyOperator(pOperator);
3✔
769
  }
770
}
3✔
771

772
int32_t getOperatorExplainExecInfo(SOperatorInfo* operatorInfo, SArray* pExecInfoList) {
×
773
  SExplainExecInfo  execInfo = {0};
×
774
  SExplainExecInfo* pExplainInfo = taosArrayPush(pExecInfoList, &execInfo);
×
775
  if (pExplainInfo == NULL) {
×
776
    return terrno;
×
777
  }
778

779
  pExplainInfo->numOfRows = operatorInfo->resultInfo.totalRows;
×
780
  pExplainInfo->startupCost = operatorInfo->cost.openCost;
×
781
  pExplainInfo->totalCost = operatorInfo->cost.totalCost;
×
782
  pExplainInfo->verboseLen = 0;
×
783
  pExplainInfo->verboseInfo = NULL;
×
784

785
  if (operatorInfo->fpSet.getExplainFn) {
×
786
    int32_t code =
787
        operatorInfo->fpSet.getExplainFn(operatorInfo, &pExplainInfo->verboseInfo, &pExplainInfo->verboseLen);
×
788
    if (code) {
×
789
      qError("%s operator getExplainFn failed, code:%s", GET_TASKID(operatorInfo->pTaskInfo), tstrerror(code));
×
790
      return code;
×
791
    }
792
  }
793

794
  int32_t code = 0;
×
795
  for (int32_t i = 0; i < operatorInfo->numOfDownstream; ++i) {
×
796
    code = getOperatorExplainExecInfo(operatorInfo->pDownstream[i], pExecInfoList);
×
797
    if (code != TSDB_CODE_SUCCESS) {
×
798
      //      taosMemoryFreeClear(*pRes);
799
      return code;
×
800
    }
801
  }
802

803
  return TSDB_CODE_SUCCESS;
×
804
}
805

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

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

823
          pBatch->multiParams = true;
×
824
          pBatch->pBatchs = tSimpleHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
×
825
          if (NULL == pBatch->pBatchs) {
×
826
            taosMemoryFree(pBatch);
×
827
            return terrno;
×
828
          }
829

830
          tSimpleHashSetFreeFp(pBatch->pBatchs, freeExchangeGetBasicOperatorParam);
×
831

832
          int32_t code = tSimpleHashPut(pBatch->pBatchs, &pDExc->basic.vgId, sizeof(pDExc->basic.vgId), &pDExc->basic,
×
833
                                        sizeof(pDExc->basic));
834
          if (code) {
×
835
            return code;
×
836
          }
837

838
          code = tSimpleHashPut(pBatch->pBatchs, &pSExc->basic.vgId, sizeof(pSExc->basic.vgId), &pSExc->basic,
×
839
                                sizeof(pSExc->basic));
840
          if (code) {
×
841
            return code;
×
842
          }
843

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

876
  return TSDB_CODE_SUCCESS;
×
877
}
878

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

895
  freeResetOperatorParams(pOperator, type, false);
×
896

897
  if (NULL == pInput) {
×
898
    return TSDB_CODE_SUCCESS;
×
899
  }
900

901
  *ppParam = (pInput->opType == pOperator->operatorType) ? pInput : NULL;
×
902

903
  if (NULL == *pppDownstramParam) {
×
904
    *pppDownstramParam = taosMemoryCalloc(pOperator->numOfDownstream, POINTER_BYTES);
×
905
    if (NULL == *pppDownstramParam) {
×
906
      return terrno;
×
907
    }
908
  }
909

910
  if (NULL == *ppParam) {
×
911
    for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
×
912
      (*pppDownstramParam)[i] = pInput;
×
913
    }
914
    return TSDB_CODE_SUCCESS;
×
915
  }
916

917
  memset(*pppDownstramParam, 0, pOperator->numOfDownstream * POINTER_BYTES);
×
918

919
  int32_t childrenNum = taosArrayGetSize((*ppParam)->pChildren);
×
920
  if (childrenNum <= 0) {
×
921
    return TSDB_CODE_SUCCESS;
×
922
  }
923

924
  for (int32_t i = 0; i < childrenNum; ++i) {
×
925
    SOperatorParam* pChild = *(SOperatorParam**)taosArrayGet((*ppParam)->pChildren, i);
×
926
    if (pChild == NULL) {
×
927
      return terrno;
×
928
    }
929

930
    if ((*pppDownstramParam)[pChild->downstreamIdx]) {
×
931
      int32_t code = mergeOperatorParams((*pppDownstramParam)[pChild->downstreamIdx], pChild);
×
932
      if (code) {
×
933
        return code;
×
934
      }
935
    } else {
936
      (*pppDownstramParam)[pChild->downstreamIdx] = pChild;
×
937
    }
938
  }
939

940
  taosArrayDestroy((*ppParam)->pChildren);
×
941
  (*ppParam)->pChildren = NULL;
×
942

943
  return TSDB_CODE_SUCCESS;
×
944
}
945

946
SSDataBlock* getNextBlockFromDownstream(struct SOperatorInfo* pOperator, int32_t idx) {
105✔
947
  SSDataBlock* p = NULL;
105✔
948
  int32_t      code = getNextBlockFromDownstreamImpl(pOperator, idx, true, &p);
105✔
949
  if (code == TSDB_CODE_SUCCESS) {
105!
950
    code = blockDataCheck(p);
105✔
951
    if (code != TSDB_CODE_SUCCESS) {
105!
952
      qError("blockDataCheck failed, code:%s", tstrerror(code));
×
953
    }
954
  }
955
  return (code == 0) ? p : NULL;
105!
956
}
957

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

970
int32_t optrDefaultGetNextExtFn(struct SOperatorInfo* pOperator, SOperatorParam* pParam, SSDataBlock** pRes) {
×
971
  QRY_PARAM_CHECK(pRes);
×
972

973
  int32_t lino = 0;
×
974
  int32_t code = setOperatorParams(pOperator, pParam, OP_GET_PARAM);
×
975
  QUERY_CHECK_CODE(code, lino, _end);
×
976
  code = pOperator->fpSet.getNextFn(pOperator, pRes);
×
977
  QUERY_CHECK_CODE(code, lino, _end);
×
978

979
_end:
×
980
  if (code != TSDB_CODE_SUCCESS) {
×
981
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
982
    pOperator->pTaskInfo->code = code;
×
983
    T_LONG_JMP(pOperator->pTaskInfo->env, code);
×
984
  }
985

986
  return code;
×
987
}
988

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

1010
  return code;
×
1011
}
1012

1013
int16_t getOperatorResultBlockId(struct SOperatorInfo* pOperator, int32_t idx) {
×
1014
  if (pOperator->transparent) {
×
1015
    return getOperatorResultBlockId(pOperator->pDownstream[idx], 0);
×
1016
  }
1017
  return pOperator->resultDataBlockId;
×
1018
}
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