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

taosdata / TDengine / #4942

29 Jan 2026 06:55AM UTC coverage: 66.788% (-0.08%) from 66.868%
#4942

push

travis-ci

web-flow
fix: explain no response issue (#34435)

17 of 18 new or added lines in 3 files covered. (94.44%)

672 existing lines in 122 files now uncovered.

204165 of 305690 relevant lines covered (66.79%)

125991233.79 hits per line

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

75.81
/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) {
959,081,623✔
66
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->limit) ? -1 : ((SLimitNode*)pLimit)->limit->datum.i;
959,081,623✔
67
}
68
static int64_t getOffset(const SNode* pLimit) {
959,024,664✔
69
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->offset) ? -1 : ((SLimitNode*)pLimit)->offset->datum.i;
959,024,664✔
70
}
71
static void releaseColInfoData(void* pCol);
72
int32_t sendFetchRemoteNodeReq(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes);
73

74
void initResultRowInfo(SResultRowInfo* pResultRowInfo) {
331,961,416✔
75
  pResultRowInfo->size = 0;
331,961,416✔
76
  pResultRowInfo->cur.pageId = -1;
331,997,688✔
77
}
332,047,628✔
78

79
void closeResultRow(SResultRow* pResultRow) { pResultRow->closed = true; }
5,299,134✔
80

81
void resetResultRow(SResultRow* pResultRow, size_t entrySize) {
870,810,437✔
82
  pResultRow->numOfRows = 0;
870,810,437✔
83
  pResultRow->closed = false;
870,810,650✔
84
  pResultRow->endInterp = false;
870,810,650✔
85
  pResultRow->startInterp = false;
870,810,863✔
86

87
  if (entrySize > 0) {
870,809,915✔
88
    memset(pResultRow->pEntryInfo, 0, entrySize);
870,809,354✔
89
  }
90
}
870,812,180✔
91

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

97
size_t getResultRowSize(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
201,908,103✔
98
  int32_t rowSize = (numOfOutput * sizeof(SResultRowEntryInfo)) + sizeof(SResultRow);
201,908,103✔
99

100
  for (int32_t i = 0; i < numOfOutput; ++i) {
951,308,253✔
101
    rowSize += pCtx[i].resDataInfo.interBufSize;
749,431,487✔
102
  }
103

104
  return rowSize;
201,876,766✔
105
}
106

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

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

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

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

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

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

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

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

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

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

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

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

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

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

220
void cleanupGroupResInfo(SGroupResInfo* pGroupResInfo) {
108,896,991✔
221
  taosMemoryFreeClear(pGroupResInfo->pBuf);
108,896,991✔
222
  if (pGroupResInfo->freeItem) {
108,907,754✔
223
    //    taosArrayDestroy(pGroupResInfo->pRows);
224
    taosArrayDestroyEx(pGroupResInfo->pRows, freeEx);
×
225
    pGroupResInfo->freeItem = false;
×
226
    pGroupResInfo->pRows = NULL;
×
227
  } else {
228
    taosArrayDestroy(pGroupResInfo->pRows);
108,893,007✔
229
    pGroupResInfo->pRows = NULL;
108,867,714✔
230
  }
231
  pGroupResInfo->index = 0;
108,873,122✔
232
  pGroupResInfo->delIndex = 0;
108,866,216✔
233
}
108,887,853✔
234

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

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

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

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

255
int32_t initGroupedResultInfo(SGroupResInfo* pGroupResInfo, SSHashObj* pHashmap, int32_t order) {
80,427,198✔
256
  int32_t code = TSDB_CODE_SUCCESS;
80,427,198✔
257
  int32_t lino = 0;
80,427,198✔
258
  if (pGroupResInfo->pRows != NULL) {
80,427,198✔
259
    taosArrayDestroy(pGroupResInfo->pRows);
5,096,428✔
260
  }
261
  if (pGroupResInfo->pBuf) {
80,429,604✔
262
    taosMemoryFree(pGroupResInfo->pBuf);
5,096,922✔
263
    pGroupResInfo->pBuf = NULL;
5,094,954✔
264
  }
265

266
  // extract the result rows information from the hash map
267
  int32_t size = tSimpleHashGetSize(pHashmap);
80,425,831✔
268

269
  void* pData = NULL;
80,428,009✔
270
  pGroupResInfo->pRows = taosArrayInit(size, POINTER_BYTES);
80,428,009✔
271
  QUERY_CHECK_NULL(pGroupResInfo->pRows, code, lino, _end, terrno);
80,422,767✔
272

273
  size_t  keyLen = 0;
80,428,252✔
274
  int32_t iter = 0;
80,429,751✔
275
  int64_t bufLen = 0, offset = 0;
80,426,031✔
276

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

283
  pGroupResInfo->pBuf = taosMemoryMalloc(bufLen);
80,403,883✔
284
  QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
80,427,641✔
285

286
  iter = 0;
80,422,225✔
287
  while ((pData = tSimpleHashIterate(pHashmap, pData, &iter)) != NULL) {
2,147,483,647✔
288
    void* key = tSimpleHashGetKey(pData, &keyLen);
2,147,483,647✔
289

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

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

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

301
  if (order == TSDB_ORDER_ASC || order == TSDB_ORDER_DESC) {
80,403,110✔
302
    __compar_fn_t fn = (order == TSDB_ORDER_ASC) ? resultrowComparAsc : resultrowComparDesc;
5,125,820✔
303
    size = POINTER_BYTES;
5,125,820✔
304
    taosSort(pGroupResInfo->pRows->pData, taosArrayGetSize(pGroupResInfo->pRows), size, fn);
5,125,820✔
305
  }
306

307
  pGroupResInfo->index = 0;
80,403,110✔
308

309
_end:
80,429,025✔
310
  if (code != TSDB_CODE_SUCCESS) {
80,436,188✔
311
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
312
  }
313
  return code;
80,436,188✔
314
}
315

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

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

327
bool hasRemainResults(SGroupResInfo* pGroupResInfo) {
320,902,380✔
328
  if (pGroupResInfo->pRows == NULL) {
320,902,380✔
329
    return false;
×
330
  }
331

332
  return pGroupResInfo->index < taosArrayGetSize(pGroupResInfo->pRows);
320,902,988✔
333
}
334

335
int32_t getNumOfTotalRes(SGroupResInfo* pGroupResInfo) {
172,210,449✔
336
  if (pGroupResInfo->pRows == 0) {
172,210,449✔
337
    return 0;
×
338
  }
339

340
  return (int32_t)taosArrayGetSize(pGroupResInfo->pRows);
172,222,146✔
341
}
342

343
SArray* createSortInfo(SNodeList* pNodeList) {
49,767,719✔
344
  size_t numOfCols = 0;
49,767,719✔
345

346
  if (pNodeList != NULL) {
49,767,719✔
347
    numOfCols = LIST_LENGTH(pNodeList);
49,573,688✔
348
  } else {
349
    numOfCols = 0;
194,059✔
350
  }
351

352
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
49,769,058✔
353
  if (pList == NULL) {
49,755,371✔
354
    return pList;
×
355
  }
356

357
  for (int32_t i = 0; i < numOfCols; ++i) {
114,073,974✔
358
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
64,300,857✔
359
    if (!pSortKey) {
64,309,297✔
UNCOV
360
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
UNCOV
361
      taosArrayDestroy(pList);
×
362
      pList = NULL;
×
363
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
364
      break;
×
365
    }
366
    SBlockOrderInfo bi = {0};
64,309,297✔
367
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
64,306,848✔
368
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);
64,300,164✔
369

370
    if (nodeType(pSortKey->pExpr) != QUERY_NODE_COLUMN) {
64,296,805✔
371
      qError("invalid order by expr type:%d", nodeType(pSortKey->pExpr));
×
372
      taosArrayDestroy(pList);
×
373
      pList = NULL;
×
374
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
375
      break;
×
376
    }
377
    
378
    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
64,276,517✔
379
    bi.slotId = pColNode->slotId;
64,306,012✔
380
    void* tmp = taosArrayPush(pList, &bi);
64,318,260✔
381
    if (!tmp) {
64,318,260✔
382
      taosArrayDestroy(pList);
×
383
      pList = NULL;
×
384
      break;
×
385
    }
386
  }
387

388
  return pList;
49,772,774✔
389
}
390

391
SSDataBlock* createDataBlockFromDescNode(void* p) {
581,493,035✔
392
  SDataBlockDescNode* pNode = (SDataBlockDescNode*)p;
581,493,035✔
393
  int32_t      numOfCols = LIST_LENGTH(pNode->pSlots);
581,493,035✔
394
  SSDataBlock* pBlock = NULL;
581,677,307✔
395
  int32_t      code = createDataBlock(&pBlock);
581,650,720✔
396
  if (code) {
581,486,508✔
397
    terrno = code;
×
398
    return NULL;
×
399
  }
400

401
  pBlock->info.id.blockId = pNode->dataBlockId;
581,486,508✔
402
  pBlock->info.type = STREAM_INVALID;
581,528,011✔
403
  pBlock->info.calWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
581,520,464✔
404
  pBlock->info.watermark = INT64_MIN;
581,647,617✔
405

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

421
    code = blockDataAppendColInfo(pBlock, &idata);
2,147,483,647✔
422
    if (code != TSDB_CODE_SUCCESS) {
2,147,483,647✔
UNCOV
423
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
UNCOV
424
      blockDataDestroy(pBlock);
×
425
      pBlock = NULL;
×
426
      terrno = code;
×
427
      break;
×
428
    }
429
  }
430

431
  return pBlock;
581,898,163✔
432
}
433

434
int32_t prepareDataBlockBuf(SSDataBlock* pDataBlock, SColMatchInfo* pMatchInfo) {
182,291,421✔
435
  SDataBlockInfo* pBlockInfo = &pDataBlock->info;
182,291,421✔
436

437
  for (int32_t i = 0; i < taosArrayGetSize(pMatchInfo->pList); ++i) {
912,981,979✔
438
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
751,530,283✔
439
    if (!pItem) {
751,383,594✔
UNCOV
440
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
UNCOV
441
      return terrno;
×
442
    }
443

444
    if (pItem->isPk) {
751,383,594✔
445
      SColumnInfoData* pInfoData = taosArrayGet(pDataBlock->pDataBlock, pItem->dstSlotId);
20,943,803✔
446
      if (!pInfoData) {
20,538,567✔
447
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
448
        return terrno;
×
449
      }
450
      pBlockInfo->pks[0].type = pInfoData->info.type;
20,538,567✔
451
      pBlockInfo->pks[1].type = pInfoData->info.type;
20,571,228✔
452

453
      // allocate enough buffer size, which is pInfoData->info.bytes
454
      if (IS_VAR_DATA_TYPE(pItem->dataType.type)) {
20,568,702✔
455
        pBlockInfo->pks[0].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
6,275,861✔
456
        if (pBlockInfo->pks[0].pData == NULL) {
6,280,164✔
457
          return terrno;
×
458
        }
459

460
        pBlockInfo->pks[1].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
6,282,572✔
461
        if (pBlockInfo->pks[1].pData == NULL) {
6,282,015✔
462
          taosMemoryFreeClear(pBlockInfo->pks[0].pData);
×
463
          return terrno;
×
464
        }
465

466
        pBlockInfo->pks[0].nData = pInfoData->info.bytes;
6,285,605✔
467
        pBlockInfo->pks[1].nData = pInfoData->info.bytes;
6,286,218✔
468
      }
469

470
      break;
20,575,573✔
471
    }
472
  }
473

474
  return TSDB_CODE_SUCCESS;
182,079,061✔
475
}
476

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

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

500
    res->translate = true;
671,684✔
501
    res->node.resType = pSColumnNode->node.resType;
671,684✔
502

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

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

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

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

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

561
  return DEAL_RES_CONTINUE;
2,350,894✔
562
}
563

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

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

574
    return TSDB_CODE_SUCCESS;
×
575
  }
576

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

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

601
    return code;
×
602
  }
603

604
  SValueNode* pValue = (SValueNode*)pNew;
335,842✔
605
  *pQualified = pValue->datum.b;
335,842✔
606

607
  nodesDestroyNode(pNew);
335,842✔
608
  return TSDB_CODE_SUCCESS;
335,842✔
609
}
610

611
static EDealRes getColumn(SNode** pNode, void* pContext) {
56,082,986✔
612
  tagFilterAssist* pData = (tagFilterAssist*)pContext;
56,082,986✔
613
  SColumnNode*     pSColumnNode = NULL;
56,082,986✔
614
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
56,089,786✔
615
    pSColumnNode = *(SColumnNode**)pNode;
18,866,191✔
616
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
37,233,799✔
617
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
740,258✔
618
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
740,045✔
619
      pData->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
695,944✔
620
      if (NULL == pSColumnNode) {
695,944✔
621
        return DEAL_RES_ERROR;
×
622
      }
623
      pSColumnNode->colId = -1;
695,944✔
624
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
695,944✔
625
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
695,944✔
626
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
695,944✔
627
      nodesDestroyNode(*pNode);
695,944✔
628
      *pNode = (SNode*)pSColumnNode;
695,379✔
629
    } else {
630
      return DEAL_RES_CONTINUE;
44,314✔
631
    }
632
  } else {
633
    return DEAL_RES_CONTINUE;
36,493,021✔
634
  }
635

636
  void* data = taosHashGet(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
19,559,113✔
637
  if (!data) {
19,556,165✔
638
    int32_t tempRes =
639
        taosHashPut(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
16,838,132✔
640
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
16,840,205✔
641
      return DEAL_RES_ERROR;
×
642
    }
643
    pSColumnNode->slotId = pData->index++;
16,840,205✔
644
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
16,840,517✔
645
                         .type = pSColumnNode->node.resType.type,
16,833,484✔
646
                         .bytes = pSColumnNode->node.resType.bytes,
16,837,724✔
647
                         .pk = pSColumnNode->isPk};
16,830,300✔
648
#if TAG_FILTER_DEBUG
649
    qDebug("tagfilter build column info, slotId:%d, colId:%d, type:%d", pSColumnNode->slotId, cInfo.colId, cInfo.type);
650
#endif
651
    void* tmp = taosArrayPush(pData->cInfoList, &cInfo);
16,839,667✔
652
    if (!tmp) {
16,843,381✔
653
      return DEAL_RES_ERROR;
×
654
    }
655
  } else {
656
    SColumnNode* col = *(SColumnNode**)data;
2,718,033✔
657
    pSColumnNode->slotId = col->slotId;
2,718,866✔
658
  }
659

660
  return DEAL_RES_CONTINUE;
19,554,859✔
661
}
662

663
static int32_t createResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
15,650,191✔
664
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
15,650,191✔
665
  if (pColumnData == NULL) {
15,649,315✔
666
    return terrno;
×
667
  }
668

669
  pColumnData->info.type = pType->type;
15,649,315✔
670
  pColumnData->info.bytes = pType->bytes;
15,651,844✔
671
  pColumnData->info.scale = pType->scale;
15,650,946✔
672
  pColumnData->info.precision = pType->precision;
15,645,136✔
673

674
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
15,646,672✔
675
  if (code != TSDB_CODE_SUCCESS) {
15,639,641✔
676
    terrno = code;
×
677
    releaseColInfoData(pColumnData);
×
678
    return terrno;
×
679
  }
680

681
  pParam->columnData = pColumnData;
15,639,641✔
682
  pParam->colAlloced = true;
15,648,077✔
683
  return TSDB_CODE_SUCCESS;
15,642,240✔
684
}
685

686
static void releaseColInfoData(void* pCol) {
4,193,050✔
687
  if (pCol) {
4,193,050✔
688
    SColumnInfoData* col = (SColumnInfoData*)pCol;
4,193,050✔
689
    colDataDestroy(col);
4,193,050✔
690
    taosMemoryFree(col);
4,193,220✔
691
  }
692
}
4,192,661✔
693

694
void freeItem(void* p) {
187,961,273✔
695
  STUidTagInfo* pInfo = p;
187,961,273✔
696
  if (pInfo->pTagVal != NULL) {
187,961,273✔
697
    taosMemoryFree(pInfo->pTagVal);
187,618,157✔
698
  }
699
}
187,950,196✔
700

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

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

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

728
    (void)memcpy(pStart, &entry->colId, sizeof(col_id_t));
18,150✔
729
    pStart += sizeof(col_id_t);
18,150✔
730

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

776
  return TSDB_CODE_SUCCESS;
9,075✔
777
}
778

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

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

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

804
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
9,075✔
805
    extractTagDataEntry((SOperatorNode*)pTagCond, pIdWithVal);
×
806
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
9,075✔
807
    SNode* pChild = NULL;
9,075✔
808
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
27,225✔
809
      extractTagDataEntry((SOperatorNode*)pChild, pIdWithVal);
18,150✔
810
    }
811
  }
812

813
  taosArraySort(pIdWithVal, compareTagDataEntry);
9,075✔
814

815
  return TSDB_CODE_SUCCESS;
9,075✔
816
}
817

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

823
  char*   payload = NULL;
9,075✔
824
  int32_t len = 0;
9,075✔
825
  int32_t code = TSDB_CODE_SUCCESS;
9,075✔
826
  int32_t lino = 0;
9,075✔
827

828
  SArray* pIdWithVal = taosArrayInit(TARRAY_MIN_SIZE, sizeof(STagDataEntry));
9,075✔
829
  code = extractTagFilterTagDataEntries(pTagCond, pIdWithVal);
9,075✔
830
  QUERY_CHECK_CODE(code, lino, _end);
9,075✔
831
  for (int32_t i = 0; i < taosArrayGetSize(pIdWithVal); ++i) {
27,225✔
832
    STagDataEntry* pEntry = taosArrayGet(pIdWithVal, i);
18,150✔
833
    len += sizeof(col_id_t) + pEntry->bytes;
18,150✔
834
  }
835
  code = buildTagDataEntryKey(pIdWithVal, &payload, len);
9,075✔
836
  QUERY_CHECK_CODE(code, lino, _end);
9,075✔
837

838
  tMD5Init(pContext);
9,075✔
839
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
9,075✔
840
  tMD5Final(pContext);
9,075✔
841

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

852
static int32_t genTagFilterDigest(const SNode* pTagCond, T_MD5_CTX* pContext) {
30,845✔
853
  if (pTagCond == NULL) {
30,845✔
854
    return TSDB_CODE_SUCCESS;
29,115✔
855
  }
856

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

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

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

875
  taosMemoryFree(payload);
1,730✔
876
  return TSDB_CODE_SUCCESS;
1,730✔
877
}
878

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

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

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

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

906
int32_t qGetColumnsFromNodeList(void* data, bool isList, SArray** pColList) {
15,478,995✔
907
  int32_t code = TSDB_CODE_SUCCESS;
15,478,995✔
908
  tagFilterAssist ctx = {0};
15,478,995✔
909
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
15,479,933✔
910
  if (ctx.colHash == NULL) {
15,477,842✔
911
    code = terrno;
×
912
    goto end;
×
913
  }
914

915
  ctx.index = 0;
15,477,842✔
916
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
15,477,842✔
917
  if (ctx.cInfoList == NULL) {
15,479,248✔
918
    code = terrno;
1,906✔
919
    goto end;
×
920
  }
921

922
  if (isList) {
15,477,342✔
923
    SNode* pNode = NULL;
4,017,382✔
924
    FOREACH(pNode, (SNodeList*)data) {
8,211,535✔
925
      nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
4,194,153✔
926
      if (TSDB_CODE_SUCCESS != ctx.code) {
4,193,940✔
927
        code = ctx.code;
×
928
        goto end;
×
929
      }
930
      REPLACE_NODE(pNode);
4,193,940✔
931
    }
932
  } else {
933
    SNode* pNode = (SNode*)data;
11,459,960✔
934
    nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
11,459,960✔
935
    if (TSDB_CODE_SUCCESS != ctx.code) {
11,461,105✔
936
      code = ctx.code;
×
937
      goto end;
×
938
    }
939
  }
940
  
941
  if (pColList != NULL) *pColList = ctx.cInfoList;
15,470,599✔
942
  ctx.cInfoList = NULL;
15,476,043✔
943

944
end:
15,484,301✔
945
  taosHashCleanup(ctx.colHash);
15,478,191✔
946
  taosArrayDestroy(ctx.cInfoList);
15,461,308✔
947
  return code;
15,462,743✔
948
}
949

