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

taosdata / TDengine / #3778

28 Mar 2025 06:51AM UTC coverage: 63.374% (+0.4%) from 62.934%
#3778

push

travis-ci

web-flow
fix(tdb): disable page recycling (#30529)

155771 of 313582 branches covered (49.67%)

Branch coverage included in aggregate %.

241569 of 313390 relevant lines covered (77.08%)

20705705.27 hits per line

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

67.8
/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 "tarray.h"
23
#include "tdatablock.h"
24
#include "thash.h"
25
#include "tmsg.h"
26
#include "ttime.h"
27

28
#include "executil.h"
29
#include "executorInt.h"
30
#include "querytask.h"
31
#include "storageapi.h"
32
#include "tcompression.h"
33

34
typedef struct tagFilterAssist {
35
  SHashObj* colHash;
36
  int32_t   index;
37
  SArray*   cInfoList;
38
  int32_t   code;
39
} tagFilterAssist;
40

41
typedef struct STransTagExprCtx {
42
  int32_t      code;
43
  SMetaReader* pReader;
44
} STransTagExprCtx;
45

46
typedef enum {
47
  FILTER_NO_LOGIC = 1,
48
  FILTER_AND,
49
  FILTER_OTHER,
50
} FilterCondType;
51

52
static FilterCondType checkTagCond(SNode* cond);
53
static int32_t optimizeTbnameInCond(void* metaHandle, int64_t suid, SArray* list, SNode* pTagCond, SStorageAPI* pAPI);
54
static int32_t optimizeTbnameInCondImpl(void* metaHandle, SArray* list, SNode* pTagCond, SStorageAPI* pStoreAPI, uint64_t suid);
55

56
static int32_t getTableList(void* pVnode, SScanPhysiNode* pScanNode, SNode* pTagCond, SNode* pTagIndexCond,
57
                            STableListInfo* pListInfo, uint8_t* digest, const char* idstr, SStorageAPI* pStorageAPI);
58

59
static int64_t getLimit(const SNode* pLimit) { return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->limit) ? -1 : ((SLimitNode*)pLimit)->limit->datum.i; }
28,916,484!
60
static int64_t getOffset(const SNode* pLimit) { return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->offset) ? -1 : ((SLimitNode*)pLimit)->offset->datum.i; }
28,922,935✔
61
static void    releaseColInfoData(void* pCol);
62

63
void initResultRowInfo(SResultRowInfo* pResultRowInfo) {
12,073,517✔
64
  pResultRowInfo->size = 0;
12,073,517✔
65
  pResultRowInfo->cur.pageId = -1;
12,073,517✔
66
}
12,073,517✔
67

68
void closeResultRow(SResultRow* pResultRow) { pResultRow->closed = true; }
8,088,404✔
69

70
void resetResultRow(SResultRow* pResultRow, size_t entrySize) {
67,209,940✔
71
  pResultRow->numOfRows = 0;
67,209,940✔
72
  pResultRow->closed = false;
67,209,940✔
73
  pResultRow->endInterp = false;
67,209,940✔
74
  pResultRow->startInterp = false;
67,209,940✔
75

76
  if (entrySize > 0) {
67,209,940!
77
    memset(pResultRow->pEntryInfo, 0, entrySize);
67,215,800✔
78
  }
79
}
67,209,940✔
80

81
// TODO refactor: use macro
82
SResultRowEntryInfo* getResultEntryInfo(const SResultRow* pRow, int32_t index, const int32_t* offset) {
2,147,483,647✔
83
  return (SResultRowEntryInfo*)((char*)pRow->pEntryInfo + offset[index]);
2,147,483,647✔
84
}
85

86
size_t getResultRowSize(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
8,612,809✔
87
  int32_t rowSize = (numOfOutput * sizeof(SResultRowEntryInfo)) + sizeof(SResultRow);
8,612,809✔
88

89
  for (int32_t i = 0; i < numOfOutput; ++i) {
34,567,256✔
90
    rowSize += pCtx[i].resDataInfo.interBufSize;
25,954,447✔
91
  }
92

93
  return rowSize;
8,612,809✔
94
}
95

96
// Convert buf read from rocksdb to result row
97
int32_t getResultRowFromBuf(SExprSupp *pSup, const char* inBuf, size_t inBufSize, char **outBuf, size_t *outBufSize) {
2,799✔
98
  if (inBuf == NULL || pSup == NULL) {
2,799!
99
    qError("invalid input parameters, inBuf:%p, pSup:%p", inBuf, pSup);
23!
100
    return TSDB_CODE_INVALID_PARA;
23✔
101
  }
102
  SqlFunctionCtx *pCtx = pSup->pCtx;
2,776✔
103
  int32_t        *offset = pSup->rowEntryInfoOffset;
2,776✔
104
  SResultRow     *pResultRow  = NULL;
2,776✔
105
  size_t          processedSize = 0;
2,776✔
106
  int32_t         code = TSDB_CODE_SUCCESS;
2,776✔
107

108
  // calculate the size of output buffer
109
  *outBufSize = getResultRowSize(pCtx, pSup->numOfExprs);
2,776✔
110
  *outBuf = taosMemoryMalloc(*outBufSize);
2,776!
111
  if (*outBuf == NULL) {
2,776!
112
    qError("failed to allocate memory for output buffer, size:%zu", *outBufSize);
×
113
    return terrno;
×
114
  }
115
  pResultRow = (SResultRow*)*outBuf;
2,776✔
116
  (void)memcpy(pResultRow, inBuf, sizeof(SResultRow));
2,776✔
117
  inBuf += sizeof(SResultRow);
2,776✔
118
  processedSize += sizeof(SResultRow);
2,776✔
119

120
  for (int32_t i = 0; i < pSup->numOfExprs; ++i) {
43,742✔
121
    int32_t len = *(int32_t*)inBuf;
40,966✔
122
    inBuf += sizeof(int32_t);
40,966✔
123
    processedSize += sizeof(int32_t);
40,966✔
124
    if (pResultRow->version != FUNCTION_RESULT_INFO_VERSION && pCtx->fpSet.decode) {
40,966!
125
      code = pCtx->fpSet.decode(&pCtx[i], inBuf, getResultEntryInfo(pResultRow, i, offset), pResultRow->version);
×
126
      if (code != TSDB_CODE_SUCCESS) {
×
127
        qError("failed to decode result row, code:%d", code);
×
128
        return code;
×
129
      }
130
    } else {
131
      (void)memcpy(getResultEntryInfo(pResultRow, i, offset), inBuf, len);
40,966✔
132
    }
133
    inBuf += len;
40,966✔
134
    processedSize += len;
40,966✔
135
  }
136

137
  if (processedSize < inBufSize) {
2,776✔
138
    // stream stores extra data after result row
139
    size_t leftLen = inBufSize - processedSize;
1,033✔
140
    TAOS_MEMORY_REALLOC(*outBuf, *outBufSize + leftLen);
1,033!
141
    if (*outBuf == NULL) {
1,033!
142
      qError("failed to reallocate memory for output buffer, size:%zu", *outBufSize + leftLen);
×
143
      return terrno;
×
144
    }
145
    (void)memcpy(*outBuf + *outBufSize, inBuf, leftLen);
1,033✔
146
    inBuf += leftLen;
1,033✔
147
    processedSize += leftLen;
1,033✔
148
    *outBufSize += leftLen;
1,033✔
149
  }
150
  return TSDB_CODE_SUCCESS;
2,776✔
151
}
152

153
// Convert result row to buf for rocksdb
154
int32_t putResultRowToBuf(SExprSupp *pSup, const char* inBuf, size_t inBufSize, char **outBuf, size_t *outBufSize) {
1,428,693✔
155
  if (pSup == NULL || inBuf == NULL || outBuf == NULL || outBufSize == NULL) {
1,428,693!
156
    qError("invalid input parameters, inBuf:%p, pSup:%p, outBufSize:%p, outBuf:%p", inBuf, pSup, outBufSize, outBuf);
×
157
    return TSDB_CODE_INVALID_PARA;
×
158
  }
159

160
  SqlFunctionCtx *pCtx = pSup->pCtx;
1,428,706✔
161
  int32_t        *offset = pSup->rowEntryInfoOffset;
1,428,706✔
162
  SResultRow     *pResultRow = (SResultRow*)inBuf;
1,428,706✔
163
  size_t          rowSize = getResultRowSize(pCtx, pSup->numOfExprs);
1,428,706✔
164

165
  if (rowSize > inBufSize) {
1,428,702!
166
    qError("invalid input buffer size, rowSize:%zu, inBufSize:%zu", rowSize, inBufSize);
×
167
    return TSDB_CODE_INVALID_PARA;
×
168
  }
169

170
  // calculate the size of output buffer
171
  *outBufSize = rowSize + sizeof(int32_t) * pSup->numOfExprs;
1,428,702✔
172
  if (rowSize < inBufSize) {
1,428,702✔
173
    *outBufSize += inBufSize - rowSize;
1,294✔
174
  }
175

176
  *outBuf = taosMemoryMalloc(*outBufSize);
1,428,702!
177
  if (*outBuf == NULL) {
1,428,691!
178
    qError("failed to allocate memory for output buffer, size:%zu", *outBufSize);
×
179
    return terrno;
×
180
  }
181

182
  char *pBuf = *outBuf;
1,428,691✔
183
  pResultRow->version = FUNCTION_RESULT_INFO_VERSION;
1,428,691✔
184
  (void)memcpy(pBuf, pResultRow, sizeof(SResultRow));
1,428,691✔
185
  pBuf += sizeof(SResultRow);
1,428,691✔
186
  for (int32_t i = 0; i < pSup->numOfExprs; ++i) {
5,782,865✔
187
    size_t len = sizeof(SResultRowEntryInfo) + pCtx[i].resDataInfo.interBufSize;
4,354,175✔
188
    *(int32_t *) pBuf = (int32_t)len;
4,354,175✔
189
    pBuf += sizeof(int32_t);
4,354,175✔
190
    (void)memcpy(pBuf, getResultEntryInfo(pResultRow, i, offset), len);
4,354,175✔
191
    pBuf += len;
4,354,174✔
192
  }
193

194
  if (rowSize < inBufSize) {
1,428,690✔
195
    // stream stores extra data after result row
196
    size_t leftLen = inBufSize - rowSize;
1,294✔
197
    (void)memcpy(pBuf, inBuf + rowSize, leftLen);
1,294✔
198
    pBuf += leftLen;
1,294✔
199
  }
200
  return TSDB_CODE_SUCCESS;
1,428,690✔
201
}
202

203
static void freeEx(void* p) { taosMemoryFree(*(void**)p); }
×
204

205
void cleanupGroupResInfo(SGroupResInfo* pGroupResInfo) {
4,918,149✔
206
  taosMemoryFreeClear(pGroupResInfo->pBuf);
4,918,149!
207
  if (pGroupResInfo->freeItem) {
4,919,240!
208
    //    taosArrayDestroy(pGroupResInfo->pRows);
209
    taosArrayDestroyEx(pGroupResInfo->pRows, freeEx);
×
210
    pGroupResInfo->freeItem = false;
×
211
    pGroupResInfo->pRows = NULL;
×
212
  } else {
213
    taosArrayDestroy(pGroupResInfo->pRows);
4,919,240✔
214
    pGroupResInfo->pRows = NULL;
4,920,145✔
215
  }
216
  pGroupResInfo->index = 0;
4,920,145✔
217
  pGroupResInfo->delIndex = 0;
4,920,145✔
218
}
4,920,145✔
219

220
int32_t resultrowComparAsc(const void* p1, const void* p2) {
2,142,192,130✔
221
  SResKeyPos* pp1 = *(SResKeyPos**)p1;
2,142,192,130✔
222
  SResKeyPos* pp2 = *(SResKeyPos**)p2;
2,142,192,130✔
223

224
  if (pp1->groupId == pp2->groupId) {
2,142,192,130✔
225
    int64_t pts1 = *(int64_t*)pp1->key;
1,953,294,055✔
226
    int64_t pts2 = *(int64_t*)pp2->key;
1,953,294,055✔
227

228
    if (pts1 == pts2) {
1,953,294,055!
229
      return 0;
×
230
    } else {
231
      return pts1 < pts2 ? -1 : 1;
1,953,294,055✔
232
    }
233
  } else {
234
    return pp1->groupId < pp2->groupId ? -1 : 1;
188,898,075✔
235
  }
236
}
237

238
static int32_t resultrowComparDesc(const void* p1, const void* p2) { return resultrowComparAsc(p2, p1); }
76,301,148✔
239

240
int32_t initGroupedResultInfo(SGroupResInfo* pGroupResInfo, SSHashObj* pHashmap, int32_t order) {
4,004,211✔
241
  int32_t code = TSDB_CODE_SUCCESS;
4,004,211✔
242
  int32_t lino = 0;
4,004,211✔
243
  if (pGroupResInfo->pRows != NULL) {
4,004,211✔
244
    taosArrayDestroy(pGroupResInfo->pRows);
49,902✔
245
  }
246
  if (pGroupResInfo->pBuf) {
4,004,211✔
247
    taosMemoryFree(pGroupResInfo->pBuf);
49,902!
248
    pGroupResInfo->pBuf = NULL;
49,902✔
249
  }
250

251
  // extract the result rows information from the hash map
252
  int32_t size = tSimpleHashGetSize(pHashmap);
4,004,211✔
253

254
  void* pData = NULL;
4,004,408✔
255
  pGroupResInfo->pRows = taosArrayInit(size, POINTER_BYTES);
4,004,408✔
256
  QUERY_CHECK_NULL(pGroupResInfo->pRows, code, lino, _end, terrno);
4,005,180!
257

258
  size_t  keyLen = 0;
4,005,180✔
259
  int32_t iter = 0;
4,005,180✔
260
  int64_t bufLen = 0, offset = 0;
4,005,180✔
261

262
  // todo move away and record this during create window
263
  while ((pData = tSimpleHashIterate(pHashmap, pData, &iter)) != NULL) {
235,442,935✔
264
    /*void* key = */ (void)tSimpleHashGetKey(pData, &keyLen);
265
    bufLen += keyLen + sizeof(SResultRowPosition);
231,437,755✔
266
  }
267

268
  pGroupResInfo->pBuf = taosMemoryMalloc(bufLen);
3,978,070!
269
  QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
4,005,546!
270

271
  iter = 0;
4,005,546✔
272
  while ((pData = tSimpleHashIterate(pHashmap, pData, &iter)) != NULL) {
235,097,743✔
273
    void* key = tSimpleHashGetKey(pData, &keyLen);
231,125,450✔
274

275
    SResKeyPos* p = (SResKeyPos*)(pGroupResInfo->pBuf + offset);
231,125,450✔
276

277
    p->groupId = *(uint64_t*)key;
231,125,450✔
278
    p->pos = *(SResultRowPosition*)pData;
231,125,450✔
279
    memcpy(p->key, (char*)key + sizeof(uint64_t), keyLen - sizeof(uint64_t));
231,125,450✔
280
    void* tmp = taosArrayPush(pGroupResInfo->pRows, &p);
231,125,450✔
281
    QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
231,092,197!
282

283
    offset += keyLen + sizeof(struct SResultRowPosition);
231,092,197✔
284
  }
285

286
  if (order == TSDB_ORDER_ASC || order == TSDB_ORDER_DESC) {
3,949,002✔
287
    __compar_fn_t fn = (order == TSDB_ORDER_ASC) ? resultrowComparAsc : resultrowComparDesc;
1,662,484✔
288
    size = POINTER_BYTES;
1,662,484✔
289
    taosSort(pGroupResInfo->pRows->pData, taosArrayGetSize(pGroupResInfo->pRows), size, fn);
1,662,484✔
290
  }
291

292
  pGroupResInfo->index = 0;
4,005,388✔
293

294
_end:
4,005,388✔
295
  if (code != TSDB_CODE_SUCCESS) {
4,005,388!
296
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
297
  }
298
  return code;
4,004,869✔
299
}
300

301
void initMultiResInfoFromArrayList(SGroupResInfo* pGroupResInfo, SArray* pArrayList) {
36,237✔
302
  if (pGroupResInfo->pRows != NULL) {
36,237✔
303
    taosArrayDestroy(pGroupResInfo->pRows);
32,162✔
304
  }
305

306
  pGroupResInfo->freeItem = true;
36,237✔
307
  pGroupResInfo->pRows = pArrayList;
36,237✔
308
  pGroupResInfo->index = 0;
36,237✔
309
  pGroupResInfo->delIndex = 0;
36,237✔
310
}
36,237✔
311

312
bool hasRemainResults(SGroupResInfo* pGroupResInfo) {
12,138,559✔
313
  if (pGroupResInfo->pRows == NULL) {
12,138,559✔
314
    return false;
1,550✔
315
  }
316

317
  return pGroupResInfo->index < taosArrayGetSize(pGroupResInfo->pRows);
12,137,009✔
318
}
319

320
int32_t getNumOfTotalRes(SGroupResInfo* pGroupResInfo) {
7,813,063✔
321
  if (pGroupResInfo->pRows == 0) {
7,813,063✔
322
    return 0;
2,519✔
323
  }
324

325
  return (int32_t)taosArrayGetSize(pGroupResInfo->pRows);
7,810,544✔
326
}
327

328
SArray* createSortInfo(SNodeList* pNodeList) {
2,303,663✔
329
  size_t numOfCols = 0;
2,303,663✔
330

331
  if (pNodeList != NULL) {
2,303,663✔
332
    numOfCols = LIST_LENGTH(pNodeList);
2,302,276!
333
  } else {
334
    numOfCols = 0;
1,387✔
335
  }
336

337
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
2,303,663✔
338
  if (pList == NULL) {
2,303,670!
339
    return pList;
×
340
  }
341

342
  for (int32_t i = 0; i < numOfCols; ++i) {
4,759,586✔
343
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
2,455,830✔
344
    if (!pSortKey) {
2,455,853✔
345
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
24!
346
      taosArrayDestroy(pList);
24✔
347
      pList = NULL;
×
348
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
349
      break;
×
350
    }
351
    SBlockOrderInfo   bi = {0};
2,455,829✔
352
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
2,455,829✔
353
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);
2,455,829✔
354

355
    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
2,455,829✔
356
    bi.slotId = pColNode->slotId;
2,455,829✔
357
    void* tmp = taosArrayPush(pList, &bi);
2,455,916✔
358
    if (!tmp) {
2,455,916!
359
      taosArrayDestroy(pList);
×
360
      pList = NULL;
×
361
      break;
×
362
    }
363
  }
364

365
  return pList;
2,303,756✔
366
}
367

368
SSDataBlock* createDataBlockFromDescNode(SDataBlockDescNode* pNode) {
21,892,197✔
369
  int32_t      numOfCols = LIST_LENGTH(pNode->pSlots);
21,892,197!
370
  SSDataBlock* pBlock = NULL;
21,892,197✔
371
  int32_t      code = createDataBlock(&pBlock);
21,892,197✔
372
  if (code) {
21,899,790!
373
    terrno = code;
×
374
    return NULL;
×
375
  }
376

377
  pBlock->info.id.blockId = pNode->dataBlockId;
21,902,471✔
378
  pBlock->info.type = STREAM_INVALID;
21,902,471✔
379
  pBlock->info.calWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
21,902,471✔
380
  pBlock->info.watermark = INT64_MIN;
21,902,471✔
381

382
  for (int32_t i = 0; i < numOfCols; ++i) {
119,168,360✔
383
    SSlotDescNode*  pDescNode = (SSlotDescNode*)nodesListGetNode(pNode->pSlots, i);
97,242,564✔
384
    if (!pDescNode) {
97,189,095!
385
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
386
      blockDataDestroy(pBlock);
×
387
      pBlock = NULL;
×
388
      terrno = TSDB_CODE_INVALID_PARA;
×
389
      break;
×
390
    }
391
    SColumnInfoData idata =
392
        createColumnInfoData(pDescNode->dataType.type, pDescNode->dataType.bytes, pDescNode->slotId);
97,189,095✔
393
    idata.info.scale = pDescNode->dataType.scale;
97,425,085✔
394
    idata.info.precision = pDescNode->dataType.precision;
97,425,085✔
395
    idata.info.noData = pDescNode->reserve;
97,425,085✔
396

397
    code = blockDataAppendColInfo(pBlock, &idata);
97,425,085✔
398
    if (code != TSDB_CODE_SUCCESS) {
97,265,889!
399
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
400
      blockDataDestroy(pBlock);
×
401
      pBlock = NULL;
×
402
      terrno = code;
×
403
      break;
×
404
    }
405
  }
406

407
  return pBlock;
21,925,796✔
408
}
409

