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

taosdata / TDengine / #4911

04 Jan 2026 09:05AM UTC coverage: 65.028% (-0.8%) from 65.864%
#4911

push

travis-ci

web-flow
merge: from main to 3.0 branch #34156

1206 of 4524 new or added lines in 22 files covered. (26.66%)

1517 existing lines in 134 files now uncovered.

195276 of 300296 relevant lines covered (65.03%)

116931714.52 hits per line

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

48.21
/source/libs/executor/src/dynqueryctrloperator.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 "executorInt.h"
17
#include "filter.h"
18
#include "nodes.h"
19
#include "operator.h"
20
#include "os.h"
21
#include "plannodes.h"
22
#include "query.h"
23
#include "querynodes.h"
24
#include "querytask.h"
25
#include "tarray.h"
26
#include "tcompare.h"
27
#include "tdatablock.h"
28
#include "thash.h"
29
#include "tmsg.h"
30
#include "trpc.h"
31
#include "ttypes.h"
32
#include "tdataformat.h"
33
#include "dynqueryctrl.h"
34

35
int64_t gSessionId = 0;
36

37
void freeVgTableList(void* ptr) { 
×
38
  taosArrayDestroy(*(SArray**)ptr); 
×
39
}
×
40

41
static void destroyStbJoinTableList(SStbJoinTableList* pListHead) {
1,111,761✔
42
  SStbJoinTableList* pNext = NULL;
1,111,761✔
43
  
44
  while (pListHead) {
1,112,244✔
45
    taosMemoryFree(pListHead->pLeftVg);
483✔
46
    taosMemoryFree(pListHead->pLeftUid);
483✔
47
    taosMemoryFree(pListHead->pRightVg);
483✔
48
    taosMemoryFree(pListHead->pRightUid);
483✔
49
    pNext = pListHead->pNext;
483✔
50
    taosMemoryFree(pListHead);
483✔
51
    pListHead = pNext;
483✔
52
  }
53
}
1,111,761✔
54

55
static void destroyStbJoinDynCtrlInfo(SStbJoinDynCtrlInfo* pStbJoin) {
1,111,761✔
56
  qDebug("dynQueryCtrl exec info, prevBlk:%" PRId64 ", prevRows:%" PRId64 ", postBlk:%" PRId64 ", postRows:%" PRId64 ", leftCacheNum:%" PRId64 ", rightCacheNum:%" PRId64, 
1,111,761✔
57
         pStbJoin->execInfo.prevBlkNum, pStbJoin->execInfo.prevBlkRows, pStbJoin->execInfo.postBlkNum, 
58
         pStbJoin->execInfo.postBlkRows, pStbJoin->execInfo.leftCacheNum, pStbJoin->execInfo.rightCacheNum);
59

60
  if (pStbJoin->basic.batchFetch) {
1,111,761✔
61
    if (pStbJoin->ctx.prev.leftHash) {
1,110,702✔
62
      tSimpleHashSetFreeFp(pStbJoin->ctx.prev.leftHash, freeVgTableList);
1,043,768✔
63
      tSimpleHashCleanup(pStbJoin->ctx.prev.leftHash);
1,043,768✔
64
    }
65
    if (pStbJoin->ctx.prev.rightHash) {
1,110,702✔
66
      tSimpleHashSetFreeFp(pStbJoin->ctx.prev.rightHash, freeVgTableList);
1,043,768✔
67
      tSimpleHashCleanup(pStbJoin->ctx.prev.rightHash);
1,043,768✔
68
    }
69
  } else {
70
    if (pStbJoin->ctx.prev.leftCache) {
1,059✔
71
      tSimpleHashCleanup(pStbJoin->ctx.prev.leftCache);
1,059✔
72
    }
73
    if (pStbJoin->ctx.prev.rightCache) {
1,059✔
74
      tSimpleHashCleanup(pStbJoin->ctx.prev.rightCache);
1,059✔
75
    }
76
    if (pStbJoin->ctx.prev.onceTable) {
1,059✔
77
      tSimpleHashCleanup(pStbJoin->ctx.prev.onceTable);
1,059✔
78
    }
79
  }
80

81
  destroyStbJoinTableList(pStbJoin->ctx.prev.pListHead);
1,111,761✔
82
}
1,111,761✔
83

84
void destroyColRefInfo(void *info) {
34,071,569✔
85
  SColRefInfo *pColRefInfo = (SColRefInfo *)info;
34,071,569✔
86
  if (pColRefInfo) {
34,071,569✔
87
    taosMemoryFree(pColRefInfo->colName);
34,071,569✔
88
    taosMemoryFree(pColRefInfo->colrefName);
34,071,569✔
89
  }
90
}
34,071,569✔
91

92
void destroyColRefArray(void *info) {
711,716✔
93
  SArray *pColRefArray = *(SArray **)info;
711,716✔
94
  if (pColRefArray) {
711,716✔
95
    taosArrayDestroyEx(pColRefArray, destroyColRefInfo);
711,716✔
96
  }
97
}
711,716✔
98

99
void freeUseDbOutput(void* pOutput) {
240,173✔
100
  SUseDbOutput *pOut = *(SUseDbOutput**)pOutput;
240,173✔
101
  if (NULL == pOutput) {
240,173✔
102
    return;
×
103
  }
104

105
  if (pOut->dbVgroup) {
240,173✔
106
    freeVgInfo(pOut->dbVgroup);
240,173✔
107
  }
108
  taosMemFree(pOut);
240,173✔
109
}
110

UNCOV
111
void destroyOtbInfoArray(void *info) {
×
UNCOV
112
  SArray *pOtbInfoArray = *(SArray **)info;
×
UNCOV
113
  if (pOtbInfoArray) {
×
UNCOV
114
    taosArrayDestroyEx(pOtbInfoArray, destroySOrgTbInfo);
×
115
  }
UNCOV
116
}
×
117

UNCOV
118
void destroyOtbVgIdToOtbInfoArrayMap(void *info) {
×
UNCOV
119
  SHashObj* pOtbVgIdToOtbInfoArrayMap = *(SHashObj **)info;
×
UNCOV
120
  if (pOtbVgIdToOtbInfoArrayMap) {
×
UNCOV
121
    taosHashSetFreeFp(pOtbVgIdToOtbInfoArrayMap, destroyOtbInfoArray);
×
UNCOV
122
    taosHashCleanup(pOtbVgIdToOtbInfoArrayMap);
×
123
  }
UNCOV
124
}
×
125

UNCOV
126
void destroyTagList(void *info) {
×
UNCOV
127
  SArray *pTagList = *(SArray **)info;
×
UNCOV
128
  if (pTagList) {
×
UNCOV
129
    taosArrayDestroyEx(pTagList, destroyTagVal);
×
130
  }
UNCOV
131
}
×
132

UNCOV
133
void destroyVtbUidTagListMap(void *info) {
×
UNCOV
134
  SHashObj* pVtbUidTagListMap = *(SHashObj **)info;
×
UNCOV
135
  if (pVtbUidTagListMap) {
×
UNCOV
136
    taosHashSetFreeFp(pVtbUidTagListMap, destroyTagList);
×
UNCOV
137
    taosHashCleanup(pVtbUidTagListMap);
×
138
  }
UNCOV
139
}
×
140

141
static void destroyVtbScanDynCtrlInfo(SVtbScanDynCtrlInfo* pVtbScan) {
274,284✔
142
  if (pVtbScan->dbName) {
274,284✔
143
    taosMemoryFreeClear(pVtbScan->dbName);
274,284✔
144
  }
145
  if (pVtbScan->tbName) {
274,284✔
146
    taosMemoryFreeClear(pVtbScan->tbName);
274,284✔
147
  }
148
  if (pVtbScan->childTableList) {
274,284✔
149
    taosArrayDestroyEx(pVtbScan->childTableList, destroyColRefArray);
274,284✔
150
  }
151
  if (pVtbScan->colRefInfo) {
274,284✔
152
    taosArrayDestroyEx(pVtbScan->colRefInfo, destroyColRefInfo);
×
153
    pVtbScan->colRefInfo = NULL;
×
154
  }
155
  if (pVtbScan->childTableMap) {
274,284✔
156
    taosHashCleanup(pVtbScan->childTableMap);
242,262✔
157
  }
158
  if (pVtbScan->readColList) {
274,284✔
159
    taosArrayDestroy(pVtbScan->readColList);
274,284✔
160
  }
161
  if (pVtbScan->dbVgInfoMap) {
274,284✔
162
    taosHashSetFreeFp(pVtbScan->dbVgInfoMap, freeUseDbOutput);
274,284✔
163
    taosHashCleanup(pVtbScan->dbVgInfoMap);
274,284✔
164
  }
165
  if (pVtbScan->otbNameToOtbInfoMap) {
274,284✔
166
    taosHashSetFreeFp(pVtbScan->otbNameToOtbInfoMap, destroySOrgTbInfo);
609✔
167
    taosHashCleanup(pVtbScan->otbNameToOtbInfoMap);
609✔
168
  }
169
  if (pVtbScan->pRsp) {
274,284✔
170
    tFreeSUsedbRsp(pVtbScan->pRsp);
×
171
    taosMemoryFreeClear(pVtbScan->pRsp);
×
172
  }
173
  if (pVtbScan->existOrgTbVg) {
274,284✔
174
    taosHashCleanup(pVtbScan->existOrgTbVg);
274,284✔
175
  }
176
  if (pVtbScan->curOrgTbVg) {
274,284✔
177
    taosHashCleanup(pVtbScan->curOrgTbVg);
1,600✔
178
  }
179
  if (pVtbScan->newAddedVgInfo) {
274,284✔
180
    taosHashCleanup(pVtbScan->newAddedVgInfo);
796✔
181
  }
182
  if (pVtbScan->otbVgIdToOtbInfoArrayMap) {
274,284✔
UNCOV
183
    taosHashSetFreeFp(pVtbScan->otbVgIdToOtbInfoArrayMap, destroyOtbInfoArray);
×
UNCOV
184
    taosHashCleanup(pVtbScan->otbVgIdToOtbInfoArrayMap);
×
185
  }
186
  if (pVtbScan->vtbUidToVgIdMapMap) {
274,284✔
UNCOV
187
    taosHashSetFreeFp(pVtbScan->vtbUidToVgIdMapMap, destroyOtbVgIdToOtbInfoArrayMap);
×
UNCOV
188
    taosHashCleanup(pVtbScan->vtbUidToVgIdMapMap);
×
189
  }
190
  if (pVtbScan->vtbGroupIdToVgIdMapMap) {
274,284✔
UNCOV
191
    taosHashSetFreeFp(pVtbScan->vtbGroupIdToVgIdMapMap, destroyOtbVgIdToOtbInfoArrayMap);
×
UNCOV
192
    taosHashCleanup(pVtbScan->vtbGroupIdToVgIdMapMap);
×
193
  }
194
  if (pVtbScan->vtbUidTagListMap) {
274,284✔
UNCOV
195
    taosHashSetFreeFp(pVtbScan->vtbUidTagListMap, destroyTagList);
×
UNCOV
196
    taosHashCleanup(pVtbScan->vtbUidTagListMap);
×
197
  }
198
  if (pVtbScan->vtbGroupIdTagListMap) {
274,284✔
UNCOV
199
    taosHashCleanup(pVtbScan->vtbGroupIdTagListMap);
×
200
  }
201
  if (pVtbScan->vtbUidToGroupIdMap) {
274,284✔
UNCOV
202
    taosHashCleanup(pVtbScan->vtbUidToGroupIdMap);
×
203
  }
204
}
274,284✔
205

206
void destroyWinArray(void *info) {
×
207
  SArray *pWinArray = *(SArray **)info;
×
208
  if (pWinArray) {
×
209
    taosArrayDestroy(pWinArray);
×
210
  }
211
}
×
212

213
static void destroyVtbWindowDynCtrlInfo(SVtbWindowDynCtrlInfo* pVtbWindow) {
×
214
  if (pVtbWindow->pRes) {
×
215
    blockDataDestroy(pVtbWindow->pRes);
×
216
  }
217
  if (pVtbWindow->pWins) {
×
218
    taosArrayDestroyEx(pVtbWindow->pWins, destroyWinArray);
×
219
  }
220
}
×
221

222
static void destroyDynQueryCtrlOperator(void* param) {
1,385,337✔
223
  SDynQueryCtrlOperatorInfo* pDyn = (SDynQueryCtrlOperatorInfo*)param;
1,385,337✔
224

225
  switch (pDyn->qType) {
1,385,337✔
226
    case DYN_QTYPE_STB_HASH:
1,111,053✔
227
      destroyStbJoinDynCtrlInfo(&pDyn->stbJoin);
1,111,053✔
228
      break;
1,111,053✔
229
    case DYN_QTYPE_VTB_AGG:
274,284✔
230
    case DYN_QTYPE_VTB_SCAN:
231
      destroyVtbScanDynCtrlInfo(&pDyn->vtbScan);
274,284✔
232
      break;
274,284✔
233
    case DYN_QTYPE_VTB_WINDOW:
×
234
      destroyVtbWindowDynCtrlInfo(&pDyn->vtbWindow);
×
235
      break;
×
236
    default:
×
237
      qError("unsupported dynamic query ctrl type: %d", pDyn->qType);
×
238
      break;
×
239
  }
240

241
  taosMemoryFreeClear(param);
1,385,337✔
242
}
1,385,337✔
243

244
static FORCE_INLINE bool tableNeedCache(int64_t uid, SStbJoinPrevJoinCtx* pPrev, SStbJoinPostJoinCtx* pPost, bool rightTable, bool batchFetch) {
245
  if (batchFetch) {
7,139,396✔
246
    return true;
7,135,160✔
247
  }
248
  
249
  if (rightTable) {
4,236✔
250
    return pPost->rightCurrUid == pPost->rightNextUid;
2,118✔
251
  }
252

253
  uint32_t* num = tSimpleHashGet(pPrev->leftCache, &uid, sizeof(uid));
2,118✔
254

255
  return (NULL == num) ? false : true;
2,118✔
256
}
257

258
static int32_t updatePostJoinCurrTableInfo(SStbJoinDynCtrlInfo*          pStbJoin) {
3,569,698✔
259
  SStbJoinPrevJoinCtx*       pPrev = &pStbJoin->ctx.prev;
3,569,698✔
260
  SStbJoinPostJoinCtx*       pPost = &pStbJoin->ctx.post;
3,569,698✔
261
  SStbJoinTableList*         pNode = pPrev->pListHead;
3,569,698✔
262
  int32_t*                   leftVgId = pNode->pLeftVg + pNode->readIdx;
3,569,698✔
263
  int32_t*                   rightVgId = pNode->pRightVg + pNode->readIdx;
3,569,698✔
264
  int64_t*                   leftUid = pNode->pLeftUid + pNode->readIdx;
3,569,698✔
265
  int64_t*                   rightUid = pNode->pRightUid + pNode->readIdx;
3,569,698✔
266
  int64_t                    readIdx = pNode->readIdx + 1;
3,569,698✔
267
  int64_t                    rightPrevUid = pPost->rightCurrUid;
3,569,698✔
268

269
  pPost->leftCurrUid = *leftUid;
3,569,698✔
270
  pPost->rightCurrUid = *rightUid;
3,569,698✔
271

272
  pPost->leftVgId = *leftVgId;
3,569,698✔
273
  pPost->rightVgId = *rightVgId;
3,569,698✔
274

275
  while (true) {
276
    if (readIdx < pNode->uidNum) {
3,569,698✔
277
      pPost->rightNextUid = *(pNode->pRightUid + readIdx);
3,502,188✔
278
      break;
3,502,188✔
279
    }
280
    
281
    pNode = pNode->pNext;
67,510✔
282
    if (NULL == pNode) {
67,510✔
283
      pPost->rightNextUid = 0;
67,510✔
284
      break;
67,510✔
285
    }
286
    
287
    rightUid = pNode->pRightUid;
×
288
    readIdx = 0;
×
289
  }
290

291
  pPost->leftNeedCache = tableNeedCache(*leftUid, pPrev, pPost, false, pStbJoin->basic.batchFetch);
7,139,396✔
292
  pPost->rightNeedCache = tableNeedCache(*rightUid, pPrev, pPost, true, pStbJoin->basic.batchFetch);
7,139,396✔
293

294
  if (!pStbJoin->basic.batchFetch && pPost->rightNeedCache && rightPrevUid != pPost->rightCurrUid) {
3,569,698✔
295
    QRY_ERR_RET(tSimpleHashPut(pPrev->rightCache, &pPost->rightCurrUid, sizeof(pPost->rightCurrUid), NULL, 0));
×
296
    pStbJoin->execInfo.rightCacheNum++;
×
297
  }  
298

299
  return TSDB_CODE_SUCCESS;
3,569,698✔
300
}
301

302
static int32_t copyOrgTbInfo(SOrgTbInfo* pSrc, SOrgTbInfo** ppDst) {
1,287,949✔
303
  int32_t     code = TSDB_CODE_SUCCESS;
1,287,949✔
304
  int32_t     lino = 0;
1,287,949✔
305
  SOrgTbInfo* pTbInfo = NULL;
1,287,949✔
306

307
  qDebug("start to copy org table info, vgId:%d, tbName:%s", pSrc->vgId, pSrc->tbName);
1,287,949✔
308

309
  pTbInfo = taosMemoryMalloc(sizeof(SOrgTbInfo));
1,287,949✔
310
  QUERY_CHECK_NULL(pTbInfo, code, lino, _return, terrno)
1,287,949✔
311

312
  pTbInfo->vgId = pSrc->vgId;
1,287,949✔
313
  tstrncpy(pTbInfo->tbName, pSrc->tbName, TSDB_TABLE_FNAME_LEN);
1,287,949✔
314

315
  pTbInfo->colMap = taosArrayDup(pSrc->colMap, NULL);
1,287,949✔
316
  QUERY_CHECK_NULL(pTbInfo->colMap, code, lino, _return, terrno)
1,287,949✔
317

318
  *ppDst = pTbInfo;
1,287,949✔
319

320
  return code;
1,287,949✔
321
_return:
×
322
  qError("failed to copy org table info, code:%d, line:%d", code, lino);
×
323
  if (pTbInfo) {
×
324
    if (pTbInfo->colMap) {
×
325
      taosArrayDestroy(pTbInfo->colMap);
×
326
    }
327
    taosMemoryFreeClear(pTbInfo);
×
328
  }
329
  return code;
×
330
}
331

UNCOV
332
static int32_t buildTagListForExchangeBasicParam(SExchangeOperatorBasicParam* pBasic, SArray* pTagList) {
×
UNCOV
333
  int32_t  code = TSDB_CODE_SUCCESS;
×
UNCOV
334
  int32_t  lino = 0;
×
UNCOV
335
  STagVal  tmpTag;
×
336

UNCOV
337
  pBasic->tagList = taosArrayInit(1, sizeof(STagVal));
×
UNCOV
338
  QUERY_CHECK_NULL(pBasic->tagList, code, lino, _return, terrno)
×
339

UNCOV
340
  for (int32_t i = 0; i < taosArrayGetSize(pTagList); ++i) {
×
UNCOV
341
    STagVal* pSrcTag = (STagVal*)taosArrayGet(pTagList, i);
×
UNCOV
342
    QUERY_CHECK_NULL(pSrcTag, code, lino, _return, terrno)
×
UNCOV
343
    tmpTag.type = pSrcTag->type;
×
UNCOV
344
    tmpTag.cid = pSrcTag->cid;
×
UNCOV
345
    if (IS_VAR_DATA_TYPE(pSrcTag->type)) {
×
UNCOV
346
      tmpTag.nData = pSrcTag->nData;
×
UNCOV
347
      tmpTag.pData = taosMemoryMalloc(tmpTag.nData);
×
UNCOV
348
      QUERY_CHECK_NULL(tmpTag.pData, code, lino, _return, terrno)
×
UNCOV
349
      memcpy(tmpTag.pData, pSrcTag->pData, tmpTag.nData);
×
350
    } else {
UNCOV
351
      tmpTag.i64 = pSrcTag->i64;
×
352
    }
353

UNCOV
354
    QUERY_CHECK_NULL(taosArrayPush(pBasic->tagList, &tmpTag), code, lino, _return, terrno)
×
UNCOV
355
    tmpTag = (STagVal){0};
×
356
  }
357

UNCOV
358
  return code;
×
359
_return:
×
360
  if (pBasic->tagList) {
×
361
    taosArrayDestroyEx(pBasic->tagList, destroyTagVal);
×
362
    pBasic->tagList = NULL;
×
363
  }
364
  if (tmpTag.pData) {
×
365
    taosMemoryFree(tmpTag.pData);
×
366
  }
367
  qError("%s failed at line: %d, code: %d", __func__, lino, code);
×
368
  return code;
×
369
}
370

UNCOV
371
static int32_t buildBatchOrgTbInfoForExchangeBasicParam(SExchangeOperatorBasicParam* pBasic, SArray* pOrgTbInfoArray) {
×
UNCOV
372
  int32_t     code = TSDB_CODE_SUCCESS;
×
UNCOV
373
  int32_t     lino = 0;
×
UNCOV
374
  SOrgTbInfo  batchInfo;
×
375

UNCOV
376
  pBasic->batchOrgTbInfo = taosArrayInit(1, sizeof(SOrgTbInfo));
×
UNCOV
377
  QUERY_CHECK_NULL(pBasic->batchOrgTbInfo, code, lino, _return, terrno)
×
378

UNCOV
379
  for (int32_t i = 0; i < taosArrayGetSize(pOrgTbInfoArray); ++i) {
×
UNCOV
380
    SOrgTbInfo* pSrc = (SOrgTbInfo*)taosArrayGet(pOrgTbInfoArray, i);
×
UNCOV
381
    QUERY_CHECK_NULL(pSrc, code, lino, _return, terrno)
×
UNCOV
382
    batchInfo.vgId = pSrc->vgId;
×
UNCOV
383
    tstrncpy(batchInfo.tbName, pSrc->tbName, TSDB_TABLE_FNAME_LEN);
×
UNCOV
384
    batchInfo.colMap = taosArrayDup(pSrc->colMap, NULL);
×
UNCOV
385
    QUERY_CHECK_NULL(batchInfo.colMap, code, lino, _return, terrno)
×
UNCOV
386
    QUERY_CHECK_NULL(taosArrayPush(pBasic->batchOrgTbInfo, &batchInfo), code, lino, _return, terrno)
×
UNCOV
387
    batchInfo = (SOrgTbInfo){0};
×
388
  }
389

UNCOV
390
  return code;
×
391
_return:
×
392
  qError("%s failed at line: %d, code: %d", __func__, lino, code);
×
393
  if (pBasic->batchOrgTbInfo) {
×
394
    taosArrayDestroyEx(pBasic->batchOrgTbInfo, destroySOrgTbInfo);
×
395
    pBasic->batchOrgTbInfo = NULL;
×
396
  }
397
  if (batchInfo.colMap) {
×
398
    taosArrayDestroy(batchInfo.colMap);
×
399
    batchInfo.colMap = NULL;
×
400
  }
401
  return code;
×
402
}
403