950
static int32_t buildGroupInfo(SColumnInfoData* pValue, int32_t i, SArray* gInfo) {
431,957✔
951
  int32_t code = TSDB_CODE_SUCCESS;
431,957✔
952
  SStreamGroupValue* v = taosArrayReserve(gInfo, 1);
431,957✔
953
  if (v == NULL) {
432,181✔
954
    code = terrno;
×
955
    goto end;
×
956
  }
957
  if (colDataIsNull_s(pValue, i)) {
864,362✔
958
    v->isNull = true;
5,775✔
959
  } else {
960
    v->isNull = false;
426,406✔
961
    char* data = colDataGetData(pValue, i);
426,406✔
962
    if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
426,193✔
963
      if (tTagIsJson(data)) {
×
964
        code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
×
965
        goto end;
×
966
      }
967
      if (tTagIsJsonNull(data)) {
×
968
        v->isNull = true;
×
969
        goto end;
×
970
      }
971
      int32_t len = getJsonValueLen(data);
×
972
      v->data.type = pValue->info.type;
×
973
      v->data.nData = len;
×
974
      v->data.pData = taosMemoryCalloc(1, len + 1);
×
975
      if (v->data.pData == NULL) {
×
976
        code = terrno;
×
977
        goto end;
×
978
      }
979
      memcpy(v->data.pData, data, len);
×
980
      qDebug("buildGroupInfo:%d add json data len:%d, data:%s", i, len, (char*)v->data.pData);
×
981
    } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
426,241✔
982
      if (varDataTLen(data) > pValue->info.bytes) {
312,584✔
983
        code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
984
        goto end;
×
985
      }
986
      v->data.type = pValue->info.type;
312,866✔
987
      v->data.nData = varDataLen(data);
313,079✔
988
      v->data.pData = taosMemoryCalloc(1, varDataLen(data) + 1);
313,079✔
989
      if (v->data.pData == NULL) {
313,079✔
990
        code = terrno;
×
991
        goto end;
×
992
      }
993
      memcpy(v->data.pData, varDataVal(data), varDataLen(data));
313,079✔
994
      qDebug("buildGroupInfo:%d add var data type:%d, len:%d, data:%s", i, pValue->info.type, varDataLen(data), (char*)v->data.pData);
313,079✔
995
    } else if (pValue->info.type == TSDB_DATA_TYPE_DECIMAL) {  // reader todo decimal
113,327✔
996
      v->data.type = pValue->info.type;
×
997
      v->data.nData = pValue->info.bytes;
×
998
      v->data.pData = taosMemoryCalloc(1, pValue->info.bytes);
×
999
      if (v->data.pData == NULL) {
×
1000
        code = terrno;
×
1001
        goto end;
×
1002
      }
1003
      memcpy(&v->data.pData, data, pValue->info.bytes);
×
1004
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
×
1005
    } else {  // reader todo decimal
1006
      v->data.type = pValue->info.type;
113,327✔
1007
      memcpy(&v->data.val, data, pValue->info.bytes);
113,327✔
1008
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
113,162✔
1009
    }
1010
  }
1011
end:
41,851✔
1012
  if (code != TSDB_CODE_SUCCESS) {
432,181✔
1013
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1014
    v->isNull = true;
×
1015
  }
1016
  return code;
432,181✔
1017
}
1018

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

1030
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
67,622✔
1031
  if (rows == 0) {
67,622✔
1032
    return;
×
1033
  }
1034

1035
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
67,622✔
1036
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
67,622✔
1037

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

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

1061
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
177,141✔
1062
    SColumnInfo* tmp = (SColumnInfo*)taosArrayGet(pColList, i);
109,732✔
1063
    if (tmp != NULL && tmp->colId == -1) {
109,519✔
1064
      tbNameIndex = i;
67,409✔
1065
    }
1066
  }
1067
  
1068
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
67,409✔
1069
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
67,409✔
1070
  taosArrayDestroy(pColList);
67,622✔
1071
  if (pResBlock == NULL) {
67,622✔
1072
    code = terrno;
247✔
1073
    goto end;
247✔
1074
  }
1075

1076
  pBlockList = taosArrayInit(2, POINTER_BYTES);
67,375✔
1077
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
67,375✔
1078

1079
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
67,375✔
1080
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
67,375✔
1081

1082
  groupData = taosArrayInit(2, POINTER_BYTES);
67,375✔
1083
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
67,375✔
1084

1085
  SNode* pNode = NULL;
67,375✔
1086
  FOREACH(pNode, group) {
176,613✔
1087
    SScalarParam output = {0};
109,025✔
1088

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

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

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

1121
    if (code != TSDB_CODE_SUCCESS) {
109,025✔
1122
      releaseColInfoData(output.columnData);
×
1123
      goto end;
×
1124
    }
1125

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

1130
  for (int i = 0; i < rows; i++) {
303,893✔
1131
    gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
236,518✔
1132
    QUERY_CHECK_NULL(gInfo, code, lino, end, terrno);
236,518✔
1133

1134
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
236,518✔
1135
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
236,518✔
1136

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

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

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

1169
  if (code != TSDB_CODE_SUCCESS) {
67,622✔
1170
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
247✔
1171
  }
1172
}
1173

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

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

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

1202
    nodesFree(listNode);
×
1203

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

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

1217
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
3,950,260✔
1218
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
3,950,260✔
1219

1220
  for (int32_t i = 0; i < rows; ++i) {
26,094,734✔
1221
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
22,144,947✔
1222
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
22,144,449✔
1223
    STUidTagInfo info = {.uid = pkeyInfo->uid};
22,144,449✔
1224
    void*        tmp = taosArrayPush(pUidTagList, &info);
22,144,972✔
1225
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
22,144,972✔
1226
  }
1227

1228
  if (taosArrayGetSize(pUidTagList) > 0) {
3,949,787✔
1229
    code = pAPI->metaFn.getTableTagsByUid(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
3,950,260✔
1230
  } else {
UNCOV
1231
    code = pAPI->metaFn.getTableTags(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
×
1232
  }
1233
  if (code != TSDB_CODE_SUCCESS) {
3,949,640✔
1234
    goto end;
×
1235
  }
1236

1237
  SArray* pColList = NULL;
3,949,640✔
1238
  code = qGetColumnsFromNodeList(group, true, &pColList); 
3,949,640✔
1239
  if (code != TSDB_CODE_SUCCESS) {
3,948,789✔
1240
    goto end;
×
1241
  }
1242

1243
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
3,948,789✔
1244
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
3,948,653✔
1245
  taosArrayDestroy(pColList);
3,950,260✔
1246
  if (pResBlock == NULL) {
3,949,762✔
1247
    code = terrno;
×
1248
    goto end;
×
1249
  }
1250

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

1254
  pBlockList = taosArrayInit(2, POINTER_BYTES);
3,949,762✔
1255
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
3,949,412✔
1256

1257
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
3,950,260✔
1258
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
3,950,260✔
1259

1260
  groupData = taosArrayInit(2, POINTER_BYTES);
3,950,260✔
1261
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
3,949,247✔
1262

1263
  SNode* pNode = NULL;
3,949,247✔
1264
  FOREACH(pNode, group) {
8,032,783✔
1265
    SScalarParam output = {0};
4,082,953✔
1266

1267
    switch (nodeType(pNode)) {
4,082,953✔
1268
      case QUERY_NODE_VALUE:
×
1269
        break;
×
1270
      case QUERY_NODE_COLUMN:
4,082,455✔
1271
      case QUERY_NODE_OPERATOR:
1272
      case QUERY_NODE_FUNCTION: {
1273
        SExprNode* expNode = (SExprNode*)pNode;
4,082,455✔
1274
        code = createResultData(&expNode->resType, rows, &output);
4,082,455✔
1275
        if (code != TSDB_CODE_SUCCESS) {
4,081,681✔
1276
          goto end;
×
1277
        }
1278
        break;
4,081,681✔
1279
      }
UNCOV
1280
      case QUERY_NODE_REMOTE_VALUE: {
×
UNCOV
1281
        SRemoteValueNode* pRemote = (SRemoteValueNode*)pNode;
×
UNCOV
1282
        code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, pNode);
×
UNCOV
1283
        QUERY_CHECK_CODE(code, lino, end);
×
UNCOV
1284
        break;
×
1285
      }
1286
      
1287
      default:
×
1288
        code = TSDB_CODE_OPS_NOT_SUPPORT;
×
1289
        goto end;
×
1290
    }
1291

1292
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
4,081,681✔
1293
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
4,065,486✔
1294
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
4,065,486✔
1295
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
4,063,366✔
1296
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
4,063,366✔
1297
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
17,321✔
UNCOV
1298
      continue;
×
1299
    } else {
1300
      gTaskScalarExtra.pStreamInfo = NULL;
18,315✔
1301
      gTaskScalarExtra.pStreamRange = NULL;
18,315✔
1302
      code = scalarCalculate(pNode, pBlockList, &output, &gTaskScalarExtra);
18,315✔
1303
    }
1304

1305
    if (code != TSDB_CODE_SUCCESS) {
4,083,314✔
1306
      releaseColInfoData(output.columnData);
×
1307
      goto end;
×
1308
    }
1309

1310
    void* tmp = taosArrayPush(groupData, &output.columnData);
4,084,034✔
1311
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
4,084,034✔
1312
  }
1313

1314
  int32_t keyLen = 0;
3,949,762✔
1315
  SNode*  node;
1316
  FOREACH(node, group) {
8,028,370✔
1317
    SExprNode* pExpr = (SExprNode*)node;
4,081,869✔
1318
    keyLen += pExpr->resType.bytes;
4,081,869✔
1319
  }
1320

1321
  int32_t nullFlagSize = sizeof(int8_t) * LIST_LENGTH(group);
3,946,788✔
1322
  keyLen += nullFlagSize;
3,947,397✔
1323

1324
  keyBuf = taosMemoryCalloc(1, keyLen);
3,947,397✔
1325
  if (keyBuf == NULL) {
3,946,628✔
1326
    code = terrno;
×
1327
    goto end;
×
1328
  }
1329

1330
  if (initRemainGroups) {
3,946,628✔
1331
    pTableListInfo->remainGroups =
1,908,243✔
1332
        taosHashInit(rows, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
1,906,749✔
1333
    if (pTableListInfo->remainGroups == NULL) {
1,908,243✔
1334
      code = terrno;
×
1335
      goto end;
×
1336
    }
1337
  }
1338

1339
  for (int i = 0; i < rows; i++) {
26,087,927✔
1340
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
22,138,688✔
1341
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
22,139,046✔
1342

1343
    if (groupIdMap != NULL){
22,139,046✔
1344
      gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
62,770✔
1345
    }
1346
    
1347
    char* isNull = (char*)keyBuf;
22,139,210✔
1348
    char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(group);
22,139,210✔
1349
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
45,353,587✔
1350
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
23,213,779✔
1351

1352
      if (groupIdMap != NULL && gInfo != NULL) {
23,216,680✔
1353
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
74,909✔
1354
        if (ret != TSDB_CODE_SUCCESS) {
75,133✔
1355
          qError("buildGroupInfo failed at line %d since %s", __LINE__, tstrerror(ret));
×
1356
          taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1357
          gInfo = NULL;
×
1358
        }
1359
      }
1360
      
1361
      if (colDataIsNull_s(pValue, i)) {
46,433,283✔
1362
        isNull[j] = 1;
96,191✔
1363
      } else {
1364
        isNull[j] = 0;
23,120,188✔
1365
        char* data = colDataGetData(pValue, i);
23,119,715✔
1366
        if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
23,117,767✔
1367
          // if (tTagIsJson(data)) {
1368
          //   code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
1369
          //   goto end;
1370
          // }
1371
          if (tTagIsJsonNull(data)) {
74,817✔
1372
            isNull[j] = 1;
×
1373
            continue;
×
1374
          }
1375
          int32_t len = getJsonValueLen(data);
74,817✔
1376
          memcpy(pStart, data, len);
74,817✔
1377
          pStart += len;
74,817✔
1378
        } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
23,042,059✔
1379
          if (IS_STR_DATA_BLOB(pValue->info.type)) {
20,223,331✔
1380
            if (blobDataTLen(data) > TSDB_MAX_BLOB_LEN) {
3,601✔
1381
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1382
              goto end;
×
1383
            }
1384
            memcpy(pStart, data, blobDataTLen(data));
×
1385
            pStart += blobDataTLen(data);
×
1386
          } else {
1387
            if (varDataTLen(data) > pValue->info.bytes) {
20,220,411✔
1388
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1389
              goto end;
×
1390
            }
1391
            memcpy(pStart, data, varDataTLen(data));
20,222,993✔
1392
            pStart += varDataTLen(data);
20,222,993✔
1393
          }
1394
        } else {
1395
          memcpy(pStart, data, pValue->info.bytes);
2,824,542✔
1396
          pStart += pValue->info.bytes;
2,824,028✔
1397
        }
1398
      }
1399
    }
1400

1401
    int32_t len = (int32_t)(pStart - (char*)keyBuf);
22,131,919✔
1402
    info->groupId = calcGroupId(keyBuf, len);
22,131,919✔
1403
    if (groupIdMap != NULL && gInfo != NULL) {
22,138,759✔
1404
      int32_t ret = taosHashPut(groupIdMap, &info->groupId, sizeof(info->groupId), &gInfo, POINTER_BYTES);
62,935✔
1405
      if (ret != TSDB_CODE_SUCCESS) {
62,935✔
1406
        qError("put groupid to map failed at line %d since %s", __LINE__, tstrerror(ret));
×
1407
        taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1408
      }
1409
      qDebug("put groupid to map gid:%" PRIu64, info->groupId);
62,935✔
1410
      gInfo = NULL;
62,935✔
1411
    }
1412
    if (initRemainGroups) {
22,138,759✔
1413
      // groupId ~ table uid
1414
      code = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId), &(info->uid),
10,885,485✔
1415
                         sizeof(info->uid));
1416
      if (code == TSDB_CODE_DUP_KEY) {
10,885,002✔
1417
        code = TSDB_CODE_SUCCESS;
749,347✔
1418
      }
1419
      QUERY_CHECK_CODE(code, lino, end);
10,885,002✔
1420
    }
1421
  }
1422

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

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

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

1436
end:
3,948,630✔
1437
  taosMemoryFreeClear(keyBuf);
3,950,260✔
1438
  blockDataDestroy(pResBlock);
3,950,260✔
1439
  taosArrayDestroy(pBlockList);
3,949,260✔
1440
  taosArrayDestroyEx(pUidTagList, freeItem);
3,948,821✔
1441
  taosArrayDestroyP(groupData, releaseColInfoData);
3,949,153✔
1442
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
3,948,823✔
1443

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

1450
static int32_t nameComparFn(const void* p1, const void* p2) {
1,762,853✔
1451
  const char* pName1 = *(const char**)p1;
1,762,853✔
1452
  const char* pName2 = *(const char**)p2;
1,763,347✔
1453

1454
  int32_t ret = strcmp(pName1, pName2);
1,763,882✔
1455
  if (ret == 0) {
1,763,882✔
1456
    return 0;
15,744✔
1457
  } else {
1458
    return (ret > 0) ? 1 : -1;
1,748,138✔
1459
  }
1460
}
1461

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

1468
  SArray* pTbList = taosArrayInit(len, POINTER_BYTES);
913,020✔
1469
  QUERY_CHECK_NULL(pTbList, code, lino, _end, terrno);
912,526✔
1470

1471
  for (int i = 0; i < pList->pNodeList->length; i++) {
2,653,301✔
1472
    SValueNode* valueNode = (SValueNode*)cell->pNode;
1,740,281✔
1473
    if (!IS_VAR_DATA_TYPE(valueNode->node.resType.type)) {
1,739,787✔
1474
      terrno = TSDB_CODE_INVALID_PARA;
×
1475
      taosArrayDestroy(pTbList);
×
1476
      return NULL;
×
1477
    }
1478

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

1485
  size_t numOfTables = taosArrayGetSize(pTbList);
913,020✔
1486

1487
  // order the name
1488
  taosArraySort(pTbList, nameComparFn);
913,020✔
1489

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

1498
  for (int32_t i = 1; i < numOfTables; ++i) {
1,740,775✔
1499
    char** name = taosArrayGetLast(pNewList);
827,755✔
1500
    char** nameInOldList = taosArrayGet(pTbList, i);
827,755✔
1501
    QUERY_CHECK_NULL(nameInOldList, code, lino, _end, terrno);
827,755✔
1502
    if (strcmp(*name, *nameInOldList) == 0) {
827,755✔
1503
      continue;
8,492✔
1504
    }
1505

1506
    tmp = taosArrayPush(pNewList, nameInOldList);
819,263✔
1507
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
819,263✔
1508
  }
1509

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

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

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

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

1530
static int32_t filterTableInfoCompare(const void* a, const void* b) {
15,338,288✔
1531
  STUidTagInfo* p1 = (STUidTagInfo*)a;
15,338,288✔
1532
  STUidTagInfo* p2 = (STUidTagInfo*)b;
15,338,288✔
1533

1534
  if (p1->uid == p2->uid) {
15,338,288✔
1535
    return 0;
×
1536
  }
1537

1538
  return p1->uid < p2->uid ? -1 : 1;
15,328,731✔
1539
}
1540

1541
static FilterCondType checkTagCond(SNode* cond) {
13,405,534✔
1542
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
13,405,534✔
1543
    return FILTER_NO_LOGIC;
9,771,389✔
1544
  }
1545
  if (nodeType(cond) == QUERY_NODE_LOGIC_CONDITION && ((SLogicConditionNode*)cond)->condType == LOGIC_COND_TYPE_AND) {
3,636,283✔
1546
    return FILTER_AND;
3,084,424✔
1547
  }
1548
  return FILTER_OTHER;
544,027✔
1549
}
1550

1551
static bool optimizeTbnameInCond(void* pVnode, int64_t suid, SArray* list, SNode* cond, SStorageAPI* pAPI) {
14,220,823✔
1552
  int32_t code = 0;
14,220,823✔
1553
  int32_t ntype = nodeType(cond);
14,220,823✔
1554

1555
  if (ntype == QUERY_NODE_OPERATOR) {
14,223,224✔
1556
    code = optimizeTbnameInCondImpl(pVnode, list, cond, pAPI, suid);
10,580,557✔
1557
    return code == 0;
10,576,561✔
1558
  }
1559
  if (ntype != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
3,642,667✔
1560
    return false;
545,840✔
1561
  }
1562

1563
  bool                 hasTbnameCond = false;
3,095,334✔
1564
  SLogicConditionNode* pNode = (SLogicConditionNode*)cond;
3,095,334✔
1565
  SNodeList*           pList = (SNodeList*)pNode->pParameterList;
3,095,334✔
1566

1567
  int32_t len = LIST_LENGTH(pList);
3,095,998✔
1568
  if (len <= 0) {
3,094,675✔
1569
    return false;
×
1570
  }
1571

1572
  SListCell* cell = pList->pHead;
3,094,675✔
1573
  for (int i = 0; i < len; i++) {
9,660,657✔
1574
    if (cell == NULL) break;
6,563,848✔
1575
    if (optimizeTbnameInCondImpl(pVnode, list, cell->pNode, pAPI, suid) == 0) {
6,563,848✔
1576
      hasTbnameCond = true;
7,515✔
1577
      break;
7,515✔
1578
    }
1579
    cell = cell->pNext;
6,556,212✔
1580
  }
1581

1582
  taosArraySort(list, filterTableInfoCompare);
3,104,324✔
1583
  taosArrayRemoveDuplicate(list, filterTableInfoCompare, NULL);
3,096,488✔
1584

1585
  if (hasTbnameCond) {
3,095,477✔
1586
    code = pAPI->metaFn.getTableTagsByUid(pVnode, suid, list);
7,515✔
1587
    return code == 0;
7,515✔
1588
  }
1589

1590
  return false;
3,087,962✔
1591
}
1592

