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

taosdata / TDengine / #4887

16 Dec 2025 08:27AM UTC coverage: 65.289% (-0.003%) from 65.292%
#4887

push

travis-ci

web-flow
feat[TS-7233]: audit (#33850)

377 of 536 new or added lines in 28 files covered. (70.34%)

1025 existing lines in 111 files now uncovered.

178977 of 274129 relevant lines covered (65.29%)

102580217.43 hits per line

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

76.0
/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 "tutil.h"
35
#include "tjson.h"
36

37
typedef struct tagFilterAssist {
38
  SHashObj* colHash;
39
  int32_t   index;
40
  SArray*   cInfoList;
41
  int32_t   code;
42
} tagFilterAssist;
43

44
typedef struct STransTagExprCtx {
45
  int32_t      code;
46
  SMetaReader* pReader;
47
} STransTagExprCtx;
48

49
typedef enum {
50
  FILTER_NO_LOGIC = 1,
51
  FILTER_AND,
52
  FILTER_OTHER,
53
} FilterCondType;
54

55
static FilterCondType checkTagCond(SNode* cond);
56
static int32_t optimizeTbnameInCond(void* metaHandle, int64_t suid, SArray* list, SNode* pTagCond, SStorageAPI* pAPI);
57
static int32_t optimizeTbnameInCondImpl(void* metaHandle, SArray* list, SNode* pTagCond, SStorageAPI* pStoreAPI,
58
                                        uint64_t suid);
59

60
static int32_t getTableList(void* pVnode, SScanPhysiNode* pScanNode, SNode* pTagCond, SNode* pTagIndexCond,
61
                            STableListInfo* pListInfo, uint8_t* digest, const char* idstr, SStorageAPI* pStorageAPI, void* pStreamInfo);
62

63
static int64_t getLimit(const SNode* pLimit) {
556,265,990✔
64
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->limit) ? -1 : ((SLimitNode*)pLimit)->limit->datum.i;
556,265,990✔
65
}
66
static int64_t getOffset(const SNode* pLimit) {
556,246,001✔
67
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->offset) ? -1 : ((SLimitNode*)pLimit)->offset->datum.i;
556,246,001✔
68
}
69
static void releaseColInfoData(void* pCol);
70

71
void initResultRowInfo(SResultRowInfo* pResultRowInfo) {
220,602,978✔
72
  pResultRowInfo->size = 0;
220,602,978✔
73
  pResultRowInfo->cur.pageId = -1;
220,628,264✔
74
}
220,640,805✔
75

76
void closeResultRow(SResultRow* pResultRow) { pResultRow->closed = true; }
5,918,894✔
77

78
void resetResultRow(SResultRow* pResultRow, size_t entrySize) {
388,105,292✔
79
  pResultRow->numOfRows = 0;
388,105,292✔
80
  pResultRow->closed = false;
388,105,292✔
81
  pResultRow->endInterp = false;
388,105,292✔
82
  pResultRow->startInterp = false;
388,105,292✔
83

84
  if (entrySize > 0) {
388,105,475✔
85
    memset(pResultRow->pEntryInfo, 0, entrySize);
388,105,475✔
86
  }
87
}
388,106,939✔
88

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

94
size_t getResultRowSize(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
129,874,675✔
95
  int32_t rowSize = (numOfOutput * sizeof(SResultRowEntryInfo)) + sizeof(SResultRow);
129,874,675✔
96

97
  for (int32_t i = 0; i < numOfOutput; ++i) {
655,662,608✔
98
    rowSize += pCtx[i].resDataInfo.interBufSize;
525,796,917✔
99
  }
100

101
  return rowSize;
129,865,691✔
102
}
103

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

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

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

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

159
  qTrace("[StreamInternal] get result inBufSize:%zu, outBufSize:%zu", inBufSize, *outBufSize);
×
160
  return TSDB_CODE_SUCCESS;
×
161
}
162

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

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

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

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

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

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

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

211
  qTrace("[StreamInternal] put result inBufSize:%zu, outBufSize:%zu", inBufSize, *outBufSize);
×
212
  return TSDB_CODE_SUCCESS;
×
213
}
214

215
static void freeEx(void* p) { taosMemoryFree(*(void**)p); }
×
216

217
void cleanupGroupResInfo(SGroupResInfo* pGroupResInfo) {
63,397,093✔
218
  taosMemoryFreeClear(pGroupResInfo->pBuf);
63,397,093✔
219
  if (pGroupResInfo->freeItem) {
63,400,121✔
220
    //    taosArrayDestroy(pGroupResInfo->pRows);
221
    taosArrayDestroyEx(pGroupResInfo->pRows, freeEx);
×
222
    pGroupResInfo->freeItem = false;
×
223
    pGroupResInfo->pRows = NULL;
×
224
  } else {
225
    taosArrayDestroy(pGroupResInfo->pRows);
63,400,747✔
226
    pGroupResInfo->pRows = NULL;
63,395,617✔
227
  }
228
  pGroupResInfo->index = 0;
63,396,820✔
229
  pGroupResInfo->delIndex = 0;
63,396,188✔
230
}
63,394,763✔
231

232
int32_t resultrowComparAsc(const void* p1, const void* p2) {
2,147,483,647✔
233
  SResKeyPos* pp1 = *(SResKeyPos**)p1;
2,147,483,647✔
234
  SResKeyPos* pp2 = *(SResKeyPos**)p2;
2,147,483,647✔
235

236
  if (pp1->groupId == pp2->groupId) {
2,147,483,647✔
237
    int64_t pts1 = *(int64_t*)pp1->key;
2,147,483,647✔
238
    int64_t pts2 = *(int64_t*)pp2->key;
2,147,483,647✔
239

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

250
static int32_t resultrowComparDesc(const void* p1, const void* p2) { return resultrowComparAsc(p2, p1); }
1,921,691,514✔
251

252
int32_t initGroupedResultInfo(SGroupResInfo* pGroupResInfo, SSHashObj* pHashmap, int32_t order) {
44,974,574✔
253
  int32_t code = TSDB_CODE_SUCCESS;
44,974,574✔
254
  int32_t lino = 0;
44,974,574✔
255
  if (pGroupResInfo->pRows != NULL) {
44,974,574✔
256
    taosArrayDestroy(pGroupResInfo->pRows);
3,261,413✔
257
  }
258
  if (pGroupResInfo->pBuf) {
44,978,626✔
259
    taosMemoryFree(pGroupResInfo->pBuf);
3,261,413✔
260
    pGroupResInfo->pBuf = NULL;
3,261,413✔
261
  }
262

263
  // extract the result rows information from the hash map
264
  int32_t size = tSimpleHashGetSize(pHashmap);
44,975,862✔
265

266
  void* pData = NULL;
44,972,680✔
267
  pGroupResInfo->pRows = taosArrayInit(size, POINTER_BYTES);
44,972,680✔
268
  QUERY_CHECK_NULL(pGroupResInfo->pRows, code, lino, _end, terrno);
44,980,154✔
269

270
  size_t  keyLen = 0;
44,972,486✔
271
  int32_t iter = 0;
44,971,014✔
272
  int64_t bufLen = 0, offset = 0;
44,975,142✔
273

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

280
  pGroupResInfo->pBuf = taosMemoryMalloc(bufLen);
44,950,624✔
281
  QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
44,981,895✔
282

283
  iter = 0;
44,973,411✔
284
  while ((pData = tSimpleHashIterate(pHashmap, pData, &iter)) != NULL) {
2,147,483,647✔
285
    void* key = tSimpleHashGetKey(pData, &keyLen);
2,147,483,647✔
286

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

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

295
    offset += keyLen + sizeof(struct SResultRowPosition);
2,147,483,647✔
296
  }
297

298
  if (order == TSDB_ORDER_ASC || order == TSDB_ORDER_DESC) {
44,963,096✔
299
    __compar_fn_t fn = (order == TSDB_ORDER_ASC) ? resultrowComparAsc : resultrowComparDesc;
2,718,233✔
300
    size = POINTER_BYTES;
2,718,233✔
301
    taosSort(pGroupResInfo->pRows->pData, taosArrayGetSize(pGroupResInfo->pRows), size, fn);
2,718,233✔
302
  }
303

304
  pGroupResInfo->index = 0;
44,962,614✔
305

306
_end:
44,969,925✔
307
  if (code != TSDB_CODE_SUCCESS) {
44,983,941✔
308
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
309
  }
310
  return code;
44,983,941✔
311
}
312

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

318
  pGroupResInfo->freeItem = true;
×
319
  pGroupResInfo->pRows = pArrayList;
×
320
  pGroupResInfo->index = 0;
×
321
  pGroupResInfo->delIndex = 0;
×
322
}
×
323

324
bool hasRemainResults(SGroupResInfo* pGroupResInfo) {
162,245,586✔
325
  if (pGroupResInfo->pRows == NULL) {
162,245,586✔
326
    return false;
×
327
  }
328

329
  return pGroupResInfo->index < taosArrayGetSize(pGroupResInfo->pRows);
162,252,975✔
330
}
331

332
int32_t getNumOfTotalRes(SGroupResInfo* pGroupResInfo) {
88,412,315✔
333
  if (pGroupResInfo->pRows == 0) {
88,412,315✔
334
    return 0;
×
335
  }
336

337
  return (int32_t)taosArrayGetSize(pGroupResInfo->pRows);
88,415,701✔
338
}
339

340
SArray* createSortInfo(SNodeList* pNodeList) {
20,827,355✔
341
  size_t numOfCols = 0;
20,827,355✔
342

343
  if (pNodeList != NULL) {
20,827,355✔
344
    numOfCols = LIST_LENGTH(pNodeList);
20,792,187✔
345
  } else {
346
    numOfCols = 0;
35,308✔
347
  }
348

349
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
20,828,492✔
350
  if (pList == NULL) {
20,834,738✔
351
    return pList;
×
352
  }
353

354
  for (int32_t i = 0; i < numOfCols; ++i) {
45,345,542✔
355
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
24,505,198✔
356
    if (!pSortKey) {
24,508,718✔
357
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
358
      taosArrayDestroy(pList);
×
359
      pList = NULL;
×
360
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
361
      break;
×
362
    }
363
    SBlockOrderInfo bi = {0};
24,508,718✔
364
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
24,505,057✔
365
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);
24,505,655✔
366

367
    if (nodeType(pSortKey->pExpr) != QUERY_NODE_COLUMN) {
24,504,794✔
368
      qError("invalid order by expr type:%d", nodeType(pSortKey->pExpr));
×
369
      taosArrayDestroy(pList);
×
370
      pList = NULL;
×
371
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
372
      break;
×
373
    }
374
    
375
    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
24,497,789✔
376
    bi.slotId = pColNode->slotId;
24,502,100✔
377
    void* tmp = taosArrayPush(pList, &bi);
24,511,347✔
378
    if (!tmp) {
24,511,347✔
379
      taosArrayDestroy(pList);
×
380
      pList = NULL;
×
381
      break;
×
382
    }
383
  }
384

385
  return pList;
20,840,210✔
386
}
387

388
SSDataBlock* createDataBlockFromDescNode(void* p) {
329,933,646✔
389
  SDataBlockDescNode* pNode = (SDataBlockDescNode*)p;
329,933,646✔
390
  int32_t      numOfCols = LIST_LENGTH(pNode->pSlots);
329,933,646✔
391
  SSDataBlock* pBlock = NULL;
329,902,909✔
392
  int32_t      code = createDataBlock(&pBlock);
329,926,826✔
393
  if (code) {
329,923,715✔
394
    terrno = code;
×
395
    return NULL;
×
396
  }
397

398
  pBlock->info.id.blockId = pNode->dataBlockId;
329,923,747✔
399
  pBlock->info.type = STREAM_INVALID;
329,921,520✔
400
  pBlock->info.calWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
329,920,650✔
401
  pBlock->info.watermark = INT64_MIN;
329,886,256✔
402

403
  for (int32_t i = 0; i < numOfCols; ++i) {
1,986,329,522✔
404
    SSlotDescNode* pDescNode = (SSlotDescNode*)nodesListGetNode(pNode->pSlots, i);
1,656,263,777✔
405
    if (!pDescNode) {
1,656,046,131✔
406
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
407
      blockDataDestroy(pBlock);
×
408
      pBlock = NULL;
×
409
      terrno = TSDB_CODE_INVALID_PARA;
×
410
      break;
×
411
    }
412
    SColumnInfoData idata =
1,655,993,375✔
413
        createColumnInfoData(pDescNode->dataType.type, pDescNode->dataType.bytes, pDescNode->slotId);
1,656,078,552✔
414
    idata.info.scale = pDescNode->dataType.scale;
1,656,371,489✔
415
    idata.info.precision = pDescNode->dataType.precision;
1,656,395,912✔
416
    idata.info.noData = pDescNode->reserve;
1,656,384,625✔
417

418
    code = blockDataAppendColInfo(pBlock, &idata);
1,656,392,622✔
419
    if (code != TSDB_CODE_SUCCESS) {
1,656,334,091✔
420
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
52,058✔
421
      blockDataDestroy(pBlock);
52,058✔
422
      pBlock = NULL;
×
423
      terrno = code;
×
424
      break;
×
425
    }
426
  }
427

428
  return pBlock;
330,039,869✔
429
}
430

431
int32_t prepareDataBlockBuf(SSDataBlock* pDataBlock, SColMatchInfo* pMatchInfo) {
113,154,789✔
432
  SDataBlockInfo* pBlockInfo = &pDataBlock->info;
113,154,789✔
433

434
  for (int32_t i = 0; i < taosArrayGetSize(pMatchInfo->pList); ++i) {
699,178,017✔
435
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
593,545,146✔
436
    if (!pItem) {
593,539,954✔
437
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
438
      return terrno;
×
439
    }
440

441
    if (pItem->isPk) {
593,539,986✔
442
      SColumnInfoData* pInfoData = taosArrayGet(pDataBlock->pDataBlock, pItem->dstSlotId);
7,597,865✔
443
      if (!pInfoData) {
7,534,845✔
444
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
445
        return terrno;
×
446
      }
447
      pBlockInfo->pks[0].type = pInfoData->info.type;
7,534,845✔
448
      pBlockInfo->pks[1].type = pInfoData->info.type;
7,566,434✔
449

450
      // allocate enough buffer size, which is pInfoData->info.bytes
451
      if (IS_VAR_DATA_TYPE(pItem->dataType.type)) {
7,536,199✔
452
        pBlockInfo->pks[0].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
2,550,106✔
453
        if (pBlockInfo->pks[0].pData == NULL) {
2,522,051✔
454
          return terrno;
×
455
        }
456

457
        pBlockInfo->pks[1].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
2,523,554✔
458
        if (pBlockInfo->pks[1].pData == NULL) {
2,525,964✔
459
          taosMemoryFreeClear(pBlockInfo->pks[0].pData);
×
460
          return terrno;
×
461
        }
462

463
        pBlockInfo->pks[0].nData = pInfoData->info.bytes;
2,526,790✔
464
        pBlockInfo->pks[1].nData = pInfoData->info.bytes;
2,525,436✔
465
      }
466

467
      break;
7,553,124✔
468
    }
469
  }
470

471
  return TSDB_CODE_SUCCESS;
113,157,227✔
472
}
473

474
EDealRes doTranslateTagExpr(SNode** pNode, void* pContext) {
336,588✔
475
  STransTagExprCtx* pCtx = pContext;
336,588✔
476
  SMetaReader*      mr = pCtx->pReader;
336,588✔
477
  bool              isTagCol = false, isTbname = false;
336,588✔
478
  if (nodeType(*pNode) == QUERY_NODE_COLUMN) {
336,588✔
479
    SColumnNode* pCol = (SColumnNode*)*pNode;
96,168✔
480
    if (pCol->colType == COLUMN_TYPE_TBNAME)
96,168✔
481
      isTbname = true;
×
482
    else
483
      isTagCol = true;
96,168✔
484
  } else if (nodeType(*pNode) == QUERY_NODE_FUNCTION) {
240,420✔
485
    SFunctionNode* pFunc = (SFunctionNode*)*pNode;
×
486
    if (pFunc->funcType == FUNCTION_TYPE_TBNAME) isTbname = true;
×
487
  }
488
  if (isTagCol) {
336,588✔
489
    SColumnNode* pSColumnNode = *(SColumnNode**)pNode;
96,168✔
490

491
    SValueNode* res = NULL;
96,168✔
492
    pCtx->code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
96,168✔
493
    if (NULL == res) {
96,168✔
494
      return DEAL_RES_ERROR;
×
495
    }
496

497
    res->translate = true;
96,168✔
498
    res->node.resType = pSColumnNode->node.resType;
96,168✔
499

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

517
      res->datum.p = taosMemoryCalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1, 1);
96,168✔
518
      if (NULL == res->datum.p) {
96,168✔
519
        return DEAL_RES_ERROR;
×
520
      }
521

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

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

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

558
  return DEAL_RES_CONTINUE;
336,588✔
559
}
560

561
int32_t isQualifiedTable(STableKeyInfo* info, SNode* pTagCond, void* metaHandle, bool* pQualified, SStorageAPI* pAPI) {
48,084✔
562
  int32_t     code = TSDB_CODE_SUCCESS;
48,084✔
563
  SMetaReader mr = {0};
48,084✔
564

565
  pAPI->metaReaderFn.initReader(&mr, metaHandle, META_READER_LOCK, &pAPI->metaFn);
48,084✔
566
  code = pAPI->metaReaderFn.getEntryGetUidCache(&mr, info->uid);
48,084✔
567
  if (TSDB_CODE_SUCCESS != code) {
48,084✔
568
    pAPI->metaReaderFn.clearReader(&mr);
×
569
    *pQualified = false;
×
570

571
    return TSDB_CODE_SUCCESS;
×
572
  }
573

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

590
  SNode* pNew = NULL;
48,084✔
591
  code = scalarCalculateConstants(pTagCondTmp, &pNew);
48,084✔
592
  if (TSDB_CODE_SUCCESS != code) {
48,084✔
593
    terrno = code;
×
594
    nodesDestroyNode(pTagCondTmp);
×
595
    *pQualified = false;
×
596

597
    return code;
×
598
  }
599

600
  SValueNode* pValue = (SValueNode*)pNew;
48,084✔
601
  *pQualified = pValue->datum.b;
48,084✔
602

603
  nodesDestroyNode(pNew);
48,084✔
604
  return TSDB_CODE_SUCCESS;
48,084✔
605
}
606

607
static EDealRes getColumn(SNode** pNode, void* pContext) {
44,759,129✔
608
  tagFilterAssist* pData = (tagFilterAssist*)pContext;
44,759,129✔
609
  SColumnNode*     pSColumnNode = NULL;
44,759,129✔
610
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
44,762,917✔
611
    pSColumnNode = *(SColumnNode**)pNode;
14,759,281✔
612
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
30,015,935✔
613
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
631,460✔
614
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
631,460✔
615
      pData->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
585,206✔
616
      if (NULL == pSColumnNode) {
585,340✔
617
        return DEAL_RES_ERROR;
×
618
      }
619
      pSColumnNode->colId = -1;
585,340✔
620
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
585,340✔
621
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
585,206✔
622
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
585,340✔
623
      nodesDestroyNode(*pNode);
585,206✔
624
      *pNode = (SNode*)pSColumnNode;
585,340✔
625
    } else {
626
      return DEAL_RES_CONTINUE;
46,120✔
627
    }
628
  } else {
629
    return DEAL_RES_CONTINUE;
29,377,589✔
630
  }
631

632
  void* data = taosHashGet(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
15,343,929✔
633
  if (!data) {
15,340,693✔
634
    int32_t tempRes =
635
        taosHashPut(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
14,088,613✔
636
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
14,091,917✔
637
      return DEAL_RES_ERROR;
×
638
    }
639
    pSColumnNode->slotId = pData->index++;
14,091,917✔
640
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
14,092,270✔
641
                         .type = pSColumnNode->node.resType.type,
14,086,496✔
642
                         .bytes = pSColumnNode->node.resType.bytes,
14,088,977✔
643
                         .pk = pSColumnNode->isPk};
14,087,228✔
644
#if TAG_FILTER_DEBUG
645
    qDebug("tagfilter build column info, slotId:%d, colId:%d, type:%d", pSColumnNode->slotId, cInfo.colId, cInfo.type);
646
#endif
647
    void* tmp = taosArrayPush(pData->cInfoList, &cInfo);
14,088,841✔
648
    if (!tmp) {
14,089,631✔
649
      return DEAL_RES_ERROR;
×
650
    }
651
  } else {
652
    SColumnNode* col = *(SColumnNode**)data;
1,252,080✔
653
    pSColumnNode->slotId = col->slotId;
1,252,274✔
654
  }
655

656
  return DEAL_RES_CONTINUE;
15,340,789✔
657
}
658

659
static int32_t createResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
13,070,725✔
660
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
13,070,725✔
661
  if (pColumnData == NULL) {
13,069,247✔
662
    return terrno;
×
663
  }
664

665
  pColumnData->info.type = pType->type;
13,069,247✔
666
  pColumnData->info.bytes = pType->bytes;
13,069,059✔
667
  pColumnData->info.scale = pType->scale;
13,070,107✔
668
  pColumnData->info.precision = pType->precision;
13,069,942✔
669

670
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
13,068,080✔
671
  if (code != TSDB_CODE_SUCCESS) {
13,066,920✔
672
    terrno = code;
×
673
    releaseColInfoData(pColumnData);
×
674
    return terrno;
×
675
  }
676

677
  pParam->columnData = pColumnData;
13,066,920✔
678
  pParam->colAlloced = true;
13,067,916✔
679
  return TSDB_CODE_SUCCESS;
13,069,571✔
680
}
681

682
static void releaseColInfoData(void* pCol) {
2,102,328✔
683
  if (pCol) {
2,102,328✔
684
    SColumnInfoData* col = (SColumnInfoData*)pCol;
2,102,810✔
685
    colDataDestroy(col);
2,102,810✔
686
    taosMemoryFree(col);
2,103,186✔
687
  }
688
}
2,103,380✔
689

690
void freeItem(void* p) {
185,201,557✔
691
  STUidTagInfo* pInfo = p;
185,201,557✔
692
  if (pInfo->pTagVal != NULL) {
185,201,557✔
693
    taosMemoryFree(pInfo->pTagVal);
184,795,109✔
694
  }
695
}
185,198,890✔
696

697
typedef struct {
698
  col_id_t  colId;
699
  SNode*    pValueNode;
700
  int32_t   bytes;  // length defined in schema
701
} STagDataEntry;
702

703
static int compareTagDataEntry(const void* a, const void* b) {
48,848✔
704
  STagDataEntry* p1 = (STagDataEntry*)a;
48,848✔
705
  STagDataEntry* p2 = (STagDataEntry*)b;
48,848✔
706
  return compareInt16Val(&p1->colId, &p2->colId);
48,848✔
707
}
708

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

724
    (void)memcpy(pStart, &entry->colId, sizeof(col_id_t));
48,848✔
725
    pStart += sizeof(col_id_t);
48,848✔
726

