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

taosdata / TDengine / #4717

04 Sep 2025 12:33PM UTC coverage: 58.697% (+0.1%) from 58.593%
#4717

push

travis-ci

web-flow
test: update case description (#32878)

134624 of 291691 branches covered (46.15%)

Branch coverage included in aggregate %.

202924 of 283375 relevant lines covered (71.61%)

26047439.86 hits per line

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

59.41
/source/libs/executor/src/executil.c
1
/*
2
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
3
 *
4
 * This program is free software: you can use, redistribute, and/or modify
5
 * it under the terms of the GNU Affero General Public License, version 3
6
 * or later ("AGPL"), as published by the Free Software Foundation.
7
 *
8
 * This program is distributed in the hope that it will be useful, but WITHOUT
9
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10
 * FITNESS FOR A PARTICULAR PURPOSE.
11
 *
12
 * You should have received a copy of the GNU Affero General Public License
13
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
14
 */
15

16
#include "function.h"
17
#include "functionMgt.h"
18
#include "index.h"
19
#include "os.h"
20
#include "query.h"
21
#include "querynodes.h"
22
#include "tarray.h"
23
#include "tdatablock.h"
24
#include "thash.h"
25
#include "tmsg.h"
26
#include "ttime.h"
27

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

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

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

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

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

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

60
static int64_t getLimit(const SNode* pLimit) {
48,628,769✔
61
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->limit) ? -1 : ((SLimitNode*)pLimit)->limit->datum.i;
48,628,769!
62
}
63
static int64_t getOffset(const SNode* pLimit) {
48,642,745✔
64
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->offset) ? -1 : ((SLimitNode*)pLimit)->offset->datum.i;
48,642,745✔
65
}
66
static void releaseColInfoData(void* pCol);
67

68
void initResultRowInfo(SResultRowInfo* pResultRowInfo) {
20,376,533✔
69
  pResultRowInfo->size = 0;
20,376,533✔
70
  pResultRowInfo->cur.pageId = -1;
20,376,533✔
71
}
20,376,533✔
72

73
void closeResultRow(SResultRow* pResultRow) { pResultRow->closed = true; }
18,120,196✔
74

75
void resetResultRow(SResultRow* pResultRow, size_t entrySize) {
115,952,188✔
76
  pResultRow->numOfRows = 0;
115,952,188✔
77
  pResultRow->closed = false;
115,952,188✔
78
  pResultRow->endInterp = false;
115,952,188✔
79
  pResultRow->startInterp = false;
115,952,188✔
80

81
  if (entrySize > 0) {
115,952,188!
82
    memset(pResultRow->pEntryInfo, 0, entrySize);
115,968,710✔
83
  }
84
}
115,952,188✔
85

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

91
size_t getResultRowSize(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
11,965,206✔
92
  int32_t rowSize = (numOfOutput * sizeof(SResultRowEntryInfo)) + sizeof(SResultRow);
11,965,206✔
93

94
  for (int32_t i = 0; i < numOfOutput; ++i) {
47,315,040✔
95
    rowSize += pCtx[i].resDataInfo.interBufSize;
35,349,834✔
96
  }
97

98
  return rowSize;
11,965,206✔
99
}
100

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

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

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

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

156
  qTrace("[StreamInternal] get result inBufSize:%zu, outBufSize:%zu", inBufSize, *outBufSize);
×
157
  return TSDB_CODE_SUCCESS;
×
158
}
159

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

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

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

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

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

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

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

208
  qTrace("[StreamInternal] put result inBufSize:%zu, outBufSize:%zu", inBufSize, *outBufSize);
×
209
  return TSDB_CODE_SUCCESS;
×
210
}
211

212
static void freeEx(void* p) { taosMemoryFree(*(void**)p); }
×
213

214
void cleanupGroupResInfo(SGroupResInfo* pGroupResInfo) {
8,258,914✔
215
  taosMemoryFreeClear(pGroupResInfo->pBuf);
8,258,914!
216
  if (pGroupResInfo->freeItem) {
8,259,744!
217
    //    taosArrayDestroy(pGroupResInfo->pRows);
218
    taosArrayDestroyEx(pGroupResInfo->pRows, freeEx);
×
219
    pGroupResInfo->freeItem = false;
×
220
    pGroupResInfo->pRows = NULL;
×
221
  } else {
222
    taosArrayDestroy(pGroupResInfo->pRows);
8,259,744✔
223
    pGroupResInfo->pRows = NULL;
8,260,847✔
224
  }
225
  pGroupResInfo->index = 0;
8,260,847✔
226
  pGroupResInfo->delIndex = 0;
8,260,847✔
227
}
8,260,847✔
228

229
int32_t resultrowComparAsc(const void* p1, const void* p2) {
2,147,483,647✔
230
  SResKeyPos* pp1 = *(SResKeyPos**)p1;
2,147,483,647✔
231
  SResKeyPos* pp2 = *(SResKeyPos**)p2;
2,147,483,647✔
232

233
  if (pp1->groupId == pp2->groupId) {
2,147,483,647✔
234
    int64_t pts1 = *(int64_t*)pp1->key;
2,147,483,647✔
235
    int64_t pts2 = *(int64_t*)pp2->key;
2,147,483,647✔
236

237
    if (pts1 == pts2) {
2,147,483,647!
238
      return 0;
×
239
    } else {
240
      return pts1 < pts2 ? -1 : 1;
2,147,483,647✔
241
    }
242
  } else {
243
    return pp1->groupId < pp2->groupId ? -1 : 1;
184,302,905✔
244
  }
245
}
246

247
static int32_t resultrowComparDesc(const void* p1, const void* p2) { return resultrowComparAsc(p2, p1); }
47,878,519✔
248

249
int32_t initGroupedResultInfo(SGroupResInfo* pGroupResInfo, SSHashObj* pHashmap, int32_t order) {
6,660,230✔
250
  int32_t code = TSDB_CODE_SUCCESS;
6,660,230✔
251
  int32_t lino = 0;
6,660,230✔
252
  if (pGroupResInfo->pRows != NULL) {
6,660,230✔
253
    taosArrayDestroy(pGroupResInfo->pRows);
1,617✔
254
  }
255
  if (pGroupResInfo->pBuf) {
6,660,230✔
256
    taosMemoryFree(pGroupResInfo->pBuf);
1,616!
257
    pGroupResInfo->pBuf = NULL;
1,617✔
258
  }
259

260
  // extract the result rows information from the hash map
261
  int32_t size = tSimpleHashGetSize(pHashmap);
6,660,231✔
262

263
  void* pData = NULL;
6,660,524✔
264
  pGroupResInfo->pRows = taosArrayInit(size, POINTER_BYTES);
6,660,524✔
265
  QUERY_CHECK_NULL(pGroupResInfo->pRows, code, lino, _end, terrno);
6,661,963!
266

267
  size_t  keyLen = 0;
6,661,963✔
268
  int32_t iter = 0;
6,661,963✔
269
  int64_t bufLen = 0, offset = 0;
6,661,963✔
270

271
  // todo move away and record this during create window
272
  while ((pData = tSimpleHashIterate(pHashmap, pData, &iter)) != NULL) {
346,077,630✔
273
    /*void* key = */ (void)tSimpleHashGetKey(pData, &keyLen);
274
    bufLen += keyLen + sizeof(SResultRowPosition);
339,415,667✔
275
  }
276

277
  pGroupResInfo->pBuf = taosMemoryMalloc(bufLen);
6,649,639!
278
  QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
6,661,543!
279

280
  iter = 0;
6,661,543✔
281
  while ((pData = tSimpleHashIterate(pHashmap, pData, &iter)) != NULL) {
345,528,691✔
282
    void* key = tSimpleHashGetKey(pData, &keyLen);
338,960,603✔
283

284
    SResKeyPos* p = (SResKeyPos*)(pGroupResInfo->pBuf + offset);
338,960,603✔
285

286
    p->groupId = *(uint64_t*)key;
338,960,603✔
287
    p->pos = *(SResultRowPosition*)pData;
338,960,603✔
288
    memcpy(p->key, (char*)key + sizeof(uint64_t), keyLen - sizeof(uint64_t));
338,960,603✔
289
    void* tmp = taosArrayPush(pGroupResInfo->pRows, &p);
338,960,603✔
290
    QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
338,867,148!
291

292
    offset += keyLen + sizeof(struct SResultRowPosition);
338,867,148✔
293
  }
294

295
  if (order == TSDB_ORDER_ASC || order == TSDB_ORDER_DESC) {
6,587,385✔
296
    __compar_fn_t fn = (order == TSDB_ORDER_ASC) ? resultrowComparAsc : resultrowComparDesc;
3,094,732✔
297
    size = POINTER_BYTES;
3,094,732✔
298
    taosSort(pGroupResInfo->pRows->pData, taosArrayGetSize(pGroupResInfo->pRows), size, fn);
3,094,732✔
299
  }
300

301
  pGroupResInfo->index = 0;
6,661,880✔
302

303
_end:
6,661,880✔
304
  if (code != TSDB_CODE_SUCCESS) {
6,661,880!
305
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
306
  }
307
  return code;
6,660,937✔
308
}
309

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

315
  pGroupResInfo->freeItem = true;
×
316
  pGroupResInfo->pRows = pArrayList;
×
317
  pGroupResInfo->index = 0;
×
318
  pGroupResInfo->delIndex = 0;
×
319
}
×
320

321
bool hasRemainResults(SGroupResInfo* pGroupResInfo) {
18,060,350✔
322
  if (pGroupResInfo->pRows == NULL) {
18,060,350!
323
    return false;
×
324
  }
325

326
  return pGroupResInfo->index < taosArrayGetSize(pGroupResInfo->pRows);
18,060,350✔
327
}
328

329
int32_t getNumOfTotalRes(SGroupResInfo* pGroupResInfo) {
12,291,815✔
330
  if (pGroupResInfo->pRows == 0) {
12,291,815!
331
    return 0;
×
332
  }
333

334
  return (int32_t)taosArrayGetSize(pGroupResInfo->pRows);
12,291,815✔
335
}
336

337
SArray* createSortInfo(SNodeList* pNodeList) {
4,179,653✔
338
  size_t numOfCols = 0;
4,179,653✔
339

340
  if (pNodeList != NULL) {
4,179,653✔
341
    numOfCols = LIST_LENGTH(pNodeList);
4,179,271!
342
  } else {
343
    numOfCols = 0;
382✔
344
  }
345

346
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
4,179,653✔
347
  if (pList == NULL) {
4,179,717!
348
    return pList;
×
349
  }
350

351
  for (int32_t i = 0; i < numOfCols; ++i) {
8,444,579✔
352
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
4,264,975✔
353
    if (!pSortKey) {
4,264,999!
354
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
355
      taosArrayDestroy(pList);
×
356
      pList = NULL;
×
357
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
358
      break;
×
359
    }
360
    SBlockOrderInfo bi = {0};
4,265,005✔
361
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
4,265,005✔
362
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);
4,265,005✔
363

364
    if (nodeType(pSortKey->pExpr) != QUERY_NODE_COLUMN) {
4,265,005!
365
      qError("invalid order by expr type:%d", nodeType(pSortKey->pExpr));
×
366
      taosArrayDestroy(pList);
×
367
      pList = NULL;
×
368
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
369
      break;
×
370
    }
371
    
372
    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
4,265,005✔
373
    bi.slotId = pColNode->slotId;
4,265,005✔
374
    void* tmp = taosArrayPush(pList, &bi);
4,264,862✔
375
    if (!tmp) {
4,264,862!
376
      taosArrayDestroy(pList);
×
377
      pList = NULL;
×
378
      break;
×
379
    }
380
  }
381

382
  return pList;
4,179,604✔
383
}
384

385
SSDataBlock* createDataBlockFromDescNode(void* p) {
37,044,861✔
386
  SDataBlockDescNode* pNode = (SDataBlockDescNode*)p;
37,044,861✔
387
  int32_t      numOfCols = LIST_LENGTH(pNode->pSlots);
37,044,861!
388
  SSDataBlock* pBlock = NULL;
37,044,861✔
389
  int32_t      code = createDataBlock(&pBlock);
37,044,861✔
390
  if (code) {
37,061,302!
391
    terrno = code;
×
392
    return NULL;
×
393
  }
394

395
  pBlock->info.id.blockId = pNode->dataBlockId;
37,064,028✔
396
  pBlock->info.type = STREAM_INVALID;
37,064,028✔
397
  pBlock->info.calWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
37,064,028✔
398
  pBlock->info.watermark = INT64_MIN;
37,064,028✔
399

400
  for (int32_t i = 0; i < numOfCols; ++i) {
203,072,332✔
401
    SSlotDescNode* pDescNode = (SSlotDescNode*)nodesListGetNode(pNode->pSlots, i);
166,019,893✔
402
    if (!pDescNode) {
166,082,242!
403
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
404
      blockDataDestroy(pBlock);
×
405
      pBlock = NULL;
×
406
      terrno = TSDB_CODE_INVALID_PARA;
×
407
      break;
×
408
    }
409
    SColumnInfoData idata =
410
        createColumnInfoData(pDescNode->dataType.type, pDescNode->dataType.bytes, pDescNode->slotId);
166,082,242✔
411
    idata.info.scale = pDescNode->dataType.scale;
166,579,159✔
412
    idata.info.precision = pDescNode->dataType.precision;
166,579,159✔
413
    idata.info.noData = pDescNode->reserve;
166,579,159✔
414

415
    code = blockDataAppendColInfo(pBlock, &idata);
166,579,159✔
416
    if (code != TSDB_CODE_SUCCESS) {
166,008,304!
417
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
418
      blockDataDestroy(pBlock);
×
419
      pBlock = NULL;
×
420
      terrno = code;
×
421
      break;
×
422
    }
423
  }
424

425
  return pBlock;
37,052,439✔
426
}
427

428
int32_t prepareDataBlockBuf(SSDataBlock* pDataBlock, SColMatchInfo* pMatchInfo) {
8,431,694✔
429
  SDataBlockInfo* pBlockInfo = &pDataBlock->info;
8,431,694✔
430

431
  for (int32_t i = 0; i < taosArrayGetSize(pMatchInfo->pList); ++i) {
52,499,989✔
432
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
44,414,565✔
433
    if (!pItem) {
44,437,521✔
434
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
120!
435
      return terrno;
120✔
436
    }
437

438
    if (pItem->isPk) {
44,437,401✔
439
      SColumnInfoData* pInfoData = taosArrayGet(pDataBlock->pDataBlock, pItem->dstSlotId);
369,106✔
440
      if (!pInfoData) {
369,111!
441
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
442
        return terrno;
×
443
      }
444
      pBlockInfo->pks[0].type = pInfoData->info.type;
369,111✔
445
      pBlockInfo->pks[1].type = pInfoData->info.type;
369,111✔
446

447
      // allocate enough buffer size, which is pInfoData->info.bytes
448
      if (IS_VAR_DATA_TYPE(pItem->dataType.type)) {
369,111!
449
        pBlockInfo->pks[0].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
17,242!
450
        if (pBlockInfo->pks[0].pData == NULL) {
17,248!
451
          return terrno;
×
452
        }
453

454
        pBlockInfo->pks[1].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
17,248!
455
        if (pBlockInfo->pks[1].pData == NULL) {
17,245!
456
          taosMemoryFreeClear(pBlockInfo->pks[0].pData);
×
457
          return terrno;
×
458
        }
459

460
        pBlockInfo->pks[0].nData = pInfoData->info.bytes;
17,245✔
461
        pBlockInfo->pks[1].nData = pInfoData->info.bytes;
17,245✔
462
      }
463

464
      break;
369,114✔
465
    }
466
  }
467

468
  return TSDB_CODE_SUCCESS;
8,414,755✔
469
}
470

471
EDealRes doTranslateTagExpr(SNode** pNode, void* pContext) {
637✔
472
  STransTagExprCtx* pCtx = pContext;
637✔
473
  SMetaReader*      mr = pCtx->pReader;
637✔
474
  bool              isTagCol = false, isTbname = false;
637✔
475
  if (nodeType(*pNode) == QUERY_NODE_COLUMN) {
637✔
476
    SColumnNode* pCol = (SColumnNode*)*pNode;
182✔
477
    if (pCol->colType == COLUMN_TYPE_TBNAME)
182!
478
      isTbname = true;
×
479
    else
480
      isTagCol = true;
182✔
481
  } else if (nodeType(*pNode) == QUERY_NODE_FUNCTION) {
455!
482
    SFunctionNode* pFunc = (SFunctionNode*)*pNode;
×
483
    if (pFunc->funcType == FUNCTION_TYPE_TBNAME) isTbname = true;
×
484
  }
485
  if (isTagCol) {
637✔
486
    SColumnNode* pSColumnNode = *(SColumnNode**)pNode;
182✔
487

488
    SValueNode* res = NULL;
182✔
489
    pCtx->code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
182✔
490
    if (NULL == res) {
182!
491
      return DEAL_RES_ERROR;
×
492
    }
493

494
    res->translate = true;
182✔
495
    res->node.resType = pSColumnNode->node.resType;
182✔
496

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

514
      res->datum.p = taosMemoryCalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1, 1);
182!
515
      if (NULL == res->datum.p) {
182!
516
        return DEAL_RES_ERROR;
×
517
      }
518

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

541
    res->translate = true;
×
542
    res->node.resType = ((SExprNode*)(*pNode))->resType;
×
543

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

555
  return DEAL_RES_CONTINUE;
637✔
556
}
557

558
int32_t isQualifiedTable(STableKeyInfo* info, SNode* pTagCond, void* metaHandle, bool* pQualified, SStorageAPI* pAPI) {
91✔
559
  int32_t     code = TSDB_CODE_SUCCESS;
91✔
560
  SMetaReader mr = {0};
91✔
561

562
  pAPI->metaReaderFn.initReader(&mr, metaHandle, META_READER_LOCK, &pAPI->metaFn);
91✔
563
  code = pAPI->metaReaderFn.getEntryGetUidCache(&mr, info->uid);
91✔
564
  if (TSDB_CODE_SUCCESS != code) {
91!
565
    pAPI->metaReaderFn.clearReader(&mr);
×
566
    *pQualified = false;
×
567

568
    return TSDB_CODE_SUCCESS;
×
569
  }
570

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

587
  SNode* pNew = NULL;
91✔
588
  code = scalarCalculateConstants(pTagCondTmp, &pNew);
91✔
589
  if (TSDB_CODE_SUCCESS != code) {
91!
590
    terrno = code;
×
591
    nodesDestroyNode(pTagCondTmp);
×
592
    *pQualified = false;
×
593

594
    return code;
×
595
  }
596

597
  SValueNode* pValue = (SValueNode*)pNew;
91✔
598
  *pQualified = pValue->datum.b;
91✔
599

600
  nodesDestroyNode(pNew);
91✔
601
  return TSDB_CODE_SUCCESS;
91✔
602
}
603

604
static EDealRes getColumn(SNode** pNode, void* pContext) {
93,543,991✔
605
  tagFilterAssist* pData = (tagFilterAssist*)pContext;
93,543,991✔
606
  SColumnNode*     pSColumnNode = NULL;
93,543,991✔
607
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
93,543,991✔
608
    pSColumnNode = *(SColumnNode**)pNode;
6,597,351✔
609
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
86,946,640✔
610
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
862,877✔
611
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
862,877✔
612
      pData->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
862,752✔
613
      if (NULL == pSColumnNode) {
862,742!
614
        return DEAL_RES_ERROR;
×
615
      }
616
      pSColumnNode->colId = -1;
862,742✔
617
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
862,742✔
618
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
862,742✔
619
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
862,742✔
620
      nodesDestroyNode(*pNode);
862,742✔
621
      *pNode = (SNode*)pSColumnNode;
862,755✔
622
    } else {
623
      return DEAL_RES_CONTINUE;
125✔
624
    }