1593
// only return uid that does not contained in pExistedUidList
1594
static int32_t optimizeTbnameInCondImpl(void* pVnode, SArray* pExistedUidList, SNode* pTagCond, SStorageAPI* pStoreAPI,
17,148,923✔
1595
                                        uint64_t suid) {
1596
  if (nodeType(pTagCond) != QUERY_NODE_OPERATOR) {
17,148,923✔
1597
    return -1;
5,048✔
1598
  }
1599

1600
  SOperatorNode* pNode = (SOperatorNode*)pTagCond;
17,144,218✔
1601
  if (pNode->opType != OP_TYPE_IN) {
17,144,218✔
1602
    return -1;
14,662,011✔
1603
  }
1604

1605
  if ((pNode->pLeft != NULL && ((nodeType(pNode->pLeft) == QUERY_NODE_FUNCTION &&
2,482,354✔
1606
                                 ((SFunctionNode*)pNode->pLeft)->funcType == FUNCTION_TYPE_TBNAME)) ||
914,396✔
1607
       (nodeType(pNode->pLeft) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pLeft)->colType == COLUMN_TYPE_TBNAME)) &&
1,566,456✔
1608
      (pNode->pRight != NULL && nodeType(pNode->pRight) == QUERY_NODE_NODE_LIST)) {
913,559✔
1609
    SNodeListNode* pList = (SNodeListNode*)pNode->pRight;
913,020✔
1610

1611
    int32_t len = LIST_LENGTH(pList->pNodeList);
913,020✔
1612
    if (len <= 0) {
912,526✔
1613
      return -1;
×
1614
    }
1615

1616
    SArray*   pTbList = getTableNameList(pList);
912,526✔
1617
    int32_t   numOfTables = taosArrayGetSize(pTbList);
913,020✔
1618
    SHashObj* uHash = NULL;
913,020✔
1619

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

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

1642
    for (int i = 0; i < numOfTables; i++) {
2,447,297✔
1643
      char* name = taosArrayGetP(pTbList, i);
1,632,159✔
1644

1645
      uint64_t uid = 0, csuid = 0;
1,632,159✔
1646
      if (pStoreAPI->metaFn.getTableUidByName(pVnode, name, &uid) == 0) {
1,632,159✔
1647
        ETableType tbType = TSDB_TABLE_MAX;
458,724✔
1648
        if (pStoreAPI->metaFn.getTableTypeSuidByName(pVnode, name, &tbType, &csuid) == 0 &&
458,724✔
1649
            tbType == TSDB_CHILD_TABLE) {
458,724✔
1650
          if (suid != csuid) {
360,842✔
1651
            continue;
2,004✔
1652
          }
1653
          if (NULL == uHash || taosHashGet(uHash, &uid, sizeof(uid)) == NULL) {
358,838✔
1654
            STUidTagInfo s = {.uid = uid, .name = name, .pTagVal = NULL};
357,836✔
1655
            void*        tmp = taosArrayPush(pExistedUidList, &s);
357,836✔
1656
            if (!tmp) {
357,836✔
1657
              return terrno;
×
1658
            }
1659
          }
1660
        } else {
1661
          taosArrayDestroy(pTbList);
97,882✔
1662
          taosHashCleanup(uHash);
97,882✔
1663
          return -1;
97,882✔
1664
        }
1665
      } else {
1666
        //        qWarn("failed to get tableIds from by table name: %s, reason: %s", name, tstrerror(terrno));
1667
        terrno = 0;
1,173,435✔
1668
      }
1669
    }
1670

1671
    taosHashCleanup(uHash);
815,138✔
1672
    taosArrayDestroy(pTbList);
815,138✔
1673
    return 0;
815,138✔
1674
  }
1675

1676
  return -1;
1,569,334✔
1677
}
1678

1679
SSDataBlock* createTagValBlockForFilter(SArray* pColList, int32_t numOfTables, SArray* pUidTagList, void* pVnode,
17,062,000✔
1680
                                        SStorageAPI* pStorageAPI) {
1681
  int32_t      code = TSDB_CODE_SUCCESS;
17,062,000✔
1682
  int32_t      lino = 0;
17,062,000✔
1683
  SSDataBlock* pResBlock = NULL;
17,062,000✔
1684
  code = createDataBlock(&pResBlock);
17,066,111✔
1685
  QUERY_CHECK_CODE(code, lino, _end);
17,061,537✔
1686

1687
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
35,489,681✔
1688
    SColumnInfoData colInfo = {0};
18,429,233✔
1689
    void*           tmp = taosArrayGet(pColList, i);
18,428,743✔
1690
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
18,426,361✔
1691
    colInfo.info = *(SColumnInfo*)tmp;
18,426,361✔
1692
    code = blockDataAppendColInfo(pResBlock, &colInfo);
18,421,892✔
1693
    QUERY_CHECK_CODE(code, lino, _end);
18,426,506✔
1694
  }
1695

1696
  code = blockDataEnsureCapacity(pResBlock, numOfTables);
17,059,280✔
1697
  if (code != TSDB_CODE_SUCCESS) {
17,065,611✔
1698
    terrno = code;
×
1699
    blockDataDestroy(pResBlock);
×
1700
    return NULL;
×
1701
  }
1702

1703
  pResBlock->info.rows = numOfTables;
17,065,611✔
1704

1705
  int32_t numOfCols = taosArrayGetSize(pResBlock->pDataBlock);
17,065,188✔
1706

1707
  for (int32_t i = 0; i < numOfTables; i++) {
207,288,539✔
1708
    STUidTagInfo* p1 = taosArrayGet(pUidTagList, i);
190,217,105✔
1709
    QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
190,184,573✔
1710

1711
    for (int32_t j = 0; j < numOfCols; j++) {
388,853,899✔
1712
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, j);
198,561,483✔
1713
      QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
198,625,322✔
1714

1715
      if (pColInfo->info.colId == -1) {  // tbname
198,625,322✔
1716
        char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
7,306,932✔
1717
        if (p1->name != NULL) {
7,306,154✔
1718
          STR_TO_VARSTR(str, p1->name);
357,836✔
1719
        } else {  // name is not retrieved during filter
1720
          code = pStorageAPI->metaFn.getTableNameByUid(pVnode, p1->uid, str);
6,949,320✔
1721
          QUERY_CHECK_CODE(code, lino, _end);
6,922,309✔
1722
        }
1723

1724
        code = colDataSetVal(pColInfo, i, str, false);
7,279,898✔
1725
        QUERY_CHECK_CODE(code, lino, _end);
7,291,922✔
1726
#if TAG_FILTER_DEBUG
1727
        qDebug("tagfilter uid:%ld, tbname:%s", *uid, str + 2);
1728
#endif
1729
      } else {
1730
        STagVal tagVal = {0};
191,252,324✔
1731
        tagVal.cid = pColInfo->info.colId;
191,253,026✔
1732
        if (p1->pTagVal == NULL) {
191,330,759✔
1733
          colDataSetNULL(pColInfo, i);
3,300✔
1734
        } else {
1735
          const char* p = pStorageAPI->metaFn.extractTagVal(p1->pTagVal, pColInfo->info.type, &tagVal);
191,354,037✔
1736

1737
          if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
191,299,759✔
1738
            colDataSetNULL(pColInfo, i);
1,992,395✔
1739
          } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
189,324,045✔
1740
            code = colDataSetVal(pColInfo, i, p, false);
591,480✔
1741
            QUERY_CHECK_CODE(code, lino, _end);
591,480✔
1742
          } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
317,929,691✔
1743
            if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
129,258,902✔
1744
              QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
1745
            }
1746
            char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
129,227,528✔
1747
            QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
129,222,032✔
1748
            varDataSetLen(tmp, tagVal.nData);
129,222,032✔
1749
            memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
129,185,623✔
1750
            code = colDataSetVal(pColInfo, i, tmp, false);
129,197,840✔
1751
#if TAG_FILTER_DEBUG
1752
            qDebug("tagfilter varch:%s", tmp + 2);
1753
#endif
1754
            taosMemoryFree(tmp);
129,217,835✔
1755
            QUERY_CHECK_CODE(code, lino, _end);
129,168,051✔
1756
          } else {
1757
            code = colDataSetVal(pColInfo, i, (const char*)&tagVal.i64, false);
59,524,932✔
1758
            QUERY_CHECK_CODE(code, lino, _end);
59,543,465✔
1759
#if TAG_FILTER_DEBUG
1760
            if (pColInfo->info.type == TSDB_DATA_TYPE_INT) {
1761
              qDebug("tagfilter int:%d", *(int*)(&tagVal.i64));
1762
            } else if (pColInfo->info.type == TSDB_DATA_TYPE_DOUBLE) {
1763
              qDebug("tagfilter double:%f", *(double*)(&tagVal.i64));
1764
            }
1765
#endif
1766
          }
1767
        }
1768
      }
1769
    }
1770
  }
1771

1772
_end:
17,034,775✔
1773
  if (code != TSDB_CODE_SUCCESS) {
17,071,681✔
1774
    blockDataDestroy(pResBlock);
2,212✔
1775
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
247✔
1776
    terrno = code;
247✔
1777
    return NULL;
247✔
1778
  }
1779
  return pResBlock;
17,069,470✔
1780
}
1781

1782
static int32_t doSetQualifiedUid(STableListInfo* pListInfo, SArray* pUidList, const SArray* pUidTagList,
11,440,785✔
1783
                                 bool* pResultList, bool addUid) {
1784
  taosArrayClear(pUidList);
11,440,785✔
1785

1786
  STableKeyInfo info = {.uid = 0, .groupId = 0};
11,434,871✔
1787
  int32_t       numOfTables = taosArrayGetSize(pUidTagList);
11,436,366✔
1788
  for (int32_t i = 0; i < numOfTables; ++i) {
177,049,970✔
1789
    if (pResultList[i]) {
165,592,756✔
1790
      STUidTagInfo* tmpTag = (STUidTagInfo*)taosArrayGet(pUidTagList, i);
75,681,674✔
1791
      if (!tmpTag) {
75,683,391✔
1792
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1793
        return terrno;
×
1794
      }
1795
      uint64_t uid = tmpTag->uid;
75,683,391✔
1796
      qDebug("tagfilter get uid:%" PRId64 ", res:%d", uid, pResultList[i]);
75,679,918✔
1797

1798
      info.uid = uid;
75,691,615✔
1799
      //qInfo("doSetQualifiedUid row:%d added to pTableList", i);
1800
      void* p = taosArrayPush(pListInfo->pTableList, &info);
75,691,615✔
1801
      if (p == NULL) {
75,689,832✔
1802
        return terrno;
×
1803
      }
1804

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

1817
  return TSDB_CODE_SUCCESS;
11,457,214✔
1818
}
1819

1820
static int32_t copyExistedUids(SArray* pUidTagList, const SArray* pUidList) {
14,220,642✔
1821
  int32_t code = TSDB_CODE_SUCCESS;
14,220,642✔
1822
  int32_t numOfExisted = taosArrayGetSize(pUidList);
14,220,642✔
1823
  if (numOfExisted == 0) {
14,220,983✔
1824
    return code;
11,691,195✔
1825
  }
1826

1827
  for (int32_t i = 0; i < numOfExisted; ++i) {
28,768,132✔
1828
    uint64_t* uid = taosArrayGet(pUidList, i);
26,238,344✔
1829
    if (!uid) {
26,237,342✔
1830
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1831
      return terrno;
×
1832
    }
1833
    STUidTagInfo info = {.uid = *uid};
26,237,342✔
1834
    void*        tmp = taosArrayPush(pUidTagList, &info);
26,238,344✔
1835
    if (!tmp) {
26,238,344✔
1836
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1837
      return code;
×
1838
    }
1839
  }
1840
  return code;
2,529,788✔
1841
}
1842

1843
int32_t doFilterByTagCond(STableListInfo* pListInfo, SArray* pUidList, SNode* pTagCond, void* pVnode,
85,059,024✔
1844
                                 SIdxFltStatus status, SStorageAPI* pAPI, bool addUid, bool* listAdded, void* pStreamInfo) {
1845
  *listAdded = false;
85,059,024✔
1846
  if (pTagCond == NULL) {
85,063,394✔
1847
    return TSDB_CODE_SUCCESS;
70,838,990✔
1848
  }
1849

1850
  terrno = TSDB_CODE_SUCCESS;
14,224,404✔
1851

1852
  int32_t      lino = 0;
14,218,546✔
1853
  int32_t      code = TSDB_CODE_SUCCESS;
14,218,546✔
1854
  SArray*      pBlockList = NULL;
14,218,546✔
1855
  SSDataBlock* pResBlock = NULL;
14,218,546✔
1856
  SScalarParam output = {0};
14,213,134✔
1857
  SArray*      pUidTagList = NULL;
14,210,992✔
1858

1859
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
14,210,992✔
1860

1861
  //  int64_t stt = taosGetTimestampUs();
1862
  pUidTagList = taosArrayInit(10, sizeof(STUidTagInfo));
14,210,550✔
1863
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
14,213,579✔
1864

1865
  code = copyExistedUids(pUidTagList, pUidList);
14,213,579✔
1866
  QUERY_CHECK_CODE(code, lino, end);
14,212,484✔
1867

1868
  // Narrow down the scope of the tablelist set if there is tbname in condition and And Logical operator
1869
  bool narrowed = optimizeTbnameInCond(pVnode, pListInfo->idInfo.suid, pUidTagList, pTagCond, pAPI);
14,212,484✔
1870
  if (narrowed) {  // tbname in filter is activated, do nothing and return
14,216,163✔
1871
    taosArrayClear(pUidList);
815,138✔
1872

1873
    int32_t numOfRows = taosArrayGetSize(pUidTagList);
815,138✔
1874
    code = taosArrayEnsureCap(pUidList, numOfRows);
815,138✔
1875
    QUERY_CHECK_CODE(code, lino, end);
815,138✔
1876

1877
    for (int32_t i = 0; i < numOfRows; ++i) {
3,173,968✔
1878
      STUidTagInfo* pInfo = taosArrayGet(pUidTagList, i);
2,358,830✔
1879
      QUERY_CHECK_NULL(pInfo, code, lino, end, terrno);
2,358,830✔
1880
      void* tmp = taosArrayPush(pUidList, &pInfo->uid);
2,358,830✔
1881
      QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
2,358,830✔
1882
    }
1883
    terrno = 0;
815,138✔
1884
  } else {
1885
    qDebug("pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
13,401,025✔
1886

1887
    FilterCondType condType = checkTagCond(pTagCond);
13,401,428✔
1888
    if (((condType == FILTER_NO_LOGIC || condType == FILTER_AND) && status != SFLT_NOT_INDEX) || // (super table) use tagIndex and operator is and
13,398,735✔
1889
        (status == SFLT_NOT_INDEX && taosArrayGetSize(pUidTagList) > 0)) {                       // (child table with tagCond)
10,539,668✔
1890
      code = pAPI->metaFn.getTableTagsByUid(pVnode, pListInfo->idInfo.suid, pUidTagList);
3,114,516✔
1891
    } else {
1892
      taosArrayClear(pUidTagList);        // clear tablelist if using tagIndex and or condition
10,282,690✔
1893
      code = pAPI->metaFn.getTableTags(pVnode, pListInfo->idInfo.suid, pUidTagList);
10,291,837✔
1894
    }
1895
    if (code != TSDB_CODE_SUCCESS) {
13,405,871✔
1896
      qError("failed to get table tags from meta, reason:%s, suid:%" PRIu64, tstrerror(code), pListInfo->idInfo.suid);
×
1897
      terrno = code;
×
1898
      QUERY_CHECK_CODE(code, lino, end);
×
1899
    }
1900
  }
1901

1902
  qDebug("final pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
14,221,009✔
1903

1904
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
14,222,914✔
1905
  if (numOfTables == 0) {
14,222,859✔
1906
    goto end;
2,761,158✔
1907
  }
1908

1909
  SArray* pColList = NULL;
11,461,701✔
1910
  code = qGetColumnsFromNodeList(pTagCond, false, &pColList); 
11,460,886✔
1911
  if (code != TSDB_CODE_SUCCESS) {
11,442,791✔
1912
    goto end;
×
1913
  }
1914
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
11,442,791✔
1915
  taosArrayDestroy(pColList);
11,457,354✔
1916
  if (pResBlock == NULL) {
11,455,287✔
1917
    code = terrno;
×
1918
    QUERY_CHECK_CODE(code, lino, end);
×
1919
  }
1920

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

1923
  //  int64_t st1 = taosGetTimestampUs();
1924
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1925
  pBlockList = taosArrayInit(2, POINTER_BYTES);
11,455,287✔
1926
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
11,458,481✔
1927

1928
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
11,462,052✔
1929
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
11,462,052✔
1930

1931
  code = createResultData(&type, numOfTables, &output);
11,462,052✔
1932
  if (code != TSDB_CODE_SUCCESS) {
11,453,908✔
1933
    terrno = code;
×
1934
    QUERY_CHECK_CODE(code, lino, end);
×
1935
  }
1936

1937
  gTaskScalarExtra.pStreamInfo = pStreamInfo;
11,453,908✔
1938
  gTaskScalarExtra.pStreamRange = NULL;
11,453,908✔
1939
  code = scalarCalculate(pTagCond, pBlockList, &output, &gTaskScalarExtra);
11,439,982✔
1940
  if (code != TSDB_CODE_SUCCESS) {
11,441,212✔
1941
    qError("failed to calculate scalar, reason:%s", tstrerror(code));
918✔
1942
    terrno = code;
918✔
1943
    QUERY_CHECK_CODE(code, lino, end);
918✔
1944
  }
1945

1946
  code = doSetQualifiedUid(pListInfo, pUidList, pUidTagList, (bool*)output.columnData->pData, addUid);
11,440,294✔
1947
  if (code != TSDB_CODE_SUCCESS) {
11,457,878✔
1948
    terrno = code;
×
1949
    QUERY_CHECK_CODE(code, lino, end);
×
1950
  }
1951
  *listAdded = true;
11,457,878✔
1952

1953
end:
14,218,783✔
1954
  if (code != TSDB_CODE_SUCCESS) {
14,222,821✔
1955
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
918✔
1956
  }
1957
  blockDataDestroy(pResBlock);
14,222,821✔
1958
  taosArrayDestroy(pBlockList);
14,206,018✔
1959
  taosArrayDestroyEx(pUidTagList, freeItem);
14,203,966✔
1960

1961
  colDataDestroy(output.columnData);
14,216,731✔
1962
  taosMemoryFreeClear(output.columnData);
14,218,059✔
1963
  return code;
14,216,137✔
1964
}
1965

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

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

1994
  return DEAL_RES_CONTINUE;
11,631✔
1995
}
1996

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

2003
  col_id_t colId = pColNode->colId;
18,150✔
2004
  void* _tmp = taosArrayPush(pColIdArray, &colId);
18,150✔
2005
}
18,150✔
2006

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

2020
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
9,075✔
2021
    extractTagColId((SOperatorNode*)pTagCond, *pTagColIds);
×
2022
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
9,075✔
2023
    SNode* pChild = NULL;
9,075✔
2024
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
27,225✔
2025
      extractTagColId((SOperatorNode*)pChild, *pTagColIds);
18,150✔
2026
    }
2027
  }
2028

2029
  taosArraySort(*pTagColIds, compareUint16Val);
9,075✔
2030

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

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

2056
static EDealRes canOptimizeTagCondFilter(SNode* pTagCond, void* pContext) {
84,810✔
2057
  if (NULL == pTagCond) {
84,810✔
2058
    *(bool*)pContext = false;
×
2059
    return DEAL_RES_END;
×
2060
  }
2061
  if (nodeType(pTagCond) == QUERY_NODE_VALUE ||
84,810✔
2062
    nodeType(pTagCond) == QUERY_NODE_COLUMN) {
56,265✔
2063
    return DEAL_RES_CONTINUE;
46,695✔
2064
  }
2065
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR &&
38,115✔
2066
    ((SOperatorNode*)pTagCond)->opType == OP_TYPE_EQUAL) {
18,645✔
2067
    return DEAL_RES_CONTINUE;
18,150✔
2068
  }
2069
  if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION &&
19,965✔
2070
    ((SLogicConditionNode*)pTagCond)->condType == LOGIC_COND_TYPE_AND) {
9,075✔
2071
    return DEAL_RES_CONTINUE;
9,075✔
2072
  }
2073
  if (nodeType(pTagCond) == QUERY_NODE_FUNCTION &&
21,285✔
2074
    fmIsStreamPesudoColVal(((SFunctionNode*)pTagCond)->funcId)) {
10,395✔
2075
    return DEAL_RES_CONTINUE;
10,395✔
2076
  }
