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

taosdata / TDengine / #4911

04 Jan 2026 09:05AM UTC coverage: 65.028% (-0.8%) from 65.864%
#4911

push

travis-ci

web-flow
merge: from main to 3.0 branch #34156

1206 of 4524 new or added lines in 22 files covered. (26.66%)

1517 existing lines in 134 files now uncovered.

195276 of 300296 relevant lines covered (65.03%)

116931714.52 hits per line

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

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

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

30
#include "executil.h"
31
#include "executorInt.h"
32
#include "querytask.h"
33
#include "storageapi.h"
34
#include "tutil.h"
35
#include "tjson.h"
36
#include "trpc.h"
37
#include "filter.h"
38

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

46
typedef struct STransTagExprCtx {
47
  int32_t      code;
48
  SMetaReader* pReader;
49
} STransTagExprCtx;
50

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

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

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

65
static int64_t getLimit(const SNode* pLimit) {
1,048,908,138✔
66
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->limit) ? -1 : ((SLimitNode*)pLimit)->limit->datum.i;
1,048,908,138✔
67
}
68
static int64_t getOffset(const SNode* pLimit) {
1,048,823,080✔
69
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->offset) ? -1 : ((SLimitNode*)pLimit)->offset->datum.i;
1,048,823,080✔
70
}
71
static void releaseColInfoData(void* pCol);
72

73
void initResultRowInfo(SResultRowInfo* pResultRowInfo) {
431,929,245✔
74
  pResultRowInfo->size = 0;
431,929,245✔
75
  pResultRowInfo->cur.pageId = -1;
431,963,910✔
76
}
432,062,021✔
77

78
void closeResultRow(SResultRow* pResultRow) { pResultRow->closed = true; }
5,202,426✔
79

80
void resetResultRow(SResultRow* pResultRow, size_t entrySize) {
840,283,835✔
81
  pResultRow->numOfRows = 0;
840,283,835✔
82
  pResultRow->closed = false;
840,284,204✔
83
  pResultRow->endInterp = false;
840,284,204✔
84
  pResultRow->startInterp = false;
840,284,204✔
85

86
  if (entrySize > 0) {
840,284,942✔
87
    memset(pResultRow->pEntryInfo, 0, entrySize);
840,284,942✔
88
  }
89
}
840,289,370✔
90

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

96
size_t getResultRowSize(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
252,554,927✔
97
  int32_t rowSize = (numOfOutput * sizeof(SResultRowEntryInfo)) + sizeof(SResultRow);
252,554,927✔
98

99
  for (int32_t i = 0; i < numOfOutput; ++i) {
867,175,818✔
100
    rowSize += pCtx[i].resDataInfo.interBufSize;
614,655,461✔
101
  }
102

103
  return rowSize;
252,520,357✔
104
}
105

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

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

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

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

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

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

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

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

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

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

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

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

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

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

219
void cleanupGroupResInfo(SGroupResInfo* pGroupResInfo) {
127,201,632✔
220
  taosMemoryFreeClear(pGroupResInfo->pBuf);
127,201,632✔
221
  if (pGroupResInfo->freeItem) {
127,204,331✔
222
    //    taosArrayDestroy(pGroupResInfo->pRows);
223
    taosArrayDestroyEx(pGroupResInfo->pRows, freeEx);
×
224
    pGroupResInfo->freeItem = false;
×
225
    pGroupResInfo->pRows = NULL;
×
226
  } else {
227
    taosArrayDestroy(pGroupResInfo->pRows);
127,201,356✔
228
    pGroupResInfo->pRows = NULL;
127,198,294✔
229
  }
230
  pGroupResInfo->index = 0;
127,201,205✔
231
  pGroupResInfo->delIndex = 0;
127,199,631✔
232
}
127,197,710✔
233

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

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

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

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

254
int32_t initGroupedResultInfo(SGroupResInfo* pGroupResInfo, SSHashObj* pHashmap, int32_t order) {
86,198,256✔
255
  int32_t code = TSDB_CODE_SUCCESS;
86,198,256✔
256
  int32_t lino = 0;
86,198,256✔
257
  if (pGroupResInfo->pRows != NULL) {
86,198,256✔
258
    taosArrayDestroy(pGroupResInfo->pRows);
638,828✔
259
  }
260
  if (pGroupResInfo->pBuf) {
86,201,963✔
261
    taosMemoryFree(pGroupResInfo->pBuf);
638,828✔
262
    pGroupResInfo->pBuf = NULL;
638,828✔
263
  }
264

265
  // extract the result rows information from the hash map
266
  int32_t size = tSimpleHashGetSize(pHashmap);
86,198,004✔
267

268
  void* pData = NULL;
86,199,147✔
269
  pGroupResInfo->pRows = taosArrayInit(size, POINTER_BYTES);
86,199,147✔
270
  QUERY_CHECK_NULL(pGroupResInfo->pRows, code, lino, _end, terrno);
86,199,630✔
271

272
  size_t  keyLen = 0;
86,196,808✔
273
  int32_t iter = 0;
86,198,277✔
274
  int64_t bufLen = 0, offset = 0;
86,199,333✔
275

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

282
  pGroupResInfo->pBuf = taosMemoryMalloc(bufLen);
86,160,141✔
283
  QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
86,199,688✔
284

285
  iter = 0;
86,198,037✔
286
  while ((pData = tSimpleHashIterate(pHashmap, pData, &iter)) != NULL) {
2,147,483,647✔
287
    void* key = tSimpleHashGetKey(pData, &keyLen);
2,147,483,647✔
288

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

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

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

300
  if (order == TSDB_ORDER_ASC || order == TSDB_ORDER_DESC) {
86,188,064✔
301
    __compar_fn_t fn = (order == TSDB_ORDER_ASC) ? resultrowComparAsc : resultrowComparDesc;
12,765,208✔
302
    size = POINTER_BYTES;
12,765,208✔
303
    taosSort(pGroupResInfo->pRows->pData, taosArrayGetSize(pGroupResInfo->pRows), size, fn);
12,765,208✔
304
  }
305

306
  pGroupResInfo->index = 0;
86,187,722✔
307

308
_end:
86,197,784✔
309
  if (code != TSDB_CODE_SUCCESS) {
86,197,174✔
310
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
311
  }
312
  return code;
86,197,174✔
313
}
314

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

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

326
bool hasRemainResults(SGroupResInfo* pGroupResInfo) {
334,432,586✔
327
  if (pGroupResInfo->pRows == NULL) {
334,432,586✔
328
    return false;
×
329
  }
330

331
  return pGroupResInfo->index < taosArrayGetSize(pGroupResInfo->pRows);
334,434,299✔
332
}
333

334
int32_t getNumOfTotalRes(SGroupResInfo* pGroupResInfo) {
178,913,053✔
335
  if (pGroupResInfo->pRows == 0) {
178,913,053✔
336
    return 0;
×
337
  }
338

339
  return (int32_t)taosArrayGetSize(pGroupResInfo->pRows);
178,918,321✔
340
}
341

342
SArray* createSortInfo(SNodeList* pNodeList) {
56,034,776✔
343
  size_t numOfCols = 0;
56,034,776✔
344

345
  if (pNodeList != NULL) {
56,034,776✔
346
    numOfCols = LIST_LENGTH(pNodeList);
55,831,794✔
347
  } else {
348
    numOfCols = 0;
202,983✔
349
  }
350

351
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
56,034,982✔
352
  if (pList == NULL) {
56,028,219✔
353
    return pList;
×
354
  }
355

356
  for (int32_t i = 0; i < numOfCols; ++i) {
127,095,597✔
357
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
71,058,327✔
358
    if (!pSortKey) {
71,060,736✔
359
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
360
      taosArrayDestroy(pList);
×
361
      pList = NULL;
×
362
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
363
      break;
×
364
    }
365
    SBlockOrderInfo bi = {0};
71,060,736✔
366
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
71,062,145✔
367
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);
71,062,145✔
368

369
    if (nodeType(pSortKey->pExpr) != QUERY_NODE_COLUMN) {
71,061,513✔
370
      qError("invalid order by expr type:%d", nodeType(pSortKey->pExpr));
×
371
      taosArrayDestroy(pList);
×
372
      pList = NULL;
×
373
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
374
      break;
×
375
    }
376
    
377
    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
71,048,945✔
378
    bi.slotId = pColNode->slotId;
71,061,866✔
379
    void* tmp = taosArrayPush(pList, &bi);
71,067,378✔
380
    if (!tmp) {
71,067,378✔
381
      taosArrayDestroy(pList);
×
382
      pList = NULL;
×
383
      break;
×
384
    }
385
  }
386

387
  return pList;
56,037,270✔
388
}
389

390
SSDataBlock* createDataBlockFromDescNode(void* p) {
672,096,681✔
391
  SDataBlockDescNode* pNode = (SDataBlockDescNode*)p;
672,096,681✔
392
  int32_t      numOfCols = LIST_LENGTH(pNode->pSlots);
672,096,681✔
393
  SSDataBlock* pBlock = NULL;
672,132,048✔
394
  int32_t      code = createDataBlock(&pBlock);
672,136,658✔
395
  if (code) {
672,074,466✔
396
    terrno = code;
×
397
    return NULL;
×
398
  }
399

400
  pBlock->info.id.blockId = pNode->dataBlockId;
672,074,466✔
401
  pBlock->info.type = STREAM_INVALID;
672,074,507✔
402
  pBlock->info.calWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
672,141,142✔
403
  pBlock->info.watermark = INT64_MIN;
672,115,702✔
404

405
  for (int32_t i = 0; i < numOfCols; ++i) {
2,147,483,647✔
406
    SSlotDescNode* pDescNode = (SSlotDescNode*)nodesListGetNode(pNode->pSlots, i);
2,023,121,273✔
407
    if (!pDescNode) {
2,023,112,126✔
408
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
409
      blockDataDestroy(pBlock);
×
410
      pBlock = NULL;
×
411
      terrno = TSDB_CODE_INVALID_PARA;
×
412
      break;
×
413
    }
414
    SColumnInfoData idata =
2,023,078,849✔
415
        createColumnInfoData(pDescNode->dataType.type, pDescNode->dataType.bytes, pDescNode->slotId);
2,023,205,023✔
416
    idata.info.scale = pDescNode->dataType.scale;
2,023,260,879✔
417
    idata.info.precision = pDescNode->dataType.precision;
2,023,316,694✔
418
    idata.info.noData = pDescNode->reserve;
2,023,330,960✔
419

420
    code = blockDataAppendColInfo(pBlock, &idata);
2,023,318,740✔
421
    if (code != TSDB_CODE_SUCCESS) {
2,023,252,778✔
422
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
107,813✔
423
      blockDataDestroy(pBlock);
107,813✔
424
      pBlock = NULL;
×
425
      terrno = code;
×
426
      break;
×
427
    }
428
  }
429

430
  return pBlock;
672,312,097✔
431
}
432

433
int32_t prepareDataBlockBuf(SSDataBlock* pDataBlock, SColMatchInfo* pMatchInfo) {
230,172,973✔
434
  SDataBlockInfo* pBlockInfo = &pDataBlock->info;
230,172,973✔
435

436
  for (int32_t i = 0; i < taosArrayGetSize(pMatchInfo->pList); ++i) {
799,598,911✔
437
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
586,778,278✔
438
    if (!pItem) {
586,823,627✔
439
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
440
      return terrno;
×
441
    }
442

443
    if (pItem->isPk) {
586,823,627✔
444
      SColumnInfoData* pInfoData = taosArrayGet(pDataBlock->pDataBlock, pItem->dstSlotId);
17,462,449✔
445
      if (!pInfoData) {
17,422,693✔
446
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
447
        return terrno;
×
448
      }
449
      pBlockInfo->pks[0].type = pInfoData->info.type;
17,422,693✔
450
      pBlockInfo->pks[1].type = pInfoData->info.type;
17,431,800✔
451

452
      // allocate enough buffer size, which is pInfoData->info.bytes
453
      if (IS_VAR_DATA_TYPE(pItem->dataType.type)) {
17,436,057✔
454
        pBlockInfo->pks[0].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
5,336,830✔
455
        if (pBlockInfo->pks[0].pData == NULL) {
5,330,850✔
456
          return terrno;
×
457
        }
458

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

465
        pBlockInfo->pks[0].nData = pInfoData->info.bytes;
5,331,464✔
466
        pBlockInfo->pks[1].nData = pInfoData->info.bytes;
5,334,422✔
467
      }
468

469
      break;
17,433,076✔
470
    }
471
  }
472

473
  return TSDB_CODE_SUCCESS;
230,227,221✔
474
}
475

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

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

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

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

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

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

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

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

560
  return DEAL_RES_CONTINUE;
2,235,184✔
561
}
562

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

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

573
    return TSDB_CODE_SUCCESS;
×
574
  }
575

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

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

600
    return code;
×
601
  }
602

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

606
  nodesDestroyNode(pNew);
319,312✔
607
  return TSDB_CODE_SUCCESS;
319,312✔
608
}
609

610
static EDealRes getColumn(SNode** pNode, void* pContext) {
50,920,665✔
611
  tagFilterAssist* pData = (tagFilterAssist*)pContext;
50,920,665✔
612
  SColumnNode*     pSColumnNode = NULL;
50,920,665✔
613
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
50,925,107✔
614
    pSColumnNode = *(SColumnNode**)pNode;
17,502,712✔
615
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
33,441,824✔
616
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
604,598✔
617
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
604,598✔
618
      pData->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
560,009✔
619
      if (NULL == pSColumnNode) {
560,009✔
620
        return DEAL_RES_ERROR;
×
621
      }
622
      pSColumnNode->colId = -1;
560,009✔
623
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
560,009✔
624
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
560,009✔
625
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
560,009✔
626
      nodesDestroyNode(*pNode);
560,009✔
627
      *pNode = (SNode*)pSColumnNode;
560,009✔
628
    } else {
629
      return DEAL_RES_CONTINUE;
44,589✔
630
    }
631
  } else {
632
    return DEAL_RES_CONTINUE;
32,831,475✔
633
  }
634

635
  void* data = taosHashGet(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
18,058,716✔
636
  if (!data) {
18,055,263✔
637
    int32_t tempRes =
638
        taosHashPut(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
15,476,742✔
639
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
15,481,523✔
640
      return DEAL_RES_ERROR;
×
641
    }
642
    pSColumnNode->slotId = pData->index++;
15,481,523✔
643
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
15,477,000✔
644
                         .type = pSColumnNode->node.resType.type,
15,478,028✔
645
                         .bytes = pSColumnNode->node.resType.bytes,
15,472,865✔
646
                         .pk = pSColumnNode->isPk};
15,477,750✔
647
#if TAG_FILTER_DEBUG
648
    qDebug("tagfilter build column info, slotId:%d, colId:%d, type:%d", pSColumnNode->slotId, cInfo.colId, cInfo.type);
649
#endif
650
    void* tmp = taosArrayPush(pData->cInfoList, &cInfo);
15,475,573✔
651
    if (!tmp) {
15,479,447✔
652
      return DEAL_RES_ERROR;
×
653
    }
654
  } else {
655
    SColumnNode* col = *(SColumnNode**)data;
2,578,521✔
656
    pSColumnNode->slotId = col->slotId;
2,578,521✔
657
  }
658

659
  return DEAL_RES_CONTINUE;
18,056,474✔
660
}
661

662
static int32_t createResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
14,405,935✔
663
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
14,405,935✔
664
  if (pColumnData == NULL) {
14,406,039✔
665
    return terrno;
×
666
  }
667

668
  pColumnData->info.type = pType->type;
14,406,039✔
669
  pColumnData->info.bytes = pType->bytes;
14,407,226✔
670
  pColumnData->info.scale = pType->scale;
14,402,532✔
671
  pColumnData->info.precision = pType->precision;
14,406,507✔
672

673
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
14,402,643✔
674
  if (code != TSDB_CODE_SUCCESS) {
14,403,480✔
675
    terrno = code;
×
676
    releaseColInfoData(pColumnData);
×
677
    return terrno;
×
678
  }
679

680
  pParam->columnData = pColumnData;
14,403,480✔
681
  pParam->colAlloced = true;
14,402,558✔
682
  return TSDB_CODE_SUCCESS;
14,398,309✔
683
}
684

685
static void releaseColInfoData(void* pCol) {
4,024,120✔
686
  if (pCol) {
4,024,120✔
687
    SColumnInfoData* col = (SColumnInfoData*)pCol;
4,024,120✔
688
    colDataDestroy(col);
4,024,120✔
689
    taosMemoryFree(col);
4,022,885✔
690
  }
691
}
4,022,901✔
692

693
void freeItem(void* p) {
181,125,330✔
694
  STUidTagInfo* pInfo = p;
181,125,330✔
695
  if (pInfo->pTagVal != NULL) {
181,125,330✔
696
    taosMemoryFree(pInfo->pTagVal);
180,804,978✔
697
  }
698
}
181,089,968✔
699

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

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

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

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

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

775
  return TSDB_CODE_SUCCESS;
8,745✔
776
}
777

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

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

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

803
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
8,745✔
804
    extractTagDataEntry((SOperatorNode*)pTagCond, pIdWithVal);
×
805
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
8,745✔
806
    SNode* pChild = NULL;
8,745✔
807
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
26,235✔
808
      extractTagDataEntry((SOperatorNode*)pChild, pIdWithVal);
17,490✔
809
    }
810
  }
811

812
  taosArraySort(pIdWithVal, compareTagDataEntry);
8,745✔
813

814
  return TSDB_CODE_SUCCESS;
8,745✔
815
}
816

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

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

827
  SArray* pIdWithVal = taosArrayInit(TARRAY_MIN_SIZE, sizeof(STagDataEntry));
8,745✔
828
  code = extractTagFilterTagDataEntries(pTagCond, pIdWithVal);
8,745✔
829
  QUERY_CHECK_CODE(code, lino, _end);
8,745✔
830
  for (int32_t i = 0; i < taosArrayGetSize(pIdWithVal); ++i) {
26,235✔
831
    STagDataEntry* pEntry = taosArrayGet(pIdWithVal, i);
17,490✔
832
    len += sizeof(col_id_t) + pEntry->bytes;
17,490✔
833
  }
834
  code = buildTagDataEntryKey(pIdWithVal, &payload, len);
8,745✔
835
  QUERY_CHECK_CODE(code, lino, _end);
8,745✔
836

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

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