404
static int32_t buildGroupCacheOperatorParam(SOperatorParam** ppRes, int32_t downstreamIdx, int32_t vgId, int64_t tbUid, bool needCache, SOperatorParam* pChild) {
7,139,396✔
405
  int32_t code = TSDB_CODE_SUCCESS;
7,139,396✔
406
  *ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
7,139,396✔
407
  if (NULL == *ppRes) {
7,139,396✔
408
    code = terrno;
×
409
    freeOperatorParam(pChild, OP_GET_PARAM);
×
410
    return code;
×
411
  }
412
  if (pChild) {
7,139,396✔
413
    (*ppRes)->pChildren = taosArrayInit(1, POINTER_BYTES);
138,104✔
414
    if (NULL == (*ppRes)->pChildren) {
138,104✔
415
      code = terrno;
×
416
      freeOperatorParam(pChild, OP_GET_PARAM);
×
417
      freeOperatorParam(*ppRes, OP_GET_PARAM);
×
418
      *ppRes = NULL;
×
419
      return code;
×
420
    }
421
    if (NULL == taosArrayPush((*ppRes)->pChildren, &pChild)) {
276,208✔
422
      code = terrno;
×
423
      freeOperatorParam(pChild, OP_GET_PARAM);
×
424
      freeOperatorParam(*ppRes, OP_GET_PARAM);
×
425
      *ppRes = NULL;
×
426
      return code;
×
427
    }
428
  } else {
429
    (*ppRes)->pChildren = NULL;
7,001,292✔
430
  }
431

432
  SGcOperatorParam* pGc = taosMemoryMalloc(sizeof(SGcOperatorParam));
7,139,396✔
433
  if (NULL == pGc) {
7,139,396✔
434
    code = terrno;
×
435
    freeOperatorParam(*ppRes, OP_GET_PARAM);
×
436
    *ppRes = NULL;
×
437
    return code;
×
438
  }
439

440
  pGc->sessionId = atomic_add_fetch_64(&gSessionId, 1);
7,139,396✔
441
  pGc->downstreamIdx = downstreamIdx;
7,139,396✔
442
  pGc->vgId = vgId;
7,139,396✔
443
  pGc->tbUid = tbUid;
7,139,396✔
444
  pGc->needCache = needCache;
7,139,396✔
445

446
  (*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_GROUP_CACHE;
7,139,396✔
447
  (*ppRes)->downstreamIdx = downstreamIdx;
7,139,396✔
448
  (*ppRes)->value = pGc;
7,139,396✔
449
  (*ppRes)->reUse = false;
7,139,396✔
450

451
  return TSDB_CODE_SUCCESS;
7,139,396✔
452
}
453

454

455
static int32_t buildGroupCacheNotifyOperatorParam(SOperatorParam** ppRes, int32_t downstreamIdx, int32_t vgId, int64_t tbUid) {
×
456
  int32_t code = TSDB_CODE_SUCCESS;
×
457
  *ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
×
458
  if (NULL == *ppRes) {
×
459
    return terrno;
×
460
  }
461
  (*ppRes)->pChildren = NULL;
×
462

463
  SGcNotifyOperatorParam* pGc = taosMemoryMalloc(sizeof(SGcNotifyOperatorParam));
×
464
  if (NULL == pGc) {
×
465
    code = terrno;
×
466
    freeOperatorParam(*ppRes, OP_NOTIFY_PARAM);
×
467
    return code;
×
468
  }
469

470
  pGc->downstreamIdx = downstreamIdx;
×
471
  pGc->vgId = vgId;
×
472
  pGc->tbUid = tbUid;
×
473

474
  (*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_GROUP_CACHE;
×
475
  (*ppRes)->downstreamIdx = downstreamIdx;
×
476
  (*ppRes)->value = pGc;
×
477
  (*ppRes)->reUse = false;
×
478

479
  return TSDB_CODE_SUCCESS;
×
480
}
481

482
static int32_t buildExchangeOperatorBasicParam(SExchangeOperatorBasicParam* pBasic, ENodeType srcOpType,
2,323,097✔
483
                                               EExchangeSourceType exchangeType, int32_t vgId, uint64_t groupId,
484
                                               SArray* pUidList, SOrgTbInfo* pOrgTbInfo, SArray* pTagList,
485
                                               SArray* pOrgTbInfoArray, STimeWindow window,
486
                                               SDownstreamSourceNode* pDownstreamSourceNode,
487
                                               bool tableSeq, bool isNewParam, bool isNewDeployed) {
488
  int32_t code = TSDB_CODE_SUCCESS;
2,323,097✔
489
  int32_t lino = 0;
2,323,097✔
490

491
  qDebug("buildExchangeOperatorBasicParam, srcOpType:%d, exchangeType:%d, vgId:%d, groupId:%" PRIu64 ", tableSeq:%d, "
2,323,097✔
492
         "isNewParam:%d, isNewDeployed:%d", srcOpType, exchangeType, vgId, groupId, tableSeq, isNewParam, isNewDeployed);
493

494
  pBasic->srcOpType = srcOpType;
2,323,097✔
495
  pBasic->vgId = vgId;
2,323,097✔
496
  pBasic->groupid = groupId;
2,323,097✔
497
  pBasic->window = window;
2,323,097✔
498
  pBasic->tableSeq = tableSeq;
2,323,097✔
499
  pBasic->type = exchangeType;
2,323,097✔
500
  pBasic->isNewParam = isNewParam;
2,323,097✔
501

502
  if (pDownstreamSourceNode) {
2,323,097✔
503
    pBasic->isNewDeployed = true;
2,189✔
504
    pBasic->newDeployedSrc.type = QUERY_NODE_DOWNSTREAM_SOURCE;
2,189✔
505
    pBasic->newDeployedSrc.clientId = pDownstreamSourceNode->clientId;// current task's taskid
2,189✔
506
    pBasic->newDeployedSrc.taskId = pDownstreamSourceNode->taskId;
2,189✔
507
    pBasic->newDeployedSrc.fetchMsgType = TDMT_STREAM_FETCH;
2,189✔
508
    pBasic->newDeployedSrc.localExec = false;
2,189✔
509
    pBasic->newDeployedSrc.addr.nodeId = pDownstreamSourceNode->addr.nodeId;
2,189✔
510
    memcpy(&pBasic->newDeployedSrc.addr.epSet, &pDownstreamSourceNode->addr.epSet, sizeof(SEpSet));
2,189✔
511
  } else {
512
    pBasic->isNewDeployed = false;
2,320,908✔
513
    pBasic->newDeployedSrc = (SDownstreamSourceNode){0};
2,320,908✔
514
  }
515

516
  if (pUidList) {
2,323,097✔
517
    pBasic->uidList = taosArrayDup(pUidList, NULL);
1,035,148✔
518
    QUERY_CHECK_NULL(pBasic->uidList, code, lino, _return, terrno)
1,035,148✔
519
  } else {
520
    pBasic->uidList = taosArrayInit(1, sizeof(int64_t));
1,287,949✔
521
    QUERY_CHECK_NULL(pBasic->uidList, code, lino, _return, terrno)
1,287,949✔
522
  }
523

524
  if (pOrgTbInfo) {
2,323,097✔
525
    code = copyOrgTbInfo(pOrgTbInfo, &pBasic->orgTbInfo);
1,287,949✔
526
    QUERY_CHECK_CODE(code, lino, _return);
1,287,949✔
527
  } else {
528
    pBasic->orgTbInfo = NULL;
1,035,148✔
529
  }
530

531
  if (pTagList) {
2,323,097✔
UNCOV
532
    code = buildTagListForExchangeBasicParam(pBasic, pTagList);
×
UNCOV
533
    QUERY_CHECK_CODE(code, lino, _return);
×
534
  } else {
535
    pBasic->tagList = NULL;
2,323,097✔
536
  }
537

538
  if (pOrgTbInfoArray) {
2,323,097✔
UNCOV
539
    code = buildBatchOrgTbInfoForExchangeBasicParam(pBasic, pOrgTbInfoArray);
×
UNCOV
540
    QUERY_CHECK_CODE(code, lino, _return);
×
541
  } else {
542
    pBasic->batchOrgTbInfo = NULL;
2,323,097✔
543
  }
544
  return code;
2,323,097✔
545

546
_return:
×
547
  qError("%s failed at line: %d, code: %d", __func__, lino, code);
×
548
  freeExchangeGetBasicOperatorParam(pBasic);
×
549
  return code;
×
550
}
551

552
static int32_t buildExchangeOperatorParamImpl(SOperatorParam** ppRes, int32_t downstreamIdx, ENodeType srcOpType,
2,104,719✔
553
                                              EExchangeSourceType exchangeType, int32_t vgId, uint64_t groupId,
554
                                              SArray* pUidList, SOrgTbInfo* pOrgTbInfo, SArray* pTagList,
555
                                              SArray* pOrgTbInfoArray, STimeWindow window,
556
                                              SDownstreamSourceNode* pDownstreamSourceNode,
557
                                              bool tableSeq, bool isNewParam, bool reUse, bool isNewDeployed) {
558

559
  int32_t                      code = TSDB_CODE_SUCCESS;
2,104,719✔
560
  int32_t                      lino = 0;
2,104,719✔
561
  SOperatorParam*              pParam = NULL;
2,104,719✔
562
  SExchangeOperatorParam*      pExc = NULL;
2,104,719✔
563

564
  pParam = taosMemoryMalloc(sizeof(SOperatorParam));
2,104,719✔
565
  QUERY_CHECK_NULL(pParam, code, lino, _return, terrno)
2,104,719✔
566

567
  pParam->opType = QUERY_NODE_PHYSICAL_PLAN_EXCHANGE;
2,104,719✔
568
  pParam->downstreamIdx = downstreamIdx;
2,104,719✔
569
  pParam->reUse = reUse;
2,104,719✔
570
  pParam->pChildren = NULL;
2,104,719✔
571
  pParam->value = taosMemoryMalloc(sizeof(SExchangeOperatorParam));
2,104,719✔
572
  QUERY_CHECK_NULL(pParam->value, code, lino, _return, terrno)
2,104,719✔
573

574
  pExc = (SExchangeOperatorParam*)pParam->value;
2,104,719✔
575
  pExc->multiParams = false;
2,104,719✔
576

577
  code = buildExchangeOperatorBasicParam(&pExc->basic, srcOpType, exchangeType, vgId, groupId,
2,104,719✔
578
                                         pUidList, pOrgTbInfo, pTagList, pOrgTbInfoArray,
579
                                         window, pDownstreamSourceNode, tableSeq, isNewParam, isNewDeployed);
580

581
  *ppRes = pParam;
2,104,719✔
582
  return code;
2,104,719✔
583
_return:
×
584
  qError("%s failed at line: %d, code: %d", __func__, lino, code);
×
585
  if (pParam) {
×
586
    freeOperatorParam(pParam, OP_GET_PARAM);
×
587
  }
588
  return code;
×
589
}
590

591
static int32_t buildExchangeOperatorParam(SOperatorParam** ppRes, int32_t downstreamIdx, const int32_t* pVgId, int64_t* pUid) {
4,236✔
592
  int32_t code = TSDB_CODE_SUCCESS;
4,236✔
593
  int32_t lino = 0;
4,236✔
594

595
  SArray* pUidList = taosArrayInit(1, sizeof(int64_t));
4,236✔
596
  QUERY_CHECK_NULL(pUidList, code, lino, _return, terrno)
4,236✔
597

598
  QUERY_CHECK_NULL(taosArrayPush(pUidList, pUid), code, lino, _return, terrno);
4,236✔
599

600
  code = buildExchangeOperatorParamImpl(ppRes, downstreamIdx, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, EX_SRC_TYPE_STB_JOIN_SCAN,
4,236✔
601
                                        *pVgId, 0, pUidList, NULL, NULL, NULL, (STimeWindow){0}, NULL, true, false, false, false);
4,236✔
602
  QUERY_CHECK_CODE(code, lino, _return);
4,236✔
603

604
_return:
4,236✔
605
  if (code) {
4,236✔
606
    qError("failed to build exchange operator param, code:%d", code);
×
607
  }
608
  taosArrayDestroy(pUidList);
4,236✔
609
  return code;
4,236✔
610
}
611

612
static int32_t buildExchangeOperatorParamForExternalWindow(SOperatorParam** ppRes, int32_t downstreamIdx, STimeWindow win) {
×
613
  int32_t                   code = TSDB_CODE_SUCCESS;
×
614
  int32_t                   lino = 0;
×
615

616
  code = buildExchangeOperatorParamImpl(ppRes, downstreamIdx, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, EX_SRC_TYPE_VSTB_WIN_SCAN,
×
617
                                        0, 0, NULL, NULL, NULL, NULL, win, NULL, true, true, true, false);
618
  QUERY_CHECK_CODE(code, lino, _return);
×
619

620
  return code;
×
621
_return:
×
622
  qError("failed to build exchange operator param for external window, code:%d", code);
×
623
  return code;
×
624
}
625

626
static int32_t buildExchangeOperatorParamForVTagScan(SOperatorParam** ppRes, int32_t downstreamIdx, int32_t vgId, tb_uid_t uid) {
812,534✔
627
  int32_t                      code = TSDB_CODE_SUCCESS;
812,534✔
628
  int32_t                      lino = 0;
812,534✔
629
  SArray*                      pUidList = NULL;
812,534✔
630

631
  pUidList = taosArrayInit(1, sizeof(int64_t));
812,534✔
632
  QUERY_CHECK_NULL(pUidList, code, lino, _return, terrno)
812,534✔
633

634
  QUERY_CHECK_NULL(taosArrayPush(pUidList, &uid), code, lino, _return, terrno)
812,534✔
635

636
  code = buildExchangeOperatorParamImpl(ppRes, downstreamIdx, QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN, EX_SRC_TYPE_VSTB_TAG_SCAN,
812,534✔
637
                                        vgId, 0, pUidList, NULL, NULL, NULL, (STimeWindow){0}, NULL, false, false, true, false);
812,534✔
638
  QUERY_CHECK_CODE(code, lino, _return);
812,534✔
639

640
_return:
812,534✔
641
  if (code) {
812,534✔
642
    qError("failed to build exchange operator param for tag scan, code:%d", code);
×
643
  }
644
  taosArrayDestroy(pUidList);
812,534✔
645
  return code;
812,534✔
646
}
647

648
static int32_t buildExchangeOperatorParamForVScan(SOperatorParam** ppRes, int32_t downstreamIdx, SOrgTbInfo* pOrgTbInfo,
1,287,949✔
649
                                                  SDownstreamSourceNode* pNewSource) {
650
  int32_t                      code = TSDB_CODE_SUCCESS;
1,287,949✔
651
  int32_t                      lino = 0;
1,287,949✔
652

653
  code = buildExchangeOperatorParamImpl(ppRes, downstreamIdx, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, EX_SRC_TYPE_VSTB_SCAN,
1,287,949✔
654
                                        pOrgTbInfo->vgId, 0, NULL, pOrgTbInfo, NULL, NULL, (STimeWindow){0}, pNewSource, false, true, true, true);
1,287,949✔
655
  QUERY_CHECK_CODE(code, lino, _return);
1,287,949✔
656

657
  return code;
1,287,949✔
658
_return:
×
659
  qError("failed to build exchange operator param for vscan, code:%d", code);
×
660
  return code;
×
661
}
662

663
static int32_t buildBatchExchangeOperatorParam(SOperatorParam** ppRes, int32_t downstreamIdx, SSHashObj* pVg) {
123,068✔
664
  int32_t                       code = TSDB_CODE_SUCCESS;
123,068✔
665
  int32_t                       line = 0;
123,068✔
666
  SOperatorParam*               pParam = NULL;
123,068✔
667
  SExchangeOperatorBatchParam*  pExc = NULL;
123,068✔
668
  SExchangeOperatorBasicParam   basic = {0};
123,068✔
669

670
  pParam = taosMemoryMalloc(sizeof(SOperatorParam));
123,068✔
671
  QUERY_CHECK_NULL(pParam, code, line, _return, terrno);
123,068✔
672

673
  pParam->opType = QUERY_NODE_PHYSICAL_PLAN_EXCHANGE;
123,068✔
674
  pParam->downstreamIdx = downstreamIdx;
123,068✔
675
  pParam->reUse = false;
123,068✔
676
  pParam->pChildren = NULL;
123,068✔
677
  pParam->value = taosMemoryMalloc(sizeof(SExchangeOperatorParam));
123,068✔
678
  QUERY_CHECK_NULL(pParam->value, code, line, _return, terrno);
123,068✔
679

680
  pExc = pParam->value;
123,068✔
681
  pExc->multiParams = true;
123,068✔
682
  pExc->pBatchs = tSimpleHashInit(tSimpleHashGetSize(pVg), taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
123,068✔
683
  QUERY_CHECK_NULL(pExc->pBatchs, code, line, _return, terrno)
123,068✔
684

685
  tSimpleHashSetFreeFp(pExc->pBatchs, freeExchangeGetBasicOperatorParam);
123,068✔
686

687
  int32_t iter = 0;
123,068✔
688
  void*   p = NULL;
123,068✔
689
  while (NULL != (p = tSimpleHashIterate(pVg, p, &iter))) {
341,446✔
690
    int32_t* pVgId = tSimpleHashGetKey(p, NULL);
218,378✔
691
    SArray*  pUidList = *(SArray**)p;
218,378✔
692

693
    code = buildExchangeOperatorBasicParam(&basic, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN,
218,378✔
694
                                           EX_SRC_TYPE_STB_JOIN_SCAN, *pVgId, 0,
695
                                           pUidList, NULL, NULL, NULL,
696
                                           (STimeWindow){0}, NULL, false, false, false);
218,378✔
697
    QUERY_CHECK_CODE(code, line, _return);
218,378✔
698

699
    // already transferred to batch param, can free here
700
    taosArrayDestroy(pUidList);
218,378✔
701

702
    QRY_ERR_RET(tSimpleHashPut(pExc->pBatchs, pVgId, sizeof(*pVgId), &basic, sizeof(basic)));
218,378✔
703

704
    basic = (SExchangeOperatorBasicParam){0};
218,378✔
705
    qTrace("build downstreamIdx %d batch scan, vgId:%d, uidNum:%" PRId64, downstreamIdx, *pVgId, (int64_t)taosArrayGetSize(pUidList));
218,378✔
706
    *(SArray**)p = NULL;
218,378✔
707
  }
708
  *ppRes = pParam;
123,068✔
709

710
  return code;
123,068✔
711
  
712
_return:
×
713
  qError("failed to build batch exchange operator param, code:%d", code);
×
714
  freeOperatorParam(pParam, OP_GET_PARAM);
×
715
  freeExchangeGetBasicOperatorParam(&basic);
×
716
  return code;
×
717
}
718

UNCOV
719
static int32_t buildBatchExchangeOperatorParamForVSAgg(SOperatorParam** ppRes, int32_t downstreamIdx, SArray* pTagList, uint64_t groupid,  SHashObj* pBatchMaps) {
×
UNCOV
720
  int32_t                       code = TSDB_CODE_SUCCESS;
×
UNCOV
721
  int32_t                       lino = 0;
×
UNCOV
722
  SOperatorParam*               pParam = NULL;
×
UNCOV
723
  SExchangeOperatorBatchParam*  pExc = NULL;
×
UNCOV
724
  SExchangeOperatorBasicParam   basic = {0};
×
725

UNCOV
726
  pParam = taosMemoryMalloc(sizeof(SOperatorParam));
×
UNCOV
727
  QUERY_CHECK_NULL(pParam, code, lino, _return, terrno)
×
728

UNCOV
729
  pParam->value = taosMemoryMalloc(sizeof(SExchangeOperatorParam));
×
UNCOV
730
  QUERY_CHECK_NULL(pParam->value, code, lino, _return, terrno)
×
731

UNCOV
732
  pExc = pParam->value;
×
UNCOV
733
  pExc->multiParams = true;
×
734

UNCOV
735
  pExc->pBatchs = tSimpleHashInit(taosHashGetSize(pBatchMaps), taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
×
UNCOV
736
  QUERY_CHECK_NULL(pExc->pBatchs, code, lino, _return, terrno)
×
UNCOV
737
  tSimpleHashSetFreeFp(pExc->pBatchs, freeExchangeGetBasicOperatorParam);
×
738

UNCOV
739
  size_t keyLen = 0;
×
UNCOV
740
  void*  pIter = taosHashIterate(pBatchMaps, NULL);
×
UNCOV
741
  while (pIter != NULL) {
×
UNCOV
742
    SArray*          pOrgTbInfoArray = *(SArray**)pIter;
×
UNCOV
743
    int32_t*         vgId = (int32_t*)taosHashGetKey(pIter, &keyLen);
×
UNCOV
744
    STimeWindow      win = {.skey = INT64_MAX, .ekey = INT64_MIN};
×
745

UNCOV
746
    code = buildExchangeOperatorBasicParam(&basic, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN,
×
747
                                           EX_SRC_TYPE_VSTB_AGG_SCAN, *vgId, groupid,
748
                                           NULL, NULL, pTagList, pOrgTbInfoArray,
749
                                           win, NULL, false, true, false);
UNCOV
750
    QUERY_CHECK_CODE(code, lino, _return);
×
751

UNCOV
752
    code = tSimpleHashPut(pExc->pBatchs, vgId, sizeof(*vgId), &basic, sizeof(basic));
×
UNCOV
753
    QUERY_CHECK_CODE(code, lino, _return);
×
754

UNCOV
755
    basic = (SExchangeOperatorBasicParam){0};
×
UNCOV
756
    pIter = taosHashIterate(pBatchMaps, pIter);
×
757
  }
758

UNCOV
759
  pParam->pChildren = NULL;
×
UNCOV
760
  pParam->opType = QUERY_NODE_PHYSICAL_PLAN_EXCHANGE;
×
UNCOV
761
  pParam->downstreamIdx = downstreamIdx;
×
UNCOV
762
  pParam->reUse = false;
×
763

UNCOV
764
  *ppRes = pParam;
×
UNCOV
765
  return code;
×
766

767
_return:
×
768
  qError("failed to build exchange operator param for vscan, code:%d", code);
×
769
  freeOperatorParam(pParam, OP_GET_PARAM);
×
770
  freeExchangeGetBasicOperatorParam(&basic);
×
771
  return code;
×
772
}
773

774
static int32_t buildMergeJoinOperatorParam(SOperatorParam** ppRes, bool initParam, SOperatorParam** ppChild0, SOperatorParam** ppChild1) {
3,569,698✔
775
  int32_t code = TSDB_CODE_SUCCESS;
3,569,698✔
776
  *ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
3,569,698✔
777
  if (NULL == *ppRes) {
3,569,698✔
778
    code = terrno;
×
779
    return code;
×
780
  }
781
  (*ppRes)->pChildren = taosArrayInit(2, POINTER_BYTES);
3,569,698✔
782
  if (NULL == (*ppRes)->pChildren) {
3,569,698✔
783
    code = terrno;
×
784
    freeOperatorParam(*ppRes, OP_GET_PARAM);
×
785
    *ppRes = NULL;
×
786
    return code;
×
787
  }
788
  if (NULL == taosArrayPush((*ppRes)->pChildren, ppChild0)) {
7,139,396✔
789
    code = terrno;
×
790
    freeOperatorParam(*ppRes, OP_GET_PARAM);
×
791
    *ppRes = NULL;
×
792
    return code;
×
793
  }
794
  *ppChild0 = NULL;
3,569,698✔
795
  if (NULL == taosArrayPush((*ppRes)->pChildren, ppChild1)) {
7,139,396✔
796
    code = terrno;
×
797
    freeOperatorParam(*ppRes, OP_GET_PARAM);
×
798
    *ppRes = NULL;
×
799
    return code;
×
800
  }
801
  *ppChild1 = NULL;
3,569,698✔
802
  
803
  SSortMergeJoinOperatorParam* pJoin = taosMemoryMalloc(sizeof(SSortMergeJoinOperatorParam));
3,569,698✔
804
  if (NULL == pJoin) {
3,569,698✔
805
    code = terrno;
×
806
    freeOperatorParam(*ppRes, OP_GET_PARAM);
×
807
    *ppRes = NULL;
×
808
    return code;
×
809
  }
810

811
  pJoin->initDownstream = initParam;
3,569,698✔
812
  
813
  (*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_MERGE_JOIN;
3,569,698✔
814
  (*ppRes)->value = pJoin;
3,569,698✔
815
  (*ppRes)->reUse = false;
3,569,698✔
816

817
  return TSDB_CODE_SUCCESS;
3,569,698✔
818
}
819

820
static int32_t buildMergeJoinNotifyOperatorParam(SOperatorParam** ppRes, SOperatorParam* pChild0, SOperatorParam* pChild1) {
×
821
  int32_t code = TSDB_CODE_SUCCESS;
×
822
  *ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
×
823
  if (NULL == *ppRes) {
×
824
    code = terrno;
×
825
    freeOperatorParam(pChild0, OP_NOTIFY_PARAM);
×
826
    freeOperatorParam(pChild1, OP_NOTIFY_PARAM);
×
827
    return code;
×
828
  }
829
  (*ppRes)->pChildren = taosArrayInit(2, POINTER_BYTES);
×
830
  if (NULL == *ppRes) {
×
831
    code = terrno;
×
832
    taosMemoryFreeClear(*ppRes);
×
833
    freeOperatorParam(pChild0, OP_NOTIFY_PARAM);
×
834
    freeOperatorParam(pChild1, OP_NOTIFY_PARAM);
×
835
    return code;
×
836
  }
837
  if (pChild0 && NULL == taosArrayPush((*ppRes)->pChildren, &pChild0)) {
×
838
    code = terrno;
×
839
    freeOperatorParam(*ppRes, OP_NOTIFY_PARAM);
×
840
    freeOperatorParam(pChild0, OP_NOTIFY_PARAM);
×
841
    freeOperatorParam(pChild1, OP_NOTIFY_PARAM);
×
842
    *ppRes = NULL;
×
843
    return code;
×
844
  }
845
  if (pChild1 && NULL == taosArrayPush((*ppRes)->pChildren, &pChild1)) {
×
846
    code = terrno;
×
847
    freeOperatorParam(*ppRes, OP_NOTIFY_PARAM);
×
848
    freeOperatorParam(pChild1, OP_NOTIFY_PARAM);
×
849
    *ppRes = NULL;
×
850
    return code;
×
851
  }
852
  
853
  (*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_MERGE_JOIN;
×
854
  (*ppRes)->value = NULL;
×
855
  (*ppRes)->reUse = false;
×
856

857
  return TSDB_CODE_SUCCESS;
×
858
}
859

860
static int32_t buildBatchTableScanOperatorParam(SOperatorParam** ppRes, int32_t downstreamIdx, SSHashObj* pVg) {
10,800✔
861
  int32_t code = TSDB_CODE_SUCCESS;
10,800✔
862
  int32_t vgNum = tSimpleHashGetSize(pVg);
10,800✔
863
  if (vgNum <= 0 || vgNum > 1) {
10,800✔
864
    qError("Invalid vgroup num %d to build table scan operator param", vgNum);
×
865
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
866
  }
867

868
  int32_t iter = 0;
10,800✔
869
  void* p = NULL;
10,800✔
870
  while (NULL != (p = tSimpleHashIterate(pVg, p, &iter))) {
21,600✔
871
    SArray* pUidList = *(SArray**)p;
10,800✔
872

873
    code = buildTableScanOperatorParam(ppRes, pUidList, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false);
10,800✔
874
    if (code) {
10,800✔
875
      return code;
×
876
    }
877
    taosArrayDestroy(pUidList);
10,800✔
878
    *(SArray**)p = NULL;
10,800✔
879
  }
880
  
881
  return TSDB_CODE_SUCCESS;
10,800✔
882
}
883

884
static int32_t buildSingleTableScanOperatorParam(SOperatorParam** ppRes, int32_t downstreamIdx, int32_t* pVgId, int64_t* pUid) {
×
885
  SArray* pUidList = taosArrayInit(1, sizeof(int64_t));
×
886
  if (NULL == pUidList) {
×
887
    return terrno;
×
888
  }
889
  if (NULL == taosArrayPush(pUidList, pUid)) {
×
890
    return terrno;
×
891
  }
892

893
  int32_t code = buildTableScanOperatorParam(ppRes, pUidList, QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, true);
×
894
  taosArrayDestroy(pUidList);
×
895
  if (code) {
×
896
    return code;
×
897
  }
898
  
899
  return TSDB_CODE_SUCCESS;
×
900
}
901

902
static int32_t buildSeqStbJoinOperatorParam(SDynQueryCtrlOperatorInfo* pInfo, SStbJoinPrevJoinCtx* pPrev, SStbJoinPostJoinCtx* pPost, SOperatorParam** ppParam) {
3,569,698✔
903
  int64_t                     rowIdx = pPrev->pListHead->readIdx;
3,569,698✔
904
  SOperatorParam*             pSrcParam0 = NULL;
3,569,698✔
905
  SOperatorParam*             pSrcParam1 = NULL;
3,569,698✔
906
  SOperatorParam*             pGcParam0 = NULL;
3,569,698✔
907
  SOperatorParam*             pGcParam1 = NULL;  
3,569,698✔
908
  int32_t*                    leftVg = pPrev->pListHead->pLeftVg + rowIdx;
3,569,698✔
909
  int64_t*                    leftUid = pPrev->pListHead->pLeftUid + rowIdx;
3,569,698✔
910
  int32_t*                    rightVg = pPrev->pListHead->pRightVg + rowIdx;
3,569,698✔
911
  int64_t*                    rightUid = pPrev->pListHead->pRightUid + rowIdx;
3,569,698✔
912
  int32_t                     code = TSDB_CODE_SUCCESS;
3,569,698✔
913

914
  qDebug("start %" PRId64 ":%" PRId64 "th stbJoin, left:%d,%" PRIu64 " - right:%d,%" PRIu64, 
3,569,698✔
915
      rowIdx, pPrev->tableNum, *leftVg, *leftUid, *rightVg, *rightUid);
916

917
  QRY_ERR_RET(updatePostJoinCurrTableInfo(&pInfo->stbJoin));
3,569,698✔
918
  
919
  if (pInfo->stbJoin.basic.batchFetch) {
3,569,698✔
920
    if (pPrev->leftHash) {
3,567,580✔
921
      code = pInfo->stbJoin.basic.srcScan[0] ? buildBatchTableScanOperatorParam(&pSrcParam0, 0, pPrev->leftHash) : buildBatchExchangeOperatorParam(&pSrcParam0, 0, pPrev->leftHash);
66,934✔
922
      if (TSDB_CODE_SUCCESS == code) {
66,934✔
923
        code = pInfo->stbJoin.basic.srcScan[1] ? buildBatchTableScanOperatorParam(&pSrcParam1, 1, pPrev->rightHash) : buildBatchExchangeOperatorParam(&pSrcParam1, 1, pPrev->rightHash);
66,934✔
924
      }
925
      if (TSDB_CODE_SUCCESS == code) {
66,934✔
926
        tSimpleHashCleanup(pPrev->leftHash);
66,934✔
927
        tSimpleHashCleanup(pPrev->rightHash);
66,934✔
928
        pPrev->leftHash = NULL;
66,934✔
929
        pPrev->rightHash = NULL;
66,934✔
930
      }
931
    }
932
  } else {
933
    code = pInfo->stbJoin.basic.srcScan[0] ? buildSingleTableScanOperatorParam(&pSrcParam0, 0, leftVg, leftUid) : buildExchangeOperatorParam(&pSrcParam0, 0, leftVg, leftUid);
2,118✔
934
    if (TSDB_CODE_SUCCESS == code) {
2,118✔
935
      code = pInfo->stbJoin.basic.srcScan[1] ? buildSingleTableScanOperatorParam(&pSrcParam1, 1, rightVg, rightUid) : buildExchangeOperatorParam(&pSrcParam1, 1, rightVg, rightUid);
2,118✔
936
    }
937
  }
938

939
  bool initParam = pSrcParam0 ? true : false;
3,569,698✔
940
  if (TSDB_CODE_SUCCESS == code) {
3,569,698✔
941
    code = buildGroupCacheOperatorParam(&pGcParam0, 0, *leftVg, *leftUid, pPost->leftNeedCache, pSrcParam0);
3,569,698✔
942
    pSrcParam0 = NULL;
3,569,698✔
943
  }
944
  if (TSDB_CODE_SUCCESS == code) {
3,569,698✔
945
    code = buildGroupCacheOperatorParam(&pGcParam1, 1, *rightVg, *rightUid, pPost->rightNeedCache, pSrcParam1);
3,569,698✔
946
    pSrcParam1 = NULL;
3,569,698✔
947
  }
948
  if (TSDB_CODE_SUCCESS == code) {
3,569,698✔
949
    code = buildMergeJoinOperatorParam(ppParam, initParam, &pGcParam0, &pGcParam1);
3,569,698✔
950
  }
951
  if (TSDB_CODE_SUCCESS != code) {
3,569,698✔
952
    if (pSrcParam0) {
×
953
      freeOperatorParam(pSrcParam0, OP_GET_PARAM);
×
954
    }
955
    if (pSrcParam1) {
×
956
      freeOperatorParam(pSrcParam1, OP_GET_PARAM);
×
957
    }
958
    if (pGcParam0) {
×
959
      freeOperatorParam(pGcParam0, OP_GET_PARAM);
×
960
    }
961
    if (pGcParam1) {
×
962
      freeOperatorParam(pGcParam1, OP_GET_PARAM);
×
963
    }
964
    if (*ppParam) {
×
965
      freeOperatorParam(*ppParam, OP_GET_PARAM);
×
966
      *ppParam = NULL;
×
967
    }
968
  }
969
  
970
  return code;
3,569,698✔
971
}
972

973
static int32_t buildVtbScanOperatorParam(SDynQueryCtrlOperatorInfo* pInfo, SOperatorParam** ppRes, uint64_t uid) {
812,534✔
974
  int32_t                   code = TSDB_CODE_SUCCESS;
812,534✔
975
  int32_t                   lino = 0;
812,534✔
976
  SVTableScanOperatorParam* pVScan = NULL;
812,534✔
977
  *ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
812,534✔
978
  QUERY_CHECK_NULL(*ppRes, code, lino, _return, terrno)
812,534✔
979

980
  (*ppRes)->pChildren = taosArrayInit(1, POINTER_BYTES);
812,534✔
981
  QUERY_CHECK_NULL((*ppRes)->pChildren, code, lino, _return, terrno)
812,534✔
982

983
  pVScan = taosMemoryMalloc(sizeof(SVTableScanOperatorParam));
812,534✔
984
  QUERY_CHECK_NULL(pVScan, code, lino, _return, terrno)
812,534✔
985
  pVScan->pOpParamArray = taosArrayInit(1, POINTER_BYTES);
812,534✔
986
  QUERY_CHECK_NULL(pVScan->pOpParamArray, code, lino, _return, terrno)
812,534✔
987
  pVScan->uid = uid;
812,534✔
988
  pVScan->window = pInfo->vtbScan.window;
812,534✔
989

990
  (*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_VIRTUAL_TABLE_SCAN;
812,534✔
991
  (*ppRes)->downstreamIdx = 0;
812,534✔
992
  (*ppRes)->value = pVScan;
812,534✔
993
  (*ppRes)->reUse = false;
812,534✔
994

995
  return TSDB_CODE_SUCCESS;
812,534✔
996
_return:
×
997
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
998
  if (pVScan) {
×
999
    taosArrayDestroy(pVScan->pOpParamArray);
×
1000
    taosMemoryFreeClear(pVScan);
×
1001
  }
1002
  if (*ppRes) {
×
1003
    taosArrayDestroy((*ppRes)->pChildren);
×
1004
    taosMemoryFreeClear(*ppRes);
×
1005
  }
1006
  return code;
×
1007
}
1008

1009
static int32_t buildExternalWindowOperatorParam(SDynQueryCtrlOperatorInfo* pInfo, SOperatorParam** ppRes, SArray* pWins, int32_t idx) {
×
1010
  int32_t                       code = TSDB_CODE_SUCCESS;
×
1011
  int32_t                       lino = 0;
×
1012
  SExternalWindowOperatorParam* pExtWinOp = NULL;
×
1013

1014
  *ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
×
1015
  QUERY_CHECK_NULL(*ppRes, code, lino, _return, terrno)
×
1016

1017
  pExtWinOp = taosMemoryMalloc(sizeof(SExternalWindowOperatorParam));
×
1018
  QUERY_CHECK_NULL(pExtWinOp, code, lino, _return, terrno)
×
1019

1020
  pExtWinOp->ExtWins = taosArrayDup(pWins, NULL);
×
1021
  QUERY_CHECK_NULL(pExtWinOp->ExtWins, code, lino, _return, terrno)
×
1022

1023
  SExtWinTimeWindow *firstWin = (SExtWinTimeWindow *)taosArrayGet(pWins, 0);
×
1024
  SExtWinTimeWindow *lastWin = (SExtWinTimeWindow *)taosArrayGet(pWins, taosArrayGetSize(pWins) - 1);
×
1025

1026
  (*ppRes)->pChildren = taosArrayInit(1, POINTER_BYTES);
×
1027
  QUERY_CHECK_NULL((*ppRes)->pChildren, code, lino, _return, terrno)
×
1028

1029
  SOperatorParam* pExchangeOperator = NULL;
×
1030
  STimeWindow     twin = {.skey = firstWin->tw.skey, .ekey = lastWin->tw.ekey};
×
1031
  code = buildExchangeOperatorParamForExternalWindow(&pExchangeOperator, 0, twin);
×
1032
  QUERY_CHECK_CODE(code, lino, _return);
×
1033
  QUERY_CHECK_NULL(taosArrayPush((*ppRes)->pChildren, &pExchangeOperator), code, lino, _return, terrno)
×
1034

1035
  (*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW;
×
1036
  (*ppRes)->downstreamIdx = idx;
×
1037
  (*ppRes)->value = pExtWinOp;
×
1038
  (*ppRes)->reUse = false;
×
1039

1040
  return code;
×
1041
_return:
×
1042
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1043
  if (pExtWinOp) {
×
1044
    if (pExtWinOp->ExtWins) {
×
1045
      taosArrayDestroy(pExtWinOp->ExtWins);
×
1046
    }
1047
    taosMemoryFree(pExtWinOp);
×
1048
  }
1049
  if (*ppRes) {
×
1050
    if ((*ppRes)->pChildren) {
×
1051
      taosArrayDestroy((*ppRes)->pChildren);
×
1052
    }
1053
    taosMemoryFree(*ppRes);
×
1054
    *ppRes = NULL;
×
1055
  }
1056
  return code;
×
1057
}
1058

1059
static int32_t buildMergeOperatorParam(SDynQueryCtrlOperatorInfo* pInfo, SOperatorParam** ppRes, SArray* pWins,
×
1060
                                       int32_t numOfDownstream, int32_t numOfWins) {
1061
  int32_t                   code = TSDB_CODE_SUCCESS;
×
1062
  int32_t                   lino = 0;
×
1063
  SMergeOperatorParam*      pMergeOp = NULL;
×
1064

1065
  *ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
×
1066
  QUERY_CHECK_NULL(*ppRes, code, lino, _return, terrno)
×
1067

1068
  (*ppRes)->pChildren = taosArrayInit(numOfDownstream, POINTER_BYTES);
×
1069
  QUERY_CHECK_NULL((*ppRes)->pChildren, code, lino, _return, terrno)
×
1070

1071
  pMergeOp = taosMemoryMalloc(sizeof(SMergeOperatorParam));
×
1072
  QUERY_CHECK_NULL(pMergeOp, code, lino, _return, terrno)
×
1073

1074
  pMergeOp->winNum = numOfWins;
×
1075

1076
  for (int32_t i = 0; i < numOfDownstream; i++) {
×
1077
    SOperatorParam* pExternalWinParam = NULL;
×
1078
    code = buildExternalWindowOperatorParam(pInfo, &pExternalWinParam, pWins, i);
×
1079
    QUERY_CHECK_CODE(code, lino, _return);
×
1080
    QUERY_CHECK_NULL(taosArrayPush((*ppRes)->pChildren, &pExternalWinParam), code, lino, _return, terrno)
×
1081
  }
1082

1083
  (*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_MERGE;
×
1084
  (*ppRes)->downstreamIdx = 0;
×
1085
  (*ppRes)->value = pMergeOp;
×
1086
  (*ppRes)->reUse = false;
×
1087

1088
  return TSDB_CODE_SUCCESS;
×
1089
_return:
×
1090
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1091
  if (pMergeOp) {
×
1092
    taosMemoryFree(pMergeOp);
×
1093
  }
1094
  if (*ppRes) {
×
1095
    if ((*ppRes)->pChildren) {
×
1096
      for (int32_t i = 0; i < taosArrayGetSize((*ppRes)->pChildren); i++) {
×
1097
        SOperatorParam* pChildParam = (SOperatorParam*)taosArrayGet((*ppRes)->pChildren, i);
×
1098
        if (pChildParam) {
×
1099
          SExternalWindowOperatorParam* pExtWinOp = (SExternalWindowOperatorParam*)pChildParam->value;
×
1100
          if (pExtWinOp) {
×
1101
            if (pExtWinOp->ExtWins) {
×
1102
              taosArrayDestroy(pExtWinOp->ExtWins);
×
1103
            }
1104
            taosMemoryFree(pExtWinOp);
×
1105
          }
1106
          taosMemoryFree(pChildParam);
×
1107
        }
1108
      }
1109
      taosArrayDestroy((*ppRes)->pChildren);
×
1110
    }
1111
    taosMemoryFree(*ppRes);
×
1112
    *ppRes = NULL;
×
1113
  }
1114
  return code;
×
1115
}
1116

UNCOV
1117
static int32_t buildAggOperatorParam(SDynQueryCtrlOperatorInfo* pInfo, SOperatorParam** ppRes) {
×
UNCOV
1118
  int32_t                   code = TSDB_CODE_SUCCESS;
×
UNCOV
1119
  int32_t                   lino = 0;
×
UNCOV
1120
  SOperatorParam*           pParam = NULL;
×
UNCOV
1121
  SOperatorParam*           pExchangeParam = NULL;
×
UNCOV
1122
  SVtbScanDynCtrlInfo*      pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
×
UNCOV
1123
  bool                      freeExchange = false;
×
1124

UNCOV
1125
  pParam = taosMemoryMalloc(sizeof(SOperatorParam));
×
UNCOV
1126
  QUERY_CHECK_NULL(pParam, code, lino, _return, terrno)
×
1127

UNCOV
1128
  pParam->pChildren = taosArrayInit(1, POINTER_BYTES);
×
UNCOV
1129
  QUERY_CHECK_NULL(pParam->pChildren, code, lino, _return, terrno)
×
1130

UNCOV
1131
  pParam->value = taosMemoryMalloc(sizeof(SAggOperatorParam));
×
UNCOV
1132
  QUERY_CHECK_NULL(pParam->value, code, lino, _return, terrno)
×
1133

UNCOV
1134
  code = buildBatchExchangeOperatorParamForVSAgg(&pExchangeParam, 0, NULL, 0, pVtbScan->otbVgIdToOtbInfoArrayMap);
×
UNCOV
1135
  QUERY_CHECK_CODE(code, lino, _return);
×
1136

UNCOV
1137
  freeExchange = true;
×
1138

UNCOV
1139
  QUERY_CHECK_NULL(taosArrayPush(pParam->pChildren, &pExchangeParam), code, lino, _return, terrno)
×
1140

UNCOV
1141
  freeExchange = false;
×
1142

UNCOV
1143
  pParam->opType = QUERY_NODE_PHYSICAL_PLAN_HASH_AGG;
×
UNCOV
1144
  pParam->downstreamIdx = 0;
×
UNCOV
1145
  pParam->reUse = false;
×
1146

UNCOV
1147
  *ppRes = pParam;
×
1148

UNCOV
1149
  return code;
×
1150
_return:
×
1151
  if (freeExchange) {
×
1152
    freeOperatorParam(pExchangeParam, OP_GET_PARAM);
×
1153
  }
1154
  if (pParam) {
×
1155
    freeOperatorParam(pParam, OP_GET_PARAM);
×
1156
  }
1157
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1158
  return code;
×
1159
}
1160

UNCOV
1161
static int32_t buildAggOperatorParamWithGroupId(SDynQueryCtrlOperatorInfo* pInfo, uint64_t groupid, SOperatorParam** ppRes) {
×
UNCOV
1162
  int32_t                   code = TSDB_CODE_SUCCESS;
×
UNCOV
1163
  int32_t                   lino = 0;
×
UNCOV
1164
  SVtbScanDynCtrlInfo*      pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
×
UNCOV
1165
  SOperatorParam*           pParam = NULL;
×
UNCOV
1166
  SOperatorParam*           pExchangeParam = NULL;
×
UNCOV
1167
  SHashObj*                 otbVgIdToOtbInfoArrayMap = NULL;
×
UNCOV
1168
  bool                      freeExchange = false;
×
UNCOV
1169
  void*                     pIter = taosHashGet(pVtbScan->vtbGroupIdToVgIdMapMap, &groupid, sizeof(groupid));
×
1170

UNCOV
1171
  if (!pIter) {
×
UNCOV
1172
    *ppRes = NULL;
×
UNCOV
1173
    return code;
×
1174
  }
1175

UNCOV
1176
  otbVgIdToOtbInfoArrayMap = *(SHashObj**)pIter;
×
1177

UNCOV
1178
  pParam = taosMemoryMalloc(sizeof(SOperatorParam));
×
UNCOV
1179
  QUERY_CHECK_NULL(pParam, code, lino, _return, terrno)
×
1180

UNCOV
1181
  pParam->pChildren = taosArrayInit(1, POINTER_BYTES);
×
UNCOV
1182
  QUERY_CHECK_NULL(pParam->pChildren, code, lino, _return, terrno)
×
1183

UNCOV
1184
  code = buildBatchExchangeOperatorParamForVSAgg(&pExchangeParam, 0, NULL, groupid, otbVgIdToOtbInfoArrayMap);
×
UNCOV
1185
  QUERY_CHECK_CODE(code, lino, _return);
×
1186

UNCOV
1187
  freeExchange = true;
×
1188

UNCOV
1189
  QUERY_CHECK_NULL(taosArrayPush(pParam->pChildren, &pExchangeParam), code, lino, _return, terrno)
×
1190

UNCOV
1191
  freeExchange = false;
×
1192

UNCOV
1193
  pParam->opType = QUERY_NODE_PHYSICAL_PLAN_HASH_AGG;
×
UNCOV
1194
  pParam->downstreamIdx = 0;
×
UNCOV
1195
  pParam->value = NULL;
×
UNCOV
1196
  pParam->reUse = false;
×
1197

UNCOV
1198
  *ppRes = pParam;
×
1199

UNCOV
1200
  return code;
×
1201
_return:
×
1202
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1203
  if (freeExchange) {
×
1204
    freeOperatorParam(pExchangeParam, OP_GET_PARAM);
×
1205
  }
1206
  if (pParam) {
×
1207
    freeOperatorParam(pParam, OP_GET_PARAM);
×
1208
  }
1209
  return code;
×
1210
}
1211

UNCOV
1212
static int32_t buildAggOperatorParamForSingleChild(SDynQueryCtrlOperatorInfo* pInfo, tb_uid_t uid, uint64_t groupid, SArray* pTagList, SOperatorParam** ppRes) {
×
UNCOV
1213
  int32_t                   code = TSDB_CODE_SUCCESS;
×
UNCOV
1214
  int32_t                   lino = 0;
×
UNCOV
1215
  SVtbScanDynCtrlInfo*      pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
×
UNCOV
1216
  SOperatorParam*           pParam = NULL;
×
UNCOV
1217
  SHashObj*                 pOtbVgIdToOtbInfoArrayMap = NULL;
×
UNCOV
1218
  void*                     pIter = taosHashGet(pVtbScan->vtbUidToVgIdMapMap, &uid, sizeof(uid));
×
1219

UNCOV
1220
  if (pIter) {
×
UNCOV
1221
    pOtbVgIdToOtbInfoArrayMap = *(SHashObj**)taosHashGet(pVtbScan->vtbUidToVgIdMapMap, &uid, sizeof(uid));
×
1222

UNCOV
1223
    code = buildBatchExchangeOperatorParamForVSAgg(&pParam, 0, pTagList, groupid, pOtbVgIdToOtbInfoArrayMap);
×
UNCOV
1224
    QUERY_CHECK_CODE(code, lino, _return);
×
1225

UNCOV
1226
    *ppRes = pParam;
×
1227
  } else {
UNCOV
1228
    *ppRes = NULL;
×
1229
  }
1230

UNCOV
1231
  return code;
×
1232
_return:
×
1233
  if (pParam) {
×
1234
    freeOperatorParam(pParam, OP_GET_PARAM);
×
1235
  }
1236
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1237
  return code;
×
1238
}
1239

1240
static void seqJoinLaunchNewRetrieveImpl(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
3,569,698✔
1241
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
3,569,698✔
1242
  SStbJoinDynCtrlInfo*       pStbJoin = (SStbJoinDynCtrlInfo*)&pInfo->stbJoin;
3,569,698✔
1243
  SStbJoinPrevJoinCtx*       pPrev = &pStbJoin->ctx.prev;
3,569,698✔
1244
  SStbJoinPostJoinCtx*       pPost = &pStbJoin->ctx.post;
3,569,698✔
1245
  SOperatorParam*            pParam = NULL;
3,569,698✔
1246
  int32_t                    code  = buildSeqStbJoinOperatorParam(pInfo, pPrev, pPost, &pParam);
3,569,698✔
1247
  if (TSDB_CODE_SUCCESS != code) {
3,569,698✔
1248
    pOperator->pTaskInfo->code = code;
×
1249
    T_LONG_JMP(pOperator->pTaskInfo->env, pOperator->pTaskInfo->code);
×
1250
  }
1251

1252
  qDebug("%s dynamic post task begin", GET_TASKID(pOperator->pTaskInfo));
3,569,698✔
1253
  code = pOperator->pDownstream[1]->fpSet.getNextExtFn(pOperator->pDownstream[1], pParam, ppRes);
3,569,698✔
1254
  if (*ppRes && (code == 0)) {
3,569,698✔
1255
    code = blockDataCheck(*ppRes);
198,934✔
1256
    if (code) {
198,934✔
1257
      qError("Invalid block data, blockDataCheck failed, error:%s", tstrerror(code));
×
1258
      pOperator->pTaskInfo->code = code;
×
1259
      T_LONG_JMP(pOperator->pTaskInfo->env, pOperator->pTaskInfo->code);
×
1260
    }
1261
    pPost->isStarted = true;
198,934✔
1262
    pStbJoin->execInfo.postBlkNum++;
198,934✔
1263
    pStbJoin->execInfo.postBlkRows += (*ppRes)->info.rows;
198,934✔
1264
    qDebug("%s join res block retrieved", GET_TASKID(pOperator->pTaskInfo));
198,934✔
1265
  } else {
1266
    qDebug("%s Empty join res block retrieved", GET_TASKID(pOperator->pTaskInfo));
3,370,764✔
1267
  }
1268
}
3,569,698✔
1269

1270

1271
static int32_t notifySeqJoinTableCacheEnd(SOperatorInfo* pOperator, SStbJoinPostJoinCtx* pPost, bool leftTable) {
×
1272
  SOperatorParam* pGcParam = NULL;
×
1273
  SOperatorParam* pMergeJoinParam = NULL;
×
1274
  int32_t         downstreamId = leftTable ? 0 : 1;
×
1275
  int32_t         vgId = leftTable ? pPost->leftVgId : pPost->rightVgId;
×
1276
  int64_t         uid = leftTable ? pPost->leftCurrUid : pPost->rightCurrUid;
×
1277

1278
  qDebug("notify table %" PRIu64 " in vgId %d downstreamId %d cache end", uid, vgId, downstreamId);
×
1279

1280
  int32_t code = buildGroupCacheNotifyOperatorParam(&pGcParam, downstreamId, vgId, uid);
×
1281
  if (TSDB_CODE_SUCCESS != code) {
×
1282
    return code;
×
1283
  }
1284
  code = buildMergeJoinNotifyOperatorParam(&pMergeJoinParam, pGcParam, NULL);
×
1285
  if (TSDB_CODE_SUCCESS != code) {
×
1286
    return code;
×
1287
  }
1288

1289
  return optrDefaultNotifyFn(pOperator->pDownstream[1], pMergeJoinParam);
×
1290
}
1291

1292
static int32_t handleSeqJoinCurrRetrieveEnd(SOperatorInfo* pOperator, SStbJoinDynCtrlInfo*          pStbJoin) {
3,569,215✔
1293
  SStbJoinPostJoinCtx* pPost = &pStbJoin->ctx.post;
3,569,215✔
1294
  int32_t code = 0;
3,569,215✔
1295
  
1296
  pPost->isStarted = false;
3,569,215✔
1297
  
1298
  if (pStbJoin->basic.batchFetch) {
3,569,215✔
1299
    return TSDB_CODE_SUCCESS;
3,567,097✔
1300
  }
1301
  
1302
  if (pPost->leftNeedCache) {
2,118✔
1303
    uint32_t* num = tSimpleHashGet(pStbJoin->ctx.prev.leftCache, &pPost->leftCurrUid, sizeof(pPost->leftCurrUid));
×
1304
    if (num && --(*num) <= 0) {
×
1305
      code = tSimpleHashRemove(pStbJoin->ctx.prev.leftCache, &pPost->leftCurrUid, sizeof(pPost->leftCurrUid));
×
1306
      if (code) {
×
1307
        qError("tSimpleHashRemove leftCurrUid %" PRId64 " from leftCache failed, error:%s", pPost->leftCurrUid, tstrerror(code));
×
1308
        QRY_ERR_RET(code);
×
1309
      }
1310
      QRY_ERR_RET(notifySeqJoinTableCacheEnd(pOperator, pPost, true));
×
1311
    }
1312
  }
1313
  
1314
  if (!pPost->rightNeedCache) {
2,118✔
1315
    void* v = tSimpleHashGet(pStbJoin->ctx.prev.rightCache, &pPost->rightCurrUid, sizeof(pPost->rightCurrUid));
2,118✔
1316
    if (NULL != v) {
2,118✔
1317
      code = tSimpleHashRemove(pStbJoin->ctx.prev.rightCache, &pPost->rightCurrUid, sizeof(pPost->rightCurrUid));
×
1318
      if (code) {
×
1319
        qError("tSimpleHashRemove rightCurrUid %" PRId64 " from rightCache failed, error:%s", pPost->rightCurrUid, tstrerror(code));
×
1320
        QRY_ERR_RET(code);
×
1321
      }
1322
      QRY_ERR_RET(notifySeqJoinTableCacheEnd(pOperator, pPost, false));
×
1323
    }
1324
  }
1325

1326
  return TSDB_CODE_SUCCESS;
2,118✔
1327
}
1328

1329

1330
static FORCE_INLINE int32_t seqJoinContinueCurrRetrieve(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
1331
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
266,444✔
1332
  SStbJoinPostJoinCtx*       pPost = &pInfo->stbJoin.ctx.post;
266,444✔
1333
  SStbJoinPrevJoinCtx*       pPrev = &pInfo->stbJoin.ctx.prev;
266,444✔
1334

1335
  if (!pPost->isStarted) {
266,444✔
1336
    return TSDB_CODE_SUCCESS;
67,993✔
1337
  }
1338
  
1339
  qDebug("%s dynQueryCtrl continue to retrieve block from post op", GET_TASKID(pOperator->pTaskInfo));
198,451✔
1340
  
1341
  *ppRes = getNextBlockFromDownstream(pOperator, 1);
198,451✔
1342
  if (NULL == *ppRes) {
198,451✔
1343
    QRY_ERR_RET(handleSeqJoinCurrRetrieveEnd(pOperator, &pInfo->stbJoin));
198,451✔
1344
    pPrev->pListHead->readIdx++;
198,451✔
1345
  } else {
1346
    pInfo->stbJoin.execInfo.postBlkNum++;
×
1347
    pInfo->stbJoin.execInfo.postBlkRows += (*ppRes)->info.rows;
×
1348
  }
1349

1350
  return TSDB_CODE_SUCCESS;
198,451✔
1351
}
1352

1353
static FORCE_INLINE int32_t addToJoinVgroupHash(SSHashObj* pHash, void* pKey, int32_t keySize, void* pVal, int32_t valSize) {
1354
  SArray** ppArray = tSimpleHashGet(pHash, pKey, keySize);
7,139,024✔
1355
  if (NULL == ppArray) {
7,139,024✔
1356
    SArray* pArray = taosArrayInit(10, valSize);
229,178✔
1357
    if (NULL == pArray) {
229,178✔
1358
      return terrno;
×
1359
    }
1360
    if (NULL == taosArrayPush(pArray, pVal)) {
458,356✔
1361
      taosArrayDestroy(pArray);
×
1362
      return terrno;
×
1363
    }
1364
    if (tSimpleHashPut(pHash, pKey, keySize, &pArray, POINTER_BYTES)) {
229,178✔
1365
      taosArrayDestroy(pArray);      
×
1366
      return terrno;
×
1367
    }
1368
    return TSDB_CODE_SUCCESS;
229,178✔
1369
  }
1370

1371
  if (NULL == taosArrayPush(*ppArray, pVal)) {
13,819,692✔
1372
    return terrno;
×
1373
  }
1374
  
1375
  return TSDB_CODE_SUCCESS;
6,909,846✔
1376
}
1377

1378
static FORCE_INLINE int32_t addToJoinTableHash(SSHashObj* pHash, SSHashObj* pOnceHash, void* pKey, int32_t keySize) {
1379
  int32_t code = TSDB_CODE_SUCCESS;
2,118✔
1380
  uint32_t* pNum = tSimpleHashGet(pHash, pKey, keySize);
2,118✔
1381
  if (NULL == pNum) {
2,118✔
1382
    uint32_t n = 1;
2,118✔
1383
    code = tSimpleHashPut(pHash, pKey, keySize, &n, sizeof(n));
2,118✔
1384
    if (code) {
2,118✔
1385
      return code;
×
1386
    }
1387
    code = tSimpleHashPut(pOnceHash, pKey, keySize, NULL, 0);
2,118✔
1388
    if (code) {
2,118✔
1389
      return code;
×
1390
    }
1391
    return TSDB_CODE_SUCCESS;
2,118✔
1392
  }
1393

1394
  switch (*pNum) {
×
1395
    case 0:
×
1396
      break;
×
1397
    case UINT32_MAX:
×
1398
      *pNum = 0;
×
1399
      break;
×
1400
    default:
×
1401
      if (1 == (*pNum)) {
×
1402
        code = tSimpleHashRemove(pOnceHash, pKey, keySize);
×
1403
        if (code) {
×
1404
          qError("tSimpleHashRemove failed in addToJoinTableHash, error:%s", tstrerror(code));
×
1405
          QRY_ERR_RET(code);
×
1406
        }
1407
      }
1408
      (*pNum)++;
×
1409
      break;
×
1410
  }
1411
  
1412
  return TSDB_CODE_SUCCESS;
×
1413
}
1414

1415

1416
static void freeStbJoinTableList(SStbJoinTableList* pList) {
67,510✔
1417
  if (NULL == pList) {
67,510✔
1418
    return;
×
1419
  }
1420
  taosMemoryFree(pList->pLeftVg);
67,510✔
1421
  taosMemoryFree(pList->pLeftUid);
67,510✔
1422
  taosMemoryFree(pList->pRightVg);
67,510✔
1423
  taosMemoryFree(pList->pRightUid);
67,510✔
1424
  taosMemoryFree(pList);
67,510✔
1425
}
1426

1427
static int32_t appendStbJoinTableList(SStbJoinPrevJoinCtx* pCtx, int64_t rows, int32_t* pLeftVg, int64_t* pLeftUid, int32_t* pRightVg, int64_t* pRightUid) {
67,993✔
1428
  int32_t code = TSDB_CODE_SUCCESS;
67,993✔
1429
  SStbJoinTableList* pNew = taosMemoryCalloc(1, sizeof(SStbJoinTableList));
67,993✔
1430
  if (NULL == pNew) {
67,993✔
1431
    return terrno;
×
1432
  }
1433
  pNew->pLeftVg = taosMemoryMalloc(rows * sizeof(*pLeftVg));
67,993✔
1434
  if (NULL == pNew->pLeftVg) {
67,993✔
1435
    code = terrno;
×
1436
    freeStbJoinTableList(pNew);
×
1437
    return code;
×
1438
  }
1439
  pNew->pLeftUid = taosMemoryMalloc(rows * sizeof(*pLeftUid));
67,993✔
1440
  if (NULL == pNew->pLeftUid) {
67,993✔
1441
    code = terrno;
×
1442
    freeStbJoinTableList(pNew);
×
1443
    return code;
×
1444
  }
1445
  pNew->pRightVg = taosMemoryMalloc(rows * sizeof(*pRightVg));
67,993✔
1446
  if (NULL == pNew->pRightVg) {
67,993✔
1447
    code = terrno;
×
1448
    freeStbJoinTableList(pNew);
×
1449
    return code;
×
1450
  }
1451
  pNew->pRightUid = taosMemoryMalloc(rows * sizeof(*pRightUid));
67,993✔
1452
  if (NULL == pNew->pRightUid) {
67,993✔
1453
    code = terrno;
×
1454
    freeStbJoinTableList(pNew);
×
1455
    return code;
×
1456
  }
1457

1458
  TAOS_MEMCPY(pNew->pLeftVg, pLeftVg, rows * sizeof(*pLeftVg));
67,993✔
1459
  TAOS_MEMCPY(pNew->pLeftUid, pLeftUid, rows * sizeof(*pLeftUid));
67,993✔
1460
  TAOS_MEMCPY(pNew->pRightVg, pRightVg, rows * sizeof(*pRightVg));
67,993✔
1461
  TAOS_MEMCPY(pNew->pRightUid, pRightUid, rows * sizeof(*pRightUid));
67,993✔
1462

1463
  pNew->readIdx = 0;
67,993✔
1464
  pNew->uidNum = rows;
67,993✔
1465
  pNew->pNext = NULL;
67,993✔
1466
  
1467
  if (pCtx->pListTail) {
67,993✔
1468
    pCtx->pListTail->pNext = pNew;
×
1469
    pCtx->pListTail = pNew;
×
1470
  } else {
1471
    pCtx->pListHead = pNew;
67,993✔
1472
    pCtx->pListTail= pNew;
67,993✔
1473
  }
1474

1475
  return TSDB_CODE_SUCCESS;
67,993✔
1476
}
1477

1478
static void doBuildStbJoinTableHash(SOperatorInfo* pOperator, SSDataBlock* pBlock) {
67,993✔
1479
  int32_t                    code = TSDB_CODE_SUCCESS;
67,993✔
1480
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
67,993✔
1481
  SStbJoinDynCtrlInfo*       pStbJoin = (SStbJoinDynCtrlInfo*)&pInfo->stbJoin;
67,993✔
1482
  SColumnInfoData*           pVg0 = taosArrayGet(pBlock->pDataBlock, pStbJoin->basic.vgSlot[0]);
67,993✔
1483
  if (NULL == pVg0) {
67,993✔
1484
    QRY_ERR_JRET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
1485
  }
1486
  SColumnInfoData*           pVg1 = taosArrayGet(pBlock->pDataBlock, pStbJoin->basic.vgSlot[1]);
67,993✔
1487
  if (NULL == pVg1) {
67,993✔
1488
    QRY_ERR_JRET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
1489
  }
1490
  SColumnInfoData*           pUid0 = taosArrayGet(pBlock->pDataBlock, pStbJoin->basic.uidSlot[0]);
67,993✔
1491
  if (NULL == pUid0) {
67,993✔
1492
    QRY_ERR_JRET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
1493
  }
1494
  SColumnInfoData*           pUid1 = taosArrayGet(pBlock->pDataBlock, pStbJoin->basic.uidSlot[1]);
67,993✔
1495
  if (NULL == pUid1) {
67,993✔
1496
    QRY_ERR_JRET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
1497
  }
1498

1499
  if (pStbJoin->basic.batchFetch) {
67,993✔
1500
    for (int32_t i = 0; i < pBlock->info.rows; ++i) {
3,636,446✔
1501
      int32_t* leftVg = (int32_t*)(pVg0->pData + pVg0->info.bytes * i);
3,569,512✔
1502
      int64_t* leftUid = (int64_t*)(pUid0->pData + pUid0->info.bytes * i);
3,569,512✔
1503
      int32_t* rightVg = (int32_t*)(pVg1->pData + pVg1->info.bytes * i);
3,569,512✔
1504
      int64_t* rightUid = (int64_t*)(pUid1->pData + pUid1->info.bytes * i);
3,569,512✔
1505

1506
      code = addToJoinVgroupHash(pStbJoin->ctx.prev.leftHash, leftVg, sizeof(*leftVg), leftUid, sizeof(*leftUid));
3,569,512✔
1507
      if (TSDB_CODE_SUCCESS != code) {
3,569,512✔
1508
        break;
×
1509
      }
1510
      code = addToJoinVgroupHash(pStbJoin->ctx.prev.rightHash, rightVg, sizeof(*rightVg), rightUid, sizeof(*rightUid));
3,569,512✔
1511
      if (TSDB_CODE_SUCCESS != code) {
3,569,512✔
1512
        break;
×
1513
      }
1514
    }
1515
  } else {
1516
    for (int32_t i = 0; i < pBlock->info.rows; ++i) {
3,177✔
1517
      int64_t* leftUid = (int64_t*)(pUid0->pData + pUid0->info.bytes * i);
2,118✔
1518
    
1519
      code = addToJoinTableHash(pStbJoin->ctx.prev.leftCache, pStbJoin->ctx.prev.onceTable, leftUid, sizeof(*leftUid));
2,118✔
1520
      if (TSDB_CODE_SUCCESS != code) {
2,118✔
1521
        break;
×
1522
      }
1523
    }
1524
  }
1525

1526
  if (TSDB_CODE_SUCCESS == code) {
67,993✔
1527
    code = appendStbJoinTableList(&pStbJoin->ctx.prev, pBlock->info.rows, (int32_t*)pVg0->pData, (int64_t*)pUid0->pData, (int32_t*)pVg1->pData, (int64_t*)pUid1->pData);
67,993✔
1528
    if (TSDB_CODE_SUCCESS == code) {
67,993✔
1529
      pStbJoin->ctx.prev.tableNum += pBlock->info.rows;
67,993✔
1530
    }
1531
  }
1532

1533
_return:
×
1534

1535
  if (TSDB_CODE_SUCCESS != code) {
67,993✔
1536
    pOperator->pTaskInfo->code = code;
×
1537
    T_LONG_JMP(pOperator->pTaskInfo->env, pOperator->pTaskInfo->code);
×
1538
  }
1539
}
67,993✔
1540

1541

1542
static void postProcessStbJoinTableHash(SOperatorInfo* pOperator) {
1,111,525✔
1543
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
1,111,525✔
1544
  SStbJoinDynCtrlInfo*       pStbJoin = (SStbJoinDynCtrlInfo*)&pInfo->stbJoin;
1,111,525✔
1545

1546
  if (pStbJoin->basic.batchFetch) {
1,111,525✔
1547
    return;
1,110,466✔
1548
  }
1549

1550
  if (tSimpleHashGetSize(pStbJoin->ctx.prev.leftCache) == tSimpleHashGetSize(pStbJoin->ctx.prev.onceTable)) {
1,059✔
1551
    tSimpleHashClear(pStbJoin->ctx.prev.leftCache);
1,059✔
1552
    return;
1,059✔
1553
  }
1554

1555
  uint64_t* pUid = NULL;
×
1556
  int32_t iter = 0;
×
1557
  int32_t code = 0;
×
1558
  while (NULL != (pUid = tSimpleHashIterate(pStbJoin->ctx.prev.onceTable, pUid, &iter))) {
×
1559
    code = tSimpleHashRemove(pStbJoin->ctx.prev.leftCache, pUid, sizeof(*pUid));
×
1560
    if (code) {
×
1561
      qError("tSimpleHashRemove failed in postProcessStbJoinTableHash, error:%s", tstrerror(code));
×
1562
    }
1563
  }
1564

1565
  pStbJoin->execInfo.leftCacheNum = tSimpleHashGetSize(pStbJoin->ctx.prev.leftCache);
×
1566
  qDebug("more than 1 ref build table num %" PRId64, (int64_t)tSimpleHashGetSize(pStbJoin->ctx.prev.leftCache));
×
1567

1568
/*
1569
  // debug only
1570
  iter = 0;
1571
  uint32_t* num = NULL;
1572
  while (NULL != (num = tSimpleHashIterate(pStbJoin->ctx.prev.leftCache, num, &iter))) {
1573
    A S S E R T(*num > 1);
1574
  }
1575
*/  
1576
}
1577

1578
static void buildStbJoinTableList(SOperatorInfo* pOperator) {
1,111,525✔
1579
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
1,111,525✔
1580
  SStbJoinDynCtrlInfo*       pStbJoin = (SStbJoinDynCtrlInfo*)&pInfo->stbJoin;
1,111,525✔
1581

1582
  while (true) {
67,993✔
1583
    SSDataBlock* pBlock = getNextBlockFromDownstream(pOperator, 0);
1,179,518✔
1584
    if (NULL == pBlock) {
1,179,518✔
1585
      break;
1,111,525✔
1586
    }
1587

1588
    pStbJoin->execInfo.prevBlkNum++;
67,993✔
1589
    pStbJoin->execInfo.prevBlkRows += pBlock->info.rows;
67,993✔
1590
    
1591
    doBuildStbJoinTableHash(pOperator, pBlock);
67,993✔
1592
  }
1593

1594
  postProcessStbJoinTableHash(pOperator);
1,111,525✔
1595

1596
  pStbJoin->ctx.prev.joinBuild = true;
1,111,525✔
1597
}
1,111,525✔
1598

1599
static int32_t seqJoinLaunchNewRetrieve(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
266,444✔
1600
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
266,444✔
1601
  SStbJoinDynCtrlInfo*       pStbJoin = (SStbJoinDynCtrlInfo*)&pInfo->stbJoin;
266,444✔
1602
  SStbJoinPrevJoinCtx*       pPrev = &pStbJoin->ctx.prev;
266,444✔
1603
  SStbJoinTableList*         pNode = pPrev->pListHead;
266,444✔
1604

1605
  while (pNode) {
3,704,718✔
1606
    if (pNode->readIdx >= pNode->uidNum) {
3,637,208✔
1607
      pPrev->pListHead = pNode->pNext;
67,510✔
1608
      freeStbJoinTableList(pNode);
67,510✔
1609
      pNode = pPrev->pListHead;
67,510✔
1610
      continue;
67,510✔
1611
    }
1612
    
1613
    seqJoinLaunchNewRetrieveImpl(pOperator, ppRes);
3,569,698✔
1614
    if (*ppRes) {
3,569,698✔
1615
      return TSDB_CODE_SUCCESS;
198,934✔
1616
    }
1617

1618
    QRY_ERR_RET(handleSeqJoinCurrRetrieveEnd(pOperator, pStbJoin));
3,370,764✔
1619
    pPrev->pListHead->readIdx++;
3,370,764✔
1620
  }
1621

1622
  *ppRes = NULL;
67,510✔
1623
  setOperatorCompleted(pOperator);
67,510✔
1624

1625
  return TSDB_CODE_SUCCESS;
67,510✔
1626
}
1627

1628
static int32_t seqStableJoinComposeRes(SStbJoinDynCtrlInfo* pStbJoin, SSDataBlock* pBlock) {
1,309,976✔
1629
  if (pBlock) {
1,309,976✔
1630
    if (pStbJoin && pStbJoin->pOutputDataBlockDesc) {
198,934✔
1631
      pBlock->info.id.blockId = pStbJoin->pOutputDataBlockDesc->dataBlockId;
198,934✔
1632
      if (!pBlock->pDataBlock) return TSDB_CODE_SUCCESS;
198,934✔
1633

1634
      for (int i = (int)pBlock->pDataBlock->size; i < pStbJoin->pOutputDataBlockDesc->pSlots->length; i++) {
201,052✔
1635
        SSlotDescNode* pSlot = (SSlotDescNode*)nodesListGetNode(pStbJoin->pOutputDataBlockDesc->pSlots, i);
2,118✔
1636
        if (pSlot == NULL) {
2,118✔
1637
          qError("seqStableJoinComposeRes: pSlot is NULL, i:%d", i);
×
1638
          return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1639
        }
1640
        SColumnInfoData colInfo = createColumnInfoData(pSlot->dataType.type, pSlot->dataType.bytes, pSlot->slotId);
2,118✔
1641
        int32_t code = colInfoDataEnsureCapacity(&colInfo, pBlock->info.rows, true);
2,118✔
1642
        if (code != TSDB_CODE_SUCCESS) {
2,118✔
1643
          return code;
×
1644
        }
1645
        code = blockDataAppendColInfo(pBlock, &colInfo);
2,118✔
1646
        if (code != TSDB_CODE_SUCCESS) {
2,118✔
1647
          return code;
×
1648
        }
1649
      }
1650
    } else {
1651
      qError("seqStableJoinComposeRes: pBlock or pStbJoin is NULL");
×
1652
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1653
    }
1654
  }
1655
  return TSDB_CODE_SUCCESS;
1,309,976✔
1656
}
1657

1658
int32_t seqStableJoin(SOperatorInfo* pOperator, SSDataBlock** pRes) {
1,333,164✔
1659
  int32_t                    code = TSDB_CODE_SUCCESS;
1,333,164✔
1660
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
1,333,164✔
1661
  SStbJoinDynCtrlInfo*       pStbJoin = (SStbJoinDynCtrlInfo*)&pInfo->stbJoin;
1,333,164✔
1662

1663
  QRY_PARAM_CHECK(pRes);
1,333,164✔
1664
  if (pOperator->status == OP_EXEC_DONE) {
1,333,164✔
1665
    return code;
23,188✔
1666
  }
1667

1668
  int64_t st = 0;
1,309,976✔
1669
  if (pOperator->cost.openCost == 0) {
1,309,976✔
1670
    st = taosGetTimestampUs();
1,111,053✔
1671
  }
1672

1673
  if (!pStbJoin->ctx.prev.joinBuild) {
1,309,976✔
1674
    buildStbJoinTableList(pOperator);
1,111,525✔
1675
    if (pStbJoin->execInfo.prevBlkRows <= 0) {
1,111,525✔
1676
      setOperatorCompleted(pOperator);
1,043,532✔
1677
      goto _return;
1,043,532✔
1678
    }
1679
  }
1680

1681
  QRY_ERR_JRET(seqJoinContinueCurrRetrieve(pOperator, pRes));
266,444✔
1682
  if (*pRes) {
266,444✔
1683
    goto _return;
×
1684
  }
1685

1686
  QRY_ERR_JRET(seqJoinLaunchNewRetrieve(pOperator, pRes));
266,444✔
1687

1688
_return:
266,444✔
1689
  if (pOperator->cost.openCost == 0) {
1,309,976✔
1690
    pOperator->cost.openCost = (double)(taosGetTimestampUs() - st) / 1000.0;
1,111,053✔
1691
  }
1692

1693
  if (code) {
1,309,976✔
1694
    qError("%s failed since %s", __func__, tstrerror(code));
×
1695
    pOperator->pTaskInfo->code = code;
×
1696
    T_LONG_JMP(pOperator->pTaskInfo->env, code);
×
1697
  } else {
1698
    code = seqStableJoinComposeRes(pStbJoin, *pRes);
1,309,976✔
1699
  }
1700
  return code;
1,309,976✔
1701
}
1702

1703
int32_t dynProcessUseDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
240,173✔
1704
  int32_t                    lino = 0;
240,173✔
1705
  SOperatorInfo*             operator=(SOperatorInfo*) param;
240,173✔
1706
  SDynQueryCtrlOperatorInfo* pScanResInfo = (SDynQueryCtrlOperatorInfo*)operator->info;
240,173✔
1707

1708
  if (TSDB_CODE_SUCCESS != code) {
240,173✔
1709
    operator->pTaskInfo->code = rpcCvtErrCode(code);
×
1710
    if (operator->pTaskInfo->code != code) {
×
1711
      qError("load systable rsp received, error:%s, cvted error:%s", tstrerror(code),
×
1712
             tstrerror(operator->pTaskInfo->code));
1713
    } else {
1714
      qError("load systable rsp received, error:%s", tstrerror(code));
×
1715
    }
1716
    goto _return;
×
1717
  }
1718

1719
  pScanResInfo->vtbScan.pRsp = taosMemoryMalloc(sizeof(SUseDbRsp));
240,173✔
1720
  QUERY_CHECK_NULL(pScanResInfo->vtbScan.pRsp, code, lino, _return, terrno)
240,173✔
1721

1722
  code = tDeserializeSUseDbRsp(pMsg->pData, (int32_t)pMsg->len, pScanResInfo->vtbScan.pRsp);
240,173✔
1723
  QUERY_CHECK_CODE(code, lino, _return);
240,173✔
1724

1725
  taosMemoryFreeClear(pMsg->pData);
240,173✔
1726

1727
  code = tsem_post(&pScanResInfo->vtbScan.ready);
240,173✔
1728
  QUERY_CHECK_CODE(code, lino, _return);
240,173✔
1729

1730
  return code;
240,173✔
1731
_return:
×
1732
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1733
  return code;
×
1734
}
1735

1736
static int32_t buildDbVgInfoMap(SOperatorInfo* pOperator, SMsgCb* pMsgCb, SName* name, SExecTaskInfo* pTaskInfo, SUseDbOutput* output) {
240,173✔
1737
  int32_t                    code = TSDB_CODE_SUCCESS;
240,173✔
1738
  int32_t                    lino = 0;
240,173✔
1739
  char*                      buf1 = NULL;
240,173✔
1740
  SUseDbReq*                 pReq = NULL;
240,173✔
1741
  SDynQueryCtrlOperatorInfo* pScanResInfo = (SDynQueryCtrlOperatorInfo*)pOperator->info;
240,173✔
1742

1743
  pReq = taosMemoryMalloc(sizeof(SUseDbReq));
240,173✔
1744
  QUERY_CHECK_NULL(pReq, code, lino, _return, terrno)
240,173✔
1745
  code = tNameGetFullDbName(name, pReq->db);
240,173✔
1746
  QUERY_CHECK_CODE(code, lino, _return);
240,173✔
1747
  int32_t contLen = tSerializeSUseDbReq(NULL, 0, pReq);
240,173✔
1748
  buf1 = taosMemoryCalloc(1, contLen);
240,173✔
1749
  QUERY_CHECK_NULL(buf1, code, lino, _return, terrno)
240,173✔
1750
  int32_t tempRes = tSerializeSUseDbReq(buf1, contLen, pReq);
240,173✔
1751
  if (tempRes < 0) {
240,173✔
1752
    QUERY_CHECK_CODE(terrno, lino, _return);
×
1753
  }
1754

1755
  // send the fetch remote task result request
1756
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
240,173✔
1757
  QUERY_CHECK_NULL(pMsgSendInfo, code, lino, _return, terrno)
240,173✔
1758

1759
  pMsgSendInfo->param = pOperator;
240,173✔
1760
  pMsgSendInfo->msgInfo.pData = buf1;
240,173✔
1761
  pMsgSendInfo->msgInfo.len = contLen;
240,173✔
1762
  pMsgSendInfo->msgType = TDMT_MND_GET_DB_INFO;
240,173✔
1763
  pMsgSendInfo->fp = dynProcessUseDbRsp;
240,173✔
1764
  pMsgSendInfo->requestId = pTaskInfo->id.queryId;
240,173✔
1765

1766
  code = asyncSendMsgToServer(pMsgCb->clientRpc, &pScanResInfo->vtbScan.epSet, NULL, pMsgSendInfo);
240,173✔
1767
  QUERY_CHECK_CODE(code, lino, _return);
240,173✔
1768

1769
  code = tsem_wait(&pScanResInfo->vtbScan.ready);
240,173✔
1770
  QUERY_CHECK_CODE(code, lino, _return);
240,173✔
1771

1772
  code = queryBuildUseDbOutput(output, pScanResInfo->vtbScan.pRsp);
240,173✔
1773
  QUERY_CHECK_CODE(code, lino, _return);
240,173✔
1774

1775
_return:
240,173✔
1776
  if (code) {
240,173✔
1777
     qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1778
     taosMemoryFree(buf1);
×
1779
  }
1780
  taosMemoryFree(pReq);
240,173✔
1781
  tFreeSUsedbRsp(pScanResInfo->vtbScan.pRsp);
240,173✔
1782
  taosMemoryFreeClear(pScanResInfo->vtbScan.pRsp);
240,173✔
1783
  return code;
240,173✔
1784
}
1785

1786
int dynVgInfoComp(const void* lp, const void* rp) {
471,550✔
1787
  SVgroupInfo* pLeft = (SVgroupInfo*)lp;
471,550✔
1788
  SVgroupInfo* pRight = (SVgroupInfo*)rp;
471,550✔
1789
  if (pLeft->hashBegin < pRight->hashBegin) {
471,550✔
1790
    return -1;
471,550✔
1791
  } else if (pLeft->hashBegin > pRight->hashBegin) {
×
1792
    return 1;
×
1793
  }
1794

1795
  return 0;
×
1796
}
1797

1798
int32_t dynMakeVgArraySortBy(SDBVgInfo* dbInfo, __compar_fn_t sort_func) {
1,311,527✔
1799
  if (NULL == dbInfo) {
1,311,527✔
1800
    return TSDB_CODE_SUCCESS;
×
1801
  }
1802

1803
  if (dbInfo->vgHash && NULL == dbInfo->vgArray) {
1,311,527✔
1804
    int32_t vgSize = taosHashGetSize(dbInfo->vgHash);
240,173✔
1805
    dbInfo->vgArray = taosArrayInit(vgSize, sizeof(SVgroupInfo));
240,173✔
1806
    if (NULL == dbInfo->vgArray) {
240,173✔
1807
      return terrno;
×
1808
    }
1809

1810
    void* pIter = taosHashIterate(dbInfo->vgHash, NULL);
240,173✔
1811
    while (pIter) {
716,121✔
1812
      if (NULL == taosArrayPush(dbInfo->vgArray, pIter)) {
951,896✔
1813
        taosHashCancelIterate(dbInfo->vgHash, pIter);
×
1814
        return terrno;
×
1815
      }
1816

1817
      pIter = taosHashIterate(dbInfo->vgHash, pIter);
475,948✔
1818
    }
1819

1820
    taosArraySort(dbInfo->vgArray, sort_func);
240,173✔
1821
  }
1822

1823
  return TSDB_CODE_SUCCESS;
1,311,527✔
1824
}
1825

1826
int32_t dynHashValueComp(void const* lp, void const* rp) {
1,971,492✔
1827
  uint32_t*    key = (uint32_t*)lp;
1,971,492✔
1828
  SVgroupInfo* pVg = (SVgroupInfo*)rp;
1,971,492✔
1829

1830
  if (*key < pVg->hashBegin) {
1,971,492✔
1831
    return -1;
×
1832
  } else if (*key > pVg->hashEnd) {
1,971,492✔
1833
    return 1;
659,965✔
1834
  }
1835

1836
  return 0;
1,311,527✔
1837
}
1838

1839
int32_t getVgId(SDBVgInfo* dbInfo, char* dbFName, int32_t* vgId, char *tbName) {
1,311,527✔
1840
  int32_t code = 0;
1,311,527✔
1841
  int32_t lino = 0;
1,311,527✔
1842
  code = dynMakeVgArraySortBy(dbInfo, dynVgInfoComp);
1,311,527✔
1843
  QUERY_CHECK_CODE(code, lino, _return);
1,311,527✔
1844

1845
  int32_t vgNum = (int32_t)taosArrayGetSize(dbInfo->vgArray);
1,311,527✔
1846
  if (vgNum <= 0) {
1,311,527✔
1847
    qError("db vgroup cache invalid, db:%s, vgroup number:%d", dbFName, vgNum);
×
1848
    QUERY_CHECK_CODE(code = TSDB_CODE_TSC_DB_NOT_SELECTED, lino, _return);
×
1849
  }
1850

1851
  SVgroupInfo* vgInfo = NULL;
1,311,527✔
1852
  char         tbFullName[TSDB_TABLE_FNAME_LEN];
1,311,527✔
1853
  (void)snprintf(tbFullName, sizeof(tbFullName), "%s.", dbFName);
1,311,527✔
1854
  int32_t offset = (int32_t)strlen(tbFullName);
1,311,527✔
1855

1856
  (void)snprintf(tbFullName + offset, sizeof(tbFullName) - offset, "%s", tbName);
1,311,527✔
1857
  uint32_t hashValue = taosGetTbHashVal(tbFullName, (int32_t)strlen(tbFullName), dbInfo->hashMethod,
2,623,054✔
1858
                                        dbInfo->hashPrefix, dbInfo->hashSuffix);
1,311,527✔
1859

1860
  vgInfo = taosArraySearch(dbInfo->vgArray, &hashValue, dynHashValueComp, TD_EQ);
1,311,527✔
1861
  if (NULL == vgInfo) {
1,311,527✔
1862
    qError("no hash range found for hash value [%u], db:%s, numOfVgId:%d", hashValue, dbFName,
×
1863
           (int32_t)taosArrayGetSize(dbInfo->vgArray));
1864
    return TSDB_CODE_CTG_INTERNAL_ERROR;
×
1865
  }
1866

1867
  *vgId = vgInfo->vgId;
1,311,527✔
1868

1869
_return:
1,311,527✔
1870
  return code;
1,311,527✔
1871
}
1872

1873
bool colNeedScan(SOperatorInfo* pOperator, col_id_t colId) {
29,592,779✔
1874
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
29,592,779✔
1875
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
29,592,779✔
1876
  SArray *                   pColList = pVtbScan->readColList;
29,592,779✔
1877
  if (pVtbScan->scanAllCols) {
29,592,779✔
1878
    return true;
235,950✔
1879
  }
1880
  for (int32_t i = 0; i < taosArrayGetSize(pColList); i++) {
158,455,981✔
1881
    if (colId == *(col_id_t*)taosArrayGet(pColList, i)) {
133,259,511✔
1882
      return true;
4,160,359✔
1883
    }
1884
  }
1885
  return false;
25,196,470✔
1886
}
1887

1888
int32_t getDbVgInfo(SOperatorInfo* pOperator, SName *name, SDBVgInfo **dbVgInfo) {
4,419,887✔
1889
  int32_t                    code = TSDB_CODE_SUCCESS;
4,419,887✔
1890
  int32_t                    line = 0;
4,419,887✔
1891
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
4,419,887✔
1892
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
4,419,887✔
1893
  SExecTaskInfo*             pTaskInfo = pOperator->pTaskInfo;
4,419,887✔
1894
  SMsgCb*                    pMsgCb = pVtbScan->pMsgCb;
4,419,887✔
1895
  SUseDbOutput*              output = NULL;
4,419,887✔
1896
  SUseDbOutput**             find = (SUseDbOutput**)taosHashGet(pInfo->vtbScan.dbVgInfoMap, name->dbname, strlen(name->dbname));
4,419,887✔
1897

1898
  QRY_PARAM_CHECK(dbVgInfo);
4,419,887✔
1899

1900
  if (find == NULL) {
4,419,887✔
1901
    output = taosMemoryMalloc(sizeof(SUseDbOutput));
240,173✔
1902
    code = buildDbVgInfoMap(pOperator, pMsgCb, name, pTaskInfo, output);
240,173✔
1903
    QUERY_CHECK_CODE(code, line, _return);
240,173✔
1904
    code = taosHashPut(pInfo->vtbScan.dbVgInfoMap, name->dbname, strlen(name->dbname), &output, POINTER_BYTES);
240,173✔
1905
    QUERY_CHECK_CODE(code, line, _return);
240,173✔
1906
  } else {
1907
    output = *find;
4,179,714✔
1908
  }
1909

1910
  *dbVgInfo = output->dbVgroup;
4,419,887✔
1911
  return code;
4,419,887✔
1912
_return:
×
1913
  qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
×
1914
  freeUseDbOutput(output);
×
1915
  return code;
×
1916
}
1917

1918
int32_t extractColRefName(const char *colref, char **refDb, char** refTb, char** refCol) {
4,419,887✔
1919
  int32_t     code = TSDB_CODE_SUCCESS;
4,419,887✔
1920
  int32_t     line = 0;
4,419,887✔
1921

1922
  const char *first_dot = strchr(colref, '.');
4,419,887✔
1923
  QUERY_CHECK_NULL(first_dot, code, line, _return, terrno)
4,419,887✔
1924

1925
  const char *second_dot = strchr(first_dot + 1, '.');
4,419,887✔
1926
  QUERY_CHECK_NULL(second_dot, code, line, _return, terrno)
4,419,887✔
1927

1928
  size_t db_len = first_dot - colref;
4,419,887✔
1929
  size_t table_len = second_dot - first_dot - 1;
4,419,887✔
1930
  size_t col_len = strlen(second_dot + 1);
4,419,887✔
1931

1932
  *refDb = taosMemoryMalloc(db_len + 1);
4,419,887✔
1933
  *refTb = taosMemoryMalloc(table_len + 1);
4,419,887✔
1934
  *refCol = taosMemoryMalloc(col_len + 1);
4,419,887✔
1935
  QUERY_CHECK_NULL(*refDb, code, line, _return, terrno)
4,419,887✔
1936
  QUERY_CHECK_NULL(*refTb, code, line, _return, terrno)
4,419,887✔
1937
  QUERY_CHECK_NULL(*refCol, code, line, _return, terrno)
4,419,887✔
1938

1939
  tstrncpy(*refDb, colref, db_len + 1);
4,419,887✔
1940
  tstrncpy(*refTb, first_dot + 1, table_len + 1);
4,419,887✔
1941
  tstrncpy(*refCol, second_dot + 1, col_len + 1);
4,419,887✔
1942

1943
  return TSDB_CODE_SUCCESS;
4,419,887✔
1944
_return:
×
1945
  qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
×
1946
  if (*refDb) {
×
1947
    taosMemoryFree(*refDb);
×
1948
    *refDb = NULL;
×
1949
  }
1950
  if (*refTb) {
×
1951
    taosMemoryFree(*refTb);
×
1952
    *refTb = NULL;
×
1953
  }
1954
  if (*refCol) {
×
1955
    taosMemoryFree(*refCol);
×
1956
    *refCol = NULL;
×
1957
  }
1958
  return code;
×
1959
}
1960

1961
bool tableInfoNeedCollect(char *dbName, char *tbName, char *expectDbName, char *expectTbName) {
52,388,899✔
1962
  if (strncmp(varDataVal(tbName), expectTbName, varDataLen(tbName)) == 0 &&
52,388,899✔
1963
      strlen(expectTbName) == varDataLen(tbName) &&
34,071,569✔
1964
      strncmp(varDataVal(dbName), expectDbName, varDataLen(dbName)) == 0 &&
34,071,569✔
1965
      strlen(expectDbName) == varDataLen(dbName)) {
34,071,569✔
1966
    return true;
34,071,569✔
1967
  }
1968
  return false;
18,317,330✔
1969
}
1970

1971
int32_t getColRefInfo(SColRefInfo *pInfo, SArray* pDataBlock, int32_t index) {
34,071,569✔
1972
  int32_t          code = TSDB_CODE_SUCCESS;
34,071,569✔
1973
  int32_t          line = 0;
34,071,569✔
1974

1975
  SColumnInfoData *pColNameCol = taosArrayGet(pDataBlock, 3);
34,071,569✔
1976
  SColumnInfoData *pUidCol = taosArrayGet(pDataBlock, 4);
34,071,569✔
1977
  SColumnInfoData *pColIdCol = taosArrayGet(pDataBlock, 5);
34,071,569✔
1978
  SColumnInfoData *pRefCol = taosArrayGet(pDataBlock, 6);
34,071,569✔
1979
  SColumnInfoData *pVgIdCol = taosArrayGet(pDataBlock, 7);
34,071,569✔
1980
  SColumnInfoData *pRefVerCol = taosArrayGet(pDataBlock, 8);
34,071,569✔
1981

1982
  QUERY_CHECK_NULL(pColNameCol, code, line, _return, terrno)
34,071,569✔
1983
  QUERY_CHECK_NULL(pUidCol, code, line, _return, terrno)
34,071,569✔
1984
  QUERY_CHECK_NULL(pColIdCol, code, line, _return, terrno)
34,071,569✔
1985
  QUERY_CHECK_NULL(pRefCol, code, line, _return, terrno)
34,071,569✔
1986
  QUERY_CHECK_NULL(pVgIdCol, code, line, _return, terrno)
34,071,569✔
1987
  QUERY_CHECK_NULL(pRefVerCol, code, line, _return, terrno)
34,071,569✔
1988

1989
  if (colDataIsNull_s(pRefCol, index)) {
68,143,138✔
1990
    pInfo->colrefName = NULL;
4,473,019✔
1991
  } else {
1992
    pInfo->colrefName = taosMemoryCalloc(varDataTLen(colDataGetData(pRefCol, index)), 1);
29,598,550✔
1993
    QUERY_CHECK_NULL(pInfo->colrefName, code, line, _return, terrno)
29,598,550✔
1994
    memcpy(pInfo->colrefName, varDataVal(colDataGetData(pRefCol, index)), varDataLen(colDataGetData(pRefCol, index)));
29,598,550✔
1995
    pInfo->colrefName[varDataLen(colDataGetData(pRefCol, index))] = 0;
29,598,550✔
1996
  }
1997

1998
  pInfo->colName = taosMemoryCalloc(varDataTLen(colDataGetData(pColNameCol, index)), 1);
34,071,569✔
1999
  QUERY_CHECK_NULL(pInfo->colName, code, line, _return, terrno)
34,071,569✔
2000
  memcpy(pInfo->colName, varDataVal(colDataGetData(pColNameCol, index)), varDataLen(colDataGetData(pColNameCol, index)));
34,071,569✔
2001
  pInfo->colName[varDataLen(colDataGetData(pColNameCol, index))] = 0;
34,071,569✔
2002

2003
  if (!colDataIsNull_s(pUidCol, index)) {
68,143,138✔
2004
    GET_TYPED_DATA(pInfo->uid, int64_t, TSDB_DATA_TYPE_BIGINT, colDataGetNumData(pUidCol, index), 0);
34,071,569✔
2005
  }
2006
  if (!colDataIsNull_s(pColIdCol, index)) {
68,143,138✔
2007
    GET_TYPED_DATA(pInfo->colId, int32_t, TSDB_DATA_TYPE_INT, colDataGetNumData(pColIdCol, index), 0);
29,598,550✔
2008
  }
2009
  if (!colDataIsNull_s(pVgIdCol, index)) {
68,143,138✔
2010
    GET_TYPED_DATA(pInfo->vgId, int32_t, TSDB_DATA_TYPE_INT, colDataGetNumData(pVgIdCol, index), 0);
34,071,569✔
2011
  }
2012

2013
_return:
×
2014
  return code;
34,071,569✔
2015
}
2016

2017
int32_t processOrgTbVg(SVtbScanDynCtrlInfo* pVtbScan, SExecTaskInfo* pTaskInfo, int32_t rversion) {
351,056✔
2018
  int32_t                    code = TSDB_CODE_SUCCESS;
351,056✔
2019
  int32_t                    line = 0;
351,056✔
2020

2021
  if (pTaskInfo->pStreamRuntimeInfo == NULL) {
351,056✔
2022
    return code;
242,262✔
2023
  }
2024

2025
  if (pVtbScan->existOrgTbVg == NULL) {
108,794✔
2026
    pVtbScan->existOrgTbVg = pVtbScan->curOrgTbVg;
×
2027
    pVtbScan->curOrgTbVg = NULL;
×
2028
  }
2029

2030
  if (pVtbScan->curOrgTbVg != NULL) {
108,794✔
2031
    // which means rversion has changed
2032
    void*   pCurIter = NULL;
9,170✔
2033
    SArray* tmpArray = NULL;
9,170✔
2034
    while ((pCurIter = taosHashIterate(pVtbScan->curOrgTbVg, pCurIter))) {
26,499✔
2035
      int32_t* vgId = (int32_t*)taosHashGetKey(pCurIter, NULL);
17,329✔
2036
      if (taosHashGet(pVtbScan->existOrgTbVg, vgId, sizeof(int32_t)) == NULL) {
17,329✔
2037
        if (tmpArray == NULL) {
2,189✔
2038
          tmpArray = taosArrayInit(1, sizeof(int32_t));
2,189✔
2039
          QUERY_CHECK_NULL(tmpArray, code, line, _return, terrno)
2,189✔
2040
        }
2041
        QUERY_CHECK_NULL(taosArrayPush(tmpArray, vgId), code, line, _return, terrno)
2,189✔
2042
      }
2043
    }
2044
    if (tmpArray == NULL) {
9,170✔
2045
      return TSDB_CODE_SUCCESS;
6,981✔
2046
    }
2047
    if (tmpArray != NULL && pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.addVgIds == NULL) {
2,189✔
2048
      SArray* expiredInfo = atomic_load_ptr(&pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.addedVgInfo);
2,189✔
2049
      if (expiredInfo && expiredInfo == atomic_val_compare_exchange_ptr(&pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.addedVgInfo, expiredInfo, NULL)) {
2,189✔
2050
        for (int32_t i = 0; i < taosArrayGetSize(expiredInfo); i++) {
×
2051
          SStreamTaskAddr* vgInfo = (SStreamTaskAddr*)taosArrayGet(expiredInfo, i);
×
2052
          QUERY_CHECK_NULL(taosArrayPush(tmpArray, &vgInfo->nodeId), code, line, _return, terrno)
×
2053
        }
2054
        taosArrayDestroy(expiredInfo);
×
2055
      }
2056
      if (atomic_val_compare_exchange_ptr(&pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.addVgIds, NULL, tmpArray)) {
2,189✔
2057
        taosArrayDestroy(tmpArray);
×
2058
      }
2059
    }
2060
    atomic_store_64(&pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.uid, (int64_t)(pVtbScan->isSuperTable ? pVtbScan->suid : pVtbScan->uid));
2,189✔
2061
    (void)atomic_val_compare_exchange_8(pTaskInfo->pStreamRuntimeInfo->vtableDeployGot, 0, 1);
2,189✔
2062
    taosHashClear(pVtbScan->curOrgTbVg);
2,189✔
2063
    pVtbScan->needRedeploy = true;
2,189✔
2064
    pVtbScan->rversion = rversion;
2,189✔
2065
    return TSDB_CODE_STREAM_VTABLE_NEED_REDEPLOY;
2,189✔
2066
  }
2067
  return code;
99,624✔
2068
_return:
×
2069
  qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
×
2070
  return code;
×
2071
}
2072

2073
int32_t getVgIdFromColref(SOperatorInfo* pOperator, const char* colRef, int32_t* vgId) {
23,578✔
2074
  int32_t                    code =TSDB_CODE_SUCCESS;
23,578✔
2075
  int32_t                    line = 0;
23,578✔
2076
  char*                      refDbName = NULL;
23,578✔
2077
  char*                      refTbName = NULL;
23,578✔
2078
  char*                      refColName = NULL;
23,578✔
2079
  SDBVgInfo*                 dbVgInfo = NULL;
23,578✔
2080
  SName                      name = {0};
23,578✔
2081
  char                       dbFname[TSDB_DB_FNAME_LEN] = {0};
23,578✔
2082
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
23,578✔
2083

2084
  code = extractColRefName(colRef, &refDbName, &refTbName, &refColName);
23,578✔
2085
  QUERY_CHECK_CODE(code, line, _return);
23,578✔
2086

2087
  toName(pInfo->vtbScan.acctId, refDbName, refTbName, &name);
23,578✔
2088

2089
  code = getDbVgInfo(pOperator, &name, &dbVgInfo);
23,578✔
2090
  QUERY_CHECK_CODE(code, line, _return);
23,578✔
2091

2092
  code = tNameGetFullDbName(&name, dbFname);
23,578✔
2093
  QUERY_CHECK_CODE(code, line, _return);
23,578✔
2094

2095
  code = getVgId(dbVgInfo, dbFname, vgId, name.tname);
23,578✔
2096
  QUERY_CHECK_CODE(code, line, _return);
23,578✔
2097

2098
_return:
23,578✔
2099
  if (code) {
23,578✔
2100
    qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
×
2101
  }
2102
  taosMemoryFree(refDbName);
23,578✔
2103
  taosMemoryFree(refTbName);
23,578✔
2104
  taosMemoryFree(refColName);
23,578✔
2105
  return code;
23,578✔
2106
}
2107

UNCOV
2108
static int32_t generateTagArrayByTagBlockAndSave(SHashObj* vtbUidTagListMap, tb_uid_t uid, SSDataBlock *pTagVal, int32_t rowIdx) {
×
UNCOV
2109
  int32_t code = TSDB_CODE_SUCCESS;
×
UNCOV
2110
  int32_t line = 0;
×
UNCOV
2111
  STagVal tagVal = {0};
×
2112
  // last col is uid
2113

UNCOV
2114
  SArray* pTagList = taosArrayInit(1, sizeof(STagVal));
×
UNCOV
2115
  QUERY_CHECK_NULL(pTagList, code, line, _return, terrno)
×
2116

UNCOV
2117
  for (int32_t k = 0; k < taosArrayGetSize(pTagVal->pDataBlock) - 1; k++) {
×
UNCOV
2118
    SColumnInfoData *pTagCol = taosArrayGet(pTagVal->pDataBlock, k);
×
UNCOV
2119
    QUERY_CHECK_NULL(pTagCol, code, line, _return, terrno)
×
UNCOV
2120
    tagVal.type = pTagCol->info.type;
×
UNCOV
2121
    tagVal.cid = pTagCol->info.colId;
×
UNCOV
2122
    if (!colDataIsNull_s(pTagCol, rowIdx)) {
×
UNCOV
2123
      char*   pData = colDataGetData(pTagCol, rowIdx);
×
UNCOV
2124
      if (IS_VAR_DATA_TYPE(pTagCol->info.type)) {
×
UNCOV
2125
        tagVal.nData = varDataLen(pData);
×
UNCOV
2126
        tagVal.pData = taosMemoryMalloc(tagVal.nData);
×
UNCOV
2127
        QUERY_CHECK_NULL(tagVal.pData, code, line, _return, terrno)
×
UNCOV
2128
        memcpy(tagVal.pData, varDataVal(pData), varDataLen(pData));
×
UNCOV
2129
        QUERY_CHECK_NULL(taosArrayPush(pTagList, &tagVal), code, line, _return, terrno)
×
2130
      } else {
UNCOV
2131
        memcpy(&tagVal.i64, pData, tDataTypes[pTagCol->info.type].bytes);
×
UNCOV
2132
        QUERY_CHECK_NULL(taosArrayPush(pTagList, &tagVal), code, line, _return, terrno)
×
2133
      }
2134
    } else {
2135
      tagVal.pData = NULL;
×
2136
      tagVal.nData = 0;
×
2137
      QUERY_CHECK_NULL(taosArrayPush(pTagList, &tagVal), code, line, _return, terrno)
×
2138
    }
UNCOV
2139
    tagVal = (STagVal){0};
×
2140
  }
UNCOV
2141
  code = taosHashPut(vtbUidTagListMap, &uid, sizeof(uid), &pTagList, POINTER_BYTES);
×
UNCOV
2142
  QUERY_CHECK_CODE(code, line, _return);
×
2143

UNCOV
2144
  return code;
×
2145
_return:
×
2146
  if (tagVal.pData) {
×
2147
    taosMemoryFreeClear(tagVal.pData);
×
2148
  }
2149
  if (pTagList) {
×
2150
    taosArrayDestroyEx(pTagList, destroyTagVal);
×
2151
  }
2152
  qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
×
2153
  return code;
×
2154
}
2155

2156
int32_t virtualTableScanProcessColRefInfo(SOperatorInfo* pOperator, SArray* pColRefInfo, tb_uid_t* uid, int32_t* vgId) {
812,534✔
2157
  int32_t                    code = TSDB_CODE_SUCCESS;
812,534✔
2158
  int32_t                    line = 0;
812,534✔
2159
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
812,534✔
2160
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
812,534✔
2161
  SDBVgInfo*                 dbVgInfo = NULL;
812,534✔
2162

2163
  for (int32_t j = 0; j < taosArrayGetSize(pColRefInfo); j++) {
34,872,959✔
2164
    SColRefInfo *pKV = (SColRefInfo*)taosArrayGet(pColRefInfo, j);
34,060,425✔
2165
    *uid = pKV->uid;
34,060,425✔
2166
    *vgId = pKV->vgId;
34,060,425✔
2167
    if (pKV->colrefName != NULL && colNeedScan(pOperator, pKV->colId)) {
34,060,425✔
2168
      char*   refDbName = NULL;
4,396,309✔
2169
      char*   refTbName = NULL;
4,396,309✔
2170
      char*   refColName = NULL;
4,396,309✔
2171
      SName   name = {0};
4,396,309✔
2172
      char    dbFname[TSDB_DB_FNAME_LEN] = {0};
4,396,309✔
2173
      char    orgTbFName[TSDB_TABLE_FNAME_LEN] = {0};
4,396,309✔
2174

2175
      code = extractColRefName(pKV->colrefName, &refDbName, &refTbName, &refColName);
4,396,309✔
2176
      QUERY_CHECK_CODE(code, line, _return);
4,396,309✔
2177

2178
      toName(pInfo->vtbScan.acctId, refDbName, refTbName, &name);
4,396,309✔
2179

2180
      code = getDbVgInfo(pOperator, &name, &dbVgInfo);
4,396,309✔
2181
      QUERY_CHECK_CODE(code, line, _return);
4,396,309✔
2182
      code = tNameGetFullDbName(&name, dbFname);
4,396,309✔
2183
      QUERY_CHECK_CODE(code, line, _return);
4,396,309✔
2184
      code = tNameGetFullTableName(&name, orgTbFName);
4,396,309✔
2185
      QUERY_CHECK_CODE(code, line, _return);
4,396,309✔
2186

2187
      void *pVal = taosHashGet(pVtbScan->otbNameToOtbInfoMap, orgTbFName, sizeof(orgTbFName));
4,396,309✔
2188
      if (!pVal) {
4,396,309✔
2189
        SOrgTbInfo orgTbInfo = {0};
1,287,949✔
2190
        code = getVgId(dbVgInfo, dbFname, &orgTbInfo.vgId, name.tname);
1,287,949✔
2191
        QUERY_CHECK_CODE(code, line, _return);
1,287,949✔
2192
        tstrncpy(orgTbInfo.tbName, orgTbFName, sizeof(orgTbInfo.tbName));
1,287,949✔
2193
        orgTbInfo.colMap = taosArrayInit(10, sizeof(SColIdNameKV));
1,287,949✔
2194
        QUERY_CHECK_NULL(orgTbInfo.colMap, code, line, _return, terrno)
1,287,949✔
2195
        SColIdNameKV colIdNameKV = {0};
1,287,949✔
2196
        colIdNameKV.colId = pKV->colId;
1,287,949✔
2197
        tstrncpy(colIdNameKV.colName, refColName, sizeof(colIdNameKV.colName));
1,287,949✔
2198
        QUERY_CHECK_NULL(taosArrayPush(orgTbInfo.colMap, &colIdNameKV), code, line, _return, terrno)
2,575,898✔
2199
        code = taosHashPut(pVtbScan->otbNameToOtbInfoMap, orgTbFName, sizeof(orgTbFName), &orgTbInfo, sizeof(orgTbInfo));
1,287,949✔
2200
        QUERY_CHECK_CODE(code, line, _return);
1,287,949✔
2201
      } else {
2202
        SOrgTbInfo *tbInfo = (SOrgTbInfo *)pVal;
3,108,360✔
2203
        SColIdNameKV colIdNameKV = {0};
3,108,360✔
2204
        colIdNameKV.colId = pKV->colId;
3,108,360✔
2205
        tstrncpy(colIdNameKV.colName, refColName, sizeof(colIdNameKV.colName));
3,108,360✔
2206
        QUERY_CHECK_NULL(taosArrayPush(tbInfo->colMap, &colIdNameKV), code, line, _return, terrno)
6,216,720✔
2207
      }
2208
      taosMemoryFree(refDbName);
4,396,309✔
2209
      taosMemoryFree(refTbName);
4,396,309✔
2210
      taosMemoryFree(refColName);
4,396,309✔
2211
    }
2212
  }
2213

2214
_return:
812,534✔
2215
  if (code) {
812,534✔
2216
    qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
×
2217
  }
2218
  return code;
812,534✔
2219
}
2220

UNCOV
2221
static int32_t getTagBlockAndProcess(SOperatorInfo* pOperator, bool hasPartition) {
×
UNCOV
2222
  int32_t                    code = TSDB_CODE_SUCCESS;
×
UNCOV
2223
  int32_t                    line = 0;
×
UNCOV
2224
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
×
UNCOV
2225
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
×
UNCOV
2226
  SExecTaskInfo*             pTaskInfo = pOperator->pTaskInfo;
×
UNCOV
2227
  SArray*                    pColRefArray = NULL;
×
UNCOV
2228
  SOperatorInfo*             pSystableScanOp = pOperator->pDownstream[0];
×
UNCOV
2229
  SOperatorInfo*             pTagScanOp = pOperator->pDownstream[1];
×
2230

UNCOV
2231
  pVtbScan->vtbUidTagListMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
×
UNCOV
2232
  QUERY_CHECK_NULL(pVtbScan->vtbUidTagListMap, code, line, _return, terrno)
×
UNCOV
2233
  taosHashSetFreeFp(pVtbScan->vtbUidTagListMap, destroyTagList);
×
UNCOV
2234
  if (hasPartition) {
×
UNCOV
2235
    pVtbScan->vtbUidToGroupIdMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
×
UNCOV
2236
    pVtbScan->vtbGroupIdTagListMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_ENTRY_LOCK);
×
UNCOV
2237
    QUERY_CHECK_NULL(pVtbScan->vtbUidToGroupIdMap, code, line, _return, terrno)
×
UNCOV
2238
    QUERY_CHECK_NULL(pVtbScan->vtbGroupIdTagListMap, code, line, _return, terrno)
×
UNCOV
2239
    taosHashSetFreeFp(pVtbScan->vtbGroupIdTagListMap, destroyVtbUidTagListMap);
×
2240
  }
2241

UNCOV
2242
  while (true) {
×
UNCOV
2243
    SSDataBlock *pTagVal = NULL;
×
UNCOV
2244
    code = pTagScanOp->fpSet.getNextFn(pTagScanOp, &pTagVal);
×
UNCOV
2245
    QUERY_CHECK_CODE(code, line, _return);
×
UNCOV
2246
    if (pTagVal == NULL) {
×
UNCOV
2247
      break;
×
2248
    }
UNCOV
2249
    SHashObj *vtbUidTagListMap = NULL;
×
UNCOV
2250
    if (hasPartition) {
×
UNCOV
2251
      void* pIter = taosHashGet(pVtbScan->vtbGroupIdTagListMap, &pTagVal->info.id.groupId, sizeof(pTagVal->info.id.groupId));
×
UNCOV
2252
      if (pIter) {
×
UNCOV
2253
        vtbUidTagListMap = *(SHashObj**)pIter;
×
2254
      } else {
UNCOV
2255
        vtbUidTagListMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
×
UNCOV
2256
        QUERY_CHECK_NULL(vtbUidTagListMap, code, line, _return, terrno)
×
UNCOV
2257
        taosHashSetFreeFp(vtbUidTagListMap, destroyTagList);
×
2258

UNCOV
2259
        code = taosHashPut(pVtbScan->vtbGroupIdTagListMap, &pTagVal->info.id.groupId, sizeof(pTagVal->info.id.groupId), &vtbUidTagListMap, POINTER_BYTES);
×
UNCOV
2260
        QUERY_CHECK_CODE(code, line, _return);
×
2261
      }
2262
    } else {
UNCOV
2263
      vtbUidTagListMap = pVtbScan->vtbUidTagListMap;
×
2264
    }
2265

UNCOV
2266
    SColumnInfoData *pUidCol = taosArrayGetLast(pTagVal->pDataBlock);
×
UNCOV
2267
    QUERY_CHECK_NULL(pUidCol, code, line, _return, terrno)
×
UNCOV
2268
    for (int32_t i = 0; i < pTagVal->info.rows; i++) {
×
UNCOV
2269
      tb_uid_t uid = 0;
×
UNCOV
2270
      if (!colDataIsNull_s(pUidCol, i)) {
×
UNCOV
2271
        GET_TYPED_DATA(uid, int64_t, TSDB_DATA_TYPE_BIGINT, colDataGetNumData(pUidCol, i), 0);
×
UNCOV
2272
        QUERY_CHECK_CODE(code, line, _return);
×
2273
      }
2274

UNCOV
2275
      code = generateTagArrayByTagBlockAndSave(vtbUidTagListMap, uid, pTagVal, i);
×
UNCOV
2276
      QUERY_CHECK_CODE(code, line, _return);
×
2277

UNCOV
2278
      if (hasPartition) {
×
UNCOV
2279
        code = taosHashPut(pVtbScan->vtbUidToGroupIdMap, &uid, sizeof(uid), &pTagVal->info.id.groupId, sizeof(pTagVal->info.id.groupId));
×
UNCOV
2280
        QUERY_CHECK_CODE(code, line, _return);
×
2281
      }
2282
    }
2283
  }
2284

UNCOV
2285
  return code;
×
2286

2287
_return:
×
2288
  qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
×
2289
  return code;
×
2290
}
2291

UNCOV
2292
static int32_t processChildTableListAndGenerateOrgTbInfoMap(SOperatorInfo* pOperator) {
×
UNCOV
2293
  int32_t                    code = TSDB_CODE_SUCCESS;
×
UNCOV
2294
  int32_t                    line = 0;
×
UNCOV
2295
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
×
UNCOV
2296
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
×
UNCOV
2297
  SExecTaskInfo*             pTaskInfo = pOperator->pTaskInfo;
×
UNCOV
2298
  SArray*                    pColRefArray = NULL;
×
UNCOV
2299
  SOperatorInfo*             pSystableScanOp = pOperator->pDownstream[0];
×
UNCOV
2300
  SOperatorInfo*             pTagScanOp = pOperator->pDownstream[1];
×
2301

UNCOV
2302
  pVtbScan->vtbUidToVgIdMapMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
×
UNCOV
2303
  QUERY_CHECK_NULL(pVtbScan->vtbUidToVgIdMapMap, code, line, _return, terrno)
×
2304

UNCOV
2305
  for (int32_t i = 0; i < taosArrayGetSize(pVtbScan->childTableList); i++) {
×
UNCOV
2306
    SHashObj* otbVgIdToOtbInfoArrayMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_ENTRY_LOCK);
×
UNCOV
2307
    QUERY_CHECK_NULL(otbVgIdToOtbInfoArrayMap, code, line, _return, terrno)
×
2308

UNCOV
2309
    SArray* pColRefInfo = (SArray*)taosArrayGetP(pVtbScan->childTableList, i);
×
UNCOV
2310
    QUERY_CHECK_NULL(pColRefInfo, code, line, _return, terrno)
×
2311

UNCOV
2312
    tb_uid_t uid = 0;
×
UNCOV
2313
    int32_t  vgId = 0;
×
UNCOV
2314
    code = virtualTableScanProcessColRefInfo(pOperator, pColRefInfo, &uid, &vgId);
×
UNCOV
2315
    QUERY_CHECK_CODE(code, line, _return);
×
2316

UNCOV
2317
    size_t len = 0;
×
UNCOV
2318
    void*  pOrgTbInfo = taosHashIterate(pVtbScan->otbNameToOtbInfoMap, NULL);
×
UNCOV
2319
    while (pOrgTbInfo != NULL) {
×
UNCOV
2320
      char*       key = taosHashGetKey(pOrgTbInfo, &len);
×
UNCOV
2321
      SOrgTbInfo* orgTbInfo = (SOrgTbInfo*)pOrgTbInfo;
×
2322

UNCOV
2323
      void* pIter = taosHashGet(otbVgIdToOtbInfoArrayMap, &orgTbInfo->vgId, sizeof(orgTbInfo->vgId));
×
UNCOV
2324
      if (!pIter) {
×
UNCOV
2325
        SArray* pOrgTbInfoArray = taosArrayInit(1, sizeof(SOrgTbInfo));
×
UNCOV
2326
        QUERY_CHECK_NULL(pOrgTbInfoArray, code, line, _return, terrno)
×
UNCOV
2327
        QUERY_CHECK_NULL(taosArrayPush(pOrgTbInfoArray, orgTbInfo), code, line, _return, terrno)
×
UNCOV
2328
        code = taosHashPut(otbVgIdToOtbInfoArrayMap, &orgTbInfo->vgId, sizeof(orgTbInfo->vgId), &pOrgTbInfoArray, POINTER_BYTES);
×
UNCOV
2329
        QUERY_CHECK_CODE(code, line, _return);
×
2330
      } else {
UNCOV
2331
        SArray* pOrgTbInfoArray = *(SArray**)pIter;
×
UNCOV
2332
        QUERY_CHECK_NULL(pOrgTbInfoArray, code, line, _return, terrno)
×
UNCOV
2333
        QUERY_CHECK_NULL(taosArrayPush(pOrgTbInfoArray, orgTbInfo), code, line, _return, terrno)
×
2334
      }
2335

UNCOV
2336
      pOrgTbInfo = taosHashIterate(pVtbScan->otbNameToOtbInfoMap, pOrgTbInfo);
×
2337

UNCOV
2338
      code = taosHashRemove(pVtbScan->otbNameToOtbInfoMap, key, len);
×
UNCOV
2339
      QUERY_CHECK_CODE(code, line, _return);
×
2340
    }
2341

UNCOV
2342
    code = taosHashPut(pVtbScan->vtbUidToVgIdMapMap, &uid, sizeof(uid), &otbVgIdToOtbInfoArrayMap, POINTER_BYTES);
×
UNCOV
2343
    QUERY_CHECK_CODE(code, line, _return);
×
2344
  }
2345

UNCOV
2346
  return code;
×
2347
_return:
×
2348
  qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
×
2349
  return code;
×
2350
}
2351

UNCOV
2352
static int32_t buildOrgTbInfoSingle(SOperatorInfo* pOperator, bool hasPartition) {
×
UNCOV
2353
  int32_t                    code = TSDB_CODE_SUCCESS;
×
UNCOV
2354
  int32_t                    line = 0;
×
2355

UNCOV
2356
  code = processChildTableListAndGenerateOrgTbInfoMap(pOperator);
×
UNCOV
2357
  QUERY_CHECK_CODE(code, line, _return);
×
2358

2359
  // process tag
UNCOV
2360
  code = getTagBlockAndProcess(pOperator, hasPartition);
×
UNCOV
2361
  QUERY_CHECK_CODE(code, line, _return);
×
2362

UNCOV
2363
  return code;
×
2364
_return:
×
2365
  qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
×
2366
  return code;
×
2367
}
2368

UNCOV
2369
static int32_t buildOrgTbInfoBatch(SOperatorInfo* pOperator, bool hasPartition) {
×
UNCOV
2370
  int32_t                    code = TSDB_CODE_SUCCESS;
×
UNCOV
2371
  int32_t                    line = 0;
×
UNCOV
2372
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
×
UNCOV
2373
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
×
UNCOV
2374
  SExecTaskInfo*             pTaskInfo = pOperator->pTaskInfo;
×
UNCOV
2375
  SArray*                    pColRefArray = NULL;
×
UNCOV
2376
  SOperatorInfo*             pSystableScanOp = pOperator->pDownstream[0];
×
UNCOV
2377
  SOperatorInfo*             pTagScanOp = pOperator->pDownstream[1];
×
2378

UNCOV
2379
  if (hasPartition) {
×
UNCOV
2380
    pVtbScan->vtbUidToGroupIdMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
×
UNCOV
2381
    pVtbScan->vtbGroupIdTagListMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_ENTRY_LOCK);
×
UNCOV
2382
    pVtbScan->vtbGroupIdToVgIdMapMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_ENTRY_LOCK);
×
2383

UNCOV
2384
    QUERY_CHECK_NULL(pVtbScan->vtbGroupIdToVgIdMapMap, code, line, _return, terrno)
×
UNCOV
2385
    QUERY_CHECK_NULL(pVtbScan->vtbUidToGroupIdMap, code, line, _return, terrno)
×
UNCOV
2386
    QUERY_CHECK_NULL(pVtbScan->vtbGroupIdTagListMap, code, line, _return, terrno)
×
UNCOV
2387
    taosHashSetFreeFp(pVtbScan->vtbGroupIdToVgIdMapMap, destroyOtbVgIdToOtbInfoArrayMap);
×
2388
  } else {
UNCOV
2389
    pVtbScan->otbVgIdToOtbInfoArrayMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_ENTRY_LOCK);
×
UNCOV
2390
    QUERY_CHECK_NULL(pVtbScan->otbVgIdToOtbInfoArrayMap, code, line, _return, terrno)
×
2391
  }
2392

UNCOV
2393
  while (true && hasPartition) {
×
UNCOV
2394
    SSDataBlock* pTagVal = NULL;
×
UNCOV
2395
    code = pTagScanOp->fpSet.getNextFn(pTagScanOp, &pTagVal);
×
UNCOV
2396
    QUERY_CHECK_CODE(code, line, _return);
×
UNCOV
2397
    if (pTagVal == NULL) {
×
UNCOV
2398
      break;
×
2399
    }
2400

UNCOV
2401
    SColumnInfoData *pUidCol = taosArrayGetLast(pTagVal->pDataBlock);
×
UNCOV
2402
    QUERY_CHECK_NULL(pUidCol, code, line, _return, terrno)
×
UNCOV
2403
    for (int32_t i = 0; i < pTagVal->info.rows; i++) {
×
UNCOV
2404
      tb_uid_t uid = 0;
×
UNCOV
2405
      if (!colDataIsNull_s(pUidCol, i)) {
×
UNCOV
2406
        GET_TYPED_DATA(uid, int64_t, TSDB_DATA_TYPE_BIGINT, colDataGetNumData(pUidCol, i), 0);
×
UNCOV
2407
        QUERY_CHECK_CODE(code, line, _return);
×
2408
      }
UNCOV
2409
      code = taosHashPut(pVtbScan->vtbUidToGroupIdMap, &uid, sizeof(uid), &pTagVal->info.id.groupId, sizeof(pTagVal->info.id.groupId));
×
UNCOV
2410
      QUERY_CHECK_CODE(code, line, _return);
×
2411
    }
UNCOV
2412
    code = taosHashPut(pVtbScan->vtbGroupIdTagListMap, &pTagVal->info.id.groupId, sizeof(pTagVal->info.id.groupId), NULL, 0);
×
UNCOV
2413
    QUERY_CHECK_CODE(code, line, _return);
×
2414
  }
2415

UNCOV
2416
  for (int32_t i = 0; i < taosArrayGetSize(pVtbScan->childTableList); i++) {
×
UNCOV
2417
    SArray* pColRefInfo = (SArray*)taosArrayGetP(pVtbScan->childTableList, i);
×
UNCOV
2418
    QUERY_CHECK_NULL(pColRefInfo, code, line, _return, terrno)
×
UNCOV
2419
    tb_uid_t uid = 0;
×
UNCOV
2420
    int32_t  vgId = 0;
×
UNCOV
2421
    code = virtualTableScanProcessColRefInfo(pOperator, pColRefInfo, &uid, &vgId);
×
UNCOV
2422
    QUERY_CHECK_CODE(code, line, _return);
×
2423

UNCOV
2424
    SHashObj* otbVgIdToOtbInfoArrayMap = NULL;
×
UNCOV
2425
    if (hasPartition) {
×
UNCOV
2426
      uint64_t* groupId = (uint64_t *)taosHashGet(pVtbScan->vtbUidToGroupIdMap, &uid, sizeof(uid));
×
UNCOV
2427
      QUERY_CHECK_NULL(groupId, code, line, _return, terrno)
×
2428

UNCOV
2429
      void* pHashIter = taosHashGet(pVtbScan->vtbGroupIdToVgIdMapMap, groupId, sizeof(*groupId));
×
UNCOV
2430
      if (pHashIter) {
×
UNCOV
2431
        otbVgIdToOtbInfoArrayMap = *(SHashObj**)pHashIter;
×
2432
      } else {
UNCOV
2433
        otbVgIdToOtbInfoArrayMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_ENTRY_LOCK);
×
UNCOV
2434
        QUERY_CHECK_NULL(otbVgIdToOtbInfoArrayMap, code, line, _return, terrno)
×
UNCOV
2435
        code = taosHashPut(pVtbScan->vtbGroupIdToVgIdMapMap, groupId, sizeof(*groupId), &otbVgIdToOtbInfoArrayMap, POINTER_BYTES);
×
UNCOV
2436
        QUERY_CHECK_CODE(code, line, _return);
×
2437
      }
2438
    } else {
UNCOV
2439
      otbVgIdToOtbInfoArrayMap = pVtbScan->otbVgIdToOtbInfoArrayMap;
×
2440
    }
2441

UNCOV
2442
    size_t len = 0;
×
UNCOV
2443
    void*  pOrgTbInfo = taosHashIterate(pVtbScan->otbNameToOtbInfoMap, NULL);
×
UNCOV
2444
    while (pOrgTbInfo != NULL) {
×
UNCOV
2445
      char*       key = taosHashGetKey(pOrgTbInfo, &len);
×
UNCOV
2446
      SOrgTbInfo* orgTbInfo = (SOrgTbInfo*)pOrgTbInfo;
×
UNCOV
2447
      void* pIter = taosHashGet(otbVgIdToOtbInfoArrayMap, &orgTbInfo->vgId, sizeof(orgTbInfo->vgId));
×
UNCOV
2448
      if (!pIter) {
×
UNCOV
2449
        SArray* pOrgTbInfoArray = taosArrayInit(1, sizeof(SOrgTbInfo));
×
UNCOV
2450
        QUERY_CHECK_NULL(pOrgTbInfoArray, code, line, _return, terrno)
×
UNCOV
2451
        QUERY_CHECK_NULL(taosArrayPush(pOrgTbInfoArray, orgTbInfo), code, line, _return, terrno)
×
UNCOV
2452
        code = taosHashPut(otbVgIdToOtbInfoArrayMap, &orgTbInfo->vgId, sizeof(orgTbInfo->vgId), &pOrgTbInfoArray, POINTER_BYTES);
×
UNCOV
2453
        QUERY_CHECK_CODE(code, line, _return);
×
2454
      } else {
UNCOV
2455
        SArray* pOrgTbInfoArray = *(SArray**)pIter;
×
UNCOV
2456
        QUERY_CHECK_NULL(pOrgTbInfoArray, code, line, _return, terrno)
×
UNCOV
2457
        QUERY_CHECK_NULL(taosArrayPush(pOrgTbInfoArray, orgTbInfo), code, line, _return, terrno)
×
2458
      }
2459

UNCOV
2460
      pOrgTbInfo = taosHashIterate(pVtbScan->otbNameToOtbInfoMap, pOrgTbInfo);
×
2461

UNCOV
2462
      code = taosHashRemove(pVtbScan->otbNameToOtbInfoMap, key, len);
×
UNCOV
2463
      QUERY_CHECK_CODE(code, line, _return);
×
2464
    }
2465
  }
UNCOV
2466
  return code;
×
2467
_return:
×
2468
  qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
×
2469
  return code;
×
2470
}
2471

2472
int32_t buildVirtualSuperTableScanChildTableMap(SOperatorInfo* pOperator) {
247,452✔
2473
  int32_t                    code = TSDB_CODE_SUCCESS;
247,452✔
2474
  int32_t                    line = 0;
247,452✔
2475
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
247,452✔
2476
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
247,452✔
2477
  SExecTaskInfo*             pTaskInfo = pOperator->pTaskInfo;
247,452✔
2478
  SArray*                    pColRefArray = NULL;
247,452✔
2479
  SOperatorInfo*             pSystableScanOp = NULL;
247,452✔
2480
  
2481
  pVtbScan->childTableMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
247,452✔
2482
  QUERY_CHECK_NULL(pVtbScan->childTableMap, code, line, _return, terrno)
247,452✔
2483

2484
  if (pInfo->qType == DYN_QTYPE_VTB_AGG) {
247,452✔
UNCOV
2485
    pVtbScan->otbNameToOtbInfoMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
×
UNCOV
2486
    QUERY_CHECK_NULL(pVtbScan->otbNameToOtbInfoMap, code, line, _return, terrno)
×
UNCOV
2487
    pSystableScanOp = pOperator->pDownstream[0];
×
2488
  } else {
2489
    pSystableScanOp = pOperator->pDownstream[1];
247,452✔
2490
  }
2491

2492
  while (true) {
496,967✔
2493
    SSDataBlock *pChildInfo = NULL;
744,419✔
2494
    code = pSystableScanOp->fpSet.getNextFn(pSystableScanOp, &pChildInfo);
744,419✔
2495
    QUERY_CHECK_CODE(code, line, _return);
744,419✔
2496
    if (pChildInfo == NULL) {
744,419✔
2497
      break;
247,452✔
2498
    }
2499
    SColumnInfoData *pTableNameCol = taosArrayGet(pChildInfo->pDataBlock, 0);
496,967✔
2500
    SColumnInfoData *pStbNameCol = taosArrayGet(pChildInfo->pDataBlock, 1);
496,967✔
2501
    SColumnInfoData *pDbNameCol = taosArrayGet(pChildInfo->pDataBlock, 2);
496,967✔
2502

2503
    QUERY_CHECK_NULL(pTableNameCol, code, line, _return, terrno)
496,967✔
2504
    QUERY_CHECK_NULL(pStbNameCol, code, line, _return, terrno)
496,967✔
2505
    QUERY_CHECK_NULL(pDbNameCol, code, line, _return, terrno)
496,967✔
2506

2507
    for (int32_t i = 0; i < pChildInfo->info.rows; i++) {
45,524,215✔
2508
      if (!colDataIsNull_s(pStbNameCol, i)) {
90,054,496✔
2509
        char* stbrawname = colDataGetData(pStbNameCol, i);
45,027,248✔
2510
        char* dbrawname = colDataGetData(pDbNameCol, i);
45,027,248✔
2511
        char *ctbName = colDataGetData(pTableNameCol, i);
45,027,248✔
2512

2513
        if (tableInfoNeedCollect(dbrawname, stbrawname, pInfo->vtbScan.dbName, pInfo->vtbScan.tbName)) {
45,027,248✔
2514
          SColRefInfo info = {0};
33,556,980✔
2515
          code = getColRefInfo(&info, pChildInfo->pDataBlock, i);
33,556,980✔
2516
          QUERY_CHECK_CODE(code, line, _return);
33,556,980✔
2517

2518
          if (pInfo->qType == DYN_QTYPE_VTB_SCAN) {
33,556,980✔
2519
            if (pInfo->vtbScan.dynTbUid != 0 && info.uid != pInfo->vtbScan.dynTbUid) {
33,556,980✔
2520
              qTrace("dynQueryCtrl tb uid filter, info uid:%" PRIu64 ", dyn tb uid:%" PRIu64, info.uid,
×
2521
                     pInfo->vtbScan.dynTbUid);
2522
              destroyColRefInfo(&info);
×
2523
              continue;
×
2524
            }
2525

2526
            if (pTaskInfo->pStreamRuntimeInfo) {
33,556,980✔
2527
              if (pVtbScan->curOrgTbVg == NULL) {
31,968✔
2528
                pVtbScan->curOrgTbVg = taosHashInit(1, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_ENTRY_LOCK);
1,202✔
2529
                QUERY_CHECK_NULL(pVtbScan->curOrgTbVg, code, line, _return, terrno)
1,202✔
2530
              }
2531

2532
              if (info.colrefName) {
31,968✔
2533
                int32_t vgId;
17,210✔
2534
                code = getVgIdFromColref(pOperator, info.colrefName, &vgId);
17,210✔
2535
                QUERY_CHECK_CODE(code, line, _return);
17,210✔
2536
                code = taosHashPut(pVtbScan->curOrgTbVg, &vgId, sizeof(vgId), NULL, 0);
17,210✔
2537
                QUERY_CHECK_CODE(code, line, _return);
17,210✔
2538
              }
2539
            }
2540
          }
2541

2542
          if (taosHashGet(pVtbScan->childTableMap, varDataVal(ctbName), varDataLen(ctbName)) == NULL) {
33,556,980✔
2543
            pColRefArray = taosArrayInit(1, sizeof(SColRefInfo));
711,716✔
2544
            QUERY_CHECK_NULL(pColRefArray, code, line, _return, terrno)
711,716✔
2545
            QUERY_CHECK_NULL(taosArrayPush(pColRefArray, &info), code, line, _return, terrno)
1,423,432✔
2546
            int32_t tableIdx = (int32_t)taosArrayGetSize(pVtbScan->childTableList);
711,716✔
2547
            QUERY_CHECK_NULL(taosArrayPush(pVtbScan->childTableList, &pColRefArray), code, line, _return, terrno)
1,423,432✔
2548
            code = taosHashPut(pVtbScan->childTableMap, varDataVal(ctbName), varDataLen(ctbName), &tableIdx, sizeof(tableIdx));
711,716✔
2549
            QUERY_CHECK_CODE(code, line, _return);
711,716✔
2550
          } else {
2551
            int32_t *tableIdx = (int32_t*)taosHashGet(pVtbScan->childTableMap, varDataVal(ctbName), varDataLen(ctbName));
32,845,264✔
2552
            QUERY_CHECK_NULL(tableIdx, code, line, _return, terrno)
32,845,264✔
2553
            pColRefArray = (SArray *)taosArrayGetP(pVtbScan->childTableList, *tableIdx);
32,845,264✔
2554
            QUERY_CHECK_NULL(pColRefArray, code, line, _return, terrno)
32,845,264✔
2555
            QUERY_CHECK_NULL(taosArrayPush(pColRefArray, &info), code, line, _return, terrno)
65,690,528✔
2556
          }
2557
        }
2558
      }
2559
    }
2560
  }
2561

2562
  switch (pInfo->qType) {
247,452✔
UNCOV
2563
    case DYN_QTYPE_VTB_AGG: {
×
UNCOV
2564
      if (pVtbScan->batchProcessChild) {
×
UNCOV
2565
        code = buildOrgTbInfoBatch(pOperator, pVtbScan->hasPartition);
×
2566
      } else {
UNCOV
2567
        code = buildOrgTbInfoSingle(pOperator, pVtbScan->hasPartition);
×
2568
      }
UNCOV
2569
      break;
×
2570
    }
2571
    case DYN_QTYPE_VTB_SCAN: {
247,452✔
2572
      code = processOrgTbVg(pVtbScan, pTaskInfo, 1);
247,452✔
2573
      break;
247,452✔
2574
    }
2575
    default: {
×
2576
      code = TSDB_CODE_PLAN_INVALID_DYN_CTRL_TYPE;
×
2577
      break;
×
2578
    }
2579
  }
2580

2581
  QUERY_CHECK_CODE(code, line, _return);
247,452✔
2582

2583
_return:
247,452✔
2584
  if (code) {
247,452✔
2585
    qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
995✔
2586
  }
2587
  return code;
247,452✔
2588
}
2589

2590
int32_t buildVirtualNormalChildTableScanChildTableMap(SOperatorInfo* pOperator) {
103,604✔
2591
  int32_t                    code = TSDB_CODE_SUCCESS;
103,604✔
2592
  int32_t                    line = 0;
103,604✔
2593
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
103,604✔
2594
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
103,604✔
2595
  SExecTaskInfo*             pTaskInfo = pOperator->pTaskInfo;
103,604✔
2596
  SArray*                    pColRefInfo = pInfo->vtbScan.colRefInfo;
103,604✔
2597
  SOperatorInfo*             pSystableScanOp = pOperator->pDownstream[1];
103,604✔
2598
  int32_t                    rversion = 0;
103,604✔
2599

2600
  pInfo->vtbScan.colRefInfo = taosArrayInit(1, sizeof(SColRefInfo));
103,604✔
2601
  QUERY_CHECK_NULL(pInfo->vtbScan.colRefInfo, code, line, _return, terrno)
103,604✔
2602

2603
  while (true) {
203,809✔
2604
    SSDataBlock *pTableInfo = NULL;
307,413✔
2605
    code = pSystableScanOp->fpSet.getNextFn(pSystableScanOp, &pTableInfo);
307,413✔
2606
    if (pTableInfo == NULL) {
307,413✔
2607
      break;
103,604✔
2608
    }
2609

2610
    SColumnInfoData *pTableNameCol = taosArrayGet(pTableInfo->pDataBlock, 0);
203,809✔
2611
    SColumnInfoData *pDbNameCol = taosArrayGet(pTableInfo->pDataBlock, 2);
203,809✔
2612
    SColumnInfoData *pRefVerCol = taosArrayGet(pTableInfo->pDataBlock, 8);
203,809✔
2613

2614
    QUERY_CHECK_NULL(pTableNameCol, code, line, _return, terrno)
203,809✔
2615
    QUERY_CHECK_NULL(pDbNameCol, code, line, _return, terrno)
203,809✔
2616
    QUERY_CHECK_NULL(pRefVerCol, code, line, _return, terrno)
203,809✔
2617

2618
    for (int32_t i = 0; i < pTableInfo->info.rows; i++) {
7,565,460✔
2619
      if (!colDataIsNull_s(pRefVerCol, i)) {
14,723,302✔
2620
        GET_TYPED_DATA(rversion, int32_t, TSDB_DATA_TYPE_INT, colDataGetNumData(pRefVerCol, i), 0);
7,361,651✔
2621
      }
2622

2623
      if (!colDataIsNull_s(pTableNameCol, i)) {
14,723,302✔
2624
        char* tbrawname = colDataGetData(pTableNameCol, i);
7,361,651✔
2625
        char* dbrawname = colDataGetData(pDbNameCol, i);
7,361,651✔
2626
        QUERY_CHECK_NULL(tbrawname, code, line, _return, terrno)
7,361,651✔
2627
        QUERY_CHECK_NULL(dbrawname, code, line, _return, terrno)
7,361,651✔
2628

2629
        if (tableInfoNeedCollect(dbrawname, tbrawname, pInfo->vtbScan.dbName, pInfo->vtbScan.tbName)) {
7,361,651✔
2630
          SColRefInfo info = {0};
514,589✔
2631
          code = getColRefInfo(&info, pTableInfo->pDataBlock, i);
514,589✔
2632
          QUERY_CHECK_CODE(code, line, _return);
514,589✔
2633

2634
          if ((rversion != pVtbScan->rversion || pVtbScan->existOrgTbVg == NULL) && info.colrefName) {
514,589✔
2635
            if (pVtbScan->curOrgTbVg == NULL) {
6,368✔
2636
              pVtbScan->curOrgTbVg = taosHashInit(1, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_ENTRY_LOCK);
398✔
2637
              QUERY_CHECK_NULL(pVtbScan->curOrgTbVg, code, line, _return, terrno)
398✔
2638
            }
2639
            int32_t vgId;
6,368✔
2640
            code = getVgIdFromColref(pOperator, info.colrefName, &vgId);
6,368✔
2641
            QUERY_CHECK_CODE(code, line, _return);
6,368✔
2642
            code = taosHashPut(pVtbScan->curOrgTbVg, &vgId, sizeof(vgId), NULL, 0);
6,368✔
2643
            QUERY_CHECK_CODE(code, line, _return);
6,368✔
2644
          }
2645

2646
          QUERY_CHECK_NULL(taosArrayPush(pInfo->vtbScan.colRefInfo, &info), code, line, _return, terrno)
1,029,178✔
2647
        }
2648
      }
2649
    }
2650
  }
2651
  code = processOrgTbVg(pVtbScan, pTaskInfo, rversion);
103,604✔
2652
  QUERY_CHECK_CODE(code, line, _return);
103,604✔
2653

2654
_return:
102,410✔
2655
  if (code) {
103,604✔
2656
    qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
1,194✔
2657
  }
2658
  return code;
103,604✔
2659
}
2660

2661
int32_t virtualTableScanCheckNeedRedeploy(SOperatorInfo* pOperator) {
1,521,753✔
2662
  int32_t                    code = TSDB_CODE_SUCCESS;
1,521,753✔
2663
  int32_t                    line = 0;
1,521,753✔
2664
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
1,521,753✔
2665
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
1,521,753✔
2666
  SExecTaskInfo*             pTaskInfo = pOperator->pTaskInfo;
1,521,753✔
2667

2668
  SArray *tmpArray = NULL;
1,521,753✔
2669
  tmpArray = atomic_load_ptr(&pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.addedVgInfo);
1,521,753✔
2670
  if (tmpArray && tmpArray == atomic_val_compare_exchange_ptr(&pTaskInfo->pStreamRuntimeInfo->vtableDeployInfo.addedVgInfo, tmpArray, NULL)) {
1,521,753✔
2671
    for (int32_t i = 0; i < taosArrayGetSize(tmpArray); i++) {
4,378✔
2672
      SStreamTaskAddr* pTaskAddr = (SStreamTaskAddr*)taosArrayGet(tmpArray, i);
2,189✔
2673
      code = taosHashPut(pVtbScan->existOrgTbVg, &pTaskAddr->nodeId, sizeof(pTaskAddr->nodeId), NULL, 0);
2,189✔
2674
      QUERY_CHECK_CODE(code, line, _return);
2,189✔
2675
      if (pVtbScan->newAddedVgInfo == NULL) {
2,189✔
2676
        pVtbScan->newAddedVgInfo = taosHashInit(1, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_ENTRY_LOCK);
796✔
2677
        QUERY_CHECK_NULL(pVtbScan->newAddedVgInfo, code, line, _return, terrno)
796✔
2678
      }
2679
      code = taosHashPut(pVtbScan->newAddedVgInfo, &pTaskAddr->nodeId, sizeof(pTaskAddr->nodeId), pTaskAddr, sizeof(SStreamTaskAddr));
2,189✔
2680
      QUERY_CHECK_CODE(code, line, _return);
2,189✔
2681
    }
2682
    pVtbScan->needRedeploy = false;
2,189✔
2683
  } else {
2684
    code = TSDB_CODE_STREAM_VTABLE_NEED_REDEPLOY;
1,519,564✔
2685
    QUERY_CHECK_CODE(code, line, _return);
1,519,564✔
2686
  }
2687

2688
_return:
×
2689
  taosArrayClear(tmpArray);
1,521,753✔
2690
  taosArrayDestroy(tmpArray);
1,521,753✔
2691
  if (code) {
1,521,753✔
2692
    qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
1,519,564✔
2693
  }
2694
  return code;
1,521,753✔
2695
}
2696

2697
int32_t virtualTableScanBuildDownStreamOpParam(SOperatorInfo* pOperator, tb_uid_t uid, int32_t vgId) {
812,534✔
2698
  int32_t                    code = TSDB_CODE_SUCCESS;
812,534✔
2699
  int32_t                    line = 0;
812,534✔
2700
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
812,534✔
2701
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
812,534✔
2702
  SExecTaskInfo*             pTaskInfo = pOperator->pTaskInfo;
812,534✔
2703

2704
  pVtbScan->vtbScanParam = NULL;
812,534✔
2705
  code = buildVtbScanOperatorParam(pInfo, &pVtbScan->vtbScanParam, uid);
812,534✔
2706
  QUERY_CHECK_CODE(code, line, _return);
812,534✔
2707

2708
  void* pIter = taosHashIterate(pVtbScan->otbNameToOtbInfoMap, NULL);
812,534✔
2709
  while (pIter != NULL) {
2,100,483✔
2710
    SOrgTbInfo*      pMap = (SOrgTbInfo*)pIter;
1,287,949✔
2711
    SOperatorParam*  pExchangeParam = NULL;
1,287,949✔
2712
    SStreamTaskAddr* addr = taosHashGet(pVtbScan->newAddedVgInfo, &pMap->vgId, sizeof(pMap->vgId));
1,287,949✔
2713
    if (addr != NULL) {
1,287,949✔
2714
      SDownstreamSourceNode newSource = {0};
2,189✔
2715
      newSource.type = QUERY_NODE_DOWNSTREAM_SOURCE;
2,189✔
2716
      newSource.clientId = pTaskInfo->id.taskId;// current task's taskid
2,189✔
2717
      newSource.taskId = addr->taskId;
2,189✔
2718
      newSource.fetchMsgType = TDMT_STREAM_FETCH;
2,189✔
2719
      newSource.localExec = false;
2,189✔
2720
      newSource.addr.nodeId = addr->nodeId;
2,189✔
2721
      memcpy(&newSource.addr.epSet, &addr->epset, sizeof(SEpSet));
2,189✔
2722

2723
      code = buildExchangeOperatorParamForVScan(&pExchangeParam, 0, pMap, &newSource);
2,189✔
2724
      QUERY_CHECK_CODE(code, line, _return);
2,189✔
2725
      code = taosHashRemove(pVtbScan->newAddedVgInfo, &pMap->vgId, sizeof(pMap->vgId));
2,189✔
2726
      QUERY_CHECK_CODE(code, line, _return);
2,189✔
2727
    } else {
2728
      code = buildExchangeOperatorParamForVScan(&pExchangeParam, 0, pMap, NULL);
1,285,760✔
2729
      QUERY_CHECK_CODE(code, line, _return);
1,285,760✔
2730
    }
2731
    QUERY_CHECK_NULL(taosArrayPush(((SVTableScanOperatorParam*)pVtbScan->vtbScanParam->value)->pOpParamArray, &pExchangeParam), code, line, _return, terrno)
2,575,898✔
2732
    pIter = taosHashIterate(pVtbScan->otbNameToOtbInfoMap, pIter);
1,287,949✔
2733
  }
2734

2735
  SOperatorParam*  pExchangeParam = NULL;
812,534✔
2736
  code = buildExchangeOperatorParamForVTagScan(&pExchangeParam, 0, vgId, uid);
812,534✔
2737
  QUERY_CHECK_CODE(code, line, _return);
812,534✔
2738
  ((SVTableScanOperatorParam*)pVtbScan->vtbScanParam->value)->pTagScanOp = pExchangeParam;
812,534✔
2739

2740
_return:
812,534✔
2741
  if (code) {
812,534✔
2742
    qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
×
2743
  }
2744
  return code;
812,534✔
2745
}
2746

2747
int32_t virtualTableScanGetNext(SOperatorInfo* pOperator, SSDataBlock** pRes) {
1,329,068✔
2748
  int32_t                    code = TSDB_CODE_SUCCESS;
1,329,068✔
2749
  int32_t                    line = 0;
1,329,068✔
2750
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
1,329,068✔
2751
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
1,329,068✔
2752
  SOperatorInfo*             pVtbScanOp = pOperator->pDownstream[0];
1,329,068✔
2753

2754
  pVtbScan->otbNameToOtbInfoMap = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
1,329,068✔
2755
  QUERY_CHECK_NULL(pVtbScan->otbNameToOtbInfoMap, code, line, _return, terrno)
1,329,068✔
2756
  taosHashSetFreeFp(pVtbScan->otbNameToOtbInfoMap, destroySOrgTbInfo);
1,329,068✔
2757

2758
  while (true) {
2759
    if (pVtbScan->curTableIdx == pVtbScan->lastTableIdx) {
1,812,535✔
2760
      code = pVtbScanOp->fpSet.getNextFn(pVtbScanOp, pRes);
1,000,001✔
2761
      QUERY_CHECK_CODE(code, line, _return);
1,000,001✔
2762
    } else {
2763
      taosHashClear(pVtbScan->otbNameToOtbInfoMap);
812,534✔
2764
      SArray* pColRefInfo = NULL;
812,534✔
2765
      if (pVtbScan->isSuperTable) {
812,534✔
2766
        pColRefInfo = (SArray*)taosArrayGetP(pVtbScan->childTableList, pVtbScan->curTableIdx);
710,124✔
2767
      } else {
2768
        pColRefInfo = pInfo->vtbScan.colRefInfo;
102,410✔
2769
      }
2770
      QUERY_CHECK_NULL(pColRefInfo, code, line, _return, terrno)
812,534✔
2771

2772
      tb_uid_t uid = 0;
812,534✔
2773
      int32_t  vgId = 0;
812,534✔
2774
      code = virtualTableScanProcessColRefInfo(pOperator, pColRefInfo, &uid, &vgId);
812,534✔
2775
      QUERY_CHECK_CODE(code, line, _return);
812,534✔
2776

2777
      code = virtualTableScanBuildDownStreamOpParam(pOperator, uid, vgId);
812,534✔
2778
      QUERY_CHECK_CODE(code, line, _return);
812,534✔
2779

2780
      // reset downstream operator's status
2781
      pVtbScanOp->status = OP_NOT_OPENED;
812,534✔
2782
      code = pVtbScanOp->fpSet.getNextExtFn(pVtbScanOp, pVtbScan->vtbScanParam, pRes);
812,534✔
2783
      QUERY_CHECK_CODE(code, line, _return);
811,925✔
2784
    }
2785

2786
    if (*pRes) {
1,811,926✔
2787
      // has result, still read data from this table.
2788
      pVtbScan->lastTableIdx = pVtbScan->curTableIdx;
1,000,001✔
2789
      break;
1,000,001✔
2790
    } else {
2791
      // no result, read next table.
2792
      pVtbScan->curTableIdx++;
811,925✔
2793
      if (pVtbScan->isSuperTable) {
811,925✔
2794
        if (pVtbScan->curTableIdx >= taosArrayGetSize(pVtbScan->childTableList)) {
709,515✔
2795
          setOperatorCompleted(pOperator);
226,048✔
2796
          break;
226,048✔
2797
        }
2798
      } else {
2799
        setOperatorCompleted(pOperator);
102,410✔
2800
        break;
102,410✔
2801
      }
2802
    }
2803
  }
2804

2805
_return:
1,328,459✔
2806
  taosHashCleanup(pVtbScan->otbNameToOtbInfoMap);
1,328,459✔
2807
  pVtbScan->otbNameToOtbInfoMap = NULL;
1,328,459✔
2808
  if (code) {
1,328,459✔
2809
    qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
×
2810
  }
2811
  return code;
1,328,459✔
2812
}
2813

2814
int32_t vtbScanOpen(SOperatorInfo* pOperator) {
1,351,057✔
2815
  int32_t                    code = TSDB_CODE_SUCCESS;
1,351,057✔
2816
  int32_t                    line = 0;
1,351,057✔
2817
  int64_t                    st = 0;
1,351,057✔
2818
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
1,351,057✔
2819
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
1,351,057✔
2820

2821
  if (OPTR_IS_OPENED(pOperator)) {
1,351,057✔
2822
    return code;
1,000,001✔
2823
  }
2824

2825
  if (pOperator->cost.openCost == 0) {
351,056✔
2826
    st = taosGetTimestampUs();
274,284✔
2827
  }
2828

2829
  if (pVtbScan->isSuperTable) {
351,056✔
2830
    code = buildVirtualSuperTableScanChildTableMap(pOperator);
247,452✔
2831
    QUERY_CHECK_CODE(code, line, _return);
247,452✔
2832
  } else {
2833
    code = buildVirtualNormalChildTableScanChildTableMap(pOperator);
103,604✔
2834
    QUERY_CHECK_CODE(code, line, _return);
103,604✔
2835
  }
2836

2837
  OPTR_SET_OPENED(pOperator);
348,867✔
2838

2839
_return:
351,056✔
2840
  if (pOperator->cost.openCost == 0) {
351,056✔
2841
    pOperator->cost.openCost = (double)(taosGetTimestampUs() - st) / 1000.0;
274,284✔
2842
  }
2843
  if (code) {
351,056✔
2844
    qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
2,189✔
2845
    pOperator->pTaskInfo->code = code;
2,189✔
2846
    T_LONG_JMP(pOperator->pTaskInfo->env, code);
2,189✔
2847
  }
2848
  return code;
348,867✔
2849
}
2850

2851
int32_t vtbScanNext(SOperatorInfo* pOperator, SSDataBlock** pRes) {
2,870,621✔
2852
  int32_t                    code = TSDB_CODE_SUCCESS;
2,870,621✔
2853
  int32_t                    line = 0;
2,870,621✔
2854
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
2,870,621✔
2855
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
2,870,621✔
2856

2857
  QRY_PARAM_CHECK(pRes);
2,870,621✔
2858
  if (pOperator->status == OP_EXEC_DONE && !pOperator->pOperatorGetParam) {
2,870,621✔
2859
    return code;
×
2860
  }
2861
  if (pOperator->pOperatorGetParam) {
2,870,621✔
2862
    if (pOperator->status == OP_EXEC_DONE) {
×
2863
      pOperator->status = OP_OPENED;
×
2864
    }
2865
    pVtbScan->curTableIdx = 0;
×
2866
    pVtbScan->lastTableIdx = -1;
×
2867
    pVtbScan->window = ((SDynQueryCtrlOperatorParam *)(pOperator->pOperatorGetParam)->value)->window;
×
2868
    pOperator->pOperatorGetParam = NULL;
×
2869
  } else {
2870
    pVtbScan->window.skey = INT64_MAX;
2,870,621✔
2871
    pVtbScan->window.ekey = INT64_MIN;
2,870,621✔
2872
  }
2873

2874
  if (pVtbScan->needRedeploy) {
2,870,621✔
2875
    code = virtualTableScanCheckNeedRedeploy(pOperator);
1,521,753✔
2876
    QUERY_CHECK_CODE(code, line, _return);
1,521,753✔
2877
  }
2878

2879
  code = pOperator->fpSet._openFn(pOperator);
1,351,057✔
2880
  QUERY_CHECK_CODE(code, line, _return);
1,348,868✔
2881

2882
  if (pVtbScan->isSuperTable && taosArrayGetSize(pVtbScan->childTableList) == 0) {
1,348,868✔
2883
    setOperatorCompleted(pOperator);
19,800✔
2884
    return code;
19,800✔
2885
  }
2886

2887
  code = virtualTableScanGetNext(pOperator, pRes);
1,329,068✔
2888
  QUERY_CHECK_CODE(code, line, _return);
1,328,459✔
2889

2890
  return code;
1,328,459✔
2891

2892
_return:
1,519,564✔
2893
  if (code) {
1,519,564✔
2894
    qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
1,519,564✔
2895
    pOperator->pTaskInfo->code = code;
1,519,564✔
2896
    T_LONG_JMP(pOperator->pTaskInfo->env, code);
1,519,564✔
2897
  }
2898
  return code;
×
2899
}
2900

2901
int32_t initSeqStbJoinTableHash(SStbJoinPrevJoinCtx* pPrev, bool batchFetch) {
1,111,761✔
2902
  if (batchFetch) {
1,111,761✔
2903
    pPrev->leftHash = tSimpleHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
1,110,702✔
2904
    if (NULL == pPrev->leftHash) {
1,110,702✔
2905
      return terrno;
×
2906
    }
2907
    pPrev->rightHash = tSimpleHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
1,110,702✔
2908
    if (NULL == pPrev->rightHash) {
1,110,702✔
2909
      return terrno;
×
2910
    }
2911
  } else {
2912
    pPrev->leftCache = tSimpleHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT));
1,059✔
2913
    if (NULL == pPrev->leftCache) {
1,059✔
2914
      return terrno;
×
2915
    }
2916
    pPrev->rightCache = tSimpleHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT));