2077
  *(bool*)pContext = false;
495✔
2078
  return DEAL_RES_END;
495✔
2079
}
2080

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

2088
  pListInfo->idInfo.suid = pScanNode->suid;
187,239,124✔
2089
  pListInfo->idInfo.tableType = pScanNode->tableType;
187,230,871✔
2090

2091
  SArray* pUidList = taosArrayInit(8, sizeof(uint64_t));
187,087,791✔
2092
  QUERY_CHECK_NULL(pUidList, code, lino, _error, terrno);
187,123,517✔
2093

2094
  char*   pTagCondKey = NULL;
187,123,517✔
2095
  int32_t tagCondKeyLen;
187,064,146✔
2096
  SArray* pTagColIds = NULL;
187,132,614✔
2097
  char*   pPayload = NULL;
187,153,996✔
2098
  qTrace("getTableList called, suid:%" PRIu64
187,153,996✔
2099
    ", tagCond:%p, tagIndexCond:%p, %d %d", pScanNode->suid, pTagCond,
2100
    pTagIndexCond, pScanNode->tableType, pScanNode->virtualStableScan);
2101
  if (pScanNode->tableType != TSDB_SUPER_TABLE && !pScanNode->virtualStableScan) {
187,153,996✔
2102
    pListInfo->idInfo.uid = pScanNode->uid;
71,054,464✔
2103
    if (pStorageAPI->metaFn.isTableExisted(pVnode, pScanNode->uid)) {
71,024,504✔
2104
      void* tmp = taosArrayPush(pUidList, &pScanNode->uid);
71,021,177✔
2105
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
71,020,572✔
2106
    }
2107
    code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, SFLT_NOT_INDEX, pStorageAPI, false, &listAdded, pStreamInfo);
71,067,205✔
2108
    QUERY_CHECK_CODE(code, lino, _end);
71,062,565✔
2109
  } else {
2110
    bool      isStream = (pStreamInfo != NULL);
116,234,355✔
2111
    bool      hasTagCond = (pTagCond != NULL);
116,234,355✔
2112
    bool      canCacheTagEqCondFilter = false;
116,234,355✔
2113
    T_MD5_CTX context = {0};
116,174,215✔
2114

2115
    qTrace("start to get table list by tag filter, suid:%" PRIu64
116,245,570✔
2116
      ",tsStableTagFilterCache:%d, tsTagFilterCache:%d", 
2117
      pScanNode->suid, tsStableTagFilterCache, tsTagFilterCache);
2118

2119
    bool acquired = false;
116,245,570✔
2120
    // first, check whether we can use stable tag filter cache
2121
    if (tsStableTagFilterCache && isStream && hasTagCond) {
116,170,168✔
2122
      canCacheTagEqCondFilter = true;
9,570✔
2123
      nodesWalkExpr(pTagCond, canOptimizeTagCondFilter,
9,570✔
2124
        (void*)&canCacheTagEqCondFilter);
2125
    }
2126
    if (canCacheTagEqCondFilter) {
116,088,682✔
2127
      qDebug("%s, stable tag filter condition can be optimized", idstr);
9,075✔
2128
      if (((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
9,075✔
2129
        SNode* tmp = NULL;
9,075✔
2130
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
9,075✔
2131
        QUERY_CHECK_CODE(code, lino, _error);
9,075✔
2132

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

2147
      code = buildTagCondKey(
9,075✔
2148
        pTagCond, &pTagCondKey, &tagCondKeyLen, &pTagColIds);
2149
      QUERY_CHECK_CODE(code, lino, _error);
9,075✔
2150
      code = pStorageAPI->metaFn.getStableCachedTableList(
9,075✔
2151
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
9,075✔
2152
        context.digest, tListLen(context.digest), pUidList, &acquired);
2153
      QUERY_CHECK_CODE(code, lino, _error);
9,075✔
2154
    } else if (tsTagFilterCache) {
116,079,607✔
2155
      // second, try to use normal tag filter cache
2156
      qDebug("%s using normal tag filter cache", idstr);
30,845✔
2157
      if (pStreamInfo != NULL && ((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
32,081✔
2158
        SNode* tmp = NULL;
1,236✔
2159
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
1,236✔
2160
        QUERY_CHECK_CODE(code, lino, _error);
1,236✔
2161

2162
        PlaceHolderContext ctx = {.code = TSDB_CODE_SUCCESS, .pStreamRuntimeInfo = (SStreamRuntimeFuncInfo*)pStreamInfo};
1,236✔
2163
        nodesRewriteExpr(&tmp, replacePlaceHolderColumn, (void*)&ctx);
1,236✔
2164
        if (TSDB_CODE_SUCCESS != ctx.code) {
1,236✔
2165
          nodesDestroyNode(tmp);
×
2166
          code = ctx.code;
×
2167
          goto _error;
×
2168
        }
2169
        code = genTagFilterDigest(tmp, &context);
1,236✔
2170
        nodesDestroyNode(tmp);
1,236✔
2171
      } else {
2172
        code = genTagFilterDigest(pTagCond, &context);
29,609✔
2173
      }
2174
      // try to retrieve the result from meta cache
2175
      QUERY_CHECK_CODE(code, lino, _error);      
30,845✔
2176
      code = pStorageAPI->metaFn.getCachedTableList(
30,845✔
2177
        pVnode, pScanNode->suid, context.digest,
30,845✔
2178
        tListLen(context.digest), pUidList, &acquired);
2179
      QUERY_CHECK_CODE(code, lino, _error);
80,482✔
2180
    }
2181
    if (acquired) {
116,138,319✔
2182
      taosArrayDestroy(pTagColIds);
27,632✔
2183
      pTagColIds = NULL;
27,632✔
2184
      
2185
      digest[0] = 1;
27,632✔
2186
      memcpy(
55,264✔
2187
        digest + 1, context.digest, tListLen(context.digest));
27,632✔
2188
      qDebug("suid:%" PRIu64 ", %s retrieve table uid list from cache,"
27,632✔
2189
        " numOfTables:%d", 
2190
        pScanNode->suid, idstr, (int32_t)taosArrayGetSize(pUidList));
2191
      goto _end;
27,632✔
2192
    } else {
2193
      qDebug("suid:%" PRIu64 
116,110,687✔
2194
        ", failed to get table uid list from cache", pScanNode->suid);
2195
    }
2196

2197
    if (!pTagCond) {  // no tag filter condition exists, let's fetch all tables of this super table
116,262,811✔
2198
      code = pStorageAPI->metaFn.getChildTableList(pVnode, pScanNode->suid, pUidList);
102,300,904✔
2199
      QUERY_CHECK_CODE(code, lino, _error);
102,271,808✔
2200
      qTrace("no tag filter, get all child tables, numOfTables:%d", (int32_t)taosArrayGetSize(pUidList));
102,271,808✔
2201
    } else {
2202
      SIdxFltStatus status = SFLT_NOT_INDEX;
13,961,907✔
2203
      if (pTagIndexCond) {
13,962,534✔
2204
        void* pIndex = pStorageAPI->metaFn.getInvertIndex(pVnode);
3,917,358✔
2205

2206
        SIndexMetaArg metaArg = {.metaEx = pVnode,
3,917,031✔
2207
                                 .idx = pStorageAPI->metaFn.storeGetIndexInfo(pVnode),
3,917,005✔
2208
                                 .ivtIdx = pIndex,
2209
                                 .suid = pScanNode->uid};
3,917,358✔
2210
        code = doFilterTag(pTagIndexCond, &metaArg, pUidList, &status, &pStorageAPI->metaFilter);
3,917,358✔
2211
        if (code != 0 || status == SFLT_NOT_INDEX) {  // temporarily disable it for performance sake
3,907,192✔
2212
          qDebug("failed to get tableIds from index, suid:%" PRIu64 ", uidListSize:%d", pScanNode->uid, (int32_t)taosArrayGetSize(pUidList));
1,046,734✔
2213
        } else {
2214
          qDebug("succ to get filter result, table num: %d", (int)taosArrayGetSize(pUidList));
2,860,458✔
2215
        }
2216
      }
2217
      qTrace("after index filter, pTagCond:%p uidListSize:%d", pTagCond, (int32_t)taosArrayGetSize(pUidList));
13,962,534✔
2218
      code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, status,
13,964,492✔
2219
        pStorageAPI, tsTagFilterCache || tsStableTagFilterCache,
13,964,492✔
2220
        &listAdded, pStreamInfo);
2221
      QUERY_CHECK_CODE(code, lino, _error);
13,954,192✔
2222
    }
2223
    // let's add the filter results into meta-cache
2224
    numOfTables = taosArrayGetSize(pUidList);
116,224,192✔
2225

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

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

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

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

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

2277

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

2283
_end:
187,273,629✔
2284
  if (!listAdded) {
187,326,614✔
2285
    numOfTables = taosArrayGetSize(pUidList);
175,845,807✔
2286
    for (int i = 0; i < numOfTables; i++) {
678,510,739✔
2287
      void* tmp = taosArrayGet(pUidList, i);
502,624,892✔
2288
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
502,711,504✔
2289
      STableKeyInfo info = {.uid = *(uint64_t*)tmp, .groupId = 0};
502,711,504✔
2290

2291
      void* p = taosArrayPush(pListInfo->pTableList, &info);
502,723,579✔
2292
      if (p == NULL) {
502,724,995✔
2293
        taosArrayDestroy(pUidList);
×
2294
        return terrno;
×
2295
      }
2296

2297
      qTrace("tagfilter get uid:%" PRIu64 ", %s", info.uid, idstr);
502,724,995✔
2298
    }
2299
  }
2300

2301
  qDebug("%s, table list with %d uids built", idstr, (int32_t)numOfTables);
187,366,654✔
2302

2303
_error:
187,360,876✔
2304
  taosArrayDestroy(pUidList);
187,384,684✔
2305
  taosArrayDestroy(pTagColIds);
187,322,658✔
2306
  taosMemFreeClear(pTagCondKey);
187,328,006✔
2307
  taosMemFreeClear(pPayload);
187,328,006✔
2308
  if (code != TSDB_CODE_SUCCESS) {
187,328,006✔
2309
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
918✔
2310
  }
2311
  return code;
187,335,857✔
2312
}
2313

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

2436
  return TSDB_CODE_SUCCESS;
×
2437
}
2438

2439
SArray* makeColumnArrayFromList(SNodeList* pNodeList) {
3,326,825✔
2440
  if (!pNodeList) {
3,326,825✔
2441
    return NULL;
×
2442
  }
2443

2444
  size_t  numOfCols = LIST_LENGTH(pNodeList);
3,326,825✔
2445
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
3,325,399✔
2446
  if (pList == NULL) {
3,325,045✔
2447
    return NULL;
×
2448
  }
2449

2450
  for (int32_t i = 0; i < numOfCols; ++i) {
7,542,233✔
2451
    SColumnNode* pColNode = (SColumnNode*)nodesListGetNode(pNodeList, i);
4,216,820✔
2452
    if (!pColNode) {
4,218,246✔
2453
      taosArrayDestroy(pList);
×
2454
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2455
      return NULL;
×
2456
    }
2457

2458
    // todo extract method
2459
    SColumn c = {0};
4,218,246✔
2460
    c.slotId = pColNode->slotId;
4,218,246✔
2461
    c.colId = pColNode->colId;
4,217,892✔
2462
    c.type = pColNode->node.resType.type;
4,218,260✔
2463
    c.bytes = pColNode->node.resType.bytes;
4,218,000✔
2464
    c.precision = pColNode->node.resType.precision;
4,217,103✔
2465
    c.scale = pColNode->node.resType.scale;
4,218,246✔
2466

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

2475
  return pList;
3,325,413✔
2476
}
2477

2478
int32_t extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols,
253,736,884✔
2479
                            int32_t type, SColMatchInfo* pMatchInfo) {
2480
  size_t  numOfCols = LIST_LENGTH(pNodeList);
253,736,884✔
2481
  int32_t code = TSDB_CODE_SUCCESS;
253,730,342✔
2482
  int32_t lino = 0;
253,730,342✔
2483

2484
  pMatchInfo->matchType = type;
253,730,342✔
2485

2486
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchItem));
253,754,334✔
2487
  if (pList == NULL) {
253,678,466✔
2488
    code = terrno;
×
2489
    return code;
×
2490
  }
2491

2492
  for (int32_t i = 0; i < numOfCols; ++i) {
1,345,414,585✔
2493
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
1,091,638,570✔
2494
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,091,743,710✔
2495
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
1,091,743,710✔
2496
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
1,079,968,264✔
2497

2498
      SColMatchItem c = {.needOutput = true};
1,080,006,055✔
2499
      c.colId = pColNode->colId;
1,079,992,360✔
2500
      c.srcSlotId = pColNode->slotId;
1,079,969,788✔
2501
      c.dstSlotId = pNode->slotId;
1,079,947,380✔
2502
      c.isPk = pColNode->isPk;
1,080,021,282✔
2503
      c.dataType = pColNode->node.resType;
1,079,971,610✔
2504
      void* tmp = taosArrayPush(pList, &c);
1,080,000,209✔
2505
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,080,000,209✔
2506
    }
2507
  }
2508

2509
  // set the output flag for each column in SColMatchInfo, according to the
2510
  *numOfOutputCols = 0;
253,776,015✔
2511
  int32_t num = LIST_LENGTH(pOutputNodeList->pSlots);
253,811,636✔
2512
  for (int32_t i = 0; i < num; ++i) {
1,515,547,171✔
2513
    SSlotDescNode* pNode = (SSlotDescNode*)nodesListGetNode(pOutputNodeList->pSlots, i);
1,261,779,800✔
2514
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,261,897,772✔
2515

2516
    // todo: add reserve flag check
2517
    // it is a column reserved for the arithmetic expression calculation
2518
    if (pNode->slotId >= numOfCols) {
1,261,897,772✔
2519
      (*numOfOutputCols) += 1;
170,230,060✔
2520
      continue;
170,232,966✔
2521
    }
2522

2523
    SColMatchItem* info = NULL;
1,091,731,043✔
2524
    for (int32_t j = 0; j < taosArrayGetSize(pList); ++j) {
2,147,483,647✔
2525
      info = taosArrayGet(pList, j);
2,147,483,647✔
2526
      QUERY_CHECK_NULL(info, code, lino, _end, terrno);
2,147,483,647✔
2527
      if (info->dstSlotId == pNode->slotId) {
2,147,483,647✔
2528
        break;
1,079,111,704✔
2529
      }
2530
    }
2531

2532
    if (pNode->output) {
14,811,992✔
2533
      (*numOfOutputCols) += 1;
1,082,196,792✔
2534
    } else if (info != NULL) {
9,393,857✔
2535
      // select distinct tbname from stb where tbname='abc';
2536
      info->needOutput = false;
9,452,491✔
2537
    }
2538
  }
2539

2540
  pMatchInfo->pList = pList;
253,767,371✔
2541

2542
_end:
253,823,182✔
2543
  if (code != TSDB_CODE_SUCCESS) {
253,823,182✔
2544
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2545
  }
2546
  return code;
253,748,530✔
2547
}
2548

2549
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
996,399,888✔
2550
                                  const char* name) {
2551
  SResSchema s = {0};
996,399,888✔
2552
  s.scale = scale;
996,480,521✔
2553
  s.type = type;
996,480,521✔
2554
  s.bytes = bytes;
996,480,521✔
2555
  s.slotId = slotId;
996,480,521✔
2556
  s.precision = precision;
996,480,521✔
2557
  tstrncpy(s.name, name, tListLen(s.name));
996,480,521✔
2558

2559
  return s;
996,480,521✔
2560
}
2561

2562
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType, EColumnType colType) {
956,601,990✔
2563
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
956,601,990✔
2564
  if (pCol == NULL) {
955,982,313✔
2565
    return NULL;
×
2566
  }
2567

2568
  pCol->slotId = slotId;
955,982,313✔
2569
  pCol->colId = colId;
956,018,060✔
2570
  pCol->bytes = pType->bytes;
956,113,214✔
2571
  pCol->type = pType->type;
956,152,893✔
2572
  pCol->scale = pType->scale;
956,400,093✔
2573
  pCol->precision = pType->precision;
956,391,552✔
2574
  pCol->dataBlockId = blockId;
956,628,053✔
2575
  pCol->colType = colType;
956,536,692✔
2576
  return pCol;
956,584,338✔
2577
}
2578

2579
int32_t createExprFromOneNode(SExprInfo* pExp, SNode* pNode, int16_t slotId) {
996,928,333✔
2580
  int32_t code = TSDB_CODE_SUCCESS;
996,928,333✔
2581
  int32_t lino = 0;
996,928,333✔
2582
  pExp->base.numOfParams = 0;
996,928,333✔
2583
  pExp->base.pParam = NULL;
996,993,430✔
2584
  pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
996,941,832✔
2585
  QUERY_CHECK_NULL(pExp->pExpr, code, lino, _end, terrno);
996,167,680✔
2586

2587
  pExp->pExpr->_function.num = 1;
996,249,419✔
2588
  pExp->pExpr->_function.functionId = -1;
996,411,369✔
2589

2590
  int32_t type = nodeType(pNode);
996,478,592✔
2591
  // it is a project query, or group by column
2592
  if (type == QUERY_NODE_COLUMN) {
996,772,750✔
2593
    pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
565,085,788✔
2594
    SColumnNode* pColNode = (SColumnNode*)pNode;
565,120,907✔
2595

2596
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
565,120,907✔
2597
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
565,026,860✔
2598

2599
    pExp->base.numOfParams = 1;
565,054,057✔
2600

2601
    SDataType* pType = &pColNode->node.resType;
565,038,495✔
2602
    pExp->base.resSchema =
2603
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pColNode->colName);
565,083,157✔
2604

2605
    pExp->base.pParam[0].pCol =
1,130,086,482✔
2606
        createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType, pColNode->colType);
1,130,202,444✔
2607
    QUERY_CHECK_NULL(pExp->base.pParam[0].pCol, code, lino, _end, terrno);
565,056,551✔
2608

2609
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
564,900,490✔
2610
  } else if (type == QUERY_NODE_VALUE) {
431,686,962✔
2611
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
16,298,294✔
2612
    SValueNode* pValNode = (SValueNode*)pNode;
16,300,994✔
2613

2614
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
16,300,994✔
2615
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
16,294,838✔
2616

2617
    pExp->base.numOfParams = 1;
16,295,430✔
2618

2619
    SDataType* pType = &pValNode->node.resType;
16,295,709✔
2620
    pExp->base.resSchema =
2621
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
16,291,563✔
2622
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
16,293,500✔
2623
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
16,292,017✔
2624
    QUERY_CHECK_CODE(code, lino, _end);
16,292,138✔
2625
  } else if (type == QUERY_NODE_REMOTE_VALUE) {
415,388,668✔
2626
    SRemoteValueNode* pRemote = (SRemoteValueNode*)pNode;
1,373,899✔
2627
    code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, pNode);
1,373,899✔
2628
    QUERY_CHECK_CODE(code, lino, _end);
1,375,486✔
2629

2630
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
968,156✔
2631
    SValueNode* pValNode = (SValueNode*)pNode;
968,156✔
2632

2633
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
968,156✔
2634
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
968,156✔
2635

2636
    pExp->base.numOfParams = 1;
968,156✔
2637

2638
    SDataType* pType = &pValNode->node.resType;
968,156✔
2639
    pExp->base.resSchema =
2640
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
968,156✔
2641
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
968,156✔
2642
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
968,156✔
2643
    QUERY_CHECK_CODE(code, lino, _end);
968,156✔
2644
  } else if (type == QUERY_NODE_FUNCTION) {
414,014,769✔
2645
    pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
391,734,148✔
2646
    SFunctionNode* pFuncNode = (SFunctionNode*)pNode;
391,741,330✔
2647

2648
    SDataType* pType = &pFuncNode->node.resType;
391,741,330✔
2649
    pExp->base.resSchema =
2650
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pFuncNode->node.aliasName);
391,708,452✔
2651
    tExprNode* pExprNode = pExp->pExpr;
391,730,675✔
2652

2653
    pExprNode->_function.functionId = pFuncNode->funcId;
391,730,316✔
2654
    pExprNode->_function.pFunctNode = pFuncNode;
391,736,432✔
2655
    pExprNode->_function.functionType = pFuncNode->funcType;