410
int32_t prepareDataBlockBuf(SSDataBlock* pDataBlock, SColMatchInfo* pMatchInfo) {
5,272,506✔
411
  SDataBlockInfo* pBlockInfo = &pDataBlock->info;
5,272,506✔
412

413
  for (int32_t i = 0; i < taosArrayGetSize(pMatchInfo->pList); ++i) {
30,116,462✔
414
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
25,335,956✔
415
    if (!pItem) {
25,346,205✔
416
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
1,711!
417
      return terrno;
1,711✔
418
    }
419

420
    if (pItem->isPk) {
25,344,494✔
421
      SColumnInfoData* pInfoData = taosArrayGet(pDataBlock->pDataBlock, pItem->dstSlotId);
500,538✔
422
      if (!pInfoData) {
500,555!
423
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
424
        return terrno;
×
425
      }
426
      pBlockInfo->pks[0].type = pInfoData->info.type;
500,555✔
427
      pBlockInfo->pks[1].type = pInfoData->info.type;
500,555✔
428

429
      // allocate enough buffer size, which is pInfoData->info.bytes
430
      if (IS_VAR_DATA_TYPE(pItem->dataType.type)) {
500,555!
431
        pBlockInfo->pks[0].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
27,206!
432
        if (pBlockInfo->pks[0].pData == NULL) {
27,114!
433
          return terrno;
×
434
        }
435

436
        pBlockInfo->pks[1].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
27,114!
437
        if (pBlockInfo->pks[1].pData == NULL) {
27,120!
438
          taosMemoryFreeClear(pBlockInfo->pks[0].pData);
×
439
          return terrno;
×
440
        }
441

442
        pBlockInfo->pks[0].nData = pInfoData->info.bytes;
27,120✔
443
        pBlockInfo->pks[1].nData = pInfoData->info.bytes;
27,120✔
444
      }
445

446
      break;
500,469✔
447
    }
448
  }
449

450
  return TSDB_CODE_SUCCESS;
5,271,563✔
451
}
452

453
EDealRes doTranslateTagExpr(SNode** pNode, void* pContext) {
6,594✔
454
  STransTagExprCtx* pCtx = pContext;
6,594✔
455
  SMetaReader*      mr = pCtx->pReader;
6,594✔
456
  bool              isTagCol = false, isTbname = false;
6,594✔
457
  if (nodeType(*pNode) == QUERY_NODE_COLUMN) {
6,594✔
458
    SColumnNode* pCol = (SColumnNode*)*pNode;
1,897✔
459
    if (pCol->colType == COLUMN_TYPE_TBNAME)
1,897!
460
      isTbname = true;
×
461
    else
462
      isTagCol = true;
1,897✔
463
  } else if (nodeType(*pNode) == QUERY_NODE_FUNCTION) {
4,697!
464
    SFunctionNode* pFunc = (SFunctionNode*)*pNode;
×
465
    if (pFunc->funcType == FUNCTION_TYPE_TBNAME) isTbname = true;
×
466
  }
467
  if (isTagCol) {
6,594✔
468
    SColumnNode* pSColumnNode = *(SColumnNode**)pNode;
1,897✔
469

470
    SValueNode* res = NULL;
1,897✔
471
    pCtx->code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
1,897✔
472
    if (NULL == res) {
1,897!
473
      return DEAL_RES_ERROR;
×
474
    }
475

476
    res->translate = true;
1,897✔
477
    res->node.resType = pSColumnNode->node.resType;
1,897✔
478

479
    STagVal tagVal = {0};
1,897✔
480
    tagVal.cid = pSColumnNode->colId;
1,897✔
481
    const char* p = mr->pAPI->extractTagVal(mr->me.ctbEntry.pTags, pSColumnNode->node.resType.type, &tagVal);
1,897✔
482
    if (p == NULL) {
1,897!
483
      res->node.resType.type = TSDB_DATA_TYPE_NULL;
×
484
    } else if (pSColumnNode->node.resType.type == TSDB_DATA_TYPE_JSON) {
1,897!
485
      int32_t len = ((const STag*)p)->len;
×
486
      res->datum.p = taosMemoryCalloc(len + 1, 1);
×
487
      if (NULL == res->datum.p) {
×
488
        return DEAL_RES_ERROR;
×
489
      }
490
      memcpy(res->datum.p, p, len);
×
491
    } else if (IS_VAR_DATA_TYPE(pSColumnNode->node.resType.type)) {
1,897!
492
      res->datum.p = taosMemoryCalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1, 1);
1,874!
493
      if (NULL == res->datum.p) {
1,874!
494
        return DEAL_RES_ERROR;
×
495
      }
496
      memcpy(varDataVal(res->datum.p), tagVal.pData, tagVal.nData);
1,874✔
497
      varDataSetLen(res->datum.p, tagVal.nData);
1,874✔
498
    } else {
499
      int32_t code = nodesSetValueNodeValue(res, &(tagVal.i64));
23✔
500
      if (code != TSDB_CODE_SUCCESS) {
23!
501
        return DEAL_RES_ERROR;
×
502
      }
503
    }
504
    nodesDestroyNode(*pNode);
1,897✔
505
    *pNode = (SNode*)res;
1,897✔
506
  } else if (isTbname) {
4,697!
507
    SValueNode* res = NULL;
×
508
    pCtx->code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
×
509
    if (NULL == res) {
×
510
      return DEAL_RES_ERROR;
×
511
    }
512

513
    res->translate = true;
×
514
    res->node.resType = ((SExprNode*)(*pNode))->resType;
×
515

516
    int32_t len = strlen(mr->me.name);
×
517
    res->datum.p = taosMemoryCalloc(len + VARSTR_HEADER_SIZE + 1, 1);
×
518
    if (NULL == res->datum.p) {
×
519
      return DEAL_RES_ERROR;
×
520
    }
521
    memcpy(varDataVal(res->datum.p), mr->me.name, len);
×
522
    varDataSetLen(res->datum.p, len);
×
523
    nodesDestroyNode(*pNode);
×
524
    *pNode = (SNode*)res;
×
525
  }
526

527
  return DEAL_RES_CONTINUE;
6,594✔
528
}
529

530
int32_t isQualifiedTable(STableKeyInfo* info, SNode* pTagCond, void* metaHandle, bool* pQualified, SStorageAPI* pAPI) {
946✔
531
  int32_t     code = TSDB_CODE_SUCCESS;
946✔
532
  SMetaReader mr = {0};
946✔
533

534
  pAPI->metaReaderFn.initReader(&mr, metaHandle, META_READER_LOCK, &pAPI->metaFn);
946✔
535
  code = pAPI->metaReaderFn.getEntryGetUidCache(&mr, info->uid);
946✔
536
  if (TSDB_CODE_SUCCESS != code) {
946!
537
    pAPI->metaReaderFn.clearReader(&mr);
×
538
    *pQualified = false;
×
539

540
    return TSDB_CODE_SUCCESS;
×
541
  }
542

543
  SNode* pTagCondTmp = NULL;
946✔
544
  code = nodesCloneNode(pTagCond, &pTagCondTmp);
946✔
545
  if (TSDB_CODE_SUCCESS != code) {
946!
546
    *pQualified = false;
×
547
    pAPI->metaReaderFn.clearReader(&mr);
×
548
    return code;
×
549
  }
550
  STransTagExprCtx ctx = {.code = 0, .pReader = &mr};
946✔
551
  nodesRewriteExprPostOrder(&pTagCondTmp, doTranslateTagExpr, &ctx);
946✔
552
  pAPI->metaReaderFn.clearReader(&mr);
946✔
553
  if (TSDB_CODE_SUCCESS != ctx.code) {
946!
554
    *pQualified = false;
×
555
    terrno = code;
×
556
    return code;
×
557
  }
558

559
  SNode* pNew = NULL;
946✔
560
  code = scalarCalculateConstants(pTagCondTmp, &pNew);
946✔
561
  if (TSDB_CODE_SUCCESS != code) {
946!
562
    terrno = code;
×
563
    nodesDestroyNode(pTagCondTmp);
×
564
    *pQualified = false;
×
565

566
    return code;
×
567
  }
568

569
  SValueNode* pValue = (SValueNode*)pNew;
946✔
570
  *pQualified = pValue->datum.b;
946✔
571

572
  nodesDestroyNode(pNew);
946✔
573
  return TSDB_CODE_SUCCESS;
946✔
574
}
575

576
static EDealRes getColumn(SNode** pNode, void* pContext) {
137,332,295✔
577
  tagFilterAssist* pData = (tagFilterAssist*)pContext;
137,332,295✔
578
  SColumnNode*     pSColumnNode = NULL;
137,332,295✔
579
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
137,332,295✔
580
    pSColumnNode = *(SColumnNode**)pNode;
3,224,471✔
581
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
134,107,824✔
582
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
451,972✔
583
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
451,972✔
584
      pData->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
439,890✔
585
      if (NULL == pSColumnNode) {
439,890!
586
        return DEAL_RES_ERROR;
×
587
      }
588
      pSColumnNode->colId = -1;
439,890✔
589
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
439,890✔
590
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
439,890✔
591
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
439,890✔
592
      nodesDestroyNode(*pNode);
439,890✔
593
      *pNode = (SNode*)pSColumnNode;
439,891✔
594
    } else {
595
      return DEAL_RES_CONTINUE;
12,082✔
596
    }
597
  } else {
598
    return DEAL_RES_CONTINUE;
133,655,852✔
599
  }
600

601
  void* data = taosHashGet(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
3,664,362✔
602
  if (!data) {
3,662,539✔
603
    int32_t tempRes =
604
        taosHashPut(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
3,065,773✔
605
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
3,068,452!
606
      return DEAL_RES_ERROR;
×
607
    }
608
    pSColumnNode->slotId = pData->index++;
3,068,452✔
609
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
3,068,452✔
610
                         .type = pSColumnNode->node.resType.type,
3,068,452✔
611
                         .bytes = pSColumnNode->node.resType.bytes,
3,068,452✔
612
                         .pk = pSColumnNode->isPk};
3,068,452✔
613
#if TAG_FILTER_DEBUG
614
    qDebug("tagfilter build column info, slotId:%d, colId:%d, type:%d", pSColumnNode->slotId, cInfo.colId, cInfo.type);
615
#endif
616
    void* tmp = taosArrayPush(pData->cInfoList, &cInfo);
3,068,452✔
617
    if (!tmp) {
3,067,126!
618
      return DEAL_RES_ERROR;
×
619
    }
620
  } else {
621
    SColumnNode* col = *(SColumnNode**)data;
596,766✔
622
    pSColumnNode->slotId = col->slotId;
596,766✔
623
  }
624

625
  return DEAL_RES_CONTINUE;
3,663,892✔
626
}
627

628
static int32_t createResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
1,391,395✔
629
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
1,391,395!
630
  if (pColumnData == NULL) {
1,392,300!
631
    return terrno;
×
632
  }
633

634
  pColumnData->info.type = pType->type;
1,392,300✔
635
  pColumnData->info.bytes = pType->bytes;
1,392,300✔
636
  pColumnData->info.scale = pType->scale;
1,392,300✔
637
  pColumnData->info.precision = pType->precision;
1,392,300✔
638

639
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
1,392,300✔
640
  if (code != TSDB_CODE_SUCCESS) {
1,392,198!
641
    terrno = code;
×
642
    releaseColInfoData(pColumnData);
×
643
    return terrno;
×
644
  }
645

646
  pParam->columnData = pColumnData;
1,392,220✔
647
  pParam->colAlloced = true;
1,392,220✔
648
  return TSDB_CODE_SUCCESS;
1,392,220✔
649
}
650

651
static void releaseColInfoData(void* pCol) {
78,274✔
652
  if (pCol) {
78,274!
653
    SColumnInfoData* col = (SColumnInfoData*)pCol;
78,274✔
654
    colDataDestroy(col);
78,274✔
655
    taosMemoryFree(col);
78,303!
656
  }
657
}
78,306✔
658

659
void freeItem(void* p) {
3,300,343✔
660
  STUidTagInfo* pInfo = p;
3,300,343✔
661
  if (pInfo->pTagVal != NULL) {
3,300,343✔
662
    taosMemoryFree(pInfo->pTagVal);
3,128,762!
663
  }
664
}
3,301,515✔
665

666
static int32_t genTagFilterDigest(const SNode* pTagCond, T_MD5_CTX* pContext) {
×
667
  if (pTagCond == NULL) {
×
668
    return TSDB_CODE_SUCCESS;
×
669
  }
670

671
  char*   payload = NULL;
×
672
  int32_t len = 0;
×
673
  int32_t code = nodesNodeToMsg(pTagCond, &payload, &len);
×
674
  if (code != TSDB_CODE_SUCCESS) {
×
675
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
676
    return code;
×
677
  }
678

679
  tMD5Init(pContext);
×
680
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
×
681
  tMD5Final(pContext);
×
682

683
  taosMemoryFree(payload);
×
684
  return TSDB_CODE_SUCCESS;
×
685
}
686

687
static int32_t genTbGroupDigest(const SNode* pGroup, uint8_t* filterDigest, T_MD5_CTX* pContext) {
×
688
  int32_t code = TSDB_CODE_SUCCESS;
×
689
  int32_t lino = 0;
×
690
  char*   payload = NULL;
×
691
  int32_t len = 0;
×
692
  code = nodesNodeToMsg(pGroup, &payload, &len);
×
693
  QUERY_CHECK_CODE(code, lino, _end);
×
694

695
  if (filterDigest[0]) {
×
696
    payload = taosMemoryRealloc(payload, len + tListLen(pContext->digest));
×
697
    QUERY_CHECK_NULL(payload, code, lino, _end, terrno);
×
698
    memcpy(payload + len, filterDigest + 1, tListLen(pContext->digest));
×
699
    len += tListLen(pContext->digest);
×
700
  }
701

702
  tMD5Init(pContext);
×
703
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
×
704
  tMD5Final(pContext);
×
705

706
_end:
×
707
  taosMemoryFree(payload);
×
708
  if (code != TSDB_CODE_SUCCESS) {
×
709
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
710
  }
711
  return code;
×
712
}
713

714
int32_t getColInfoResultForGroupby(void* pVnode, SNodeList* group, STableListInfo* pTableListInfo, uint8_t* digest,
66,066✔
715
                                   SStorageAPI* pAPI, bool initRemainGroups) {
716
  int32_t      code = TSDB_CODE_SUCCESS;
66,066✔
717
  int32_t      lino = 0;
66,066✔
718
  SArray*      pBlockList = NULL;
66,066✔
719
  SSDataBlock* pResBlock = NULL;
66,066✔
720
  void*        keyBuf = NULL;
66,066✔
721
  SArray*      groupData = NULL;
66,066✔
722
  SArray*      pUidTagList = NULL;
66,066✔
723
  SArray*      tableList = NULL;
66,066✔
724

725
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
66,066✔
726
  if (rows == 0) {
66,093!
727
    return TSDB_CODE_SUCCESS;
×
728
  }
729

730
  tagFilterAssist ctx = {0};
66,093✔
731
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
66,093✔
732
  if (ctx.colHash == NULL) {
66,100!
733
    code = terrno;
×
734
    goto end;
×
735
  }
736

737
  ctx.index = 0;
66,100✔
738
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
66,100✔
739
  if (ctx.cInfoList == NULL) {
66,122!
740
    code = terrno;
×
741
    goto end;
×
742
  }
743

744
  SNode* pNode = NULL;
66,122✔
745
  FOREACH(pNode, group) {
144,403!
746
    nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
78,271✔
747
    if (TSDB_CODE_SUCCESS != ctx.code) {
78,281!
748
      code = ctx.code;
×
749
      goto end;
×
750
    }
751
    REPLACE_NODE(pNode);
78,281✔
752
  }
753

754
  T_MD5_CTX context = {0};
66,132✔
755
  if (tsTagFilterCache) {
66,132!
756
    SNodeListNode* listNode = NULL;
×
757
    code = nodesMakeNode(QUERY_NODE_NODE_LIST, (SNode**)&listNode);
×
758
    if (TSDB_CODE_SUCCESS != code) {
×
759
      goto end;
×
760
    }
761
    listNode->pNodeList = group;
×
762
    code = genTbGroupDigest((SNode*)listNode, digest, &context);
×
763
    QUERY_CHECK_CODE(code, lino, end);
×
764

765
    nodesFree(listNode);
×
766

767
    code = pAPI->metaFn.metaGetCachedTbGroup(pVnode, pTableListInfo->idInfo.suid, context.digest,
×
768
                                             tListLen(context.digest), &tableList);
769
    QUERY_CHECK_CODE(code, lino, end);
×
770

771
    if (tableList) {
×
772
      taosArrayDestroy(pTableListInfo->pTableList);
×
773
      pTableListInfo->pTableList = tableList;
×
774
      qDebug("retrieve tb group list from cache, numOfTables:%d",
×
775
             (int32_t)taosArrayGetSize(pTableListInfo->pTableList));
776
      goto end;
×
777
    }
778
  }
779

780
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
66,132✔
781
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
66,062!
782

783
  for (int32_t i = 0; i < rows; ++i) {
328,280✔
784
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
262,090✔
785
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
262,011!
786
    STUidTagInfo   info = {.uid = pkeyInfo->uid};
262,011✔
787
    void*          tmp = taosArrayPush(pUidTagList, &info);
262,209✔
788
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
262,209!
789
  }
790

791
  code = pAPI->metaFn.getTableTags(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
66,190✔
792
  if (code != TSDB_CODE_SUCCESS) {
66,136!
793
    goto end;
×
794
  }
795

796
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
66,136✔
797
  pResBlock = createTagValBlockForFilter(ctx.cInfoList, numOfTables, pUidTagList, pVnode, pAPI);
66,127✔
798
  if (pResBlock == NULL) {
66,099!
799
    code = terrno;
×
800
    goto end;
×
801
  }
802

803
  //  int64_t st1 = taosGetTimestampUs();
804
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
805

806
  pBlockList = taosArrayInit(2, POINTER_BYTES);
66,099✔
807
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
66,082!
808

809
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
66,103✔
810
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
66,103!
811

812
  groupData = taosArrayInit(2, POINTER_BYTES);
66,103✔
813
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
66,126✔
814

815
  FOREACH(pNode, group) {
144,359!
816
    SScalarParam output = {0};
78,232✔
817

818
    switch (nodeType(pNode)) {
78,232!
819
      case QUERY_NODE_VALUE:
×
820
        break;
×
821
      case QUERY_NODE_COLUMN:
78,232✔
822
      case QUERY_NODE_OPERATOR:
823
      case QUERY_NODE_FUNCTION: {
824
        SExprNode* expNode = (SExprNode*)pNode;
78,232✔
825
        code = createResultData(&expNode->resType, rows, &output);
78,232✔
826
        if (code != TSDB_CODE_SUCCESS) {
78,291!
827
          goto end;
×
828
        }
829
        break;
78,291✔
830
      }
831

832
      default:
×
833
        code = TSDB_CODE_OPS_NOT_SUPPORT;
×
834
        goto end;
×
835
    }
836

837
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
78,291✔
838
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
73,950✔
839
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
73,950✔
840
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
73,901!
841
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
73,901✔
842
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
4,341!
843
      continue;
×
844
    } else {
845
      code = scalarCalculate(pNode, pBlockList, &output);
4,341✔
846
    }
847

848
    if (code != TSDB_CODE_SUCCESS) {
78,261!
849
      releaseColInfoData(output.columnData);
×
850
      goto end;
×
851
    }
852

853
    void* tmp = taosArrayPush(groupData, &output.columnData);
78,264✔
854
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
78,264!
855
  }
856

857
  int32_t keyLen = 0;
66,127✔
858
  SNode*  node;
859
  FOREACH(node, group) {
144,403✔
860
    SExprNode* pExpr = (SExprNode*)node;
78,276✔
861
    keyLen += pExpr->resType.bytes;
78,276✔
862
  }
863

864
  int32_t nullFlagSize = sizeof(int8_t) * LIST_LENGTH(group);
66,127✔
865
  keyLen += nullFlagSize;
66,127✔
866

867
  keyBuf = taosMemoryCalloc(1, keyLen);
66,127✔
868
  if (keyBuf == NULL) {
66,109!
869
    code = terrno;
×
870
    goto end;
×
871
  }
872

873
  if (initRemainGroups) {
66,109✔
874
    pTableListInfo->remainGroups =
13,118✔
875
        taosHashInit(rows, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
13,141✔
876
    if (pTableListInfo->remainGroups == NULL) {
13,118!
877
      code = terrno;
×
878
      goto end;
×
879
    }
880
  }
881

882
  for (int i = 0; i < rows; i++) {
328,236✔
883
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
262,059✔
884
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
261,950!
885

886
    char* isNull = (char*)keyBuf;
261,970✔
887
    char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(group);
261,970!
888
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
553,383✔
889
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
291,423✔
890

891
      if (colDataIsNull_s(pValue, i)) {
582,858✔
892
        isNull[j] = 1;
568✔
893
      } else {
894
        isNull[j] = 0;
290,861✔
895
        char* data = colDataGetData(pValue, i);
290,861!
896
        if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
290,861✔
897
          if (tTagIsJson(data)) {
360✔
898
            code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
12✔
899
            goto end;
12✔
900
          }
901
          if (tTagIsJsonNull(data)) {
348!
902
            isNull[j] = 1;
×
903
            continue;
×
904
          }
905
          int32_t len = getJsonValueLen(data);
348✔
906
          memcpy(pStart, data, len);
348✔
907
          pStart += len;
348✔
908
        } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
290,501!
909
          if (varDataTLen(data) > pValue->info.bytes) {
259,007✔
910
            code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
4✔
911
            goto end;
4✔
912
          }
913
          memcpy(pStart, data, varDataTLen(data));
259,003✔
914
          pStart += varDataTLen(data);
259,003✔
915
        } else {
916
          memcpy(pStart, data, pValue->info.bytes);
31,494✔
917
          pStart += pValue->info.bytes;
31,494✔
918
        }
919
      }
920
    }