1,059✔
2917
    if (NULL == pPrev->rightCache) {
1,059✔
2918
      return terrno;
×
2919
    }
2920
    pPrev->onceTable = tSimpleHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT));
1,059✔
2921
    if (NULL == pPrev->onceTable) {
1,059✔
2922
      return terrno;
×
2923
    }
2924
  }
2925

2926
  return TSDB_CODE_SUCCESS;
1,111,761✔
2927
}
2928

2929
static void updateDynTbUidIfNeeded(SVtbScanDynCtrlInfo* pVtbScan, SStreamRuntimeInfo* pStreamRuntimeInfo) {
×
2930
  if (pStreamRuntimeInfo == NULL) {
×
2931
    return;
×
2932
  }
2933

2934
  SArray* vals = pStreamRuntimeInfo->funcInfo.pStreamPartColVals;
×
2935
  for (int32_t i = 0; i < taosArrayGetSize(vals); ++i) {
×
2936
    SStreamGroupValue* pValue = taosArrayGet(vals, i);
×
2937
    if (pValue != NULL && pValue->isTbname && pValue->uid != pVtbScan->dynTbUid) {
×
2938
      qTrace("dynQueryCtrl dyn tb uid:%" PRIu64 " reset to:%" PRIu64, pVtbScan->dynTbUid, pValue->uid);
×
2939

2940
      pVtbScan->dynTbUid = pValue->uid;
×
2941
      break;
×
2942
    }
2943
  }
2944
}
2945

