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

taosdata / TDengine / #5053

13 May 2026 12:00PM UTC coverage: 73.397% (+0.06%) from 73.338%
#5053

push

travis-ci

web-flow
feat: taosdump support stream backup/restore (#35326)

139 of 170 new or added lines in 3 files covered. (81.76%)

627 existing lines in 131 files now uncovered.

281694 of 383795 relevant lines covered (73.4%)

132505311.38 hits per line

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

74.15
/source/libs/executor/src/executil.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 "function.h"
17
#include "functionMgt.h"
18
#include "index.h"
19
#include "os.h"
20
#include "query.h"
21
#include "querynodes.h"
22
#include "taoserror.h"
23
#include "tarray.h"
24
#include "tcompare.h"
25
#include "tdatablock.h"
26
#include "thash.h"
27
#include "tmsg.h"
28
#include "ttime.h"
29

30
#include "executil.h"
31
#include "executorInt.h"
32
#include "querytask.h"
33
#include "storageapi.h"
34
#include "ttypes.h"
35
#include "tutil.h"
36
#include "tjson.h"
37
#include "trpc.h"
38
#include "filter.h"
39
#include "operator.h"
40
#include "tref.h"
41
#include "streamMsg.h"
42

43
typedef struct tagFilterAssist {
44
  SHashObj* colHash;
45
  int32_t   index;
46
  SArray*   cInfoList;
47
  int32_t   code;
48
} tagFilterAssist;
49

50
typedef struct STransTagExprCtx {
51
  int32_t      code;
52
  SMetaReader* pReader;
53
} STransTagExprCtx;
54

55
typedef enum {
56
  FILTER_NO_LOGIC = 1,
57
  FILTER_AND,
58
  FILTER_OTHER,
59
} FilterCondType;
60

61
static int32_t optimizeTbnameInCondImpl(void* pVnode, SArray* list, SNode* pTagCond, SStorageAPI* pStoreAPI,
62
                                        uint64_t suid);
63

64
static int64_t getLimit(const SNode* pLimit) {
1,449,870,273✔
65
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->limit) ? -1 : ((SLimitNode*)pLimit)->limit->datum.i;
1,449,870,273✔
66
}
67
static int64_t getOffset(const SNode* pLimit) {
1,449,742,163✔
68
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->offset) ? -1 : ((SLimitNode*)pLimit)->offset->datum.i;
1,449,742,163✔
69
}
70
static void releaseColInfoData(void* pCol);
71
int32_t sendFetchRemoteNodeReq(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes, bool reset);
72

73
void initResultRowInfo(SResultRowInfo* pResultRowInfo) {
548,748,886✔
74
  pResultRowInfo->size = 0;
548,748,886✔
75
  pResultRowInfo->cur.pageId = -1;
548,778,612✔
76
}
548,745,243✔
77

78
void closeResultRow(SResultRow* pResultRow) { pResultRow->closed = true; }
6,244,555✔
79

80
void resetResultRow(SResultRow* pResultRow, size_t entrySize) {
1,070,411,877✔
81
  pResultRow->numOfRows = 0;
1,070,411,877✔
82
  pResultRow->nOrigRows = 0;
1,070,414,207✔
83
  pResultRow->closed = false;
1,070,419,244✔
84
  pResultRow->endInterp = false;
1,070,420,176✔
85
  pResultRow->startInterp = false;
1,070,419,991✔
86

87
  if (entrySize > 0) {
1,070,422,787✔
88
    memset(pResultRow->pEntryInfo, 0, entrySize);
1,070,423,253✔
89
  }
90
}
1,070,421,204✔
91

92
// TODO refactor: use macro
93
SResultRowEntryInfo* getResultEntryInfo(const SResultRow* pRow, int32_t index, const int32_t* offset) {
2,147,483,647✔
94
  return (SResultRowEntryInfo*)((char*)pRow->pEntryInfo + offset[index]);
2,147,483,647✔
95
}
96

97
size_t getResultRowSize(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
327,354,838✔
98
  int32_t rowSize = (numOfOutput * sizeof(SResultRowEntryInfo)) + sizeof(SResultRow);
327,354,838✔
99

100
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,387,344,080✔
101
    rowSize += pCtx[i].resDataInfo.interBufSize;
1,060,074,768✔
102
  }
103

104
  return rowSize;
327,269,312✔
105
}
106

107
// Convert buf read from rocksdb to result row
108
int32_t getResultRowFromBuf(SExprSupp* pSup, const char* inBuf, size_t inBufSize, char** outBuf, size_t* outBufSize) {
×
109
  if (inBuf == NULL || pSup == NULL) {
×
110
    qError("invalid input parameters, inBuf:%p, pSup:%p", inBuf, pSup);
×
111
    return TSDB_CODE_INVALID_PARA;
×
112
  }
113
  SqlFunctionCtx* pCtx = pSup->pCtx;
×
114
  int32_t*        offset = pSup->rowEntryInfoOffset;
×
115
  SResultRow*     pResultRow = NULL;
×
116
  size_t          processedSize = 0;
×
117
  int32_t         code = TSDB_CODE_SUCCESS;
×
118

119
  // calculate the size of output buffer
120
  *outBufSize = getResultRowSize(pCtx, pSup->numOfExprs);
×
121
  *outBuf = taosMemoryMalloc(*outBufSize);
×
122
  if (*outBuf == NULL) {
×
123
    qError("failed to allocate memory for output buffer, size:%zu", *outBufSize);
×
124
    return terrno;
×
125
  }
126
  pResultRow = (SResultRow*)*outBuf;
×
127
  (void)memcpy(pResultRow, inBuf, sizeof(SResultRow));
×
128
  inBuf += sizeof(SResultRow);
×
129
  processedSize += sizeof(SResultRow);
×
130

131
  for (int32_t i = 0; i < pSup->numOfExprs; ++i) {
×
132
    int32_t len = *(int32_t*)inBuf;
×
133
    inBuf += sizeof(int32_t);
×
134
    processedSize += sizeof(int32_t);
×
135
    if (pResultRow->version != FUNCTION_RESULT_INFO_VERSION && pCtx->fpSet.decode) {
×
136
      code = pCtx->fpSet.decode(&pCtx[i], inBuf, getResultEntryInfo(pResultRow, i, offset), pResultRow->version);
×
137
      if (code != TSDB_CODE_SUCCESS) {
×
138
        qError("failed to decode result row, code:%d", code);
×
139
        return code;
×
140
      }
141
    } else {
142
      (void)memcpy(getResultEntryInfo(pResultRow, i, offset), inBuf, len);
×
143
    }
144
    inBuf += len;
×
145
    processedSize += len;
×
146
  }
147

148
  if (processedSize < inBufSize) {
×
149
    // stream stores extra data after result row
150
    size_t leftLen = inBufSize - processedSize;
×
151
    TAOS_MEMORY_REALLOC(*outBuf, *outBufSize + leftLen);
×
152
    if (*outBuf == NULL) {
×
153
      qError("failed to reallocate memory for output buffer, size:%zu", *outBufSize + leftLen);
×
154
      return terrno;
×
155
    }
156
    (void)memcpy(*outBuf + *outBufSize, inBuf, leftLen);
×
157
    inBuf += leftLen;
×
158
    processedSize += leftLen;
×
159
    *outBufSize += leftLen;
×
160
  }
161

162
  qTrace("[StreamInternal] get result inBufSize:%zu, outBufSize:%zu", inBufSize, *outBufSize);
×
163
  return TSDB_CODE_SUCCESS;
×
164
}
165

166
// Convert result row to buf for rocksdb
167
int32_t putResultRowToBuf(SExprSupp* pSup, const char* inBuf, size_t inBufSize, char** outBuf, size_t* outBufSize) {
×
168
  if (pSup == NULL || inBuf == NULL || outBuf == NULL || outBufSize == NULL) {
×
169
    qError("invalid input parameters, inBuf:%p, pSup:%p, outBufSize:%p, outBuf:%p", inBuf, pSup, outBufSize, outBuf);
×
170
    return TSDB_CODE_INVALID_PARA;
×
171
  }
172

173
  SqlFunctionCtx* pCtx = pSup->pCtx;
×
174
  int32_t*        offset = pSup->rowEntryInfoOffset;
×
175
  SResultRow*     pResultRow = (SResultRow*)inBuf;
×
176
  size_t          rowSize = getResultRowSize(pCtx, pSup->numOfExprs);
×
177

178
  if (rowSize > inBufSize) {
×
179
    qError("invalid input buffer size, rowSize:%zu, inBufSize:%zu", rowSize, inBufSize);
×
180
    return TSDB_CODE_INVALID_PARA;
×
181
  }
182

183
  // calculate the size of output buffer
184
  *outBufSize = rowSize + sizeof(int32_t) * pSup->numOfExprs;
×
185
  if (rowSize < inBufSize) {
×
186
    *outBufSize += inBufSize - rowSize;
×
187
  }
188

189
  *outBuf = taosMemoryMalloc(*outBufSize);
×
190
  if (*outBuf == NULL) {
×
191
    qError("failed to allocate memory for output buffer, size:%zu", *outBufSize);
×
192
    return terrno;
×
193
  }
194

195
  char* pBuf = *outBuf;
×
196
  pResultRow->version = FUNCTION_RESULT_INFO_VERSION;
×
197
  (void)memcpy(pBuf, pResultRow, sizeof(SResultRow));
×
198
  pBuf += sizeof(SResultRow);
×
199
  for (int32_t i = 0; i < pSup->numOfExprs; ++i) {
×
200
    size_t len = sizeof(SResultRowEntryInfo) + pCtx[i].resDataInfo.interBufSize;
×
201
    *(int32_t*)pBuf = (int32_t)len;
×
202
    pBuf += sizeof(int32_t);
×
203
    (void)memcpy(pBuf, getResultEntryInfo(pResultRow, i, offset), len);
×
204
    pBuf += len;
×
205
  }
206

207
  if (rowSize < inBufSize) {
×
208
    // stream stores extra data after result row
209
    size_t leftLen = inBufSize - rowSize;
×
210
    (void)memcpy(pBuf, inBuf + rowSize, leftLen);
×
211
    pBuf += leftLen;
×
212
  }
213

214
  qTrace("[StreamInternal] put result inBufSize:%zu, outBufSize:%zu", inBufSize, *outBufSize);
×
215
  return TSDB_CODE_SUCCESS;
×
216
}
217

218
static void freeEx(void* p) { taosMemoryFree(*(void**)p); }
×
219

220
void cleanupGroupResInfo(SGroupResInfo* pGroupResInfo) {
150,950,616✔
221
  taosMemoryFreeClear(pGroupResInfo->pBuf);
150,950,616✔
222
  if (pGroupResInfo->freeItem) {
150,954,109✔
223
    //    taosArrayDestroy(pGroupResInfo->pRows);
224
    taosArrayDestroyEx(pGroupResInfo->pRows, freeEx);
×
225
    pGroupResInfo->freeItem = false;
×
226
    pGroupResInfo->pRows = NULL;
×
227
  } else {
228
    taosArrayDestroy(pGroupResInfo->pRows);
150,947,300✔
229
    pGroupResInfo->pRows = NULL;
150,937,606✔
230
  }
231
  pGroupResInfo->index = 0;
150,944,615✔
232
  pGroupResInfo->delIndex = 0;
150,944,146✔
233
}
150,939,748✔
234

235
int32_t resultrowComparAsc(const void* p1, const void* p2) {
2,147,483,647✔
236
  SResKeyPos* pp1 = *(SResKeyPos**)p1;
2,147,483,647✔
237
  SResKeyPos* pp2 = *(SResKeyPos**)p2;
2,147,483,647✔
238

239
  if (pp1->groupId == pp2->groupId) {
2,147,483,647✔
240
    int64_t pts1 = *(int64_t*)pp1->key;
2,147,483,647✔
241
    int64_t pts2 = *(int64_t*)pp2->key;
2,147,483,647✔
242

243
    if (pts1 == pts2) {
2,147,483,647✔
244
      return 0;
×
245
    } else {
246
      return pts1 < pts2 ? -1 : 1;
2,147,483,647✔
247
    }
248
  } else {
249
    return pp1->groupId < pp2->groupId ? -1 : 1;
2,147,483,647✔
250
  }
251
}
252

253
static int32_t resultrowComparDesc(const void* p1, const void* p2) { return resultrowComparAsc(p2, p1); }
2,147,483,647✔
254

255
int32_t initGroupedResultInfo(SGroupResInfo* pGroupResInfo, SSHashObj* pHashmap, int32_t order) {
113,736,497✔
256
  int32_t code = TSDB_CODE_SUCCESS;
113,736,497✔
257
  int32_t lino = 0;
113,736,497✔
258
  if (pGroupResInfo->pRows != NULL) {
113,736,497✔
259
    taosArrayDestroy(pGroupResInfo->pRows);
2,630,342✔
260
  }
261
  if (pGroupResInfo->pBuf) {
113,744,819✔
262
    taosMemoryFree(pGroupResInfo->pBuf);
2,630,342✔
263
    pGroupResInfo->pBuf = NULL;
2,630,342✔
264
  }
265

266
  // extract the result rows information from the hash map
267
  int32_t size = tSimpleHashGetSize(pHashmap);
113,736,638✔
268

269
  void* pData = NULL;
113,738,393✔
270
  pGroupResInfo->pRows = taosArrayInit(size, POINTER_BYTES);
113,738,393✔
271
  QUERY_CHECK_NULL(pGroupResInfo->pRows, code, lino, _end, terrno);
113,746,274✔
272

273
  size_t  keyLen = 0;
113,735,836✔
274
  int32_t iter = 0;
113,735,713✔
275
  int64_t bufLen = 0, offset = 0;
113,742,935✔
276

277
  // todo move away and record this during create window
278
  while ((pData = tSimpleHashIterate(pHashmap, pData, &iter)) != NULL) {
2,147,483,647✔
279
    /*void* key = */ (void)tSimpleHashGetKey(pData, &keyLen);
280
    bufLen += keyLen + sizeof(SResultRowPosition);
2,147,483,647✔
281
  }
282

283
  pGroupResInfo->pBuf = taosMemoryMalloc(bufLen);
113,678,420✔
284
  QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
113,746,169✔
285

286
  iter = 0;
113,737,085✔
287
  while ((pData = tSimpleHashIterate(pHashmap, pData, &iter)) != NULL) {
2,147,483,647✔
288
    void* key = tSimpleHashGetKey(pData, &keyLen);
2,147,483,647✔
289

290
    SResKeyPos* p = (SResKeyPos*)(pGroupResInfo->pBuf + offset);
2,147,483,647✔
291

292
    p->groupId = *(uint64_t*)key;
2,147,483,647✔
293
    p->pos = *(SResultRowPosition*)pData;
2,147,483,647✔
294
    memcpy(p->key, (char*)key + sizeof(uint64_t), keyLen - sizeof(uint64_t));
2,147,483,647✔
295
    void* tmp = taosArrayPush(pGroupResInfo->pRows, &p);
2,147,483,647✔
296
    QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
2,147,483,647✔
297

298
    offset += keyLen + sizeof(struct SResultRowPosition);
2,147,483,647✔
299
  }
300

301
  if (order == TSDB_ORDER_ASC || order == TSDB_ORDER_DESC) {
113,707,748✔
302
    __compar_fn_t fn = (order == TSDB_ORDER_ASC) ? resultrowComparAsc : resultrowComparDesc;
7,111,879✔
303
    size = POINTER_BYTES;
7,111,879✔
304
    taosSort(pGroupResInfo->pRows->pData, taosArrayGetSize(pGroupResInfo->pRows), size, fn);
7,111,879✔
305
  }
306

307
  pGroupResInfo->index = 0;
113,707,748✔
308

309
_end:
113,728,428✔
310
  if (code != TSDB_CODE_SUCCESS) {
113,741,626✔
311
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
312
  }
313
  return code;
113,741,626✔
314
}
315

316
void initMultiResInfoFromArrayList(SGroupResInfo* pGroupResInfo, SArray* pArrayList) {
×
317
  if (pGroupResInfo->pRows != NULL) {
×
318
    taosArrayDestroy(pGroupResInfo->pRows);
×
319
  }
320

321
  pGroupResInfo->freeItem = true;
×
322
  pGroupResInfo->pRows = pArrayList;
×
323
  pGroupResInfo->index = 0;
×
324
  pGroupResInfo->delIndex = 0;
×
325
}
×
326

327
bool hasRemainResults(SGroupResInfo* pGroupResInfo) {
457,942,338✔
328
  if (pGroupResInfo->pRows == NULL) {
457,942,338✔
329
    return false;
×
330
  }
331

332
  return pGroupResInfo->index < taosArrayGetSize(pGroupResInfo->pRows);
457,947,450✔
333
}
334

335
int32_t getNumOfTotalRes(SGroupResInfo* pGroupResInfo) {
242,246,911✔
336
  if (pGroupResInfo->pRows == 0) {
242,246,911✔
337
    return 0;
×
338
  }
339

340
  return (int32_t)taosArrayGetSize(pGroupResInfo->pRows);
242,249,170✔
341
}
342

343
SArray* createSortInfo(SNodeList* pNodeList) {
59,290,456✔
344
  size_t numOfCols = 0;
59,290,456✔
345

346
  if (pNodeList != NULL) {
59,290,456✔
347
    numOfCols = LIST_LENGTH(pNodeList);
59,229,547✔
348
  } else {
349
    numOfCols = 0;
60,909✔
350
  }
351

352
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
59,292,729✔
353
  if (pList == NULL) {
59,283,536✔
354
    return pList;
×
355
  }
356

357
  for (int32_t i = 0; i < numOfCols; ++i) {
134,098,112✔
358
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
74,809,385✔
359
    if (!pSortKey) {
74,811,639✔
360
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
361
      taosArrayDestroy(pList);
×
362
      pList = NULL;
×
363
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
364
      break;
×
365
    }
366
    SBlockOrderInfo bi = {0};
74,811,639✔
367
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
74,809,012✔
368
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);
74,808,708✔
369

370
    if (nodeType(pSortKey->pExpr) != QUERY_NODE_COLUMN) {
74,810,800✔
371
      qError("invalid order by expr type:%d", nodeType(pSortKey->pExpr));
×
372
      taosArrayDestroy(pList);
×
373
      pList = NULL;
×
374
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
375
      break;
×
376
    }
377
    
378
    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
74,794,624✔
379
    bi.slotId = pColNode->slotId;
74,813,866✔
380
    void* tmp = taosArrayPush(pList, &bi);
74,814,603✔
381
    if (!tmp) {
74,814,603✔
382
      taosArrayDestroy(pList);
×
383
      pList = NULL;
×
384
      break;
×
385
    }
386
  }
387

388
  return pList;
59,288,140✔
389
}
390

391
SSDataBlock* createDataBlockFromDescNode(void* p) {
864,813,748✔
392
  SDataBlockDescNode* pNode = (SDataBlockDescNode*)p;
864,813,748✔
393
  int32_t      numOfCols = LIST_LENGTH(pNode->pSlots);
864,813,748✔
394
  SSDataBlock* pBlock = NULL;
864,672,529✔
395
  int32_t      code = TSDB_CODE_SUCCESS;
864,660,858✔
396
  int32_t      lino = 0;
864,660,858✔
397

398
  code = createDataBlock(&pBlock);
864,660,858✔
399
  QUERY_CHECK_CODE(code, lino, _return);
864,741,465✔
400

401
  pBlock->info.id.blockId = pNode->dataBlockId;
864,741,465✔
402
  pBlock->info.calWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
864,735,087✔
403
  pBlock->info.watermark = INT64_MIN;
864,898,931✔
404

405
  int32_t i = 0;
864,803,185✔
406
  SNode*  node = NULL;
864,803,185✔
407
  FOREACH(node, pNode->pSlots) {
2,147,483,647✔
408
    SSlotDescNode* pDescNode = (SSlotDescNode*)node;
2,147,483,647✔
409
    SColumnInfoData idata =
2,147,483,647✔
410
        createColumnInfoData(pDescNode->dataType.type, pDescNode->dataType.bytes, pDescNode->slotId);
2,147,483,647✔
411
    idata.info.scale = pDescNode->dataType.scale;
2,147,483,647✔
412
    idata.info.precision = pDescNode->dataType.precision;
2,147,483,647✔
413
    idata.info.noData = pDescNode->reserve;
2,147,483,647✔
414

415
    code = blockDataAppendColInfo(pBlock, &idata);
2,147,483,647✔
416
    QUERY_CHECK_CODE(code, lino, _return);
2,147,483,647✔
417
    ++i;
2,147,483,647✔
418
  }
419

420
  if (pBlock != NULL && i != numOfCols) {
865,088,056✔
421
    code = TSDB_CODE_INVALID_PARA;
×
422
    QUERY_CHECK_CODE(code, lino, _return);
×
423
  }
424

425
  return pBlock;
865,088,056✔
UNCOV
426
_return:
×
427
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
428
  blockDataDestroy(pBlock);
×
429
  terrno = code;
×
430
  return NULL;
14✔
431
}
432

433
int32_t prepareDataBlockBuf(SSDataBlock* pDataBlock, SColMatchInfo* pMatchInfo) {
296,350,998✔
434
  SDataBlockInfo* pBlockInfo = &pDataBlock->info;
296,350,998✔
435

436
  for (int32_t i = 0; i < taosArrayGetSize(pMatchInfo->pList); ++i) {
1,334,860,510✔
437
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
1,054,598,485✔
438
    if (!pItem) {
1,054,686,838✔
439
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
440
      return terrno;
×
441
    }
442

443
    if (pItem->isPk) {
1,054,686,838✔
444
      SColumnInfoData* pInfoData = taosArrayGet(pDataBlock->pDataBlock, pItem->dstSlotId);
16,239,648✔
445
      if (!pInfoData) {
16,161,384✔
446
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
447
        return terrno;
×
448
      }
449
      pBlockInfo->pks[0].type = pInfoData->info.type;
16,161,384✔
450
      pBlockInfo->pks[1].type = pInfoData->info.type;
16,180,294✔
451

452
      // allocate enough buffer size, which is pInfoData->info.bytes
453
      if (IS_VAR_DATA_TYPE(pItem->dataType.type)) {
16,175,163✔
454
        pBlockInfo->pks[0].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
5,094,720✔
455
        if (pBlockInfo->pks[0].pData == NULL) {
5,090,359✔
456
          return terrno;
×
457
        }
458

459
        pBlockInfo->pks[1].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
5,090,359✔
460
        if (pBlockInfo->pks[1].pData == NULL) {
5,094,098✔
461
          taosMemoryFreeClear(pBlockInfo->pks[0].pData);
×
462
          return terrno;
×
463
        }
464

465
        pBlockInfo->pks[0].nData = pInfoData->info.bytes;
5,095,527✔
466
        pBlockInfo->pks[1].nData = pInfoData->info.bytes;
5,096,186✔
467
      }
468

469
      break;
16,173,112✔
470
    }
471
  }
472

473
  return TSDB_CODE_SUCCESS;
296,391,126✔
474
}
475

476
EDealRes doTranslateTagExpr(SNode** pNode, void* pContext) {
2,946,596✔
477
  STransTagExprCtx* pCtx = pContext;
2,946,596✔
478
  SMetaReader*      mr = pCtx->pReader;
2,946,596✔
479
  bool              isTagCol = false, isTbname = false;
2,946,596✔
480
  if (nodeType(*pNode) == QUERY_NODE_COLUMN) {
2,946,596✔
481
    SColumnNode* pCol = (SColumnNode*)*pNode;
842,044✔
482
    if (pCol->colType == COLUMN_TYPE_TBNAME)
842,044✔
483
      isTbname = true;
×
484
    else
485
      isTagCol = true;
842,044✔
486
  } else if (nodeType(*pNode) == QUERY_NODE_FUNCTION) {
2,104,552✔
487
    SFunctionNode* pFunc = (SFunctionNode*)*pNode;
744✔
488
    if (pFunc->funcType == FUNCTION_TYPE_TBNAME) isTbname = true;
744✔
489
  }
490
  if (isTagCol) {
2,946,596✔
491
    SColumnNode* pSColumnNode = *(SColumnNode**)pNode;
842,044✔
492

493
    SValueNode* res = NULL;
842,044✔
494
    pCtx->code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
842,044✔
495
    if (NULL == res) {
842,044✔
496
      return DEAL_RES_ERROR;
×
497
    }
498

499
    res->translate = true;
842,044✔
500
    res->node.resType = pSColumnNode->node.resType;
842,044✔
501

502
    STagVal tagVal = {0};
842,044✔
503
    tagVal.cid = pSColumnNode->colId;
842,044✔
504
    const char* p = mr->pAPI->extractTagVal(mr->me.ctbEntry.pTags, pSColumnNode->node.resType.type, &tagVal);
842,044✔
505
    if (p == NULL) {
842,044✔
506
      res->node.resType.type = TSDB_DATA_TYPE_NULL;
372✔
507
    } else if (pSColumnNode->node.resType.type == TSDB_DATA_TYPE_JSON) {
841,672✔
508
      int32_t len = ((const STag*)p)->len;
×
509
      res->datum.p = taosMemoryCalloc(len + 1, 1);
×
510
      if (NULL == res->datum.p) {
×
511
        return DEAL_RES_ERROR;
×
512
      }
513
      memcpy(res->datum.p, p, len);
×
514
    } else if (IS_VAR_DATA_TYPE(pSColumnNode->node.resType.type)) {
841,672✔
515
      if (IS_STR_DATA_BLOB(pSColumnNode->node.resType.type)) {
840,184✔
516
        return TSDB_CODE_BLOB_NOT_SUPPORT_TAG;
×
517
      }
518

519
      res->datum.p = taosMemoryCalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1, 1);
840,184✔
520
      if (NULL == res->datum.p) {
840,184✔
521
        return DEAL_RES_ERROR;
×
522
      }
523

524
      if (IS_STR_DATA_BLOB(pSColumnNode->node.resType.type)) {
840,184✔
525
        memcpy(blobDataVal(res->datum.p), tagVal.pData, tagVal.nData);
×
526
        blobDataSetLen(res->datum.p, tagVal.nData);
×
527
      } else {
528
        memcpy(varDataVal(res->datum.p), tagVal.pData, tagVal.nData);
840,184✔
529
        varDataSetLen(res->datum.p, tagVal.nData);
840,184✔
530
      }
531
    } else {
532
      int32_t code = nodesSetValueNodeValue(res, &(tagVal.i64));
1,488✔
533
      if (code != TSDB_CODE_SUCCESS) {
1,488✔
534
        return DEAL_RES_ERROR;
×
535
      }
536
    }
537
    nodesDestroyNode(*pNode);
842,044✔
538
    *pNode = (SNode*)res;
842,044✔
539
  } else if (isTbname) {
2,104,552✔
540
    SValueNode* res = NULL;
×
541
    pCtx->code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
×
542
    if (NULL == res) {
×
543
      return DEAL_RES_ERROR;
×
544
    }
545

546
    res->translate = true;
×
547
    res->node.resType = ((SExprNode*)(*pNode))->resType;
×
548

549
    int32_t len = strlen(mr->me.name);
×
550
    res->datum.p = taosMemoryCalloc(len + VARSTR_HEADER_SIZE + 1, 1);
×
551
    if (NULL == res->datum.p) {
×
552
      return DEAL_RES_ERROR;
×
553
    }
554
    memcpy(varDataVal(res->datum.p), mr->me.name, len);
×
555
    varDataSetLen(res->datum.p, len);
×
556
    nodesDestroyNode(*pNode);
×
557
    *pNode = (SNode*)res;
×
558
  }
559

560
  return DEAL_RES_CONTINUE;
2,946,596✔
561
}
562

563
int32_t isQualifiedTable(int64_t uid, SNode* pTagCond, void* vnode, bool* pQualified, SStorageAPI* pAPI) {
422,324✔
564
  int32_t     code = TSDB_CODE_SUCCESS;
422,324✔
565
  SMetaReader mr = {0};
422,324✔
566

567
  pAPI->metaReaderFn.initReader(&mr, vnode, META_READER_LOCK, &pAPI->metaFn);
422,324✔
568
  code = pAPI->metaReaderFn.getEntryGetUidCache(&mr, uid);
422,324✔
569
  if (TSDB_CODE_SUCCESS != code) {
422,324✔
570
    pAPI->metaReaderFn.clearReader(&mr);
×
571
    *pQualified = false;
×
572

573
    return TSDB_CODE_SUCCESS;
×
574
  }
575

576
  SNode* pTagCondTmp = NULL;
422,324✔
577
  code = nodesCloneNode(pTagCond, &pTagCondTmp);
422,324✔
578
  if (TSDB_CODE_SUCCESS != code) {
422,324✔
579
    *pQualified = false;
×
580
    pAPI->metaReaderFn.clearReader(&mr);
×
581
    return code;
×
582
  }
583
  STransTagExprCtx ctx = {.code = 0, .pReader = &mr};
422,324✔
584
  nodesRewriteExprPostOrder(&pTagCondTmp, doTranslateTagExpr, &ctx);
422,324✔
585
  pAPI->metaReaderFn.clearReader(&mr);
422,324✔
586
  if (TSDB_CODE_SUCCESS != ctx.code) {
422,324✔
587
    *pQualified = false;
×
588
    nodesDestroyNode(pTagCondTmp);
×
589
    terrno = code;
×
590
    return code;
×
591
  }
592

593
  SNode* pNew = NULL;
422,324✔
594
  code = scalarCalculateConstants(pTagCondTmp, &pNew);
422,324✔
595
  if (TSDB_CODE_SUCCESS != code) {
422,324✔
596
    terrno = code;
×
597
    nodesDestroyNode(pTagCondTmp);
×
598
    *pQualified = false;
×
599

600
    return code;
×
601
  }
602

603
  SValueNode* pValue = (SValueNode*)pNew;
422,324✔
604
  *pQualified = pValue->datum.b;
422,324✔
605

606
  nodesDestroyNode(pNew);
422,324✔
607
  return TSDB_CODE_SUCCESS;
422,324✔
608
}
609

610
static EDealRes getColumn(SNode** pNode, void* pContext) {
70,838,493✔
611
  tagFilterAssist* pData = (tagFilterAssist*)pContext;
70,838,493✔
612
  SColumnNode*     pSColumnNode = NULL;
70,838,493✔
613
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
70,846,768✔
614
    pSColumnNode = *(SColumnNode**)pNode;
24,394,935✔
615
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
46,480,818✔
616
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
1,475,279✔
617
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
1,475,716✔
618
      pData->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
922,716✔
619
      if (NULL == pSColumnNode) {
923,153✔
620
        return DEAL_RES_ERROR;
×
621
      }
622
      pSColumnNode->colId = -1;
923,153✔
623
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
923,153✔
624
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
923,153✔
625
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
923,153✔
626
      nodesDestroyNode(*pNode);
923,153✔
627
      *pNode = (SNode*)pSColumnNode;
923,153✔
628
    } else {
629
      return DEAL_RES_CONTINUE;
552,563✔
630
    }
631
  } else {
632
    return DEAL_RES_CONTINUE;
44,973,774✔
633
  }
634

635
  void* data = taosHashGet(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
25,317,665✔
636
  if (!data) {
25,316,411✔
637
    int32_t tempRes =
638
        taosHashPut(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
21,800,890✔
639
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
21,808,759✔
640
      return DEAL_RES_ERROR;
×
641
    }
642
    pSColumnNode->slotId = pData->index++;
21,808,759✔
643
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
21,804,230✔
644
                         .type = pSColumnNode->node.resType.type,
21,799,694✔
645
                         .bytes = pSColumnNode->node.resType.bytes,
21,792,995✔
646
                         .pk = pSColumnNode->isPk};
21,796,603✔
647
#if TAG_FILTER_DEBUG
648
    qDebug("tagfilter build column info, slotId:%d, colId:%d, type:%d", pSColumnNode->slotId, cInfo.colId, cInfo.type);
649
#endif
650
    void* tmp = taosArrayPush(pData->cInfoList, &cInfo);
21,794,576✔
651
    if (!tmp) {
21,798,481✔
652
      return DEAL_RES_ERROR;
×
653
    }
654
  } else {
655
    SColumnNode* col = *(SColumnNode**)data;
3,515,521✔
656
    pSColumnNode->slotId = col->slotId;
3,515,098✔
657
  }
658

659
  return DEAL_RES_CONTINUE;
25,308,582✔
660
}
661

662
static int32_t createResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
20,333,211✔
663
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
20,333,211✔
664
  if (pColumnData == NULL) {
20,327,187✔
665
    return terrno;
×
666
  }
667

668
  pColumnData->info.type = pType->type;
20,327,187✔
669
  pColumnData->info.bytes = pType->bytes;
20,332,748✔
670
  pColumnData->info.scale = pType->scale;
20,320,292✔
671
  pColumnData->info.precision = pType->precision;
20,330,310✔
672

673
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
20,317,341✔
674
  if (code != TSDB_CODE_SUCCESS) {
20,321,193✔
675
    terrno = code;
×
676
    releaseColInfoData(pColumnData);
×
677
    return terrno;
×
678
  }
679

680
  pParam->columnData = pColumnData;
20,321,193✔
681
  pParam->colAlloced = true;
20,322,638✔
682
  return TSDB_CODE_SUCCESS;
20,323,560✔
683
}
684

685
static void releaseColInfoData(void* pCol) {
5,672,065✔
686
  if (pCol) {
5,672,065✔
687
    SColumnInfoData* col = (SColumnInfoData*)pCol;
5,672,242✔
688
    colDataDestroy(col);
5,672,242✔
689
    taosMemoryFree(col);
5,672,419✔
690
  }
691
}
5,670,835✔
692

693
void freeItem(void* p) {
235,653,580✔
694
  STUidTagInfo* pInfo = p;
235,653,580✔
695
  if (pInfo->pTagVal != NULL) {
235,653,580✔
696
    taosMemoryFree(pInfo->pTagVal);
235,221,778✔
697
  }
698
}
235,642,901✔
699