391,800,933✔
2656

2657
    tstrncpy(pExprNode->_function.functionName, pFuncNode->functionName, tListLen(pExprNode->_function.functionName));
391,713,747✔
2658

2659
    pExp->base.pParamList = pFuncNode->pParameterList;
391,768,539✔
2660
#if 1
2661
    // todo refactor: add the parameter for tbname function
2662
    const char* name = "tbname";
391,806,600✔
2663
    int32_t     len = strlen(name);
391,806,600✔
2664

2665
    if (!pFuncNode->pParameterList && (memcmp(pExprNode->_function.functionName, name, len) == 0) &&
391,806,600✔
2666
        pExprNode->_function.functionName[len] == 0) {
28,496,278✔
2667
      pFuncNode->pParameterList = NULL;
28,485,616✔
2668
      int32_t     code = nodesMakeList(&pFuncNode->pParameterList);
28,498,315✔
2669
      SValueNode* res = NULL;
28,500,153✔
2670
      if (TSDB_CODE_SUCCESS == code) {
28,501,657✔
2671
        code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
28,499,658✔
2672
      }
2673
      QUERY_CHECK_CODE(code, lino, _end);
28,518,757✔
2674
      res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
28,518,757✔
2675
      code = nodesListAppend(pFuncNode->pParameterList, (SNode*)res);
28,499,860✔
2676
      if (code != TSDB_CODE_SUCCESS) {
28,505,873✔
2677
        nodesDestroyNode((SNode*)res);
×
2678
        res = NULL;
×
2679
      }
2680
      QUERY_CHECK_CODE(code, lino, _end);
28,505,873✔
2681
    }
2682
#endif
2683

2684
    int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
391,815,818✔
2685

2686
    pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
391,837,585✔
2687
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
391,620,299✔
2688
    pExp->base.numOfParams = numOfParam;
391,611,935✔
2689

2690
    for (int32_t j = 0; j < numOfParam && TSDB_CODE_SUCCESS == code; ++j) {
942,094,085✔
2691
      SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
550,547,242✔
2692
      QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
550,557,302✔
2693
      if (p1->type == QUERY_NODE_COLUMN) {
550,557,302✔
2694
        SColumnNode* pcn = (SColumnNode*)p1;
391,454,678✔
2695

2696
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
391,454,678✔
2697
        pExp->base.pParam[j].pCol =
782,814,026✔
2698
            createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType, pcn->colType);
782,960,725✔
2699
        QUERY_CHECK_NULL(pExp->base.pParam[j].pCol, code, lino, _end, terrno);
391,398,665✔
2700
      } else if (p1->type == QUERY_NODE_VALUE) {
159,093,424✔
2701
        SValueNode* pvn = (SValueNode*)p1;
105,487,797✔
2702
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
105,487,797✔
2703
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
105,465,815✔
2704
        QUERY_CHECK_CODE(code, lino, _end);
105,435,724✔
2705
      } else if (p1->type == QUERY_NODE_REMOTE_VALUE) {
53,707,454✔
2706
        SRemoteValueNode* pRemote = (SRemoteValueNode*)p1;
70,886✔
2707
        code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, p1);
70,886✔
2708
        QUERY_CHECK_CODE(code, lino, _end);
70,886✔
2709

2710
        SValueNode* pvn = (SValueNode*)pRemote;
51,313✔
2711
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
51,313✔
2712
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
51,313✔
2713
        QUERY_CHECK_CODE(code, lino, _end);
779✔
2714
      }
2715
    }
2716
    pExp->pExpr->_function.bindExprID = ((SExprNode*)pNode)->bindExprID;
391,546,843✔
2717
  } else if (type == QUERY_NODE_OPERATOR) {
22,280,621✔
2718
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
21,073,446✔
2719
    SOperatorNode* pOpNode = (SOperatorNode*)pNode;
21,039,338✔
2720

2721
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
21,039,338✔
2722
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
21,035,335✔
2723
    pExp->base.numOfParams = 1;
21,038,997✔
2724

2725
    SDataType* pType = &pOpNode->node.resType;
21,052,822✔
2726
    pExp->base.resSchema =
2727
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pOpNode->node.aliasName);
21,017,402✔
2728
    pExp->pExpr->_optrRoot.pRootNode = pNode;
21,047,464✔
2729
  } else if (type == QUERY_NODE_CASE_WHEN) {
1,208,034✔
2730
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
1,207,568✔
2731
    SCaseWhenNode* pCaseNode = (SCaseWhenNode*)pNode;
1,207,568✔
2732

2733
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
1,207,568✔
2734
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
1,207,135✔
2735
    pExp->base.numOfParams = 1;
1,207,033✔
2736

2737
    SDataType* pType = &pCaseNode->node.resType;
1,207,568✔
2738
    pExp->base.resSchema =
2739
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCaseNode->node.aliasName);
1,207,568✔
2740
    pExp->pExpr->_optrRoot.pRootNode = pNode;
1,207,568✔
2741
  } else if (type == QUERY_NODE_LOGIC_CONDITION) {
1,140✔
2742
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
3,034✔
2743
    SLogicConditionNode* pCond = (SLogicConditionNode*)pNode;
3,034✔
2744
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
3,034✔
2745
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
3,034✔
2746
    pExp->base.numOfParams = 1;
3,034✔
2747
    SDataType* pType = &pCond->node.resType;
3,034✔
2748
    pExp->base.resSchema =
2749
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCond->node.aliasName);
3,034✔
2750
    pExp->pExpr->_optrRoot.pRootNode = pNode;
3,034✔
2751
  } else {
UNCOV
2752
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
UNCOV
2753
    QUERY_CHECK_CODE(code, lino, _end);
×
2754
  }
2755
  pExp->pExpr->relatedTo = ((SExprNode*)pNode)->relatedTo;
996,261,426✔
2756
_end:
996,678,368✔
2757
  if (code != TSDB_CODE_SUCCESS) {
996,678,368✔
2758
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
426,903✔
2759
  }
2760
  return code;
996,748,590✔
2761
}
2762

2763
int32_t createExprFromTargetNode(SExprInfo* pExp, STargetNode* pTargetNode) {
996,651,231✔
2764
  return createExprFromOneNode(pExp, pTargetNode->pExpr, pTargetNode->slotId);
996,651,231✔
2765
}
2766

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

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

2785
  return pExprs;
×
2786
}
2787

2788
int32_t createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, SExprInfo** pExprInfo, int32_t* numOfExprs) {
348,901,866✔
2789
  QRY_PARAM_CHECK(pExprInfo);
348,901,866✔
2790

2791
  int32_t code = 0;
348,932,507✔
2792
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
348,932,507✔
2793
  int32_t numOfGroupKeys = 0;
348,893,685✔
2794
  if (pGroupKeys != NULL) {
348,893,685✔
2795
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
29,603,589✔
2796
  }
2797

2798
  *numOfExprs = numOfFuncs + numOfGroupKeys;
348,897,291✔
2799
  if (*numOfExprs == 0) {
348,996,942✔
2800
    return code;
36,261,989✔
2801
  }
2802

2803
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
312,795,768✔
2804
  if (pExprs == NULL) {
312,389,073✔
2805
    return terrno;
×
2806
  }
2807

2808
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
1,308,449,837✔
2809
    STargetNode* pTargetNode = NULL;
996,296,619✔
2810
    if (i < numOfFuncs) {
996,296,619✔
2811
      pTargetNode = (STargetNode*)nodesListGetNode(pNodeList, i);
948,678,965✔
2812
    } else {
2813
      pTargetNode = (STargetNode*)nodesListGetNode(pGroupKeys, i - numOfFuncs);
47,617,654✔
2814
    }
2815
    if (!pTargetNode) {
996,653,199✔
2816
      destroyExprInfo(pExprs, *numOfExprs);
×
2817
      taosMemoryFreeClear(pExprs);
×
2818
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2819
      return terrno;
×
2820
    }
2821

2822
    SExprInfo* pExp = &pExprs[i];
996,653,199✔
2823
    code = createExprFromTargetNode(pExp, pTargetNode);
996,693,938✔
2824
    if (code != TSDB_CODE_SUCCESS) {
996,487,667✔
2825
      destroyExprInfo(pExprs, *numOfExprs);
426,903✔
2826
      taosMemoryFreeClear(pExprs);
426,903✔
2827
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
426,903✔
2828
      return code;
426,903✔
2829
    }
2830
  }
2831

2832
  *pExprInfo = pExprs;
312,255,316✔
2833
  return code;
312,254,497✔
2834
}
2835

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

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

2849
  SArray* pValCtxArray = NULL;
338,416,741✔
2850
  for (int32_t i = numOfOutput - 1; i > 0; --i) {  // select Func is at the end of the list
1,013,625,764✔
2851
    int32_t funcIdx = pCtx[i].pExpr->pExpr->_function.bindExprID;
675,334,002✔
2852
    if (funcIdx > 0) {
675,414,119✔
2853
      if (pValCtxArray == NULL) {
1,476,401✔
2854
        // the end of the list is the select function of biggest index
2855
        pValCtxArray = taosArrayInit_s(sizeof(SSubsidiaryResInfo*), funcIdx);
1,061,546✔
2856
        if (pValCtxArray == NULL) {
1,060,337✔
2857
          return terrno;
×
2858
        }
2859
      }
2860
      if (funcIdx > pValCtxArray->size) {
1,475,192✔
2861
        qError("funcIdx:%d is out of range", funcIdx);
×
2862
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2863
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2864
      }
2865
      SSubsidiaryResInfo* pSubsidiary = &pCtx[i].subsidiaries;
1,475,998✔
2866
      pSubsidiary->pCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
1,477,610✔
2867
      if (pSubsidiary->pCtx == NULL) {
1,475,998✔
2868
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2869
        return terrno;
×
2870
      }
2871
      pSubsidiary->num = 0;
1,475,595✔
2872
      taosArraySet(pValCtxArray, funcIdx - 1, &pSubsidiary);
1,474,789✔
2873
    }
2874
  }
2875

2876
  SqlFunctionCtx*  p = NULL;
338,291,762✔
2877
  SqlFunctionCtx** pValCtx = NULL;
338,291,762✔
2878
  if (pValCtxArray == NULL) {
338,291,762✔
2879
    pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
337,302,247✔
2880
    if (pValCtx == NULL) {
337,280,912✔
2881
      QUERY_CHECK_CODE(terrno, lino, _end);
×
2882
    }
2883
  }
2884

2885
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,321,642,952✔
2886
    const char* pName = pCtx[i].pExpr->pExpr->_function.functionName;
983,280,465✔
2887
    if ((strcmp(pName, "_select_value") == 0)) {
983,484,722✔
2888
      if (pValCtxArray == NULL) {
8,002,518✔
2889
        pValCtx[num++] = &pCtx[i];
5,929,930✔
2890
      } else {
2891
        int32_t bindFuncIndex = pCtx[i].pExpr->pExpr->relatedTo;  // start from index 1;
2,072,684✔
2892
        if (bindFuncIndex > 0) {                                  // 0 is default index related to the select function
2,073,135✔
2893
          bindFuncIndex -= 1;
2,024,372✔
2894
        }
2895
        SSubsidiaryResInfo** pSubsidiary = taosArrayGet(pValCtxArray, bindFuncIndex);
2,073,135✔
2896
        if (pSubsidiary == NULL) {
2,073,135✔
2897
          QUERY_CHECK_CODE(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR, lino, _end);
×
2898
        }
2899
        (*pSubsidiary)->pCtx[(*pSubsidiary)->num] = &pCtx[i];
2,073,135✔
2900
        (*pSubsidiary)->num++;
2,071,120✔
2901
      }
2902
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
975,482,204✔
2903
      if (pValCtxArray == NULL) {
94,399,742✔
2904
        p = &pCtx[i];
92,599,401✔
2905
      }
2906
    }
2907
  }
2908

2909
  if (p != NULL) {
338,362,487✔
2910
    p->subsidiaries.pCtx = pValCtx;
28,482,384✔
2911
    p->subsidiaries.num = num;
28,481,979✔
2912
  } else {
2913
    taosMemoryFreeClear(pValCtx);
309,880,103✔
2914
  }
2915

2916
_end:
1,074,194✔
2917
  if (code != TSDB_CODE_SUCCESS) {
338,344,054✔
2918
    taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2919
    taosMemoryFreeClear(pValCtx);
×
2920
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2921
  } else {
2922
    taosArrayDestroy(pValCtxArray);
338,344,054✔
2923
  }
2924
  return code;
338,388,833✔
2925
}
2926

2927
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowEntryInfoOffset,
338,458,805✔
2928
                                     SFunctionStateStore* pStore) {
2929
  int32_t         code = TSDB_CODE_SUCCESS;
338,458,805✔
2930
  int32_t         lino = 0;
338,458,805✔
2931
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
338,458,805✔
2932
  if (pFuncCtx == NULL) {
338,094,102✔
2933
    return NULL;
×
2934
  }
2935

2936
  *rowEntryInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
338,094,102✔
2937
  if (*rowEntryInfoOffset == 0) {
338,432,600✔
2938
    taosMemoryFreeClear(pFuncCtx);
×
2939
    return NULL;
×
2940
  }
2941

2942
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,321,835,056✔
2943
    SExprInfo* pExpr = &pExprInfo[i];
983,479,019✔
2944

2945
    SExprBasicInfo* pFunct = &pExpr->base;
983,397,361✔
2946
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
983,469,641✔
2947

2948
    pCtx->functionId = -1;
983,487,266✔
2949
    pCtx->pExpr = pExpr;
983,522,183✔
2950

2951
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
983,475,721✔
2952
      SFuncExecEnv env = {0};
390,764,189✔
2953
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;
390,787,029✔
2954
      pCtx->isPseudoFunc = fmIsWindowPseudoColumnFunc(pCtx->functionId) || fmIsPlaceHolderFunc(pCtx->functionId);
390,781,597✔
2955
      pCtx->isNotNullFunc = fmIsNotNullOutputFunc(pCtx->functionId);
390,741,491✔
2956

2957
      bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
390,728,884✔
2958
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId)) {
643,533,905✔
2959
        if (!isUdaf) {
253,092,175✔
2960
          code = fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
252,992,945✔
2961
          QUERY_CHECK_CODE(code, lino, _end);
252,839,995✔
2962
        } else {
2963
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
99,230✔
2964
          pCtx->udfName = taosStrdup(udfName);
99,230✔
2965
          QUERY_CHECK_NULL(pCtx->udfName, code, lino, _end, terrno);
99,230✔
2966

2967
          code = fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
99,230✔
2968
          QUERY_CHECK_CODE(code, lino, _end);
99,230✔
2969
        }
2970
        bool tmp = pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
252,939,225✔
2971
        if (!tmp) {
252,883,824✔
2972
          code = terrno;
×
2973
          QUERY_CHECK_CODE(code, lino, _end);
9,604✔
2974
        }
2975
      } else {
2976
        if (fmIsPlaceHolderFunc(pCtx->functionId)) {
137,600,014✔
2977
          code = fmGetStreamPesudoFuncEnv(pCtx->functionId, pExpr->base.pParamList, &env);
5,853,204✔
2978
          QUERY_CHECK_CODE(code, lino, _end);
5,853,417✔
2979
        }      
2980
        
2981
        code = fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
137,636,229✔
2982
        if (code != TSDB_CODE_SUCCESS && isUdaf) {
137,626,449✔
2983
          code = TSDB_CODE_SUCCESS;
114,582✔
2984
        }
2985
        QUERY_CHECK_CODE(code, lino, _end);
137,626,449✔
2986

2987
        if (pCtx->sfp.getEnv != NULL) {
137,626,449✔
2988
          bool tmp = pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
24,033,127✔
2989
          if (!tmp) {
24,035,842✔
2990
            code = terrno;
×
2991
            QUERY_CHECK_CODE(code, lino, _end);
×
2992
          }
2993
        }
2994
      }
2995
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
390,550,387✔
2996
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
592,661,647✔
2997
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
17,249,213✔
2998
      // for simple column, the result buffer needs to hold at least one element.
2999
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
592,843,411✔
3000
    }
3001

3002
    pCtx->input.numOfInputCols = pFunct->numOfParams;
983,457,046✔
3003
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
983,435,537✔
3004
    QUERY_CHECK_NULL(pCtx->input.pData, code, lino, _end, terrno);
983,574,651✔
3005
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
983,355,703✔
3006
    QUERY_CHECK_NULL(pCtx->input.pColumnDataAgg, code, lino, _end, terrno);
983,582,264✔
3007

3008
    pCtx->pTsOutput = NULL;
983,485,952✔
3009
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
983,641,809✔
3010
    pCtx->resDataInfo.type = pFunct->resSchema.type;
983,496,527✔
3011
    pCtx->order = TSDB_ORDER_ASC;
983,592,408✔
3012
    pCtx->start.key = INT64_MIN;
983,654,160✔
3013
    pCtx->end.key = INT64_MIN;
983,522,817✔
3014
    pCtx->numOfParams = pExpr->base.numOfParams;
983,527,581✔
3015
    pCtx->param = pFunct->pParam;
983,671,050✔
3016
    pCtx->saveHandle.currentPage = -1;
983,453,186✔
3017
    pCtx->pStore = pStore;
983,649,998✔
3018
    pCtx->hasWindowOrGroup = false;
983,632,063✔
3019
    pCtx->needCleanup = false;
983,630,952✔
3020
    pCtx->skipDynDataCheck = false;
983,478,401✔
3021
  }
3022

3023
  for (int32_t i = 1; i < numOfOutput; ++i) {
1,013,943,715✔
3024
    (*rowEntryInfoOffset)[i] = (int32_t)((*rowEntryInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) +
1,350,857,571✔
3025
                                         pFuncCtx[i - 1].resDataInfo.interBufSize);
675,477,173✔
3026
  }
3027

3028
  code = setSelectValueColumnInfo(pFuncCtx, numOfOutput);
338,460,008✔
3029
  QUERY_CHECK_CODE(code, lino, _end);
338,406,022✔
3030

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

3041
    terrno = code;
×
3042
    return NULL;
×
3043
  }
3044
  return pFuncCtx;
338,324,639✔
3045
}
3046

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

3053
  int32_t i = 0, j = 0;
13,939,755✔
3054
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
109,725,747✔
3055
    SColumnInfoData* p = taosArrayGet(pCols, i);
95,786,494✔
3056
    if (!p) {
95,788,637✔
3057
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3058
      return terrno;
×
3059
    }
3060
    SColMatchItem* pmInfo = taosArrayGet(pColMatchInfo, j);
95,788,637✔
3061
    if (!pmInfo) {
95,788,593✔
3062
      return terrno;
×
3063
    }
3064

3065
    if (p->info.colId == pmInfo->colId) {
95,788,593✔
3066
      SColumnInfoData* pDst = taosArrayGet(pBlock->pDataBlock, pmInfo->dstSlotId);
89,576,459✔
3067
      if (!pDst) {
89,574,377✔
3068
        return terrno;
×
3069
      }
3070
      code = colDataAssign(pDst, p, pBlock->info.rows, &pBlock->info);
89,574,377✔
3071
      if (code != TSDB_CODE_SUCCESS) {
89,572,782✔
3072
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3073
        return code;
×
3074
      }
3075
      i++;
89,572,782✔
3076
      j++;
89,572,782✔
3077
    } else if (p->info.colId < pmInfo->colId) {
6,213,210✔
3078
      i++;
6,213,210✔
3079
    } else {
3080
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3081
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3082
    }
3083
  }
3084
  return code;
13,939,755✔
3085
}
3086

3087
SInterval extractIntervalInfo(const STableScanPhysiNode* pTableScanNode) {
157,023,019✔
3088
  SInterval interval = {
313,873,128✔
3089
      .interval = pTableScanNode->interval,
156,941,890✔
3090
      .sliding = pTableScanNode->sliding,
156,908,438✔
3091
      .intervalUnit = pTableScanNode->intervalUnit,
157,068,769✔
3092
      .slidingUnit = pTableScanNode->slidingUnit,
156,967,937✔
3093
      .offset = pTableScanNode->offset,
156,996,007✔
3094
      .precision = pTableScanNode->scan.node.pOutputDataBlockDesc->precision,
157,047,139✔
3095
      .timeRange = pTableScanNode->scanRange,
3096
  };
3097
  calcIntervalAutoOffset(&interval);
156,835,929✔
3098

3099
  return interval;
156,854,016✔
3100
}
3101