2946
static int32_t initVtbScanInfo(SDynQueryCtrlOperatorInfo* pInfo, SMsgCb* pMsgCb,
274,284✔
2947
                               SDynQueryCtrlPhysiNode* pPhyciNode, SExecTaskInfo* pTaskInfo) {
2948
  int32_t      code = TSDB_CODE_SUCCESS;
274,284✔
2949
  int32_t      line = 0;
274,284✔
2950

2951
  code = tsem_init(&pInfo->vtbScan.ready, 0, 0);
274,284✔
2952
  QUERY_CHECK_CODE(code, line, _return);
274,284✔
2953

2954
  pInfo->vtbScan.genNewParam = true;
274,284✔
2955
  pInfo->vtbScan.batchProcessChild = pPhyciNode->vtbScan.batchProcessChild;
274,284✔
2956
  pInfo->vtbScan.hasPartition = pPhyciNode->vtbScan.hasPartition;
274,284✔
2957
  pInfo->vtbScan.scanAllCols = pPhyciNode->vtbScan.scanAllCols;
274,284✔
2958
  pInfo->vtbScan.isSuperTable = pPhyciNode->vtbScan.isSuperTable;
274,284✔
2959
  pInfo->vtbScan.rversion = pPhyciNode->vtbScan.rversion;
274,284✔
2960
  pInfo->vtbScan.uid = pPhyciNode->vtbScan.uid;
274,284✔
2961
  pInfo->vtbScan.suid = pPhyciNode->vtbScan.suid;
274,284✔
2962
  pInfo->vtbScan.epSet = pPhyciNode->vtbScan.mgmtEpSet;
274,284✔
2963
  pInfo->vtbScan.acctId = pPhyciNode->vtbScan.accountId;
274,284✔
2964
  pInfo->vtbScan.needRedeploy = false;
274,284✔
2965
  pInfo->vtbScan.pMsgCb = pMsgCb;
274,284✔
2966
  pInfo->vtbScan.curTableIdx = 0;
274,284✔
2967
  pInfo->vtbScan.lastTableIdx = -1;
274,284✔
2968
  pInfo->vtbScan.dynTbUid = 0;
274,284✔
2969
  pInfo->vtbScan.dbName = taosStrdup(pPhyciNode->vtbScan.dbName);
274,284✔
2970
  pInfo->vtbScan.tbName = taosStrdup(pPhyciNode->vtbScan.tbName);
274,284✔
2971
  QUERY_CHECK_NULL(pInfo->vtbScan.dbName, code, line, _return, terrno)
274,284✔
2972
  QUERY_CHECK_NULL(pInfo->vtbScan.tbName, code, line, _return, terrno)
274,284✔
2973
  pInfo->vtbScan.existOrgTbVg = taosHashInit(1, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_ENTRY_LOCK);
274,284✔
2974
  QUERY_CHECK_NULL(pInfo->vtbScan.existOrgTbVg, code, line, _return, terrno)
274,284✔
2975
  for (int32_t i = 0; i < LIST_LENGTH(pPhyciNode->vtbScan.pOrgVgIds); ++i) {
790,830✔
2976
    SValueNode* valueNode = (SValueNode*)nodesListGetNode(pPhyciNode->vtbScan.pOrgVgIds, i);
516,546✔
2977
    int32_t vgId = (int32_t)valueNode->datum.i;
516,546✔
2978
    code = taosHashPut(pInfo->vtbScan.existOrgTbVg, &vgId, sizeof(vgId), NULL, 0);
516,546✔
2979
    QUERY_CHECK_CODE(code, line, _return);
516,546✔
2980
  }
2981

2982
  if (pPhyciNode->dynTbname && pTaskInfo) {
274,284✔
2983
    updateDynTbUidIfNeeded(&pInfo->vtbScan, pTaskInfo->pStreamRuntimeInfo);
×
2984
  }
2985

2986
  pInfo->vtbScan.readColList = taosArrayInit(LIST_LENGTH(pPhyciNode->vtbScan.pScanCols), sizeof(col_id_t));
274,284✔
2987
  QUERY_CHECK_NULL(pInfo->vtbScan.readColList, code, line, _return, terrno)
274,284✔
2988

2989
  for (int32_t i = 0; i < LIST_LENGTH(pPhyciNode->vtbScan.pScanCols); ++i) {
2,996,194✔
2990
    SColumnNode* pNode = (SColumnNode*)nodesListGetNode(pPhyciNode->vtbScan.pScanCols, i);
2,721,910✔
2991
    QUERY_CHECK_NULL(pNode, code, line, _return, terrno)
2,721,910✔
2992
    QUERY_CHECK_NULL(taosArrayPush(pInfo->vtbScan.readColList, &pNode->colId), code, line, _return, terrno)
5,443,820✔
2993
  }
2994

2995
  pInfo->vtbScan.childTableList = taosArrayInit(10, POINTER_BYTES);
274,284✔
2996
  QUERY_CHECK_NULL(pInfo->vtbScan.childTableList, code, line, _return, terrno)
274,284✔
2997

2998
  pInfo->vtbScan.dbVgInfoMap = taosHashInit(taosArrayGetSize(pInfo->vtbScan.childTableList), taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
274,284✔
2999
  QUERY_CHECK_NULL(pInfo->vtbScan.dbVgInfoMap, code, line, _return, terrno)
274,284✔
3000

3001
  pInfo->vtbScan.otbNameToOtbInfoMap = NULL;
274,284✔
3002
  pInfo->vtbScan.otbVgIdToOtbInfoArrayMap = NULL;
274,284✔
3003
  pInfo->vtbScan.vtbUidToVgIdMapMap = NULL;
274,284✔
3004
  pInfo->vtbScan.vtbGroupIdToVgIdMapMap = NULL;
274,284✔
3005
  pInfo->vtbScan.vtbUidTagListMap = NULL;
274,284✔
3006
  pInfo->vtbScan.vtbGroupIdTagListMap = NULL;
274,284✔
3007
  pInfo->vtbScan.vtbUidToGroupIdMap = NULL;
274,284✔
3008

3009
  return code;
274,284✔
3010
_return:
×
3011
  // no need to destroy array and hashmap allocated in this function,
3012
  // since the operator's destroy function will take care of it
3013
  qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
×
3014
  return code;
×
3015
}
3016

3017
static int32_t initVtbWindowInfo(SDynQueryCtrlOperatorInfo* pInfo, SDynQueryCtrlPhysiNode* pPhyciNode,
×
3018
                                 SExecTaskInfo* pTaskInfo, SOperatorInfo* pOperator) {
3019
  int32_t              code = TSDB_CODE_SUCCESS;
×
3020
  int32_t              line = 0;
×
3021
  SDataBlockDescNode*  pDescNode = pPhyciNode->node.pOutputDataBlockDesc;
×
3022

3023
  pInfo->vtbWindow.wstartSlotId = pPhyciNode->vtbWindow.wstartSlotId;
×
3024
  pInfo->vtbWindow.wendSlotId = pPhyciNode->vtbWindow.wendSlotId;
×
3025
  pInfo->vtbWindow.wdurationSlotId = pPhyciNode->vtbWindow.wdurationSlotId;
×
3026
  pInfo->vtbWindow.pTargets = pPhyciNode->vtbWindow.pTargets;
×
3027
  pInfo->vtbWindow.isVstb = pPhyciNode->vtbWindow.isVstb;
×
3028
  pInfo->vtbWindow.singleWinMode = pPhyciNode->vtbWindow.singleWinMode;
×
3029
  pInfo->vtbWindow.extendOption = pPhyciNode->vtbWindow.extendOption;
×
3030

3031
  pInfo->vtbWindow.pRes = createDataBlockFromDescNode(pDescNode);
×
3032
  QUERY_CHECK_NULL(pInfo->vtbWindow.pRes, code, line, _return, terrno)
×
3033

3034
  pInfo->vtbWindow.pWins = taosArrayInit(TARRAY_MIN_SIZE, POINTER_BYTES);
×
3035
  QUERY_CHECK_NULL(pInfo->vtbWindow.pWins, code, line, _return, terrno)
×
3036

3037
  pInfo->vtbWindow.outputWstartSlotId = -1;
×
3038
  pInfo->vtbWindow.outputWendSlotId = -1;
×
3039
  pInfo->vtbWindow.outputWdurationSlotId = -1;
×
3040
  pInfo->vtbWindow.curWinBatchIdx = 0;
×
3041

3042
  initResultSizeInfo(&pOperator->resultInfo, 1);
×
3043
  code = blockDataEnsureCapacity(pInfo->vtbWindow.pRes, pOperator->resultInfo.capacity);
×
3044
  QUERY_CHECK_CODE(code, line, _return);
×
3045

3046
  return code;
×
3047
_return:
×
3048
  qError("%s failed at line %d since %s", __func__, line, tstrerror(code));
×
3049
  return code;
×
3050
}
3051

3052
static int32_t extractTsCol(SSDataBlock* pBlock, int32_t slotId, TSKEY** ppTsCols) {
×
3053
  int32_t code = TSDB_CODE_SUCCESS;
×
3054
  int32_t lino = 0;
×
3055

3056
  if (pBlock->pDataBlock != NULL && pBlock->info.dataLoad) {
×
3057
    SColumnInfoData* pColDataInfo = taosArrayGet(pBlock->pDataBlock, slotId);
×
3058
    QUERY_CHECK_NULL(pColDataInfo, code, lino, _return, terrno)
×
3059

3060
    *ppTsCols = (int64_t*)pColDataInfo->pData;
×
3061

3062
    if ((*ppTsCols)[0] != 0 && (pBlock->info.window.skey == 0 && pBlock->info.window.ekey == 0)) {
×
3063
      code = blockDataUpdateTsWindow(pBlock, slotId);
×
3064
      QUERY_CHECK_CODE(code, lino, _return);
×
3065
    }
3066
  }
3067

3068
  return code;
×
3069
_return:
×
3070
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3071
  return code;
×
3072
}
3073

3074
int32_t vtbWindowOpen(SOperatorInfo* pOperator) {
×
3075
  int32_t                    code = TSDB_CODE_SUCCESS;
×
3076
  int32_t                    lino = 0;
×
3077
  SDynQueryCtrlOperatorInfo* pDynInfo = pOperator->info;
×
3078
  SExecTaskInfo*             pTaskInfo = pOperator->pTaskInfo;
×
3079
  SVtbWindowDynCtrlInfo*     pInfo = &pDynInfo->vtbWindow;
×
3080
  int64_t                    st = 0;
×
3081

3082
  if (OPTR_IS_OPENED(pOperator)) {
×
3083
    return code;
×
3084
  }
3085

3086
  if (pOperator->cost.openCost == 0) {
×
3087
    st = taosGetTimestampUs();
×
3088
  }
3089

3090
  while (1) {
×
3091
    SSDataBlock* pBlock = getNextBlockFromDownstream(pOperator, 0);
×
3092
    if (pBlock == NULL) {
×
3093
      break;
×
3094
    }
3095

3096
    if (pInfo->outputWendSlotId == -1 && pInfo->outputWstartSlotId == -1 && pInfo->outputWdurationSlotId == -1) {
×
3097
      for (int32_t i = 0; i < LIST_LENGTH(pInfo->pTargets); ++i) {
×
3098
        STargetNode* pNode = (STargetNode*)nodesListGetNode(pInfo->pTargets, i);
×
3099
        if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pExpr)->dataBlockId == pBlock->info.id.blockId) {
×
3100
          if (((SColumnNode*)pNode->pExpr)->slotId == pDynInfo->vtbWindow.wstartSlotId) {
×
3101
            pInfo->outputWstartSlotId = i;
×
3102
          } else if (((SColumnNode*)pNode->pExpr)->slotId == pDynInfo->vtbWindow.wendSlotId) {
×
3103
            pInfo->outputWendSlotId = i;
×
3104
          } else if (((SColumnNode*)pNode->pExpr)->slotId == pDynInfo->vtbWindow.wdurationSlotId) {
×
3105
            pInfo->outputWdurationSlotId = i;
×
3106
          }
3107
        }
3108
      }
3109
    }
3110

3111
    TSKEY* wstartCol = NULL;
×
3112
    TSKEY* wendCol = NULL;
×
3113

3114
    code = extractTsCol(pBlock, pDynInfo->vtbWindow.wstartSlotId, &wstartCol);
×
3115
    QUERY_CHECK_CODE(code, lino, _return);
×
3116
    code = extractTsCol(pBlock, pDynInfo->vtbWindow.wendSlotId, &wendCol);
×
3117
    QUERY_CHECK_CODE(code, lino, _return);
×
3118

3119
    if (pDynInfo->vtbWindow.singleWinMode) {
×
3120
      for (int32_t i = 0; i < pBlock->info.rows; i++) {
×
3121
        SArray* pWin = taosArrayInit(pBlock->info.rows, sizeof(SExtWinTimeWindow));
×
3122
        QUERY_CHECK_NULL(pWin, code, lino, _return, terrno)
×
3123

3124
        QUERY_CHECK_NULL(taosArrayReserve(pWin, 1), code, lino, _return, terrno);
×
3125

3126
        SExtWinTimeWindow* pWindow = taosArrayGet(pWin, 0);
×
3127
        QUERY_CHECK_NULL(pWindow, code, lino, _return, terrno)
×
3128
        pWindow->tw.skey = wstartCol[i];
×
3129
        pWindow->tw.ekey = wendCol[i] + 1;
×
3130
        pWindow->winOutIdx = -1;
×
3131

3132
        QUERY_CHECK_NULL(taosArrayPush(pDynInfo->vtbWindow.pWins, &pWin), code, lino, _return, terrno);
×
3133
      }
3134
    } else {
3135
      SArray* pWin = taosArrayInit(pBlock->info.rows, sizeof(SExtWinTimeWindow));
×
3136
      QUERY_CHECK_NULL(pWin, code, lino, _return, terrno)
×
3137

3138
      QUERY_CHECK_NULL(taosArrayReserve(pWin, pBlock->info.rows), code, lino, _return, terrno);
×
3139

3140
      for (int32_t i = 0; i < pBlock->info.rows; i++) {
×
3141
        SExtWinTimeWindow* pWindow = taosArrayGet(pWin, i);
×
3142
        QUERY_CHECK_NULL(pWindow, code, lino, _return, terrno)
×
3143
        pWindow->tw.skey = wstartCol[i];
×
3144
        pWindow->tw.ekey = wendCol[i] + 1;
×
3145
        pWindow->winOutIdx = -1;
×
3146
      }
3147

3148
      QUERY_CHECK_NULL(taosArrayPush(pDynInfo->vtbWindow.pWins, &pWin), code, lino, _return, terrno);
×
3149
    }
3150
  }