851
static int32_t genTagFilterDigest(const SNode* pTagCond, T_MD5_CTX* pContext) {
29,447✔
852
  if (pTagCond == NULL) {
29,447✔
853
    return TSDB_CODE_SUCCESS;
27,790✔
854
  }
855

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

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

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

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

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

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

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

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

905
int32_t qGetColumnsFromNodeList(void* data, bool isList, SArray** pColList) {
14,245,527✔
906
  int32_t code = TSDB_CODE_SUCCESS;
14,245,527✔
907
  tagFilterAssist ctx = {0};
14,245,527✔
908
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
14,246,056✔
909
  if (ctx.colHash == NULL) {
14,245,238✔
910
    code = terrno;
×
911
    goto end;
×
912
  }
913

914
  ctx.index = 0;
14,245,238✔
915
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
14,245,238✔
916
  if (ctx.cInfoList == NULL) {
14,244,903✔
917
    code = terrno;
1,835✔
918
    goto end;
×
919
  }
920

921
  if (isList) {
14,243,068✔
922
    SNode* pNode = NULL;
3,860,912✔
923
    FOREACH(pNode, (SNodeList*)data) {
7,890,878✔
924
      nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
4,029,966✔
925
      if (TSDB_CODE_SUCCESS != ctx.code) {
4,029,877✔
926
        code = ctx.code;
×
927
        goto end;
×
928
      }
929
      REPLACE_NODE(pNode);
4,029,877✔
930
    }
931
  } else {
932
    SNode* pNode = (SNode*)data;
10,382,156✔
933
    nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
10,383,991✔
934
    if (TSDB_CODE_SUCCESS != ctx.code) {
10,383,831✔
935
      code = ctx.code;
×
936
      goto end;
×
937
    }
938
  }
939
  
940
  if (pColList != NULL) *pColList = ctx.cInfoList;
14,244,394✔
941
  ctx.cInfoList = NULL;
14,241,548✔
942

943
end:
14,241,503✔
944
  taosHashCleanup(ctx.colHash);
14,241,323✔
945
  taosArrayDestroy(ctx.cInfoList);
14,234,953✔
946
  return code;
14,235,991✔
947
}
948

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

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

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

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

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

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

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

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

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

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

1084
  SNode* pNode = NULL;
61,960✔
1085
  FOREACH(pNode, group) {
161,467✔
1086
    SScalarParam output = {0};
99,507✔
1087

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

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

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

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

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

1129
  for (int i = 0; i < rows; i++) {
287,713✔
1130
    gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
225,753✔
1131
    QUERY_CHECK_NULL(gInfo, code, lino, end, terrno);
225,753✔
1132

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

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

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

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

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

1173
int32_t getColInfoResultForGroupby(void* pVnode, SNodeList* group, STableListInfo* pTableListInfo, uint8_t* digest,
3,799,213✔
1174
                                   SStorageAPI* pAPI, bool initRemainGroups, SHashObj* groupIdMap) {
1175
  int32_t      code = TSDB_CODE_SUCCESS;
3,799,213✔
1176
  int32_t      lino = 0;
3,799,213✔
1177
  SArray*      pBlockList = NULL;
3,799,213✔
1178
  SSDataBlock* pResBlock = NULL;
3,799,213✔
1179
  void*        keyBuf = NULL;
3,799,435✔
1180
  SArray*      groupData = NULL;
3,799,435✔
1181
  SArray*      pUidTagList = NULL;
3,799,435✔
1182
  SArray*      tableList = NULL;
3,799,435✔
1183
  SArray*      gInfo = NULL;
3,799,435✔
1184

1185
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
3,799,435✔
1186
  if (rows == 0) {
3,799,435✔
1187
    return TSDB_CODE_SUCCESS;
×
1188
  } 
1189

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

1201
    nodesFree(listNode);
×
1202

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

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

1216
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
3,799,435✔
1217
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
3,799,435✔
1218

1219
  for (int32_t i = 0; i < rows; ++i) {
25,282,452✔
1220
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
21,483,017✔
1221
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
21,483,017✔
1222
    STUidTagInfo info = {.uid = pkeyInfo->uid};
21,483,017✔
1223
    void*        tmp = taosArrayPush(pUidTagList, &info);
21,483,017✔
1224
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
21,483,017✔
1225
  }
1226

1227
  if (taosArrayGetSize(pUidTagList) > 0) {
3,799,435✔
1228
    code = pAPI->metaFn.getTableTagsByUid(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
3,799,435✔
1229
  } else {
1230
    code = pAPI->metaFn.getTableTags(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
×
1231
  }
1232
  if (code != TSDB_CODE_SUCCESS) {
3,799,435✔
1233
    goto end;
×
1234
  }
1235

1236
  SArray* pColList = NULL;
3,799,435✔
1237
  code = qGetColumnsFromNodeList(group, true, &pColList); 
3,799,435✔
1238
  if (code != TSDB_CODE_SUCCESS) {
3,798,558✔
1239
    goto end;
×
1240
  }
1241

1242
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
3,798,558✔
1243
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
3,798,558✔
1244
  taosArrayDestroy(pColList);
3,798,469✔
1245
  if (pResBlock == NULL) {
3,799,041✔
1246
    code = terrno;
×
1247
    goto end;
×
1248
  }
1249

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

1253
  pBlockList = taosArrayInit(2, POINTER_BYTES);
3,799,041✔
1254
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
3,799,022✔
1255

1256
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
3,799,435✔
1257
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
3,799,435✔
1258

1259
  groupData = taosArrayInit(2, POINTER_BYTES);
3,799,435✔
1260
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
3,798,588✔
1261

1262
  SNode* pNode = NULL;
3,798,952✔
1263
  FOREACH(pNode, group) {
7,729,488✔
1264
    SScalarParam output = {0};
3,929,852✔
1265

1266
    switch (nodeType(pNode)) {
3,929,510✔
1267
      case QUERY_NODE_VALUE:
×
1268
        break;
×
1269
      case QUERY_NODE_COLUMN:
3,924,059✔
1270
      case QUERY_NODE_OPERATOR:
1271
      case QUERY_NODE_FUNCTION: {
1272
        SExprNode* expNode = (SExprNode*)pNode;
3,924,059✔
1273
        code = createResultData(&expNode->resType, rows, &output);
3,924,059✔
1274
        if (code != TSDB_CODE_SUCCESS) {
3,925,096✔
1275
          goto end;
×
1276
        }
1277
        break;
3,925,096✔
1278
      }
1279
      case QUERY_NODE_REMOTE_VALUE: {
6,240✔
1280
        SRemoteValueNode* pRemote = (SRemoteValueNode*)pNode;
6,240✔
1281
        code = qFetchRemoteValue(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, pRemote);
6,240✔
1282
        QUERY_CHECK_CODE(code, lino, end);
6,240✔
1283
        break;
6,240✔
1284
      }
1285
      
1286
      default:
×
1287
        code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1288
        goto end;
×
1289
    }
1290

1291
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
3,931,336✔
1292
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
3,901,315✔
1293
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
3,901,315✔
1294
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
3,898,844✔
1295
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
3,898,844✔
1296
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
29,538✔
1297
      continue;
6,240✔
1298
    } else {
1299
      gTaskScalarExtra.pStreamInfo = NULL;
23,781✔
1300
      gTaskScalarExtra.pStreamRange = NULL;
23,781✔
1301
      code = scalarCalculate(pNode, pBlockList, &output, &gTaskScalarExtra);
23,781✔
1302
    }
1303

1304
    if (code != TSDB_CODE_SUCCESS) {
3,924,313✔
1305
      releaseColInfoData(output.columnData);
×
1306
      goto end;
×
1307
    }
1308

1309
    void* tmp = taosArrayPush(groupData, &output.columnData);
3,924,906✔
1310
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
3,924,906✔
1311
  }
1312

1313
  int32_t keyLen = 0;
3,798,042✔
1314
  SNode*  node;
1315
  FOREACH(node, group) {
7,724,507✔
1316
    SExprNode* pExpr = (SExprNode*)node;
3,928,861✔
1317
    keyLen += pExpr->resType.bytes;
3,928,861✔
1318
  }
1319

1320
  int32_t nullFlagSize = sizeof(int8_t) * LIST_LENGTH(group);
3,795,493✔
1321
  keyLen += nullFlagSize;
3,795,493✔
1322

1323
  keyBuf = taosMemoryCalloc(1, keyLen);
3,795,493✔
1324
  if (keyBuf == NULL) {
3,797,548✔
1325
    code = terrno;
×
1326
    goto end;
×
1327
  }
1328

1329
  if (initRemainGroups) {
3,797,548✔
1330
    pTableListInfo->remainGroups =
1,855,135✔
1331
        taosHashInit(rows, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
1,855,281✔
1332
    if (pTableListInfo->remainGroups == NULL) {
1,855,135✔
1333
      code = terrno;
×
1334
      goto end;
×
1335
    }
1336
  }
1337

1338
  for (int i = 0; i < rows; i++) {
25,275,630✔
1339
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
21,476,957✔
1340
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
21,477,461✔
1341

1342
    if (groupIdMap != NULL){
21,477,461✔
1343
      gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
60,529✔
1344
    }
1345
    
1346
    char* isNull = (char*)keyBuf;
21,477,626✔
1347
    char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(group);
21,477,626✔
1348
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
44,015,974✔
1349
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
22,536,575✔
1350

1351
      if (groupIdMap != NULL && gInfo != NULL) {
22,537,969✔
1352
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
72,297✔
1353
        if (ret != TSDB_CODE_SUCCESS) {
72,297✔
1354
          qError("buildGroupInfo failed at line %d since %s", __LINE__, tstrerror(ret));
×
1355
          taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1356
          gInfo = NULL;
×
1357
        }
1358
      }
1359
      
1360
      if (colDataIsNull_s(pValue, i)) {
45,075,203✔
1361
        isNull[j] = 1;
94,524✔
1362
      } else {
1363
        isNull[j] = 0;
22,442,710✔
1364
        char* data = colDataGetData(pValue, i);
22,443,787✔
1365
        if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
22,446,115✔
1366
          // if (tTagIsJson(data)) {
1367
          //   code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
1368
          //   goto end;
1369
          // }
1370
          if (tTagIsJsonNull(data)) {
71,394✔
1371
            isNull[j] = 1;
×
1372
            continue;
×
1373
          }
1374
          int32_t len = getJsonValueLen(data);
71,394✔
1375
          memcpy(pStart, data, len);
71,394✔
1376
          pStart += len;
71,394✔
1377
        } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
22,371,579✔
1378
          if (IS_STR_DATA_BLOB(pValue->info.type)) {
19,551,111✔
UNCOV
1379
            if (blobDataTLen(data) > TSDB_MAX_BLOB_LEN) {
×
1380
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1381
              goto end;
×
1382
            }
1383
            memcpy(pStart, data, blobDataTLen(data));
×
1384
            pStart += blobDataTLen(data);
×
1385
          } else {
1386
            if (varDataTLen(data) > pValue->info.bytes) {
19,550,288✔
1387
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1388
              goto end;
×
1389
            }
1390
            memcpy(pStart, data, varDataTLen(data));
19,551,364✔
1391
            pStart += varDataTLen(data);
19,551,364✔
1392
          }
1393
        } else {
1394
          memcpy(pStart, data, pValue->info.bytes);
2,823,467✔
1395
          pStart += pValue->info.bytes;
2,823,467✔
1396
        }
1397
      }
1398
    }
1399

1400
    int32_t len = (int32_t)(pStart - (char*)keyBuf);
21,478,356✔
1401
    info->groupId = calcGroupId(keyBuf, len);
21,478,356✔
1402
    if (groupIdMap != NULL && gInfo != NULL) {
21,481,263✔
1403
      int32_t ret = taosHashPut(groupIdMap, &info->groupId, sizeof(info->groupId), &gInfo, POINTER_BYTES);
60,085✔
1404
      if (ret != TSDB_CODE_SUCCESS) {
60,529✔
1405
        qError("put groupid to map failed at line %d since %s", __LINE__, tstrerror(ret));
×
1406
        taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1407
      }
1408
      qDebug("put groupid to map gid:%" PRIu64, info->groupId);
60,529✔
1409
      gInfo = NULL;
60,529✔
1410
    }
1411
    if (initRemainGroups) {
21,481,707✔
1412
      // groupId ~ table uid
1413
      code = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId), &(info->uid),
10,589,334✔
1414
                         sizeof(info->uid));
1415
      if (code == TSDB_CODE_DUP_KEY) {
10,585,508✔
1416
        code = TSDB_CODE_SUCCESS;
734,065✔
1417
      }
1418
      QUERY_CHECK_CODE(code, lino, end);
10,585,508✔
1419
    }
1420
  }
1421

1422
  if (tsTagFilterCache && groupIdMap == NULL) {
3,798,673✔
1423
    tableList = taosArrayDup(pTableListInfo->pTableList, NULL);
×
1424
    QUERY_CHECK_NULL(tableList, code, lino, end, terrno);
×
1425

1426
    code = pAPI->metaFn.metaPutTbGroupToCache(pVnode, pTableListInfo->idInfo.suid, context.digest,
×
1427
                                              tListLen(context.digest), tableList,
1428
                                              taosArrayGetSize(tableList) * sizeof(STableKeyInfo));
×
1429
    QUERY_CHECK_CODE(code, lino, end);
×
1430
  }
1431

1432
  //  int64_t st2 = taosGetTimestampUs();
1433
  //  qDebug("calculate tag block rows:%d, cost:%ld us", rows, st2-st1);
1434

1435
end:
3,798,673✔
1436
  taosMemoryFreeClear(keyBuf);
3,799,093✔
1437
  blockDataDestroy(pResBlock);
3,799,093✔
1438
  taosArrayDestroy(pBlockList);
3,798,075✔
1439
  taosArrayDestroyEx(pUidTagList, freeItem);
3,797,198✔
1440
  taosArrayDestroyP(groupData, releaseColInfoData);
3,798,127✔
1441
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
3,798,164✔
1442

1443
  if (code != TSDB_CODE_SUCCESS) {
3,798,164✔
1444
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1445
  }
1446
  return code;
3,798,952✔
1447
}
1448

1449
static int32_t nameComparFn(const void* p1, const void* p2) {
635,402✔
1450
  const char* pName1 = *(const char**)p1;
635,402✔
1451
  const char* pName2 = *(const char**)p2;
635,402✔
1452

1453
  int32_t ret = strcmp(pName1, pName2);
635,402✔
1454
  if (ret == 0) {
635,402✔
1455
    return 0;
15,534✔
1456
  } else {
1457
    return (ret > 0) ? 1 : -1;
619,868✔
1458
  }
1459
}
1460

1461
static SArray* getTableNameList(const SNodeListNode* pList) {
350,372✔
1462
  int32_t    code = TSDB_CODE_SUCCESS;
350,372✔
1463
  int32_t    lino = 0;
350,372✔
1464
  int32_t    len = LIST_LENGTH(pList->pNodeList);
350,372✔
1465
  SListCell* cell = pList->pNodeList->pHead;
350,372✔
1466

1467
  SArray* pTbList = taosArrayInit(len, POINTER_BYTES);
350,372✔
1468
  QUERY_CHECK_NULL(pTbList, code, lino, _end, terrno);
350,372✔
1469

1470
  for (int i = 0; i < pList->pNodeList->length; i++) {
967,892✔
1471
    SValueNode* valueNode = (SValueNode*)cell->pNode;
617,520✔
1472
    if (!IS_VAR_DATA_TYPE(valueNode->node.resType.type)) {
617,520✔
1473
      terrno = TSDB_CODE_INVALID_PARA;
×
1474
      taosArrayDestroy(pTbList);
×
1475
      return NULL;
×
1476
    }
1477

1478
    char* name = varDataVal(valueNode->datum.p);
617,520✔
1479
    void* tmp = taosArrayPush(pTbList, &name);
617,520✔
1480
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
617,520✔
1481
    cell = cell->pNext;
617,520✔
1482
  }
1483

1484
  size_t numOfTables = taosArrayGetSize(pTbList);
350,372✔
1485

1486
  // order the name
1487
  taosArraySort(pTbList, nameComparFn);
350,372✔
1488

1489
  // remove the duplicates
1490
  SArray* pNewList = taosArrayInit(taosArrayGetSize(pTbList), sizeof(void*));
350,372✔
1491
  QUERY_CHECK_NULL(pNewList, code, lino, _end, terrno);
350,372✔
1492
  void* tmpTbl = taosArrayGet(pTbList, 0);
350,372✔
1493
  QUERY_CHECK_NULL(tmpTbl, code, lino, _end, terrno);
350,372✔
1494
  void* tmp = taosArrayPush(pNewList, tmpTbl);
350,372✔
1495
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
350,372✔
1496

1497
  for (int32_t i = 1; i < numOfTables; ++i) {
617,520✔
1498
    char** name = taosArrayGetLast(pNewList);
267,148✔
1499
    char** nameInOldList = taosArrayGet(pTbList, i);
267,148✔
1500
    QUERY_CHECK_NULL(nameInOldList, code, lino, _end, terrno);
267,148✔
1501
    if (strcmp(*name, *nameInOldList) == 0) {
267,148✔
1502
      continue;
8,376✔
1503
    }
1504

1505
    tmp = taosArrayPush(pNewList, nameInOldList);
258,772✔
1506
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
258,772✔
1507
  }
1508

1509
_end:
350,372✔
1510
  taosArrayDestroy(pTbList);
350,372✔
1511
  if (code != TSDB_CODE_SUCCESS) {
350,372✔
1512
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1513
    return NULL;
×
1514
  }
1515
  return pNewList;
350,372✔
1516
}
1517

1518
static int tableUidCompare(const void* a, const void* b) {
×
1519
  uint64_t u1 = *(uint64_t*)a;
×
1520
  uint64_t u2 = *(uint64_t*)b;
×
1521

1522
  if (u1 == u2) {
×
1523
    return 0;
×
1524
  }
1525

1526
  return u1 < u2 ? -1 : 1;
×
1527
}
1528

1529
static int32_t filterTableInfoCompare(const void* a, const void* b) {
16,078,898✔
1530
  STUidTagInfo* p1 = (STUidTagInfo*)a;
16,078,898✔
1531
  STUidTagInfo* p2 = (STUidTagInfo*)b;
16,078,898✔
1532

1533
  if (p1->uid == p2->uid) {
16,078,898✔
1534
    return 0;
×
1535
  }
1536

1537
  return p1->uid < p2->uid ? -1 : 1;
16,078,898✔
1538
}
1539

1540
static FilterCondType checkTagCond(SNode* cond) {
11,540,371✔
1541
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
11,540,371✔
1542
    return FILTER_NO_LOGIC;
8,146,260✔
1543
  }
1544
  if (nodeType(cond) == QUERY_NODE_LOGIC_CONDITION && ((SLogicConditionNode*)cond)->condType == LOGIC_COND_TYPE_AND) {
3,394,111✔
1545
    return FILTER_AND;
2,878,013✔
1546
  }
1547
  return FILTER_OTHER;
512,150✔
1548
}
1549

1550
static bool optimizeTbnameInCond(void* pVnode, int64_t suid, SArray* list, SNode* cond, SStorageAPI* pAPI) {
11,884,341✔
1551
  int32_t code = 0;
11,884,341✔
1552
  int32_t ntype = nodeType(cond);
11,884,341✔
1553

1554
  if (ntype == QUERY_NODE_OPERATOR) {
11,883,375✔
1555
    code = optimizeTbnameInCondImpl(pVnode, list, cond, pAPI, suid);
8,482,565✔
1556
    return code == 0;
8,480,842✔
1557
  }
1558
  if (ntype != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
3,400,810✔
1559
    return false;
512,968✔
1560
  }
1561

1562
  bool                 hasTbnameCond = false;
2,885,618✔
1563
  SLogicConditionNode* pNode = (SLogicConditionNode*)cond;
2,885,618✔
1564
  SNodeList*           pList = (SNodeList*)pNode->pParameterList;
2,885,618✔
1565

1566
  int32_t len = LIST_LENGTH(pList);
2,884,420✔
1567
  if (len <= 0) {
2,884,613✔
1568
    return false;
×
1569
  }
1570

1571
  SListCell* cell = pList->pHead;
2,884,613✔
1572
  for (int i = 0; i < len; i++) {
9,020,702✔
1573
    if (cell == NULL) break;
6,137,697✔
1574
    if (optimizeTbnameInCondImpl(pVnode, list, cell->pNode, pAPI, suid) == 0) {
6,137,697✔
1575
      hasTbnameCond = true;
7,418✔
1576
      break;
7,418✔
1577
    }
1578
    cell = cell->pNext;
6,131,960✔
1579
  }
1580

1581
  taosArraySort(list, filterTableInfoCompare);
2,890,423✔
1582
  taosArrayRemoveDuplicate(list, filterTableInfoCompare, NULL);
2,886,095✔
1583

1584
  if (hasTbnameCond) {
2,887,743✔
1585
    code = pAPI->metaFn.getTableTagsByUid(pVnode, suid, list);
7,418✔
1586
    return code == 0;
7,418✔
1587
  }
1588

1589
  return false;
2,880,325✔
1590
}
1591

1592
// only return uid that does not contained in pExistedUidList
1593
static int32_t optimizeTbnameInCondImpl(void* pVnode, SArray* pExistedUidList, SNode* pTagCond, SStorageAPI* pStoreAPI,
14,624,886✔
1594
                                        uint64_t suid) {
1595
  if (nodeType(pTagCond) != QUERY_NODE_OPERATOR) {
14,624,886✔
1596
    return -1;
4,988✔
1597
  }
1598

1599
  SOperatorNode* pNode = (SOperatorNode*)pTagCond;
14,620,233✔
1600
  if (pNode->opType != OP_TYPE_IN) {
14,620,233✔
1601
    return -1;
13,817,337✔
1602
  }
1603

1604
  if ((pNode->pLeft != NULL && ((nodeType(pNode->pLeft) == QUERY_NODE_FUNCTION &&
803,083✔
1605
                                 ((SFunctionNode*)pNode->pLeft)->funcType == FUNCTION_TYPE_TBNAME)) ||
350,372✔
1606
       (nodeType(pNode->pLeft) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pLeft)->colType == COLUMN_TYPE_TBNAME)) &&
452,711✔
1607
      (pNode->pRight != NULL && nodeType(pNode->pRight) == QUERY_NODE_NODE_LIST)) {
349,889✔
1608
    SNodeListNode* pList = (SNodeListNode*)pNode->pRight;
350,372✔
1609

1610
    int32_t len = LIST_LENGTH(pList->pNodeList);
350,372✔
1611
    if (len <= 0) {
350,372✔
1612
      return -1;
×
1613
    }
1614

1615
    SArray*   pTbList = getTableNameList(pList);
350,372✔
1616
    int32_t   numOfTables = taosArrayGetSize(pTbList);
350,372✔
1617
    SHashObj* uHash = NULL;
350,372✔
1618

1619
    size_t numOfExisted = taosArrayGetSize(pExistedUidList);  // len > 0 means there already have uids
350,372✔
1620
    if (numOfExisted > 0) {
350,372✔
1621
      uHash = taosHashInit(numOfExisted / 0.7, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
1,980✔
1622
      if (!uHash) {
1,980✔
1623
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1624
        return terrno;
×
1625
      }
1626

1627
      for (int i = 0; i < numOfExisted; i++) {
1,979,010✔
1628
        STUidTagInfo* pTInfo = taosArrayGet(pExistedUidList, i);
1,977,030✔
1629
        if (!pTInfo) {
1,977,030✔
1630
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1631
          return terrno;
×
1632
        }
1633
        int32_t tempRes = taosHashPut(uHash, &pTInfo->uid, sizeof(uint64_t), &i, sizeof(i));
1,977,030✔
1634
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
1,977,030✔
1635
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
1636
          return tempRes;
×
1637
        }
1638
      }
1639
    }
1640

1641
    for (int i = 0; i < numOfTables; i++) {
944,974✔
1642
      char* name = taosArrayGetP(pTbList, i);
600,768✔
1643

1644
      uint64_t uid = 0, csuid = 0;
600,768✔
1645
      if (pStoreAPI->metaFn.getTableUidByName(pVnode, name, &uid) == 0) {
600,768✔
1646
        ETableType tbType = TSDB_TABLE_MAX;
351,487✔
1647
        if (pStoreAPI->metaFn.getTableTypeSuidByName(pVnode, name, &tbType, &csuid) == 0 &&
351,487✔
1648
            tbType == TSDB_CHILD_TABLE) {
351,487✔
1649
          if (suid != csuid) {
345,321✔
1650
            continue;
1,976✔
1651
          }
1652
          if (NULL == uHash || taosHashGet(uHash, &uid, sizeof(uid)) == NULL) {
343,345✔
1653
            STUidTagInfo s = {.uid = uid, .name = name, .pTagVal = NULL};
342,355✔
1654
            void*        tmp = taosArrayPush(pExistedUidList, &s);
342,355✔
1655
            if (!tmp) {
342,355✔
1656
              return terrno;
×
1657
            }
1658
          }
1659
        } else {
1660
          taosArrayDestroy(pTbList);
6,166✔
1661
          taosHashCleanup(uHash);
6,166✔
1662
          return -1;
6,166✔
1663
        }
1664
      } else {
1665
        //        qWarn("failed to get tableIds from by table name: %s, reason: %s", name, tstrerror(terrno));
1666
        terrno = 0;
249,281✔
1667
      }
1668
    }
1669

1670
    taosHashCleanup(uHash);
344,206✔
1671
    taosArrayDestroy(pTbList);
344,206✔
1672
    return 0;
344,206✔
1673
  }
1674

1675
  return -1;
452,711✔
1676
}
1677

1678
SSDataBlock* createTagValBlockForFilter(SArray* pColList, int32_t numOfTables, SArray* pUidTagList, void* pVnode,
15,839,919✔
1679
                                        SStorageAPI* pStorageAPI) {
1680
  int32_t      code = TSDB_CODE_SUCCESS;
15,839,919✔
1681
  int32_t      lino = 0;
15,839,919✔
1682
  SSDataBlock* pResBlock = NULL;
15,839,919✔
1683
  code = createDataBlock(&pResBlock);
15,842,510✔
1684
  QUERY_CHECK_CODE(code, lino, _end);
15,837,786✔
1685

1686
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
32,912,173✔
1687
    SColumnInfoData colInfo = {0};
17,078,762✔
1688
    void*           tmp = taosArrayGet(pColList, i);
17,077,944✔
1689
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
17,070,162✔
1690
    colInfo.info = *(SColumnInfo*)tmp;
17,070,162✔
1691
    code = blockDataAppendColInfo(pResBlock, &colInfo);
17,070,713✔
1692
    QUERY_CHECK_CODE(code, lino, _end);
17,075,190✔
1693
  }
1694

1695
  code = blockDataEnsureCapacity(pResBlock, numOfTables);
15,836,936✔
1696
  if (code != TSDB_CODE_SUCCESS) {
15,837,072✔
1697
    terrno = code;
×
1698
    blockDataDestroy(pResBlock);
×
1699
    return NULL;
×
1700
  }
1701

1702
  pResBlock->info.rows = numOfTables;
15,837,072✔
1703

1704
  int32_t numOfCols = taosArrayGetSize(pResBlock->pDataBlock);
15,839,083✔
1705

1706
  for (int32_t i = 0; i < numOfTables; i++) {
199,273,643✔
1707
    STUidTagInfo* p1 = taosArrayGet(pUidTagList, i);
183,430,811✔
1708
    QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
183,418,791✔
1709

1710
    for (int32_t j = 0; j < numOfCols; j++) {
375,067,730✔
1711
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, j);
191,569,977✔
1712
      QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
191,556,400✔
1713

1714
      if (pColInfo->info.colId == -1) {  // tbname
191,556,400✔
1715
        char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
6,795,539✔
1716
        if (p1->name != NULL) {
6,793,569✔
1717
          STR_TO_VARSTR(str, p1->name);
342,355✔
1718
        } else {  // name is not retrieved during filter
1719
          code = pStorageAPI->metaFn.getTableNameByUid(pVnode, p1->uid, str);
6,451,645✔
1720
          QUERY_CHECK_CODE(code, lino, _end);
6,451,243✔
1721
        }
1722

1723
        code = colDataSetVal(pColInfo, i, str, false);
6,793,598✔
1724
        QUERY_CHECK_CODE(code, lino, _end);
6,797,523✔
1725
#if TAG_FILTER_DEBUG
1726
        qDebug("tagfilter uid:%ld, tbname:%s", *uid, str + 2);
1727
#endif
1728
      } else {
1729
        STagVal tagVal = {0};
184,725,891✔
1730
        tagVal.cid = pColInfo->info.colId;
184,724,990✔
1731
        if (p1->pTagVal == NULL) {
184,753,258✔
1732
          colDataSetNULL(pColInfo, i);
3,975✔
1733
        } else {
1734
          const char* p = pStorageAPI->metaFn.extractTagVal(p1->pTagVal, pColInfo->info.type, &tagVal);
184,771,748✔
1735

1736
          if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
184,790,628✔
1737
            colDataSetNULL(pColInfo, i);
2,115,831✔
1738
          } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
182,692,275✔
1739
            code = colDataSetVal(pColInfo, i, p, false);
564,447✔
1740
            QUERY_CHECK_CODE(code, lino, _end);
564,447✔
1741
          } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
306,390,992✔
1742
            if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
124,213,434✔
1743
              QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
1744
            }