625
  } else {
626
    return DEAL_RES_CONTINUE;
86,083,763✔
627
  }
628

629
  void* data = taosHashGet(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
7,460,106✔
630
  if (!data) {
7,456,612✔
631
    int32_t tempRes =
632
        taosHashPut(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
5,997,790✔
633
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
6,006,484!
634
      return DEAL_RES_ERROR;
×
635
    }
636
    pSColumnNode->slotId = pData->index++;
6,006,484✔
637
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
6,006,484✔
638
                         .type = pSColumnNode->node.resType.type,
6,006,484✔
639
                         .bytes = pSColumnNode->node.resType.bytes,
6,006,484✔
640
                         .pk = pSColumnNode->isPk};
6,006,484✔
641
#if TAG_FILTER_DEBUG
642
    qDebug("tagfilter build column info, slotId:%d, colId:%d, type:%d", pSColumnNode->slotId, cInfo.colId, cInfo.type);
643
#endif
644
    void* tmp = taosArrayPush(pData->cInfoList, &cInfo);
6,006,484✔
645
    if (!tmp) {
5,994,521!
646
      return DEAL_RES_ERROR;
×
647
    }
648
  } else {
649
    SColumnNode* col = *(SColumnNode**)data;
1,458,822✔
650
    pSColumnNode->slotId = col->slotId;
1,458,822✔
651
  }
652

653
  return DEAL_RES_CONTINUE;
7,453,343✔
654
}
655

656
static int32_t createResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
2,623,432✔
657
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
2,623,432!
658
  if (pColumnData == NULL) {
2,625,317!
659
    return terrno;
×
660
  }
661

662
  pColumnData->info.type = pType->type;
2,625,317✔
663
  pColumnData->info.bytes = pType->bytes;
2,625,317✔
664
  pColumnData->info.scale = pType->scale;
2,625,317✔
665
  pColumnData->info.precision = pType->precision;
2,625,317✔
666

667
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
2,625,317✔
668
  if (code != TSDB_CODE_SUCCESS) {
2,625,079!
669
    terrno = code;
×
670
    releaseColInfoData(pColumnData);
×
671
    return terrno;
×
672
  }
673

674
  pParam->columnData = pColumnData;
2,625,104✔
675
  pParam->colAlloced = true;
2,625,104✔
676
  return TSDB_CODE_SUCCESS;
2,625,104✔
677
}
678

679
static void releaseColInfoData(void* pCol) {
197,387✔
680
  if (pCol) {
197,387✔
681
    SColumnInfoData* col = (SColumnInfoData*)pCol;
197,386✔
682
    colDataDestroy(col);
197,386✔
683
    taosMemoryFree(col);
197,394!
684
  }
685
}
197,415✔
686

687
void freeItem(void* p) {
8,141,877✔
688
  STUidTagInfo* pInfo = p;
8,141,877✔
689
  if (pInfo->pTagVal != NULL) {
8,141,877!
690
    taosMemoryFree(pInfo->pTagVal);
8,142,418!
691
  }
692
}
8,142,609✔
693

694
static int32_t genTagFilterDigest(const SNode* pTagCond, T_MD5_CTX* pContext) {
×
695
  if (pTagCond == NULL) {
×
696
    return TSDB_CODE_SUCCESS;
×
697
  }
698

699
  char*   payload = NULL;
×
700
  int32_t len = 0;
×
701
  int32_t code = nodesNodeToMsg(pTagCond, &payload, &len);
×
702
  if (code != TSDB_CODE_SUCCESS) {
×
703
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
704
    return code;
×
705
  }
706

707
  tMD5Init(pContext);
×
708
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
×
709
  tMD5Final(pContext);
×
710

711
  taosMemoryFree(payload);
×
712
  return TSDB_CODE_SUCCESS;
×
713
}
714

715
static int32_t genTbGroupDigest(const SNode* pGroup, uint8_t* filterDigest, T_MD5_CTX* pContext) {
×
716
  int32_t code = TSDB_CODE_SUCCESS;
×
717
  int32_t lino = 0;
×
718
  char*   payload = NULL;
×
719
  int32_t len = 0;
×
720
  code = nodesNodeToMsg(pGroup, &payload, &len);
×
721
  QUERY_CHECK_CODE(code, lino, _end);
×
722

723
  if (filterDigest[0]) {
×
724
    payload = taosMemoryRealloc(payload, len + tListLen(pContext->digest));
×
725
    QUERY_CHECK_NULL(payload, code, lino, _end, terrno);
×
726
    memcpy(payload + len, filterDigest + 1, tListLen(pContext->digest));
×
727
    len += tListLen(pContext->digest);
×
728
  }
729

730
  tMD5Init(pContext);
×
731
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
×
732
  tMD5Final(pContext);
×
733

734
_end:
×
735
  taosMemoryFree(payload);
×
736
  if (code != TSDB_CODE_SUCCESS) {
×
737
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
738
  }
739
  return code;
×
740
}
741

742
int32_t qGetColumnsFromNodeList(void* data, bool isList, SArray** pColList) {
2,538,978✔
743
  int32_t code = TSDB_CODE_SUCCESS;
2,538,978✔
744
  tagFilterAssist ctx = {0};
2,538,978✔
745
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
2,538,978✔
746
  if (ctx.colHash == NULL) {
2,538,664!
747
    code = terrno;
×
748
    goto end;
×
749
  }
750

751
  ctx.index = 0;
2,538,664✔
752
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
2,538,664✔
753
  if (ctx.cInfoList == NULL) {
2,540,775✔
754
    code = terrno;
1,976✔
755
    goto end;
×
756
  }
757

758
  if (isList) {
2,538,799✔
759
    SNode* pNode = NULL;
113,343✔
760
    FOREACH(pNode, (SNodeList*)data) {
310,725✔
761
      nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
197,412✔
762
      if (TSDB_CODE_SUCCESS != ctx.code) {
197,382!
763
        code = ctx.code;
×
764
        goto end;
×
765
      }
766
      REPLACE_NODE(pNode);
197,382✔
767
    }
768
  } else {
769
    SNode* pNode = (SNode*)data;
2,425,456✔
770
    nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
2,425,456✔
771
    if (TSDB_CODE_SUCCESS != ctx.code) {
2,426,227!
772
      code = ctx.code;
×
773
      goto end;
×
774
    }
775
  }
776
  
777
  if (pColList != NULL) *pColList = ctx.cInfoList;
2,539,540✔
778
  ctx.cInfoList = NULL;
2,539,540✔
779

780
end:
2,539,540✔
781
  taosHashCleanup(ctx.colHash);
2,539,540✔
782
  taosArrayDestroy(ctx.cInfoList);
2,539,951✔
783
  return code;
2,540,086✔
784
}
785

786
static int32_t buildGroupInfo(SColumnInfoData* pValue, int32_t i, SArray* gInfo) {
280,261✔
787
  int32_t code = TSDB_CODE_SUCCESS;
280,261✔
788
  SStreamGroupValue* v = taosArrayReserve(gInfo, 1);
280,261✔
789
  if (v == NULL) {
280,234!
790
    code = terrno;
×
791
    goto end;
×
792
  }
793
  if (colDataIsNull_s(pValue, i)) {
560,482!
794
    v->isNull = true;
×
795
  } else {
796
    v->isNull = false;
280,241✔
797
    char* data = colDataGetData(pValue, i);
280,241!
798
    if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
280,241!
799
      if (tTagIsJson(data)) {
×
800
        code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
×
801
        goto end;
×
802
      }
803
      if (tTagIsJsonNull(data)) {
×
804
        v->isNull = true;
×
805
        goto end;
×
806
      }
807
      int32_t len = getJsonValueLen(data);
×
808
      v->data.type = pValue->info.type;
×
809
      v->data.nData = len;
×
810
      v->data.pData = taosMemoryCalloc(1, len + 1);
×
811
      if (v->data.pData == NULL) {
×
812
        code = terrno;
×
813
        goto end;
×
814
      }
815
      memcpy(v->data.pData, data, len);
×
816
      qDebug("buildGroupInfo:%d add json data len:%d, data:%s", i, len, (char*)v->data.pData);
×
817
    } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
280,241!
818
      if (varDataTLen(data) > pValue->info.bytes) {
272,902!
819
        code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
820
        goto end;
×
821
      }
822
      v->data.type = pValue->info.type;
272,902✔
823
      v->data.nData = varDataLen(data);
272,902✔
824
      v->data.pData = taosMemoryCalloc(1, varDataLen(data) + 1);
272,902!
825
      if (v->data.pData == NULL) {
273,023!
826
        code = terrno;
×
827
        goto end;
×
828
      }
829
      memcpy(v->data.pData, varDataVal(data), varDataLen(data));
273,023✔
830
      qDebug("buildGroupInfo:%d add var data type:%d, len:%d, data:%s", i, pValue->info.type, varDataLen(data), (char*)v->data.pData);
273,023✔
831
    } else if (pValue->info.type == TSDB_DATA_TYPE_DECIMAL) {  // reader todo decimal
7,339!
832
      v->data.type = pValue->info.type;
×
833
      v->data.nData = pValue->info.bytes;
×
834
      v->data.pData = taosMemoryCalloc(1, pValue->info.bytes);
×
835
      if (v->data.pData == NULL) {
×
836
        code = terrno;
×
837
        goto end;
×
838
      }
839
      memcpy(&v->data.pData, data, pValue->info.bytes);
×
840
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
×
841
    } else {  // reader todo decimal
842
      v->data.type = pValue->info.type;
7,339✔
843
      memcpy(&v->data.val, data, pValue->info.bytes);
7,339✔
844
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
7,339✔
845
    }
846
  }
847
end:
5,957✔
848
  if (code != TSDB_CODE_SUCCESS) {
280,292!
849
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
850
    v->isNull = true;
×
851
  }
852
  return code;
280,292✔
853
}
854

855
static void getColInfoResultForGroupbyForStream(void* pVnode, SNodeList* group, STableListInfo* pTableListInfo,
32,806✔
856
                                   SStorageAPI* pAPI, SHashObj* groupIdMap) {
857
  int32_t      code = TSDB_CODE_SUCCESS;
32,806✔
858
  int32_t      lino = 0;
32,806✔
859
  SArray*      pBlockList = NULL;
32,806✔
860
  SSDataBlock* pResBlock = NULL;
32,806✔
861
  SArray*      groupData = NULL;
32,806✔
862
  SArray*      pUidTagList = NULL;
32,806✔
863
  SArray*      gInfo = NULL;
32,806✔
864
  int32_t      tbNameIndex = 0;
32,806✔
865

866
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
32,806✔
867
  if (rows == 0) {
32,807!
868
    return;
×
869
  }
870

871
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
32,807✔
872
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
32,819✔
873

874
  for (int32_t i = 0; i < rows; ++i) {
227,830✔
875
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
195,021✔
876
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
195,013!
877
    STUidTagInfo info = {.uid = pkeyInfo->uid};
195,013✔
878
    void*        tmp = taosArrayPush(pUidTagList, &info);
195,012✔
879
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
195,012!
880
  }
881

882
  code = pAPI->metaFn.getTableTags(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
32,809✔
883
  if (code != TSDB_CODE_SUCCESS) {
32,818!
884
    goto end;
×
885
  }
886

887
  SArray* pColList = NULL;
32,818✔
888
  code = qGetColumnsFromNodeList(group, true, &pColList);
32,818✔
889
  if (code != TSDB_CODE_SUCCESS) {
32,807!
890
    goto end;
×
891
  }
892

893
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
144,446✔
894
    SColumnInfo* tmp = (SColumnInfo*)taosArrayGet(pColList, i);
111,637✔
895
    if (tmp != NULL && tmp->colId == -1) {
111,639!
896
      tbNameIndex = i;
32,805✔
897
    }
898
  }
899
  
900
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
32,792✔
901
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
32,804✔
902
  taosArrayDestroy(pColList);
32,810✔
903
  if (pResBlock == NULL) {
32,811!
904
    code = terrno;
×
905
    goto end;
×
906
  }
907

908
  pBlockList = taosArrayInit(2, POINTER_BYTES);
32,811✔
909
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
32,817!
910

911
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
32,814✔
912
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
32,814!
913

914
  groupData = taosArrayInit(2, POINTER_BYTES);
32,814✔
915
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
32,813✔
916

917
  SNode* pNode = NULL;
32,812✔
918
  FOREACH(pNode, group) {
144,459!
919
    SScalarParam output = {0};
111,649✔
920

921
    switch (nodeType(pNode)) {
111,649!
922
      case QUERY_NODE_VALUE:
×
923
        break;
×
924
      case QUERY_NODE_COLUMN:
111,649✔
925
      case QUERY_NODE_OPERATOR:
926
      case QUERY_NODE_FUNCTION: {
927
        SExprNode* expNode = (SExprNode*)pNode;
111,649✔
928
        code = createResultData(&expNode->resType, rows, &output);
111,649✔
929
        if (code != TSDB_CODE_SUCCESS) {
111,663!
930
          goto end;
×
931
        }
932
        break;
111,663✔
933
      }
934

935
      default:
×
936
        code = TSDB_CODE_OPS_NOT_SUPPORT;
×
937
        goto end;
×
938
    }
939

940
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
111,663✔
941
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
111,661✔
942
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
111,661✔
943
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
111,654!
944
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
111,654✔
945
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
2!
946
      continue;
×
947
    } else {
948
      code = scalarCalculate(pNode, pBlockList, &output, NULL, NULL);
2✔
949
    }
950

951
    if (code != TSDB_CODE_SUCCESS) {
111,651!
952
      releaseColInfoData(output.columnData);
×
953
      goto end;
×
954
    }
955

956
    void* tmp = taosArrayPush(groupData, &output.columnData);
111,647✔
957
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
111,647!
958
  }
959

960
  for (int i = 0; i < rows; i++) {
227,836✔
961
    gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
195,013✔
962
    QUERY_CHECK_NULL(gInfo, code, lino, end, terrno);
195,013!
963

964
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
195,013✔
965
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
195,000!
966

967
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
473,827✔
968
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
278,832✔
969
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
278,796✔
970
        if (ret != TSDB_CODE_SUCCESS) {
278,832!
971
          qError("buildGroupInfo failed at line %d since %s", __LINE__, tstrerror(ret));
×
972
          goto end;
×
973
        }
974
        if (j == tbNameIndex) {
278,832✔
975
          SStreamGroupValue* v = taosArrayGetLast(gInfo);
195,015✔
976
          if (v != NULL){
195,006!
977
            v->isTbname = true;
195,014✔
978
            v->uid = info->uid;
195,014✔
979
          }
980
        }
981
    }
982

983
    int32_t ret = taosHashPut(groupIdMap, &info->uid, sizeof(info->uid), &gInfo, POINTER_BYTES);
194,846✔
984
    if (ret != TSDB_CODE_SUCCESS) {
195,028!
985
      qError("put groupid to map failed at line %d since %s", __LINE__, tstrerror(ret));
×
986
      goto end;
×
987
    }
988
    qDebug("put groupid to map gid:%" PRIu64, info->uid);
195,028✔
989
    gInfo = NULL;
195,026✔
990
  }
991

992
end:
32,823✔
993
  blockDataDestroy(pResBlock);
32,823✔
994
  taosArrayDestroy(pBlockList);
32,819✔
995
  taosArrayDestroyEx(pUidTagList, freeItem);
32,820✔
996
  taosArrayDestroyP(groupData, releaseColInfoData);
32,819✔
997
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
32,820✔
998

999
  if (code != TSDB_CODE_SUCCESS) {
32,820!
1000
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1001
  }
1002
}
1003