3151

3152
  // handle first window's start key and last window's end key
3153
  SArray* firstBatch = (SArray*)taosArrayGetP(pDynInfo->vtbWindow.pWins, 0);
×
3154
  SArray* lastBatch = (SArray*)taosArrayGetP(pDynInfo->vtbWindow.pWins, taosArrayGetSize(pDynInfo->vtbWindow.pWins) - 1);
×
3155

3156
  QUERY_CHECK_NULL(firstBatch, code, lino, _return, terrno)
×
3157
  QUERY_CHECK_NULL(lastBatch, code, lino, _return, terrno)
×
3158

3159
  SExtWinTimeWindow* firstWin = (SExtWinTimeWindow*)taosArrayGet(firstBatch, 0);
×
3160
  SExtWinTimeWindow* lastWin = (SExtWinTimeWindow*)taosArrayGetLast(lastBatch);
×
3161

3162
  QUERY_CHECK_NULL(firstWin, code, lino, _return, terrno)
×
3163
  QUERY_CHECK_NULL(lastWin, code, lino, _return, terrno)
×
3164

3165
  if (pInfo->extendOption == STATE_WIN_EXTEND_OPTION_BACKWARD) {
×
3166
    lastWin->tw.ekey = INT64_MAX;
×
3167
  }
3168
  if (pInfo->extendOption == STATE_WIN_EXTEND_OPTION_FORWARD) {
×
3169
    firstWin->tw.skey = INT64_MIN;
×
3170
  }
