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

taosdata / TDengine / #3532

20 Nov 2024 07:11AM UTC coverage: 60.78% (+0.6%) from 60.213%
#3532

push

travis-ci

web-flow
Merge pull request #28823 from taosdata/fix/3.0/TD-32587

fix:[TD-32587]fix stmt segmentation fault

119943 of 252352 branches covered (47.53%)

Branch coverage included in aggregate %.

1 of 4 new or added lines in 1 file covered. (25.0%)

463 existing lines in 99 files now uncovered.

200682 of 275165 relevant lines covered (72.93%)

15642683.31 hits per line

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

70.77
/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 "tdatablock.h"
22
#include "thash.h"
23
#include "tmsg.h"
24
#include "ttime.h"
25

26
#include "executil.h"
27
#include "executorInt.h"
28
#include "querytask.h"
29
#include "storageapi.h"
30
#include "tcompression.h"
31

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

39
typedef struct STransTagExprCtx {
40
  int32_t      code;
41
  SMetaReader* pReader;
42
} STransTagExprCtx;
43

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

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

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

57
static int64_t getLimit(const SNode* pLimit) { return NULL == pLimit ? -1 : ((SLimitNode*)pLimit)->limit; }
25,904,139✔
58
static int64_t getOffset(const SNode* pLimit) { return NULL == pLimit ? -1 : ((SLimitNode*)pLimit)->offset; }
25,909,656✔
59
static void    releaseColInfoData(void* pCol);
60

61
void initResultRowInfo(SResultRowInfo* pResultRowInfo) {
10,169,081✔
62
  pResultRowInfo->size = 0;
10,169,081✔
63
  pResultRowInfo->cur.pageId = -1;
10,169,081✔
64
}
10,169,081✔
65

66
void closeResultRow(SResultRow* pResultRow) { pResultRow->closed = true; }
9,570,344✔
67

68
void resetResultRow(SResultRow* pResultRow, size_t entrySize) {
63,971,227✔
69
  pResultRow->numOfRows = 0;
63,971,227✔
70
  pResultRow->closed = false;
63,971,227✔
71
  pResultRow->endInterp = false;
63,971,227✔
72
  pResultRow->startInterp = false;
63,971,227✔
73

74
  if (entrySize > 0) {
63,971,227!
75
    memset(pResultRow->pEntryInfo, 0, entrySize);
63,981,006✔
76
  }
77
}
63,971,227✔
78

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

84
size_t getResultRowSize(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
6,718,011✔
85
  int32_t rowSize = (numOfOutput * sizeof(SResultRowEntryInfo)) + sizeof(SResultRow);
6,718,011✔
86

87
  for (int32_t i = 0; i < numOfOutput; ++i) {
31,063,700✔
88
    rowSize += pCtx[i].resDataInfo.interBufSize;
24,345,689✔
89
  }
90

91
  return rowSize;
6,718,011✔
92
}
93

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

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

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

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

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

158
  SqlFunctionCtx *pCtx = pSup->pCtx;
1,230,355✔
159
  int32_t        *offset = pSup->rowEntryInfoOffset;
1,230,355✔
160
  SResultRow     *pResultRow = (SResultRow*)inBuf;
1,230,355✔
161
  size_t          rowSize = getResultRowSize(pCtx, pSup->numOfExprs);
1,230,355✔
162

163
  if (rowSize > inBufSize) {
1,230,355!
164
    qError("invalid input buffer size, rowSize:%zu, inBufSize:%zu", rowSize, inBufSize);
×
165
    return TSDB_CODE_INVALID_PARA;
×
166
  }
167

168
  // calculate the size of output buffer
169
  *outBufSize = rowSize + sizeof(int32_t) * pSup->numOfExprs;
1,230,355✔
170
  if (rowSize < inBufSize) {
1,230,355✔
171
    *outBufSize += inBufSize - rowSize;
1,251✔
172
  }
173

174
  *outBuf = taosMemoryMalloc(*outBufSize);
1,230,355✔
175
  if (*outBuf == NULL) {
1,230,355!
176
    qError("failed to allocate memory for output buffer, size:%zu", *outBufSize);
×
177
    return terrno;
×
178
  }
179

180
  char *pBuf = *outBuf;
1,230,355✔
181
  pResultRow->version = FUNCTION_RESULT_INFO_VERSION;
1,230,355✔
182
  (void)memcpy(pBuf, pResultRow, sizeof(SResultRow));
1,230,355✔
183
  pBuf += sizeof(SResultRow);
1,230,355✔
184
  for (int32_t i = 0; i < pSup->numOfExprs; ++i) {
4,981,047✔
185
    size_t len = sizeof(SResultRowEntryInfo) + pCtx[i].resDataInfo.interBufSize;
3,750,966✔
186
    *(int32_t *) pBuf = (int32_t)len;
3,750,966✔
187
    pBuf += sizeof(int32_t);
3,750,966✔
188
    (void)memcpy(pBuf, getResultEntryInfo(pResultRow, i, offset), len);
3,750,966✔
189
    pBuf += len;
3,750,692✔
190
  }
191

192
  if (rowSize < inBufSize) {
1,230,081✔
193
    // stream stores extra data after result row
194
    size_t leftLen = inBufSize - rowSize;
1,251✔
195
    (void)memcpy(pBuf, inBuf + rowSize, leftLen);
1,251✔
196
    pBuf += leftLen;
1,251✔
197
  }
198
  return TSDB_CODE_SUCCESS;
1,230,081✔
199
}
200

201
static void freeEx(void* p) { taosMemoryFree(*(void**)p); }
×
202

203
void cleanupGroupResInfo(SGroupResInfo* pGroupResInfo) {
3,380,429✔
204
  taosMemoryFreeClear(pGroupResInfo->pBuf);
3,380,429✔
205
  if (pGroupResInfo->freeItem) {
3,380,576!
206
    //    taosArrayDestroy(pGroupResInfo->pRows);
207
    taosArrayDestroyEx(pGroupResInfo->pRows, freeEx);
×
208
    pGroupResInfo->freeItem = false;
×
209
    pGroupResInfo->pRows = NULL;
×
210
  } else {
211
    taosArrayDestroy(pGroupResInfo->pRows);
3,380,576✔
212
    pGroupResInfo->pRows = NULL;
3,380,657✔
213
  }
214
  pGroupResInfo->index = 0;
3,380,657✔
215
}
3,380,657✔
216

217
int32_t resultrowComparAsc(const void* p1, const void* p2) {
2,018,107,253✔
218
  SResKeyPos* pp1 = *(SResKeyPos**)p1;
2,018,107,253✔
219
  SResKeyPos* pp2 = *(SResKeyPos**)p2;
2,018,107,253✔
220

221
  if (pp1->groupId == pp2->groupId) {
2,018,107,253✔
222
    int64_t pts1 = *(int64_t*)pp1->key;
1,829,080,060✔
223
    int64_t pts2 = *(int64_t*)pp2->key;
1,829,080,060✔
224

225
    if (pts1 == pts2) {
1,829,080,060!
226
      return 0;
×
227
    } else {
228
      return pts1 < pts2 ? -1 : 1;
1,829,080,060✔
229
    }
230
  } else {
231
    return pp1->groupId < pp2->groupId ? -1 : 1;
189,027,193✔
232
  }
233
}
234

235
static int32_t resultrowComparDesc(const void* p1, const void* p2) { return resultrowComparAsc(p2, p1); }
67,634,088✔
236

237
int32_t initGroupedResultInfo(SGroupResInfo* pGroupResInfo, SSHashObj* pHashmap, int32_t order) {
2,587,312✔
238
  int32_t code = TSDB_CODE_SUCCESS;
2,587,312✔
239
  int32_t lino = 0;
2,587,312✔
240
  if (pGroupResInfo->pRows != NULL) {
2,587,312✔
241
    taosArrayDestroy(pGroupResInfo->pRows);
49,943✔
242
  }
243
  if (pGroupResInfo->pBuf) {
2,587,312✔
244
    taosMemoryFree(pGroupResInfo->pBuf);
49,943✔
245
    pGroupResInfo->pBuf = NULL;
49,943✔
246
  }
247

248
  // extract the result rows information from the hash map
249
  int32_t size = tSimpleHashGetSize(pHashmap);
2,587,312✔
250

251
  void* pData = NULL;
2,587,525✔
252
  pGroupResInfo->pRows = taosArrayInit(size, POINTER_BYTES);
2,587,525✔
253
  QUERY_CHECK_NULL(pGroupResInfo->pRows, code, lino, _end, terrno);
2,587,981!
254

255
  size_t  keyLen = 0;
2,587,981✔
256
  int32_t iter = 0;
2,587,981✔
257
  int64_t bufLen = 0, offset = 0;
2,587,981✔
258

259
  // todo move away and record this during create window
260
  while ((pData = tSimpleHashIterate(pHashmap, pData, &iter)) != NULL) {
219,491,621✔
261
    /*void* key = */ (void)tSimpleHashGetKey(pData, &keyLen);
262
    bufLen += keyLen + sizeof(SResultRowPosition);
216,903,640✔
263
  }
264

265
  pGroupResInfo->pBuf = taosMemoryMalloc(bufLen);
2,573,306✔
266
  QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
2,587,935!
267

268
  iter = 0;
2,588,038✔
269
  while ((pData = tSimpleHashIterate(pHashmap, pData, &iter)) != NULL) {
219,310,984✔
270
    void* key = tSimpleHashGetKey(pData, &keyLen);
216,744,640✔
271

272
    SResKeyPos* p = (SResKeyPos*)(pGroupResInfo->pBuf + offset);
216,744,640✔
273

274
    p->groupId = *(uint64_t*)key;
216,744,640✔
275
    p->pos = *(SResultRowPosition*)pData;
216,744,640✔
276
    memcpy(p->key, (char*)key + sizeof(uint64_t), keyLen - sizeof(uint64_t));
216,744,640✔
277
    void* tmp = taosArrayPush(pGroupResInfo->pRows, &p);
216,744,640✔
278
    QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
216,722,946!
279

280
    offset += keyLen + sizeof(struct SResultRowPosition);
216,722,946✔
281
  }
282

283
  if (order == TSDB_ORDER_ASC || order == TSDB_ORDER_DESC) {
2,599,126✔
284
    __compar_fn_t fn = (order == TSDB_ORDER_ASC) ? resultrowComparAsc : resultrowComparDesc;
1,727,615✔
285
    size = POINTER_BYTES;
1,727,615✔
286
    taosSort(pGroupResInfo->pRows->pData, taosArrayGetSize(pGroupResInfo->pRows), size, fn);
1,727,615✔
287
  }
288

289
  pGroupResInfo->index = 0;
2,587,948✔
290

291
_end:
2,587,948✔
292
  if (code != TSDB_CODE_SUCCESS) {
2,587,948!
293
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
294
  }
295
  return code;
2,587,771✔
296
}
297

298
void initMultiResInfoFromArrayList(SGroupResInfo* pGroupResInfo, SArray* pArrayList) {
37,384✔
299
  if (pGroupResInfo->pRows != NULL) {
37,384✔
300
    taosArrayDestroy(pGroupResInfo->pRows);
33,689✔
301
  }
302

303
  pGroupResInfo->freeItem = true;
37,386✔
304
  pGroupResInfo->pRows = pArrayList;
37,386✔
305
  pGroupResInfo->index = 0;
37,386✔
306
}
37,386✔
307

308
bool hasRemainResults(SGroupResInfo* pGroupResInfo) {
8,114,668✔
309
  if (pGroupResInfo->pRows == NULL) {
8,114,668✔
310
    return false;
1,676✔
311
  }
312

313
  return pGroupResInfo->index < taosArrayGetSize(pGroupResInfo->pRows);
8,112,992✔
314
}
315

316
int32_t getNumOfTotalRes(SGroupResInfo* pGroupResInfo) {
4,784,945✔
317
  if (pGroupResInfo->pRows == 0) {
4,784,945✔
318
    return 0;
1,677✔
319
  }
320

321
  return (int32_t)taosArrayGetSize(pGroupResInfo->pRows);
4,783,268✔
322
}
323

324
SArray* createSortInfo(SNodeList* pNodeList) {
2,333,547✔
325
  size_t numOfCols = 0;
2,333,547✔
326

327
  if (pNodeList != NULL) {
2,333,547✔
328
    numOfCols = LIST_LENGTH(pNodeList);
2,331,977!
329
  } else {
330
    numOfCols = 0;
1,570✔
331
  }
332

333
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
2,333,547✔
334
  if (pList == NULL) {
2,333,611!
UNCOV
335
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
336
    return pList;
×
337
  }
338

339
  for (int32_t i = 0; i < numOfCols; ++i) {
4,821,865✔
340
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
2,488,165✔
341
    if (!pSortKey) {
2,488,211✔
342
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
11!
343
      taosArrayDestroy(pList);
11✔
344
      pList = NULL;
×
345
      terrno = TSDB_CODE_OUT_OF_MEMORY;
×
346
      break;
×
347
    }
348
    SBlockOrderInfo   bi = {0};
2,488,200✔
349
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
2,488,200✔
350
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);
2,488,200✔
351

352
    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
2,488,200✔
353
    bi.slotId = pColNode->slotId;
2,488,200✔
354
    void* tmp = taosArrayPush(pList, &bi);
2,488,249✔
355
    if (!tmp) {
2,488,249!
356
      taosArrayDestroy(pList);
×
357
      pList = NULL;
×
358
      terrno = TSDB_CODE_OUT_OF_MEMORY;
×
359
      break;
×
360
    }
361
  }
362

363
  return pList;
2,333,700✔
364
}
365

366
SSDataBlock* createDataBlockFromDescNode(SDataBlockDescNode* pNode) {
18,859,690✔
367
  int32_t      numOfCols = LIST_LENGTH(pNode->pSlots);
18,859,690!
368
  SSDataBlock* pBlock = NULL;
18,859,690✔
369
  int32_t      code = createDataBlock(&pBlock);
18,859,690✔
370
  if (code) {
18,868,841!
371
    terrno = code;
×
372
    return NULL;
×
373
  }
374

375
  pBlock->info.id.blockId = pNode->dataBlockId;
18,871,615✔
376
  pBlock->info.type = STREAM_INVALID;
18,871,615✔
377
  pBlock->info.calWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
18,871,615✔
378
  pBlock->info.watermark = INT64_MIN;
18,871,615✔
379

380
  for (int32_t i = 0; i < numOfCols; ++i) {
115,181,300✔
381
    SSlotDescNode*  pDescNode = (SSlotDescNode*)nodesListGetNode(pNode->pSlots, i);
96,302,519✔
382
    if (!pDescNode) {
96,355,285!
383
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
384
      blockDataDestroy(pBlock);
×
385
      pBlock = NULL;
×
386
      terrno = TSDB_CODE_INVALID_PARA;
×
387
      break;
×
388
    }
389
    SColumnInfoData idata =
390
        createColumnInfoData(pDescNode->dataType.type, pDescNode->dataType.bytes, pDescNode->slotId);
96,355,285✔
391
    idata.info.scale = pDescNode->dataType.scale;
96,453,311✔
392
    idata.info.precision = pDescNode->dataType.precision;
96,453,311✔
393
    idata.info.noData = pDescNode->reserve;
96,453,311✔
394

395
    code = blockDataAppendColInfo(pBlock, &idata);
96,453,311✔
396
    if (code != TSDB_CODE_SUCCESS) {
96,309,685!
397
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
398
      blockDataDestroy(pBlock);
×
399
      pBlock = NULL;
×
400
      terrno = code;
×
401
      break;
×
402
    }
403
  }
404

405
  return pBlock;
18,878,781✔
406
}
407