1004
int32_t getColInfoResultForGroupby(void* pVnode, SNodeList* group, STableListInfo* pTableListInfo, uint8_t* digest,
80,475✔
1005
                                   SStorageAPI* pAPI, bool initRemainGroups, SHashObj* groupIdMap) {
1006
  int32_t      code = TSDB_CODE_SUCCESS;
80,475✔
1007
  int32_t      lino = 0;
80,475✔
1008
  SArray*      pBlockList = NULL;
80,475✔
1009
  SSDataBlock* pResBlock = NULL;
80,475✔
1010
  void*        keyBuf = NULL;
80,475✔
1011
  SArray*      groupData = NULL;
80,475✔
1012
  SArray*      pUidTagList = NULL;
80,475✔
1013
  SArray*      tableList = NULL;
80,475✔
1014
  SArray*      gInfo = NULL;
80,475✔
1015

1016
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
80,475✔
1017
  if (rows == 0) {
80,473!
1018
    return TSDB_CODE_SUCCESS;
×
1019
  } 
1020

1021
  T_MD5_CTX context = {0};
80,473✔
1022
  if (tsTagFilterCache) {
80,473!
1023
    SNodeListNode* listNode = NULL;
×
1024
    code = nodesMakeNode(QUERY_NODE_NODE_LIST, (SNode**)&listNode);
×
1025
    if (TSDB_CODE_SUCCESS != code) {
×
1026
      goto end;
×
1027
    }
1028
    listNode->pNodeList = group;
×
1029
    code = genTbGroupDigest((SNode*)listNode, digest, &context);
×
1030
    QUERY_CHECK_CODE(code, lino, end);
×
1031

1032
    nodesFree(listNode);
×
1033

1034
    code = pAPI->metaFn.metaGetCachedTbGroup(pVnode, pTableListInfo->idInfo.suid, context.digest,
×
1035
                                             tListLen(context.digest), &tableList);
1036
    QUERY_CHECK_CODE(code, lino, end);
×
1037

1038
    if (tableList) {
×
1039
      taosArrayDestroy(pTableListInfo->pTableList);
×
1040
      pTableListInfo->pTableList = tableList;
×
1041
      qDebug("retrieve tb group list from cache, numOfTables:%d",
×
1042
             (int32_t)taosArrayGetSize(pTableListInfo->pTableList));
1043
      goto end;
×
1044
    }
1045
  }
1046

1047
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
80,473✔
1048
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
80,477!
1049

1050
  for (int32_t i = 0; i < rows; ++i) {
379,448✔
1051
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
299,000✔
1052
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
298,678!
1053
    STUidTagInfo info = {.uid = pkeyInfo->uid};
298,678✔
1054
    void*        tmp = taosArrayPush(pUidTagList, &info);
298,968✔
1055
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
298,968!
1056
  }
1057

1058
  code = pAPI->metaFn.getTableTags(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
80,448✔
1059
  if (code != TSDB_CODE_SUCCESS) {
80,526!
1060
    goto end;
×
1061
  }
1062

1063
  SArray* pColList = NULL;
80,526✔
1064
  code = qGetColumnsFromNodeList(group, true, &pColList); 
80,526✔
1065
  if (code != TSDB_CODE_SUCCESS) {
80,507!
1066
    goto end;
×
1067
  }
1068

1069
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
80,507✔
1070
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
80,486✔
1071
  taosArrayDestroy(pColList);
80,505✔
1072
  if (pResBlock == NULL) {
80,503!
1073
    code = terrno;
×
1074
    goto end;
×
1075
  }
1076

1077
  //  int64_t st1 = taosGetTimestampUs();
1078
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1079

1080
  pBlockList = taosArrayInit(2, POINTER_BYTES);
80,503✔
1081
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
80,508!
1082

1083
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
80,502✔
1084
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
80,502!
1085

1086
  groupData = taosArrayInit(2, POINTER_BYTES);
80,502✔
1087
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
80,504!
1088

1089
  SNode* pNode = NULL;
80,507✔
1090
  FOREACH(pNode, group) {
166,208!
1091
    SScalarParam output = {0};
85,716✔
1092

1093
    switch (nodeType(pNode)) {
85,716!
1094
      case QUERY_NODE_VALUE:
×
1095
        break;
×
1096
      case QUERY_NODE_COLUMN:
85,716✔
1097
      case QUERY_NODE_OPERATOR:
1098
      case QUERY_NODE_FUNCTION: {
1099
        SExprNode* expNode = (SExprNode*)pNode;
85,716✔
1100
        code = createResultData(&expNode->resType, rows, &output);
85,716✔
1101
        if (code != TSDB_CODE_SUCCESS) {
85,714!
1102
          goto end;
×
1103
        }
1104
        break;
85,714✔
1105
      }
1106

1107
      default:
×
1108
        code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1109
        goto end;
×
1110
    }
1111

1112
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
85,714✔
1113
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
85,677✔
1114
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
85,677✔
1115
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
85,630!
1116
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
85,630✔
1117
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
37!
1118
      continue;
×
1119
    } else {
1120
      code = scalarCalculate(pNode, pBlockList, &output, NULL, NULL);
37✔
1121
    }
1122

1123
    if (code != TSDB_CODE_SUCCESS) {
85,715!
1124
      releaseColInfoData(output.columnData);
×
1125
      goto end;
×
1126
    }
1127

1128
    void* tmp = taosArrayPush(groupData, &output.columnData);
85,701✔
1129
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
85,701!
1130
  }
1131

1132
  int32_t keyLen = 0;
80,492✔
1133
  SNode*  node;
1134
  FOREACH(node, group) {
166,203✔
1135
    SExprNode* pExpr = (SExprNode*)node;
85,711✔
1136
    keyLen += pExpr->resType.bytes;
85,711✔
1137
  }
1138

1139
  int32_t nullFlagSize = sizeof(int8_t) * LIST_LENGTH(group);
80,492✔
1140
  keyLen += nullFlagSize;
80,492✔
1141

1142
  keyBuf = taosMemoryCalloc(1, keyLen);
80,492✔
1143
  if (keyBuf == NULL) {
80,501!
1144
    code = terrno;
×
1145
    goto end;
×
1146
  }
1147

1148
  if (initRemainGroups) {
80,501✔
1149
    pTableListInfo->remainGroups =
13,697✔
1150
        taosHashInit(rows, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
13,707✔
1151
    if (pTableListInfo->remainGroups == NULL) {
13,697!
1152
      code = terrno;
×
1153
      goto end;
×
1154
    }
1155
  }
1156

1157
  for (int i = 0; i < rows; i++) {
379,457✔
1158
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
298,853✔
1159
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
298,492!
1160

1161
    if (groupIdMap != NULL){
298,492✔
1162
      gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
956✔
1163
    }
1164
    
1165
    char* isNull = (char*)keyBuf;
298,484✔
1166
    char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(group);
298,484!
1167
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
609,543✔
1168
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
311,274✔
1169

1170
      if (groupIdMap != NULL && gInfo != NULL) {
311,036!
1171
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
1,466✔
1172
        if (ret != TSDB_CODE_SUCCESS) {
1,466!
1173
          qError("buildGroupInfo failed at line %d since %s", __LINE__, tstrerror(ret));
×
1174
          taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1175
          gInfo = NULL;
×
1176
        }
1177
      }
1178
      
1179
      if (colDataIsNull_s(pValue, i)) {
622,148✔
1180
        isNull[j] = 1;
520✔
1181
      } else {
1182
        isNull[j] = 0;
310,554✔
1183
        char* data = colDataGetData(pValue, i);
310,554!
1184
        if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
310,554✔
1185
          if (tTagIsJson(data)) {
264✔
1186
            code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
12✔
1187
            goto end;
12✔
1188
          }
1189
          if (tTagIsJsonNull(data)) {
252!
1190
            isNull[j] = 1;
×
1191
            continue;
×
1192
          }
1193
          int32_t len = getJsonValueLen(data);
252✔
1194
          memcpy(pStart, data, len);
252✔
1195
          pStart += len;
252✔
1196
        } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
310,290!
1197
          if (IS_STR_DATA_BLOB(pValue->info.type)) {
296,990!
1198
            if (blobDataTLen(data) > TSDB_MAX_BLOB_LEN) {
54!
1199
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1200
              goto end;
×
1201
            }
1202
            memcpy(pStart, data, blobDataTLen(data));
54✔
1203
            pStart += blobDataTLen(data);
54✔
1204
          } else {
1205
            if (varDataTLen(data) > pValue->info.bytes) {
296,936✔
1206
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
3✔
1207
              goto end;
3✔
1208
            }
1209
            memcpy(pStart, data, varDataTLen(data));
296,933✔
1210
            pStart += varDataTLen(data);
296,933✔
1211
          }
1212
        } else {
1213
          memcpy(pStart, data, pValue->info.bytes);
13,300✔
1214
          pStart += pValue->info.bytes;
13,300✔
1215
        }
1216
      }
1217
    }
1218

1219
    int32_t len = (int32_t)(pStart - (char*)keyBuf);
296,731✔
1220
    info->groupId = calcGroupId(keyBuf, len);
296,731✔
1221
    if (groupIdMap != NULL && gInfo != NULL) {
298,958!
1222
      int32_t ret = taosHashPut(groupIdMap, &info->groupId, sizeof(info->groupId), &gInfo, POINTER_BYTES);
956✔
1223
      if (ret != TSDB_CODE_SUCCESS) {
956!
1224
        qError("put groupid to map failed at line %d since %s", __LINE__, tstrerror(ret));
×
1225
        taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1226
      }
1227
      qDebug("put groupid to map gid:%" PRIu64, info->groupId);
956!
1228
      gInfo = NULL;
956✔
1229
    }
1230
    if (initRemainGroups) {
298,958✔
1231
      // groupId ~ table uid
1232
      code = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId), &(info->uid),
80,080✔
1233
                         sizeof(info->uid));
1234
      if (code == TSDB_CODE_DUP_KEY) {
80,088✔
1235
        code = TSDB_CODE_SUCCESS;
1,227✔
1236
      }
1237
      QUERY_CHECK_CODE(code, lino, end);
80,088!
1238
    }
1239
  }
1240

1241
  if (tsTagFilterCache) {
80,604!
1242
    tableList = taosArrayDup(pTableListInfo->pTableList, NULL);
×
1243
    QUERY_CHECK_NULL(tableList, code, lino, end, terrno);
×
1244

1245
    code = pAPI->metaFn.metaPutTbGroupToCache(pVnode, pTableListInfo->idInfo.suid, context.digest,
×
1246
                                              tListLen(context.digest), tableList,
1247
                                              taosArrayGetSize(tableList) * sizeof(STableKeyInfo));
×
1248
    QUERY_CHECK_CODE(code, lino, end);
×
1249
  }
1250

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

1254
end:
80,604✔
1255
  taosMemoryFreeClear(keyBuf);
80,619!
1256
  blockDataDestroy(pResBlock);
80,598✔
1257
  taosArrayDestroy(pBlockList);
80,519✔
1258
  taosArrayDestroyEx(pUidTagList, freeItem);
80,506✔
1259
  taosArrayDestroyP(groupData, releaseColInfoData);
80,498✔
1260
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
80,486✔
1261

1262
  if (code != TSDB_CODE_SUCCESS) {
80,497✔
1263
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
15!
1264
  }
1265
  return code;
80,494✔
1266
}
1267

1268
static int32_t nameComparFn(const void* p1, const void* p2) {
×
1269
  const char* pName1 = *(const char**)p1;
×
1270
  const char* pName2 = *(const char**)p2;
×
1271

1272
  int32_t ret = strcmp(pName1, pName2);
×
1273
  if (ret == 0) {
×
1274
    return 0;
×
1275
  } else {
1276
    return (ret > 0) ? 1 : -1;
×
1277
  }
1278
}
1279

1280
static SArray* getTableNameList(const SNodeListNode* pList) {
×
1281
  int32_t    code = TSDB_CODE_SUCCESS;
×
1282
  int32_t    lino = 0;
×
1283
  int32_t    len = LIST_LENGTH(pList->pNodeList);
×
1284
  SListCell* cell = pList->pNodeList->pHead;
×
1285

1286
  SArray* pTbList = taosArrayInit(len, POINTER_BYTES);
×
1287
  QUERY_CHECK_NULL(pTbList, code, lino, _end, terrno);
×
1288

1289
  for (int i = 0; i < pList->pNodeList->length; i++) {
×
1290
    SValueNode* valueNode = (SValueNode*)cell->pNode;
×
1291
    if (!IS_VAR_DATA_TYPE(valueNode->node.resType.type)) {
×
1292
      terrno = TSDB_CODE_INVALID_PARA;
×
1293
      taosArrayDestroy(pTbList);
×
1294
      return NULL;
×
1295
    }
1296

1297
    char* name = varDataVal(valueNode->datum.p);
×
1298
    void* tmp = taosArrayPush(pTbList, &name);
×
1299
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
1300
    cell = cell->pNext;
×
1301
  }
1302

1303
  size_t numOfTables = taosArrayGetSize(pTbList);
×
1304

1305
  // order the name
1306
  taosArraySort(pTbList, nameComparFn);
×
1307

1308
  // remove the duplicates
1309
  SArray* pNewList = taosArrayInit(taosArrayGetSize(pTbList), sizeof(void*));
×
1310
  QUERY_CHECK_NULL(pNewList, code, lino, _end, terrno);
×
1311
  void* tmpTbl = taosArrayGet(pTbList, 0);
×
1312
  QUERY_CHECK_NULL(tmpTbl, code, lino, _end, terrno);
×
1313
  void* tmp = taosArrayPush(pNewList, tmpTbl);
×
1314
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
1315

1316
  for (int32_t i = 1; i < numOfTables; ++i) {
×
1317
    char** name = taosArrayGetLast(pNewList);
×
1318
    char** nameInOldList = taosArrayGet(pTbList, i);
×
1319
    QUERY_CHECK_NULL(nameInOldList, code, lino, _end, terrno);
×
1320
    if (strcmp(*name, *nameInOldList) == 0) {
×
1321
      continue;
×
1322
    }
1323

1324
    tmp = taosArrayPush(pNewList, nameInOldList);
×
1325
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
1326
  }
1327

1328
_end:
×
1329
  taosArrayDestroy(pTbList);
×
1330
  if (code != TSDB_CODE_SUCCESS) {
×
1331
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1332
    return NULL;
×
1333
  }
1334
  return pNewList;
×
1335
}
1336

1337
static int tableUidCompare(const void* a, const void* b) {
×
1338
  uint64_t u1 = *(uint64_t*)a;
×
1339
  uint64_t u2 = *(uint64_t*)b;
×
1340

1341
  if (u1 == u2) {
×
1342
    return 0;
×
1343
  }
1344

1345
  return u1 < u2 ? -1 : 1;
×
1346
}
1347

1348
static int32_t filterTableInfoCompare(const void* a, const void* b) {
58,863✔
1349
  STUidTagInfo* p1 = (STUidTagInfo*)a;
58,863✔
1350
  STUidTagInfo* p2 = (STUidTagInfo*)b;
58,863✔
1351

1352
  if (p1->uid == p2->uid) {
58,863!
1353
    return 0;
×
1354
  }
1355

1356
  return p1->uid < p2->uid ? -1 : 1;
58,863!
1357
}
1358

1359
static FilterCondType checkTagCond(SNode* cond) {
2,517,570✔
1360
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
2,517,570✔
1361
    return FILTER_NO_LOGIC;
591,164✔
1362
  }
1363
  if (nodeType(cond) != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
1,926,406!
1364
    return FILTER_AND;
94,444✔
1365
  }
1366
  return FILTER_OTHER;
1,831,962✔
1367
}
1368

1369
static int32_t optimizeTbnameInCond(void* pVnode, int64_t suid, SArray* list, SNode* cond, SStorageAPI* pAPI) {
2,518,255✔
1370
  int32_t ret = -1;
2,518,255✔
1371
  int32_t ntype = nodeType(cond);
2,518,255✔
1372

1373
  if (ntype == QUERY_NODE_OPERATOR) {
2,518,255✔
1374
    ret = optimizeTbnameInCondImpl(pVnode, list, cond, pAPI, suid);
591,201✔
1375
  }
1376

1377
  if (ntype != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
2,520,005✔
1378
    return ret;
688,134✔
1379
  }
1380

1381
  bool                 hasTbnameCond = false;
1,831,871✔
1382
  SLogicConditionNode* pNode = (SLogicConditionNode*)cond;
1,831,871✔
1383
  SNodeList*           pList = (SNodeList*)pNode->pParameterList;
1,831,871✔
1384

1385
  int32_t len = LIST_LENGTH(pList);
1,831,871!
1386
  if (len <= 0) {
1,831,871!
1387
    return ret;
×
1388
  }
1389

1390
  SListCell* cell = pList->pHead;
1,831,871✔
1391
  for (int i = 0; i < len; i++) {
8,016,550✔
1392
    if (cell == NULL) break;
6,185,246!
1393
    if (optimizeTbnameInCondImpl(pVnode, list, cell->pNode, pAPI, suid) == 0) {
6,185,246!
1394
      hasTbnameCond = true;
×
1395
      break;
×
1396
    }
1397
    cell = cell->pNext;
6,184,679✔
1398
  }
1399

1400
  taosArraySort(list, filterTableInfoCompare);
1,831,304✔
1401
  taosArrayRemoveDuplicate(list, filterTableInfoCompare, NULL);
1,828,510✔
1402

1403
  if (hasTbnameCond) {
1,829,497!
1404
    ret = pAPI->metaFn.getTableTagsByUid(pVnode, suid, list);
×
1405
  }
1406

1407
  return ret;
1,829,636✔
1408
}
1409

1410
// only return uid that does not contained in pExistedUidList
1411
static int32_t optimizeTbnameInCondImpl(void* pVnode, SArray* pExistedUidList, SNode* pTagCond, SStorageAPI* pStoreAPI,
6,774,245✔
1412
                                        uint64_t suid) {
1413
  if (nodeType(pTagCond) != QUERY_NODE_OPERATOR) {
6,774,245✔
1414
    return -1;
537,371✔
1415
  }
1416

1417
  SOperatorNode* pNode = (SOperatorNode*)pTagCond;
6,236,874✔
1418
  if (pNode->opType != OP_TYPE_IN) {
6,236,874✔
1419
    return -1;
4,547,025✔
1420
  }
1421

1422
  if ((pNode->pLeft != NULL && nodeType(pNode->pLeft) == QUERY_NODE_COLUMN &&
1,689,849!
1423
       ((SColumnNode*)pNode->pLeft)->colType == COLUMN_TYPE_TBNAME) &&
864,136!
1424
      (pNode->pRight != NULL && nodeType(pNode->pRight) == QUERY_NODE_NODE_LIST)) {
×
1425
    SNodeListNode* pList = (SNodeListNode*)pNode->pRight;
×
1426

1427
    int32_t len = LIST_LENGTH(pList->pNodeList);
×
1428
    if (len <= 0) {
×
1429
      return -1;
×
1430
    }
1431

1432
    SArray*   pTbList = getTableNameList(pList);
×
1433
    int32_t   numOfTables = taosArrayGetSize(pTbList);
×
1434
    SHashObj* uHash = NULL;
×
1435

1436
    size_t numOfExisted = taosArrayGetSize(pExistedUidList);  // len > 0 means there already have uids
×
1437
    if (numOfExisted > 0) {
×
1438
      uHash = taosHashInit(numOfExisted / 0.7, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
×
1439
      if (!uHash) {
×
1440
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1441
        return terrno;
×
1442
      }
1443

1444
      for (int i = 0; i < numOfExisted; i++) {
×
1445
        STUidTagInfo* pTInfo = taosArrayGet(pExistedUidList, i);
×
1446
        if (!pTInfo) {
×
1447
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1448
          return terrno;
×
1449
        }
1450
        int32_t tempRes = taosHashPut(uHash, &pTInfo->uid, sizeof(uint64_t), &i, sizeof(i));
×
1451
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
×
1452
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
1453
          return tempRes;
×
1454
        }
1455
      }
1456
    }
1457

1458
    for (int i = 0; i < numOfTables; i++) {
×
1459
      char* name = taosArrayGetP(pTbList, i);
×
1460

1461
      uint64_t uid = 0, csuid = 0;
×
1462
      if (pStoreAPI->metaFn.getTableUidByName(pVnode, name, &uid) == 0) {
×
1463
        ETableType tbType = TSDB_TABLE_MAX;
×
1464
        if (pStoreAPI->metaFn.getTableTypeSuidByName(pVnode, name, &tbType, &csuid) == 0 &&
×
1465
            tbType == TSDB_CHILD_TABLE) {
×
1466
          if (suid != csuid) {
×
1467
            continue;
×
1468
          }
1469
          if (NULL == uHash || taosHashGet(uHash, &uid, sizeof(uid)) == NULL) {
×
1470
            STUidTagInfo s = {.uid = uid, .name = name, .pTagVal = NULL};
×
1471
            void*        tmp = taosArrayPush(pExistedUidList, &s);
×
1472
            if (!tmp) {
×
1473
              return terrno;
×
1474
            }
1475
          }
1476
        } else {
1477
          taosArrayDestroy(pTbList);
×
1478
          taosHashCleanup(uHash);
×
1479
          return -1;
×
1480
        }
1481
      } else {
1482
        //        qWarn("failed to get tableIds from by table name: %s, reason: %s", name, tstrerror(terrno));
1483
        terrno = 0;
×
1484
      }
1485
    }
1486

1487
    taosHashCleanup(uHash);
×
1488
    taosArrayDestroy(pTbList);
×
1489
    return 0;
×
1490
  }
1491

1492
  return -1;
1,689,849✔
1493
}
1494

1495
SSDataBlock* createTagValBlockForFilter(SArray* pColList, int32_t numOfTables, SArray* pUidTagList, void* pVnode,
2,547,161✔
1496
                                        SStorageAPI* pStorageAPI) {
1497
  int32_t      code = TSDB_CODE_SUCCESS;
2,547,161✔
1498
  int32_t      lino = 0;
2,547,161✔
1499
  SSDataBlock* pResBlock = NULL;
2,547,161✔
1500
  code = createDataBlock(&pResBlock);
2,547,161✔
1501
  QUERY_CHECK_CODE(code, lino, _end);
2,549,988!
1502

1503
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
8,552,656✔
1504
    SColumnInfoData colInfo = {0};
6,002,391✔
1505
    void*           tmp = taosArrayGet(pColList, i);
6,002,391✔
1506
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
5,989,449!
1507
    colInfo.info = *(SColumnInfo*)tmp;
5,989,449✔
1508
    code = blockDataAppendColInfo(pResBlock, &colInfo);
5,989,449✔
1509
    QUERY_CHECK_CODE(code, lino, _end);
6,002,668!
1510
  }
1511

1512
  code = blockDataEnsureCapacity(pResBlock, numOfTables);
2,539,811✔
1513
  if (code != TSDB_CODE_SUCCESS) {
2,551,306!
1514
    terrno = code;
×
1515
    blockDataDestroy(pResBlock);
×
1516
    return NULL;
×
1517
  }
1518

1519
  pResBlock->info.rows = numOfTables;
2,551,306✔
1520

1521
  int32_t numOfCols = taosArrayGetSize(pResBlock->pDataBlock);
2,551,306✔
1522