727
    if (!pValueNode->isNull) {
48,848✔
728
      switch (pValueNode->node.resType.type) {
41,968✔
729
        case TSDB_DATA_TYPE_BOOL:
5,504✔
730
          (void)memcpy(
5,504✔
731
            pStart, &pValueNode->datum.b, bytes);
5,504✔
732
          pStart += bytes;
5,504✔
733
          break;
5,504✔
734
        case TSDB_DATA_TYPE_TINYINT:
19,952✔
735
        case TSDB_DATA_TYPE_SMALLINT:
736
        case TSDB_DATA_TYPE_INT:
737
        case TSDB_DATA_TYPE_BIGINT:
738
        case TSDB_DATA_TYPE_TIMESTAMP:
739
          (void)memcpy(
19,952✔
740
            pStart, &pValueNode->datum.i, bytes);
19,952✔
741
          pStart += bytes;
19,952✔
742
          break;
19,952✔
743
        case TSDB_DATA_TYPE_UTINYINT:
×
744
        case TSDB_DATA_TYPE_USMALLINT:
745
        case TSDB_DATA_TYPE_UINT:
746
        case TSDB_DATA_TYPE_UBIGINT:
747
          (void)memcpy(
×
748
            pStart, &pValueNode->datum.u, bytes);
×
749
          pStart += bytes;
×
750
          break;
×
751
        case TSDB_DATA_TYPE_FLOAT:
5,160✔
752
        case TSDB_DATA_TYPE_DOUBLE:
753
          (void)memcpy(
5,160✔
754
            pStart, &pValueNode->datum.d, bytes);
5,160✔
755
          pStart += bytes;
5,160✔
756
          break;
5,160✔
757
        case TSDB_DATA_TYPE_VARCHAR:
11,352✔
758
        case TSDB_DATA_TYPE_VARBINARY:
759
        case TSDB_DATA_TYPE_NCHAR:
760
          (void)memcpy(pStart,
11,352✔
761
            varDataVal(pValueNode->datum.p), varDataLen(pValueNode->datum.p));
11,352✔
762
          pStart += varDataLen(pValueNode->datum.p);
11,352✔
763
          break;
11,352✔
764
        default:
×
765
          qError("unsupported tag data type %d in tag filter optimization",
×
766
            pValueNode->node.resType.type);
767
          return TSDB_CODE_STREAM_INTERNAL_ERROR;
×
768
      }
769
    }
770
  }
771

772
  return TSDB_CODE_SUCCESS;
24,424✔
773
}
774

775
static void extractTagDataEntry(
48,848✔
776
  SOperatorNode* pOpNode, SArray* pIdWithValue) {
777
  SNode* pLeft = pOpNode->pLeft;
48,848✔
778
  SNode* pRight = pOpNode->pRight;
48,848✔
779
  SColumnNode* pColNode = nodeType(pLeft) == QUERY_NODE_COLUMN ?
48,848✔
780
    (SColumnNode*)pLeft : (SColumnNode*)pRight;
48,848✔
781
  SValueNode* pValueNode = nodeType(pLeft) == QUERY_NODE_VALUE ?
48,848✔
782
    (SValueNode*)pLeft : (SValueNode*)pRight;
48,848✔
783

784
  STagDataEntry entry = {0};
48,848✔
785
  entry.colId = pColNode->colId;
48,848✔
786
  entry.pValueNode = (SNode*)pValueNode;
48,848✔
787
  entry.bytes = pColNode->node.resType.bytes;
48,848✔
788
  void* _tmp = taosArrayPush(pIdWithValue, &entry);
48,848✔
789
}
48,848✔
790

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

800
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
24,424✔
801
    extractTagDataEntry((SOperatorNode*)pTagCond, pIdWithVal);
×
802
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
24,424✔
803
    SNode* pChild = NULL;
24,424✔
804
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
73,272✔
805
      extractTagDataEntry((SOperatorNode*)pChild, pIdWithVal);
48,848✔
806
    }
807
  }
808

809
  taosArraySort(pIdWithVal, compareTagDataEntry);
24,424✔
810

811
  return TSDB_CODE_SUCCESS;
24,424✔
812
}
813

814
static int32_t genStableTagFilterDigest(const SNode* pTagCond, T_MD5_CTX* pContext) {
24,424✔
815
  if (pTagCond == NULL) {
24,424✔
816
    return TSDB_CODE_SUCCESS;
×
817
  }
818

819
  char*   payload = NULL;
24,424✔
820
  int32_t len = 0;
24,424✔
821
  int32_t code = TSDB_CODE_SUCCESS;
24,424✔
822
  int32_t lino = 0;
24,424✔
823

824
  SArray* pIdWithVal = taosArrayInit(TARRAY_MIN_SIZE, sizeof(STagDataEntry));
24,424✔
825
  code = extractTagFilterTagDataEntries(pTagCond, pIdWithVal);
24,424✔
826
  QUERY_CHECK_CODE(code, lino, _end);
24,424✔
827
  for (int32_t i = 0; i < taosArrayGetSize(pIdWithVal); ++i) {
73,272✔
828
    STagDataEntry* pEntry = taosArrayGet(pIdWithVal, i);
48,848✔
829
    len += sizeof(col_id_t) + pEntry->bytes;
48,848✔
830
  }
831
  code = buildTagDataEntryKey(pIdWithVal, &payload, len);
24,424✔
832
  QUERY_CHECK_CODE(code, lino, _end);
24,424✔
833

834
  tMD5Init(pContext);
24,424✔
835
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
24,424✔
836
  tMD5Final(pContext);
24,424✔
837

838
_end:
24,424✔
839
  if (TSDB_CODE_SUCCESS != code) {
24,424✔
840
    qError("%s failed at line %d since %s",
×
841
      __func__, __LINE__, tstrerror(code));
842
  }
843
  taosArrayDestroy(pIdWithVal);
24,424✔
844
  taosMemoryFree(payload);
24,424✔
845
  return code;
24,424✔
846
}
847

848
static int32_t genTagFilterDigest(const SNode* pTagCond, T_MD5_CTX* pContext) {
64,700✔
849
  if (pTagCond == NULL) {
64,700✔
850
    return TSDB_CODE_SUCCESS;
61,488✔
851
  }
852

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

861
  tMD5Init(pContext);
3,212✔
862
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
3,212✔
863
  tMD5Final(pContext);
3,212✔
864

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

871
  taosMemoryFree(payload);
3,212✔
872
  return TSDB_CODE_SUCCESS;
3,212✔
873
}
874

875
static int32_t genTbGroupDigest(const SNode* pGroup, uint8_t* filterDigest, T_MD5_CTX* pContext) {
1,376✔
876
  int32_t code = TSDB_CODE_SUCCESS;
1,376✔
877
  int32_t lino = 0;
1,376✔
878
  char*   payload = NULL;
1,376✔
879
  int32_t len = 0;
1,376✔
880
  code = nodesNodeToMsg(pGroup, &payload, &len);
1,376✔
881
  QUERY_CHECK_CODE(code, lino, _end);
1,376✔
882

883
  if (filterDigest[0]) {
1,376✔
884
    payload = taosMemoryRealloc(payload, len + tListLen(pContext->digest));
1,376✔
885
    QUERY_CHECK_NULL(payload, code, lino, _end, terrno);
1,376✔
886
    memcpy(payload + len, filterDigest + 1, tListLen(pContext->digest));
1,376✔
887
    len += tListLen(pContext->digest);
1,376✔
888
  }
889

890
  tMD5Init(pContext);
1,376✔
891
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
1,376✔
892
  tMD5Final(pContext);
1,376✔
893

894
_end:
1,376✔
895
  taosMemoryFree(payload);
1,376✔
896
  if (code != TSDB_CODE_SUCCESS) {
1,376✔
897
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
898
  }
899
  return code;
1,376✔
900
}
901

902
int32_t qGetColumnsFromNodeList(void* data, bool isList, SArray** pColList) {
12,886,268✔
903
  int32_t code = TSDB_CODE_SUCCESS;
12,886,268✔
904
  tagFilterAssist ctx = {0};
12,886,268✔
905
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
12,887,004✔
906
  if (ctx.colHash == NULL) {
12,887,168✔
907
    code = terrno;
×
908
    goto end;
×
909
  }
910

911
  ctx.index = 0;
12,887,168✔
912
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
12,887,168✔
913
  if (ctx.cInfoList == NULL) {
12,887,953✔
914
    code = terrno;
1,767✔
915
    goto end;
×
916
  }
917

918
  if (isList) {
12,886,186✔
919
    SNode* pNode = NULL;
1,917,075✔
920
    FOREACH(pNode, (SNodeList*)data) {
4,020,124✔
921
      nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
2,103,728✔
922
      if (TSDB_CODE_SUCCESS != ctx.code) {
2,103,594✔
923
        code = ctx.code;
×
924
        goto end;
×
925
      }
926
      REPLACE_NODE(pNode);
2,103,594✔
927
    }
928
  } else {
929
    SNode* pNode = (SNode*)data;
10,969,111✔
930
    nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
10,968,221✔
931
    if (TSDB_CODE_SUCCESS != ctx.code) {
10,969,612✔
932
      code = ctx.code;
×
933
      goto end;
×
934
    }
935
  }
936
  
937
  if (pColList != NULL) *pColList = ctx.cInfoList;
12,884,694✔
938
  ctx.cInfoList = NULL;
12,880,720✔
939

940
end:
12,882,243✔
941
  taosHashCleanup(ctx.colHash);
12,879,721✔
942
  taosArrayDestroy(ctx.cInfoList);
12,877,299✔
943
  return code;
12,879,866✔
944
}
945

946
static int32_t buildGroupInfo(SColumnInfoData* pValue, int32_t i, SArray* gInfo) {
754,715✔
947
  int32_t code = TSDB_CODE_SUCCESS;
754,715✔
948
  SStreamGroupValue* v = taosArrayReserve(gInfo, 1);
754,715✔
949
  if (v == NULL) {
754,715✔
950
    code = terrno;
×
951
    goto end;
×
952
  }
953
  if (colDataIsNull_s(pValue, i)) {
1,509,166✔
954
    v->isNull = true;
8,600✔
955
  } else {
956
    v->isNull = false;
745,851✔
957
    char* data = colDataGetData(pValue, i);
746,115✔
958
    if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
745,869✔
959
      if (tTagIsJson(data)) {
×
960
        code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
×
961
        goto end;
×
962
      }
963
      if (tTagIsJsonNull(data)) {
×
964
        v->isNull = true;
×
965
        goto end;
×
966
      }
967
      int32_t len = getJsonValueLen(data);
×
968
      v->data.type = pValue->info.type;
×
969
      v->data.nData = len;
×
970
      v->data.pData = taosMemoryCalloc(1, len + 1);
×
971
      if (v->data.pData == NULL) {
×
972
        code = terrno;
×
973
        goto end;
×
974
      }
975
      memcpy(v->data.pData, data, len);
×
976
      qDebug("buildGroupInfo:%d add json data len:%d, data:%s", i, len, (char*)v->data.pData);
×
977
    } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
746,249✔
978
      if (varDataTLen(data) > pValue->info.bytes) {
446,841✔
979
        code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
980
        goto end;
×
981
      }
982
      v->data.type = pValue->info.type;
446,595✔
983
      v->data.nData = varDataLen(data);
446,197✔
984
      v->data.pData = taosMemoryCalloc(1, varDataLen(data) + 1);
446,049✔
985
      if (v->data.pData == NULL) {
446,183✔
986
        code = terrno;
×
987
        goto end;
×
988
      }
989
      memcpy(v->data.pData, varDataVal(data), varDataLen(data));
446,183✔
990
      qDebug("buildGroupInfo:%d add var data type:%d, len:%d, data:%s", i, pValue->info.type, varDataLen(data), (char*)v->data.pData);
446,461✔
991
    } else if (pValue->info.type == TSDB_DATA_TYPE_DECIMAL) {  // reader todo decimal
299,654✔
992
      v->data.type = pValue->info.type;
×
993
      v->data.nData = pValue->info.bytes;
×
994
      v->data.pData = taosMemoryCalloc(1, pValue->info.bytes);
×
995
      if (v->data.pData == NULL) {
×
996
        code = terrno;
×
997
        goto end;
×
998
      }
999
      memcpy(&v->data.pData, data, pValue->info.bytes);
×
1000
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
×
1001
    } else {  // reader todo decimal
1002
      v->data.type = pValue->info.type;
299,654✔
1003
      memcpy(&v->data.val, data, pValue->info.bytes);
299,654✔
1004
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
299,654✔
1005
    }
1006
  }
1007
end:
78,744✔
1008
  if (code != TSDB_CODE_SUCCESS) {
754,849✔
1009
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1010
    v->isNull = true;
×
1011
  }
1012
  return code;
754,849✔
1013
}
1014

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

1026
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
105,896✔
1027
  if (rows == 0) {
105,896✔
1028
    return;
×
1029
  }
1030

1031
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
105,896✔
1032
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
105,896✔
1033

1034
  for (int32_t i = 0; i < rows; ++i) {
444,151✔
1035
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
338,255✔
1036
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
338,255✔
1037
    STUidTagInfo info = {.uid = pkeyInfo->uid};
338,255✔
1038
    void*        tmp = taosArrayPush(pUidTagList, &info);
338,255✔
1039
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
338,255✔
1040
  }
1041
 
1042
  if (taosArrayGetSize(pUidTagList) > 0) {
105,896✔
1043
    code = pAPI->metaFn.getTableTagsByUid(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
105,896✔
1044
  } else {
1045
    code = pAPI->metaFn.getTableTags(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
×
1046
  }
1047
  if (code != TSDB_CODE_SUCCESS) {
105,762✔
1048
    goto end;
×
1049
  }
1050

1051
  SArray* pColList = NULL;
105,762✔
1052
  code = qGetColumnsFromNodeList(group, true, &pColList);
105,762✔
1053
  if (code != TSDB_CODE_SUCCESS) {
105,358✔
1054
    goto end;
×
1055
  }
1056

1057
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
277,007✔
1058
    SColumnInfo* tmp = (SColumnInfo*)taosArrayGet(pColList, i);
172,053✔
1059
    if (tmp != NULL && tmp->colId == -1) {
172,053✔
1060
      tbNameIndex = i;
105,358✔
1061
    }
1062
  }
1063
  
1064
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
105,358✔
1065
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
105,358✔
1066
  taosArrayDestroy(pColList);
105,896✔
1067
  if (pResBlock == NULL) {
105,896✔
1068
    code = terrno;
×
1069
    goto end;
×
1070
  }
1071

1072
  pBlockList = taosArrayInit(2, POINTER_BYTES);
105,896✔
1073
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
105,762✔
1074

1075
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
105,896✔
1076
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
105,896✔
1077

1078
  groupData = taosArrayInit(2, POINTER_BYTES);
105,896✔
1079
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
105,896✔
1080

1081
  SNode* pNode = NULL;
105,896✔
1082
  FOREACH(pNode, group) {
277,685✔
1083
    SScalarParam output = {0};
172,187✔
1084

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

1099
      default:
×
1100
        code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1101
        goto end;
×
1102
    }
1103

1104
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
172,187✔
1105
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
172,187✔
1106
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
172,187✔
1107
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
172,187✔
1108
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
172,187✔
1109
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
×
1110
      continue;
×
1111
    } else {
1112
      code = scalarCalculate(pNode, pBlockList, &output, NULL, NULL);
×
1113
    }
1114

1115
    if (code != TSDB_CODE_SUCCESS) {
171,655✔
1116
      releaseColInfoData(output.columnData);
×
1117
      goto end;
×
1118
    }
1119

1120
    void* tmp = taosArrayPush(groupData, &output.columnData);
171,789✔
1121
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
171,789✔
1122
  }
1123

1124
  for (int i = 0; i < rows; i++) {
443,753✔
1125
    gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
337,857✔
1126
    QUERY_CHECK_NULL(gInfo, code, lino, end, terrno);
338,121✔
1127

1128
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
338,121✔
1129
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
338,121✔
1130

1131
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
846,908✔
1132
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
508,787✔
1133
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
508,653✔
1134
        if (ret != TSDB_CODE_SUCCESS) {
508,787✔
1135
          qError("buildGroupInfo failed at line %d since %s", __LINE__, tstrerror(ret));
×
1136
          goto end;
×
1137
        }
1138
        if (j == tbNameIndex) {
508,787✔
1139
          SStreamGroupValue* v = taosArrayGetLast(gInfo);
338,255✔
1140
          if (v != NULL){
338,255✔
1141
            v->isTbname = true;
338,255✔
1142
            v->uid = info->uid;
338,255✔
1143
          }
1144
        }
1145
    }
1146

1147
    int32_t ret = taosHashPut(groupIdMap, &info->uid, sizeof(info->uid), &gInfo, POINTER_BYTES);
338,255✔
1148
    if (ret != TSDB_CODE_SUCCESS) {
338,255✔
1149
      qError("put groupid to map failed at line %d since %s", __LINE__, tstrerror(ret));
×
1150
      goto end;
×
1151
    }
1152
    qDebug("put groupid to map gid:%" PRIu64, info->uid);
338,255✔
1153
    gInfo = NULL;
338,255✔
1154
  }
1155

1156
end:
105,896✔
1157
  blockDataDestroy(pResBlock);
105,896✔
1158
  taosArrayDestroy(pBlockList);
105,896✔
1159
  taosArrayDestroyEx(pUidTagList, freeItem);
105,896✔
1160
  taosArrayDestroyP(groupData, releaseColInfoData);
105,896✔
1161
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
105,896✔
1162

1163
  if (code != TSDB_CODE_SUCCESS) {
105,896✔
1164
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1165
  }
1166
}
1167

1168
int32_t getColInfoResultForGroupby(void* pVnode, SNodeList* group, STableListInfo* pTableListInfo, uint8_t* digest,
1,810,767✔
1169
                                   SStorageAPI* pAPI, bool initRemainGroups, SHashObj* groupIdMap) {
1170
  int32_t      code = TSDB_CODE_SUCCESS;
1,810,767✔
1171
  int32_t      lino = 0;
1,810,767✔
1172
  SArray*      pBlockList = NULL;
1,810,767✔
1173
  SSDataBlock* pResBlock = NULL;
1,810,767✔
1174
  void*        keyBuf = NULL;
1,811,179✔
1175
  SArray*      groupData = NULL;
1,811,179✔
1176
  SArray*      pUidTagList = NULL;
1,811,179✔
1177
  SArray*      tableList = NULL;
1,811,179✔
1178
  SArray*      gInfo = NULL;
1,811,179✔
1179

1180
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
1,811,179✔
1181
  if (rows == 0) {
1,811,179✔
1182
    return TSDB_CODE_SUCCESS;
×
1183
  } 
1184

1185
  T_MD5_CTX context = {0};
1,811,179✔
1186
  if (tsTagFilterCache && groupIdMap == NULL) {
1,811,179✔
1187
    SNodeListNode* listNode = NULL;
1,376✔
1188
    code = nodesMakeNode(QUERY_NODE_NODE_LIST, (SNode**)&listNode);
1,376✔
1189
    if (TSDB_CODE_SUCCESS != code) {
1,376✔
1190
      goto end;
×
1191
    }
1192
    listNode->pNodeList = group;
1,376✔
1193
    code = genTbGroupDigest((SNode*)listNode, digest, &context);
1,376✔
1194
    QUERY_CHECK_CODE(code, lino, end);
1,376✔
1195

1196
    nodesFree(listNode);
1,376✔
1197

1198
    code = pAPI->metaFn.metaGetCachedTbGroup(pVnode, pTableListInfo->idInfo.suid, context.digest,
1,376✔
1199
                                             tListLen(context.digest), &tableList);
1200
    QUERY_CHECK_CODE(code, lino, end);
1,376✔
1201

1202
    if (tableList) {
1,376✔
1203
      taosArrayDestroy(pTableListInfo->pTableList);
×
1204
      pTableListInfo->pTableList = tableList;
×
1205
      qDebug("retrieve tb group list from cache, numOfTables:%d",
×
1206
             (int32_t)taosArrayGetSize(pTableListInfo->pTableList));
1207
      goto end;
×
1208
    }
1209
  }
1210

1211
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
1,811,179✔
1212
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
1,811,179✔
1213

1214
  for (int32_t i = 0; i < rows; ++i) {
10,425,090✔
1215
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
8,613,911✔
1216
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
8,613,911✔
1217
    STUidTagInfo info = {.uid = pkeyInfo->uid};
8,613,911✔
1218
    void*        tmp = taosArrayPush(pUidTagList, &info);
8,613,911✔
1219
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
8,613,911✔
1220
  }
1221

1222
  if (taosArrayGetSize(pUidTagList) > 0) {
1,811,179✔
1223
    code = pAPI->metaFn.getTableTagsByUid(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
1,811,179✔
1224
  } else {
1225
    code = pAPI->metaFn.getTableTags(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
×
1226
  }
1227
  if (code != TSDB_CODE_SUCCESS) {
1,811,179✔
1228
    goto end;
×
1229
  }
1230

1231
  SArray* pColList = NULL;
1,811,179✔
1232
  code = qGetColumnsFromNodeList(group, true, &pColList); 
1,811,179✔
1233
  if (code != TSDB_CODE_SUCCESS) {
1,810,115✔
1234
    goto end;
×
1235
  }
1236

1237
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
1,810,115✔
1238
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
1,810,249✔
1239
  taosArrayDestroy(pColList);
1,810,503✔
1240
  if (pResBlock == NULL) {
1,811,179✔
1241
    code = terrno;
×
1242
    goto end;
×
1243
  }
1244

1245
  //  int64_t st1 = taosGetTimestampUs();
1246
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1247

1248
  pBlockList = taosArrayInit(2, POINTER_BYTES);
1,811,179✔
1249
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
1,811,179✔
1250

1251
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
1,811,038✔
1252
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
1,811,038✔
1253

1254
  groupData = taosArrayInit(2, POINTER_BYTES);
1,811,038✔
1255
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
1,810,503✔
1256

1257
  SNode* pNode = NULL;
1,810,556✔
1258
  FOREACH(pNode, group) {
3,740,879✔
1259
    SScalarParam output = {0};
1,930,554✔
1260

1261
    switch (nodeType(pNode)) {
1,930,554✔
1262
      case QUERY_NODE_VALUE:
×
1263
        break;
×
1264
      case QUERY_NODE_COLUMN:
1,931,295✔
1265
      case QUERY_NODE_OPERATOR:
1266
      case QUERY_NODE_FUNCTION: {
1267
        SExprNode* expNode = (SExprNode*)pNode;
1,931,295✔
1268
        code = createResultData(&expNode->resType, rows, &output);
1,931,295✔
1269
        if (code != TSDB_CODE_SUCCESS) {
1,929,938✔
1270
          goto end;
×
1271
        }
1272
        break;
1,929,938✔
1273
      }
1274

1275
      default:
×
1276
        code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1277
        goto end;
×
1278
    }
1279

1280
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
1,929,938✔
1281
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
1,919,132✔
1282
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
1,919,132✔
1283
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
1,917,640✔
1284
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
1,917,640✔
1285
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
10,806✔
1286
      continue;
×
1287
    } else {
1288
      code = scalarCalculate(pNode, pBlockList, &output, NULL, NULL);
10,129✔
1289
    }
1290

1291
    if (code != TSDB_CODE_SUCCESS) {
1,927,902✔
1292
      releaseColInfoData(output.columnData);
×
1293
      goto end;
×
1294
    }
1295

1296
    void* tmp = taosArrayPush(groupData, &output.columnData);
1,930,182✔
1297
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
1,930,182✔
1298
  }
1299

1300
  int32_t keyLen = 0;
1,809,382✔
1301
  SNode*  node;
1302
  FOREACH(node, group) {
3,738,776✔
1303
    SExprNode* pExpr = (SExprNode*)node;
1,929,371✔
1304
    keyLen += pExpr->resType.bytes;
1,929,371✔
1305
  }
1306

1307
  int32_t nullFlagSize = sizeof(int8_t) * LIST_LENGTH(group);
1,808,227✔
1308
  keyLen += nullFlagSize;
1,808,361✔
1309

1310
  keyBuf = taosMemoryCalloc(1, keyLen);
1,808,361✔
1311
  if (keyBuf == NULL) {
1,809,425✔
1312
    code = terrno;
×
1313
    goto end;
×
1314
  }
1315

1316
  if (initRemainGroups) {
1,809,425✔
1317
    pTableListInfo->remainGroups =
785,254✔
1318
        taosHashInit(rows, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
782,378✔
1319
    if (pTableListInfo->remainGroups == NULL) {
785,254✔
1320
      code = terrno;
×
1321
      goto end;
×
1322
    }
1323
  }
1324

1325
  for (int i = 0; i < rows; i++) {
10,419,342✔
1326
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
8,609,178✔
1327
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
8,609,350✔
1328

1329
    if (groupIdMap != NULL){
8,609,350✔
1330
      gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
222,734✔
1331
    }
1332
    
1333
    char* isNull = (char*)keyBuf;
8,606,838✔
1334
    char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(group);
8,606,838✔
1335
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
17,863,858✔
1336
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
9,253,379✔
1337

1338
      if (groupIdMap != NULL && gInfo != NULL) {
9,255,934✔
1339
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
246,062✔
1340
        if (ret != TSDB_CODE_SUCCESS) {
246,062✔
1341
          qError("buildGroupInfo failed at line %d since %s", __LINE__, tstrerror(ret));
×
1342
          taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1343
          gInfo = NULL;
×
1344
        }
1345
      }
1346
      
1347
      if (colDataIsNull_s(pValue, i)) {
18,512,855✔
1348
        isNull[j] = 1;
88,592✔
1349
      } else {
1350
        isNull[j] = 0;
9,168,329✔
1351
        char* data = colDataGetData(pValue, i);
9,167,608✔
1352
        if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
9,167,397✔
1353
          // if (tTagIsJson(data)) {
1354
          //   code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
1355
          //   goto end;
1356
          // }
1357
          if (tTagIsJsonNull(data)) {
89,324✔
1358
            isNull[j] = 1;
×
1359
            continue;
×
1360
          }
1361
          int32_t len = getJsonValueLen(data);
89,324✔
1362
          memcpy(pStart, data, len);
89,324✔
1363
          pStart += len;
89,324✔
1364
        } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
9,075,202✔
1365
          if (IS_STR_DATA_BLOB(pValue->info.type)) {
6,016,426✔
UNCOV
1366
            if (blobDataTLen(data) > TSDB_MAX_BLOB_LEN) {
×
1367
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1368
              goto end;
×
1369
            }
1370
            memcpy(pStart, data, blobDataTLen(data));
×
1371
            pStart += blobDataTLen(data);
×
1372
          } else {
1373
            if (varDataTLen(data) > pValue->info.bytes) {
6,016,589✔
1374
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1375
              goto end;
×
1376
            }
1377
            memcpy(pStart, data, varDataTLen(data));
6,017,733✔
1378
            pStart += varDataTLen(data);
6,017,733✔
1379
          }
1380
        } else {
1381
          memcpy(pStart, data, pValue->info.bytes);
3,060,891✔
1382
          pStart += pValue->info.bytes;
3,060,891✔
1383
        }
1384
      }
1385
    }
1386

1387
    int32_t len = (int32_t)(pStart - (char*)keyBuf);
8,605,999✔
1388
    info->groupId = calcGroupId(keyBuf, len);
8,605,999✔
1389
    if (groupIdMap != NULL && gInfo != NULL) {
8,607,746✔
1390
      int32_t ret = taosHashPut(groupIdMap, &info->groupId, sizeof(info->groupId), &gInfo, POINTER_BYTES);
222,336✔
1391
      if (ret != TSDB_CODE_SUCCESS) {
222,734✔
1392
        qError("put groupid to map failed at line %d since %s", __LINE__, tstrerror(ret));
×
1393
        taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1394
      }
1395
      qDebug("put groupid to map gid:%" PRIu64, info->groupId);
222,734✔
1396
      gInfo = NULL;
222,734✔
1397
    }
1398
    if (initRemainGroups) {
8,608,144✔
1399
      // groupId ~ table uid
1400
      code = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId), &(info->uid),
4,164,049✔
1401
                         sizeof(info->uid));