408
int32_t prepareDataBlockBuf(SSDataBlock* pDataBlock, SColMatchInfo* pMatchInfo) {
5,116,605✔
409
  SDataBlockInfo* pBlockInfo = &pDataBlock->info;
5,116,605✔
410

411
  for (int32_t i = 0; i < taosArrayGetSize(pMatchInfo->pList); ++i) {
31,015,993✔
412
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
26,159,340✔
413
    if (!pItem) {
26,167,352✔
414
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
2,463!
415
      return terrno;
2,463✔
416
    }
417

418
    if (pItem->isPk) {
26,164,889✔
419
      SColumnInfoData* pInfoData = taosArrayGet(pDataBlock->pDataBlock, pItem->dstSlotId);
265,501✔
420
      if (!pInfoData) {
265,508!
421
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
422
        return terrno;
×
423
      }
424
      pBlockInfo->pks[0].type = pInfoData->info.type;
265,508✔
425
      pBlockInfo->pks[1].type = pInfoData->info.type;
265,508✔
426

427
      // allocate enough buffer size, which is pInfoData->info.bytes
428
      if (IS_VAR_DATA_TYPE(pItem->dataType.type)) {
265,508!
429
        pBlockInfo->pks[0].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
35,373✔
430
        if (pBlockInfo->pks[0].pData == NULL) {
35,347!
431
          return terrno;
×
432
        }
433

434
        pBlockInfo->pks[1].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
35,347✔
435
        if (pBlockInfo->pks[1].pData == NULL) {
35,351!
436
          taosMemoryFreeClear(pBlockInfo->pks[0].pData);
×
437
          return terrno;
×
438
        }
439

440
        pBlockInfo->pks[0].nData = pInfoData->info.bytes;
35,352✔
441
        pBlockInfo->pks[1].nData = pInfoData->info.bytes;
35,352✔
442
      }
443

444
      break;
265,487✔
445
    }
446
  }
447

448
  return TSDB_CODE_SUCCESS;
5,117,672✔
449
}
450

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

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

474
    res->translate = true;
1,510✔
475
    res->node.resType = pSColumnNode->node.resType;
1,510✔
476

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

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

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

525
  return DEAL_RES_CONTINUE;
5,250✔
526
}
527

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

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

538
    return TSDB_CODE_SUCCESS;
×
539
  }
540

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

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

564
    return code;
×
565
  }
566

567
  SValueNode* pValue = (SValueNode*)pNew;
748✔
568
  *pQualified = pValue->datum.b;
748✔
569

570
  nodesDestroyNode(pNew);
748✔
571
  return TSDB_CODE_SUCCESS;
748✔
572
}
573

574
static EDealRes getColumn(SNode** pNode, void* pContext) {
50,869,922✔
575
  tagFilterAssist* pData = (tagFilterAssist*)pContext;
50,869,922✔
576
  SColumnNode*     pSColumnNode = NULL;
50,869,922✔
577
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
50,869,922✔
578
    pSColumnNode = *(SColumnNode**)pNode;
3,583,582✔
579
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
47,286,340✔
580
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
477,319✔
581
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
477,319✔
582
      pData->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
465,707✔
583
      if (NULL == pSColumnNode) {
465,701!
584
        return DEAL_RES_ERROR;
×
585
      }
586
      pSColumnNode->colId = -1;
465,701✔
587
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
465,701✔
588
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
465,701✔
589
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
465,701✔
590
      nodesDestroyNode(*pNode);
465,701✔
591
      *pNode = (SNode*)pSColumnNode;
465,703✔
592
    } else {
593
      return DEAL_RES_CONTINUE;
11,612✔
594
    }
595
  } else {
596
    return DEAL_RES_CONTINUE;
46,809,021✔
597
  }
598

599
  void* data = taosHashGet(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
4,049,285✔
600
  if (!data) {
4,048,148✔
601
    int32_t tempRes =
602
        taosHashPut(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
3,390,706✔
603
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
3,392,778!
604
      return DEAL_RES_ERROR;
×
605
    }
606
    pSColumnNode->slotId = pData->index++;
3,392,778✔
607
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
3,392,778✔
608
                         .type = pSColumnNode->node.resType.type,
3,392,778✔
609
                         .bytes = pSColumnNode->node.resType.bytes,
3,392,778✔
610
                         .pk = pSColumnNode->isPk};
3,392,778✔
611
#if TAG_FILTER_DEBUG
612
    qDebug("tagfilter build column info, slotId:%d, colId:%d, type:%d", pSColumnNode->slotId, cInfo.colId, cInfo.type);
613
#endif
614
    void* tmp = taosArrayPush(pData->cInfoList, &cInfo);
3,392,778✔
615
    if (!tmp) {
3,392,657!
616
      return DEAL_RES_ERROR;
×
617
    }
618
  } else {
619
    SColumnNode* col = *(SColumnNode**)data;
657,442✔
620
    pSColumnNode->slotId = col->slotId;
657,442✔
621
  }
622

623
  return DEAL_RES_CONTINUE;
4,050,099✔
624
}
625

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

632
  pColumnData->info.type = pType->type;
1,392,382✔
633
  pColumnData->info.bytes = pType->bytes;
1,392,382✔
634
  pColumnData->info.scale = pType->scale;
1,392,382✔
635
  pColumnData->info.precision = pType->precision;
1,392,382✔
636

637
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
1,392,382✔
638
  if (code != TSDB_CODE_SUCCESS) {
1,391,978!
639
    terrno = code;
×
640
    releaseColInfoData(pColumnData);
×
641
    return terrno;
×
642
  }
643

644
  pParam->columnData = pColumnData;
1,392,166✔
645
  pParam->colAlloced = true;
1,392,166✔
646
  return TSDB_CODE_SUCCESS;
1,392,166✔
647
}
648

649
static void releaseColInfoData(void* pCol) {
79,016✔
650
  if (pCol) {
79,016!
651
    SColumnInfoData* col = (SColumnInfoData*)pCol;
79,019✔
652
    colDataDestroy(col);
79,019✔
653
    taosMemoryFree(col);
79,021✔
654
  }
655
}
79,024✔
656

657
void freeItem(void* p) {
3,278,289✔
658
  STUidTagInfo* pInfo = p;
3,278,289✔
659
  if (pInfo->pTagVal != NULL) {
3,278,289✔
660
    taosMemoryFree(pInfo->pTagVal);
3,090,942✔
661
  }
662
}
3,277,869✔
663

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

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

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

681
  taosMemoryFree(payload);
×
682
  return TSDB_CODE_SUCCESS;
×
683
}
684

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

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

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

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

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

723
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
66,322✔
724
  if (rows == 0) {
66,351!
725
    return TSDB_CODE_SUCCESS;
×
726
  }
727

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

735
  ctx.index = 0;
66,430✔
736
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
66,430✔
737
  if (ctx.cInfoList == NULL) {
66,463✔
738
    code = terrno;
20✔
739
    goto end;
×
740
  }
741

742
  SNode* pNode = NULL;
66,443✔
743
  FOREACH(pNode, group) {
145,433!
744
    nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
78,921✔
745
    if (TSDB_CODE_SUCCESS != ctx.code) {
78,990!
746
      code = ctx.code;
×
747
      goto end;
×
748
    }
749
    REPLACE_NODE(pNode);
78,990✔
750
  }
751

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

763
    nodesFree(listNode);
×
764

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

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

778
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
66,512✔
779
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
66,401!
780

781
  for (int32_t i = 0; i < rows; ++i) {
329,553✔
782
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
263,023✔
783
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
262,918!
784
    STUidTagInfo   info = {.uid = pkeyInfo->uid};
262,918✔
785
    void*          tmp = taosArrayPush(pUidTagList, &info);
263,141✔
786
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
263,141!
787
  }
788

789
  code = pAPI->metaFn.getTableTags(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
66,530✔
790
  if (code != TSDB_CODE_SUCCESS) {
66,472!
791
    goto end;
×
792
  }
793

794
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
66,472✔
795
  pResBlock = createTagValBlockForFilter(ctx.cInfoList, numOfTables, pUidTagList, pVnode, pAPI);
66,416✔
796
  if (pResBlock == NULL) {
66,441!
797
    code = terrno;
×
798
    goto end;
×
799
  }
800

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

804
  pBlockList = taosArrayInit(2, POINTER_BYTES);
66,441✔
805
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
66,472!
806

807
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
66,409✔
808
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
66,409!
809

810
  groupData = taosArrayInit(2, POINTER_BYTES);
66,409✔
811
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
66,468!
812

813
  FOREACH(pNode, group) {
145,443!
814
    SScalarParam output = {0};
78,927✔
815

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

830
      default:
×
831
        code = TSDB_CODE_OPS_NOT_SUPPORT;
×
832
        goto end;
×
833
    }
834

835
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
78,996✔
836
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
74,756✔
837
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
74,756✔
838
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
74,728!
839
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
74,728✔
840
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
4,240!
841
      continue;
×
842
    } else {
843
      code = scalarCalculate(pNode, pBlockList, &output);
4,240✔
844
    }
845

846
    if (code != TSDB_CODE_SUCCESS) {
78,992!
847
      releaseColInfoData(output.columnData);
×
848
      goto end;
×
849
    }
850

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

855
  int32_t keyLen = 0;
66,516✔
856
  SNode*  node;
857
  FOREACH(node, group) {
145,497✔
858
    SExprNode* pExpr = (SExprNode*)node;
78,981✔
859
    keyLen += pExpr->resType.bytes;
78,981✔
860
  }
861

862
  int32_t nullFlagSize = sizeof(int8_t) * LIST_LENGTH(group);
66,516✔
863
  keyLen += nullFlagSize;
66,516✔
864

865
  keyBuf = taosMemoryCalloc(1, keyLen);
66,516✔
866
  if (keyBuf == NULL) {
66,417!
867
    code = terrno;
×
868
    goto end;
×
869
  }
870

871
  if (initRemainGroups) {
66,417✔
872
    pTableListInfo->remainGroups =
13,978✔
873
        taosHashInit(rows, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
13,984✔
874
    if (pTableListInfo->remainGroups == NULL) {
13,978!
875
      code = terrno;
×
876
      goto end;
×
877
    }
878
  }
879

880
  for (int i = 0; i < rows; i++) {
329,506✔
881
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
262,971✔
882
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
262,857!
883

884
    char* isNull = (char*)keyBuf;
262,875✔
885
    char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(group);
262,875!
886
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
556,282✔
887
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
293,416✔
888

889
      if (colDataIsNull_s(pValue, i)) {
586,844✔
890
        isNull[j] = 1;
488✔
891
      } else {
892
        isNull[j] = 0;
292,934✔
893
        char* data = colDataGetData(pValue, i);
292,934!
894
        if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
292,934✔
895
          if (tTagIsJson(data)) {
264✔
896
            code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
12✔
897
            goto end;
12✔
898
          }
899
          if (tTagIsJsonNull(data)) {
252!
900
            isNull[j] = 1;
×
901
            continue;
×
902
          }
903
          int32_t len = getJsonValueLen(data);
252✔
904
          memcpy(pStart, data, len);
252✔
905
          pStart += len;
252✔
906
        } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
292,670!
907
          if (varDataTLen(data) > pValue->info.bytes) {
256,877✔
908
            code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
3✔
909
            goto end;
3✔
910
          }
911
          memcpy(pStart, data, varDataTLen(data));
256,874✔
912
          pStart += varDataTLen(data);
256,874✔
913
        } else {
914
          memcpy(pStart, data, pValue->info.bytes);
35,793✔
915
          pStart += pValue->info.bytes;
35,793✔
916
        }
917
      }
918
    }
919

920
    int32_t len = (int32_t)(pStart - (char*)keyBuf);
261,927✔
921
    info->groupId = calcGroupId(keyBuf, len);
261,927✔
922
    if (initRemainGroups) {
263,126✔
923
      // groupId ~ table uid
924
      code = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId), &(info->uid),
80,721✔
925
                         sizeof(info->uid));
926
      if (code == TSDB_CODE_DUP_KEY) {
80,690✔
927
        code = TSDB_CODE_SUCCESS;
1,291✔
928
      }
929
      QUERY_CHECK_CODE(code, lino, end);
80,690!
930
    }
931
  }
932

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

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

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

946
end:
66,535✔
947
  taosMemoryFreeClear(keyBuf);
66,550✔
948
  taosHashCleanup(ctx.colHash);
66,551✔
949
  taosArrayDestroy(ctx.cInfoList);
66,452✔
950
  blockDataDestroy(pResBlock);
66,458✔
951
  taosArrayDestroy(pBlockList);
66,471✔
952
  taosArrayDestroyEx(pUidTagList, freeItem);
66,476✔
953
  taosArrayDestroyP(groupData, releaseColInfoData);
66,475✔
954
  if (code != TSDB_CODE_SUCCESS) {
66,421✔
955
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
15!
956
  }
957
  return code;
66,435✔
958
}
959

960
static int32_t nameComparFn(const void* p1, const void* p2) {
3,861✔
961
  const char* pName1 = *(const char**)p1;
3,861✔
962
  const char* pName2 = *(const char**)p2;
3,861✔
963

964
  int32_t ret = strcmp(pName1, pName2);
3,861✔
965
  if (ret == 0) {
3,861✔
966
    return 0;
16✔
967
  } else {
968
    return (ret > 0) ? 1 : -1;
3,845✔
969
  }
970
}
971

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

978
  SArray* pTbList = taosArrayInit(len, POINTER_BYTES);
460,294✔
979
  QUERY_CHECK_NULL(pTbList, code, lino, _end, terrno);
460,303✔
980

981
  for (int i = 0; i < pList->pNodeList->length; i++) {
923,256✔
982
    SValueNode* valueNode = (SValueNode*)cell->pNode;
462,950✔
983
    if (!IS_VAR_DATA_TYPE(valueNode->node.resType.type)) {
462,950!
984
      terrno = TSDB_CODE_INVALID_PARA;
×
985
      taosArrayDestroy(pTbList);
×
986
      return NULL;
×
987
    }
988

989
    char* name = varDataVal(valueNode->datum.p);
462,950✔
990
    void* tmp = taosArrayPush(pTbList, &name);
462,954✔
991
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
462,954!
992
    cell = cell->pNext;
462,954✔
993
  }
994

995
  size_t numOfTables = taosArrayGetSize(pTbList);
460,306✔
996

997
  // order the name
998
  taosArraySort(pTbList, nameComparFn);
460,297✔
999

1000
  // remove the duplicates
1001
  SArray* pNewList = taosArrayInit(taosArrayGetSize(pTbList), sizeof(void*));
460,298✔
1002
  QUERY_CHECK_NULL(pNewList, code, lino, _end, terrno);
460,302!
1003
  void*   tmpTbl = taosArrayGet(pTbList, 0);
460,302✔
1004
  QUERY_CHECK_NULL(tmpTbl, code, lino, _end, terrno);
460,298!
1005
  void*   tmp = taosArrayPush(pNewList, tmpTbl);
460,298✔
1006
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
460,298!
1007

1008
  for (int32_t i = 1; i < numOfTables; ++i) {
462,953✔
1009
    char** name = taosArrayGetLast(pNewList);
2,653✔
1010
    char** nameInOldList = taosArrayGet(pTbList, i);
2,654✔
1011
    QUERY_CHECK_NULL(nameInOldList, code, lino, _end, terrno);
2,655!
1012
    if (strcmp(*name, *nameInOldList) == 0) {
2,655✔
1013
      continue;
16✔
1014
    }
1015

1016
    tmp = taosArrayPush(pNewList, nameInOldList);
2,639✔
1017
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
2,639!
1018
  }
1019

1020
_end:
460,300✔
1021
  taosArrayDestroy(pTbList);
460,300✔
1022
  if (code != TSDB_CODE_SUCCESS) {
460,298!
1023
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1024
    return NULL;
×
1025
  }
1026
  return pNewList;
460,298✔
1027
}
1028

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

1033
  if (u1 == u2) {
×
1034
    return 0;
×
1035
  }
1036

1037
  return u1 < u2 ? -1 : 1;
×
1038
}
1039

1040
static int32_t filterTableInfoCompare(const void* a, const void* b) {
5,388✔
1041
  STUidTagInfo* p1 = (STUidTagInfo*)a;
5,388✔
1042
  STUidTagInfo* p2 = (STUidTagInfo*)b;
5,388✔
1043

1044
  if (p1->uid == p2->uid) {
5,388!
1045
    return 0;
×
1046
  }
1047

1048
  return p1->uid < p2->uid ? -1 : 1;
5,388!
1049
}
1050