1523
  for (int32_t i = 0; i < numOfTables; i++) {
10,714,894✔
1524
    STUidTagInfo* p1 = taosArrayGet(pUidTagList, i);
8,134,788✔
1525
    QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
8,118,693✔
1526

1527
    for (int32_t j = 0; j < numOfCols; j++) {
26,499,062✔
1528
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, j);
18,333,345✔
1529
      QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
18,280,844!
1530

1531
      if (pColInfo->info.colId == -1) {  // tbname
18,288,202✔
1532
        char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
3,102,036✔
1533
        if (p1->name != NULL) {
3,102,036!
1534
          STR_TO_VARSTR(str, p1->name);
×
1535
        } else {  // name is not retrieved during filter
1536
          code = pStorageAPI->metaFn.getTableNameByUid(pVnode, p1->uid, str);
3,102,036✔
1537
          QUERY_CHECK_CODE(code, lino, _end);
3,103,724!
1538
        }
1539

1540
        code = colDataSetVal(pColInfo, i, str, false);
3,103,724✔
1541
        QUERY_CHECK_CODE(code, lino, _end);
3,102,818!
1542
#if TAG_FILTER_DEBUG
1543
        qDebug("tagfilter uid:%ld, tbname:%s", *uid, str + 2);
1544
#endif
1545
      } else {
1546
        STagVal tagVal = {0};
15,186,166✔
1547
        tagVal.cid = pColInfo->info.colId;
15,186,166✔
1548
        if (p1->pTagVal == NULL) {
15,186,166!
1549
          colDataSetNULL(pColInfo, i);
×
1550
        } else {
1551
          const char* p = pStorageAPI->metaFn.extractTagVal(p1->pTagVal, pColInfo->info.type, &tagVal);
15,186,166✔
1552

1553
          if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
15,377,829✔
1554
            colDataSetNULL(pColInfo, i);
13,065✔
1555
          } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
15,364,764✔
1556
            code = colDataSetVal(pColInfo, i, p, false);
4,862✔
1557
            QUERY_CHECK_CODE(code, lino, _end);
6,730!
1558
          } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
18,771,525!
1559
            if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
3,417,181✔
1560
              QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
1,868✔
1561
            }
1562
            char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
3,415,313!
1563
            QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
3,413,043!
1564
            varDataSetLen(tmp, tagVal.nData);
3,413,043✔
1565
            memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
3,413,043✔
1566
            code = colDataSetVal(pColInfo, i, tmp, false);
3,413,043✔
1567
#if TAG_FILTER_DEBUG
1568
            qDebug("tagfilter varch:%s", tmp + 2);
1569
#endif
1570
            taosMemoryFree(tmp);
3,405,354!
1571
            QUERY_CHECK_CODE(code, lino, _end);
3,411,623!
1572
          } else {
1573
            code = colDataSetVal(pColInfo, i, (const char*)&tagVal.i64, false);
11,942,721✔
1574
            QUERY_CHECK_CODE(code, lino, _end);
11,848,377!
1575
#if TAG_FILTER_DEBUG
1576
            if (pColInfo->info.type == TSDB_DATA_TYPE_INT) {
1577
              qDebug("tagfilter int:%d", *(int*)(&tagVal.i64));
1578
            } else if (pColInfo->info.type == TSDB_DATA_TYPE_DOUBLE) {
1579
              qDebug("tagfilter double:%f", *(double*)(&tagVal.i64));
1580
            }
1581
#endif
1582
          }
1583
        }
1584
      }
1585
    }
1586
  }
1587

1588
_end:
2,580,106✔
1589
  if (code != TSDB_CODE_SUCCESS) {
2,581,974!
1590
    blockDataDestroy(pResBlock);
×
1591
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1592
    terrno = code;
×
1593
    return NULL;
×
1594
  }
1595
  return pResBlock;
2,581,974✔
1596
}
1597

1598
static int32_t doSetQualifiedUid(STableListInfo* pListInfo, SArray* pUidList, const SArray* pUidTagList,
2,425,226✔
1599
                                 bool* pResultList, bool addUid) {
1600
  taosArrayClear(pUidList);
2,425,226✔
1601

1602
  STableKeyInfo info = {.uid = 0, .groupId = 0};
2,425,198✔
1603
  int32_t       numOfTables = taosArrayGetSize(pUidTagList);
2,425,198✔
1604
  for (int32_t i = 0; i < numOfTables; ++i) {
10,068,839✔
1605
    if (pResultList[i]) {
7,642,428✔
1606
      STUidTagInfo* tmpTag = (STUidTagInfo*)taosArrayGet(pUidTagList, i);
4,418,151✔
1607
      if (!tmpTag) {
4,410,129!
1608
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1609
        return terrno;
×
1610
      }
1611
      uint64_t uid = tmpTag->uid;
4,410,129✔
1612
      qDebug("tagfilter get uid:%" PRId64 ", res:%d", uid, pResultList[i]);
4,410,129✔
1613

1614
      info.uid = uid;
4,410,235✔
1615
      void* p = taosArrayPush(pListInfo->pTableList, &info);
4,410,235✔
1616
      if (p == NULL) {
4,417,521!
1617
        return terrno;
×
1618
      }
1619

1620
      if (addUid) {
4,417,521!
1621
        void* tmp = taosArrayPush(pUidList, &uid);
×
1622
        if (tmp == NULL) {
×
1623
          return terrno;
×
1624
        }
1625
      }
1626
    }
1627
  }
1628

1629
  return TSDB_CODE_SUCCESS;
2,426,411✔
1630
}
1631

1632
static int32_t copyExistedUids(SArray* pUidTagList, const SArray* pUidList) {
2,518,114✔
1633
  int32_t code = TSDB_CODE_SUCCESS;
2,518,114✔
1634
  int32_t numOfExisted = taosArrayGetSize(pUidList);
2,518,114✔
1635
  if (numOfExisted == 0) {
2,518,641✔
1636
    return code;
2,451,976✔
1637
  }
1638

1639
  for (int32_t i = 0; i < numOfExisted; ++i) {
214,801✔
1640
    uint64_t* uid = taosArrayGet(pUidList, i);
148,120✔
1641
    if (!uid) {
148,056!
1642
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1643
      return terrno;
×
1644
    }
1645
    STUidTagInfo info = {.uid = *uid};
148,056✔
1646
    void*        tmp = taosArrayPush(pUidTagList, &info);
148,136✔
1647
    if (!tmp) {
148,136!
1648
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1649
      return code;
×
1650
    }
1651
  }
1652
  return code;
66,681✔
1653
}
1654

1655
static int32_t doFilterByTagCond(STableListInfo* pListInfo, SArray* pUidList, SNode* pTagCond, void* pVnode,
8,555,771✔
1656
                                 SIdxFltStatus status, SStorageAPI* pAPI, bool addUid, bool* listAdded) {
1657
  *listAdded = false;
8,555,771✔
1658
  if (pTagCond == NULL) {
8,555,771✔
1659
    return TSDB_CODE_SUCCESS;
6,040,797✔
1660
  }
1661

1662
  terrno = TSDB_CODE_SUCCESS;
2,514,974✔
1663

1664
  int32_t      lino = 0;
2,518,969✔
1665
  int32_t      code = TSDB_CODE_SUCCESS;
2,518,969✔
1666
  SArray*      pBlockList = NULL;
2,518,969✔
1667
  SSDataBlock* pResBlock = NULL;
2,518,969✔
1668
  SScalarParam output = {0};
2,518,969✔
1669
  SArray*      pUidTagList = NULL;
2,518,969✔
1670

1671
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
2,518,969✔
1672

1673
  //  int64_t stt = taosGetTimestampUs();
1674
  pUidTagList = taosArrayInit(10, sizeof(STUidTagInfo));
2,518,969✔
1675
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
2,518,873!
1676

1677
  code = copyExistedUids(pUidTagList, pUidList);
2,518,873✔
1678
  QUERY_CHECK_CODE(code, lino, end);
2,518,997!
1679

1680
  FilterCondType condType = checkTagCond(pTagCond);
2,518,997✔
1681

1682
  int32_t filter = optimizeTbnameInCond(pVnode, pListInfo->idInfo.suid, pUidTagList, pTagCond, pAPI);
2,519,837✔
1683
  if (filter == 0) {  // tbname in filter is activated, do nothing and return
2,516,936!
1684
    taosArrayClear(pUidList);
×
1685

1686
    int32_t numOfRows = taosArrayGetSize(pUidTagList);
×
1687
    code = taosArrayEnsureCap(pUidList, numOfRows);
×
1688
    QUERY_CHECK_CODE(code, lino, end);
×
1689

1690
    for (int32_t i = 0; i < numOfRows; ++i) {
×
1691
      STUidTagInfo* pInfo = taosArrayGet(pUidTagList, i);
×
1692
      QUERY_CHECK_NULL(pInfo, code, lino, end, terrno);
×
1693
      void* tmp = taosArrayPush(pUidList, &pInfo->uid);
×
1694
      QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
×
1695
    }
1696
    terrno = 0;
×
1697
  } else {
1698
    if ((condType == FILTER_NO_LOGIC || condType == FILTER_AND) && status != SFLT_NOT_INDEX) {
2,516,936✔
1699
      code = pAPI->metaFn.getTableTagsByUid(pVnode, pListInfo->idInfo.suid, pUidTagList);
6,647✔
1700
    } else {
1701
      code = pAPI->metaFn.getTableTags(pVnode, pListInfo->idInfo.suid, pUidTagList);
2,510,289✔
1702
    }
1703
    if (code != TSDB_CODE_SUCCESS) {
2,521,478!
1704
      qError("failed to get table tags from meta, reason:%s, suid:%" PRIu64, tstrerror(code), pListInfo->idInfo.suid);
×
1705
      terrno = code;
×
1706
      QUERY_CHECK_CODE(code, lino, end);
×
1707
    }
1708
  }
1709

1710
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
2,521,478✔
1711
  if (numOfTables == 0) {
2,519,902✔
1712
    goto end;
93,825✔
1713
  }
1714

1715
  SArray* pColList = NULL;
2,426,077✔
1716
  code = qGetColumnsFromNodeList(pTagCond, false, &pColList); 
2,426,077✔
1717
  if (code != TSDB_CODE_SUCCESS) {
2,426,189!
1718
    goto end;
×
1719
  }
1720
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
2,426,189✔
1721
  taosArrayDestroy(pColList);
2,425,012✔
1722
  if (pResBlock == NULL) {
2,427,553!
1723
    code = terrno;
×
1724
    QUERY_CHECK_CODE(code, lino, end);
×
1725
  }
1726

1727
  //  int64_t st1 = taosGetTimestampUs();
1728
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1729
  pBlockList = taosArrayInit(2, POINTER_BYTES);
2,427,553✔
1730
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
2,427,353!
1731

1732
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
2,426,423✔
1733
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
2,426,423!
1734

1735
  code = createResultData(&type, numOfTables, &output);
2,426,423✔
1736
  if (code != TSDB_CODE_SUCCESS) {
2,427,715!
1737
    terrno = code;
×
1738
    QUERY_CHECK_CODE(code, lino, end);
×
1739
  }
1740

1741
  code = scalarCalculate(pTagCond, pBlockList, &output, NULL, NULL);
2,427,715✔
1742
  if (code != TSDB_CODE_SUCCESS) {
2,426,586✔
1743
    qError("failed to calculate scalar, reason:%s", tstrerror(code));
8!
1744
    terrno = code;
8✔
1745
    QUERY_CHECK_CODE(code, lino, end);
8!
1746
  }
1747

1748
  code = doSetQualifiedUid(pListInfo, pUidList, pUidTagList, (bool*)output.columnData->pData, addUid);
2,426,578✔
1749
  if (code != TSDB_CODE_SUCCESS) {
2,426,286✔
1750
    terrno = code;
14✔
1751
    QUERY_CHECK_CODE(code, lino, end);
×
1752
  }
1753
  *listAdded = true;
2,426,272✔
1754

1755
end:
2,520,105✔
1756
  if (code != TSDB_CODE_SUCCESS) {
2,520,105✔
1757
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
8!
1758
  }
1759
  blockDataDestroy(pResBlock);
2,520,105✔
1760
  taosArrayDestroy(pBlockList);
2,521,432✔
1761
  taosArrayDestroyEx(pUidTagList, freeItem);
2,521,747✔
1762

1763
  colDataDestroy(output.columnData);
2,521,833✔
1764
  taosMemoryFreeClear(output.columnData);
2,521,842!
1765
  return code;
2,521,889✔
1766
}
1767

1768
int32_t getTableList(void* pVnode, SScanPhysiNode* pScanNode, SNode* pTagCond, SNode* pTagIndexCond,
8,550,821✔
1769
                     STableListInfo* pListInfo, uint8_t* digest, const char* idstr, SStorageAPI* pStorageAPI) {
1770
  int32_t code = TSDB_CODE_SUCCESS;
8,550,821✔
1771
  int32_t lino = 0;
8,550,821✔
1772
  size_t  numOfTables = 0;
8,550,821✔
1773
  bool    listAdded = false;
8,550,821✔
1774

1775
  pListInfo->idInfo.suid = pScanNode->suid;
8,550,821✔
1776
  pListInfo->idInfo.tableType = pScanNode->tableType;
8,550,821✔
1777

1778
  SArray* pUidList = taosArrayInit(8, sizeof(uint64_t));
8,550,821✔
1779
  QUERY_CHECK_NULL(pUidList, code, lino, _error, terrno);
8,563,302✔
1780

1781
  SIdxFltStatus status = SFLT_NOT_INDEX;
8,556,129✔
1782
  if (pScanNode->tableType != TSDB_SUPER_TABLE) {
8,556,129✔
1783
    pListInfo->idInfo.uid = pScanNode->uid;
1,091,654✔
1784
    if (pStorageAPI->metaFn.isTableExisted(pVnode, pScanNode->uid)) {
1,091,654✔
1785
      void* tmp = taosArrayPush(pUidList, &pScanNode->uid);
1,091,647✔
1786
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
1,091,636!
1787
    }
1788
    code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, status, pStorageAPI, false, &listAdded);
1,091,659✔
1789
    QUERY_CHECK_CODE(code, lino, _end);
1,091,646!
1790
  } else {
1791
    T_MD5_CTX context = {0};
7,464,475✔
1792

1793
    if (tsTagFilterCache) {
7,464,475!
1794
      // try to retrieve the result from meta cache
1795
      code = genTagFilterDigest(pTagCond, &context);
×
1796
      QUERY_CHECK_CODE(code, lino, _error);
×
1797

1798
      bool acquired = false;
×
1799
      code = pStorageAPI->metaFn.getCachedTableList(pVnode, pScanNode->suid, context.digest, tListLen(context.digest),
×
1800
                                                    pUidList, &acquired);
1801
      QUERY_CHECK_CODE(code, lino, _error);
×
1802

1803
      if (acquired) {
×
1804
        digest[0] = 1;
×
1805
        memcpy(digest + 1, context.digest, tListLen(context.digest));
×
1806
        qDebug("retrieve table uid list from cache, numOfTables:%d", (int32_t)taosArrayGetSize(pUidList));
×
1807
        goto _end;
×
1808
      }
1809
    }
1810

1811
    if (!pTagCond) {  // no tag filter condition exists, let's fetch all tables of this super table
7,464,475✔
1812
      code = pStorageAPI->metaFn.getChildTableList(pVnode, pScanNode->suid, pUidList);
4,997,358✔
1813
      QUERY_CHECK_CODE(code, lino, _error);
5,007,120!
1814
    } else {
1815
      // failed to find the result in the cache, let try to calculate the results
1816
      if (pTagIndexCond) {
2,467,117✔
1817
        void* pIndex = pStorageAPI->metaFn.getInvertIndex(pVnode);
97,860✔
1818

1819
        SIndexMetaArg metaArg = {.metaEx = pVnode,
195,801✔
1820
                                 .idx = pStorageAPI->metaFn.storeGetIndexInfo(pVnode),
97,917✔
1821
                                 .ivtIdx = pIndex,
1822
                                 .suid = pScanNode->uid};
97,884✔
1823

1824
        status = SFLT_NOT_INDEX;
97,884✔
1825
        code = doFilterTag(pTagIndexCond, &metaArg, pUidList, &status, &pStorageAPI->metaFilter);
97,884✔
1826
        if (code != 0 || status == SFLT_NOT_INDEX) {  // temporarily disable it for performance sake
97,801✔
1827
          qDebug("failed to get tableIds from index, suid:%" PRIu64, pScanNode->uid);
84,214✔
1828
        } else {
1829
          qDebug("succ to get filter result, table num: %d", (int)taosArrayGetSize(pUidList));
13,587✔
1830
        }
1831
      }
1832
    }
1833

1834
    code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, status, pStorageAPI, tsTagFilterCache, &listAdded);
7,474,208✔
1835
    QUERY_CHECK_CODE(code, lino, _end);
7,469,680✔
1836

1837
    // let's add the filter results into meta-cache
1838
    numOfTables = taosArrayGetSize(pUidList);
7,469,672✔
1839

1840
    if (tsTagFilterCache) {
7,466,242✔
1841
      size_t size = numOfTables * sizeof(uint64_t) + sizeof(int32_t);
326✔
1842
      char*  pPayload = taosMemoryMalloc(size);
326!
1843
      QUERY_CHECK_NULL(pPayload, code, lino, _end, terrno);
×
1844

1845
      *(int32_t*)pPayload = numOfTables;
×
1846
      if (numOfTables > 0) {
×
1847
        void* tmp = taosArrayGet(pUidList, 0);
×
1848
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
1849
        memcpy(pPayload + sizeof(int32_t), tmp, numOfTables * sizeof(uint64_t));
×
1850
      }
1851

1852
      code = pStorageAPI->metaFn.putCachedTableList(pVnode, pScanNode->suid, context.digest, tListLen(context.digest),
×
1853
                                                    pPayload, size, 1);
1854
      QUERY_CHECK_CODE(code, lino, _error);
×
1855

1856
      digest[0] = 1;
×
1857
      memcpy(digest + 1, context.digest, tListLen(context.digest));
×
1858
    }
1859
  }
1860

1861
_end:
8,557,570✔
1862
  if (!listAdded) {
8,557,570✔
1863
    numOfTables = taosArrayGetSize(pUidList);
6,131,711✔
1864
    for (int i = 0; i < numOfTables; i++) {
26,283,294✔
1865
      void* tmp = taosArrayGet(pUidList, i);
20,155,293✔
1866
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
20,119,685!
1867
      STableKeyInfo info = {.uid = *(uint64_t*)tmp, .groupId = 0};
20,119,685✔
1868

1869
      void* p = taosArrayPush(pListInfo->pTableList, &info);
20,119,685✔
1870
      if (p == NULL) {
20,151,825!
1871
        taosArrayDestroy(pUidList);
×
1872
        return terrno;
×
1873
      }
1874

1875
      qTrace("tagfilter get uid:%" PRIu64 ", %s", info.uid, idstr);
20,151,825✔
1876
    }
1877
  }
1878

1879
_error:
8,553,860✔
1880
  taosArrayDestroy(pUidList);
8,553,860✔
1881
  if (code != TSDB_CODE_SUCCESS) {
8,565,024✔
1882
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
8!
1883
  }
1884
  return code;
8,564,230✔
1885
}
1886

1887
int32_t qGetTableList(int64_t suid, void* pVnode, void* node, SArray** tableList, void* pTaskInfo) {
75✔
1888
  int32_t        code = TSDB_CODE_SUCCESS;
75✔
1889
  int32_t        lino = 0;
75✔
1890
  SSubplan*      pSubplan = (SSubplan*)node;
75✔
1891
  SScanPhysiNode pNode = {0};
75✔
1892
  pNode.suid = suid;
75✔
1893
  pNode.uid = suid;
75✔
1894
  pNode.tableType = TSDB_SUPER_TABLE;
75✔
1895

1896
  STableListInfo* pTableListInfo = tableListCreate();
75✔
1897
  QUERY_CHECK_NULL(pTableListInfo, code, lino, _end, terrno);
75!
1898
  uint8_t digest[17] = {0};
75✔
1899
  code = getTableList(pVnode, &pNode, pSubplan ? pSubplan->pTagCond : NULL, pSubplan ? pSubplan->pTagIndexCond : NULL,
75✔
1900
                      pTableListInfo, digest, "qGetTableList", &((SExecTaskInfo*)pTaskInfo)->storageAPI);
1901
  QUERY_CHECK_CODE(code, lino, _end);
75!
1902
  *tableList = pTableListInfo->pTableList;
75✔
1903
  pTableListInfo->pTableList = NULL;
75✔
1904
  tableListDestroy(pTableListInfo);
75✔
1905

1906
_end:
75✔
1907
  if (code != TSDB_CODE_SUCCESS) {
75!
1908
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1909
  }
1910
  return code;
75✔
1911
}
1912