921

922
    int32_t len = (int32_t)(pStart - (char*)keyBuf);
261,011✔
923
    info->groupId = calcGroupId(keyBuf, len);
261,011✔
924
    if (initRemainGroups) {
262,174✔
925
      // groupId ~ table uid
926
      code = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId), &(info->uid),
76,732✔
927
                         sizeof(info->uid));
928
      if (code == TSDB_CODE_DUP_KEY) {
76,708✔
929
        code = TSDB_CODE_SUCCESS;
1,287✔
930
      }
931
      QUERY_CHECK_CODE(code, lino, end);
76,708!
932
    }
933
  }
934

935
  if (tsTagFilterCache) {
66,177!
936
    tableList = taosArrayDup(pTableListInfo->pTableList, NULL);
×
937
    QUERY_CHECK_NULL(tableList, code, lino, end, terrno);
×
938

939
    code = pAPI->metaFn.metaPutTbGroupToCache(pVnode, pTableListInfo->idInfo.suid, context.digest,
×
940
                                              tListLen(context.digest), tableList,
941
                                              taosArrayGetSize(tableList) * sizeof(STableKeyInfo));
×
942
    QUERY_CHECK_CODE(code, lino, end);
×
943
  }
944

945
  //  int64_t st2 = taosGetTimestampUs();
946
  //  qDebug("calculate tag block rows:%d, cost:%ld us", rows, st2-st1);
947

948
end:
66,177✔
949
  taosMemoryFreeClear(keyBuf);
66,193!
950
  taosHashCleanup(ctx.colHash);
66,170✔
951
  taosArrayDestroy(ctx.cInfoList);
66,116✔
952
  blockDataDestroy(pResBlock);
66,118✔
953
  taosArrayDestroy(pBlockList);
66,136✔
954
  taosArrayDestroyEx(pUidTagList, freeItem);
66,099✔
955
  taosArrayDestroyP(groupData, releaseColInfoData);
66,138✔
956
  if (code != TSDB_CODE_SUCCESS) {
66,144✔
957
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
16!
958
  }
959
  return code;
66,143✔
960
}
961

962
static int32_t nameComparFn(const void* p1, const void* p2) {
4,008✔
963
  const char* pName1 = *(const char**)p1;
4,008✔
964
  const char* pName2 = *(const char**)p2;
4,008✔
965

966
  int32_t ret = strcmp(pName1, pName2);
4,008✔
967
  if (ret == 0) {
4,008✔
968
    return 0;
16✔
969
  } else {
970
    return (ret > 0) ? 1 : -1;
3,992✔
971
  }
972
}
973

974
static SArray* getTableNameList(const SNodeListNode* pList) {
434,128✔
975
  int32_t    code = TSDB_CODE_SUCCESS;
434,128✔
976
  int32_t    lino = 0;
434,128✔
977
  int32_t    len = LIST_LENGTH(pList->pNodeList);
434,128!
978
  SListCell* cell = pList->pNodeList->pHead;
434,128✔
979

980
  SArray* pTbList = taosArrayInit(len, POINTER_BYTES);
434,128✔
981
  QUERY_CHECK_NULL(pTbList, code, lino, _end, terrno);
434,135!
982

983
  for (int i = 0; i < pList->pNodeList->length; i++) {
871,063✔
984
    SValueNode* valueNode = (SValueNode*)cell->pNode;
436,931✔
985
    if (!IS_VAR_DATA_TYPE(valueNode->node.resType.type)) {
436,931!
986
      terrno = TSDB_CODE_INVALID_PARA;
×
987
      taosArrayDestroy(pTbList);
×
988
      return NULL;
×
989
    }
990

991
    char* name = varDataVal(valueNode->datum.p);
436,931✔
992
    void* tmp = taosArrayPush(pTbList, &name);
436,928✔
993
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
436,928!
994
    cell = cell->pNext;
436,928✔
995
  }
996

997
  size_t numOfTables = taosArrayGetSize(pTbList);
434,132✔
998

999
  // order the name
1000
  taosArraySort(pTbList, nameComparFn);
434,130✔
1001

1002
  // remove the duplicates
1003
  SArray* pNewList = taosArrayInit(taosArrayGetSize(pTbList), sizeof(void*));
434,129✔
1004
  QUERY_CHECK_NULL(pNewList, code, lino, _end, terrno);
434,134!
1005
  void*   tmpTbl = taosArrayGet(pTbList, 0);
434,134✔
1006
  QUERY_CHECK_NULL(tmpTbl, code, lino, _end, terrno);
434,131!
1007
  void*   tmp = taosArrayPush(pNewList, tmpTbl);
434,132✔
1008
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
434,132!
1009

1010
  for (int32_t i = 1; i < numOfTables; ++i) {
436,930✔
1011
    char** name = taosArrayGetLast(pNewList);
2,795✔
1012
    char** nameInOldList = taosArrayGet(pTbList, i);
2,795✔
1013
    QUERY_CHECK_NULL(nameInOldList, code, lino, _end, terrno);
2,797!
1014
    if (strcmp(*name, *nameInOldList) == 0) {
2,797✔
1015
      continue;
16✔
1016
    }
1017

1018
    tmp = taosArrayPush(pNewList, nameInOldList);
2,782✔
1019
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
2,782!
1020
  }
1021

1022
_end:
434,135✔
1023
  taosArrayDestroy(pTbList);
434,135✔
1024
  if (code != TSDB_CODE_SUCCESS) {
434,134!
1025
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1026
    return NULL;
×
1027
  }
1028
  return pNewList;
434,134✔
1029
}
1030

1031
static int tableUidCompare(const void* a, const void* b) {
×
1032
  uint64_t u1 = *(uint64_t*)a;
×
1033
  uint64_t u2 = *(uint64_t*)b;
×
1034

1035
  if (u1 == u2) {
×
1036
    return 0;
×
1037
  }
1038

1039
  return u1 < u2 ? -1 : 1;
×
1040
}
1041

1042
static int32_t filterTableInfoCompare(const void* a, const void* b) {
150✔
1043
  STUidTagInfo* p1 = (STUidTagInfo*)a;
150✔
1044
  STUidTagInfo* p2 = (STUidTagInfo*)b;
150✔
1045

1046
  if (p1->uid == p2->uid) {
150!
1047
    return 0;
×
1048
  }
1049

1050
  return p1->uid < p2->uid ? -1 : 1;
150!
1051
}
1052

1053
static FilterCondType checkTagCond(SNode* cond) {
1,323,485✔
1054
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
1,323,485✔
1055
    return FILTER_NO_LOGIC;
383,497✔
1056
  }
1057
  if (nodeType(cond) != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
939,988!
1058
    return FILTER_AND;
8,845✔
1059
  }
1060
  return FILTER_OTHER;
931,143✔
1061
}
1062

1063
static int32_t optimizeTbnameInCond(void* pVnode, int64_t suid, SArray* list, SNode* cond, SStorageAPI* pAPI) {
1,323,835✔
1064
  int32_t ret = -1;
1,323,835✔
1065
  int32_t ntype = nodeType(cond);
1,323,835✔
1066

1067
  if (ntype == QUERY_NODE_OPERATOR) {
1,323,835✔
1068
    ret = optimizeTbnameInCondImpl(pVnode, list, cond, pAPI, suid);
383,468✔
1069
  }
1070

1071
  if (ntype != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
1,324,390✔
1072
    return ret;
393,998✔
1073
  }
1074

1075
  bool                 hasTbnameCond = false;
930,392✔
1076
  SLogicConditionNode* pNode = (SLogicConditionNode*)cond;
930,392✔
1077
  SNodeList*           pList = (SNodeList*)pNode->pParameterList;
930,392✔
1078

1079
  int32_t len = LIST_LENGTH(pList);
930,392!
1080
  if (len <= 0) {
930,392!
1081
    return ret;
×
1082
  }
1083

1084
  SListCell* cell = pList->pHead;
930,392✔
1085
  for (int i = 0; i < len; i++) {
2,931,059✔
1086
    if (cell == NULL) break;
2,261,396!
1087
    if (optimizeTbnameInCondImpl(pVnode, list, cell->pNode, pAPI, suid) == 0) {
2,261,396✔
1088
      hasTbnameCond = true;
261,564✔
1089
      break;
261,564✔
1090
    }
1091
    cell = cell->pNext;
2,000,667✔
1092
  }
1093

1094
  taosArraySort(list, filterTableInfoCompare);
931,227✔
1095
  taosArrayRemoveDuplicate(list, filterTableInfoCompare, NULL);
929,482✔
1096

1097
  if (hasTbnameCond) {
929,795✔
1098
    ret = pAPI->metaFn.getTableTagsByUid(pVnode, suid, list);
261,564✔
1099
  }
1100

1101
  return ret;
929,914✔
1102
}
1103

1104
// only return uid that does not contained in pExistedUidList
1105
static int32_t optimizeTbnameInCondImpl(void* pVnode, SArray* pExistedUidList, SNode* pTagCond,
2,644,025✔
1106
                                        SStorageAPI* pStoreAPI, uint64_t suid) {
1107
  if (nodeType(pTagCond) != QUERY_NODE_OPERATOR) {
2,644,025✔
1108
    return -1;
107,651✔
1109
  }
1110

1111
  SOperatorNode* pNode = (SOperatorNode*)pTagCond;
2,536,374✔
1112
  if (pNode->opType != OP_TYPE_IN) {
2,536,374✔
1113
    return -1;
1,658,355✔
1114
  }
1115

1116
  if ((pNode->pLeft != NULL && nodeType(pNode->pLeft) == QUERY_NODE_COLUMN &&
878,019!
1117
       ((SColumnNode*)pNode->pLeft)->colType == COLUMN_TYPE_TBNAME) &&
880,549✔
1118
      (pNode->pRight != NULL && nodeType(pNode->pRight) == QUERY_NODE_NODE_LIST)) {
434,133!
1119
    SNodeListNode* pList = (SNodeListNode*)pNode->pRight;
434,133✔
1120

1121
    int32_t len = LIST_LENGTH(pList->pNodeList);
434,133✔
1122
    if (len <= 0) {
434,133!
1123
      return -1;
×
1124
    }
1125

1126
    SArray*   pTbList = getTableNameList(pList);
434,133✔
1127
    int32_t   numOfTables = taosArrayGetSize(pTbList);
434,133✔
1128
    SHashObj* uHash = NULL;
434,134✔
1129

1130
    size_t numOfExisted = taosArrayGetSize(pExistedUidList);  // len > 0 means there already have uids
434,134✔
1131
    if (numOfExisted > 0) {
434,128!
1132
      uHash = taosHashInit(numOfExisted / 0.7, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
×
1133
      if (!uHash) {
×
1134
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1135
        return terrno;
×
1136
      }
1137

1138
      for (int i = 0; i < numOfExisted; i++) {
×
1139
        STUidTagInfo* pTInfo = taosArrayGet(pExistedUidList, i);
×
1140
        if (!pTInfo) {
×
1141
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1142
          return terrno;
×
1143
        }
1144
        int32_t       tempRes = taosHashPut(uHash, &pTInfo->uid, sizeof(uint64_t), &i, sizeof(i));
×
1145
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
×
1146
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
1147
          return tempRes;
×
1148
        }
1149
      }
1150
    }
1151

1152
    for (int i = 0; i < numOfTables; i++) {
871,004✔
1153
      char* name = taosArrayGetP(pTbList, i);
436,890✔
1154

1155
      uint64_t uid = 0, csuid = 0;
436,894✔
1156
      if (pStoreAPI->metaFn.getTableUidByName(pVnode, name, &uid) == 0) {
436,894✔
1157
        ETableType tbType = TSDB_TABLE_MAX;
434,134✔
1158
        if (pStoreAPI->metaFn.getTableTypeSuidByName(pVnode, name, &tbType, &csuid) == 0 && tbType == TSDB_CHILD_TABLE) {
434,134!
1159
          if (suid != csuid) {
434,114✔
1160
            continue;
20✔
1161
          }
1162
          if (NULL == uHash || taosHashGet(uHash, &uid, sizeof(uid)) == NULL) {
434,094!
1163
            STUidTagInfo s = {.uid = uid, .name = name, .pTagVal = NULL};
434,094✔
1164
            void*        tmp = taosArrayPush(pExistedUidList, &s);
434,092✔
1165
            if (!tmp) {
434,092!
1166
              return terrno;
×
1167
            }
1168
          }
1169
        } else {
1170
          taosArrayDestroy(pTbList);
22✔
1171
          taosHashCleanup(uHash);
21✔
1172
          return -1;
21✔
1173
        }
1174
      } else {
1175
        //        qWarn("failed to get tableIds from by table name: %s, reason: %s", name, tstrerror(terrno));
1176
        terrno = 0;
2,764✔
1177
      }
1178
    }
1179

1180
    taosHashCleanup(uHash);
434,114✔
1181
    taosArrayDestroy(pTbList);
434,110✔
1182
    return 0;
434,114✔
1183
  }
1184

1185
  return -1;
443,886✔
1186
}
1187

1188
SSDataBlock* createTagValBlockForFilter(SArray* pColList, int32_t numOfTables, SArray* pUidTagList, void* pVnode,
1,394,258✔
1189
                                        SStorageAPI* pStorageAPI) {
1190
  int32_t      code = TSDB_CODE_SUCCESS;
1,394,258✔
1191
  int32_t      lino = 0;
1,394,258✔
1192
  SSDataBlock* pResBlock = NULL;
1,394,258✔
1193
  code = createDataBlock(&pResBlock);
1,394,258✔
1194
  QUERY_CHECK_CODE(code, lino, _end);
1,394,548!
1195

1196
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
4,450,248✔
1197
    SColumnInfoData colInfo = {0};
3,053,816✔
1198
    void* tmp = taosArrayGet(pColList, i);
3,053,816✔
1199
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
3,052,114!
1200
    colInfo.info = *(SColumnInfo*)tmp;
3,052,114✔
1201
    code = blockDataAppendColInfo(pResBlock, &colInfo);
3,052,114✔
1202
    QUERY_CHECK_CODE(code, lino, _end);
3,055,700!
1203
  }
1204

1205
  code = blockDataEnsureCapacity(pResBlock, numOfTables);
1,392,679✔
1206
  if (code != TSDB_CODE_SUCCESS) {
1,394,987!
1207
    terrno = code;
×
1208
    blockDataDestroy(pResBlock);
×
1209
    return NULL;
×
1210
  }
1211

1212
  pResBlock->info.rows = numOfTables;
1,394,987✔
1213

1214
  int32_t numOfCols = taosArrayGetSize(pResBlock->pDataBlock);
1,394,987✔
1215

1216
  for (int32_t i = 0; i < numOfTables; i++) {
4,717,032✔
1217
    STUidTagInfo* p1 = taosArrayGet(pUidTagList, i);
3,316,325✔
1218
    QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
3,313,964✔
1219

1220
    for (int32_t j = 0; j < numOfCols; j++) {
10,065,774✔
1221
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, j);
6,743,227✔
1222
      QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
6,735,998!
1223

1224
      if (pColInfo->info.colId == -1) {  // tbname
6,738,502✔
1225
        char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
451,224✔
1226
        if (p1->name != NULL) {
451,224✔
1227
          STR_TO_VARSTR(str, p1->name);
434,095✔
1228
        } else {  // name is not retrieved during filter
1229
          code = pStorageAPI->metaFn.getTableNameByUid(pVnode, p1->uid, str);
17,129✔
1230
          QUERY_CHECK_CODE(code, lino, _end);
17,250!
1231
        }
1232

1233
        code = colDataSetVal(pColInfo, i, str, false);
451,345✔
1234
        QUERY_CHECK_CODE(code, lino, _end);
451,247!
1235
#if TAG_FILTER_DEBUG
1236
        qDebug("tagfilter uid:%ld, tbname:%s", *uid, str + 2);
1237
#endif
1238
      } else {
1239
        STagVal tagVal = {0};
6,287,278✔
1240
        tagVal.cid = pColInfo->info.colId;
6,287,278✔
1241
        if (p1->pTagVal == NULL) {
6,287,278!
1242
          colDataSetNULL(pColInfo, i);
×
1243
        } else {
1244
          const char* p = pStorageAPI->metaFn.extractTagVal(p1->pTagVal, pColInfo->info.type, &tagVal);
6,287,278✔
1245

1246
          if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
6,317,130✔
1247
            colDataSetNULL(pColInfo, i);
4,174✔
1248
          } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
6,312,956✔
1249
            code = colDataSetVal(pColInfo, i, p, false);
6,382✔
1250
            QUERY_CHECK_CODE(code, lino, _end);
6,386!
1251
          } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
7,624,479!
1252
            char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
1,320,444✔
1253
            QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,317,352!
1254
            varDataSetLen(tmp, tagVal.nData);
1,317,352✔
1255
            memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
1,317,352✔
1256
            code = colDataSetVal(pColInfo, i, tmp, false);
1,317,352✔
1257
#if TAG_FILTER_DEBUG
1258
            qDebug("tagfilter varch:%s", tmp + 2);
1259
#endif
1260
            taosMemoryFree(tmp);
1,316,585!
1261
            QUERY_CHECK_CODE(code, lino, _end);
1,317,905!
1262
          } else {
1263
            code = colDataSetVal(pColInfo, i, (const char*)&tagVal.i64, false);
4,986,130✔
1264
            QUERY_CHECK_CODE(code, lino, _end);
4,972,567!
1265
#if TAG_FILTER_DEBUG
1266
            if (pColInfo->info.type == TSDB_DATA_TYPE_INT) {
1267
              qDebug("tagfilter int:%d", *(int*)(&tagVal.i64));
1268
            } else if (pColInfo->info.type == TSDB_DATA_TYPE_DOUBLE) {
1269
              qDebug("tagfilter double:%f", *(double*)(&tagVal.i64));
1270
            }
1271
#endif
1272
          }