1051
static FilterCondType checkTagCond(SNode* cond) {
1,368,491✔
1052
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
1,368,491✔
1053
    return FILTER_NO_LOGIC;
348,477✔
1054
  }
1055
  if (nodeType(cond) != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
1,020,014!
1056
    return FILTER_AND;
10,631✔
1057
  }
1058
  return FILTER_OTHER;
1,009,383✔
1059
}
1060

1061
static int32_t optimizeTbnameInCond(void* pVnode, int64_t suid, SArray* list, SNode* cond, SStorageAPI* pAPI) {
1,368,452✔
1062
  int32_t ret = -1;
1,368,452✔
1063
  int32_t ntype = nodeType(cond);
1,368,452✔
1064

1065
  if (ntype == QUERY_NODE_OPERATOR) {
1,368,452✔
1066
    ret = optimizeTbnameInCondImpl(pVnode, list, cond, pAPI);
348,465✔
1067
  }
1068

1069
  if (ntype != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
1,369,504✔
1070
    return ret;
360,242✔
1071
  }
1072

1073
  bool                 hasTbnameCond = false;
1,009,262✔
1074
  SLogicConditionNode* pNode = (SLogicConditionNode*)cond;
1,009,262✔
1075
  SNodeList*           pList = (SNodeList*)pNode->pParameterList;
1,009,262✔
1076

1077
  int32_t len = LIST_LENGTH(pList);
1,009,262✔
1078
  if (len <= 0) {
1,009,262!
1079
    return ret;
×
1080
  }
1081

1082
  SListCell* cell = pList->pHead;
1,009,262✔
1083
  for (int i = 0; i < len; i++) {
3,330,710✔
1084
    if (cell == NULL) break;
2,594,288!
1085
    if (optimizeTbnameInCondImpl(pVnode, list, cell->pNode, pAPI) == 0) {
2,594,288✔
1086
      hasTbnameCond = true;
272,218✔
1087
      break;
272,218✔
1088
    }
1089
    cell = cell->pNext;
2,321,448✔
1090
  }
1091

1092
  taosArraySort(list, filterTableInfoCompare);
1,008,640✔
1093
  taosArrayRemoveDuplicate(list, filterTableInfoCompare, NULL);
1,007,642✔
1094

1095
  if (hasTbnameCond) {
1,007,962✔
1096
    ret = pAPI->metaFn.getTableTagsByUid(pVnode, suid, list);
272,216✔
1097
  }
1098

1099
  return ret;
1,008,061✔
1100
}
1101

1102
// only return uid that does not contained in pExistedUidList
1103
static int32_t optimizeTbnameInCondImpl(void* pVnode, SArray* pExistedUidList, SNode* pTagCond,
2,941,644✔
1104
                                        SStorageAPI* pStoreAPI) {
1105
  if (nodeType(pTagCond) != QUERY_NODE_OPERATOR) {
2,941,644✔
1106
    return -1;
168,951✔
1107
  }
1108

1109
  SOperatorNode* pNode = (SOperatorNode*)pTagCond;
2,772,693✔
1110
  if (pNode->opType != OP_TYPE_IN) {
2,772,693✔
1111
    return -1;
1,952,403✔
1112
  }
1113

1114
  if ((pNode->pLeft != NULL && nodeType(pNode->pLeft) == QUERY_NODE_COLUMN &&
820,290!
1115
       ((SColumnNode*)pNode->pLeft)->colType == COLUMN_TYPE_TBNAME) &&
821,853✔
1116
      (pNode->pRight != NULL && nodeType(pNode->pRight) == QUERY_NODE_NODE_LIST)) {
460,298!
1117
    SNodeListNode* pList = (SNodeListNode*)pNode->pRight;
460,298✔
1118

1119
    int32_t len = LIST_LENGTH(pList->pNodeList);
460,298!
1120
    if (len <= 0) {
460,298!
1121
      return -1;
×
1122
    }
1123

1124
    SArray*   pTbList = getTableNameList(pList);
460,298✔
1125
    int32_t   numOfTables = taosArrayGetSize(pTbList);
460,298✔
1126
    SHashObj* uHash = NULL;
460,298✔
1127

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

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

1150
    for (int i = 0; i < numOfTables; i++) {
923,204✔
1151
      char* name = taosArrayGetP(pTbList, i);
462,914✔
1152

1153
      uint64_t uid = 0;
462,915✔
1154
      if (pStoreAPI->metaFn.getTableUidByName(pVnode, name, &uid) == 0) {
462,915✔
1155
        ETableType tbType = TSDB_TABLE_MAX;
460,312✔
1156
        if (pStoreAPI->metaFn.getTableTypeByName(pVnode, name, &tbType) == 0 && tbType == TSDB_CHILD_TABLE) {
460,312!
1157
          if (NULL == uHash || taosHashGet(uHash, &uid, sizeof(uid)) == NULL) {
460,301!
1158
            STUidTagInfo s = {.uid = uid, .name = name, .pTagVal = NULL};
460,303✔
1159
            void*        tmp = taosArrayPush(pExistedUidList, &s);
460,298✔
1160
            if (!tmp) {
460,298!
1161
              return terrno;
×
1162
            }
1163
          }
1164
        } else {
1165
          taosArrayDestroy(pTbList);
10✔
1166
          taosHashCleanup(uHash);
11✔
1167
          return -1;
11✔
1168
        }
1169
      } else {
1170
        //        qWarn("failed to get tableIds from by table name: %s, reason: %s", name, tstrerror(terrno));
1171
        terrno = 0;
2,614✔
1172
      }
1173
    }
1174

1175
    taosHashCleanup(uHash);
460,290✔
1176
    taosArrayDestroy(pTbList);
460,286✔
1177
    return 0;
460,288✔
1178
  }
1179

1180
  return -1;
359,992✔
1181
}
1182

1183
SSDataBlock* createTagValBlockForFilter(SArray* pColList, int32_t numOfTables, SArray* pUidTagList, void* pVnode,
1,394,012✔
1184
                                        SStorageAPI* pStorageAPI) {
1185
  int32_t      code = TSDB_CODE_SUCCESS;
1,394,012✔
1186
  int32_t      lino = 0;
1,394,012✔
1187
  SSDataBlock* pResBlock = NULL;
1,394,012✔
1188
  code = createDataBlock(&pResBlock);
1,394,012✔
1189
  QUERY_CHECK_CODE(code, lino, _end);
1,394,316!
1190

1191
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
4,664,373✔
1192
    SColumnInfoData colInfo = {0};
3,269,457✔
1193
    void* tmp = taosArrayGet(pColList, i);
3,269,457✔
1194
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
3,268,367!
1195
    colInfo.info = *(SColumnInfo*)tmp;
3,268,367✔
1196
    code = blockDataAppendColInfo(pResBlock, &colInfo);
3,268,367✔
1197
    QUERY_CHECK_CODE(code, lino, _end);
3,270,057!
1198
  }
1199

1200
  code = blockDataEnsureCapacity(pResBlock, numOfTables);
1,393,007✔
1201
  if (code != TSDB_CODE_SUCCESS) {
1,394,767!
1202
    terrno = code;
×
1203
    blockDataDestroy(pResBlock);
×
1204
    return NULL;
×
1205
  }
1206

1207
  pResBlock->info.rows = numOfTables;
1,394,767✔
1208

1209
  int32_t numOfCols = taosArrayGetSize(pResBlock->pDataBlock);
1,394,767✔
1210

1211
  for (int32_t i = 0; i < numOfTables; i++) {
4,697,007✔
1212
    STUidTagInfo* p1 = taosArrayGet(pUidTagList, i);
3,295,276✔
1213
    QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
3,294,180✔
1214

1215
    for (int32_t j = 0; j < numOfCols; j++) {
10,531,146✔
1216
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, j);
7,228,686✔
1217
      QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
7,224,260!
1218

1219
      if (pColInfo->info.colId == -1) {  // tbname
7,224,592✔
1220
        char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
477,224✔
1221
        if (p1->name != NULL) {
477,224✔
1222
          STR_TO_VARSTR(str, p1->name);
460,299✔
1223
        } else {  // name is not retrieved during filter
1224
          code = pStorageAPI->metaFn.getTableNameByUid(pVnode, p1->uid, str);
16,925✔
1225
          QUERY_CHECK_CODE(code, lino, _end);
16,975!
1226
        }
1227

1228
        code = colDataSetVal(pColInfo, i, str, false);
477,274✔
1229
        QUERY_CHECK_CODE(code, lino, _end);
477,263!
1230
#if TAG_FILTER_DEBUG
1231
        qDebug("tagfilter uid:%ld, tbname:%s", *uid, str + 2);
1232
#endif
1233
      } else {
1234
        STagVal tagVal = {0};
6,747,368✔
1235
        tagVal.cid = pColInfo->info.colId;
6,747,368✔
1236
        if (p1->pTagVal == NULL) {
6,747,368!
1237
          colDataSetNULL(pColInfo, i);
×
1238
        } else {
1239
          const char* p = pStorageAPI->metaFn.extractTagVal(p1->pTagVal, pColInfo->info.type, &tagVal);
6,747,368✔
1240

1241
          if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
6,766,995✔
1242
            colDataSetNULL(pColInfo, i);
3,118✔
1243
          } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
6,763,877✔
1244
            code = colDataSetVal(pColInfo, i, p, false);
4,860✔
1245
            QUERY_CHECK_CODE(code, lino, _end);
4,860!
1246
          } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
8,968,838!
1247
            char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
2,211,806✔
1248
            QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
2,211,546!
1249
            varDataSetLen(tmp, tagVal.nData);
2,211,546✔
1250
            memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
2,211,546✔
1251
            code = colDataSetVal(pColInfo, i, tmp, false);
2,211,546✔
1252
#if TAG_FILTER_DEBUG
1253
            qDebug("tagfilter varch:%s", tmp + 2);
1254
#endif
1255
            taosMemoryFree(tmp);
2,209,660✔
1256
            QUERY_CHECK_CODE(code, lino, _end);
2,209,821!
1257
          } else {
1258
            code = colDataSetVal(pColInfo, i, (const char*)&tagVal.i64, false);
4,547,211✔
1259
            QUERY_CHECK_CODE(code, lino, _end);
4,542,229!
1260
#if TAG_FILTER_DEBUG
1261
            if (pColInfo->info.type == TSDB_DATA_TYPE_INT) {
1262
              qDebug("tagfilter int:%d", *(int*)(&tagVal.i64));
1263
            } else if (pColInfo->info.type == TSDB_DATA_TYPE_DOUBLE) {
1264
              qDebug("tagfilter double:%f", *(double*)(&tagVal.i64));
1265
            }
1266
#endif
1267
          }
1268
        }
1269
      }
1270
    }
1271
  }
1272

1273
_end:
1,401,731✔
1274
  if (code != TSDB_CODE_SUCCESS) {
1,401,731!
1275
    blockDataDestroy(pResBlock);
×
1276
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1277
    terrno = code;
×
1278
    return NULL;
×
1279
  }
1280
  return pResBlock;
1,401,731✔
1281
}
1282

1283
static int32_t doSetQualifiedUid(STableListInfo* pListInfo, SArray* pUidList, const SArray* pUidTagList,
1,311,889✔
1284
                                 bool* pResultList, bool addUid) {
1285
  taosArrayClear(pUidList);
1,311,889✔
1286

1287
  STableKeyInfo info = {.uid = 0, .groupId = 0};
1,311,857✔
1288
  int32_t       numOfTables = taosArrayGetSize(pUidTagList);
1,311,857✔
1289
  for (int32_t i = 0; i < numOfTables; ++i) {
4,329,580✔
1290
    if (pResultList[i]) {
3,016,370✔
1291
      STUidTagInfo* tmpTag = (STUidTagInfo*)taosArrayGet(pUidTagList, i);
2,311,203✔
1292
      if (!tmpTag) {
2,310,411!
1293
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1294
        return terrno;
×
1295
      }
1296
      uint64_t uid = tmpTag->uid;
2,310,411✔
1297
      qDebug("tagfilter get uid:%" PRId64 ", res:%d", uid, pResultList[i]);
2,310,411✔
1298

1299
      info.uid = uid;
2,310,432✔
1300
      void* p = taosArrayPush(pListInfo->pTableList, &info);
2,310,432✔
1301
      if (p == NULL) {
2,312,017!
1302
        return terrno;
×
1303
      }
1304

1305
      if (addUid) {
2,312,017!
1306
        void* tmp = taosArrayPush(pUidList, &uid);
×
1307
        if (tmp == NULL) {
×
1308
          return terrno;
×
1309
        }
1310
      }
1311
    }
1312
  }
1313

1314
  return TSDB_CODE_SUCCESS;
1,313,210✔
1315
}
1316

1317
static int32_t copyExistedUids(SArray* pUidTagList, const SArray* pUidList) {
1,368,591✔
1318
  int32_t code = TSDB_CODE_SUCCESS;
1,368,591✔
1319
  int32_t numOfExisted = taosArrayGetSize(pUidList);
1,368,591✔
1320
  if (numOfExisted == 0) {
1,368,979✔
1321
    return code;
1,337,930✔
1322
  }
1323

1324
  for (int32_t i = 0; i < numOfExisted; ++i) {
64,875✔
1325
    uint64_t*    uid = taosArrayGet(pUidList, i);
33,825✔
1326
    if (!uid) {
33,824!
1327
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1328
      return terrno;
×
1329
    }
1330
    STUidTagInfo info = {.uid = *uid};
33,824✔
1331
    void*        tmp = taosArrayPush(pUidTagList, &info);
33,826✔
1332
    if (!tmp) {
33,826!
1333
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_OUT_OF_MEMORY));
×
1334
      return code;
×
1335
    }
1336
  }
1337
  return code;
31,050✔
1338
}
1339