700
typedef struct {
701
  col_id_t  colId;
702
  SNode*    pValueNode;
703
  int32_t   bytes;  // length defined in schema
704
} STagDataEntry;
705

706
static int compareTagDataEntry(const void* a, const void* b) {
20,984✔
707
  STagDataEntry* p1 = (STagDataEntry*)a;
20,984✔
708
  STagDataEntry* p2 = (STagDataEntry*)b;
20,984✔
709
  return compareInt16Val(&p1->colId, &p2->colId);
20,984✔
710
}
711

712
static int32_t buildTagDataEntryKey(SArray* pIdWithValue, char** keyBuf, int32_t keyLen) {
10,492✔
713
  *keyBuf = (char*)taosMemoryCalloc(1, keyLen);
10,492✔
714
  if (NULL == *keyBuf) {
10,492✔
715
    qError(
×
716
      "failed to allocate memory for tag filter optimization key, size:%d",
717
      keyLen);
718
    return terrno;
×
719
  }
720
  char* pStart = *keyBuf;
10,492✔
721
  for (int32_t i = 0; i < taosArrayGetSize(pIdWithValue); ++i) {
31,476✔
722
    STagDataEntry* entry      = (STagDataEntry*)taosArrayGet(pIdWithValue, i);
20,984✔
723
    SValueNode*    pValueNode = (SValueNode*)entry->pValueNode;
20,984✔
724
    // num type may have different bytes length, use the smaller one
725
    int32_t        bytes = TMIN(entry->bytes, pValueNode->node.resType.bytes);
20,984✔
726

727
    (void)memcpy(pStart, &entry->colId, sizeof(col_id_t));
20,984✔
728
    pStart += sizeof(col_id_t);
20,984✔
729

730
    if (!pValueNode->isNull) {
20,984✔
731
      switch (pValueNode->node.resType.type) {
18,404✔
732
        case TSDB_DATA_TYPE_BOOL:
2,236✔
733
          (void)memcpy(
2,236✔
734
            pStart, &pValueNode->datum.b, bytes);
2,236✔
735
          pStart += bytes;
2,236✔
736
          break;
2,236✔
737
        case TSDB_DATA_TYPE_TINYINT:
8,944✔
738
        case TSDB_DATA_TYPE_SMALLINT:
739
        case TSDB_DATA_TYPE_INT:
740
        case TSDB_DATA_TYPE_BIGINT:
741
        case TSDB_DATA_TYPE_TIMESTAMP:
742
          (void)memcpy(
8,944✔
743
            pStart, &pValueNode->datum.i, bytes);
8,944✔
744
          pStart += bytes;
8,944✔
745
          break;
9,116✔
746
        case TSDB_DATA_TYPE_UTINYINT:
×
747
        case TSDB_DATA_TYPE_USMALLINT:
748
        case TSDB_DATA_TYPE_UINT:
749
        case TSDB_DATA_TYPE_UBIGINT:
750
          (void)memcpy(
×
751
            pStart, &pValueNode->datum.u, bytes);
×
752
          pStart += bytes;
×
753
          break;
×
754
        case TSDB_DATA_TYPE_FLOAT:
2,236✔
755
        case TSDB_DATA_TYPE_DOUBLE:
756
          (void)memcpy(
2,236✔
757
            pStart, &pValueNode->datum.d, bytes);
2,236✔
758
          pStart += bytes;
2,236✔
759
          break;
2,236✔
760
        case TSDB_DATA_TYPE_VARCHAR:
4,816✔
761
        case TSDB_DATA_TYPE_VARBINARY:
762
        case TSDB_DATA_TYPE_NCHAR:
763
          (void)memcpy(pStart,
4,816✔
764
            varDataVal(pValueNode->datum.p), varDataLen(pValueNode->datum.p));
4,816✔
765
          pStart += varDataLen(pValueNode->datum.p);
4,816✔
766
          break;
4,816✔
767
        default:
×
768
          qError("unsupported tag data type %d in tag filter optimization",
×
769
            pValueNode->node.resType.type);
770
          return TSDB_CODE_STREAM_INTERNAL_ERROR;
×
771
      }
772
    }
773
  }
774

775
  return TSDB_CODE_SUCCESS;
10,492✔
776
}
777

778
static void extractTagDataEntry(
20,984✔
779
  SOperatorNode* pOpNode, SArray* pIdWithValue) {
780
  SNode* pLeft = pOpNode->pLeft;
20,984✔
781
  SNode* pRight = pOpNode->pRight;
20,984✔
782
  SColumnNode* pColNode = nodeType(pLeft) == QUERY_NODE_COLUMN ?
20,984✔
783
    (SColumnNode*)pLeft : (SColumnNode*)pRight;
20,984✔
784
  SValueNode* pValueNode = nodeType(pLeft) == QUERY_NODE_VALUE ?
20,984✔
785
    (SValueNode*)pLeft : (SValueNode*)pRight;
20,984✔
786

787
  STagDataEntry entry = {0};
20,984✔
788
  entry.colId = pColNode->colId;
20,984✔
789
  entry.pValueNode = (SNode*)pValueNode;
20,984✔
790
  entry.bytes = pColNode->node.resType.bytes;
20,984✔
791
  void* _tmp = taosArrayPush(pIdWithValue, &entry);
20,984✔
792
}
20,984✔
793

794
static int32_t extractTagFilterTagDataEntries(
10,492✔
795
  const SNode* pTagCond, SArray* pIdWithVal) {
796
  if (NULL == pTagCond || NULL == pIdWithVal ||
10,492✔
797
    (nodeType(pTagCond) != QUERY_NODE_OPERATOR &&
10,492✔
798
      nodeType(pTagCond) != QUERY_NODE_LOGIC_CONDITION)) {
10,492✔
799
    qError("invalid parameter to extract tag filter symbol");
×
800
    return TSDB_CODE_STREAM_INTERNAL_ERROR;
×
801
  }
802

803
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
10,492✔
804
    extractTagDataEntry((SOperatorNode*)pTagCond, pIdWithVal);
×
805
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
10,492✔
806
    SNode* pChild = NULL;
10,492✔
807
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
31,476✔
808
      extractTagDataEntry((SOperatorNode*)pChild, pIdWithVal);
20,984✔
809
    }
810
  }
811

812
  taosArraySort(pIdWithVal, compareTagDataEntry);
10,492✔
813

814
  return TSDB_CODE_SUCCESS;
10,320✔
815
}
816

817
static int32_t genStableTagFilterDigest(const SNode* pTagCond, T_MD5_CTX* pContext) {
10,492✔
818
  if (pTagCond == NULL) {
10,492✔
819
    return TSDB_CODE_SUCCESS;
×
820
  }
821

822
  char*   payload = NULL;
10,492✔
823
  int32_t len = 0;
10,492✔
824
  int32_t code = TSDB_CODE_SUCCESS;
10,492✔
825
  int32_t lino = 0;
10,492✔
826

827
  SArray* pIdWithVal = taosArrayInit(TARRAY_MIN_SIZE, sizeof(STagDataEntry));
10,492✔
828
  code = extractTagFilterTagDataEntries(pTagCond, pIdWithVal);
10,492✔
829
  QUERY_CHECK_CODE(code, lino, _end);
10,320✔
830
  for (int32_t i = 0; i < taosArrayGetSize(pIdWithVal); ++i) {
31,132✔
831
    STagDataEntry* pEntry = taosArrayGet(pIdWithVal, i);
20,812✔
832
    len += sizeof(col_id_t) + pEntry->bytes;
20,812✔
833
  }
834
  code = buildTagDataEntryKey(pIdWithVal, &payload, len);
10,320✔
835
  QUERY_CHECK_CODE(code, lino, _end);
10,492✔
836

837
  tMD5Init(pContext);
10,492✔
838
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
10,492✔
839
  tMD5Final(pContext);
10,492✔
840

841
_end:
10,492✔
842
  if (TSDB_CODE_SUCCESS != code) {
10,492✔
843
    qError("%s failed at line %d since %s",
×
844
      __func__, __LINE__, tstrerror(code));
845
  }
846
  taosArrayDestroy(pIdWithVal);
10,492✔
847
  taosMemoryFree(payload);
10,492✔
848
  return code;
10,492✔
849
}
850

851
static int32_t genTagFilterDigest(const SNode* pTagCond, T_MD5_CTX* pContext) {
32,936✔
852
  if (pTagCond == NULL) {
32,936✔
853
    return TSDB_CODE_SUCCESS;
31,140✔
854
  }
855

856
  char*   payload = NULL;
1,796✔
857
  int32_t len = 0;
1,796✔
858
  int32_t code = nodesNodeToMsg(pTagCond, &payload, &len);
1,796✔
859
  if (code != TSDB_CODE_SUCCESS) {
1,796✔
860
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
861
    return code;
×
862
  }
863

864
  tMD5Init(pContext);
1,796✔
865
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
1,796✔
866
  tMD5Final(pContext);
1,796✔
867

868
  // void* tmp = NULL;
869
  // uint32_t size = 0;
870
  // (void)taosAscii2Hex((const char*)pContext->digest, 16, &tmp, &size);
871
  // qInfo("tag filter digest payload: %s", tmp);
872
  // taosMemoryFree(tmp);
873

874
  taosMemoryFree(payload);
1,796✔
875
  return TSDB_CODE_SUCCESS;
1,796✔
876
}
877

878
static int32_t genTbGroupDigest(const SNode* pGroup, uint8_t* filterDigest, T_MD5_CTX* pContext) {
×
879
  int32_t code = TSDB_CODE_SUCCESS;
×
880
  int32_t lino = 0;
×
881
  char*   payload = NULL;
×
882
  int32_t len = 0;
×
883
  code = nodesNodeToMsg(pGroup, &payload, &len);
×
884
  QUERY_CHECK_CODE(code, lino, _end);
×
885

886
  if (filterDigest[0]) {
×
887
    payload = taosMemoryRealloc(payload, len + tListLen(pContext->digest));
×
888
    QUERY_CHECK_NULL(payload, code, lino, _end, terrno);
×
889
    memcpy(payload + len, filterDigest + 1, tListLen(pContext->digest));
×
890
    len += tListLen(pContext->digest);
×
891
  }
892

893
  tMD5Init(pContext);
×
894
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
×
895
  tMD5Final(pContext);
×
896

897
_end:
×
898
  taosMemoryFree(payload);
×
899
  if (code != TSDB_CODE_SUCCESS) {
×
900
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
901
  }
902
  return code;
×
903
}
904

905
int32_t qGetColumnsFromNodeList(void* data, bool isList, SArray** pColList) {
19,831,876✔
906
  int32_t code = TSDB_CODE_SUCCESS;
19,831,876✔
907
  tagFilterAssist ctx = {0};
19,831,876✔
908
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
19,833,713✔
909
  if (ctx.colHash == NULL) {
19,827,424✔
910
    code = terrno;
×
911
    goto end;
×
912
  }
913

914
  ctx.index = 0;
19,827,424✔
915
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
19,827,424✔
916
  if (ctx.cInfoList == NULL) {
19,831,826✔
917
    code = terrno;
8,994✔
918
    goto end;
×
919
  }
920

921
  if (isList) {
19,822,832✔
922
    SNode* pNode = NULL;
5,167,708✔
923
    FOREACH(pNode, (SNodeList*)data) {
10,840,083✔
924
      nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
5,672,463✔
925
      if (TSDB_CODE_SUCCESS != ctx.code) {
5,672,419✔
926
        code = ctx.code;
×
927
        goto end;
×
928
      }
929
      REPLACE_NODE(pNode);
5,672,419✔
930
    }
931
  } else {
932
    SNode* pNode = (SNode*)data;
14,655,124✔
933
    nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
14,657,933✔
934
    if (TSDB_CODE_SUCCESS != ctx.code) {
14,661,944✔
935
      code = ctx.code;
×
936
      goto end;
×
937
    }
938
  }
939
  
940
  if (pColList != NULL) *pColList = ctx.cInfoList;
19,831,054✔
941
  ctx.cInfoList = NULL;
19,816,831✔
942

943
end:
19,817,167✔
944
  taosHashCleanup(ctx.colHash);
19,829,937✔
945
  taosArrayDestroy(ctx.cInfoList);
19,794,009✔
946
  return code;
19,803,688✔
947
}
948

949
static int32_t buildGroupInfo(SColumnInfoData* pValue, int32_t i, SArray* gInfo) {
824,835✔
950
  int32_t code = TSDB_CODE_SUCCESS;
824,835✔
951
  SStreamGroupValue* v = taosArrayReserve(gInfo, 1);
824,835✔
952
  if (v == NULL) {
824,835✔
953
    code = terrno;
×
954
    goto end;
×
955
  }
956
  if (colDataIsNull_s(pValue, i)) {
1,649,670✔
957
    v->isNull = true;
6,880✔
958
  } else {
959
    v->isNull = false;
817,955✔
960
    char* data = colDataGetData(pValue, i);
818,240✔
961
    if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
818,525✔
962
      if (tTagIsJson(data)) {
×
963
        code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
×
964
        goto end;
×
965
      }
966
      if (tTagIsJsonNull(data)) {
×
967
        v->isNull = true;
×
968
        goto end;
×
969
      }
970
      int32_t len = getJsonValueLen(data);
×
971
      v->data.type = pValue->info.type;
×
972
      v->data.nData = len;
×
973
      v->data.pData = taosMemoryCalloc(1, len + 1);
×
974
      if (v->data.pData == NULL) {
×
975
        code = terrno;
×
976
        goto end;
×
977
      }
978
      memcpy(v->data.pData, data, len);
×
979
      qDebug("buildGroupInfo:%d add json data len:%d, data:%s", i, len, (char*)v->data.pData);
×
980
    } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
818,525✔
981
      if (varDataTLen(data) > pValue->info.bytes) {
517,481✔
982
        code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
983
        goto end;
×
984
      }
985
      v->data.type = pValue->info.type;
517,481✔
986
      v->data.nData = varDataLen(data);
517,190✔
987
      v->data.pData = taosMemoryCalloc(1, varDataLen(data) + 1);
517,481✔
988
      if (v->data.pData == NULL) {
517,190✔
989
        code = terrno;
×
990
        goto end;
×
991
      }
992
      memcpy(v->data.pData, varDataVal(data), varDataLen(data));
517,190✔
993
      qDebug("buildGroupInfo:%d add var data type:%d, len:%d, data:%s", i, pValue->info.type, varDataLen(data), (char*)v->data.pData);
516,899✔
994
    } else if (pValue->info.type == TSDB_DATA_TYPE_DECIMAL) {  // reader todo decimal
301,044✔
995
      v->data.type = pValue->info.type;
×
996
      v->data.nData = pValue->info.bytes;
×
997
      v->data.pData = taosMemoryCalloc(1, pValue->info.bytes);
×
998
      if (v->data.pData == NULL) {
×
999
        code = terrno;
×
1000
        goto end;
×
1001
      }
1002
      memcpy(&v->data.pData, data, pValue->info.bytes);
×
1003
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
×
1004
    } else {  // reader todo decimal
1005
      v->data.type = pValue->info.type;
301,044✔
1006
      memcpy(&v->data.val, data, pValue->info.bytes);
301,044✔
1007
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
301,044✔
1008
    }
1009
  }
1010
end:
43,165✔
1011
  if (code != TSDB_CODE_SUCCESS) {
825,228✔
1012
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1013
    v->isNull = true;
×
1014
  }
1015
  return code;
825,228✔
1016
}
1017

1018
static void getColInfoResultForGroupbyForStream(void* pVnode, SNodeList* group, STableListInfo* pTableListInfo,
105,028✔
1019
                                   SStorageAPI* pAPI, SHashObj* groupIdMap) {
1020
  int32_t      code = TSDB_CODE_SUCCESS;
105,028✔
1021
  int32_t      lino = 0;
105,028✔
1022
  SArray*      pBlockList = NULL;
105,028✔
1023
  SSDataBlock* pResBlock = NULL;
105,028✔
1024
  SArray*      groupData = NULL;
105,028✔
1025
  SArray*      pUidTagList = NULL;
105,028✔
1026
  SArray*      gInfo = NULL;
105,028✔
1027
  int32_t      tbNameIndex = 0;
105,028✔
1028

1029
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
105,028✔
1030
  if (rows == 0) {
105,028✔
1031
    return;
×
1032
  }
1033

1034
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
105,028✔
1035
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
105,028✔
1036

1037
  for (int32_t i = 0; i < rows; ++i) {
472,422✔
1038
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
367,394✔
1039
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
367,394✔
1040
    STUidTagInfo info = {.uid = pkeyInfo->uid};
367,394✔
1041
    void*        tmp = taosArrayPush(pUidTagList, &info);
367,394✔
1042
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
367,394✔
1043
  }
1044
 
1045
  if (taosArrayGetSize(pUidTagList) > 0) {
105,028✔
1046
    code = pAPI->metaFn.getTableTagsByUidVersion(pVnode, pTableListInfo->idInfo.suid, pUidTagList, -1);
105,028✔
1047
  } else {
1048
    code = pAPI->metaFn.getTableTags(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
×
1049
  }
1050
  if (code != TSDB_CODE_SUCCESS) {
105,028✔
1051
    goto end;
×
1052
  }
1053

1054
  SArray* pColList = NULL;
105,028✔
1055
  code = qGetColumnsFromNodeList(group, true, &pColList);
105,028✔
1056
  if (code != TSDB_CODE_SUCCESS) {
105,028✔
1057
    goto end;
×
1058
  }
1059

1060
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
314,438✔
1061
    SColumnInfo* tmp = (SColumnInfo*)taosArrayGet(pColList, i);
209,410✔
1062
    if (tmp != NULL && tmp->colId == -1) {
209,410✔
1063
      tbNameIndex = i;
105,028✔
1064
    }
1065
  }
1066
  
1067
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
105,028✔
1068
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
105,028✔
1069
  taosArrayDestroy(pColList);
105,028✔
1070
  if (pResBlock == NULL) {
105,028✔
UNCOV
1071
    code = terrno;
×
UNCOV
1072
    goto end;
×
1073
  }
1074

1075
  pBlockList = taosArrayInit(2, POINTER_BYTES);
105,028✔
1076
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
105,028✔
1077

1078
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
105,028✔
1079
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
105,028✔
1080

1081
  groupData = taosArrayInit(2, POINTER_BYTES);
105,028✔
1082
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
105,028✔
1083

1084
  SNode* pNode = NULL;
105,028✔
1085
  FOREACH(pNode, group) {
314,438✔
1086
    SScalarParam output = {0};
209,410✔
1087

1088
    switch (nodeType(pNode)) {
209,410✔
1089
      case QUERY_NODE_VALUE:
×
1090
        break;
×
1091
      case QUERY_NODE_COLUMN:
209,410✔
1092
      case QUERY_NODE_OPERATOR:
1093
      case QUERY_NODE_FUNCTION: {
1094
        SExprNode* expNode = (SExprNode*)pNode;
209,410✔
1095
        code = createResultData(&expNode->resType, rows, &output);
209,410✔
1096
        if (code != TSDB_CODE_SUCCESS) {
209,410✔
1097
          goto end;
×
1098
        }
1099
        break;
209,410✔
1100
      }
1101

1102
      default:
×
1103
        code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1104
        goto end;
×
1105
    }
1106

1107
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
209,410✔
1108
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
209,410✔
1109
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
209,410✔
1110
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
209,410✔
1111
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
209,410✔
1112
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
×
1113
      continue;
×
1114
    } else {
1115
      gTaskScalarExtra.pStreamInfo = NULL;
×
1116
      gTaskScalarExtra.pStreamRange = NULL;
×
1117
      code = scalarCalculate(pNode, pBlockList, &output, &gTaskScalarExtra);
×
1118
    }
1119

1120
    if (code != TSDB_CODE_SUCCESS) {
209,410✔
1121
      releaseColInfoData(output.columnData);
×
1122
      goto end;
×
1123
    }
1124

1125
    void* tmp = taosArrayPush(groupData, &output.columnData);
209,410✔
1126
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
209,410✔
1127
  }
1128

1129
  for (int i = 0; i < rows; i++) {
472,422✔
1130
    gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
367,394✔
1131
    QUERY_CHECK_NULL(gInfo, code, lino, end, terrno);
367,394✔
1132

1133
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
367,394✔
1134
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
367,394✔
1135

1136
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
1,058,624✔
1137
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
691,230✔
1138
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
691,230✔
1139
        if (ret != TSDB_CODE_SUCCESS) {
691,515✔
1140
          qError("buildGroupInfo failed at line %d since %s", __LINE__, tstrerror(ret));
×
1141
          goto end;
×
1142
        }
1143
        if (j == tbNameIndex) {
691,515✔
1144
          SStreamGroupValue* v = taosArrayGetLast(gInfo);
367,394✔
1145
          if (v != NULL){
367,394✔
1146
            v->isTbname = true;
367,394✔
1147
            v->uid = info->uid;
367,394✔
1148
          }
1149
        }
1150
    }
1151

1152
    int32_t ret = taosHashPut(groupIdMap, &info->uid, sizeof(info->uid), &gInfo, POINTER_BYTES);
367,394✔
1153
    if (ret != TSDB_CODE_SUCCESS) {
367,394✔
1154
      qError("put groupid to map failed at line %d since %s", __LINE__, tstrerror(ret));
×
1155
      goto end;
×
1156
    }
1157
    qDebug("put groupid to map gid:%" PRIu64, info->uid);
367,394✔
1158
    gInfo = NULL;
367,394✔
1159
  }
1160

1161
end:
105,028✔
1162
  blockDataDestroy(pResBlock);
105,028✔
1163
  taosArrayDestroy(pBlockList);
105,028✔
1164
  taosArrayDestroyEx(pUidTagList, freeItem);
105,028✔
1165
  taosArrayDestroyP(groupData, releaseColInfoData);
105,028✔
1166
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
105,028✔
1167

1168
  if (code != TSDB_CODE_SUCCESS) {
105,028✔
UNCOV
1169
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1170
  }
1171
}
1172

1173
int32_t getColInfoResultForGroupby(SReadHandle* pHandle, SNodeList* group, STableListInfo* pTableListInfo, uint8_t* digest,
5,062,389✔
1174
                                   SStorageAPI* pAPI, bool initRemainGroups, SHashObj* groupIdMap, bool gIdFromBaseId) {
1175
  int32_t      code = TSDB_CODE_SUCCESS;
5,062,389✔
1176
  int32_t      lino = 0;
5,062,389✔
1177
  SArray*      pBlockList = NULL;
5,062,389✔
1178
  SSDataBlock* pResBlock = NULL;
5,062,389✔
1179
  void*        keyBuf = NULL;
5,062,680✔
1180
  SArray*      groupData = NULL;
5,062,680✔
1181
  SArray*      pUidTagList = NULL;
5,062,680✔
1182
  SArray*      tableList = NULL;
5,062,680✔
1183
  SArray*      gInfo = NULL;
5,062,680✔
1184
  void*        pVnode = pHandle->vnode;
5,062,680✔
1185

1186
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
5,062,680✔
1187
  if (rows == 0) {
5,062,680✔
1188
    return TSDB_CODE_SUCCESS;
×
1189
  } 
1190

1191
  T_MD5_CTX context = {0};
5,062,680✔
1192
  if (tsTagFilterCache && groupIdMap == NULL && digest) {
5,062,680✔
1193
    SNodeListNode* listNode = NULL;
×
1194
    code = nodesMakeNode(QUERY_NODE_NODE_LIST, (SNode**)&listNode);
×
1195
    if (TSDB_CODE_SUCCESS != code) {
×
1196
      goto end;
×
1197
    }
1198
    listNode->pNodeList = group;
×
1199
    code = genTbGroupDigest((SNode*)listNode, digest, &context);
×
1200
    QUERY_CHECK_CODE(code, lino, end);
×
1201

1202
    nodesFree(listNode);
×
1203

1204
    code = pAPI->metaFn.metaGetCachedTbGroup(pVnode, pTableListInfo->idInfo.suid, context.digest,
×
1205
                                             tListLen(context.digest), &tableList);
1206
    QUERY_CHECK_CODE(code, lino, end);
×
1207

1208
    if (tableList) {
×
1209
      taosArrayDestroy(pTableListInfo->pTableList);
×
1210
      pTableListInfo->pTableList = tableList;
×
1211
      qDebug("retrieve tb group list from cache, numOfTables:%d",
×
1212
             (int32_t)taosArrayGetSize(pTableListInfo->pTableList));
1213
      goto end;
×
1214
    }
1215
  }
1216

1217
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
5,062,680✔
1218
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
5,062,680✔
1219

1220
  for (int32_t i = 0; i < rows; ++i) {
32,228,628✔
1221
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
27,165,948✔
1222
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
27,165,948✔
1223
    STUidTagInfo info = {.uid = pkeyInfo->uid};
27,165,948✔
1224
    void*        tmp = taosArrayPush(pUidTagList, &info);
27,165,948✔
1225
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
27,165,948✔
1226
  }
1227

1228
  if (taosArrayGetSize(pUidTagList) > 0) {
5,062,680✔
1229
    code = pAPI->metaFn.getTableTagsByUidVersion(pVnode, pTableListInfo->idInfo.suid, pUidTagList, -1);
5,062,680✔
1230
  } else {
1231
    code = pAPI->metaFn.getTableTags(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
×
1232
  }
1233
  if (code != TSDB_CODE_SUCCESS) {
5,062,680✔
1234
    goto end;
×
1235
  }
1236

1237
  SArray* pColList = NULL;
5,062,680✔
1238
  code = qGetColumnsFromNodeList(group, true, &pColList); 
5,062,680✔
1239
  if (code != TSDB_CODE_SUCCESS) {
5,061,395✔
1240
    goto end;
×
1241
  }
1242

1243
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
5,061,395✔
1244
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
5,061,439✔
1245
  taosArrayDestroy(pColList);
5,062,680✔
1246
  if (pResBlock == NULL) {
5,061,907✔
1247
    code = terrno;
×
1248
    goto end;
×
1249
  }
1250

1251
  //  int64_t st1 = taosGetTimestampUs();
1252
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1253

1254
  pBlockList = taosArrayInit(2, POINTER_BYTES);
5,061,907✔
1255
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
5,062,503✔
1256

1257
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
5,062,680✔
1258
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
5,062,680✔
1259

1260
  groupData = taosArrayInit(2, POINTER_BYTES);
5,062,680✔
1261
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
5,062,040✔
1262

1263
  SNode* pNode = NULL;
5,062,680✔
1264
  FOREACH(pNode, group) {
10,525,733✔
1265
    SScalarParam output = {0};
5,462,663✔
1266

1267
    switch (nodeType(pNode)) {
5,462,019✔
1268
      case QUERY_NODE_VALUE:
×
1269
        break;
×
1270
      case QUERY_NODE_COLUMN:
5,462,792✔
1271
      case QUERY_NODE_OPERATOR:
1272
      case QUERY_NODE_FUNCTION: {
1273
        SExprNode* expNode = (SExprNode*)pNode;
5,462,792✔
1274
        code = createResultData(&expNode->resType, rows, &output);
5,462,792✔
1275
        if (code != TSDB_CODE_SUCCESS) {
5,462,961✔
1276
          goto end;
×
1277
        }
1278
        break;
5,462,961✔
1279
      }
1280
      case QUERY_NODE_REMOTE_VALUE: {
×
1281
        SRemoteValueNode* pRemote = (SRemoteValueNode*)pNode;
×
1282
        code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, pNode);
×
1283
        QUERY_CHECK_CODE(code, lino, end);
×
1284
        break;
×
1285
      }
1286
      
1287
      default:
×
1288
        code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1289
        goto end;
×
1290
    }
1291

1292
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
5,462,961✔
1293
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
5,314,208✔
1294
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
5,314,208✔
1295
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
5,312,695✔
1296
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
5,312,695✔
1297
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
148,249✔
1298
      continue;
×
1299
    } else {
1300
      gTaskScalarExtra.pStreamInfo = NULL;
148,801✔
1301
      gTaskScalarExtra.pStreamRange = NULL;
148,801✔
1302
      code = scalarCalculate(pNode, pBlockList, &output, &gTaskScalarExtra);
148,801✔
1303
    }
1304

1305
    if (code != TSDB_CODE_SUCCESS) {
5,461,030✔
1306
      releaseColInfoData(output.columnData);
×
1307
      goto end;
×
1308
    }
1309

1310
    void* tmp = taosArrayPush(groupData, &output.columnData);
5,461,803✔
1311
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
5,461,803✔
1312
  }
1313

1314
  int32_t baseIdLen = gIdFromBaseId ? sizeof(int64_t) : 0;
5,062,214✔
1315
  int32_t keyLen = baseIdLen;
5,062,214✔
1316
  SNode*  node;
1317
  FOREACH(node, group) {
10,525,196✔
1318
    SExprNode* pExpr = (SExprNode*)node;
5,460,758✔
1319
    keyLen += pExpr->resType.bytes;
5,460,758✔
1320
  }
1321

1322
  int32_t nullFlagSize = sizeof(int8_t) * LIST_LENGTH(group);
5,061,396✔
1323
  keyLen += nullFlagSize;
5,062,125✔
1324

1325
  keyBuf = taosMemoryCalloc(1, keyLen);
5,062,125✔
1326
  if (keyBuf == NULL) {
5,060,598✔
1327
    code = terrno;
×
1328
    goto end;
×
1329
  }
1330

1331
  if (initRemainGroups) {
5,060,598✔
1332
    pTableListInfo->remainGroups =
2,289,295✔
1333
        taosHashInit(rows, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
2,286,505✔
1334
    if (pTableListInfo->remainGroups == NULL) {
2,289,295✔
1335
      code = terrno;
×
1336
      goto end;
×
1337
    }
1338
  }
1339

1340
  for (int i = 0; i < rows; i++) {
32,224,271✔
1341
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
27,161,965✔
1342
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
27,162,827✔
1343

1344
    if (groupIdMap != NULL){
27,162,827✔
1345
      gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
76,224✔
1346
    }
1347

1348
    if (baseIdLen > 0) {
27,162,827✔
1349
      *(uint64_t*)keyBuf = info->baseGId;
×
1350
    }
1351
    char* isNull = (char*)keyBuf + baseIdLen;
27,162,827✔
1352
    char* pStart = (char*)keyBuf + baseIdLen + sizeof(int8_t) * LIST_LENGTH(group);
27,163,646✔
1353
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
56,129,294✔
1354
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
28,966,684✔
1355

1356
      if (groupIdMap != NULL && gInfo != NULL) {
28,966,566✔
1357
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
133,599✔
1358
        if (ret != TSDB_CODE_SUCCESS) {
133,713✔
1359
          qError("buildGroupInfo failed at line %d since %s", __LINE__, tstrerror(ret));
×
1360
          taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1361
          gInfo = NULL;
×
1362
        }
1363
      }
1364
      
1365
      if (colDataIsNull_s(pValue, i)) {
57,932,468✔
1366
        isNull[j] = 1;
116,319✔
1367
      } else {
1368
        isNull[j] = 0;
28,849,469✔
1369
        char* data = colDataGetData(pValue, i);
28,851,670✔
1370
        if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
28,849,051✔
1371
          // if (tTagIsJson(data)) {
1372
          //   code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
1373
          //   goto end;
1374
          // }
1375
          if (tTagIsJsonNull(data)) {
90,302✔
1376
            isNull[j] = 1;
×
1377
            continue;
×
1378
          }
1379
          int32_t len = getJsonValueLen(data);
90,302✔
1380
          memcpy(pStart, data, len);
90,302✔
1381
          pStart += len;
90,302✔
1382
        } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
28,755,880✔
1383
          if (IS_STR_DATA_BLOB(pValue->info.type)) {
24,630,738✔
1384
            if (blobDataTLen(data) > TSDB_MAX_BLOB_LEN) {
660✔
1385
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1386
              goto end;
×
1387
            }
1388
            memcpy(pStart, data, blobDataTLen(data));
×
1389
            pStart += blobDataTLen(data);
×
1390
          } else {
1391
            if (varDataTLen(data) > pValue->info.bytes) {
24,629,611✔
1392
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1393
              goto end;
×
1394
            }
1395
            memcpy(pStart, data, varDataTLen(data));
24,631,867✔
1396
            pStart += varDataTLen(data);
24,631,270✔
1397
          }
1398
        } else {
1399
          memcpy(pStart, data, pValue->info.bytes);
4,126,707✔
1400
          pStart += pValue->info.bytes;
4,128,306✔
1401
        }
1402
      }
1403
    }
1404

1405
    int32_t len = (int32_t)(pStart - (char*)keyBuf);
27,161,141✔
1406
    info->groupId = calcGroupId(keyBuf, len);
27,161,141✔
1407
    if (groupIdMap != NULL && gInfo != NULL) {
27,157,792✔
1408
      int32_t ret = taosHashPut(groupIdMap, &info->groupId, sizeof(info->groupId), &gInfo, POINTER_BYTES);
75,972✔
1409
      if (ret != TSDB_CODE_SUCCESS) {
76,515✔
1410
        qError("put groupid to map failed at line %d since %s", __LINE__, tstrerror(ret));
×
1411
        taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1412
      }
1413
      qDebug("put groupid to map gid:%" PRIu64, info->groupId);
76,515✔
1414
      gInfo = NULL;
76,515✔
1415
    }
1416
    if (initRemainGroups) {
27,158,335✔
1417
      // groupId ~ table uid
1418
      code = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId), &(info->uid),
13,040,745✔
1419
                         sizeof(info->uid));
1420
      if (code == TSDB_CODE_DUP_KEY) {
13,044,045✔
1421
        code = TSDB_CODE_SUCCESS;
892,962✔
1422
      }
1423
      QUERY_CHECK_CODE(code, lino, end);
13,044,045✔
1424
    }
1425
  }
1426