1402
      if (code == TSDB_CODE_DUP_KEY) {
4,164,550✔
1403
        code = TSDB_CODE_SUCCESS;
828,070✔
1404
      }
1405
      QUERY_CHECK_CODE(code, lino, end);
4,164,550✔
1406
    }
1407
  }
1408

1409
  if (tsTagFilterCache && groupIdMap == NULL) {
1,810,164✔
1410
    tableList = taosArrayDup(pTableListInfo->pTableList, NULL);
1,376✔
1411
    QUERY_CHECK_NULL(tableList, code, lino, end, terrno);
1,376✔
1412

1413
    code = pAPI->metaFn.metaPutTbGroupToCache(pVnode, pTableListInfo->idInfo.suid, context.digest,
2,752✔
1414
                                              tListLen(context.digest), tableList,
1415
                                              taosArrayGetSize(tableList) * sizeof(STableKeyInfo));
1,376✔
1416
    QUERY_CHECK_CODE(code, lino, end);
1,376✔
1417
  }
1418

1419
  //  int64_t st2 = taosGetTimestampUs();
1420
  //  qDebug("calculate tag block rows:%d, cost:%ld us", rows, st2-st1);
1421

1422
end:
1,810,023✔
1423
  taosMemoryFreeClear(keyBuf);
1,811,179✔
1424
  blockDataDestroy(pResBlock);
1,811,765✔
1425
  taosArrayDestroy(pBlockList);
1,810,572✔
1426
  taosArrayDestroyEx(pUidTagList, freeItem);
1,809,576✔
1427
  taosArrayDestroyP(groupData, releaseColInfoData);
1,810,697✔
1428
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
1,809,645✔
1429

1430
  if (code != TSDB_CODE_SUCCESS) {
1,810,572✔
1431
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1432
  }
1433
  return code;
1,808,969✔
1434
}
1435

1436
static int32_t nameComparFn(const void* p1, const void* p2) {
722,266✔
1437
  const char* pName1 = *(const char**)p1;
722,266✔
1438
  const char* pName2 = *(const char**)p2;
722,266✔
1439

1440
  int32_t ret = strcmp(pName1, pName2);
722,266✔
1441
  if (ret == 0) {
722,266✔
1442
    return 0;
18,108✔
1443
  } else {
1444
    return (ret > 0) ? 1 : -1;
704,158✔
1445
  }
1446
}
1447

1448
static SArray* getTableNameList(const SNodeListNode* pList) {
418,278✔
1449
  int32_t    code = TSDB_CODE_SUCCESS;
418,278✔
1450
  int32_t    lino = 0;
418,278✔
1451
  int32_t    len = LIST_LENGTH(pList->pNodeList);
418,278✔
1452
  SListCell* cell = pList->pNodeList->pHead;
418,278✔
1453

1454
  SArray* pTbList = taosArrayInit(len, POINTER_BYTES);
418,278✔
1455
  QUERY_CHECK_NULL(pTbList, code, lino, _end, terrno);
418,278✔
1456

1457
  for (int i = 0; i < pList->pNodeList->length; i++) {
1,135,682✔
1458
    SValueNode* valueNode = (SValueNode*)cell->pNode;
717,404✔
1459
    if (!IS_VAR_DATA_TYPE(valueNode->node.resType.type)) {
717,404✔
1460
      terrno = TSDB_CODE_INVALID_PARA;
×
1461
      taosArrayDestroy(pTbList);
×
1462
      return NULL;
×
1463
    }
1464

1465
    char* name = varDataVal(valueNode->datum.p);
717,404✔
1466
    void* tmp = taosArrayPush(pTbList, &name);
717,404✔
1467
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
717,404✔
1468
    cell = cell->pNext;
717,404✔
1469
  }
1470

1471
  size_t numOfTables = taosArrayGetSize(pTbList);
418,278✔
1472

1473
  // order the name
1474
  taosArraySort(pTbList, nameComparFn);
418,278✔
1475

1476
  // remove the duplicates
1477
  SArray* pNewList = taosArrayInit(taosArrayGetSize(pTbList), sizeof(void*));
418,278✔
1478
  QUERY_CHECK_NULL(pNewList, code, lino, _end, terrno);
418,278✔
1479
  void* tmpTbl = taosArrayGet(pTbList, 0);
418,278✔
1480
  QUERY_CHECK_NULL(tmpTbl, code, lino, _end, terrno);
418,278✔
1481
  void* tmp = taosArrayPush(pNewList, tmpTbl);
418,278✔
1482
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
418,278✔
1483

1484
  for (int32_t i = 1; i < numOfTables; ++i) {
717,404✔
1485
    char** name = taosArrayGetLast(pNewList);
299,126✔
1486
    char** nameInOldList = taosArrayGet(pTbList, i);
299,126✔
1487
    QUERY_CHECK_NULL(nameInOldList, code, lino, _end, terrno);
299,126✔
1488
    if (strcmp(*name, *nameInOldList) == 0) {
299,126✔
1489
      continue;
9,736✔
1490
    }
1491

1492
    tmp = taosArrayPush(pNewList, nameInOldList);
289,390✔
1493
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
289,390✔
1494
  }
1495

1496
_end:
418,278✔
1497
  taosArrayDestroy(pTbList);
418,278✔
1498
  if (code != TSDB_CODE_SUCCESS) {
418,278✔
1499
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1500
    return NULL;
×
1501
  }
1502
  return pNewList;
418,278✔
1503
}
1504

1505
static int tableUidCompare(const void* a, const void* b) {
×
1506
  uint64_t u1 = *(uint64_t*)a;
×
1507
  uint64_t u2 = *(uint64_t*)b;
×
1508

1509
  if (u1 == u2) {
×
1510
    return 0;
×
1511
  }
1512

1513
  return u1 < u2 ? -1 : 1;
×
1514
}
1515

1516
static int32_t filterTableInfoCompare(const void* a, const void* b) {
18,250,792✔
1517
  STUidTagInfo* p1 = (STUidTagInfo*)a;
18,250,792✔
1518
  STUidTagInfo* p2 = (STUidTagInfo*)b;
18,250,792✔
1519

1520
  if (p1->uid == p2->uid) {
18,250,792✔
1521
    return 0;
×
1522
  }
1523

1524
  return p1->uid < p2->uid ? -1 : 1;
18,250,792✔
1525
}
1526

1527
static FilterCondType checkTagCond(SNode* cond) {
12,744,328✔
1528
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
12,744,328✔
1529
    return FILTER_NO_LOGIC;
10,875,962✔
1530
  }
1531
  if (nodeType(cond) != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
1,870,171✔
1532
    return FILTER_AND;
216,295✔
1533
  }
1534
  return FILTER_OTHER;
1,652,688✔
1535
}
1536

1537
static int32_t optimizeTbnameInCond(void* pVnode, int64_t suid, SArray* list, SNode* cond, SStorageAPI* pAPI) {
12,744,076✔
1538
  int32_t ret = -1;
12,744,076✔
1539
  int32_t ntype = nodeType(cond);
12,744,076✔
1540

1541
  if (ntype == QUERY_NODE_OPERATOR) {
12,745,071✔
1542
    ret = optimizeTbnameInCondImpl(pVnode, list, cond, pAPI, suid);
10,876,347✔
1543
  }
1544

1545
  if (ntype != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
12,743,226✔
1546
    return ret;
11,092,839✔
1547
  }
1548

1549
  bool                 hasTbnameCond = false;
1,649,775✔
1550
  SLogicConditionNode* pNode = (SLogicConditionNode*)cond;
1,649,775✔
1551
  SNodeList*           pList = (SNodeList*)pNode->pParameterList;
1,649,775✔
1552

1553
  int32_t len = LIST_LENGTH(pList);
1,650,848✔
1554
  if (len <= 0) {
1,651,331✔
1555
    return ret;
×
1556
  }
1557

1558
  SListCell* cell = pList->pHead;
1,651,331✔
1559
  for (int i = 0; i < len; i++) {
5,370,443✔
1560
    if (cell == NULL) break;
3,723,008✔
1561
    if (optimizeTbnameInCondImpl(pVnode, list, cell->pNode, pAPI, suid) == 0) {
3,723,008✔
1562
      hasTbnameCond = true;
6,210✔
1563
      break;
6,210✔
1564
    }
1565
    cell = cell->pNext;
3,717,856✔
1566
  }
1567

1568
  taosArraySort(list, filterTableInfoCompare);
1,653,645✔
1569
  taosArrayRemoveDuplicate(list, filterTableInfoCompare, NULL);
1,652,594✔
1570

1571
  if (hasTbnameCond) {
1,651,020✔
1572
    ret = pAPI->metaFn.getTableTagsByUid(pVnode, suid, list);
6,210✔
1573
  }
1574

1575
  return ret;
1,650,717✔
1576
}
1577

1578
// only return uid that does not contained in pExistedUidList
1579
static int32_t optimizeTbnameInCondImpl(void* pVnode, SArray* pExistedUidList, SNode* pTagCond, SStorageAPI* pStoreAPI,
14,601,113✔
1580
                                        uint64_t suid) {
1581
  if (nodeType(pTagCond) != QUERY_NODE_OPERATOR) {
14,601,113✔
1582
    return -1;
6,272✔
1583
  }
1584

1585
  SOperatorNode* pNode = (SOperatorNode*)pTagCond;
14,594,650✔
1586
  if (pNode->opType != OP_TYPE_IN) {
14,594,650✔
1587
    return -1;
13,890,964✔
1588
  }
1589

1590
  if ((pNode->pLeft != NULL && ((nodeType(pNode->pLeft) == QUERY_NODE_FUNCTION &&
705,222✔
1591
                                 ((SFunctionNode*)pNode->pLeft)->funcType == FUNCTION_TYPE_TBNAME)) ||
418,278✔
1592
       (nodeType(pNode->pLeft) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pLeft)->colType == COLUMN_TYPE_TBNAME)) &&
286,750✔
1593
      (pNode->pRight != NULL && nodeType(pNode->pRight) == QUERY_NODE_NODE_LIST)) {
418,278✔
1594
    SNodeListNode* pList = (SNodeListNode*)pNode->pRight;
418,278✔
1595

1596
    int32_t len = LIST_LENGTH(pList->pNodeList);
418,278✔
1597
    if (len <= 0) {
418,278✔
1598
      return -1;
×
1599
    }
1600

1601
    SArray*   pTbList = getTableNameList(pList);
418,278✔
1602
    int32_t   numOfTables = taosArrayGetSize(pTbList);
418,278✔
1603
    SHashObj* uHash = NULL;
418,278✔
1604

1605
    size_t numOfExisted = taosArrayGetSize(pExistedUidList);  // len > 0 means there already have uids
418,278✔
1606
    if (numOfExisted > 0) {
418,278✔
1607
      uHash = taosHashInit(numOfExisted / 0.7, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
2,336✔
1608
      if (!uHash) {
2,336✔
1609
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1610
        return terrno;
×
1611
      }
1612

1613
      for (int i = 0; i < numOfExisted; i++) {
2,331,912✔
1614
        STUidTagInfo* pTInfo = taosArrayGet(pExistedUidList, i);
2,329,576✔
1615
        if (!pTInfo) {
2,329,576✔
1616
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1617
          return terrno;
×
1618
        }
1619
        int32_t tempRes = taosHashPut(uHash, &pTInfo->uid, sizeof(uint64_t), &i, sizeof(i));
2,329,576✔
1620
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
2,329,576✔
1621
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
1622
          return tempRes;
×
1623
        }
1624
      }
1625
    }
1626

1627
    for (int i = 0; i < numOfTables; i++) {
1,109,738✔
1628
      char* name = taosArrayGetP(pTbList, i);
697,932✔
1629

1630
      uint64_t uid = 0, csuid = 0;
697,932✔
1631
      if (pStoreAPI->metaFn.getTableUidByName(pVnode, name, &uid) == 0) {
697,932✔
1632
        ETableType tbType = TSDB_TABLE_MAX;
418,404✔
1633
        if (pStoreAPI->metaFn.getTableTypeSuidByName(pVnode, name, &tbType, &csuid) == 0 &&
418,404✔
1634
            tbType == TSDB_CHILD_TABLE) {
418,211✔
1635
          if (suid != csuid) {
411,739✔
1636
            continue;
872✔
1637
          }
1638
          if (NULL == uHash || taosHashGet(uHash, &uid, sizeof(uid)) == NULL) {
410,867✔
1639
            STUidTagInfo s = {.uid = uid, .name = name, .pTagVal = NULL};
409,699✔
1640
            void*        tmp = taosArrayPush(pExistedUidList, &s);
409,892✔
1641
            if (!tmp) {
409,892✔
1642
              return terrno;
×
1643
            }
1644
          }
1645
        } else {
1646
          taosArrayDestroy(pTbList);
6,472✔
1647
          taosHashCleanup(uHash);
6,472✔
1648
          return -1;
6,472✔
1649
        }
1650
      } else {
1651
        //        qWarn("failed to get tableIds from by table name: %s, reason: %s", name, tstrerror(terrno));
1652
        terrno = 0;
279,528✔
1653
      }
1654
    }
1655

1656
    taosHashCleanup(uHash);
411,806✔
1657
    taosArrayDestroy(pTbList);
411,806✔
1658
    return 0;
411,806✔
1659
  }
1660

1661
  return -1;
286,944✔
1662
}
1663

1664
SSDataBlock* createTagValBlockForFilter(SArray* pColList, int32_t numOfTables, SArray* pUidTagList, void* pVnode,
13,519,151✔
1665
                                        SStorageAPI* pStorageAPI) {
1666
  int32_t      code = TSDB_CODE_SUCCESS;
13,519,151✔
1667
  int32_t      lino = 0;
13,519,151✔
1668
  SSDataBlock* pResBlock = NULL;
13,519,151✔
1669
  code = createDataBlock(&pResBlock);
13,520,927✔
1670
  QUERY_CHECK_CODE(code, lino, _end);
13,518,440✔
1671

1672
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
28,245,551✔
1673
    SColumnInfoData colInfo = {0};
14,727,361✔
1674
    void*           tmp = taosArrayGet(pColList, i);
14,727,361✔
1675
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
14,718,381✔
1676
    colInfo.info = *(SColumnInfo*)tmp;
14,718,381✔
1677
    code = blockDataAppendColInfo(pResBlock, &colInfo);
14,722,781✔
1678
    QUERY_CHECK_CODE(code, lino, _end);
14,726,823✔
1679
  }
1680

1681
  code = blockDataEnsureCapacity(pResBlock, numOfTables);
13,516,171✔
1682
  if (code != TSDB_CODE_SUCCESS) {
13,518,602✔
1683
    terrno = code;
×
1684
    blockDataDestroy(pResBlock);
×
1685
    return NULL;
×
1686
  }
1687

1688
  pResBlock->info.rows = numOfTables;
13,518,602✔
1689

1690
  int32_t numOfCols = taosArrayGetSize(pResBlock->pDataBlock);
13,523,718✔
1691

1692
  for (int32_t i = 0; i < numOfTables; i++) {
199,633,572✔
1693
    STUidTagInfo* p1 = taosArrayGet(pUidTagList, i);
186,107,289✔
1694
    QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
186,137,997✔
1695

1696
    for (int32_t j = 0; j < numOfCols; j++) {
379,835,605✔
1697
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, j);
193,684,754✔
1698
      QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
193,660,177✔
1699

1700
      if (pColInfo->info.colId == -1) {  // tbname
193,660,177✔
1701
        char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
7,886,005✔
1702
        if (p1->name != NULL) {
7,886,749✔
1703
          STR_TO_VARSTR(str, p1->name);
409,892✔
1704
        } else {  // name is not retrieved during filter
1705
          code = pStorageAPI->metaFn.getTableNameByUid(pVnode, p1->uid, str);
7,476,273✔
1706
          QUERY_CHECK_CODE(code, lino, _end);
7,478,383✔
1707
        }
1708

1709
        code = colDataSetVal(pColInfo, i, str, false);
7,888,275✔
1710
        QUERY_CHECK_CODE(code, lino, _end);
7,887,757✔
1711
#if TAG_FILTER_DEBUG
1712
        qDebug("tagfilter uid:%ld, tbname:%s", *uid, str + 2);
1713
#endif
1714
      } else {
1715
        STagVal tagVal = {0};
185,769,199✔
1716
        tagVal.cid = pColInfo->info.colId;
185,780,047✔
1717
        if (p1->pTagVal == NULL) {
185,776,400✔
1718
          colDataSetNULL(pColInfo, i);
1,720✔
1719
        } else {
1720
          const char* p = pStorageAPI->metaFn.extractTagVal(p1->pTagVal, pColInfo->info.type, &tagVal);
185,770,548✔
1721

1722
          if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
185,794,706✔
1723
            colDataSetNULL(pColInfo, i);
4,237,275✔
1724
          } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
181,560,449✔
1725
            code = colDataSetVal(pColInfo, i, p, false);
706,497✔
1726
            QUERY_CHECK_CODE(code, lino, _end);
706,497✔
1727
          } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
292,076,191✔
1728
            if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
111,211,888✔
1729
              QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
1730
            }
1731
            char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
111,226,094✔
1732
            QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
111,232,484✔
1733
            varDataSetLen(tmp, tagVal.nData);
111,232,484✔
1734
            memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
111,228,049✔
1735
            code = colDataSetVal(pColInfo, i, tmp, false);
111,227,314✔
1736
#if TAG_FILTER_DEBUG
1737
            qDebug("tagfilter varch:%s", tmp + 2);
1738
#endif
1739
            taosMemoryFree(tmp);
111,230,522✔
1740
            QUERY_CHECK_CODE(code, lino, _end);
111,235,236✔
1741
          } else {
1742
            code = colDataSetVal(pColInfo, i, (const char*)&tagVal.i64, false);
69,622,134✔
1743
            QUERY_CHECK_CODE(code, lino, _end);
69,629,507✔
1744
#if TAG_FILTER_DEBUG
1745
            if (pColInfo->info.type == TSDB_DATA_TYPE_INT) {
1746
              qDebug("tagfilter int:%d", *(int*)(&tagVal.i64));
1747
            } else if (pColInfo->info.type == TSDB_DATA_TYPE_DOUBLE) {
1748
              qDebug("tagfilter double:%f", *(double*)(&tagVal.i64));
1749
            }
1750
#endif
1751
          }
1752
        }
1753
      }
1754
    }
1755
  }
1756

1757
_end:
13,523,982✔
1758
  if (code != TSDB_CODE_SUCCESS) {
13,526,283✔
1759
    blockDataDestroy(pResBlock);
1,431✔
1760
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1761
    terrno = code;
×
1762
    return NULL;
×
1763
  }
1764
  return pResBlock;
13,524,852✔
1765
}
1766