1340
static int32_t doFilterByTagCond(STableListInfo* pListInfo, SArray* pUidList, SNode* pTagCond, void* pVnode,
5,089,974✔
1341
                                 SIdxFltStatus status, SStorageAPI* pAPI, bool addUid, bool* listAdded) {
1342
  *listAdded = false;
5,089,974✔
1343
  if (pTagCond == NULL) {
5,089,974✔
1344
    return TSDB_CODE_SUCCESS;
3,722,166✔
1345
  }
1346

1347
  terrno = TSDB_CODE_SUCCESS;
1,367,808✔
1348

1349
  int32_t      lino = 0;
1,369,482✔
1350
  int32_t      code = TSDB_CODE_SUCCESS;
1,369,482✔
1351
  SArray*      pBlockList = NULL;
1,369,482✔
1352
  SSDataBlock* pResBlock = NULL;
1,369,482✔
1353
  SScalarParam output = {0};
1,369,482✔
1354
  SArray*      pUidTagList = NULL;
1,369,482✔
1355

1356
  tagFilterAssist ctx = {0};
1,369,482✔
1357
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
1,369,482✔
1358
  if (ctx.colHash == NULL) {
1,370,185!
1359
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1360
    code = TSDB_CODE_OUT_OF_MEMORY;
×
1361
    QUERY_CHECK_CODE(code, lino, end);
×
1362
  }
1363

1364
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
1,370,185✔
1365
  if (ctx.cInfoList == NULL) {
1,370,501!
1366
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1367
    code = TSDB_CODE_OUT_OF_MEMORY;
×
1368
    QUERY_CHECK_CODE(code, lino, end);
×
1369
  }
1370

1371
  nodesRewriteExprPostOrder(&pTagCond, getColumn, (void*)&ctx);
1,370,501✔
1372
  if (TSDB_CODE_SUCCESS != ctx.code) {
1,368,637!
1373
    terrno = code = ctx.code;
×
1374
    goto end;
×
1375
  }
1376

1377
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
1,368,637✔
1378

1379
  //  int64_t stt = taosGetTimestampUs();
1380
  pUidTagList = taosArrayInit(10, sizeof(STUidTagInfo));
1,368,637✔
1381
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
1,368,991!
1382

1383
  code = copyExistedUids(pUidTagList, pUidList);
1,368,991✔
1384
  QUERY_CHECK_CODE(code, lino, end);
1,369,601!
1385

1386
  FilterCondType condType = checkTagCond(pTagCond);
1,369,601✔
1387

1388
  int32_t filter = optimizeTbnameInCond(pVnode, pListInfo->idInfo.suid, pUidTagList, pTagCond, pAPI);
1,368,839✔
1389
  if (filter == 0) {  // tbname in filter is activated, do nothing and return
1,367,727✔
1390
    taosArrayClear(pUidList);
460,290✔
1391

1392
    int32_t numOfRows = taosArrayGetSize(pUidTagList);
460,290✔
1393
    code = taosArrayEnsureCap(pUidList, numOfRows);
460,290✔
1394
    QUERY_CHECK_CODE(code, lino, end);
460,290!
1395

1396
    for (int32_t i = 0; i < numOfRows; ++i) {
920,589✔
1397
      STUidTagInfo* pInfo = taosArrayGet(pUidTagList, i);
460,298✔
1398
      QUERY_CHECK_NULL(pInfo, code, lino, end, terrno);
460,298!
1399
      void*         tmp = taosArrayPush(pUidList, &pInfo->uid);
460,298✔
1400
      QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
460,299!
1401
    }
1402
    terrno = 0;
460,291✔
1403
  } else {
1404
    if ((condType == FILTER_NO_LOGIC || condType == FILTER_AND) && status != SFLT_NOT_INDEX) {
907,437✔
1405
      code = pAPI->metaFn.getTableTagsByUid(pVnode, pListInfo->idInfo.suid, pUidTagList);
32✔
1406
    } else {
1407
      code = pAPI->metaFn.getTableTags(pVnode, pListInfo->idInfo.suid, pUidTagList);
907,405✔
1408
    }
1409
    if (code != TSDB_CODE_SUCCESS) {
909,997!
1410
      qError("failed to get table tags from meta, reason:%s, suid:%" PRIu64, tstrerror(code), pListInfo->idInfo.suid);
×
1411
      terrno = code;
×
1412
      QUERY_CHECK_CODE(code, lino, end);
×
1413
    }
1414
  }
1415

1416
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
1,370,285✔
1417
  if (numOfTables == 0) {
1,369,693✔
1418
    goto end;
57,168✔
1419
  }
1420

1421
  pResBlock = createTagValBlockForFilter(ctx.cInfoList, numOfTables, pUidTagList, pVnode, pAPI);
1,312,525✔
1422
  if (pResBlock == NULL) {
1,312,921!
1423
    code = terrno;
×
1424
    QUERY_CHECK_CODE(code, lino, end);
×
1425
  }
1426

1427
  //  int64_t st1 = taosGetTimestampUs();
1428
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1429
  pBlockList = taosArrayInit(2, POINTER_BYTES);
1,312,921✔
1430
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
1,313,165!
1431

1432
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
1,313,155✔
1433
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
1,313,155!
1434

1435
  code = createResultData(&type, numOfTables, &output);
1,313,155✔
1436
  if (code != TSDB_CODE_SUCCESS) {
1,313,164!
1437
    terrno = code;
×
1438
    QUERY_CHECK_CODE(code, lino, end);
×
1439
  }
1440

1441
  code = scalarCalculate(pTagCond, pBlockList, &output);
1,313,164✔
1442
  if (code != TSDB_CODE_SUCCESS) {
1,312,681✔
1443
    qError("failed to calculate scalar, reason:%s", tstrerror(code));
8!
1444
    terrno = code;
8✔
1445
    QUERY_CHECK_CODE(code, lino, end);
8!
1446
  }
1447

1448
  code = doSetQualifiedUid(pListInfo, pUidList, pUidTagList, (bool*)output.columnData->pData, addUid);
1,312,673✔
1449
  if (code != TSDB_CODE_SUCCESS) {
1,312,817!
1450
    terrno = code;
×
1451
    QUERY_CHECK_CODE(code, lino, end);
×
1452
  }
1453
  *listAdded = true;
1,312,853✔
1454

1455
end:
1,370,029✔
1456
  if (code != TSDB_CODE_SUCCESS) {
1,370,029✔
1457
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
8!
1458
  }
1459
  taosHashCleanup(ctx.colHash);
1,370,029✔
1460
  taosArrayDestroy(ctx.cInfoList);
1,370,328✔
1461
  blockDataDestroy(pResBlock);
1,370,494✔
1462
  taosArrayDestroy(pBlockList);
1,370,546✔
1463
  taosArrayDestroyEx(pUidTagList, freeItem);
1,370,551✔
1464

1465
  colDataDestroy(output.columnData);
1,370,475✔
1466
  taosMemoryFreeClear(output.columnData);
1,370,564✔
1467
  return code;
1,370,593✔
1468
}
1469

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

1477
  pListInfo->idInfo.suid = pScanNode->suid;
5,085,388✔
1478
  pListInfo->idInfo.tableType = pScanNode->tableType;
5,085,388✔
1479

1480
  SArray* pUidList = taosArrayInit(8, sizeof(uint64_t));
5,085,388✔
1481
  QUERY_CHECK_NULL(pUidList, code, lino, _error, terrno);
5,094,302✔
1482

1483
  SIdxFltStatus status = SFLT_NOT_INDEX;
5,091,752✔
1484
  if (pScanNode->tableType != TSDB_SUPER_TABLE) {
5,091,752✔
1485
    pListInfo->idInfo.uid = pScanNode->uid;
779,034✔
1486
    if (pStorageAPI->metaFn.isTableExisted(pVnode, pScanNode->uid)) {
779,034✔
1487
      void* tmp = taosArrayPush(pUidList, &pScanNode->uid);
777,320✔
1488
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
777,318!
1489
    }
1490
    code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, status, pStorageAPI, false, &listAdded);
779,040✔
1491
    QUERY_CHECK_CODE(code, lino, _end);
779,038!
1492
  } else {
1493
    T_MD5_CTX context = {0};
4,312,718✔
1494

1495
    if (tsTagFilterCache) {
4,312,718!
1496
      // try to retrieve the result from meta cache
1497
      code = genTagFilterDigest(pTagCond, &context);
×
1498
      QUERY_CHECK_CODE(code, lino, _error);
×
1499

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

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

1513
    if (!pTagCond) {  // no tag filter condition exists, let's fetch all tables of this super table
4,312,718✔
1514
      code = pStorageAPI->metaFn.getChildTableList(pVnode, pScanNode->suid, pUidList);
2,970,347✔
1515
      QUERY_CHECK_CODE(code, lino, _error);
2,975,975!
1516
    } else {
1517
      // failed to find the result in the cache, let try to calculate the results
1518
      if (pTagIndexCond) {
1,342,371✔
1519
        void* pIndex = pStorageAPI->metaFn.getInvertIndex(pVnode);
66,233✔
1520

1521
        SIndexMetaArg metaArg = {.metaEx = pVnode,
132,517✔
1522
                                 .idx = pStorageAPI->metaFn.storeGetIndexInfo(pVnode),
66,258✔
1523
                                 .ivtIdx = pIndex,
1524
                                 .suid = pScanNode->uid};
66,259✔
1525

1526
        status = SFLT_NOT_INDEX;
66,259✔
1527
        code = doFilterTag(pTagIndexCond, &metaArg, pUidList, &status, &pStorageAPI->metaFilter);
66,259✔
1528
        if (code != 0 || status == SFLT_NOT_INDEX) {  // temporarily disable it for performance sake
66,286✔
1529
          qDebug("failed to get tableIds from index, suid:%" PRIu64, pScanNode->uid);
66,186✔
1530
        } else {
1531
          qDebug("succ to get filter result, table num: %d", (int)taosArrayGetSize(pUidList));
100✔
1532
        }
1533
      }
1534
    }
1535

1536
    code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, status, pStorageAPI, tsTagFilterCache, &listAdded);
4,318,323✔
1537
    QUERY_CHECK_CODE(code, lino, _end);
4,312,193✔
1538

1539
    // let's add the filter results into meta-cache
1540
    numOfTables = taosArrayGetSize(pUidList);
4,312,185✔
1541

1542
    if (tsTagFilterCache) {
4,312,040✔
1543
      size_t size = numOfTables * sizeof(uint64_t) + sizeof(int32_t);
470✔
1544
      char*  pPayload = taosMemoryMalloc(size);
470✔
1545
      QUERY_CHECK_NULL(pPayload, code, lino, _end, terrno);
×
1546

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

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

1558
      digest[0] = 1;
×
1559
      memcpy(digest + 1, context.digest, tListLen(context.digest));
×
1560
    }
1561
  }
1562

1563
_end:
5,090,616✔
1564
  if (!listAdded) {
5,090,616✔
1565
    numOfTables = taosArrayGetSize(pUidList);
3,778,291✔
1566
    for (int i = 0; i < numOfTables; i++) {
14,354,913✔
1567
      void* tmp = taosArrayGet(pUidList, i);
10,573,784✔
1568
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
10,569,610!
1569
      STableKeyInfo info = {.uid = *(uint64_t*)tmp, .groupId = 0};
10,569,610✔
1570

1571
      void* p = taosArrayPush(pListInfo->pTableList, &info);
10,569,610✔
1572
      if (p == NULL) {
10,576,277!
1573
        taosArrayDestroy(pUidList);
×
1574
        return terrno;
×
1575
      }
1576

1577
      qTrace("tagfilter get uid:%" PRIu64 ", %s", info.uid, idstr);
10,576,277✔
1578
    }
1579
  }
1580

1581
_error:
5,093,454✔
1582
  taosArrayDestroy(pUidList);
5,093,454✔
1583
  if (code != TSDB_CODE_SUCCESS) {
5,093,430✔
1584
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
8!
1585
  }
1586
  return code;
5,093,204✔
1587
}
1588

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

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

1609
_end:
80✔
1610
  if (code != TSDB_CODE_SUCCESS) {
80!
1611
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1612
  }
1613
  return code;
81✔
1614
}
1615

1616
size_t getTableTagsBufLen(const SNodeList* pGroups) {
8,837✔
1617
  size_t keyLen = 0;
8,837✔
1618

1619
  SNode* node;
1620
  FOREACH(node, pGroups) {
53,034!
1621
    SExprNode* pExpr = (SExprNode*)node;
44,197✔
1622
    keyLen += pExpr->resType.bytes;
44,197✔
1623
  }
1624

1625
  keyLen += sizeof(int8_t) * LIST_LENGTH(pGroups);
8,837✔
1626
  return keyLen;
8,837✔
1627
}
1628

1629
int32_t getGroupIdFromTagsVal(void* pVnode, uint64_t uid, SNodeList* pGroupNode, char* keyBuf, uint64_t* pGroupId,
14✔
1630
                              SStorageAPI* pAPI) {
1631
  SMetaReader mr = {0};
14✔
1632

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

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

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

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

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

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

1703
  int32_t len = (int32_t)(pStart - (char*)keyBuf);
14✔
1704
  *pGroupId = calcGroupId(keyBuf, len);
14✔
1705

1706
  nodesDestroyList(groupNew);
14✔
1707
  pAPI->metaReaderFn.clearReader(&mr);
14✔
1708
  return TSDB_CODE_SUCCESS;
14✔
1709
}
1710

1711
SArray* makeColumnArrayFromList(SNodeList* pNodeList) {
27,925✔
1712
  if (!pNodeList) {
27,925!
1713
    return NULL;
×
1714
  }
1715

1716
  size_t  numOfCols = LIST_LENGTH(pNodeList);
27,925!
1717
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
27,925✔
1718
  if (pList == NULL) {
27,919!
1719
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1720
    return NULL;
×
1721
  }
1722

1723
  for (int32_t i = 0; i < numOfCols; ++i) {
105,906✔
1724
    SColumnNode* pColNode = (SColumnNode*)nodesListGetNode(pNodeList, i);
77,937✔
1725
    if (!pColNode) {
77,885!
1726
      taosArrayDestroy(pList);
×
1727
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_OUT_OF_MEMORY));
×
1728
      return NULL;
×
1729
    }
1730

1731
    // todo extract method
1732
    SColumn c = {0};
77,885✔
1733
    c.slotId = pColNode->slotId;
77,885✔
1734
    c.colId = pColNode->colId;
77,885✔
1735
    c.type = pColNode->node.resType.type;
77,885✔
1736
    c.bytes = pColNode->node.resType.bytes;
77,885✔
1737
    c.precision = pColNode->node.resType.precision;
77,885✔
1738
    c.scale = pColNode->node.resType.scale;
77,885✔
1739

1740
    void* tmp = taosArrayPush(pList, &c);
77,987✔
1741
    if (!tmp) {
77,987!
1742
      taosArrayDestroy(pList);
×
1743
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_OUT_OF_MEMORY));
×
1744
      return NULL;
×
1745
    }
1746
  }
1747

1748
  return pList;
27,969✔
1749
}
1750

1751
int32_t extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols,
8,064,404✔
1752
                            int32_t type, SColMatchInfo* pMatchInfo) {
1753
  size_t  numOfCols = LIST_LENGTH(pNodeList);
8,064,404!
1754
  int32_t code = TSDB_CODE_SUCCESS;
8,064,404✔
1755
  int32_t lino = 0;
8,064,404✔
1756

1757
  pMatchInfo->matchType = type;
8,064,404✔
1758

1759
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchItem));
8,064,404✔
1760
  if (pList == NULL) {
8,066,086!
1761
    code = terrno;
×
1762
    return code;
×
1763
  }
1764

1765
  for (int32_t i = 0; i < numOfCols; ++i) {
46,813,088✔
1766
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
38,736,955✔
1767
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
38,729,693!
1768
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
38,733,224✔
1769
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
38,640,210✔
1770

1771
      SColMatchItem c = {.needOutput = true};
38,640,210✔
1772
      c.colId = pColNode->colId;
38,640,210✔
1773
      c.srcSlotId = pColNode->slotId;
38,640,210✔
1774
      c.dstSlotId = pNode->slotId;
38,640,210✔
1775
      c.isPk = pColNode->isPk;
38,640,210✔
1776
      c.dataType = pColNode->node.resType;
38,640,210✔
1777
      void* tmp = taosArrayPush(pList, &c);
38,653,794✔
1778
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
38,653,794!
1779
    }
1780
  }
1781

1782
  // set the output flag for each column in SColMatchInfo, according to the
1783
  *numOfOutputCols = 0;
8,076,133✔
1784
  int32_t num = LIST_LENGTH(pOutputNodeList->pSlots);
8,076,133✔
1785
  for (int32_t i = 0; i < num; ++i) {
54,834,565✔
1786
    SSlotDescNode* pNode = (SSlotDescNode*)nodesListGetNode(pOutputNodeList->pSlots, i);
46,722,983✔
1787
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
46,839,616!
1788

1789
    // todo: add reserve flag check
1790
    // it is a column reserved for the arithmetic expression calculation
1791
    if (pNode->slotId >= numOfCols) {
46,860,668✔
1792
      (*numOfOutputCols) += 1;
8,133,996✔
1793
      continue;
8,133,996✔
1794
    }
1795

1796
    SColMatchItem* info = NULL;
38,726,672✔
1797
    for (int32_t j = 0; j < taosArrayGetSize(pList); ++j) {
2,147,483,647✔
1798
      info = taosArrayGet(pList, j);
2,147,483,647✔
1799
      QUERY_CHECK_NULL(info, code, lino, _end, terrno);
2,147,483,647✔
1800
      if (info->dstSlotId == pNode->slotId) {
2,147,483,647✔
1801
        break;
38,348,738✔
1802
      }
1803
    }
1804

1805
    if (pNode->output) {
38,624,436✔
1806
      (*numOfOutputCols) += 1;
38,550,997✔
1807
    } else if (info != NULL) {
73,439!
1808
      // select distinct tbname from stb where tbname='abc';
1809
      info->needOutput = false;
73,439✔
1810
    }
1811
  }