1427
  if (tsTagFilterCache && groupIdMap == NULL && digest) {
5,062,306✔
1428
    tableList = taosArrayDup(pTableListInfo->pTableList, NULL);
×
1429
    QUERY_CHECK_NULL(tableList, code, lino, end, terrno);
×
1430

1431
    code = pAPI->metaFn.metaPutTbGroupToCache(pVnode, pTableListInfo->idInfo.suid, context.digest,
×
1432
                                              tListLen(context.digest), tableList,
1433
                                              taosArrayGetSize(tableList) * sizeof(STableKeyInfo));
×
1434
    QUERY_CHECK_CODE(code, lino, end);
×
1435
  }
1436

1437
  //  int64_t st2 = taosGetTimestampUs();
1438
  //  qDebug("calculate tag block rows:%d, cost:%ld us", rows, st2-st1);
1439

1440
end:
5,061,522✔
1441
  taosMemoryFreeClear(keyBuf);
5,062,680✔
1442
  blockDataDestroy(pResBlock);
5,062,680✔
1443
  taosArrayDestroy(pBlockList);
5,061,730✔
1444
  taosArrayDestroyEx(pUidTagList, freeItem);
5,061,907✔
1445
  taosArrayDestroyP(groupData, releaseColInfoData);
5,061,100✔
1446
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
5,061,859✔
1447

1448
  if (code != TSDB_CODE_SUCCESS) {
5,061,859✔
1449
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1450
  }
1451
  return code;
5,062,411✔
1452
}
1453

1454
static int32_t nameComparFn(const void* p1, const void* p2) {
833,440✔
1455
  const char* pName1 = *(const char**)p1;
833,440✔
1456
  const char* pName2 = *(const char**)p2;
833,440✔
1457

1458
  int32_t ret = strcmp(pName1, pName2);
833,440✔
1459
  if (ret == 0) {
833,440✔
1460
    return 0;
16,524✔
1461
  } else {
1462
    return (ret > 0) ? 1 : -1;
816,916✔
1463
  }
1464
}
1465

1466
static SArray* getTableNameList(const SNodeListNode* pList) {
443,603✔
1467
  int32_t    code = TSDB_CODE_SUCCESS;
443,603✔
1468
  int32_t    lino = 0;
443,603✔
1469
  int32_t    len = LIST_LENGTH(pList->pNodeList);
443,603✔
1470
  SListCell* cell = pList->pNodeList->pHead;
443,603✔
1471

1472
  SArray* pTbList = taosArrayInit(len, POINTER_BYTES);
443,603✔
1473
  QUERY_CHECK_NULL(pTbList, code, lino, _end, terrno);
443,603✔
1474

1475
  for (int i = 0; i < pList->pNodeList->length; i++) {
1,234,508✔
1476
    SValueNode* valueNode = (SValueNode*)cell->pNode;
790,905✔
1477
    if (!IS_VAR_DATA_TYPE(valueNode->node.resType.type)) {
790,905✔
1478
      terrno = TSDB_CODE_INVALID_PARA;
×
1479
      taosArrayDestroy(pTbList);
×
1480
      return NULL;
×
1481
    }
1482

1483
    char* name = varDataVal(valueNode->datum.p);
790,905✔
1484
    void* tmp = taosArrayPush(pTbList, &name);
790,905✔
1485
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
790,905✔
1486
    cell = cell->pNext;
790,905✔
1487
  }
1488

1489
  size_t numOfTables = taosArrayGetSize(pTbList);
443,603✔
1490

1491
  // order the name
1492
  taosArraySort(pTbList, nameComparFn);
443,603✔
1493

1494
  // remove the duplicates
1495
  SArray* pNewList = taosArrayInit(taosArrayGetSize(pTbList), sizeof(void*));
443,603✔
1496
  QUERY_CHECK_NULL(pNewList, code, lino, _end, terrno);
443,603✔
1497
  void* tmpTbl = taosArrayGet(pTbList, 0);
443,603✔
1498
  QUERY_CHECK_NULL(tmpTbl, code, lino, _end, terrno);
443,603✔
1499
  void* tmp = taosArrayPush(pNewList, tmpTbl);
443,603✔
1500
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
443,603✔
1501

1502
  for (int32_t i = 1; i < numOfTables; ++i) {
790,905✔
1503
    char** name = taosArrayGetLast(pNewList);
347,302✔
1504
    char** nameInOldList = taosArrayGet(pTbList, i);
347,302✔
1505
    QUERY_CHECK_NULL(nameInOldList, code, lino, _end, terrno);
347,302✔
1506
    if (strcmp(*name, *nameInOldList) == 0) {
347,302✔
1507
      continue;
8,632✔
1508
    }
1509

1510
    tmp = taosArrayPush(pNewList, nameInOldList);
338,670✔
1511
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
338,670✔
1512
  }
1513

1514
_end:
443,603✔
1515
  taosArrayDestroy(pTbList);
443,603✔
1516
  if (code != TSDB_CODE_SUCCESS) {
443,603✔
1517
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1518
    return NULL;
×
1519
  }
1520
  return pNewList;
443,603✔
1521
}
1522

1523
static int tableUidCompare(const void* a, const void* b) {
×
1524
  uint64_t u1 = *(uint64_t*)a;
×
1525
  uint64_t u2 = *(uint64_t*)b;
×
1526

1527
  if (u1 == u2) {
×
1528
    return 0;
×
1529
  }
1530

1531
  return u1 < u2 ? -1 : 1;
×
1532
}
1533

1534
static int32_t filterTableInfoCompare(const void* a, const void* b) {
44,016✔
1535
  STUidTagInfo* p1 = (STUidTagInfo*)a;
44,016✔
1536
  STUidTagInfo* p2 = (STUidTagInfo*)b;
44,016✔
1537

1538
  if (p1->uid == p2->uid) {
44,016✔
1539
    return 0;
×
1540
  }
1541

1542
  return p1->uid < p2->uid ? -1 : 1;
44,016✔
1543
}
1544

1545
static FilterCondType checkTagCond(SNode* cond) {
17,391,096✔
1546
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
17,391,096✔
1547
    return FILTER_NO_LOGIC;
12,415,943✔
1548
  }
1549
  if (nodeType(cond) == QUERY_NODE_LOGIC_CONDITION && ((SLogicConditionNode*)cond)->condType == LOGIC_COND_TYPE_AND) {
4,974,695✔
1550
    return FILTER_AND;
4,304,382✔
1551
  }
1552
  return FILTER_OTHER;
667,349✔
1553
}
1554

1555
static int32_t doInWithAnd(SNode* cond, void* pVnode, SArray* list, SStorageAPI* pAPI, uint64_t suid) {
4,984,904✔
1556
  if (nodeType(cond) != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
4,984,904✔
1557
    return -1;
666,762✔
1558
  }
1559

1560
  SLogicConditionNode* pNode = (SLogicConditionNode*)cond;
4,308,205✔
1561
  SNodeList*           pList = (SNodeList*)pNode->pParameterList;
4,308,205✔
1562

1563
  int32_t len = LIST_LENGTH(pList);
4,315,178✔
1564
  if (len <= 0) {
4,316,968✔
1565
    return -1;
×
1566
  }
1567

1568
  SListCell* cell = pList->pHead;
4,316,968✔
1569
  for (int i = 0; i < len; i++) {
13,378,357✔
1570
    if (cell == NULL) break;
9,073,325✔
1571
    if (nodeType(cell->pNode) == QUERY_NODE_OPERATOR && optimizeTbnameInCondImpl(pVnode, list, cell->pNode, pAPI, suid) == 0) {
9,073,325✔
1572
      return 0;
8,940✔
1573
    }
1574
    cell = cell->pNext;
9,064,847✔
1575
  }
1576
  return -1;
4,305,032✔
1577
}
1578

1579
static int32_t optimizeTbnameInCond(void* pVnode, int64_t suid, SArray* list, SNode* cond, SStorageAPI* pAPI, int64_t version) {
17,834,699✔
1580
  int32_t code = 0;
17,834,699✔
1581
  int32_t lino = 0;
17,834,699✔
1582
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
17,834,699✔
1583
    code = optimizeTbnameInCondImpl(pVnode, list, cond, pAPI, suid);
12,848,036✔
1584
  } else {
1585
    code = doInWithAnd(cond, pVnode, list, pAPI, suid);
4,987,879✔
1586
    QUERY_CHECK_CODE(code, lino, end);
4,980,147✔
1587
    code = pAPI->metaFn.getTableTagsByUidVersion(pVnode, suid, list, version);
7,507✔
1588
  }
1589
  
1590
end:
17,831,679✔
1591
  return code;
17,831,679✔
1592
}
1593

1594
static int32_t getTableListInInOperator(void* pVnode, SArray* pExistedUidList, SNodeListNode* pList, SStorageAPI* pStoreAPI,
443,603✔
1595
                                        uint64_t suid) {
1596
  int32_t   code = 0;                                          
443,603✔
1597
  SArray*   pTbList = getTableNameList(pList);
443,603✔
1598
  int32_t   numOfTables = taosArrayGetSize(pTbList);
443,603✔
1599
  SHashObj* uHash = NULL;
443,603✔
1600

1601
  size_t numOfExisted = taosArrayGetSize(pExistedUidList);  // len > 0 means there already have uids
443,603✔
1602
  if (numOfExisted > 0) {
443,603✔
1603
    uHash = taosHashInit(numOfExisted / 0.7, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
2,384✔
1604
    if (!uHash) {
2,384✔
1605
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1606
      code = terrno;
×
1607
      goto end;
×
1608
    }
1609

1610
    for (int i = 0; i < numOfExisted; i++) {
2,381,616✔
1611
      STUidTagInfo* pTInfo = taosArrayGet(pExistedUidList, i);
2,379,232✔
1612
      if (!pTInfo) {
2,378,636✔
1613
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1614
        code = terrno;
×
1615
        goto end;
×
1616
      }
1617
      code = taosHashPut(uHash, &pTInfo->uid, sizeof(uint64_t), &i, sizeof(i));
2,378,636✔
1618
      if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_DUP_KEY) {
2,379,232✔
1619
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1620
        goto end;
×
1621
      }
1622
    }
1623
  }
1624
  taosArrayClear(pExistedUidList);
443,603✔
1625

1626
  for (int i = 0; i < numOfTables; i++) {
1,225,876✔
1627
    char* name = taosArrayGetP(pTbList, i);
782,273✔
1628
    uint64_t uid = 0, csuid = 0;
782,273✔
1629
    if (pStoreAPI->metaFn.getTableUidByName(pVnode, name, &uid) != 0) {
782,273✔
1630
      continue;
330,166✔
1631
    }
1632

1633
    ETableType tbType = TSDB_TABLE_MAX;
452,107✔
1634
    if (pStoreAPI->metaFn.getTableTypeSuidByName(pVnode, name, &tbType, &csuid) == 0 &&
452,107✔
1635
        (tbType == TSDB_CHILD_TABLE || tbType == TSDB_VIRTUAL_CHILD_TABLE) &&
452,107✔
1636
        csuid == suid) {
444,378✔
1637
      if (NULL == uHash || taosHashGet(uHash, &uid, sizeof(uid)) != NULL) {
441,868✔
1638
        STUidTagInfo s = {.uid = uid, .name = name, .pTagVal = NULL};
440,676✔
1639
        void*        tmp = taosArrayPush(pExistedUidList, &s);
440,676✔
1640
        if (!tmp) {
440,676✔
1641
          code = terrno;
×
1642
          goto end;
×
1643
        }
1644
      }
1645
    } 
1646
  }
1647
  taosArraySort(pExistedUidList, filterTableInfoCompare);
443,603✔
1648
  taosArrayRemoveDuplicate(pExistedUidList, filterTableInfoCompare, NULL);
443,603✔
1649
end:
443,603✔
1650
  taosArrayDestroy(pTbList);
443,603✔
1651
  taosHashCleanup(uHash);
443,603✔
1652
  return code;
443,603✔
1653
}
1654

1655
// only return uid that does not contained in pExistedUidList
1656
static int32_t optimizeTbnameInCondImpl(void* pVnode, SArray* pExistedUidList, SNode* pTagCond, SStorageAPI* pStoreAPI,
21,923,387✔
1657
                                        uint64_t suid) {
1658
  SOperatorNode* pNode = (SOperatorNode*)pTagCond;
21,923,387✔
1659
  if (pNode->opType != OP_TYPE_IN) {
21,923,387✔
1660
    return -1;
20,890,941✔
1661
  }
1662

1663
  if ((pNode->pLeft != NULL && ((nodeType(pNode->pLeft) == QUERY_NODE_FUNCTION &&
1,028,410✔
1664
                                 ((SFunctionNode*)pNode->pLeft)->funcType == FUNCTION_TYPE_TBNAME)) ||
446,059✔
1665
       (nodeType(pNode->pLeft) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pLeft)->colType == COLUMN_TYPE_TBNAME)) &&
582,774✔
1666
      (pNode->pRight != NULL && nodeType(pNode->pRight) == QUERY_NODE_NODE_LIST)) {
446,810✔
1667
    SNodeListNode* pList = (SNodeListNode*)pNode->pRight;
443,603✔
1668

1669
    int32_t len = LIST_LENGTH(pList->pNodeList);
443,603✔
1670
    if (len <= 0) {
443,603✔
1671
      return -1;
×
1672
    }
1673

1674
    return getTableListInInOperator(pVnode, pExistedUidList, pList, pStoreAPI, suid);
443,603✔
1675
  }
1676

1677
  return -1;
586,090✔
1678
}
1679

1680
SSDataBlock* createTagValBlockForFilter(SArray* pColList, int32_t numOfTables, SArray* pUidTagList, void* pVnode,
21,739,284✔
1681
                                        SStorageAPI* pStorageAPI) {
1682
  int32_t      code = TSDB_CODE_SUCCESS;
21,739,284✔
1683
  int32_t      lino = 0;
21,739,284✔
1684
  SSDataBlock* pResBlock = NULL;
21,739,284✔
1685
  code = createDataBlock(&pResBlock);
21,744,445✔
1686
  QUERY_CHECK_CODE(code, lino, _end);
21,739,243✔
1687

1688
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
45,462,228✔
1689
    SColumnInfoData colInfo = {0};
23,721,425✔
1690
    void*           tmp = taosArrayGet(pColList, i);
23,719,785✔
1691
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
23,693,932✔
1692
    colInfo.info = *(SColumnInfo*)tmp;
23,693,932✔
1693
    code = blockDataAppendColInfo(pResBlock, &colInfo);
23,706,950✔
1694
    QUERY_CHECK_CODE(code, lino, _end);
23,720,724✔
1695
  }
1696

1697
  code = blockDataEnsureCapacity(pResBlock, numOfTables);
21,736,154✔
1698
  if (code != TSDB_CODE_SUCCESS) {
21,725,619✔
1699
    terrno = code;
×
1700
    blockDataDestroy(pResBlock);
×
1701
    return NULL;
×
1702
  }
1703

1704
  pResBlock->info.rows = numOfTables;
21,725,619✔
1705

1706
  int32_t numOfCols = taosArrayGetSize(pResBlock->pDataBlock);
21,741,462✔
1707

1708
  for (int32_t i = 0; i < numOfTables; i++) {
259,751,334✔
1709
    STUidTagInfo* p1 = taosArrayGet(pUidTagList, i);
237,995,894✔
1710
    QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
238,039,237✔
1711

1712
    for (int32_t j = 0; j < numOfCols; j++) {
487,911,971✔
1713
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, j);
249,883,341✔
1714
      QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
249,810,803✔
1715

1716
      if (pColInfo->info.colId == -1) {  // tbname
249,810,803✔
1717
        char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
8,552,374✔
1718
        if (p1->name != NULL) {
8,553,064✔
1719
          STR_TO_VARSTR(str, p1->name);
6,556✔
1720
        } else {  // name is not retrieved during filter
1721
          code = pStorageAPI->metaFn.getTableNameByUid(pVnode, p1->uid, str);
8,545,204✔
1722
          QUERY_CHECK_CODE(code, lino, _end);
8,541,678✔
1723
        }
1724

1725
        code = colDataSetVal(pColInfo, i, str, false);
8,548,234✔
1726
        QUERY_CHECK_CODE(code, lino, _end);
8,552,811✔
1727
#if TAG_FILTER_DEBUG
1728
        qDebug("tagfilter uid:%ld, tbname:%s", *uid, str + 2);
1729
#endif
1730
      } else {
1731
        STagVal tagVal = {0};
241,261,418✔
1732
        tagVal.cid = pColInfo->info.colId;
241,267,782✔
1733
        if (p1->pTagVal == NULL) {
241,283,396✔
1734
          colDataSetNULL(pColInfo, i);
4,300✔
1735
        } else {
1736
          const char* p = pStorageAPI->metaFn.extractTagVal(p1->pTagVal, pColInfo->info.type, &tagVal);
241,242,760✔
1737

1738
          if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
241,378,725✔
1739
            colDataSetNULL(pColInfo, i);
2,728,422✔
1740
          } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
238,654,618✔
1741
            code = colDataSetVal(pColInfo, i, p, false);
713,921✔
1742
            QUERY_CHECK_CODE(code, lino, _end);
713,921✔
1743
          } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
392,987,436✔
1744
            if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
155,139,901✔
1745
              QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
1746
            }
1747
            char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
155,139,546✔
1748
            QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
155,093,904✔
1749
            varDataSetLen(tmp, tagVal.nData);
155,093,904✔
1750
            memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
155,080,154✔
1751
            code = colDataSetVal(pColInfo, i, tmp, false);
155,091,099✔
1752
#if TAG_FILTER_DEBUG
1753
            qDebug("tagfilter varch:%s", tmp + 2);
1754
#endif
1755
            taosMemoryFree(tmp);
155,111,349✔
1756
            QUERY_CHECK_CODE(code, lino, _end);
155,073,043✔
1757
          } else {
1758
            code = colDataSetVal(pColInfo, i, (const char*)&tagVal.i64, false);
82,778,652✔
1759
            QUERY_CHECK_CODE(code, lino, _end);
82,807,693✔
1760
#if TAG_FILTER_DEBUG
1761
            if (pColInfo->info.type == TSDB_DATA_TYPE_INT) {
1762
              qDebug("tagfilter int:%d", *(int*)(&tagVal.i64));
1763
            } else if (pColInfo->info.type == TSDB_DATA_TYPE_DOUBLE) {
1764
              qDebug("tagfilter double:%f", *(double*)(&tagVal.i64));
1765
            }
1766
#endif
1767
          }
1768
        }
1769
      }
1770
    }
1771
  }
1772

1773
_end:
21,750,642✔
1774
  if (code != TSDB_CODE_SUCCESS) {
21,755,440✔
1775
    blockDataDestroy(pResBlock);
4,043✔
UNCOV
1776
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
UNCOV
1777
    terrno = code;
×
UNCOV
1778
    return NULL;
×
1779
  }
1780
  return pResBlock;
21,751,397✔
1781
}
1782

1783
static int32_t doSetQualifiedUid(SArray* pUidList, const SArray* pUidTagList, SColumnInfoData* colData) {
17,804,635✔
1784
  taosArrayClear(pUidList);
17,804,635✔
1785

1786
  int32_t       numOfTables = taosArrayGetSize(pUidTagList);
17,796,965✔
1787
  bool* pResultList = colData == NULL ? NULL : (bool*)colData->pData;
17,802,567✔
1788
  for (int32_t i = 0; i < numOfTables; ++i) {
225,880,324✔
1789
    if (pResultList != NULL && !pResultList[i]) {
208,049,770✔
1790
       continue;
118,736,529✔
1791
    }
1792
    STUidTagInfo* tmpTag = (STUidTagInfo*)taosArrayGet(pUidTagList, i);
89,347,543✔
1793
    if (!tmpTag) {
89,313,233✔
1794
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1795
      return terrno;
×
1796
    }
1797
    uint64_t uid = tmpTag->uid;
89,313,233✔
1798
    qDebug("tagfilter get uid:%" PRId64, uid);
89,331,439✔
1799

1800
    void* tmp = taosArrayPush(pUidList, &uid);
89,345,854✔
1801
    if (tmp == NULL) {
89,345,854✔
1802
      return terrno;
×
1803
    }
1804
  }
1805

1806
  return TSDB_CODE_SUCCESS;
17,830,554✔
1807
}
1808

1809
static int32_t copyExistedUids(SArray* pUidTagList, const SArray* pUidList) {
17,833,861✔
1810
  int32_t code = TSDB_CODE_SUCCESS;
17,833,861✔
1811
  int32_t numOfExisted = taosArrayGetSize(pUidList);
17,833,861✔
1812
  if (numOfExisted == 0) {
17,833,437✔
1813
    return code;
14,743,290✔
1814
  }
1815

1816
  for (int32_t i = 0; i < numOfExisted; ++i) {
34,820,600✔
1817
    uint64_t* uid = taosArrayGet(pUidList, i);
31,729,614✔
1818
    if (!uid) {
31,730,038✔
1819
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1820
      return terrno;
×
1821
    }
1822
    STUidTagInfo info = {.uid = *uid};
31,730,038✔
1823
    void*        tmp = taosArrayPush(pUidTagList, &info);
31,730,877✔
1824
    if (!tmp) {
31,730,877✔
1825
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1826
      return code;
×
1827
    }
1828
  }
1829
  return code;
3,090,986✔
1830
}
1831

1832
static EDealRes getCids(SNode** pNode, void* pContext) {
18,768✔
1833
  SHashObj* colHash = (SHashObj*)pContext;
18,768✔
1834
  col_id_t colId = 0;
18,768✔
1835
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
18,768✔
1836
    colId = (*(SColumnNode**)pNode)->colId;
6,008✔
1837
  } else {
1838
    return DEAL_RES_CONTINUE;
12,760✔
1839
  }
1840

1841
  int32_t tempRes = taosHashPut(colHash, &colId, sizeof(colId), pNode, sizeof((*pNode)));
6,008✔
1842
  if (tempRes != TSDB_CODE_SUCCESS ) {
6,008✔
1843
    return DEAL_RES_ERROR;
×
1844
  }
1845

1846
  return DEAL_RES_CONTINUE;
6,008✔
1847
}
1848

1849
SNode* getTagCondNodeForQueryTmq(void* tinfo) {
3,762✔
1850
  SExecTaskInfo* pTaskInfo = (SExecTaskInfo*)tinfo;
3,762✔
1851
  const char*    id = GET_TASKID(pTaskInfo);
3,762✔
1852
  int32_t        code = 0;
3,762✔
1853

1854
  // traverse to the stream scanner node to add this table id
1855
  SOperatorInfo* pInfo = NULL;
3,762✔
1856
  code = extractOperatorInTree(pTaskInfo->pRoot, QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN, id, &pInfo);
3,762✔
1857
  if (code != 0 || pInfo == NULL) {
3,762✔
1858
    return NULL;
×
1859
  }
1860

1861
  STmqQueryScanInfo* pScanInfo = pInfo->info;
3,762✔
1862
  return pScanInfo->pTagCond;
3,762✔
1863
}
1864

1865
SNode* getTagCondNodeForStableTmq(void* node) {
9,090✔
1866
  return node == NULL ? NULL : ((SSubplan*)node)->pTagCond;
9,090✔
1867
}
1868

1869
bool checkCidInTagCondition(SNode* pTagCond, SArray* cidList) {
10,676✔
1870
  if (pTagCond == NULL) { 
10,676✔
1871
    return false;
4,668✔
1872
  }
1873
  bool result = true;
6,008✔
1874
  SHashObj *colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), true, HASH_NO_LOCK);
6,008✔
1875
  if (colHash == NULL) {
6,008✔
1876
    goto end;
×
1877
  }
1878

1879
  nodesRewriteExprPostOrder(&pTagCond, getCids, (void*)colHash);
6,008✔
1880
  if (cidList != NULL) {
6,008✔
1881
    int32_t numOfCols = taosArrayGetSize(cidList);
678✔
1882
    int32_t i = 0;
678✔
1883
    for (; i < numOfCols; ++i) {
678✔
1884
      col_id_t* pCid = taosArrayGet(cidList, i);
678✔
1885
      void* data = taosHashGet(colHash, pCid, sizeof(*pCid));
678✔
1886
      if (data != NULL) break;
678✔
1887
    }
1888
    if (i == numOfCols) { // no tag column involved in condition
678✔
1889
      result = false;
×
1890
      goto end;
×
1891
    }
1892
  }
1893

1894
end:
6,008✔
1895
  taosHashCleanup(colHash);
6,008✔
1896
  return result;
6,008✔
1897
}
1898

1899
int32_t doFilterByTagCond(int64_t suid, SArray* pUidList, int64_t version, SNode* pTagCond, void* pVnode,
299,233,942✔
1900
                                 SIdxFltStatus status, SStorageAPI* pAPI, void* pStreamInfo) {
1901
  if (pTagCond == NULL) {
299,233,942✔
1902
    return TSDB_CODE_SUCCESS;
281,409,948✔
1903
  }
1904

1905
  terrno = TSDB_CODE_SUCCESS;
17,824,000✔
1906

1907
  int32_t      lino = 0;
17,832,449✔
1908
  int32_t      code = TSDB_CODE_SUCCESS;
17,832,449✔
1909
  SArray*      pBlockList = NULL;
17,832,449✔
1910
  SSDataBlock* pResBlock = NULL;
17,832,449✔
1911
  SScalarParam output = {0};
17,832,228✔
1912
  SArray*      pUidTagList = NULL;
17,832,850✔
1913

1914
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
17,832,850✔
1915

1916
  pUidTagList = taosArrayInit(10, sizeof(STUidTagInfo));
17,834,664✔
1917
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
17,822,910✔
1918

1919
  // pUidList size is 0 if tagIndex is NULL and in query condition
1920
  code = copyExistedUids(pUidTagList, pUidList);
17,822,910✔
1921
  QUERY_CHECK_CODE(code, lino, end);
17,824,354✔
1922

1923
  // Narrow down the scope of the tablelist set if there is tbname in condition and And Logical operator
1924
  code = optimizeTbnameInCond(pVnode, suid, pUidTagList, pTagCond, pAPI, version);
17,824,354✔
1925
  if (code == 0) {
17,831,430✔
1926
    if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
443,603✔
1927
      goto end;
434,663✔
1928
    }
1929
  } else {
1930
    qDebug("pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
17,387,827✔
1931

1932
    FilterCondType condType = checkTagCond(pTagCond);
17,388,397✔
1933
    if (((condType == FILTER_NO_LOGIC || condType == FILTER_AND) && status != SFLT_NOT_INDEX) || // (super table) use tagIndex and operator is and
17,389,240✔
1934
        (status == SFLT_NOT_INDEX && taosArrayGetSize(pUidTagList) > 0)) {                       // (child table with tagCond)
17,391,709✔
1935
      code = pAPI->metaFn.getTableTagsByUidVersion(pVnode, suid, pUidTagList, version);
3,088,787✔
1936
    } else {
1937
      taosArrayClearEx(pUidTagList, freeItem);       // clear tablelist if using tagIndex and or condition
14,299,231✔
1938
      code = pAPI->metaFn.getTableTags(pVnode, suid, pUidTagList);
14,298,538✔
1939
    }
1940
    if (code != TSDB_CODE_SUCCESS) {
17,386,225✔
1941
      qError("failed to get table tags from meta, reason:%s, suid:%" PRIu64, tstrerror(code),suid);
×
1942
      terrno = code;
×
1943
      QUERY_CHECK_CODE(code, lino, end);
×
1944
    }
1945
  }
1946

1947
  qDebug("final pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
17,395,165✔
1948

1949
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
17,399,175✔
1950
  if (numOfTables == 0) {
17,401,423✔
1951
    goto end;
2,737,068✔
1952
  }
1953

1954
  SArray* pColList = NULL;
14,664,355✔
1955
  code = qGetColumnsFromNodeList(pTagCond, false, &pColList); 
14,665,432✔
1956
  if (code != TSDB_CODE_SUCCESS) {
14,642,116✔
1957
    goto end;
×
1958
  }
1959
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
14,642,116✔
1960
  taosArrayDestroy(pColList);
14,659,027✔
1961
  if (pResBlock == NULL) {
14,657,906✔
1962
    code = terrno;
×
1963
    QUERY_CHECK_CODE(code, lino, end);
×
1964
  }
1965

1966
  //fprintDataBlock(pResBlock, "tagFilter", "", 0);
1967

1968
  //  int64_t st1 = taosGetTimestampUs();
1969
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1970
  pBlockList = taosArrayInit(2, POINTER_BYTES);
14,657,906✔
1971
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
14,662,360✔
1972

1973
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
14,661,215✔
1974
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
14,661,215✔
1975

1976
  code = createResultData(&type, numOfTables, &output);
14,661,215✔
1977
  if (code != TSDB_CODE_SUCCESS) {
14,650,645✔
1978
    terrno = code;
×
1979
    QUERY_CHECK_CODE(code, lino, end);
×
1980
  }
1981

1982
  gTaskScalarExtra.pStreamInfo = pStreamInfo;
14,650,645✔
1983
  gTaskScalarExtra.pStreamRange = NULL;
14,650,645✔
1984
  code = scalarCalculate(pTagCond, pBlockList, &output, &gTaskScalarExtra);
14,649,706✔
1985
  if (code != TSDB_CODE_SUCCESS) {
14,628,359✔
1986
    qError("failed to calculate scalar, reason:%s", tstrerror(code));
1,108✔
1987
    terrno = code;
1,108✔
1988
    QUERY_CHECK_CODE(code, lino, end);
1,108✔
1989
  }
1990

1991
end:
17,800,090✔
1992
  if (code == 0) {
17,821,182✔
1993
    code = doSetQualifiedUid(pUidList, pUidTagList, output.columnData);
17,821,244✔
1994
    lino = __LINE__;
17,830,797✔
1995
  }
1996
  
1997
  if (code != 0) {
17,830,735✔
1998
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1,108✔
1999
  }
2000
  blockDataDestroy(pResBlock);
17,830,735✔
2001
  taosArrayDestroy(pBlockList);
17,808,104✔
2002
  taosArrayDestroyEx(pUidTagList, freeItem);
17,807,429✔
2003

2004
  colDataDestroy(output.columnData);
17,828,247✔
2005
  taosMemoryFreeClear(output.columnData);
17,822,158✔
2006
  return code;
17,827,019✔
2007
}
2008

2009
typedef struct {
2010
  int32_t code;
2011
  SStreamRuntimeFuncInfo* pStreamRuntimeInfo;
2012
} PlaceHolderContext;
2013

2014
static EDealRes replacePlaceHolderColumn(SNode** pNode, void* pContext) {
76,912✔
2015
  PlaceHolderContext* pData = (PlaceHolderContext*)pContext;
76,912✔
2016
  if (QUERY_NODE_FUNCTION != nodeType((*pNode))) {
76,912✔
2017
    return DEAL_RES_CONTINUE;
63,248✔
2018
  }
2019
  SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
13,664✔
2020
  if (!fmIsStreamPesudoColVal(pFuncNode->funcId)) {
13,664✔
2021
    return DEAL_RES_CONTINUE;
640✔
2022
  }
2023
  pData->code = fmSetStreamPseudoFuncParamVal(pFuncNode->funcId, pFuncNode->pParameterList, pData->pStreamRuntimeInfo);
13,024✔
2024
  if (pData->code != TSDB_CODE_SUCCESS) {
13,024✔
2025
    return DEAL_RES_ERROR;
×
2026
  }
2027
  SNode* pFirstParam = nodesListGetNode(pFuncNode->pParameterList, 0);
13,024✔
2028
  ((SValueNode*)pFirstParam)->translate = true;
13,024✔
2029
  SValueNode* res = NULL;
13,024✔
2030
  pData->code = nodesCloneNode(pFirstParam, (SNode**)&res);
13,024✔
2031
  if (NULL == res) {
13,024✔
2032
    return DEAL_RES_ERROR;
×
2033
  }
2034
  nodesDestroyNode(*pNode);
13,024✔
2035
  *pNode = (SNode*)res;
13,024✔
2036

2037
  return DEAL_RES_CONTINUE;
13,024✔
2038
}
2039

2040
static void extractTagColId(SOperatorNode* pOpNode, SArray* pColIdArray) {
20,984✔
2041
  SNode* pLeft = pOpNode->pLeft;
20,984✔
2042
  SNode* pRight = pOpNode->pRight;
20,984✔
2043
  SColumnNode* pColNode = nodeType(pLeft) == QUERY_NODE_COLUMN ?
20,984✔
2044
    (SColumnNode*)pLeft : (SColumnNode*)pRight;
20,984✔
2045

2046
  col_id_t colId = pColNode->colId;
20,984✔
2047
  void* _tmp = taosArrayPush(pColIdArray, &colId);
20,984✔
2048
}
20,984✔
2049

2050
static int32_t buildTagCondKey(
10,492✔
2051
  const SNode* pTagCond, char** pTagCondKey,
2052
  int32_t* tagCondKeyLen, SArray** pTagColIds) {
2053
  if (NULL == pTagCond ||
10,492✔
2054
    (nodeType(pTagCond) != QUERY_NODE_OPERATOR &&
10,492✔
2055
      nodeType(pTagCond) != QUERY_NODE_LOGIC_CONDITION)) {
10,492✔
2056
    qError("invalid parameter to extract tag filter symbol");
×
2057
    return TSDB_CODE_INTERNAL_ERROR;
×
2058
  }
2059
  int32_t code = TSDB_CODE_SUCCESS;
10,492✔
2060
  int32_t lino = 0;
10,492✔
2061
  *pTagColIds = taosArrayInit(4, sizeof(col_id_t));
10,492✔
2062

2063
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
10,492✔
2064
    extractTagColId((SOperatorNode*)pTagCond, *pTagColIds);
×
2065
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
10,492✔
2066
    SNode* pChild = NULL;
10,492✔
2067
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
31,476✔
2068
      extractTagColId((SOperatorNode*)pChild, *pTagColIds);
20,984✔
2069
    }
2070
  }
2071

2072
  taosArraySort(*pTagColIds, compareUint16Val);
10,492✔
2073

2074
  // encode ordered colIds into key string, separated by ','
2075
  *tagCondKeyLen =
20,984✔
2076
    (int32_t)(taosArrayGetSize(*pTagColIds) * (sizeof(col_id_t) + 1) - 1);