3102
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
51,884,400✔
3103
  SColumn c = {0};
51,884,400✔
3104

3105
  c.slotId = pColNode->slotId;
51,884,400✔
3106
  c.colId = pColNode->colId;
51,882,282✔
3107
  c.type = pColNode->node.resType.type;
51,885,767✔
3108
  c.bytes = pColNode->node.resType.bytes;
51,876,810✔
3109
  c.scale = pColNode->node.resType.scale;
51,885,015✔
3110
  c.precision = pColNode->node.resType.precision;
51,886,278✔
3111
  return c;
51,876,254✔
3112
}
3113

3114

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

3129
  if (cond->skey > cond->ekey || range->skey > range->ekey) {
3,075,742✔
3130
    *twindow = extTwindows[0] = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
5,228✔
3131
    return code;
4,755✔
3132
  }
3133

3134
  if (range->ekey < cond->skey) {
3,070,987✔
3135
    extTwindows[1] = *cond;
482,473✔
3136
    *twindow = extTwindows[0] = TSWINDOW_DESC_INITIALIZER;
482,473✔
3137
    return code;
482,473✔
3138
  }
3139

3140
  if (cond->ekey < range->skey) {
2,588,514✔
3141
    extTwindows[0] = *cond;
346,543✔
3142
    *twindow = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
346,543✔
3143
    return code;
346,543✔
3144
  }
3145

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

3153
  return code;
2,241,971✔
3154
}
3155

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

3168
int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, STableScanPhysiNode* pTableScanNode,
184,894,832✔
3169
                               const SReadHandle* readHandle, bool applyExtWin) {
3170
  int32_t code = 0;                             
184,894,832✔
3171
  pCond->order = pTableScanNode->scanSeq[0] > 0 ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
184,894,832✔
3172
  pCond->numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
184,893,939✔
3173

3174
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
184,914,257✔
3175
  if (!pCond->colList) {
184,859,923✔
3176
    return terrno;
×
3177
  }
3178
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
184,830,470✔
3179
  if (pCond->pSlotList == NULL) {
184,854,904✔
3180
    taosMemoryFreeClear(pCond->colList);
×
3181
    return terrno;
×
3182
  }
3183

3184
  // TODO: get it from stable scan node
3185
  pCond->twindows = pTableScanNode->scanRange;
184,704,394✔
3186
  pCond->suid = pTableScanNode->scan.suid;
184,885,888✔
3187
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
184,818,809✔
3188
  pCond->startVersion = -1;
184,844,558✔
3189
  pCond->endVersion = -1;
184,906,263✔
3190
  pCond->skipRollup = readHandle->skipRollup;
184,766,743✔
3191
  if (readHandle->winRangeValid) {
184,844,660✔
3192
    pCond->twindows = readHandle->winRange;
269,625✔
3193
  }
3194
  pCond->cacheSttStatis = readHandle->cacheSttStatis;
184,944,556✔
3195
  // allowed read stt file optimization mode
3196
  pCond->notLoadData = (pTableScanNode->dataRequired == FUNC_DATA_REQUIRED_NOT_LOAD) &&
369,752,605✔
3197
                       (pTableScanNode->scan.node.pConditions == NULL) && (pTableScanNode->interval == 0);
184,856,526✔
3198

3199
  int32_t j = 0;
184,832,743✔
3200
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
985,407,491✔
3201
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pTableScanNode->scan.pScanCols, i);
800,693,697✔
3202
    if (!pNode) {
800,367,475✔
UNCOV
3203
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
UNCOV
3204
      return terrno;
×
3205
    }
3206
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
800,367,475✔
3207
    if (pColNode->colType == COLUMN_TYPE_TAG) {
800,645,437✔
3208
      continue;
×
3209
    }
3210

3211
    pCond->colList[j].type = pColNode->node.resType.type;
800,664,586✔
3212
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
800,544,731✔
3213
    pCond->colList[j].colId = pColNode->colId;
800,610,092✔
3214
    pCond->colList[j].pk = pColNode->isPk;
800,718,531✔
3215

3216
    pCond->pSlotList[j] = pNode->slotId;
800,559,074✔
3217
    j += 1;
800,574,748✔
3218
  }
3219

3220
  pCond->numOfCols = j;
184,940,704✔
3221

3222
  if (applyExtWin) {
184,941,384✔
3223
    if (NULL != pTableScanNode->pExtScanRange) {
157,410,930✔
3224
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
3,011,074✔
3225
      code = getQueryExtWindow(&pCond->twindows, pTableScanNode->pExtScanRange, &pCond->twindows, pCond->extTwindows);
3,011,074✔
3226
    } else if (readHandle->extWinRangeValid) {
154,233,572✔
3227
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
64,668✔
3228
      code = getQueryExtWindow(&pCond->twindows, &readHandle->extWinRange, &pCond->twindows, pCond->extTwindows);
64,668✔
3229
    }
3230
  }
3231

3232
  if (pTableScanNode->pPrimaryCond) {
184,866,054✔
3233
    bool isStrict = false;
4,128✔
3234
    code = getPrimaryTimeRange((SNode**)&pTableScanNode->pPrimaryCond, &pCond->twindows, &isStrict);
4,128✔
3235
    if (code || !isStrict) {
4,128✔
3236
      code = nodesMergeNode((SNode**)&pTableScanNode->scan.node.pConditions, &pTableScanNode->pPrimaryCond);
1,376✔
3237
    }
3238
  }
3239

3240
  return code;
184,845,785✔
3241
}
3242

3243
int32_t initQueryTableDataCondWithColArray(SQueryTableDataCond* pCond, SQueryTableDataCond* pOrgCond,
44,617,026✔
3244
                                           const SReadHandle* readHandle, SArray* colArray) {
3245
  int32_t code = TSDB_CODE_SUCCESS;
44,617,026✔
3246
  int32_t lino = 0;
44,617,026✔
3247

3248
  pCond->order = TSDB_ORDER_ASC;
44,617,026✔
3249
  pCond->numOfCols = (int32_t)taosArrayGetSize(colArray);
44,623,057✔
3250

3251
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
44,608,011✔
3252
  QUERY_CHECK_NULL(pCond->colList, code, lino, _return, terrno);
44,610,437✔
3253

3254
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
44,603,950✔
3255
  QUERY_CHECK_NULL(pCond->pSlotList, code, lino, _return, terrno);
44,609,098✔
3256

3257
  pCond->twindows = pOrgCond->twindows;
44,593,488✔
3258
  pCond->order = pOrgCond->order;
44,605,065✔
3259
  pCond->type = pOrgCond->type;
44,614,993✔
3260
  pCond->startVersion = -1;
44,600,387✔
3261
  pCond->endVersion = -1;
44,604,015✔
3262
  pCond->skipRollup = true;
44,622,063✔
3263
  pCond->notLoadData = false;
44,609,999✔
3264

3265
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
217,435,968✔
3266
    SColIdPair* pColPair = taosArrayGet(colArray, i);
172,843,628✔
3267
    QUERY_CHECK_NULL(pColPair, code, lino, _return, terrno);
172,843,078✔
3268

3269
    bool find = false;
172,834,937✔
3270
    for (int32_t j = 0; j < pOrgCond->numOfCols; ++j) {
1,050,937,274✔
3271
      if (pOrgCond->colList[j].colId == pColPair->vtbColId) {
1,050,654,186✔
3272
        pCond->colList[i].type = pOrgCond->colList[j].type;
172,867,988✔
3273
        pCond->colList[i].bytes = pOrgCond->colList[j].bytes;
172,876,460✔
3274
        pCond->colList[i].colId = pColPair->orgColId;
172,888,483✔
3275
        pCond->colList[i].pk = pOrgCond->colList[j].pk;
172,884,987✔
3276
        pCond->pSlotList[i] = i;
172,895,087✔
3277
        find = true;
172,861,504✔
3278
        qDebug("%s mapped vtb colId:%d to org colId:%d", __func__, pColPair->vtbColId, pColPair->orgColId);
172,861,504✔
3279
        break;
172,843,998✔
3280
      }
3281
    }
3282
    QUERY_CHECK_CONDITION(find, code, lino, _return, TSDB_CODE_NOT_FOUND);
172,838,929✔
3283
  }
3284

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

3293
void cleanupQueryTableDataCond(SQueryTableDataCond* pCond) {
431,555,190✔
3294
  taosMemoryFreeClear(pCond->colList);
431,555,190✔
3295
  taosMemoryFreeClear(pCond->pSlotList);
431,543,791✔
3296
}
431,505,858✔
3297

3298
int32_t convertFillType(int32_t mode) {
3,650,722✔
3299
  int32_t type = TSDB_FILL_NONE;
3,650,722✔
3300
  switch (mode) {
3,650,722✔
3301
    case FILL_MODE_PREV:
174,342✔
3302
      type = TSDB_FILL_PREV;
174,342✔
3303
      break;
174,342✔
3304
    case FILL_MODE_NONE:
×
3305
      type = TSDB_FILL_NONE;
×
3306
      break;
×
3307
    case FILL_MODE_NULL:
193,961✔
3308
      type = TSDB_FILL_NULL;
193,961✔
3309
      break;
193,961✔
3310
    case FILL_MODE_NULL_F:
32,355✔
3311
      type = TSDB_FILL_NULL_F;
32,355✔
3312
      break;
32,355✔
3313
    case FILL_MODE_NEXT:
160,349✔
3314
      type = TSDB_FILL_NEXT;
160,349✔
3315
      break;
160,349✔
3316
    case FILL_MODE_VALUE:
155,163✔
3317
      type = TSDB_FILL_SET_VALUE;
155,163✔
3318
      break;
155,163✔
3319
    case FILL_MODE_VALUE_F:
11,282✔
3320
      type = TSDB_FILL_SET_VALUE_F;
11,282✔
3321
      break;
11,282✔
3322
    case FILL_MODE_LINEAR:
207,354✔
3323
      type = TSDB_FILL_LINEAR;
207,354✔
3324
      break;
207,354✔
3325
    case FILL_MODE_NEAR:
2,715,916✔
3326
      type = TSDB_FILL_NEAR;
2,715,916✔
3327
      break;
2,715,916✔
3328
    default:
×
3329
      type = TSDB_FILL_NONE;
×
3330
  }
3331

3332
  return type;
3,650,722✔
3333
}
3334

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

3342
    int64_t key = w->skey;
280,535✔
3343
    while (key < ts) {  // moving towards end
296,334✔
3344
      key = getNextTimeWindowStart(pInterval, key, TSDB_ORDER_ASC);
139,584✔
3345
      if (key > ts) {
139,725✔
3346
        break;
123,926✔
3347
      }
3348

3349
      w->skey = key;
15,799✔
3350
    }
3351
    w->ekey = taosTimeAdd(w->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
280,676✔
3352
  }
3353
}
2,147,483,647✔
3354

3355
static STimeWindow doCalculateTimeWindow(int64_t ts, SInterval* pInterval) {
69,515,851✔
3356
  STimeWindow w = {0};
69,515,851✔
3357

3358
  w.skey = taosTimeTruncate(ts, pInterval);
69,515,851✔
3359
  w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
69,517,050✔
3360
  return w;
69,519,836✔
3361
}
3362

3363
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order) {
1,753,903✔
3364
  STimeWindow win = *pWindow;
1,753,903✔
3365
  STimeWindow save = win;
1,753,903✔
3366
  while (win.skey <= ts && win.ekey >= ts) {
10,372,722✔
3367
    save = win;
8,618,819✔
3368
    // get previous time window
3369
    getNextTimeWindow(pInterval, &win, order == TSDB_ORDER_DESC ? TSDB_ORDER_ASC : TSDB_ORDER_DESC);
8,618,819✔
3370
  }
3371

3372
  return save;
1,753,903✔
3373
}
3374

3375
// get the correct time window according to the handled timestamp
3376
// todo refactor
3377
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
107,361,321✔
3378
                                int32_t order) {
3379
  STimeWindow w = {0};
107,361,321✔
3380
  if (pResultRowInfo->cur.pageId == -1) {  // the first window, from the previous stored value
107,362,824✔
3381
    getInitialStartTimeWindow(pInterval, ts, &w, (order != TSDB_ORDER_DESC));
2,753,022✔
3382
    return w;
2,754,468✔
3383
  }
3384

3385
  SResultRow* pRow = getResultRowByPos(pBuf, &pResultRowInfo->cur, false);
104,611,547✔
3386
  if (pRow) {
104,608,715✔
3387
    TAOS_SET_OBJ_ALIGNED(&w, pRow->win);
104,611,308✔
3388
  }
3389

3390
  // in case of typical time window, we can calculate time window directly.
3391
  if (w.skey > ts || w.ekey < ts) {
104,609,642✔
3392
    w = doCalculateTimeWindow(ts, pInterval);
69,515,048✔
3393
  }
3394

3395
  if (pInterval->interval != pInterval->sliding) {
104,614,430✔
3396
    // it is an sliding window query, in which sliding value is not equalled to
3397
    // interval value, and we need to find the first qualified time window.
3398
    w = getFirstQualifiedTimeWindow(ts, &w, pInterval, order);
1,753,903✔
3399
  }
3400

3401
  return w;
104,607,355✔
3402
}
3403

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

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

3417
bool hasLimitOffsetInfo(SLimitInfo* pLimitInfo) {
405,209,929✔
3418
  return (pLimitInfo->limit.limit != -1 || pLimitInfo->limit.offset != -1 || pLimitInfo->slimit.limit != -1 ||
806,637,305✔
3419
          pLimitInfo->slimit.offset != -1);
401,425,893✔
3420
}
3421

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

3426
void initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo) {
479,698,773✔
3427
  SLimit limit = {.limit = getLimit(pLimit), .offset = getOffset(pLimit)};
479,698,773✔
3428
  SLimit slimit = {.limit = getLimit(pSLimit), .offset = getOffset(pSLimit)};
479,515,192✔
3429

3430
  pLimitInfo->limit = limit;
479,505,228✔
3431
  pLimitInfo->slimit = slimit;
479,525,294✔
3432
  pLimitInfo->remainOffset = limit.offset;
479,534,824✔
3433
  pLimitInfo->remainGroupOffset = slimit.offset;
479,507,905✔
3434
  pLimitInfo->numOfOutputRows = 0;
479,541,269✔
3435
  pLimitInfo->numOfOutputGroups = 0;
479,689,511✔
3436
  pLimitInfo->currentGroupId = 0;
479,676,714✔
3437
}
479,626,376✔
3438

3439
void resetLimitInfoForNextGroup(SLimitInfo* pLimitInfo) {
89,997,112✔
3440
  pLimitInfo->numOfOutputRows = 0;
89,997,112✔
3441
  pLimitInfo->remainOffset = pLimitInfo->limit.offset;
90,023,086✔
3442
}
89,984,804✔
3443

3444
int32_t tableListGetSize(const STableListInfo* pTableList, int32_t* pRes) {
509,571,991✔
3445
  if (taosArrayGetSize(pTableList->pTableList) != taosHashGetSize(pTableList->map)) {
509,571,991✔
3446
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3447
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3448
  }
3449
  (*pRes) = taosArrayGetSize(pTableList->pTableList);
509,486,561✔
3450
  return TSDB_CODE_SUCCESS;
509,493,378✔
3451
}
3452

3453
uint64_t tableListGetSuid(const STableListInfo* pTableList) { return pTableList->idInfo.suid; }
4,606,503✔
3454

3455
STableKeyInfo* tableListGetInfo(const STableListInfo* pTableList, int32_t index) {
204,741,810✔
3456
  if (taosArrayGetSize(pTableList->pTableList) == 0) {
204,741,810✔
3457
    return NULL;
2,029✔
3458
  }
3459

3460
  return taosArrayGet(pTableList->pTableList, index);
204,718,577✔
3461
}
3462

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

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

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

3488
uint64_t tableListGetTableGroupId(const STableListInfo* pTableList, uint64_t tableUid) {
751,999,722✔
3489
  int32_t* slot = taosHashGet(pTableList->map, &tableUid, sizeof(tableUid));
751,999,722✔
3490
  if (slot == NULL) {
752,235,043✔
3491
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3492
    return -1;
×
3493
  }
3494

3495
  STableKeyInfo* pKeyInfo = taosArrayGet(pTableList->pTableList, *slot);
752,235,043✔
3496
  if (pKeyInfo == NULL) {
752,241,401✔
3497
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3498
    return -1;
×
3499
  }
3500
  return pKeyInfo->groupId;
752,241,401✔
3501
}
3502

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

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

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

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

3524
//   return code;
3525
// }
3526

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

3535
  STableKeyInfo keyInfo = {.uid = uid, .groupId = gid};
1,021,839✔
3536
  void*         p = taosHashGet(pTableList->map, &uid, sizeof(uid));
1,021,863✔
3537
  if (p != NULL) {
1,021,564✔
3538
    qInfo("table:%" PRId64 " already in tableIdList, ignore it", uid);
645✔
3539
    goto _end;
645✔
3540
  }
3541

3542
  void* tmp = taosArrayPush(pTableList->pTableList, &keyInfo);
1,020,919✔
3543
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,020,911✔
3544

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

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

3560
  return code;
1,022,760✔
3561
}
3562

3563
int32_t tableListGetGroupList(const STableListInfo* pTableList, int32_t ordinalGroupIndex, STableKeyInfo** pKeyInfo,
176,981,364✔
3564
                              int32_t* size) {
3565
  int32_t totalGroups = tableListGetOutputGroups(pTableList);
176,981,364✔
3566
  int32_t numOfTables = 0;
176,960,456✔
3567
  int32_t code = tableListGetSize(pTableList, &numOfTables);
176,984,558✔
3568
  if (code != TSDB_CODE_SUCCESS) {
176,931,194✔
3569
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3570
    return code;
×
3571
  }
3572

3573
  if (ordinalGroupIndex < 0 || ordinalGroupIndex >= totalGroups) {
176,931,194✔
3574
    return TSDB_CODE_INVALID_PARA;
×
3575
  }
3576

3577
  // here handle two special cases:
3578
  // 1. only one group exists, and 2. one table exists for each group.
3579
  if (totalGroups == 1) {
176,931,194✔
3580
    *size = numOfTables;
176,450,091✔
3581
    *pKeyInfo = (*size == 0) ? NULL : taosArrayGet(pTableList->pTableList, 0);
176,384,561✔
3582
    return TSDB_CODE_SUCCESS;
176,465,204✔
3583
  } else if (totalGroups == numOfTables) {
481,164✔
3584
    *size = 1;
457,327✔
3585
    *pKeyInfo = taosArrayGet(pTableList->pTableList, ordinalGroupIndex);
457,327✔
3586
    return TSDB_CODE_SUCCESS;
457,841✔
3587
  }
3588

3589
  int32_t offset = pTableList->groupOffset[ordinalGroupIndex];
27,589✔
3590
  if (ordinalGroupIndex < totalGroups - 1) {
55,074✔
3591
    *size = pTableList->groupOffset[ordinalGroupIndex + 1] - offset;
41,394✔
3592
  } else {
3593
    *size = numOfTables - offset;
13,680✔
3594
  }
3595

3596
  *pKeyInfo = taosArrayGet(pTableList->pTableList, offset);
55,074✔
3597
  return TSDB_CODE_SUCCESS;
55,074✔
3598
}
3599

3600
int32_t tableListGetOutputGroups(const STableListInfo* pTableList) { return pTableList->numOfOuputGroups; }
482,202,899✔
3601

3602
bool oneTableForEachGroup(const STableListInfo* pTableList) { return pTableList->oneTableForEachGroup; }
1,289,190✔
3603