1745
            char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
124,235,928✔
1746
            QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
124,245,405✔
1747
            varDataSetLen(tmp, tagVal.nData);
124,245,405✔
1748
            memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
124,242,934✔
1749
            code = colDataSetVal(pColInfo, i, tmp, false);
124,253,737✔
1750
#if TAG_FILTER_DEBUG
1751
            qDebug("tagfilter varch:%s", tmp + 2);
1752
#endif
1753
            taosMemoryFree(tmp);
124,278,269✔
1754
            QUERY_CHECK_CODE(code, lino, _end);
124,253,189✔
1755
          } else {
1756
            code = colDataSetVal(pColInfo, i, (const char*)&tagVal.i64, false);
57,937,087✔
1757
            QUERY_CHECK_CODE(code, lino, _end);
57,941,557✔
1758
#if TAG_FILTER_DEBUG
1759
            if (pColInfo->info.type == TSDB_DATA_TYPE_INT) {
1760
              qDebug("tagfilter int:%d", *(int*)(&tagVal.i64));
1761
            } else if (pColInfo->info.type == TSDB_DATA_TYPE_DOUBLE) {
1762
              qDebug("tagfilter double:%f", *(double*)(&tagVal.i64));
1763
            }
1764
#endif
1765
          }
1766
        }
1767
      }
1768
    }
1769
  }
1770

1771
_end:
15,859,058✔
1772
  if (code != TSDB_CODE_SUCCESS) {
15,842,832✔
1773
    blockDataDestroy(pResBlock);
1,101✔
1774
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1775
    terrno = code;
×
1776
    return NULL;
×
1777
  }
1778
  return pResBlock;
15,841,731✔
1779
}
1780

1781
static int32_t doSetQualifiedUid(STableListInfo* pListInfo, SArray* pUidList, const SArray* pUidTagList,
10,369,144✔
1782
                                 bool* pResultList, bool addUid) {
1783
  taosArrayClear(pUidList);
10,369,144✔
1784

1785
  STableKeyInfo info = {.uid = 0, .groupId = 0};
10,373,009✔
1786
  int32_t       numOfTables = taosArrayGetSize(pUidTagList);
10,374,846✔
1787
  for (int32_t i = 0; i < numOfTables; ++i) {
169,874,445✔
1788
    if (pResultList[i]) {
159,493,952✔
1789
      STUidTagInfo* tmpTag = (STUidTagInfo*)taosArrayGet(pUidTagList, i);
72,275,713✔
1790
      if (!tmpTag) {
72,277,765✔
1791
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1792
        return terrno;
×
1793
      }
1794
      uint64_t uid = tmpTag->uid;
72,277,765✔
1795
      qDebug("tagfilter get uid:%" PRId64 ", res:%d", uid, pResultList[i]);
72,269,358✔
1796

1797
      info.uid = uid;
72,276,086✔
1798
      //qInfo("doSetQualifiedUid row:%d added to pTableList", i);
1799
      void* p = taosArrayPush(pListInfo->pTableList, &info);
72,276,086✔
1800
      if (p == NULL) {
72,276,100✔
1801
        return terrno;
×
1802
      }
1803

1804
      if (addUid) {
72,276,100✔
1805
        //qInfo("doSetQualifiedUid row:%d added to pUidList", i);
1806
        void* tmp = taosArrayPush(pUidList, &uid);
9,432✔
1807
        if (tmp == NULL) {
9,432✔
1808
          return terrno;
×
1809
        }
1810
      }
1811
    } else {
1812
      //qInfo("doSetQualifiedUid row:%d failed", i);
1813
    }
1814
  }
1815

1816
  return TSDB_CODE_SUCCESS;
10,380,493✔
1817
}
1818

1819
static int32_t copyExistedUids(SArray* pUidTagList, const SArray* pUidList) {
11,884,577✔
1820
  int32_t code = TSDB_CODE_SUCCESS;
11,884,577✔
1821
  int32_t numOfExisted = taosArrayGetSize(pUidList);
11,884,577✔
1822
  if (numOfExisted == 0) {
11,884,429✔
1823
    return code;
9,448,154✔
1824
  }
1825

1826
  for (int32_t i = 0; i < numOfExisted; ++i) {
28,158,494✔
1827
    uint64_t* uid = taosArrayGet(pUidList, i);
25,722,219✔
1828
    if (!uid) {
25,722,219✔
1829
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1830
      return terrno;
×
1831
    }
1832
    STUidTagInfo info = {.uid = *uid};
25,722,219✔
1833
    void*        tmp = taosArrayPush(pUidTagList, &info);
25,722,219✔
1834
    if (!tmp) {
25,722,219✔
1835
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1836
      return code;
×
1837
    }
1838
  }
1839
  return code;
2,436,275✔
1840
}
1841

1842
int32_t doFilterByTagCond(STableListInfo* pListInfo, SArray* pUidList, SNode* pTagCond, void* pVnode,
145,393,156✔
1843
                                 SIdxFltStatus status, SStorageAPI* pAPI, bool addUid, bool* listAdded, void* pStreamInfo) {
1844
  *listAdded = false;
145,393,156✔
1845
  if (pTagCond == NULL) {
145,402,440✔
1846
    return TSDB_CODE_SUCCESS;
133,514,934✔
1847
  }
1848

1849
  terrno = TSDB_CODE_SUCCESS;
11,887,506✔
1850

1851
  int32_t      lino = 0;
11,881,862✔
1852
  int32_t      code = TSDB_CODE_SUCCESS;
11,881,862✔
1853
  SArray*      pBlockList = NULL;
11,881,862✔
1854
  SSDataBlock* pResBlock = NULL;
11,881,862✔
1855
  SScalarParam output = {0};
11,881,044✔
1856
  SArray*      pUidTagList = NULL;
11,882,055✔
1857

1858
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
11,882,055✔
1859

1860
  //  int64_t stt = taosGetTimestampUs();
1861
  pUidTagList = taosArrayInit(10, sizeof(STUidTagInfo));
11,877,485✔
1862
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
11,879,180✔
1863

1864
  code = copyExistedUids(pUidTagList, pUidList);
11,879,180✔
1865
  QUERY_CHECK_CODE(code, lino, end);
11,879,166✔
1866

1867
  // Narrow down the scope of the tablelist set if there is tbname in condition and And Logical operator
1868
  bool narrowed = optimizeTbnameInCond(pVnode, pListInfo->idInfo.suid, pUidTagList, pTagCond, pAPI);
11,879,166✔
1869
  if (narrowed) {  // tbname in filter is activated, do nothing and return
11,882,083✔
1870
    taosArrayClear(pUidList);
344,206✔
1871

1872
    int32_t numOfRows = taosArrayGetSize(pUidTagList);
344,206✔
1873
    code = taosArrayEnsureCap(pUidList, numOfRows);
344,206✔
1874
    QUERY_CHECK_CODE(code, lino, end);
344,206✔
1875

1876
    for (int32_t i = 0; i < numOfRows; ++i) {
2,663,591✔
1877
      STUidTagInfo* pInfo = taosArrayGet(pUidTagList, i);
2,319,385✔
1878
      QUERY_CHECK_NULL(pInfo, code, lino, end, terrno);
2,319,385✔
1879
      void* tmp = taosArrayPush(pUidList, &pInfo->uid);
2,319,385✔
1880
      QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
2,319,385✔
1881
    }
1882
    terrno = 0;
344,206✔
1883
  } else {
1884
    qDebug("pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
11,537,877✔
1885

1886
    FilterCondType condType = checkTagCond(pTagCond);
11,538,470✔
1887
    if (((condType == FILTER_NO_LOGIC || condType == FILTER_AND) && status != SFLT_NOT_INDEX) || // (super table) use tagIndex and operator is and
11,536,127✔
1888
        (status == SFLT_NOT_INDEX && taosArrayGetSize(pUidTagList) > 0)) {                       // (child table with tagCond)
8,821,310✔
1889
      code = pAPI->metaFn.getTableTagsByUid(pVnode, pListInfo->idInfo.suid, pUidTagList);
2,961,848✔
1890
    } else {
1891
      taosArrayClear(pUidTagList);        // clear tablelist if using tagIndex and or condition
8,573,313✔
1892
      code = pAPI->metaFn.getTableTags(pVnode, pListInfo->idInfo.suid, pUidTagList);
8,575,743✔
1893
    }
1894
    if (code != TSDB_CODE_SUCCESS) {
11,538,981✔
1895
      qError("failed to get table tags from meta, reason:%s, suid:%" PRIu64, tstrerror(code), pListInfo->idInfo.suid);
×
1896
      terrno = code;
×
1897
      QUERY_CHECK_CODE(code, lino, end);
×
1898
    }
1899
  }
1900

1901
  qDebug("final pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
11,883,187✔
1902

1903
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
11,884,390✔
1904
  if (numOfTables == 0) {
11,884,049✔
1905
    goto end;
1,500,064✔
1906
  }
1907

1908
  SArray* pColList = NULL;
10,383,985✔
1909
  code = qGetColumnsFromNodeList(pTagCond, false, &pColList); 
10,384,326✔
1910
  if (code != TSDB_CODE_SUCCESS) {
10,376,545✔
1911
    goto end;
×
1912
  }
1913
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
10,376,545✔
1914
  taosArrayDestroy(pColList);
10,380,657✔
1915
  if (pResBlock == NULL) {
10,380,833✔
1916
    code = terrno;
×
1917
    QUERY_CHECK_CODE(code, lino, end);
×
1918
  }
1919

1920
  //fprintDataBlock(pResBlock, "tagFilter", "", 0);
1921

1922
  //  int64_t st1 = taosGetTimestampUs();
1923
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1924
  pBlockList = taosArrayInit(2, POINTER_BYTES);
10,380,833✔
1925
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
10,380,678✔
1926

1927
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
10,384,032✔
1928
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
10,384,032✔
1929

1930
  code = createResultData(&type, numOfTables, &output);
10,384,032✔
1931
  if (code != TSDB_CODE_SUCCESS) {
10,379,276✔
1932
    terrno = code;
×
1933
    QUERY_CHECK_CODE(code, lino, end);
×
1934
  }
1935

1936
  gTaskScalarExtra.pStreamInfo = pStreamInfo;
10,379,276✔
1937
  gTaskScalarExtra.pStreamRange = NULL;
10,379,276✔
1938
  code = scalarCalculate(pTagCond, pBlockList, &output, &gTaskScalarExtra);
10,372,507✔
1939
  if (code != TSDB_CODE_SUCCESS) {
10,373,181✔
1940
    qError("failed to calculate scalar, reason:%s", tstrerror(code));
876✔
1941
    terrno = code;
876✔
1942
    QUERY_CHECK_CODE(code, lino, end);
876✔
1943
  }
1944

1945
  code = doSetQualifiedUid(pListInfo, pUidList, pUidTagList, (bool*)output.columnData->pData, addUid);
10,372,305✔
1946
  if (code != TSDB_CODE_SUCCESS) {
10,380,493✔
1947
    terrno = code;
×
1948
    QUERY_CHECK_CODE(code, lino, end);
×
1949
  }
1950
  *listAdded = true;
10,380,493✔
1951

1952
end:
11,880,948✔
1953
  if (code != TSDB_CODE_SUCCESS) {
11,881,892✔
1954
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
876✔
1955
  }
1956
  blockDataDestroy(pResBlock);
11,881,892✔
1957
  taosArrayDestroy(pBlockList);
11,873,289✔
1958
  taosArrayDestroyEx(pUidTagList, freeItem);
11,869,550✔
1959

1960
  colDataDestroy(output.columnData);
11,880,266✔
1961
  taosMemoryFreeClear(output.columnData);
11,879,886✔
1962
  return code;
11,876,905✔
1963
}
1964

1965
typedef struct {
1966
  int32_t code;
1967
  SStreamRuntimeFuncInfo* pStreamRuntimeInfo;
1968
} PlaceHolderContext;
1969

1970
static EDealRes replacePlaceHolderColumn(SNode** pNode, void* pContext) {
64,770✔
1971
  PlaceHolderContext* pData = (PlaceHolderContext*)pContext;
64,770✔
1972
  if (QUERY_NODE_FUNCTION != nodeType((*pNode))) {
64,770✔
1973
    return DEAL_RES_CONTINUE;
53,096✔
1974
  }
1975
  SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
11,674✔
1976
  if (!fmIsStreamPesudoColVal(pFuncNode->funcId)) {
11,674✔
1977
    return DEAL_RES_CONTINUE;
472✔
1978
  }
1979
  pData->code = fmSetStreamPseudoFuncParamVal(pFuncNode->funcId, pFuncNode->pParameterList, pData->pStreamRuntimeInfo);
11,202✔
1980
  if (pData->code != TSDB_CODE_SUCCESS) {
11,202✔
1981
    return DEAL_RES_ERROR;
×
1982
  }
1983
  SNode* pFirstParam = nodesListGetNode(pFuncNode->pParameterList, 0);
11,202✔
1984
  ((SValueNode*)pFirstParam)->translate = true;
11,202✔
1985
  SValueNode* res = NULL;
11,202✔
1986
  pData->code = nodesCloneNode(pFirstParam, (SNode**)&res);
11,202✔
1987
  if (NULL == res) {
11,202✔
1988
    return DEAL_RES_ERROR;
×
1989
  }
1990
  nodesDestroyNode(*pNode);
11,202✔
1991
  *pNode = (SNode*)res;
11,202✔
1992

1993
  return DEAL_RES_CONTINUE;
11,202✔
1994
}
1995

1996
static void extractTagColId(SOperatorNode* pOpNode, SArray* pColIdArray) {
17,490✔
1997
  SNode* pLeft = pOpNode->pLeft;
17,490✔
1998
  SNode* pRight = pOpNode->pRight;
17,490✔
1999
  SColumnNode* pColNode = nodeType(pLeft) == QUERY_NODE_COLUMN ?
17,490✔
2000
    (SColumnNode*)pLeft : (SColumnNode*)pRight;
17,490✔
2001

2002
  col_id_t colId = pColNode->colId;
17,490✔
2003
  void* _tmp = taosArrayPush(pColIdArray, &colId);
17,490✔
2004
}
17,490✔
2005

2006
static int32_t buildTagCondKey(
8,745✔
2007
  const SNode* pTagCond, char** pTagCondKey,
2008
  int32_t* tagCondKeyLen, SArray** pTagColIds) {
2009
  if (NULL == pTagCond ||
8,745✔
2010
    (nodeType(pTagCond) != QUERY_NODE_OPERATOR &&
8,745✔
2011
      nodeType(pTagCond) != QUERY_NODE_LOGIC_CONDITION)) {
8,745✔
2012
    qError("invalid parameter to extract tag filter symbol");
×
2013
    return TSDB_CODE_INTERNAL_ERROR;
×
2014
  }
2015
  int32_t code = TSDB_CODE_SUCCESS;
8,745✔
2016
  int32_t lino = 0;
8,745✔
2017
  *pTagColIds = taosArrayInit(4, sizeof(col_id_t));
8,745✔
2018

2019
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
8,745✔
2020
    extractTagColId((SOperatorNode*)pTagCond, *pTagColIds);
×
2021
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
8,745✔
2022
    SNode* pChild = NULL;
8,745✔
2023
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
26,235✔
2024
      extractTagColId((SOperatorNode*)pChild, *pTagColIds);
17,490✔
2025
    }
2026
  }
2027

2028
  taosArraySort(*pTagColIds, compareUint16Val);
8,745✔
2029

2030
  // encode ordered colIds into key string, separated by ','
2031
  *tagCondKeyLen =
17,490✔
2032
    (int32_t)(taosArrayGetSize(*pTagColIds) * (sizeof(col_id_t) + 1) - 1);
8,745✔
2033
  *pTagCondKey = (char*)taosMemoryCalloc(1, *tagCondKeyLen);
8,745✔
2034
  TSDB_CHECK_NULL(*pTagCondKey, code, lino, _end, terrno);
8,745✔
2035
  char* pStart = *pTagCondKey;
8,745✔
2036
  for (int32_t i = 0; i < taosArrayGetSize(*pTagColIds); ++i) {
26,235✔
2037
    col_id_t* pColId = (col_id_t*)taosArrayGet(*pTagColIds, i);
17,490✔
2038
    TSDB_CHECK_NULL(pColId, code, lino, _end, terrno);
17,490✔
2039
    memcpy(pStart, pColId, sizeof(col_id_t));
17,490✔
2040
    pStart += sizeof(col_id_t);
17,490✔
2041
    if (i != taosArrayGetSize(*pTagColIds) - 1) {
17,490✔
2042
      *pStart = ',';
8,745✔
2043
      pStart += 1;
8,745✔
2044
    }
2045
  }
2046

2047
_end:
8,745✔
2048
  if (TSDB_CODE_SUCCESS != code) {
8,745✔
2049
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2050
    terrno = code;
×
2051
  }
2052
  return code;
8,745✔
2053
}
2054

2055
static EDealRes canOptimizeTagCondFilter(SNode* pTagCond, void* pContext) {
81,090✔
2056
  if (NULL == pTagCond) {
81,090✔
2057
    *(bool*)pContext = false;
×
2058
    return DEAL_RES_END;
×
2059
  }
2060
  if (nodeType(pTagCond) == QUERY_NODE_VALUE ||
81,090✔
2061
    nodeType(pTagCond) == QUERY_NODE_COLUMN) {
53,742✔
2062
    return DEAL_RES_CONTINUE;
44,997✔
2063
  }
2064
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR &&
36,093✔
2065
    ((SOperatorNode*)pTagCond)->opType == OP_TYPE_EQUAL) {
17,808✔
2066
    return DEAL_RES_CONTINUE;
17,490✔
2067
  }
2068
  if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION &&
19,239✔
2069
    ((SLogicConditionNode*)pTagCond)->condType == LOGIC_COND_TYPE_AND) {
8,745✔
2070
    return DEAL_RES_CONTINUE;
8,745✔
2071
  }
2072
  if (nodeType(pTagCond) == QUERY_NODE_FUNCTION &&
20,511✔
2073
    fmIsStreamPesudoColVal(((SFunctionNode*)pTagCond)->funcId)) {
10,017✔
2074
    return DEAL_RES_CONTINUE;
10,017✔
2075
  }
2076
  *(bool*)pContext = false;
477✔
2077
  return DEAL_RES_END;
477✔
2078
}
2079