1913
size_t getTableTagsBufLen(const SNodeList* pGroups) {
×
1914
  size_t keyLen = 0;
×
1915

1916
  SNode* node;
1917
  FOREACH(node, pGroups) {
×
1918
    SExprNode* pExpr = (SExprNode*)node;
×
1919
    keyLen += pExpr->resType.bytes;
×
1920
  }
1921

1922
  keyLen += sizeof(int8_t) * LIST_LENGTH(pGroups);
×
1923
  return keyLen;
×
1924
}
1925

1926
int32_t getGroupIdFromTagsVal(void* pVnode, uint64_t uid, SNodeList* pGroupNode, char* keyBuf, uint64_t* pGroupId,
×
1927
                              SStorageAPI* pAPI) {
1928
  SMetaReader mr = {0};
×
1929

1930
  pAPI->metaReaderFn.initReader(&mr, pVnode, META_READER_LOCK, &pAPI->metaFn);
×
1931
  if (pAPI->metaReaderFn.getEntryGetUidCache(&mr, uid) != 0) {  // table not exist
×
1932
    pAPI->metaReaderFn.clearReader(&mr);
×
1933
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
1934
  }
1935

1936
  SNodeList* groupNew = NULL;
×
1937
  int32_t    code = nodesCloneList(pGroupNode, &groupNew);
×
1938
  if (TSDB_CODE_SUCCESS != code) {
×
1939
    pAPI->metaReaderFn.clearReader(&mr);
×
1940
    return code;
×
1941
  }
1942

1943
  STransTagExprCtx ctx = {.code = 0, .pReader = &mr};
×
1944
  nodesRewriteExprsPostOrder(groupNew, doTranslateTagExpr, &ctx);
×
1945
  if (TSDB_CODE_SUCCESS != ctx.code) {
×
1946
    nodesDestroyList(groupNew);
×
1947
    pAPI->metaReaderFn.clearReader(&mr);
×
1948
    return code;
×
1949
  }
1950
  char* isNull = (char*)keyBuf;
×
1951
  char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(pGroupNode);
×
1952

1953
  SNode*  pNode;
1954
  int32_t index = 0;
×
1955
  FOREACH(pNode, groupNew) {
×
1956
    SNode*  pNew = NULL;
×
1957
    int32_t code = scalarCalculateConstants(pNode, &pNew);
×
1958
    if (TSDB_CODE_SUCCESS == code) {
×
1959
      REPLACE_NODE(pNew);
×
1960
    } else {
1961
      nodesDestroyList(groupNew);
×
1962
      pAPI->metaReaderFn.clearReader(&mr);
×
1963
      return code;
×
1964
    }
1965

1966
    if (nodeType(pNew) != QUERY_NODE_VALUE) {
×
1967
      nodesDestroyList(groupNew);
×
1968
      pAPI->metaReaderFn.clearReader(&mr);
×
1969
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
1970
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1971
    }
1972
    SValueNode* pValue = (SValueNode*)pNew;
×
1973

1974
    if (pValue->node.resType.type == TSDB_DATA_TYPE_NULL || pValue->isNull) {
×
1975
      isNull[index++] = 1;
×
1976
      continue;
×
1977
    } else {
1978
      isNull[index++] = 0;
×
1979
      char* data = nodesGetValueFromNode(pValue);
×
1980
      if (pValue->node.resType.type == TSDB_DATA_TYPE_JSON) {
×
1981
        if (tTagIsJson(data)) {
×
1982
          terrno = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
×
1983
          nodesDestroyList(groupNew);
×
1984
          pAPI->metaReaderFn.clearReader(&mr);
×
1985
          return terrno;
×
1986
        }
1987
        int32_t len = getJsonValueLen(data);
×
1988
        memcpy(pStart, data, len);
×
1989
        pStart += len;
×
1990
      } else if (IS_VAR_DATA_TYPE(pValue->node.resType.type)) {
×
1991
        if (IS_STR_DATA_BLOB(pValue->node.resType.type)) {
×
1992
          return TSDB_CODE_BLOB_NOT_SUPPORT_TAG;
×
1993
        }
1994
        memcpy(pStart, data, varDataTLen(data));
×
1995
        pStart += varDataTLen(data);
×
1996
      } else {
1997
        memcpy(pStart, data, pValue->node.resType.bytes);
×
1998
        pStart += pValue->node.resType.bytes;
×
1999
      }
2000
    }
2001
  }
2002

2003
  int32_t len = (int32_t)(pStart - (char*)keyBuf);
×
2004
  *pGroupId = calcGroupId(keyBuf, len);
×
2005

2006
  nodesDestroyList(groupNew);
×
2007
  pAPI->metaReaderFn.clearReader(&mr);
×
2008

2009
  return TSDB_CODE_SUCCESS;
×
2010
}
2011

2012
SArray* makeColumnArrayFromList(SNodeList* pNodeList) {
16,007✔
2013
  if (!pNodeList) {
16,007!
2014
    return NULL;
×
2015
  }
2016

2017
  size_t  numOfCols = LIST_LENGTH(pNodeList);
16,007!
2018
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
16,007✔
2019
  if (pList == NULL) {
16,009!
2020
    return NULL;
×
2021
  }
2022

2023
  for (int32_t i = 0; i < numOfCols; ++i) {
37,390✔
2024
    SColumnNode* pColNode = (SColumnNode*)nodesListGetNode(pNodeList, i);
21,378✔
2025
    if (!pColNode) {
21,384!
2026
      taosArrayDestroy(pList);
×
2027
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2028
      return NULL;
×
2029
    }
2030

2031
    // todo extract method
2032
    SColumn c = {0};
21,384✔
2033
    c.slotId = pColNode->slotId;
21,384✔
2034
    c.colId = pColNode->colId;
21,384✔
2035
    c.type = pColNode->node.resType.type;
21,384✔
2036
    c.bytes = pColNode->node.resType.bytes;
21,384✔
2037
    c.precision = pColNode->node.resType.precision;
21,384✔
2038
    c.scale = pColNode->node.resType.scale;
21,384✔
2039

2040
    void* tmp = taosArrayPush(pList, &c);
21,381✔
2041
    if (!tmp) {
21,381!
2042
      taosArrayDestroy(pList);
×
2043
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2044
      return NULL;
×
2045
    }
2046
  }
2047

2048
  return pList;
16,012✔
2049
}
2050

2051
int32_t extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols,
16,003,392✔
2052
                            int32_t type, SColMatchInfo* pMatchInfo) {
2053
  size_t  numOfCols = LIST_LENGTH(pNodeList);
16,003,392!
2054
  int32_t code = TSDB_CODE_SUCCESS;
16,003,392✔
2055
  int32_t lino = 0;
16,003,392✔
2056

2057
  pMatchInfo->matchType = type;
16,003,392✔
2058

2059
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchItem));
16,003,392✔
2060
  if (pList == NULL) {
16,001,786!
2061
    code = terrno;
×
2062
    return code;
×
2063
  }
2064

2065
  for (int32_t i = 0; i < numOfCols; ++i) {
83,976,834✔
2066
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
67,968,874✔
2067
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
67,966,749!
2068
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
67,979,842✔
2069
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
67,900,548✔
2070

2071
      SColMatchItem c = {.needOutput = true};
67,900,548✔
2072
      c.colId = pColNode->colId;
67,900,548✔
2073
      c.srcSlotId = pColNode->slotId;
67,900,548✔
2074
      c.dstSlotId = pNode->slotId;
67,900,548✔
2075
      c.isPk = pColNode->isPk;
67,900,548✔
2076
      c.dataType = pColNode->node.resType;
67,900,548✔
2077
      void* tmp = taosArrayPush(pList, &c);
67,895,542✔
2078
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
67,895,542!
2079
    }
2080
  }
2081

2082
  // set the output flag for each column in SColMatchInfo, according to the
2083
  *numOfOutputCols = 0;
16,007,960✔
2084
  int32_t num = LIST_LENGTH(pOutputNodeList->pSlots);
16,007,960✔
2085
  for (int32_t i = 0; i < num; ++i) {
96,580,890✔
2086
    SSlotDescNode* pNode = (SSlotDescNode*)nodesListGetNode(pOutputNodeList->pSlots, i);
80,509,770✔
2087
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
81,085,796!
2088

2089
    // todo: add reserve flag check
2090
    // it is a column reserved for the arithmetic expression calculation
2091
    if (pNode->slotId >= numOfCols) {
81,159,158✔
2092
      (*numOfOutputCols) += 1;
13,201,988✔
2093
      continue;
13,201,988✔
2094
    }
2095

2096
    SColMatchItem* info = NULL;
67,957,170✔
2097
    for (int32_t j = 0; j < taosArrayGetSize(pList); ++j) {
2,147,483,647✔
2098
      info = taosArrayGet(pList, j);
2,147,483,647✔
2099
      QUERY_CHECK_NULL(info, code, lino, _end, terrno);
2,147,483,647!
2100
      if (info->dstSlotId == pNode->slotId) {
2,147,483,647✔
2101
        break;
66,944,004✔
2102
      }
2103
    }
2104

2105
    if (pNode->output) {
67,370,942✔
2106
      (*numOfOutputCols) += 1;
67,311,483✔
2107
    } else if (info != NULL) {
59,459!
2108
      // select distinct tbname from stb where tbname='abc';
2109
      info->needOutput = false;
59,459✔
2110
    }
2111
  }
2112

2113
  pMatchInfo->pList = pList;
16,071,120✔
2114

2115
_end:
16,071,120✔
2116
  if (code != TSDB_CODE_SUCCESS) {
16,071,120!
2117
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2118
  }
2119
  return code;
15,982,743✔
2120
}
2121

2122
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
50,444,170✔
2123
                                  const char* name) {
2124
  SResSchema s = {0};
50,444,170✔
2125
  s.scale = scale;
50,444,170✔
2126
  s.type = type;
50,444,170✔
2127
  s.bytes = bytes;
50,444,170✔
2128
  s.slotId = slotId;
50,444,170✔
2129
  s.precision = precision;
50,444,170✔
2130
  tstrncpy(s.name, name, tListLen(s.name));
50,444,170✔
2131

2132
  return s;
50,444,170✔
2133
}
2134

2135
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType, EColumnType colType) {
48,587,280✔
2136
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
48,587,280!
2137
  if (pCol == NULL) {
48,584,556!
2138
    return NULL;
×
2139
  }
2140

2141
  pCol->slotId = slotId;
48,584,556✔
2142
  pCol->colId = colId;
48,584,556✔
2143
  pCol->bytes = pType->bytes;
48,584,556✔
2144
  pCol->type = pType->type;
48,584,556✔
2145
  pCol->scale = pType->scale;
48,584,556✔
2146
  pCol->precision = pType->precision;
48,584,556✔
2147
  pCol->dataBlockId = blockId;
48,584,556✔
2148
  pCol->colType = colType;
48,584,556✔
2149
  return pCol;
48,584,556✔
2150
}
2151

2152
int32_t createExprFromOneNode(SExprInfo* pExp, SNode* pNode, int16_t slotId) {
50,407,358✔
2153
  int32_t code = TSDB_CODE_SUCCESS;
50,407,358✔
2154
  int32_t lino = 0;
50,407,358✔
2155
  pExp->base.numOfParams = 0;
50,407,358✔
2156
  pExp->base.pParam = NULL;
50,407,358✔
2157
  pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
50,407,358!
2158
  QUERY_CHECK_NULL(pExp->pExpr, code, lino, _end, terrno);
50,453,248!
2159

2160
  pExp->pExpr->_function.num = 1;
50,453,248✔
2161
  pExp->pExpr->_function.functionId = -1;
50,453,248✔
2162

2163
  int32_t type = nodeType(pNode);
50,453,248✔
2164
  // it is a project query, or group by column
2165
  if (type == QUERY_NODE_COLUMN) {
50,453,248✔
2166
    pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
32,724,811✔
2167
    SColumnNode* pColNode = (SColumnNode*)pNode;
32,724,811✔
2168

2169
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
32,724,811!
2170
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
32,725,377!
2171

2172
    pExp->base.numOfParams = 1;
32,725,377✔
2173

2174
    SDataType* pType = &pColNode->node.resType;
32,725,377✔
2175
    pExp->base.resSchema =
2176
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pColNode->colName);
32,725,377✔
2177

2178
    pExp->base.pParam[0].pCol =
65,445,455✔
2179
        createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType, pColNode->colType);
32,729,244✔
2180
    QUERY_CHECK_NULL(pExp->base.pParam[0].pCol, code, lino, _end, terrno);
32,716,211!
2181

2182
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
32,716,211✔
2183
  } else if (type == QUERY_NODE_VALUE) {
17,728,437✔
2184
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
301,516✔
2185
    SValueNode* pValNode = (SValueNode*)pNode;
301,516✔
2186

2187
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
301,516!
2188
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
301,549!
2189

2190
    pExp->base.numOfParams = 1;
301,549✔
2191

2192
    SDataType* pType = &pValNode->node.resType;
301,549✔
2193
    pExp->base.resSchema =
2194
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
301,549✔
2195
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
301,537✔
2196
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
301,537✔
2197
    QUERY_CHECK_CODE(code, lino, _end);
285,170!
2198
  } else if (type == QUERY_NODE_FUNCTION) {
17,426,921✔
2199
    pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
17,139,768✔
2200
    SFunctionNode* pFuncNode = (SFunctionNode*)pNode;
17,139,768✔
2201

2202
    SDataType* pType = &pFuncNode->node.resType;
17,139,768✔
2203
    pExp->base.resSchema =
2204
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pFuncNode->node.aliasName);
17,139,768✔
2205
    tExprNode* pExprNode = pExp->pExpr;
17,138,737✔
2206

2207
    pExprNode->_function.functionId = pFuncNode->funcId;
17,138,737✔
2208
    pExprNode->_function.pFunctNode = pFuncNode;
17,138,737✔
2209
    pExprNode->_function.functionType = pFuncNode->funcType;
17,138,737✔
2210

2211
    tstrncpy(pExprNode->_function.functionName, pFuncNode->functionName, tListLen(pExprNode->_function.functionName));
17,138,737✔
2212

2213
    pExp->base.pParamList = pFuncNode->pParameterList;
17,138,737✔
2214
#if 1
2215
    // todo refactor: add the parameter for tbname function
2216
    const char* name = "tbname";
17,138,737✔
2217
    int32_t     len = strlen(name);
17,138,737✔
2218

2219
    if (!pFuncNode->pParameterList && (memcmp(pExprNode->_function.functionName, name, len) == 0) &&
17,138,737✔
2220
        pExprNode->_function.functionName[len] == 0) {
1,040,414!
2221
      pFuncNode->pParameterList = NULL;
1,040,454✔
2222
      int32_t     code = nodesMakeList(&pFuncNode->pParameterList);
1,040,454✔
2223
      SValueNode* res = NULL;
1,040,443✔
2224
      if (TSDB_CODE_SUCCESS == code) {
1,040,443!
2225
        code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
1,040,447✔
2226
      }
2227
      QUERY_CHECK_CODE(code, lino, _end);
1,040,396!
2228
      res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
1,040,396✔
2229
      code = nodesListAppend(pFuncNode->pParameterList, (SNode*)res);
1,040,396✔
2230
      if (code != TSDB_CODE_SUCCESS) {
1,040,402!
2231
        nodesDestroyNode((SNode*)res);
×
2232
        res = NULL;
×
2233
      }
2234
      QUERY_CHECK_CODE(code, lino, _end);
1,040,419!
2235
    }
2236
#endif
2237

2238
    int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
17,138,702✔
2239

2240
    pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
17,138,702!
2241
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
17,143,991!
2242
    pExp->base.numOfParams = numOfParam;
17,143,991✔
2243

2244
    for (int32_t j = 0; j < numOfParam && TSDB_CODE_SUCCESS == code; ++j) {
36,474,346!
2245
      SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
19,326,580✔
2246
      QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
19,326,954!
2247
      if (p1->type == QUERY_NODE_COLUMN) {
19,328,977✔
2248
        SColumnNode* pcn = (SColumnNode*)p1;
15,869,713✔
2249

2250
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
15,869,713✔
2251
        pExp->base.pParam[j].pCol =
31,741,342✔
2252
            createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType, pcn->colType);
15,869,713✔
2253
        QUERY_CHECK_NULL(pExp->base.pParam[j].pCol, code, lino, _end, terrno);
15,871,629✔
2254
      } else if (p1->type == QUERY_NODE_VALUE) {
3,459,264✔
2255
        SValueNode* pvn = (SValueNode*)p1;
2,922,632✔
2256
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
2,922,632✔
2257
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
2,922,632✔
2258
        QUERY_CHECK_CODE(code, lino, _end);
2,922,448!
2259
      }
2260
    }
2261
    pExp->pExpr->_function.bindExprID = ((SExprNode*)pNode)->bindExprID;
17,147,766✔
2262
  } else if (type == QUERY_NODE_OPERATOR) {
287,153✔
2263
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
285,373✔
2264
    SOperatorNode* pOpNode = (SOperatorNode*)pNode;
285,373✔
2265

2266
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
285,373✔
2267
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
285,372!
2268
    pExp->base.numOfParams = 1;
285,372✔
2269

2270
    SDataType* pType = &pOpNode->node.resType;
285,372✔
2271
    pExp->base.resSchema =
2272
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pOpNode->node.aliasName);
285,372✔
2273
    pExp->pExpr->_optrRoot.pRootNode = pNode;
285,337✔
2274
  } else if (type == QUERY_NODE_CASE_WHEN) {
1,780!
2275
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
4,190✔
2276
    SCaseWhenNode* pCaseNode = (SCaseWhenNode*)pNode;
4,190✔
2277

2278
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
4,190!
2279
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
4,191!
2280
    pExp->base.numOfParams = 1;
4,191✔
2281

2282
    SDataType* pType = &pCaseNode->node.resType;
4,191✔
2283
    pExp->base.resSchema =
2284
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCaseNode->node.aliasName);
4,191✔
2285
    pExp->pExpr->_optrRoot.pRootNode = pNode;
4,189✔
2286
  } else if (type == QUERY_NODE_LOGIC_CONDITION) {
×
2287
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
10✔
2288
    SLogicConditionNode* pCond = (SLogicConditionNode*)pNode;
10✔
2289
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
10!
2290
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
10!
2291
    pExp->base.numOfParams = 1;
10✔
2292
    SDataType* pType = &pCond->node.resType;
10✔
2293
    pExp->base.resSchema =
2294
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCond->node.aliasName);
10✔
2295
    pExp->pExpr->_optrRoot.pRootNode = pNode;
10✔
2296
  } else {
2297
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2298
    QUERY_CHECK_CODE(code, lino, _end);
×
2299
  }
2300
  pExp->pExpr->relatedTo = ((SExprNode*)pNode)->relatedTo;
50,436,263✔
2301
_end:
50,436,263✔
2302
  if (code != TSDB_CODE_SUCCESS) {
50,436,263!
2303
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2304
  }
2305
  return code;
50,439,063✔
2306
}
2307

2308
int32_t createExprFromTargetNode(SExprInfo* pExp, STargetNode* pTargetNode) {
50,379,780✔
2309
  return createExprFromOneNode(pExp, pTargetNode->pExpr, pTargetNode->slotId);
50,379,780✔
2310
}
2311