1767
static int32_t doSetQualifiedUid(STableListInfo* pListInfo, SArray* pUidList, const SArray* pUidTagList,
10,953,189✔
1768
                                 bool* pResultList, bool addUid) {
1769
  taosArrayClear(pUidList);
10,953,189✔
1770

1771
  STableKeyInfo info = {.uid = 0, .groupId = 0};
10,945,413✔
1772
  int32_t       numOfTables = taosArrayGetSize(pUidTagList);
10,948,300✔
1773
  for (int32_t i = 0; i < numOfTables; ++i) {
187,084,674✔
1774
    if (pResultList[i]) {
176,117,559✔
1775
      STUidTagInfo* tmpTag = (STUidTagInfo*)taosArrayGet(pUidTagList, i);
78,009,342✔
1776
      if (!tmpTag) {
78,013,752✔
1777
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1778
        return terrno;
×
1779
      }
1780
      uint64_t uid = tmpTag->uid;
78,013,752✔
1781
      qDebug("tagfilter get uid:%" PRId64 ", res:%d", uid, pResultList[i]);
77,996,031✔
1782

1783
      info.uid = uid;
78,011,732✔
1784
      //qInfo("doSetQualifiedUid row:%d added to pTableList", i);
1785
      void* p = taosArrayPush(pListInfo->pTableList, &info);
78,011,732✔
1786
      if (p == NULL) {
78,024,362✔
1787
        return terrno;
×
1788
      }
1789

1790
      if (addUid) {
78,024,362✔
1791
        //qInfo("doSetQualifiedUid row:%d added to pUidList", i);
1792
        void* tmp = taosArrayPush(pUidList, &uid);
19,606✔
1793
        if (tmp == NULL) {
19,606✔
1794
          return terrno;
×
1795
        }
1796
      }
1797
    } else {
1798
      //qInfo("doSetQualifiedUid row:%d failed", i);
1799
    }
1800
  }
1801

1802
  return TSDB_CODE_SUCCESS;
10,967,115✔
1803
}
1804

1805
static int32_t copyExistedUids(SArray* pUidTagList, const SArray* pUidList) {
12,744,959✔
1806
  int32_t code = TSDB_CODE_SUCCESS;
12,744,959✔
1807
  int32_t numOfExisted = taosArrayGetSize(pUidList);
12,744,959✔
1808
  if (numOfExisted == 0) {
12,743,373✔
1809
    return code;
9,873,293✔
1810
  }
1811

1812
  for (int32_t i = 0; i < numOfExisted; ++i) {
35,389,573✔
1813
    uint64_t* uid = taosArrayGet(pUidList, i);
32,519,034✔
1814
    if (!uid) {
32,519,249✔
1815
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1816
      return terrno;
×
1817
    }
1818
    STUidTagInfo info = {.uid = *uid};
32,519,249✔
1819
    void*        tmp = taosArrayPush(pUidTagList, &info);
32,519,952✔
1820
    if (!tmp) {
32,519,952✔
1821
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1822
      return code;
×
1823
    }
1824
  }
1825
  return code;
2,870,539✔
1826
}
1827

1828
int32_t doFilterByTagCond(STableListInfo* pListInfo, SArray* pUidList, SNode* pTagCond, void* pVnode,
118,041,700✔
1829
                                 SIdxFltStatus status, SStorageAPI* pAPI, bool addUid, bool* listAdded, void* pStreamInfo) {
1830
  *listAdded = false;
118,041,700✔
1831
  if (pTagCond == NULL) {
118,057,314✔
1832
    return TSDB_CODE_SUCCESS;
105,276,195✔
1833
  }
1834

1835
  terrno = TSDB_CODE_SUCCESS;
12,781,119✔
1836

1837
  int32_t      lino = 0;
12,745,439✔
1838
  int32_t      code = TSDB_CODE_SUCCESS;
12,745,439✔
1839
  SArray*      pBlockList = NULL;
12,745,439✔
1840
  SSDataBlock* pResBlock = NULL;
12,745,439✔
1841
  SScalarParam output = {0};
12,745,136✔
1842
  SArray*      pUidTagList = NULL;
12,744,745✔
1843

1844
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
12,744,745✔
1845

1846
  //  int64_t stt = taosGetTimestampUs();
1847
  pUidTagList = taosArrayInit(10, sizeof(STUidTagInfo));
12,742,951✔
1848
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
12,743,663✔
1849

1850
  code = copyExistedUids(pUidTagList, pUidList);
12,743,663✔
1851
  QUERY_CHECK_CODE(code, lino, end);
12,743,274✔
1852

1853
  FilterCondType condType = checkTagCond(pTagCond);
12,743,274✔
1854

1855
  int32_t filter = optimizeTbnameInCond(pVnode, pListInfo->idInfo.suid, pUidTagList, pTagCond, pAPI);
12,743,959✔
1856
  if (filter == 0) {  // tbname in filter is activated, do nothing and return
12,741,701✔
1857
    taosArrayClear(pUidList);
411,806✔
1858

1859
    int32_t numOfRows = taosArrayGetSize(pUidTagList);
411,806✔
1860
    code = taosArrayEnsureCap(pUidList, numOfRows);
411,806✔
1861
    QUERY_CHECK_CODE(code, lino, end);
411,806✔
1862

1863
    for (int32_t i = 0; i < numOfRows; ++i) {
3,154,194✔
1864
      STUidTagInfo* pInfo = taosArrayGet(pUidTagList, i);
2,742,388✔
1865
      QUERY_CHECK_NULL(pInfo, code, lino, end, terrno);
2,742,388✔
1866
      void* tmp = taosArrayPush(pUidList, &pInfo->uid);
2,742,388✔
1867
      QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
2,742,388✔
1868
    }
1869
    terrno = 0;
411,806✔
1870
  } else {
1871
    qDebug("pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
12,329,895✔
1872
    
1873
    if (((condType == FILTER_NO_LOGIC || condType == FILTER_AND) && status != SFLT_NOT_INDEX) ||
21,606,173✔
1874
          taosArrayGetSize(pUidTagList) > 0) {
9,273,916✔
1875
      code = pAPI->metaFn.getTableTagsByUid(pVnode, pListInfo->idInfo.suid, pUidTagList);
3,550,088✔
1876
    } else {
1877
      code = pAPI->metaFn.getTableTags(pVnode, pListInfo->idInfo.suid, pUidTagList);
8,782,169✔
1878
    }
1879
    if (code != TSDB_CODE_SUCCESS) {
12,330,647✔
1880
      qError("failed to get table tags from meta, reason:%s, suid:%" PRIu64, tstrerror(code), pListInfo->idInfo.suid);
×
1881
      terrno = code;
×
1882
      QUERY_CHECK_CODE(code, lino, end);
×
1883
    }
1884
  }
1885

1886
  qDebug("final pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
12,742,453✔
1887

1888
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
12,744,977✔
1889
  if (numOfTables == 0) {
12,745,530✔
1890
    goto end;
1,775,640✔
1891
  }
1892

1893
  SArray* pColList = NULL;
10,969,890✔
1894
  code = qGetColumnsFromNodeList(pTagCond, false, &pColList); 
10,970,081✔
1895
  if (code != TSDB_CODE_SUCCESS) {
10,965,986✔
1896
    goto end;
×
1897
  }
1898
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
10,965,986✔
1899
  taosArrayDestroy(pColList);
10,968,531✔
1900
  if (pResBlock == NULL) {
10,967,778✔
1901
    code = terrno;
×
1902
    QUERY_CHECK_CODE(code, lino, end);
×
1903
  }
1904

1905
  //fprintDataBlock(pResBlock, "tagFilter", "", 0);
1906

1907
  //  int64_t st1 = taosGetTimestampUs();
1908
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1909
  pBlockList = taosArrayInit(2, POINTER_BYTES);
10,967,778✔
1910
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
10,967,945✔
1911

1912
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
10,968,470✔
1913
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
10,968,470✔
1914

1915
  code = createResultData(&type, numOfTables, &output);
10,968,470✔
1916
  if (code != TSDB_CODE_SUCCESS) {
10,965,752✔
1917
    terrno = code;
×
1918
    QUERY_CHECK_CODE(code, lino, end);
×
1919
  }
1920

1921
  code = scalarCalculate(pTagCond, pBlockList, &output, pStreamInfo, NULL);
10,965,752✔
1922
  if (code != TSDB_CODE_SUCCESS) {
10,953,046✔
1923
    qError("failed to calculate scalar, reason:%s", tstrerror(code));
1,096✔
1924
    terrno = code;
1,096✔
1925
    QUERY_CHECK_CODE(code, lino, end);
1,096✔
1926
  }
1927

1928
  code = doSetQualifiedUid(pListInfo, pUidList, pUidTagList, (bool*)output.columnData->pData, addUid);
10,951,950✔
1929
  if (code != TSDB_CODE_SUCCESS) {
10,966,733✔
1930
    terrno = code;
×
1931
    QUERY_CHECK_CODE(code, lino, end);
×
1932
  }
1933
  *listAdded = true;
10,966,733✔
1934

1935
end:
12,744,036✔
1936
  if (code != TSDB_CODE_SUCCESS) {
12,743,151✔
1937
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1,096✔
1938
  }
1939
  blockDataDestroy(pResBlock);
12,743,151✔
1940
  taosArrayDestroy(pBlockList);
12,742,751✔
1941
  taosArrayDestroyEx(pUidTagList, freeItem);
12,739,987✔
1942

1943
  colDataDestroy(output.columnData);
12,745,221✔
1944
  taosMemoryFreeClear(output.columnData);
12,743,525✔
1945
  return code;
12,742,261✔
1946
}
1947

1948
typedef struct {
1949
  int32_t code;
1950
  SStreamRuntimeFuncInfo* pStreamRuntimeInfo;
1951
} PlaceHolderContext;
1952

1953
static EDealRes replacePlaceHolderColumn(SNode** pNode, void* pContext) {
177,988✔
1954
  PlaceHolderContext* pData = (PlaceHolderContext*)pContext;
177,988✔
1955
  if (QUERY_NODE_FUNCTION != nodeType((*pNode))) {
177,988✔
1956
    return DEAL_RES_CONTINUE;
147,600✔
1957
  }
1958
  SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
30,388✔
1959
  if (!fmIsStreamPesudoColVal(pFuncNode->funcId)) {
30,388✔
1960
    return DEAL_RES_CONTINUE;
872✔
1961
  }
1962
  pData->code = fmSetStreamPseudoFuncParamVal(pFuncNode->funcId, pFuncNode->pParameterList, pData->pStreamRuntimeInfo);
29,516✔
1963
  if (pData->code != TSDB_CODE_SUCCESS) {
29,516✔
1964
    return DEAL_RES_ERROR;
×
1965
  }
1966
  SNode* pFirstParam = nodesListGetNode(pFuncNode->pParameterList, 0);
29,516✔
1967
  ((SValueNode*)pFirstParam)->translate = true;
29,516✔
1968
  SValueNode* res = NULL;
29,516✔
1969
  pData->code = nodesCloneNode(pFirstParam, (SNode**)&res);
29,516✔
1970
  if (NULL == res) {
29,516✔
1971
    return DEAL_RES_ERROR;
×
1972
  }
1973
  nodesDestroyNode(*pNode);
29,516✔
1974
  *pNode = (SNode*)res;
29,516✔
1975

1976
  return DEAL_RES_CONTINUE;
29,516✔
1977
}
1978

1979
static void extractTagColId(SOperatorNode* pOpNode, SArray* pColIdArray) {
48,848✔
1980
  SNode* pLeft = pOpNode->pLeft;
48,848✔
1981
  SNode* pRight = pOpNode->pRight;
48,848✔
1982
  SColumnNode* pColNode = nodeType(pLeft) == QUERY_NODE_COLUMN ?
48,848✔
1983
    (SColumnNode*)pLeft : (SColumnNode*)pRight;
48,848✔
1984

1985
  col_id_t colId = pColNode->colId;
48,848✔
1986
  void* _tmp = taosArrayPush(pColIdArray, &colId);
48,848✔
1987
}
48,848✔
1988

1989
static int32_t buildTagCondKey(
24,424✔
1990
  const SNode* pTagCond, char** pTagCondKey,
1991
  int32_t* tagCondKeyLen, SArray** pTagColIds) {
1992
  if (NULL == pTagCond ||
24,424✔
1993
    (nodeType(pTagCond) != QUERY_NODE_OPERATOR &&
24,424✔
1994
      nodeType(pTagCond) != QUERY_NODE_LOGIC_CONDITION)) {
24,424✔
1995
    qError("invalid parameter to extract tag filter symbol");
×
1996
    return TSDB_CODE_INTERNAL_ERROR;
×
1997
  }
1998
  int32_t code = TSDB_CODE_SUCCESS;
24,424✔
1999
  int32_t lino = 0;
24,424✔
2000
  *pTagColIds = taosArrayInit(4, sizeof(col_id_t));
24,424✔
2001

2002
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
24,424✔
2003
    extractTagColId((SOperatorNode*)pTagCond, *pTagColIds);
×
2004
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
24,424✔
2005
    SNode* pChild = NULL;
24,424✔
2006
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
73,272✔
2007
      extractTagColId((SOperatorNode*)pChild, *pTagColIds);
48,848✔
2008
    }
2009
  }
2010

2011
  taosArraySort(*pTagColIds, compareUint16Val);
24,424✔
2012

2013
  // encode ordered colIds into key string, separated by ','
2014
  *tagCondKeyLen =
48,848✔
2015
    (int32_t)(taosArrayGetSize(*pTagColIds) * (sizeof(col_id_t) + 1) - 1);
24,424✔
2016
  *pTagCondKey = (char*)taosMemoryCalloc(1, *tagCondKeyLen);
24,424✔
2017
  TSDB_CHECK_NULL(*pTagCondKey, code, lino, _end, terrno);
24,424✔
2018
  char* pStart = *pTagCondKey;
24,424✔
2019
  for (int32_t i = 0; i < taosArrayGetSize(*pTagColIds); ++i) {
73,272✔
2020
    col_id_t* pColId = (col_id_t*)taosArrayGet(*pTagColIds, i);
48,848✔
2021
    TSDB_CHECK_NULL(pColId, code, lino, _end, terrno);
48,848✔
2022
    memcpy(pStart, pColId, sizeof(col_id_t));
48,848✔
2023
    pStart += sizeof(col_id_t);
48,848✔
2024
    if (i != taosArrayGetSize(*pTagColIds) - 1) {
48,848✔
2025
      *pStart = ',';
24,424✔
2026
      pStart += 1;
24,424✔
2027
    }
2028
  }
2029

2030
_end:
24,424✔
2031
  if (TSDB_CODE_SUCCESS != code) {
24,424✔
2032
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2033
    terrno = code;
×
2034
  }
2035
  return code;
24,424✔
2036
}
2037

2038
static EDealRes canOptimizeTagCondFilter(SNode* pTagCond, void* pContext) {
226,352✔
2039
  if (NULL == pTagCond) {
226,352✔
2040
    *(bool*)pContext = false;
×
2041
    return DEAL_RES_END;
×
2042
  }
2043
  if (nodeType(pTagCond) == QUERY_NODE_VALUE ||
226,352✔
2044
    nodeType(pTagCond) == QUERY_NODE_COLUMN) {
150,328✔
2045
    return DEAL_RES_CONTINUE;
124,872✔
2046
  }
2047
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR &&
101,480✔
2048
    ((SOperatorNode*)pTagCond)->opType == OP_TYPE_EQUAL) {
49,880✔
2049
    return DEAL_RES_CONTINUE;
48,848✔
2050
  }
2051
  if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION &&
52,632✔
2052
    ((SLogicConditionNode*)pTagCond)->condType == LOGIC_COND_TYPE_AND) {
24,424✔
2053
    return DEAL_RES_CONTINUE;
24,424✔
2054
  }
2055
  if (nodeType(pTagCond) == QUERY_NODE_FUNCTION &&
55,384✔
2056
    fmIsStreamPesudoColVal(((SFunctionNode*)pTagCond)->funcId)) {
27,176✔
2057
    return DEAL_RES_CONTINUE;
27,176✔
2058
  }
2059
  *(bool*)pContext = false;
1,032✔
2060
  return DEAL_RES_END;
1,032✔
2061
}
2062