1812

1813
  pMatchInfo->pList = pList;
8,111,582✔
1814

1815
_end:
8,111,582✔
1816
  if (code != TSDB_CODE_SUCCESS) {
8,111,582!
1817
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1818
  }
1819
  return code;
8,066,254✔
1820
}
1821

1822
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
29,927,611✔
1823
                                  const char* name) {
1824
  SResSchema s = {0};
29,927,611✔
1825
  s.scale = scale;
29,927,611✔
1826
  s.type = type;
29,927,611✔
1827
  s.bytes = bytes;
29,927,611✔
1828
  s.slotId = slotId;
29,927,611✔
1829
  s.precision = precision;
29,927,611✔
1830
  tstrncpy(s.name, name, tListLen(s.name));
29,927,611✔
1831

1832
  return s;
29,927,611✔
1833
}
1834

1835
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType, EColumnType colType) {
28,541,611✔
1836
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
28,541,611✔
1837
  if (pCol == NULL) {
28,544,509!
1838
    terrno = TSDB_CODE_OUT_OF_MEMORY;
×
1839
    return NULL;
×
1840
  }
1841

1842
  pCol->slotId = slotId;
28,551,765✔
1843
  pCol->colId = colId;
28,551,765✔
1844
  pCol->bytes = pType->bytes;
28,551,765✔
1845
  pCol->type = pType->type;
28,551,765✔
1846
  pCol->scale = pType->scale;
28,551,765✔
1847
  pCol->precision = pType->precision;
28,551,765✔
1848
  pCol->dataBlockId = blockId;
28,551,765✔
1849
  pCol->colType = colType;
28,551,765✔
1850
  return pCol;
28,551,765✔
1851
}
1852

1853
int32_t createExprFromOneNode(SExprInfo* pExp, SNode* pNode, int16_t slotId) {
29,915,104✔
1854
  int32_t code = TSDB_CODE_SUCCESS;
29,915,104✔
1855
  int32_t lino = 0;
29,915,104✔
1856
  pExp->base.numOfParams = 0;
29,915,104✔
1857
  pExp->base.pParam = NULL;
29,915,104✔
1858
  pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
29,915,104✔
1859
  QUERY_CHECK_NULL(pExp->pExpr, code, lino, _end, terrno);
29,938,241!
1860

1861
  pExp->pExpr->_function.num = 1;
29,942,651✔
1862
  pExp->pExpr->_function.functionId = -1;
29,942,651✔
1863

1864
  int32_t type = nodeType(pNode);
29,942,651✔
1865
  // it is a project query, or group by column
1866
  if (type == QUERY_NODE_COLUMN) {
29,942,651✔
1867
    pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
19,582,616✔
1868
    SColumnNode* pColNode = (SColumnNode*)pNode;
19,582,616✔
1869

1870
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
19,582,616✔
1871
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
19,579,466!
1872

1873
    pExp->base.numOfParams = 1;
19,579,466✔
1874

1875
    SDataType* pType = &pColNode->node.resType;
19,579,466✔
1876
    pExp->base.resSchema =
1877
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pColNode->colName);
19,579,466✔
1878

1879
    pExp->base.pParam[0].pCol =
39,155,691✔
1880
        createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType, pColNode->colType);
19,578,576✔
1881
    QUERY_CHECK_NULL(pExp->base.pParam[0].pCol, code, lino, _end, terrno);
19,577,115!
1882

1883
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
19,577,115✔
1884
  } else if (type == QUERY_NODE_VALUE) {
10,360,035✔
1885
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
285,952✔
1886
    SValueNode* pValNode = (SValueNode*)pNode;
285,952✔
1887

1888
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
285,952✔
1889
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
286,029!
1890

1891
    pExp->base.numOfParams = 1;
286,029✔
1892

1893
    SDataType* pType = &pValNode->node.resType;
286,029✔
1894
    pExp->base.resSchema =
1895
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
286,029✔
1896
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
285,913✔
1897
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
285,913✔
1898
    QUERY_CHECK_CODE(code, lino, _end);
285,936!
1899
  } else if (type == QUERY_NODE_FUNCTION) {
10,074,083✔
1900
    pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
9,919,882✔
1901
    SFunctionNode* pFuncNode = (SFunctionNode*)pNode;
9,919,882✔
1902

1903
    SDataType* pType = &pFuncNode->node.resType;
9,919,882✔
1904
    pExp->base.resSchema =
1905
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pFuncNode->node.aliasName);
9,919,882✔
1906
    tExprNode* pExprNode = pExp->pExpr;
9,921,223✔
1907

1908
    pExprNode->_function.functionId = pFuncNode->funcId;
9,921,223✔
1909
    pExprNode->_function.pFunctNode = pFuncNode;
9,921,223✔
1910
    pExprNode->_function.functionType = pFuncNode->funcType;
9,921,223✔
1911

1912
    tstrncpy(pExprNode->_function.functionName, pFuncNode->functionName, tListLen(pExprNode->_function.functionName));
9,921,223✔
1913

1914
#if 1
1915
    // todo refactor: add the parameter for tbname function
1916
    const char* name = "tbname";
9,921,223✔
1917
    int32_t     len = strlen(name);
9,921,223✔
1918

1919
    if (!pFuncNode->pParameterList && (memcmp(pExprNode->_function.functionName, name, len) == 0) &&
9,921,223✔
1920
        pExprNode->_function.functionName[len] == 0) {
789,757!
1921
      pFuncNode->pParameterList = NULL;
789,918✔
1922
      int32_t     code = nodesMakeList(&pFuncNode->pParameterList);
789,918✔
1923
      SValueNode* res = NULL;
789,733✔
1924
      if (TSDB_CODE_SUCCESS == code) {
789,733!
1925
        code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
789,790✔
1926
      }
1927
      QUERY_CHECK_CODE(code, lino, _end);
789,835!
1928
      res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
789,835✔
1929
      code = nodesListAppend(pFuncNode->pParameterList, (SNode*)res);
789,835✔
1930
      if (code != TSDB_CODE_SUCCESS) {
789,984✔
1931
        nodesDestroyNode((SNode*)res);
14✔
1932
        res = NULL;
×
1933
      }
1934
      QUERY_CHECK_CODE(code, lino, _end);
789,970!
1935
    }
1936
#endif
1937

1938
    int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
9,921,275✔
1939

1940
    pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
9,921,275✔
1941
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
9,922,039!
1942
    pExp->base.numOfParams = numOfParam;
9,922,354✔
1943

1944
    for (int32_t j = 0; j < numOfParam && TSDB_CODE_SUCCESS == code; ++j) {
21,824,303!
1945
      SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
11,901,864✔
1946
      QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
11,901,332!
1947
      if (p1->type == QUERY_NODE_COLUMN) {
11,902,591✔
1948
        SColumnNode* pcn = (SColumnNode*)p1;
8,972,851✔
1949

1950
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
8,972,851✔
1951
        pExp->base.pParam[j].pCol =
17,946,795✔
1952
            createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType, pcn->colType);
8,972,851✔
1953
        QUERY_CHECK_NULL(pExp->base.pParam[j].pCol, code, lino, _end, terrno);
8,973,944✔
1954
      } else if (p1->type == QUERY_NODE_VALUE) {
2,929,740✔
1955
        SValueNode* pvn = (SValueNode*)p1;
2,343,091✔
1956
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
2,343,091✔
1957
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
2,343,091✔
1958
        QUERY_CHECK_CODE(code, lino, _end);
2,342,600!
1959
      }
1960
    }
1961
  } else if (type == QUERY_NODE_OPERATOR) {
154,201✔
1962
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
147,312✔
1963
    SOperatorNode* pOpNode = (SOperatorNode*)pNode;
147,312✔
1964

1965
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
147,312✔
1966
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
147,344!
1967
    pExp->base.numOfParams = 1;
147,344✔
1968

1969
    SDataType* pType = &pOpNode->node.resType;
147,344✔
1970
    pExp->base.resSchema =
1971
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pOpNode->node.aliasName);
147,344✔
1972
    pExp->pExpr->_optrRoot.pRootNode = pNode;
147,320✔
1973
  } else if (type == QUERY_NODE_CASE_WHEN) {
6,889✔
1974
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
4,455✔
1975
    SCaseWhenNode* pCaseNode = (SCaseWhenNode*)pNode;
4,455✔
1976

1977
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
4,455✔
1978
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
4,455!
1979
    pExp->base.numOfParams = 1;
4,455✔
1980

1981
    SDataType* pType = &pCaseNode->node.resType;
4,455✔
1982
    pExp->base.resSchema =
1983
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCaseNode->node.aliasName);
4,455✔
1984
    pExp->pExpr->_optrRoot.pRootNode = pNode;
4,455✔
1985
  } else if (type == QUERY_NODE_LOGIC_CONDITION) {
2,434✔
1986
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
10✔
1987
    SLogicConditionNode* pCond = (SLogicConditionNode*)pNode;
10✔
1988
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
10✔
1989
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
10!
1990
    pExp->base.numOfParams = 1;
10✔
1991
    SDataType* pType = &pCond->node.resType;
10✔
1992
    pExp->base.resSchema = createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCond->node.aliasName);
10✔
1993
    pExp->pExpr->_optrRoot.pRootNode = pNode;
10✔
1994
  } else {
1995
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
2,424✔
1996
    QUERY_CHECK_CODE(code, lino, _end);
2,424!
1997
  }
1998

1999
_end:
2,424✔
2000
  if (code != TSDB_CODE_SUCCESS) {
29,939,699!
2001
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2002
  }
2003
  return code;
29,926,460✔
2004
}
2005

2006
int32_t createExprFromTargetNode(SExprInfo* pExp, STargetNode* pTargetNode) {
29,891,289✔
2007
  return createExprFromOneNode(pExp, pTargetNode->pExpr, pTargetNode->slotId);
29,891,289✔
2008
}
2009

2010
SExprInfo* createExpr(SNodeList* pNodeList, int32_t* numOfExprs) {
2,590✔
2011
  *numOfExprs = LIST_LENGTH(pNodeList);
2,590!
2012
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
2,590✔
2013
  if (!pExprs) {
2,590!
2014
    return NULL;
×
2015
  }
2016

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

2028
  return pExprs;
2,590✔
2029
}
2030

2031
int32_t createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, SExprInfo** pExprInfo, int32_t* numOfExprs) {
10,362,004✔
2032
  QRY_PARAM_CHECK(pExprInfo);
10,362,004!
2033

2034
  int32_t code = 0;
10,362,004✔
2035
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
10,362,004✔
2036
  int32_t numOfGroupKeys = 0;
10,362,004✔
2037
  if (pGroupKeys != NULL) {
10,362,004✔
2038
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
270,000✔
2039
  }
2040

2041
  *numOfExprs = numOfFuncs + numOfGroupKeys;
10,362,004✔
2042
  if (*numOfExprs == 0) {
10,362,004✔
2043
    return code;
1,634,053✔
2044
  }
2045

2046
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
8,727,951✔
2047
  if (pExprs == NULL) {
8,729,500!
2048
    return terrno;
×
2049
  }
2050

2051
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
38,231,340✔
2052
    STargetNode* pTargetNode = NULL;
29,496,366✔
2053
    if (i < numOfFuncs) {
29,496,366✔
2054
      pTargetNode = (STargetNode*)nodesListGetNode(pNodeList, i);
28,974,156✔
2055
    } else {
2056
      pTargetNode = (STargetNode*)nodesListGetNode(pGroupKeys, i - numOfFuncs);
522,210✔
2057
    }
2058
    if (!pTargetNode) {
29,500,295!
2059
      destroyExprInfo(pExprs, *numOfExprs);
×
2060
      taosMemoryFreeClear(pExprs);
×
2061
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2062
      return terrno;
×
2063
    }
2064

2065
    SExprInfo* pExp = &pExprs[i];
29,500,295✔
2066
    code = createExprFromTargetNode(pExp, pTargetNode);
29,500,295✔
2067
    if (code != TSDB_CODE_SUCCESS) {
29,504,514✔
2068
      destroyExprInfo(pExprs, *numOfExprs);
3,039✔
2069
      taosMemoryFreeClear(pExprs);
×
2070
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2071
      return code;
×
2072
    }
2073
  }
2074

2075
  *pExprInfo = pExprs;
8,734,974✔
2076
  return code;
8,734,974✔
2077
}
2078

2079
// set the output buffer for the selectivity + tag query
2080
static int32_t setSelectValueColumnInfo(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
10,015,948✔
2081
  int32_t num = 0;
10,015,948✔
2082
  int32_t code = TSDB_CODE_SUCCESS;
10,015,948✔
2083
  int32_t lino = 0;
10,015,948✔
2084

2085
  SqlFunctionCtx*  p = NULL;
10,015,948✔
2086
  SqlFunctionCtx** pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
10,015,948✔
2087
  if (pValCtx == NULL) {
10,019,596!
2088
    return terrno;
×
2089
  }
2090

2091
  SHashObj* pSelectFuncs = taosHashInit(8, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
10,019,596✔
2092
  QUERY_CHECK_NULL(pSelectFuncs, code, lino, _end, terrno);
10,016,609!
2093

2094
  for (int32_t i = 0; i < numOfOutput; ++i) {
38,965,615✔
2095
    const char* pName = pCtx[i].pExpr->pExpr->_function.functionName;
29,088,994✔
2096
    if ((strcmp(pName, "_select_value") == 0) || (strcmp(pName, "_group_key") == 0) ||
29,088,994✔
2097
        (strcmp(pName, "_group_const_value") == 0)) {
27,668,148✔
2098
      pValCtx[num++] = &pCtx[i];
1,538,407✔
2099
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
27,550,587✔
2100
      void* data = taosHashGet(pSelectFuncs, pName, strlen(pName));
2,480,524✔
2101
      if (taosHashGetSize(pSelectFuncs) != 0 && data == NULL) {
2,474,556✔
2102
        p = NULL;
131,302✔
2103
        break;
131,302✔
2104
      } else {
2105
        int32_t tempRes = taosHashPut(pSelectFuncs, pName, strlen(pName), &num, sizeof(num));
2,343,595✔
2106
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
2,345,323!
2107
          code = tempRes;
×
2108
          QUERY_CHECK_CODE(code, lino, _end);
×
2109
        }
2110
        p = &pCtx[i];
2,345,323✔
2111
      }
2112
    }
2113
  }
2114
  taosHashCleanup(pSelectFuncs);
10,007,923✔
2115

2116
  if (p != NULL) {
10,017,861✔
2117
    p->subsidiaries.pCtx = pValCtx;
2,081,868✔
2118
    p->subsidiaries.num = num;
2,081,868✔
2119
  } else {
2120
    taosMemoryFreeClear(pValCtx);
7,935,993✔
2121
  }
2122

2123
_end:
2,716✔
2124
  if (code != TSDB_CODE_SUCCESS) {
10,020,432!
2125
    taosMemoryFreeClear(pValCtx);
×
2126
    taosHashCleanup(pSelectFuncs);
×
2127
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2128
  }
2129
  return code;
10,017,395✔
2130
}
2131