1273
        }
1274
      }
1275
    }
1276
  }
1277

1278
_end:
1,400,707✔
1279
  if (code != TSDB_CODE_SUCCESS) {
1,400,707!
1280
    blockDataDestroy(pResBlock);
×
1281
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1282
    terrno = code;
×
1283
    return NULL;
×
1284
  }
1285
  return pResBlock;
1,400,707✔
1286
}
1287

1288
static int32_t doSetQualifiedUid(STableListInfo* pListInfo, SArray* pUidList, const SArray* pUidTagList,
1,311,511✔
1289
                                 bool* pResultList, bool addUid) {
1290
  taosArrayClear(pUidList);
1,311,511✔
1291

1292
  STableKeyInfo info = {.uid = 0, .groupId = 0};
1,311,659✔
1293
  int32_t       numOfTables = taosArrayGetSize(pUidTagList);
1,311,659✔
1294
  for (int32_t i = 0; i < numOfTables; ++i) {
4,350,590✔
1295
    if (pResultList[i]) {
3,036,950✔
1296
      STUidTagInfo* tmpTag = (STUidTagInfo*)taosArrayGet(pUidTagList, i);
2,472,895✔
1297
      if (!tmpTag) {
2,471,160!
1298
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1299
        return terrno;
×
1300
      }
1301
      uint64_t uid = tmpTag->uid;
2,471,160✔
1302
      qDebug("tagfilter get uid:%" PRId64 ", res:%d", uid, pResultList[i]);
2,471,160✔
1303

1304
      info.uid = uid;
2,471,192✔
1305
      void* p = taosArrayPush(pListInfo->pTableList, &info);
2,471,192✔
1306
      if (p == NULL) {
2,474,207!
1307
        return terrno;
×
1308
      }
1309

1310
      if (addUid) {
2,474,207!
1311
        void* tmp = taosArrayPush(pUidList, &uid);
×
1312
        if (tmp == NULL) {
×
1313
          return terrno;
×
1314
        }
1315
      }
1316
    }
1317
  }
1318

1319
  return TSDB_CODE_SUCCESS;
1,313,640✔
1320
}
1321

1322
static int32_t copyExistedUids(SArray* pUidTagList, const SArray* pUidList) {
1,323,659✔
1323
  int32_t code = TSDB_CODE_SUCCESS;
1,323,659✔
1324
  int32_t numOfExisted = taosArrayGetSize(pUidList);
1,323,659✔
1325
  if (numOfExisted == 0) {
1,324,026✔
1326
    return code;
1,293,049✔
1327
  }
1328

1329
  for (int32_t i = 0; i < numOfExisted; ++i) {
62,687✔
1330
    uint64_t*    uid = taosArrayGet(pUidList, i);
31,711✔
1331
    if (!uid) {
31,711!
1332
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1333
      return terrno;
×
1334
    }
1335
    STUidTagInfo info = {.uid = *uid};
31,711✔
1336
    void*        tmp = taosArrayPush(pUidTagList, &info);
31,710✔
1337
    if (!tmp) {
31,710!
1338
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1339
      return code;
×
1340
    }
1341
  }
1342
  return code;
30,976✔
1343
}
1344

1345
static int32_t doFilterByTagCond(STableListInfo* pListInfo, SArray* pUidList, SNode* pTagCond, void* pVnode,
5,245,727✔
1346
                                 SIdxFltStatus status, SStorageAPI* pAPI, bool addUid, bool* listAdded) {
1347
  *listAdded = false;
5,245,727✔
1348
  if (pTagCond == NULL) {
5,245,727✔
1349
    return TSDB_CODE_SUCCESS;
3,923,106✔
1350
  }
1351

1352
  terrno = TSDB_CODE_SUCCESS;
1,322,621✔
1353

1354
  int32_t      lino = 0;
1,324,526✔
1355
  int32_t      code = TSDB_CODE_SUCCESS;
1,324,526✔
1356
  SArray*      pBlockList = NULL;
1,324,526✔
1357
  SSDataBlock* pResBlock = NULL;
1,324,526✔
1358
  SScalarParam output = {0};
1,324,526✔
1359
  SArray*      pUidTagList = NULL;
1,324,526✔
1360

1361
  tagFilterAssist ctx = {0};
1,324,526✔
1362
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
1,324,526✔
1363
  if (ctx.colHash == NULL) {
1,325,670!
1364
    code = terrno;
×
1365
    QUERY_CHECK_CODE(code, lino, end);
×
1366
  }
1367

1368
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
1,325,670✔
1369
  if (ctx.cInfoList == NULL) {
1,325,893!
1370
    code = terrno;
×
1371
    QUERY_CHECK_CODE(code, lino, end);
×
1372
  }
1373

1374
  nodesRewriteExprPostOrder(&pTagCond, getColumn, (void*)&ctx);
1,325,893✔
1375
  if (TSDB_CODE_SUCCESS != ctx.code) {
1,325,599!
1376
    terrno = code = ctx.code;
×
1377
    goto end;
×
1378
  }
1379

1380
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
1,325,599✔
1381

1382
  //  int64_t stt = taosGetTimestampUs();
1383
  pUidTagList = taosArrayInit(10, sizeof(STUidTagInfo));
1,325,599✔
1384
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
1,324,064!
1385

1386
  code = copyExistedUids(pUidTagList, pUidList);
1,324,064✔
1387
  QUERY_CHECK_CODE(code, lino, end);
1,324,449!
1388

1389
  FilterCondType condType = checkTagCond(pTagCond);
1,324,449✔
1390

1391
  int32_t filter = optimizeTbnameInCond(pVnode, pListInfo->idInfo.suid, pUidTagList, pTagCond, pAPI);
1,325,061✔
1392
  if (filter == 0) {  // tbname in filter is activated, do nothing and return
1,322,997✔
1393
    taosArrayClear(pUidList);
434,116✔
1394

1395
    int32_t numOfRows = taosArrayGetSize(pUidTagList);
434,115✔
1396
    code = taosArrayEnsureCap(pUidList, numOfRows);
434,115✔
1397
    QUERY_CHECK_CODE(code, lino, end);
434,113!
1398

1399
    for (int32_t i = 0; i < numOfRows; ++i) {
868,207✔
1400
      STUidTagInfo* pInfo = taosArrayGet(pUidTagList, i);
434,093✔
1401
      QUERY_CHECK_NULL(pInfo, code, lino, end, terrno);
434,091!
1402
      void*         tmp = taosArrayPush(pUidList, &pInfo->uid);
434,091✔
1403
      QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
434,094!
1404
    }
1405
    terrno = 0;
434,114✔
1406
  } else {
1407
    if ((condType == FILTER_NO_LOGIC || condType == FILTER_AND) && status != SFLT_NOT_INDEX) {
888,881✔
1408
      code = pAPI->metaFn.getTableTagsByUid(pVnode, pListInfo->idInfo.suid, pUidTagList);
165✔
1409
    } else {
1410
      code = pAPI->metaFn.getTableTags(pVnode, pListInfo->idInfo.suid, pUidTagList);
888,716✔
1411
    }
1412
    if (code != TSDB_CODE_SUCCESS) {
891,922!
1413
      qError("failed to get table tags from meta, reason:%s, suid:%" PRIu64, tstrerror(code), pListInfo->idInfo.suid);
×
1414
      terrno = code;
×
1415
      QUERY_CHECK_CODE(code, lino, end);
×
1416
    }
1417
  }
1418

1419
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
1,326,036✔
1420
  if (numOfTables == 0) {
1,325,092✔
1421
    goto end;
12,132✔
1422
  }
1423

1424
  pResBlock = createTagValBlockForFilter(ctx.cInfoList, numOfTables, pUidTagList, pVnode, pAPI);
1,312,960✔
1425
  if (pResBlock == NULL) {
1,313,138!
1426
    code = terrno;
×
1427
    QUERY_CHECK_CODE(code, lino, end);
×
1428
  }
1429

1430
  //  int64_t st1 = taosGetTimestampUs();
1431
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1432
  pBlockList = taosArrayInit(2, POINTER_BYTES);
1,313,138✔
1433
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
1,313,799!
1434

1435
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
1,313,347✔
1436
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
1,313,347!
1437

1438
  code = createResultData(&type, numOfTables, &output);
1,313,347✔
1439
  if (code != TSDB_CODE_SUCCESS) {
1,313,925!
1440
    terrno = code;
×
1441
    QUERY_CHECK_CODE(code, lino, end);
×
1442
  }
1443

1444
  code = scalarCalculate(pTagCond, pBlockList, &output);
1,313,925✔
1445
  if (code != TSDB_CODE_SUCCESS) {
1,313,373✔
1446
    qError("failed to calculate scalar, reason:%s", tstrerror(code));
16!
1447
    terrno = code;
16✔
1448
    QUERY_CHECK_CODE(code, lino, end);
16!
1449
  }
1450

1451
  code = doSetQualifiedUid(pListInfo, pUidList, pUidTagList, (bool*)output.columnData->pData, addUid);
1,313,357✔
1452
  if (code != TSDB_CODE_SUCCESS) {
1,313,271✔
1453
    terrno = code;
483✔
1454
    QUERY_CHECK_CODE(code, lino, end);
×
1455
  }
1456
  *listAdded = true;
1,312,788✔
1457

1458
end:
1,324,936✔
1459
  if (code != TSDB_CODE_SUCCESS) {
1,324,936✔
1460
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
16!
1461
  }
1462
  taosHashCleanup(ctx.colHash);
1,324,936✔
1463
  taosArrayDestroy(ctx.cInfoList);
1,326,007✔
1464
  blockDataDestroy(pResBlock);
1,326,149✔
1465
  taosArrayDestroy(pBlockList);
1,326,118✔
1466
  taosArrayDestroyEx(pUidTagList, freeItem);
1,326,180✔
1467

1468
  colDataDestroy(output.columnData);
1,326,245✔
1469
  taosMemoryFreeClear(output.columnData);
1,326,270!
1470
  return code;
1,326,222✔
1471
}
1472

1473
int32_t getTableList(void* pVnode, SScanPhysiNode* pScanNode, SNode* pTagCond, SNode* pTagIndexCond,
5,240,979✔
1474
                     STableListInfo* pListInfo, uint8_t* digest, const char* idstr, SStorageAPI* pStorageAPI) {
1475
  int32_t code = TSDB_CODE_SUCCESS;
5,240,979✔
1476
  int32_t lino = 0;
5,240,979✔
1477
  size_t  numOfTables = 0;
5,240,979✔
1478
  bool    listAdded = false;
5,240,979✔
1479

1480
  pListInfo->idInfo.suid = pScanNode->suid;
5,240,979✔
1481
  pListInfo->idInfo.tableType = pScanNode->tableType;
5,240,979✔
1482

1483
  SArray* pUidList = taosArrayInit(8, sizeof(uint64_t));
5,240,979✔
1484
  QUERY_CHECK_NULL(pUidList, code, lino, _error, terrno);
5,249,384✔
1485

1486
  SIdxFltStatus status = SFLT_NOT_INDEX;
5,247,152✔
1487
  if (pScanNode->tableType != TSDB_SUPER_TABLE) {
5,247,152✔
1488
    pListInfo->idInfo.uid = pScanNode->uid;
1,043,340✔
1489
    if (pStorageAPI->metaFn.isTableExisted(pVnode, pScanNode->uid)) {
1,043,340✔
1490
      void* tmp = taosArrayPush(pUidList, &pScanNode->uid);
1,041,530✔
1491
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
1,041,522!
1492
    }
1493
    code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, status, pStorageAPI, false, &listAdded);
1,043,339✔
1494
    QUERY_CHECK_CODE(code, lino, _end);
1,043,328!
1495
  } else {
1496
    T_MD5_CTX context = {0};
4,203,812✔
1497

1498
    if (tsTagFilterCache) {
4,203,812!
1499
      // try to retrieve the result from meta cache
1500
      code = genTagFilterDigest(pTagCond, &context);
×
1501
      QUERY_CHECK_CODE(code, lino, _error);
×
1502

1503
      bool acquired = false;
×
1504
      code = pStorageAPI->metaFn.getCachedTableList(pVnode, pScanNode->suid, context.digest, tListLen(context.digest),
×
1505
                                                    pUidList, &acquired);
1506
      QUERY_CHECK_CODE(code, lino, _error);
×
1507

1508
      if (acquired) {
×
1509
        digest[0] = 1;
×
1510
        memcpy(digest + 1, context.digest, tListLen(context.digest));
×
1511
        qDebug("retrieve table uid list from cache, numOfTables:%d", (int32_t)taosArrayGetSize(pUidList));
×
1512
        goto _end;
×
1513
      }
1514
    }
1515

1516
    if (!pTagCond) {  // no tag filter condition exists, let's fetch all tables of this super table
4,203,812✔
1517
      code = pStorageAPI->metaFn.getChildTableList(pVnode, pScanNode->suid, pUidList);
2,905,639✔
1518
      QUERY_CHECK_CODE(code, lino, _error);
2,913,901!
1519
    } else {
1520
      // failed to find the result in the cache, let try to calculate the results
1521
      if (pTagIndexCond) {
1,298,173✔
1522
        void* pIndex = pStorageAPI->metaFn.getInvertIndex(pVnode);
75,605✔
1523

1524
        SIndexMetaArg metaArg = {.metaEx = pVnode,
151,399✔
1525
                                 .idx = pStorageAPI->metaFn.storeGetIndexInfo(pVnode),
75,689✔
1526
                                 .ivtIdx = pIndex,
1527
                                 .suid = pScanNode->uid};
75,710✔
1528

1529
        status = SFLT_NOT_INDEX;
75,710✔
1530
        code = doFilterTag(pTagIndexCond, &metaArg, pUidList, &status, &pStorageAPI->metaFilter);
75,710✔
1531
        if (code != 0 || status == SFLT_NOT_INDEX) {  // temporarily disable it for performance sake
75,551✔
1532
          qDebug("failed to get tableIds from index, suid:%" PRIu64, pScanNode->uid);
75,330✔
1533
        } else {
1534
          qDebug("succ to get filter result, table num: %d", (int)taosArrayGetSize(pUidList));
221✔
1535
        }
1536
      }
1537
    }
1538

1539
    code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, status, pStorageAPI, tsTagFilterCache, &listAdded);
4,212,062✔
1540
    QUERY_CHECK_CODE(code, lino, _end);
4,204,387✔
1541

1542
    // let's add the filter results into meta-cache
1543
    numOfTables = taosArrayGetSize(pUidList);
4,204,371✔
1544

1545
    if (tsTagFilterCache) {
4,203,066✔
1546
      size_t size = numOfTables * sizeof(uint64_t) + sizeof(int32_t);
549✔
1547
      char*  pPayload = taosMemoryMalloc(size);
549!
1548
      QUERY_CHECK_NULL(pPayload, code, lino, _end, terrno);
×
1549

1550
      *(int32_t*)pPayload = numOfTables;
×
1551
      if (numOfTables > 0) {
×
1552
        void* tmp = taosArrayGet(pUidList, 0);
×
1553
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
1554
        memcpy(pPayload + sizeof(int32_t), tmp, numOfTables * sizeof(uint64_t));
×
1555
      }
1556

1557
      code = pStorageAPI->metaFn.putCachedTableList(pVnode, pScanNode->suid, context.digest, tListLen(context.digest),
×
1558
                                                    pPayload, size, 1);
1559
      QUERY_CHECK_CODE(code, lino, _error);
×
1560

1561
      digest[0] = 1;
×
1562
      memcpy(digest + 1, context.digest, tListLen(context.digest));
×
1563
    }
1564
  }
1565

1566
_end:
5,245,861✔
1567
  if (!listAdded) {
5,245,861✔
1568
    numOfTables = taosArrayGetSize(pUidList);
3,934,135✔
1569
    for (int i = 0; i < numOfTables; i++) {
14,368,667✔
1570
      void* tmp = taosArrayGet(pUidList, i);
10,430,782✔
1571
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
10,419,351!
1572
      STableKeyInfo info = {.uid = *(uint64_t*)tmp, .groupId = 0};
10,419,351✔
1573

1574
      void* p = taosArrayPush(pListInfo->pTableList, &info);
10,419,351✔
1575
      if (p == NULL) {
10,432,294!
1576
        taosArrayDestroy(pUidList);
×
1577
        return terrno;
×
1578
      }
1579

1580
      qTrace("tagfilter get uid:%" PRIu64 ", %s", info.uid, idstr);
10,432,294✔
1581
    }
1582
  }
1583

1584
_error:
5,249,611✔
1585
  taosArrayDestroy(pUidList);
5,249,611✔
1586
  if (code != TSDB_CODE_SUCCESS) {
5,251,703✔
1587
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
16!
1588
  }
1589
  return code;
5,248,042✔
1590
}
1591

1592
int32_t qGetTableList(int64_t suid, void* pVnode, void* node, SArray** tableList, void* pTaskInfo) {
99✔
1593
  int32_t        code = TSDB_CODE_SUCCESS;
99✔
1594
  int32_t        lino = 0;
99✔
1595
  SSubplan*      pSubplan = (SSubplan*)node;
99✔
1596
  SScanPhysiNode pNode = {0};
99✔
1597
  pNode.suid = suid;
99✔
1598
  pNode.uid = suid;
99✔
1599
  pNode.tableType = TSDB_SUPER_TABLE;
99✔
1600

1601
  STableListInfo* pTableListInfo = tableListCreate();
99✔
1602
  QUERY_CHECK_NULL(pTableListInfo, code, lino, _end, terrno);
99!
1603
  uint8_t         digest[17] = {0};
99✔
1604
  code =
1605
      getTableList(pVnode, &pNode, pSubplan ? pSubplan->pTagCond : NULL, pSubplan ? pSubplan->pTagIndexCond : NULL,
99✔
1606
                   pTableListInfo, digest, "qGetTableList", &((SExecTaskInfo*)pTaskInfo)->storageAPI);
1607
  QUERY_CHECK_CODE(code, lino, _end);
99!
1608
  *tableList = pTableListInfo->pTableList;
99✔
1609
  pTableListInfo->pTableList = NULL;
99✔
1610
  tableListDestroy(pTableListInfo);
99✔
1611

1612
_end:
99✔
1613
  if (code != TSDB_CODE_SUCCESS) {
99!
1614
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1615
  }
1616
  return code;
99✔
1617
}
1618