3171

3172
  OPTR_SET_OPENED(pOperator);
×
3173

3174
  if (pOperator->cost.openCost == 0) {
×
3175
    pOperator->cost.openCost = (double)(taosGetTimestampUs() - st) / 1000.0;
×
3176
  }
3177

3178
_return:
×
3179
  if (code != TSDB_CODE_SUCCESS) {
×
3180
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3181
    pTaskInfo->code = code;
×
3182
    T_LONG_JMP(pTaskInfo->env, code);
×
3183
  }
3184
  return code;
×
3185
}
3186

3187
static int32_t buildDynQueryCtrlOperatorParamForExternalWindow(SOperatorParam** ppRes, int32_t downstreamIdx, int64_t skey, int64_t ekey) {
×
3188
  int32_t                     code = TSDB_CODE_SUCCESS;
×
3189
  int32_t                     lino = 0;
×
3190
  SDynQueryCtrlOperatorParam* pDyn = NULL;
×
3191

3192
  *ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
×
3193
  QUERY_CHECK_NULL(*ppRes, code, lino, _return, terrno)
×
3194

3195
  (*ppRes)->pChildren = taosArrayInit(1, POINTER_BYTES);
×
3196
  QUERY_CHECK_NULL((*ppRes)->pChildren, code, lino, _return, terrno)
×
3197

3198
  pDyn = taosMemoryMalloc(sizeof(SDynQueryCtrlOperatorParam));
×
3199
  QUERY_CHECK_NULL(pDyn, code, lino, _return, terrno);
×
3200

3201
  pDyn->window.skey = skey;
×
3202
  pDyn->window.ekey = ekey;
×
3203

3204
  (*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_DYN_QUERY_CTRL;
×
3205
  (*ppRes)->downstreamIdx = 0;
×
3206
  (*ppRes)->reUse = false;
×
3207
  (*ppRes)->value = pDyn;
×
3208

3209
  return code;
×
3210
_return:
×
3211
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3212
  if (pDyn) {
×
3213
    taosMemoryFree(pDyn);
×
3214
  }
3215
  if (*ppRes) {
×
3216
    if ((*ppRes)->pChildren) {
×
3217
      taosArrayDestroy((*ppRes)->pChildren);
×
3218
    }
3219
    taosMemoryFree(*ppRes);
×
3220
    *ppRes = NULL;
×
3221
  }
3222
  return code;
×
3223
}
3224

3225
static int32_t buildExternalWindowOperatorParamEx(SDynQueryCtrlOperatorInfo* pInfo, SOperatorParam** ppRes, SArray* pWins, int32_t idx) {
×
3226
  int32_t                       code = TSDB_CODE_SUCCESS;
×
3227
  int32_t                       lino = 0;
×
3228
  SExternalWindowOperatorParam* pExtWinOp = NULL;
×
3229

3230
  *ppRes = taosMemoryMalloc(sizeof(SOperatorParam));
×
3231
  QUERY_CHECK_NULL(*ppRes, code, lino, _return, terrno)
×
3232

3233
  pExtWinOp = taosMemoryMalloc(sizeof(SExternalWindowOperatorParam));
×
3234
  QUERY_CHECK_NULL(pExtWinOp, code, lino, _return, terrno)
×
3235

3236
  pExtWinOp->ExtWins = taosArrayDup(pWins, NULL);
×
3237
  QUERY_CHECK_NULL(pExtWinOp->ExtWins, code, lino, _return, terrno)
×
3238

3239
  SExtWinTimeWindow *firstWin = (SExtWinTimeWindow *)taosArrayGet(pWins, 0);
×
3240
  SExtWinTimeWindow *lastWin = (SExtWinTimeWindow *)taosArrayGet(pWins, taosArrayGetSize(pWins) - 1);
×
3241

3242
  (*ppRes)->pChildren = taosArrayInit(1, POINTER_BYTES);
×
3243
  QUERY_CHECK_NULL((*ppRes)->pChildren, code, lino, _return, terrno)
×
3244

3245
  SOperatorParam* pDynQueryCtrlParam = NULL;
×
3246
  code = buildDynQueryCtrlOperatorParamForExternalWindow(&pDynQueryCtrlParam, 0, firstWin->tw.skey, lastWin->tw.ekey);
×
3247
  QUERY_CHECK_CODE(code, lino, _return);
×
3248
  QUERY_CHECK_NULL(taosArrayPush((*ppRes)->pChildren, &pDynQueryCtrlParam), code, lino, _return, terrno)
×
3249

3250
  (*ppRes)->opType = QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW;
×
3251
  (*ppRes)->downstreamIdx = idx;
×
3252
  (*ppRes)->value = pExtWinOp;
×
3253
  (*ppRes)->reUse = false;
×
3254

3255
  return code;
×
3256
_return:
×
3257
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3258
  if (pExtWinOp) {
×
3259
    if (pExtWinOp->ExtWins) {
×
3260
      taosArrayDestroy(pExtWinOp->ExtWins);
×
3261
    }
3262
    taosMemoryFree(pExtWinOp);
×
3263
  }
3264
  if (*ppRes) {
×
3265
    if ((*ppRes)->pChildren) {
×
3266
      for (int32_t i = 0; i < taosArrayGetSize((*ppRes)->pChildren); i++) {
×
3267
        SOperatorParam* pChildParam = (SOperatorParam*)taosArrayGet((*ppRes)->pChildren, i);
×
3268
        if (pChildParam) {
×
3269
          SDynQueryCtrlOperatorParam* pDynParam = (SDynQueryCtrlOperatorParam*)pChildParam->value;
×
3270
          if (pDynParam) {
×
3271
            taosMemoryFree(pDynParam);
×
3272
          }
3273
          taosMemoryFree(pChildParam);
×
3274
        }
3275
      }
3276
      taosArrayDestroy((*ppRes)->pChildren);
×
3277
    }
3278
    taosMemoryFree(*ppRes);
×
3279
    *ppRes = NULL;
×
3280
  }
3281
  return code;
×
3282
}
3283

3284
int32_t vtbWindowNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
×
3285
  int32_t                    code = TSDB_CODE_SUCCESS;
×
3286
  int32_t                    lino = 0;
×
3287
  SDynQueryCtrlOperatorInfo* pDynInfo = pOperator->info;
×
3288
  SExecTaskInfo*             pTaskInfo = pOperator->pTaskInfo;
×
3289
  int64_t                    st = taosGetTimestampUs();
×
3290
  int32_t                    numOfWins = 0;
×
3291
  SOperatorInfo*             mergeOp = NULL;
×
3292
  SOperatorInfo*             extWinOp = NULL;
×
3293
  SOperatorParam*            pMergeParam = NULL;
×
3294
  SOperatorParam*            pExtWinParam = NULL;
×
3295
  SVtbWindowDynCtrlInfo*     pInfo = &pDynInfo->vtbWindow;
×
3296
  SSDataBlock*               pRes = pInfo->pRes;
×
3297

3298
  code = pOperator->fpSet._openFn(pOperator);
×
3299
  QUERY_CHECK_CODE(code, lino, _return);
×
3300

3301
  if (pInfo->curWinBatchIdx >= taosArrayGetSize(pInfo->pWins)) {
×
3302
    *ppRes = NULL;
×
3303
    return code;
×
3304
  }
3305

3306
  SArray* pWinArray = (SArray*)taosArrayGetP(pInfo->pWins, pInfo->curWinBatchIdx);
×
3307
  QUERY_CHECK_NULL(pWinArray, code, lino, _return, terrno)
×
3308

3309
  numOfWins = (int32_t)taosArrayGetSize(pWinArray);
×
3310

3311
  if (pInfo->isVstb) {
×
3312
    extWinOp = pOperator->pDownstream[1];
×
3313
    code = buildExternalWindowOperatorParamEx(pDynInfo, &pExtWinParam, pWinArray, extWinOp->numOfDownstream);
×
3314
    QUERY_CHECK_CODE(code, lino, _return);
×
3315

3316
    SSDataBlock* pExtWinBlock = NULL;
×
3317
    code = extWinOp->fpSet.getNextExtFn(extWinOp, pExtWinParam, &pExtWinBlock);
×
3318
    QUERY_CHECK_CODE(code, lino, _return);
×
3319

3320
    blockDataCleanup(pRes);
×
3321
    code = blockDataEnsureCapacity(pRes, numOfWins);
×
3322
    QUERY_CHECK_CODE(code, lino, _return);
×
3323

3324
    if (pExtWinBlock) {
×
3325
      code = copyColumnsValue(pInfo->pTargets, pExtWinBlock->info.id.blockId, pRes, pExtWinBlock, numOfWins);
×
3326
      QUERY_CHECK_CODE(code, lino, _return);
×
3327

3328
      if (pInfo->curWinBatchIdx == 0) {
×
3329
        // first batch, get _wstart from pMergedBlock
3330
        SExtWinTimeWindow* firstWin = (SExtWinTimeWindow*)taosArrayGet(taosArrayGetP(pInfo->pWins, 0), 0);
×
3331
        QUERY_CHECK_NULL(firstWin, code, lino, _return, terrno)
×
3332

3333
        firstWin->tw.skey = pExtWinBlock->info.window.skey;
×
3334
      }
3335
      if (pInfo->curWinBatchIdx == taosArrayGetSize(pInfo->pWins) - 1) {
×
3336
        // last batch, get _wend from pMergedBlock
3337
        SExtWinTimeWindow* lastWin = (SExtWinTimeWindow*)taosArrayGetLast(taosArrayGetP(pInfo->pWins, taosArrayGetSize(pInfo->pWins) - 1));
×
3338
        QUERY_CHECK_NULL(lastWin, code, lino, _return, terrno)
×
3339

3340
        lastWin->tw.ekey = pExtWinBlock->info.window.ekey + 1;
×
3341
      }
3342
    }
3343
  } else {
3344
    mergeOp = pOperator->pDownstream[1];
×
3345
    code = buildMergeOperatorParam(pDynInfo, &pMergeParam, pWinArray, mergeOp->numOfDownstream, numOfWins);
×
3346
    QUERY_CHECK_CODE(code, lino, _return);
×
3347

3348
    SSDataBlock* pMergedBlock = NULL;
×
3349
    code = mergeOp->fpSet.getNextExtFn(mergeOp, pMergeParam, &pMergedBlock);
×
3350
    QUERY_CHECK_CODE(code, lino, _return);
×
3351

3352
    blockDataCleanup(pRes);
×
3353
    code = blockDataEnsureCapacity(pRes, numOfWins);
×
3354
    QUERY_CHECK_CODE(code, lino, _return);
×
3355

3356
    if (pMergedBlock) {
×
3357
      code = copyColumnsValue(pInfo->pTargets, pMergedBlock->info.id.blockId, pRes, pMergedBlock, numOfWins);
×
3358
      QUERY_CHECK_CODE(code, lino, _return);
×
3359

3360
      if (pInfo->curWinBatchIdx == 0) {
×
3361
        // first batch, get _wstart from pMergedBlock
3362
        SExtWinTimeWindow* firstWin = (SExtWinTimeWindow*)taosArrayGet(taosArrayGetP(pInfo->pWins, 0), 0);
×
3363
        QUERY_CHECK_NULL(firstWin, code, lino, _return, terrno)
×
3364

3365
        firstWin->tw.skey = pMergedBlock->info.window.skey;
×
3366
      }
3367
      if (pInfo->curWinBatchIdx == taosArrayGetSize(pInfo->pWins) - 1) {
×
3368
        // last batch, get _wend from pMergedBlock
3369
        SExtWinTimeWindow* lastWin = (SExtWinTimeWindow*)taosArrayGetLast(taosArrayGetP(pInfo->pWins, taosArrayGetSize(pInfo->pWins) - 1));
×
3370
        QUERY_CHECK_NULL(lastWin, code, lino, _return, terrno)
×
3371

3372
        lastWin->tw.ekey = pMergedBlock->info.window.ekey + 1;
×
3373
      }
3374
    }
3375
  }
3376

3377

3378
  if (pInfo->outputWstartSlotId != -1) {
×
3379
    SColumnInfoData* pWstartCol = taosArrayGet(pRes->pDataBlock, pInfo->outputWstartSlotId);
×
3380
    QUERY_CHECK_NULL(pWstartCol, code, lino, _return, terrno)
×
3381

3382
    for (int32_t i = 0; i < numOfWins; i++) {
×
3383
      SExtWinTimeWindow* pWindow = (SExtWinTimeWindow*)taosArrayGet(pWinArray, i);
×
3384
      QUERY_CHECK_NULL(pWindow, code, lino, _return, terrno)
×
3385
      code = colDataSetVal(pWstartCol, i, (const char*)&pWindow->tw.skey, false);
×
3386
      QUERY_CHECK_CODE(code, lino, _return);
×
3387
    }
3388
  }
3389
  if (pInfo->outputWendSlotId != -1) {
×
3390
    SColumnInfoData* pWendCol = taosArrayGet(pRes->pDataBlock, pInfo->outputWendSlotId);
×
3391
    QUERY_CHECK_NULL(pWendCol, code, lino, _return, terrno)
×
3392

3393
    for (int32_t i = 0; i < numOfWins; i++) {
×
3394
      SExtWinTimeWindow* pWindow = (SExtWinTimeWindow*)taosArrayGet(pWinArray, i);
×
3395
      QUERY_CHECK_NULL(pWindow, code, lino, _return, terrno)
×
3396
      TSKEY ekey = pWindow->tw.ekey - 1;
×
3397
      code = colDataSetVal(pWendCol, i, (const char*)&ekey, false);
×
3398
      QUERY_CHECK_CODE(code, lino, _return);
×
3399
    }
3400
  }
3401
  if (pInfo->outputWdurationSlotId != -1) {
×
3402
    SColumnInfoData* pWdurationCol = taosArrayGet(pRes->pDataBlock, pInfo->outputWdurationSlotId);
×
3403
    QUERY_CHECK_NULL(pWdurationCol, code, lino, _return, terrno)
×
3404

3405
    for (int32_t i = 0; i < numOfWins; i++) {
×
3406
      SExtWinTimeWindow* pWindow = (SExtWinTimeWindow*)taosArrayGet(pWinArray, i);
×
3407
      QUERY_CHECK_NULL(pWindow, code, lino, _return, terrno)
×
3408
      int64_t duration = pWindow->tw.ekey - 1 - pWindow->tw.skey;
×
3409
      code = colDataSetVal(pWdurationCol, i, (const char*)&duration, false);
×
3410
      QUERY_CHECK_CODE(code, lino, _return);
×
3411
    }
3412
  }
3413

3414
  pRes->info.rows = numOfWins;
×
3415
  *ppRes = pRes;
×
3416
  pInfo->curWinBatchIdx++;
×
3417

3418
  return code;
×
3419

3420
_return:
×
3421
  if (code != TSDB_CODE_SUCCESS) {
×
3422
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3423
    pTaskInfo->code = code;
×
3424
    T_LONG_JMP(pTaskInfo->env, code);
×
3425
  }
3426
  return code;
×
3427
}
3428