10,492✔
2077
  *pTagCondKey = (char*)taosMemoryCalloc(1, *tagCondKeyLen);
10,492✔
2078
  TSDB_CHECK_NULL(*pTagCondKey, code, lino, _end, terrno);
10,492✔
2079
  char* pStart = *pTagCondKey;
10,492✔
2080
  for (int32_t i = 0; i < taosArrayGetSize(*pTagColIds); ++i) {
31,476✔
2081
    col_id_t* pColId = (col_id_t*)taosArrayGet(*pTagColIds, i);
20,984✔
2082
    TSDB_CHECK_NULL(pColId, code, lino, _end, terrno);
20,984✔
2083
    memcpy(pStart, pColId, sizeof(col_id_t));
20,984✔
2084
    pStart += sizeof(col_id_t);
20,984✔
2085
    if (i != taosArrayGetSize(*pTagColIds) - 1) {
20,984✔
2086
      *pStart = ',';
10,492✔
2087
      pStart += 1;
10,492✔
2088
    }
2089
  }
2090

2091
_end:
10,492✔
2092
  if (TSDB_CODE_SUCCESS != code) {
10,492✔
2093
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2094
    terrno = code;
×
2095
  }
2096
  return code;
10,492✔
2097
}
2098

2099
static EDealRes canOptimizeTagCondFilter(SNode* pTagCond, void* pContext) {
97,696✔
2100
  if (NULL == pTagCond) {
97,696✔
2101
    *(bool*)pContext = false;
×
2102
    return DEAL_RES_END;
×
2103
  }
2104
  if (nodeType(pTagCond) == QUERY_NODE_VALUE ||
97,696✔
2105
    nodeType(pTagCond) == QUERY_NODE_COLUMN) {
64,844✔
2106
    return DEAL_RES_CONTINUE;
53,836✔
2107
  }
2108
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR &&
43,860✔
2109
    ((SOperatorNode*)pTagCond)->opType == OP_TYPE_EQUAL) {
21,500✔
2110
    return DEAL_RES_CONTINUE;
20,984✔
2111
  }
2112
  if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION &&
22,876✔
2113
    ((SLogicConditionNode*)pTagCond)->condType == LOGIC_COND_TYPE_AND) {
10,492✔
2114
    return DEAL_RES_CONTINUE;
10,492✔
2115
  }
2116
  if (nodeType(pTagCond) == QUERY_NODE_FUNCTION &&
24,252✔
2117
    fmIsStreamPesudoColVal(((SFunctionNode*)pTagCond)->funcId)) {
11,868✔
2118
    return DEAL_RES_CONTINUE;
11,868✔
2119
  }
2120
  *(bool*)pContext = false;
516✔
2121
  return DEAL_RES_END;
516✔
2122
}
2123

2124
int32_t getTableList(void* pVnode, SScanPhysiNode* pScanNode, SNode* pTagCond, SNode* pTagIndexCond,
299,089,778✔
2125
                     STableListInfo* pListInfo, uint8_t* digest, const char* idstr, SStorageAPI* pStorageAPI, void* pStreamInfo) {
2126
  int32_t code = TSDB_CODE_SUCCESS;
299,089,778✔
2127
  int32_t lino = 0;
299,089,778✔
2128
  size_t  numOfTables = 0;
299,089,778✔
2129

2130
  pListInfo->idInfo.suid = pScanNode->suid;
299,089,778✔
2131
  pListInfo->idInfo.tableType = pScanNode->tableType;
299,026,289✔
2132

2133
  SArray* pUidList = taosArrayInit(8, sizeof(uint64_t));
299,060,888✔
2134
  QUERY_CHECK_NULL(pUidList, code, lino, end, terrno);
298,981,719✔
2135

2136
  char*   pTagCondKey = NULL;
298,981,719✔
2137
  int32_t tagCondKeyLen;
298,992,056✔
2138
  SArray* pTagColIds = NULL;
298,963,808✔
2139
  char*   pPayload = NULL;
298,972,766✔
2140
  SIdxFltStatus status = SFLT_NOT_INDEX;
298,972,766✔
2141

2142
  qTrace("getTableList called, suid:%" PRIu64
298,972,766✔
2143
    ", tagCond:%p, tagIndexCond:%p, %d %d", pScanNode->suid, pTagCond,
2144
    pTagIndexCond, pScanNode->tableType, pScanNode->virtualStableScan);
2145
  if (pScanNode->tableType != TSDB_SUPER_TABLE && !pScanNode->virtualStableScan) {
298,975,481✔
2146
    pListInfo->idInfo.uid = pScanNode->uid;
166,659,172✔
2147
    if (pStorageAPI->metaFn.isTableExisted(pVnode, pScanNode->uid)) {
166,710,492✔
2148
      void* tmp = taosArrayPush(pUidList, &pScanNode->uid);
166,671,763✔
2149
      QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
166,672,271✔
2150
    }
2151
    code = doFilterByTagCond(pScanNode->suid, pUidList, -1, pTagCond, pVnode, status, pStorageAPI, pStreamInfo);
166,722,287✔
2152
    QUERY_CHECK_CODE(code, lino, end);
166,715,709✔
2153
  } else {
2154
    bool      isStream = (pStreamInfo != NULL);
132,511,411✔
2155
    bool      hasTagCond = (pTagCond != NULL);
132,511,411✔
2156
    bool      canCacheTagEqCondFilter = false;
132,511,411✔
2157
    T_MD5_CTX context = {0};
132,386,056✔
2158

2159
    qTrace("start to get table list by tag filter, suid:%" PRIu64
132,325,816✔
2160
      ",tsStableTagFilterCache:%d, tsTagFilterCache:%d", 
2161
      pScanNode->suid, tsStableTagFilterCache, tsTagFilterCache);
2162

2163
    bool acquired = false;
132,325,816✔
2164
    // first, check whether we can use stable tag filter cache
2165
    if (tsStableTagFilterCache && isStream && hasTagCond) {
132,388,496✔
2166
      canCacheTagEqCondFilter = true;
11,008✔
2167
      nodesWalkExpr(pTagCond, canOptimizeTagCondFilter,
11,008✔
2168
        (void*)&canCacheTagEqCondFilter);
2169
    }
2170
    if (canCacheTagEqCondFilter) {
132,123,290✔
2171
      qDebug("%s, stable tag filter condition can be optimized", idstr);
10,492✔
2172
      if (((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
10,492✔
2173
        SNode* tmp = NULL;
10,492✔
2174
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
10,492✔
2175
        QUERY_CHECK_CODE(code, lino, end);
10,492✔
2176

2177
        PlaceHolderContext ctx = {.code = TSDB_CODE_SUCCESS, .pStreamRuntimeInfo = (SStreamRuntimeFuncInfo*)pStreamInfo};
10,492✔
2178
        nodesRewriteExpr(&tmp, replacePlaceHolderColumn, (void*)&ctx);
10,492✔
2179
        if (TSDB_CODE_SUCCESS != ctx.code) {
10,492✔
2180
          nodesDestroyNode(tmp);
×
2181
          code = ctx.code;
×
2182
          goto end;
×
2183
        }
2184
        code = genStableTagFilterDigest(tmp, &context);
10,492✔
2185
        nodesDestroyNode(tmp);
10,492✔
2186
      } else {
2187
        code = genStableTagFilterDigest(pTagCond, &context);
×
2188
      }
2189
      QUERY_CHECK_CODE(code, lino, end);
10,492✔
2190

2191
      code = buildTagCondKey(
10,492✔
2192
        pTagCond, &pTagCondKey, &tagCondKeyLen, &pTagColIds);
2193
      QUERY_CHECK_CODE(code, lino, end);
10,492✔
2194
      code = pStorageAPI->metaFn.getStableCachedTableList(
10,492✔
2195
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
10,492✔
2196
        context.digest, tListLen(context.digest), pUidList, &acquired);
2197
      QUERY_CHECK_CODE(code, lino, end);
10,492✔
2198
    } else if (tsTagFilterCache) {
132,112,798✔
2199
      // second, try to use normal tag filter cache
2200
      qDebug("%s using normal tag filter cache", idstr);
32,936✔
2201
      if (pStreamInfo != NULL && ((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
34,092✔
2202
        SNode* tmp = NULL;
1,156✔
2203
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
1,156✔
2204
        QUERY_CHECK_CODE(code, lino, end);
1,156✔
2205

2206
        PlaceHolderContext ctx = {.code = TSDB_CODE_SUCCESS, .pStreamRuntimeInfo = (SStreamRuntimeFuncInfo*)pStreamInfo};
1,156✔
2207
        nodesRewriteExpr(&tmp, replacePlaceHolderColumn, (void*)&ctx);
1,156✔
2208
        if (TSDB_CODE_SUCCESS != ctx.code) {
1,156✔
2209
          nodesDestroyNode(tmp);
×
2210
          code = ctx.code;
×
2211
          goto end;
×
2212
        }
2213
        code = genTagFilterDigest(tmp, &context);
1,156✔
2214
        nodesDestroyNode(tmp);
1,156✔
2215
      } else {
2216
        code = genTagFilterDigest(pTagCond, &context);
31,780✔
2217
      }
2218
      // try to retrieve the result from meta cache
2219
      QUERY_CHECK_CODE(code, lino, end);      
32,936✔
2220
      code = pStorageAPI->metaFn.getCachedTableList(
32,936✔
2221
        pVnode, pScanNode->suid, context.digest,
32,936✔
2222
        tListLen(context.digest), pUidList, &acquired);
2223
      QUERY_CHECK_CODE(code, lino, end);
286,508✔
2224
    }
2225
    if (acquired) {
132,376,862✔
2226
      taosArrayDestroy(pTagColIds);
30,376✔
2227
      pTagColIds = NULL;
30,204✔
2228
      
2229
      if (digest != NULL) {
30,204✔
2230
        digest[0] = 1;
30,376✔
2231
        memcpy(digest + 1, context.digest, tListLen(context.digest));
30,376✔
2232
      }
2233
      
2234
      qDebug("suid:%" PRIu64 ", %s retrieve table uid list from cache, numOfTables:%d", 
30,204✔
2235
        pScanNode->suid, idstr, (int32_t)taosArrayGetSize(pUidList));
2236
      goto end;
30,376✔
2237
    } else {
2238
      qDebug("suid:%" PRIu64 
132,346,486✔
2239
        ", failed to get table uid list from cache", pScanNode->suid);
2240
    }
2241

2242
    if (!pTagCond) {  // no tag filter condition exists, let's fetch all tables of this super table
132,423,299✔
2243
      code = pStorageAPI->metaFn.getChildTableList(pVnode, pScanNode->suid, pUidList);
114,981,698✔
2244
      QUERY_CHECK_CODE(code, lino, end);
114,959,404✔
2245
      qTrace("no tag filter, get all child tables, numOfTables:%d", (int32_t)taosArrayGetSize(pUidList));
114,959,404✔
2246
    } else {
2247
      SIdxFltStatus status = SFLT_NOT_INDEX;
17,441,601✔
2248
      if (pTagIndexCond) {
17,444,890✔
2249
        void* pIndex = pStorageAPI->metaFn.getInvertIndex(pVnode);
4,990,976✔
2250

2251
        SIndexMetaArg metaArg = {.metaEx = pVnode,
4,991,004✔
2252
                                 .idx = pStorageAPI->metaFn.storeGetIndexInfo(pVnode),
4,990,976✔
2253
                                 .ivtIdx = pIndex,
2254
                                 .suid = pScanNode->uid};
4,990,976✔
2255
        code = doFilterTag(pTagIndexCond, &metaArg, pUidList, &status, &pStorageAPI->metaFilter);
4,990,976✔
2256
        if (code != 0 || status == SFLT_NOT_INDEX) {  // temporarily disable it for performance sake
4,972,451✔
2257
          qDebug("failed to get tableIds from index, suid:%" PRIu64 ", uidListSize:%d", pScanNode->uid, (int32_t)taosArrayGetSize(pUidList));
1,380,535✔
2258
        } else {
2259
          qDebug("succ to get filter result, table num: %d", (int)taosArrayGetSize(pUidList));
3,591,916✔
2260
        }
2261
      }
2262
    }
2263
    qTrace("after index filter, pTagCond:%p uidListSize:%d", pTagCond, (int32_t)taosArrayGetSize(pUidList));
132,412,881✔
2264
    code = doFilterByTagCond(pScanNode->suid, pUidList, -1, pTagCond, pVnode, status, pStorageAPI, pStreamInfo);
132,412,881✔
2265
    QUERY_CHECK_CODE(code, lino, end);
132,407,998✔
2266

2267
    // let's add the filter results into meta-cache
2268
    numOfTables = taosArrayGetSize(pUidList);
132,406,890✔
2269

2270
    if (canCacheTagEqCondFilter) {
132,389,315✔
2271
      qInfo("%s, suid:%" PRIu64 ", add uid list to stable tag filter cache, "
4,816✔
2272
            "uidListSize:%d, origin key:%" PRIu64 ",%" PRIu64,
2273
            idstr, pScanNode->suid, (int32_t)numOfTables,
2274
            *(uint64_t*)context.digest, *(uint64_t*)(context.digest + 8));
2275

2276
      code = pStorageAPI->metaFn.putStableCachedTableList(
4,816✔
2277
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
2278
        context.digest, tListLen(context.digest),
2279
        pUidList, &pTagColIds);
2280
      QUERY_CHECK_CODE(code, lino, end);
4,816✔
2281

2282
      if (digest != NULL) {
4,816✔
2283
        digest[0] = 1;
4,816✔
2284
        memcpy(digest + 1, context.digest, tListLen(context.digest));
4,816✔
2285
      }
2286
    } else if (tsTagFilterCache) {
132,384,499✔
2287
      qInfo("%s, suid:%" PRIu64 ", add uid list to normal tag filter cache, "
8,236✔
2288
            "uidListSize:%d, origin key:%" PRIu64 ",%" PRIu64,
2289
            idstr, pScanNode->suid, (int32_t)numOfTables,
2290
            *(uint64_t*)context.digest, *(uint64_t*)(context.digest + 8));
2291
      size_t size = numOfTables * sizeof(uint64_t) + sizeof(int32_t);
8,236✔
2292
      pPayload = taosMemoryMalloc(size);
8,236✔
2293
      QUERY_CHECK_NULL(pPayload, code, lino, end, terrno);
8,236✔
2294

2295
      *(int32_t*)pPayload = (int32_t)numOfTables;
8,236✔
2296
      if (numOfTables > 0) {
8,236✔
2297
        void* tmp = taosArrayGet(pUidList, 0);
6,736✔
2298
        QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
6,736✔
2299
        memcpy(pPayload + sizeof(int32_t), tmp, numOfTables * sizeof(uint64_t));
6,736✔
2300
      }
2301

2302
      code = pStorageAPI->metaFn.putCachedTableList(pVnode, pScanNode->suid,
8,236✔
2303
                                                    context.digest,
2304
                                                    tListLen(context.digest),
2305
                                                    pPayload, size, 1);
2306
      if (TSDB_CODE_SUCCESS == code) {
8,236✔
2307
        /*
2308
          data referenced by pPayload is used in lru cache,
2309
          reset pPayload to NULL to avoid being freed in _error block
2310
        */
2311
        pPayload = NULL;
8,064✔
2312
      } else {
2313
        if (TSDB_CODE_DUP_KEY == code) {
172✔
2314
          /*
2315
            another thread has already put the same key into cache,
2316
            we can just ignore this error
2317
          */
2318
          code = TSDB_CODE_SUCCESS;
172✔
2319
        }
2320
        QUERY_CHECK_CODE(code, lino, end);
172✔
2321
      }
2322

2323
      if (digest != NULL) {
8,236✔
2324
        digest[0] = 1;
8,236✔
2325
        memcpy(digest + 1, context.digest, tListLen(context.digest));
8,236✔
2326
      }
2327
    }
2328
  }
2329

2330
  qDebug("%s, table list with %d uids built", idstr, (int32_t)numOfTables);
299,100,637✔
2331

2332
end:
299,155,158✔
2333
  if (code == 0) {
299,150,008✔
2334
     STableKeyInfo info = {.uid = 0, .groupId = 0};
299,148,900✔
2335
    for (int32_t i = 0; i < taosArrayGetSize(pUidList); ++i) {
1,071,727,795✔
2336
      int64_t* uid = (int64_t*)taosArrayGet(pUidList, i);
772,474,427✔
2337
      QUERY_CHECK_NULL(uid, code, lino, end, terrno);
772,545,344✔
2338

2339
      info.uid = *uid;
772,545,344✔
2340
        //qInfo("doSetQualifiedUid row:%d added to pTableList", i);
2341
      void* p = taosArrayPush(pListInfo->pTableList, &info);
772,518,244✔
2342
      QUERY_CHECK_NULL(p, code, lino, end, terrno);
772,567,951✔
2343
    }
2344
  } else {
2345
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1,108✔
2346
  }
2347
  taosArrayDestroy(pUidList);
299,211,828✔
2348
  taosArrayDestroy(pTagColIds);
299,078,757✔
2349
  taosMemFreeClear(pTagCondKey);
299,101,437✔
2350
  taosMemFreeClear(pPayload);
299,101,437✔
2351
  return code;
299,101,437✔
2352
}
2353

2354
int32_t qGetTableList(int64_t suid, void* pVnode, void* node, SArray** tableList, void* pTaskInfo) {
28,972✔
2355
  int32_t        code = TSDB_CODE_SUCCESS;
28,972✔
2356
  int32_t        lino = 0;
28,972✔
2357
  SSubplan*      pSubplan = (SSubplan*)node;
28,972✔
2358
  SScanPhysiNode pNode = {0};
28,972✔
2359
  pNode.suid = suid;
28,972✔
2360
  pNode.uid = suid;
28,972✔
2361
  pNode.tableType = TSDB_SUPER_TABLE;
28,972✔
2362

2363
  STableListInfo* pTableListInfo = tableListCreate();
28,972✔
2364
  QUERY_CHECK_NULL(pTableListInfo, code, lino, _end, terrno);
28,972✔
2365
  code = getTableList(pVnode, &pNode, pSubplan ? pSubplan->pTagCond : NULL, pSubplan ? pSubplan->pTagIndexCond : NULL,
28,972✔
2366
                      pTableListInfo, NULL, "qGetTableList", &((SExecTaskInfo*)pTaskInfo)->storageAPI, NULL);
2367
  QUERY_CHECK_CODE(code, lino, _end);
28,972✔
2368
  *tableList = pTableListInfo->pTableList;
28,972✔
2369
  pTableListInfo->pTableList = NULL;
28,972✔
2370

2371
_end:
28,972✔
2372
  tableListDestroy(pTableListInfo);
28,972✔
2373
  if (code != TSDB_CODE_SUCCESS) {
28,972✔
2374
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2375
  }
2376
  return code;
28,972✔
2377
}
2378

2379
size_t getTableTagsBufLen(const SNodeList* pGroups) {
×
2380
  size_t keyLen = 0;
×
2381

2382
  SNode* node;
2383
  FOREACH(node, pGroups) {
×
2384
    SExprNode* pExpr = (SExprNode*)node;
×
2385
    keyLen += pExpr->resType.bytes;
×
2386
  }
2387

2388
  keyLen += sizeof(int8_t) * LIST_LENGTH(pGroups);
×
2389
  return keyLen;
×
2390
}
2391

2392
int32_t getGroupIdFromTagsVal(void* pVnode, uint64_t uid, SNodeList* pGroupNode, char* keyBuf, uint64_t* pGroupId,
×
2393
                              SStorageAPI* pAPI) {
2394
  SMetaReader mr = {0};
×
2395

2396
  pAPI->metaReaderFn.initReader(&mr, pVnode, META_READER_LOCK, &pAPI->metaFn);
×
2397
  if (pAPI->metaReaderFn.getEntryGetUidCache(&mr, uid) != 0) {  // table not exist
×
2398
    pAPI->metaReaderFn.clearReader(&mr);
×
2399
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
2400
  }
2401

2402
  SNodeList* groupNew = NULL;
×
2403
  int32_t    code = nodesCloneList(pGroupNode, &groupNew);
×
2404
  if (TSDB_CODE_SUCCESS != code) {
×
2405
    pAPI->metaReaderFn.clearReader(&mr);
×
2406
    return code;
×
2407
  }
2408

2409
  STransTagExprCtx ctx = {.code = 0, .pReader = &mr};
×
2410
  nodesRewriteExprsPostOrder(groupNew, doTranslateTagExpr, &ctx);
×
2411
  if (TSDB_CODE_SUCCESS != ctx.code) {
×
2412
    nodesDestroyList(groupNew);
×
2413
    pAPI->metaReaderFn.clearReader(&mr);
×
2414
    return code;
×
2415
  }
2416
  char* isNull = (char*)keyBuf;
×
2417
  char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(pGroupNode);
×
2418

2419
  SNode*  pNode;
2420
  int32_t index = 0;
×
2421
  FOREACH(pNode, groupNew) {
×
2422
    SNode*  pNew = NULL;
×
2423
    int32_t code = scalarCalculateConstants(pNode, &pNew);
×
2424
    if (TSDB_CODE_SUCCESS == code) {
×
2425
      REPLACE_NODE(pNew);
×
2426
    } else {
2427
      nodesDestroyList(groupNew);
×
2428
      pAPI->metaReaderFn.clearReader(&mr);
×
2429
      return code;
×
2430
    }
2431

2432
    if (nodeType(pNew) != QUERY_NODE_VALUE) {
×
2433
      nodesDestroyList(groupNew);
×
2434
      pAPI->metaReaderFn.clearReader(&mr);
×
2435
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2436
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2437
    }
2438
    SValueNode* pValue = (SValueNode*)pNew;
×
2439

2440
    if (pValue->node.resType.type == TSDB_DATA_TYPE_NULL || pValue->isNull) {
×
2441
      isNull[index++] = 1;
×
2442
      continue;
×
2443
    } else {
2444
      isNull[index++] = 0;
×
2445
      char* data = nodesGetValueFromNode(pValue);
×
2446
      if (pValue->node.resType.type == TSDB_DATA_TYPE_JSON) {
×
2447
        if (tTagIsJson(data)) {
×
2448
          terrno = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
×
2449
          nodesDestroyList(groupNew);
×
2450
          pAPI->metaReaderFn.clearReader(&mr);
×
2451
          return terrno;
×
2452
        }
2453
        int32_t len = getJsonValueLen(data);
×
2454
        memcpy(pStart, data, len);
×
2455
        pStart += len;
×
2456
      } else if (IS_VAR_DATA_TYPE(pValue->node.resType.type)) {
×
2457
        if (IS_STR_DATA_BLOB(pValue->node.resType.type)) {
×
2458
          return TSDB_CODE_BLOB_NOT_SUPPORT_TAG;
×
2459
        }
2460
        memcpy(pStart, data, varDataTLen(data));
×
2461
        pStart += varDataTLen(data);
×
2462
      } else {
2463
        memcpy(pStart, data, pValue->node.resType.bytes);
×
2464
        pStart += pValue->node.resType.bytes;
×
2465
      }
2466
    }
2467
  }
2468

2469
  int32_t len = (int32_t)(pStart - (char*)keyBuf);
×
2470
  *pGroupId = calcGroupId(keyBuf, len);
×
2471

2472
  nodesDestroyList(groupNew);
×
2473
  pAPI->metaReaderFn.clearReader(&mr);
×
2474

2475
  return TSDB_CODE_SUCCESS;
×
2476
}
2477

2478
SArray* makeColumnArrayFromList(SNodeList* pNodeList) {
3,495,408✔
2479
  if (!pNodeList) {
3,495,408✔
2480
    return NULL;
×
2481
  }
2482

2483
  size_t  numOfCols = LIST_LENGTH(pNodeList);
3,495,408✔
2484
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
3,495,401✔
2485
  if (pList == NULL) {
3,493,598✔
2486
    return NULL;
×
2487
  }
2488

2489
  for (int32_t i = 0; i < numOfCols; ++i) {
8,057,478✔
2490
    SColumnNode* pColNode = (SColumnNode*)nodesListGetNode(pNodeList, i);
4,565,388✔
2491
    if (!pColNode) {
4,565,733✔
2492
      taosArrayDestroy(pList);
×
2493
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2494
      return NULL;
×
2495
    }
2496

2497
    // todo extract method
2498
    SColumn c = {0};
4,565,733✔
2499
    c.slotId = pColNode->slotId;
4,565,388✔
2500
    c.colId = pColNode->colId;
4,566,378✔
2501
    c.type = pColNode->node.resType.type;
4,567,257✔
2502
    c.bytes = pColNode->node.resType.bytes;
4,564,242✔
2503
    c.precision = pColNode->node.resType.precision;
4,561,729✔
2504
    c.scale = pColNode->node.resType.scale;
4,566,820✔
2505

2506
    void* tmp = taosArrayPush(pList, &c);
4,566,128✔
2507
    if (!tmp) {
4,566,128✔
2508
      taosArrayDestroy(pList);
×
2509
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2510
      return NULL;
×
2511
    }
2512
  }
2513

2514
  return pList;
3,492,090✔
2515
}
2516

2517
int32_t extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols,
383,861,506✔
2518
                            int32_t type, SColMatchInfo* pMatchInfo) {
2519
  size_t  numOfCols = LIST_LENGTH(pNodeList);
383,861,506✔
2520
  int32_t code = TSDB_CODE_SUCCESS;
383,884,916✔
2521
  int32_t lino = 0;
383,884,916✔
2522
  bool    colIdOrdered = true;
383,884,916✔
2523
  bool    hasPrevColId = false;
383,884,916✔
2524
  int32_t prevColId = 0;
383,884,916✔
2525

2526
  pMatchInfo->matchType = type;
383,884,916✔
2527

2528
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchItem));
383,833,718✔
2529
  if (pList == NULL) {
383,735,499✔
2530
    code = terrno;
×
2531
    return code;
×
2532
  }
2533

2534
  SColMatchItem** infoBySlot = NULL;
383,735,499✔
2535
  if (numOfCols > 0) {
383,735,499✔
2536
    infoBySlot = taosMemoryCalloc(numOfCols, sizeof(*infoBySlot));
383,621,975✔
2537
    if (infoBySlot == NULL) {
383,692,370✔
2538
      code = terrno;
×
2539
      goto _end;
×
2540
    }
2541
  }
2542

2543
  SNode* node = NULL;
383,805,894✔
2544
  FOREACH(node, pNodeList) {
1,857,710,961✔
2545
    STargetNode* pNode = (STargetNode*)node;
1,473,752,176✔
2546
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,473,752,176✔
2547
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
1,473,752,176✔
2548
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
1,461,483,326✔
2549

2550
      SColMatchItem c = {.needOutput = true};
1,461,475,479✔
2551
      c.colId = pColNode->colId;
1,461,475,552✔
2552
      c.srcSlotId = pColNode->slotId;
1,461,545,170✔
2553
      c.dstSlotId = pNode->slotId;
1,461,510,444✔
2554
      c.isPk = pColNode->isPk;
1,461,488,306✔
2555
      c.dataType = pColNode->node.resType;
1,461,469,284✔
2556
      void* tmp = taosArrayPush(pList, &c);
1,461,597,432✔
2557
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,461,597,432✔
2558
      if (hasPrevColId) {
1,461,597,432✔
2559
        if (c.colId < prevColId) {
1,078,115,884✔
2560
          colIdOrdered = false;
16,145,850✔
2561
        }
2562
      } else {
2563
        hasPrevColId = true;
383,481,548✔
2564
      }
2565
      prevColId = c.colId;
1,461,597,432✔
2566
      if (pNode->slotId >= 0 && pNode->slotId < numOfCols) {
1,461,597,432✔
2567
        infoBySlot[pNode->slotId] = (SColMatchItem*)taosArrayGet(pList, taosArrayGetSize(pList) - 1);
1,460,917,205✔
2568
      }
2569
    }
2570
  }
2571
  // set the output flag for each column in SColMatchInfo, according to the
2572
  *numOfOutputCols = 0;
383,913,689✔
2573
  SNode* slotNode = NULL;
383,917,585✔
2574
  FOREACH(slotNode, pOutputNodeList->pSlots) {
2,060,902,736✔
2575
    SSlotDescNode* pNode = (SSlotDescNode*)slotNode;
1,677,034,594✔
2576
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,677,034,594✔
2577

2578
    // todo: add reserve flag check
2579
    // it is a column reserved for the arithmetic expression calculation
2580
    if (pNode->slotId >= numOfCols) {
1,677,034,594✔
2581
      (*numOfOutputCols) += 1;
203,461,319✔
2582
      continue;
203,471,208✔
2583
    }
2584

2585
    SColMatchItem* info = infoBySlot ? infoBySlot[pNode->slotId] : NULL;
1,473,731,755✔
2586

2587
    if (pNode->output) {
1,473,762,336✔
2588
      (*numOfOutputCols) += 1;
1,451,427,805✔
2589
    } else if (info != NULL) {
22,363,430✔
2590
      // select distinct tbname from stb where tbname='abc';
2591
      info->needOutput = false;
22,333,504✔
2592
    }
2593
  }
2594

2595
  pMatchInfo->pList = pList;
383,845,209✔
2596
  pMatchInfo->colIdOrdered = colIdOrdered;
383,847,532✔
2597

2598
_end:
383,863,671✔
2599
  taosMemoryFree(infoBySlot);
383,863,671✔
2600
  if (code != TSDB_CODE_SUCCESS) {
383,786,840✔
2601
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2602
  }
2603
  return code;
383,806,807✔
2604
}
2605

2606
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
1,352,789,397✔
2607
                                  const char* name) {
2608
  SResSchema s = {0};
1,352,789,397✔
2609
  s.scale = scale;
1,352,856,227✔
2610
  s.type = type;
1,352,856,227✔
2611
  s.bytes = bytes;
1,352,856,227✔
2612
  s.slotId = slotId;
1,352,856,227✔
2613
  s.precision = precision;
1,352,856,227✔
2614
  tstrncpy(s.name, name, tListLen(s.name));
1,352,856,227✔
2615

2616
  return s;
1,352,916,403✔
2617
}
2618

2619
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType, EColumnType colType) {
1,272,851,811✔
2620
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
1,272,851,811✔
2621
  if (pCol == NULL) {
1,272,477,491✔
2622
    return NULL;
×
2623
  }
2624

2625
  pCol->slotId = slotId;
1,272,477,491✔
2626
  pCol->colId = colId;
1,272,560,060✔
2627
  pCol->bytes = pType->bytes;
1,272,611,172✔
2628
  pCol->type = pType->type;
1,272,630,390✔
2629
  pCol->scale = pType->scale;
1,272,845,768✔
2630
  pCol->precision = pType->precision;
1,272,782,790✔
2631
  pCol->dataBlockId = blockId;
1,272,961,971✔
2632
  pCol->colType = colType;
1,272,925,395✔
2633
  return pCol;
1,272,880,519✔
2634
}
2635

2636
int32_t createExprFromOneNode(SExprInfo* pExp, SNode* pNode, int16_t slotId) {
1,353,276,812✔
2637
  int32_t code = TSDB_CODE_SUCCESS;
1,353,276,812✔
2638
  int32_t lino = 0;
1,353,276,812✔
2639
  pExp->base.numOfParams = 0;
1,353,276,812✔
2640
  pExp->base.pParam = NULL;
1,353,351,192✔
2641
  pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
1,353,277,109✔
2642
  QUERY_CHECK_NULL(pExp->pExpr, code, lino, _end, terrno);
1,352,758,951✔
2643

2644
  pExp->pExpr->_function.num = 1;
1,352,862,321✔
2645
  pExp->pExpr->_function.functionId = -1;
1,352,979,964✔
2646

2647
  int32_t type = nodeType(pNode);
1,353,126,702✔
2648
  // it is a project query, or group by column
2649
  if (type == QUERY_NODE_COLUMN) {
1,353,171,839✔
2650
    pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
768,857,623✔
2651
    SColumnNode* pColNode = (SColumnNode*)pNode;
768,901,439✔
2652

2653
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
768,901,439✔
2654
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
768,821,717✔
2655

2656
    pExp->base.numOfParams = 1;
768,835,633✔
2657

2658
    SDataType* pType = &pColNode->node.resType;
768,830,068✔
2659
    pExp->base.resSchema =
2660
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pColNode->colName);
768,895,322✔
2661

2662
    pExp->base.pParam[0].pCol =
1,537,776,217✔
2663
        createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType, pColNode->colType);
1,537,668,760✔
2664
    QUERY_CHECK_NULL(pExp->base.pParam[0].pCol, code, lino, _end, terrno);
768,915,105✔
2665

2666
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
768,787,058✔
2667
  } else if (type == QUERY_NODE_VALUE) {
584,314,216✔
2668
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
20,454,425✔
2669
    SValueNode* pValNode = (SValueNode*)pNode;
20,438,377✔
2670

2671
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
20,438,377✔
2672
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
20,448,392✔
2673

2674
    pExp->base.numOfParams = 1;
20,452,719✔
2675

2676
    SDataType* pType = &pValNode->node.resType;
20,452,999✔
2677
    pExp->base.resSchema =
2678
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
20,445,563✔
2679
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
20,440,865✔
2680
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
20,441,750✔
2681
    QUERY_CHECK_CODE(code, lino, _end);
20,443,504✔
2682
  } else if (type == QUERY_NODE_REMOTE_VALUE) {
563,859,791✔
2683
    SRemoteValueNode* pRemote = (SRemoteValueNode*)pNode;
1,604,738✔
2684
    code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, pNode);
1,604,738✔
2685
    QUERY_CHECK_CODE(code, lino, _end);
1,605,966✔
2686

2687
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
1,130,730✔
2688
    SValueNode* pValNode = (SValueNode*)pNode;
1,130,730✔
2689

2690
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
1,130,730✔
2691
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
1,130,730✔
2692

2693
    pExp->base.numOfParams = 1;
1,130,730✔
2694

2695
    SDataType* pType = &pValNode->node.resType;
1,130,730✔
2696
    pExp->base.resSchema =