3604
STableListInfo* tableListCreate() {
203,492,729✔
3605
  STableListInfo* pListInfo = taosMemoryCalloc(1, sizeof(STableListInfo));
203,492,729✔
3606
  if (pListInfo == NULL) {
203,365,606✔
3607
    return NULL;
×
3608
  }
3609

3610
  pListInfo->remainGroups = NULL;
203,365,606✔
3611
  pListInfo->pTableList = taosArrayInit(4, sizeof(STableKeyInfo));
203,371,064✔
3612
  if (pListInfo->pTableList == NULL) {
203,437,068✔
3613
    goto _error;
×
3614
  }
3615

3616
  pListInfo->map = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
203,444,205✔
3617
  if (pListInfo->map == NULL) {
203,606,563✔
3618
    goto _error;
×
3619
  }
3620

3621
  pListInfo->numOfOuputGroups = 1;
203,609,322✔
3622
  return pListInfo;
203,604,709✔
3623

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

3629
void tableListDestroy(STableListInfo* pTableListInfo) {
213,619,349✔
3630
  if (pTableListInfo == NULL) {
213,619,349✔
3631
    return;
10,103,075✔
3632
  }
3633

3634
  taosArrayDestroy(pTableListInfo->pTableList);
203,516,274✔
3635
  taosMemoryFreeClear(pTableListInfo->groupOffset);
203,473,645✔
3636

3637
  taosHashCleanup(pTableListInfo->map);
203,471,019✔
3638
  taosHashCleanup(pTableListInfo->remainGroups);
203,568,638✔
3639
  pTableListInfo->pTableList = NULL;
203,558,036✔
3640
  pTableListInfo->map = NULL;
203,581,722✔
3641
  taosMemoryFree(pTableListInfo);
203,578,272✔
3642
}
3643

3644
void tableListClear(STableListInfo* pTableListInfo) {
769,270✔
3645
  if (pTableListInfo == NULL) {
769,270✔
3646
    return;
×
3647
  }
3648

3649
  taosArrayClear(pTableListInfo->pTableList);
769,270✔
3650
  taosHashClear(pTableListInfo->map);
770,749✔
3651
  taosHashClear(pTableListInfo->remainGroups);
771,363✔
3652
  taosMemoryFree(pTableListInfo->groupOffset);
771,363✔
3653
  pTableListInfo->numOfOuputGroups = 1;
771,363✔
3654
  pTableListInfo->oneTableForEachGroup = false;
771,363✔
3655
}
3656

3657
static int32_t orderbyGroupIdComparFn(const void* p1, const void* p2) {
517,853,358✔
3658
  STableKeyInfo* pInfo1 = (STableKeyInfo*)p1;
517,853,358✔
3659
  STableKeyInfo* pInfo2 = (STableKeyInfo*)p2;
517,853,358✔
3660

3661
  if (pInfo1->groupId == pInfo2->groupId) {
517,853,358✔
3662
    return 0;
484,696,526✔
3663
  } else {
3664
    return pInfo1->groupId < pInfo2->groupId ? -1 : 1;
33,183,510✔
3665
  }
3666
}
3667

3668
int32_t sortTableGroup(STableListInfo* pTableListInfo) {
28,183,839✔
3669
  int32_t code = TSDB_CODE_SUCCESS;
28,183,839✔
3670
  taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
28,183,839✔
3671
  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
28,196,110✔
3672
  if (size == 0) {
28,181,166✔
3673
    pTableListInfo->numOfOuputGroups = 0;
×
3674
    return code;
×
3675
  }
3676

3677
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
28,181,166✔
3678
  if (!pList) {
28,183,773✔
3679
    code = terrno;
×
3680
    goto end;
×
3681
  }
3682

3683
  STableKeyInfo* pInfo = taosArrayGet(pTableListInfo->pTableList, 0);
28,183,773✔
3684
  if (pInfo == NULL) {
28,178,685✔
3685
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3686
    code = terrno;
×
3687
    goto end;
×
3688
  }
3689
  uint64_t gid = pInfo->groupId;
28,178,685✔
3690

3691
  int32_t start = 0;
28,187,474✔
3692
  void*   tmp = taosArrayPush(pList, &start);
28,190,079✔
3693
  if (tmp == NULL) {
28,190,079✔
3694
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3695
    code = terrno;
×
3696
    goto end;
×
3697
  }
3698

3699
  for (int32_t i = 1; i < size; ++i) {
148,296,084✔
3700
    pInfo = taosArrayGet(pTableListInfo->pTableList, i);
120,115,345✔
3701
    if (pInfo == NULL) {
120,105,098✔
3702
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3703
      code = terrno;
×
3704
      goto end;
×
3705
    }
3706
    if (pInfo->groupId != gid) {
120,105,098✔
3707
      tmp = taosArrayPush(pList, &i);
6,809,044✔
3708
      if (tmp == NULL) {
6,809,044✔
3709
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3710
        code = terrno;
×
3711
        goto end;
×
3712
      }
3713
      gid = pInfo->groupId;
6,809,044✔
3714
    }
3715
  }
3716

3717
  pTableListInfo->numOfOuputGroups = taosArrayGetSize(pList);
28,194,680✔
3718
  pTableListInfo->groupOffset = taosMemoryMalloc(sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
28,195,294✔
3719
  if (pTableListInfo->groupOffset == NULL) {
28,177,170✔
3720
    code = terrno;
×
3721
    goto end;
×
3722
  }
3723

3724
  memcpy(pTableListInfo->groupOffset, taosArrayGet(pList, 0), sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
28,181,265✔
3725

3726
end:
28,172,490✔
3727
  taosArrayDestroy(pList);
28,175,836✔
3728
  return code;
28,149,371✔
3729
}
3730

3731
int32_t buildGroupIdMapForAllTables(STableListInfo* pTableListInfo, SReadHandle* pHandle, SScanPhysiNode* pScanNode,
169,377,712✔
3732
                                    SNodeList* group, bool groupSort, uint8_t* digest, SStorageAPI* pAPI, SHashObj* groupIdMap) {
3733
  int32_t code = TSDB_CODE_SUCCESS;
169,377,712✔
3734

3735
  bool   groupByTbname = groupbyTbname(group);
169,377,712✔
3736
  size_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
169,390,487✔
3737
  if (!numOfTables) {
169,380,825✔
3738
    return code;
3,358✔
3739
  }
3740
  qDebug("numOfTables:%zu, groupByTbname:%d, group:%p", numOfTables, groupByTbname, group);
169,377,467✔
3741
  if (group == NULL || groupByTbname) {
169,330,337✔
3742
    if (tsCountAlwaysReturnValue && QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode) &&
165,380,077✔
3743
        ((STableScanPhysiNode*)pScanNode)->needCountEmptyTable) {
122,987,010✔
3744
      pTableListInfo->remainGroups =
13,068,376✔
3745
          taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
13,068,376✔
3746
      if (pTableListInfo->remainGroups == NULL) {
13,068,376✔
3747
        return terrno;
×
3748
      }
3749

3750
      for (int i = 0; i < numOfTables; i++) {
65,973,276✔
3751
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
52,905,398✔
3752
        if (!info) {
52,902,831✔
3753
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3754
          return terrno;
×
3755
        }
3756
        info->groupId = groupByTbname ? info->uid : 0;
52,902,831✔
3757
        int32_t tempRes = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId),
52,906,394✔
3758
                                      &(info->uid), sizeof(info->uid));
52,905,896✔
3759
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
52,904,900✔
3760
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3761
          return tempRes;
×
3762
        }
3763
      }
3764
    } else {
3765
      for (int32_t i = 0; i < numOfTables; i++) {
655,305,391✔
3766
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
503,084,874✔
3767
        if (!info) {
503,074,411✔
UNCOV
3768
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
UNCOV
3769
          return terrno;
×
3770
        }
3771
        info->groupId = groupByTbname ? info->uid : 0;
503,074,411✔
3772
        
3773
      }
3774
    }
3775
    if (groupIdMap && group != NULL){
165,288,395✔
3776
      getColInfoResultForGroupbyForStream(pHandle->vnode, group, pTableListInfo, pAPI, groupIdMap);
67,622✔
3777
    }
3778

3779
    pTableListInfo->oneTableForEachGroup = groupByTbname;
165,288,395✔
3780
    if (numOfTables == 1 && pTableListInfo->idInfo.tableType == TSDB_CHILD_TABLE) {
165,391,247✔
3781
      pTableListInfo->oneTableForEachGroup = true;
36,507,061✔
3782
    }
3783

3784
    if (groupSort && groupByTbname) {
165,394,523✔
3785
      taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
2,239,896✔
3786
      pTableListInfo->numOfOuputGroups = numOfTables;
2,240,386✔
3787
    } else if (groupByTbname && pScanNode->groupOrderScan) {
163,154,627✔
3788
      pTableListInfo->numOfOuputGroups = numOfTables;
29,508✔
3789
    } else {
3790
      pTableListInfo->numOfOuputGroups = 1;
163,128,289✔
3791
    }
3792
    if (groupSort || pScanNode->groupOrderScan) {
165,437,174✔
3793
      code = sortTableGroup(pTableListInfo);
28,066,734✔
3794
    }
3795
  } else {
3796
    bool initRemainGroups = false;
3,950,260✔
3797
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode)) {
3,950,260✔
3798
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pScanNode;
3,861,343✔
3799
      if (tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable &&
3,861,343✔
3800
          !(groupSort || pScanNode->groupOrderScan)) {
1,931,635✔
3801
        initRemainGroups = true;
1,908,243✔
3802
      }
3803
    }
3804

3805
    code = getColInfoResultForGroupby(pHandle->vnode, group, pTableListInfo, digest, pAPI, initRemainGroups, groupIdMap);
3,950,260✔
3806
    if (code != TSDB_CODE_SUCCESS) {
3,948,542✔
3807
      return code;
×
3808
    }
3809

3810
    if (pScanNode->groupOrderScan) pTableListInfo->numOfOuputGroups = taosArrayGetSize(pTableListInfo->pTableList);
3,948,542✔
3811

3812
    if (groupSort || pScanNode->groupOrderScan) {
3,948,542✔
3813
      code = sortTableGroup(pTableListInfo);
134,976✔
3814
    }
3815
  }
3816

3817
  // add all table entry in the hash map
3818
  size_t size = taosArrayGetSize(pTableListInfo->pTableList);
169,314,595✔
3819
  for (int32_t i = 0; i < size; ++i) {
747,679,607✔
3820
    STableKeyInfo* p = taosArrayGet(pTableListInfo->pTableList, i);
578,307,391✔
3821
    if (!p) {
578,141,086✔
3822
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3823
      return terrno;
×
3824
    }
3825
    int32_t tempRes = taosHashPut(pTableListInfo->map, &p->uid, sizeof(uint64_t), &i, sizeof(int32_t));
578,141,086✔
3826
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
578,341,911✔
3827
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3828
      return tempRes;
×
3829
    }
3830
  }
3831

3832
  return code;
169,413,928✔
3833
}
3834

3835
int32_t createScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
187,206,397✔
3836
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
3837
                                SExecTaskInfo* pTaskInfo, SHashObj* groupIdMap) {
3838
  int64_t     st = taosGetTimestampUs();
187,231,703✔
3839
  const char* idStr = GET_TASKID(pTaskInfo);
187,231,703✔
3840

3841
  if (pHandle == NULL) {
187,029,134✔
3842
    qError("invalid handle, in creating operator tree, %s", idStr);
×
3843
    return TSDB_CODE_INVALID_PARA;
×
3844
  }
3845

3846
  if (pHandle->uid != 0) {
187,029,134✔
3847
    pScanNode->uid = pHandle->uid;
62,468✔
3848
    pScanNode->tableType = TSDB_CHILD_TABLE;
62,468✔
3849
  }
3850
  uint8_t digest[17] = {0};
187,183,013✔
3851
  int32_t code = getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, digest, idStr,
187,144,271✔
3852
                              &pTaskInfo->storageAPI, pTaskInfo->pStreamRuntimeInfo);
187,196,771✔
3853
  if (code != TSDB_CODE_SUCCESS) {
187,324,878✔
3854
    qError("failed to getTableList, code:%s", tstrerror(code));
918✔
3855
    return code;
918✔
3856
  }
3857

3858
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
187,323,960✔
3859

3860
  int64_t st1 = taosGetTimestampUs();
187,335,784✔
3861
  pTaskInfo->cost.extractListTime = (st1 - st) / 1000.0;
187,335,784✔
3862
  qDebug("extract queried table list completed, %d tables, elapsed time:%.2f ms %s", numOfTables,
187,283,468✔
3863
         pTaskInfo->cost.extractListTime, idStr);
3864

3865
  if (numOfTables == 0) {
187,281,527✔
3866
    qDebug("no table qualified for query, %s", idStr);
17,980,288✔
3867
    return TSDB_CODE_SUCCESS;
17,981,276✔
3868
  }
3869

3870
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, digest, &pTaskInfo->storageAPI, groupIdMap);
169,301,239✔
3871
  if (code != TSDB_CODE_SUCCESS) {
169,386,796✔
3872
    return code;
×
3873
  }
3874

3875
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
169,390,268✔
3876
  qDebug("generate group id map completed, elapsed time:%.2f ms %s", pTaskInfo->cost.groupIdMapTime, idStr);
169,375,474✔
3877

3878
  return TSDB_CODE_SUCCESS;
169,376,014✔
3879
}
3880

3881
char* getStreamOpName(uint16_t opType) {
5,774,601✔
3882
  switch (opType) {
5,774,601✔
3883
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
×
3884
      return "stream scan";
×
3885
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
5,608,280✔
3886
      return "project";
5,608,280✔
3887
    case QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW:
166,321✔
3888
      return "external window";
166,321✔
3889
  }
3890
  return "error name";
×
3891
}
3892

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

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

3934
TSKEY getStartTsKey(STimeWindow* win, const TSKEY* tsCols) { return tsCols == NULL ? win->skey : tsCols[0]; }
13,151,383✔
3935

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

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

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

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

3956
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
2,147,483,647✔
3957
      if (isNull[i] != 1) return 1;
256,466,434✔
3958
    } else {
3959
      if (isNull[i] != 0) return 1;
2,147,483,647✔
3960
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,147,483,647✔
3961
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,147,483,647✔
3962
        int32_t len = getJsonValueLen(val);
×
3963
        if (memcmp(p, val, len) != 0) return 1;
×
3964
        p += len;
×
3965
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,147,483,647✔
3966
        if (IS_STR_DATA_BLOB(pCol->type)) {
1,184,821,929✔
3967
          if (memcmp(p, val, blobDataTLen(val)) != 0) return 1;
×
3968
          p += blobDataTLen(val);
×
3969
        } else {
3970
          if (memcmp(p, val, varDataTLen(val)) != 0) return 1;
1,186,042,349✔
3971
          p += varDataTLen(val);
1,185,482,624✔
3972
        }
3973
      } else {
3974
        if (0 != memcmp(p, val, pCol->bytes)) return 1;
2,147,483,647✔
3975
        p += pCol->bytes;
2,147,483,647✔
3976
      }
3977
    }
3978
  }
3979
  if ((int32_t)(p - oldkeyBuf) != oldKeysLen) return 1;
2,147,483,647✔
3980
  return 0;
2,147,483,647✔
3981
}
3982

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

3989
  for (int32_t i = 0; i < colNum; ++i) {
4,283,762✔
3990
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
2,719,166✔
3991
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
2,719,166✔
3992
    if (pCol->slotId > pBlock->pDataBlock->size) continue;
2,719,166✔
3993

3994
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
2,719,166✔
3995

3996
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
5,438,332✔
3997
      isNull[i] = 1;
131,700✔
3998
    } else {
3999
      isNull[i] = 0;
2,587,466✔
4000
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,587,027✔
4001
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,587,466✔
4002
        int32_t len = getJsonValueLen(val);
×
4003
        memcpy(p, val, len);
×
4004
        p += len;
×
4005
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,587,466✔
4006
        if (IS_STR_DATA_BLOB(pCol->type)) {
968,434✔
4007
          blobDataCopy(p, val);
×
4008
          p += blobDataTLen(val);
×
4009
        } else {
4010
          varDataCopy(p, val);
967,556✔
4011
          p += varDataTLen(val);
967,556✔
4012
        }
4013
      } else {
4014
        memcpy(p, val, pCol->bytes);
1,619,471✔
4015
        p += pCol->bytes;
1,619,910✔
4016
      }
4017
    }
4018
  }
4019
  return (int32_t)(p - keyBuf);
1,564,596✔
4020
}
4021

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

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

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

4050
int32_t extractKeysLen(const SArray* keys, int32_t* pLen) {
103,604✔
4051
  int32_t code = TSDB_CODE_SUCCESS;
103,604✔
4052
  int32_t lino = 0;
103,604✔
4053
  int32_t len = 0;
103,604✔
4054
  int32_t keyNum = taosArrayGetSize(keys);
103,604✔
4055
  for (int32_t i = 0; i < keyNum; ++i) {
263,400✔
4056
    SColumn* pCol = (SColumn*)taosArrayGet(keys, i);
159,796✔
4057
    QUERY_CHECK_NULL(pCol, code, lino, _end, terrno);
159,796✔
4058
    len += pCol->bytes;
159,796✔
4059
  }
4060
  len += sizeof(int8_t) * keyNum;  // null flag
103,604✔
4061
  *pLen = len;
103,604✔
4062

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

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

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

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

4092
  return code;
×
4093
}
4094

4095