3429
static int32_t resetDynQueryCtrlOperState(SOperatorInfo* pOper) {
1,701,850✔
3430
  SDynQueryCtrlOperatorInfo*    pDyn = pOper->info;
1,701,850✔
3431
  SDynQueryCtrlPhysiNode const* pPhyciNode = pOper->pPhyNode;
1,701,850✔
3432
  SExecTaskInfo*                pTaskInfo = pOper->pTaskInfo;
1,702,049✔
3433

3434
  pOper->status = OP_NOT_OPENED;
1,701,850✔
3435

3436
  switch (pDyn->qType) {
1,701,850✔
3437
    case DYN_QTYPE_STB_HASH:{
708✔
3438
      pDyn->stbJoin.execInfo = (SDynQueryCtrlExecInfo){0};
708✔
3439
      SStbJoinDynCtrlInfo* pStbJoin = &pDyn->stbJoin;
708✔
3440
      destroyStbJoinDynCtrlInfo(&pDyn->stbJoin);
708✔
3441
      
3442
      int32_t code = initSeqStbJoinTableHash(&pDyn->stbJoin.ctx.prev, pDyn->stbJoin.basic.batchFetch);
708✔
3443
      if (TSDB_CODE_SUCCESS != code) {
708✔
3444
        qError("initSeqStbJoinTableHash failed since %s", tstrerror(code));
×
3445
        return code;
×
3446
      }
3447
      pStbJoin->ctx.prev.pListHead = NULL;
708✔
3448
      pStbJoin->ctx.prev.joinBuild = false;
708✔
3449
      pStbJoin->ctx.prev.pListTail = NULL;
708✔
3450
      pStbJoin->ctx.prev.tableNum = 0;
708✔
3451

3452
      pStbJoin->ctx.post = (SStbJoinPostJoinCtx){0};
708✔
3453
      break; 
708✔
3454
    }
3455
    case DYN_QTYPE_VTB_SCAN: {
1,701,341✔
3456
      SVtbScanDynCtrlInfo* pVtbScan = &pDyn->vtbScan;
1,701,341✔
3457
      
3458
      if (pVtbScan->otbNameToOtbInfoMap) {
1,701,341✔
3459
        taosHashSetFreeFp(pVtbScan->otbNameToOtbInfoMap, destroySOrgTbInfo);
×
3460
        taosHashCleanup(pVtbScan->otbNameToOtbInfoMap);
×
3461
        pVtbScan->otbNameToOtbInfoMap = NULL;
×
3462
      }
3463
      if (pVtbScan->pRsp) {
1,701,341✔
3464
        tFreeSUsedbRsp(pVtbScan->pRsp);
×
3465
        taosMemoryFreeClear(pVtbScan->pRsp);
×
3466
      }
3467
      if (pVtbScan->colRefInfo) {
1,701,341✔
3468
        taosArrayDestroyEx(pVtbScan->colRefInfo, destroyColRefInfo);
103,604✔
3469
        pVtbScan->colRefInfo = NULL;
103,604✔
3470
      }
3471
      if (pVtbScan->childTableMap) {
1,701,341✔
3472
        taosHashCleanup(pVtbScan->childTableMap);
5,190✔
3473
        pVtbScan->childTableMap = NULL;
5,190✔
3474
      }
3475
      if (pVtbScan->childTableList) {
1,701,142✔
3476
        taosArrayClearEx(pVtbScan->childTableList, destroyColRefArray);
1,701,142✔
3477
      }
3478
      if (pPhyciNode->dynTbname && pTaskInfo) {
1,700,943✔
3479
        updateDynTbUidIfNeeded(pVtbScan, pTaskInfo->pStreamRuntimeInfo);
×
3480
      }
3481
      pVtbScan->curTableIdx = 0;
1,700,943✔
3482
      pVtbScan->lastTableIdx = -1;
1,701,142✔
3483
      break;
1,701,341✔
3484
    }
3485
    case DYN_QTYPE_VTB_WINDOW: {
×
3486
      SVtbWindowDynCtrlInfo* pVtbWindow = &pDyn->vtbWindow;
×
3487
      if (pVtbWindow->pRes) {
×
3488
        blockDataDestroy(pVtbWindow->pRes);
×
3489
        pVtbWindow->pRes = NULL;
×
3490
      }
3491
      if (pVtbWindow->pWins) {
×
3492
        taosArrayDestroyEx(pVtbWindow->pWins, destroyWinArray);
×
3493
        pVtbWindow->pWins = NULL;
×
3494
      }
3495
      pVtbWindow->outputWdurationSlotId = -1;
×
3496
      pVtbWindow->outputWendSlotId = -1;
×
3497
      pVtbWindow->outputWstartSlotId = -1;
×
3498
      pVtbWindow->curWinBatchIdx = 0;
×
3499
      break;
×
3500
    }
3501
    default:
×
3502
      qError("unsupported dynamic query ctrl type: %d", pDyn->qType);
×
3503
      break;
×
3504
  }
3505
  return 0;
1,701,850✔
3506
}
3507