2697
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
1,130,730✔
2698
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
1,130,730✔
2699
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
1,130,730✔
2700
    QUERY_CHECK_CODE(code, lino, _end);
1,130,116✔
2701
  } else if (type == QUERY_NODE_FUNCTION) {
562,255,053✔
2702
    pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
516,286,467✔
2703
    SFunctionNode* pFuncNode = (SFunctionNode*)pNode;
516,265,733✔
2704

2705
    SDataType* pType = &pFuncNode->node.resType;
516,265,733✔
2706
    pExp->base.resSchema =
2707
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pFuncNode->node.aliasName);
516,276,170✔
2708
    tExprNode* pExprNode = pExp->pExpr;
516,251,600✔
2709

2710
    pExprNode->_function.functionId = pFuncNode->funcId;
516,267,689✔
2711
    pExprNode->_function.pFunctNode = pFuncNode;
516,287,412✔
2712
    pExprNode->_function.functionType = pFuncNode->funcType;
516,345,024✔
2713

2714
    tstrncpy(pExprNode->_function.functionName, pFuncNode->functionName, tListLen(pExprNode->_function.functionName));
516,268,982✔
2715

2716
    pExp->base.pParamList = pFuncNode->pParameterList;
516,284,888✔
2717
#if 1
2718
    // todo refactor: add the parameter for tbname function
2719
    const char* name = "tbname";
516,329,619✔
2720
    int32_t     len = strlen(name);
516,329,619✔
2721

2722
    if (!pFuncNode->pParameterList && (memcmp(pExprNode->_function.functionName, name, len) == 0) &&
516,329,619✔
2723
        pExprNode->_function.functionName[len] == 0) {
33,334,144✔
2724
      pFuncNode->pParameterList = NULL;
33,325,327✔
2725
      int32_t     code = nodesMakeList(&pFuncNode->pParameterList);
33,337,946✔
2726
      SValueNode* res = NULL;
33,339,374✔
2727
      if (TSDB_CODE_SUCCESS == code) {
33,340,566✔
2728
        code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
33,339,437✔
2729
      }
2730
      QUERY_CHECK_CODE(code, lino, _end);
33,338,385✔
2731
      res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
33,338,385✔
2732
      code = nodesListAppend(pFuncNode->pParameterList, (SNode*)res);
33,332,445✔
2733
      if (code != TSDB_CODE_SUCCESS) {
33,339,322✔
2734
        nodesDestroyNode((SNode*)res);
×
2735
        res = NULL;
×
2736
      }
2737
      QUERY_CHECK_CODE(code, lino, _end);
33,339,322✔
2738
    }
2739
#endif
2740

2741
    int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
516,295,799✔
2742

2743
    pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
516,309,354✔
2744
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
516,177,962✔
2745
    pExp->base.numOfParams = numOfParam;
516,161,320✔
2746

2747
    for (int32_t j = 0; j < numOfParam && TSDB_CODE_SUCCESS == code; ++j) {
1,221,059,460✔
2748
      SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
704,887,461✔
2749
      QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
704,878,825✔
2750
      if (p1->type == QUERY_NODE_COLUMN) {
704,878,825✔
2751
        SColumnNode* pcn = (SColumnNode*)p1;
503,992,838✔
2752

2753
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
503,992,838✔
2754
        pExp->base.pParam[j].pCol =
1,007,909,382✔
2755
            createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType, pcn->colType);
1,007,939,101✔
2756
        QUERY_CHECK_NULL(pExp->base.pParam[j].pCol, code, lino, _end, terrno);
503,935,169✔
2757
      } else if (p1->type == QUERY_NODE_VALUE) {
200,913,771✔
2758
        SValueNode* pvn = (SValueNode*)p1;
134,967,802✔
2759
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
134,967,802✔
2760
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
134,952,091✔
2761
        QUERY_CHECK_CODE(code, lino, _end);
134,949,622✔
2762
      } else if (p1->type == QUERY_NODE_REMOTE_VALUE) {
66,009,302✔
2763
        SRemoteValueNode* pRemote = (SRemoteValueNode*)p1;
81,662✔
2764
        code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, p1);
81,662✔
2765
        QUERY_CHECK_CODE(code, lino, _end);
81,662✔
2766

2767
        SValueNode* pvn = (SValueNode*)pRemote;
59,558✔
2768
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
59,558✔
2769
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
59,558✔
2770
        QUERY_CHECK_CODE(code, lino, _end);
423✔
2771
      }
2772
    }
2773
    pExp->pExpr->_function.bindExprID = ((SExprNode*)pNode)->bindExprID;
516,171,999✔
2774
  } else if (type == QUERY_NODE_OPERATOR) {
45,968,586✔
2775
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
44,247,776✔
2776
    SOperatorNode* pOpNode = (SOperatorNode*)pNode;
44,249,564✔
2777

2778
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
44,249,564✔
2779
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
44,241,797✔
2780
    pExp->base.numOfParams = 1;
44,235,052✔
2781

2782
    SDataType* pType = &pOpNode->node.resType;
44,240,465✔
2783
    pExp->base.resSchema =
2784
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pOpNode->node.aliasName);
44,247,189✔
2785
    pExp->pExpr->_optrRoot.pRootNode = pNode;
44,251,264✔
2786
  } else if (type == QUERY_NODE_CASE_WHEN) {
1,720,912✔
2787
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
1,616,563✔
2788
    SCaseWhenNode* pCaseNode = (SCaseWhenNode*)pNode;
1,617,085✔
2789

2790
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
1,617,085✔
2791
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
1,616,563✔
2792
    pExp->base.numOfParams = 1;
1,616,563✔
2793

2794
    SDataType* pType = &pCaseNode->node.resType;
1,616,563✔
2795
    pExp->base.resSchema =
2796
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCaseNode->node.aliasName);
1,616,563✔
2797
    pExp->pExpr->_optrRoot.pRootNode = pNode;
1,617,085✔
2798
  } else if (type == QUERY_NODE_LOGIC_CONDITION) {
104,463✔
2799
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
117,380✔
2800
    SLogicConditionNode* pCond = (SLogicConditionNode*)pNode;
117,380✔
2801
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
117,380✔
2802
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
117,380✔
2803
    pExp->base.numOfParams = 1;
117,380✔
2804
    SDataType* pType = &pCond->node.resType;
117,380✔
2805
    pExp->base.resSchema =
2806
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCond->node.aliasName);
117,380✔
2807
    pExp->pExpr->_optrRoot.pRootNode = pNode;
117,380✔
2808
  } else {
2809
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
31✔
2810
    QUERY_CHECK_CODE(code, lino, _end);
31✔
2811
  }
2812
  pExp->pExpr->relatedTo = ((SExprNode*)pNode)->relatedTo;
1,352,682,589✔
2813
_end:
1,353,130,265✔
2814
  if (code != TSDB_CODE_SUCCESS) {
1,353,130,265✔
2815
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
497,340✔
2816
  }
2817
  return code;
1,353,192,484✔
2818
}
2819

2820
int32_t createExprFromTargetNode(SExprInfo* pExp, STargetNode* pTargetNode) {
1,353,144,488✔
2821
  return createExprFromOneNode(pExp, pTargetNode->pExpr, pTargetNode->slotId);
1,353,144,488✔
2822
}
2823

2824
SExprInfo* createExpr(SNodeList* pNodeList, int32_t* numOfExprs) {
×
2825
  *numOfExprs = LIST_LENGTH(pNodeList);
×
2826
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
×
2827
  if (!pExprs) {
×
2828
    return NULL;
×
2829
  }
2830

2831
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
×
2832
    SExprInfo* pExp = &pExprs[i];
×
2833
    int32_t    code = createExprFromOneNode(pExp, nodesListGetNode(pNodeList, i), i + UD_TAG_COLUMN_INDEX);
×
2834
    if (code != TSDB_CODE_SUCCESS) {
×
2835
      taosMemoryFreeClear(pExprs);
×
2836
      terrno = code;
×
2837
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2838
      return NULL;
×
2839
    }
2840
  }
2841

2842
  return pExprs;
×
2843
}
2844

2845
int32_t createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, SExprInfo** pExprInfo, int32_t* numOfExprs) {
504,984,100✔
2846
  QRY_PARAM_CHECK(pExprInfo);
504,984,100✔
2847

2848
  int32_t code = 0;
505,038,706✔
2849
  int32_t lino = 0;
505,038,706✔
2850
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
505,038,706✔
2851
  int32_t numOfGroupKeys = 0;
504,929,719✔
2852
  if (pGroupKeys != NULL) {
504,929,719✔
2853
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
35,223,982✔
2854
  }
2855

2856
  *numOfExprs = numOfFuncs + numOfGroupKeys;
504,929,103✔
2857
  if (*numOfExprs == 0) {
505,003,709✔
2858
    return code;
43,946,710✔
2859
  }
2860

2861
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
461,059,244✔
2862
  QUERY_CHECK_NULL(pExprs, code, lino, _return, terrno);
460,850,996✔
2863

2864
  int32_t i = 0;
460,850,996✔
2865
  if (pNodeList != NULL) {
460,850,996✔
2866
    SNode* node = NULL;
429,908,681✔
2867
    FOREACH(node, pNodeList) {
1,720,896,248✔
2868
      STargetNode* pTargetNode = (STargetNode*)node;
1,291,496,991✔
2869
      SExprInfo*   pExp = &pExprs[i++];
1,291,496,991✔
2870
      code = createExprFromTargetNode(pExp, pTargetNode);
1,291,386,787✔
2871
      QUERY_CHECK_CODE(code, lino, _return);
1,291,484,907✔
2872
    }
2873
  }
2874

2875
  if (pGroupKeys != NULL) {
460,609,516✔
2876
    SNode* node = NULL;
35,216,093✔
2877
    FOREACH(node, pGroupKeys) {
96,794,301✔
2878
      STargetNode* pTargetNode = (STargetNode*)node;
61,573,626✔
2879
      SExprInfo*   pExp = &pExprs[i++];
61,573,626✔
2880
      code = createExprFromTargetNode(pExp, pTargetNode);
61,567,143✔
2881
      QUERY_CHECK_CODE(code, lino, _return);
61,578,208✔
2882
    }
2883
  }
2884

2885
  *pExprInfo = pExprs;
460,608,386✔
2886
  return code;
460,567,235✔
2887

2888
_return:
497,340✔
2889
  qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
497,340✔
2890
  destroyExprInfo(pExprs, *numOfExprs);
497,340✔
2891
  taosMemoryFreeClear(pExprs);
497,340✔
2892
  return code;
497,340✔
2893
}
2894

2895
static void deleteSubsidiareCtx(void* pData) {
×
2896
  SSubsidiaryResInfo* pCtx = (SSubsidiaryResInfo*)pData;
×
2897
  if (pCtx->pCtx) {
×
2898
    taosMemoryFreeClear(pCtx->pCtx);
×
2899
  }
2900
}
×
2901

2902
// set the output buffer for the selectivity + tag query
2903
static int32_t setSelectValueColumnInfo(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
491,214,458✔
2904
  int32_t num = 0;
491,214,458✔
2905
  int32_t code = TSDB_CODE_SUCCESS;
491,214,458✔
2906
  int32_t lino = 0;
491,214,458✔
2907

2908
  SArray* pValCtxArray = NULL;
491,214,458✔
2909
  for (int32_t i = numOfOutput - 1; i > 0; --i) {  // select Func is at the end of the list
1,375,108,962✔
2910
    int32_t funcIdx = pCtx[i].pExpr->pExpr->_function.bindExprID;
884,014,485✔
2911
    if (funcIdx > 0) {
884,028,736✔
2912
      if (pValCtxArray == NULL) {
1,807,012✔
2913
        // the end of the list is the select function of biggest index
2914
        pValCtxArray = taosArrayInit_s(sizeof(SSubsidiaryResInfo*), funcIdx);
1,295,772✔
2915
        if (pValCtxArray == NULL) {
1,296,264✔
2916
          return terrno;
×
2917
        }
2918
      }
2919
      if (funcIdx > pValCtxArray->size) {
1,807,504✔
2920
        qError("funcIdx:%d is out of range", funcIdx);
×
2921
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2922
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2923
      }
2924
      SSubsidiaryResInfo* pSubsidiary = &pCtx[i].subsidiaries;
1,807,996✔
2925
      pSubsidiary->pCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
1,808,980✔
2926
      if (pSubsidiary->pCtx == NULL) {
1,808,488✔
2927
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2928
        return terrno;
×
2929
      }
2930
      pSubsidiary->num = 0;
1,806,028✔
2931
      taosArraySet(pValCtxArray, funcIdx - 1, &pSubsidiary);
1,806,028✔
2932
    }
2933
  }
2934

2935
  SqlFunctionCtx*  p = NULL;
491,094,477✔
2936
  SqlFunctionCtx** pValCtx = NULL;
491,094,477✔
2937
  if (pValCtxArray == NULL) {
491,094,477✔
2938
    pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
489,838,009✔
2939
    if (pValCtx == NULL) {
489,875,508✔
2940
      QUERY_CHECK_CODE(terrno, lino, _end);
×
2941
    }
2942
  }
2943

2944
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,830,590,855✔
2945
    const char* pName = pCtx[i].pExpr->pExpr->_function.functionName;
1,339,475,448✔
2946
    if ((strcmp(pName, "_select_value") == 0)) {
1,339,610,293✔
2947
      if (pValCtxArray == NULL) {
8,645,252✔
2948
        pValCtx[num++] = &pCtx[i];
6,104,870✔
2949
      } else {
2950
        int32_t bindFuncIndex = pCtx[i].pExpr->pExpr->relatedTo;  // start from index 1;
2,540,415✔
2951
        if (bindFuncIndex > 0) {                                  // 0 is default index related to the select function
2,539,659✔
2952
          bindFuncIndex -= 1;
2,480,619✔
2953
        }
2954
        SSubsidiaryResInfo** pSubsidiary = taosArrayGet(pValCtxArray, bindFuncIndex);
2,539,659✔
2955
        if (pSubsidiary == NULL) {
2,538,675✔
2956
          QUERY_CHECK_CODE(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR, lino, _end);
×
2957
        }
2958
        (*pSubsidiary)->pCtx[(*pSubsidiary)->num] = &pCtx[i];
2,538,675✔
2959
        (*pSubsidiary)->num++;
2,539,167✔
2960
      }
2961
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
1,330,965,041✔
2962
      if (pValCtxArray == NULL) {
116,605,341✔
2963
        p = &pCtx[i];
114,402,530✔
2964
      }
2965
    }
2966
  }
2967

2968
  if (p != NULL) {
491,115,407✔
2969
    p->subsidiaries.pCtx = pValCtx;
56,649,209✔
2970
    p->subsidiaries.num = num;
56,648,524✔
2971
  } else {
2972
    taosMemoryFreeClear(pValCtx);
434,466,198✔
2973
  }
2974

2975
_end:
1,313,158✔
2976
  if (code != TSDB_CODE_SUCCESS) {
491,037,875✔
2977
    taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2978
    taosMemoryFreeClear(pValCtx);
×
2979
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2980
  } else {
2981
    taosArrayDestroy(pValCtxArray);
491,037,875✔
2982
  }
2983
  return code;
491,145,980✔
2984
}
2985

2986
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowEntryInfoOffset,
491,197,960✔
2987
                                     SFunctionStateStore* pStore) {
2988
  int32_t         code = TSDB_CODE_SUCCESS;
491,197,960✔
2989
  int32_t         lino = 0;
491,197,960✔
2990
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
491,197,960✔
2991
  if (pFuncCtx == NULL) {
490,916,369✔
2992
    return NULL;
×
2993
  }
2994

2995
  *rowEntryInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
490,916,369✔
2996
  if (*rowEntryInfoOffset == 0) {
491,216,731✔
2997
    taosMemoryFreeClear(pFuncCtx);
×
2998
    return NULL;
×
2999
  }
3000

3001
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,830,715,600✔
3002
    SExprInfo* pExpr = &pExprInfo[i];
1,339,593,180✔
3003

3004
    SExprBasicInfo* pFunct = &pExpr->base;
1,339,535,349✔
3005
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
1,339,553,126✔
3006

3007
    pCtx->functionId = -1;
1,339,576,605✔
3008
    pCtx->pExpr = pExpr;
1,339,636,166✔
3009

3010
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
1,339,580,683✔
3011
      SFuncExecEnv env = {0};
514,631,878✔
3012
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;
514,651,008✔
3013
      pCtx->isPseudoFunc = fmIsWindowPseudoColumnFunc(pCtx->functionId) || fmIsPlaceHolderFunc(pCtx->functionId);
514,682,723✔
3014
      pCtx->isNotNullFunc = fmIsNotNullOutputFunc(pCtx->functionId);
514,642,172✔
3015

3016
      bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
514,695,156✔
3017
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId) ||
687,913,308✔
3018
          fmIsProcessByRowFunc(pCtx->functionId)) {
514,584,527✔
3019
        if (!isUdaf) {
341,341,116✔
3020
          code = fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
341,287,234✔
3021
          QUERY_CHECK_CODE(code, lino, _end);
341,180,491✔
3022
        } else {
3023
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
53,882✔
3024
          pCtx->udfName = taosStrdup(udfName);
53,882✔
3025
          QUERY_CHECK_NULL(pCtx->udfName, code, lino, _end, terrno);
53,882✔
3026

3027
          code = fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
53,882✔
3028
          QUERY_CHECK_CODE(code, lino, _end);
53,882✔
3029
        }
3030
        bool tmp = pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
341,234,373✔
3031
        if (!tmp) {
341,263,996✔
3032
          code = terrno;
×
3033
          QUERY_CHECK_CODE(code, lino, _end);
4,213✔
3034
        }
3035
      } else {
3036
        if (fmIsPlaceHolderFunc(pCtx->functionId)) {
173,276,073✔
3037
          code = fmGetStreamPesudoFuncEnv(pCtx->functionId, pExpr->base.pParamList, &env);
8,515,684✔
3038
          QUERY_CHECK_CODE(code, lino, _end);
8,515,684✔
3039
        }      
3040
        
3041
        code = fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
173,297,113✔
3042
        if (code != TSDB_CODE_SUCCESS && isUdaf) {
173,309,736✔
3043
          code = TSDB_CODE_SUCCESS;
71,100✔
3044
        }
3045
        QUERY_CHECK_CODE(code, lino, _end);
173,309,736✔
3046

3047
        if (pCtx->sfp.getEnv != NULL) {
173,309,736✔
3048
          bool tmp = pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
32,739,951✔
3049
          if (!tmp) {
32,741,390✔
3050
            code = terrno;
×
3051
            QUERY_CHECK_CODE(code, lino, _end);
×
3052
          }
3053
        }
3054
      }
3055
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
514,590,463✔
3056
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
824,856,400✔
3057
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
21,562,380✔
3058
      // for simple column, the result buffer needs to hold at least one element.
3059
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
825,004,315✔
3060
    }
3061

3062
    pCtx->input.numOfInputCols = pFunct->numOfParams;
1,339,617,474✔
3063
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
1,339,566,958✔
3064
    QUERY_CHECK_NULL(pCtx->input.pData, code, lino, _end, terrno);
1,339,628,316✔
3065
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
1,339,517,364✔
3066
    QUERY_CHECK_NULL(pCtx->input.pColumnDataAgg, code, lino, _end, terrno);
1,339,657,394✔
3067

3068
    pCtx->pTsOutput = NULL;
1,339,569,208✔
3069
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
1,339,661,616✔
3070
    pCtx->resDataInfo.type = pFunct->resSchema.type;
1,339,546,940✔
3071
    pCtx->order = TSDB_ORDER_ASC;
1,339,673,111✔
3072
    pCtx->start.key = INT64_MIN;
1,339,704,905✔
3073
    pCtx->end.key = INT64_MIN;
1,339,556,546✔
3074
    pCtx->numOfParams = pExpr->base.numOfParams;
1,339,619,599✔
3075
    pCtx->param = pFunct->pParam;
1,339,775,693✔
3076
    pCtx->saveHandle.currentPage = -1;
1,339,696,283✔
3077
    pCtx->pStore = pStore;
1,339,685,546✔
3078
    pCtx->hasWindowOrGroup = false;
1,339,721,657✔
3079
    pCtx->needCleanup = false;
1,339,643,363✔
3080
    pCtx->skipDynDataCheck = false;
1,339,498,113✔
3081
  }
3082

3083
  for (int32_t i = 1; i < numOfOutput; ++i) {
1,375,319,097✔
3084
    (*rowEntryInfoOffset)[i] = (int32_t)((*rowEntryInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) +
1,768,181,292✔
3085
                                         pFuncCtx[i - 1].resDataInfo.interBufSize);
884,118,424✔
3086
  }
3087

3088
  code = setSelectValueColumnInfo(pFuncCtx, numOfOutput);
491,195,306✔
3089
  QUERY_CHECK_CODE(code, lino, _end);
491,160,663✔
3090

3091
_end:
491,160,663✔
3092
  if (code != TSDB_CODE_SUCCESS) {
491,065,810✔
3093
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3094
    for (int32_t i = 0; i < numOfOutput; ++i) {
×
3095
      taosMemoryFree(pFuncCtx[i].input.pData);
×
3096
      taosMemoryFree(pFuncCtx[i].input.pColumnDataAgg);
×
3097
    }
3098
    taosMemoryFreeClear(*rowEntryInfoOffset);
×
3099
    taosMemoryFreeClear(pFuncCtx);
×
3100

3101
    terrno = code;
×
3102
    return NULL;
×
3103
  }
3104
  return pFuncCtx;
491,065,810✔
3105
}
3106

3107
// NOTE: sources columns are more than the destination SSDatablock columns.
3108
// doFilter in table scan needs every column even its output is false
3109
int32_t relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray* pCols, bool outputEveryColumn) {
17,047,991✔
3110
  int32_t code = TSDB_CODE_SUCCESS;
17,047,991✔
3111
  size_t  numOfSrcCols = taosArrayGetSize(pCols);
17,047,991✔
3112

3113
  int32_t i = 0, j = 0;
17,047,991✔
3114
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
137,965,005✔
3115
    SColumnInfoData* p = taosArrayGet(pCols, i);
120,917,593✔
3116
    if (!p) {
120,918,247✔
3117
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3118
      return terrno;
×
3119
    }
3120
    SColMatchItem* pmInfo = taosArrayGet(pColMatchInfo, j);
120,918,247✔
3121
    if (!pmInfo) {
120,918,241✔
3122
      return terrno;
×
3123
    }
3124

3125
    if (p->info.colId == pmInfo->colId) {
120,918,241✔
3126
      SColumnInfoData* pDst = taosArrayGet(pBlock->pDataBlock, pmInfo->dstSlotId);
112,031,093✔
3127
      if (!pDst) {
112,030,344✔
3128
        return terrno;
×
3129
      }
3130
      code = colDataAssign(pDst, p, pBlock->info.rows, &pBlock->info);
112,030,344✔
3131
      if (code != TSDB_CODE_SUCCESS) {
112,029,839✔
3132
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3133
        return code;
×
3134
      }
3135
      i++;
112,029,839✔
3136
      j++;
112,029,839✔
3137
    } else if (p->info.colId < pmInfo->colId) {
8,887,169✔
3138
      i++;
8,887,175✔
3139
    } else {
3140
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3141
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3142
    }
3143
  }
3144
  return code;
17,047,406✔
3145
}
3146

3147
SInterval extractIntervalInfo(const STableScanPhysiNode* pTableScanNode) {
266,599,725✔
3148
  SInterval interval = {
533,184,371✔
3149
      .interval = pTableScanNode->interval,
266,551,493✔
3150
      .sliding = pTableScanNode->sliding,
266,544,940✔
3151
      .intervalUnit = pTableScanNode->intervalUnit,
266,634,730✔
3152
      .slidingUnit = pTableScanNode->slidingUnit,
266,506,129✔
3153
      .offset = pTableScanNode->offset,
266,435,846✔
3154
      .precision = pTableScanNode->scan.node.pOutputDataBlockDesc->precision,
266,368,366✔
3155
      .timeRange = pTableScanNode->scanRange,
3156
  };
3157
  calcIntervalAutoOffset(&interval);
266,611,944✔
3158

3159
  return interval;
266,428,765✔
3160
}
3161

3162
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
67,157,203✔
3163
  SColumn c = {0};
67,157,203✔
3164

3165
  c.slotId = pColNode->slotId;
67,157,203✔
3166
  c.colId = pColNode->colId;
67,149,259✔
3167
  c.type = pColNode->node.resType.type;
67,156,084✔
3168
  c.bytes = pColNode->node.resType.bytes;
67,159,886✔
3169
  c.scale = pColNode->node.resType.scale;
67,152,675✔
3170
  c.precision = pColNode->node.resType.precision;
67,145,141✔
3171
  return c;
67,156,058✔
3172
}
3173

3174

3175
/**
3176
 * @brief Determine the actual time range for reading data based on the RANGE clause and the WHERE conditions.
3177
 * @param[in] cond The range specified by WHERE condition.
3178
 * @param[in] range The range specified by RANGE clause.
3179
 * @param[out] twindow The range to be read in DESC order, and only one record is needed.
3180
 * @param[out] extTwindow The external range to read for only one record, which is used for FILL clause.
3181
 * @note `cond` and `twindow` may be the same address.
3182
 */
3183
static int32_t getQueryExtWindow(const STimeWindow* cond, const STimeWindow* range, STimeWindow* twindow,
3,711,725✔
3184
                                 STimeWindow* extTwindows) {
3185
  int32_t     code = TSDB_CODE_SUCCESS;
3,711,725✔
3186
  int32_t     lino = 0;
3,711,725✔
3187
  STimeWindow tempWindow;
3188

3189
  if (cond->skey > cond->ekey || range->skey > range->ekey) {
3,711,725✔
3190
    *twindow = extTwindows[0] = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
6,160✔
3191
    return code;
5,680✔
3192
  }
3193

3194
  if (range->ekey < cond->skey) {
3,706,045✔
3195
    extTwindows[1] = *cond;
580,703✔
3196
    *twindow = extTwindows[0] = TSWINDOW_DESC_INITIALIZER;
580,703✔
3197
    return code;
580,703✔
3198
  }
3199

3200
  if (cond->ekey < range->skey) {
3,125,342✔
3201
    extTwindows[0] = *cond;
416,943✔
3202
    *twindow = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
416,943✔
3203
    return code;
416,943✔
3204
  }
3205

3206
  // Only scan data in the time range intersecion.
3207
  extTwindows[0] = extTwindows[1] = *cond;
2,708,399✔
3208
  twindow->skey = TMAX(cond->skey, range->skey);
2,708,399✔
3209
  twindow->ekey = TMIN(cond->ekey, range->ekey);
2,708,399✔
3210
  extTwindows[0].ekey = twindow->skey - 1;
2,708,399✔
3211
  extTwindows[1].skey = twindow->ekey + 1;
2,708,399✔
3212

3213
  return code;
2,708,399✔
3214
}
3215

3216
static int32_t getPrimaryTimeRange(SNode** pPrimaryKeyCond, STimeWindow* pTimeRange, bool* isStrict) {
712,152✔
3217
  SNode*  pNew = NULL;
712,152✔
3218
  int32_t code = scalarCalculateRemoteConstants(*pPrimaryKeyCond, &pNew);
712,152✔
3219
  if (TSDB_CODE_SUCCESS == code) {
712,152✔
3220
    *pPrimaryKeyCond = pNew;
712,152✔
3221
    if (nodeType(pNew) != QUERY_NODE_VALUE) {
712,152✔
3222
      code = filterGetTimeRange(*pPrimaryKeyCond, pTimeRange, isStrict, NULL);
712,152✔
3223
    }
3224
  }
3225
  return code;
712,152✔
3226
}
3227

3228
int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, STableScanPhysiNode* pTableScanNode,
300,053,089✔
3229
                               const SReadHandle* readHandle, bool applyExtWin) {
3230
  int32_t code = 0;                             
300,053,089✔
3231
  pCond->order = pTableScanNode->scanSeq[0] > 0 ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
300,053,089✔
3232
  pCond->numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
299,941,572✔
3233

3234
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
300,010,107✔
3235
  if (!pCond->colList) {
300,001,667✔
3236
    return terrno;
×
3237
  }
3238
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
299,976,308✔
3239
  if (pCond->pSlotList == NULL) {
300,014,348✔
3240
    taosMemoryFreeClear(pCond->colList);
×
3241
    return terrno;
×
3242
  }
3243

3244
  // TODO: get it from stable scan node
3245
  pCond->twindows = pTableScanNode->scanRange;
299,910,350✔
3246
  pCond->suid = pTableScanNode->scan.suid;
300,087,190✔
3247
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
300,042,626✔
3248
  pCond->startVersion = -1;
299,980,985✔
3249
  pCond->endVersion = -1;
299,885,687✔
3250
  pCond->skipRollup = readHandle->skipRollup;
300,002,411✔
3251
  if (readHandle->winRangeValid) {
299,821,955✔
3252
    pCond->twindows = readHandle->winRange;
561,685✔
3253
  }
3254
  pCond->cacheSttStatis = readHandle->cacheSttStatis;
299,970,218✔
3255
  // allowed read stt file optimization mode
3256
  pCond->notLoadData = (pTableScanNode->dataRequired == FUNC_DATA_REQUIRED_NOT_LOAD) &&
600,126,272✔
3257
                       (pTableScanNode->scan.node.pConditions == NULL) && (pTableScanNode->interval == 0);
300,027,447✔
3258

3259
  int32_t j = 0;
300,077,654✔
3260
  SNode*  node = NULL;
300,077,654✔
3261
  FOREACH(node, pTableScanNode->scan.pScanCols) {
1,400,490,134✔
3262
    STargetNode* pNode = (STargetNode*)node;
1,100,168,981✔
3263
    if (!pNode) {
1,100,168,981✔
3264
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3265
      return terrno;
×
3266
    }
3267
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
1,100,168,981✔
3268
    if (pColNode->colType == COLUMN_TYPE_TAG) {
1,100,425,954✔
3269
      continue;
×
3270
    }
3271

3272
    pCond->colList[j].type = pColNode->node.resType.type;
1,100,433,775✔
3273
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
1,100,427,908✔
3274
    pCond->colList[j].colId = pColNode->colId;
1,100,300,415✔
3275
    pCond->colList[j].pk = pColNode->isPk;
1,100,457,553✔
3276

3277
    pCond->pSlotList[j] = pNode->slotId;
1,100,486,366✔
3278
    j += 1;
1,100,412,480✔
3279
  }
3280

3281
  pCond->numOfCols = j;
300,019,328✔
3282

3283
  if (applyExtWin) {
300,080,903✔
3284
    if (NULL != pTableScanNode->pExtScanRange) {
267,310,047✔
3285
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
3,631,069✔
3286
      code = getQueryExtWindow(&pCond->twindows, pTableScanNode->pExtScanRange, &pCond->twindows, pCond->extTwindows);
3,631,069✔
3287
    } else if (readHandle->extWinRangeValid) {
263,425,739✔
3288
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
80,656✔
3289
      code = getQueryExtWindow(&pCond->twindows, &readHandle->extWinRange, &pCond->twindows, pCond->extTwindows);
80,656✔
3290
    }
3291
  }
3292

3293
  if (pTableScanNode->pPrimaryCond) {
300,034,516✔
3294
    bool isStrict = false;
712,152✔
3295
    code = getPrimaryTimeRange((SNode**)&pTableScanNode->pPrimaryCond, &pCond->twindows, &isStrict);
712,152✔
3296
    if (code || !isStrict) {
712,152✔
3297
      code = nodesMergeNode((SNode**)&pTableScanNode->scan.node.pConditions, &pTableScanNode->pPrimaryCond);
698,840✔
3298
    }
3299
  }
3300

3301
  return code;
300,016,174✔
3302
}
3303

3304
int32_t initQueryTableDataCondWithColArray(SQueryTableDataCond* pCond, SQueryTableDataCond* pOrgCond,
122,740,341✔
3305
                                           const SReadHandle* readHandle, SArray* colArray) {
3306
  int32_t code = TSDB_CODE_SUCCESS;
122,740,341✔
3307
  int32_t lino = 0;
122,740,341✔
3308

3309
  pCond->order = TSDB_ORDER_ASC;
122,740,341✔
3310
  pCond->numOfCols = (int32_t)taosArrayGetSize(colArray);
122,795,608✔
3311

3312
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
122,804,040✔
3313
  QUERY_CHECK_NULL(pCond->colList, code, lino, _return, terrno);
122,772,581✔
3314

3315
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
122,769,365✔
3316
  QUERY_CHECK_NULL(pCond->pSlotList, code, lino, _return, terrno);
122,782,197✔
3317

3318
  pCond->twindows = pOrgCond->twindows;
122,780,510✔
3319
  pCond->order = pOrgCond->order;
122,806,737✔
3320
  pCond->type = pOrgCond->type;
122,808,280✔
3321
  pCond->startVersion = -1;
122,812,894✔
3322
  pCond->endVersion = -1;
122,810,343✔
3323
  pCond->skipRollup = true;
122,814,837✔
3324
  pCond->notLoadData = false;
122,813,637✔
3325

3326
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
503,812,906✔
3327
    SColIdPair* pColPair = taosArrayGet(colArray, i);
381,032,062✔
3328
    QUERY_CHECK_NULL(pColPair, code, lino, _return, terrno);
381,055,498✔
3329

3330
    bool find = false;
381,027,610✔
3331
    for (int32_t j = 0; j < pOrgCond->numOfCols; ++j) {
1,950,530,724✔
3332
      if (pOrgCond->colList[j].colId == pColPair->vtbColId) {
1,950,311,774✔
3333
        pCond->colList[i].type = pOrgCond->colList[j].type;
381,055,973✔
3334
        // For variable-length types (nchar/binary/varchar/varbinary), use the source table's bytes
3335
        // to avoid tsdb reader doCopyColVal length check failure when source data is longer
3336
        // than the virtual table's defined length.
3337
        if (IS_VAR_DATA_TYPE(pOrgCond->colList[j].type) && pColPair->type.bytes > 0) {
381,070,324✔
3338
          pCond->colList[i].bytes = TMAX(pOrgCond->colList[j].bytes, pColPair->type.bytes);
44,423,738✔
3339
        } else {
3340
          pCond->colList[i].bytes = pOrgCond->colList[j].bytes;
336,658,110✔
3341
        }
3342
        pCond->colList[i].colId = pColPair->orgColId;
381,081,916✔
3343
        pCond->colList[i].pk = pOrgCond->colList[j].pk;
381,066,144✔
3344
        pCond->pSlotList[i] = i;
381,074,355✔
3345
        find = true;
381,073,101✔
3346
        qDebug("%s mapped vtb colId:%d to org colId:%d, type:%d, bytes:%d", __func__, pColPair->vtbColId,
381,073,101✔
3347
               pColPair->orgColId, pCond->colList[i].type, pCond->colList[i].bytes);
3348
        break;
380,978,016✔
3349
      }
3350
    }
3351
    QUERY_CHECK_CONDITION(find, code, lino, _return, TSDB_CODE_NOT_FOUND);
381,003,286✔
3352
  }