2063
int32_t getTableList(void* pVnode, SScanPhysiNode* pScanNode, SNode* pTagCond, SNode* pTagIndexCond,
117,928,729✔
2064
                     STableListInfo* pListInfo, uint8_t* digest, const char* idstr, SStorageAPI* pStorageAPI, void* pStreamInfo) {
2065
  int32_t code = TSDB_CODE_SUCCESS;
117,928,729✔
2066
  int32_t lino = 0;
117,928,729✔
2067
  size_t  numOfTables = 0;
117,928,729✔
2068
  bool    listAdded = false;
117,928,729✔
2069

2070
  pListInfo->idInfo.suid = pScanNode->suid;
117,937,940✔
2071
  pListInfo->idInfo.tableType = pScanNode->tableType;
117,797,318✔
2072

2073
  SArray* pUidList = taosArrayInit(8, sizeof(uint64_t));
117,894,627✔
2074
  QUERY_CHECK_NULL(pUidList, code, lino, _error, terrno);
117,827,716✔
2075

2076
  SIdxFltStatus status = SFLT_NOT_INDEX;
117,827,716✔
2077
  char*   pTagCondKey = NULL;
117,850,488✔
2078
  int32_t tagCondKeyLen;
117,875,295✔
2079
  SArray* pTagColIds = NULL;
117,900,681✔
2080
  char*   pPayload = NULL;
117,911,581✔
2081
  qTrace("getTableList called, suid:%" PRIu64
117,911,581✔
2082
    ", tagCond:%p, tagIndexCond:%p, %d %d", pScanNode->suid, pTagCond,
2083
    pTagIndexCond, pScanNode->tableType, pScanNode->virtualStableScan);
2084
  if (pScanNode->tableType != TSDB_SUPER_TABLE && !pScanNode->virtualStableScan) {
117,911,581✔
2085
    pListInfo->idInfo.uid = pScanNode->uid;
51,920,351✔
2086
    if (pStorageAPI->metaFn.isTableExisted(pVnode, pScanNode->uid)) {
51,934,530✔
2087
      void* tmp = taosArrayPush(pUidList, &pScanNode->uid);
51,900,457✔
2088
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
51,899,532✔
2089
    }
2090
    code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, status, pStorageAPI, false, &listAdded, pStreamInfo);
51,951,804✔
2091
    QUERY_CHECK_CODE(code, lino, _end);
51,951,121✔
2092
  } else {
2093
    bool      isStream = (pStreamInfo != NULL);
65,952,496✔
2094
    bool      hasTagCond = (pTagCond != NULL);
65,952,496✔
2095
    bool      canCacheTagEqCondFilter = false;
65,952,496✔
2096
    T_MD5_CTX context = {0};
65,968,516✔
2097

2098
    qTrace("start to get table list by tag filter, suid:%" PRIu64
66,024,946✔
2099
      ",tsStableTagFilterCache:%d, tsTagFilterCache:%d", 
2100
      pScanNode->suid, tsStableTagFilterCache, tsTagFilterCache);
2101

2102
    bool acquired = false;
66,024,946✔
2103
    // first, check whether we can use stable tag filter cache
2104
    if (tsStableTagFilterCache && isStream && hasTagCond) {
65,971,789✔
2105
      canCacheTagEqCondFilter = true;
25,456✔
2106
      nodesWalkExpr(pTagCond, canOptimizeTagCondFilter,
25,456✔
2107
        (void*)&canCacheTagEqCondFilter);
2108
    }
2109
    if (canCacheTagEqCondFilter) {
65,821,333✔
2110
      qDebug("%s, stable tag filter condition can be optimized", idstr);
24,424✔
2111
      if (((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
24,424✔
2112
        SNode* tmp = NULL;
24,424✔
2113
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
24,424✔
2114
        QUERY_CHECK_CODE(code, lino, _error);
24,424✔
2115

2116
        PlaceHolderContext ctx = {.code = TSDB_CODE_SUCCESS, .pStreamRuntimeInfo = (SStreamRuntimeFuncInfo*)pStreamInfo};
24,424✔
2117
        nodesRewriteExpr(&tmp, replacePlaceHolderColumn, (void*)&ctx);
24,424✔
2118
        if (TSDB_CODE_SUCCESS != ctx.code) {
24,424✔
2119
          nodesDestroyNode(tmp);
×
2120
          code = ctx.code;
×
2121
          goto _error;
×
2122
        }
2123
        code = genStableTagFilterDigest(tmp, &context);
24,424✔
2124
        nodesDestroyNode(tmp);
24,424✔
2125
      } else {
2126
        code = genStableTagFilterDigest(pTagCond, &context);
×
2127
      }
2128
      QUERY_CHECK_CODE(code, lino, _error);
24,424✔
2129

2130
      code = buildTagCondKey(
24,424✔
2131
        pTagCond, &pTagCondKey, &tagCondKeyLen, &pTagColIds);
2132
      QUERY_CHECK_CODE(code, lino, _error);
24,424✔
2133
      code = pStorageAPI->metaFn.getStableCachedTableList(
24,424✔
2134
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
24,424✔
2135
        context.digest, tListLen(context.digest), pUidList, &acquired);
2136
      QUERY_CHECK_CODE(code, lino, _error);
24,424✔
2137
    } else if (tsTagFilterCache) {
65,796,909✔
2138
      // second, try to use normal tag filter cache
2139
      qDebug("%s using normal tag filter cache", idstr);
64,700✔
2140
      if (pStreamInfo != NULL && ((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
67,040✔
2141
        SNode* tmp = NULL;
2,340✔
2142
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
2,340✔
2143
        QUERY_CHECK_CODE(code, lino, _error);
2,340✔
2144

2145
        PlaceHolderContext ctx = {.code = TSDB_CODE_SUCCESS, .pStreamRuntimeInfo = (SStreamRuntimeFuncInfo*)pStreamInfo};
2,340✔
2146
        nodesRewriteExpr(&tmp, replacePlaceHolderColumn, (void*)&ctx);
2,340✔
2147
        if (TSDB_CODE_SUCCESS != ctx.code) {
2,340✔
2148
          nodesDestroyNode(tmp);
×
2149
          code = ctx.code;
×
2150
          goto _error;
×
2151
        }
2152
        code = genTagFilterDigest(tmp, &context);
2,340✔
2153
        nodesDestroyNode(tmp);
2,340✔
2154
      } else {
2155
        code = genTagFilterDigest(pTagCond, &context);
62,360✔
2156
      }
2157
      // try to retrieve the result from meta cache
2158
      QUERY_CHECK_CODE(code, lino, _error);      
64,700✔
2159
      code = pStorageAPI->metaFn.getCachedTableList(
64,700✔
2160
        pVnode, pScanNode->suid, context.digest,
64,700✔
2161
        tListLen(context.digest), pUidList, &acquired);
2162
      QUERY_CHECK_CODE(code, lino, _error);
105,940✔
2163
    }
2164
    if (acquired) {
65,862,573✔
2165
      taosArrayDestroy(pTagColIds);
64,560✔
2166
      pTagColIds = NULL;
64,560✔
2167
      
2168
      digest[0] = 1;
64,560✔
2169
      memcpy(
129,120✔
2170
        digest + 1, context.digest, tListLen(context.digest));
64,560✔
2171
      qDebug("suid:%" PRIu64 ", %s retrieve table uid list from cache,"
64,560✔
2172
        " numOfTables:%d", 
2173
        pScanNode->suid, idstr, (int32_t)taosArrayGetSize(pUidList));
2174
      goto _end;
64,560✔
2175
    } else {
2176
      qDebug("suid:%" PRIu64 
65,798,013✔
2177
        ", failed to get table uid list from cache", pScanNode->suid);
2178
    }
2179

2180
    if (!pTagCond) {  // no tag filter condition exists, let's fetch all tables of this super table
65,966,684✔
2181
      code = pStorageAPI->metaFn.getChildTableList(pVnode, pScanNode->suid, pUidList);
53,492,781✔
2182
      QUERY_CHECK_CODE(code, lino, _error);
53,480,150✔
2183
      qTrace("no tag filter, get all child tables, numOfTables:%d", (int32_t)taosArrayGetSize(pUidList));
53,480,150✔
2184
    } else {
2185
      // failed to find the result in the cache, let try to calculate the results
2186
      if (pTagIndexCond) {
12,473,903✔
2187
        void* pIndex = pStorageAPI->metaFn.getInvertIndex(pVnode);
4,360,902✔
2188

2189
        SIndexMetaArg metaArg = {.metaEx = pVnode,
4,361,078✔
2190
                                 .idx = pStorageAPI->metaFn.storeGetIndexInfo(pVnode),
4,361,014✔
2191
                                 .ivtIdx = pIndex,
2192
                                 .suid = pScanNode->uid};
4,361,014✔
2193

2194
        status = SFLT_NOT_INDEX;
4,361,014✔
2195
        code = doFilterTag(pTagIndexCond, &metaArg, pUidList, &status, &pStorageAPI->metaFilter);
4,361,014✔
2196
        if (code != 0 || status == SFLT_NOT_INDEX) {  // temporarily disable it for performance sake
4,348,589✔
2197
          qDebug("failed to get tableIds from index, suid:%" PRIu64 ", uidListSize:%d", pScanNode->uid, (int32_t)taosArrayGetSize(pUidList));
1,057,917✔
2198
        } else {
2199
          qDebug("succ to get filter result, table num: %d", (int)taosArrayGetSize(pUidList));
3,290,672✔
2200
        }
2201
      }
2202
    }
2203
    qTrace("after index filter, pTagCond:%p uidListSize:%d", pTagCond, (int32_t)taosArrayGetSize(pUidList));
65,945,980✔
2204
    code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, status,
65,966,622✔
2205
      pStorageAPI, tsTagFilterCache || tsStableTagFilterCache,
65,966,622✔
2206
      &listAdded, pStreamInfo);
2207
    QUERY_CHECK_CODE(code, lino, _error);
65,947,416✔
2208

2209
    // let's add the filter results into meta-cache
2210
    numOfTables = taosArrayGetSize(pUidList);
65,946,320✔
2211

2212
    if (canCacheTagEqCondFilter) {
65,938,318✔
2213
      qInfo("%s, suid:%" PRIu64 ", add uid list to stable tag filter cache, "
9,632✔
2214
            "uidListSize:%d, origin key:%" PRIu64 ",%" PRIu64,
2215
            idstr, pScanNode->suid, (int32_t)numOfTables,
2216
            *(uint64_t*)context.digest, *(uint64_t*)(context.digest + 8));
2217

2218
      code = pStorageAPI->metaFn.putStableCachedTableList(
9,632✔
2219
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
2220
        context.digest, tListLen(context.digest),
2221
        pUidList, &pTagColIds);
2222
      QUERY_CHECK_CODE(code, lino, _end);
9,632✔
2223

2224
      digest[0] = 1;
9,632✔
2225
      memcpy(digest + 1, context.digest, tListLen(context.digest));
9,632✔
2226
    } else if (tsTagFilterCache) {
65,928,686✔
2227
      qInfo("%s, suid:%" PRIu64 ", add uid list to normal tag filter cache, "
14,932✔
2228
            "uidListSize:%d, origin key:%" PRIu64 ",%" PRIu64,
2229
            idstr, pScanNode->suid, (int32_t)numOfTables,
2230
            *(uint64_t*)context.digest, *(uint64_t*)(context.digest + 8));
2231
      size_t size = numOfTables * sizeof(uint64_t) + sizeof(int32_t);
14,932✔
2232
      pPayload = taosMemoryMalloc(size);
14,932✔
2233
      QUERY_CHECK_NULL(pPayload, code, lino, _end, terrno);
14,932✔
2234

2235
      *(int32_t*)pPayload = (int32_t)numOfTables;
14,932✔
2236
      if (numOfTables > 0) {
14,932✔
2237
        void* tmp = taosArrayGet(pUidList, 0);
12,340✔
2238
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
12,340✔
2239
        memcpy(pPayload + sizeof(int32_t), tmp, numOfTables * sizeof(uint64_t));
12,340✔
2240
      }
2241

2242
      code = pStorageAPI->metaFn.putCachedTableList(pVnode, pScanNode->suid,
14,932✔
2243
                                                    context.digest,
2244
                                                    tListLen(context.digest),
2245
                                                    pPayload, size, 1);
2246
      if (TSDB_CODE_SUCCESS == code) {
14,932✔
2247
        /*
2248
          data referenced by pPayload is used in lru cache,
2249
          reset pPayload to NULL to avoid being freed in _error block
2250
        */
2251
        pPayload = NULL;
14,932✔
2252
      } else {
2253
        if (TSDB_CODE_DUP_KEY == code) {
×
2254
          /*
2255
            another thread has already put the same key into cache,
2256
            we can just ignore this error
2257
          */
2258
          code = TSDB_CODE_SUCCESS;
×
2259
        }
2260
        QUERY_CHECK_CODE(code, lino, _end);
×
2261
      }
2262

2263

2264
      digest[0] = 1;
14,932✔
2265
      memcpy(digest + 1, context.digest, tListLen(context.digest));
14,932✔
2266
    }
2267
  }
2268

2269
_end:
117,939,172✔
2270
  if (!listAdded) {
117,981,707✔
2271
    numOfTables = taosArrayGetSize(pUidList);
106,984,124✔
2272
    for (int i = 0; i < numOfTables; i++) {
415,941,252✔
2273
      void* tmp = taosArrayGet(pUidList, i);
308,970,822✔
2274
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
308,960,804✔
2275
      STableKeyInfo info = {.uid = *(uint64_t*)tmp, .groupId = 0};
308,960,804✔
2276

2277
      void* p = taosArrayPush(pListInfo->pTableList, &info);
308,911,421✔
2278
      if (p == NULL) {
308,993,262✔
2279
        taosArrayDestroy(pUidList);
×
2280
        return terrno;
×
2281
      }
2282

2283
      qTrace("tagfilter get uid:%" PRIu64 ", %s", info.uid, idstr);
308,993,262✔
2284
    }
2285
  }
2286

2287
  qDebug("%s, table list with %d uids built", idstr, (int32_t)numOfTables);
117,968,013✔
2288

2289
_error:
117,981,946✔
2290
  taosArrayDestroy(pUidList);
117,991,700✔
2291
  taosArrayDestroy(pTagColIds);
117,986,423✔
2292
  taosMemFreeClear(pTagCondKey);
117,985,722✔
2293
  taosMemFreeClear(pPayload);
117,985,722✔
2294
  if (code != TSDB_CODE_SUCCESS) {
117,985,722✔
2295
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1,096✔
2296
  }
2297
  return code;
117,976,618✔
2298
}
2299

2300
int32_t qGetTableList(int64_t suid, void* pVnode, void* node, SArray** tableList, void* pTaskInfo) {
4,973✔
2301
  int32_t        code = TSDB_CODE_SUCCESS;
4,973✔
2302
  int32_t        lino = 0;
4,973✔
2303
  SSubplan*      pSubplan = (SSubplan*)node;
4,973✔
2304
  SScanPhysiNode pNode = {0};
4,973✔
2305
  pNode.suid = suid;
4,973✔
2306
  pNode.uid = suid;
4,973✔
2307
  pNode.tableType = TSDB_SUPER_TABLE;
4,973✔
2308

2309
  STableListInfo* pTableListInfo = tableListCreate();
4,973✔
2310
  QUERY_CHECK_NULL(pTableListInfo, code, lino, _end, terrno);
4,973✔
2311
  uint8_t digest[17] = {0};
4,973✔
2312
  code = getTableList(pVnode, &pNode, pSubplan ? pSubplan->pTagCond : NULL, pSubplan ? pSubplan->pTagIndexCond : NULL,
4,973✔
2313
                      pTableListInfo, digest, "qGetTableList", &((SExecTaskInfo*)pTaskInfo)->storageAPI, NULL);
2314
  QUERY_CHECK_CODE(code, lino, _end);
4,973✔
2315
  *tableList = pTableListInfo->pTableList;
4,973✔
2316
  pTableListInfo->pTableList = NULL;
4,973✔
2317
  tableListDestroy(pTableListInfo);
4,973✔
2318

2319
_end:
4,973✔
2320
  if (code != TSDB_CODE_SUCCESS) {
4,973✔
2321
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2322
  }
2323
  return code;
4,973✔
2324
}
2325

2326
size_t getTableTagsBufLen(const SNodeList* pGroups) {
×
2327
  size_t keyLen = 0;
×
2328

2329
  SNode* node;
2330
  FOREACH(node, pGroups) {
×
2331
    SExprNode* pExpr = (SExprNode*)node;
×
2332
    keyLen += pExpr->resType.bytes;
×
2333
  }
2334

2335
  keyLen += sizeof(int8_t) * LIST_LENGTH(pGroups);
×
2336
  return keyLen;
×
2337
}
2338

2339
int32_t getGroupIdFromTagsVal(void* pVnode, uint64_t uid, SNodeList* pGroupNode, char* keyBuf, uint64_t* pGroupId,
×
2340
                              SStorageAPI* pAPI) {
2341
  SMetaReader mr = {0};
×
2342

2343
  pAPI->metaReaderFn.initReader(&mr, pVnode, META_READER_LOCK, &pAPI->metaFn);
×
2344
  if (pAPI->metaReaderFn.getEntryGetUidCache(&mr, uid) != 0) {  // table not exist
×
2345
    pAPI->metaReaderFn.clearReader(&mr);
×
2346
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
2347
  }
2348

2349
  SNodeList* groupNew = NULL;
×
2350
  int32_t    code = nodesCloneList(pGroupNode, &groupNew);
×
2351
  if (TSDB_CODE_SUCCESS != code) {
×
2352
    pAPI->metaReaderFn.clearReader(&mr);
×
2353
    return code;
×
2354
  }
2355

2356
  STransTagExprCtx ctx = {.code = 0, .pReader = &mr};
×
2357
  nodesRewriteExprsPostOrder(groupNew, doTranslateTagExpr, &ctx);
×
2358
  if (TSDB_CODE_SUCCESS != ctx.code) {
×
2359
    nodesDestroyList(groupNew);
×
2360
    pAPI->metaReaderFn.clearReader(&mr);
×
2361
    return code;
×
2362
  }
2363
  char* isNull = (char*)keyBuf;
×
2364
  char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(pGroupNode);
×
2365

2366
  SNode*  pNode;
2367
  int32_t index = 0;
×
2368
  FOREACH(pNode, groupNew) {
×
2369
    SNode*  pNew = NULL;
×
2370
    int32_t code = scalarCalculateConstants(pNode, &pNew);
×
2371
    if (TSDB_CODE_SUCCESS == code) {
×
2372
      REPLACE_NODE(pNew);
×
2373
    } else {
2374
      nodesDestroyList(groupNew);
×
2375
      pAPI->metaReaderFn.clearReader(&mr);
×
2376
      return code;
×
2377
    }
2378

2379
    if (nodeType(pNew) != QUERY_NODE_VALUE) {
×
2380
      nodesDestroyList(groupNew);
×
2381
      pAPI->metaReaderFn.clearReader(&mr);
×
2382
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2383
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2384
    }
2385
    SValueNode* pValue = (SValueNode*)pNew;
×
2386

2387
    if (pValue->node.resType.type == TSDB_DATA_TYPE_NULL || pValue->isNull) {
×
2388
      isNull[index++] = 1;
×
2389
      continue;
×
2390
    } else {
2391
      isNull[index++] = 0;
×
2392
      char* data = nodesGetValueFromNode(pValue);
×
2393
      if (pValue->node.resType.type == TSDB_DATA_TYPE_JSON) {
×
2394
        if (tTagIsJson(data)) {
×
2395
          terrno = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
×
2396
          nodesDestroyList(groupNew);
×
2397
          pAPI->metaReaderFn.clearReader(&mr);
×
2398
          return terrno;
×
2399
        }
2400
        int32_t len = getJsonValueLen(data);
×
2401
        memcpy(pStart, data, len);
×
2402
        pStart += len;
×
2403
      } else if (IS_VAR_DATA_TYPE(pValue->node.resType.type)) {
×
2404
        if (IS_STR_DATA_BLOB(pValue->node.resType.type)) {
×
2405
          return TSDB_CODE_BLOB_NOT_SUPPORT_TAG;
×
2406
        }
2407
        memcpy(pStart, data, varDataTLen(data));
×
2408
        pStart += varDataTLen(data);
×
2409
      } else {
2410
        memcpy(pStart, data, pValue->node.resType.bytes);
×
2411
        pStart += pValue->node.resType.bytes;
×
2412
      }
2413
    }
2414
  }
2415

2416
  int32_t len = (int32_t)(pStart - (char*)keyBuf);
×
2417
  *pGroupId = calcGroupId(keyBuf, len);
×
2418

2419
  nodesDestroyList(groupNew);
×
2420
  pAPI->metaReaderFn.clearReader(&mr);
×
2421

2422
  return TSDB_CODE_SUCCESS;
×
2423
}
2424

2425
SArray* makeColumnArrayFromList(SNodeList* pNodeList) {
1,739,217✔
2426
  if (!pNodeList) {
1,739,217✔
2427
    return NULL;
×
2428
  }
2429

2430
  size_t  numOfCols = LIST_LENGTH(pNodeList);
1,739,217✔
2431
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
1,738,762✔
2432
  if (pList == NULL) {
1,737,620✔
2433
    return NULL;
×
2434
  }
2435

2436
  for (int32_t i = 0; i < numOfCols; ++i) {
3,906,144✔
2437
    SColumnNode* pColNode = (SColumnNode*)nodesListGetNode(pNodeList, i);
2,169,888✔
2438
    if (!pColNode) {
2,170,330✔
2439
      taosArrayDestroy(pList);
×
2440
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2441
      return NULL;
×
2442
    }
2443

2444
    // todo extract method
2445
    SColumn c = {0};
2,170,330✔
2446
    c.slotId = pColNode->slotId;
2,169,901✔
2447
    c.colId = pColNode->colId;
2,169,459✔
2448
    c.type = pColNode->node.resType.type;
2,169,446✔
2449
    c.bytes = pColNode->node.resType.bytes;
2,167,614✔
2450
    c.precision = pColNode->node.resType.precision;
2,168,524✔
2451
    c.scale = pColNode->node.resType.scale;
2,168,966✔
2452

2453
    void* tmp = taosArrayPush(pList, &c);
2,170,760✔
2454
    if (!tmp) {
2,170,760✔
2455
      taosArrayDestroy(pList);
×
2456
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2457
      return NULL;
×
2458
    }
2459
  }
2460

2461
  return pList;
1,736,256✔
2462
}
2463

2464
int32_t extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols,
145,426,692✔
2465
                            int32_t type, SColMatchInfo* pMatchInfo) {
2466
  size_t  numOfCols = LIST_LENGTH(pNodeList);
145,426,692✔
2467
  int32_t code = TSDB_CODE_SUCCESS;
145,423,844✔
2468
  int32_t lino = 0;
145,423,844✔
2469

2470
  pMatchInfo->matchType = type;
145,423,844✔
2471

2472
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchItem));
145,422,522✔
2473
  if (pList == NULL) {
145,416,302✔
2474
    code = terrno;
×
2475
    return code;
×
2476
  }
2477

2478
  for (int32_t i = 0; i < numOfCols; ++i) {
888,350,207✔
2479
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
742,911,243✔
2480
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
742,870,297✔
2481
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
742,870,297✔
2482
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
738,674,291✔
2483

2484
      SColMatchItem c = {.needOutput = true};
738,663,152✔
2485
      c.colId = pColNode->colId;
738,693,040✔
2486
      c.srcSlotId = pColNode->slotId;
738,678,146✔
2487
      c.dstSlotId = pNode->slotId;
738,679,805✔
2488
      c.isPk = pColNode->isPk;
738,726,303✔
2489
      c.dataType = pColNode->node.resType;
738,711,316✔
2490
      void* tmp = taosArrayPush(pList, &c);
738,709,663✔
2491
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
738,709,663✔
2492
    }
2493
  }
2494

2495
  // set the output flag for each column in SColMatchInfo, according to the
2496
  *numOfOutputCols = 0;
145,438,964✔
2497
  int32_t num = LIST_LENGTH(pOutputNodeList->pSlots);
145,439,392✔
2498
  for (int32_t i = 0; i < num; ++i) {
968,501,307✔
2499
    SSlotDescNode* pNode = (SSlotDescNode*)nodesListGetNode(pOutputNodeList->pSlots, i);
823,065,240✔
2500
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
823,039,407✔
2501

2502
    // todo: add reserve flag check
2503
    // it is a column reserved for the arithmetic expression calculation
2504
    if (pNode->slotId >= numOfCols) {
823,039,407✔
2505
      (*numOfOutputCols) += 1;
80,149,198✔
2506
      continue;
80,150,074✔
2507
    }
2508

2509
    SColMatchItem* info = NULL;
742,910,527✔
2510
    for (int32_t j = 0; j < taosArrayGetSize(pList); ++j) {
2,147,483,647✔
2511
      info = taosArrayGet(pList, j);
2,147,483,647✔
2512
      QUERY_CHECK_NULL(info, code, lino, _end, terrno);
2,147,483,647✔
2513
      if (info->dstSlotId == pNode->slotId) {
2,147,483,647✔
2514
        break;
737,891,322✔
2515
      }
2516
    }
2517

2518
    if (pNode->output) {
13,011,677✔
2519
      (*numOfOutputCols) += 1;
737,326,243✔
2520
    } else if (info != NULL) {
5,573,087✔
2521
      // select distinct tbname from stb where tbname='abc';
2522
      info->needOutput = false;
5,570,487✔
2523
    }
2524
  }
2525

2526
  pMatchInfo->pList = pList;
145,436,067✔
2527

2528
_end:
145,422,461✔
2529
  if (code != TSDB_CODE_SUCCESS) {
145,422,461✔
2530
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2531
  }
2532
  return code;
145,413,405✔
2533
}
2534

2535
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
633,301,407✔
2536
                                  const char* name) {
2537
  SResSchema s = {0};
633,301,407✔
2538
  s.scale = scale;
633,312,496✔
2539
  s.type = type;
633,312,496✔
2540
  s.bytes = bytes;
633,312,496✔
2541
  s.slotId = slotId;
633,312,496✔
2542
  s.precision = precision;
633,312,496✔
2543
  tstrncpy(s.name, name, tListLen(s.name));
633,312,496✔
2544

2545
  return s;
633,312,496✔
2546
}
2547

2548
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType, EColumnType colType) {
606,160,884✔
2549
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
606,160,884✔
2550
  if (pCol == NULL) {
605,991,572✔
2551
    return NULL;
×
2552
  }
2553

2554
  pCol->slotId = slotId;
605,991,572✔
2555
  pCol->colId = colId;
606,001,517✔
2556
  pCol->bytes = pType->bytes;
606,048,327✔
2557
  pCol->type = pType->type;
606,078,554✔
2558
  pCol->scale = pType->scale;
606,061,655✔
2559
  pCol->precision = pType->precision;
606,105,315✔
2560
  pCol->dataBlockId = blockId;
606,135,654✔
2561
  pCol->colType = colType;
606,144,083✔
2562
  return pCol;
606,107,322✔
2563
}
2564

2565
int32_t createExprFromOneNode(SExprInfo* pExp, SNode* pNode, int16_t slotId) {
633,308,650✔
2566
  int32_t code = TSDB_CODE_SUCCESS;
633,308,650✔
2567
  int32_t lino = 0;
633,308,650✔
2568
  pExp->base.numOfParams = 0;
633,308,650✔
2569
  pExp->base.pParam = NULL;
633,329,227✔
2570
  pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
633,296,154✔
2571
  QUERY_CHECK_NULL(pExp->pExpr, code, lino, _end, terrno);
633,046,864✔
2572

2573
  pExp->pExpr->_function.num = 1;
633,132,970✔
2574
  pExp->pExpr->_function.functionId = -1;
633,153,971✔
2575

2576
  int32_t type = nodeType(pNode);
633,179,381✔
2577
  // it is a project query, or group by column
2578
  if (type == QUERY_NODE_COLUMN) {
633,249,617✔
2579
    pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
412,860,097✔
2580
    SColumnNode* pColNode = (SColumnNode*)pNode;
412,863,548✔
2581

2582
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
412,863,548✔
2583
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
412,835,285✔
2584

2585
    pExp->base.numOfParams = 1;
412,824,241✔
2586

2587
    SDataType* pType = &pColNode->node.resType;
412,815,881✔
2588
    pExp->base.resSchema =
2589
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pColNode->colName);
412,818,547✔
2590

2591
    pExp->base.pParam[0].pCol =
825,652,110✔
2592
        createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType, pColNode->colType);
825,652,027✔
2593
    QUERY_CHECK_NULL(pExp->base.pParam[0].pCol, code, lino, _end, terrno);
412,830,767✔
2594

2595
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
412,795,893✔
2596
  } else if (type == QUERY_NODE_VALUE) {
220,389,520✔
2597
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
7,768,924✔
2598
    SValueNode* pValNode = (SValueNode*)pNode;
7,770,747✔
2599

2600
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
7,770,747✔
2601
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
7,770,616✔
2602

2603
    pExp->base.numOfParams = 1;
7,771,590✔
2604

2605
    SDataType* pType = &pValNode->node.resType;
7,770,230✔
2606
    pExp->base.resSchema =
2607
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
7,771,932✔
2608
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
7,770,545✔
2609
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
7,774,705✔
2610
    QUERY_CHECK_CODE(code, lino, _end);
7,773,361✔
2611
  } else if (type == QUERY_NODE_FUNCTION) {
212,620,596✔
2612
    pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
200,333,924✔
2613
    SFunctionNode* pFuncNode = (SFunctionNode*)pNode;
200,349,931✔
2614

2615
    SDataType* pType = &pFuncNode->node.resType;
200,349,931✔
2616
    pExp->base.resSchema =
2617
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pFuncNode->node.aliasName);
200,334,586✔
2618
    tExprNode* pExprNode = pExp->pExpr;
200,331,167✔
2619

2620
    pExprNode->_function.functionId = pFuncNode->funcId;
200,348,034✔
2621
    pExprNode->_function.pFunctNode = pFuncNode;
200,336,422✔
2622
    pExprNode->_function.functionType = pFuncNode->funcType;
200,309,488✔
2623

2624
    tstrncpy(pExprNode->_function.functionName, pFuncNode->functionName, tListLen(pExprNode->_function.functionName));
200,330,204✔
2625

2626
    pExp->base.pParamList = pFuncNode->pParameterList;
200,357,820✔
2627
#if 1
2628
    // todo refactor: add the parameter for tbname function
2629
    const char* name = "tbname";
200,326,791✔
2630
    int32_t     len = strlen(name);
200,326,791✔
2631

2632
    if (!pFuncNode->pParameterList && (memcmp(pExprNode->_function.functionName, name, len) == 0) &&
200,326,791✔
2633
        pExprNode->_function.functionName[len] == 0) {
8,179,273✔
2634
      pFuncNode->pParameterList = NULL;
8,174,006✔
2635
      int32_t     code = nodesMakeList(&pFuncNode->pParameterList);
8,174,883✔
2636
      SValueNode* res = NULL;
8,181,536✔
2637
      if (TSDB_CODE_SUCCESS == code) {
8,181,536✔
2638
        code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
8,181,261✔
2639
      }
2640
      QUERY_CHECK_CODE(code, lino, _end);
8,183,084✔
2641
      res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
8,183,084✔
2642
      code = nodesListAppend(pFuncNode->pParameterList, (SNode*)res);
8,179,982✔
2643
      if (code != TSDB_CODE_SUCCESS) {
8,182,275✔
2644
        nodesDestroyNode((SNode*)res);
×
2645
        res = NULL;
×
2646
      }
2647
      QUERY_CHECK_CODE(code, lino, _end);
8,182,275✔
2648
    }