2132
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowEntryInfoOffset,
10,012,374✔
2133
                                     SFunctionStateStore* pStore) {
2134
  int32_t         code = TSDB_CODE_SUCCESS;
10,012,374✔
2135
  int32_t         lino = 0;
10,012,374✔
2136
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
10,012,374✔
2137
  if (pFuncCtx == NULL) {
10,011,844!
2138
    return NULL;
×
2139
  }
2140

2141
  *rowEntryInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
10,011,844✔
2142
  if (*rowEntryInfoOffset == 0) {
10,020,191!
2143
    taosMemoryFreeClear(pFuncCtx);
×
2144
    return NULL;
×
2145
  }
2146

2147
  for (int32_t i = 0; i < numOfOutput; ++i) {
39,691,090✔
2148
    SExprInfo* pExpr = &pExprInfo[i];
29,668,609✔
2149

2150
    SExprBasicInfo* pFunct = &pExpr->base;
29,668,609✔
2151
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
29,668,609✔
2152

2153
    pCtx->functionId = -1;
29,668,609✔
2154
    pCtx->pExpr = pExpr;
29,668,609✔
2155

2156
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
29,668,609✔
2157
      SFuncExecEnv env = {0};
9,917,565✔
2158
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;
9,917,565✔
2159
      pCtx->isPseudoFunc = fmIsWindowPseudoColumnFunc(pCtx->functionId);
9,917,565✔
2160
      pCtx->isNotNullFunc = fmIsNotNullOutputFunc(pCtx->functionId);
9,912,220✔
2161

2162
      bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
9,914,325✔
2163
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId)) {
16,186,109✔
2164
        if (!isUdaf) {
6,276,084✔
2165
          code = fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
6,275,485✔
2166
          QUERY_CHECK_CODE(code, lino, _end);
6,271,164!
2167
        } else {
2168
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
599✔
2169
          pCtx->udfName = taosStrdup(udfName);
599✔
2170
          QUERY_CHECK_NULL(pCtx->udfName, code, lino, _end, terrno);
599!
2171

2172
          code = fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
599✔
2173
          QUERY_CHECK_CODE(code, lino, _end);
599!
2174
        }
2175
        bool tmp = pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
6,271,763✔
2176
        if (!tmp) {
6,272,623!
2177
          code = TSDB_CODE_OUT_OF_MEMORY;
×
2178
          QUERY_CHECK_CODE(code, lino, _end);
×
2179
        }
2180
      } else {
2181
        code = fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
3,632,428✔
2182
        if (code != TSDB_CODE_SUCCESS && isUdaf) {
3,633,266!
2183
          code = TSDB_CODE_SUCCESS;
677✔
2184
        }
2185
        QUERY_CHECK_CODE(code, lino, _end);
3,633,266!
2186

2187
        if (pCtx->sfp.getEnv != NULL) {
3,633,266✔
2188
          bool tmp = pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
1,847,566✔
2189
          if (!tmp) {
1,846,624!
2190
            code = TSDB_CODE_OUT_OF_MEMORY;
×
2191
            QUERY_CHECK_CODE(code, lino, _end);
×
2192
          }
2193
        }
2194
      }
2195
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
9,904,947✔
2196
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
19,751,044✔
2197
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
285,951✔
2198
      // for simple column, the result buffer needs to hold at least one element.
2199
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
19,746,874✔
2200
    }
2201

2202
    pCtx->input.numOfInputCols = pFunct->numOfParams;
29,655,991✔
2203
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
29,655,991✔
2204
    QUERY_CHECK_NULL(pCtx->input.pData, code, lino, _end, terrno);
29,679,205!
2205
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
29,679,205✔
2206
    QUERY_CHECK_NULL(pCtx->input.pColumnDataAgg, code, lino, _end, terrno);
29,678,819✔
2207

2208
    pCtx->pTsOutput = NULL;
29,670,899✔
2209
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
29,670,899✔
2210
    pCtx->resDataInfo.type = pFunct->resSchema.type;
29,670,899✔
2211
    pCtx->order = TSDB_ORDER_ASC;
29,670,899✔
2212
    pCtx->start.key = INT64_MIN;
29,670,899✔
2213
    pCtx->end.key = INT64_MIN;
29,670,899✔
2214
    pCtx->numOfParams = pExpr->base.numOfParams;
29,670,899✔
2215
    pCtx->param = pFunct->pParam;
29,670,899✔
2216
    pCtx->saveHandle.currentPage = -1;
29,670,899✔
2217
    pCtx->pStore = pStore;
29,670,899✔
2218
    pCtx->hasWindowOrGroup = false;
29,670,899✔
2219
    pCtx->needCleanup = false;
29,670,899✔
2220
  }
2221

2222
  for (int32_t i = 1; i < numOfOutput; ++i) {
30,608,131✔
2223
    (*rowEntryInfoOffset)[i] = (int32_t)((*rowEntryInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) +
20,585,650✔
2224
                                         pFuncCtx[i - 1].resDataInfo.interBufSize);
20,585,650✔
2225
  }
2226

2227
  code = setSelectValueColumnInfo(pFuncCtx, numOfOutput);
10,022,481✔
2228
  QUERY_CHECK_CODE(code, lino, _end);
10,017,008!
2229

2230
_end:
10,017,008✔
2231
  if (code != TSDB_CODE_SUCCESS) {
10,017,008!
2232
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2233
    for (int32_t i = 0; i < numOfOutput; ++i) {
×
2234
      taosMemoryFree(pFuncCtx[i].input.pData);
×
2235
      taosMemoryFree(pFuncCtx[i].input.pColumnDataAgg);
×
2236
    }
2237
    taosMemoryFreeClear(*rowEntryInfoOffset);
×
2238
    taosMemoryFreeClear(pFuncCtx);
×
2239
    return NULL;
×
2240
  }
2241
  return pFuncCtx;
10,017,008✔
2242
}
2243

2244
// NOTE: sources columns are more than the destination SSDatablock columns.
2245
// doFilter in table scan needs every column even its output is false
2246
int32_t relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray* pCols, bool outputEveryColumn) {
50,888✔
2247
  int32_t code = TSDB_CODE_SUCCESS;
50,888✔
2248
  size_t  numOfSrcCols = taosArrayGetSize(pCols);
50,888✔
2249

2250
  int32_t i = 0, j = 0;
50,887✔
2251
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
275,658✔
2252
    SColumnInfoData* p = taosArrayGet(pCols, i);
224,781✔
2253
    if (!p) {
224,792!
2254
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2255
      return terrno;
×
2256
    }
2257
    SColMatchItem* pmInfo = taosArrayGet(pColMatchInfo, j);
224,792✔
2258
    if (!pmInfo) {
224,788✔
2259
      return terrno;
3✔
2260
    }
2261

2262
    if (p->info.colId == pmInfo->colId) {
224,785✔
2263
      SColumnInfoData* pDst = taosArrayGet(pBlock->pDataBlock, pmInfo->dstSlotId);
200,872✔
2264
      if (!pDst) {
200,867!
2265
        return terrno;
×
2266
      }
2267
      code = colDataAssign(pDst, p, pBlock->info.rows, &pBlock->info);
200,867✔
2268
      if (code != TSDB_CODE_SUCCESS) {
200,858!
UNCOV
2269
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2270
        return code;
×
2271
      }
2272
      i++;
200,858✔
2273
      j++;
200,858✔
2274
    } else if (p->info.colId < pmInfo->colId) {
23,913!
2275
      i++;
23,913✔
2276
    } else {
2277
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2278
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2279
    }
2280
  }
2281
  return code;
50,882✔
2282
}
2283

2284
SInterval extractIntervalInfo(const STableScanPhysiNode* pTableScanNode) {
3,035,024✔
2285
  SInterval interval = {
3,035,024✔
2286
      .interval = pTableScanNode->interval,
3,035,024✔
2287
      .sliding = pTableScanNode->sliding,
3,035,024✔
2288
      .intervalUnit = pTableScanNode->intervalUnit,
3,035,024✔
2289
      .slidingUnit = pTableScanNode->slidingUnit,
3,035,024✔
2290
      .offset = pTableScanNode->offset,
3,035,024✔
2291
      .precision = pTableScanNode->scan.node.pOutputDataBlockDesc->precision,
3,035,024✔
2292
  };
2293

2294
  return interval;
3,035,024✔
2295
}
2296

2297
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
675,696✔
2298
  SColumn c = {0};
675,696✔
2299

2300
  c.slotId = pColNode->slotId;
675,696✔
2301
  c.colId = pColNode->colId;
675,696✔
2302
  c.type = pColNode->node.resType.type;
675,696✔
2303
  c.bytes = pColNode->node.resType.bytes;
675,696✔
2304
  c.scale = pColNode->node.resType.scale;
675,696✔
2305
  c.precision = pColNode->node.resType.precision;
675,696✔
2306
  return c;
675,696✔
2307
}
2308

2309
int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, const STableScanPhysiNode* pTableScanNode,
5,114,919✔
2310
                               const SReadHandle* readHandle) {
2311
  pCond->order = pTableScanNode->scanSeq[0] > 0 ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
5,114,919✔
2312
  pCond->numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
5,114,919!
2313

2314
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
5,114,919✔
2315
  if (!pCond->colList) {
5,122,983!
2316
    return terrno;
×
2317
  }
2318
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
5,122,983✔
2319
  if (pCond->pSlotList == NULL) {
5,122,021✔
2320
    taosMemoryFreeClear(pCond->colList);
416!
2321
    return terrno;
416✔
2322
  }
2323

2324
  // TODO: get it from stable scan node
2325
  pCond->twindows = pTableScanNode->scanRange;
5,121,605✔
2326
  pCond->suid = pTableScanNode->scan.suid;
5,121,605✔
2327
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
5,121,605✔
2328
  pCond->startVersion = -1;
5,121,605✔
2329
  pCond->endVersion = -1;
5,121,605✔
2330
  pCond->skipRollup = readHandle->skipRollup;
5,121,605✔
2331

2332
  // allowed read stt file optimization mode
2333
  pCond->notLoadData = (pTableScanNode->dataRequired == FUNC_DATA_REQUIRED_NOT_LOAD) &&
10,429,234✔
2334
                       (pTableScanNode->scan.node.pConditions == NULL) && (pTableScanNode->interval == 0);
5,121,605✔
2335

2336
  int32_t j = 0;
5,121,605✔
2337
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
31,631,082✔
2338
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pTableScanNode->scan.pScanCols, i);
26,506,564✔
2339
    if (!pNode) {
26,510,275✔
2340
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
798!
2341
      return terrno;
798✔
2342
    }
2343
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
26,509,477✔
2344
    if (pColNode->colType == COLUMN_TYPE_TAG) {
26,509,477!
2345
      continue;
×
2346
    }
2347

2348
    pCond->colList[j].type = pColNode->node.resType.type;
26,509,477✔
2349
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
26,509,477✔
2350
    pCond->colList[j].colId = pColNode->colId;
26,509,477✔
2351
    pCond->colList[j].pk = pColNode->isPk;
26,509,477✔
2352

2353
    pCond->pSlotList[j] = pNode->slotId;
26,509,477✔
2354
    j += 1;
26,509,477✔
2355
  }
2356

2357
  pCond->numOfCols = j;
5,124,518✔
2358
  return TSDB_CODE_SUCCESS;
5,124,518✔
2359
}
2360

2361
void cleanupQueryTableDataCond(SQueryTableDataCond* pCond) {
7,213,486✔
2362
  taosMemoryFreeClear(pCond->colList);
7,213,486✔
2363
  taosMemoryFreeClear(pCond->pSlotList);
7,213,927✔
2364
}
7,213,913✔
2365

2366
int32_t convertFillType(int32_t mode) {
561,165✔
2367
  int32_t type = TSDB_FILL_NONE;
561,165✔
2368
  switch (mode) {
561,165!
2369
    case FILL_MODE_PREV:
16,280✔
2370
      type = TSDB_FILL_PREV;
16,280✔
2371
      break;
16,280✔
2372
    case FILL_MODE_NONE:
×
2373
      type = TSDB_FILL_NONE;
×
2374
      break;
×
2375
    case FILL_MODE_NULL:
32,506✔
2376
      type = TSDB_FILL_NULL;
32,506✔
2377
      break;
32,506✔
2378
    case FILL_MODE_NULL_F:
30,381✔
2379
      type = TSDB_FILL_NULL_F;
30,381✔
2380
      break;
30,381✔
2381
    case FILL_MODE_NEXT:
16,639✔
2382
      type = TSDB_FILL_NEXT;
16,639✔
2383
      break;
16,639✔
2384
    case FILL_MODE_VALUE:
152,445✔
2385
      type = TSDB_FILL_SET_VALUE;
152,445✔
2386
      break;
152,445✔
2387
    case FILL_MODE_VALUE_F:
298,103✔
2388
      type = TSDB_FILL_SET_VALUE_F;
298,103✔
2389
      break;
298,103✔
2390
    case FILL_MODE_LINEAR:
14,814✔
2391
      type = TSDB_FILL_LINEAR;
14,814✔
2392
      break;
14,814✔
2393
    default:
×
2394
      type = TSDB_FILL_NONE;
×
2395
  }
2396

2397
  return type;
561,165✔
2398
}
2399

2400
void getInitialStartTimeWindow(SInterval* pInterval, TSKEY ts, STimeWindow* w, bool ascQuery) {
22,347,809✔
2401
  if (ascQuery) {
22,347,809✔
2402
    *w = getAlignQueryTimeWindow(pInterval, ts);
22,119,377✔
2403
  } else {
2404
    // the start position of the first time window in the endpoint that spreads beyond the queried last timestamp
2405
    *w = getAlignQueryTimeWindow(pInterval, ts);
228,432✔
2406

2407
    int64_t key = w->skey;
228,471✔
2408
    while (key < ts) {  // moving towards end
748,514✔
2409
      key = taosTimeAdd(key, pInterval->sliding, pInterval->slidingUnit, pInterval->precision);
732,434✔
2410
      if (key > ts) {
732,443✔
2411
        break;
212,400✔
2412
      }
2413

2414
      w->skey = key;
520,043✔
2415
    }
2416
  }
2417
}
22,347,802✔
2418

2419
static STimeWindow doCalculateTimeWindow(int64_t ts, SInterval* pInterval) {
1,040,936✔
2420
  STimeWindow w = {0};
1,040,936✔
2421

2422
  w.skey = taosTimeTruncate(ts, pInterval);
1,040,936✔
2423
  w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
1,040,994✔
2424
  return w;
1,040,904✔
2425
}
2426

2427
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order) {
414,004✔
2428
  int32_t factor = (order == TSDB_ORDER_ASC) ? -1 : 1;
414,004✔
2429

2430
  STimeWindow win = *pWindow;
414,004✔
2431
  STimeWindow save = win;
414,004✔
2432
  while (win.skey <= ts && win.ekey >= ts) {
1,812,792✔
2433
    save = win;
1,398,789✔
2434
    win.skey = taosTimeAdd(win.skey, factor * pInterval->sliding, pInterval->slidingUnit, pInterval->precision);
1,398,789✔
2435
    win.ekey = taosTimeAdd(win.ekey, factor * pInterval->sliding, pInterval->slidingUnit, pInterval->precision);
1,398,792✔
2436
  }
2437

2438
  return save;
414,003✔
2439
}
2440

2441
// get the correct time window according to the handled timestamp
2442
// todo refactor
2443
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
23,986,996✔
2444
                                int32_t order) {
2445
  STimeWindow w = {0};
23,986,996✔
2446
  if (pResultRowInfo->cur.pageId == -1) {  // the first window, from the previous stored value
23,986,996✔
2447
    getInitialStartTimeWindow(pInterval, ts, &w, (order == TSDB_ORDER_ASC));
22,347,825✔
2448
    w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
22,347,795✔
2449
    return w;
22,347,687✔
2450
  }
2451

2452
  SResultRow* pRow = getResultRowByPos(pBuf, &pResultRowInfo->cur, false);
1,639,171✔
2453
  if (pRow) {
1,639,076!
2454
    w = pRow->win;
1,639,081✔
2455
  }
2456

2457
  // in case of typical time window, we can calculate time window directly.
2458
  if (w.skey > ts || w.ekey < ts) {
1,639,076✔
2459
    w = doCalculateTimeWindow(ts, pInterval);
1,040,937✔
2460
  }
2461

2462
  if (pInterval->interval != pInterval->sliding) {
1,639,056✔
2463
    // it is an sliding window query, in which sliding value is not equalled to
2464
    // interval value, and we need to find the first qualified time window.
2465
    w = getFirstQualifiedTimeWindow(ts, &w, pInterval, order);
414,004✔
2466
  }
2467

2468
  return w;
1,639,054✔
2469
}
2470