2312
SExprInfo* createExpr(SNodeList* pNodeList, int32_t* numOfExprs) {
×
2313
  *numOfExprs = LIST_LENGTH(pNodeList);
×
2314
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
×
2315
  if (!pExprs) {
×
2316
    return NULL;
×
2317
  }
2318

2319
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
×
2320
    SExprInfo* pExp = &pExprs[i];
×
2321
    int32_t    code = createExprFromOneNode(pExp, nodesListGetNode(pNodeList, i), i + UD_TAG_COLUMN_INDEX);
×
2322
    if (code != TSDB_CODE_SUCCESS) {
×
2323
      taosMemoryFreeClear(pExprs);
×
2324
      terrno = code;
×
2325
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2326
      return NULL;
×
2327
    }
2328
  }
2329

2330
  return pExprs;
×
2331
}
2332

2333
int32_t createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, SExprInfo** pExprInfo, int32_t* numOfExprs) {
20,256,364✔
2334
  QRY_PARAM_CHECK(pExprInfo);
20,256,364!
2335

2336
  int32_t code = 0;
20,256,364✔
2337
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
20,256,364✔
2338
  int32_t numOfGroupKeys = 0;
20,256,364✔
2339
  if (pGroupKeys != NULL) {
20,256,364✔
2340
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
539,759!
2341
  }
2342

2343
  *numOfExprs = numOfFuncs + numOfGroupKeys;
20,256,364✔
2344
  if (*numOfExprs == 0) {
20,256,364✔
2345
    return code;
2,916,771✔
2346
  }
2347

2348
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
17,339,593!
2349
  if (pExprs == NULL) {
17,344,088!
2350
    return terrno;
×
2351
  }
2352

2353
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
66,999,832✔
2354
    STargetNode* pTargetNode = NULL;
49,651,027✔
2355
    if (i < numOfFuncs) {
49,651,027✔
2356
      pTargetNode = (STargetNode*)nodesListGetNode(pNodeList, i);
48,800,520✔
2357
    } else {
2358
      pTargetNode = (STargetNode*)nodesListGetNode(pGroupKeys, i - numOfFuncs);
850,507✔
2359
    }
2360
    if (!pTargetNode) {
49,602,396!
2361
      destroyExprInfo(pExprs, *numOfExprs);
×
2362
      taosMemoryFreeClear(pExprs);
×
2363
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2364
      return terrno;
×
2365
    }
2366

2367
    SExprInfo* pExp = &pExprs[i];
49,602,396✔
2368
    code = createExprFromTargetNode(pExp, pTargetNode);
49,602,396✔
2369
    if (code != TSDB_CODE_SUCCESS) {
49,658,115✔
2370
      destroyExprInfo(pExprs, *numOfExprs);
2,371✔
2371
      taosMemoryFreeClear(pExprs);
×
2372
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2373
      return code;
×
2374
    }
2375
  }
2376

2377
  *pExprInfo = pExprs;
17,348,805✔
2378
  return code;
17,348,805✔
2379
}
2380

2381
static void deleteSubsidiareCtx(void* pData) {
×
2382
  SSubsidiaryResInfo* pCtx = (SSubsidiaryResInfo*)pData;
×
2383
  if (pCtx->pCtx) {
×
2384
    taosMemoryFreeClear(pCtx->pCtx);
×
2385
  }
2386
}
×
2387

2388
// set the output buffer for the selectivity + tag query
2389
static int32_t setSelectValueColumnInfo(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
19,621,959✔
2390
  int32_t num = 0;
19,621,959✔
2391
  int32_t code = TSDB_CODE_SUCCESS;
19,621,959✔
2392
  int32_t lino = 0;
19,621,959✔
2393

2394
  SArray* pValCtxArray = NULL;
19,621,959✔
2395
  for (int32_t i = numOfOutput - 1; i > 0; --i) {  // select Func is at the end of the list
51,846,529✔
2396
    int32_t funcIdx = pCtx[i].pExpr->pExpr->_function.bindExprID;
32,224,570✔
2397
    if (funcIdx > 0) {
32,224,570✔
2398
      if (pValCtxArray == NULL) {
13,859✔
2399
        // the end of the list is the select function of biggest index
2400
        pValCtxArray = taosArrayInit_s(sizeof(SSubsidiaryResInfo*), funcIdx);
9,872✔
2401
        if (pValCtxArray == NULL) {
9,874✔
2402
          return terrno;
2✔
2403
        }
2404
      }
2405
      if (funcIdx > pValCtxArray->size) {
13,859!
2406
        qError("funcIdx:%d is out of range", funcIdx);
×
2407
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2408
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2409
      }
2410
      SSubsidiaryResInfo* pSubsidiary = &pCtx[i].subsidiaries;
13,859✔
2411
      pSubsidiary->pCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
13,859!
2412
      if (pSubsidiary->pCtx == NULL) {
13,862!
2413
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2414
        return terrno;
×
2415
      }
2416
      pSubsidiary->num = 0;
13,862✔
2417
      taosArraySet(pValCtxArray, funcIdx - 1, &pSubsidiary);
13,862✔
2418
    }
2419
  }
2420

2421
  SqlFunctionCtx*  p = NULL;
19,621,959✔
2422
  SqlFunctionCtx** pValCtx = NULL;
19,621,959✔
2423
  if (pValCtxArray == NULL) {
19,621,959✔
2424
    pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
19,613,120!
2425
    if (pValCtx == NULL) {
19,620,547!
2426
      QUERY_CHECK_CODE(terrno, lino, _end);
×
2427
    }
2428
  }
2429

2430
  for (int32_t i = 0; i < numOfOutput; ++i) {
69,923,175✔
2431
    const char* pName = pCtx[i].pExpr->pExpr->_function.functionName;
50,294,512✔
2432
    if ((strcmp(pName, "_select_value") == 0)) {
50,294,512✔
2433
      if (pValCtxArray == NULL) {
2,239,571✔
2434
        pValCtx[num++] = &pCtx[i];
2,220,879✔
2435
      } else {
2436
        int32_t bindFuncIndex = pCtx[i].pExpr->pExpr->relatedTo;  // start from index 1;
18,692✔
2437
        if (bindFuncIndex > 0) {                                  // 0 is default index related to the select function
18,692!
2438
          bindFuncIndex -= 1;
18,947✔
2439
        }
2440
        SSubsidiaryResInfo** pSubsidiary = taosArrayGet(pValCtxArray, bindFuncIndex);
18,692✔
2441
        if (pSubsidiary == NULL) {
19,370!
2442
          QUERY_CHECK_CODE(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR, lino, _end);
×
2443
        }
2444
        (*pSubsidiary)->pCtx[(*pSubsidiary)->num] = &pCtx[i];
19,370✔
2445
        (*pSubsidiary)->num++;
19,370✔
2446
      }
2447
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
48,054,941✔
2448
      if (pValCtxArray == NULL) {
4,421,230✔
2449
        p = &pCtx[i];
4,404,907✔
2450
      }
2451
    }
2452
  }
2453

2454
  if (p != NULL) {
19,628,663✔
2455
    p->subsidiaries.pCtx = pValCtx;
4,111,247✔
2456
    p->subsidiaries.num = num;
4,111,247✔
2457
  } else {
2458
    taosMemoryFreeClear(pValCtx);
15,517,416!
2459
  }
2460

2461
_end:
12,916✔
2462
  if (code != TSDB_CODE_SUCCESS) {
19,627,186!
2463
    taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2464
    taosMemoryFreeClear(pValCtx);
×
2465
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2466
  } else {
2467
    taosArrayDestroy(pValCtxArray);
19,627,186✔
2468
  }
2469
  return code;
19,610,703✔
2470
}
2471

2472
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowEntryInfoOffset,
19,619,009✔
2473
                                     SFunctionStateStore* pStore) {
2474
  int32_t         code = TSDB_CODE_SUCCESS;
19,619,009✔
2475
  int32_t         lino = 0;
19,619,009✔
2476
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
19,619,009!
2477
  if (pFuncCtx == NULL) {
19,617,263!
2478
    return NULL;
×
2479
  }
2480

2481
  *rowEntryInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
19,617,263!
2482
  if (*rowEntryInfoOffset == 0) {
19,630,680!
2483
    taosMemoryFreeClear(pFuncCtx);
×
2484
    return NULL;
×
2485
  }
2486

2487
  for (int32_t i = 0; i < numOfOutput; ++i) {
69,941,494✔
2488
    SExprInfo* pExpr = &pExprInfo[i];
50,304,961✔
2489

2490
    SExprBasicInfo* pFunct = &pExpr->base;
50,304,961✔
2491
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
50,304,961✔
2492

2493
    pCtx->functionId = -1;
50,304,961✔
2494
    pCtx->pExpr = pExpr;
50,304,961✔
2495

2496
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
50,304,961✔
2497
      SFuncExecEnv env = {0};
17,139,181✔
2498
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;
17,139,181✔
2499
      pCtx->isPseudoFunc = fmIsWindowPseudoColumnFunc(pCtx->functionId) || fmIsPlaceHolderFunc(pCtx->functionId);
17,139,181✔
2500
      pCtx->isNotNullFunc = fmIsNotNullOutputFunc(pCtx->functionId);
17,130,669✔
2501

2502
      bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
17,135,458✔
2503
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId)) {
28,811,543✔
2504
        if (!isUdaf) {
11,680,932✔
2505
          code = fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
11,680,847✔
2506
          QUERY_CHECK_CODE(code, lino, _end);
11,668,140!
2507
        } else {
2508
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
85✔
2509
          pCtx->udfName = taosStrdup(udfName);
85!
2510
          QUERY_CHECK_NULL(pCtx->udfName, code, lino, _end, terrno);
85!
2511

2512
          code = fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
85✔
2513
          QUERY_CHECK_CODE(code, lino, _end);
85!
2514
        }
2515
        bool tmp = pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
11,668,225✔
2516
        if (!tmp) {
11,674,346!
2517
          code = terrno;
×
2518
          QUERY_CHECK_CODE(code, lino, _end);
×
2519
        }
2520
      } else {
2521
        if (fmIsPlaceHolderFunc(pCtx->functionId)) {
5,443,737✔
2522
          code = fmGetStreamPesudoFuncEnv(pCtx->functionId, pExpr->base.pParamList, &env);
14,320✔
2523
          QUERY_CHECK_CODE(code, lino, _end);
14,320!
2524
        }      
2525
        
2526
        code = fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
5,449,239✔
2527
        if (code != TSDB_CODE_SUCCESS && isUdaf) {
5,446,104!
2528
          code = TSDB_CODE_SUCCESS;
180✔
2529
        }
2530
        QUERY_CHECK_CODE(code, lino, _end);
5,446,104!
2531

2532
        if (pCtx->sfp.getEnv != NULL) {
5,446,104✔
2533
          bool tmp = pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
3,407,875✔
2534
          if (!tmp) {
3,405,356!
2535
            code = terrno;
×
2536
            QUERY_CHECK_CODE(code, lino, _end);
×
2537
          }
2538
        }
2539
      }
2540
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
17,123,153✔
2541
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
33,165,780✔
2542
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
301,540!
2543
      // for simple column, the result buffer needs to hold at least one element.
2544
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
33,169,145✔
2545
    }
2546

2547
    pCtx->input.numOfInputCols = pFunct->numOfParams;
50,288,933✔
2548
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
50,288,933!
2549
    QUERY_CHECK_NULL(pCtx->input.pData, code, lino, _end, terrno);
50,313,558!
2550
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
50,313,558!
2551
    QUERY_CHECK_NULL(pCtx->input.pColumnDataAgg, code, lino, _end, terrno);
50,310,814!
2552

2553
    pCtx->pTsOutput = NULL;
50,310,814✔
2554
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
50,310,814✔
2555
    pCtx->resDataInfo.type = pFunct->resSchema.type;
50,310,814✔
2556
    pCtx->order = TSDB_ORDER_ASC;
50,310,814✔
2557
    pCtx->start.key = INT64_MIN;
50,310,814✔
2558
    pCtx->end.key = INT64_MIN;
50,310,814✔
2559
    pCtx->numOfParams = pExpr->base.numOfParams;
50,310,814✔
2560
    pCtx->param = pFunct->pParam;
50,310,814✔
2561
    pCtx->saveHandle.currentPage = -1;
50,310,814✔
2562
    pCtx->pStore = pStore;
50,310,814✔
2563
    pCtx->hasWindowOrGroup = false;
50,310,814✔
2564
    pCtx->needCleanup = false;
50,310,814✔
2565
  }
2566

2567
  for (int32_t i = 1; i < numOfOutput; ++i) {
51,867,969✔
2568
    (*rowEntryInfoOffset)[i] = (int32_t)((*rowEntryInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) +
32,231,436✔
2569
                                         pFuncCtx[i - 1].resDataInfo.interBufSize);
32,231,436✔
2570
  }
2571

2572
  code = setSelectValueColumnInfo(pFuncCtx, numOfOutput);
19,636,533✔
2573
  QUERY_CHECK_CODE(code, lino, _end);
19,612,701!
2574

2575
_end:
19,612,701✔
2576
  if (code != TSDB_CODE_SUCCESS) {
19,612,701!
2577
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2578
    for (int32_t i = 0; i < numOfOutput; ++i) {
×
2579
      taosMemoryFree(pFuncCtx[i].input.pData);
×
2580
      taosMemoryFree(pFuncCtx[i].input.pColumnDataAgg);
×
2581
    }
2582
    taosMemoryFreeClear(*rowEntryInfoOffset);
×
2583
    taosMemoryFreeClear(pFuncCtx);
×
2584

2585
    terrno = code;
×
2586
    return NULL;
×
2587
  }
2588
  return pFuncCtx;
19,612,701✔
2589
}
2590

2591
// NOTE: sources columns are more than the destination SSDatablock columns.
2592
// doFilter in table scan needs every column even its output is false
2593
int32_t relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray* pCols, bool outputEveryColumn) {
2,027,233✔
2594
  int32_t code = TSDB_CODE_SUCCESS;
2,027,233✔
2595
  size_t  numOfSrcCols = taosArrayGetSize(pCols);
2,027,233✔
2596

2597
  int32_t i = 0, j = 0;
2,027,304✔
2598
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
4,234,817✔
2599
    SColumnInfoData* p = taosArrayGet(pCols, i);
2,207,542✔
2600
    if (!p) {
2,206,266!
2601
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2602
      return terrno;
×
2603
    }
2604
    SColMatchItem* pmInfo = taosArrayGet(pColMatchInfo, j);
2,206,266✔
2605
    if (!pmInfo) {
2,205,913!
2606
      return terrno;
×
2607
    }
2608

2609
    if (p->info.colId == pmInfo->colId) {
2,206,044✔
2610
      SColumnInfoData* pDst = taosArrayGet(pBlock->pDataBlock, pmInfo->dstSlotId);
2,174,789✔
2611
      if (!pDst) {
2,173,923!
2612
        return terrno;
×
2613
      }
2614
      code = colDataAssign(pDst, p, pBlock->info.rows, &pBlock->info);
2,173,923✔
2615
      if (code != TSDB_CODE_SUCCESS) {
2,176,353✔
2616
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
95!
2617
        return code;
×
2618
      }
2619
      i++;
2,176,258✔
2620
      j++;
2,176,258✔
2621
    } else if (p->info.colId < pmInfo->colId) {
31,255!
2622
      i++;
31,255✔
2623
    } else {
2624
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2625
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2626
    }
2627
  }
2628
  return code;
2,027,390✔
2629
}
2630

2631
SInterval extractIntervalInfo(const STableScanPhysiNode* pTableScanNode) {
4,707,666✔
2632
  SInterval interval = {
4,707,666✔
2633
      .interval = pTableScanNode->interval,
4,707,666✔
2634
      .sliding = pTableScanNode->sliding,
4,707,666✔
2635
      .intervalUnit = pTableScanNode->intervalUnit,
4,707,666✔
2636
      .slidingUnit = pTableScanNode->slidingUnit,
4,707,666✔
2637
      .offset = pTableScanNode->offset,
4,707,666✔
2638
      .precision = pTableScanNode->scan.node.pOutputDataBlockDesc->precision,
4,707,666✔
2639
      .timeRange = pTableScanNode->scanRange,
2640
  };
2641
  calcIntervalAutoOffset(&interval);
4,707,666✔
2642

2643
  return interval;
4,709,112✔
2644
}
2645

2646
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
1,120,164✔
2647
  SColumn c = {0};
1,120,164✔
2648

2649
  c.slotId = pColNode->slotId;
1,120,164✔
2650
  c.colId = pColNode->colId;
1,120,164✔
2651
  c.type = pColNode->node.resType.type;
1,120,164✔
2652
  c.bytes = pColNode->node.resType.bytes;
1,120,164✔
2653
  c.scale = pColNode->node.resType.scale;
1,120,164✔
2654
  c.precision = pColNode->node.resType.precision;
1,120,164✔
2655
  return c;
1,120,164✔
2656
}
2657

2658
int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, const STableScanPhysiNode* pTableScanNode,
8,428,883✔
2659
                               const SReadHandle* readHandle) {
2660
  pCond->order = pTableScanNode->scanSeq[0] > 0 ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
8,428,883✔
2661
  pCond->numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
8,428,883!
2662

2663
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
8,428,883!
2664
  if (!pCond->colList) {
8,439,641!
2665
    return terrno;
×
2666
  }
2667
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
8,439,641!
2668
  if (pCond->pSlotList == NULL) {
8,434,462!
2669
    taosMemoryFreeClear(pCond->colList);
×
2670
    return terrno;
×
2671
  }
2672

2673
  // TODO: get it from stable scan node
2674
  pCond->twindows = pTableScanNode->scanRange;
8,434,462✔
2675
  pCond->suid = pTableScanNode->scan.suid;
8,434,462✔
2676
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
8,434,462✔
2677
  pCond->startVersion = -1;
8,434,462✔
2678
  pCond->endVersion = -1;
8,434,462✔
2679
  pCond->skipRollup = readHandle->skipRollup;
8,434,462✔
2680
  if (readHandle->winRangeValid) {
8,434,462✔
2681
    pCond->twindows = readHandle->winRange;
89,660✔
2682
  }
2683
  // allowed read stt file optimization mode
2684
  pCond->notLoadData = (pTableScanNode->dataRequired == FUNC_DATA_REQUIRED_NOT_LOAD) &&
17,129,340✔
2685
                       (pTableScanNode->scan.node.pConditions == NULL) && (pTableScanNode->interval == 0);
8,434,462✔
2686

2687
  int32_t j = 0;
8,434,462✔
2688
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
53,345,604✔
2689
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pTableScanNode->scan.pScanCols, i);
44,914,421✔
2690
    if (!pNode) {
44,911,124!
2691
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2692
      return terrno;
×
2693
    }
2694
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
44,911,142✔
2695
    if (pColNode->colType == COLUMN_TYPE_TAG) {
44,911,142!
2696
      continue;
×
2697
    }
2698

2699
    pCond->colList[j].type = pColNode->node.resType.type;
44,911,142✔
2700
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
44,911,142✔
2701
    pCond->colList[j].colId = pColNode->colId;
44,911,142✔
2702
    pCond->colList[j].pk = pColNode->isPk;
44,911,142✔
2703

2704
    pCond->pSlotList[j] = pNode->slotId;
44,911,142✔
2705
    j += 1;
44,911,142✔
2706
  }
2707

2708
  pCond->numOfCols = j;
8,431,183✔
2709
  return TSDB_CODE_SUCCESS;
8,431,183✔
2710
}
2711