1619
size_t getTableTagsBufLen(const SNodeList* pGroups) {
12,331✔
1620
  size_t keyLen = 0;
12,331✔
1621

1622
  SNode* node;
1623
  FOREACH(node, pGroups) {
60,290!
1624
    SExprNode* pExpr = (SExprNode*)node;
47,959✔
1625
    keyLen += pExpr->resType.bytes;
47,959✔
1626
  }
1627

1628
  keyLen += sizeof(int8_t) * LIST_LENGTH(pGroups);
12,331!
1629
  return keyLen;
12,331✔
1630
}
1631

1632
int32_t getGroupIdFromTagsVal(void* pVnode, uint64_t uid, SNodeList* pGroupNode, char* keyBuf, uint64_t* pGroupId,
16✔
1633
                              SStorageAPI* pAPI) {
1634
  SMetaReader mr = {0};
16✔
1635

1636
  pAPI->metaReaderFn.initReader(&mr, pVnode, META_READER_LOCK, &pAPI->metaFn);
16✔
1637
  if (pAPI->metaReaderFn.getEntryGetUidCache(&mr, uid) != 0) {  // table not exist
16!
1638
    pAPI->metaReaderFn.clearReader(&mr);
×
1639
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
1640
  }
1641

1642
  SNodeList* groupNew = NULL;
16✔
1643
  int32_t    code = nodesCloneList(pGroupNode, &groupNew);
16✔
1644
  if (TSDB_CODE_SUCCESS != code) {
16!
1645
    pAPI->metaReaderFn.clearReader(&mr);
×
1646
    return code;
×
1647
  }
1648

1649
  STransTagExprCtx ctx = {.code = 0, .pReader = &mr};
16✔
1650
  nodesRewriteExprsPostOrder(groupNew, doTranslateTagExpr, &ctx);
16✔
1651
  if (TSDB_CODE_SUCCESS != ctx.code) {
16!
1652
    nodesDestroyList(groupNew);
×
1653
    pAPI->metaReaderFn.clearReader(&mr);
×
1654
    return code;
×
1655
  }
1656
  char* isNull = (char*)keyBuf;
16✔
1657
  char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(pGroupNode);
16!
1658

1659
  SNode*  pNode;
1660
  int32_t index = 0;
16✔
1661
  FOREACH(pNode, groupNew) {
32!
1662
    SNode*  pNew = NULL;
16✔
1663
    int32_t code = scalarCalculateConstants(pNode, &pNew);
16✔
1664
    if (TSDB_CODE_SUCCESS == code) {
16!
1665
      REPLACE_NODE(pNew);
16✔
1666
    } else {
1667
      nodesDestroyList(groupNew);
×
1668
      pAPI->metaReaderFn.clearReader(&mr);
×
1669
      return code;
×
1670
    }
1671

1672
    if (nodeType(pNew) != QUERY_NODE_VALUE) {
16!
1673
      nodesDestroyList(groupNew);
×
1674
      pAPI->metaReaderFn.clearReader(&mr);
×
1675
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
1676
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1677
    }
1678
    SValueNode* pValue = (SValueNode*)pNew;
16✔
1679

1680
    if (pValue->node.resType.type == TSDB_DATA_TYPE_NULL || pValue->isNull) {
16!
1681
      isNull[index++] = 1;
×
1682
      continue;
×
1683
    } else {
1684
      isNull[index++] = 0;
16✔
1685
      char* data = nodesGetValueFromNode(pValue);
16✔
1686
      if (pValue->node.resType.type == TSDB_DATA_TYPE_JSON) {
16!
1687
        if (tTagIsJson(data)) {
×
1688
          terrno = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
×
1689
          nodesDestroyList(groupNew);
×
1690
          pAPI->metaReaderFn.clearReader(&mr);
×
1691
          return terrno;
×
1692
        }
1693
        int32_t len = getJsonValueLen(data);
×
1694
        memcpy(pStart, data, len);
×
1695
        pStart += len;
×
1696
      } else if (IS_VAR_DATA_TYPE(pValue->node.resType.type)) {
16!
1697
        memcpy(pStart, data, varDataTLen(data));
4✔
1698
        pStart += varDataTLen(data);
4✔
1699
      } else {
1700
        memcpy(pStart, data, pValue->node.resType.bytes);
12✔
1701
        pStart += pValue->node.resType.bytes;
12✔
1702
      }
1703
    }
1704
  }
1705

1706
  int32_t len = (int32_t)(pStart - (char*)keyBuf);
16✔
1707
  *pGroupId = calcGroupId(keyBuf, len);
16✔
1708

1709
  nodesDestroyList(groupNew);
16✔
1710
  pAPI->metaReaderFn.clearReader(&mr);
16✔
1711
  
1712
  return TSDB_CODE_SUCCESS;
16✔
1713
}
1714

1715
SArray* makeColumnArrayFromList(SNodeList* pNodeList) {
28,158✔
1716
  if (!pNodeList) {
28,158!
1717
    return NULL;
×
1718
  }
1719

1720
  size_t  numOfCols = LIST_LENGTH(pNodeList);
28,158!
1721
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
28,158✔
1722
  if (pList == NULL) {
28,171!
1723
    return NULL;
×
1724
  }
1725

1726
  for (int32_t i = 0; i < numOfCols; ++i) {
107,487✔
1727
    SColumnNode* pColNode = (SColumnNode*)nodesListGetNode(pNodeList, i);
79,285✔
1728
    if (!pColNode) {
79,158!
1729
      taosArrayDestroy(pList);
×
1730
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
1731
      return NULL;
×
1732
    }
1733

1734
    // todo extract method
1735
    SColumn c = {0};
79,158✔
1736
    c.slotId = pColNode->slotId;
79,158✔
1737
    c.colId = pColNode->colId;
79,158✔
1738
    c.type = pColNode->node.resType.type;
79,158✔
1739
    c.bytes = pColNode->node.resType.bytes;
79,158✔
1740
    c.precision = pColNode->node.resType.precision;
79,158✔
1741
    c.scale = pColNode->node.resType.scale;
79,158✔
1742

1743
    void* tmp = taosArrayPush(pList, &c);
79,316✔
1744
    if (!tmp) {
79,316!
1745
      taosArrayDestroy(pList);
×
1746
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1747
      return NULL;
×
1748
    }
1749
  }
1750

1751
  return pList;
28,202✔
1752
}
1753

1754
int32_t extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols,
9,354,455✔
1755
                            int32_t type, SColMatchInfo* pMatchInfo) {
1756
  size_t  numOfCols = LIST_LENGTH(pNodeList);
9,354,455!
1757
  int32_t code = TSDB_CODE_SUCCESS;
9,354,455✔
1758
  int32_t lino = 0;
9,354,455✔
1759

1760
  pMatchInfo->matchType = type;
9,354,455✔
1761

1762
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchItem));
9,354,455✔
1763
  if (pList == NULL) {
9,355,122!
1764
    code = terrno;
×
1765
    return code;
×
1766
  }
1767

1768
  for (int32_t i = 0; i < numOfCols; ++i) {
48,660,193✔
1769
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
39,289,843✔
1770
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
39,254,911!
1771
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
39,261,065✔
1772
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
39,167,075✔
1773

1774
      SColMatchItem c = {.needOutput = true};
39,167,075✔
1775
      c.colId = pColNode->colId;
39,167,075✔
1776
      c.srcSlotId = pColNode->slotId;
39,167,075✔
1777
      c.dstSlotId = pNode->slotId;
39,167,075✔
1778
      c.isPk = pColNode->isPk;
39,167,075✔
1779
      c.dataType = pColNode->node.resType;
39,167,075✔
1780
      void* tmp = taosArrayPush(pList, &c);
39,210,858✔
1781
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
39,210,858!
1782
    }
1783
  }
1784

1785
  // set the output flag for each column in SColMatchInfo, according to the
1786
  *numOfOutputCols = 0;
9,370,350✔
1787
  int32_t num = LIST_LENGTH(pOutputNodeList->pSlots);
9,370,350✔
1788
  for (int32_t i = 0; i < num; ++i) {
56,268,392✔
1789
    SSlotDescNode* pNode = (SSlotDescNode*)nodesListGetNode(pOutputNodeList->pSlots, i);
46,873,195✔
1790
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
46,949,924!
1791

1792
    // todo: add reserve flag check
1793
    // it is a column reserved for the arithmetic expression calculation
1794
    if (pNode->slotId >= numOfCols) {
46,984,173✔
1795
      (*numOfOutputCols) += 1;
7,720,811✔
1796
      continue;
7,720,811✔
1797
    }
1798

1799
    SColMatchItem* info = NULL;
39,263,362✔
1800
    for (int32_t j = 0; j < taosArrayGetSize(pList); ++j) {
883,142,599✔
1801
      info = taosArrayGet(pList, j);
883,057,798✔
1802
      QUERY_CHECK_NULL(info, code, lino, _end, terrno);
883,250,369✔
1803
      if (info->dstSlotId == pNode->slotId) {
882,778,944✔
1804
        break;
38,899,707✔
1805
      }
1806
    }
1807

1808
    if (pNode->output) {
39,177,231✔
1809
      (*numOfOutputCols) += 1;
39,100,400✔
1810
    } else if (info != NULL) {
76,831!
1811
      // select distinct tbname from stb where tbname='abc';
1812
      info->needOutput = false;
76,833✔
1813
    }
1814
  }
1815

1816
  pMatchInfo->pList = pList;
9,395,197✔
1817

1818
_end:
9,395,197✔
1819
  if (code != TSDB_CODE_SUCCESS) {
9,395,197!
1820
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1821
  }
1822
  return code;
9,355,026✔
1823
}
1824

1825
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
30,694,318✔
1826
                                  const char* name) {
1827
  SResSchema s = {0};
30,694,318✔
1828
  s.scale = scale;
30,694,318✔
1829
  s.type = type;
30,694,318✔
1830
  s.bytes = bytes;
30,694,318✔
1831
  s.slotId = slotId;
30,694,318✔
1832
  s.precision = precision;
30,694,318✔
1833
  tstrncpy(s.name, name, tListLen(s.name));
30,694,318✔
1834

1835
  return s;
30,694,318✔
1836
}
1837

1838
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType, EColumnType colType) {
29,465,011✔
1839
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
29,465,011!
1840
  if (pCol == NULL) {
29,457,639!
1841
    return NULL;
×
1842
  }
1843

1844
  pCol->slotId = slotId;
29,457,639✔
1845
  pCol->colId = colId;
29,457,639✔
1846
  pCol->bytes = pType->bytes;
29,457,639✔
1847
  pCol->type = pType->type;
29,457,639✔
1848
  pCol->scale = pType->scale;
29,457,639✔
1849
  pCol->precision = pType->precision;
29,457,639✔
1850
  pCol->dataBlockId = blockId;
29,457,639✔
1851
  pCol->colType = colType;
29,457,639✔
1852
  return pCol;
29,457,639✔
1853
}
1854

1855
int32_t createExprFromOneNode(SExprInfo* pExp, SNode* pNode, int16_t slotId) {
30,673,776✔
1856
  int32_t code = TSDB_CODE_SUCCESS;
30,673,776✔
1857
  int32_t lino = 0;
30,673,776✔
1858
  pExp->base.numOfParams = 0;
30,673,776✔
1859
  pExp->base.pParam = NULL;
30,673,776✔
1860
  pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
30,673,776!
1861
  QUERY_CHECK_NULL(pExp->pExpr, code, lino, _end, terrno);
30,705,424!
1862

1863
  pExp->pExpr->_function.num = 1;
30,705,424✔
1864
  pExp->pExpr->_function.functionId = -1;
30,705,424✔
1865

1866
  int32_t type = nodeType(pNode);
30,705,424✔
1867
  // it is a project query, or group by column
1868
  if (type == QUERY_NODE_COLUMN) {
30,705,424✔
1869
    pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
18,600,164✔
1870
    SColumnNode* pColNode = (SColumnNode*)pNode;
18,600,164✔
1871

1872
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
18,600,164!
1873
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
18,598,752!
1874

1875
    pExp->base.numOfParams = 1;
18,598,752✔
1876

1877
    SDataType* pType = &pColNode->node.resType;
18,598,752✔
1878
    pExp->base.resSchema =
1879
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pColNode->colName);
18,598,752✔
1880

1881
    pExp->base.pParam[0].pCol =
37,203,378✔
1882
        createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType, pColNode->colType);
18,605,728✔
1883
    QUERY_CHECK_NULL(pExp->base.pParam[0].pCol, code, lino, _end, terrno);
18,597,650!
1884

1885
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
18,597,650✔
1886
  } else if (type == QUERY_NODE_VALUE) {
12,105,260✔
1887
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
275,496✔
1888
    SValueNode* pValNode = (SValueNode*)pNode;
275,496✔
1889

1890
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
275,496!
1891
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
275,592!
1892

1893
    pExp->base.numOfParams = 1;
275,592✔
1894

1895
    SDataType* pType = &pValNode->node.resType;
275,592✔
1896
    pExp->base.resSchema =
1897
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
275,592✔
1898
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
275,564✔
1899
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
275,564✔
1900
    QUERY_CHECK_CODE(code, lino, _end);
261,991!
1901
  } else if (type == QUERY_NODE_FUNCTION) {
11,829,764✔
1902
    pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
11,586,778✔
1903
    SFunctionNode* pFuncNode = (SFunctionNode*)pNode;
11,586,778✔
1904

1905
    SDataType* pType = &pFuncNode->node.resType;
11,586,778✔
1906
    pExp->base.resSchema =
1907
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pFuncNode->node.aliasName);
11,586,778✔
1908
    tExprNode* pExprNode = pExp->pExpr;
11,586,055✔
1909

1910
    pExprNode->_function.functionId = pFuncNode->funcId;
11,586,055✔
1911
    pExprNode->_function.pFunctNode = pFuncNode;
11,586,055✔
1912
    pExprNode->_function.functionType = pFuncNode->funcType;
11,586,055✔
1913

1914
    tstrncpy(pExprNode->_function.functionName, pFuncNode->functionName, tListLen(pExprNode->_function.functionName));
11,586,055✔
1915

1916
#if 1
1917
    // todo refactor: add the parameter for tbname function
1918
    const char* name = "tbname";
11,586,055✔
1919
    int32_t     len = strlen(name);
11,586,055✔
1920

1921
    if (!pFuncNode->pParameterList && (memcmp(pExprNode->_function.functionName, name, len) == 0) &&
11,586,055✔
1922
        pExprNode->_function.functionName[len] == 0) {
675,432!
1923
      pFuncNode->pParameterList = NULL;
675,474✔
1924
      int32_t     code = nodesMakeList(&pFuncNode->pParameterList);
675,474✔
1925
      SValueNode* res = NULL;
675,484✔
1926
      if (TSDB_CODE_SUCCESS == code) {
675,484!
1927
        code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
675,489✔
1928
      }
1929
      QUERY_CHECK_CODE(code, lino, _end);
675,394!
1930
      res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
675,394✔
1931
      code = nodesListAppend(pFuncNode->pParameterList, (SNode*)res);
675,394✔
1932
      if (code != TSDB_CODE_SUCCESS) {
675,466✔
1933
        nodesDestroyNode((SNode*)res);
2✔
1934
        res = NULL;
×
1935
      }
1936
      QUERY_CHECK_CODE(code, lino, _end);
675,464!
1937
    }
1938
#endif
1939

1940
    int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
11,586,045✔
1941

1942
    pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
11,586,045!
1943
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
11,590,484!
1944
    pExp->base.numOfParams = numOfParam;
11,590,484✔
1945

1946
    for (int32_t j = 0; j < numOfParam && TSDB_CODE_SUCCESS == code; ++j) {
25,162,005!
1947
      SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
13,570,995✔
1948
      QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
13,570,172!
1949
      if (p1->type == QUERY_NODE_COLUMN) {
13,572,740✔
1950
        SColumnNode* pcn = (SColumnNode*)p1;
10,865,625✔
1951

1952
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
10,865,625✔
1953
        pExp->base.pParam[j].pCol =
21,733,190✔
1954
            createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType, pcn->colType);
10,865,625✔
1955
        QUERY_CHECK_NULL(pExp->base.pParam[j].pCol, code, lino, _end, terrno);
10,867,565✔
1956
      } else if (p1->type == QUERY_NODE_VALUE) {
2,707,115✔
1957
        SValueNode* pvn = (SValueNode*)p1;
2,263,873✔
1958
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
2,263,873✔
1959
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
2,263,873✔
1960
        QUERY_CHECK_CODE(code, lino, _end);
2,263,495!
1961
      }
1962
    }
1963
    pExp->pExpr->_function.bindExprID = ((SExprNode*)pNode)->bindExprID;
11,591,010✔
1964
  } else if (type == QUERY_NODE_OPERATOR) {
242,986✔
1965
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
236,162✔
1966
    SOperatorNode* pOpNode = (SOperatorNode*)pNode;
236,162✔
1967

1968
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
236,162✔
1969
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
236,193!
1970
    pExp->base.numOfParams = 1;
236,193✔
1971

1972
    SDataType* pType = &pOpNode->node.resType;
236,193✔
1973
    pExp->base.resSchema =
1974
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pOpNode->node.aliasName);
236,193✔
1975
    pExp->pExpr->_optrRoot.pRootNode = pNode;
236,143✔
1976
  } else if (type == QUERY_NODE_CASE_WHEN) {
6,824✔
1977
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
5,990✔
1978
    SCaseWhenNode* pCaseNode = (SCaseWhenNode*)pNode;
5,990✔
1979

1980
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
5,990✔
1981
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
5,988!
1982
    pExp->base.numOfParams = 1;
5,988✔
1983

1984
    SDataType* pType = &pCaseNode->node.resType;
5,988✔
1985
    pExp->base.resSchema =
1986
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCaseNode->node.aliasName);
5,988✔
1987
    pExp->pExpr->_optrRoot.pRootNode = pNode;
5,990✔
1988
  } else if (type == QUERY_NODE_LOGIC_CONDITION) {
834✔
1989
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
10✔
1990
    SLogicConditionNode* pCond = (SLogicConditionNode*)pNode;
10✔
1991
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
10!
1992
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
10!
1993
    pExp->base.numOfParams = 1;
10✔
1994
    SDataType* pType = &pCond->node.resType;
10✔
1995
    pExp->base.resSchema = createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCond->node.aliasName);
10✔
1996
    pExp->pExpr->_optrRoot.pRootNode = pNode;
10✔
1997
  } else {
1998
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
824✔
1999
    QUERY_CHECK_CODE(code, lino, _end);
824!
2000
  }
2001
  pExp->pExpr->relatedTo = ((SExprNode*)pNode)->relatedTo;
30,693,618✔
2002
_end:
30,693,618✔
2003
  if (code != TSDB_CODE_SUCCESS) {
30,693,618!
2004
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2005
  }
2006
  return code;
30,695,436✔
2007
}
2008

2009
int32_t createExprFromTargetNode(SExprInfo* pExp, STargetNode* pTargetNode) {
30,634,781✔
2010
  return createExprFromOneNode(pExp, pTargetNode->pExpr, pTargetNode->slotId);
30,634,781✔
2011
}
2012