3353

3354
  return code;
122,822,216✔
3355
_return:
×
3356
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(terrno));
×
3357
  taosMemoryFreeClear(pCond->colList);
×
3358
  taosMemoryFreeClear(pCond->pSlotList);
×
3359
  return code;
×
3360
}
3361

3362
void cleanupQueryTableDataCond(SQueryTableDataCond* pCond) {
747,823,561✔
3363
  taosMemoryFreeClear(pCond->colList);
747,823,561✔
3364
  taosMemoryFreeClear(pCond->pSlotList);
747,852,619✔
3365
}
747,805,962✔
3366

3367
int32_t convertFillType(int32_t mode) {
4,537,594✔
3368
  int32_t type = TSDB_FILL_NONE;
4,537,594✔
3369
  switch (mode) {
4,537,594✔
3370
    case FILL_MODE_PREV:
238,082✔
3371
      type = TSDB_FILL_PREV;
238,082✔
3372
      break;
238,082✔
3373
    case FILL_MODE_NONE:
×
3374
      type = TSDB_FILL_NONE;
×
3375
      break;
×
3376
    case FILL_MODE_NULL:
277,381✔
3377
      type = TSDB_FILL_NULL;
277,381✔
3378
      break;
277,381✔
3379
    case FILL_MODE_NULL_F:
45,026✔
3380
      type = TSDB_FILL_NULL_F;
45,026✔
3381
      break;
45,026✔
3382
    case FILL_MODE_NEXT:
212,764✔
3383
      type = TSDB_FILL_NEXT;
212,764✔
3384
      break;
212,764✔
3385
    case FILL_MODE_VALUE:
202,369✔
3386
      type = TSDB_FILL_SET_VALUE;
202,369✔
3387
      break;
202,369✔
3388
    case FILL_MODE_VALUE_F:
21,092✔
3389
      type = TSDB_FILL_SET_VALUE_F;
21,092✔
3390
      break;
21,092✔
3391
    case FILL_MODE_LINEAR:
272,047✔
3392
      type = TSDB_FILL_LINEAR;
272,047✔
3393
      break;
272,047✔
3394
    case FILL_MODE_NEAR:
3,268,833✔
3395
      type = TSDB_FILL_NEAR;
3,268,833✔
3396
      break;
3,268,833✔
UNCOV
3397
    default:
×
UNCOV
3398
      type = TSDB_FILL_NONE;
×
3399
  }
3400

3401
  return type;
4,537,594✔
3402
}
3403

3404
void getInitialStartTimeWindow(SInterval* pInterval, TSKEY ts, STimeWindow* w, bool ascQuery) {
2,147,483,647✔
3405
  if (ascQuery) {
2,147,483,647✔
3406
    *w = getAlignQueryTimeWindow(pInterval, ts);
2,147,483,647✔
3407
  } else {
3408
    // the start position of the first time window in the endpoint that spreads beyond the queried last timestamp
3409
    *w = getAlignQueryTimeWindow(pInterval, ts);
14,086✔
3410

3411
    int64_t key = w->skey;
278,251✔
3412
    while (key < ts) {  // moving towards end
297,328✔
3413
      key = getNextTimeWindowStart(pInterval, key, TSDB_ORDER_ASC);
153,665✔
3414
      if (key > ts) {
153,073✔
3415
        break;
133,996✔
3416
      }
3417

3418
      w->skey = key;
19,077✔
3419
    }
3420
    w->ekey = taosTimeAdd(w->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
277,659✔
3421
  }
3422
}
2,147,483,647✔
3423

3424
static STimeWindow doCalculateTimeWindow(int64_t ts, SInterval* pInterval) {
83,389,395✔
3425
  STimeWindow w = {0};
83,389,395✔
3426

3427
  w.skey = taosTimeTruncate(ts, pInterval);
83,389,395✔
3428
  w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
83,410,283✔
3429
  return w;
83,416,187✔
3430
}
3431

3432
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order) {
1,792,643✔
3433
  STimeWindow win = *pWindow;
1,792,643✔
3434
  STimeWindow save = win;
1,792,643✔
3435
  while (win.skey <= ts && win.ekey >= ts) {
13,448,671✔
3436
    save = win;
11,656,028✔
3437
    // get previous time window
3438
    getNextTimeWindow(pInterval, &win, order == TSDB_ORDER_DESC ? TSDB_ORDER_ASC : TSDB_ORDER_DESC);
11,656,028✔
3439
  }
3440

3441
  return save;
1,792,643✔
3442
}
3443

3444
// get the correct time window according to the handled timestamp
3445
// todo refactor
3446
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
128,753,804✔
3447
                                int32_t order) {
3448
  STimeWindow w = {0};
128,753,804✔
3449
  if (pResultRowInfo->cur.pageId == -1) {  // the first window, from the previous stored value
128,759,333✔
3450
    getInitialStartTimeWindow(pInterval, ts, &w, (order != TSDB_ORDER_DESC));
4,095,355✔
3451
    return w;
4,100,581✔
3452
  }
3453

3454
  SResultRow* pRow = getResultRowByPos(pBuf, &pResultRowInfo->cur, false);
124,662,271✔
3455
  if (pRow) {
124,663,865✔
3456
    TAOS_SET_OBJ_ALIGNED(&w, pRow->win);
124,666,000✔
3457
  }
3458

3459
  // in case of typical time window, we can calculate time window directly.
3460
  if (w.skey > ts || w.ekey < ts) {
124,663,390✔
3461
    w = doCalculateTimeWindow(ts, pInterval);
83,414,021✔
3462
  }
3463

3464
  if (pInterval->interval != pInterval->sliding) {
124,664,488✔
3465
    // it is an sliding window query, in which sliding value is not equalled to
3466
    // interval value, and we need to find the first qualified time window.
3467
    w = getFirstQualifiedTimeWindow(ts, &w, pInterval, order);
1,792,643✔
3468
  }
3469

3470
  return w;
124,656,920✔
3471
}
3472

3473
void getNextTimeWindow(const SInterval* pInterval, STimeWindow* tw, int32_t order) {
2,147,483,647✔
3474
  tw->skey = getNextTimeWindowStart(pInterval, tw->skey, order);
2,147,483,647✔
3475
  tw->ekey = taosTimeGetIntervalEnd(tw->skey, pInterval);
2,147,483,647✔
3476
}
2,147,483,647✔
3477

3478
bool hasLimitOffsetInfo(SLimitInfo* pLimitInfo) {
610,527,366✔
3479
  return (pLimitInfo->limit.limit != -1 || pLimitInfo->limit.offset != -1 || pLimitInfo->slimit.limit != -1 ||
1,216,246,211✔
3480
          pLimitInfo->slimit.offset != -1);
605,718,845✔
3481
}
3482

3483
bool hasSlimitOffsetInfo(SLimitInfo* pLimitInfo) {
×
3484
  return (pLimitInfo->slimit.limit != -1 || pLimitInfo->slimit.offset != -1);
×
3485
}
3486

3487
void initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo) {
725,099,588✔
3488
  SLimit limit = {.limit = getLimit(pLimit), .offset = getOffset(pLimit)};
725,099,588✔
3489
  SLimit slimit = {.limit = getLimit(pSLimit), .offset = getOffset(pSLimit)};
724,886,514✔
3490

3491
  pLimitInfo->limit = limit;
724,900,989✔
3492
  pLimitInfo->slimit = slimit;
724,919,249✔
3493
  pLimitInfo->remainOffset = limit.offset;
724,906,926✔
3494
  pLimitInfo->remainGroupOffset = slimit.offset;
724,977,511✔
3495
  pLimitInfo->numOfOutputRows = 0;
724,978,805✔
3496
  pLimitInfo->numOfOutputGroups = 0;
724,967,992✔
3497
  pLimitInfo->currentGroupId = 0;
724,975,441✔
3498
}
725,145,705✔
3499

3500
void resetLimitInfoForNextGroup(SLimitInfo* pLimitInfo) {
92,544,597✔
3501
  pLimitInfo->numOfOutputRows = 0;
92,544,597✔
3502
  pLimitInfo->remainOffset = pLimitInfo->limit.offset;
92,560,729✔
3503
}
92,517,168✔
3504

3505
int32_t tableListGetSize(const STableListInfo* pTableList, int32_t* pRes) {
764,979,205✔
3506
  if (taosArrayGetSize(pTableList->pTableList) != taosHashGetSize(pTableList->map)) {
764,979,205✔
3507
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3508
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3509
  }
3510
  (*pRes) = taosArrayGetSize(pTableList->pTableList);
764,901,565✔
3511
  return TSDB_CODE_SUCCESS;
764,977,539✔
3512
}
3513

3514
uint64_t tableListGetSuid(const STableListInfo* pTableList) { return pTableList->idInfo.suid; }
4,323,179✔
3515

3516
STableKeyInfo* tableListGetInfo(const STableListInfo* pTableList, int32_t index) {
238,309,758✔
3517
  if (taosArrayGetSize(pTableList->pTableList) == 0) {
238,309,758✔
3518
    return NULL;
2,586✔
3519
  }
3520

3521
  return taosArrayGet(pTableList->pTableList, index);
238,255,578✔
3522
}
3523

3524
int32_t tableListFind(const STableListInfo* pTableList, uint64_t uid, int32_t startIndex) {
54,932✔
3525
  int32_t numOfTables = taosArrayGetSize(pTableList->pTableList);
54,932✔
3526
  if (startIndex >= numOfTables) {
54,932✔
3527
    return -1;
×
3528
  }
3529

3530
  for (int32_t i = startIndex; i < numOfTables; ++i) {
246,600✔
3531
    STableKeyInfo* p = taosArrayGet(pTableList->pTableList, i);
246,600✔
3532
    if (!p) {
246,600✔
3533
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3534
      return -1;
×
3535
    }
3536
    if (p->uid == uid) {
246,600✔
3537
      return i;
54,932✔
3538
    }
3539
  }
3540
  return -1;
×
3541
}
3542

3543
void tableListGetSourceTableInfo(const STableListInfo* pTableList, uint64_t* psuid, uint64_t* uid, int32_t* type) {
486,709✔
3544
  *psuid = pTableList->idInfo.suid;
486,709✔
3545
  *uid = pTableList->idInfo.uid;
486,709✔
3546
  *type = pTableList->idInfo.tableType;
486,709✔
3547
}
486,709✔
3548

3549
int32_t tableListGetTableGroupId(const STableListInfo* pTableList, uint64_t tableUid, uint64_t* gid, uint64_t* baseGid) {
945,654,716✔
3550
  int32_t* slot = taosHashGet(pTableList->map, &tableUid, sizeof(tableUid));
945,654,716✔
3551
  if (slot == NULL) {
946,069,467✔
3552
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3553
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
3554
  }
3555

3556
  STableKeyInfo* pKeyInfo = taosArrayGet(pTableList->pTableList, *slot);
946,069,467✔
3557
  if (pKeyInfo == NULL) {
946,196,214✔
3558
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3559
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
3560
  }
3561

3562
  *gid = pKeyInfo->groupId;
946,196,214✔
3563
  *baseGid = pKeyInfo->baseGId;
946,221,090✔
3564
  
3565
  return 0;
946,215,211✔
3566
}
3567

3568
// TODO handle the group offset info, fix it, the rule of group output will be broken by this function
3569
// int32_t tableListRemoveTableInfo(STableListInfo* pTableList, uint64_t uid) {
3570
//   int32_t code = TSDB_CODE_SUCCESS;
3571
//   int32_t lino = 0;
3572

3573
//   int32_t* slot = taosHashGet(pTableList->map, &uid, sizeof(uid));
3574
//   if (slot == NULL) {
3575
//     qDebug("table:%" PRIu64 " not found in table list", uid);
3576
//     return 0;
3577
//   }
3578

3579
//   taosArrayRemove(pTableList->pTableList, *slot);
3580
//   code = taosHashRemove(pTableList->map, &uid, sizeof(uid));
3581

3582
//   _end:
3583
//   if (code != TSDB_CODE_SUCCESS) {
3584
//     qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
3585
//   } else {
3586
//     qDebug("uid:%" PRIu64 ", remove from table list", uid);
3587
//   }
3588

3589
//   return code;
3590
// }
3591

3592
int32_t tableListAddTableInfo(STableListInfo* pTableList, uint64_t uid, uint64_t gid) {
1,254,955✔
3593
  int32_t code = TSDB_CODE_SUCCESS;
1,254,955✔
3594
  int32_t lino = 0;
1,254,955✔
3595
  if (pTableList->map == NULL) {
1,254,955✔
3596
    pTableList->map = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
×
3597
    QUERY_CHECK_NULL(pTableList->map, code, lino, _end, terrno);
×
3598
  }
3599

3600
  STableKeyInfo keyInfo = {.uid = uid, .groupId = gid};
1,255,693✔
3601
  void*         p = taosHashGet(pTableList->map, &uid, sizeof(uid));
1,255,324✔
3602
  if (p != NULL) {
1,254,212✔
3603
    qInfo("table:%" PRId64 " already in tableIdList, ignore it", uid);
×
3604
    goto _end;
×
3605
  }
3606

3607
  void* tmp = taosArrayPush(pTableList->pTableList, &keyInfo);
1,254,212✔
3608
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,255,319✔
3609

3610
  int32_t slot = (int32_t)taosArrayGetSize(pTableList->pTableList) - 1;
1,255,319✔
3611
  code = taosHashPut(pTableList->map, &uid, sizeof(uid), &slot, sizeof(slot));
1,255,319✔
3612
  if (code != TSDB_CODE_SUCCESS) {
1,255,693✔
3613
    // we have checked the existence of uid in hash map above
3614
    QUERY_CHECK_CONDITION((code != TSDB_CODE_DUP_KEY), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
3615
    taosArrayPopTailBatch(pTableList->pTableList, 1);  // let's pop the last element in the array list
×
3616
  }
3617

3618
_end:
1,255,693✔
3619
  if (code != TSDB_CODE_SUCCESS) {
1,254,576✔
3620
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3621
  } else {
3622
    qDebug("uid:%" PRIu64 ", groupId:%" PRIu64 " added into table list, slot:%d, total:%d", uid, gid, slot, slot + 1);
1,254,576✔
3623
  }
3624

3625
  return code;
1,256,062✔
3626
}
3627

3628
int32_t tableListGetGroupList(const STableListInfo* pTableList, int32_t ordinalGroupIndex, STableKeyInfo** pKeyInfo,
286,819,627✔
3629
                              int32_t* size) {
3630
  int32_t totalGroups = tableListGetOutputGroups(pTableList);
286,819,627✔
3631
  int32_t numOfTables = 0;
286,829,334✔
3632
  int32_t code = tableListGetSize(pTableList, &numOfTables);
286,845,752✔
3633
  if (code != TSDB_CODE_SUCCESS) {
286,820,721✔
3634
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3635
    return code;
×
3636
  }
3637

3638
  if (ordinalGroupIndex < 0 || ordinalGroupIndex >= totalGroups) {
286,820,721✔
3639
    return TSDB_CODE_INVALID_PARA;
×
3640
  }
3641

3642
  // here handle two special cases:
3643
  // 1. only one group exists, and 2. one table exists for each group.
3644
  if (totalGroups == 1) {
286,820,721✔
3645
    *size = numOfTables;
286,228,896✔
3646
    *pKeyInfo = (*size == 0) ? NULL : taosArrayGet(pTableList->pTableList, 0);
286,167,180✔
3647
    return TSDB_CODE_SUCCESS;
286,203,537✔
3648
  } else if (totalGroups == numOfTables) {
591,825✔
3649
    *size = 1;
540,950✔
3650
    *pKeyInfo = taosArrayGet(pTableList->pTableList, ordinalGroupIndex);
540,950✔
3651
    return TSDB_CODE_SUCCESS;
541,011✔
3652
  }
3653

3654
  int32_t offset = pTableList->groupOffset[ordinalGroupIndex];
51,252✔
3655
  if (ordinalGroupIndex < totalGroups - 1) {
67,189✔
3656
    *size = pTableList->groupOffset[ordinalGroupIndex + 1] - offset;
50,562✔
3657
  } else {
3658
    *size = numOfTables - offset;
16,627✔
3659
  }
3660

3661
  *pKeyInfo = taosArrayGet(pTableList->pTableList, offset);
67,189✔
3662
  return TSDB_CODE_SUCCESS;
67,189✔
3663
}
3664

3665
int32_t tableListGetOutputGroups(const STableListInfo* pTableList) { return pTableList->numOfOuputGroups; }
801,330,393✔
3666

3667
bool oneTableForEachGroup(const STableListInfo* pTableList) { return pTableList->oneTableForEachGroup; }
875,494✔
3668

3669
STableListInfo* tableListCreate() {
323,893,370✔
3670
  STableListInfo* pListInfo = taosMemoryCalloc(1, sizeof(STableListInfo));
323,893,370✔
3671
  if (pListInfo == NULL) {
323,738,790✔
3672
    return NULL;
×
3673
  }
3674

3675
  pListInfo->remainGroups = NULL;
323,738,790✔
3676
  pListInfo->pTableList = taosArrayInit(4, sizeof(STableKeyInfo));
323,773,216✔
3677
  if (pListInfo->pTableList == NULL) {
323,777,835✔
3678
    goto _error;
×
3679
  }
3680

3681
  pListInfo->map = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
323,819,922✔
3682
  if (pListInfo->map == NULL) {
323,998,785✔
3683
    goto _error;
×
3684
  }
3685

3686
  pListInfo->numOfOuputGroups = 1;
323,995,498✔
3687
  return pListInfo;
323,995,129✔
3688

3689
_error:
×
3690
  tableListDestroy(pListInfo);
×
3691
  return NULL;
×
3692
}
3693

3694
void tableListDestroy(STableListInfo* pTableListInfo) {
337,216,505✔
3695
  if (pTableListInfo == NULL) {
337,216,505✔
3696
    return;
13,280,894✔
3697
  }
3698

3699
  taosArrayDestroy(pTableListInfo->pTableList);
323,935,611✔
3700
  taosMemoryFreeClear(pTableListInfo->groupOffset);
323,896,826✔
3701

3702
  taosHashCleanup(pTableListInfo->map);
323,894,308✔
3703
  taosHashCleanup(pTableListInfo->remainGroups);
323,961,383✔
3704
  pTableListInfo->pTableList = NULL;
323,944,002✔
3705
  pTableListInfo->map = NULL;
323,960,777✔
3706
  taosMemoryFree(pTableListInfo);
323,957,598✔
3707
}
3708

3709
void tableListClear(STableListInfo* pTableListInfo) {
952,521✔
3710
  if (pTableListInfo == NULL) {
952,521✔
3711
    return;
×
3712
  }
3713

3714
  taosArrayClear(pTableListInfo->pTableList);
952,521✔
3715
  taosHashClear(pTableListInfo->map);
954,361✔
3716
  taosHashClear(pTableListInfo->remainGroups);
954,735✔
3717
  taosMemoryFree(pTableListInfo->groupOffset);
954,735✔
3718
  pTableListInfo->numOfOuputGroups = 1;
954,735✔
3719
  pTableListInfo->oneTableForEachGroup = false;
954,735✔
3720
}
3721

3722
static int32_t orderbyGroupIdComparFn(const void* p1, const void* p2) {
632,316,125✔
3723
  STableKeyInfo* pInfo1 = (STableKeyInfo*)p1;
632,316,125✔
3724
  STableKeyInfo* pInfo2 = (STableKeyInfo*)p2;
632,316,125✔
3725

3726
  if (pInfo1->groupId == pInfo2->groupId) {
632,316,125✔
3727
    return 0;
591,875,120✔
3728
  } else {
3729
    return pInfo1->groupId < pInfo2->groupId ? -1 : 1;
40,444,846✔
3730
  }
3731
}
3732

3733
int32_t sortTableGroup(STableListInfo* pTableListInfo) {
32,757,939✔
3734
  int32_t code = TSDB_CODE_SUCCESS;
32,757,939✔
3735
  taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
32,757,939✔
3736
  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
32,777,824✔
3737
  if (size == 0) {
32,753,391✔
3738
    pTableListInfo->numOfOuputGroups = 0;
×
3739
    return code;
×
3740
  }
3741

3742
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
32,753,391✔
3743
  if (!pList) {
32,762,395✔
3744
    code = terrno;
×
3745
    goto end;
×
3746
  }
3747

3748
  STableKeyInfo* pInfo = taosArrayGet(pTableListInfo->pTableList, 0);
32,762,395✔
3749
  if (pInfo == NULL) {
32,762,250✔
3750
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3751
    code = terrno;
×
3752
    goto end;
×
3753
  }
3754
  uint64_t gid = pInfo->groupId;
32,762,250✔
3755

3756
  int32_t start = 0;
32,763,012✔
3757
  void*   tmp = taosArrayPush(pList, &start);
32,778,339✔
3758
  if (tmp == NULL) {
32,778,339✔
3759
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3760
    code = terrno;
×
3761
    goto end;
×
3762
  }
3763

3764
  for (int32_t i = 1; i < size; ++i) {
176,885,733✔
3765
    pInfo = taosArrayGet(pTableListInfo->pTableList, i);
144,115,942✔
3766
    if (pInfo == NULL) {
144,104,445✔
3767
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3768
      code = terrno;
×
3769
      goto end;
×
3770
    }
3771
    if (pInfo->groupId != gid) {
144,104,445✔
3772
      tmp = taosArrayPush(pList, &i);
8,292,006✔
3773
      if (tmp == NULL) {
8,292,006✔
3774
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3775
        code = terrno;
×
3776
        goto end;
×
3777
      }
3778
      gid = pInfo->groupId;
8,292,006✔
3779
    }
3780
  }
3781

3782
  pTableListInfo->numOfOuputGroups = taosArrayGetSize(pList);
32,779,432✔
3783
  pTableListInfo->groupOffset = taosMemoryMalloc(sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
32,775,554✔
3784
  if (pTableListInfo->groupOffset == NULL) {
32,751,417✔
3785
    code = terrno;
×
3786
    goto end;
×
3787
  }
3788

3789
  memcpy(pTableListInfo->groupOffset, taosArrayGet(pList, 0), sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
32,753,721✔
3790

3791
end:
32,750,503✔
3792
  taosArrayDestroy(pList);
32,752,202✔
3793
  return code;
32,731,958✔
3794
}
3795

3796
int32_t buildGroupIdMapForAllTables(STableListInfo* pTableListInfo, SReadHandle* pHandle, SScanPhysiNode* pScanNode,
275,954,822✔
3797
                                    SNodeList* group, bool groupSort, uint8_t* digest, SStorageAPI* pAPI, SHashObj* groupIdMap, bool gIdFromBaseId) {
3798
  int32_t code = TSDB_CODE_SUCCESS;
275,954,822✔
3799

3800
  bool   groupByTbname = groupbyTbname(group);
275,954,822✔
3801
  size_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
275,955,080✔
3802
  if (!numOfTables) {
275,927,042✔
3803
    return code;
×
3804
  }
3805
  qDebug("numOfTables:%zu, groupByTbname:%d, group:%p", numOfTables, groupByTbname, group);
275,927,042✔
3806
  if (group == NULL || groupByTbname) {
275,881,618✔
3807
    if (tsCountAlwaysReturnValue && QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode) &&
270,818,938✔
3808
        ((STableScanPhysiNode*)pScanNode)->needCountEmptyTable) {
224,959,677✔
3809
      pTableListInfo->remainGroups =
15,683,420✔
3810
          taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
15,683,420✔
3811
      if (pTableListInfo->remainGroups == NULL) {
15,683,420✔
3812
        return terrno;
×
3813
      }
3814

3815
      for (int i = 0; i < numOfTables; i++) {
78,878,567✔
3816
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
63,195,247✔
3817
        if (!info) {
63,195,843✔
3818
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3819
          return terrno;
×
3820
        }
3821
        info->groupId = groupByTbname ? info->uid : 0;
63,195,843✔
3822
        int32_t tempRes = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId),
63,195,943✔
3823
                                      &(info->uid), sizeof(info->uid));
63,196,539✔
3824
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
63,195,843✔
3825
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3826
          return tempRes;
×
3827
        }
3828
      }
3829
    } else {
3830
      for (int32_t i = 0; i < numOfTables; i++) {
936,969,476✔
3831
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
681,886,304✔
3832
        if (!info) {
681,733,891✔
3833
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3834
          return terrno;
×
3835
        }
3836
        info->groupId = groupByTbname ? info->uid : 0;
681,733,891✔
3837
        
3838
      }
3839
    }
3840
    if (groupIdMap && group != NULL){
270,766,492✔
3841
      getColInfoResultForGroupbyForStream(pHandle->vnode, group, pTableListInfo, pAPI, groupIdMap);
105,028✔
3842
    }
3843

3844
    pTableListInfo->oneTableForEachGroup = groupByTbname;
270,766,492✔
3845
    if (numOfTables == 1 && pTableListInfo->idInfo.tableType == TSDB_CHILD_TABLE) {
270,885,797✔
3846
      pTableListInfo->oneTableForEachGroup = true;
117,432,498✔
3847
    }
3848

3849
    if (groupSort && groupByTbname) {
270,876,563✔
3850
      taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
2,691,904✔
3851
      pTableListInfo->numOfOuputGroups = numOfTables;
2,691,904✔
3852
    } else if (groupByTbname && pScanNode->groupOrderScan) {
268,184,659✔
3853
      pTableListInfo->numOfOuputGroups = numOfTables;
30,072✔
3854
    } else {
3855
      pTableListInfo->numOfOuputGroups = 1;
268,155,024✔
3856
    }
3857
    if (groupSort || pScanNode->groupOrderScan) {
270,827,810✔
3858
      code = sortTableGroup(pTableListInfo);
32,521,231✔
3859
    }
3860
  } else {
3861
    bool initRemainGroups = false;
5,062,680✔
3862
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode)) {
5,062,680✔
3863
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pScanNode;
4,933,248✔
3864
      if (tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable &&
4,933,248✔
3865
          !(groupSort || pScanNode->groupOrderScan)) {
2,317,125✔
3866
        initRemainGroups = true;
2,289,295✔
3867
      }
3868
    }
3869

3870
    code = getColInfoResultForGroupby(pHandle, group, pTableListInfo, digest, pAPI, initRemainGroups, groupIdMap, gIdFromBaseId);
5,062,680✔
3871
    if (code != TSDB_CODE_SUCCESS) {
5,062,459✔
3872
      return code;
×
3873
    }
3874

3875
    if (pScanNode->groupOrderScan) pTableListInfo->numOfOuputGroups = taosArrayGetSize(pTableListInfo->pTableList);
5,062,459✔
3876

3877
    if (groupSort || pScanNode->groupOrderScan) {
5,062,282✔
3878
      code = sortTableGroup(pTableListInfo);
165,962✔
3879
    }
3880
  }
3881

3882
  qTrace("EXEC_GROUP_TRACE build_group_map done scanType:%d group:%p groupByTbname:%d groupSort:%d groupOrderScan:%d "
275,925,948✔
3883
         "oneTableEachGroup:%d numTables:%zu numOutputGroups:%d code:%d",
3884
         nodeType((SNode*)pScanNode), group, groupByTbname, groupSort, pScanNode->groupOrderScan,
3885
         pTableListInfo->oneTableForEachGroup, taosArrayGetSize(pTableListInfo->pTableList),
3886
         pTableListInfo->numOfOuputGroups, code);
3887

3888
  // add all table entry in the hash map
3889
  size_t size = taosArrayGetSize(pTableListInfo->pTableList);
275,928,214✔
3890
  for (int32_t i = 0; i < size; ++i) {
1,048,245,915✔
3891
    STableKeyInfo* p = taosArrayGet(pTableListInfo->pTableList, i);
772,334,369✔
3892
    if (!p) {
772,316,070✔
3893
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3894
      return terrno;
×
3895
    }
3896
    int32_t tempRes = taosHashPut(pTableListInfo->map, &p->uid, sizeof(uint64_t), &i, sizeof(int32_t));
772,316,070✔
3897
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
772,382,909✔
3898
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3899
      return tempRes;
×
3900
    }
3901
  }
3902

3903
  return code;
276,000,540✔
3904
}
3905

3906
int32_t createNonStreamScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
299,006,991✔
3907
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
3908
                                SExecTaskInfo* pTaskInfo, SHashObj* groupIdMap) {
3909
  int64_t     st = taosGetTimestampUs();
299,066,874✔
3910
  const char* idStr = GET_TASKID(pTaskInfo);
299,066,874✔
3911

3912
  if (pHandle == NULL) {
298,829,684✔
3913
    qError("invalid handle, in creating operator tree, %s", idStr);
×
3914
    return TSDB_CODE_INVALID_PARA;
×
3915
  }
3916

3917
  if (pHandle->uid != 0) {
298,829,684✔
3918
    pScanNode->uid = pHandle->uid;
160,913✔
3919
    pScanNode->tableType = TSDB_CHILD_TABLE;
160,913✔
3920
  }
3921
  uint8_t digest[17] = {0};
299,072,699✔
3922
  int32_t code = getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, digest, idStr,
298,983,912✔
3923
                              &pTaskInfo->storageAPI, pTaskInfo->pStreamRuntimeInfo);
299,062,818✔
3924
  if (code != TSDB_CODE_SUCCESS) {
299,083,086✔
3925
    qError("failed to getTableList, code:%s", tstrerror(code));
1,108✔
3926
    return code;
1,108✔
3927
  }
3928

3929
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
299,081,978✔
3930

3931
  int64_t st1 = taosGetTimestampUs();
299,151,987✔
3932
  pTaskInfo->cost.extractListTime = (st1 - st) / 1000.0;
299,151,987✔
3933
  qDebug("extract queried table list completed, %d tables, elapsed time:%.2f ms %s", numOfTables,
299,174,439✔
3934
         pTaskInfo->cost.extractListTime, idStr);
3935

3936
  if (numOfTables == 0) {
299,131,434✔
3937
    qDebug("no table qualified for query, %s", idStr);
23,213,667✔
3938
    return TSDB_CODE_SUCCESS;
23,214,447✔
3939
  }
3940

3941
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, digest, &pTaskInfo->storageAPI, groupIdMap, false);
275,917,767✔
3942
  if (code != TSDB_CODE_SUCCESS) {
275,972,103✔
3943
    return code;
×
3944
  }
3945

3946
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
275,973,872✔
3947
  qDebug("generate group id map completed, elapsed time:%.2f ms %s", pTaskInfo->cost.groupIdMapTime, idStr);
275,970,703✔
3948

3949
  return TSDB_CODE_SUCCESS;
275,968,484✔
3950
}
3951

3952
int32_t createStreamGrpTableListFromCond(SScanPhysiNode* pScanNode, SReadHandle* pHandle, SExecTaskInfo* pTaskInfo, SStreamRuntimeFuncInfo* pStream, STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond, int32_t grpNum, bool hasCalcGrp) {
×
3953
  int32_t code = 0, lino = 0;
×
3954

3955
  for (int32_t i = 0; i < grpNum; ++i) {
×
3956
    SSTriggerGroupReadInfo* pGrpRead = taosArrayGet(pStream->curGrpRead, i);
×
3957
    SSTriggerGroupCalcInfo* pGrpCalc = tSimpleHashGet(pStream->pGroupCalcInfos, &pGrpRead->gid, sizeof(pGrpRead->gid));
×
3958

3959
    pStream->pStreamPartColVals = pGrpCalc->pGroupColVals;
×
3960

3961
    int32_t oldSize = taosArrayGetSize(pTableListInfo->pTableList);
×
3962
    TAOS_CHECK_EXIT(getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, NULL, GET_TASKID(pTaskInfo),
×
3963
                                &pTaskInfo->storageAPI, pTaskInfo->pStreamRuntimeInfo));
3964
    int32_t newSize = taosArrayGetSize(pTableListInfo->pTableList);
×
3965
    for (int32_t i = oldSize; i < newSize; ++i) {
×
3966
      STableKeyInfo* pKey = taosArrayGet(pTableListInfo->pTableList, i);
×
3967
      pKey->baseGId = pGrpRead->gid;
×
3968
      if (!hasCalcGrp) {
×
3969
        pKey->groupId = pGrpRead->gid;
×
3970
      }
3971
    }
3972
    
3973
    qDebug("%s get %d tables for tgrp %" PRIu64, GET_TASKID(pTaskInfo), newSize - oldSize, pGrpRead->gid);
×
3974
  }
3975

3976
  qDebug("%s get total %d tables in table list from cond", GET_TASKID(pTaskInfo), (int32_t)taosArrayGetSize(pTableListInfo->pTableList));
×
3977

3978
_exit:
×
3979

3980
  if (code) {
×
3981
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, __LINE__, tstrerror(code));
×
3982
  }
3983
  
3984
  return code;}
×
3985