4096
int32_t createExprSubQResBlock(SSDataBlock** ppBlock, SDataType* pResType) {
1,953,628✔
4097
  int32_t code = 0;
1,953,628✔
4098
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
1,953,628✔
4099
  if (pBlock == NULL) {
1,953,628✔
4100
    return terrno;
×
4101
  }
4102

4103
  pBlock->pDataBlock = taosArrayInit(1, sizeof(SColumnInfoData));
1,953,628✔
4104
  if (pBlock->pDataBlock == NULL) {
1,952,570✔
4105
    code = terrno;
×
4106
    taosMemoryFree(pBlock);
×
4107
    return code;
×
4108
  }
4109

4110
  SColumnInfoData idata =
1,953,099✔
4111
      createColumnInfoData(pResType->type, pResType->bytes, 0);
1,953,628✔
4112
  idata.info.scale = pResType->scale;
1,953,628✔
4113
  idata.info.precision = pResType->precision;
1,953,628✔
4114

4115
  code = blockDataAppendColInfo(pBlock, &idata);
1,953,628✔
4116
  if (code != TSDB_CODE_SUCCESS) {
1,953,099✔
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;
1,953,099✔
4124

4125
  return code;
1,953,628✔
4126
}
4127

4128

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

4134
  char* pNextStart = pRetrieveRsp->data;
1,953,628✔
4135
  char* pStart = pNextStart;
1,953,628✔
4136

4137
  int32_t index = 0;
1,953,628✔
4138

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

4144
  int32_t compLen = *(int32_t*)pStart;
1,953,628✔
4145
  pStart += sizeof(int32_t);
1,953,628✔
4146

4147
  int32_t rawLen = *(int32_t*)pStart;
1,953,099✔
4148
  pStart += sizeof(int32_t);
1,953,099✔
4149
  QUERY_CHECK_CONDITION((compLen <= rawLen && compLen != 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
1,953,099✔
4150

4151
  pNextStart = pStart + compLen;
1,953,099✔
4152
  if (pRetrieveRsp->compressed && (compLen < rawLen)) {
1,953,099✔
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);
1,953,628✔
4159
  if (code != 0) {
1,953,099✔
4160
    taosMemoryFreeClear(pRetrieveRsp);
×
4161
    goto _end;
×
4162
  }
4163

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

4172
int32_t setValueFromResBlock(STaskSubJobCtx* ctx, SRemoteValueNode* pRes, SSDataBlock* pBlock) {
1,768,114✔
4173
  int32_t code = 0;
1,768,114✔
4174
  bool needFree = true;
1,768,114✔
4175
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
1,768,114✔
4176
  if (NULL == pBlock->pDataBlock || 1 != colNum || pBlock->info.rows > 1) {
1,768,114✔
4177
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d, rows:%" PRId64, 
529✔
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;
1,767,056✔
4183
  pRes->val.flag &= (~VALUE_FLAG_VAL_UNSET);
1,768,114✔
4184
  pRes->val.translate = true;
1,768,114✔
4185
  
4186
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
1,768,114✔
4187
  if (colDataIsNull_s(pCol, 0)) {
1,768,114✔
4188
    pRes->val.isNull = true;
109,503✔
4189
  } else {
4190
    code = nodesSetValueNodeValueExt(&pRes->val, colDataGetData(pCol, 0), &needFree);
1,658,611✔
4191
  }
4192

4193
  if (!needFree) {
1,768,114✔
UNCOV
4194
    pCol->pData = NULL;
×
4195
  }
4196

4197
  return code;
1,768,114✔
4198
}
4199

4200
void handleRemoteValueRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp) {
1,934,220✔
4201
  SSDataBlock* pResBlock = NULL;
1,934,220✔
4202

4203
  qDebug("%s scl fetch rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
1,934,220✔
4204

4205
  if (pRsp->numOfRows > 1 || pRsp->numOfBlocks > 1 || !pRsp->completed) {
1,934,220✔
4206
    qError("%s invalid scl fetch rsp received, subQIdx:%d, rows:%" PRId64 ", blocks:%d, completed:%d", 
1,058✔
4207
      ctx->idStr, pParam->subQIdx, pRsp->numOfRows, pRsp->numOfBlocks, pRsp->completed);
4208
    ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
1,058✔
4209

UNCOV
4210
    return;
×
4211
  }
4212

4213
  if (0 == pRsp->numOfRows) {
1,933,162✔
4214
    SRemoteValueNode* pRemote = (SRemoteValueNode*)pParam->pRes;
166,106✔
4215
    pRemote->val.node.type = QUERY_NODE_VALUE;
166,106✔
4216
    pRemote->val.isNull = true;
166,106✔
4217
    pRemote->val.translate = true;
166,106✔
4218
    pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
166,106✔
4219
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
166,106✔
4220

4221
    return;
166,106✔
4222
  }
4223
  
4224
  ctx->code = createExprSubQResBlock(&pResBlock, &((SRemoteValueNode*)pParam->pRes)->val.node.resType);
1,767,056✔
4225
  if (TSDB_CODE_SUCCESS == ctx->code) {
1,768,114✔
4226
    ctx->code = blockDataEnsureCapacity(pResBlock, 1);
1,768,114✔
4227
  }
4228
  if (TSDB_CODE_SUCCESS == ctx->code) {
1,768,114✔
4229
    ctx->code = extractSingleRspBlock(pRsp, pResBlock);
1,768,114✔
4230
  }
4231
  if (TSDB_CODE_SUCCESS == ctx->code) {
1,768,114✔
4232
    ctx->code = setValueFromResBlock(ctx, (SRemoteValueNode*)pParam->pRes, pResBlock);
1,768,114✔
4233
  }
4234
  if (TSDB_CODE_SUCCESS == ctx->code) {
1,768,114✔
4235
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
1,768,114✔
4236
  }
4237

4238
  blockDataDestroy(pResBlock);  
1,767,056✔
4239
}
4240

4241

4242
int32_t updateValueListFromResBlock(STaskSubJobCtx* ctx, SRemoteValueListNode* pRes, SSDataBlock* pBlock) {
185,514✔
4243
  int32_t code = 0, lino = 0;
185,514✔
4244
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
185,514✔
4245
  if (NULL == pBlock->pDataBlock || 1 != colNum) {
185,514✔
UNCOV
4246
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d", ctx->idStr, pBlock->pDataBlock, colNum);
×
4247
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4248
  }
4249

4250
  pRes->hasValue = true;
185,514✔
4251
  
4252
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
185,514✔
4253
  TAOS_CHECK_EXIT(scalarBuildRemoteListHash(pRes, pCol, pBlock->info.rows));
185,514✔
4254

4255
_exit:
185,514✔
4256

4257
  if (code) {
185,514✔
4258
    qError("%s %s failed with error: %s", ctx->idStr, __func__, tstrerror(code));
×
4259
  }
4260
  
4261
  return code;
185,514✔
4262
}
4263

4264

4265

4266
void handleRemoteValueListRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
185,514✔
4267
  SSDataBlock* pResBlock = NULL;
185,514✔
4268
  SRemoteValueListNode* pRemote = (SRemoteValueListNode*)pParam->pRes;
185,514✔
4269

4270
  qDebug("%s scl fetch rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
185,514✔
4271

4272
  if (pRsp->numOfRows > 0) {
185,514✔
4273
    ctx->code = createExprSubQResBlock(&pResBlock, &((SExprNode*)pParam->pRes)->resType);
185,514✔
4274
    if (TSDB_CODE_SUCCESS == ctx->code) {
185,514✔
4275
      ctx->code = blockDataEnsureCapacity(pResBlock, pRsp->numOfRows);
185,514✔
4276
    }
4277
    if (TSDB_CODE_SUCCESS == ctx->code) {
185,514✔
4278
      ctx->code = extractSingleRspBlock(pRsp, pResBlock);
185,514✔
4279
    }
4280
    if (TSDB_CODE_SUCCESS == ctx->code) {
185,514✔
4281
      ctx->code = updateValueListFromResBlock(ctx, pRemote, pResBlock);
185,514✔
4282
    }
4283
    if (TSDB_CODE_SUCCESS == ctx->code && pRsp->completed) {
185,514✔
4284
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
144,480✔
4285
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
144,480✔
4286
    }
4287

4288
    blockDataDestroy(pResBlock);  
185,514✔
UNCOV
4289
  } else if (0 == pRsp->numOfRows && pRsp->completed) {
×
UNCOV
4290
    if (!pRemote->hasValue) {
×
UNCOV
4291
      ctx->code = scalarBuildRemoteListHash(pRemote, NULL, 0);
×
4292
    }
UNCOV
4293
    if (TSDB_CODE_SUCCESS == ctx->code) {    
×
UNCOV
4294
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
×
UNCOV
4295
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
×
4296
    }
4297
  }
4298

4299
  *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
185,514✔
4300

4301
  if (!(*fetchDone)) {
185,514✔
4302
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes);
40,505✔
4303
    if (TSDB_CODE_SUCCESS != code) {
41,034✔
4304
      ctx->code = code;
×
4305
      *fetchDone = true;
×
4306
    }
4307
  }
4308
}
185,514✔
4309

4310

4311
int32_t remoteFetchCallBack(void* param, SDataBuf* pMsg, int32_t code) {
2,959,786✔
4312
  SScalarFetchParam* pParam = (SScalarFetchParam*)param;
2,959,786✔
4313
  STaskSubJobCtx* ctx = pParam->pSubJobCtx;
2,959,786✔
4314
  char idStr[64];
2,959,786✔
4315
  
4316
  taosMemoryFreeClear(pMsg->pEpSet);
2,959,786✔
4317

4318
  if (NULL == ctx) {
2,959,786✔
4319
    qWarn("scl fetch ctx not exists since it may have been released");
7,406✔
4320
    goto _exit;
7,406✔
4321
  }
4322

4323
  if (qDebugFlag & DEBUG_DEBUG) {
2,952,380✔
4324
    strncpy(idStr, ctx->idStr, sizeof(idStr));
89,432✔
4325
  }
4326
  
4327
  qDebug("%s subQIdx %d got rsp, blockIdx:%" PRId64 ", code:%d, rsp:%p", ctx->idStr, pParam->subQIdx, ctx->blockIdx, code, pMsg->pData);
2,952,380✔
4328

4329
  taosWLockLatch(&ctx->lock);
2,952,380✔
4330
  ctx->param = NULL;
2,952,380✔
4331
  taosWUnLockLatch(&ctx->lock);
2,952,380✔
4332

4333
  if (ctx->transporterId > 0) {
2,952,380✔
4334
    int32_t ret = asyncFreeConnById(ctx->rpcHandle, ctx->transporterId);
2,952,380✔
4335
    if (ret != 0) {
2,952,380✔
4336
      qDebug("%s failed to free subQ rpc handle, code:%s, subQIdx:%d", ctx->idStr, tstrerror(ret), pParam->subQIdx);
×
4337
    }
4338
    ctx->transporterId = -1;
2,952,380✔
4339
  }
4340

4341
  if (0 == code && NULL == pMsg->pData) {
2,952,380✔
4342
    qError("%s invalid rsp msg, msgType:%d, len:%d", ctx->idStr, pMsg->msgType, pMsg->len);
×
4343
    code = TSDB_CODE_QRY_INVALID_MSG;
×
4344
  }
4345

4346
  if (code == TSDB_CODE_SUCCESS) {
2,952,380✔
4347
    SRetrieveTableRsp* pRsp = pMsg->pData;
2,119,734✔
4348
    pRsp->numOfRows = htobe64(pRsp->numOfRows);
2,119,734✔
4349
    pRsp->compLen = htonl(pRsp->compLen);
2,119,734✔
4350
    pRsp->payloadLen = htonl(pRsp->payloadLen);
2,119,734✔
4351
    pRsp->numOfCols = htonl(pRsp->numOfCols);
2,119,734✔
4352
    pRsp->useconds = htobe64(pRsp->useconds);
2,119,734✔
4353
    pRsp->numOfBlocks = htonl(pRsp->numOfBlocks);
2,119,734✔
4354

4355
    qDebug("%s subQIdx %d blockIdx:%" PRIu64 " rsp detail, numOfBlocks:%d, numOfRows:%" PRId64 ", completed:%d", 
2,119,734✔
4356
      ctx->idStr, pParam->subQIdx, ctx->blockIdx, pRsp->numOfBlocks, pRsp->numOfRows, pRsp->completed);
4357

4358
    ctx->blockIdx++;
2,119,734✔
4359

4360
    switch (nodeType(pParam->pRes)) {
2,119,734✔
4361
      case QUERY_NODE_REMOTE_VALUE:
1,934,220✔
4362
        handleRemoteValueRes(pParam, ctx, pRsp);
1,934,220✔
4363
        break;
1,933,691✔
4364
      case QUERY_NODE_REMOTE_VALUE_LIST: {
185,514✔
4365
        bool fetchDone = false;
185,514✔
4366
        handleRemoteValueListRes(pParam, ctx, pRsp, &fetchDone);
185,514✔
4367
        qDebug("%s subQIdx %d handle remote value list finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
185,514✔
4368
        if (!fetchDone) {
185,514✔
4369
          goto _exit;
41,034✔
4370
        }
4371
        break;
144,480✔
4372
      }  
4373
      default:
×
4374
        qError("%s invalid scl fetch res node %d, subQIdx:%d", ctx->idStr, nodeType(pParam->pRes), pParam->subQIdx);
×
4375
        ctx->code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4376
        break;
×
4377
    }
4378
  } else {
4379
    ctx->code = rpcCvtErrCode(code);
832,646✔
4380
    if (ctx->code != code) {
832,646✔
4381
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s, cvted error: %s", ctx->idStr, pParam->subQIdx,
×
4382
             tstrerror(code), tstrerror(ctx->code));
4383
    } else {
4384
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s", ctx->idStr, pParam->subQIdx, tstrerror(code));
832,646✔
4385
    }
4386
  }
4387

4388
  qDebug("%s subQIdx %d sem_post subQ ready", ctx->idStr, pParam->subQIdx);
2,911,346✔
4389
  
4390
  code = tsem_post(&pParam->pSubJobCtx->ready);
2,911,346✔
4391
  if (code != TSDB_CODE_SUCCESS) {
2,911,346✔
4392
    qError("failed to invoke post when scl fetch rsp is ready, code:%s", tstrerror(code));
×
4393
  }
4394

4395
_exit:
2,959,257✔
4396

4397
  taosMemoryFree(pMsg->pData);
2,959,786✔
4398

4399
  return code;
2,959,786✔
4400
}
4401

4402

4403
int32_t sendFetchRemoteNodeReq(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes) {
2,959,627✔
4404
  int32_t          code = TSDB_CODE_SUCCESS;
2,959,627✔
4405
  int32_t          lino = 0;
2,959,627✔
4406
  SDownstreamSourceNode* pSource = (SDownstreamSourceNode*)taosArrayGetP(ctx->subEndPoints, subQIdx);
2,959,627✔
4407

4408
  SResFetchReq req = {0};
2,954,496✔
4409
  req.header.vgId = pSource->addr.nodeId;
2,954,496✔
4410
  req.sId = pSource->sId;
2,955,924✔
4411
  req.clientId = pSource->clientId;
2,958,199✔
4412
  req.taskId = pSource->taskId;
2,957,855✔
4413
  req.srcTaskId = ctx->taskId;
2,951,507✔
4414
  req.blockIdx = ctx->blockIdx;
2,952,565✔
4415
  req.queryId = ctx->queryId;
2,954,337✔
4416
  req.execId = pSource->execId;
2,954,152✔
4417

4418
  int32_t msgSize = tSerializeSResFetchReq(NULL, 0, &req, false);
2,952,565✔
4419
  if (msgSize < 0) {
2,954,866✔
4420
    return msgSize;
×
4421
  }
4422

4423
  void* msg = taosMemoryCalloc(1, msgSize);
2,954,866✔
4424
  if (NULL == msg) {
2,944,815✔
4425
    return terrno;
×
4426
  }
4427

4428
  msgSize = tSerializeSResFetchReq(msg, msgSize, &req, false);
2,944,815✔
4429
  if (msgSize < 0) {
2,953,094✔
4430
    taosMemoryFree(msg);
×
4431
    return msgSize;
×
4432
  }
4433

4434
  qDebug("%s scl build fetch msg and send to nodeId:%d, ep:%s, clientId:0x%" PRIx64 " taskId:0x%" PRIx64
2,953,094✔
4435
         ", execId:%d, blockIdx:%" PRId64,
4436
         ctx->idStr, pSource->addr.nodeId, pSource->addr.epSet.eps[0].fqdn, pSource->clientId,
4437
         pSource->taskId, pSource->execId, req.blockIdx);
4438

4439
  // send the fetch remote task result reques
4440
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
2,960,315✔
4441
  if (NULL == pMsgSendInfo) {
2,958,728✔
4442
    taosMemoryFreeClear(msg);
×
4443
    qError("%s prepare message %d failed", ctx->idStr, (int32_t)sizeof(SMsgSendInfo));
×
4444
    return terrno;
×
4445
  }
4446

4447
  SScalarFetchParam* param = taosMemoryMalloc(sizeof(SScalarFetchParam));
2,958,728✔
4448
  if (NULL == param) {
2,950,793✔
4449
    taosMemoryFreeClear(msg);
×
4450
    taosMemoryFreeClear(pMsgSendInfo);
×
4451
    qError("%s prepare param %d failed", ctx->idStr, (int32_t)sizeof(SScalarFetchParam));
×
4452
    return terrno;
×
4453
  }
4454

4455
  taosWLockLatch(&ctx->lock);
2,950,793✔
4456
  
4457
  if (ctx->code) {
2,956,612✔
4458
    qError("task has been killed, error:%s", tstrerror(ctx->code));
529✔
4459
    taosMemoryFree(param);
529✔
4460
    taosMemoryFreeClear(msg);
529✔
4461
    taosMemoryFreeClear(pMsgSendInfo);
529✔
4462
    code = ctx->code;
529✔
4463
    taosWUnLockLatch(&ctx->lock);
529✔
4464
    goto _end;
529✔
4465
  } else {
4466
    ctx->param = param;
2,953,967✔
4467
  }
4468
  
4469
  taosWUnLockLatch(&ctx->lock);
2,957,670✔
4470

4471
  param->subQIdx = subQIdx;
2,956,612✔
4472
  param->pRes = pRes;
2,957,141✔
4473
  param->pSubJobCtx = ctx;
2,954,496✔
4474

4475
  pMsgSendInfo->param = param;
2,954,496✔
4476
  pMsgSendInfo->paramFreeFp = taosAutoMemoryFree;
2,952,380✔
4477
  pMsgSendInfo->msgInfo.pData = msg;
2,957,141✔
4478
  pMsgSendInfo->msgInfo.len = msgSize;
2,954,496✔
4479
  pMsgSendInfo->msgType = pSource->fetchMsgType;
2,954,496✔
4480
  pMsgSendInfo->fp = remoteFetchCallBack;
2,954,496✔
4481
  pMsgSendInfo->requestId = ctx->queryId;
2,953,438✔
4482

4483
  code = asyncSendMsgToServer(ctx->rpcHandle, &pSource->addr.epSet, &ctx->transporterId, pMsgSendInfo);
2,955,025✔
4484
  QUERY_CHECK_CODE(code, lino, _end);
2,960,844✔
4485
      
4486
_end:
2,961,373✔
4487

4488
  if (code != TSDB_CODE_SUCCESS) {
2,961,373✔
4489
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
529✔
4490
  }
4491
  
4492
  return code;
2,960,844✔
4493
}
4494

4495
int32_t fetchRemoteNodeImpl(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes) {
2,919,122✔
4496
  int32_t          code = TSDB_CODE_SUCCESS;
2,919,122✔
4497
  int32_t          lino = 0;
2,919,122✔
4498

4499
  ctx->blockIdx = 0;
2,919,122✔
4500

4501
  code = sendFetchRemoteNodeReq(ctx, subQIdx, pRes);
2,916,636✔
4502
  QUERY_CHECK_CODE(code, lino, _end);
2,919,810✔
4503

4504
  code = qSemWait(ctx->pTaskInfo, &ctx->ready);
2,919,281✔
4505
  if (isTaskKilled(ctx->pTaskInfo)) {
2,919,810✔
4506
    code = getTaskCode(ctx->pTaskInfo);
12,167✔
4507
  } else {
4508
    code = ctx->code;
2,907,643✔
4509
  }
4510
      
4511
_end:
2,920,339✔
4512

4513
  taosWLockLatch(&ctx->lock);
2,920,339✔
4514
  ctx->param = NULL;
2,920,339✔
4515
  taosWUnLockLatch(&ctx->lock);
2,920,339✔
4516

4517
  if (code != TSDB_CODE_SUCCESS) {
2,920,339✔
4518
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
841,639✔
4519
  }
4520
  return code;
2,920,339✔
4521
}
4522

4523
int32_t remoteNodeCopy(SNode* pSrc, SNode* pDst) {
56,074✔
4524
  int32_t code = 0, lino = 0;
56,074✔
4525
  
4526
  switch (nodeType(pSrc)) {
56,074✔
4527
    case QUERY_NODE_VALUE:
56,074✔
4528
      TAOS_CHECK_EXIT(valueNodeCopy((SValueNode*)pSrc, &((SRemoteValueNode*)pDst)->val));
56,074✔
4529
      ((SRemoteValueNode*)pDst)->val.node.type = QUERY_NODE_VALUE;
56,074✔
4530
      break;
56,074✔
UNCOV
4531
    case QUERY_NODE_REMOTE_VALUE_LIST: {
×
UNCOV
4532
      SRemoteValueListNode* pDstNode = (SRemoteValueListNode*)pDst;
×
UNCOV
4533
      memcpy(pDst, pSrc, sizeof(SRemoteValueListNode));
×
UNCOV
4534
      pDstNode->hashAllocated = false;      
×
UNCOV
4535
      break;
×
4536
    } 
4537
    default:
×
4538
      break;
×
4539
  }
4540

4541
_exit:
56,074✔
4542

4543
  if (code) {
56,074✔
4544
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4545
  }
4546

4547
  return code;
56,074✔
4548
}
4549

4550
int32_t qFetchRemoteNode(void* pCtx, int32_t subQIdx, SNode* pRes) {
2,975,196✔
4551
  STaskSubJobCtx*  ctx = (STaskSubJobCtx*)pCtx;
2,975,196✔
4552
  int32_t code = 0, lino = 0;
2,975,196✔
4553
  int32_t       subEndPoinsNum = taosArrayGetSize(ctx->subEndPoints);
2,975,196✔
4554
  if (subQIdx >= subEndPoinsNum) {
2,969,536✔
4555
    qError("%s invalid subQIdx %d, subEndPointsNum:%d", ctx->idStr, subQIdx, subEndPoinsNum);
×
4556
    return TSDB_CODE_QRY_SUBQ_NOT_FOUND;
×
4557
  }
4558

4559
  SNode** ppRes = taosArrayGet(ctx->subResNodes, subQIdx);
2,969,536✔
4560
  if (NULL == *ppRes) {
2,969,721✔
4561
    TAOS_CHECK_EXIT(fetchRemoteNodeImpl(ctx, subQIdx, pRes));
2,916,636✔
4562
    *ppRes = pRes;
2,078,700✔
4563
  } else {
4564
    TAOS_CHECK_EXIT(remoteNodeCopy(*ppRes, pRes));
56,074✔
4565
  }
4566

4567
_exit:
56,074✔
4568

4569
  if (code) {
2,976,413✔
4570
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
841,639✔
4571
  } else {
4572
    qDebug("%s %s subQIdx %d succeed", ctx->idStr, __func__, subQIdx);
2,134,774✔
4573
  }
4574

4575
  return code;
2,976,413✔
4576
}
4577

4578

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