2013
SExprInfo* createExpr(SNodeList* pNodeList, int32_t* numOfExprs) {
2,718✔
2014
  *numOfExprs = LIST_LENGTH(pNodeList);
2,718!
2015
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
2,718!
2016
  if (!pExprs) {
2,719!
2017
    return NULL;
×
2018
  }
2019

2020
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
22,840✔
2021
    SExprInfo* pExp = &pExprs[i];
20,123✔
2022
    int32_t    code = createExprFromOneNode(pExp, nodesListGetNode(pNodeList, i), i + UD_TAG_COLUMN_INDEX);
20,123✔
2023
    if (code != TSDB_CODE_SUCCESS) {
20,121!
2024
      taosMemoryFreeClear(pExprs);
×
2025
      terrno = code;
×
2026
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2027
      return NULL;
×
2028
    }
2029
  }
2030

2031
  return pExprs;
2,717✔
2032
}
2033

2034
int32_t createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, SExprInfo** pExprInfo, int32_t* numOfExprs) {
11,975,971✔
2035
  QRY_PARAM_CHECK(pExprInfo);
11,975,971!
2036

2037
  int32_t code = 0;
11,975,971✔
2038
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
11,975,971✔
2039
  int32_t numOfGroupKeys = 0;
11,975,971✔
2040
  if (pGroupKeys != NULL) {
11,975,971✔
2041
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
395,812!
2042
  }
2043

2044
  *numOfExprs = numOfFuncs + numOfGroupKeys;
11,975,971✔
2045
  if (*numOfExprs == 0) {
11,975,971✔
2046
    return code;
1,629,225✔
2047
  }
2048

2049
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
10,346,746!
2050
  if (pExprs == NULL) {
10,350,251!
2051
    return terrno;
×
2052
  }
2053

2054
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
40,619,684✔
2055
    STargetNode* pTargetNode = NULL;
30,267,288✔
2056
    if (i < numOfFuncs) {
30,267,288✔
2057
      pTargetNode = (STargetNode*)nodesListGetNode(pNodeList, i);
29,612,680✔
2058
    } else {
2059
      pTargetNode = (STargetNode*)nodesListGetNode(pGroupKeys, i - numOfFuncs);
654,608✔
2060
    }
2061
    if (!pTargetNode) {
30,241,127!
2062
      destroyExprInfo(pExprs, *numOfExprs);
×
2063
      taosMemoryFreeClear(pExprs);
×
2064
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2065
      return terrno;
×
2066
    }
2067

2068
    SExprInfo* pExp = &pExprs[i];
30,241,127✔
2069
    code = createExprFromTargetNode(pExp, pTargetNode);
30,241,127✔
2070
    if (code != TSDB_CODE_SUCCESS) {
30,272,701✔
2071
      destroyExprInfo(pExprs, *numOfExprs);
3,268✔
2072
      taosMemoryFreeClear(pExprs);
×
2073
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2074
      return code;
×
2075
    }
2076
  }
2077

2078
  *pExprInfo = pExprs;
10,352,396✔
2079
  return code;
10,352,396✔
2080
}
2081

2082
static void deleteSubsidiareCtx(void* pData) {
×
2083
  SSubsidiaryResInfo* pCtx = (SSubsidiaryResInfo*)pData;
×
2084
  if (pCtx->pCtx) {
×
2085
    taosMemoryFreeClear(pCtx->pCtx);
×
2086
  }
2087
}
×
2088

2089
// set the output buffer for the selectivity + tag query
2090
static int32_t setSelectValueColumnInfo(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
11,633,177✔
2091
  int32_t num = 0;
11,633,177✔
2092
  int32_t code = TSDB_CODE_SUCCESS;
11,633,177✔
2093
  int32_t lino = 0;
11,633,177✔
2094

2095
  SArray* pValCtxArray = NULL;
11,633,177✔
2096
  for (int32_t i = numOfOutput - 1; i > 0; --i) {  // select Func is at the end of the list
31,368,473✔
2097
    int32_t funcIdx = pCtx[i].pExpr->pExpr->_function.bindExprID;
19,735,294✔
2098
    if (funcIdx > 0) {
19,735,294✔
2099
      if (pValCtxArray == NULL) {
3,493✔
2100
        // the end of the list is the select function of biggest index
2101
        pValCtxArray = taosArrayInit_s(sizeof(SSubsidiaryResInfo*), funcIdx);
2,478✔
2102
        if (pValCtxArray == NULL) {
2,480✔
2103
          return terrno;
1✔
2104
        }
2105
      }
2106
      if (funcIdx > pValCtxArray->size) {
3,494!
2107
        qError("funcIdx:%d is out of range", funcIdx);
×
2108
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2109
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2110
      }
2111
      SSubsidiaryResInfo* pSubsidiary = &pCtx[i].subsidiaries;
3,494✔
2112
      pSubsidiary->pCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
3,494!
2113
      if (pSubsidiary->pCtx == NULL) {
3,496!
2114
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2115
        return terrno;
×
2116
      }
2117
      pSubsidiary->num = 0;
3,496✔
2118
      taosArraySet(pValCtxArray, funcIdx - 1, &pSubsidiary);
3,496✔
2119
    }
2120
  }
2121

2122
  SqlFunctionCtx*  p = NULL;
11,633,179✔
2123
  SqlFunctionCtx** pValCtx = NULL;
11,633,179✔
2124
  if (pValCtxArray == NULL) {
11,633,179✔
2125
    pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
11,630,805!
2126
    if (pValCtx == NULL) {
11,636,052!
2127
      QUERY_CHECK_CODE(terrno, lino, _end);
×
2128
    }
2129
  }
2130

2131
  for (int32_t i = 0; i < numOfOutput; ++i) {
42,070,169✔
2132
    const char* pName = pCtx[i].pExpr->pExpr->_function.functionName;
30,431,137✔
2133
    if ((strcmp(pName, "_select_value") == 0)) {
30,431,137✔
2134
      if (pValCtxArray == NULL) {
1,153,564✔
2135
        pValCtx[num++] = &pCtx[i];
1,149,151✔
2136
      } else {
2137
        int32_t bindFuncIndex = pCtx[i].pExpr->pExpr->relatedTo;  // start from index 1;
4,413✔
2138
        if (bindFuncIndex > 0) {  // 0 is default index related to the select function
4,413!
2139
          bindFuncIndex -= 1;
4,712✔
2140
        }
2141
        SSubsidiaryResInfo** pSubsidiary = taosArrayGet(pValCtxArray, bindFuncIndex);
4,413✔
2142
        if(pSubsidiary == NULL) {
4,834!
2143
          QUERY_CHECK_CODE(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR, lino, _end);
×
2144
        }
2145
        (*pSubsidiary)->pCtx[(*pSubsidiary)->num] = &pCtx[i];
4,834✔
2146
        (*pSubsidiary)->num++;
4,834✔
2147
      }
2148
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
29,277,573✔
2149
       if (pValCtxArray == NULL) {
2,929,612✔
2150
        p = &pCtx[i];
2,925,761✔
2151
       }
2152
    }
2153
  }
2154

2155
  if (p != NULL) {
11,639,032✔
2156
    p->subsidiaries.pCtx = pValCtx;
2,303,532✔
2157
    p->subsidiaries.num = num;
2,303,532✔
2158
  } else {
2159
    taosMemoryFreeClear(pValCtx);
9,335,500!
2160
  }
2161

2162
_end:
6,234✔
2163
  if (code != TSDB_CODE_SUCCESS) {
11,638,716!
2164
    taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2165
    taosMemoryFreeClear(pValCtx);
×
2166
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2167
  } else {
2168
    taosArrayDestroy(pValCtxArray);
11,638,716✔
2169
  }
2170
  return code;
11,623,067✔
2171
}
2172

2173
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowEntryInfoOffset,
11,627,890✔
2174
                                     SFunctionStateStore* pStore) {
2175
  int32_t         code = TSDB_CODE_SUCCESS;
11,627,890✔
2176
  int32_t         lino = 0;
11,627,890✔
2177
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
11,627,890!
2178
  if (pFuncCtx == NULL) {
11,628,836!
2179
    return NULL;
×
2180
  }
2181

2182
  *rowEntryInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
11,628,836!
2183
  if (*rowEntryInfoOffset == 0) {
11,638,574!
2184
    taosMemoryFreeClear(pFuncCtx);
×
2185
    return NULL;
×
2186
  }
2187

2188
  for (int32_t i = 0; i < numOfOutput; ++i) {
42,087,954✔
2189
    SExprInfo* pExpr = &pExprInfo[i];
30,443,204✔
2190

2191
    SExprBasicInfo* pFunct = &pExpr->base;
30,443,204✔
2192
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
30,443,204✔
2193

2194
    pCtx->functionId = -1;
30,443,204✔
2195
    pCtx->pExpr = pExpr;
30,443,204✔
2196

2197
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
30,443,204✔
2198
      SFuncExecEnv env = {0};
11,587,681✔
2199
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;
11,587,681✔
2200
      pCtx->isPseudoFunc = fmIsWindowPseudoColumnFunc(pCtx->functionId);
11,587,681✔
2201
      pCtx->isNotNullFunc = fmIsNotNullOutputFunc(pCtx->functionId);
11,580,497✔
2202

2203
      bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
11,585,205✔
2204
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId)) {
19,578,759✔
2205
        if (!isUdaf) {
8,004,903✔
2206
          code = fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
8,004,871✔
2207
          QUERY_CHECK_CODE(code, lino, _end);
7,997,549!
2208
        } else {
2209
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
32✔
2210
          pCtx->udfName = taosStrdup(udfName);
32!
2211
          QUERY_CHECK_NULL(pCtx->udfName, code, lino, _end, terrno);
32!
2212

2213
          code = fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
32✔
2214
          QUERY_CHECK_CODE(code, lino, _end);
32!
2215
        }
2216
        bool tmp = pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
7,997,581✔
2217
        if (!tmp) {
7,996,165!
2218
          code = terrno;
×
2219
          QUERY_CHECK_CODE(code, lino, _end);
×
2220
        }
2221
      } else {
2222
        code = fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
3,573,011✔
2223
        if (code != TSDB_CODE_SUCCESS && isUdaf) {
3,573,692!
2224
          code = TSDB_CODE_SUCCESS;
262✔
2225
        }
2226
        QUERY_CHECK_CODE(code, lino, _end);
3,573,692!
2227

2228
        if (pCtx->sfp.getEnv != NULL) {
3,573,692✔
2229
          bool tmp = pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
1,916,671✔
2230
          if (!tmp) {
1,914,830✔
2231
            code = terrno;
1,684✔
2232
            QUERY_CHECK_CODE(code, lino, _end);
×
2233
          }
2234
        }
2235
      }
2236
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
11,569,159✔
2237
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
18,855,523✔
2238
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
275,542!
2239
      // for simple column, the result buffer needs to hold at least one element.
2240
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
18,858,671✔
2241
    }
2242

2243
    pCtx->input.numOfInputCols = pFunct->numOfParams;
30,424,682✔
2244
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
30,424,682✔
2245
    QUERY_CHECK_NULL(pCtx->input.pData, code, lino, _end, terrno);
30,446,010!
2246
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
30,446,010!
2247
    QUERY_CHECK_NULL(pCtx->input.pColumnDataAgg, code, lino, _end, terrno);
30,449,380!
2248

2249
    pCtx->pTsOutput = NULL;
30,449,380✔
2250
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
30,449,380✔
2251
    pCtx->resDataInfo.type = pFunct->resSchema.type;
30,449,380✔
2252
    pCtx->order = TSDB_ORDER_ASC;
30,449,380✔
2253
    pCtx->start.key = INT64_MIN;
30,449,380✔
2254
    pCtx->end.key = INT64_MIN;
30,449,380✔
2255
    pCtx->numOfParams = pExpr->base.numOfParams;
30,449,380✔
2256
    pCtx->param = pFunct->pParam;
30,449,380✔
2257
    pCtx->saveHandle.currentPage = -1;
30,449,380✔
2258
    pCtx->pStore = pStore;
30,449,380✔
2259
    pCtx->hasWindowOrGroup = false;
30,449,380✔
2260
    pCtx->needCleanup = false;
30,449,380✔
2261
  }
2262

2263
  for (int32_t i = 1; i < numOfOutput; ++i) {
31,385,713✔
2264
    (*rowEntryInfoOffset)[i] = (int32_t)((*rowEntryInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) +
19,740,963✔
2265
                                         pFuncCtx[i - 1].resDataInfo.interBufSize);
19,740,963✔
2266
  }
2267

2268
  code = setSelectValueColumnInfo(pFuncCtx, numOfOutput);
11,644,750✔
2269
  QUERY_CHECK_CODE(code, lino, _end);
11,625,651!
2270

2271
_end:
11,625,651✔
2272
  if (code != TSDB_CODE_SUCCESS) {
11,625,651!
2273
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2274
    for (int32_t i = 0; i < numOfOutput; ++i) {
×
2275
      taosMemoryFree(pFuncCtx[i].input.pData);
×
2276
      taosMemoryFree(pFuncCtx[i].input.pColumnDataAgg);
×
2277
    }
2278
    taosMemoryFreeClear(*rowEntryInfoOffset);
×
2279
    taosMemoryFreeClear(pFuncCtx);
×
2280

2281
    terrno = code;
×
2282
    return NULL;
×
2283
  }
2284
  return pFuncCtx;
11,625,651✔
2285
}
2286

2287
// NOTE: sources columns are more than the destination SSDatablock columns.
2288
// doFilter in table scan needs every column even its output is false
2289
int32_t relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray* pCols, bool outputEveryColumn) {
1,099,176✔
2290
  int32_t code = TSDB_CODE_SUCCESS;
1,099,176✔
2291
  size_t  numOfSrcCols = taosArrayGetSize(pCols);
1,099,176✔
2292

2293
  int32_t i = 0, j = 0;
1,099,287✔
2294
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
2,666,739✔
2295
    SColumnInfoData* p = taosArrayGet(pCols, i);
1,567,724✔
2296
    if (!p) {
1,567,205!
2297
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2298
      return terrno;
×
2299
    }
2300
    SColMatchItem* pmInfo = taosArrayGet(pColMatchInfo, j);
1,567,205✔
2301
    if (!pmInfo) {
1,567,100!
2302
      return terrno;
×
2303
    }
2304

2305
    if (p->info.colId == pmInfo->colId) {
1,567,273✔
2306
      SColumnInfoData* pDst = taosArrayGet(pBlock->pDataBlock, pmInfo->dstSlotId);
1,449,556✔
2307
      if (!pDst) {
1,449,327!
2308
        return terrno;
×
2309
      }
2310
      code = colDataAssign(pDst, p, pBlock->info.rows, &pBlock->info);
1,449,327✔
2311
      if (code != TSDB_CODE_SUCCESS) {
1,449,804✔
2312
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
65!
2313
        return code;
×
2314
      }
2315
      i++;
1,449,739✔
2316
      j++;
1,449,739✔
2317
    } else if (p->info.colId < pmInfo->colId) {
117,717✔
2318
      i++;
117,713✔
2319
    } else {
2320
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
4!
2321
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2322
    }
2323
  }
2324
  return code;
1,099,313✔
2325
}
2326

2327
SInterval extractIntervalInfo(const STableScanPhysiNode* pTableScanNode) {
3,247,457✔
2328
  SInterval interval = {
3,247,457✔
2329
      .interval = pTableScanNode->interval,
3,247,457✔
2330
      .sliding = pTableScanNode->sliding,
3,247,457✔
2331
      .intervalUnit = pTableScanNode->intervalUnit,
3,247,457✔
2332
      .slidingUnit = pTableScanNode->slidingUnit,
3,247,457✔
2333
      .offset = pTableScanNode->offset,
3,247,457✔
2334
      .precision = pTableScanNode->scan.node.pOutputDataBlockDesc->precision,
3,247,457✔
2335
      .timeRange = pTableScanNode->scanRange,
2336
  };
2337
  calcIntervalAutoOffset(&interval);
3,247,457✔
2338

2339
  return interval;
3,248,208✔
2340
}
2341

2342
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
805,444✔
2343
  SColumn c = {0};
805,444✔
2344

2345
  c.slotId = pColNode->slotId;
805,444✔
2346
  c.colId = pColNode->colId;
805,444✔
2347
  c.type = pColNode->node.resType.type;
805,444✔
2348
  c.bytes = pColNode->node.resType.bytes;
805,444✔
2349
  c.scale = pColNode->node.resType.scale;
805,444✔
2350
  c.precision = pColNode->node.resType.precision;
805,444✔
2351
  return c;
805,444✔
2352
}
2353

2354
int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, const STableScanPhysiNode* pTableScanNode,
5,268,630✔
2355
                               const SReadHandle* readHandle) {
2356
  pCond->order = pTableScanNode->scanSeq[0] > 0 ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
5,268,630✔
2357
  pCond->numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
5,268,630!
2358

2359
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
5,268,630!
2360
  if (!pCond->colList) {
5,278,950!
2361
    return terrno;
×
2362
  }
2363
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
5,278,950!
2364
  if (pCond->pSlotList == NULL) {
5,275,202!
2365
    taosMemoryFreeClear(pCond->colList);
×
2366
    return terrno;
×
2367
  }
2368

2369
  // TODO: get it from stable scan node
2370
  pCond->twindows = pTableScanNode->scanRange;
5,275,202✔
2371
  pCond->suid = pTableScanNode->scan.suid;
5,275,202✔
2372
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
5,275,202✔
2373
  pCond->startVersion = -1;
5,275,202✔
2374
  pCond->endVersion = -1;
5,275,202✔
2375
  pCond->skipRollup = readHandle->skipRollup;
5,275,202✔
2376

2377
  // allowed read stt file optimization mode
2378
  pCond->notLoadData = (pTableScanNode->dataRequired == FUNC_DATA_REQUIRED_NOT_LOAD) &&
10,906,098✔
2379
                       (pTableScanNode->scan.node.pConditions == NULL) && (pTableScanNode->interval == 0);
5,275,202✔
2380

2381
  int32_t j = 0;
5,275,202✔
2382
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
30,889,172✔
2383
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pTableScanNode->scan.pScanCols, i);
25,610,431✔
2384
    if (!pNode) {
25,612,686!
2385
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2386
      return terrno;
×
2387
    }
2388
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
25,613,970✔
2389
    if (pColNode->colType == COLUMN_TYPE_TAG) {
25,613,970!
2390
      continue;
×
2391
    }
2392

2393
    pCond->colList[j].type = pColNode->node.resType.type;
25,613,970✔
2394
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
25,613,970✔
2395
    pCond->colList[j].colId = pColNode->colId;
25,613,970✔
2396
    pCond->colList[j].pk = pColNode->isPk;
25,613,970✔
2397

2398
    pCond->pSlotList[j] = pNode->slotId;
25,613,970✔
2399
    j += 1;
25,613,970✔
2400
  }
2401

2402
  pCond->numOfCols = j;
5,278,741✔
2403
  return TSDB_CODE_SUCCESS;
5,278,741✔
2404
}
2405