2080
int32_t getTableList(void* pVnode, SScanPhysiNode* pScanNode, SNode* pTagCond, SNode* pTagIndexCond,
233,190,693✔
2081
                     STableListInfo* pListInfo, uint8_t* digest, const char* idstr, SStorageAPI* pStorageAPI, void* pStreamInfo) {
2082
  int32_t code = TSDB_CODE_SUCCESS;
233,190,693✔
2083
  int32_t lino = 0;
233,190,693✔
2084
  size_t  numOfTables = 0;
233,190,693✔
2085
  bool    listAdded = false;
233,190,693✔
2086

2087
  pListInfo->idInfo.suid = pScanNode->suid;
233,236,256✔
2088
  pListInfo->idInfo.tableType = pScanNode->tableType;
233,206,229✔
2089

2090
  SArray* pUidList = taosArrayInit(8, sizeof(uint64_t));
233,070,252✔
2091
  QUERY_CHECK_NULL(pUidList, code, lino, _error, terrno);
233,065,563✔
2092

2093
  char*   pTagCondKey = NULL;
233,065,563✔
2094
  int32_t tagCondKeyLen;
233,056,321✔
2095
  SArray* pTagColIds = NULL;
233,096,846✔
2096
  char*   pPayload = NULL;
233,114,660✔
2097
  qTrace("getTableList called, suid:%" PRIu64
233,114,660✔
2098
    ", tagCond:%p, tagIndexCond:%p, %d %d", pScanNode->suid, pTagCond,
2099
    pTagIndexCond, pScanNode->tableType, pScanNode->virtualStableScan);
2100
  if (pScanNode->tableType != TSDB_SUPER_TABLE && !pScanNode->virtualStableScan) {
233,114,660✔
2101
    pListInfo->idInfo.uid = pScanNode->uid;
133,705,938✔
2102
    if (pStorageAPI->metaFn.isTableExisted(pVnode, pScanNode->uid)) {
133,668,593✔
2103
      void* tmp = taosArrayPush(pUidList, &pScanNode->uid);
133,705,521✔
2104
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
133,704,054✔
2105
    }
2106
    code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, SFLT_NOT_INDEX, pStorageAPI, false, &listAdded, pStreamInfo);
133,749,651✔
2107
    QUERY_CHECK_CODE(code, lino, _end);
133,735,952✔
2108
  } else {
2109
    bool      isStream = (pStreamInfo != NULL);
99,491,260✔
2110
    bool      hasTagCond = (pTagCond != NULL);
99,491,260✔
2111
    bool      canCacheTagEqCondFilter = false;
99,491,260✔
2112
    T_MD5_CTX context = {0};
99,496,814✔
2113

2114
    qTrace("start to get table list by tag filter, suid:%" PRIu64
99,543,611✔
2115
      ",tsStableTagFilterCache:%d, tsTagFilterCache:%d", 
2116
      pScanNode->suid, tsStableTagFilterCache, tsTagFilterCache);
2117

2118
    bool acquired = false;
99,543,611✔
2119
    // first, check whether we can use stable tag filter cache
2120
    if (tsStableTagFilterCache && isStream && hasTagCond) {
99,497,041✔
2121
      canCacheTagEqCondFilter = true;
9,063✔
2122
      nodesWalkExpr(pTagCond, canOptimizeTagCondFilter,
9,063✔
2123
        (void*)&canCacheTagEqCondFilter);
2124
    }
2125
    if (canCacheTagEqCondFilter) {
99,444,457✔
2126
      qDebug("%s, stable tag filter condition can be optimized", idstr);
8,745✔
2127
      if (((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
8,745✔
2128
        SNode* tmp = NULL;
8,745✔
2129
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
8,745✔
2130
        QUERY_CHECK_CODE(code, lino, _error);
8,745✔
2131

2132
        PlaceHolderContext ctx = {.code = TSDB_CODE_SUCCESS, .pStreamRuntimeInfo = (SStreamRuntimeFuncInfo*)pStreamInfo};
8,745✔
2133
        nodesRewriteExpr(&tmp, replacePlaceHolderColumn, (void*)&ctx);
8,745✔
2134
        if (TSDB_CODE_SUCCESS != ctx.code) {
8,745✔
2135
          nodesDestroyNode(tmp);
×
2136
          code = ctx.code;
×
2137
          goto _error;
×
2138
        }
2139
        code = genStableTagFilterDigest(tmp, &context);
8,745✔
2140
        nodesDestroyNode(tmp);
8,745✔
2141
      } else {
2142
        code = genStableTagFilterDigest(pTagCond, &context);
×
2143
      }
2144
      QUERY_CHECK_CODE(code, lino, _error);
8,745✔
2145

2146
      code = buildTagCondKey(
8,745✔
2147
        pTagCond, &pTagCondKey, &tagCondKeyLen, &pTagColIds);
2148
      QUERY_CHECK_CODE(code, lino, _error);
8,745✔
2149
      code = pStorageAPI->metaFn.getStableCachedTableList(
8,745✔
2150
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
8,745✔
2151
        context.digest, tListLen(context.digest), pUidList, &acquired);
2152
      QUERY_CHECK_CODE(code, lino, _error);
8,745✔
2153
    } else if (tsTagFilterCache) {
99,435,712✔
2154
      // second, try to use normal tag filter cache
2155
      qDebug("%s using normal tag filter cache", idstr);
29,447✔
2156
      if (pStreamInfo != NULL && ((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
30,632✔
2157
        SNode* tmp = NULL;
1,185✔
2158
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
1,185✔
2159
        QUERY_CHECK_CODE(code, lino, _error);
1,185✔
2160

2161
        PlaceHolderContext ctx = {.code = TSDB_CODE_SUCCESS, .pStreamRuntimeInfo = (SStreamRuntimeFuncInfo*)pStreamInfo};
1,185✔
2162
        nodesRewriteExpr(&tmp, replacePlaceHolderColumn, (void*)&ctx);
1,185✔
2163
        if (TSDB_CODE_SUCCESS != ctx.code) {
1,185✔
2164
          nodesDestroyNode(tmp);
×
2165
          code = ctx.code;
×
2166
          goto _error;
×
2167
        }
2168
        code = genTagFilterDigest(tmp, &context);
1,185✔
2169
        nodesDestroyNode(tmp);
1,185✔
2170
      } else {
2171
        code = genTagFilterDigest(pTagCond, &context);
28,262✔
2172
      }
2173
      // try to retrieve the result from meta cache
2174
      QUERY_CHECK_CODE(code, lino, _error);      
29,447✔
2175
      code = pStorageAPI->metaFn.getCachedTableList(
29,447✔
2176
        pVnode, pScanNode->suid, context.digest,
29,447✔
2177
        tListLen(context.digest), pUidList, &acquired);
2178
      QUERY_CHECK_CODE(code, lino, _error);
63,663✔
2179
    }
2180
    if (acquired) {
99,478,673✔
2181
      taosArrayDestroy(pTagColIds);
26,528✔
2182
      pTagColIds = NULL;
26,528✔
2183
      
2184
      digest[0] = 1;
26,528✔
2185
      memcpy(
53,056✔
2186
        digest + 1, context.digest, tListLen(context.digest));
26,528✔
2187
      qDebug("suid:%" PRIu64 ", %s retrieve table uid list from cache,"
26,528✔
2188
        " numOfTables:%d", 
2189
        pScanNode->suid, idstr, (int32_t)taosArrayGetSize(pUidList));
2190
      goto _end;
26,528✔
2191
    } else {
2192
      qDebug("suid:%" PRIu64 
99,452,145✔
2193
        ", failed to get table uid list from cache", pScanNode->suid);
2194
    }
2195

2196
    if (!pTagCond) {  // no tag filter condition exists, let's fetch all tables of this super table
99,519,810✔
2197
      code = pStorageAPI->metaFn.getChildTableList(pVnode, pScanNode->suid, pUidList);
87,889,617✔
2198
      QUERY_CHECK_CODE(code, lino, _error);
87,873,593✔
2199
      qTrace("no tag filter, get all child tables, numOfTables:%d", (int32_t)taosArrayGetSize(pUidList));
87,873,593✔
2200
    } else {
2201
      SIdxFltStatus status = SFLT_NOT_INDEX;
11,630,193✔
2202
      if (pTagIndexCond) {
11,631,681✔
2203
        void* pIndex = pStorageAPI->metaFn.getInvertIndex(pVnode);
3,743,787✔
2204

2205
        SIndexMetaArg metaArg = {.metaEx = pVnode,
3,743,815✔
2206
                                 .idx = pStorageAPI->metaFn.storeGetIndexInfo(pVnode),
3,743,787✔
2207
                                 .ivtIdx = pIndex,
2208
                                 .suid = pScanNode->uid};
3,743,319✔
2209
        code = doFilterTag(pTagIndexCond, &metaArg, pUidList, &status, &pStorageAPI->metaFilter);
3,743,319✔
2210
        if (code != 0 || status == SFLT_NOT_INDEX) {  // temporarily disable it for performance sake
3,739,198✔
2211
          qDebug("failed to get tableIds from index, suid:%" PRIu64 ", uidListSize:%d", pScanNode->uid, (int32_t)taosArrayGetSize(pUidList));
1,021,889✔
2212
        } else {
2213
          qDebug("succ to get filter result, table num: %d", (int)taosArrayGetSize(pUidList));
2,717,309✔
2214
        }
2215
      }
2216
      qTrace("after index filter, pTagCond:%p uidListSize:%d", pTagCond, (int32_t)taosArrayGetSize(pUidList));
11,631,681✔
2217
      code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, status,
11,631,005✔
2218
        pStorageAPI, tsTagFilterCache || tsStableTagFilterCache,
11,631,005✔
2219
        &listAdded, pStreamInfo);
2220
      QUERY_CHECK_CODE(code, lino, _error);
11,625,243✔
2221
    }
2222
    // let's add the filter results into meta-cache
2223
    numOfTables = taosArrayGetSize(pUidList);
99,498,859✔
2224

2225
    if (canCacheTagEqCondFilter) {
99,506,063✔
2226
      qInfo("%s, suid:%" PRIu64 ", add uid list to stable tag filter cache, "
4,452✔
2227
            "uidListSize:%d, origin key:%" PRIu64 ",%" PRIu64,
2228
            idstr, pScanNode->suid, (int32_t)numOfTables,
2229
            *(uint64_t*)context.digest, *(uint64_t*)(context.digest + 8));
2230

2231
      code = pStorageAPI->metaFn.putStableCachedTableList(
4,452✔
2232
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
2233
        context.digest, tListLen(context.digest),
2234
        pUidList, &pTagColIds);
2235
      QUERY_CHECK_CODE(code, lino, _end);
4,452✔
2236

2237
      digest[0] = 1;
4,452✔
2238
      memcpy(digest + 1, context.digest, tListLen(context.digest));
4,452✔
2239
    } else if (tsTagFilterCache) {
99,501,611✔
2240
      qInfo("%s, suid:%" PRIu64 ", add uid list to normal tag filter cache, "
7,212✔
2241
            "uidListSize:%d, origin key:%" PRIu64 ",%" PRIu64,
2242
            idstr, pScanNode->suid, (int32_t)numOfTables,
2243
            *(uint64_t*)context.digest, *(uint64_t*)(context.digest + 8));
2244
      size_t size = numOfTables * sizeof(uint64_t) + sizeof(int32_t);
7,212✔
2245
      pPayload = taosMemoryMalloc(size);
7,212✔
2246
      QUERY_CHECK_NULL(pPayload, code, lino, _end, terrno);
7,212✔
2247

2248
      *(int32_t*)pPayload = (int32_t)numOfTables;
7,212✔
2249
      if (numOfTables > 0) {
7,212✔
2250
        void* tmp = taosArrayGet(pUidList, 0);
5,945✔
2251
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
5,945✔
2252
        memcpy(pPayload + sizeof(int32_t), tmp, numOfTables * sizeof(uint64_t));
5,945✔
2253
      }
2254

2255
      code = pStorageAPI->metaFn.putCachedTableList(pVnode, pScanNode->suid,
7,212✔
2256
                                                    context.digest,
2257
                                                    tListLen(context.digest),
2258
                                                    pPayload, size, 1);
2259
      if (TSDB_CODE_SUCCESS == code) {
7,212✔
2260
        /*
2261
          data referenced by pPayload is used in lru cache,
2262
          reset pPayload to NULL to avoid being freed in _error block
2263
        */
2264
        pPayload = NULL;
7,212✔
2265
      } else {
2266
        if (TSDB_CODE_DUP_KEY == code) {
×
2267
          /*
2268
            another thread has already put the same key into cache,
2269
            we can just ignore this error
2270
          */
2271
          code = TSDB_CODE_SUCCESS;
×
2272
        }
2273
        QUERY_CHECK_CODE(code, lino, _end);
×
2274
      }
2275

2276

2277
      digest[0] = 1;
7,212✔
2278
      memcpy(digest + 1, context.digest, tListLen(context.digest));
7,212✔
2279
    }
2280
  }
2281

2282
_end:
233,242,096✔
2283
  if (!listAdded) {
233,247,378✔
2284
    numOfTables = taosArrayGetSize(pUidList);
222,887,063✔
2285
    for (int i = 0; i < numOfTables; i++) {
759,936,878✔
2286
      void* tmp = taosArrayGet(pUidList, i);
537,002,944✔
2287
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
537,027,172✔
2288
      STableKeyInfo info = {.uid = *(uint64_t*)tmp, .groupId = 0};
537,027,172✔
2289

2290
      void* p = taosArrayPush(pListInfo->pTableList, &info);
537,026,371✔
2291
      if (p == NULL) {
537,065,072✔
2292
        taosArrayDestroy(pUidList);
×
2293
        return terrno;
×
2294
      }
2295

2296
      qTrace("tagfilter get uid:%" PRIu64 ", %s", info.uid, idstr);
537,065,072✔
2297
    }
2298
  }
2299

2300
  qDebug("%s, table list with %d uids built", idstr, (int32_t)numOfTables);
233,294,249✔
2301

2302
_error:
233,292,617✔
2303
  taosArrayDestroy(pUidList);
233,294,029✔
2304
  taosArrayDestroy(pTagColIds);
233,234,634✔
2305
  taosMemFreeClear(pTagCondKey);
233,242,212✔
2306
  taosMemFreeClear(pPayload);
233,242,212✔
2307
  if (code != TSDB_CODE_SUCCESS) {
233,242,212✔
2308
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
876✔
2309
  }
2310
  return code;
233,233,073✔
2311
}
2312

2313
int32_t qGetTableList(int64_t suid, void* pVnode, void* node, SArray** tableList, void* pTaskInfo) {
19,584✔
2314
  int32_t        code = TSDB_CODE_SUCCESS;
19,584✔
2315
  int32_t        lino = 0;
19,584✔
2316
  SSubplan*      pSubplan = (SSubplan*)node;
19,584✔
2317
  SScanPhysiNode pNode = {0};
19,584✔
2318
  pNode.suid = suid;
19,584✔
2319
  pNode.uid = suid;
19,584✔
2320
  pNode.tableType = TSDB_SUPER_TABLE;
19,584✔
2321

2322
  STableListInfo* pTableListInfo = tableListCreate();
19,584✔
2323
  QUERY_CHECK_NULL(pTableListInfo, code, lino, _end, terrno);
19,584✔
2324
  uint8_t digest[17] = {0};
19,584✔
2325
  code = getTableList(pVnode, &pNode, pSubplan ? pSubplan->pTagCond : NULL, pSubplan ? pSubplan->pTagIndexCond : NULL,
19,584✔
2326
                      pTableListInfo, digest, "qGetTableList", &((SExecTaskInfo*)pTaskInfo)->storageAPI, NULL);
2327
  QUERY_CHECK_CODE(code, lino, _end);
19,584✔
2328
  *tableList = pTableListInfo->pTableList;
19,584✔
2329
  pTableListInfo->pTableList = NULL;
19,584✔
2330
  tableListDestroy(pTableListInfo);
19,584✔
2331

2332
_end:
19,584✔
2333
  if (code != TSDB_CODE_SUCCESS) {
19,584✔
2334
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2335
  }
2336
  return code;
19,584✔
2337
}
2338

2339
size_t getTableTagsBufLen(const SNodeList* pGroups) {
×
2340
  size_t keyLen = 0;
×
2341

2342
  SNode* node;
2343
  FOREACH(node, pGroups) {
×
2344
    SExprNode* pExpr = (SExprNode*)node;
×
2345
    keyLen += pExpr->resType.bytes;
×
2346
  }
2347

2348
  keyLen += sizeof(int8_t) * LIST_LENGTH(pGroups);
×
2349
  return keyLen;
×
2350
}
2351

2352
int32_t getGroupIdFromTagsVal(void* pVnode, uint64_t uid, SNodeList* pGroupNode, char* keyBuf, uint64_t* pGroupId,
×
2353
                              SStorageAPI* pAPI) {
2354
  SMetaReader mr = {0};
×
2355

2356
  pAPI->metaReaderFn.initReader(&mr, pVnode, META_READER_LOCK, &pAPI->metaFn);
×
2357
  if (pAPI->metaReaderFn.getEntryGetUidCache(&mr, uid) != 0) {  // table not exist
×
2358
    pAPI->metaReaderFn.clearReader(&mr);
×
2359
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
2360
  }
2361

2362
  SNodeList* groupNew = NULL;
×
2363
  int32_t    code = nodesCloneList(pGroupNode, &groupNew);
×
2364
  if (TSDB_CODE_SUCCESS != code) {
×
2365
    pAPI->metaReaderFn.clearReader(&mr);
×
2366
    return code;
×
2367
  }
2368

2369
  STransTagExprCtx ctx = {.code = 0, .pReader = &mr};
×
2370
  nodesRewriteExprsPostOrder(groupNew, doTranslateTagExpr, &ctx);
×
2371
  if (TSDB_CODE_SUCCESS != ctx.code) {
×
2372
    nodesDestroyList(groupNew);
×
2373
    pAPI->metaReaderFn.clearReader(&mr);
×
2374
    return code;
×
2375
  }
2376
  char* isNull = (char*)keyBuf;
×
2377
  char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(pGroupNode);
×
2378

2379
  SNode*  pNode;
2380
  int32_t index = 0;
×
2381
  FOREACH(pNode, groupNew) {
×
2382
    SNode*  pNew = NULL;
×
2383
    int32_t code = scalarCalculateConstants(pNode, &pNew);
×
2384
    if (TSDB_CODE_SUCCESS == code) {
×
2385
      REPLACE_NODE(pNew);
×
2386
    } else {
2387
      nodesDestroyList(groupNew);
×
2388
      pAPI->metaReaderFn.clearReader(&mr);
×
2389
      return code;
×
2390
    }
2391

2392
    if (nodeType(pNew) != QUERY_NODE_VALUE) {
×
2393
      nodesDestroyList(groupNew);
×
2394
      pAPI->metaReaderFn.clearReader(&mr);
×
2395
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2396
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2397
    }
2398
    SValueNode* pValue = (SValueNode*)pNew;
×
2399

2400
    if (pValue->node.resType.type == TSDB_DATA_TYPE_NULL || pValue->isNull) {
×
2401
      isNull[index++] = 1;
×
2402
      continue;
×
2403
    } else {
2404
      isNull[index++] = 0;
×
2405
      char* data = nodesGetValueFromNode(pValue);
×
2406
      if (pValue->node.resType.type == TSDB_DATA_TYPE_JSON) {
×
2407
        if (tTagIsJson(data)) {
×
2408
          terrno = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
×
2409
          nodesDestroyList(groupNew);
×
2410
          pAPI->metaReaderFn.clearReader(&mr);
×
2411
          return terrno;
×
2412
        }
2413
        int32_t len = getJsonValueLen(data);
×
2414
        memcpy(pStart, data, len);
×
2415
        pStart += len;
×
2416
      } else if (IS_VAR_DATA_TYPE(pValue->node.resType.type)) {
×
2417
        if (IS_STR_DATA_BLOB(pValue->node.resType.type)) {
×
2418
          return TSDB_CODE_BLOB_NOT_SUPPORT_TAG;
×
2419
        }
2420
        memcpy(pStart, data, varDataTLen(data));
×
2421
        pStart += varDataTLen(data);
×
2422
      } else {
2423
        memcpy(pStart, data, pValue->node.resType.bytes);
×
2424
        pStart += pValue->node.resType.bytes;
×
2425
      }
2426
    }
2427
  }
2428

2429
  int32_t len = (int32_t)(pStart - (char*)keyBuf);
×
2430
  *pGroupId = calcGroupId(keyBuf, len);
×
2431

2432
  nodesDestroyList(groupNew);
×
2433
  pAPI->metaReaderFn.clearReader(&mr);
×
2434

2435
  return TSDB_CODE_SUCCESS;
×
2436
}
2437

2438
SArray* makeColumnArrayFromList(SNodeList* pNodeList) {
7,380,177✔
2439
  if (!pNodeList) {
7,380,177✔
2440
    return NULL;
×
2441
  }
2442

2443
  size_t  numOfCols = LIST_LENGTH(pNodeList);
7,380,177✔
2444
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
7,379,820✔
2445
  if (pList == NULL) {
7,378,880✔
2446
    return NULL;
×
2447
  }
2448

2449
  for (int32_t i = 0; i < numOfCols; ++i) {
16,907,824✔
2450
    SColumnNode* pColNode = (SColumnNode*)nodesListGetNode(pNodeList, i);
9,529,301✔
2451
    if (!pColNode) {
9,529,999✔
2452
      taosArrayDestroy(pList);
×
2453
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2454
      return NULL;
×
2455
    }
2456

2457
    // todo extract method
2458
    SColumn c = {0};
9,529,999✔
2459
    c.slotId = pColNode->slotId;
9,529,301✔
2460
    c.colId = pColNode->colId;
9,528,896✔
2461
    c.type = pColNode->node.resType.type;
9,530,089✔
2462
    c.bytes = pColNode->node.resType.bytes;
9,530,089✔
2463
    c.precision = pColNode->node.resType.precision;
9,528,306✔
2464
    c.scale = pColNode->node.resType.scale;
9,529,748✔
2465

2466
    void* tmp = taosArrayPush(pList, &c);
9,528,854✔
2467
    if (!tmp) {
9,528,854✔
2468
      taosArrayDestroy(pList);
×
2469
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2470
      return NULL;
×
2471
    }
2472
  }
2473

2474
  return pList;
7,378,523✔
2475
}
2476

2477
int32_t extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols,
304,468,632✔
2478
                            int32_t type, SColMatchInfo* pMatchInfo) {
2479
  size_t  numOfCols = LIST_LENGTH(pNodeList);
304,468,632✔
2480
  int32_t code = TSDB_CODE_SUCCESS;
304,479,805✔
2481
  int32_t lino = 0;
304,479,805✔
2482

2483
  pMatchInfo->matchType = type;
304,479,805✔
2484

2485
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchItem));
304,473,741✔
2486
  if (pList == NULL) {
304,354,390✔
2487
    code = terrno;
×
2488
    return code;
×
2489
  }
2490

2491
  for (int32_t i = 0; i < numOfCols; ++i) {
1,142,181,453✔
2492
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
837,722,546✔
2493
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
837,826,741✔
2494
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
837,826,741✔
2495
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
828,683,265✔
2496

2497
      SColMatchItem c = {.needOutput = true};
828,724,216✔
2498
      c.colId = pColNode->colId;
828,716,960✔
2499
      c.srcSlotId = pColNode->slotId;
828,687,336✔
2500
      c.dstSlotId = pNode->slotId;
828,669,306✔
2501
      c.isPk = pColNode->isPk;
828,726,425✔
2502
      c.dataType = pColNode->node.resType;
828,673,416✔
2503
      void* tmp = taosArrayPush(pList, &c);
828,685,128✔
2504
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
828,685,128✔
2505
    }
2506
  }
2507

2508
  // set the output flag for each column in SColMatchInfo, according to the
2509
  *numOfOutputCols = 0;
304,458,907✔
2510
  int32_t num = LIST_LENGTH(pOutputNodeList->pSlots);
304,532,424✔
2511
  for (int32_t i = 0; i < num; ++i) {
1,314,169,456✔
2512
    SSlotDescNode* pNode = (SSlotDescNode*)nodesListGetNode(pOutputNodeList->pSlots, i);
1,009,676,173✔
2513
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,009,739,924✔
2514

2515
    // todo: add reserve flag check
2516
    // it is a column reserved for the arithmetic expression calculation
2517
    if (pNode->slotId >= numOfCols) {
1,009,739,924✔
2518
      (*numOfOutputCols) += 1;
171,942,249✔
2519
      continue;
171,942,425✔
2520
    }
2521

2522
    SColMatchItem* info = NULL;
837,838,623✔
2523
    for (int32_t j = 0; j < taosArrayGetSize(pList); ++j) {
2,147,483,647✔
2524
      info = taosArrayGet(pList, j);
2,147,483,647✔
2525
      QUERY_CHECK_NULL(info, code, lino, _end, terrno);
2,147,483,647✔
2526
      if (info->dstSlotId == pNode->slotId) {
2,147,483,647✔
2527
        break;
827,893,429✔
2528
      }
2529
    }
2530

2531
    if (pNode->output) {
146,528,244✔
2532
      (*numOfOutputCols) += 1;
826,477,313✔
2533
    } else if (info != NULL) {
11,302,481✔
2534
      // select distinct tbname from stb where tbname='abc';
2535
      info->needOutput = false;
11,310,471✔
2536
    }
2537
  }
2538

2539
  pMatchInfo->pList = pList;
304,493,283✔
2540

2541
_end:
304,520,876✔
2542
  if (code != TSDB_CODE_SUCCESS) {
304,520,876✔
2543
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2544
  }
2545
  return code;
304,478,641✔
2546
}
2547

2548
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
840,502,502✔
2549
                                  const char* name) {
2550
  SResSchema s = {0};
840,502,502✔
2551
  s.scale = scale;
840,554,887✔
2552
  s.type = type;
840,554,887✔
2553
  s.bytes = bytes;
840,554,887✔
2554
  s.slotId = slotId;
840,554,887✔
2555
  s.precision = precision;
840,554,887✔
2556
  tstrncpy(s.name, name, tListLen(s.name));
840,554,887✔
2557

2558
  return s;
840,554,887✔
2559
}
2560

2561
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType, EColumnType colType) {
745,144,910✔
2562
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
745,144,910✔
2563
  if (pCol == NULL) {
744,777,560✔
2564
    return NULL;
×
2565
  }
2566

2567
  pCol->slotId = slotId;
744,777,560✔
2568
  pCol->colId = colId;
744,811,900✔
2569
  pCol->bytes = pType->bytes;
744,860,559✔
2570
  pCol->type = pType->type;
744,941,384✔
2571
  pCol->scale = pType->scale;
745,092,930✔
2572
  pCol->precision = pType->precision;
745,049,201✔
2573
  pCol->dataBlockId = blockId;
745,173,177✔
2574
  pCol->colType = colType;
745,159,298✔
2575
  return pCol;
745,168,481✔
2576
}
2577

2578
int32_t createExprFromOneNode(SExprInfo* pExp, SNode* pNode, int16_t slotId) {
844,542,724✔
2579
  int32_t code = TSDB_CODE_SUCCESS;
844,542,724✔
2580
  int32_t lino = 0;
844,542,724✔
2581
  pExp->base.numOfParams = 0;
844,542,724✔
2582
  pExp->base.pParam = NULL;
844,595,539✔
2583
  pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
844,532,987✔
2584
  QUERY_CHECK_NULL(pExp->pExpr, code, lino, _end, terrno);
844,049,881✔
2585

2586
  pExp->pExpr->_function.num = 1;
844,173,099✔
2587
  pExp->pExpr->_function.functionId = -1;
844,293,904✔
2588

2589
  int32_t type = nodeType(pNode);
844,390,122✔
2590
  // it is a project query, or group by column
2591
  if (type == QUERY_NODE_COLUMN) {
844,451,960✔
2592
    pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
412,015,464✔
2593
    SColumnNode* pColNode = (SColumnNode*)pNode;
412,054,211✔
2594

2595
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
412,054,211✔
2596
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
411,933,307✔
2597

2598
    pExp->base.numOfParams = 1;
411,960,997✔
2599

2600
    SDataType* pType = &pColNode->node.resType;
411,972,794✔
2601
    pExp->base.resSchema =
2602
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pColNode->colName);
412,016,228✔
2603

2604
    pExp->base.pParam[0].pCol =
824,049,178✔
2605
        createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType, pColNode->colType);
824,039,075✔
2606
    QUERY_CHECK_NULL(pExp->base.pParam[0].pCol, code, lino, _end, terrno);
412,034,871✔
2607