2471
void getNextTimeWindow(const SInterval* pInterval, STimeWindow* tw, int32_t order) {
241,932,741✔
2472
  int64_t slidingStart = 0;
241,932,741✔
2473
  if (pInterval->offset > 0) {
241,932,741✔
2474
    slidingStart = taosTimeAdd(tw->skey, -1 * pInterval->offset, pInterval->offsetUnit, pInterval->precision);
55,262,666✔
2475
  } else {
2476
    slidingStart = tw->skey;
186,670,075✔
2477
  }
2478
  int32_t factor = GET_FORWARD_DIRECTION_FACTOR(order);
242,083,710✔
2479
  slidingStart = taosTimeAdd(slidingStart, factor * pInterval->sliding, pInterval->slidingUnit, pInterval->precision);
242,083,710✔
2480
  tw->skey = taosTimeAdd(slidingStart, pInterval->offset, pInterval->offsetUnit, pInterval->precision);
242,097,923✔
2481
  int64_t slidingEnd =
242,191,967✔
2482
      taosTimeAdd(slidingStart, pInterval->interval, pInterval->intervalUnit, pInterval->precision) - 1;
242,231,773✔
2483
  tw->ekey = taosTimeAdd(slidingEnd, pInterval->offset, pInterval->offsetUnit, pInterval->precision);
242,191,967✔
2484
}
242,284,509✔
2485

2486
bool hasLimitOffsetInfo(SLimitInfo* pLimitInfo) {
7,537,953✔
2487
  return (pLimitInfo->limit.limit != -1 || pLimitInfo->limit.offset != -1 || pLimitInfo->slimit.limit != -1 ||
15,052,454!
2488
          pLimitInfo->slimit.offset != -1);
7,514,501✔
2489
}
2490

2491
bool hasSlimitOffsetInfo(SLimitInfo* pLimitInfo) {
×
2492
  return (pLimitInfo->slimit.limit != -1 || pLimitInfo->slimit.offset != -1);
×
2493
}
2494

2495
void initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo) {
12,950,746✔
2496
  SLimit limit = {.limit = getLimit(pLimit), .offset = getOffset(pLimit)};
12,950,746✔
2497
  SLimit slimit = {.limit = getLimit(pSLimit), .offset = getOffset(pSLimit)};
12,956,411✔
2498

2499
  pLimitInfo->limit = limit;
12,958,660✔
2500
  pLimitInfo->slimit = slimit;
12,958,660✔
2501
  pLimitInfo->remainOffset = limit.offset;
12,958,660✔
2502
  pLimitInfo->remainGroupOffset = slimit.offset;
12,958,660✔
2503
}
12,958,660✔
2504

2505
void resetLimitInfoForNextGroup(SLimitInfo* pLimitInfo) {
2,479,529✔
2506
  pLimitInfo->numOfOutputRows = 0;
2,479,529✔
2507
  pLimitInfo->remainOffset = pLimitInfo->limit.offset;
2,479,529✔
2508
}
2,479,529✔
2509

2510
int32_t tableListGetSize(const STableListInfo* pTableList, int32_t* pRes) {
13,444,412✔
2511
  if (taosArrayGetSize(pTableList->pTableList) != taosHashGetSize(pTableList->map)) {
13,444,412!
2512
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2513
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2514
  }
2515
  (*pRes) = taosArrayGetSize(pTableList->pTableList);
13,459,129✔
2516
  return TSDB_CODE_SUCCESS;
13,458,465✔
2517
}
2518

2519
uint64_t tableListGetSuid(const STableListInfo* pTableList) { return pTableList->idInfo.suid; }
89,911✔
2520

2521
STableKeyInfo* tableListGetInfo(const STableListInfo* pTableList, int32_t index) {
7,638,224✔
2522
  if (taosArrayGetSize(pTableList->pTableList) == 0) {
7,638,224✔
2523
    return NULL;
544✔
2524
  }
2525

2526
  return taosArrayGet(pTableList->pTableList, index);
7,641,311✔
2527
}
2528

2529
int32_t tableListFind(const STableListInfo* pTableList, uint64_t uid, int32_t startIndex) {
181✔
2530
  int32_t numOfTables = taosArrayGetSize(pTableList->pTableList);
181✔
2531
  if (startIndex >= numOfTables) {
181!
2532
    return -1;
×
2533
  }
2534

2535
  for (int32_t i = startIndex; i < numOfTables; ++i) {
2,425!
2536
    STableKeyInfo* p = taosArrayGet(pTableList->pTableList, i);
2,425✔
2537
    if (!p) {
2,425!
2538
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2539
      return -1;
×
2540
    }
2541
    if (p->uid == uid) {
2,425✔
2542
      return i;
181✔
2543
    }
2544
  }
2545
  return -1;
×
2546
}
2547

2548
void tableListGetSourceTableInfo(const STableListInfo* pTableList, uint64_t* psuid, uint64_t* uid, int32_t* type) {
17,933✔
2549
  *psuid = pTableList->idInfo.suid;
17,933✔
2550
  *uid = pTableList->idInfo.uid;
17,933✔
2551
  *type = pTableList->idInfo.tableType;
17,933✔
2552
}
17,933✔
2553

2554
uint64_t tableListGetTableGroupId(const STableListInfo* pTableList, uint64_t tableUid) {
10,789,154✔
2555
  int32_t* slot = taosHashGet(pTableList->map, &tableUid, sizeof(tableUid));
10,789,154✔
2556
  if (slot == NULL) {
10,792,919!
2557
    return -1;
×
2558
  }
2559

2560
  STableKeyInfo* pKeyInfo = taosArrayGet(pTableList->pTableList, *slot);
10,792,919✔
2561

2562
  return pKeyInfo->groupId;
10,790,842✔
2563
}
2564

2565
// TODO handle the group offset info, fix it, the rule of group output will be broken by this function
2566
int32_t tableListAddTableInfo(STableListInfo* pTableList, uint64_t uid, uint64_t gid) {
10,345✔
2567
  int32_t code = TSDB_CODE_SUCCESS;
10,345✔
2568
  int32_t lino = 0;
10,345✔
2569
  if (pTableList->map == NULL) {
10,345!
2570
    pTableList->map = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
×
2571
    QUERY_CHECK_NULL(pTableList->map, code, lino, _end, terrno);
×
2572
  }
2573

2574
  STableKeyInfo keyInfo = {.uid = uid, .groupId = gid};
10,345✔
2575
  void*         p = taosHashGet(pTableList->map, &uid, sizeof(uid));
10,345✔
2576
  if (p != NULL) {
10,358!
2577
    qInfo("table:%" PRId64 " already in tableIdList, ignore it", uid);
×
2578
    goto _end;
×
2579
  }
2580

2581
  void* tmp = taosArrayPush(pTableList->pTableList, &keyInfo);
10,358✔
2582
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
10,354!
2583

2584
  int32_t slot = (int32_t)taosArrayGetSize(pTableList->pTableList) - 1;
10,354✔
2585
  code = taosHashPut(pTableList->map, &uid, sizeof(uid), &slot, sizeof(slot));
10,354✔
2586
  if (code != TSDB_CODE_SUCCESS) {
10,369!
2587
    // we have checked the existence of uid in hash map above
2588
    QUERY_CHECK_CONDITION((code != TSDB_CODE_DUP_KEY), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
2589
    taosArrayPopTailBatch(pTableList->pTableList, 1);  // let's pop the last element in the array list
×
2590
  }
2591

2592
_end:
10,369✔
2593
  if (code != TSDB_CODE_SUCCESS) {
10,369!
2594
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2595
  } else {
2596
    qDebug("uid:%" PRIu64 ", groupId:%" PRIu64 " added into table list, slot:%d, total:%d", uid, gid, slot, slot + 1);
10,369✔
2597
  }
2598

2599
  return code;
10,368✔
2600
}
2601

2602
int32_t tableListGetGroupList(const STableListInfo* pTableList, int32_t ordinalGroupIndex, STableKeyInfo** pKeyInfo,
2,998,294✔
2603
                              int32_t* size) {
2604
  int32_t totalGroups = tableListGetOutputGroups(pTableList);
2,998,294✔
2605
  int32_t numOfTables = 0;
3,003,774✔
2606
  int32_t code = tableListGetSize(pTableList, &numOfTables);
3,003,774✔
2607
  if (code != TSDB_CODE_SUCCESS) {
3,002,292!
2608
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2609
    return code;
×
2610
  }
2611

2612
  if (ordinalGroupIndex < 0 || ordinalGroupIndex >= totalGroups) {
3,002,292!
2613
    return TSDB_CODE_INVALID_PARA;
×
2614
  }
2615

2616
  // here handle two special cases:
2617
  // 1. only one group exists, and 2. one table exists for each group.
2618
  if (totalGroups == 1) {
3,002,292✔
2619
    *size = numOfTables;
3,001,139✔
2620
    *pKeyInfo = (*size == 0) ? NULL : taosArrayGet(pTableList->pTableList, 0);
3,001,139✔
2621
    return TSDB_CODE_SUCCESS;
3,001,435✔
2622
  } else if (totalGroups == numOfTables) {
1,153✔
2623
    *size = 1;
1,066✔
2624
    *pKeyInfo = taosArrayGet(pTableList->pTableList, ordinalGroupIndex);
1,066✔
2625
    return TSDB_CODE_SUCCESS;
1,066✔
2626
  }
2627

2628
  int32_t offset = pTableList->groupOffset[ordinalGroupIndex];
87✔
2629
  if (ordinalGroupIndex < totalGroups - 1) {
87!
2630
    *size = pTableList->groupOffset[ordinalGroupIndex + 1] - offset;
536✔
2631
  } else {
2632
    *size = numOfTables - offset;
×
2633
  }
2634

2635
  *pKeyInfo = taosArrayGet(pTableList->pTableList, offset);
87✔
2636
  return TSDB_CODE_SUCCESS;
670✔
2637
}
2638

2639
int32_t tableListGetOutputGroups(const STableListInfo* pTableList) { return pTableList->numOfOuputGroups; }
8,963,802✔
2640

2641
bool oneTableForEachGroup(const STableListInfo* pTableList) { return pTableList->oneTableForEachGroup; }
11,510✔
2642

2643
STableListInfo* tableListCreate() {
5,178,899✔
2644
  STableListInfo* pListInfo = taosMemoryCalloc(1, sizeof(STableListInfo));
5,178,899✔
2645
  if (pListInfo == NULL) {
5,185,982!
2646
    return NULL;
×
2647
  }
2648

2649
  pListInfo->remainGroups = NULL;
5,185,982✔
2650
  pListInfo->pTableList = taosArrayInit(4, sizeof(STableKeyInfo));
5,185,982✔
2651
  if (pListInfo->pTableList == NULL) {
5,186,561!
2652
    goto _error;
×
2653
  }
2654

2655
  pListInfo->map = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
5,186,561✔
2656
  if (pListInfo->map == NULL) {
5,185,099!
2657
    goto _error;
×
2658
  }
2659

2660
  pListInfo->numOfOuputGroups = 1;
5,185,099✔
2661
  return pListInfo;
5,185,099✔
2662

2663
_error:
×
2664
  tableListDestroy(pListInfo);
×
2665
  return NULL;
×
2666
}
2667

2668
void tableListDestroy(STableListInfo* pTableListInfo) {
5,186,676✔
2669
  if (pTableListInfo == NULL) {
5,186,676✔
2670
    return;
2✔
2671
  }
2672

2673
  taosArrayDestroy(pTableListInfo->pTableList);
5,186,674✔
2674
  taosMemoryFreeClear(pTableListInfo->groupOffset);
5,186,943✔
2675

2676
  taosHashCleanup(pTableListInfo->map);
5,186,943✔
2677
  taosHashCleanup(pTableListInfo->remainGroups);
5,187,247✔
2678
  pTableListInfo->pTableList = NULL;
5,187,258✔
2679
  pTableListInfo->map = NULL;
5,187,258✔
2680
  taosMemoryFree(pTableListInfo);
5,187,258✔
2681
}
2682

2683
void tableListClear(STableListInfo* pTableListInfo) {
2,531✔
2684
  if (pTableListInfo == NULL) {
2,531!
2685
    return;
×
2686
  }
2687

2688
  taosArrayClear(pTableListInfo->pTableList);
2,531✔
2689
  taosHashClear(pTableListInfo->map);
2,531✔
2690
  taosHashClear(pTableListInfo->remainGroups);
2,531✔
2691
  taosMemoryFree(pTableListInfo->groupOffset);
2,531✔
2692
  pTableListInfo->numOfOuputGroups = 1;
2,531✔
2693
  pTableListInfo->oneTableForEachGroup = false;
2,531✔
2694
}
2695

2696
static int32_t orderbyGroupIdComparFn(const void* p1, const void* p2) {
94,293✔
2697
  STableKeyInfo* pInfo1 = (STableKeyInfo*)p1;
94,293✔
2698
  STableKeyInfo* pInfo2 = (STableKeyInfo*)p2;
94,293✔
2699

2700
  if (pInfo1->groupId == pInfo2->groupId) {
94,293✔
2701
    return 0;
170✔
2702
  } else {
2703
    return pInfo1->groupId < pInfo2->groupId ? -1 : 1;
94,123✔
2704
  }
2705
}
2706

2707
static int32_t sortTableGroup(STableListInfo* pTableListInfo) {
887✔
2708
  taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
887✔
2709
  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
886✔
2710

2711
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
886✔
2712
  if (!pList) {
887!
2713
    return terrno;
×
2714
  }
2715

2716
  STableKeyInfo* pInfo = taosArrayGet(pTableListInfo->pTableList, 0);
887✔
2717
  if (!pInfo) {
887!
2718
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2719
    return terrno;
×
2720
  }
2721
  uint64_t       gid = pInfo->groupId;
887✔
2722

2723
  int32_t start = 0;
887✔
2724
  void*   tmp = taosArrayPush(pList, &start);
887✔
2725
  if (!tmp) {
887!
2726
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2727
    return terrno;
×
2728
  }
2729

2730
  for (int32_t i = 1; i < size; ++i) {
2,628✔
2731
    pInfo = taosArrayGet(pTableListInfo->pTableList, i);
1,742✔
2732
    if (!pInfo) {
1,742✔
2733
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
1!
2734
      return terrno;
1✔
2735
    }
2736
    if (pInfo->groupId != gid) {
1,741✔
2737
      tmp = taosArrayPush(pList, &i);
1,572✔
2738
      if (!tmp) {
1,572!
2739
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2740
        return terrno;
×
2741
      }
2742
      gid = pInfo->groupId;
1,572✔
2743
    }
2744
  }
2745

2746
  pTableListInfo->numOfOuputGroups = taosArrayGetSize(pList);
886✔
2747
  pTableListInfo->groupOffset = taosMemoryMalloc(sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
887✔
2748
  if (pTableListInfo->groupOffset == NULL) {
887!
2749
    taosArrayDestroy(pList);
×
2750
    return terrno;
×
2751
  }
2752

2753
  memcpy(pTableListInfo->groupOffset, taosArrayGet(pList, 0), sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
887✔
2754
  taosArrayDestroy(pList);
887✔
2755
  return TSDB_CODE_SUCCESS;
887✔
2756
}
2757

2758
int32_t buildGroupIdMapForAllTables(STableListInfo* pTableListInfo, SReadHandle* pHandle, SScanPhysiNode* pScanNode,
4,781,672✔
2759
                                    SNodeList* group, bool groupSort, uint8_t* digest, SStorageAPI* pAPI) {
2760
  int32_t code = TSDB_CODE_SUCCESS;
4,781,672✔
2761

2762
  bool   groupByTbname = groupbyTbname(group);
4,781,672✔
2763
  size_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
4,781,526✔
2764
  if (!numOfTables) {
4,780,534!
2765
    return code;
×
2766
  }
2767
  if (group == NULL || groupByTbname) {
4,780,534✔
2768
    if (tsCountAlwaysReturnValue && QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode) &&
4,714,184✔
2769
        ((STableScanPhysiNode*)pScanNode)->needCountEmptyTable) {
2,701,758✔
2770
      pTableListInfo->remainGroups =
212,569✔
2771
          taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
212,571✔
2772
      if (pTableListInfo->remainGroups == NULL) {
212,569!
2773
        return terrno;
×
2774
      }
2775

2776
      for (int i = 0; i < numOfTables; i++) {
749,549✔
2777
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
536,908✔
2778
        if (!info) {
536,888✔
2779
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
3!
2780
          return terrno;
3✔
2781
        }
2782
        info->groupId = groupByTbname ? info->uid : 0;
536,885✔
2783

2784
        int32_t tempRes = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId),
536,885✔
2785
                                      &(info->uid), sizeof(info->uid));
536,885✔
2786
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
536,980!
2787
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
2788
          return tempRes;
×
2789
        }
2790
      }
2791
    } else {
2792
      for (int32_t i = 0; i < numOfTables; i++) {
16,582,312✔
2793
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
12,079,710✔
2794
        if (!info) {
12,080,998✔
2795
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
299!
2796
          return terrno;
299✔
2797
        }
2798
        info->groupId = groupByTbname ? info->uid : 0;
12,080,699✔
2799
      }
2800
    }
2801

2802
    pTableListInfo->oneTableForEachGroup = groupByTbname;
4,715,243✔
2803
    if (numOfTables == 1 && pTableListInfo->idInfo.tableType == TSDB_CHILD_TABLE) {
4,715,243✔
2804
      pTableListInfo->oneTableForEachGroup = true;
474,912✔
2805
    }
2806

2807
    if (groupSort && groupByTbname) {
4,715,243✔
2808
      taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
28,723✔
2809
      pTableListInfo->numOfOuputGroups = numOfTables;
28,723✔
2810
    } else if (groupByTbname && pScanNode->groupOrderScan) {
4,686,520✔
2811
      pTableListInfo->numOfOuputGroups = numOfTables;
159✔
2812
    } else {
2813
      pTableListInfo->numOfOuputGroups = 1;
4,686,361✔
2814
    }
2815
  } else {
2816
    bool initRemainGroups = false;
66,350✔
2817
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode)) {
66,350✔
2818
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pScanNode;
65,788✔
2819
      if (tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable &&
65,788✔
2820
          !(groupSort || pScanNode->groupOrderScan)) {
14,031!
2821
        initRemainGroups = true;
13,987✔
2822
      }
2823
    }