3986
int32_t createStreamGrpTableListFromTrig(SExecTaskInfo* pTaskInfo, SStreamRuntimeFuncInfo* pStream, STableListInfo* pTableListInfo, int32_t grpNum, bool hasCalcGrp) {
×
3987
  STableKeyInfo tableKey = {0};
×
3988
  int32_t tblNum = 0, idx = 0;
×
3989
  int32_t code = 0, lino = 0;
×
3990

3991
  if (!hasCalcGrp) {
×
3992
    pTableListInfo->numOfOuputGroups = grpNum;
×
3993
  }
3994
  
3995
  for (int32_t i = 0; i < grpNum; ++i) {
×
3996
    SSTriggerGroupReadInfo* pGrp = taosArrayGet(pStream->curGrpRead, i);
×
3997
    tableKey.baseGId = pGrp->gid;
×
3998
    tblNum = taosArrayGetSize(pGrp->pTables);
×
3999

4000
    if (pStream->stbPartByTbname) {
×
4001
      tableKey.uid = pGrp->gid;
×
4002
      if (!hasCalcGrp) {
×
4003
        tableKey.groupId = pGrp->gid;
×
4004
        pTableListInfo->oneTableForEachGroup = true;
×
4005
      }
4006
      
4007
      TSDB_CHECK_NULL(taosArrayPush(pTableListInfo->pTableList, &tableKey), code, lino, _exit, terrno);
×
4008
      if (!hasCalcGrp) {
×
4009
        int32_t tempRes = taosHashPut(pTableListInfo->map, &tableKey.uid, sizeof(uint64_t), &idx, sizeof(int32_t));
×
4010
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
×
4011
          TAOS_CHECK_EXIT(tempRes);
×
4012
        }      
4013
      }      
4014
      
4015
      qDebug("%s gid:%" PRIu64 " table:%" PRIu64 " added to stream table list, partByTbname", GET_TASKID(pTaskInfo), tableKey.groupId, tableKey.uid);
×
4016
      idx++;
×
4017
      continue;
×
4018
    }
4019
    
4020
    for (int32_t n = 0; n < tblNum; ++n, ++idx) {
×
4021
      uint64_t* pUid = taosArrayGet(pGrp->pTables, n);
×
4022
      tableKey.uid = *pUid;
×
4023
      if (!hasCalcGrp) {
×
4024
        tableKey.groupId = pGrp->gid;
×
4025
      }
4026
      
4027
      TSDB_CHECK_NULL(taosArrayPush(pTableListInfo->pTableList, &tableKey), code, lino, _exit, terrno);
×
4028
      if (!hasCalcGrp) {
×
4029
        int32_t tempRes = taosHashPut(pTableListInfo->map, &tableKey.uid, sizeof(uint64_t), &idx, sizeof(int32_t));
×
4030
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
×
4031
          TAOS_CHECK_EXIT(tempRes);
×
4032
        }
4033
      }
4034

4035
      qDebug("%s gid:%" PRIu64 " table:%" PRIu64 " added to stream table list", GET_TASKID(pTaskInfo), tableKey.groupId, tableKey.uid);
×
4036
    }
4037
  }
4038

4039
  qDebug("%s get total %d tables in table list from trig", GET_TASKID(pTaskInfo), (int32_t)taosArrayGetSize(pTableListInfo->pTableList));
×
4040

4041
_exit:
×
4042

4043
  if (code) {
×
4044
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, __LINE__, tstrerror(code));
×
4045
  }
4046
  
4047
  return code;
×
4048
}
4049

4050
int32_t createStreamMultiGrpTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
×
4051
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
4052
                                SExecTaskInfo* pTaskInfo, SHashObj* groupIdMap) {
4053
  int32_t code = 0, lino = 0;
×
4054
  const char* idStr = GET_TASKID(pTaskInfo);
×
4055
  SStreamRuntimeFuncInfo* pStream = &pTaskInfo->pStreamRuntimeInfo->funcInfo;
×
4056
  bool hasCalcGrp = (pGroupTags && pGroupTags->length > 0);
×
4057
  int32_t grpNum = taosArrayGetSize(pStream->curGrpRead);
×
4058
  SSTriggerGroupReadInfo* pGrp = taosArrayGet(pStream->curGrpRead, 0);
×
4059
  int64_t     st = taosGetTimestampUs();
×
4060

4061
  qDebug("%s start to create table list, hasCalcGrp:%d, tgrpNum:%d", idStr, hasCalcGrp, grpNum);
×
4062
  
4063
  if (taosArrayGetSize(pGrp->pTables) <= 0) {
×
4064
    TAOS_CHECK_EXIT(createStreamGrpTableListFromCond(pScanNode, pHandle, pTaskInfo, pStream, pTableListInfo, pTagCond, pTagIndexCond, grpNum, hasCalcGrp));
×
4065
  } else {
4066
    TAOS_CHECK_EXIT(createStreamGrpTableListFromTrig(pTaskInfo, pStream, pTableListInfo, grpNum, hasCalcGrp));
×
4067
  }
4068
  
4069
  pTableListInfo->idInfo.suid = pScanNode->suid;
×
4070
  pTableListInfo->idInfo.tableType = pScanNode->tableType;
×
4071

4072
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
×
4073

4074
  int64_t st1 = taosGetTimestampUs();
×
4075
  qDebug("%s build multi tgrp table list completed, elapsed time:%.2f ms", idStr, (st1 - st) / 1000.0);
×
4076

4077
  if (numOfTables == 0) {
×
4078
    qDebug("%s no table qualified for stream multi tgrp", idStr);
×
4079
    return TSDB_CODE_SUCCESS;
×
4080
  }
4081

4082
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, NULL, &pTaskInfo->storageAPI, groupIdMap, true);
×
4083
  if (code != TSDB_CODE_SUCCESS) {
×
4084
    return code;
×
4085
  }
4086

4087
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
×
4088
  qDebug("%s generate stream group id map completed, elapsed time:%.2f ms", idStr, pTaskInfo->cost.groupIdMapTime);
×
4089

4090
_exit:
×
4091

4092
  if (code) {
×
4093
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, lino, tstrerror(code));
×
4094
  }
4095
  
4096
  return code;
×
4097
}
4098

4099

4100
int32_t createScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
301,652,300✔
4101
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
4102
                                SExecTaskInfo* pTaskInfo, SHashObj* groupIdMap) {
4103
  int32_t code = 0;
301,652,300✔
4104
  // Ensure curGrpRead is prepared from pre-initialized group calc infos for non-stream external-window initialization.
4105
  if (pTaskInfo->pStreamRuntimeInfo) {
301,652,300✔
4106
    SStreamRuntimeFuncInfo* pStream = &pTaskInfo->pStreamRuntimeInfo->funcInfo;
5,226,321✔
4107
    if (pStream->isMultiGroupCalc && pStream->curGrpRead == NULL && pStream->pGroupCalcInfos != NULL) {
5,225,587✔
4108
      int32_t n = tSimpleHashGetSize(pStream->pGroupCalcInfos);
×
4109
      if (n < 0) n = 0;
×
4110
      pStream->curGrpRead = taosArrayInit_s(sizeof(SSTriggerGroupReadInfo), n > 0 ? n : 4);
×
4111
      if (pStream->curGrpRead == NULL) {
×
4112
        return terrno;
×
4113
      }
4114
      int32_t iter = 0;
×
4115
      SSTriggerGroupCalcInfo* pGrp = tSimpleHashIterate(pStream->pGroupCalcInfos, NULL, &iter);
×
4116
      while (pGrp != NULL) {
×
4117
        uint64_t gid = *(uint64_t*)tSimpleHashGetKey(pGrp, NULL);
×
4118
        SSTriggerGroupReadInfo readInfo = {0};
×
4119
        readInfo.gid = gid;
×
4120
        readInfo.pTables = NULL;  // tables decided by getTableList when needed
×
4121
        if (taosArrayPush(pStream->curGrpRead, &readInfo) == NULL) {
×
4122
          return terrno;
×
4123
        }
4124
        pGrp = tSimpleHashIterate(pStream->pGroupCalcInfos, pGrp, &iter);
×
4125
      }
4126
    }
4127
  }
4128
  if (pTaskInfo->pStreamRuntimeInfo && pTaskInfo->pStreamRuntimeInfo->funcInfo.isMultiGroupCalc /*&&got table list*/) {
301,641,103✔
4129
    code = createStreamMultiGrpTableListInfo(pScanNode, pGroupTags, groupSort,
×
4130
                                    pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo, groupIdMap);
4131
    if (code) {
×
4132
      qError("%s failed to createStreamMultiGrpTableListInfo, code:%s", __func__, tstrerror(code));
×
4133
      return code;
×
4134
    }
4135
  } else if (!pScanNode->node.dynamicOp) {
301,692,381✔
4136
    code = createNonStreamScanTableListInfo(pScanNode, pGroupTags, groupSort,
299,119,541✔
4137
                                    pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo, groupIdMap);
4138
    if (code) {
299,177,849✔
4139
      qError("%s failed to createScanTableListInfo, code:%s", __func__, tstrerror(code));
5✔
4140
      return code;
1,108✔
4141
    }
4142
  }
4143

4144
  return TSDB_CODE_SUCCESS;
301,755,342✔
4145
}
4146

4147

4148
char* getStreamOpName(uint16_t opType) {
9,505,345✔
4149
  switch (opType) {
9,505,345✔
4150
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
×
4151
      return "stream scan";
×
4152
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
9,028,692✔
4153
      return "project";
9,028,692✔
4154
    case QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW:
476,653✔
4155
      return "external window";
476,653✔
4156
  }
4157
  return "error name";
×
4158
}
4159

4160
void printDataBlock(SSDataBlock* pBlock, const char* flag, const char* taskIdStr, int64_t qId) {
708,158,829✔
4161
  if (qDebugFlag & DEBUG_TRACE) {
708,158,829✔
4162
    if (!pBlock) {
1,707,958✔
4163
      qDebug("%" PRIx64 " %s %s %s: Block is Null", qId, taskIdStr, flag, __func__);
59,682✔
4164
      return;
61,110✔
4165
    } else if (pBlock->info.rows == 0) {
1,648,276✔
4166
      qDebug("%" PRIx64 " %s %s %s: Block is Empty.", qId, taskIdStr, flag, __func__);
11,254✔
4167
      return;
11,254✔
4168
    }
4169
    
4170
    char*   pBuf = NULL;
1,637,022✔
4171
    int32_t code = dumpBlockData(pBlock, flag, &pBuf, taskIdStr, qId);
1,637,022✔
4172
    if (code == 0) {
1,637,022✔
4173
      qDebugL("%" PRIx64 " %s %s", qId, __func__, pBuf);
1,637,022✔
4174
      taosMemoryFree(pBuf);
1,637,022✔
4175
    }
4176
  }
4177
}
4178

4179
void printSpecDataBlock(SSDataBlock* pBlock, const char* flag, const char* opStr, const char* taskIdStr) {
×
4180
  if (!pBlock) {
×
4181
    qDebug("%s===stream===%s %s: Block is Null", taskIdStr, flag, opStr);
×
4182
    return;
×
4183
  } else if (pBlock->info.rows == 0) {
×
4184
    qDebug("%s===stream===%s %s: Block is Empty.skey:%" PRId64 ",ekey:%" PRId64 ",version%" PRId64,
×
4185
           taskIdStr, flag, opStr, pBlock->info.window.skey, pBlock->info.window.ekey,
4186
           pBlock->info.version);
4187
    return;
×
4188
  }
4189
  if (qDebugFlag & DEBUG_TRACE) {
×
4190
    char* pBuf = NULL;
×
4191
    char  flagBuf[64];
×
4192
    snprintf(flagBuf, sizeof(flagBuf), "%s %s", flag, opStr);
×
4193
    int32_t code = dumpBlockData(pBlock, flagBuf, &pBuf, taskIdStr, 0);
×
4194
    if (code == 0) {
×
4195
      qDebug("%s", pBuf);
×
4196
      taosMemoryFree(pBuf);
×
4197
    }
4198
  }
4199
}
4200

4201
TSKEY getStartTsKey(STimeWindow* win, const TSKEY* tsCols) { return tsCols == NULL ? win->skey : tsCols[0]; }
14,148,668✔
4202

4203
void updateTimeWindowInfo(SColumnInfoData* pColData, const STimeWindow* pWin, int64_t delta) {
2,147,483,647✔
4204
  int64_t* ts = (int64_t*)pColData->pData;
2,147,483,647✔
4205

4206
  int64_t duration = pWin->ekey > pWin->skey ? pWin->ekey - pWin->skey + delta : pWin->skey - pWin->ekey + delta;
2,147,483,647✔
4207
  ts[2] = duration;            // set the duration
2,147,483,647✔
4208
  ts[3] = pWin->skey;          // window start key
2,147,483,647✔
4209
  ts[4] = pWin->ekey + delta;  // window end key
2,147,483,647✔
4210
}
2,147,483,647✔
4211

4212
int32_t compKeys(const SArray* pSortGroupCols, const char* oldkeyBuf, int32_t oldKeysLen, const SSDataBlock* pBlock,
2,147,483,647✔
4213
                 int32_t rowIndex) {
4214
  SColumnDataAgg* pColAgg = NULL;
2,147,483,647✔
4215
  const char*     isNull = oldkeyBuf;
2,147,483,647✔
4216
  const char*     p = oldkeyBuf + sizeof(int8_t) * pSortGroupCols->size;
2,147,483,647✔
4217

4218
  for (int32_t i = 0; i < pSortGroupCols->size; ++i) {
2,147,483,647✔
4219
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
2,147,483,647✔
4220
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
2,147,483,647✔
4221
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
2,147,483,647✔
4222

4223
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
2,147,483,647✔
4224
      if (isNull[i] != 1) return 1;
314,153,802✔
4225
    } else {
4226
      if (isNull[i] != 0) return 1;
2,147,483,647✔
4227
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,147,483,647✔
4228
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,147,483,647✔
4229
        int32_t len = getJsonValueLen(val);
×
4230
        if (memcmp(p, val, len) != 0) return 1;
×
4231
        p += len;
×
4232
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,147,483,647✔
4233
        if (IS_STR_DATA_BLOB(pCol->type)) {
1,443,211,362✔
UNCOV
4234
          if (memcmp(p, val, blobDataTLen(val)) != 0) return 1;
×
4235
          p += blobDataTLen(val);
×
4236
        } else {
4237
          if (memcmp(p, val, varDataTLen(val)) != 0) return 1;
1,442,836,494✔
4238
          p += varDataTLen(val);
1,442,172,198✔
4239
        }
4240
      } else {
4241
        if (0 != memcmp(p, val, pCol->bytes)) return 1;
2,147,483,647✔
4242
        p += pCol->bytes;
2,147,483,647✔
4243
      }
4244
    }
4245
  }
4246
  if ((int32_t)(p - oldkeyBuf) != oldKeysLen) return 1;
2,147,483,647✔
4247
  return 0;
2,147,483,647✔
4248
}
4249

4250
int32_t buildKeys(char* keyBuf, const SArray* pSortGroupCols, const SSDataBlock* pBlock, int32_t rowIndex) {
1,904,244✔
4251
  uint32_t        colNum = pSortGroupCols->size;
1,904,244✔
4252
  SColumnDataAgg* pColAgg = NULL;
1,904,244✔
4253
  char*           isNull = keyBuf;
1,904,244✔
4254
  char*           p = keyBuf + sizeof(int8_t) * colNum;
1,904,244✔
4255

4256
  for (int32_t i = 0; i < colNum; ++i) {
5,212,908✔
4257
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
3,308,664✔
4258
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
3,308,664✔
4259
    if (pCol->slotId > pBlock->pDataBlock->size) continue;
3,308,664✔
4260

4261
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
3,308,664✔
4262

4263
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
6,617,328✔
4264
      isNull[i] = 1;
160,200✔
4265
    } else {
4266
      isNull[i] = 0;
3,148,464✔
4267
      const char* val = colDataGetData(pColInfoData, rowIndex);
3,148,464✔
4268
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
3,148,464✔
4269
        int32_t len = getJsonValueLen(val);
×
4270
        memcpy(p, val, len);
×
4271
        p += len;
×
4272
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
3,148,464✔
4273
        if (IS_STR_DATA_BLOB(pCol->type)) {
1,176,936✔
4274
          blobDataCopy(p, val);
×
4275
          p += blobDataTLen(val);
×
4276
        } else {
4277
          varDataCopy(p, val);
1,176,936✔
4278
          p += varDataTLen(val);
1,176,936✔
4279
        }
4280
      } else {
4281
        memcpy(p, val, pCol->bytes);
1,971,528✔
4282
        p += pCol->bytes;
1,971,528✔
4283
      }
4284
    }
4285
  }
4286
  return (int32_t)(p - keyBuf);
1,904,244✔
4287
}
4288

4289
uint64_t calcGroupId(char* pData, int32_t len) {
2,147,483,647✔
4290
  T_MD5_CTX context;
2,147,483,647✔
4291
  tMD5Init(&context);
2,147,483,647✔
4292
  tMD5Update(&context, (uint8_t*)pData, len);
2,147,483,647✔
4293
  tMD5Final(&context);
2,147,483,647✔
4294

4295
  // NOTE: only extract the initial 8 bytes of the final MD5 digest
4296
  uint64_t id = 0;
2,147,483,647✔
4297
  memcpy(&id, context.digest, sizeof(uint64_t));
2,147,483,647✔
4298
  if (0 == id) memcpy(&id, context.digest + 8, sizeof(uint64_t));
2,147,483,647✔
4299
  return id;
2,147,483,647✔
4300
}
4301

4302
SNodeList* makeColsNodeArrFromSortKeys(SNodeList* pSortKeys) {
126,024✔
4303
  SNode*     node;
4304
  SNodeList* ret = NULL;
126,024✔
4305
  FOREACH(node, pSortKeys) {
384,480✔
4306
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)node;
258,456✔
4307
    int32_t           code = nodesListMakeAppend(&ret, pSortKey->pExpr);
258,456✔
4308
    if (code != TSDB_CODE_SUCCESS) {
258,456✔
4309
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4310
      terrno = code;
×
4311
      return NULL;
×
4312
    }
4313
  }
4314
  return ret;
126,024✔
4315
}
4316

4317
int32_t extractKeysLen(const SArray* keys, int32_t* pLen) {
126,024✔
4318
  int32_t code = TSDB_CODE_SUCCESS;
126,024✔
4319
  int32_t lino = 0;
126,024✔
4320
  int32_t len = 0;
126,024✔
4321
  int32_t keyNum = taosArrayGetSize(keys);
126,024✔
4322
  for (int32_t i = 0; i < keyNum; ++i) {
320,400✔
4323
    SColumn* pCol = (SColumn*)taosArrayGet(keys, i);
194,376✔
4324
    QUERY_CHECK_NULL(pCol, code, lino, _end, terrno);
194,376✔
4325
    len += pCol->bytes;
194,376✔
4326
  }
4327
  len += sizeof(int8_t) * keyNum;  // null flag
126,024✔
4328
  *pLen = len;
126,024✔
4329

4330
_end:
126,024✔
4331
  if (code != TSDB_CODE_SUCCESS) {
126,024✔
4332
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4333
  }
4334
  return code;
126,024✔
4335
}
4336

4337
int32_t parseErrorMsgFromAnalyticServer(SJson* pJson, const char* typeStr, const char* pId) {
×
4338
  int32_t code = TSDB_CODE_ANA_ANODE_RETURN_ERROR;
×
4339
  char*   pMsg = NULL;
×
4340
  if (pJson == NULL) {
×
4341
    return code;
×
4342
  }
4343

4344
  int32_t ret = tjsonDupStringValue(pJson, "msg", &pMsg);
×
4345
  if (ret == 0 && pMsg != NULL) {
×
4346
    qError("%s failed to exec %s operation, msg:%s", pId, typeStr, pMsg);
×
4347
    if (strstr(pMsg, "white noise") != NULL) {
×
4348
      code = TSDB_CODE_ANA_WN_DATA;
×
4349
    } else if (strstr(pMsg, "white-noise") != NULL) {
×
4350
      code = TSDB_CODE_ANA_WN_DATA;
×
4351
    } else if (strstr(pMsg, "[Errno 111] Connection refused") != NULL) {
×
4352
      code = TSDB_CODE_ANA_ALGO_NOT_LOAD;
×
4353
    } else if (strstr(pMsg, "failed to load model") != NULL) {
×
4354
      code = TSDB_CODE_ANA_ALGO_NOT_LOAD;
×
4355
    }
4356
  } else {
4357
    qError("%s failed to extract msg from server, unknown error", pId);
×
4358
  }
4359

4360
  taosMemoryFreeClear(pMsg);
×
4361
  return code;
×
4362
}
4363

4364

4365
int32_t createExprSubQResBlock(SSDataBlock** ppBlock, SDataType* pResType, int32_t cols) {
32,962,017✔
4366
  int32_t code = 0;
32,962,017✔
4367
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
32,962,017✔
4368
  if (pBlock == NULL) {
32,961,415✔
4369
    return terrno;
×
4370
  }
4371

4372
  pBlock->pDataBlock = taosArrayInit(cols, sizeof(SColumnInfoData));
32,961,415✔
4373
  if (pBlock->pDataBlock == NULL) {
32,960,789✔
4374
    code = terrno;
×
4375
    taosMemoryFree(pBlock);
×
4376
    return code;
×
4377
  }
4378

4379
  for (int32_t i = 0; i < cols; ++i) {
67,282,415✔
4380
    SColumnInfoData idata = {0};
34,320,398✔
4381
    if (pResType) {
34,319,182✔
4382
      idata = createColumnInfoData(pResType->type, pResType->bytes, 0);
32,260,546✔
4383
      idata.info.scale = pResType->scale;
32,261,762✔
4384
      idata.info.precision = pResType->precision;
32,261,762✔
4385
    }
4386
    
4387
    code = blockDataAppendColInfo(pBlock, &idata);
34,319,784✔
4388
    if (code != TSDB_CODE_SUCCESS) {
34,320,398✔
UNCOV
4389
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
UNCOV
4390
      blockDataDestroy(pBlock);
×
4391
      *ppBlock = NULL;
×
4392
      return code;
×
4393
    }
4394
  }
4395

4396
  *ppBlock = pBlock;
32,962,017✔
4397

4398
  return code;
32,962,017✔
4399
}
4400

4401

4402
int32_t extractSingleRspBlock(SRetrieveTableRsp* pRetrieveRsp, SSDataBlock* pb) {
32,962,017✔
4403
  int32_t            code = TSDB_CODE_SUCCESS;
32,962,017✔
4404
  int32_t            lino = 0;
32,962,017✔
4405
  void*              decompBuf = NULL;
32,962,017✔
4406

4407
  char* pNextStart = pRetrieveRsp->data;
32,962,017✔
4408
  char* pStart = pNextStart;
32,962,017✔
4409

4410
  int32_t index = 0;
32,961,403✔
4411

4412
  if (pRetrieveRsp->compressed) {  // decompress the data
32,961,403✔
4413
    decompBuf = taosMemoryMalloc(pRetrieveRsp->payloadLen);
×
4414
    QUERY_CHECK_NULL(decompBuf, code, lino, _end, terrno);
×
4415
  }
4416

4417
  int32_t compLen = *(int32_t*)pStart;
32,962,017✔
4418
  pStart += sizeof(int32_t);
32,962,017✔
4419

4420
  int32_t rawLen = *(int32_t*)pStart;
32,961,403✔
4421
  pStart += sizeof(int32_t);
32,961,403✔
4422
  QUERY_CHECK_CONDITION((compLen <= rawLen && compLen != 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
32,960,789✔
4423

4424
  pNextStart = pStart + compLen;
32,960,789✔
4425
  if (pRetrieveRsp->compressed && (compLen < rawLen)) {
32,960,801✔
4426
    int32_t t = tsDecompressString(pStart, compLen, 1, decompBuf, rawLen, ONE_STAGE_COMP, NULL, 0);
×
4427
    QUERY_CHECK_CONDITION((t == rawLen), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
4428
    pStart = decompBuf;
×
4429
  }
4430

4431
  code = blockDecodeInternal(pb, pStart, (const char**)&pStart);
32,962,017✔
4432
  if (code != 0) {
32,961,413✔
4433
    taosMemoryFreeClear(pRetrieveRsp);
×
4434
    goto _end;
×
4435
  }
4436

4437
_end:
32,961,413✔
4438
  if (code != TSDB_CODE_SUCCESS) {
32,961,413✔
4439
    blockDataDestroy(pb);
×
4440
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4441
  }
4442
  return code;
32,961,413✔
4443
}
4444

4445
int32_t setValueFromResBlock(STaskSubJobCtx* ctx, SValueNode* pRes, SSDataBlock* pBlock) {
21,124,604✔
4446
  int32_t code = 0;
21,124,604✔
4447
  bool needFree = true;
21,124,604✔
4448
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
21,124,604✔
4449
  if (NULL == pBlock->pDataBlock || 1 > colNum || pBlock->info.rows > 1) {
21,122,774✔
UNCOV
4450
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d, rows:%" PRId64, 
×
4451
      ctx->idStr, pBlock->pDataBlock, colNum, pBlock->info.rows);
4452
    return TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
×
4453
  }
4454
  
4455
  pRes->flag &= (~VALUE_FLAG_VAL_UNSET);
21,124,002✔
4456
  pRes->translate = true;
21,124,604✔
4457
  
4458
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
21,123,990✔
4459
  if (colDataIsNull_s(pCol, 0)) {
21,123,990✔
4460
    pRes->isNull = true;
5,704,698✔
4461
  } else {
4462
    code = nodesSetValueNodeValueExt(pRes, colDataGetData(pCol, 0), &needFree);
15,419,292✔
4463
  }
4464

4465
  if (!needFree) {
21,123,388✔
4466
    pCol->pData = NULL;
2,096,640✔
4467
  }
4468

4469
  return code;
21,123,388✔
4470
}
4471

4472
void handleRemoteValueRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
3,791,024✔
4473
  SSDataBlock*      pResBlock = NULL;
3,791,024✔
4474
  SExecTaskInfo*    pTaskInfo = ctx->pTaskInfo;
3,791,024✔
4475
  SRemoteValueNode* pRemote = (SRemoteValueNode*)pParam->pRes;
3,791,024✔
4476

4477
  qDebug("%s scl fetch value rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
3,791,024✔
4478

4479
  if (IS_STREAM_MODE(pTaskInfo)) {
3,791,024✔
4480
    SNode** ppRes = taosArrayGet(ctx->subResNodes, pParam->subQIdx);
115,256✔
4481
    if (NULL == *ppRes && 0 == pRsp->numOfRows) {
115,256✔
4482
      pRemote->val.node.type = QUERY_NODE_VALUE;
×
4483
      pRemote->val.isNull = true;
×
4484
      pRemote->val.translate = true;
×
4485
      pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
×
4486
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
×
4487
    } else if (pRsp->numOfRows > 0 || pRsp->numOfBlocks > 0) {
115,256✔
4488
      ctx->code = createExprSubQResBlock(&pResBlock, &((SRemoteValueNode*)pParam->pRes)->val.node.resType, 1);
57,628✔
4489
      if (TSDB_CODE_SUCCESS == ctx->code) {
57,628✔
4490
        ctx->code = blockDataEnsureCapacity(pResBlock, 1);
57,628✔
4491
      }
4492
      if (TSDB_CODE_SUCCESS == ctx->code) {
57,628✔
4493
        ctx->code = extractSingleRspBlock(pRsp, pResBlock);
57,628✔
4494
      }
4495
      if (TSDB_CODE_SUCCESS == ctx->code) {
57,628✔
4496
        ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
57,628✔
4497
        pRemote->val.node.type = QUERY_NODE_REMOTE_VALUE;
57,628✔
4498
      }
4499
      if (TSDB_CODE_SUCCESS == ctx->code) {
57,628✔
4500
        taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
57,628✔
4501
      }
4502

4503
      blockDataDestroy(pResBlock);
57,628✔
4504
    } else if (NULL != *ppRes && 0 == pRsp->numOfRows) {
57,628✔
4505
      pRemote->val.node.type = QUERY_NODE_VALUE;
57,628✔
4506
      pRsp->completed = true;
57,628✔
4507
    }
4508

4509
    *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
115,256✔
4510

4511
    if (!(*fetchDone)) {
115,256✔
4512
      int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
57,628✔
4513
      if (TSDB_CODE_SUCCESS != code) {
57,628✔
4514
        ctx->code = code;
×
4515
        *fetchDone = true;
×
4516
      }
4517
    }
4518

4519
    return;
115,256✔
4520
  }
4521

4522
  *fetchDone = true;
3,675,154✔
4523

4524
  if (pRsp->numOfRows > 1 || pRsp->numOfBlocks > 1 || !pRsp->completed) {
3,675,768✔
4525
    qError("%s invalid scl value fetch rsp received, subQIdx:%d, rows:%" PRId64 ", blocks:%d, completed:%d", 
614✔
4526
      ctx->idStr, pParam->subQIdx, pRsp->numOfRows, pRsp->numOfBlocks, pRsp->completed);
4527
    ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
614✔
4528

4529
    return;
×
4530
  }
4531

4532
  if (1 != pRsp->numOfCols && pRsp->numOfRows > 0) {
3,675,154✔
4533
    qError("%s invalid scl value fetch rsp received, subQIdx:%d, cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfCols);
×
4534
    ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_COLS;
×
4535

4536
    return;
×
4537
  }
4538

4539
  if (0 == pRsp->numOfRows) {
3,675,768✔
4540
    pRemote->val.node.type = QUERY_NODE_VALUE;
195,826✔
4541
    pRemote->val.isNull = true;
195,826✔
4542
    pRemote->val.translate = true;
195,826✔
4543
    pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
195,826✔
4544
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
195,826✔
4545

4546
    return;
195,826✔
4547
  }
4548
  
4549
  ctx->code = createExprSubQResBlock(&pResBlock, &((SRemoteValueNode*)pParam->pRes)->val.node.resType, 1);
3,478,714✔
4550
  if (TSDB_CODE_SUCCESS == ctx->code) {
3,479,942✔
4551
    ctx->code = blockDataEnsureCapacity(pResBlock, 1);
3,479,328✔
4552
  }
4553
  if (TSDB_CODE_SUCCESS == ctx->code) {
3,479,942✔
4554
    ctx->code = extractSingleRspBlock(pRsp, pResBlock);
3,479,942✔
4555
  }
4556
  if (TSDB_CODE_SUCCESS == ctx->code) {
3,479,328✔
4557
    ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
3,479,328✔
4558
  }
4559
  if (TSDB_CODE_SUCCESS == ctx->code) {
3,479,328✔
4560
    pRemote->val.node.type = QUERY_NODE_VALUE;
3,479,942✔
4561
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
3,479,942✔
4562
  }
4563

4564
  blockDataDestroy(pResBlock);  
3,479,942✔
4565
}
4566

4567

4568
int32_t setValueListFromResBlock(STaskSubJobCtx* ctx, SRemoteValueListNode* pRes, SSDataBlock* pBlock) {
11,137,158✔
4569
  int32_t code = 0, lino = 0;
11,137,158✔
4570
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
11,137,158✔
4571
  if (NULL == pBlock->pDataBlock || 1 != colNum) {
11,137,158✔
UNCOV
4572
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d", ctx->idStr, pBlock->pDataBlock, colNum);
×
4573
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4574
  }
4575

4576
  pRes->hasValue = true;
11,137,158✔
4577
  
4578
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
11,137,158✔
4579
  TAOS_CHECK_EXIT(scalarBuildRemoteListHash(ctx->idStr, pRes, pCol, pBlock->info.rows));
11,137,158✔
4580

4581
_exit:
10,324,038✔
4582

4583
  if (code) {
11,137,158✔
4584
    qError("%s %s failed with error: %s", ctx->idStr, __func__, tstrerror(code));
813,120✔
4585
  }
4586
  
4587
  return code;
11,137,158✔
4588
}
4589

4590

4591

4592
void handleRemoteValueListRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
14,457,678✔
4593
  SSDataBlock* pResBlock = NULL;
14,457,678✔
4594
  SRemoteValueListNode* pRemote = (SRemoteValueListNode*)pParam->pRes;
14,457,678✔
4595
  SExecTaskInfo* pTaskInfo = ctx->pTaskInfo;
14,457,678✔
4596

4597
  qDebug("%s scl fetch valueList rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
14,457,678✔
4598

4599
  if (pRsp->numOfRows > 0) {
14,457,678✔
4600
    if (1 != pRsp->numOfCols) {
11,137,158✔
4601
      qError("%s invalid scl valueList fetch rsp received, subQIdx:%d, cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfCols);
×
4602
      ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_COLS;
×
4603
      *fetchDone = true;
×
4604
      return;
×
4605
    }
4606

4607
    ctx->code = createExprSubQResBlock(&pResBlock, &((SExprNode*)pParam->pRes)->resType, 1);
11,137,158✔
4608
    if (TSDB_CODE_SUCCESS == ctx->code) {
11,137,158✔
4609
      ctx->code = blockDataEnsureCapacity(pResBlock, pRsp->numOfRows);
11,137,158✔
4610
    }
4611
    if (TSDB_CODE_SUCCESS == ctx->code) {
11,137,158✔
4612
      ctx->code = extractSingleRspBlock(pRsp, pResBlock);
11,137,158✔
4613
    }
4614
    if (TSDB_CODE_SUCCESS == ctx->code) {
11,136,554✔
4615
      ctx->code = setValueListFromResBlock(ctx, pRemote, pResBlock);
11,137,158✔
4616
    }
4617
    if (TSDB_CODE_SUCCESS == ctx->code && pRsp->completed) {
11,136,865✔
4618
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
8,913,056✔
4619
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
8,913,056✔
4620
    }
4621

4622
    blockDataDestroy(pResBlock);  
11,136,865✔
4623
  } else if (0 == pRsp->numOfRows && pRsp->completed) {
3,320,520✔
4624
    if (!pRemote->hasValue) {
3,320,520✔
4625
      ctx->code = scalarBuildRemoteListHash(ctx->idStr, pRemote, NULL, 0);
3,300,077✔
4626
    }
4627
    if (TSDB_CODE_SUCCESS == ctx->code) {    
3,320,520✔
4628
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
3,045,000✔
4629
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
3,045,000✔
4630
    }
4631
  }
4632

4633
  if (IS_STREAM_MODE(pTaskInfo) && 0 == pRsp->numOfRows) {
14,457,678✔
4634
    if (!pRemote->hasValue) {
21,000✔
4635
      ctx->code = scalarBuildRemoteListHash(ctx->idStr, pRemote, NULL, 0);
557✔
4636
    }
4637
    if (TSDB_CODE_SUCCESS == ctx->code) {
21,000✔
4638
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
21,000✔
4639
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
21,000✔
4640
    }
4641

4642
    pRsp->completed = true;
21,000✔
4643
  }
4644

4645
  *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
14,457,678✔
4646

4647
  if (!(*fetchDone)) {
14,457,385✔
4648
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
1,410,689✔
4649
    if (TSDB_CODE_SUCCESS != code) {
1,410,982✔
4650
      ctx->code = code;
×
4651
      *fetchDone = true;
×
4652
    }
4653
  }
4654
}
4655

4656
int32_t setRowHasNullFromResBlock(STaskSubJobCtx* ctx, bool* hasNull, SSDataBlock* pBlock) {
17,587,034✔
4657
  int32_t code = 0;
17,587,034✔
4658
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
17,587,034✔
4659
  if (2 != colNum) {
17,586,432✔
4660
    qError("%s invalid scl fetch res block, colNum:%d", ctx->idStr, colNum);
×
4661
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4662
  }
4663
  
4664
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 1);
17,586,432✔
4665
  if (colDataIsNull_s(pCol, 0)) {
17,586,432✔
4666
    qError("%s invalid has_null res since it's null", ctx->idStr);
×
4667
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4668
  } else {
4669
    *hasNull = *(bool*)colDataGetData(pCol, 0);
17,586,432✔
4670
  }
4671

4672
  return code;
17,587,034✔
4673
}
4674

4675
void handleRemoteRowRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
23,837,234✔
4676
  SSDataBlock* pResBlock = NULL;
23,837,234✔
4677
  SExecTaskInfo* pTaskInfo = ctx->pTaskInfo;
23,837,234✔
4678

4679
  qDebug("%s scl fetch row rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
23,837,234✔
4680

4681
  SRemoteRowNode* pRemote = (SRemoteRowNode*)pParam->pRes;
23,837,234✔
4682

4683
  if (IS_STREAM_MODE(pTaskInfo)) {
23,837,234✔
4684
    SNode** ppRes = taosArrayGet(ctx->subResNodes, pParam->subQIdx);
1,200✔
4685
    if (NULL == *ppRes && 0 == pRsp->numOfRows) {
1,200✔
4686
      pRemote->valSet = true;
×
4687
      pRemote->hasValue = false;
×
4688
      pRemote->hasNull = false;
×
4689
      pRemote->val.isNull = true;
×
4690
      pRemote->val.translate = true;
×
4691
      pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
×
4692
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
×
4693
    } else if (pRsp->numOfRows > 0 || pRsp->numOfBlocks > 0) {
1,200✔
4694
      if (2 != pRsp->numOfCols) {
600✔
4695
        qError("%s invalid scl fetch row rsp received, subQIdx:%d, cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfCols);
×
4696
        ctx->code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4697
        *fetchDone = true;
×
4698
        return;
×
4699
      }
4700

4701
      ctx->code = createExprSubQResBlock(&pResBlock, &pRemote->val.node.resType, 1);
600✔
4702
      if (TSDB_CODE_SUCCESS == ctx->code) {
600✔
4703
        SColumnInfoData idata = createColumnInfoData(TSDB_DATA_TYPE_BOOL, tDataTypes[TSDB_DATA_TYPE_BOOL].bytes, 0);
600✔
4704
        ctx->code = blockDataAppendColInfo(pResBlock, &idata);
600✔
4705
      }
4706
      if (TSDB_CODE_SUCCESS == ctx->code) {
600✔
4707
        ctx->code = blockDataEnsureCapacity(pResBlock, 1);
600✔
4708
      }
4709
      if (TSDB_CODE_SUCCESS == ctx->code) {
600✔
4710
        ctx->code = extractSingleRspBlock(pRsp, pResBlock);
600✔
4711
      }
4712
      if (TSDB_CODE_SUCCESS == ctx->code) {
600✔
4713
        ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
600✔
4714
      }
4715
      if (TSDB_CODE_SUCCESS == ctx->code) {
600✔
4716
        ctx->code = setRowHasNullFromResBlock(ctx, &pRemote->hasNull, pResBlock);
600✔
4717
      }
4718
      if (TSDB_CODE_SUCCESS == ctx->code) {
600✔
4719
        pRemote->valSet = true;
600✔
4720
        pRemote->hasValue = true;
600✔
4721
        pRemote->val.node.type = QUERY_NODE_REMOTE_ROW;
600✔
4722
        taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
600✔
4723
      }
4724

4725
      blockDataDestroy(pResBlock);
600✔
4726
    } else if (NULL != *ppRes && 0 == pRsp->numOfRows) {
600✔
4727
      pRemote->val.node.type = QUERY_NODE_VALUE;
600✔
4728
      pRsp->completed = true;
600✔
4729
    }
4730

4731
    *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
1,200✔
4732

4733
    if (!(*fetchDone)) {
1,200✔
4734
      ctx->code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
600✔
4735
      if (TSDB_CODE_SUCCESS != ctx->code) {
600✔
4736
        *fetchDone = true;
×
4737
      }
4738
    }
4739

4740
    return;
1,200✔
4741
  }
4742

4743
  *fetchDone = true;
23,835,432✔
4744

4745
  if (pRsp->numOfRows > 1 || pRsp->numOfBlocks > 1 || !pRsp->completed) {
23,836,034✔
4746
    qError("%s invalid scl fetch row rsp received, subQIdx:%d, rows:%" PRId64 ", blocks:%d, completed:%d",
×
4747
      ctx->idStr, pParam->subQIdx, pRsp->numOfRows, pRsp->numOfBlocks, pRsp->completed);
4748
    ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
×
4749

4750
    return;
×
4751
  }
4752

4753
  if (0 == pRsp->numOfRows) {
23,836,034✔
4754
    pRemote->valSet = true;
6,249,600✔
4755
    pRemote->hasValue = false;
6,249,600✔
4756
    pRemote->hasNull = false;
6,249,600✔
4757
    pRemote->val.isNull = true;
6,249,600✔
4758
    pRemote->val.translate = true;
6,249,600✔
4759
    pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
6,249,600✔
4760
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
6,249,600✔
4761

4762
    return;
6,249,600✔
4763
  }
4764

4765
  if (2 != pRsp->numOfCols) {
17,585,832✔
4766
    qError("%s invalid scl fetch row rsp received, subQIdx:%d, cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfCols);
×
4767
    ctx->code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4768

4769
    return;
×
4770
  }
4771

4772
  ctx->code = createExprSubQResBlock(&pResBlock, &pRemote->val.node.resType, 1);
17,586,434✔
4773
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,585,832✔
4774
    SColumnInfoData idata = createColumnInfoData(TSDB_DATA_TYPE_BOOL, tDataTypes[TSDB_DATA_TYPE_BOOL].bytes, 0);
17,586,434✔
4775
    ctx->code = blockDataAppendColInfo(pResBlock, &idata);
17,586,434✔
4776
  }
4777
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,586,434✔
4778
    ctx->code = blockDataEnsureCapacity(pResBlock, 1);
17,586,434✔
4779
  }
4780
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,586,434✔
4781
    ctx->code = extractSingleRspBlock(pRsp, pResBlock);
17,586,434✔
4782
  }
4783
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,586,434✔
4784
    ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
17,586,434✔
4785
  }
4786
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,586,434✔
4787
    ctx->code = setRowHasNullFromResBlock(ctx, &pRemote->hasNull, pResBlock);
17,586,434✔
4788
  }
4789
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,586,434✔
4790
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
17,586,434✔
4791
  }