2649
#endif
2650

2651
    int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
200,298,618✔
2652

2653
    pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
200,349,205✔
2654
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
200,313,502✔
2655
    pExp->base.numOfParams = numOfParam;
200,323,560✔
2656

2657
    for (int32_t j = 0; j < numOfParam && TSDB_CODE_SUCCESS == code; ++j) {
515,549,068✔
2658
      SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
315,206,294✔
2659
      QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
315,208,481✔
2660
      if (p1->type == QUERY_NODE_COLUMN) {
315,208,481✔
2661
        SColumnNode* pcn = (SColumnNode*)p1;
193,307,736✔
2662

2663
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
193,307,736✔
2664
        pExp->base.pParam[j].pCol =
386,564,132✔
2665
            createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType, pcn->colType);
386,569,712✔
2666
        QUERY_CHECK_NULL(pExp->base.pParam[j].pCol, code, lino, _end, terrno);
193,287,944✔
2667
      } else if (p1->type == QUERY_NODE_VALUE) {
121,893,415✔
2668
        SValueNode* pvn = (SValueNode*)p1;
62,574,510✔
2669
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
62,574,510✔
2670
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
62,568,276✔
2671
        QUERY_CHECK_CODE(code, lino, _end);
62,571,919✔
2672
      }
2673
    }
2674
    pExp->pExpr->_function.bindExprID = ((SExprNode*)pNode)->bindExprID;
200,342,774✔
2675
  } else if (type == QUERY_NODE_OPERATOR) {
12,286,672✔
2676
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
11,152,262✔
2677
    SOperatorNode* pOpNode = (SOperatorNode*)pNode;
11,152,402✔
2678

2679
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
11,152,402✔
2680
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
11,148,336✔
2681
    pExp->base.numOfParams = 1;
11,146,305✔
2682

2683
    SDataType* pType = &pOpNode->node.resType;
11,147,345✔
2684
    pExp->base.resSchema =
2685
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pOpNode->node.aliasName);
11,147,867✔
2686
    pExp->pExpr->_optrRoot.pRootNode = pNode;
11,148,642✔
2687
  } else if (type == QUERY_NODE_CASE_WHEN) {
1,134,420✔
2688
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
1,137,305✔
2689
    SCaseWhenNode* pCaseNode = (SCaseWhenNode*)pNode;
1,137,305✔
2690

2691
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
1,137,305✔
2692
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
1,137,305✔
2693
    pExp->base.numOfParams = 1;
1,137,305✔
2694

2695
    SDataType* pType = &pCaseNode->node.resType;
1,137,305✔
2696
    pExp->base.resSchema =
2697
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCaseNode->node.aliasName);
1,137,305✔
2698
    pExp->pExpr->_optrRoot.pRootNode = pNode;
1,137,305✔
2699
  } else if (type == QUERY_NODE_LOGIC_CONDITION) {
114✔
2700
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
1,156✔
2701
    SLogicConditionNode* pCond = (SLogicConditionNode*)pNode;
1,156✔
2702
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
1,156✔
2703
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
1,156✔
2704
    pExp->base.numOfParams = 1;
1,156✔
2705
    SDataType* pType = &pCond->node.resType;
1,156✔
2706
    pExp->base.resSchema =
2707
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCond->node.aliasName);
1,156✔
2708
    pExp->pExpr->_optrRoot.pRootNode = pNode;
1,156✔
2709
  } else {
2710
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
2✔
2711
    QUERY_CHECK_CODE(code, lino, _end);
2✔
2712
  }
2713
  pExp->pExpr->relatedTo = ((SExprNode*)pNode)->relatedTo;
633,261,643✔
2714
_end:
633,244,215✔
2715
  if (code != TSDB_CODE_SUCCESS) {
633,244,215✔
2716
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2717
  }
2718
  return code;
633,173,813✔
2719
}
2720

2721
int32_t createExprFromTargetNode(SExprInfo* pExp, STargetNode* pTargetNode) {
633,246,294✔
2722
  return createExprFromOneNode(pExp, pTargetNode->pExpr, pTargetNode->slotId);
633,246,294✔
2723
}
2724

2725
SExprInfo* createExpr(SNodeList* pNodeList, int32_t* numOfExprs) {
×
2726
  *numOfExprs = LIST_LENGTH(pNodeList);
×
2727
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
×
2728
  if (!pExprs) {
×
2729
    return NULL;
×
2730
  }
2731

2732
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
×
2733
    SExprInfo* pExp = &pExprs[i];
×
2734
    int32_t    code = createExprFromOneNode(pExp, nodesListGetNode(pNodeList, i), i + UD_TAG_COLUMN_INDEX);
×
2735
    if (code != TSDB_CODE_SUCCESS) {
×
2736
      taosMemoryFreeClear(pExprs);
×
2737
      terrno = code;
×
2738
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2739
      return NULL;
×
2740
    }
2741
  }
2742

2743
  return pExprs;
×
2744
}
2745

2746
int32_t createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, SExprInfo** pExprInfo, int32_t* numOfExprs) {
202,993,659✔
2747
  QRY_PARAM_CHECK(pExprInfo);
202,993,659✔
2748

2749
  int32_t code = 0;
203,012,456✔
2750
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
203,012,456✔
2751
  int32_t numOfGroupKeys = 0;
202,977,259✔
2752
  if (pGroupKeys != NULL) {
202,977,259✔
2753
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
18,690,751✔
2754
  }
2755

2756
  *numOfExprs = numOfFuncs + numOfGroupKeys;
202,979,156✔
2757
  if (*numOfExprs == 0) {
202,940,450✔
2758
    return code;
18,641,748✔
2759
  }
2760

2761
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
184,344,915✔
2762
  if (pExprs == NULL) {
184,272,738✔
2763
    return terrno;
×
2764
  }
2765

2766
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
817,341,908✔
2767
    STargetNode* pTargetNode = NULL;
633,082,461✔
2768
    if (i < numOfFuncs) {
633,082,461✔
2769
      pTargetNode = (STargetNode*)nodesListGetNode(pNodeList, i);
606,236,266✔
2770
    } else {
2771
      pTargetNode = (STargetNode*)nodesListGetNode(pGroupKeys, i - numOfFuncs);
26,846,195✔
2772
    }
2773
    if (!pTargetNode) {
633,129,529✔
2774
      destroyExprInfo(pExprs, *numOfExprs);
×
2775
      taosMemoryFreeClear(pExprs);
×
2776
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2777
      return terrno;
×
2778
    }
2779

2780
    SExprInfo* pExp = &pExprs[i];
633,129,529✔
2781
    code = createExprFromTargetNode(pExp, pTargetNode);
633,141,082✔
2782
    if (code != TSDB_CODE_SUCCESS) {
633,069,170✔
2783
      destroyExprInfo(pExprs, *numOfExprs);
×
2784
      taosMemoryFreeClear(pExprs);
×
2785
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2786
      return code;
×
2787
    }
2788
  }
2789

2790
  *pExprInfo = pExprs;
184,350,103✔
2791
  return code;
184,345,303✔
2792
}
2793

2794
static void deleteSubsidiareCtx(void* pData) {
×
2795
  SSubsidiaryResInfo* pCtx = (SSubsidiaryResInfo*)pData;
×
2796
  if (pCtx->pCtx) {
×
2797
    taosMemoryFreeClear(pCtx->pCtx);
×
2798
  }
2799
}
×
2800

2801
// set the output buffer for the selectivity + tag query
2802
static int32_t setSelectValueColumnInfo(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
192,935,026✔
2803
  int32_t num = 0;
192,935,026✔
2804
  int32_t code = TSDB_CODE_SUCCESS;
192,935,026✔
2805
  int32_t lino = 0;
192,935,026✔
2806

2807
  SArray* pValCtxArray = NULL;
192,935,026✔
2808
  for (int32_t i = numOfOutput - 1; i > 0; --i) {  // select Func is at the end of the list
637,415,274✔
2809
    int32_t funcIdx = pCtx[i].pExpr->pExpr->_function.bindExprID;
444,501,268✔
2810
    if (funcIdx > 0) {
444,508,143✔
2811
      if (pValCtxArray == NULL) {
1,797,720✔
2812
        // the end of the list is the select function of biggest index
2813
        pValCtxArray = taosArrayInit_s(sizeof(SSubsidiaryResInfo*), funcIdx);
1,290,101✔
2814
        if (pValCtxArray == NULL) {
1,287,149✔
2815
          return terrno;
×
2816
        }
2817
      }
2818
      if (funcIdx > pValCtxArray->size) {
1,794,768✔
2819
        qError("funcIdx:%d is out of range", funcIdx);
×
2820
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2821
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2822
      }
2823
      SSubsidiaryResInfo* pSubsidiary = &pCtx[i].subsidiaries;
1,795,752✔
2824
      pSubsidiary->pCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
1,796,736✔
2825
      if (pSubsidiary->pCtx == NULL) {
1,797,720✔
2826
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2827
        return terrno;
×
2828
      }
2829
      pSubsidiary->num = 0;
1,796,736✔
2830
      taosArraySet(pValCtxArray, funcIdx - 1, &pSubsidiary);
1,797,228✔
2831
    }
2832
  }
2833

2834
  SqlFunctionCtx*  p = NULL;
192,914,006✔
2835
  SqlFunctionCtx** pValCtx = NULL;
192,914,006✔
2836
  if (pValCtxArray == NULL) {
192,914,006✔
2837
    pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
191,666,747✔
2838
    if (pValCtx == NULL) {
191,755,297✔
2839
      QUERY_CHECK_CODE(terrno, lino, _end);
×
2840
    }
2841
  }
2842

2843
  for (int32_t i = 0; i < numOfOutput; ++i) {
818,851,309✔
2844
    const char* pName = pCtx[i].pExpr->pExpr->_function.functionName;
625,824,429✔
2845
    if ((strcmp(pName, "_select_value") == 0)) {
625,894,940✔
2846
      if (pValCtxArray == NULL) {
5,640,434✔
2847
        pValCtx[num++] = &pCtx[i];
3,113,271✔
2848
      } else {
2849
        int32_t bindFuncIndex = pCtx[i].pExpr->pExpr->relatedTo;  // start from index 1;
2,527,163✔
2850
        if (bindFuncIndex > 0) {                                  // 0 is default index related to the select function
2,526,671✔
2851
          bindFuncIndex -= 1;
2,468,123✔
2852
        }
2853
        SSubsidiaryResInfo** pSubsidiary = taosArrayGet(pValCtxArray, bindFuncIndex);
2,526,671✔
2854
        if (pSubsidiary == NULL) {
2,528,147✔
2855
          QUERY_CHECK_CODE(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR, lino, _end);
×
2856
        }
2857
        (*pSubsidiary)->pCtx[(*pSubsidiary)->num] = &pCtx[i];
2,528,147✔
2858
        (*pSubsidiary)->num++;
2,527,163✔
2859
      }
2860
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
620,254,506✔
2861
      if (pValCtxArray == NULL) {
50,751,323✔
2862
        p = &pCtx[i];
48,561,070✔
2863
      }
2864
    }
2865
  }
2866

2867
  if (p != NULL) {
193,026,880✔
2868
    p->subsidiaries.pCtx = pValCtx;
20,245,461✔
2869
    p->subsidiaries.num = num;
20,244,594✔
2870
  } else {
2871
    taosMemoryFreeClear(pValCtx);
172,781,419✔
2872
  }
2873

2874
_end:
1,341,875✔
2875
  if (code != TSDB_CODE_SUCCESS) {
192,963,603✔
2876
    taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2877
    taosMemoryFreeClear(pValCtx);
×
2878
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2879
  } else {
2880
    taosArrayDestroy(pValCtxArray);
192,963,603✔
2881
  }
2882
  return code;
192,956,752✔
2883
}
2884

2885
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowEntryInfoOffset,
193,043,664✔
2886
                                     SFunctionStateStore* pStore) {
2887
  int32_t         code = TSDB_CODE_SUCCESS;
193,043,664✔
2888
  int32_t         lino = 0;
193,043,664✔
2889
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
193,043,664✔
2890
  if (pFuncCtx == NULL) {
192,969,441✔
2891
    return NULL;
×
2892
  }
2893

2894
  *rowEntryInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
192,969,441✔
2895
  if (*rowEntryInfoOffset == 0) {
193,035,410✔
2896
    taosMemoryFreeClear(pFuncCtx);
×
2897
    return NULL;
×
2898
  }
2899

2900
  for (int32_t i = 0; i < numOfOutput; ++i) {
818,935,051✔
2901
    SExprInfo* pExpr = &pExprInfo[i];
625,907,011✔
2902

2903
    SExprBasicInfo* pFunct = &pExpr->base;
625,848,436✔
2904
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
625,871,767✔
2905

2906
    pCtx->functionId = -1;
625,885,281✔
2907
    pCtx->pExpr = pExpr;
625,858,272✔
2908

2909
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
625,898,750✔
2910
      SFuncExecEnv env = {0};
199,010,305✔
2911
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;
199,016,534✔
2912
      pCtx->isPseudoFunc = fmIsWindowPseudoColumnFunc(pCtx->functionId) || fmIsPlaceHolderFunc(pCtx->functionId);
199,001,475✔
2913
      pCtx->isNotNullFunc = fmIsNotNullOutputFunc(pCtx->functionId);
199,022,421✔
2914

2915
      bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
198,994,589✔
2916
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId)) {
317,995,645✔
2917
        if (!isUdaf) {
119,052,453✔
2918
          code = fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
119,010,161✔
2919
          QUERY_CHECK_CODE(code, lino, _end);
118,956,840✔
2920
        } else {
2921
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
42,292✔
2922
          pCtx->udfName = taosStrdup(udfName);
42,292✔
2923
          QUERY_CHECK_NULL(pCtx->udfName, code, lino, _end, terrno);
42,292✔
2924

2925
          code = fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
42,292✔
2926
          QUERY_CHECK_CODE(code, lino, _end);
42,292✔
2927
        }
2928
        bool tmp = pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
118,999,132✔
2929
        if (!tmp) {
119,024,328✔
2930
          code = terrno;
×
2931
          QUERY_CHECK_CODE(code, lino, _end);
4,163✔
2932
        }
2933
      } else {
2934
        if (fmIsPlaceHolderFunc(pCtx->functionId)) {
79,958,945✔
2935
          code = fmGetStreamPesudoFuncEnv(pCtx->functionId, pExpr->base.pParamList, &env);
8,591,822✔
2936
          QUERY_CHECK_CODE(code, lino, _end);
8,592,193✔
2937
        }      
2938
        
2939
        code = fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
79,958,187✔
2940
        if (code != TSDB_CODE_SUCCESS && isUdaf) {
79,954,765✔
2941
          code = TSDB_CODE_SUCCESS;
23,976✔
2942
        }
2943
        QUERY_CHECK_CODE(code, lino, _end);
79,954,765✔
2944

2945
        if (pCtx->sfp.getEnv != NULL) {
79,954,765✔
2946
          bool tmp = pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
15,679,817✔
2947
          if (!tmp) {
15,680,020✔
2948
            code = terrno;
×
2949
            QUERY_CHECK_CODE(code, lino, _end);
×
2950
          }
2951
        }
2952
      }
2953
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
198,990,481✔
2954
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
426,887,970✔
2955
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
7,772,918✔
2956
      // for simple column, the result buffer needs to hold at least one element.
2957
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
426,929,728✔
2958
    }
2959

2960
    pCtx->input.numOfInputCols = pFunct->numOfParams;
625,988,572✔
2961
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
625,931,816✔
2962
    QUERY_CHECK_NULL(pCtx->input.pData, code, lino, _end, terrno);
625,967,298✔
2963
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
625,951,307✔
2964
    QUERY_CHECK_NULL(pCtx->input.pColumnDataAgg, code, lino, _end, terrno);
625,965,334✔
2965

2966
    pCtx->pTsOutput = NULL;
625,915,408✔
2967
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
625,938,941✔
2968
    pCtx->resDataInfo.type = pFunct->resSchema.type;
625,922,289✔
2969
    pCtx->order = TSDB_ORDER_ASC;
625,890,374✔
2970
    pCtx->start.key = INT64_MIN;
625,941,980✔
2971
    pCtx->end.key = INT64_MIN;
625,929,353✔
2972
    pCtx->numOfParams = pExpr->base.numOfParams;
625,904,435✔
2973
    pCtx->param = pFunct->pParam;
625,882,985✔
2974
    pCtx->saveHandle.currentPage = -1;
625,938,405✔
2975
    pCtx->pStore = pStore;
625,872,477✔
2976
    pCtx->hasWindowOrGroup = false;
625,933,521✔
2977
    pCtx->needCleanup = false;
625,895,669✔
2978
  }
2979

2980
  for (int32_t i = 1; i < numOfOutput; ++i) {
637,583,492✔
2981
    (*rowEntryInfoOffset)[i] = (int32_t)((*rowEntryInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) +
889,176,561✔
2982
                                         pFuncCtx[i - 1].resDataInfo.interBufSize);
444,589,197✔
2983
  }
2984

2985
  code = setSelectValueColumnInfo(pFuncCtx, numOfOutput);
192,993,349✔
2986
  QUERY_CHECK_CODE(code, lino, _end);
192,967,807✔
2987

2988
_end:
192,967,807✔
2989
  if (code != TSDB_CODE_SUCCESS) {
192,874,212✔
2990
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2991
    for (int32_t i = 0; i < numOfOutput; ++i) {
×
2992
      taosMemoryFree(pFuncCtx[i].input.pData);
×
2993
      taosMemoryFree(pFuncCtx[i].input.pColumnDataAgg);
×
2994
    }
2995
    taosMemoryFreeClear(*rowEntryInfoOffset);
×
2996
    taosMemoryFreeClear(pFuncCtx);
×
2997

2998
    terrno = code;
×
2999
    return NULL;
×
3000
  }
3001
  return pFuncCtx;
192,874,212✔
3002
}
3003

3004
// NOTE: sources columns are more than the destination SSDatablock columns.
3005
// doFilter in table scan needs every column even its output is false
3006
int32_t relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray* pCols, bool outputEveryColumn) {
8,792,299✔
3007
  int32_t code = TSDB_CODE_SUCCESS;
8,792,299✔
3008
  size_t  numOfSrcCols = taosArrayGetSize(pCols);
8,792,299✔
3009

3010
  int32_t i = 0, j = 0;
8,792,977✔
3011
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
80,801,687✔
3012
    SColumnInfoData* p = taosArrayGet(pCols, i);
72,009,956✔
3013
    if (!p) {
72,009,278✔
3014
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3015
      return terrno;
×
3016
    }
3017
    SColMatchItem* pmInfo = taosArrayGet(pColMatchInfo, j);
72,009,278✔
3018
    if (!pmInfo) {
72,008,600✔
3019
      return terrno;
×
3020
    }
3021

3022
    if (p->info.colId == pmInfo->colId) {
72,008,600✔
3023
      SColumnInfoData* pDst = taosArrayGet(pBlock->pDataBlock, pmInfo->dstSlotId);
63,195,887✔
3024
      if (!pDst) {
63,195,155✔
3025
        return terrno;
×
3026
      }
3027
      code = colDataAssign(pDst, p, pBlock->info.rows, &pBlock->info);
63,195,155✔
3028
      if (code != TSDB_CODE_SUCCESS) {
63,195,317✔
3029
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3030
        return code;
×
3031
      }
3032
      i++;
63,195,317✔
3033
      j++;
63,195,317✔
3034
    } else if (p->info.colId < pmInfo->colId) {
8,813,758✔
3035
      i++;
8,813,393✔
3036
    } else {
3037
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3038
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3039
    }
3040
  }
3041
  return code;
8,792,774✔
3042
}
3043

3044
SInterval extractIntervalInfo(const STableScanPhysiNode* pTableScanNode) {
96,138,596✔
3045
  SInterval interval = {
192,279,457✔
3046
      .interval = pTableScanNode->interval,
96,139,424✔
3047
      .sliding = pTableScanNode->sliding,
96,137,220✔
3048
      .intervalUnit = pTableScanNode->intervalUnit,
96,149,815✔
3049
      .slidingUnit = pTableScanNode->slidingUnit,
96,150,883✔
3050
      .offset = pTableScanNode->offset,
96,160,378✔
3051
      .precision = pTableScanNode->scan.node.pOutputDataBlockDesc->precision,
96,145,219✔
3052
      .timeRange = pTableScanNode->scanRange,
3053
  };
3054
  calcIntervalAutoOffset(&interval);
96,138,725✔
3055

3056
  return interval;
96,159,646✔
3057
}
3058

3059
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
29,062,007✔
3060
  SColumn c = {0};
29,062,007✔
3061

3062
  c.slotId = pColNode->slotId;
29,062,007✔
3063
  c.colId = pColNode->colId;
29,062,100✔
3064
  c.type = pColNode->node.resType.type;
29,063,040✔
3065
  c.bytes = pColNode->node.resType.bytes;
29,061,423✔
3066
  c.scale = pColNode->node.resType.scale;
29,057,620✔
3067
  c.precision = pColNode->node.resType.precision;
29,056,514✔
3068
  return c;
29,057,084✔
3069
}
3070

3071

3072
/**
3073
 * @brief Determine the actual time range for reading data based on the RANGE clause and the WHERE conditions.
3074
 * @param[in] cond The range specified by WHERE condition.
3075
 * @param[in] range The range specified by RANGE clause.
3076
 * @param[out] twindow The range to be read in DESC order, and only one record is needed.
3077
 * @param[out] extTwindow The external range to read for only one record, which is used for FILL clause.
3078
 * @note `cond` and `twindow` may be the same address.
3079
 */
3080
static int32_t getQueryExtWindow(const STimeWindow* cond, const STimeWindow* range, STimeWindow* twindow,
1,606,181✔
3081
                                 STimeWindow* extTwindows) {
3082
  int32_t     code = TSDB_CODE_SUCCESS;
1,606,181✔
3083
  int32_t     lino = 0;
1,606,181✔
3084
  STimeWindow tempWindow;
3085

3086
  if (cond->skey > cond->ekey || range->skey > range->ekey) {
1,606,181✔
3087
    *twindow = extTwindows[0] = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
2,950✔
3088
    return code;
2,950✔
3089
  }
3090

3091
  if (range->ekey < cond->skey) {
1,603,231✔
3092
    extTwindows[1] = *cond;
249,121✔
3093
    *twindow = extTwindows[0] = TSWINDOW_DESC_INITIALIZER;
249,121✔
3094
    return code;
249,121✔
3095
  }
3096

3097
  if (cond->ekey < range->skey) {
1,354,110✔
3098
    extTwindows[0] = *cond;
178,781✔
3099
    *twindow = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
178,781✔
3100
    return code;
178,781✔
3101
  }
3102

3103
  // Only scan data in the time range intersecion.
3104
  extTwindows[0] = extTwindows[1] = *cond;
1,175,329✔
3105
  twindow->skey = TMAX(cond->skey, range->skey);
1,175,329✔
3106
  twindow->ekey = TMIN(cond->ekey, range->ekey);
1,175,329✔
3107
  extTwindows[0].ekey = twindow->skey - 1;
1,175,329✔
3108
  extTwindows[1].skey = twindow->ekey + 1;
1,175,329✔
3109

3110
  return code;
1,175,329✔
3111
}
3112