2608
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
411,906,349✔
2609
  } else if (type == QUERY_NODE_VALUE) {
432,436,496✔
2610
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
21,244,077✔
2611
    SValueNode* pValNode = (SValueNode*)pNode;
21,241,750✔
2612

2613
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
21,241,750✔
2614
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
21,236,256✔
2615

2616
    pExp->base.numOfParams = 1;
21,240,361✔
2617

2618
    SDataType* pType = &pValNode->node.resType;
21,242,325✔
2619
    pExp->base.resSchema =
2620
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
21,237,686✔
2621
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
21,240,719✔
2622
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
21,240,180✔
2623
    QUERY_CHECK_CODE(code, lino, _end);
21,242,838✔
2624
  } else if (type == QUERY_NODE_REMOTE_VALUE) {
411,192,419✔
2625
    SRemoteValueNode* pRemote = (SRemoteValueNode*)pNode;
23,515,984✔
2626
    code = qFetchRemoteValue(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, pRemote);
23,515,984✔
2627
    QUERY_CHECK_CODE(code, lino, _end);
23,522,901✔
2628

2629
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
19,544,281✔
2630
    SValueNode* pValNode = (SValueNode*)pNode;
19,544,741✔
2631

2632
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
19,544,741✔
2633
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
19,544,286✔
2634

2635
    pExp->base.numOfParams = 1;
19,543,826✔
2636

2637
    SDataType* pType = &pValNode->node.resType;
19,545,201✔
2638
    pExp->base.resSchema =
2639
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
19,544,291✔
2640
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
19,544,286✔
2641
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
19,544,286✔
2642
    QUERY_CHECK_CODE(code, lino, _end);
19,544,741✔
2643
  } else if (type == QUERY_NODE_FUNCTION) {
387,676,435✔
2644
    pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
351,484,913✔
2645
    SFunctionNode* pFuncNode = (SFunctionNode*)pNode;
351,484,155✔
2646

2647
    SDataType* pType = &pFuncNode->node.resType;
351,484,155✔
2648
    pExp->base.resSchema =
2649
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pFuncNode->node.aliasName);
351,500,445✔
2650
    tExprNode* pExprNode = pExp->pExpr;
351,475,616✔
2651

2652
    pExprNode->_function.functionId = pFuncNode->funcId;
351,468,236✔
2653
    pExprNode->_function.pFunctNode = pFuncNode;
351,513,174✔
2654
    pExprNode->_function.functionType = pFuncNode->funcType;
351,521,679✔
2655

2656
    tstrncpy(pExprNode->_function.functionName, pFuncNode->functionName, tListLen(pExprNode->_function.functionName));
351,487,964✔
2657

2658
    pExp->base.pParamList = pFuncNode->pParameterList;
351,498,131✔
2659
#if 1
2660
    // todo refactor: add the parameter for tbname function
2661
    const char* name = "tbname";
351,516,849✔
2662
    int32_t     len = strlen(name);
351,516,849✔
2663

2664
    if (!pFuncNode->pParameterList && (memcmp(pExprNode->_function.functionName, name, len) == 0) &&
351,516,849✔
2665
        pExprNode->_function.functionName[len] == 0) {
24,978,044✔
2666
      pFuncNode->pParameterList = NULL;
24,976,489✔
2667
      int32_t     code = nodesMakeList(&pFuncNode->pParameterList);
24,977,197✔
2668
      SValueNode* res = NULL;
24,981,939✔
2669
      if (TSDB_CODE_SUCCESS == code) {
24,981,939✔
2670
        code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
24,982,905✔
2671
      }
2672
      QUERY_CHECK_CODE(code, lino, _end);
24,984,116✔
2673
      res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
24,984,116✔
2674
      code = nodesListAppend(pFuncNode->pParameterList, (SNode*)res);
24,970,835✔
2675
      if (code != TSDB_CODE_SUCCESS) {
24,978,527✔
2676
        nodesDestroyNode((SNode*)res);
×
2677
        res = NULL;
×
2678
      }
2679
      QUERY_CHECK_CODE(code, lino, _end);
24,978,527✔
2680
    }
2681
#endif
2682

2683
    int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
351,430,009✔
2684

2685
    pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
351,528,847✔
2686
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
351,427,369✔
2687
    pExp->base.numOfParams = numOfParam;
351,429,108✔
2688

2689
    for (int32_t j = 0; j < numOfParam && TSDB_CODE_SUCCESS == code; ++j) {
837,333,242✔
2690
      SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
486,093,354✔
2691
      QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
486,119,324✔
2692
      if (p1->type == QUERY_NODE_COLUMN) {
486,119,324✔
2693
        SColumnNode* pcn = (SColumnNode*)p1;
333,098,022✔
2694

2695
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
333,098,022✔
2696
        pExp->base.pParam[j].pCol =
666,142,645✔
2697
            createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType, pcn->colType);
666,185,514✔
2698
        QUERY_CHECK_NULL(pExp->base.pParam[j].pCol, code, lino, _end, terrno);
333,076,075✔
2699
      } else if (p1->type == QUERY_NODE_VALUE) {
153,043,454✔
2700
        SValueNode* pvn = (SValueNode*)p1;
98,350,689✔
2701
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
98,350,689✔
2702
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
98,338,469✔
2703
        QUERY_CHECK_CODE(code, lino, _end);
98,323,163✔
2704
      } else if (p1->type == QUERY_NODE_REMOTE_VALUE) {
54,727,740✔
2705
        SRemoteValueNode* pRemote = (SRemoteValueNode*)p1;
1,191,977✔
2706
        code = qFetchRemoteValue(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, pRemote);
1,191,977✔
2707
        QUERY_CHECK_CODE(code, lino, _end);
1,191,977✔
2708

2709
        SValueNode* pvn = (SValueNode*)pRemote;
1,010,206✔
2710
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
1,010,206✔
2711
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
1,010,206✔
2712
        QUERY_CHECK_CODE(code, lino, _end);
951,872✔
2713
      }
2714
    }
2715
    pExp->pExpr->_function.bindExprID = ((SExprNode*)pNode)->bindExprID;
351,239,888✔
2716
  } else if (type == QUERY_NODE_OPERATOR) {
36,191,522✔
2717
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
31,428,441✔
2718
    SOperatorNode* pOpNode = (SOperatorNode*)pNode;
31,425,543✔
2719

2720
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
31,425,543✔
2721
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
31,415,819✔
2722
    pExp->base.numOfParams = 1;
31,416,324✔
2723

2724
    SDataType* pType = &pOpNode->node.resType;
31,423,128✔
2725
    pExp->base.resSchema =
2726
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pOpNode->node.aliasName);
31,415,940✔
2727
    pExp->pExpr->_optrRoot.pRootNode = pNode;
31,419,383✔
2728
  } else if (type == QUERY_NODE_CASE_WHEN) {
4,763,081✔
2729
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
4,755,851✔
2730
    SCaseWhenNode* pCaseNode = (SCaseWhenNode*)pNode;
4,755,851✔
2731

2732
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
4,755,851✔
2733
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
4,755,851✔
2734
    pExp->base.numOfParams = 1;
4,755,851✔
2735

2736
    SDataType* pType = &pCaseNode->node.resType;
4,755,851✔
2737
    pExp->base.resSchema =
2738
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCaseNode->node.aliasName);
4,755,851✔
2739
    pExp->pExpr->_optrRoot.pRootNode = pNode;
4,755,851✔
2740
  } else if (type == QUERY_NODE_LOGIC_CONDITION) {
7,230✔
2741
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
2,970✔
2742
    SLogicConditionNode* pCond = (SLogicConditionNode*)pNode;
2,970✔
2743
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
2,970✔
2744
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
2,970✔
2745
    pExp->base.numOfParams = 1;
2,970✔
2746
    SDataType* pType = &pCond->node.resType;
2,970✔
2747
    pExp->base.resSchema =
2748
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCond->node.aliasName);
2,970✔
2749
    pExp->pExpr->_optrRoot.pRootNode = pNode;
2,970✔
2750
  } else {
2751
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
4,260✔
2752
    QUERY_CHECK_CODE(code, lino, _end);
4,260✔
2753
  }
2754
  pExp->pExpr->relatedTo = ((SExprNode*)pNode)->relatedTo;
840,315,628✔
2755
_end:
844,459,127✔
2756
  if (code != TSDB_CODE_SUCCESS) {
844,459,127✔
2757
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
4,160,391✔
2758
  }
2759
  return code;
844,456,289✔
2760
}
2761

2762
int32_t createExprFromTargetNode(SExprInfo* pExp, STargetNode* pTargetNode) {
844,487,076✔
2763
  return createExprFromOneNode(pExp, pTargetNode->pExpr, pTargetNode->slotId);
844,487,076✔
2764
}
2765

2766
SExprInfo* createExpr(SNodeList* pNodeList, int32_t* numOfExprs) {
×
2767
  *numOfExprs = LIST_LENGTH(pNodeList);
×
2768
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
×
2769
  if (!pExprs) {
×
2770
    return NULL;
×
2771
  }
2772

2773
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
×
2774
    SExprInfo* pExp = &pExprs[i];
×
2775
    int32_t    code = createExprFromOneNode(pExp, nodesListGetNode(pNodeList, i), i + UD_TAG_COLUMN_INDEX);
×
2776
    if (code != TSDB_CODE_SUCCESS) {
×
2777
      taosMemoryFreeClear(pExprs);
×
2778
      terrno = code;
×
2779
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2780
      return NULL;
×
2781
    }
2782
  }
2783

2784
  return pExprs;
×
2785
}
2786

2787
int32_t createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, SExprInfo** pExprInfo, int32_t* numOfExprs) {
421,188,768✔
2788
  QRY_PARAM_CHECK(pExprInfo);
421,188,768✔
2789

2790
  int32_t code = 0;
421,245,773✔
2791
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
421,245,773✔
2792
  int32_t numOfGroupKeys = 0;
421,184,053✔
2793
  if (pGroupKeys != NULL) {
421,184,053✔
2794
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
36,443,645✔
2795
  }
2796

2797
  *numOfExprs = numOfFuncs + numOfGroupKeys;
421,183,187✔
2798
  if (*numOfExprs == 0) {
421,235,242✔
2799
    return code;
51,883,261✔
2800
  }
2801

2802
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
369,392,754✔
2803
  if (pExprs == NULL) {
369,071,784✔
2804
    return terrno;
×
2805
  }
2806

2807
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
1,209,175,432✔
2808
    STargetNode* pTargetNode = NULL;
844,115,167✔
2809
    if (i < numOfFuncs) {
844,115,167✔
2810
      pTargetNode = (STargetNode*)nodesListGetNode(pNodeList, i);
789,151,315✔
2811
    } else {
2812
      pTargetNode = (STargetNode*)nodesListGetNode(pGroupKeys, i - numOfFuncs);
54,963,852✔
2813
    }
2814
    if (!pTargetNode) {
844,338,587✔
2815
      destroyExprInfo(pExprs, *numOfExprs);
×
2816
      taosMemoryFreeClear(pExprs);
×
2817
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2818
      return terrno;
×
2819
    }
2820

2821
    SExprInfo* pExp = &pExprs[i];
844,338,587✔
2822
    code = createExprFromTargetNode(pExp, pTargetNode);
844,373,401✔
2823
    if (code != TSDB_CODE_SUCCESS) {
844,264,039✔
2824
      destroyExprInfo(pExprs, *numOfExprs);
4,160,391✔
2825
      taosMemoryFreeClear(pExprs);
4,160,391✔
2826
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
4,160,391✔
2827
      return code;
4,160,391✔
2828
    }
2829
  }
2830

2831
  *pExprInfo = pExprs;
365,146,935✔
2832
  return code;
365,173,462✔
2833
}
2834

2835
static void deleteSubsidiareCtx(void* pData) {
×
2836
  SSubsidiaryResInfo* pCtx = (SSubsidiaryResInfo*)pData;
×
2837
  if (pCtx->pCtx) {
×
2838
    taosMemoryFreeClear(pCtx->pCtx);
×
2839
  }
2840
}
×
2841

2842
// set the output buffer for the selectivity + tag query
2843
static int32_t setSelectValueColumnInfo(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
399,747,648✔
2844
  int32_t num = 0;
399,747,648✔
2845
  int32_t code = TSDB_CODE_SUCCESS;
399,747,648✔
2846
  int32_t lino = 0;
399,747,648✔
2847

2848
  SArray* pValCtxArray = NULL;
399,747,648✔
2849
  for (int32_t i = numOfOutput - 1; i > 0; --i) {  // select Func is at the end of the list
869,191,085✔
2850
    int32_t funcIdx = pCtx[i].pExpr->pExpr->_function.bindExprID;
469,507,526✔
2851
    if (funcIdx > 0) {
469,524,215✔
2852
      if (pValCtxArray == NULL) {
1,336,429✔
2853
        // the end of the list is the select function of biggest index
2854
        pValCtxArray = taosArrayInit_s(sizeof(SSubsidiaryResInfo*), funcIdx);
959,265✔
2855
        if (pValCtxArray == NULL) {
957,809✔
2856
          return terrno;
×
2857
        }
2858
      }
2859
      if (funcIdx > pValCtxArray->size) {
1,334,973✔
2860
        qError("funcIdx:%d is out of range", funcIdx);
×
2861
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2862
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2863
      }
2864
      SSubsidiaryResInfo* pSubsidiary = &pCtx[i].subsidiaries;
1,335,701✔
2865
      pSubsidiary->pCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
1,335,701✔
2866
      if (pSubsidiary->pCtx == NULL) {
1,335,701✔
2867
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2868
        return terrno;
×
2869
      }
2870
      pSubsidiary->num = 0;
1,334,609✔
2871
      taosArraySet(pValCtxArray, funcIdx - 1, &pSubsidiary);
1,334,973✔
2872
    }
2873
  }
2874

2875
  SqlFunctionCtx*  p = NULL;
399,683,559✔
2876
  SqlFunctionCtx** pValCtx = NULL;
399,683,559✔
2877
  if (pValCtxArray == NULL) {
399,683,559✔
2878
    pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
398,758,137✔
2879
    if (pValCtx == NULL) {
398,690,601✔
2880
      QUERY_CHECK_CODE(terrno, lino, _end);
×
2881
    }
2882
  }
2883

2884
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,225,990,256✔
2885
    const char* pName = pCtx[i].pExpr->pExpr->_function.functionName;
826,321,834✔
2886
    if ((strcmp(pName, "_select_value") == 0)) {
826,448,290✔
2887
      if (pValCtxArray == NULL) {
6,994,589✔
2888
        pValCtx[num++] = &pCtx[i];
5,119,746✔
2889
      } else {
2890
        int32_t bindFuncIndex = pCtx[i].pExpr->pExpr->relatedTo;  // start from index 1;
1,875,038✔
2891
        if (bindFuncIndex > 0) {                                  // 0 is default index related to the select function
1,875,457✔
2892
          bindFuncIndex -= 1;
1,832,141✔
2893
        }
2894
        SSubsidiaryResInfo** pSubsidiary = taosArrayGet(pValCtxArray, bindFuncIndex);
1,875,457✔
2895
        if (pSubsidiary == NULL) {
1,875,093✔
2896
          QUERY_CHECK_CODE(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR, lino, _end);
×
2897
        }
2898
        (*pSubsidiary)->pCtx[(*pSubsidiary)->num] = &pCtx[i];
1,875,093✔
2899
        (*pSubsidiary)->num++;
1,875,093✔
2900
      }
2901
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
819,453,701✔
2902
      if (pValCtxArray == NULL) {
60,826,522✔
2903
        p = &pCtx[i];
59,195,815✔
2904
      }
2905
    }
2906
  }
2907

2908
  if (p != NULL) {
399,668,422✔
2909
    p->subsidiaries.pCtx = pValCtx;
26,524,692✔
2910
    p->subsidiaries.num = num;
26,524,327✔
2911
  } else {
2912
    taosMemoryFreeClear(pValCtx);
373,143,730✔
2913
  }
2914

2915
_end:
966,238✔
2916
  if (code != TSDB_CODE_SUCCESS) {
399,652,709✔
2917
    taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2918
    taosMemoryFreeClear(pValCtx);
×
2919
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2920
  } else {
2921
    taosArrayDestroy(pValCtxArray);
399,652,709✔
2922
  }
2923
  return code;
399,672,063✔
2924
}
2925

2926
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowEntryInfoOffset,
399,755,837✔
2927
                                     SFunctionStateStore* pStore) {
2928
  int32_t         code = TSDB_CODE_SUCCESS;
399,755,837✔
2929
  int32_t         lino = 0;
399,755,837✔
2930
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
399,755,837✔
2931
  if (pFuncCtx == NULL) {
399,525,564✔
2932
    return NULL;
×
2933
  }
2934

2935
  *rowEntryInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
399,525,564✔
2936
  if (*rowEntryInfoOffset == 0) {
399,705,292✔
2937
    taosMemoryFreeClear(pFuncCtx);
×
2938
    return NULL;
×
2939
  }
2940

2941
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,226,159,533✔
2942
    SExprInfo* pExpr = &pExprInfo[i];
826,458,572✔
2943

2944
    SExprBasicInfo* pFunct = &pExpr->base;
826,390,234✔
2945
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
826,444,321✔
2946

2947
    pCtx->functionId = -1;
826,455,081✔
2948
    pCtx->pExpr = pExpr;
826,452,559✔
2949

2950
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
826,510,987✔
2951
      SFuncExecEnv env = {0};
350,454,971✔
2952
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;
350,444,031✔
2953
      pCtx->isPseudoFunc = fmIsWindowPseudoColumnFunc(pCtx->functionId) || fmIsPlaceHolderFunc(pCtx->functionId);
350,474,086✔
2954
      pCtx->isNotNullFunc = fmIsNotNullOutputFunc(pCtx->functionId);
350,444,599✔
2955

2956
      bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
350,432,483✔
2957
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId)) {
579,922,378✔
2958
        if (!isUdaf) {
229,592,226✔
2959
          code = fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
229,559,594✔
2960
          QUERY_CHECK_CODE(code, lino, _end);
229,537,447✔
2961
        } else {
2962
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
32,632✔
2963
          pCtx->udfName = taosStrdup(udfName);
32,632✔
2964
          QUERY_CHECK_NULL(pCtx->udfName, code, lino, _end, terrno);
32,632✔
2965

2966
          code = fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
32,632✔
2967
          QUERY_CHECK_CODE(code, lino, _end);
32,632✔
2968
        }
2969
        bool tmp = pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
229,570,079✔
2970
        if (!tmp) {
229,528,694✔
2971
          code = terrno;
×
UNCOV
2972
          QUERY_CHECK_CODE(code, lino, _end);
×
2973
        }
2974
      } else {
2975
        if (fmIsPlaceHolderFunc(pCtx->functionId)) {
120,817,980✔
2976
          code = fmGetStreamPesudoFuncEnv(pCtx->functionId, pExpr->base.pParamList, &env);
5,620,143✔
2977
          QUERY_CHECK_CODE(code, lino, _end);
5,620,342✔
2978
        }      
2979
        
2980
        code = fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
120,834,796✔
2981
        if (code != TSDB_CODE_SUCCESS && isUdaf) {
120,823,590✔
2982
          code = TSDB_CODE_SUCCESS;
54,720✔
2983
        }
2984
        QUERY_CHECK_CODE(code, lino, _end);
120,823,590✔
2985

2986
        if (pCtx->sfp.getEnv != NULL) {
120,823,590✔
2987
          bool tmp = pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
17,445,347✔
2988
          if (!tmp) {
17,446,879✔
2989
            code = terrno;
×
2990
            QUERY_CHECK_CODE(code, lino, _end);
×
2991
          }
2992
        }
2993
      }
2994
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
350,363,722✔
2995
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
475,995,242✔
2996
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
40,742,649✔
2997
      // for simple column, the result buffer needs to hold at least one element.
2998
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
476,112,194✔
2999
    }
3000

3001
    pCtx->input.numOfInputCols = pFunct->numOfParams;
826,500,561✔
3002
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
826,438,348✔
3003
    QUERY_CHECK_NULL(pCtx->input.pData, code, lino, _end, terrno);
826,512,502✔
3004
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
826,453,869✔
3005
    QUERY_CHECK_NULL(pCtx->input.pColumnDataAgg, code, lino, _end, terrno);
826,502,948✔
3006

3007
    pCtx->pTsOutput = NULL;
826,444,701✔
3008
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
826,542,084✔
3009
    pCtx->resDataInfo.type = pFunct->resSchema.type;
826,496,263✔
3010
    pCtx->order = TSDB_ORDER_ASC;
826,508,773✔
3011
    pCtx->start.key = INT64_MIN;
826,592,377✔
3012
    pCtx->end.key = INT64_MIN;
826,558,782✔
3013
    pCtx->numOfParams = pExpr->base.numOfParams;
826,514,519✔
3014
    pCtx->param = pFunct->pParam;
826,629,210✔
3015
    pCtx->saveHandle.currentPage = -1;
826,557,891✔
3016
    pCtx->pStore = pStore;
826,558,818✔
3017
    pCtx->hasWindowOrGroup = false;
826,622,005✔
3018
    pCtx->needCleanup = false;
826,555,764✔
3019
    pCtx->skipDynDataCheck = false;
826,480,208✔
3020
  }
3021

3022
  for (int32_t i = 1; i < numOfOutput; ++i) {
869,326,587✔
3023
    (*rowEntryInfoOffset)[i] = (int32_t)((*rowEntryInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) +
939,091,001✔
3024
                                         pFuncCtx[i - 1].resDataInfo.interBufSize);
469,574,560✔
3025
  }
3026

3027
  code = setSelectValueColumnInfo(pFuncCtx, numOfOutput);
399,749,655✔
3028
  QUERY_CHECK_CODE(code, lino, _end);
399,686,714✔
3029

3030
_end:
399,686,714✔
3031
  if (code != TSDB_CODE_SUCCESS) {
399,660,317✔
3032
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3033
    for (int32_t i = 0; i < numOfOutput; ++i) {
×
3034
      taosMemoryFree(pFuncCtx[i].input.pData);
×
3035
      taosMemoryFree(pFuncCtx[i].input.pColumnDataAgg);
×
3036
    }
3037
    taosMemoryFreeClear(*rowEntryInfoOffset);
×
3038
    taosMemoryFreeClear(pFuncCtx);
×
3039

3040
    terrno = code;
×
3041
    return NULL;
×
3042
  }
3043
  return pFuncCtx;
399,660,317✔
3044
}
3045

3046
// NOTE: sources columns are more than the destination SSDatablock columns.
3047
// doFilter in table scan needs every column even its output is false
3048
int32_t relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray* pCols, bool outputEveryColumn) {
9,802,789✔
3049
  int32_t code = TSDB_CODE_SUCCESS;
9,802,789✔
3050
  size_t  numOfSrcCols = taosArrayGetSize(pCols);
9,802,789✔
3051

3052
  int32_t i = 0, j = 0;
9,802,789✔
3053
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
69,173,513✔
3054
    SColumnInfoData* p = taosArrayGet(pCols, i);
59,370,724✔
3055
    if (!p) {
59,370,724✔
3056
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3057
      return terrno;
×
3058
    }
3059
    SColMatchItem* pmInfo = taosArrayGet(pColMatchInfo, j);
59,370,724✔
3060
    if (!pmInfo) {
59,370,244✔
3061
      return terrno;
×
3062
    }
3063

3064
    if (p->info.colId == pmInfo->colId) {
59,370,244✔
3065
      SColumnInfoData* pDst = taosArrayGet(pBlock->pDataBlock, pmInfo->dstSlotId);
53,510,584✔
3066
      if (!pDst) {
53,510,846✔
3067
        return terrno;
×
3068
      }
3069
      code = colDataAssign(pDst, p, pBlock->info.rows, &pBlock->info);
53,510,846✔
3070
      if (code != TSDB_CODE_SUCCESS) {
53,511,064✔
3071
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3072
        return code;
×
3073
      }
3074
      i++;
53,511,064✔
3075
      j++;
53,511,064✔
3076
    } else if (p->info.colId < pmInfo->colId) {
5,859,660✔
3077
      i++;
5,859,660✔
3078
    } else {
3079
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3080
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3081
    }
3082
  }