2712
int32_t initQueryTableDataCondWithColArray(SQueryTableDataCond* pCond, SQueryTableDataCond* pOrgCond,
76,777✔
2713
                                           const SReadHandle* readHandle, SArray* colArray) {
2714
  int32_t code = TSDB_CODE_SUCCESS;
76,777✔
2715
  int32_t lino = 0;
76,777✔
2716

2717
  pCond->order = TSDB_ORDER_ASC;
76,777✔
2718
  pCond->numOfCols = (int32_t)taosArrayGetSize(colArray);
76,777✔
2719

2720
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
76,777!
2721
  QUERY_CHECK_NULL(pCond->colList, code, lino, _return, terrno);
76,777!
2722

2723
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
76,777!
2724
  QUERY_CHECK_NULL(pCond->pSlotList, code, lino, _return, terrno);
76,777!
2725

2726
  pCond->twindows = pOrgCond->twindows;
76,777✔
2727
  pCond->type = pOrgCond->type;
76,777✔
2728
  pCond->startVersion = -1;
76,777✔
2729
  pCond->endVersion = -1;
76,777✔
2730
  pCond->skipRollup = true;
76,777✔
2731
  pCond->notLoadData = false;
76,777✔
2732

2733
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
231,059✔
2734
    SColIdPair* pColPair = taosArrayGet(colArray, i);
154,282✔
2735
    QUERY_CHECK_NULL(pColPair, code, lino, _return, terrno);
154,282!
2736

2737
    bool find = false;
154,282✔
2738
    for (int32_t j = 0; j < pOrgCond->numOfCols; ++j) {
238,335!
2739
      if (pOrgCond->colList[j].colId == pColPair->vtbColId) {
238,335✔
2740
        pCond->colList[i].type = pOrgCond->colList[j].type;
154,282✔
2741
        pCond->colList[i].bytes = pOrgCond->colList[j].bytes;
154,282✔
2742
        pCond->colList[i].colId = pColPair->orgColId;
154,282✔
2743
        pCond->colList[i].pk = pOrgCond->colList[j].pk;
154,282✔
2744
        pCond->pSlotList[i] = i;
154,282✔
2745
        find = true;
154,282✔
2746
        break;
154,282✔
2747
      }
2748
    }
2749
    QUERY_CHECK_CONDITION(find, code, lino, _return, TSDB_CODE_NOT_FOUND);
154,282!
2750
  }
2751

2752
  return code;
76,777✔
2753
_return:
×
2754
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(terrno));
×
2755
  taosMemoryFreeClear(pCond->colList);
×
2756
  taosMemoryFreeClear(pCond->pSlotList);
×
2757
  return code;
×
2758
}
2759

2760
void cleanupQueryTableDataCond(SQueryTableDataCond* pCond) {
20,770,211✔
2761
  taosMemoryFreeClear(pCond->colList);
20,770,211!
2762
  taosMemoryFreeClear(pCond->pSlotList);
20,770,681!
2763
}
20,770,588✔
2764

2765
int32_t convertFillType(int32_t mode) {
1,083,453✔
2766
  int32_t type = TSDB_FILL_NONE;
1,083,453✔
2767
  switch (mode) {
1,083,453!
2768
    case FILL_MODE_PREV:
30,643✔
2769
      type = TSDB_FILL_PREV;
30,643✔
2770
      break;
30,643✔
2771
    case FILL_MODE_NONE:
×
2772
      type = TSDB_FILL_NONE;
×
2773
      break;
×
2774
    case FILL_MODE_NULL:
32,340✔
2775
      type = TSDB_FILL_NULL;
32,340✔
2776
      break;
32,340✔
2777
    case FILL_MODE_NULL_F:
58,811✔
2778
      type = TSDB_FILL_NULL_F;
58,811✔
2779
      break;
58,811✔
2780
    case FILL_MODE_NEXT:
31,628✔
2781
      type = TSDB_FILL_NEXT;
31,628✔
2782
      break;
31,628✔
2783
    case FILL_MODE_VALUE:
297,209✔
2784
      type = TSDB_FILL_SET_VALUE;
297,209✔
2785
      break;
297,209✔
2786
    case FILL_MODE_VALUE_F:
591,045✔
2787
      type = TSDB_FILL_SET_VALUE_F;
591,045✔
2788
      break;
591,045✔
2789
    case FILL_MODE_LINEAR:
31,721✔
2790
      type = TSDB_FILL_LINEAR;
31,721✔
2791
      break;
31,721✔
2792
    case FILL_MODE_NEAR:
10,088✔
2793
      type = TSDB_FILL_NEAR;
10,088✔
2794
      break;
10,088✔
2795
    default:
×
2796
      type = TSDB_FILL_NONE;
×
2797
  }
2798

2799
  return type;
1,083,453✔
2800
}
2801

2802
void getInitialStartTimeWindow(SInterval* pInterval, TSKEY ts, STimeWindow* w, bool ascQuery) {
1,286,487✔
2803
  if (ascQuery) {
1,286,487✔
2804
    *w = getAlignQueryTimeWindow(pInterval, ts);
705,689✔
2805
  } else {
2806
    // the start position of the first time window in the endpoint that spreads beyond the queried last timestamp
2807
    *w = getAlignQueryTimeWindow(pInterval, ts);
580,798✔
2808

2809
    int64_t key = w->skey;
580,813✔
2810
    while (key < ts) {  // moving towards end
1,981,692✔
2811
      key = getNextTimeWindowStart(pInterval, key, TSDB_ORDER_ASC);
1,951,667✔
2812
      if (key > ts) {
1,951,667✔
2813
        break;
550,788✔
2814
      }
2815

2816
      w->skey = key;
1,400,879✔
2817
    }
2818
    w->ekey = taosTimeAdd(w->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
580,813✔
2819
  }
2820
}
1,286,514✔
2821

2822
static STimeWindow doCalculateTimeWindow(int64_t ts, SInterval* pInterval) {
1,465,638✔
2823
  STimeWindow w = {0};
1,465,638✔
2824

2825
  w.skey = taosTimeTruncate(ts, pInterval);
1,465,638✔
2826
  w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
1,465,767✔
2827
  return w;
1,465,706✔
2828
}
2829

2830
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order) {
872,883✔
2831
  STimeWindow win = *pWindow;
872,883✔
2832
  STimeWindow save = win;
872,883✔
2833
  while (win.skey <= ts && win.ekey >= ts) {
3,911,185✔
2834
    save = win;
3,038,305✔
2835
    // get previous time window
2836
    getNextTimeWindow(pInterval, &win, order == TSDB_ORDER_ASC ? TSDB_ORDER_DESC : TSDB_ORDER_ASC);
3,038,305✔
2837
  }
2838

2839
  return save;
872,880✔
2840
}
2841

2842
// get the correct time window according to the handled timestamp
2843
// todo refactor
2844
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
3,723,417✔
2845
                                int32_t order) {
2846
  STimeWindow w = {0};
3,723,417✔
2847
  if (pResultRowInfo->cur.pageId == -1) {  // the first window, from the previous stored value
3,723,417✔
2848
    getInitialStartTimeWindow(pInterval, ts, &w, (order == TSDB_ORDER_ASC));
1,286,486✔
2849
    return w;
1,286,506✔
2850
  }
2851

2852
  SResultRow* pRow = getResultRowByPos(pBuf, &pResultRowInfo->cur, false);
2,436,931✔
2853
  if (pRow) {
2,436,734!
2854
    TAOS_SET_OBJ_ALIGNED(&w, pRow->win);
2,436,748✔
2855
  }
2856

2857
  // in case of typical time window, we can calculate time window directly.
2858
  if (w.skey > ts || w.ekey < ts) {
2,436,734✔
2859
    w = doCalculateTimeWindow(ts, pInterval);
1,465,634✔
2860
  }
2861

2862
  if (pInterval->interval != pInterval->sliding) {
2,436,798✔
2863
    // it is an sliding window query, in which sliding value is not equalled to
2864
    // interval value, and we need to find the first qualified time window.
2865
    w = getFirstQualifiedTimeWindow(ts, &w, pInterval, order);
872,883✔
2866
  }
2867

2868
  return w;
2,436,793✔
2869
}
2870

2871
TSKEY getNextTimeWindowStart(const SInterval* pInterval, TSKEY start, int32_t order) {
331,565,627✔
2872
  int32_t factor = GET_FORWARD_DIRECTION_FACTOR(order);
331,565,627✔
2873
  TSKEY   nextStart = taosTimeAdd(start, -1 * pInterval->offset, pInterval->offsetUnit, pInterval->precision, NULL);
331,565,627✔
2874
  nextStart = taosTimeAdd(nextStart, factor * pInterval->sliding, pInterval->slidingUnit, pInterval->precision, NULL);
331,334,304✔
2875
  nextStart = taosTimeAdd(nextStart, pInterval->offset, pInterval->offsetUnit, pInterval->precision, NULL);
331,181,420✔
2876
  return nextStart;
330,897,003✔
2877
}
2878

2879
void getNextTimeWindow(const SInterval* pInterval, STimeWindow* tw, int32_t order) {
329,622,792✔
2880
  tw->skey = getNextTimeWindowStart(pInterval, tw->skey, order);
329,622,792✔
2881
  tw->ekey = taosTimeAdd(tw->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
328,946,106✔
2882
}
328,934,909✔
2883

2884
bool hasLimitOffsetInfo(SLimitInfo* pLimitInfo) {
24,724,328✔
2885
  return (pLimitInfo->limit.limit != -1 || pLimitInfo->limit.offset != -1 || pLimitInfo->slimit.limit != -1 ||
49,431,541!
2886
          pLimitInfo->slimit.offset != -1);
24,707,213✔
2887
}
2888

2889
bool hasSlimitOffsetInfo(SLimitInfo* pLimitInfo) {
×
2890
  return (pLimitInfo->slimit.limit != -1 || pLimitInfo->slimit.offset != -1);
×
2891
}
2892

2893
void initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo) {
24,311,546✔
2894
  SLimit limit = {.limit = getLimit(pLimit), .offset = getOffset(pLimit)};
24,311,546✔
2895
  SLimit slimit = {.limit = getLimit(pSLimit), .offset = getOffset(pSLimit)};
24,324,494✔
2896

2897
  pLimitInfo->limit = limit;
24,330,665✔
2898
  pLimitInfo->slimit = slimit;
24,330,665✔
2899
  pLimitInfo->remainOffset = limit.offset;
24,330,665✔
2900
  pLimitInfo->remainGroupOffset = slimit.offset;
24,330,665✔
2901
}
24,330,665✔
2902

2903
void resetLimitInfoForNextGroup(SLimitInfo* pLimitInfo) {
4,006,509✔
2904
  pLimitInfo->numOfOutputRows = 0;
4,006,509✔
2905
  pLimitInfo->remainOffset = pLimitInfo->limit.offset;
4,006,509✔
2906
}
4,006,509✔
2907

2908
int32_t tableListGetSize(const STableListInfo* pTableList, int32_t* pRes) {
30,516,392✔
2909
  if (taosArrayGetSize(pTableList->pTableList) != taosHashGetSize(pTableList->map)) {
30,516,392!
2910
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2911
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2912
  }
2913
  (*pRes) = taosArrayGetSize(pTableList->pTableList);
30,520,316✔
2914
  return TSDB_CODE_SUCCESS;
30,515,497✔
2915
}
2916

2917
uint64_t tableListGetSuid(const STableListInfo* pTableList) { return pTableList->idInfo.suid; }
142,425✔
2918

2919
STableKeyInfo* tableListGetInfo(const STableListInfo* pTableList, int32_t index) {
13,109,533✔
2920
  if (taosArrayGetSize(pTableList->pTableList) == 0) {
13,109,533✔
2921
    return NULL;
1,093✔
2922
  }
2923

2924
  return taosArrayGet(pTableList->pTableList, index);
13,106,289✔
2925
}
2926

2927
int32_t tableListFind(const STableListInfo* pTableList, uint64_t uid, int32_t startIndex) {
143✔
2928
  int32_t numOfTables = taosArrayGetSize(pTableList->pTableList);
143✔
2929
  if (startIndex >= numOfTables) {
143!
2930
    return -1;
×
2931
  }
2932

2933
  for (int32_t i = startIndex; i < numOfTables; ++i) {
2,573!
2934
    STableKeyInfo* p = taosArrayGet(pTableList->pTableList, i);
2,573✔
2935
    if (!p) {
2,573!
2936
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2937
      return -1;
×
2938
    }
2939
    if (p->uid == uid) {
2,573✔
2940
      return i;
143✔
2941
    }
2942
  }
2943
  return -1;
×
2944
}
2945

2946
void tableListGetSourceTableInfo(const STableListInfo* pTableList, uint64_t* psuid, uint64_t* uid, int32_t* type) {
202✔
2947
  *psuid = pTableList->idInfo.suid;
202✔
2948
  *uid = pTableList->idInfo.uid;
202✔
2949
  *type = pTableList->idInfo.tableType;
202✔
2950
}
202✔
2951

2952
uint64_t tableListGetTableGroupId(const STableListInfo* pTableList, uint64_t tableUid) {
27,669,119✔
2953
  int32_t* slot = taosHashGet(pTableList->map, &tableUid, sizeof(tableUid));
27,669,119✔
2954
  if (slot == NULL) {
27,673,081✔
2955
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
27,862✔
2956
    return -1;
27,863✔
2957
  }
2958

2959
  STableKeyInfo* pKeyInfo = taosArrayGet(pTableList->pTableList, *slot);
27,645,219✔
2960

2961
  return pKeyInfo->groupId;
27,641,177✔
2962
}
2963

2964
// TODO handle the group offset info, fix it, the rule of group output will be broken by this function
2965
int32_t tableListAddTableInfo(STableListInfo* pTableList, uint64_t uid, uint64_t gid) {
17,028✔
2966
  int32_t code = TSDB_CODE_SUCCESS;
17,028✔
2967
  int32_t lino = 0;
17,028✔
2968
  if (pTableList->map == NULL) {
17,028!
2969
    pTableList->map = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
×
2970
    QUERY_CHECK_NULL(pTableList->map, code, lino, _end, terrno);
×
2971
  }
2972

2973
  STableKeyInfo keyInfo = {.uid = uid, .groupId = gid};
17,028✔
2974
  void*         p = taosHashGet(pTableList->map, &uid, sizeof(uid));
17,028✔
2975
  if (p != NULL) {
17,081✔
2976
    qInfo("table:%" PRId64 " already in tableIdList, ignore it", uid);
3!
2977
    goto _end;
3✔
2978
  }
2979

2980
  void* tmp = taosArrayPush(pTableList->pTableList, &keyInfo);
17,078✔
2981
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
17,047!
2982

2983
  int32_t slot = (int32_t)taosArrayGetSize(pTableList->pTableList) - 1;
17,047✔
2984
  code = taosHashPut(pTableList->map, &uid, sizeof(uid), &slot, sizeof(slot));
17,034✔
2985
  if (code != TSDB_CODE_SUCCESS) {
17,096!
2986
    // we have checked the existence of uid in hash map above
2987
    QUERY_CHECK_CONDITION((code != TSDB_CODE_DUP_KEY), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
2988
    taosArrayPopTailBatch(pTableList->pTableList, 1);  // let's pop the last element in the array list
×
2989
  }
2990

2991
_end:
17,096✔
2992
  if (code != TSDB_CODE_SUCCESS) {
17,093!
2993
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2994
  } else {
2995
    qDebug("uid:%" PRIu64 ", groupId:%" PRIu64 " added into table list, slot:%d, total:%d", uid, gid, slot, slot + 1);
17,093✔
2996
  }
2997

2998
  return code;
17,087✔
2999
}
3000

3001
int32_t tableListGetGroupList(const STableListInfo* pTableList, int32_t ordinalGroupIndex, STableKeyInfo** pKeyInfo,
4,756,518✔
3002
                              int32_t* size) {
3003
  int32_t totalGroups = tableListGetOutputGroups(pTableList);
4,756,518✔
3004
  int32_t numOfTables = 0;
4,763,598✔
3005
  int32_t code = tableListGetSize(pTableList, &numOfTables);
4,763,598✔
3006
  if (code != TSDB_CODE_SUCCESS) {
4,756,680!
3007
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3008
    return code;
×
3009
  }
3010

3011
  if (ordinalGroupIndex < 0 || ordinalGroupIndex >= totalGroups) {
4,756,680!
3012
    return TSDB_CODE_INVALID_PARA;
×
3013
  }
3014

3015
  // here handle two special cases:
3016
  // 1. only one group exists, and 2. one table exists for each group.
3017
  if (totalGroups == 1) {
4,756,680✔
3018
    *size = numOfTables;
4,754,391✔
3019
    *pKeyInfo = (*size == 0) ? NULL : taosArrayGet(pTableList->pTableList, 0);
4,754,391✔
3020
    return TSDB_CODE_SUCCESS;
4,749,220✔
3021
  } else if (totalGroups == numOfTables) {
2,289!
3022
    *size = 1;
2,403✔
3023
    *pKeyInfo = taosArrayGet(pTableList->pTableList, ordinalGroupIndex);
2,403✔
3024
    return TSDB_CODE_SUCCESS;
2,403✔
3025
  }
3026

3027
  int32_t offset = pTableList->groupOffset[ordinalGroupIndex];
×
3028
  if (ordinalGroupIndex < totalGroups - 1) {
×
3029
    *size = pTableList->groupOffset[ordinalGroupIndex + 1] - offset;
64✔
3030
  } else {
3031
    *size = numOfTables - offset;
×
3032
  }
3033

3034
  *pKeyInfo = taosArrayGet(pTableList->pTableList, offset);
×
3035
  return TSDB_CODE_SUCCESS;
107✔
3036
}
3037

3038
int32_t tableListGetOutputGroups(const STableListInfo* pTableList) { return pTableList->numOfOuputGroups; }
14,080,427✔
3039

3040
bool oneTableForEachGroup(const STableListInfo* pTableList) { return pTableList->oneTableForEachGroup; }
10,784✔
3041

3042
STableListInfo* tableListCreate() {
8,698,262✔
3043
  STableListInfo* pListInfo = taosMemoryCalloc(1, sizeof(STableListInfo));
8,698,262!
3044
  if (pListInfo == NULL) {
8,714,359!
3045
    return NULL;
×
3046
  }
3047

3048
  pListInfo->remainGroups = NULL;
8,714,359✔
3049
  pListInfo->pTableList = taosArrayInit(4, sizeof(STableKeyInfo));
8,714,359✔
3050
  if (pListInfo->pTableList == NULL) {
8,714,797!
3051
    goto _error;
×
3052
  }
3053

3054
  pListInfo->map = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
8,714,797✔
3055
  if (pListInfo->map == NULL) {
8,711,273!
3056
    goto _error;
×
3057
  }
3058

3059
  pListInfo->numOfOuputGroups = 1;
8,711,273✔
3060
  return pListInfo;
8,711,273✔
3061

3062
_error:
×
3063
  tableListDestroy(pListInfo);
×
3064
  return NULL;
×
3065
}
3066

3067
void tableListDestroy(STableListInfo* pTableListInfo) {
8,868,853✔
3068
  if (pTableListInfo == NULL) {
8,868,853✔
3069
    return;
151,917✔
3070
  }
3071

3072
  taosArrayDestroy(pTableListInfo->pTableList);
8,716,936✔
3073
  taosMemoryFreeClear(pTableListInfo->groupOffset);
8,718,027!
3074

3075
  taosHashCleanup(pTableListInfo->map);
8,718,029✔
3076
  taosHashCleanup(pTableListInfo->remainGroups);
8,717,122✔
3077
  pTableListInfo->pTableList = NULL;
8,717,485✔
3078
  pTableListInfo->map = NULL;
8,717,485✔
3079
  taosMemoryFree(pTableListInfo);
8,717,485!
3080
}
3081

3082
void tableListClear(STableListInfo* pTableListInfo) {
2,515✔
3083
  if (pTableListInfo == NULL) {
2,515!
3084
    return;
×
3085
  }
3086

3087
  taosArrayClear(pTableListInfo->pTableList);
2,515✔
3088
  taosHashClear(pTableListInfo->map);
2,515✔
3089
  taosHashClear(pTableListInfo->remainGroups);
2,516✔
3090
  taosMemoryFree(pTableListInfo->groupOffset);
2,516!
3091
  pTableListInfo->numOfOuputGroups = 1;
2,516✔
3092
  pTableListInfo->oneTableForEachGroup = false;
2,516✔
3093
}
3094

3095
static int32_t orderbyGroupIdComparFn(const void* p1, const void* p2) {
8,697,800✔
3096
  STableKeyInfo* pInfo1 = (STableKeyInfo*)p1;
8,697,800✔
3097
  STableKeyInfo* pInfo2 = (STableKeyInfo*)p2;
8,697,800✔
3098

3099
  if (pInfo1->groupId == pInfo2->groupId) {
8,697,800✔
3100
    return 0;
7,673,441✔
3101
  } else {
3102
    return pInfo1->groupId < pInfo2->groupId ? -1 : 1;
1,024,359!
3103
  }
3104
}
3105

3106
static int32_t sortTableGroup(STableListInfo* pTableListInfo) {
3,761,298✔
3107
  taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
3,761,298✔
3108
  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
3,766,073✔
3109

3110
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
3,765,208✔
3111
  if (!pList) {
3,765,535!
3112
    return terrno;
×
3113
  }
3114

3115
  STableKeyInfo* pInfo = taosArrayGet(pTableListInfo->pTableList, 0);
3,765,535✔
3116
  if (!pInfo) {
3,760,615!
3117
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3118
    return terrno;
×
3119
  }
3120
  uint64_t gid = pInfo->groupId;
3,760,615✔
3121

3122
  int32_t start = 0;
3,760,615✔
3123
  void*   tmp = taosArrayPush(pList, &start);
3,760,835✔
3124
  if (!tmp) {
3,760,835!
3125
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3126
    return terrno;
×
3127
  }
3128

3129
  for (int32_t i = 1; i < size; ++i) {
9,802,559✔
3130
    pInfo = taosArrayGet(pTableListInfo->pTableList, i);
6,043,977✔
3131
    if (!pInfo) {
6,041,763✔
3132
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
68!
3133
      return terrno;
68✔
3134
    }
3135
    if (pInfo->groupId != gid) {
6,041,695✔
3136
      tmp = taosArrayPush(pList, &i);
233,383✔
3137
      if (!tmp) {
233,383!
3138
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3139
        return terrno;
×
3140
      }
3141
      gid = pInfo->groupId;
233,383✔
3142
    }
3143
  }
3144

3145
  pTableListInfo->numOfOuputGroups = taosArrayGetSize(pList);
3,758,582✔
3146
  pTableListInfo->groupOffset = taosMemoryMalloc(sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
3,757,426!
3147
  if (pTableListInfo->groupOffset == NULL) {
3,764,718!
3148
    taosArrayDestroy(pList);
×
3149
    return terrno;
×
3150
  }
3151

3152
  memcpy(pTableListInfo->groupOffset, taosArrayGet(pList, 0), sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
3,764,718✔
3153
  taosArrayDestroy(pList);
3,760,458✔
3154
  return TSDB_CODE_SUCCESS;
3,767,834✔
3155
}
3156

3157
int32_t buildGroupIdMapForAllTables(STableListInfo* pTableListInfo, SReadHandle* pHandle, SScanPhysiNode* pScanNode,
8,161,677✔
3158
                                    SNodeList* group, bool groupSort, uint8_t* digest, SStorageAPI* pAPI, SHashObj* groupIdMap) {
3159
  int32_t code = TSDB_CODE_SUCCESS;
8,161,677✔
3160

3161
  bool   groupByTbname = groupbyTbname(group);
8,161,677✔
3162
  size_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
8,162,264✔
3163
  if (!numOfTables) {
8,160,972!
3164
    return code;
×
3165
  }
3166
  qDebug("numOfTables:%zu, groupByTbname:%d, group:%p", numOfTables, groupByTbname, group);
8,160,972✔
3167
  if (group == NULL || groupByTbname) {
8,160,520✔
3168
    if (tsCountAlwaysReturnValue && QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode) &&
8,080,027✔
3169
        ((STableScanPhysiNode*)pScanNode)->needCountEmptyTable) {
4,311,285✔
3170
      pTableListInfo->remainGroups =
301,655✔
3171
          taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
301,654✔
3172
      if (pTableListInfo->remainGroups == NULL) {
301,655!
3173
        return terrno;
×
3174
      }
3175

3176
      for (int i = 0; i < numOfTables; i++) {
907,786✔
3177
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
606,116✔
3178
        if (!info) {
606,108✔
3179
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
14!
3180
          return terrno;
14✔
3181
        }
3182
        info->groupId = groupByTbname ? info->uid : 0;
606,094✔
3183
        int32_t tempRes = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId),
606,094✔
3184
                                      &(info->uid), sizeof(info->uid));
606,094✔
3185
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
606,131!
3186
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3187
          return tempRes;
×
3188
        }
3189
      }
3190
    } else {
3191
      for (int32_t i = 0; i < numOfTables; i++) {
31,452,306✔
3192
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
23,679,283✔
3193
        if (!info) {
23,673,125!
3194
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3195
          return terrno;
×
3196
        }
3197
        info->groupId = groupByTbname ? info->uid : 0;
23,673,933✔
3198
        
3199
      }
3200
    }
3201
    if (groupIdMap && group != NULL){
8,074,693✔
3202
      getColInfoResultForGroupbyForStream(pHandle->vnode, group, pTableListInfo, pAPI, groupIdMap);
32,815✔
3203
    }
3204

3205
    pTableListInfo->oneTableForEachGroup = groupByTbname;
8,076,815✔
3206
    if (numOfTables == 1 && pTableListInfo->idInfo.tableType == TSDB_CHILD_TABLE) {
8,076,815✔
3207
      pTableListInfo->oneTableForEachGroup = true;
687,204✔
3208
    }
3209

3210
    if (groupSort && groupByTbname) {
8,076,815✔
3211
      taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
39,146✔
3212
      pTableListInfo->numOfOuputGroups = numOfTables;
39,170✔
3213
    } else if (groupByTbname && pScanNode->groupOrderScan) {
8,037,669✔
3214
      pTableListInfo->numOfOuputGroups = numOfTables;
143✔
3215
    } else {
3216
      pTableListInfo->numOfOuputGroups = 1;
8,037,526✔
3217
    }
3218
    if (groupSort || pScanNode->groupOrderScan) {
8,076,839✔
3219
      code = sortTableGroup(pTableListInfo);
3,756,463✔
3220
    }
3221
  } else {
3222
    bool initRemainGroups = false;
80,493✔
3223
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode)) {
80,493✔
3224
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pScanNode;
75,048✔
3225
      if (tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable &&
75,048✔
3226
          !(groupSort || pScanNode->groupOrderScan)) {
13,753!
3227
        initRemainGroups = true;
13,709✔
3228
      }
3229
    }