4792
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,585,832✔
4793
    pRemote->valSet = true;
17,586,434✔
4794
    pRemote->hasValue = true;
17,585,832✔
4795
  }
4796

4797
  blockDataDestroy(pResBlock);
17,586,434✔
4798
}
4799

4800

4801
int32_t setZeroRowsResValue(STaskSubJobCtx* ctx, SValueNode* pRes, int32_t rows) {
308,934✔
4802
  pRes->node.type = QUERY_NODE_VALUE;
308,934✔
4803
  pRes->flag &= (~VALUE_FLAG_VAL_UNSET);
308,934✔
4804
  pRes->translate = true;
308,934✔
4805
  
4806
  return nodesSetValueNodeValue(pRes, &rows);
308,934✔
4807
}
4808

4809
void handleRemoteZeroRowsRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
308,634✔
4810
  SRemoteZeroRowsNode* pRemote = (SRemoteZeroRowsNode*)pParam->pRes;
308,634✔
4811
  SExecTaskInfo* pTaskInfo = ctx->pTaskInfo;
308,634✔
4812

4813
  qDebug("%s scl fetch zeroRows rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
308,634✔
4814

4815
  int32_t resRows = (pRsp->numOfRows > 0) ? 1 : 0;
308,634✔
4816
  if (resRows > 0 || pRsp->completed) {
308,634✔
4817
    ctx->code = setZeroRowsResValue(ctx, &pRemote->val, resRows);
308,634✔
4818
    if (TSDB_CODE_SUCCESS == ctx->code) {
308,634✔
4819
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
308,634✔
4820
    }
4821

4822
    *fetchDone = true;
308,634✔
4823
  } else {
4824
    *fetchDone = false;
×
4825
  }
4826

4827
  if (IS_STREAM_MODE(pTaskInfo) && 0 == pRsp->numOfRows) {
308,634✔
4828
    ctx->code = setZeroRowsResValue(ctx, &pRemote->val, 0);
300✔
4829
    if (TSDB_CODE_SUCCESS == ctx->code) {
300✔
4830
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
300✔
4831
    }
4832

4833
    pRsp->completed = true;
300✔
4834
    *fetchDone = true;
300✔
4835
  }
4836

4837
  if (!(*fetchDone)) {
308,634✔
4838
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
×
4839
    if (TSDB_CODE_SUCCESS != code) {
×
4840
      ctx->code = code;
×
4841
      *fetchDone = true;
×
4842
    }
4843
  }
4844
}
308,634✔
4845

4846

4847

4848
int32_t setTableFromResBlock(STaskSubJobCtx* ctx, SRemoteTableNode* pRes, SSDataBlock** ppBlock) {
700,255✔
4849
  int32_t code = 0, lino = 0;
700,255✔
4850
  
4851
  if (NULL == pRes->pResBlks) {
700,255✔
4852
    pRes->pResBlks = taosArrayInit(10, POINTER_BYTES);
571,603✔
4853
    TSDB_CHECK_NULL(pRes->pResBlks, code, lino, _exit, terrno);
571,603✔
4854
  }
4855

4856
  TSDB_CHECK_NULL(taosArrayPush(pRes->pResBlks, ppBlock), code, lino, _exit, terrno);
1,400,057✔
4857

4858
_exit:
699,802✔
4859

4860
  if (code) {
699,802✔
4861
    blockDataDestroy(*ppBlock);    
×
4862
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4863
  }
4864

4865
  *ppBlock = NULL;
699,802✔
4866
  
4867
  return code;
700,255✔
4868
}
4869

4870

4871
void handleRemoteTableRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
702,973✔
4872
  SSDataBlock* pResBlock = NULL;
702,973✔
4873
  SRemoteTableNode* pRemote = (SRemoteTableNode*)pParam->pRes;
702,973✔
4874

4875
  qDebug("%s scl fetch remoteTable rsp received, subQIdx:%d, rows:%" PRId64 ", cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfRows, pRsp->numOfCols);
702,973✔
4876

4877
  if (pRsp->numOfRows > 0) {
702,973✔
4878
    ctx->code = createExprSubQResBlock(&pResBlock, NULL, pRemote->resCols);
700,255✔
4879
    if (TSDB_CODE_SUCCESS == ctx->code) {
700,255✔
4880
      ctx->code = extractSingleRspBlock(pRsp, pResBlock);
700,255✔
4881
    }
4882
    if (TSDB_CODE_SUCCESS == ctx->code) {
700,255✔
4883
      ctx->code = setTableFromResBlock(ctx, pRemote, &pResBlock);
699,802✔
4884
    }
4885
    if (TSDB_CODE_SUCCESS == ctx->code && pRsp->completed) {
700,255✔
4886
      pRemote->flag &= (~REMOTE_TABLE_FLAG_VAL_UNSET);
571,603✔
4887
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
571,603✔
4888
    }
4889

4890
    blockDataDestroy(pResBlock);  
700,255✔
4891
  } else if (0 == pRsp->numOfRows && pRsp->completed) {
2,718✔
4892
    pRemote->flag &= (~REMOTE_TABLE_FLAG_VAL_UNSET);
2,718✔
4893
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
2,718✔
4894
  }
4895

4896
  *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
702,520✔
4897

4898
  if (!(*fetchDone)) {
702,520✔
4899
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, 1);
128,652✔
4900
    if (TSDB_CODE_SUCCESS != code) {
128,652✔
4901
      ctx->code = code;
×
4902
      *fetchDone = true;
×
4903
    }
4904
  }
4905
}
702,973✔
4906

4907

4908
int32_t remoteFetchCallBack(void* param, SDataBuf* pMsg, int32_t code) {
44,079,339✔
4909
  taosMemoryFreeClear(pMsg->pEpSet);
44,079,339✔
4910

4911
  SScalarFetchParam* pParam = (SScalarFetchParam*)param;
44,079,339✔
4912
  STaskSubJobCtx* ctx = taosAcquireRef(fetchObjRefPool, pParam->subJobRefId);
44,079,339✔
4913
  if (ctx == NULL) {
44,079,339✔
4914
    qWarn("failed to acquire subJobCtx, since it may have been released, refId:%" PRIu64, pParam->subJobRefId);
3,070✔
4915
    taosMemoryFree(pMsg->pData);
3,070✔
4916
    return TSDB_CODE_SUCCESS;
3,070✔
4917
  }
4918

4919
  char idStr[64];
44,076,269✔
4920
  if (qDebugFlag & DEBUG_DEBUG) {
44,076,269✔
4921
    tstrncpy(idStr, ctx->idStr, sizeof(idStr));
2,226,119✔
4922
  }
4923
  
4924
  qDebug("%s subQIdx %d got rsp, blockIdx:%" PRId64 ", code:%d, rsp:%p", ctx->idStr, pParam->subQIdx, ctx->blockIdx, code, pMsg->pData);
44,076,269✔
4925

4926
  if (ctx->transporterId > 0) {
44,076,269✔
4927
    int32_t ret = asyncFreeConnById(ctx->rpcHandle, ctx->transporterId);
44,076,873✔
4928
    if (ret != 0) {
44,076,873✔
UNCOV
4929
      qDebug("%s failed to free subQ rpc handle, code:%s, subQIdx:%d", ctx->idStr, tstrerror(ret), pParam->subQIdx);
×
4930
    }
4931
    ctx->transporterId = -1;
44,076,873✔
4932
  }
4933

4934
  if (0 == code && NULL == pMsg->pData) {
44,076,269✔
4935
    qError("%s invalid rsp msg, msgType:%d, len:%d", ctx->idStr, pMsg->msgType, pMsg->len);
×
4936
    code = TSDB_CODE_QRY_INVALID_MSG;
×
4937
  }
4938

4939
  if (code == TSDB_CODE_SUCCESS) {
44,076,873✔
4940
    SRetrieveTableRsp* pRsp = pMsg->pData;
43,097,543✔
4941
    pRsp->numOfRows = htobe64(pRsp->numOfRows);
43,097,543✔
4942
    pRsp->compLen = htonl(pRsp->compLen);
43,097,543✔
4943
    pRsp->payloadLen = htonl(pRsp->payloadLen);
43,096,929✔
4944
    pRsp->numOfCols = htonl(pRsp->numOfCols);
43,097,543✔
4945
    pRsp->useconds = htobe64(pRsp->useconds);
43,097,543✔
4946
    pRsp->numOfBlocks = htonl(pRsp->numOfBlocks);
43,097,543✔
4947

4948
    qDebug("%s subQIdx %d blockIdx:%" PRIu64 " rsp detail, numOfBlocks:%d, numOfRows:%" PRId64 ", numOfCols:%" PRId64 ", completed:%d", 
43,096,929✔
4949
      ctx->idStr, pParam->subQIdx, ctx->blockIdx, pRsp->numOfBlocks, pRsp->numOfRows, pRsp->numOfCols, pRsp->completed);
4950

4951
    ctx->blockIdx++;
43,096,929✔
4952

4953
    switch (nodeType(pParam->pRes)) {
43,097,543✔
4954
      case QUERY_NODE_REMOTE_VALUE: {
3,791,024✔
4955
        bool fetchDone = false;
3,791,024✔
4956
        handleRemoteValueRes(pParam, ctx, pRsp, &fetchDone);
3,791,024✔
4957
        qDebug("%s subQIdx %d handle remote value, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
3,790,410✔
4958
        if (!fetchDone) {
3,789,796✔
4959
          goto _exit;
57,628✔
4960
        }
4961
        break;
3,732,168✔
4962
      }
4963
      case QUERY_NODE_REMOTE_VALUE_LIST: {
14,457,678✔
4964
        bool fetchDone = false;
14,457,678✔
4965
        handleRemoteValueListRes(pParam, ctx, pRsp, &fetchDone);
14,457,678✔
4966
        qDebug("%s subQIdx %d handle remote value list finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
14,457,678✔
4967
        if (!fetchDone) {
14,457,678✔
4968
          goto _exit;
1,410,982✔
4969
        }
4970
        break;
13,046,696✔
4971
      }  
4972
      case QUERY_NODE_REMOTE_ROW: {
23,837,234✔
4973
        bool fetchDone = false;
23,837,234✔
4974
        handleRemoteRowRes(pParam, ctx, pRsp, &fetchDone);
23,837,234✔
4975
        qDebug("%s subQIdx %d handle remote row finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
23,837,234✔
4976
        if (!fetchDone) {
23,837,234✔
4977
          goto _exit;
600✔
4978
        }
4979
        break;
23,836,634✔
4980
      }
4981
      case QUERY_NODE_REMOTE_ZERO_ROWS: {
308,634✔
4982
        bool fetchDone = false;
308,634✔
4983
        handleRemoteZeroRowsRes(pParam, ctx, pRsp, &fetchDone);
308,634✔
4984
        qDebug("%s subQIdx %d handle remote zeroRows finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
308,634✔
4985
        if (!fetchDone) {
308,634✔
4986
          goto _exit;
×
4987
        }
4988
        break;
308,634✔
4989
      }
4990
      case QUERY_NODE_REMOTE_TABLE: {
702,973✔
4991
        bool fetchDone = false;
702,973✔
4992
        handleRemoteTableRes(pParam, ctx, pRsp, &fetchDone);
702,973✔
4993
        qDebug("%s subQIdx %d handle remote table finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
702,973✔
4994
        if (!fetchDone) {
702,973✔
4995
          goto _exit;
128,652✔
4996
        }
4997
        break;
574,321✔
4998
      }
4999
      default:
×
5000
        qError("%s invalid scl fetch res node %d, subQIdx:%d", ctx->idStr, nodeType(pParam->pRes), pParam->subQIdx);
×
5001
        ctx->code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
5002
        break;
×
5003
    }
5004
  } else {
5005
    ctx->code = rpcCvtErrCode(code);
979,330✔
5006
    if (ctx->code != code) {
979,330✔
5007
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s, cvted error: %s", ctx->idStr, pParam->subQIdx,
×
5008
             tstrerror(code), tstrerror(ctx->code));
5009
    } else {
5010
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s", ctx->idStr, pParam->subQIdx, tstrerror(code));
979,330✔
5011
    }
5012
  }
5013

5014
  qDebug("%s subQIdx %d sem_post subQ ready", ctx->idStr, pParam->subQIdx);
42,479,011✔
5015
  
5016
  code = tsem_post(&ctx->ready);
42,479,011✔
5017
  if (code != TSDB_CODE_SUCCESS) {
42,479,011✔
5018
    qError("failed to invoke post when scl fetch rsp is ready, code:%s", tstrerror(code));
×
5019
  }
5020

5021
_exit:
44,076,259✔
5022

5023
  code = taosReleaseRef(fetchObjRefPool, pParam->subJobRefId);
44,076,873✔
5024
  if (code != TSDB_CODE_SUCCESS) {
44,076,873✔
5025
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
5026
  }
5027

5028
  taosMemoryFree(pMsg->pData);
44,076,873✔
5029

5030
  return code;
44,076,873✔
5031
}
5032

5033
#define STREAM_TASK_REPLICA_NUM 5
5034
#define STREAM_TASK_DEPLOY_NUM 3
5035

5036
int32_t sendFetchRemoteNodeReq(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes, bool reset) {
44,073,189✔
5037
  int32_t          code = TSDB_CODE_SUCCESS;
44,073,189✔
5038
  int32_t          lino = 0;
44,073,189✔
5039
  int32_t          innerIdx = 0;
44,073,189✔
5040
  SExecTaskInfo*   pTaskInfo = ctx->pTaskInfo;
44,073,189✔
5041
  bool             needStreamPesudoFuncVals = false;
44,075,645✔
5042

5043
  if (IS_STREAM_MODE(pTaskInfo)) {
44,075,645✔
5044
    innerIdx = pTaskInfo->pStreamRuntimeInfo->execId / STREAM_TASK_REPLICA_NUM;
1,231,465✔
5045
    subQIdx = subQIdx * STREAM_TASK_DEPLOY_NUM + innerIdx;
1,231,465✔
5046
  }
5047

5048
  SDownstreamSourceNode* pSource = (SDownstreamSourceNode*)taosArrayGetP(ctx->subEndPoints, subQIdx);
44,073,803✔
5049

5050
  SResFetchReq req = {0};
44,075,358✔
5051
  req.header.vgId = pSource->addr.nodeId;
44,074,742✔
5052
  req.sId = pSource->sId;
44,066,499✔
5053
  req.clientId = pSource->clientId;
44,074,136✔
5054
  req.taskId = pSource->taskId;
44,074,738✔
5055
  req.srcTaskId = ctx->taskId;
44,076,259✔
5056
  req.blockIdx = ctx->blockIdx;
44,065,845✔
5057
  req.queryId = ctx->queryId;
44,060,654✔
5058
  req.execId = pSource->execId;
44,066,152✔
5059

5060
  if (IS_STREAM_MODE(pTaskInfo)) {
44,078,101✔
5061
    req.queryId = pSource->clientId;
1,231,172✔
5062
    req.execId = pTaskInfo->pStreamRuntimeInfo->execId;
1,231,156✔
5063
    req.pStRtFuncInfo = &pTaskInfo->pStreamRuntimeInfo->funcInfo;
1,231,465✔
5064
    req.reset = reset;
1,231,449✔
5065

5066
    needStreamPesudoFuncVals = true;
1,231,449✔
5067
    subQIdx = (subQIdx - innerIdx) / STREAM_TASK_DEPLOY_NUM;
1,231,449✔
5068
  }
5069

5070
  int32_t msgSize = tSerializeSResFetchReq(NULL, 0, &req, needStreamPesudoFuncVals, false);
44,066,423✔
5071
  if (msgSize < 0) {
44,071,971✔
5072
    return msgSize;
×
5073
  }
5074

5075
  void* msg = taosMemoryCalloc(1, msgSize);
44,071,971✔
5076
  if (NULL == msg) {
44,061,128✔
5077
    return terrno;
×
5078
  }
5079

5080
  msgSize = tSerializeSResFetchReq(msg, msgSize, &req, needStreamPesudoFuncVals, false);
44,061,128✔
5081
  if (msgSize < 0) {
44,072,144✔
5082
    taosMemoryFree(msg);
×
5083
    return msgSize;
×
5084
  }
5085

5086
  qDebug("%s scl build fetch msg and send to nodeId:%d, ep:%s, clientId:0x%" PRIx64 " taskId:0x%" PRIx64
44,072,144✔
5087
         ", execId:%d, blockIdx:%" PRId64,
5088
         ctx->idStr, pSource->addr.nodeId, pSource->addr.epSet.eps[0].fqdn, pSource->clientId,
5089
         pSource->taskId, pSource->execId, req.blockIdx);
5090

5091
  // send the fetch remote task result reques
5092
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
44,072,585✔
5093
  if (NULL == pMsgSendInfo) {
44,061,577✔
5094
    taosMemoryFreeClear(msg);
×
5095
    qError("%s prepare message %d failed", ctx->idStr, (int32_t)sizeof(SMsgSendInfo));
×
5096
    return terrno;
×
5097
  }
5098

5099
  SScalarFetchParam* param = taosMemoryMalloc(sizeof(SScalarFetchParam));
44,061,577✔
5100
  if (NULL == param) {
44,060,975✔
5101
    taosMemoryFreeClear(msg);
×
5102
    taosMemoryFreeClear(pMsgSendInfo);
×
5103
    qError("%s prepare param %d failed", ctx->idStr, (int32_t)sizeof(SScalarFetchParam));
×
5104
    return terrno;
×
5105
  }
5106

5107
  if (ctx->code) {
44,060,975✔
5108
    qError("task has been killed, error:%s", tstrerror(ctx->code));
×
5109
    taosMemoryFree(param);
×
5110
    taosMemoryFreeClear(msg);
×
5111
    taosMemoryFreeClear(pMsgSendInfo);
×
5112
    code = ctx->code;
×
5113
    goto _end;
×
5114
  }
5115
  
5116
  param->subQIdx = subQIdx;
44,058,531✔
5117
  param->pRes = pRes;
44,072,585✔
5118
  param->subJobRefId = ctx->subJobRefId;
44,074,439✔
5119

5120
  pMsgSendInfo->param = param;
44,061,589✔
5121
  pMsgSendInfo->paramFreeFp = taosAutoMemoryFree;
44,071,983✔
5122
  pMsgSendInfo->msgInfo.pData = msg;
44,076,895✔
5123
  pMsgSendInfo->msgInfo.len = msgSize;
44,065,263✔
5124
  pMsgSendInfo->msgType = pSource->fetchMsgType;
44,057,917✔
5125
  pMsgSendInfo->fp = remoteFetchCallBack;
44,065,263✔
5126
  pMsgSendInfo->requestId = ctx->queryId;
44,064,641✔
5127

5128
  code = asyncSendMsgToServer(ctx->rpcHandle, &pSource->addr.epSet, &ctx->transporterId, pMsgSendInfo);
44,057,311✔
5129
  QUERY_CHECK_CODE(code, lino, _end);
44,079,349✔
5130
      
5131
_end:
44,079,349✔
5132

5133
  if (code != TSDB_CODE_SUCCESS) {
44,078,725✔
5134
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
×
5135
  }
5136
  
5137
  return code;
44,078,725✔
5138
}
5139

5140
int32_t fetchRemoteNodeImpl(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes) {
42,475,642✔
5141
  int32_t          code = TSDB_CODE_SUCCESS;
42,475,642✔
5142
  int32_t          lino = 0;
42,475,642✔
5143

5144
  ctx->blockIdx = 0;
42,475,642✔
5145

5146
  code = sendFetchRemoteNodeReq(ctx, subQIdx, pRes, true);
42,479,023✔
5147
  QUERY_CHECK_CODE(code, lino, _end);
42,479,021✔
5148

5149
  code = qSemWait(ctx->pTaskInfo, &ctx->ready);
42,479,021✔
5150
  if (isTaskKilled(ctx->pTaskInfo)) {
42,483,309✔
5151
    code = getTaskCode(ctx->pTaskInfo);
17,192✔
5152
  } else {
5153
    code = ctx->code;
42,466,117✔
5154
  }
5155
      
5156
_end:
42,483,309✔
5157

5158
  if (code != TSDB_CODE_SUCCESS) {
42,483,309✔
5159
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
2,072,268✔
5160
  }
5161
  return code;
42,483,309✔
5162
}
5163

5164
int32_t remoteNodeCopy(SNode* pSrc, SNode* pDst) {
131,304✔
5165
  int32_t code = 0, lino = 0;
131,304✔
5166
  
5167
  switch (nodeType(pSrc)) {
131,304✔
5168
    case QUERY_NODE_VALUE:
93,980✔
5169
      TAOS_CHECK_EXIT(valueNodeCopy((SValueNode*)pSrc, &((SRemoteValueNode*)pDst)->val));
93,980✔
5170
      ((SRemoteValueNode*)pDst)->val.node.type = QUERY_NODE_VALUE;
93,980✔
5171
      break;
93,980✔
5172
    case QUERY_NODE_REMOTE_VALUE_LIST: {
27,692✔
5173
      SRemoteValueListNode* pDstNode = (SRemoteValueListNode*)pDst;
27,692✔
5174
      memcpy(pDst, pSrc, sizeof(SRemoteValueListNode));
27,692✔
5175
      pDstNode->hashAllocated = false;      
27,692✔
5176
      break;
27,692✔
5177
    } 
5178
    case QUERY_NODE_REMOTE_ROW: {
9,632✔
5179
      SRemoteRowNode* pRemote = (SRemoteRowNode*)pDst;
9,632✔
5180
      TAOS_CHECK_EXIT(valueNodeCopy((SValueNode*)pSrc, &pRemote->val));
9,632✔
5181
      pRemote->valSet = true;
9,632✔
5182
      pRemote->hasValue = ((SRemoteRowNode*)pSrc)->hasValue;
9,632✔
5183
      pRemote->hasNull = ((SRemoteRowNode*)pSrc)->hasNull;
9,632✔
5184
      break;
9,632✔
5185
    }
5186
    case QUERY_NODE_REMOTE_TABLE: {
×
5187
      SRemoteTableNode* pDstNode = (SRemoteTableNode*)pDst;
×
5188
      SRemoteTableNode* pSrcNode = (SRemoteTableNode*)pSrc;
×
5189
      pDstNode->pResBlks = pSrcNode->pResBlks;
×
5190
      pDstNode->flag = 0;      
×
5191
      break;
×
5192
    }
5193
    case QUERY_NODE_REMOTE_ZERO_ROWS: {
×
5194
      SRemoteZeroRowsNode* pRemote = (SRemoteZeroRowsNode*)pDst;
×
5195
      TAOS_CHECK_EXIT(valueNodeCopy((SValueNode*)pSrc, &pRemote->val));
×
5196
      pRemote->val.node.type = QUERY_NODE_VALUE;
×
5197
      break;
×
5198
    }
5199
    default:
×
5200
      break;
×
5201
  }
5202

5203
_exit:
131,304✔
5204

5205
  if (code) {
131,304✔
5206
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5207
  }
5208

5209
  return code;
131,304✔
5210
}
5211

5212
int32_t qFetchRemoteNode(void* pCtx, int32_t subQIdx, SNode* pRes) {
42,608,778✔
5213
  STaskSubJobCtx*  ctx = (STaskSubJobCtx*)pCtx;
42,608,778✔
5214
  int32_t code = 0, lino = 0;
42,608,778✔
5215
  int32_t       subEndPoinsNum = taosArrayGetSize(ctx->subEndPoints);
42,608,778✔
5216
  if (subQIdx >= subEndPoinsNum) {
42,607,859✔
5217
    qError("%s invalid subQIdx %d, subEndPointsNum:%d", ctx->idStr, subQIdx, subEndPoinsNum);
×
5218
    return TSDB_CODE_QRY_SUBQ_NOT_FOUND;
×
5219
  }
5220

5221
  SNode** ppRes = taosArrayGet(ctx->subResNodes, subQIdx);
42,607,859✔
5222
  if (NULL == *ppRes) {
42,610,642✔
5223
    TAOS_CHECK_EXIT(fetchRemoteNodeImpl(ctx, subQIdx, pRes));
42,481,772✔
5224
    *ppRes = pRes;
40,411,041✔
5225
  } else {
5226
    TAOS_CHECK_EXIT(remoteNodeCopy(*ppRes, pRes));
131,304✔
5227
  }
5228

5229
_exit:
131,304✔
5230

5231
  if (code) {
42,614,613✔
5232
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
2,072,268✔
5233
  } else {
5234
    qDebug("%s %s subQIdx %d succeed", ctx->idStr, __func__, subQIdx);
40,542,345✔
5235
  }
5236

5237
  return code;
42,614,613✔
5238
}
5239

5240
int32_t findDataBlockColIndexBySlotId(const SSDataBlock* pBlock, int32_t slotId) {
210,422,103✔
5241
  if (NULL == pBlock || NULL == pBlock->pDataBlock) {
210,422,103✔
5242
    return -1;
×
5243
  }
5244

5245
  int32_t numOfCols = (int32_t)taosArrayGetSize(pBlock->pDataBlock);
210,422,103✔
5246
  if (slotId >= 0 && slotId < numOfCols) {
210,422,552✔
5247
    SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, slotId);
210,422,552✔
5248
    if (NULL != pCol && pCol->info.slotId == slotId) {
210,422,103✔
5249
      return slotId;
210,422,552✔
5250
    }
5251
  }
5252

5253
  for (int32_t i = 0; i < numOfCols; ++i) {
×
5254
    SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, i);
×
5255
    if (NULL != pCol && pCol->info.slotId == slotId) {
×
5256
      return i;
×
5257
    }
5258
  }
5259

5260
  return -1;
×
5261
}
5262

5263
SColumnInfoData* getDataBlockColBySlotId(const SSDataBlock* pBlock, int32_t slotId, int32_t* pIndex) {
210,422,103✔
5264
  int32_t index = findDataBlockColIndexBySlotId(pBlock, slotId);
210,422,103✔
5265
  if (NULL != pIndex) {
210,422,552✔
5266
    *pIndex = index;
×
5267
  }
5268
  return (index >= 0) ? taosArrayGet(pBlock->pDataBlock, index) : NULL;
210,422,552✔
5269
}
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