3083
  return code;
9,803,269✔
3084
}
3085

3086
SInterval extractIntervalInfo(const STableScanPhysiNode* pTableScanNode) {
206,691,462✔
3087
  SInterval interval = {
413,231,366✔
3088
      .interval = pTableScanNode->interval,
206,559,103✔
3089
      .sliding = pTableScanNode->sliding,
206,553,472✔
3090
      .intervalUnit = pTableScanNode->intervalUnit,
206,724,595✔
3091
      .slidingUnit = pTableScanNode->slidingUnit,
206,596,381✔
3092
      .offset = pTableScanNode->offset,
206,705,546✔
3093
      .precision = pTableScanNode->scan.node.pOutputDataBlockDesc->precision,
206,717,420✔
3094
      .timeRange = pTableScanNode->scanRange,
3095
  };
3096
  calcIntervalAutoOffset(&interval);
206,543,069✔
3097

3098
  return interval;
206,644,365✔
3099
}
3100

3101
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
59,951,101✔
3102
  SColumn c = {0};
59,951,101✔
3103

3104
  c.slotId = pColNode->slotId;
59,951,101✔
3105
  c.colId = pColNode->colId;
59,951,803✔
3106
  c.type = pColNode->node.resType.type;
59,951,921✔
3107
  c.bytes = pColNode->node.resType.bytes;
59,950,373✔
3108
  c.scale = pColNode->node.resType.scale;
59,950,138✔
3109
  c.precision = pColNode->node.resType.precision;
59,951,754✔
3110
  return c;
59,946,512✔
3111
}
3112

3113

3114
/**
3115
 * @brief Determine the actual time range for reading data based on the RANGE clause and the WHERE conditions.
3116
 * @param[in] cond The range specified by WHERE condition.
3117
 * @param[in] range The range specified by RANGE clause.
3118
 * @param[out] twindow The range to be read in DESC order, and only one record is needed.
3119
 * @param[out] extTwindow The external range to read for only one record, which is used for FILL clause.
3120
 * @note `cond` and `twindow` may be the same address.
3121
 */
3122
static int32_t getQueryExtWindow(const STimeWindow* cond, const STimeWindow* range, STimeWindow* twindow,
1,956,819✔
3123
                                 STimeWindow* extTwindows) {
3124
  int32_t     code = TSDB_CODE_SUCCESS;
1,956,819✔
3125
  int32_t     lino = 0;
1,956,819✔
3126
  STimeWindow tempWindow;
3127

3128
  if (cond->skey > cond->ekey || range->skey > range->ekey) {
1,956,819✔
3129
    *twindow = extTwindows[0] = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
2,435✔
3130
    return code;
2,435✔
3131
  }
3132

3133
  if (range->ekey < cond->skey) {
1,954,384✔
3134
    extTwindows[1] = *cond;
312,311✔
3135
    *twindow = extTwindows[0] = TSWINDOW_DESC_INITIALIZER;
312,311✔
3136
    return code;
312,311✔
3137
  }
3138

3139
  if (cond->ekey < range->skey) {
1,642,073✔
3140
    extTwindows[0] = *cond;
222,337✔
3141
    *twindow = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
222,337✔
3142
    return code;
222,337✔
3143
  }
3144

3145
  // Only scan data in the time range intersecion.
3146
  extTwindows[0] = extTwindows[1] = *cond;
1,419,736✔
3147
  twindow->skey = TMAX(cond->skey, range->skey);
1,419,736✔
3148
  twindow->ekey = TMIN(cond->ekey, range->ekey);
1,419,736✔
3149
  extTwindows[0].ekey = twindow->skey - 1;
1,419,736✔
3150
  extTwindows[1].skey = twindow->ekey + 1;
1,419,736✔
3151

3152
  return code;
1,419,736✔
3153
}
3154

3155
static int32_t getPrimaryTimeRange(SNode** pPrimaryKeyCond, STimeWindow* pTimeRange, bool* isStrict) {
10,400✔
3156
  SNode*  pNew = NULL;
10,400✔
3157
  int32_t code = scalarCalculateRemoteConstants(*pPrimaryKeyCond, &pNew);
10,400✔
3158
  if (TSDB_CODE_SUCCESS == code) {
10,400✔
3159
    *pPrimaryKeyCond = pNew;
10,400✔
3160
    if (nodeType(pNew) != QUERY_NODE_VALUE) {
10,400✔
3161
      code = filterGetTimeRange(*pPrimaryKeyCond, pTimeRange, isStrict, NULL);
10,400✔
3162
    }
3163
  }
3164
  return code;
10,400✔
3165
}
3166

3167
int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, STableScanPhysiNode* pTableScanNode,
232,763,731✔
3168
                               const SReadHandle* readHandle, bool applyExtWin) {
3169
  int32_t code = 0;                             
232,763,731✔
3170
  pCond->order = pTableScanNode->scanSeq[0] > 0 ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
232,763,731✔
3171
  pCond->numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
232,741,067✔
3172

3173
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
232,772,782✔
3174
  if (!pCond->colList) {
232,696,205✔
3175
    return terrno;
×
3176
  }
3177
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
232,648,111✔
3178
  if (pCond->pSlotList == NULL) {
232,727,548✔
3179
    taosMemoryFreeClear(pCond->colList);
×
3180
    return terrno;
×
3181
  }
3182

3183
  // TODO: get it from stable scan node
3184
  pCond->twindows = pTableScanNode->scanRange;
232,605,741✔
3185
  pCond->suid = pTableScanNode->scan.suid;
232,758,873✔
3186
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
232,689,363✔
3187
  pCond->startVersion = -1;
232,735,165✔
3188
  pCond->endVersion = -1;
232,751,830✔
3189
  pCond->skipRollup = readHandle->skipRollup;
232,585,513✔
3190
  if (readHandle->winRangeValid) {
232,709,416✔
3191
    pCond->twindows = readHandle->winRange;
276,907✔
3192
  }
3193
  pCond->cacheSttStatis = readHandle->cacheSttStatis;
232,743,665✔
3194
  // allowed read stt file optimization mode
3195
  pCond->notLoadData = (pTableScanNode->dataRequired == FUNC_DATA_REQUIRED_NOT_LOAD) &&
465,493,802✔
3196
                       (pTableScanNode->scan.node.pConditions == NULL) && (pTableScanNode->interval == 0);
232,742,541✔
3197

3198
  int32_t j = 0;
232,719,265✔
3199
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
861,877,419✔
3200
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pTableScanNode->scan.pScanCols, i);
629,244,605✔
3201
    if (!pNode) {
628,983,737✔
3202
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3203
      return terrno;
×
3204
    }
3205
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
628,983,737✔
3206
    if (pColNode->colType == COLUMN_TYPE_TAG) {
629,185,858✔
3207
      continue;
×
3208
    }
3209

3210
    pCond->colList[j].type = pColNode->node.resType.type;
629,179,787✔
3211
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
629,194,575✔
3212
    pCond->colList[j].colId = pColNode->colId;
629,147,633✔
3213
    pCond->colList[j].pk = pColNode->isPk;
629,235,497✔
3214

3215
    pCond->pSlotList[j] = pNode->slotId;
629,286,692✔
3216
    j += 1;
629,158,154✔
3217
  }
3218

3219
  pCond->numOfCols = j;
232,812,164✔
3220

3221
  if (applyExtWin) {
232,817,812✔
3222
    if (NULL != pTableScanNode->pExtScanRange) {
207,008,732✔
3223
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
1,956,819✔
3224
      code = getQueryExtWindow(&pCond->twindows, pTableScanNode->pExtScanRange, &pCond->twindows, pCond->extTwindows);
1,956,819✔
3225
    } else if (readHandle->extWinRangeValid) {
204,951,315✔
3226
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
×
3227
      code = getQueryExtWindow(&pCond->twindows, &readHandle->extWinRange, &pCond->twindows, pCond->extTwindows);
×
3228
    }
3229
  }
3230

3231
  if (pTableScanNode->pPrimaryCond) {
232,730,208✔
3232
    bool isStrict = false;
10,400✔
3233
    code = getPrimaryTimeRange((SNode**)&pTableScanNode->pPrimaryCond, &pCond->twindows, &isStrict);
10,400✔
3234
    if (code || !isStrict) {
10,400✔
3235
      code = nodesMergeNode((SNode**)&pTableScanNode->scan.node.pConditions, &pTableScanNode->pPrimaryCond);
×
3236
    }
3237
  }
3238

3239
  return code;
232,741,629✔
3240
}
3241

3242
int32_t initQueryTableDataCondWithColArray(SQueryTableDataCond* pCond, SQueryTableDataCond* pOrgCond,
3,479,972✔
3243
                                           const SReadHandle* readHandle, SArray* colArray) {
3244
  int32_t code = TSDB_CODE_SUCCESS;
3,479,972✔
3245
  int32_t lino = 0;
3,479,972✔
3246

3247
  pCond->order = TSDB_ORDER_ASC;
3,479,972✔
3248
  pCond->numOfCols = (int32_t)taosArrayGetSize(colArray);
3,479,972✔
3249

3250
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
3,479,972✔
3251
  QUERY_CHECK_NULL(pCond->colList, code, lino, _return, terrno);
3,479,972✔
3252

3253
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
3,479,972✔
3254
  QUERY_CHECK_NULL(pCond->pSlotList, code, lino, _return, terrno);
3,479,972✔
3255

3256
  pCond->twindows = pOrgCond->twindows;
3,479,972✔
3257
  pCond->order = pOrgCond->order;
3,479,972✔
3258
  pCond->type = pOrgCond->type;
3,479,972✔
3259
  pCond->startVersion = -1;
3,479,972✔
3260
  pCond->endVersion = -1;
3,479,972✔
3261
  pCond->skipRollup = true;
3,479,972✔
3262
  pCond->notLoadData = false;
3,479,972✔
3263

3264
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
19,102,355✔
3265
    SColIdPair* pColPair = taosArrayGet(colArray, i);
15,622,383✔
3266
    QUERY_CHECK_NULL(pColPair, code, lino, _return, terrno);
15,622,383✔
3267

3268
    bool find = false;
15,622,383✔
3269
    for (int32_t j = 0; j < pOrgCond->numOfCols; ++j) {
104,667,363✔
3270
      if (pOrgCond->colList[j].colId == pColPair->vtbColId) {
104,667,363✔
3271
        pCond->colList[i].type = pOrgCond->colList[j].type;
15,622,383✔
3272
        pCond->colList[i].bytes = pOrgCond->colList[j].bytes;
15,622,383✔
3273
        pCond->colList[i].colId = pColPair->orgColId;
15,622,383✔
3274
        pCond->colList[i].pk = pOrgCond->colList[j].pk;
15,622,383✔
3275
        pCond->pSlotList[i] = i;
15,622,383✔
3276
        find = true;
15,622,383✔
3277
        qDebug("%s mapped vtb colId:%d to org colId:%d", __func__, pColPair->vtbColId, pColPair->orgColId);
15,622,383✔
3278
        break;
15,622,383✔
3279
      }
3280
    }
3281
    QUERY_CHECK_CONDITION(find, code, lino, _return, TSDB_CODE_NOT_FOUND);
15,622,383✔
3282
  }
3283

3284
  return code;
3,479,972✔
3285
_return:
×
3286
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(terrno));
×
3287
  taosMemoryFreeClear(pCond->colList);
×
3288
  taosMemoryFreeClear(pCond->pSlotList);
×
3289
  return code;
×
3290
}
3291

3292
void cleanupQueryTableDataCond(SQueryTableDataCond* pCond) {
482,968,781✔
3293
  taosMemoryFreeClear(pCond->colList);
482,968,781✔
3294
  taosMemoryFreeClear(pCond->pSlotList);
482,940,285✔
3295
}
482,928,911✔
3296

3297
int32_t convertFillType(int32_t mode) {
2,520,993✔
3298
  int32_t type = TSDB_FILL_NONE;
2,520,993✔
3299
  switch (mode) {
2,520,993✔
3300
    case FILL_MODE_PREV:
128,074✔
3301
      type = TSDB_FILL_PREV;
128,074✔
3302
      break;
128,074✔
3303
    case FILL_MODE_NONE:
×
3304
      type = TSDB_FILL_NONE;
×
3305
      break;
×
3306
    case FILL_MODE_NULL:
149,559✔
3307
      type = TSDB_FILL_NULL;
149,559✔
3308
      break;
149,559✔
3309
    case FILL_MODE_NULL_F:
34,347✔
3310
      type = TSDB_FILL_NULL_F;
34,347✔
3311
      break;
34,347✔
3312
    case FILL_MODE_NEXT:
120,098✔
3313
      type = TSDB_FILL_NEXT;
120,098✔
3314
      break;
120,098✔
3315
    case FILL_MODE_VALUE:
143,127✔
3316
      type = TSDB_FILL_SET_VALUE;
143,127✔
3317
      break;
143,127✔
3318
    case FILL_MODE_VALUE_F:
10,770✔
3319
      type = TSDB_FILL_SET_VALUE_F;
10,770✔
3320
      break;
10,770✔
3321
    case FILL_MODE_LINEAR:
193,701✔
3322
      type = TSDB_FILL_LINEAR;
193,701✔
3323
      break;
193,701✔
3324
    case FILL_MODE_NEAR:
1,741,317✔
3325
      type = TSDB_FILL_NEAR;
1,741,317✔
3326
      break;
1,741,317✔
3327
    default:
×
3328
      type = TSDB_FILL_NONE;
×
3329
  }
3330

3331
  return type;
2,520,993✔
3332
}
3333

3334
void getInitialStartTimeWindow(SInterval* pInterval, TSKEY ts, STimeWindow* w, bool ascQuery) {
2,147,483,647✔
3335
  if (ascQuery) {
2,147,483,647✔
3336
    *w = getAlignQueryTimeWindow(pInterval, ts);
2,147,483,647✔
3337
  } else {
3338
    // the start position of the first time window in the endpoint that spreads beyond the queried last timestamp
3339
    *w = getAlignQueryTimeWindow(pInterval, ts);
25,499✔
3340

3341
    int64_t key = w->skey;
247,148✔
3342
    while (key < ts) {  // moving towards end
262,251✔
3343
      key = getNextTimeWindowStart(pInterval, key, TSDB_ORDER_ASC);
115,909✔
3344
      if (key > ts) {
115,909✔
3345
        break;
100,806✔
3346
      }
3347

3348
      w->skey = key;
15,103✔
3349
    }
3350
    w->ekey = taosTimeAdd(w->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
247,148✔
3351
  }
3352
}
2,147,483,647✔
3353

3354
static STimeWindow doCalculateTimeWindow(int64_t ts, SInterval* pInterval) {
65,967,406✔
3355
  STimeWindow w = {0};
65,967,406✔
3356

3357
  w.skey = taosTimeTruncate(ts, pInterval);
65,967,406✔
3358
  w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
65,965,597✔
3359
  return w;
65,970,514✔
3360
}
3361

3362
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order) {
640,034✔
3363
  STimeWindow win = *pWindow;
640,034✔
3364
  STimeWindow save = win;
640,034✔
3365
  while (win.skey <= ts && win.ekey >= ts) {
3,256,120✔
3366
    save = win;
2,616,086✔
3367
    // get previous time window
3368
    getNextTimeWindow(pInterval, &win, order == TSDB_ORDER_DESC ? TSDB_ORDER_ASC : TSDB_ORDER_DESC);
2,616,086✔
3369
  }
3370

3371
  return save;
640,034✔
3372
}
3373

3374
// get the correct time window according to the handled timestamp
3375
// todo refactor
3376
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
102,941,441✔
3377
                                int32_t order) {
3378
  STimeWindow w = {0};
102,941,441✔
3379
  if (pResultRowInfo->cur.pageId == -1) {  // the first window, from the previous stored value
102,949,099✔
3380
    getInitialStartTimeWindow(pInterval, ts, &w, (order != TSDB_ORDER_DESC));
5,834,654✔
3381
    return w;
5,836,325✔
3382
  }
3383

3384
  SResultRow* pRow = getResultRowByPos(pBuf, &pResultRowInfo->cur, false);
97,118,881✔
3385
  if (pRow) {
97,112,430✔
3386
    TAOS_SET_OBJ_ALIGNED(&w, pRow->win);
97,115,739✔
3387
  }
3388

3389
  // in case of typical time window, we can calculate time window directly.
3390
  if (w.skey > ts || w.ekey < ts) {
97,111,261✔
3391
    w = doCalculateTimeWindow(ts, pInterval);
65,966,779✔
3392
  }
3393

3394
  if (pInterval->interval != pInterval->sliding) {
97,114,996✔
3395
    // it is an sliding window query, in which sliding value is not equalled to
3396
    // interval value, and we need to find the first qualified time window.
3397
    w = getFirstQualifiedTimeWindow(ts, &w, pInterval, order);
640,034✔
3398
  }
3399

3400
  return w;
97,106,503✔
3401
}
3402

3403
TSKEY getNextTimeWindowStart(const SInterval* pInterval, TSKEY start, int32_t order) {
2,147,483,647✔
3404
  int32_t factor = GET_FORWARD_DIRECTION_FACTOR(order);
2,147,483,647✔
3405
  TSKEY   nextStart = taosTimeAdd(start, -1 * pInterval->offset, pInterval->offsetUnit, pInterval->precision, NULL);
2,147,483,647✔
3406
  nextStart = taosTimeAdd(nextStart, factor * pInterval->sliding, pInterval->slidingUnit, pInterval->precision, NULL);
2,147,483,647✔
3407
  nextStart = taosTimeAdd(nextStart, pInterval->offset, pInterval->offsetUnit, pInterval->precision, NULL);
2,147,483,647✔
3408
  return nextStart;
2,147,483,647✔
3409
}
3410

3411
void getNextTimeWindow(const SInterval* pInterval, STimeWindow* tw, int32_t order) {
2,147,483,647✔
3412
  tw->skey = getNextTimeWindowStart(pInterval, tw->skey, order);
2,147,483,647✔
3413
  tw->ekey = taosTimeAdd(tw->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
2,147,483,647✔
3414
}
2,147,483,647✔
3415

3416
bool hasLimitOffsetInfo(SLimitInfo* pLimitInfo) {
363,382,717✔
3417
  return (pLimitInfo->limit.limit != -1 || pLimitInfo->limit.offset != -1 || pLimitInfo->slimit.limit != -1 ||
722,926,097✔
3418
          pLimitInfo->slimit.offset != -1);
359,542,876✔
3419
}
3420

3421
bool hasSlimitOffsetInfo(SLimitInfo* pLimitInfo) {
×
3422
  return (pLimitInfo->slimit.limit != -1 || pLimitInfo->slimit.offset != -1);
×
3423
}
3424

3425
void initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo) {
524,588,680✔
3426
  SLimit limit = {.limit = getLimit(pLimit), .offset = getOffset(pLimit)};
524,588,680✔
3427
  SLimit slimit = {.limit = getLimit(pSLimit), .offset = getOffset(pSLimit)};
524,453,144✔
3428

3429
  pLimitInfo->limit = limit;
524,432,775✔
3430
  pLimitInfo->slimit = slimit;
524,465,814✔
3431
  pLimitInfo->remainOffset = limit.offset;
524,463,637✔
3432
  pLimitInfo->remainGroupOffset = slimit.offset;
524,482,220✔
3433
  pLimitInfo->numOfOutputRows = 0;
524,493,222✔
3434
  pLimitInfo->numOfOutputGroups = 0;
524,558,699✔
3435
  pLimitInfo->currentGroupId = 0;
524,558,077✔
3436
}
524,539,881✔
3437

3438
void resetLimitInfoForNextGroup(SLimitInfo* pLimitInfo) {
76,506,131✔
3439
  pLimitInfo->numOfOutputRows = 0;
76,506,131✔
3440
  pLimitInfo->remainOffset = pLimitInfo->limit.offset;
76,519,281✔
3441
}
76,511,278✔
3442

3443
int32_t tableListGetSize(const STableListInfo* pTableList, int32_t* pRes) {
564,394,440✔
3444
  if (taosArrayGetSize(pTableList->pTableList) != taosHashGetSize(pTableList->map)) {
564,394,440✔
3445
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3446
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3447
  }
3448
  (*pRes) = taosArrayGetSize(pTableList->pTableList);
564,398,991✔
3449
  return TSDB_CODE_SUCCESS;
564,333,707✔
3450
}
3451

3452
uint64_t tableListGetSuid(const STableListInfo* pTableList) { return pTableList->idInfo.suid; }
4,329,447✔
3453

3454
STableKeyInfo* tableListGetInfo(const STableListInfo* pTableList, int32_t index) {
191,052,664✔
3455
  if (taosArrayGetSize(pTableList->pTableList) == 0) {
191,052,664✔
3456
    return NULL;
1,955✔
3457
  }
3458

3459
  return taosArrayGet(pTableList->pTableList, index);
191,007,758✔
3460
}
3461

3462
int32_t tableListFind(const STableListInfo* pTableList, uint64_t uid, int32_t startIndex) {
43,515✔
3463
  int32_t numOfTables = taosArrayGetSize(pTableList->pTableList);
43,515✔
3464
  if (startIndex >= numOfTables) {
43,515✔
3465
    return -1;
×
3466
  }
3467

3468
  for (int32_t i = startIndex; i < numOfTables; ++i) {
318,727✔
3469
    STableKeyInfo* p = taosArrayGet(pTableList->pTableList, i);
318,727✔
3470
    if (!p) {
318,727✔
3471
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3472
      return -1;
×
3473
    }
3474
    if (p->uid == uid) {
318,727✔
3475
      return i;
43,515✔
3476
    }
3477
  }
UNCOV
3478
  return -1;
×
3479
}
3480

3481
void tableListGetSourceTableInfo(const STableListInfo* pTableList, uint64_t* psuid, uint64_t* uid, int32_t* type) {
351,980✔
3482
  *psuid = pTableList->idInfo.suid;
351,980✔
3483
  *uid = pTableList->idInfo.uid;
351,980✔
3484
  *type = pTableList->idInfo.tableType;
351,980✔
3485
}
351,980✔
3486

3487
uint64_t tableListGetTableGroupId(const STableListInfo* pTableList, uint64_t tableUid) {
759,778,457✔
3488
  int32_t* slot = taosHashGet(pTableList->map, &tableUid, sizeof(tableUid));
759,778,457✔
3489
  if (slot == NULL) {
760,046,236✔
3490
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3491
    return -1;
×
3492
  }
3493

3494
  STableKeyInfo* pKeyInfo = taosArrayGet(pTableList->pTableList, *slot);
760,046,236✔
3495
  if (pKeyInfo == NULL) {
760,045,945✔
3496
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3497
    return -1;
×
3498
  }
3499
  return pKeyInfo->groupId;
760,045,945✔
3500
}
3501

3502
// TODO handle the group offset info, fix it, the rule of group output will be broken by this function
3503
// int32_t tableListRemoveTableInfo(STableListInfo* pTableList, uint64_t uid) {
3504
//   int32_t code = TSDB_CODE_SUCCESS;
3505
//   int32_t lino = 0;
3506

3507
//   int32_t* slot = taosHashGet(pTableList->map, &uid, sizeof(uid));
3508
//   if (slot == NULL) {
3509
//     qDebug("table:%" PRIu64 " not found in table list", uid);
3510
//     return 0;
3511
//   }
3512

3513
//   taosArrayRemove(pTableList->pTableList, *slot);
3514
//   code = taosHashRemove(pTableList->map, &uid, sizeof(uid));
3515