3113
int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, const STableScanPhysiNode* pTableScanNode,
118,447,167✔
3114
                               const SReadHandle* readHandle, bool applyExtWin) {
3115
  int32_t code = 0;                             
118,447,167✔
3116
  pCond->order = pTableScanNode->scanSeq[0] > 0 ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
118,447,167✔
3117
  pCond->numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
118,445,605✔
3118

3119
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
118,416,892✔
3120
  if (!pCond->colList) {
118,413,222✔
3121
    return terrno;
×
3122
  }
3123
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
118,397,451✔
3124
  if (pCond->pSlotList == NULL) {
118,407,822✔
3125
    taosMemoryFreeClear(pCond->colList);
×
3126
    return terrno;
×
3127
  }
3128

3129
  // TODO: get it from stable scan node
3130
  pCond->twindows = pTableScanNode->scanRange;
118,381,556✔
3131
  pCond->suid = pTableScanNode->scan.suid;
118,446,236✔
3132
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
118,419,562✔
3133
  pCond->startVersion = -1;
118,398,477✔
3134
  pCond->endVersion = -1;
118,389,522✔
3135
  pCond->skipRollup = readHandle->skipRollup;
118,360,743✔
3136
  if (readHandle->winRangeValid) {
118,406,610✔
3137
    pCond->twindows = readHandle->winRange;
289,974✔
3138
  }
3139
  pCond->cacheSttStatis = readHandle->cacheSttStatis;
118,392,253✔
3140
  // allowed read stt file optimization mode
3141
  pCond->notLoadData = (pTableScanNode->dataRequired == FUNC_DATA_REQUIRED_NOT_LOAD) &&
236,796,583✔
3142
                       (pTableScanNode->scan.node.pConditions == NULL) && (pTableScanNode->interval == 0);
118,368,956✔
3143

3144
  int32_t j = 0;
118,398,175✔
3145
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
739,752,995✔
3146
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pTableScanNode->scan.pScanCols, i);
621,358,043✔
3147
    if (!pNode) {
621,197,798✔
3148
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3149
      return terrno;
×
3150
    }
3151
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
621,197,798✔
3152
    if (pColNode->colType == COLUMN_TYPE_TAG) {
621,228,865✔
3153
      continue;
×
3154
    }
3155

3156
    pCond->colList[j].type = pColNode->node.resType.type;
621,277,988✔
3157
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
621,281,823✔
3158
    pCond->colList[j].colId = pColNode->colId;
621,320,196✔
3159
    pCond->colList[j].pk = pColNode->isPk;
621,344,031✔
3160

3161
    pCond->pSlotList[j] = pNode->slotId;
621,354,795✔
3162
    j += 1;
621,354,820✔
3163
  }
3164

3165
  pCond->numOfCols = j;
118,445,990✔
3166

3167
  if (applyExtWin) {
118,450,358✔
3168
    if (NULL != pTableScanNode->pExtScanRange) {
96,486,262✔
3169
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
1,606,181✔
3170
      code = getQueryExtWindow(&pCond->twindows, pTableScanNode->pExtScanRange, &pCond->twindows, pCond->extTwindows);
1,606,181✔
3171
    } else if (readHandle->extWinRangeValid) {
94,809,529✔
3172
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
×
3173
      code = getQueryExtWindow(&pCond->twindows, &readHandle->extWinRange, &pCond->twindows, pCond->extTwindows);
×
3174
    }
3175
  }
3176
  
3177
  return code;
118,370,653✔
3178
}
3179

3180
int32_t initQueryTableDataCondWithColArray(SQueryTableDataCond* pCond, SQueryTableDataCond* pOrgCond,
385,907✔
3181
                                           const SReadHandle* readHandle, SArray* colArray) {
3182
  int32_t code = TSDB_CODE_SUCCESS;
385,907✔
3183
  int32_t lino = 0;
385,907✔
3184

3185
  pCond->order = TSDB_ORDER_ASC;
385,907✔
3186
  pCond->numOfCols = (int32_t)taosArrayGetSize(colArray);
385,907✔
3187

3188
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
385,907✔
3189
  QUERY_CHECK_NULL(pCond->colList, code, lino, _return, terrno);
385,907✔
3190

3191
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
385,907✔
3192
  QUERY_CHECK_NULL(pCond->pSlotList, code, lino, _return, terrno);
385,907✔
3193

3194
  pCond->twindows = pOrgCond->twindows;
385,907✔
3195
  pCond->type = pOrgCond->type;
385,907✔
3196
  pCond->startVersion = -1;
385,907✔
3197
  pCond->endVersion = -1;
385,907✔
3198
  pCond->skipRollup = true;
385,907✔
3199
  pCond->notLoadData = false;
385,907✔
3200

3201
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
1,802,675✔
3202
    SColIdPair* pColPair = taosArrayGet(colArray, i);
1,416,768✔
3203
    QUERY_CHECK_NULL(pColPair, code, lino, _return, terrno);
1,416,768✔
3204

3205
    bool find = false;
1,416,768✔
3206
    for (int32_t j = 0; j < pOrgCond->numOfCols; ++j) {
7,056,387✔
3207
      if (pOrgCond->colList[j].colId == pColPair->vtbColId) {
7,056,387✔
3208
        pCond->colList[i].type = pOrgCond->colList[j].type;
1,416,768✔
3209
        pCond->colList[i].bytes = pOrgCond->colList[j].bytes;
1,416,768✔
3210
        pCond->colList[i].colId = pColPair->orgColId;
1,416,768✔
3211
        pCond->colList[i].pk = pOrgCond->colList[j].pk;
1,416,768✔
3212
        pCond->pSlotList[i] = i;
1,416,768✔
3213
        find = true;
1,416,768✔
3214
        break;
1,416,768✔
3215
      }
3216
    }
3217
    QUERY_CHECK_CONDITION(find, code, lino, _return, TSDB_CODE_NOT_FOUND);
1,416,768✔
3218
  }
3219

3220
  return code;
385,907✔
3221
_return:
×
3222
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(terrno));
×
3223
  taosMemoryFreeClear(pCond->colList);
×
3224
  taosMemoryFreeClear(pCond->pSlotList);
×
3225
  return code;
×
3226
}
3227

3228
void cleanupQueryTableDataCond(SQueryTableDataCond* pCond) {
259,032,174✔
3229
  taosMemoryFreeClear(pCond->colList);
259,032,174✔
3230
  taosMemoryFreeClear(pCond->pSlotList);
259,037,634✔
3231
}
259,034,229✔
3232

3233
int32_t convertFillType(int32_t mode) {
2,030,414✔
3234
  int32_t type = TSDB_FILL_NONE;
2,030,414✔
3235
  switch (mode) {
2,030,414✔
3236
    case FILL_MODE_PREV:
108,992✔
3237
      type = TSDB_FILL_PREV;
108,992✔
3238
      break;
108,992✔
3239
    case FILL_MODE_NONE:
×
3240
      type = TSDB_FILL_NONE;
×
3241
      break;
×
3242
    case FILL_MODE_NULL:
129,085✔
3243
      type = TSDB_FILL_NULL;
129,085✔
3244
      break;
129,085✔
3245
    case FILL_MODE_NULL_F:
15,698✔
3246
      type = TSDB_FILL_NULL_F;
15,698✔
3247
      break;
15,698✔
3248
    case FILL_MODE_NEXT:
95,413✔
3249
      type = TSDB_FILL_NEXT;
95,413✔
3250
      break;
95,413✔
3251
    case FILL_MODE_VALUE:
147,742✔
3252
      type = TSDB_FILL_SET_VALUE;
147,742✔
3253
      break;
147,742✔
3254
    case FILL_MODE_VALUE_F:
4,222✔
3255
      type = TSDB_FILL_SET_VALUE_F;
4,222✔
3256
      break;
4,222✔
3257
    case FILL_MODE_LINEAR:
145,856✔
3258
      type = TSDB_FILL_LINEAR;
145,856✔
3259
      break;
145,856✔
3260
    case FILL_MODE_NEAR:
1,383,406✔
3261
      type = TSDB_FILL_NEAR;
1,383,406✔
3262
      break;
1,383,406✔
3263
    default:
×
3264
      type = TSDB_FILL_NONE;
×
3265
  }
3266

3267
  return type;
2,030,414✔
3268
}
3269

3270
void getInitialStartTimeWindow(SInterval* pInterval, TSKEY ts, STimeWindow* w, bool ascQuery) {
1,335,493✔
3271
  if (ascQuery) {
1,335,493✔
3272
    *w = getAlignQueryTimeWindow(pInterval, ts);
1,066,601✔
3273
  } else {
3274
    // the start position of the first time window in the endpoint that spreads beyond the queried last timestamp
3275
    *w = getAlignQueryTimeWindow(pInterval, ts);
268,892✔
3276

3277
    int64_t key = w->skey;
271,302✔
3278
    while (key < ts) {  // moving towards end
285,796✔
3279
      key = getNextTimeWindowStart(pInterval, key, TSDB_ORDER_ASC);
162,488✔
3280
      if (key > ts) {
161,136✔
3281
        break;
146,642✔
3282
      }
3283

3284
      w->skey = key;
14,494✔
3285
    }
3286
    w->ekey = taosTimeAdd(w->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
269,950✔
3287
  }
3288
}
1,336,164✔
3289

3290
static STimeWindow doCalculateTimeWindow(int64_t ts, SInterval* pInterval) {
26,974,317✔
3291
  STimeWindow w = {0};
26,974,317✔
3292

3293
  w.skey = taosTimeTruncate(ts, pInterval);
26,974,317✔
3294
  w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
26,970,167✔
3295
  return w;
26,981,222✔
3296
}
3297

3298
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order) {
533,558✔
3299
  STimeWindow win = *pWindow;
533,558✔
3300
  STimeWindow save = win;
533,558✔
3301
  while (win.skey <= ts && win.ekey >= ts) {
1,683,338✔
3302
    save = win;
1,149,780✔
3303
    // get previous time window
3304
    getNextTimeWindow(pInterval, &win, order == TSDB_ORDER_ASC ? TSDB_ORDER_DESC : TSDB_ORDER_ASC);
1,149,780✔
3305
  }
3306

3307
  return save;
533,558✔
3308
}
3309

3310
// get the correct time window according to the handled timestamp
3311
// todo refactor
3312
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
37,232,597✔
3313
                                int32_t order) {
3314
  STimeWindow w = {0};
37,232,597✔
3315
  if (pResultRowInfo->cur.pageId == -1) {  // the first window, from the previous stored value
37,235,897✔
3316
    getInitialStartTimeWindow(pInterval, ts, &w, (order == TSDB_ORDER_ASC));
1,334,860✔
3317
    return w;
1,335,690✔
3318
  }
3319

3320
  SResultRow* pRow = getResultRowByPos(pBuf, &pResultRowInfo->cur, false);
35,901,841✔
3321
  if (pRow) {
35,901,635✔
3322
    TAOS_SET_OBJ_ALIGNED(&w, pRow->win);
35,903,818✔
3323
  }
3324

3325
  // in case of typical time window, we can calculate time window directly.
3326
  if (w.skey > ts || w.ekey < ts) {
35,903,553✔
3327
    w = doCalculateTimeWindow(ts, pInterval);
26,978,843✔
3328
  }
3329

3330
  if (pInterval->interval != pInterval->sliding) {
35,905,932✔
3331
    // it is an sliding window query, in which sliding value is not equalled to
3332
    // interval value, and we need to find the first qualified time window.
3333
    w = getFirstQualifiedTimeWindow(ts, &w, pInterval, order);
533,558✔
3334
  }
3335

3336
  return w;
35,902,532✔
3337
}
3338

3339
TSKEY getNextTimeWindowStart(const SInterval* pInterval, TSKEY start, int32_t order) {
2,147,483,647✔
3340
  int32_t factor = GET_FORWARD_DIRECTION_FACTOR(order);
2,147,483,647✔
3341
  TSKEY   nextStart = taosTimeAdd(start, -1 * pInterval->offset, pInterval->offsetUnit, pInterval->precision, NULL);
2,147,483,647✔
3342
  nextStart = taosTimeAdd(nextStart, factor * pInterval->sliding, pInterval->slidingUnit, pInterval->precision, NULL);
2,147,483,647✔
3343
  nextStart = taosTimeAdd(nextStart, pInterval->offset, pInterval->offsetUnit, pInterval->precision, NULL);
2,147,483,647✔
3344
  return nextStart;
2,147,483,647✔
3345
}
3346

3347
void getNextTimeWindow(const SInterval* pInterval, STimeWindow* tw, int32_t order) {
2,147,483,647✔
3348
  tw->skey = getNextTimeWindowStart(pInterval, tw->skey, order);
2,147,483,647✔
3349
  tw->ekey = taosTimeAdd(tw->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
2,147,483,647✔
3350
}
2,147,483,647✔
3351

3352
bool hasLimitOffsetInfo(SLimitInfo* pLimitInfo) {
298,604,609✔
3353
  return (pLimitInfo->limit.limit != -1 || pLimitInfo->limit.offset != -1 || pLimitInfo->slimit.limit != -1 ||
595,015,194✔
3354
          pLimitInfo->slimit.offset != -1);
296,409,996✔
3355
}
3356

3357
bool hasSlimitOffsetInfo(SLimitInfo* pLimitInfo) {
×
3358
  return (pLimitInfo->slimit.limit != -1 || pLimitInfo->slimit.offset != -1);
×
3359
}
3360

3361
void initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo) {
278,176,554✔
3362
  SLimit limit = {.limit = getLimit(pLimit), .offset = getOffset(pLimit)};
278,176,554✔
3363
  SLimit slimit = {.limit = getLimit(pSLimit), .offset = getOffset(pSLimit)};
278,123,520✔
3364

3365
  pLimitInfo->limit = limit;
278,135,041✔
3366
  pLimitInfo->slimit = slimit;
278,139,542✔
3367
  pLimitInfo->remainOffset = limit.offset;
278,115,163✔
3368
  pLimitInfo->remainGroupOffset = slimit.offset;
278,101,594✔
3369
  pLimitInfo->numOfOutputRows = 0;
278,119,583✔
3370
  pLimitInfo->numOfOutputGroups = 0;
278,131,903✔
3371
  pLimitInfo->currentGroupId = 0;
278,116,336✔
3372
}
278,139,659✔
3373

3374
void resetLimitInfoForNextGroup(SLimitInfo* pLimitInfo) {
44,456,442✔
3375
  pLimitInfo->numOfOutputRows = 0;
44,456,442✔
3376
  pLimitInfo->remainOffset = pLimitInfo->limit.offset;
44,458,579✔
3377
}
44,457,183✔
3378

3379
int32_t tableListGetSize(const STableListInfo* pTableList, int32_t* pRes) {
282,085,055✔
3380
  if (taosArrayGetSize(pTableList->pTableList) != taosHashGetSize(pTableList->map)) {
282,085,055✔
3381
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3382
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3383
  }
3384
  (*pRes) = taosArrayGetSize(pTableList->pTableList);
282,093,914✔
3385
  return TSDB_CODE_SUCCESS;
282,083,267✔
3386
}
3387

3388
uint64_t tableListGetSuid(const STableListInfo* pTableList) { return pTableList->idInfo.suid; }
3,201,105✔
3389

3390
STableKeyInfo* tableListGetInfo(const STableListInfo* pTableList, int32_t index) {
146,492,638✔
3391
  if (taosArrayGetSize(pTableList->pTableList) == 0) {
146,492,638✔
3392
    return NULL;
3,655✔
3393
  }
3394

3395
  return taosArrayGet(pTableList->pTableList, index);
146,472,118✔
3396
}
3397

3398
int32_t tableListFind(const STableListInfo* pTableList, uint64_t uid, int32_t startIndex) {
9,315✔
3399
  int32_t numOfTables = taosArrayGetSize(pTableList->pTableList);
9,315✔
3400
  if (startIndex >= numOfTables) {
9,315✔
3401
    return -1;
×
3402
  }
3403

3404
  for (int32_t i = startIndex; i < numOfTables; ++i) {
126,364✔
3405
    STableKeyInfo* p = taosArrayGet(pTableList->pTableList, i);
126,364✔
3406
    if (!p) {
126,364✔
3407
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3408
      return -1;
×
3409
    }
3410
    if (p->uid == uid) {
126,364✔
3411
      return i;
9,315✔
3412
    }
3413
  }
3414
  return -1;
×
3415
}
3416

3417
void tableListGetSourceTableInfo(const STableListInfo* pTableList, uint64_t* psuid, uint64_t* uid, int32_t* type) {
55,431✔
3418
  *psuid = pTableList->idInfo.suid;
55,431✔
3419
  *uid = pTableList->idInfo.uid;
55,431✔
3420
  *type = pTableList->idInfo.tableType;
55,431✔
3421
}
55,431✔
3422

3423
uint64_t tableListGetTableGroupId(const STableListInfo* pTableList, uint64_t tableUid) {
506,053,192✔
3424
  int32_t* slot = taosHashGet(pTableList->map, &tableUid, sizeof(tableUid));
506,053,192✔
3425
  if (slot == NULL) {
506,247,359✔
3426
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3427
    return -1;
×
3428
  }
3429

3430
  STableKeyInfo* pKeyInfo = taosArrayGet(pTableList->pTableList, *slot);
506,247,359✔
3431
  if (pKeyInfo == NULL) {
506,267,880✔
3432
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
32✔
3433
    return -1;
×
3434
  }
3435
  return pKeyInfo->groupId;
506,267,848✔
3436
}
3437

3438
// TODO handle the group offset info, fix it, the rule of group output will be broken by this function
3439
// int32_t tableListRemoveTableInfo(STableListInfo* pTableList, uint64_t uid) {
3440
//   int32_t code = TSDB_CODE_SUCCESS;
3441
//   int32_t lino = 0;
3442

3443
//   int32_t* slot = taosHashGet(pTableList->map, &uid, sizeof(uid));
3444
//   if (slot == NULL) {
3445
//     qDebug("table:%" PRIu64 " not found in table list", uid);
3446
//     return 0;
3447
//   }
3448

3449
//   taosArrayRemove(pTableList->pTableList, *slot);
3450
//   code = taosHashRemove(pTableList->map, &uid, sizeof(uid));
3451

3452
//   _end:
3453
//   if (code != TSDB_CODE_SUCCESS) {
3454
//     qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
3455
//   } else {
3456
//     qDebug("uid:%" PRIu64 ", remove from table list", uid);
3457
//   }
3458

3459
//   return code;
3460
// }
3461