UNCOV
3508
int32_t vtbAggOpen(SOperatorInfo* pOperator) {
×
UNCOV
3509
  int32_t                    code = TSDB_CODE_SUCCESS;
×
UNCOV
3510
  int32_t                    line = 0;
×
UNCOV
3511
  int64_t                    st = 0;
×
UNCOV
3512
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
×
3513

UNCOV
3514
  if (OPTR_IS_OPENED(pOperator)) {
×
UNCOV
3515
    return code;
×
3516
  }
3517

UNCOV
3518
  if (pOperator->cost.openCost == 0) {
×
UNCOV
3519
    st = taosGetTimestampUs();
×
3520
  }
3521

UNCOV
3522
  code = buildVirtualSuperTableScanChildTableMap(pOperator);
×
UNCOV
3523
  QUERY_CHECK_CODE(code, line, _return);
×
UNCOV
3524
  OPTR_SET_OPENED(pOperator);
×
3525

UNCOV
3526
_return:
×
UNCOV
3527
  if (pOperator->cost.openCost == 0) {
×
UNCOV
3528
    pOperator->cost.openCost = (double)(taosGetTimestampUs() - st) / 1000.0;
×
3529
  }
UNCOV
3530
  if (code) {
×
3531
    qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
×
3532
    pOperator->pTaskInfo->code = code;
×
3533
    T_LONG_JMP(pOperator->pTaskInfo->env, code);
×
3534
  }
UNCOV
3535
  return code;
×
3536
}
3537

UNCOV
3538
int32_t virtualTableAggGetNext(SOperatorInfo* pOperator, SSDataBlock** pRes) {
×
UNCOV
3539
  int32_t                    code = TSDB_CODE_SUCCESS;
×
UNCOV
3540
  int32_t                    line = 0;
×
UNCOV
3541
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
×
UNCOV
3542
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
×
UNCOV
3543
  SOperatorInfo*             pAggOp = pOperator->pDownstream[pOperator->numOfDownstream - 1];
×
UNCOV
3544
  SOperatorInfo*             pTagScanOp = pOperator->pDownstream[1];
×
UNCOV
3545
  SOperatorParam*            pAggParam = NULL;
×
3546

UNCOV
3547
  if (pInfo->vtbScan.hasPartition) {
×
UNCOV
3548
    if (pInfo->vtbScan.batchProcessChild) {
×
UNCOV
3549
      void* pIter = taosHashIterate(pVtbScan->vtbGroupIdTagListMap, NULL);
×
UNCOV
3550
      while (pIter) {
×
UNCOV
3551
        size_t     keyLen = 0;
×
UNCOV
3552
        uint64_t   groupid = *(uint64_t*)taosHashGetKey(pIter, &keyLen);
×
3553

UNCOV
3554
        code = buildAggOperatorParamWithGroupId(pInfo, groupid, &pAggParam);
×
UNCOV
3555
        QUERY_CHECK_CODE(code, line, _return);
×
3556

UNCOV
3557
        if (pAggParam) {
×
UNCOV
3558
          code = pAggOp->fpSet.getNextExtFn(pAggOp, pAggParam, pRes);
×
UNCOV
3559
          QUERY_CHECK_CODE(code, line, _return);
×
3560
        } else {
UNCOV
3561
          *pRes = NULL;
×
3562
        }
3563

UNCOV
3564
        pIter = taosHashIterate(pVtbScan->vtbGroupIdTagListMap, pIter);
×
3565

UNCOV
3566
        if (*pRes) {
×
UNCOV
3567
          (*pRes)->info.id.groupId = groupid;
×
UNCOV
3568
          code = taosHashRemove(pVtbScan->vtbGroupIdTagListMap, &groupid, keyLen);
×
UNCOV
3569
          QUERY_CHECK_CODE(code, line, _return);
×
UNCOV
3570
          break;
×
3571
        }
3572
      }
3573
    } else {
UNCOV
3574
      void *pIter = taosHashIterate(pVtbScan->vtbGroupIdTagListMap, NULL);
×
UNCOV
3575
      while (pIter) {
×
UNCOV
3576
        size_t     keyLen = 0;
×
UNCOV
3577
        uint64_t*  groupid = (uint64_t*)taosHashGetKey(pIter, &keyLen);
×
UNCOV
3578
        SHashObj*  vtbUidTagListMap = *(SHashObj**)pIter;
×
3579

UNCOV
3580
        void* pIter2 = taosHashIterate(vtbUidTagListMap, NULL);
×
UNCOV
3581
        while (pIter2) {
×
UNCOV
3582
          size_t   keyLen2 = 0;
×
UNCOV
3583
          tb_uid_t uid = *(tb_uid_t*)taosHashGetKey(pIter2, &keyLen2);
×
UNCOV
3584
          SArray*  pTagList = *(SArray**)pIter2;
×
3585

UNCOV
3586
          if (pVtbScan->genNewParam) {
×
UNCOV
3587
            code = buildAggOperatorParamForSingleChild(pInfo, uid, *groupid, pTagList, &pAggParam);
×
UNCOV
3588
            QUERY_CHECK_CODE(code, line, _return);
×
UNCOV
3589
            if (pAggParam) {
×
UNCOV
3590
              code = pAggOp->fpSet.getNextExtFn(pAggOp, pAggParam, pRes);
×
UNCOV
3591
              QUERY_CHECK_CODE(code, line, _return);
×
3592
            } else {
UNCOV
3593
              *pRes = NULL;
×
3594
            }
3595
          } else {
UNCOV
3596
            code = pAggOp->fpSet.getNextFn(pAggOp, pRes);
×
UNCOV
3597
            QUERY_CHECK_CODE(code, line, _return);
×
3598
          }
3599

UNCOV
3600
          if (*pRes) {
×
UNCOV
3601
            pVtbScan->genNewParam = false;
×
UNCOV
3602
            (*pRes)->info.id.groupId = *groupid;
×
UNCOV
3603
            break;
×
3604
          }
UNCOV
3605
          pVtbScan->genNewParam = true;
×
UNCOV
3606
          pIter2 = taosHashIterate(vtbUidTagListMap, pIter2);
×
UNCOV
3607
          code = taosHashRemove(vtbUidTagListMap, &uid, keyLen);
×
UNCOV
3608
          QUERY_CHECK_CODE(code, line, _return);
×
3609
        }
UNCOV
3610
        if (*pRes) {
×
UNCOV
3611
          break;
×
3612
        }
UNCOV
3613
        pIter = taosHashIterate(pVtbScan->vtbGroupIdTagListMap, pIter);
×
UNCOV
3614
        code = taosHashRemove(pVtbScan->vtbGroupIdTagListMap, groupid, keyLen);
×
UNCOV
3615
        QUERY_CHECK_CODE(code, line, _return);
×
3616
      }
3617
    }
3618

3619
  } else {
UNCOV
3620
    if (pInfo->vtbScan.batchProcessChild) {
×
UNCOV
3621
      code = buildAggOperatorParam(pInfo, &pAggParam);
×
UNCOV
3622
      QUERY_CHECK_CODE(code, line, _return);
×
3623

UNCOV
3624
      code = pAggOp->fpSet.getNextExtFn(pAggOp, pAggParam, pRes);
×
UNCOV
3625
      QUERY_CHECK_CODE(code, line, _return);
×
UNCOV
3626
      setOperatorCompleted(pOperator);
×
3627
    } else {
UNCOV
3628
      void* pIter = taosHashIterate(pVtbScan->vtbUidTagListMap, NULL);
×
UNCOV
3629
      while (pIter) {
×
UNCOV
3630
        size_t   keyLen = 0;
×
UNCOV
3631
        tb_uid_t uid = *(tb_uid_t*)taosHashGetKey(pIter, &keyLen);
×
UNCOV
3632
        SArray*  pTagList = *(SArray**)pIter;
×
3633

UNCOV
3634
        if (pVtbScan->genNewParam) {
×
UNCOV
3635
          code = buildAggOperatorParamForSingleChild(pInfo, uid, 0, pTagList, &pAggParam);
×
UNCOV
3636
          QUERY_CHECK_CODE(code, line, _return);
×
3637

UNCOV
3638
          if (pAggParam) {
×
UNCOV
3639
            code = pAggOp->fpSet.getNextExtFn(pAggOp, pAggParam, pRes);
×
UNCOV
3640
            QUERY_CHECK_CODE(code, line, _return);
×
3641
          } else {
UNCOV
3642
            *pRes = NULL;
×
3643
          }
3644
        } else {
UNCOV
3645
          code = pAggOp->fpSet.getNextFn(pAggOp, pRes);
×
UNCOV
3646
          QUERY_CHECK_CODE(code, line, _return);
×
3647
        }
3648

UNCOV
3649
        if (*pRes) {
×
UNCOV
3650
          pVtbScan->genNewParam = false;
×
UNCOV
3651
          break;
×
3652
        }
UNCOV
3653
        pVtbScan->genNewParam = true;
×
UNCOV
3654
        pIter = taosHashIterate(pVtbScan->vtbUidTagListMap, pIter);
×
UNCOV
3655
        code = taosHashRemove(pVtbScan->vtbUidTagListMap, &uid, keyLen);
×
UNCOV
3656
        QUERY_CHECK_CODE(code, line, _return);
×
3657
      }
3658
    }
3659
  }
UNCOV
3660
_return:
×
UNCOV
3661
  if (code) {
×
3662
    qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
×
3663
  }
UNCOV
3664
  return code;
×
3665
}
3666

UNCOV
3667
int32_t vtbAggNext(SOperatorInfo* pOperator, SSDataBlock** pRes) {
×
UNCOV
3668
  int32_t                    code = TSDB_CODE_SUCCESS;
×
UNCOV
3669
  int32_t                    line = 0;
×
UNCOV
3670
  SDynQueryCtrlOperatorInfo* pInfo = pOperator->info;
×
UNCOV
3671
  SVtbScanDynCtrlInfo*       pVtbScan = (SVtbScanDynCtrlInfo*)&pInfo->vtbScan;
×
3672

UNCOV
3673
  QRY_PARAM_CHECK(pRes);
×
UNCOV
3674
  if (pOperator->status == OP_EXEC_DONE) {
×
UNCOV
3675
    return code;
×
3676
  }
3677

UNCOV
3678
  code = pOperator->fpSet._openFn(pOperator);
×
UNCOV
3679
  QUERY_CHECK_CODE(code, line, _return);
×
3680

UNCOV
3681
  if (pVtbScan->isSuperTable && taosArrayGetSize(pVtbScan->childTableList) == 0) {
×
3682
    setOperatorCompleted(pOperator);
×
3683
    return code;
×
3684
  }
3685

UNCOV
3686
  code = virtualTableAggGetNext(pOperator, pRes);
×
UNCOV
3687
  QUERY_CHECK_CODE(code, line, _return);
×
3688

UNCOV
3689
  return code;
×
3690

3691
_return:
×
3692
  if (code) {
×
3693
    qError("%s failed since %s, line %d", __func__, tstrerror(code), line);
×
3694
    pOperator->pTaskInfo->code = code;
×
3695
    T_LONG_JMP(pOperator->pTaskInfo->env, code);
×
3696
  }
3697
  return code;
×
3698
}
3699

3700
int32_t createDynQueryCtrlOperatorInfo(SOperatorInfo** pDownstream, int32_t numOfDownstream,
1,385,337✔
3701
                                       SDynQueryCtrlPhysiNode* pPhyciNode, SExecTaskInfo* pTaskInfo,
3702
                                       SMsgCb* pMsgCb, SOperatorInfo** pOptrInfo) {
3703
  QRY_PARAM_CHECK(pOptrInfo);
1,385,337✔
3704

3705
  int32_t                    code = TSDB_CODE_SUCCESS;
1,385,337✔
3706
  int32_t                    line = 0;
1,385,337✔
3707
  __optr_fn_t                nextFp = NULL;
1,385,337✔
3708
  __optr_open_fn_t           openFp = NULL;
1,385,337✔
3709
  SOperatorInfo*             pOperator = NULL;
1,385,337✔
3710
  SDynQueryCtrlOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SDynQueryCtrlOperatorInfo));
1,385,337✔
3711
  QUERY_CHECK_NULL(pInfo, code, line, _error, terrno)
1,385,337✔
3712

3713
  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
1,385,337✔
3714
  QUERY_CHECK_NULL(pOperator, code, line, _error, terrno)
1,385,337✔
3715

3716
  pOperator->pPhyNode = pPhyciNode;
1,385,337✔
3717
  pTaskInfo->dynamicTask = (int8_t)pPhyciNode->node.dynamicOp;
1,385,337✔
3718

3719
  code = appendDownstream(pOperator, pDownstream, numOfDownstream);
1,385,337✔
3720
  QUERY_CHECK_CODE(code, line, _error);
1,385,337✔
3721

3722
  setOperatorInfo(pOperator, "DynQueryCtrlOperator", QUERY_NODE_PHYSICAL_PLAN_DYN_QUERY_CTRL, false, OP_NOT_OPENED,
1,385,337✔
3723
                  pInfo, pTaskInfo);
3724

3725
  pInfo->qType = pPhyciNode->qType;
1,385,337✔
3726
  switch (pInfo->qType) {
1,385,337✔
3727
    case DYN_QTYPE_STB_HASH:
1,111,053✔
3728
      TAOS_MEMCPY(&pInfo->stbJoin.basic, &pPhyciNode->stbJoin, sizeof(pPhyciNode->stbJoin));
1,111,053✔
3729
      pInfo->stbJoin.pOutputDataBlockDesc = pPhyciNode->node.pOutputDataBlockDesc;
1,111,053✔
3730
      code = initSeqStbJoinTableHash(&pInfo->stbJoin.ctx.prev, pInfo->stbJoin.basic.batchFetch);
1,111,053✔
3731
      QUERY_CHECK_CODE(code, line, _error);
1,111,053✔
3732
      nextFp = seqStableJoin;
1,111,053✔
3733
      openFp = optrDummyOpenFn;
1,111,053✔
3734
      break;
1,111,053✔
3735
    case DYN_QTYPE_VTB_SCAN:
274,284✔
3736
      code = initVtbScanInfo(pInfo, pMsgCb, pPhyciNode, pTaskInfo);
274,284✔
3737
      QUERY_CHECK_CODE(code, line, _error);
274,284✔
3738
      nextFp = vtbScanNext;
274,284✔
3739
      openFp = vtbScanOpen;
274,284✔
3740
      break;
274,284✔
3741
    case DYN_QTYPE_VTB_WINDOW:
×
3742
      code = initVtbWindowInfo(pInfo, pPhyciNode, pTaskInfo, pOperator);
×
3743
      QUERY_CHECK_CODE(code, line, _error);
×
3744
      nextFp = vtbWindowNext;
×
3745
      openFp = vtbWindowOpen;
×
3746
      break;
×
UNCOV
3747
    case DYN_QTYPE_VTB_AGG:
×
UNCOV
3748
      code = initVtbScanInfo(pInfo, pMsgCb, pPhyciNode, pTaskInfo);
×
UNCOV
3749
      QUERY_CHECK_CODE(code, line, _error);
×
UNCOV
3750
      nextFp = vtbAggNext;
×
UNCOV
3751
      openFp = vtbAggOpen;
×
UNCOV
3752
      break;
×
3753
    default:
×
3754
      qError("unsupported dynamic query ctrl type: %d", pInfo->qType);
×
3755
      code = TSDB_CODE_INVALID_PARA;
×
3756
      goto _error;
×
3757
  }
3758

3759
  pOperator->fpSet = createOperatorFpSet(openFp, nextFp, NULL, destroyDynQueryCtrlOperator, optrDefaultBufFn,
1,385,337✔
3760
                                         NULL, optrDefaultGetNextExtFn, NULL);
3761

3762
  setOperatorResetStateFn(pOperator, resetDynQueryCtrlOperState);
1,385,337✔
3763
  *pOptrInfo = pOperator;
1,385,337✔
3764
  return TSDB_CODE_SUCCESS;
1,385,337✔
3765

3766
_error:
×
3767
  if (pInfo != NULL) {
×
3768
    destroyDynQueryCtrlOperator(pInfo);
×
3769
  }
3770
  qError("failed to create dyn query ctrl operator, %s code:%s, line:%d", __func__, tstrerror(code), line);
×
3771
  destroyOperatorAndDownstreams(pOperator, pDownstream, numOfDownstream);
×
3772
  pTaskInfo->code = code;
×
3773
  return code;
×
3774
}
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