2406
int32_t initQueryTableDataCondWithColArray(SQueryTableDataCond* pCond, SQueryTableDataCond* pOrgCond,
×
2407
                                           const SReadHandle* readHandle, SArray* colArray) {
2408
  int32_t code = TSDB_CODE_SUCCESS;
×
2409
  int32_t lino = 0;
×
2410

2411
  pCond->order = TSDB_ORDER_ASC;
×
2412
  pCond->numOfCols = (int32_t)taosArrayGetSize(colArray);
×
2413

2414
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
×
2415
  QUERY_CHECK_NULL(pCond->colList, code, lino, _return, terrno);
×
2416

2417
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
×
2418
  QUERY_CHECK_NULL(pCond->pSlotList, code, lino, _return, terrno);
×
2419

2420
  pCond->twindows = pOrgCond->twindows;
×
2421
  pCond->type = pOrgCond->type;
×
2422
  pCond->startVersion = -1;
×
2423
  pCond->endVersion = -1;
×
2424
  pCond->skipRollup = true;
×
2425
  pCond->notLoadData = false;
×
2426

2427
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
×
2428
    SColIdPair* pColPair = taosArrayGet(colArray, i);
×
2429
    QUERY_CHECK_NULL(pColPair, code, lino, _return, terrno);
×
2430

2431
    bool find = false;
×
2432
    for (int32_t j = 0; j < pOrgCond->numOfCols; ++j) {
×
2433
      if (pOrgCond->colList[j].colId == pColPair->vtbColId) {
×
2434
        pCond->colList[i].type = pOrgCond->colList[j].type;
×
2435
        pCond->colList[i].bytes = pOrgCond->colList[j].bytes;
×
2436
        pCond->colList[i].colId = pColPair->orgColId;
×
2437
        pCond->colList[i].pk = pOrgCond->colList[j].pk;
×
2438
        pCond->pSlotList[i] = i;
×
2439
        find = true;
×
2440
        break;
×
2441
      }
2442
    }
2443
    QUERY_CHECK_CONDITION(find, code, lino, _return, TSDB_CODE_NOT_FOUND);
×
2444
  }
2445

2446
  return code;
×
2447
_return:
×
2448
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(terrno));
×
2449
  taosMemoryFreeClear(pCond->colList);
×
2450
  taosMemoryFreeClear(pCond->pSlotList);
×
2451
  return code;
×
2452
}
2453

2454
void cleanupQueryTableDataCond(SQueryTableDataCond* pCond) {
12,596,968✔
2455
  taosMemoryFreeClear(pCond->colList);
12,596,968!
2456
  taosMemoryFreeClear(pCond->pSlotList);
12,597,613!
2457
}
12,597,580✔
2458

2459
int32_t convertFillType(int32_t mode) {
570,023✔
2460
  int32_t type = TSDB_FILL_NONE;
570,023✔
2461
  switch (mode) {
570,023!
2462
    case FILL_MODE_PREV:
17,442✔
2463
      type = TSDB_FILL_PREV;
17,442✔
2464
      break;
17,442✔
2465
    case FILL_MODE_NONE:
×
2466
      type = TSDB_FILL_NONE;
×
2467
      break;
×
2468
    case FILL_MODE_NULL:
32,214✔
2469
      type = TSDB_FILL_NULL;
32,214✔
2470
      break;
32,214✔
2471
    case FILL_MODE_NULL_F:
33,345✔
2472
      type = TSDB_FILL_NULL_F;
33,345✔
2473
      break;
33,345✔
2474
    case FILL_MODE_NEXT:
18,267✔
2475
      type = TSDB_FILL_NEXT;
18,267✔
2476
      break;
18,267✔
2477
    case FILL_MODE_VALUE:
151,980✔
2478
      type = TSDB_FILL_SET_VALUE;
151,980✔
2479
      break;
151,980✔
2480
    case FILL_MODE_VALUE_F:
289,160✔
2481
      type = TSDB_FILL_SET_VALUE_F;
289,160✔
2482
      break;
289,160✔
2483
    case FILL_MODE_LINEAR:
17,588✔
2484
      type = TSDB_FILL_LINEAR;
17,588✔
2485
      break;
17,588✔
2486
    case FILL_MODE_NEAR:
10,030✔
2487
      type = TSDB_FILL_NEAR;
10,030✔
2488
      break;
10,030✔
2489
    default:
×
2490
      type = TSDB_FILL_NONE;
×
2491
  }
2492

2493
  return type;
570,023✔
2494
}
2495

2496
void getInitialStartTimeWindow(SInterval* pInterval, TSKEY ts, STimeWindow* w, bool ascQuery) {
31,940,633✔
2497
  if (ascQuery) {
31,940,633✔
2498
    *w = getAlignQueryTimeWindow(pInterval, ts);
31,582,773✔
2499
  } else {
2500
    // the start position of the first time window in the endpoint that spreads beyond the queried last timestamp
2501
    *w = getAlignQueryTimeWindow(pInterval, ts);
357,860✔
2502

2503
    int64_t key = w->skey;
360,259✔
2504
    while (key < ts) {  // moving towards end
1,132,564✔
2505
      key = getNextTimeWindowStart(pInterval, key, TSDB_ORDER_ASC);
1,113,928✔
2506
      if (key > ts) {
1,113,906✔
2507
        break;
341,601✔
2508
      }
2509

2510
      w->skey = key;
772,305✔
2511
    }
2512
    w->ekey = taosTimeAdd(w->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
360,237✔
2513
  }
2514
}
31,903,596✔
2515

2516
static STimeWindow doCalculateTimeWindow(int64_t ts, SInterval* pInterval) {
1,133,436✔
2517
  STimeWindow w = {0};
1,133,436✔
2518

2519
  w.skey = taosTimeTruncate(ts, pInterval);
1,133,436✔
2520
  w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
1,133,570✔
2521
  return w;
1,133,486✔
2522
}
2523

2524
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order) {
456,896✔
2525
  STimeWindow win = *pWindow;
456,896✔
2526
  STimeWindow save = win;
456,896✔
2527
  while (win.skey <= ts && win.ekey >= ts) {
1,949,651✔
2528
    save = win;
1,492,755✔
2529
    // get previous time window
2530
    getNextTimeWindow(pInterval, &win, order == TSDB_ORDER_ASC ? TSDB_ORDER_DESC : TSDB_ORDER_ASC);
1,492,755✔
2531
  }
2532

2533
  return save;
456,896✔
2534
}
2535

2536
// get the correct time window according to the handled timestamp
2537
// todo refactor
2538
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
33,714,358✔
2539
                                int32_t order) {
2540
  STimeWindow w = {0};
33,714,358✔
2541
  if (pResultRowInfo->cur.pageId == -1) {  // the first window, from the previous stored value
33,714,358✔
2542
    getInitialStartTimeWindow(pInterval, ts, &w, (order == TSDB_ORDER_ASC));
31,943,388✔
2543
    return w;
31,904,293✔
2544
  }
2545

2546
  SResultRow* pRow = getResultRowByPos(pBuf, &pResultRowInfo->cur, false);
1,770,970✔
2547
  if (pRow) {
1,774,797✔
2548
    TAOS_SET_OBJ_ALIGNED(&w, pRow->win);
1,774,795✔
2549
  }
2550

2551
  // in case of typical time window, we can calculate time window directly.
2552
  if (w.skey > ts || w.ekey < ts) {
1,774,797✔
2553
    w = doCalculateTimeWindow(ts, pInterval);
1,133,443✔
2554
  }
2555

2556
  if (pInterval->interval != pInterval->sliding) {
1,774,838✔
2557
    // it is an sliding window query, in which sliding value is not equalled to
2558
    // interval value, and we need to find the first qualified time window.
2559
    w = getFirstQualifiedTimeWindow(ts, &w, pInterval, order);
456,897✔
2560
  }
2561

2562
  return w;
1,774,840✔
2563
}
2564

2565
TSKEY getNextTimeWindowStart(const SInterval* pInterval, TSKEY start, int32_t order) {
258,603,444✔
2566
  int32_t factor = GET_FORWARD_DIRECTION_FACTOR(order);
258,603,444✔
2567
  TSKEY   nextStart = taosTimeAdd(start, -1 * pInterval->offset, pInterval->offsetUnit, pInterval->precision, NULL);
258,603,444✔
2568
  nextStart = taosTimeAdd(nextStart, factor * pInterval->sliding, pInterval->slidingUnit, pInterval->precision, NULL);
258,430,796✔
2569
  nextStart = taosTimeAdd(nextStart, pInterval->offset, pInterval->offsetUnit, pInterval->precision, NULL);
258,213,778✔
2570
  return nextStart;
257,681,426✔
2571
}
2572

2573
void getNextTimeWindow(const SInterval* pInterval, STimeWindow* tw, int32_t order) {
257,559,766✔
2574
  tw->skey = getNextTimeWindowStart(pInterval, tw->skey, order);
257,559,766✔
2575
  tw->ekey = taosTimeAdd(tw->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
256,566,143✔
2576
}
256,481,406✔
2577

2578
bool hasLimitOffsetInfo(SLimitInfo* pLimitInfo) {
16,746,906✔
2579
  return (pLimitInfo->limit.limit != -1 || pLimitInfo->limit.offset != -1 || pLimitInfo->slimit.limit != -1 ||
33,470,033!
2580
          pLimitInfo->slimit.offset != -1);
16,723,127✔
2581
}
2582

2583
bool hasSlimitOffsetInfo(SLimitInfo* pLimitInfo) {
×
2584
  return (pLimitInfo->slimit.limit != -1 || pLimitInfo->slimit.offset != -1);
×
2585
}
2586

2587
void initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo) {
14,458,752✔
2588
  SLimit limit = {.limit = getLimit(pLimit), .offset = getOffset(pLimit)};
14,458,752✔
2589
  SLimit slimit = {.limit = getLimit(pSLimit), .offset = getOffset(pSLimit)};
14,462,334✔
2590

2591
  pLimitInfo->limit = limit;
14,464,832✔
2592
  pLimitInfo->slimit = slimit;
14,464,832✔
2593
  pLimitInfo->remainOffset = limit.offset;
14,464,832✔
2594
  pLimitInfo->remainGroupOffset = slimit.offset;
14,464,832✔
2595
}
14,464,832✔
2596

2597
void resetLimitInfoForNextGroup(SLimitInfo* pLimitInfo) {
2,467,541✔
2598
  pLimitInfo->numOfOutputRows = 0;
2,467,541✔
2599
  pLimitInfo->remainOffset = pLimitInfo->limit.offset;
2,467,541✔
2600
}
2,467,541✔
2601

2602
int32_t tableListGetSize(const STableListInfo* pTableList, int32_t* pRes) {
19,234,473✔
2603
  if (taosArrayGetSize(pTableList->pTableList) != taosHashGetSize(pTableList->map)) {
19,234,473!
2604
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2605
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2606
  }
2607
  (*pRes) = taosArrayGetSize(pTableList->pTableList);
19,247,548✔
2608
  return TSDB_CODE_SUCCESS;
19,246,134✔
2609
}
2610

2611
uint64_t tableListGetSuid(const STableListInfo* pTableList) { return pTableList->idInfo.suid; }
175,935✔
2612

2613
STableKeyInfo* tableListGetInfo(const STableListInfo* pTableList, int32_t index) {
7,474,500✔
2614
  if (taosArrayGetSize(pTableList->pTableList) == 0) {
7,474,500✔
2615
    return NULL;
554✔
2616
  }
2617

2618
  return taosArrayGet(pTableList->pTableList, index);
7,474,914✔
2619
}
2620

2621
int32_t tableListFind(const STableListInfo* pTableList, uint64_t uid, int32_t startIndex) {
179✔
2622
  int32_t numOfTables = taosArrayGetSize(pTableList->pTableList);
179✔
2623
  if (startIndex >= numOfTables) {
179!
2624
    return -1;
×
2625
  }
2626

2627
  for (int32_t i = startIndex; i < numOfTables; ++i) {
2,938!
2628
    STableKeyInfo* p = taosArrayGet(pTableList->pTableList, i);
2,938✔
2629
    if (!p) {
2,938✔
2630
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
1!
2631
      return -1;
×
2632
    }
2633
    if (p->uid == uid) {
2,937✔
2634
      return i;
178✔
2635
    }
2636
  }
2637
  return -1;
×
2638
}
2639

2640
void tableListGetSourceTableInfo(const STableListInfo* pTableList, uint64_t* psuid, uint64_t* uid, int32_t* type) {
22,818✔
2641
  *psuid = pTableList->idInfo.suid;
22,818✔
2642
  *uid = pTableList->idInfo.uid;
22,818✔
2643
  *type = pTableList->idInfo.tableType;
22,818✔
2644
}
22,818✔
2645

2646
uint64_t tableListGetTableGroupId(const STableListInfo* pTableList, uint64_t tableUid) {
17,644,708✔
2647
  int32_t* slot = taosHashGet(pTableList->map, &tableUid, sizeof(tableUid));
17,644,708✔
2648
  if (slot == NULL) {
17,649,257!
2649
    return -1;
×
2650
  }
2651

2652
  STableKeyInfo* pKeyInfo = taosArrayGet(pTableList->pTableList, *slot);
17,649,257✔
2653

2654
  return pKeyInfo->groupId;
17,646,744✔
2655
}
2656

2657
// TODO handle the group offset info, fix it, the rule of group output will be broken by this function
2658
int32_t tableListAddTableInfo(STableListInfo* pTableList, uint64_t uid, uint64_t gid) {
10,519✔
2659
  int32_t code = TSDB_CODE_SUCCESS;
10,519✔
2660
  int32_t lino = 0;
10,519✔
2661
  if (pTableList->map == NULL) {
10,519!
2662
    pTableList->map = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
×
2663
    QUERY_CHECK_NULL(pTableList->map, code, lino, _end, terrno);
×
2664
  }
2665

2666
  STableKeyInfo keyInfo = {.uid = uid, .groupId = gid};
10,519✔
2667
  void*         p = taosHashGet(pTableList->map, &uid, sizeof(uid));
10,519✔
2668
  if (p != NULL) {
10,534✔
2669
    qInfo("table:%" PRId64 " already in tableIdList, ignore it", uid);
3!
2670
    goto _end;
3✔
2671
  }
2672

2673
  void* tmp = taosArrayPush(pTableList->pTableList, &keyInfo);
10,531✔
2674
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
10,514!
2675

2676
  int32_t slot = (int32_t)taosArrayGetSize(pTableList->pTableList) - 1;
10,514✔
2677
  code = taosHashPut(pTableList->map, &uid, sizeof(uid), &slot, sizeof(slot));
10,512✔
2678
  if (code != TSDB_CODE_SUCCESS) {
10,532!
2679
    // we have checked the existence of uid in hash map above
2680
    QUERY_CHECK_CONDITION((code != TSDB_CODE_DUP_KEY), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
2681
    taosArrayPopTailBatch(pTableList->pTableList, 1);  // let's pop the last element in the array list
×
2682
  }
2683

2684
_end:
10,532✔
2685
  if (code != TSDB_CODE_SUCCESS) {
10,533!
2686
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2687
  } else {
2688
    qDebug("uid:%" PRIu64 ", groupId:%" PRIu64 " added into table list, slot:%d, total:%d", uid, gid, slot, slot + 1);
10,533✔
2689
  }
2690

2691
  return code;
10,529✔
2692
}
2693

2694
int32_t tableListGetGroupList(const STableListInfo* pTableList, int32_t ordinalGroupIndex, STableKeyInfo** pKeyInfo,
3,213,846✔
2695
                              int32_t* size) {
2696
  int32_t totalGroups = tableListGetOutputGroups(pTableList);
3,213,846✔
2697
  int32_t numOfTables = 0;
3,218,144✔
2698
  int32_t code = tableListGetSize(pTableList, &numOfTables);
3,218,144✔
2699
  if (code != TSDB_CODE_SUCCESS) {
3,216,774!
2700
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2701
    return code;
×
2702
  }
2703

2704
  if (ordinalGroupIndex < 0 || ordinalGroupIndex >= totalGroups) {
3,216,774!
2705
    return TSDB_CODE_INVALID_PARA;
×
2706
  }
2707

2708
  // here handle two special cases:
2709
  // 1. only one group exists, and 2. one table exists for each group.
2710
  if (totalGroups == 1) {
3,216,774!
2711
    *size = numOfTables;
3,217,384✔
2712
    *pKeyInfo = (*size == 0) ? NULL : taosArrayGet(pTableList->pTableList, 0);
3,217,384✔
2713
    return TSDB_CODE_SUCCESS;
3,214,097✔
2714
  } else if (totalGroups == numOfTables) {
×
2715
    *size = 1;
981✔
2716
    *pKeyInfo = taosArrayGet(pTableList->pTableList, ordinalGroupIndex);
981✔
2717
    return TSDB_CODE_SUCCESS;
981✔
2718
  }
2719

2720
  int32_t offset = pTableList->groupOffset[ordinalGroupIndex];
×
2721
  if (ordinalGroupIndex < totalGroups - 1) {
×
2722
    *size = pTableList->groupOffset[ordinalGroupIndex + 1] - offset;
524✔
2723
  } else {
2724
    *size = numOfTables - offset;
×
2725
  }
2726

2727
  *pKeyInfo = taosArrayGet(pTableList->pTableList, offset);
×
2728
  return TSDB_CODE_SUCCESS;
655✔
2729
}
2730

2731
int32_t tableListGetOutputGroups(const STableListInfo* pTableList) { return pTableList->numOfOuputGroups; }
9,605,199✔
2732

2733
bool oneTableForEachGroup(const STableListInfo* pTableList) { return pTableList->oneTableForEachGroup; }
12,131✔
2734

2735
STableListInfo* tableListCreate() {
5,331,795✔
2736
  STableListInfo* pListInfo = taosMemoryCalloc(1, sizeof(STableListInfo));
5,331,795!
2737
  if (pListInfo == NULL) {
5,343,545!
2738
    return NULL;
×
2739
  }
2740

2741
  pListInfo->remainGroups = NULL;
5,343,545✔
2742
  pListInfo->pTableList = taosArrayInit(4, sizeof(STableKeyInfo));
5,343,545✔
2743
  if (pListInfo->pTableList == NULL) {
5,343,060!
2744
    goto _error;
×
2745
  }
2746

2747
  pListInfo->map = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
5,343,060✔
2748
  if (pListInfo->map == NULL) {
5,343,308!
2749
    goto _error;
×
2750
  }
2751

2752
  pListInfo->numOfOuputGroups = 1;
5,343,308✔
2753
  return pListInfo;
5,343,308✔
2754

2755
_error:
×
2756
  tableListDestroy(pListInfo);
×
2757
  return NULL;
×
2758
}
2759

2760
void tableListDestroy(STableListInfo* pTableListInfo) {
5,345,288✔
2761
  if (pTableListInfo == NULL) {
5,345,288✔
2762
    return;
222✔
2763
  }
2764

2765
  taosArrayDestroy(pTableListInfo->pTableList);
5,345,066✔
2766
  taosMemoryFreeClear(pTableListInfo->groupOffset);
5,346,085!
2767

2768
  taosHashCleanup(pTableListInfo->map);
5,346,086✔
2769
  taosHashCleanup(pTableListInfo->remainGroups);
5,345,740✔
2770
  pTableListInfo->pTableList = NULL;
5,345,922✔
2771
  pTableListInfo->map = NULL;
5,345,922✔
2772
  taosMemoryFree(pTableListInfo);
5,345,922!
2773
}
2774

2775
void tableListClear(STableListInfo* pTableListInfo) {
2,507✔
2776
  if (pTableListInfo == NULL) {
2,507!
2777
    return;
×
2778
  }
2779

2780
  taosArrayClear(pTableListInfo->pTableList);
2,507✔
2781
  taosHashClear(pTableListInfo->map);
2,509✔
2782
  taosHashClear(pTableListInfo->remainGroups);
2,511✔
2783
  taosMemoryFree(pTableListInfo->groupOffset);
2,511!
2784
  pTableListInfo->numOfOuputGroups = 1;
2,511✔
2785
  pTableListInfo->oneTableForEachGroup = false;
2,511✔
2786
}
2787

2788
static int32_t orderbyGroupIdComparFn(const void* p1, const void* p2) {
94,134✔
2789
  STableKeyInfo* pInfo1 = (STableKeyInfo*)p1;
94,134✔
2790
  STableKeyInfo* pInfo2 = (STableKeyInfo*)p2;
94,134✔
2791

2792
  if (pInfo1->groupId == pInfo2->groupId) {
94,134✔
2793
    return 0;
167✔
2794
  } else {
2795
    return pInfo1->groupId < pInfo2->groupId ? -1 : 1;
93,967✔
2796
  }
2797
}
2798

2799
static int32_t sortTableGroup(STableListInfo* pTableListInfo) {
953✔
2800
  taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
953✔
2801
  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
953✔
2802

2803
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
954✔
2804
  if (!pList) {
955!
2805
    return terrno;
×
2806
  }
2807

2808
  STableKeyInfo* pInfo = taosArrayGet(pTableListInfo->pTableList, 0);
955✔
2809
  if (!pInfo) {
955!
2810
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2811
    return terrno;
×
2812
  }
2813
  uint64_t       gid = pInfo->groupId;
955✔
2814

2815
  int32_t start = 0;
955✔
2816
  void*   tmp = taosArrayPush(pList, &start);
955✔
2817
  if (!tmp) {
955!
2818
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2819
    return terrno;
×
2820
  }
2821

2822
  for (int32_t i = 1; i < size; ++i) {
2,996✔
2823
    pInfo = taosArrayGet(pTableListInfo->pTableList, i);
2,040✔
2824
    if (!pInfo) {
2,041✔
2825
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
2!
2826
      return terrno;
2✔
2827
    }
2828
    if (pInfo->groupId != gid) {
2,039✔
2829
      tmp = taosArrayPush(pList, &i);
1,875✔
2830
      if (!tmp) {
1,875!
2831
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2832
        return terrno;
×
2833
      }
2834
      gid = pInfo->groupId;
1,875✔
2835
    }
2836
  }
2837

2838
  pTableListInfo->numOfOuputGroups = taosArrayGetSize(pList);
956✔
2839
  pTableListInfo->groupOffset = taosMemoryMalloc(sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
953!
2840
  if (pTableListInfo->groupOffset == NULL) {
954!
2841
    taosArrayDestroy(pList);
×
2842
    return terrno;
×
2843
  }
2844

2845
  memcpy(pTableListInfo->groupOffset, taosArrayGet(pList, 0), sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
954✔
2846
  taosArrayDestroy(pList);
954✔
2847
  return TSDB_CODE_SUCCESS;
955✔
2848
}
2849

2850
int32_t buildGroupIdMapForAllTables(STableListInfo* pTableListInfo, SReadHandle* pHandle, SScanPhysiNode* pScanNode,
5,038,686✔
2851
                                    SNodeList* group, bool groupSort, uint8_t* digest, SStorageAPI* pAPI) {
2852
  int32_t code = TSDB_CODE_SUCCESS;
5,038,686✔
2853

2854
  bool   groupByTbname = groupbyTbname(group);
5,038,686✔
2855
  size_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
5,037,704✔
2856
  if (!numOfTables) {
5,037,493!
2857
    return code;
×
2858
  }
2859
  if (group == NULL || groupByTbname) {
5,037,493✔
2860
    if (tsCountAlwaysReturnValue && QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode) &&
4,971,414✔
2861
        ((STableScanPhysiNode*)pScanNode)->needCountEmptyTable) {
2,954,423✔
2862
      pTableListInfo->remainGroups =
387,333✔
2863
          taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
387,357✔
2864
      if (pTableListInfo->remainGroups == NULL) {
387,333!
2865
        return terrno;
×
2866
      }
2867

2868
      for (int i = 0; i < numOfTables; i++) {
1,094,451✔
2869
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
707,003✔
2870
        if (!info) {
706,971!
2871
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2872
          return terrno;
×
2873
        }
2874
        info->groupId = groupByTbname ? info->uid : 0;
706,974✔
2875

2876
        int32_t tempRes = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId),
706,974✔
2877
                                      &(info->uid), sizeof(info->uid));
706,974✔
2878
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
707,118!
2879
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
2880
          return tempRes;
×
2881
        }
2882
      }
2883
    } else {
2884
      for (int32_t i = 0; i < numOfTables; i++) {
16,509,549✔
2885
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
11,924,730✔
2886
        if (!info) {
11,925,740✔
2887
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
248!
2888
          return terrno;
248✔
2889
        }
2890
        info->groupId = groupByTbname ? info->uid : 0;
11,925,492✔
2891
      }
2892
    }
2893

2894
    pTableListInfo->oneTableForEachGroup = groupByTbname;
4,972,267✔
2895
    if (numOfTables == 1 && pTableListInfo->idInfo.tableType == TSDB_CHILD_TABLE) {
4,972,267✔
2896
      pTableListInfo->oneTableForEachGroup = true;
727,122✔
2897
    }
2898

2899
    if (groupSort && groupByTbname) {
4,972,267✔
2900
      taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
28,842✔
2901
      pTableListInfo->numOfOuputGroups = numOfTables;
28,824✔
2902
    } else if (groupByTbname && pScanNode->groupOrderScan) {
4,943,425✔
2903
      pTableListInfo->numOfOuputGroups = numOfTables;
111✔
2904
    } else {
2905
      pTableListInfo->numOfOuputGroups = 1;
4,943,314✔
2906
    }
2907
  } else {
2908
    bool initRemainGroups = false;
66,079✔
2909
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode)) {
66,079✔
2910
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pScanNode;
65,418✔
2911
      if (tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable &&
65,418✔
2912
          !(groupSort || pScanNode->groupOrderScan)) {
13,186!
2913
        initRemainGroups = true;
13,142✔
2914
      }
2915
    }