2824

2825
    code = getColInfoResultForGroupby(pHandle->vnode, group, pTableListInfo, digest, pAPI, initRemainGroups);
66,350✔
2826
    if (code != TSDB_CODE_SUCCESS) {
66,434✔
2827
      return code;
15✔
2828
    }
2829

2830
    if (pScanNode->groupOrderScan) pTableListInfo->numOfOuputGroups = taosArrayGetSize(pTableListInfo->pTableList);
66,419✔
2831

2832
    if (groupSort || pScanNode->groupOrderScan) {
66,419✔
2833
      code = sortTableGroup(pTableListInfo);
871✔
2834
    }
2835
  }
2836

2837
  // add all table entry in the hash map
2838
  size_t size = taosArrayGetSize(pTableListInfo->pTableList);
4,781,678✔
2839
  for (int32_t i = 0; i < size; ++i) {
17,675,467✔
2840
    STableKeyInfo* p = taosArrayGet(pTableListInfo->pTableList, i);
12,885,643✔
2841
    if (!p) {
12,879,269!
2842
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2843
      return terrno;
×
2844
    }
2845
    int32_t        tempRes = taosHashPut(pTableListInfo->map, &p->uid, sizeof(uint64_t), &i, sizeof(int32_t));
12,879,269✔
2846
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
12,895,207!
2847
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
2848
      return tempRes;
×
2849
    }
2850
  }
2851

2852
  return code;
4,789,824✔
2853
}
2854

2855
int32_t createScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
5,086,865✔
2856
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
2857
                                SExecTaskInfo* pTaskInfo) {
2858
  int64_t     st = taosGetTimestampUs();
5,089,218✔
2859
  const char* idStr = GET_TASKID(pTaskInfo);
5,089,218✔
2860

2861
  if (pHandle == NULL) {
5,089,218!
2862
    qError("invalid handle, in creating operator tree, %s", idStr);
×
2863
    return TSDB_CODE_INVALID_PARA;
×
2864
  }
2865

2866
  uint8_t digest[17] = {0};
5,089,218✔
2867
  int32_t code = getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, digest, idStr,
5,089,218✔
2868
                              &pTaskInfo->storageAPI);
2869
  if (code != TSDB_CODE_SUCCESS) {
5,091,089✔
2870
    qError("failed to getTableList, code: %s", tstrerror(code));
8!
2871
    return code;
8✔
2872
  }
2873

2874
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
5,091,081✔
2875

2876
  int64_t st1 = taosGetTimestampUs();
5,091,642✔
2877
  pTaskInfo->cost.extractListTime = (st1 - st) / 1000.0;
5,091,642✔
2878
  qDebug("extract queried table list completed, %d tables, elapsed time:%.2f ms %s", numOfTables,
5,091,642✔
2879
         pTaskInfo->cost.extractListTime, idStr);
2880

2881
  if (numOfTables == 0) {
5,092,882✔
2882
    qDebug("no table qualified for query, %s" PRIx64, idStr);
307,560✔
2883
    return TSDB_CODE_SUCCESS;
307,571✔
2884
  }
2885

2886
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, digest,
4,785,322✔
2887
                                     &pTaskInfo->storageAPI);
2888
  if (code != TSDB_CODE_SUCCESS) {
4,787,324✔
2889
    return code;
15✔
2890
  }
2891

2892
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
4,787,082✔
2893
  qDebug("generate group id map completed, elapsed time:%.2f ms %s", pTaskInfo->cost.groupIdMapTime, idStr);
4,787,082✔
2894

2895
  return TSDB_CODE_SUCCESS;
4,787,121✔
2896
}
2897

2898
char* getStreamOpName(uint16_t opType) {
1,296,937✔
2899
  switch (opType) {
1,296,937!
2900
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
513,672✔
2901
      return "stream scan";
513,672✔
2902
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
40,732✔
2903
      return "project";
40,732✔
2904
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL:
635,715✔
2905
      return "interval single";
635,715✔
2906
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_INTERVAL:
16,570✔
2907
      return "interval final";
16,570✔
2908
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_INTERVAL:
36,916✔
2909
      return "interval semi";
36,916✔
2910
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_MID_INTERVAL:
337✔
2911
      return "interval mid";
337✔
2912
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_FILL:
4,089✔
2913
      return "stream fill";
4,089✔
2914
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SESSION:
12,012✔
2915
      return "session single";
12,012✔
2916
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_SESSION:
1,458✔
2917
      return "session semi";
1,458✔
2918
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_SESSION:
1,230✔
2919
      return "session final";
1,230✔
2920
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_STATE:
7,327✔
2921
      return "state single";
7,327✔
2922
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_PARTITION:
13,657✔
2923
      return "stream partitionby";
13,657✔
2924
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_EVENT:
914✔
2925
      return "stream event";
914✔
2926
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_COUNT:
10,069✔
2927
      return "stream count";
10,069✔
2928
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERP_FUNC:
674✔
2929
      return "stream interp";
674✔
2930
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_INTERVAL:
1,613✔
2931
      return "interval continue";
1,613✔
2932
  }
2933
  return "";
×
2934
}
2935

2936
void printDataBlock(SSDataBlock* pBlock, const char* flag, const char* taskIdStr) {
452,859✔
2937
  if (!pBlock) {
452,859!
2938
    qDebug("%s===stream===%s: Block is Null", taskIdStr, flag);
×
2939
    return;
×
2940
  } else if (pBlock->info.rows == 0) {
452,859✔
2941
    qDebug("%s===stream===%s: Block is Empty. block type %d", taskIdStr, flag, pBlock->info.type);
30,606✔
2942
    return;
30,604✔
2943
  }
2944
  if (qDebugFlag & DEBUG_DEBUG) {
422,253✔
2945
    char*   pBuf = NULL;
48,183✔
2946
    int32_t code = dumpBlockData(pBlock, flag, &pBuf, taskIdStr);
48,183✔
2947
    if (code == 0) {
48,182!
2948
      qDebug("%s", pBuf);
48,183!
2949
      taosMemoryFree(pBuf);
48,183✔
2950
    }
2951
  }
2952
}
2953

2954
void printSpecDataBlock(SSDataBlock* pBlock, const char* flag, const char* opStr, const char* taskIdStr) {
775,793✔
2955
  if (!pBlock) {
775,793!
2956
    qDebug("%s===stream===%s %s: Block is Null", taskIdStr, flag, opStr);
×
2957
    return;
×
2958
  } else if (pBlock->info.rows == 0) {
775,793✔
2959
    qDebug("%s===stream===%s %s: Block is Empty. block type %d.skey:%" PRId64 ",ekey:%" PRId64 ",version%" PRId64,
7,352✔
2960
           taskIdStr, flag, opStr, pBlock->info.type, pBlock->info.window.skey, pBlock->info.window.ekey,
2961
           pBlock->info.version);
2962
    return;
7,352✔
2963
  }
2964
  if (qDebugFlag & DEBUG_DEBUG) {
768,441✔
2965
    char* pBuf = NULL;
58,697✔
2966
    char  flagBuf[64];
2967
    snprintf(flagBuf, sizeof(flagBuf), "%s %s", flag, opStr);
58,697✔
2968
    int32_t code = dumpBlockData(pBlock, flagBuf, &pBuf, taskIdStr);
58,697✔
2969
    if (code == 0) {
58,692!
2970
      qDebug("%s", pBuf);
58,694!
2971
      taosMemoryFree(pBuf);
58,697✔
2972
    }
2973
  }
2974
}
2975

2976
TSKEY getStartTsKey(STimeWindow* win, const TSKEY* tsCols) { return tsCols == NULL ? win->skey : tsCols[0]; }
3,195,704!
2977

2978
void updateTimeWindowInfo(SColumnInfoData* pColData, STimeWindow* pWin, int64_t delta) {
407,644,109✔
2979
  int64_t* ts = (int64_t*)pColData->pData;
407,644,109✔
2980

2981
  int64_t duration = pWin->ekey - pWin->skey + delta;
407,644,109✔
2982
  ts[2] = duration;            // set the duration
407,644,109✔
2983
  ts[3] = pWin->skey;          // window start key
407,644,109✔
2984
  ts[4] = pWin->ekey + delta;  // window end key
407,644,109✔
2985
}
407,644,109✔
2986

2987
int32_t compKeys(const SArray* pSortGroupCols, const char* oldkeyBuf, int32_t oldKeysLen, const SSDataBlock* pBlock,
21,352,681✔
2988
                 int32_t rowIndex) {
2989
  SColumnDataAgg* pColAgg = NULL;
21,352,681✔
2990
  const char*     isNull = oldkeyBuf;
21,352,681✔
2991
  const char*     p = oldkeyBuf + sizeof(int8_t) * pSortGroupCols->size;
21,352,681✔
2992

2993
  for (int32_t i = 0; i < pSortGroupCols->size; ++i) {
53,702,363✔
2994
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
32,832,135✔
2995
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
32,832,135✔
2996
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
32,832,135!
2997

2998
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
65,664,270!
2999
      if (isNull[i] != 1) return 1;
2,182,141✔
3000
    } else {
3001
      if (isNull[i] != 0) return 1;
30,649,994✔
3002
      const char* val = colDataGetData(pColInfoData, rowIndex);
30,635,773!
3003
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
30,635,773!
3004
        int32_t len = getJsonValueLen(val);
×
3005
        if (memcmp(p, val, len) != 0) return 1;
×
3006
        p += len;
×
3007
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
30,635,773!
3008
        if (memcmp(p, val, varDataTLen(val)) != 0) return 1;
10,445,293✔
3009
        p += varDataTLen(val);
10,308,949✔
3010
      } else {
3011
        if (0 != memcmp(p, val, pCol->bytes)) return 1;
20,190,480✔
3012
        p += pCol->bytes;
19,858,784✔
3013
      }
3014
    }
3015
  }
3016
  if ((int32_t)(p - oldkeyBuf) != oldKeysLen) return 1;
20,870,228✔
3017
  return 0;
20,869,932✔
3018
}
3019

3020
int32_t buildKeys(char* keyBuf, const SArray* pSortGroupCols, const SSDataBlock* pBlock, int32_t rowIndex) {
482,465✔
3021
  uint32_t        colNum = pSortGroupCols->size;
482,465✔
3022
  SColumnDataAgg* pColAgg = NULL;
482,465✔
3023
  char*           isNull = keyBuf;
482,465✔
3024
  char*           p = keyBuf + sizeof(int8_t) * colNum;
482,465✔
3025

3026
  for (int32_t i = 0; i < colNum; ++i) {
1,437,539✔
3027
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
955,074✔
3028
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
955,074✔
3029
    if (pCol->slotId > pBlock->pDataBlock->size) continue;
955,074!
3030

3031
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
955,074!
3032

3033
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
1,910,148!
3034
      isNull[i] = 1;
34,572✔
3035
    } else {
3036
      isNull[i] = 0;
920,502✔
3037
      const char* val = colDataGetData(pColInfoData, rowIndex);
920,502!
3038
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
920,502!
3039
        int32_t len = getJsonValueLen(val);
×
3040
        memcpy(p, val, len);
×
3041
        p += len;
×
3042
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
920,502!
3043
        varDataCopy(p, val);
139,805✔
3044
        p += varDataTLen(val);
139,805✔
3045
      } else {
3046
        memcpy(p, val, pCol->bytes);
780,697✔
3047
        p += pCol->bytes;
780,697✔
3048
      }
3049
    }
3050
  }
3051
  return (int32_t)(p - keyBuf);
482,465✔
3052
}
3053

3054
uint64_t calcGroupId(char* pData, int32_t len) {
17,791,262✔
3055
  T_MD5_CTX context;
3056
  tMD5Init(&context);
17,791,262✔
3057
  tMD5Update(&context, (uint8_t*)pData, len);
17,926,715✔
3058
  tMD5Final(&context);
17,954,664✔
3059

3060
  // NOTE: only extract the initial 8 bytes of the final MD5 digest
3061
  uint64_t id = 0;
18,073,666✔
3062
  memcpy(&id, context.digest, sizeof(uint64_t));
18,073,666✔
3063
  if (0 == id) memcpy(&id, context.digest + 8, sizeof(uint64_t));
18,073,666!
3064
  return id;
18,073,666✔
3065
}
3066

3067
SNodeList* makeColsNodeArrFromSortKeys(SNodeList* pSortKeys) {
944✔
3068
  SNode*     node;
3069
  SNodeList* ret = NULL;
944✔
3070
  FOREACH(node, pSortKeys) {
2,880!
3071
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)node;
1,936✔
3072
    int32_t           code = nodesListMakeAppend(&ret, pSortKey->pExpr);
1,936✔
3073
    if (code != TSDB_CODE_SUCCESS) {
1,936!
3074
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3075
      return NULL;
×
3076
    }
3077
  }
3078
  return ret;
944✔
3079
}
3080

3081
int32_t extractKeysLen(const SArray* keys, int32_t* pLen) {
944✔
3082
  int32_t code = TSDB_CODE_SUCCESS;
944✔
3083
  int32_t lino = 0;
944✔
3084
  int32_t len = 0;
944✔
3085
  int32_t keyNum = taosArrayGetSize(keys);
944✔
3086
  for (int32_t i = 0; i < keyNum; ++i) {
2,399✔
3087
    SColumn* pCol = (SColumn*)taosArrayGet(keys, i);
1,455✔
3088
    QUERY_CHECK_NULL(pCol, code, lino, _end, terrno);
1,455!
3089
    len += pCol->bytes;
1,455✔
3090
  }
3091
  len += sizeof(int8_t) * keyNum;  // null flag
944✔
3092
  *pLen = len;
944✔
3093

3094
_end:
944✔
3095
  if (code != TSDB_CODE_SUCCESS) {
944!
3096
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3097
  }
3098
  return code;
943✔
3099
}
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