3462
int32_t tableListAddTableInfo(STableListInfo* pTableList, uint64_t uid, uint64_t gid) {
170,974✔
3463
  int32_t code = TSDB_CODE_SUCCESS;
170,974✔
3464
  int32_t lino = 0;
170,974✔
3465
  if (pTableList->map == NULL) {
170,974✔
3466
    pTableList->map = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
×
3467
    QUERY_CHECK_NULL(pTableList->map, code, lino, _end, terrno);
×
3468
  }
3469

3470
  STableKeyInfo keyInfo = {.uid = uid, .groupId = gid};
171,068✔
3471
  void*         p = taosHashGet(pTableList->map, &uid, sizeof(uid));
171,068✔
3472
  if (p != NULL) {
170,998✔
3473
    qInfo("table:%" PRId64 " already in tableIdList, ignore it", uid);
192✔
3474
    goto _end;
192✔
3475
  }
3476

3477
  void* tmp = taosArrayPush(pTableList->pTableList, &keyInfo);
170,806✔
3478
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
170,782✔
3479

3480
  int32_t slot = (int32_t)taosArrayGetSize(pTableList->pTableList) - 1;
170,782✔
3481
  code = taosHashPut(pTableList->map, &uid, sizeof(uid), &slot, sizeof(slot));
170,876✔
3482
  if (code != TSDB_CODE_SUCCESS) {
170,876✔
3483
    // we have checked the existence of uid in hash map above
3484
    QUERY_CHECK_CONDITION((code != TSDB_CODE_DUP_KEY), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
3485
    taosArrayPopTailBatch(pTableList->pTableList, 1);  // let's pop the last element in the array list
×
3486
  }
3487

3488
_end:
171,068✔
3489
  if (code != TSDB_CODE_SUCCESS) {
171,068✔
3490
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3491
  } else {
3492
    qDebug("uid:%" PRIu64 ", groupId:%" PRIu64 " added into table list, slot:%d, total:%d", uid, gid, slot, slot + 1);
171,068✔
3493
  }
3494

3495
  return code;
171,068✔
3496
}
3497

3498
int32_t tableListGetGroupList(const STableListInfo* pTableList, int32_t ordinalGroupIndex, STableKeyInfo** pKeyInfo,
95,159,881✔
3499
                              int32_t* size) {
3500
  int32_t totalGroups = tableListGetOutputGroups(pTableList);
95,159,881✔
3501
  int32_t numOfTables = 0;
95,175,049✔
3502
  int32_t code = tableListGetSize(pTableList, &numOfTables);
95,177,277✔
3503
  if (code != TSDB_CODE_SUCCESS) {
95,169,379✔
3504
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3505
    return code;
×
3506
  }
3507

3508
  if (ordinalGroupIndex < 0 || ordinalGroupIndex >= totalGroups) {
95,169,379✔
3509
    return TSDB_CODE_INVALID_PARA;
×
3510
  }
3511

3512
  // here handle two special cases:
3513
  // 1. only one group exists, and 2. one table exists for each group.
3514
  if (totalGroups == 1) {
95,169,379✔
3515
    *size = numOfTables;
94,791,663✔
3516
    *pKeyInfo = (*size == 0) ? NULL : taosArrayGet(pTableList->pTableList, 0);
94,822,428✔
3517
    return TSDB_CODE_SUCCESS;
94,798,548✔
3518
  } else if (totalGroups == numOfTables) {
377,734✔
3519
    *size = 1;
323,656✔
3520
    *pKeyInfo = taosArrayGet(pTableList->pTableList, ordinalGroupIndex);
323,656✔
3521
    return TSDB_CODE_SUCCESS;
323,656✔
3522
  }
3523

3524
  int32_t offset = pTableList->groupOffset[ordinalGroupIndex];
54,089✔
3525
  if (ordinalGroupIndex < totalGroups - 1) {
54,453✔
3526
    *size = pTableList->groupOffset[ordinalGroupIndex + 1] - offset;
40,077✔
3527
  } else {
3528
    *size = numOfTables - offset;
14,376✔
3529
  }
3530

3531
  *pKeyInfo = taosArrayGet(pTableList->pTableList, offset);
54,453✔
3532
  return TSDB_CODE_SUCCESS;
54,453✔
3533
}
3534

3535
int32_t tableListGetOutputGroups(const STableListInfo* pTableList) { return pTableList->numOfOuputGroups; }
284,319,446✔
3536

3537
bool oneTableForEachGroup(const STableListInfo* pTableList) { return pTableList->oneTableForEachGroup; }
580,951✔
3538

3539
STableListInfo* tableListCreate() {
122,439,694✔
3540
  STableListInfo* pListInfo = taosMemoryCalloc(1, sizeof(STableListInfo));
122,439,694✔
3541
  if (pListInfo == NULL) {
122,402,286✔
3542
    return NULL;
×
3543
  }
3544

3545
  pListInfo->remainGroups = NULL;
122,402,286✔
3546
  pListInfo->pTableList = taosArrayInit(4, sizeof(STableKeyInfo));
122,402,608✔
3547
  if (pListInfo->pTableList == NULL) {
122,421,451✔
3548
    goto _error;
×
3549
  }
3550

3551
  pListInfo->map = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
122,428,381✔
3552
  if (pListInfo->map == NULL) {
122,467,794✔
3553
    goto _error;
×
3554
  }
3555

3556
  pListInfo->numOfOuputGroups = 1;
122,464,054✔
3557
  return pListInfo;
122,468,104✔
3558

3559
_error:
×
3560
  tableListDestroy(pListInfo);
×
3561
  return NULL;
×
3562
}
3563

3564
void tableListDestroy(STableListInfo* pTableListInfo) {
131,474,014✔
3565
  if (pTableListInfo == NULL) {
131,474,014✔
3566
    return;
9,031,997✔
3567
  }
3568

3569
  taosArrayDestroy(pTableListInfo->pTableList);
122,442,017✔
3570
  taosMemoryFreeClear(pTableListInfo->groupOffset);
122,446,186✔
3571

3572
  taosHashCleanup(pTableListInfo->map);
122,445,137✔
3573
  taosHashCleanup(pTableListInfo->remainGroups);
122,455,965✔
3574
  pTableListInfo->pTableList = NULL;
122,460,713✔
3575
  pTableListInfo->map = NULL;
122,452,791✔
3576
  taosMemoryFree(pTableListInfo);
122,442,804✔
3577
}
3578

3579
void tableListClear(STableListInfo* pTableListInfo) {
124,133✔
3580
  if (pTableListInfo == NULL) {
124,133✔
3581
    return;
×
3582
  }
3583

3584
  taosArrayClear(pTableListInfo->pTableList);
124,133✔
3585
  taosHashClear(pTableListInfo->map);
124,180✔
3586
  taosHashClear(pTableListInfo->remainGroups);
124,227✔
3587
  taosMemoryFree(pTableListInfo->groupOffset);
124,227✔
3588
  pTableListInfo->numOfOuputGroups = 1;
124,227✔
3589
  pTableListInfo->oneTableForEachGroup = false;
124,227✔
3590
}
3591

3592
static int32_t orderbyGroupIdComparFn(const void* p1, const void* p2) {
470,392,950✔
3593
  STableKeyInfo* pInfo1 = (STableKeyInfo*)p1;
470,392,950✔
3594
  STableKeyInfo* pInfo2 = (STableKeyInfo*)p2;
470,392,950✔
3595

3596
  if (pInfo1->groupId == pInfo2->groupId) {
470,392,950✔
3597
    return 0;
455,111,564✔
3598
  } else {
3599
    return pInfo1->groupId < pInfo2->groupId ? -1 : 1;
15,285,852✔
3600
  }
3601
}
3602

3603
int32_t sortTableGroup(STableListInfo* pTableListInfo) {
17,612,492✔
3604
  int32_t code = TSDB_CODE_SUCCESS;
17,612,492✔
3605
  taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
17,612,492✔
3606
  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
17,627,289✔
3607
  if (size == 0) {
17,620,619✔
3608
    pTableListInfo->numOfOuputGroups = 0;
×
3609
    return code;
×
3610
  }
3611

3612
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
17,620,619✔
3613
  if (!pList) {
17,627,068✔
3614
    code = terrno;
×
3615
    goto end;
×
3616
  }
3617

3618
  STableKeyInfo* pInfo = taosArrayGet(pTableListInfo->pTableList, 0);
17,627,068✔
3619
  if (pInfo == NULL) {
17,616,574✔
3620
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3621
    code = terrno;
×
3622
    goto end;
×
3623
  }
3624
  uint64_t gid = pInfo->groupId;
17,616,574✔
3625

3626
  int32_t start = 0;
17,616,083✔
3627
  void*   tmp = taosArrayPush(pList, &start);
17,623,553✔
3628
  if (tmp == NULL) {
17,623,553✔
3629
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3630
    code = terrno;
×
3631
    goto end;
×
3632
  }
3633

3634
  for (int32_t i = 1; i < size; ++i) {
115,034,487✔
3635
    pInfo = taosArrayGet(pTableListInfo->pTableList, i);
97,412,737✔
3636
    if (pInfo == NULL) {
97,411,687✔
3637
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3638
      code = terrno;
×
3639
      goto end;
×
3640
    }
3641
    if (pInfo->groupId != gid) {
97,411,687✔
3642
      tmp = taosArrayPush(pList, &i);
3,347,434✔
3643
      if (tmp == NULL) {
3,347,434✔
3644
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3645
        code = terrno;
×
3646
        goto end;
×
3647
      }
3648
      gid = pInfo->groupId;
3,347,434✔
3649
    }
3650
  }
3651

3652
  pTableListInfo->numOfOuputGroups = taosArrayGetSize(pList);
17,621,116✔
3653
  pTableListInfo->groupOffset = taosMemoryMalloc(sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
17,621,863✔
3654
  if (pTableListInfo->groupOffset == NULL) {
17,618,685✔
3655
    code = terrno;
×
3656
    goto end;
×
3657
  }
3658

3659
  memcpy(pTableListInfo->groupOffset, taosArrayGet(pList, 0), sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
17,610,349✔
3660

3661
end:
17,615,981✔
3662
  taosArrayDestroy(pList);
17,608,926✔
3663
  return code;
17,613,472✔
3664
}
3665

3666
int32_t buildGroupIdMapForAllTables(STableListInfo* pTableListInfo, SReadHandle* pHandle, SScanPhysiNode* pScanNode,
106,566,381✔
3667
                                    SNodeList* group, bool groupSort, uint8_t* digest, SStorageAPI* pAPI, SHashObj* groupIdMap) {
3668
  int32_t code = TSDB_CODE_SUCCESS;
106,566,381✔
3669

3670
  bool   groupByTbname = groupbyTbname(group);
106,566,381✔
3671
  size_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
106,562,777✔
3672
  if (!numOfTables) {
106,560,758✔
3673
    return code;
6,204✔
3674
  }
3675
  qDebug("numOfTables:%zu, groupByTbname:%d, group:%p", numOfTables, groupByTbname, group);
106,554,554✔
3676
  if (group == NULL || groupByTbname) {
106,546,579✔
3677
    if (tsCountAlwaysReturnValue && QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode) &&
104,735,400✔
3678
        ((STableScanPhysiNode*)pScanNode)->needCountEmptyTable) {
82,319,682✔
3679
      pTableListInfo->remainGroups =
4,291,005✔
3680
          taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
4,291,005✔
3681
      if (pTableListInfo->remainGroups == NULL) {
4,291,005✔
3682
        return terrno;
×
3683
      }
3684

3685
      for (int i = 0; i < numOfTables; i++) {
18,482,709✔
3686
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
14,191,570✔
3687
        if (!info) {
14,191,972✔
3688
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3689
          return terrno;
×
3690
        }
3691
        info->groupId = groupByTbname ? info->uid : 0;
14,191,972✔
3692
        int32_t tempRes = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId),
14,192,106✔
3693
                                      &(info->uid), sizeof(info->uid));
14,192,106✔
3694
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
14,191,838✔
3695
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3696
          return tempRes;
×
3697
        }
3698
      }
3699
    } else {
3700
      for (int32_t i = 0; i < numOfTables; i++) {
464,846,526✔
3701
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
364,401,913✔
3702
        if (!info) {
364,379,410✔
3703
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3704
          return terrno;
×
3705
        }
3706
        info->groupId = groupByTbname ? info->uid : 0;
364,379,410✔
3707
        
3708
      }
3709
    }
3710
    if (groupIdMap && group != NULL){
104,735,752✔
3711
      getColInfoResultForGroupbyForStream(pHandle->vnode, group, pTableListInfo, pAPI, groupIdMap);
105,896✔
3712
    }
3713

3714
    pTableListInfo->oneTableForEachGroup = groupByTbname;
104,735,752✔
3715
    if (numOfTables == 1 && pTableListInfo->idInfo.tableType == TSDB_CHILD_TABLE) {
104,738,547✔
3716
      pTableListInfo->oneTableForEachGroup = true;
32,193,481✔
3717
    }
3718

3719
    if (groupSort && groupByTbname) {
104,735,922✔
3720
      taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
1,401,788✔
3721
      pTableListInfo->numOfOuputGroups = numOfTables;
1,401,654✔
3722
    } else if (groupByTbname && pScanNode->groupOrderScan) {
103,334,134✔
3723
      pTableListInfo->numOfOuputGroups = numOfTables;
30,962✔
3724
    } else {
3725
      pTableListInfo->numOfOuputGroups = 1;
103,303,363✔
3726
    }
3727
    if (groupSort || pScanNode->groupOrderScan) {
104,738,450✔
3728
      code = sortTableGroup(pTableListInfo);
17,463,875✔
3729
    }
3730
  } else {
3731
    bool initRemainGroups = false;
1,811,179✔
3732
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode)) {
1,811,179✔
3733
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pScanNode;
1,592,309✔
3734
      if (tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable &&
1,592,309✔
3735
          !(groupSort || pScanNode->groupOrderScan)) {
811,538✔
3736
        initRemainGroups = true;
785,254✔
3737
      }
3738
    }
3739

3740
    code = getColInfoResultForGroupby(pHandle->vnode, group, pTableListInfo, digest, pAPI, initRemainGroups, groupIdMap);
1,811,179✔
3741
    if (code != TSDB_CODE_SUCCESS) {
1,810,697✔
3742
      return code;
×
3743
    }
3744

3745
    if (pScanNode->groupOrderScan) pTableListInfo->numOfOuputGroups = taosArrayGetSize(pTableListInfo->pTableList);
1,810,697✔
3746

3747
    if (groupSort || pScanNode->groupOrderScan) {
1,810,697✔
3748
      code = sortTableGroup(pTableListInfo);
150,353✔
3749
    }
3750
  }
3751

3752
  // add all table entry in the hash map
3753
  size_t size = taosArrayGetSize(pTableListInfo->pTableList);
106,542,998✔
3754
  for (int32_t i = 0; i < size; ++i) {
493,732,795✔
3755
    STableKeyInfo* p = taosArrayGet(pTableListInfo->pTableList, i);
387,169,013✔
3756
    if (!p) {
387,151,015✔
3757
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3758
      return terrno;
×
3759
    }
3760
    int32_t tempRes = taosHashPut(pTableListInfo->map, &p->uid, sizeof(uint64_t), &i, sizeof(int32_t));
387,151,015✔
3761
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
387,196,740✔
3762
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3763
      return tempRes;
×
3764
    }
3765
  }
3766

3767
  return code;
106,568,716✔
3768
}
3769

3770
int32_t createScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
117,927,845✔
3771
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
3772
                                SExecTaskInfo* pTaskInfo, SHashObj* groupIdMap) {
3773
  int64_t     st = taosGetTimestampUs();
117,911,541✔
3774
  const char* idStr = GET_TASKID(pTaskInfo);
117,911,541✔
3775

3776
  if (pHandle == NULL) {
117,788,651✔
3777
    qError("invalid handle, in creating operator tree, %s", idStr);
×
3778
    return TSDB_CODE_INVALID_PARA;
×
3779
  }
3780

3781
  if (pHandle->uid != 0) {
117,788,651✔
3782
    pScanNode->uid = pHandle->uid;
39,961✔
3783
    pScanNode->tableType = TSDB_CHILD_TABLE;
39,961✔
3784
  }
3785
  uint8_t digest[17] = {0};
117,839,674✔
3786
  int32_t code = getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, digest, idStr,
117,890,174✔
3787
                              &pTaskInfo->storageAPI, pTaskInfo->pStreamRuntimeInfo);
117,872,106✔
3788
  if (code != TSDB_CODE_SUCCESS) {
117,978,410✔
3789
    qError("failed to getTableList, code:%s", tstrerror(code));
1,096✔
3790
    return code;
1,096✔
3791
  }
3792

3793
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
117,977,314✔
3794

3795
  int64_t st1 = taosGetTimestampUs();
117,984,205✔
3796
  pTaskInfo->cost.extractListTime = (st1 - st) / 1000.0;
117,984,205✔
3797
  qDebug("extract queried table list completed, %d tables, elapsed time:%.2f ms %s", numOfTables,
117,989,016✔
3798
         pTaskInfo->cost.extractListTime, idStr);
3799

3800
  if (numOfTables == 0) {
117,970,983✔
3801
    qDebug("no table qualified for query, %s", idStr);
11,557,945✔
3802
    return TSDB_CODE_SUCCESS;
11,558,061✔
3803
  }
3804

3805
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, digest, &pTaskInfo->storageAPI, groupIdMap);
106,413,038✔
3806
  if (code != TSDB_CODE_SUCCESS) {
106,429,898✔
3807
    return code;
×
3808
  }
3809

3810
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
106,430,499✔
3811
  qDebug("generate group id map completed, elapsed time:%.2f ms %s", pTaskInfo->cost.groupIdMapTime, idStr);
106,428,268✔
3812

3813
  return TSDB_CODE_SUCCESS;
106,428,283✔
3814
}
3815

3816
char* getStreamOpName(uint16_t opType) {
9,840,118✔
3817
  switch (opType) {
9,840,118✔
3818
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
×
3819
      return "stream scan";
×
3820
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
9,638,049✔
3821
      return "project";
9,638,049✔
3822
    case QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW:
202,491✔
3823
      return "external window";
202,491✔
3824
  }
3825
  return "error name";
×
3826
}
3827

3828
void printDataBlock(SSDataBlock* pBlock, const char* flag, const char* taskIdStr, int64_t qId) {
247,222,546✔
3829
  if (qDebugFlag & DEBUG_TRACE) {
247,222,546✔
3830
    if (!pBlock) {
37,976✔
3831
      qDebug("%" PRIx64 " %s %s %s: Block is Null", qId, taskIdStr, flag, __func__);
6,868✔
3832
      return;
6,868✔
3833
    } else if (pBlock->info.rows == 0) {
31,108✔
3834
      qDebug("%" PRIx64 " %s %s %s: Block is Empty. block type %d", qId, taskIdStr, flag, __func__, pBlock->info.type);
×
3835
      return;
×
3836
    }
3837
    
3838
    char*   pBuf = NULL;
31,108✔
3839
    int32_t code = dumpBlockData(pBlock, flag, &pBuf, taskIdStr, qId);
31,108✔
3840
    if (code == 0) {
31,108✔
3841
      qDebugL("%" PRIx64 " %s %s", qId, __func__, pBuf);
31,108✔
3842
      taosMemoryFree(pBuf);
31,108✔
3843
    }
3844
  }
3845
}
3846

3847
void printSpecDataBlock(SSDataBlock* pBlock, const char* flag, const char* opStr, const char* taskIdStr) {
×
3848
  if (!pBlock) {
×
3849
    qDebug("%s===stream===%s %s: Block is Null", taskIdStr, flag, opStr);
×
3850
    return;
×
3851
  } else if (pBlock->info.rows == 0) {
×
3852
    qDebug("%s===stream===%s %s: Block is Empty. block type %d.skey:%" PRId64 ",ekey:%" PRId64 ",version%" PRId64,
×
3853
           taskIdStr, flag, opStr, pBlock->info.type, pBlock->info.window.skey, pBlock->info.window.ekey,
3854
           pBlock->info.version);
3855
    return;
×
3856
  }
3857
  if (qDebugFlag & DEBUG_TRACE) {
×
3858
    char* pBuf = NULL;
×
3859
    char  flagBuf[64];
×
3860
    snprintf(flagBuf, sizeof(flagBuf), "%s %s", flag, opStr);
×
3861
    int32_t code = dumpBlockData(pBlock, flagBuf, &pBuf, taskIdStr, 0);
×
3862
    if (code == 0) {
×
3863
      qDebug("%s", pBuf);
×
3864
      taosMemoryFree(pBuf);
×
3865
    }
3866
  }
3867
}
3868

3869
TSKEY getStartTsKey(STimeWindow* win, const TSKEY* tsCols) { return tsCols == NULL ? win->skey : tsCols[0]; }
39,020,593✔
3870

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

3874
  int64_t duration = pWin->ekey - pWin->skey + delta;
2,147,483,647✔
3875
  ts[2] = duration;            // set the duration
2,147,483,647✔
3876
  ts[3] = pWin->skey;          // window start key
2,147,483,647✔
3877
  ts[4] = pWin->ekey + delta;  // window end key
2,147,483,647✔
3878
}
2,147,483,647✔
3879

3880
int32_t compKeys(const SArray* pSortGroupCols, const char* oldkeyBuf, int32_t oldKeysLen, const SSDataBlock* pBlock,
905,211,279✔
3881
                 int32_t rowIndex) {
3882
  SColumnDataAgg* pColAgg = NULL;
905,211,279✔
3883
  const char*     isNull = oldkeyBuf;
905,211,279✔
3884
  const char*     p = oldkeyBuf + sizeof(int8_t) * pSortGroupCols->size;
905,211,279✔
3885

3886
  for (int32_t i = 0; i < pSortGroupCols->size; ++i) {
2,147,483,647✔
3887
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
1,403,457,006✔
3888
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
1,405,715,835✔
3889
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
1,405,823,283✔
3890

3891
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
2,147,483,647✔
3892
      if (isNull[i] != 1) return 1;
98,025,375✔
3893
    } else {
3894
      if (isNull[i] != 0) return 1;
1,308,426,369✔
3895
      const char* val = colDataGetData(pColInfoData, rowIndex);
1,307,538,261✔
3896
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
1,307,553,708✔
3897
        int32_t len = getJsonValueLen(val);
×
3898
        if (memcmp(p, val, len) != 0) return 1;
×
3899
        p += len;
×
3900
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
1,307,429,787✔
3901
        if (IS_STR_DATA_BLOB(pCol->type)) {
443,149,299✔
3902
          if (memcmp(p, val, blobDataTLen(val)) != 0) return 1;
42✔
3903
          p += blobDataTLen(val);
×
3904
        } else {
3905
          if (memcmp(p, val, varDataTLen(val)) != 0) return 1;
443,351,193✔
3906
          p += varDataTLen(val);
436,282,899✔
3907
        }
3908
      } else {
3909
        if (0 != memcmp(p, val, pCol->bytes)) return 1;
864,042,732✔
3910
        p += pCol->bytes;
846,725,508✔
3911
      }
3912
    }
3913
  }
3914
  if ((int32_t)(p - oldkeyBuf) != oldKeysLen) return 1;
880,444,929✔
3915
  return 0;
880,432,551✔
3916
}
3917

3918
int32_t buildKeys(char* keyBuf, const SArray* pSortGroupCols, const SSDataBlock* pBlock, int32_t rowIndex) {
24,931,785✔
3919
  uint32_t        colNum = pSortGroupCols->size;
24,931,785✔
3920
  SColumnDataAgg* pColAgg = NULL;
24,931,785✔
3921
  char*           isNull = keyBuf;
24,931,785✔
3922
  char*           p = keyBuf + sizeof(int8_t) * colNum;
24,931,785✔
3923

3924
  for (int32_t i = 0; i < colNum; ++i) {
74,500,347✔
3925
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
49,565,664✔
3926
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
49,570,011✔
3927
    if (pCol->slotId > pBlock->pDataBlock->size) continue;
49,571,874✔
3928

3929
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
49,573,323✔
3930

3931
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
99,147,888✔
3932
      isNull[i] = 1;
1,782,951✔
3933
    } else {
3934
      isNull[i] = 0;
47,793,063✔
3935
      const char* val = colDataGetData(pColInfoData, rowIndex);
47,783,955✔
3936
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
47,789,337✔
3937
        int32_t len = getJsonValueLen(val);
×
3938
        memcpy(p, val, len);
×
3939
        p += len;
×
3940
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
47,790,786✔
3941
        if (IS_STR_DATA_BLOB(pCol->type)) {
7,176,426✔
3942
          blobDataCopy(p, val);
×
3943
          p += blobDataTLen(val);
×
3944
        } else {
3945
          varDataCopy(p, val);
7,181,601✔
3946
          p += varDataTLen(val);
7,181,601✔
3947
        }
3948
      } else {
3949
        memcpy(p, val, pCol->bytes);
40,610,013✔
3950
        p += pCol->bytes;
40,605,459✔
3951
      }
3952
    }
3953
  }
3954
  return (int32_t)(p - keyBuf);
24,934,683✔
3955
}
3956

3957
uint64_t calcGroupId(char* pData, int32_t len) {
2,147,483,647✔
3958
  T_MD5_CTX context;
2,147,483,647✔
3959
  tMD5Init(&context);
2,147,483,647✔
3960
  tMD5Update(&context, (uint8_t*)pData, len);
2,147,483,647✔
3961
  tMD5Final(&context);
2,147,483,647✔
3962

3963
  // NOTE: only extract the initial 8 bytes of the final MD5 digest
3964
  uint64_t id = 0;
2,147,483,647✔
3965
  memcpy(&id, context.digest, sizeof(uint64_t));
2,147,483,647✔
3966
  if (0 == id) memcpy(&id, context.digest + 8, sizeof(uint64_t));
2,147,483,647✔
3967
  return id;
2,147,483,647✔
3968
}
3969

3970
SNodeList* makeColsNodeArrFromSortKeys(SNodeList* pSortKeys) {
40,116✔
3971
  SNode*     node;
3972
  SNodeList* ret = NULL;
40,116✔
3973
  FOREACH(node, pSortKeys) {
122,160✔
3974
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)node;
82,044✔
3975
    int32_t           code = nodesListMakeAppend(&ret, pSortKey->pExpr);
82,044✔
3976
    if (code != TSDB_CODE_SUCCESS) {
82,044✔
3977
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3978
      terrno = code;
×
3979
      return NULL;
×
3980
    }
3981
  }
3982
  return ret;
40,116✔
3983
}
3984

3985
int32_t extractKeysLen(const SArray* keys, int32_t* pLen) {
40,116✔
3986
  int32_t code = TSDB_CODE_SUCCESS;
40,116✔
3987
  int32_t lino = 0;
40,116✔
3988
  int32_t len = 0;
40,116✔
3989
  int32_t keyNum = taosArrayGetSize(keys);
40,116✔
3990
  for (int32_t i = 0; i < keyNum; ++i) {
102,360✔
3991
    SColumn* pCol = (SColumn*)taosArrayGet(keys, i);
62,244✔
3992
    QUERY_CHECK_NULL(pCol, code, lino, _end, terrno);
62,244✔
3993
    len += pCol->bytes;
62,244✔
3994
  }
3995
  len += sizeof(int8_t) * keyNum;  // null flag
40,116✔
3996
  *pLen = len;
40,116✔
3997

3998
_end:
40,116✔
3999
  if (code != TSDB_CODE_SUCCESS) {
40,116✔
4000
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4001
  }
4002
  return code;
40,116✔
4003
}
4004

4005
int32_t parseErrorMsgFromAnalyticServer(SJson* pJson, const char* pId) {
×
4006
  int32_t code = TSDB_CODE_ANA_ANODE_RETURN_ERROR;
×
4007
  if (pJson == NULL) {
×
4008
    return code;
×
4009
  }
4010

4011
  char    pMsg[1024] = {0};
×
4012
  int32_t ret = tjsonGetStringValue(pJson, "msg", pMsg);
×
4013

4014
  if (ret == 0) {
×
4015
    qError("%s failed to exec imputation, msg:%s", pId, pMsg);
×
4016
    if (strstr(pMsg, "white noise") != NULL) {
×
4017
      code = TSDB_CODE_ANA_WN_DATA;
×
4018
    } else if (strstr(pMsg, "white-noise") != NULL) {
×
4019
      code = TSDB_CODE_ANA_WN_DATA;
×
4020
    } else if (strstr(pMsg, "[Errno 111] Connection refused") != NULL) {
×
4021
      code = TSDB_CODE_ANA_ALGO_NOT_LOAD;
×
4022
    }
4023
  } else {
4024
    qError("%s failed to extract msg from server, unknown error", pId);
×
4025
  }
4026

4027
  return code;
×
4028
}
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