2916

2917
    code = getColInfoResultForGroupby(pHandle->vnode, group, pTableListInfo, digest, pAPI, initRemainGroups);
66,079✔
2918
    if (code != TSDB_CODE_SUCCESS) {
66,093✔
2919
      return code;
16✔
2920
    }
2921

2922
    if (pScanNode->groupOrderScan) pTableListInfo->numOfOuputGroups = taosArrayGetSize(pTableListInfo->pTableList);
66,077✔
2923

2924
    if (groupSort || pScanNode->groupOrderScan) {
66,077✔
2925
      code = sortTableGroup(pTableListInfo);
925✔
2926
    }
2927
  }
2928

2929
  // add all table entry in the hash map
2930
  size_t size = taosArrayGetSize(pTableListInfo->pTableList);
5,038,356✔
2931
  for (int32_t i = 0; i < size; ++i) {
17,954,660✔
2932
    STableKeyInfo* p = taosArrayGet(pTableListInfo->pTableList, i);
12,905,164✔
2933
    if (!p) {
12,888,473!
2934
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2935
      return terrno;
×
2936
    }
2937
    int32_t        tempRes = taosHashPut(pTableListInfo->map, &p->uid, sizeof(uint64_t), &i, sizeof(int32_t));
12,888,473✔
2938
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
12,918,539!
2939
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
2940
      return tempRes;
×
2941
    }
2942
  }
2943

2944
  return code;
5,049,496✔
2945
}
2946

2947
int32_t createScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
5,242,335✔
2948
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
2949
                                SExecTaskInfo* pTaskInfo) {
2950
  int64_t     st = taosGetTimestampUs();
5,245,724✔
2951
  const char* idStr = GET_TASKID(pTaskInfo);
5,245,724✔
2952

2953
  if (pHandle == NULL) {
5,245,724!
2954
    qError("invalid handle, in creating operator tree, %s", idStr);
×
2955
    return TSDB_CODE_INVALID_PARA;
×
2956
  }
2957

2958
  uint8_t digest[17] = {0};
5,245,724✔
2959
  int32_t code = getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, digest, idStr,
5,245,724✔
2960
                              &pTaskInfo->storageAPI);
2961
  if (code != TSDB_CODE_SUCCESS) {
5,246,836✔
2962
    qError("failed to getTableList, code:%s", tstrerror(code));
16!
2963
    return code;
16✔
2964
  }
2965

2966
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
5,246,820✔
2967

2968
  int64_t st1 = taosGetTimestampUs();
5,247,894✔
2969
  pTaskInfo->cost.extractListTime = (st1 - st) / 1000.0;
5,247,894✔
2970
  qDebug("extract queried table list completed, %d tables, elapsed time:%.2f ms %s", numOfTables,
5,247,894✔
2971
         pTaskInfo->cost.extractListTime, idStr);
2972

2973
  if (numOfTables == 0) {
5,250,261✔
2974
    qDebug("no table qualified for query, %s" PRIx64, idStr);
206,524✔
2975
    return TSDB_CODE_SUCCESS;
206,533✔
2976
  }
2977

2978
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, digest,
5,043,737✔
2979
                                     &pTaskInfo->storageAPI);
2980
  if (code != TSDB_CODE_SUCCESS) {
5,046,233✔
2981
    return code;
16✔
2982
  }
2983

2984
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
5,045,788✔
2985
  qDebug("generate group id map completed, elapsed time:%.2f ms %s", pTaskInfo->cost.groupIdMapTime, idStr);
5,045,788✔
2986

2987
  return TSDB_CODE_SUCCESS;
5,045,916✔
2988
}
2989

2990
char* getStreamOpName(uint16_t opType) {
1,262,138✔
2991
  switch (opType) {
1,262,138!
2992
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
436,530✔
2993
      return "stream scan";
436,530✔
2994
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
29,686✔
2995
      return "project";
29,686✔
2996
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL:
602,862✔
2997
      return "interval single";
602,862✔
2998
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_INTERVAL:
12,247✔
2999
      return "interval final";
12,247✔
3000
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_INTERVAL:
36,041✔
3001
      return "interval semi";
36,041✔
3002
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_MID_INTERVAL:
323✔
3003
      return "interval mid";
323✔
3004
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_FILL:
5,554✔
3005
      return "stream fill";
5,554✔
3006
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SESSION:
9,141✔
3007
      return "session single";
9,141✔
3008
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_SESSION:
1,231✔
3009
      return "session semi";
1,231✔
3010
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_SESSION:
981✔
3011
      return "session final";
981✔
3012
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_STATE:
7,179✔
3013
      return "state single";
7,179✔
3014
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_PARTITION:
91,975✔
3015
      return "stream partitionby";
91,975✔
3016
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_EVENT:
940✔
3017
      return "stream event";
940✔
3018
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_COUNT:
10,013✔
3019
      return "stream count";
10,013✔
3020
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERP_FUNC:
9,442✔
3021
      return "stream interp";
9,442✔
3022
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_INTERVAL:
6,042✔
3023
      return "interval continue";
6,042✔
3024
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_SEMI_INTERVAL:
786✔
3025
      return "interval continue semi";
786✔
3026
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_FINAL_INTERVAL:
344✔
3027
      return "interval continue final";
344✔
3028
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_SESSION:
×
3029
      return "session continue";
×
3030
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_SEMI_SESSION:
×
3031
      return "session continue semi";
×
3032
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_FINAL_SESSION:
×
3033
      return "session continue final";
×
3034
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_STATE:
×
3035
      return "state continue";
×
3036
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_EVENT:
×
3037
      return "event continue";
×
3038
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_COUNT:
×
3039
      return "count continue";
×
3040
  }
3041
  return "error name";
821✔
3042
}
3043

3044
void printDataBlock(SSDataBlock* pBlock, const char* flag, const char* taskIdStr) {
586,870✔
3045
  if (!pBlock) {
586,870✔
3046
    qInfo("%s===stream===%s: Block is Null", taskIdStr, flag);
250!
3047
    return;
250✔
3048
  } else if (pBlock->info.rows == 0) {
586,620✔
3049
    qInfo("%s===stream===%s: Block is Empty. block type %d", taskIdStr, flag, pBlock->info.type);
21,560!
3050
    return;
21,568✔
3051
  }
3052
  if (qDebugFlag & DEBUG_INFO) {
565,060✔
3053
    char*   pBuf = NULL;
564,978✔
3054
    int32_t code = dumpBlockData(pBlock, flag, &pBuf, taskIdStr);
564,978✔
3055
    if (code == 0) {
565,139!
3056
      qInfo("%s", pBuf);
565,154✔
3057
      taosMemoryFree(pBuf);
565,181!
3058
    }
3059
  }
3060
}
3061

3062
void printSpecDataBlock(SSDataBlock* pBlock, const char* flag, const char* opStr, const char* taskIdStr) {
608,416✔
3063
  if (!pBlock) {
608,416!
3064
    qDebug("%s===stream===%s %s: Block is Null", taskIdStr, flag, opStr);
×
3065
    return;
×
3066
  } else if (pBlock->info.rows == 0) {
608,416✔
3067
    qDebug("%s===stream===%s %s: Block is Empty. block type %d.skey:%" PRId64 ",ekey:%" PRId64 ",version%" PRId64,
14,649✔
3068
           taskIdStr, flag, opStr, pBlock->info.type, pBlock->info.window.skey, pBlock->info.window.ekey,
3069
           pBlock->info.version);
3070
    return;
14,649✔
3071
  }
3072
  if (qDebugFlag & DEBUG_INFO) {
593,767!
3073
    char* pBuf = NULL;
593,768✔
3074
    char  flagBuf[64];
3075
    snprintf(flagBuf, sizeof(flagBuf), "%s %s", flag, opStr);
593,768✔
3076
    int32_t code = dumpBlockData(pBlock, flagBuf, &pBuf, taskIdStr);
593,768✔
3077
    if (code == 0) {
593,748!
3078
      qInfo("%s", pBuf);
593,754!
3079
      taosMemoryFree(pBuf);
593,773!
3080
    }
3081
  }
3082
}
3083

3084
TSKEY getStartTsKey(STimeWindow* win, const TSKEY* tsCols) { return tsCols == NULL ? win->skey : tsCols[0]; }
3,443,122!
3085

3086
void updateTimeWindowInfo(SColumnInfoData* pColData, STimeWindow* pWin, int64_t delta) {
452,240,526✔
3087
  int64_t* ts = (int64_t*)pColData->pData;
452,240,526✔
3088

3089
  int64_t duration = pWin->ekey - pWin->skey + delta;
452,240,526✔
3090
  ts[2] = duration;            // set the duration
452,240,526✔
3091
  ts[3] = pWin->skey;          // window start key
452,240,526✔
3092
  ts[4] = pWin->ekey + delta;  // window end key
452,240,526✔
3093
}
452,240,526✔
3094

3095
int32_t compKeys(const SArray* pSortGroupCols, const char* oldkeyBuf, int32_t oldKeysLen, const SSDataBlock* pBlock,
21,395,041✔
3096
                 int32_t rowIndex) {
3097
  SColumnDataAgg* pColAgg = NULL;
21,395,041✔
3098
  const char*     isNull = oldkeyBuf;
21,395,041✔
3099
  const char*     p = oldkeyBuf + sizeof(int8_t) * pSortGroupCols->size;
21,395,041✔
3100

3101
  for (int32_t i = 0; i < pSortGroupCols->size; ++i) {
53,678,285✔
3102
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
32,765,256✔
3103
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
32,765,256✔
3104
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
32,765,256!
3105

3106
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
65,530,512!
3107
      if (isNull[i] != 1) return 1;
2,179,695✔
3108
    } else {
3109
      if (isNull[i] != 0) return 1;
30,585,561✔
3110
      const char* val = colDataGetData(pColInfoData, rowIndex);
30,571,337!
3111
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
30,571,337!
3112
        int32_t len = getJsonValueLen(val);
×
3113
        if (memcmp(p, val, len) != 0) return 1;
×
3114
        p += len;
×
3115
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
30,571,337!
3116
        if (memcmp(p, val, varDataTLen(val)) != 0) return 1;
10,399,355✔
3117
        p += varDataTLen(val);
10,263,067✔
3118
      } else {
3119
        if (0 != memcmp(p, val, pCol->bytes)) return 1;
20,171,982✔
3120
        p += pCol->bytes;
19,840,674✔
3121
      }
3122
    }
3123
  }
3124
  if ((int32_t)(p - oldkeyBuf) != oldKeysLen) return 1;
20,913,029✔
3125
  return 0;
20,912,733✔
3126
}
3127

3128
int32_t buildKeys(char* keyBuf, const SArray* pSortGroupCols, const SSDataBlock* pBlock, int32_t rowIndex) {
481,972✔
3129
  uint32_t        colNum = pSortGroupCols->size;
481,972✔
3130
  SColumnDataAgg* pColAgg = NULL;
481,972✔
3131
  char*           isNull = keyBuf;
481,972✔
3132
  char*           p = keyBuf + sizeof(int8_t) * colNum;
481,972✔
3133

3134
  for (int32_t i = 0; i < colNum; ++i) {
1,434,114✔
3135
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
952,142✔
3136
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
952,142✔
3137
    if (pCol->slotId > pBlock->pDataBlock->size) continue;
952,142!
3138

3139
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
952,142!
3140

3141
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
1,904,284!
3142
      isNull[i] = 1;
34,558✔
3143
    } else {
3144
      isNull[i] = 0;
917,584✔
3145
      const char* val = colDataGetData(pColInfoData, rowIndex);
917,584!
3146
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
917,584!
3147
        int32_t len = getJsonValueLen(val);
×
3148
        memcpy(p, val, len);
×
3149
        p += len;
×
3150
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
917,584!
3151
        varDataCopy(p, val);
139,764✔
3152
        p += varDataTLen(val);
139,764✔
3153
      } else {
3154
        memcpy(p, val, pCol->bytes);
777,820✔
3155
        p += pCol->bytes;
777,820✔
3156
      }
3157
    }
3158
  }
3159
  return (int32_t)(p - keyBuf);
481,972✔
3160
}
3161

3162
uint64_t calcGroupId(char* pData, int32_t len) {
28,287,714✔
3163
  T_MD5_CTX context;
3164
  tMD5Init(&context);
28,287,714✔
3165
  tMD5Update(&context, (uint8_t*)pData, len);
28,254,356✔
3166
  tMD5Final(&context);
28,405,637✔
3167

3168
  // NOTE: only extract the initial 8 bytes of the final MD5 digest
3169
  uint64_t id = 0;
28,550,188✔
3170
  memcpy(&id, context.digest, sizeof(uint64_t));
28,550,188✔
3171
  if (0 == id) memcpy(&id, context.digest + 8, sizeof(uint64_t));
28,550,188!
3172
  return id;
28,550,188✔
3173
}
3174

3175
SNodeList* makeColsNodeArrFromSortKeys(SNodeList* pSortKeys) {
944✔
3176
  SNode*     node;
3177
  SNodeList* ret = NULL;
944✔
3178
  FOREACH(node, pSortKeys) {
2,878!
3179
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)node;
1,936✔
3180
    int32_t           code = nodesListMakeAppend(&ret, pSortKey->pExpr);
1,936✔
3181
    if (code != TSDB_CODE_SUCCESS) {
1,934!
3182
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3183
      terrno = code;
×
3184
      return NULL;
×
3185
    }
3186
  }
3187
  return ret;
942✔
3188
}
3189

3190
int32_t extractKeysLen(const SArray* keys, int32_t* pLen) {
944✔
3191
  int32_t code = TSDB_CODE_SUCCESS;
944✔
3192
  int32_t lino = 0;
944✔
3193
  int32_t len = 0;
944✔
3194
  int32_t keyNum = taosArrayGetSize(keys);
944✔
3195
  for (int32_t i = 0; i < keyNum; ++i) {
2,399✔
3196
    SColumn* pCol = (SColumn*)taosArrayGet(keys, i);
1,456✔
3197
    QUERY_CHECK_NULL(pCol, code, lino, _end, terrno);
1,455!
3198
    len += pCol->bytes;
1,455✔
3199
  }
3200
  len += sizeof(int8_t) * keyNum;  // null flag
943✔
3201
  *pLen = len;
943✔
3202

3203
_end:
943✔
3204
  if (code != TSDB_CODE_SUCCESS) {
943!
3205
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3206
  }
3207
  return code;
943✔
3208
}
3209

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