3230

3231
    code = getColInfoResultForGroupby(pHandle->vnode, group, pTableListInfo, digest, pAPI, initRemainGroups, groupIdMap);
80,493✔
3232
    if (code != TSDB_CODE_SUCCESS) {
80,478✔
3233
      return code;
15✔
3234
    }
3235

3236
    if (pScanNode->groupOrderScan) pTableListInfo->numOfOuputGroups = taosArrayGetSize(pTableListInfo->pTableList);
80,463✔
3237

3238
    if (groupSort || pScanNode->groupOrderScan) {
80,463✔
3239
      code = sortTableGroup(pTableListInfo);
5,229✔
3240
    }
3241
  }
3242

3243
  // add all table entry in the hash map
3244
  size_t size = taosArrayGetSize(pTableListInfo->pTableList);
8,163,464✔
3245
  for (int32_t i = 0; i < size; ++i) {
32,784,723✔
3246
    STableKeyInfo* p = taosArrayGet(pTableListInfo->pTableList, i);
24,605,093✔
3247
    if (!p) {
24,544,621!
3248
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3249
      return terrno;
×
3250
    }
3251
    int32_t tempRes = taosHashPut(pTableListInfo->map, &p->uid, sizeof(uint64_t), &i, sizeof(int32_t));
24,544,621✔
3252
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
24,626,552!
3253
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3254
      return tempRes;
×
3255
    }
3256
  }
3257

3258
  return code;
8,179,630✔
3259
}
3260

3261
int32_t createScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
8,551,258✔
3262
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
3263
                                SExecTaskInfo* pTaskInfo, SHashObj* groupIdMap) {
3264
  int64_t     st = taosGetTimestampUs();
8,555,200✔
3265
  const char* idStr = GET_TASKID(pTaskInfo);
8,555,200✔
3266

3267
  if (pHandle == NULL) {
8,555,200!
3268
    qError("invalid handle, in creating operator tree, %s", idStr);
×
3269
    return TSDB_CODE_INVALID_PARA;
×
3270
  }
3271

3272
  if (pHandle->uid != 0) {
8,555,200✔
3273
    pScanNode->uid = pHandle->uid;
167✔
3274
    pScanNode->tableType = TSDB_CHILD_TABLE;
167✔
3275
  }
3276
  uint8_t digest[17] = {0};
8,555,200✔
3277
  int32_t code = getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, digest, idStr,
8,555,200✔
3278
                              &pTaskInfo->storageAPI);
3279
  if (code != TSDB_CODE_SUCCESS) {
8,562,126✔
3280
    qError("failed to getTableList, code:%s", tstrerror(code));
8!
3281
    return code;
8✔
3282
  }
3283

3284
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
8,562,118✔
3285

3286
  int64_t st1 = taosGetTimestampUs();
8,560,698✔
3287
  pTaskInfo->cost.extractListTime = (st1 - st) / 1000.0;
8,560,698✔
3288
  qDebug("extract queried table list completed, %d tables, elapsed time:%.2f ms %s", numOfTables,
8,560,698✔
3289
         pTaskInfo->cost.extractListTime, idStr);
3290

3291
  if (numOfTables == 0) {
8,563,459✔
3292
    qDebug("no table qualified for query, %s" PRIx64, idStr);
397,256✔
3293
    return TSDB_CODE_SUCCESS;
397,268✔
3294
  }
3295

3296
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, digest, &pTaskInfo->storageAPI, groupIdMap);
8,166,203✔
3297
  if (code != TSDB_CODE_SUCCESS) {
8,167,862✔
3298
    return code;
15✔
3299
  }
3300

3301
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
8,168,820✔
3302
  qDebug("generate group id map completed, elapsed time:%.2f ms %s", pTaskInfo->cost.groupIdMapTime, idStr);
8,168,820✔
3303

3304
  return TSDB_CODE_SUCCESS;
8,169,600✔
3305
}
3306

3307
char* getStreamOpName(uint16_t opType) {
24,341✔
3308
  switch (opType) {
24,341!
3309
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
×
3310
      return "stream scan";
×
3311
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
24,071✔
3312
      return "project";
24,071✔
3313
    case QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW:
270✔
3314
      return "external window";
270✔
3315
  }
3316
  return "error name";
×
3317
}
3318

3319
void printDataBlock(SSDataBlock* pBlock, const char* flag, const char* taskIdStr) {
454,966✔
3320
  if (!pBlock) {
454,966✔
3321
    qDebug("%s===stream===%s: Block is Null", taskIdStr, flag);
23,779✔
3322
    return;
23,779✔
3323
  } else if (pBlock->info.rows == 0) {
431,187✔
3324
    qDebug("%s===stream===%s: Block is Empty. block type %d", taskIdStr, flag, pBlock->info.type);
249,751✔
3325
    return;
249,751✔
3326
  }
3327
  if (qDebugFlag & DEBUG_TRACE) {
181,436✔
3328
    char*   pBuf = NULL;
33,839✔
3329
    int32_t code = dumpBlockData(pBlock, flag, &pBuf, taskIdStr);
33,839✔
3330
    if (code == 0) {
33,838!
3331
      qDebugL("%s", pBuf);
33,838!
3332
      taosMemoryFree(pBuf);
33,840!
3333
    }
3334
  }
3335
}
3336

3337
void printSpecDataBlock(SSDataBlock* pBlock, const char* flag, const char* opStr, const char* taskIdStr) {
×
3338
  if (!pBlock) {
×
3339
    qDebug("%s===stream===%s %s: Block is Null", taskIdStr, flag, opStr);
×
3340
    return;
×
3341
  } else if (pBlock->info.rows == 0) {
×
3342
    qDebug("%s===stream===%s %s: Block is Empty. block type %d.skey:%" PRId64 ",ekey:%" PRId64 ",version%" PRId64,
×
3343
           taskIdStr, flag, opStr, pBlock->info.type, pBlock->info.window.skey, pBlock->info.window.ekey,
3344
           pBlock->info.version);
3345
    return;
×
3346
  }
3347
  if (qDebugFlag & DEBUG_DEBUG) {
×
3348
    char* pBuf = NULL;
×
3349
    char  flagBuf[64];
3350
    snprintf(flagBuf, sizeof(flagBuf), "%s %s", flag, opStr);
×
3351
    int32_t code = dumpBlockData(pBlock, flagBuf, &pBuf, taskIdStr);
×
3352
    if (code == 0) {
×
3353
      qDebug("%s", pBuf);
×
3354
      taosMemoryFree(pBuf);
×
3355
    }
3356
  }
3357
}
3358

3359
TSKEY getStartTsKey(STimeWindow* win, const TSKEY* tsCols) { return tsCols == NULL ? win->skey : tsCols[0]; }
4,478,377!
3360

3361
void updateTimeWindowInfo(SColumnInfoData* pColData, const STimeWindow* pWin, int64_t delta) {
660,269,083✔
3362
  int64_t* ts = (int64_t*)pColData->pData;
660,269,083✔
3363

3364
  int64_t duration = pWin->ekey - pWin->skey + delta;
660,269,083✔
3365
  ts[2] = duration;            // set the duration
660,269,083✔
3366
  ts[3] = pWin->skey;          // window start key
660,269,083✔
3367
  ts[4] = pWin->ekey + delta;  // window end key
660,269,083✔
3368
}
660,269,083✔
3369

3370
int32_t compKeys(const SArray* pSortGroupCols, const char* oldkeyBuf, int32_t oldKeysLen, const SSDataBlock* pBlock,
19,469,804✔
3371
                 int32_t rowIndex) {
3372
  SColumnDataAgg* pColAgg = NULL;
19,469,804✔
3373
  const char*     isNull = oldkeyBuf;
19,469,804✔
3374
  const char*     p = oldkeyBuf + sizeof(int8_t) * pSortGroupCols->size;
19,469,804✔
3375

3376
  for (int32_t i = 0; i < pSortGroupCols->size; ++i) {
48,932,320✔
3377
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
29,475,188✔
3378
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
29,475,188✔
3379
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
29,475,188!
3380

3381
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
58,950,376!
3382
      if (isNull[i] != 1) return 1;
1,534,313✔
3383
    } else {
3384
      if (isNull[i] != 0) return 1;
27,940,875✔
3385
      const char* val = colDataGetData(pColInfoData, rowIndex);
27,940,531!
3386
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
27,940,531!
3387
        int32_t len = getJsonValueLen(val);
×
3388
        if (memcmp(p, val, len) != 0) return 1;
×
3389
        p += len;
×
3390
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
27,940,531!
3391
        if (IS_STR_DATA_BLOB(pCol->type)) {
10,068,012!
3392
          if (memcmp(p, val, blobDataTLen(val)) != 0) return 1;
×
3393
          p += blobDataTLen(val);
×
3394
        } else {
3395
          if (memcmp(p, val, varDataTLen(val)) != 0) return 1;
10,290,414✔
3396
          p += varDataTLen(val);
10,285,694✔
3397
        }
3398
      } else {
3399
        if (0 != memcmp(p, val, pCol->bytes)) return 1;
17,872,519✔
3400
        p += pCol->bytes;
17,865,039✔
3401
      }
3402
    }
3403
  }
3404
  if ((int32_t)(p - oldkeyBuf) != oldKeysLen) return 1;
19,457,132✔
3405
  return 0;
19,456,852✔
3406
}
3407

3408
int32_t buildKeys(char* keyBuf, const SArray* pSortGroupCols, const SSDataBlock* pBlock, int32_t rowIndex) {
12,952✔
3409
  uint32_t        colNum = pSortGroupCols->size;
12,952✔
3410
  SColumnDataAgg* pColAgg = NULL;
12,952✔
3411
  char*           isNull = keyBuf;
12,952✔
3412
  char*           p = keyBuf + sizeof(int8_t) * colNum;
12,952✔
3413

3414
  for (int32_t i = 0; i < colNum; ++i) {
35,224✔
3415
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
22,272✔
3416
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
22,272✔
3417
    if (pCol->slotId > pBlock->pDataBlock->size) continue;
22,272!
3418

3419
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
22,272!
3420

3421
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
44,544!
3422
      isNull[i] = 1;
704✔
3423
    } else {
3424
      isNull[i] = 0;
21,568✔
3425
      const char* val = colDataGetData(pColInfoData, rowIndex);
21,568!
3426
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
21,568!
3427
        int32_t len = getJsonValueLen(val);
×
3428
        memcpy(p, val, len);
×
3429
        p += len;
×
3430
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
21,568!
3431
        if (IS_STR_DATA_BLOB(pCol->type)) {
8,648!
3432
          blobDataCopy(p, val);
×
3433
          p += blobDataTLen(val);
×
3434
        } else {
3435
          varDataCopy(p, val);
8,648✔
3436
          p += varDataTLen(val);
8,648✔
3437
        }
3438
      } else {
3439
        memcpy(p, val, pCol->bytes);
12,920✔
3440
        p += pCol->bytes;
12,920✔
3441
      }
3442
    }
3443
  }
3444
  return (int32_t)(p - keyBuf);
12,952✔
3445
}
3446

3447
uint64_t calcGroupId(char* pData, int32_t len) {
42,527,288✔
3448
  T_MD5_CTX context;
3449
  tMD5Init(&context);
42,527,288✔
3450
  tMD5Update(&context, (uint8_t*)pData, len);
42,504,616✔
3451
  tMD5Final(&context);
42,660,320✔
3452

3453
  // NOTE: only extract the initial 8 bytes of the final MD5 digest
3454
  uint64_t id = 0;
42,756,923✔
3455
  memcpy(&id, context.digest, sizeof(uint64_t));
42,756,923✔
3456
  if (0 == id) memcpy(&id, context.digest + 8, sizeof(uint64_t));
42,756,923!
3457
  return id;
42,756,923✔
3458
}
3459

3460
SNodeList* makeColsNodeArrFromSortKeys(SNodeList* pSortKeys) {
830✔
3461
  SNode*     node;
3462
  SNodeList* ret = NULL;
830✔
3463
  FOREACH(node, pSortKeys) {
2,558!
3464
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)node;
1,726✔
3465
    int32_t           code = nodesListMakeAppend(&ret, pSortKey->pExpr);
1,726✔
3466
    if (code != TSDB_CODE_SUCCESS) {
1,728!
3467
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3468
      terrno = code;
×
3469
      return NULL;
×
3470
    }
3471
  }
3472
  return ret;
832✔
3473
}
3474

3475
int32_t extractKeysLen(const SArray* keys, int32_t* pLen) {
830✔
3476
  int32_t code = TSDB_CODE_SUCCESS;
830✔
3477
  int32_t lino = 0;
830✔
3478
  int32_t len = 0;
830✔
3479
  int32_t keyNum = taosArrayGetSize(keys);
830✔
3480
  for (int32_t i = 0; i < keyNum; ++i) {
2,076✔
3481
    SColumn* pCol = (SColumn*)taosArrayGet(keys, i);
1,245✔
3482
    QUERY_CHECK_NULL(pCol, code, lino, _end, terrno);
1,246!
3483
    len += pCol->bytes;
1,246✔
3484
  }
3485
  len += sizeof(int8_t) * keyNum;  // null flag
831✔
3486
  *pLen = len;
831✔
3487

3488
_end:
831✔
3489
  if (code != TSDB_CODE_SUCCESS) {
831!
3490
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3491
  }
3492
  return code;
832✔
3493
}
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