3516
//   _end:
3517
//   if (code != TSDB_CODE_SUCCESS) {
3518
//     qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
3519
//   } else {
3520
//     qDebug("uid:%" PRIu64 ", remove from table list", uid);
3521
//   }
3522

3523
//   return code;
3524
// }
3525

3526
int32_t tableListAddTableInfo(STableListInfo* pTableList, uint64_t uid, uint64_t gid) {
938,339✔
3527
  int32_t code = TSDB_CODE_SUCCESS;
938,339✔
3528
  int32_t lino = 0;
938,339✔
3529
  if (pTableList->map == NULL) {
938,339✔
3530
    pTableList->map = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
×
3531
    QUERY_CHECK_NULL(pTableList->map, code, lino, _end, terrno);
×
3532
  }
3533

3534
  STableKeyInfo keyInfo = {.uid = uid, .groupId = gid};
940,055✔
3535
  void*         p = taosHashGet(pTableList->map, &uid, sizeof(uid));
938,339✔
3536
  if (p != NULL) {
938,055✔
3537
    qInfo("table:%" PRId64 " already in tableIdList, ignore it", uid);
614✔
3538
    goto _end;
614✔
3539
  }
3540

3541
  void* tmp = taosArrayPush(pTableList->pTableList, &keyInfo);
937,441✔
3542
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
938,293✔
3543

3544
  int32_t slot = (int32_t)taosArrayGetSize(pTableList->pTableList) - 1;
938,293✔
3545
  code = taosHashPut(pTableList->map, &uid, sizeof(uid), &slot, sizeof(slot));
939,151✔
3546
  if (code != TSDB_CODE_SUCCESS) {
940,293✔
3547
    // we have checked the existence of uid in hash map above
3548
    QUERY_CHECK_CONDITION((code != TSDB_CODE_DUP_KEY), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
3549
    taosArrayPopTailBatch(pTableList->pTableList, 1);  // let's pop the last element in the array list
×
3550
  }
3551

3552
_end:
940,907✔
3553
  if (code != TSDB_CODE_SUCCESS) {
939,191✔
3554
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3555
  } else {
3556
    qDebug("uid:%" PRIu64 ", groupId:%" PRIu64 " added into table list, slot:%d, total:%d", uid, gid, slot, slot + 1);
939,191✔
3557
  }
3558

3559
  return code;
941,197✔
3560
}
3561

3562
int32_t tableListGetGroupList(const STableListInfo* pTableList, int32_t ordinalGroupIndex, STableKeyInfo** pKeyInfo,
201,239,742✔
3563
                              int32_t* size) {
3564
  int32_t totalGroups = tableListGetOutputGroups(pTableList);
201,239,742✔
3565
  int32_t numOfTables = 0;
201,245,709✔
3566
  int32_t code = tableListGetSize(pTableList, &numOfTables);
201,301,127✔
3567
  if (code != TSDB_CODE_SUCCESS) {
201,204,817✔
3568
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3569
    return code;
×
3570
  }
3571

3572
  if (ordinalGroupIndex < 0 || ordinalGroupIndex >= totalGroups) {
201,204,817✔
3573
    return TSDB_CODE_INVALID_PARA;
×
3574
  }
3575

3576
  // here handle two special cases:
3577
  // 1. only one group exists, and 2. one table exists for each group.
3578
  if (totalGroups == 1) {
201,204,817✔
3579
    *size = numOfTables;
200,897,021✔
3580
    *pKeyInfo = (*size == 0) ? NULL : taosArrayGet(pTableList->pTableList, 0);
200,806,879✔
3581
    return TSDB_CODE_SUCCESS;
200,917,947✔
3582
  } else if (totalGroups == numOfTables) {
307,797✔
3583
    *size = 1;
276,451✔
3584
    *pKeyInfo = taosArrayGet(pTableList->pTableList, ordinalGroupIndex);
276,451✔
3585
    return TSDB_CODE_SUCCESS;
276,451✔
3586
  }
3587

3588
  int32_t offset = pTableList->groupOffset[ordinalGroupIndex];
32,883✔
3589
  if (ordinalGroupIndex < totalGroups - 1) {
53,338✔
3590
    *size = pTableList->groupOffset[ordinalGroupIndex + 1] - offset;
40,158✔
3591
  } else {
3592
    *size = numOfTables - offset;
13,180✔
3593
  }
3594

3595
  *pKeyInfo = taosArrayGet(pTableList->pTableList, offset);
53,338✔
3596
  return TSDB_CODE_SUCCESS;
53,338✔
3597
}
3598

3599
int32_t tableListGetOutputGroups(const STableListInfo* pTableList) { return pTableList->numOfOuputGroups; }
598,087,066✔
3600

3601
bool oneTableForEachGroup(const STableListInfo* pTableList) { return pTableList->oneTableForEachGroup; }
1,091,906✔
3602

3603
STableListInfo* tableListCreate() {
243,102,393✔
3604
  STableListInfo* pListInfo = taosMemoryCalloc(1, sizeof(STableListInfo));
243,102,393✔
3605
  if (pListInfo == NULL) {
242,914,297✔
3606
    return NULL;
×
3607
  }
3608

3609
  pListInfo->remainGroups = NULL;
242,914,297✔
3610
  pListInfo->pTableList = taosArrayInit(4, sizeof(STableKeyInfo));
242,941,121✔
3611
  if (pListInfo->pTableList == NULL) {
242,984,633✔
3612
    goto _error;
×
3613
  }
3614

3615
  pListInfo->map = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
243,050,566✔
3616
  if (pListInfo->map == NULL) {
243,193,079✔
3617
    goto _error;
×
3618
  }
3619

3620
  pListInfo->numOfOuputGroups = 1;
243,190,249✔
3621
  return pListInfo;
243,188,049✔
3622

3623
_error:
×
3624
  tableListDestroy(pListInfo);
×
3625
  return NULL;
×
3626
}
3627

3628
void tableListDestroy(STableListInfo* pTableListInfo) {
252,713,523✔
3629
  if (pTableListInfo == NULL) {
252,713,523✔
3630
    return;
9,607,575✔
3631
  }
3632

3633
  taosArrayDestroy(pTableListInfo->pTableList);
243,105,948✔
3634
  taosMemoryFreeClear(pTableListInfo->groupOffset);
242,993,430✔
3635

3636
  taosHashCleanup(pTableListInfo->map);
243,004,377✔
3637
  taosHashCleanup(pTableListInfo->remainGroups);
243,120,429✔
3638
  pTableListInfo->pTableList = NULL;
243,116,686✔
3639
  pTableListInfo->map = NULL;
243,141,735✔
3640
  taosMemoryFree(pTableListInfo);
243,140,262✔
3641
}
3642

3643
void tableListClear(STableListInfo* pTableListInfo) {
735,815✔
3644
  if (pTableListInfo == NULL) {
735,815✔
3645
    return;
×
3646
  }
3647

3648
  taosArrayClear(pTableListInfo->pTableList);
735,815✔
3649
  taosHashClear(pTableListInfo->map);
737,223✔
3650
  taosHashClear(pTableListInfo->remainGroups);
738,093✔
3651
  taosMemoryFree(pTableListInfo->groupOffset);
738,093✔
3652
  pTableListInfo->numOfOuputGroups = 1;
737,809✔
3653
  pTableListInfo->oneTableForEachGroup = false;
737,809✔
3654
}
3655

3656
static int32_t orderbyGroupIdComparFn(const void* p1, const void* p2) {
515,744,978✔
3657
  STableKeyInfo* pInfo1 = (STableKeyInfo*)p1;
515,744,978✔
3658
  STableKeyInfo* pInfo2 = (STableKeyInfo*)p2;
515,744,978✔
3659

3660
  if (pInfo1->groupId == pInfo2->groupId) {
515,744,978✔
3661
    return 0;
484,185,607✔
3662
  } else {
3663
    return pInfo1->groupId < pInfo2->groupId ? -1 : 1;
31,486,525✔
3664
  }
3665
}
3666

3667
int32_t sortTableGroup(STableListInfo* pTableListInfo) {
26,526,949✔
3668
  int32_t code = TSDB_CODE_SUCCESS;
26,526,949✔
3669
  taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
26,526,949✔
3670
  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
26,534,973✔
3671
  if (size == 0) {
26,526,221✔
3672
    pTableListInfo->numOfOuputGroups = 0;
×
3673
    return code;
×
3674
  }
3675

3676
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
26,526,221✔
3677
  if (!pList) {
26,526,971✔
3678
    code = terrno;
×
3679
    goto end;
×
3680
  }
3681

3682
  STableKeyInfo* pInfo = taosArrayGet(pTableListInfo->pTableList, 0);
26,526,971✔
3683
  if (pInfo == NULL) {
26,525,532✔
3684
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3685
    code = terrno;
×
3686
    goto end;
×
3687
  }
3688
  uint64_t gid = pInfo->groupId;
26,525,532✔
3689

3690
  int32_t start = 0;
26,528,205✔
3691
  void*   tmp = taosArrayPush(pList, &start);
26,522,649✔
3692
  if (tmp == NULL) {
26,522,649✔
3693
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3694
    code = terrno;
×
3695
    goto end;
×
3696
  }
3697

3698
  for (int32_t i = 1; i < size; ++i) {
144,584,369✔
3699
    pInfo = taosArrayGet(pTableListInfo->pTableList, i);
118,067,411✔
3700
    if (pInfo == NULL) {
118,061,742✔
3701
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3702
      code = terrno;
×
3703
      goto end;
×
3704
    }
3705
    if (pInfo->groupId != gid) {
118,061,742✔
3706
      tmp = taosArrayPush(pList, &i);
6,445,656✔
3707
      if (tmp == NULL) {
6,445,656✔
3708
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3709
        code = terrno;
×
3710
        goto end;
×
3711
      }
3712
      gid = pInfo->groupId;
6,445,656✔
3713
    }
3714
  }
3715

3716
  pTableListInfo->numOfOuputGroups = taosArrayGetSize(pList);
26,528,176✔
3717
  pTableListInfo->groupOffset = taosMemoryMalloc(sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
26,530,480✔
3718
  if (pTableListInfo->groupOffset == NULL) {
26,520,548✔
3719
    code = terrno;
×
3720
    goto end;
×
3721
  }
3722

3723
  memcpy(pTableListInfo->groupOffset, taosArrayGet(pList, 0), sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
26,515,935✔
3724

3725
end:
26,517,195✔
3726
  taosArrayDestroy(pList);
26,517,612✔
3727
  return code;
26,498,378✔
3728
}
3729

3730
int32_t buildGroupIdMapForAllTables(STableListInfo* pTableListInfo, SReadHandle* pHandle, SScanPhysiNode* pScanNode,
221,746,250✔
3731
                                    SNodeList* group, bool groupSort, uint8_t* digest, SStorageAPI* pAPI, SHashObj* groupIdMap) {
3732
  int32_t code = TSDB_CODE_SUCCESS;
221,746,250✔
3733

3734
  bool   groupByTbname = groupbyTbname(group);
221,746,250✔
3735
  size_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
221,717,394✔
3736
  if (!numOfTables) {
221,721,940✔
3737
    return code;
3,202✔
3738
  }
3739
  qDebug("numOfTables:%zu, groupByTbname:%d, group:%p", numOfTables, groupByTbname, group);
221,718,738✔
3740
  if (group == NULL || groupByTbname) {
221,698,489✔
3741
    if (tsCountAlwaysReturnValue && QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode) &&
217,899,054✔
3742
        ((STableScanPhysiNode*)pScanNode)->needCountEmptyTable) {
181,378,652✔
3743
      pTableListInfo->remainGroups =
14,711,055✔
3744
          taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
14,711,055✔
3745
      if (pTableListInfo->remainGroups == NULL) {
14,711,055✔
3746
        return terrno;
×
3747
      }
3748

3749
      for (int i = 0; i < numOfTables; i++) {
68,119,244✔
3750
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
53,409,252✔
3751
        if (!info) {
53,409,153✔
3752
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3753
          return terrno;
×
3754
        }
3755
        info->groupId = groupByTbname ? info->uid : 0;
53,409,153✔
3756
        int32_t tempRes = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId),
53,409,733✔
3757
                                      &(info->uid), sizeof(info->uid));
53,409,371✔
3758
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
53,408,771✔
3759
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3760
          return tempRes;
×
3761
        }
3762
      }
3763
    } else {
3764
      for (int32_t i = 0; i < numOfTables; i++) {
737,419,171✔
3765
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
534,293,287✔
3766
        if (!info) {
534,276,116✔
3767
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3768
          return terrno;
×
3769
        }
3770
        info->groupId = groupByTbname ? info->uid : 0;
534,276,116✔
3771
        
3772
      }
3773
    }
3774
    if (groupIdMap && group != NULL){
217,835,876✔
3775
      getColInfoResultForGroupbyForStream(pHandle->vnode, group, pTableListInfo, pAPI, groupIdMap);
61,960✔
3776
    }
3777

3778
    pTableListInfo->oneTableForEachGroup = groupByTbname;
217,835,876✔
3779
    if (numOfTables == 1 && pTableListInfo->idInfo.tableType == TSDB_CHILD_TABLE) {
217,905,927✔
3780
      pTableListInfo->oneTableForEachGroup = true;
78,508,111✔
3781
    }
3782

3783
    if (groupSort && groupByTbname) {
217,925,237✔
3784
      taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
2,094,788✔
3785
      pTableListInfo->numOfOuputGroups = numOfTables;
2,095,123✔
3786
    } else if (groupByTbname && pScanNode->groupOrderScan) {
215,830,449✔
3787
      pTableListInfo->numOfOuputGroups = numOfTables;
24,642✔
3788
    } else {
3789
      pTableListInfo->numOfOuputGroups = 1;
215,806,389✔
3790
    }
3791
    if (groupSort || pScanNode->groupOrderScan) {
217,959,610✔
3792
      code = sortTableGroup(pTableListInfo);
26,403,959✔
3793
    }
3794
  } else {
3795
    bool initRemainGroups = false;
3,799,435✔
3796
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode)) {
3,799,435✔
3797
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pScanNode;
3,718,263✔
3798
      if (tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable &&
3,718,263✔
3799
          !(groupSort || pScanNode->groupOrderScan)) {
1,878,576✔
3800
        initRemainGroups = true;
1,855,728✔
3801
      }
3802
    }
3803

3804
    code = getColInfoResultForGroupby(pHandle->vnode, group, pTableListInfo, digest, pAPI, initRemainGroups, groupIdMap);
3,799,435✔
3805
    if (code != TSDB_CODE_SUCCESS) {
3,798,610✔
3806
      return code;
×
3807
    }
3808

3809
    if (pScanNode->groupOrderScan) pTableListInfo->numOfOuputGroups = taosArrayGetSize(pTableListInfo->pTableList);
3,798,610✔
3810

3811
    if (groupSort || pScanNode->groupOrderScan) {
3,798,610✔
3812
      code = sortTableGroup(pTableListInfo);
125,143✔
3813
    }
3814
  }
3815

3816
  // add all table entry in the hash map
3817
  size_t size = taosArrayGetSize(pTableListInfo->pTableList);
221,709,071✔
3818
  for (int32_t i = 0; i < size; ++i) {
830,817,367✔
3819
    STableKeyInfo* p = taosArrayGet(pTableListInfo->pTableList, i);
609,081,165✔
3820
    if (!p) {
608,888,230✔
3821
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3822
      return terrno;
×
3823
    }
3824
    int32_t tempRes = taosHashPut(pTableListInfo->map, &p->uid, sizeof(uint64_t), &i, sizeof(int32_t));
608,888,230✔
3825
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
609,111,129✔
3826
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3827
      return tempRes;
×
3828
    }
3829
  }
3830

3831
  return code;
221,782,170✔
3832
}
3833

3834
int32_t createScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
233,182,948✔
3835
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
3836
                                SExecTaskInfo* pTaskInfo, SHashObj* groupIdMap) {
3837
  int64_t     st = taosGetTimestampUs();
233,182,644✔
3838
  const char* idStr = GET_TASKID(pTaskInfo);
233,182,644✔
3839

3840
  if (pHandle == NULL) {
233,098,487✔
3841
    qError("invalid handle, in creating operator tree, %s", idStr);
×
3842
    return TSDB_CODE_INVALID_PARA;
×
3843
  }
3844

3845
  if (pHandle->uid != 0) {
233,098,487✔
3846
    pScanNode->uid = pHandle->uid;
68,348✔
3847
    pScanNode->tableType = TSDB_CHILD_TABLE;
68,348✔
3848
  }
3849
  uint8_t digest[17] = {0};
233,156,442✔
3850
  int32_t code = getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, digest, idStr,
233,152,934✔
3851
                              &pTaskInfo->storageAPI, pTaskInfo->pStreamRuntimeInfo);
233,165,132✔
3852
  if (code != TSDB_CODE_SUCCESS) {
233,245,797✔
3853
    qError("failed to getTableList, code:%s", tstrerror(code));
876✔
3854
    return code;
876✔
3855
  }
3856

3857
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
233,244,921✔
3858

3859
  int64_t st1 = taosGetTimestampUs();
233,264,457✔
3860
  pTaskInfo->cost.extractListTime = (st1 - st) / 1000.0;
233,264,457✔
3861
  qDebug("extract queried table list completed, %d tables, elapsed time:%.2f ms %s", numOfTables,
233,235,330✔
3862
         pTaskInfo->cost.extractListTime, idStr);
3863

3864
  if (numOfTables == 0) {
233,215,995✔
3865
    qDebug("no table qualified for query, %s", idStr);
11,526,894✔
3866
    return TSDB_CODE_SUCCESS;
11,526,894✔
3867
  }
3868

3869
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, digest, &pTaskInfo->storageAPI, groupIdMap);
221,689,101✔
3870
  if (code != TSDB_CODE_SUCCESS) {
221,757,532✔
3871
    return code;
×
3872
  }
3873

3874
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
221,764,533✔
3875
  qDebug("generate group id map completed, elapsed time:%.2f ms %s", pTaskInfo->cost.groupIdMapTime, idStr);
221,740,238✔
3876

3877
  return TSDB_CODE_SUCCESS;
221,732,380✔
3878
}
3879

3880
char* getStreamOpName(uint16_t opType) {
5,564,023✔
3881
  switch (opType) {
5,564,023✔
3882
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
×
3883
      return "stream scan";
×
3884
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
5,405,587✔
3885
      return "project";
5,405,587✔
3886
    case QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW:
158,436✔
3887
      return "external window";
158,436✔
3888
  }
3889
  return "error name";
×
3890
}
3891

3892
void printDataBlock(SSDataBlock* pBlock, const char* flag, const char* taskIdStr, int64_t qId) {
380,077,172✔
3893
  if (qDebugFlag & DEBUG_TRACE) {
380,077,172✔
3894
    if (!pBlock) {
18,894✔
3895
      qDebug("%" PRIx64 " %s %s %s: Block is Null", qId, taskIdStr, flag, __func__);
3,417✔
3896
      return;
3,417✔
3897
    } else if (pBlock->info.rows == 0) {
15,477✔
3898
      qDebug("%" PRIx64 " %s %s %s: Block is Empty. block type %d", qId, taskIdStr, flag, __func__, pBlock->info.type);
×
3899
      return;
×
3900
    }
3901
    
3902
    char*   pBuf = NULL;
15,477✔
3903
    int32_t code = dumpBlockData(pBlock, flag, &pBuf, taskIdStr, qId);
15,477✔
3904
    if (code == 0) {
15,477✔
3905
      qDebugL("%" PRIx64 " %s %s", qId, __func__, pBuf);
15,477✔
3906
      taosMemoryFree(pBuf);
15,477✔
3907
    }
3908
  }
3909
}
3910

3911
void printSpecDataBlock(SSDataBlock* pBlock, const char* flag, const char* opStr, const char* taskIdStr) {
×
3912
  if (!pBlock) {
×
3913
    qDebug("%s===stream===%s %s: Block is Null", taskIdStr, flag, opStr);
×
3914
    return;
×
3915
  } else if (pBlock->info.rows == 0) {
×
3916
    qDebug("%s===stream===%s %s: Block is Empty. block type %d.skey:%" PRId64 ",ekey:%" PRId64 ",version%" PRId64,
×
3917
           taskIdStr, flag, opStr, pBlock->info.type, pBlock->info.window.skey, pBlock->info.window.ekey,
3918
           pBlock->info.version);
3919
    return;
×
3920
  }
3921
  if (qDebugFlag & DEBUG_TRACE) {
×
3922
    char* pBuf = NULL;
×
3923
    char  flagBuf[64];
×
3924
    snprintf(flagBuf, sizeof(flagBuf), "%s %s", flag, opStr);
×
3925
    int32_t code = dumpBlockData(pBlock, flagBuf, &pBuf, taskIdStr, 0);
×
3926
    if (code == 0) {
×
3927
      qDebug("%s", pBuf);
×
3928
      taosMemoryFree(pBuf);
×
3929
    }
3930
  }
3931
}
3932

3933
TSKEY getStartTsKey(STimeWindow* win, const TSKEY* tsCols) { return tsCols == NULL ? win->skey : tsCols[0]; }
12,449,876✔
3934

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

3938
  int64_t duration = pWin->ekey > pWin->skey ? pWin->ekey - pWin->skey + delta : pWin->skey - pWin->ekey + delta;
2,147,483,647✔
3939
  ts[2] = duration;            // set the duration
2,147,483,647✔
3940
  ts[3] = pWin->skey;          // window start key
2,147,483,647✔
3941
  ts[4] = pWin->ekey + delta;  // window end key
2,147,483,647✔
3942
}
2,147,483,647✔
3943

3944
int32_t compKeys(const SArray* pSortGroupCols, const char* oldkeyBuf, int32_t oldKeysLen, const SSDataBlock* pBlock,
2,147,483,647✔
3945
                 int32_t rowIndex) {
3946
  SColumnDataAgg* pColAgg = NULL;
2,147,483,647✔
3947
  const char*     isNull = oldkeyBuf;
2,147,483,647✔
3948
  const char*     p = oldkeyBuf + sizeof(int8_t) * pSortGroupCols->size;
2,147,483,647✔
3949

3950
  for (int32_t i = 0; i < pSortGroupCols->size; ++i) {
2,147,483,647✔
3951
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
2,147,483,647✔
3952
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
2,147,483,647✔
3953
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
2,147,483,647✔
3954

3955
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
2,147,483,647✔
3956
      if (isNull[i] != 1) return 1;
262,936,128✔
3957
    } else {
3958
      if (isNull[i] != 0) return 1;
2,147,483,647✔
3959
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,147,483,647✔
3960
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,147,483,647✔
3961
        int32_t len = getJsonValueLen(val);
×
3962
        if (memcmp(p, val, len) != 0) return 1;
×
3963
        p += len;
×
3964
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,147,483,647✔
3965
        if (IS_STR_DATA_BLOB(pCol->type)) {
1,204,955,997✔
3966
          if (memcmp(p, val, blobDataTLen(val)) != 0) return 1;
×
3967
          p += blobDataTLen(val);
×
3968
        } else {
3969
          if (memcmp(p, val, varDataTLen(val)) != 0) return 1;
1,205,521,452✔
3970
          p += varDataTLen(val);
1,204,659,189✔
3971
        }
3972
      } else {
3973
        if (0 != memcmp(p, val, pCol->bytes)) return 1;
2,147,483,647✔
3974
        p += pCol->bytes;
2,147,483,647✔
3975
      }
3976
    }
3977
  }
3978
  if ((int32_t)(p - oldkeyBuf) != oldKeysLen) return 1;
2,147,483,647✔
3979
  return 0;
2,147,483,647✔
3980
}
3981

3982
int32_t buildKeys(char* keyBuf, const SArray* pSortGroupCols, const SSDataBlock* pBlock, int32_t rowIndex) {
1,595,343✔
3983
  uint32_t        colNum = pSortGroupCols->size;
1,595,343✔
3984
  SColumnDataAgg* pColAgg = NULL;
1,595,343✔
3985
  char*           isNull = keyBuf;
1,595,343✔
3986
  char*           p = keyBuf + sizeof(int8_t) * colNum;
1,595,343✔
3987

3988
  for (int32_t i = 0; i < colNum; ++i) {
4,366,296✔
3989
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
2,770,953✔
3990
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
2,770,953✔
3991
    if (pCol->slotId > pBlock->pDataBlock->size) continue;
2,770,953✔
3992

3993
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
2,770,953✔
3994

3995
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
5,541,906✔
3996
      isNull[i] = 1;
134,100✔
3997
    } else {
3998
      isNull[i] = 0;
2,636,853✔
3999
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,636,853✔
4000
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,636,853✔
4001
        int32_t len = getJsonValueLen(val);
×
4002
        memcpy(p, val, len);
×
4003
        p += len;
×
4004
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,636,853✔
4005
        if (IS_STR_DATA_BLOB(pCol->type)) {
984,294✔
4006
          blobDataCopy(p, val);
×
4007
          p += blobDataTLen(val);
×
4008
        } else {
4009
          varDataCopy(p, val);
984,294✔
4010
          p += varDataTLen(val);
984,294✔
4011
        }
4012
      } else {
4013
        memcpy(p, val, pCol->bytes);
1,652,559✔
4014
        p += pCol->bytes;
1,652,559✔
4015
      }
4016
    }
4017
  }
4018
  return (int32_t)(p - keyBuf);
1,595,343✔
4019
}
4020

4021
uint64_t calcGroupId(char* pData, int32_t len) {
2,147,483,647✔
4022
  T_MD5_CTX context;
2,147,483,647✔
4023
  tMD5Init(&context);
2,147,483,647✔
4024
  tMD5Update(&context, (uint8_t*)pData, len);
2,147,483,647✔
4025
  tMD5Final(&context);
2,147,483,647✔
4026

4027
  // NOTE: only extract the initial 8 bytes of the final MD5 digest
4028
  uint64_t id = 0;
2,147,483,647✔
4029
  memcpy(&id, context.digest, sizeof(uint64_t));
2,147,483,647✔
4030
  if (0 == id) memcpy(&id, context.digest + 8, sizeof(uint64_t));
2,147,483,647✔
4031
  return id;
2,147,483,647✔
4032
}
4033

4034
SNodeList* makeColsNodeArrFromSortKeys(SNodeList* pSortKeys) {
105,492✔
4035
  SNode*     node;
4036
  SNodeList* ret = NULL;
105,492✔
4037
  FOREACH(node, pSortKeys) {
321,840✔
4038
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)node;
216,348✔
4039
    int32_t           code = nodesListMakeAppend(&ret, pSortKey->pExpr);
216,348✔
4040
    if (code != TSDB_CODE_SUCCESS) {
216,348✔
4041
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4042
      terrno = code;
×
4043
      return NULL;
×
4044
    }
4045
  }
4046
  return ret;
105,492✔
4047
}
4048

4049
int32_t extractKeysLen(const SArray* keys, int32_t* pLen) {
105,492✔
4050
  int32_t code = TSDB_CODE_SUCCESS;
105,492✔
4051
  int32_t lino = 0;
105,492✔
4052
  int32_t len = 0;
105,492✔
4053
  int32_t keyNum = taosArrayGetSize(keys);
105,492✔
4054
  for (int32_t i = 0; i < keyNum; ++i) {
267,306✔
4055
    SColumn* pCol = (SColumn*)taosArrayGet(keys, i);
162,261✔
4056
    QUERY_CHECK_NULL(pCol, code, lino, _end, terrno);
161,814✔
4057
    len += pCol->bytes;
161,814✔
4058
  }
4059
  len += sizeof(int8_t) * keyNum;  // null flag
105,045✔
4060
  *pLen = len;
105,045✔
4061

4062
_end:
105,045✔
4063
  if (code != TSDB_CODE_SUCCESS) {
105,045✔
4064
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4065
  }
4066
  return code;
105,492✔
4067
}
4068

4069
int32_t parseErrorMsgFromAnalyticServer(SJson* pJson, const char* pId) {
×
4070
  int32_t code = TSDB_CODE_ANA_ANODE_RETURN_ERROR;
×
4071
  if (pJson == NULL) {
×
4072
    return code;
×
4073
  }
4074

4075
  char    pMsg[1024] = {0};
×
4076
  int32_t ret = tjsonGetStringValue(pJson, "msg", pMsg);
×
4077

4078
  if (ret == 0) {
×
4079
    qError("%s failed to exec imputation, msg:%s", pId, pMsg);
×
4080
    if (strstr(pMsg, "white noise") != NULL) {
×
4081
      code = TSDB_CODE_ANA_WN_DATA;
×
4082
    } else if (strstr(pMsg, "white-noise") != NULL) {
×
4083
      code = TSDB_CODE_ANA_WN_DATA;
×
4084
    } else if (strstr(pMsg, "[Errno 111] Connection refused") != NULL) {
×
4085
      code = TSDB_CODE_ANA_ALGO_NOT_LOAD;
×
4086
    }
4087
  } else {
4088
    qError("%s failed to extract msg from server, unknown error", pId);
×
4089
  }
4090

4091
  return code;
×
4092
}
4093

4094

4095
int32_t createBlockFromRemoteValueNode(SSDataBlock** ppBlock, SRemoteValueNode* pRemote) {
21,978,211✔
4096
  SValueNode* pVal = (SValueNode*)pRemote;
21,978,211✔
4097
  int32_t code = 0;
21,978,211✔
4098
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
21,978,211✔
4099
  if (pBlock == NULL) {
21,977,296✔
4100
    return terrno;
×
4101
  }
4102

4103
  pBlock->pDataBlock = taosArrayInit(1, sizeof(SColumnInfoData));
21,977,296✔
4104
  if (pBlock->pDataBlock == NULL) {
21,976,818✔
4105
    code = terrno;
×
4106
    taosMemoryFree(pBlock);
×
4107
    return code;
×
4108
  }
4109

4110
  SColumnInfoData idata =
21,977,278✔
4111
      createColumnInfoData(pVal->node.resType.type, pVal->node.resType.bytes, 0);
21,977,738✔
4112
  idata.info.scale = pVal->node.resType.scale;
21,978,211✔
4113
  idata.info.precision = pVal->node.resType.precision;
21,978,229✔
4114

4115
  code = blockDataAppendColInfo(pBlock, &idata);
21,977,296✔
4116
  if (code != TSDB_CODE_SUCCESS) {
21,977,738✔
4117
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4118
    blockDataDestroy(pBlock);
×
4119
    *ppBlock = NULL;
×
4120
    return code;
×
4121
  }
4122

4123
  *ppBlock = pBlock;
21,977,738✔
4124

4125
  return code;
21,978,211✔
4126
}
4127

4128

4129
int32_t extractSingleRspBlock(SRetrieveTableRsp* pRetrieveRsp, SSDataBlock* pb) {
21,978,211✔
4130
  int32_t            code = TSDB_CODE_SUCCESS;
21,978,211✔
4131
  int32_t            lino = 0;
21,978,211✔
4132
  void*              decompBuf = NULL;
21,978,211✔
4133

4134
  char* pNextStart = pRetrieveRsp->data;
21,978,211✔
4135
  char* pStart = pNextStart;
21,977,756✔
4136

4137
  int32_t index = 0;
21,978,211✔
4138

4139
  if (pRetrieveRsp->compressed) {  // decompress the data
21,978,211✔
4140
    decompBuf = taosMemoryMalloc(pRetrieveRsp->payloadLen);
×
4141
    QUERY_CHECK_NULL(decompBuf, code, lino, _end, terrno);
×
4142
  }
4143

4144
  int32_t compLen = *(int32_t*)pStart;
21,978,684✔
4145
  pStart += sizeof(int32_t);
21,978,211✔
4146

4147
  int32_t rawLen = *(int32_t*)pStart;
21,978,211✔
4148
  pStart += sizeof(int32_t);
21,978,684✔
4149
  QUERY_CHECK_CONDITION((compLen <= rawLen && compLen != 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
21,976,818✔
4150

4151
  pNextStart = pStart + compLen;
21,976,818✔
4152
  if (pRetrieveRsp->compressed && (compLen < rawLen)) {
21,976,823✔
4153
    int32_t t = tsDecompressString(pStart, compLen, 1, decompBuf, rawLen, ONE_STAGE_COMP, NULL, 0);
×
4154
    QUERY_CHECK_CONDITION((t == rawLen), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
4155
    pStart = decompBuf;
×
4156
  }
4157

4158
  code = blockDecodeInternal(pb, pStart, (const char**)&pStart);
21,978,211✔
4159
  if (code != 0) {
21,978,211✔
4160
    taosMemoryFreeClear(pRetrieveRsp);
×
4161
    goto _end;
×
4162
  }
4163

4164
_end:
21,978,211✔
4165
  if (code != TSDB_CODE_SUCCESS) {
21,978,211✔
4166
    blockDataDestroy(pb);
×
4167
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4168
  }
4169
  return code;
21,978,211✔
4170
}
4171

4172
int32_t setValueFromResBlock(STaskSubJobCtx* ctx, SRemoteValueNode* pRes, SSDataBlock* pBlock) {
21,978,211✔
4173
  int32_t code = 0;
21,978,211✔
4174
  bool needFree = true;
21,978,211✔
4175
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
21,978,211✔
4176
  if (NULL == pBlock->pDataBlock || 1 != colNum || pBlock->info.rows > 1) {
21,977,278✔
4177
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d, rows:%" PRId64, 
3,675✔
4178
      ctx->idStr, pBlock->pDataBlock, colNum, pBlock->info.rows);
4179
    return TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
×
4180
  }
4181
  
4182
  pRes->val.node.type = QUERY_NODE_VALUE;
21,973,603✔
4183
  pRes->val.flag &= (~VALUE_FLAG_VAL_UNSET);
21,977,751✔
4184
  pRes->val.translate = true;
21,976,849✔
4185
  
4186
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
21,977,751✔
4187
  if (colDataIsNull_s(pCol, 0)) {
21,977,751✔
4188
    pRes->val.isNull = true;
1,800,563✔
4189
  } else {
4190
    code = nodesSetValueNodeValueExt(&pRes->val, colDataGetData(pCol, 0), &needFree);
20,177,188✔
4191
  }
4192

4193
  if (!needFree) {
21,975,425✔
4194
    pCol->pData = NULL;
14,560✔
4195
  }
4196

4197
  return code;
21,975,425✔
4198
}
4199

4200
int32_t remoteFetchCallBack(void* param, SDataBuf* pMsg, int32_t code) {
39,638,546✔
4201
  SScalarFetchParam* pParam = (SScalarFetchParam*)param;
39,638,546✔
4202
  STaskSubJobCtx* ctx = pParam->pSubJobCtx;
39,638,546✔
4203
  SSDataBlock* pResBlock = NULL;
39,639,479✔
4204
  
4205
  taosMemoryFreeClear(pMsg->pEpSet);
39,638,065✔
4206

4207
  if (NULL == ctx) {
39,638,556✔
4208
    qWarn("scl fetch ctx not exists since it may have been released");
7,126✔
4209
    goto _exit;
7,126✔
4210
  }
4211

4212
  qDebug("%s subQIdx %d got rsp, code:%d, rsp:%p", ctx->idStr, pParam->subQIdx, code, pMsg->pData);
39,631,430✔
4213

4214
  taosWLockLatch(&ctx->lock);
39,631,430✔
4215
  ctx->param = NULL;
39,632,358✔
4216
  taosWUnLockLatch(&ctx->lock);
39,632,358✔
4217

4218
  if (ctx->transporterId > 0) {
39,632,813✔
4219
    int32_t ret = asyncFreeConnById(ctx->rpcHandle, ctx->transporterId);
39,632,340✔
4220
    if (ret != 0) {
39,631,880✔
4221
      qDebug("%s failed to free subQ rpc handle, code:%s, subQIdx:%d", ctx->idStr, tstrerror(ret), pParam->subQIdx);
×
4222
    }
4223
    ctx->transporterId = -1;
39,631,880✔
4224
  }
4225

4226
  if (0 == code && NULL == pMsg->pData) {
39,631,443✔
4227
    qError("%s invalid rsp msg, msgType:%d, len:%d", ctx->idStr, pMsg->msgType, pMsg->len);
×
4228
    code = TSDB_CODE_QRY_INVALID_MSG;
×
4229
  }
4230

4231
  if (code == TSDB_CODE_SUCCESS) {
39,631,880✔
4232
    SRetrieveTableRsp* pRsp = pMsg->pData;
32,692,800✔
4233
    pRsp->numOfRows = htobe64(pRsp->numOfRows);
32,692,800✔
4234
    pRsp->compLen = htonl(pRsp->compLen);
32,692,368✔
4235
    pRsp->payloadLen = htonl(pRsp->payloadLen);
32,693,273✔
4236
    pRsp->numOfCols = htonl(pRsp->numOfCols);
32,691,890✔
4237
    pRsp->useconds = htobe64(pRsp->useconds);
32,689,155✔
4238
    pRsp->numOfBlocks = htonl(pRsp->numOfBlocks);
32,690,980✔
4239

4240
    if (pRsp->numOfRows > 1 || pRsp->numOfBlocks > 1 || !pRsp->completed) {
32,692,350✔
4241
      qError("%s invalid scl fetch rsp received, subQIdx:%d, rows:%" PRId64 ", blocks:%d, completed:%d", 
761,881✔
4242
        ctx->idStr, pParam->subQIdx, pRsp->numOfRows, pRsp->numOfBlocks, pRsp->completed);
4243
      ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
761,881✔
4244
    } else if (0 == pRsp->numOfRows) {
31,931,415✔
4245
      SRemoteValueNode* pRemote = (SRemoteValueNode*)pParam->pRes;
9,957,345✔
4246
      pRemote->val.node.type = QUERY_NODE_VALUE;
9,956,890✔
4247
      pRemote->val.isNull = true;
9,957,340✔
4248
      pRemote->val.translate = true;
9,957,345✔
4249
      pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
9,956,430✔
4250
      taosArraySet(ctx->subResValues, pParam->subQIdx, &pParam->pRes);
9,957,340✔
4251
    } else {
4252
      qDebug("%s scl fetch rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
21,976,810✔
4253
      ctx->code = createBlockFromRemoteValueNode(&pResBlock, pParam->pRes);
21,976,810✔
4254
      if (TSDB_CODE_SUCCESS == ctx->code) {
21,977,278✔
4255
        ctx->code = blockDataEnsureCapacity(pResBlock, 1);
21,977,769✔
4256
      }
4257
      if (TSDB_CODE_SUCCESS == ctx->code) {
21,977,265✔
4258
        ctx->code = extractSingleRspBlock(pRsp, pResBlock);
21,978,211✔
4259
      }
4260
      if (TSDB_CODE_SUCCESS == ctx->code) {
21,976,818✔
4261
        ctx->code = setValueFromResBlock(ctx, pParam->pRes, pResBlock);
21,978,684✔
4262
      }
4263
      if (TSDB_CODE_SUCCESS == ctx->code) {
21,975,425✔
4264
        taosArraySet(ctx->subResValues, pParam->subQIdx, &pParam->pRes);
21,978,224✔
4265
      }
4266
    }
4267
  } else {
4268
    ctx->code = rpcCvtErrCode(code);
6,939,080✔
4269
    if (ctx->code != code) {
6,939,080✔
4270
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s, cvted error: %s", ctx->idStr, pParam->subQIdx,
×
4271
             tstrerror(code), tstrerror(ctx->code));
4272
    } else {
4273
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s", ctx->idStr, pParam->subQIdx, tstrerror(code));
6,939,080✔
4274
    }
4275
  }
4276
  
4277
  code = tsem_post(&pParam->pSubJobCtx->ready);
39,630,510✔
4278
  if (code != TSDB_CODE_SUCCESS) {
39,632,813✔
4279
    qError("failed to invoke post when scl fetch rsp is ready, code:%s", tstrerror(code));
×
4280
  }
4281

4282
_exit:
39,639,939✔
4283

4284
  taosMemoryFree(pMsg->pData);
39,639,006✔
4285
  blockDataDestroy(pResBlock);
39,637,613✔
4286

4287
  return code;
39,637,613✔
4288
}
4289

4290

4291
int32_t fetchRemoteValueImpl(STaskSubJobCtx* ctx, int32_t subQIdx, SRemoteValueNode* pRes) {
39,630,587✔
4292
  int32_t          code = TSDB_CODE_SUCCESS;
39,630,587✔
4293
  int32_t          lino = 0;
39,630,587✔
4294
  SDownstreamSourceNode* pSource = (SDownstreamSourceNode*)taosArrayGetP(ctx->subEndPoints, subQIdx);
39,630,587✔
4295

4296
  SResFetchReq req = {0};
39,623,289✔
4297
  req.header.vgId = pSource->addr.nodeId;
39,623,286✔
4298
  req.sId = pSource->sId;
39,622,804✔
4299
  req.clientId = pSource->clientId;
39,629,061✔
4300
  req.taskId = pSource->taskId;
39,625,780✔
4301
  req.queryId = ctx->queryId;
39,628,170✔
4302
  req.execId = pSource->execId;
39,612,443✔
4303

4304
  int32_t msgSize = tSerializeSResFetchReq(NULL, 0, &req, false);
39,613,873✔
4305
  if (msgSize < 0) {
39,622,634✔
4306
    return msgSize;
×
4307
  }
4308

4309
  void* msg = taosMemoryCalloc(1, msgSize);
39,622,634✔
4310
  if (NULL == msg) {
39,607,766✔
4311
    return terrno;
×
4312
  }
4313

4314
  msgSize = tSerializeSResFetchReq(msg, msgSize, &req, false);
39,607,766✔
4315
  if (msgSize < 0) {
39,626,947✔
4316
    taosMemoryFree(msg);
×
4317
    return msgSize;
×
4318
  }
4319

4320
  qDebug("%s scl build fetch msg and send to nodeId:%d, ep:%s, clientId:0x%" PRIx64 " taskId:0x%" PRIx64
39,626,947✔
4321
         ", execId:%d",
4322
         ctx->idStr, pSource->addr.nodeId, pSource->addr.epSet.eps[0].fqdn, pSource->clientId,
4323
         pSource->taskId, pSource->execId);
4324

4325
  // send the fetch remote task result reques
4326
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
39,630,049✔
4327
  if (NULL == pMsgSendInfo) {
39,617,011✔
4328
    taosMemoryFreeClear(msg);
×
4329
    qError("%s prepare message %d failed", ctx->idStr, (int32_t)sizeof(SMsgSendInfo));
×
4330
    return terrno;
×
4331
  }
4332

4333
  SScalarFetchParam* param = taosMemoryMalloc(sizeof(SScalarFetchParam));
39,617,011✔
4334
  if (NULL == param) {
39,619,384✔
4335
    taosMemoryFreeClear(msg);
×
4336
    taosMemoryFreeClear(pMsgSendInfo);
×
4337
    qError("%s prepare param %d failed", ctx->idStr, (int32_t)sizeof(SScalarFetchParam));
×
4338
    return terrno;
×
4339
  }
4340

4341
  taosWLockLatch(&ctx->lock);
39,619,384✔
4342
  
4343
  if (ctx->code) {
39,635,937✔
UNCOV
4344
    qError("task has been killed, error:%s", tstrerror(ctx->code));
×
UNCOV
4345
    taosMemoryFree(param);
×
UNCOV
4346
    taosMemoryFreeClear(msg);
×
UNCOV
4347
    taosMemoryFreeClear(pMsgSendInfo);
×
UNCOV
4348
    code = ctx->code;
×
UNCOV
4349
    taosWUnLockLatch(&ctx->lock);
×
UNCOV
4350
    goto _end;
×
4351
  } else {
4352
    ctx->param = param;
39,621,113✔
4353
  }
4354
  
4355
  taosWUnLockLatch(&ctx->lock);
39,636,798✔
4356

4357
  param->subQIdx = subQIdx;
39,625,422✔
4358
  param->pRes = pRes;
39,625,078✔
4359
  param->pSubJobCtx = ctx;
39,626,404✔
4360

4361
  pMsgSendInfo->param = param;
39,627,343✔
4362
  pMsgSendInfo->paramFreeFp = taosAutoMemoryFree;
39,615,655✔
4363
  pMsgSendInfo->msgInfo.pData = msg;
39,634,257✔
4364
  pMsgSendInfo->msgInfo.len = msgSize;
39,631,046✔
4365
  pMsgSendInfo->msgType = pSource->fetchMsgType;
39,612,328✔
4366
  pMsgSendInfo->fp = remoteFetchCallBack;
39,635,353✔
4367
  pMsgSendInfo->requestId = ctx->queryId;
39,630,206✔
4368

4369
  code = asyncSendMsgToServer(ctx->rpcHandle, &pSource->addr.epSet, &ctx->transporterId, pMsgSendInfo);
39,615,600✔
4370
  QUERY_CHECK_CODE(code, lino, _end);
39,639,582✔
4371

4372
  code = qSemWait(ctx->pTaskInfo, &ctx->ready);
39,639,582✔
4373
  if (isTaskKilled(ctx->pTaskInfo)) {
39,640,437✔
4374
    code = getTaskCode(ctx->pTaskInfo);
11,707✔
4375
  } else {
4376
    code = ctx->code;
39,628,270✔
4377
  }
4378
      
4379
_end:
39,640,497✔
4380

4381
  taosWLockLatch(&ctx->lock);
39,640,042✔
4382
  ctx->param = NULL;
39,640,957✔
4383
  taosWUnLockLatch(&ctx->lock);
39,640,437✔
4384

4385
  if (code != TSDB_CODE_SUCCESS) {
39,640,497✔
4386
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
7,704,473✔
4387
  }
4388
  return code;
39,639,517✔
4389
}
4390

4391

4392
int32_t qFetchRemoteValue(void* pCtx, int32_t subQIdx, SRemoteValueNode* pRes) {
40,943,808✔
4393
  STaskSubJobCtx*  ctx = (STaskSubJobCtx*)pCtx;
40,943,808✔
4394
  int32_t code = 0, lino = 0;
40,943,808✔
4395
  int32_t       subEndPoinsNum = taosArrayGetSize(ctx->subEndPoints);
40,943,808✔
4396
  if (subQIdx >= subEndPoinsNum) {
40,926,760✔
4397
    qError("%s invalid subQIdx %d, subEndPointsNum:%d", ctx->idStr, subQIdx, subEndPoinsNum);
×
4398
    return TSDB_CODE_QRY_SUBQ_NOT_FOUND;
×
4399
  }
4400

4401
  SValueNode** ppRes = taosArrayGet(ctx->subResValues, subQIdx);
40,926,760✔
4402
  if (NULL == *ppRes) {
40,938,103✔
4403
    TAOS_CHECK_EXIT(fetchRemoteValueImpl(ctx, subQIdx, pRes));
39,630,434✔
4404
    *ppRes = (SValueNode*)pRes;
31,935,044✔
4405
  } else {
4406
    TAOS_CHECK_EXIT(valueNodeCopy(*ppRes, &pRes->val));
1,312,782✔
4407
    pRes->val.node.type = QUERY_NODE_VALUE;
1,312,782✔
4408
  }
4409

4410
_exit:
40,953,279✔
4411

4412
  if (code) {
40,953,279✔
4413
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
7,704,473✔
4414
  }
4415

4416
  return code;
40,952,299✔
4417
}
4418

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