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

taosdata / TDengine / #4948

04 Feb 2026 09:05AM UTC coverage: 66.866% (-0.006%) from 66.872%
#4948

push

travis-ci

web-flow
enh(keeper): password information desensitization processing (#34458)

Close https://project.feishu.cn/taosdata_td/feature/detail/6600039687

Commits:

* feat: add String method to TDengineRestful for safe string representation

* feat: replace hardcoded credentials with test utility functions in tests

* feat: update test workflows to use environment variables for credentials

* test: fix failed test case

* chore: setup tmate

* feat: streamline test execution by setting environment variables for credentials

* feat: enhance test utility functions to return default values if environment variables are not set

205522 of 307364 relevant lines covered (66.87%)

124991068.44 hits per line

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

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

74
void initResultRowInfo(SResultRowInfo* pResultRowInfo) {
327,297,864✔
75
  pResultRowInfo->size = 0;
327,297,864✔
76
  pResultRowInfo->cur.pageId = -1;
327,326,843✔
77
}
327,374,782✔
78

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

81
void resetResultRow(SResultRow* pResultRow, size_t entrySize) {
864,667,928✔
82
  pResultRow->numOfRows = 0;
864,667,928✔
83
  pResultRow->nOrigRows = 0;
864,755,102✔
84
  pResultRow->closed = false;
864,763,796✔
85
  pResultRow->endInterp = false;
864,768,614✔
86
  pResultRow->startInterp = false;
864,732,086✔
87

88
  if (entrySize > 0) {
864,765,260✔
89
    memset(pResultRow->pEntryInfo, 0, entrySize);
864,767,462✔
90
  }
91
}
864,790,184✔
92

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

98
size_t getResultRowSize(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
199,745,948✔
99
  int32_t rowSize = (numOfOutput * sizeof(SResultRowEntryInfo)) + sizeof(SResultRow);
199,745,948✔
100

101
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,022,957,378✔
102
    rowSize += pCtx[i].resDataInfo.interBufSize;
823,264,668✔
103
  }
104

105
  return rowSize;
199,692,710✔
106
}
107

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

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

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

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

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

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

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

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

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

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

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

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

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

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

221
void cleanupGroupResInfo(SGroupResInfo* pGroupResInfo) {
109,211,574✔
222
  taosMemoryFreeClear(pGroupResInfo->pBuf);
109,211,574✔
223
  if (pGroupResInfo->freeItem) {
109,226,210✔
224
    //    taosArrayDestroy(pGroupResInfo->pRows);
225
    taosArrayDestroyEx(pGroupResInfo->pRows, freeEx);
×
226
    pGroupResInfo->freeItem = false;
×
227
    pGroupResInfo->pRows = NULL;
×
228
  } else {
229
    taosArrayDestroy(pGroupResInfo->pRows);
109,207,456✔
230
    pGroupResInfo->pRows = NULL;
109,184,240✔
231
  }
232
  pGroupResInfo->index = 0;
109,198,444✔
233
  pGroupResInfo->delIndex = 0;
109,201,704✔
234
}
109,193,015✔
235

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

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

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

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

256
int32_t initGroupedResultInfo(SGroupResInfo* pGroupResInfo, SSHashObj* pHashmap, int32_t order) {
80,038,746✔
257
  int32_t code = TSDB_CODE_SUCCESS;
80,038,746✔
258
  int32_t lino = 0;
80,038,746✔
259
  if (pGroupResInfo->pRows != NULL) {
80,038,746✔
260
    taosArrayDestroy(pGroupResInfo->pRows);
4,280,375✔
261
  }
262
  if (pGroupResInfo->pBuf) {
80,045,317✔
263
    taosMemoryFree(pGroupResInfo->pBuf);
4,281,005✔
264
    pGroupResInfo->pBuf = NULL;
4,280,690✔
265
  }
266

267
  // extract the result rows information from the hash map
268
  int32_t size = tSimpleHashGetSize(pHashmap);
80,041,337✔
269

270
  void* pData = NULL;
80,044,268✔
271
  pGroupResInfo->pRows = taosArrayInit(size, POINTER_BYTES);
80,044,268✔
272
  QUERY_CHECK_NULL(pGroupResInfo->pRows, code, lino, _end, terrno);
80,041,528✔
273

274
  size_t  keyLen = 0;
80,034,127✔
275
  int32_t iter = 0;
80,038,557✔
276
  int64_t bufLen = 0, offset = 0;
80,040,584✔
277

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

284
  pGroupResInfo->pBuf = taosMemoryMalloc(bufLen);
79,859,018✔
285
  QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
80,047,942✔
286

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

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

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

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

302
  if (order == TSDB_ORDER_ASC || order == TSDB_ORDER_DESC) {
80,015,330✔
303
    __compar_fn_t fn = (order == TSDB_ORDER_ASC) ? resultrowComparAsc : resultrowComparDesc;
4,962,744✔
304
    size = POINTER_BYTES;
4,962,744✔
305
    taosSort(pGroupResInfo->pRows->pData, taosArrayGetSize(pGroupResInfo->pRows), size, fn);
4,962,744✔
306
  }
307

308
  pGroupResInfo->index = 0;
80,015,330✔
309

310
_end:
80,038,659✔
311
  if (code != TSDB_CODE_SUCCESS) {
80,046,367✔
312
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
313
  }
314
  return code;
80,046,367✔
315
}
316

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

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

328
bool hasRemainResults(SGroupResInfo* pGroupResInfo) {
322,807,628✔
329
  if (pGroupResInfo->pRows == NULL) {
322,807,628✔
330
    return false;
×
331
  }
332

333
  return pGroupResInfo->index < taosArrayGetSize(pGroupResInfo->pRows);
322,814,170✔
334
}
335

336
int32_t getNumOfTotalRes(SGroupResInfo* pGroupResInfo) {
171,757,202✔
337
  if (pGroupResInfo->pRows == 0) {
171,757,202✔
338
    return 0;
×
339
  }
340

341
  return (int32_t)taosArrayGetSize(pGroupResInfo->pRows);
171,769,762✔
342
}
343

344
SArray* createSortInfo(SNodeList* pNodeList) {
49,032,189✔
345
  size_t numOfCols = 0;
49,032,189✔
346

347
  if (pNodeList != NULL) {
49,032,189✔
348
    numOfCols = LIST_LENGTH(pNodeList);
48,827,027✔
349
  } else {
350
    numOfCols = 0;
205,162✔
351
  }
352

353
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
49,026,622✔
354
  if (pList == NULL) {
49,016,660✔
355
    return pList;
×
356
  }
357

358
  for (int32_t i = 0; i < numOfCols; ++i) {
111,691,749✔
359
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
62,663,817✔
360
    if (!pSortKey) {
62,669,568✔
361
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
362
      taosArrayDestroy(pList);
×
363
      pList = NULL;
×
364
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
365
      break;
×
366
    }
367
    SBlockOrderInfo bi = {0};
62,669,568✔
368
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
62,668,349✔
369
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);
62,669,589✔
370

371
    if (nodeType(pSortKey->pExpr) != QUERY_NODE_COLUMN) {
62,668,829✔
372
      qError("invalid order by expr type:%d", nodeType(pSortKey->pExpr));
×
373
      taosArrayDestroy(pList);
×
374
      pList = NULL;
×
375
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
376
      break;
×
377
    }
378
    
379
    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
62,649,068✔
380
    bi.slotId = pColNode->slotId;
62,671,162✔
381
    void* tmp = taosArrayPush(pList, &bi);
62,675,952✔
382
    if (!tmp) {
62,675,952✔
383
      taosArrayDestroy(pList);
×
384
      pList = NULL;
×
385
      break;
×
386
    }
387
  }
388

389
  return pList;
49,027,932✔
390
}
391

392
SSDataBlock* createDataBlockFromDescNode(void* p) {
583,359,257✔
393
  SDataBlockDescNode* pNode = (SDataBlockDescNode*)p;
583,359,257✔
394
  int32_t      numOfCols = LIST_LENGTH(pNode->pSlots);
583,359,257✔
395
  SSDataBlock* pBlock = NULL;
583,449,434✔
396
  int32_t      code = TSDB_CODE_SUCCESS;
583,424,258✔
397
  int32_t      lino = 0;
583,424,258✔
398

399
  code = createDataBlock(&pBlock);
583,424,258✔
400
  QUERY_CHECK_CODE(code, lino, _return);
583,307,282✔
401

402
  pBlock->info.id.blockId = pNode->dataBlockId;
583,307,282✔
403
  pBlock->info.type = STREAM_INVALID;
583,409,176✔
404
  pBlock->info.calWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
583,335,345✔
405
  pBlock->info.watermark = INT64_MIN;
583,191,387✔
406

407
  int32_t i = 0;
583,513,511✔
408
  SNode*  node = NULL;
583,513,511✔
409
  FOREACH(node, pNode->pSlots) {
2,147,483,647✔
410
    SSlotDescNode* pDescNode = (SSlotDescNode*)node;
2,147,483,647✔
411
    SColumnInfoData idata =
2,147,483,647✔
412
        createColumnInfoData(pDescNode->dataType.type, pDescNode->dataType.bytes, pDescNode->slotId);
2,147,483,647✔
413
    idata.info.scale = pDescNode->dataType.scale;
2,147,483,647✔
414
    idata.info.precision = pDescNode->dataType.precision;
2,147,483,647✔
415
    idata.info.noData = pDescNode->reserve;
2,147,483,647✔
416

417
    code = blockDataAppendColInfo(pBlock, &idata);
2,147,483,647✔
418
    QUERY_CHECK_CODE(code, lino, _return);
2,147,483,647✔
419
    ++i;
2,147,483,647✔
420
  }
421

422
  if (pBlock != NULL && i != numOfCols) {
583,539,821✔
423
    code = TSDB_CODE_INVALID_PARA;
×
424
    QUERY_CHECK_CODE(code, lino, _return);
×
425
  }
426

427
  return pBlock;
583,539,821✔
428
_return:
×
429
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
430
  blockDataDestroy(pBlock);
×
431
  terrno = code;
×
432
  return NULL;
×
433
}
434

435
int32_t prepareDataBlockBuf(SSDataBlock* pDataBlock, SColMatchInfo* pMatchInfo) {
185,540,932✔
436
  SDataBlockInfo* pBlockInfo = &pDataBlock->info;
185,540,932✔
437

438
  for (int32_t i = 0; i < taosArrayGetSize(pMatchInfo->pList); ++i) {
916,337,165✔
439
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
750,533,752✔
440
    if (!pItem) {
750,343,537✔
441
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
442
      return terrno;
×
443
    }
444

445
    if (pItem->isPk) {
750,343,537✔
446
      SColumnInfoData* pInfoData = taosArrayGet(pDataBlock->pDataBlock, pItem->dstSlotId);
19,833,314✔
447
      if (!pInfoData) {
19,463,552✔
448
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
449
        return terrno;
×
450
      }
451
      pBlockInfo->pks[0].type = pInfoData->info.type;
19,463,552✔
452
      pBlockInfo->pks[1].type = pInfoData->info.type;
19,487,600✔
453

454
      // allocate enough buffer size, which is pInfoData->info.bytes
455
      if (IS_VAR_DATA_TYPE(pItem->dataType.type)) {
19,483,960✔
456
        pBlockInfo->pks[0].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
6,048,243✔
457
        if (pBlockInfo->pks[0].pData == NULL) {
6,037,755✔
458
          return terrno;
×
459
        }
460

461
        pBlockInfo->pks[1].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
6,041,996✔
462
        if (pBlockInfo->pks[1].pData == NULL) {
6,043,934✔
463
          taosMemoryFreeClear(pBlockInfo->pks[0].pData);
×
464
          return terrno;
×
465
        }
466

467
        pBlockInfo->pks[0].nData = pInfoData->info.bytes;
6,046,956✔
468
        pBlockInfo->pks[1].nData = pInfoData->info.bytes;
6,047,583✔
469
      }
470

471
      break;
19,483,604✔
472
    }
473
  }
474

475
  return TSDB_CODE_SUCCESS;
185,323,852✔
476
}
477

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

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

501
    res->translate = true;
675,150✔
502
    res->node.resType = pSColumnNode->node.resType;
675,150✔
503

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

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

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

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

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

562
  return DEAL_RES_CONTINUE;
2,363,025✔
563
}
564

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

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

575
    return TSDB_CODE_SUCCESS;
×
576
  }
577

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

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

602
    return code;
×
603
  }
604

605
  SValueNode* pValue = (SValueNode*)pNew;
337,575✔
606
  *pQualified = pValue->datum.b;
337,575✔
607

608
  nodesDestroyNode(pNew);
337,575✔
609
  return TSDB_CODE_SUCCESS;
337,575✔
610
}
611

612
static EDealRes getColumn(SNode** pNode, void* pContext) {
54,836,774✔
613
  tagFilterAssist* pData = (tagFilterAssist*)pContext;
54,836,774✔
614
  SColumnNode*     pSColumnNode = NULL;
54,836,774✔
615
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
54,846,008✔
616
    pSColumnNode = *(SColumnNode**)pNode;
18,531,593✔
617
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
36,322,228✔
618
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
722,558✔
619
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
722,924✔
620
      pData->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
678,814✔
621
      if (NULL == pSColumnNode) {
678,814✔
622
        return DEAL_RES_ERROR;
×
623
      }
624
      pSColumnNode->colId = -1;
678,814✔
625
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
678,814✔
626
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
678,814✔
627
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
678,814✔
628
      nodesDestroyNode(*pNode);
678,814✔
629
      *pNode = (SNode*)pSColumnNode;
678,814✔
630
    } else {
631
      return DEAL_RES_CONTINUE;
44,110✔
632
    }
633
  } else {
634
    return DEAL_RES_CONTINUE;
35,603,410✔
635
  }
636

637
  void* data = taosHashGet(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
19,213,249✔
638
  if (!data) {
19,204,081✔
639
    int32_t tempRes =
640
        taosHashPut(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
16,563,457✔
641
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
16,570,113✔
642
      return DEAL_RES_ERROR;
×
643
    }
644
    pSColumnNode->slotId = pData->index++;
16,570,113✔
645
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
16,570,489✔
646
                         .type = pSColumnNode->node.resType.type,
16,558,500✔
647
                         .bytes = pSColumnNode->node.resType.bytes,
16,559,908✔
648
                         .pk = pSColumnNode->isPk};
16,563,881✔
649
#if TAG_FILTER_DEBUG
650
    qDebug("tagfilter build column info, slotId:%d, colId:%d, type:%d", pSColumnNode->slotId, cInfo.colId, cInfo.type);
651
#endif
652
    void* tmp = taosArrayPush(pData->cInfoList, &cInfo);
16,567,643✔
653
    if (!tmp) {
16,574,451✔
654
      return DEAL_RES_ERROR;
×
655
    }
656
  } else {
657
    SColumnNode* col = *(SColumnNode**)data;
2,640,624✔
658
    pSColumnNode->slotId = col->slotId;
2,639,643✔
659
  }
660

661
  return DEAL_RES_CONTINUE;
19,204,501✔
662
}
663

664
static int32_t createResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
15,391,147✔
665
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
15,391,147✔
666
  if (pColumnData == NULL) {
15,390,126✔
667
    return terrno;
×
668
  }
669

670
  pColumnData->info.type = pType->type;
15,390,126✔
671
  pColumnData->info.bytes = pType->bytes;
15,392,171✔
672
  pColumnData->info.scale = pType->scale;
15,387,296✔
673
  pColumnData->info.precision = pType->precision;
15,379,874✔
674

675
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
15,384,517✔
676
  if (code != TSDB_CODE_SUCCESS) {
15,379,780✔
677
    terrno = code;
×
678
    releaseColInfoData(pColumnData);
×
679
    return terrno;
×
680
  }
681

682
  pParam->columnData = pColumnData;
15,379,780✔
683
  pParam->colAlloced = true;
15,388,771✔
684
  return TSDB_CODE_SUCCESS;
15,381,244✔
685
}
686

687
static void releaseColInfoData(void* pCol) {
4,205,431✔
688
  if (pCol) {
4,205,431✔
689
    SColumnInfoData* col = (SColumnInfoData*)pCol;
4,205,431✔
690
    colDataDestroy(col);
4,205,431✔
691
    taosMemoryFree(col);
4,204,305✔
692
  }
693
}
4,204,809✔
694

695
void freeItem(void* p) {
186,182,238✔
696
  STUidTagInfo* pInfo = p;
186,182,238✔
697
  if (pInfo->pTagVal != NULL) {
186,182,238✔
698
    taosMemoryFree(pInfo->pTagVal);
185,816,179✔
699
  }
700
}
186,182,751✔
701

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

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

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

729
    (void)memcpy(pStart, &entry->colId, sizeof(col_id_t));
20,664✔
730
    pStart += sizeof(col_id_t);
20,664✔
731

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

777
  return TSDB_CODE_SUCCESS;
10,332✔
778
}
779

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

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

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

805
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
10,332✔
806
    extractTagDataEntry((SOperatorNode*)pTagCond, pIdWithVal);
×
807
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
10,332✔
808
    SNode* pChild = NULL;
10,332✔
809
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
30,996✔
810
      extractTagDataEntry((SOperatorNode*)pChild, pIdWithVal);
20,664✔
811
    }
812
  }
813

814
  taosArraySort(pIdWithVal, compareTagDataEntry);
10,332✔
815

816
  return TSDB_CODE_SUCCESS;
10,332✔
817
}
818

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

824
  char*   payload = NULL;
10,332✔
825
  int32_t len = 0;
10,332✔
826
  int32_t code = TSDB_CODE_SUCCESS;
10,332✔
827
  int32_t lino = 0;
10,332✔
828

829
  SArray* pIdWithVal = taosArrayInit(TARRAY_MIN_SIZE, sizeof(STagDataEntry));
10,332✔
830
  code = extractTagFilterTagDataEntries(pTagCond, pIdWithVal);
10,332✔
831
  QUERY_CHECK_CODE(code, lino, _end);
10,332✔
832
  for (int32_t i = 0; i < taosArrayGetSize(pIdWithVal); ++i) {
30,996✔
833
    STagDataEntry* pEntry = taosArrayGet(pIdWithVal, i);
20,664✔
834
    len += sizeof(col_id_t) + pEntry->bytes;
20,664✔
835
  }
836
  code = buildTagDataEntryKey(pIdWithVal, &payload, len);
10,332✔
837
  QUERY_CHECK_CODE(code, lino, _end);
10,332✔
838

839
  tMD5Init(pContext);
10,332✔
840
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
10,332✔
841
  tMD5Final(pContext);
10,332✔
842

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

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

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

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

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

876
  taosMemoryFree(payload);
1,732✔
877
  return TSDB_CODE_SUCCESS;
1,732✔
878
}
879

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

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

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

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

907
int32_t qGetColumnsFromNodeList(void* data, bool isList, SArray** pColList) {
15,219,858✔
908
  int32_t code = TSDB_CODE_SUCCESS;
15,219,858✔
909
  tagFilterAssist ctx = {0};
15,219,858✔
910
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
15,219,717✔
911
  if (ctx.colHash == NULL) {
15,219,224✔
912
    code = terrno;
×
913
    goto end;
×
914
  }
915

916
  ctx.index = 0;
15,219,224✔
917
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
15,219,224✔
918
  if (ctx.cInfoList == NULL) {
15,220,060✔
919
    code = terrno;
3,396✔
920
    goto end;
×
921
  }
922

923
  if (isList) {
15,216,664✔
924
    SNode* pNode = NULL;
4,030,837✔
925
    FOREACH(pNode, (SNodeList*)data) {
8,237,270✔
926
      nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
4,206,433✔
927
      if (TSDB_CODE_SUCCESS != ctx.code) {
4,206,433✔
928
        code = ctx.code;
×
929
        goto end;
×
930
      }
931
      REPLACE_NODE(pNode);
4,206,433✔
932
    }
933
  } else {
934
    SNode* pNode = (SNode*)data;
11,185,827✔
935
    nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
11,185,677✔
936
    if (TSDB_CODE_SUCCESS != ctx.code) {
11,188,270✔
937
      code = ctx.code;
×
938
      goto end;
×
939
    }
940
  }
941
  
942
  if (pColList != NULL) *pColList = ctx.cInfoList;
15,210,612✔
943
  ctx.cInfoList = NULL;
15,213,656✔
944

945
end:
15,222,987✔
946
  taosHashCleanup(ctx.colHash);
15,213,869✔
947
  taosArrayDestroy(ctx.cInfoList);
15,197,252✔
948
  return code;
15,197,920✔
949
}
950

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

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

1031
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
66,922✔
1032
  if (rows == 0) {
66,922✔
1033
    return;
×
1034
  }
1035

1036
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
66,922✔
1037
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
66,922✔
1038

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

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

1062
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
175,209✔
1063
    SColumnInfo* tmp = (SColumnInfo*)taosArrayGet(pColList, i);
108,287✔
1064
    if (tmp != NULL && tmp->colId == -1) {
108,287✔
1065
      tbNameIndex = i;
66,922✔
1066
    }
1067
  }
1068
  
1069
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
66,922✔
1070
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
66,922✔
1071
  taosArrayDestroy(pColList);
66,922✔
1072
  if (pResBlock == NULL) {
66,922✔
1073
    code = terrno;
249✔
1074
    goto end;
249✔
1075
  }
1076

1077
  pBlockList = taosArrayInit(2, POINTER_BYTES);
66,673✔
1078
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
66,673✔
1079

1080
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
66,673✔
1081
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
66,673✔
1082

1083
  groupData = taosArrayInit(2, POINTER_BYTES);
66,673✔
1084
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
66,673✔
1085

1086
  SNode* pNode = NULL;
66,673✔
1087
  FOREACH(pNode, group) {
174,462✔
1088
    SScalarParam output = {0};
107,789✔
1089

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

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

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

1122
    if (code != TSDB_CODE_SUCCESS) {
107,789✔
1123
      releaseColInfoData(output.columnData);
×
1124
      goto end;
×
1125
    }
1126

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

1131
  for (int i = 0; i < rows; i++) {
299,973✔
1132
    gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
233,300✔
1133
    QUERY_CHECK_NULL(gInfo, code, lino, end, terrno);
233,300✔
1134

1135
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
233,300✔
1136
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
233,300✔
1137

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

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

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

1170
  if (code != TSDB_CODE_SUCCESS) {
66,922✔
1171
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
249✔
1172
  }
1173
}
1174

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

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

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

1203
    nodesFree(listNode);
×
1204

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

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

1218
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
3,963,915✔
1219
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
3,963,915✔
1220

1221
  for (int32_t i = 0; i < rows; ++i) {
26,306,259✔
1222
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
22,341,724✔
1223
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
22,341,221✔
1224
    STUidTagInfo info = {.uid = pkeyInfo->uid};
22,341,221✔
1225
    void*        tmp = taosArrayPush(pUidTagList, &info);
22,342,344✔
1226
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
22,342,344✔
1227
  }
1228

1229
  if (taosArrayGetSize(pUidTagList) > 0) {
3,964,535✔
1230
    code = pAPI->metaFn.getTableTagsByUid(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
3,963,915✔
1231
  } else {
1232
    code = pAPI->metaFn.getTableTags(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
×
1233
  }
1234
  if (code != TSDB_CODE_SUCCESS) {
3,963,915✔
1235
    goto end;
×
1236
  }
1237

1238
  SArray* pColList = NULL;
3,963,915✔
1239
  code = qGetColumnsFromNodeList(group, true, &pColList); 
3,963,915✔
1240
  if (code != TSDB_CODE_SUCCESS) {
3,963,412✔
1241
    goto end;
×
1242
  }
1243

1244
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
3,963,412✔
1245
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
3,963,412✔
1246
  taosArrayDestroy(pColList);
3,963,411✔
1247
  if (pResBlock == NULL) {
3,963,411✔
1248
    code = terrno;
×
1249
    goto end;
×
1250
  }
1251

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

1255
  pBlockList = taosArrayInit(2, POINTER_BYTES);
3,963,411✔
1256
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
3,963,751✔
1257

1258
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
3,963,411✔
1259
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
3,963,411✔
1260

1261
  groupData = taosArrayInit(2, POINTER_BYTES);
3,963,411✔
1262
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
3,963,751✔
1263

1264
  SNode* pNode = NULL;
3,962,743✔
1265
  FOREACH(pNode, group) {
8,060,385✔
1266
    SScalarParam output = {0};
4,097,478✔
1267

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

1293
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
4,096,636✔
1294
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
4,079,950✔
1295
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
4,079,950✔
1296
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
4,078,779✔
1297
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
4,078,779✔
1298
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
17,692✔
1299
      continue;
×
1300
    } else {
1301
      gTaskScalarExtra.pStreamInfo = NULL;
18,196✔
1302
      gTaskScalarExtra.pStreamRange = NULL;
18,196✔
1303
      code = scalarCalculate(pNode, pBlockList, &output, &gTaskScalarExtra);
18,196✔
1304
    }
1305

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

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

1315
  int32_t keyLen = 0;
3,962,907✔
1316
  SNode*  node;
1317
  FOREACH(node, group) {
8,058,031✔
1318
    SExprNode* pExpr = (SExprNode*)node;
4,096,471✔
1319
    keyLen += pExpr->resType.bytes;
4,096,471✔
1320
  }
1321

1322
  int32_t nullFlagSize = sizeof(int8_t) * LIST_LENGTH(group);
3,961,275✔
1323
  keyLen += nullFlagSize;
3,961,111✔
1324

1325
  keyBuf = taosMemoryCalloc(1, keyLen);
3,961,111✔
1326
  if (keyBuf == NULL) {
3,960,787✔
1327
    code = terrno;
×
1328
    goto end;
×
1329
  }
1330

1331
  if (initRemainGroups) {
3,960,787✔
1332
    pTableListInfo->remainGroups =
1,924,474✔
1333
        taosHashInit(rows, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
1,922,858✔
1334
    if (pTableListInfo->remainGroups == NULL) {
1,924,474✔
1335
      code = terrno;
×
1336
      goto end;
×
1337
    }
1338
  }
1339

1340
  for (int i = 0; i < rows; i++) {
26,301,561✔
1341
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
22,336,957✔
1342
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
22,339,319✔
1343

1344
    if (groupIdMap != NULL){
22,339,319✔
1345
      gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
62,340✔
1346
    }
1347
    
1348
    char* isNull = (char*)keyBuf;
22,339,319✔
1349
    char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(group);
22,339,319✔
1350
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
45,747,913✔
1351
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
23,410,738✔
1352

1353
      if (groupIdMap != NULL && gInfo != NULL) {
23,409,730✔
1354
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
74,358✔
1355
        if (ret != TSDB_CODE_SUCCESS) {
74,358✔
1356
          qError("buildGroupInfo failed at line %d since %s", __LINE__, tstrerror(ret));
×
1357
          taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1358
          gInfo = NULL;
×
1359
        }
1360
      }
1361
      
1362
      if (colDataIsNull_s(pValue, i)) {
46,818,956✔
1363
        isNull[j] = 1;
96,978✔
1364
      } else {
1365
        isNull[j] = 0;
23,312,248✔
1366
        char* data = colDataGetData(pValue, i);
23,312,650✔
1367
        if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
23,313,298✔
1368
          // if (tTagIsJson(data)) {
1369
          //   code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
1370
          //   goto end;
1371
          // }
1372
          if (tTagIsJsonNull(data)) {
74,491✔
1373
            isNull[j] = 1;
×
1374
            continue;
×
1375
          }
1376
          int32_t len = getJsonValueLen(data);
74,491✔
1377
          memcpy(pStart, data, len);
74,491✔
1378
          pStart += len;
74,491✔
1379
        } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
23,236,545✔
1380
          if (IS_STR_DATA_BLOB(pValue->info.type)) {
20,343,797✔
1381
            if (blobDataTLen(data) > TSDB_MAX_BLOB_LEN) {
4,253✔
1382
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1383
              goto end;
×
1384
            }
1385
            memcpy(pStart, data, blobDataTLen(data));
×
1386
            pStart += blobDataTLen(data);
×
1387
          } else {
1388
            if (varDataTLen(data) > pValue->info.bytes) {
20,341,903✔
1389
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1390
              goto end;
×
1391
            }
1392
            memcpy(pStart, data, varDataTLen(data));
20,344,141✔
1393
            pStart += varDataTLen(data);
20,344,644✔
1394
          }
1395
        } else {
1396
          memcpy(pStart, data, pValue->info.bytes);
2,895,740✔
1397
          pStart += pValue->info.bytes;
2,895,740✔
1398
        }
1399
      }
1400
    }
1401

1402
    int32_t len = (int32_t)(pStart - (char*)keyBuf);
22,335,287✔
1403
    info->groupId = calcGroupId(keyBuf, len);
22,335,287✔
1404
    if (groupIdMap != NULL && gInfo != NULL) {
22,341,998✔
1405
      int32_t ret = taosHashPut(groupIdMap, &info->groupId, sizeof(info->groupId), &gInfo, POINTER_BYTES);
62,340✔
1406
      if (ret != TSDB_CODE_SUCCESS) {
62,340✔
1407
        qError("put groupid to map failed at line %d since %s", __LINE__, tstrerror(ret));
×
1408
        taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1409
      }
1410
      qDebug("put groupid to map gid:%" PRIu64, info->groupId);
62,340✔
1411
      gInfo = NULL;
62,340✔
1412
    }
1413
    if (initRemainGroups) {
22,341,998✔
1414
      // groupId ~ table uid
1415
      code = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId), &(info->uid),
10,984,867✔
1416
                         sizeof(info->uid));
1417
      if (code == TSDB_CODE_DUP_KEY) {
10,981,687✔
1418
        code = TSDB_CODE_SUCCESS;
752,470✔
1419
      }
1420
      QUERY_CHECK_CODE(code, lino, end);
10,981,687✔
1421
    }
1422
  }
1423

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

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

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

1437
end:
3,964,604✔
1438
  taosMemoryFreeClear(keyBuf);
3,963,915✔
1439
  blockDataDestroy(pResBlock);
3,963,411✔
1440
  taosArrayDestroy(pBlockList);
3,962,789✔
1441
  taosArrayDestroyEx(pUidTagList, freeItem);
3,962,791✔
1442
  taosArrayDestroyP(groupData, releaseColInfoData);
3,961,938✔
1443
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
3,963,411✔
1444

1445
  if (code != TSDB_CODE_SUCCESS) {
3,963,411✔
1446
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1447
  }
1448
  return code;
3,962,907✔
1449
}
1450

1451
static int32_t nameComparFn(const void* p1, const void* p2) {
1,435,199✔
1452
  const char* pName1 = *(const char**)p1;
1,435,199✔
1453
  const char* pName2 = *(const char**)p2;
1,435,199✔
1454

1455
  int32_t ret = strcmp(pName1, pName2);
1,434,569✔
1456
  if (ret == 0) {
1,434,569✔
1457
    return 0;
15,846✔
1458
  } else {
1459
    return (ret > 0) ? 1 : -1;
1,418,723✔
1460
  }
1461
}
1462

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

1469
  SArray* pTbList = taosArrayInit(len, POINTER_BYTES);
747,207✔
1470
  QUERY_CHECK_NULL(pTbList, code, lino, _end, terrno);
746,892✔
1471

1472
  for (int i = 0; i < pList->pNodeList->length; i++) {
2,156,430✔
1473
    SValueNode* valueNode = (SValueNode*)cell->pNode;
1,408,908✔
1474
    if (!IS_VAR_DATA_TYPE(valueNode->node.resType.type)) {
1,409,223✔
1475
      terrno = TSDB_CODE_INVALID_PARA;
×
1476
      taosArrayDestroy(pTbList);
×
1477
      return NULL;
×
1478
    }
1479

1480
    char* name = varDataVal(valueNode->datum.p);
1,408,908✔
1481
    void* tmp = taosArrayPush(pTbList, &name);
1,409,223✔
1482
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,409,223✔
1483
    cell = cell->pNext;
1,409,223✔
1484
  }
1485

1486
  size_t numOfTables = taosArrayGetSize(pTbList);
747,207✔
1487

1488
  // order the name
1489
  taosArraySort(pTbList, nameComparFn);
747,207✔
1490

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

1499
  for (int32_t i = 1; i < numOfTables; ++i) {
1,409,538✔
1500
    char** name = taosArrayGetLast(pNewList);
662,331✔
1501
    char** nameInOldList = taosArrayGet(pTbList, i);
662,331✔
1502
    QUERY_CHECK_NULL(nameInOldList, code, lino, _end, terrno);
662,016✔
1503
    if (strcmp(*name, *nameInOldList) == 0) {
662,016✔
1504
      continue;
8,548✔
1505
    }
1506

1507
    tmp = taosArrayPush(pNewList, nameInOldList);
653,783✔
1508
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
653,783✔
1509
  }
1510

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

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

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

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

1531
static int32_t filterTableInfoCompare(const void* a, const void* b) {
16,385,620✔
1532
  STUidTagInfo* p1 = (STUidTagInfo*)a;
16,385,620✔
1533
  STUidTagInfo* p2 = (STUidTagInfo*)b;
16,385,620✔
1534

1535
  if (p1->uid == p2->uid) {
16,385,620✔
1536
    return 0;
×
1537
  }
1538

1539
  return p1->uid < p2->uid ? -1 : 1;
16,385,620✔
1540
}
1541

1542
static FilterCondType checkTagCond(SNode* cond) {
12,858,628✔
1543
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
12,858,628✔
1544
    return FILTER_NO_LOGIC;
9,312,478✔
1545
  }
1546
  if (nodeType(cond) == QUERY_NODE_LOGIC_CONDITION && ((SLogicConditionNode*)cond)->condType == LOGIC_COND_TYPE_AND) {
3,546,962✔
1547
    return FILTER_AND;
3,048,180✔
1548
  }
1549
  return FILTER_OTHER;
495,462✔
1550
}
1551

1552
static bool optimizeTbnameInCond(void* pVnode, int64_t suid, SArray* list, SNode* cond, SStorageAPI* pAPI) {
13,523,483✔
1553
  int32_t code = 0;
13,523,483✔
1554
  int32_t ntype = nodeType(cond);
13,523,483✔
1555

1556
  if (ntype == QUERY_NODE_OPERATOR) {
13,526,533✔
1557
    code = optimizeTbnameInCondImpl(pVnode, list, cond, pAPI, suid);
9,968,969✔
1558
    return code == 0;
9,968,298✔
1559
  }
1560
  if (ntype != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
3,557,564✔
1561
    return false;
495,805✔
1562
  }
1563

1564
  bool                 hasTbnameCond = false;
3,059,789✔
1565
  SLogicConditionNode* pNode = (SLogicConditionNode*)cond;
3,059,789✔
1566
  SNodeList*           pList = (SNodeList*)pNode->pParameterList;
3,059,789✔
1567

1568
  int32_t len = LIST_LENGTH(pList);
3,058,133✔
1569
  if (len <= 0) {
3,058,493✔
1570
    return false;
×
1571
  }
1572

1573
  SListCell* cell = pList->pHead;
3,058,493✔
1574
  for (int i = 0; i < len; i++) {
9,547,540✔
1575
    if (cell == NULL) break;
6,488,596✔
1576
    if (optimizeTbnameInCondImpl(pVnode, list, cell->pNode, pAPI, suid) == 0) {
6,488,596✔
1577
      hasTbnameCond = true;
7,553✔
1578
      break;
7,553✔
1579
    }
1580
    cell = cell->pNext;
6,485,919✔
1581
  }
1582

1583
  taosArraySort(list, filterTableInfoCompare);
3,066,497✔
1584
  taosArrayRemoveDuplicate(list, filterTableInfoCompare, NULL);
3,060,006✔
1585

1586
  if (hasTbnameCond) {
3,057,555✔
1587
    code = pAPI->metaFn.getTableTagsByUid(pVnode, suid, list);
7,553✔
1588
    return code == 0;
7,553✔
1589
  }
1590

1591
  return false;
3,050,002✔
1592
}
1593

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

1601
  SOperatorNode* pNode = (SOperatorNode*)pTagCond;
16,461,655✔
1602
  if (pNode->opType != OP_TYPE_IN) {
16,461,655✔
1603
    return -1;
14,447,880✔
1604
  }
1605

1606
  if ((pNode->pLeft != NULL && ((nodeType(pNode->pLeft) == QUERY_NODE_FUNCTION &&
2,013,430✔
1607
                                 ((SFunctionNode*)pNode->pLeft)->funcType == FUNCTION_TYPE_TBNAME)) ||
748,603✔
1608
       (nodeType(pNode->pLeft) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pLeft)->colType == COLUMN_TYPE_TBNAME)) &&
1,265,865✔
1609
      (pNode->pRight != NULL && nodeType(pNode->pRight) == QUERY_NODE_NODE_LIST)) {
745,988✔
1610
    SNodeListNode* pList = (SNodeListNode*)pNode->pRight;
747,207✔
1611

1612
    int32_t len = LIST_LENGTH(pList->pNodeList);
747,207✔
1613
    if (len <= 0) {
746,577✔
1614
      return -1;
×
1615
    }
1616

1617
    SArray*   pTbList = getTableNameList(pList);
746,577✔
1618
    int32_t   numOfTables = taosArrayGetSize(pTbList);
746,577✔
1619
    SHashObj* uHash = NULL;
746,577✔
1620

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

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

1643
    for (int i = 0; i < numOfTables; i++) {
1,982,035✔
1644
      char* name = taosArrayGetP(pTbList, i);
1,316,464✔
1645

1646
      uint64_t uid = 0, csuid = 0;
1,316,779✔
1647
      if (pStoreAPI->metaFn.getTableUidByName(pVnode, name, &uid) == 0) {
1,317,094✔
1648
        ETableType tbType = TSDB_TABLE_MAX;
440,941✔
1649
        if (pStoreAPI->metaFn.getTableTypeSuidByName(pVnode, name, &tbType, &csuid) == 0 &&
440,941✔
1650
            tbType == TSDB_CHILD_TABLE) {
440,941✔
1651
          if (suid != csuid) {
359,305✔
1652
            continue;
2,012✔
1653
          }
1654
          if (NULL == uHash || taosHashGet(uHash, &uid, sizeof(uid)) == NULL) {
357,293✔
1655
            STUidTagInfo s = {.uid = uid, .name = name, .pTagVal = NULL};
356,285✔
1656
            void*        tmp = taosArrayPush(pExistedUidList, &s);
356,285✔
1657
            if (!tmp) {
356,285✔
1658
              return terrno;
×
1659
            }
1660
          }
1661
        } else {
1662
          taosArrayDestroy(pTbList);
81,636✔
1663
          taosHashCleanup(uHash);
81,636✔
1664
          return -1;
81,636✔
1665
        }
1666
      } else {
1667
        //        qWarn("failed to get tableIds from by table name: %s, reason: %s", name, tstrerror(terrno));
1668
        terrno = 0;
875,838✔
1669
      }
1670
    }
1671

1672
    taosHashCleanup(uHash);
665,571✔
1673
    taosArrayDestroy(pTbList);
665,571✔
1674
    return 0;
665,571✔
1675
  }
1676

1677
  return -1;
1,266,032✔
1678
}
1679

1680
SSDataBlock* createTagValBlockForFilter(SArray* pColList, int32_t numOfTables, SArray* pUidTagList, void* pVnode,
16,786,751✔
1681
                                        SStorageAPI* pStorageAPI) {
1682
  int32_t      code = TSDB_CODE_SUCCESS;
16,786,751✔
1683
  int32_t      lino = 0;
16,786,751✔
1684
  SSDataBlock* pResBlock = NULL;
16,786,751✔
1685
  code = createDataBlock(&pResBlock);
16,791,176✔
1686
  QUERY_CHECK_CODE(code, lino, _end);
16,781,555✔
1687

1688
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
34,921,762✔
1689
    SColumnInfoData colInfo = {0};
18,139,506✔
1690
    void*           tmp = taosArrayGet(pColList, i);
18,135,387✔
1691
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
18,118,461✔
1692
    colInfo.info = *(SColumnInfo*)tmp;
18,118,461✔
1693
    code = blockDataAppendColInfo(pResBlock, &colInfo);
18,129,465✔
1694
    QUERY_CHECK_CODE(code, lino, _end);
18,140,534✔
1695
  }
1696

1697
  code = blockDataEnsureCapacity(pResBlock, numOfTables);
16,783,841✔
1698
  if (code != TSDB_CODE_SUCCESS) {
16,782,152✔
1699
    terrno = code;
×
1700
    blockDataDestroy(pResBlock);
×
1701
    return NULL;
×
1702
  }
1703

1704
  pResBlock->info.rows = numOfTables;
16,782,152✔
1705

1706
  int32_t numOfCols = taosArrayGetSize(pResBlock->pDataBlock);
16,789,055✔
1707

1708
  for (int32_t i = 0; i < numOfTables; i++) {
205,268,509✔
1709
    STUidTagInfo* p1 = taosArrayGet(pUidTagList, i);
188,472,752✔
1710
    QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
188,487,476✔
1711

1712
    for (int32_t j = 0; j < numOfCols; j++) {
385,525,464✔
1713
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, j);
197,000,683✔
1714
      QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
196,950,368✔
1715

1716
      if (pColInfo->info.colId == -1) {  // tbname
196,950,368✔
1717
        char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
7,272,797✔
1718
        if (p1->name != NULL) {
7,274,151✔
1719
          STR_TO_VARSTR(str, p1->name);
356,285✔
1720
        } else {  // name is not retrieved during filter
1721
          code = pStorageAPI->metaFn.getTableNameByUid(pVnode, p1->uid, str);
6,922,427✔
1722
          QUERY_CHECK_CODE(code, lino, _end);
6,914,896✔
1723
        }
1724

1725
        code = colDataSetVal(pColInfo, i, str, false);
7,270,932✔
1726
        QUERY_CHECK_CODE(code, lino, _end);
7,277,958✔
1727
#if TAG_FILTER_DEBUG
1728
        qDebug("tagfilter uid:%ld, tbname:%s", *uid, str + 2);
1729
#endif
1730
      } else {
1731
        STagVal tagVal = {0};
189,703,229✔
1732
        tagVal.cid = pColInfo->info.colId;
189,692,034✔
1733
        if (p1->pTagVal == NULL) {
189,714,716✔
1734
          colDataSetNULL(pColInfo, i);
4,100✔
1735
        } else {
1736
          const char* p = pStorageAPI->metaFn.extractTagVal(p1->pTagVal, pColInfo->info.type, &tagVal);
189,664,419✔
1737

1738
          if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
189,793,803✔
1739
            colDataSetNULL(pColInfo, i);
1,884,227✔
1740
          } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
187,913,969✔
1741
            code = colDataSetVal(pColInfo, i, p, false);
588,898✔
1742
            QUERY_CHECK_CODE(code, lino, _end);
588,898✔
1743
          } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
314,830,541✔
1744
            if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
127,576,286✔
1745
              QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
1746
            }
1747
            char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
127,569,792✔
1748
            QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
127,537,519✔
1749
            varDataSetLen(tmp, tagVal.nData);
127,537,519✔
1750
            memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
127,531,901✔
1751
            code = colDataSetVal(pColInfo, i, tmp, false);
127,535,602✔
1752
#if TAG_FILTER_DEBUG
1753
            qDebug("tagfilter varch:%s", tmp + 2);
1754
#endif
1755
            taosMemoryFree(tmp);
127,574,679✔
1756
            QUERY_CHECK_CODE(code, lino, _end);
127,541,608✔
1757
          } else {
1758
            code = colDataSetVal(pColInfo, i, (const char*)&tagVal.i64, false);
59,746,628✔
1759
            QUERY_CHECK_CODE(code, lino, _end);
59,761,211✔
1760
#if TAG_FILTER_DEBUG
1761
            if (pColInfo->info.type == TSDB_DATA_TYPE_INT) {
1762
              qDebug("tagfilter int:%d", *(int*)(&tagVal.i64));
1763
            } else if (pColInfo->info.type == TSDB_DATA_TYPE_DOUBLE) {
1764
              qDebug("tagfilter double:%f", *(double*)(&tagVal.i64));
1765
            }
1766
#endif
1767
          }
1768
        }
1769
      }
1770
    }
1771
  }
1772

1773
_end:
16,795,876✔
1774
  if (code != TSDB_CODE_SUCCESS) {
16,796,006✔
1775
    blockDataDestroy(pResBlock);
761✔
1776
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
249✔
1777
    terrno = code;
249✔
1778
    return NULL;
249✔
1779
  }
1780
  return pResBlock;
16,795,245✔
1781
}
1782

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

1787
  STableKeyInfo info = {.uid = 0, .groupId = 0};
11,161,030✔
1788
  int32_t       numOfTables = taosArrayGetSize(pUidTagList);
11,163,659✔
1789
  for (int32_t i = 0; i < numOfTables; ++i) {
174,738,778✔
1790
    if (pResultList[i]) {
163,554,098✔
1791
      STUidTagInfo* tmpTag = (STUidTagInfo*)taosArrayGet(pUidTagList, i);
74,901,227✔
1792
      if (!tmpTag) {
74,902,100✔
1793
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1794
        return terrno;
×
1795
      }
1796
      uint64_t uid = tmpTag->uid;
74,902,100✔
1797
      qDebug("tagfilter get uid:%" PRId64 ", res:%d", uid, pResultList[i]);
74,902,093✔
1798

1799
      info.uid = uid;
74,924,021✔
1800
      //qInfo("doSetQualifiedUid row:%d added to pTableList", i);
1801
      void* p = taosArrayPush(pListInfo->pTableList, &info);
74,924,021✔
1802
      if (p == NULL) {
74,918,439✔
1803
        return terrno;
×
1804
      }
1805

1806
      if (addUid) {
74,918,439✔
1807
        //qInfo("doSetQualifiedUid row:%d added to pUidList", i);
1808
        void* tmp = taosArrayPush(pUidList, &uid);
10,105✔
1809
        if (tmp == NULL) {
10,105✔
1810
          return terrno;
×
1811
        }
1812
      }
1813
    } else {
1814
      //qInfo("doSetQualifiedUid row:%d failed", i);
1815
    }
1816
  }
1817

1818
  return TSDB_CODE_SUCCESS;
11,184,680✔
1819
}
1820

1821
static int32_t copyExistedUids(SArray* pUidTagList, const SArray* pUidList) {
13,525,504✔
1822
  int32_t code = TSDB_CODE_SUCCESS;
13,525,504✔
1823
  int32_t numOfExisted = taosArrayGetSize(pUidList);
13,525,504✔
1824
  if (numOfExisted == 0) {
13,525,532✔
1825
    return code;
10,990,022✔
1826
  }
1827

1828
  for (int32_t i = 0; i < numOfExisted; ++i) {
28,884,721✔
1829
    uint64_t* uid = taosArrayGet(pUidList, i);
26,349,222✔
1830
    if (!uid) {
26,349,581✔
1831
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1832
      return terrno;
×
1833
    }
1834
    STUidTagInfo info = {.uid = *uid};
26,349,581✔
1835
    void*        tmp = taosArrayPush(pUidTagList, &info);
26,349,211✔
1836
    if (!tmp) {
26,349,211✔
1837
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1838
      return code;
×
1839
    }
1840
  }
1841
  return code;
2,535,499✔
1842
}
1843

1844
int32_t doFilterByTagCond(STableListInfo* pListInfo, SArray* pUidList, SNode* pTagCond, void* pVnode,
89,443,041✔
1845
                                 SIdxFltStatus status, SStorageAPI* pAPI, bool addUid, bool* listAdded, void* pStreamInfo) {
1846
  *listAdded = false;
89,443,041✔
1847
  if (pTagCond == NULL) {
89,448,023✔
1848
    return TSDB_CODE_SUCCESS;
75,920,863✔
1849
  }
1850

1851
  terrno = TSDB_CODE_SUCCESS;
13,527,160✔
1852

1853
  int32_t      lino = 0;
13,523,469✔
1854
  int32_t      code = TSDB_CODE_SUCCESS;
13,523,469✔
1855
  SArray*      pBlockList = NULL;
13,523,469✔
1856
  SSDataBlock* pResBlock = NULL;
13,523,469✔
1857
  SScalarParam output = {0};
13,524,097✔
1858
  SArray*      pUidTagList = NULL;
13,521,452✔
1859

1860
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
13,521,452✔
1861

1862
  //  int64_t stt = taosGetTimestampUs();
1863
  pUidTagList = taosArrayInit(10, sizeof(STUidTagInfo));
13,515,724✔
1864
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
13,521,276✔
1865

1866
  code = copyExistedUids(pUidTagList, pUidList);
13,521,276✔
1867
  QUERY_CHECK_CODE(code, lino, end);
13,515,877✔
1868

1869
  // Narrow down the scope of the tablelist set if there is tbname in condition and And Logical operator
1870
  bool narrowed = optimizeTbnameInCond(pVnode, pListInfo->idInfo.suid, pUidTagList, pTagCond, pAPI);
13,515,877✔
1871
  if (narrowed) {  // tbname in filter is activated, do nothing and return
13,522,173✔
1872
    taosArrayClear(pUidList);
665,571✔
1873

1874
    int32_t numOfRows = taosArrayGetSize(pUidTagList);
665,571✔
1875
    code = taosArrayEnsureCap(pUidList, numOfRows);
665,571✔
1876
    QUERY_CHECK_CODE(code, lino, end);
665,571✔
1877

1878
    for (int32_t i = 0; i < numOfRows; ++i) {
3,034,832✔
1879
      STUidTagInfo* pInfo = taosArrayGet(pUidTagList, i);
2,369,261✔
1880
      QUERY_CHECK_NULL(pInfo, code, lino, end, terrno);
2,369,261✔
1881
      void* tmp = taosArrayPush(pUidList, &pInfo->uid);
2,369,261✔
1882
      QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
2,369,261✔
1883
    }
1884
    terrno = 0;
665,571✔
1885
  } else {
1886
    qDebug("pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
12,856,602✔
1887

1888
    FilterCondType condType = checkTagCond(pTagCond);
12,856,602✔
1889
    if (((condType == FILTER_NO_LOGIC || condType == FILTER_AND) && status != SFLT_NOT_INDEX) || // (super table) use tagIndex and operator is and
12,853,008✔
1890
        (status == SFLT_NOT_INDEX && taosArrayGetSize(pUidTagList) > 0)) {                       // (child table with tagCond)
9,987,160✔
1891
      code = pAPI->metaFn.getTableTagsByUid(pVnode, pListInfo->idInfo.suid, pUidTagList);
3,121,221✔
1892
    } else {
1893
      taosArrayClear(pUidTagList);        // clear tablelist if using tagIndex and or condition
9,730,115✔
1894
      code = pAPI->metaFn.getTableTags(pVnode, pListInfo->idInfo.suid, pUidTagList);
9,737,674✔
1895
    }
1896
    if (code != TSDB_CODE_SUCCESS) {
12,862,205✔
1897
      qError("failed to get table tags from meta, reason:%s, suid:%" PRIu64, tstrerror(code), pListInfo->idInfo.suid);
×
1898
      terrno = code;
×
1899
      QUERY_CHECK_CODE(code, lino, end);
×
1900
    }
1901
  }
1902

1903
  qDebug("final pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
13,527,776✔
1904

1905
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
13,530,294✔
1906
  if (numOfTables == 0) {
13,528,867✔
1907
    goto end;
2,340,098✔
1908
  }
1909

1910
  SArray* pColList = NULL;
11,188,769✔
1911
  code = qGetColumnsFromNodeList(pTagCond, false, &pColList); 
11,189,021✔
1912
  if (code != TSDB_CODE_SUCCESS) {
11,167,036✔
1913
    goto end;
×
1914
  }
1915
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
11,167,036✔
1916
  taosArrayDestroy(pColList);
11,184,520✔
1917
  if (pResBlock == NULL) {
11,181,708✔
1918
    code = terrno;
×
1919
    QUERY_CHECK_CODE(code, lino, end);
×
1920
  }
1921

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

1924
  //  int64_t st1 = taosGetTimestampUs();
1925
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1926
  pBlockList = taosArrayInit(2, POINTER_BYTES);
11,181,708✔
1927
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
11,183,613✔
1928

1929
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
11,187,413✔
1930
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
11,187,413✔
1931

1932
  code = createResultData(&type, numOfTables, &output);
11,187,413✔
1933
  if (code != TSDB_CODE_SUCCESS) {
11,181,497✔
1934
    terrno = code;
×
1935
    QUERY_CHECK_CODE(code, lino, end);
×
1936
  }
1937

1938
  gTaskScalarExtra.pStreamInfo = pStreamInfo;
11,181,497✔
1939
  gTaskScalarExtra.pStreamRange = NULL;
11,181,497✔
1940
  code = scalarCalculate(pTagCond, pBlockList, &output, &gTaskScalarExtra);
11,168,561✔
1941
  if (code != TSDB_CODE_SUCCESS) {
11,159,140✔
1942
    qError("failed to calculate scalar, reason:%s", tstrerror(code));
914✔
1943
    terrno = code;
914✔
1944
    QUERY_CHECK_CODE(code, lino, end);
914✔
1945
  }
1946

1947
  code = doSetQualifiedUid(pListInfo, pUidList, pUidTagList, (bool*)output.columnData->pData, addUid);
11,158,226✔
1948
  if (code != TSDB_CODE_SUCCESS) {
11,182,673✔
1949
    terrno = code;
×
1950
    QUERY_CHECK_CODE(code, lino, end);
×
1951
  }
1952
  *listAdded = true;
11,182,673✔
1953

1954
end:
13,523,741✔
1955
  if (code != TSDB_CODE_SUCCESS) {
13,528,806✔
1956
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
914✔
1957
  }
1958
  blockDataDestroy(pResBlock);
13,528,806✔
1959
  taosArrayDestroy(pBlockList);
13,508,498✔
1960
  taosArrayDestroyEx(pUidTagList, freeItem);
13,504,721✔
1961

1962
  colDataDestroy(output.columnData);
13,524,903✔
1963
  taosMemoryFreeClear(output.columnData);
13,517,221✔
1964
  return code;
13,518,239✔
1965
}
1966

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

1972
static EDealRes replacePlaceHolderColumn(SNode** pNode, void* pContext) {
76,032✔
1973
  PlaceHolderContext* pData = (PlaceHolderContext*)pContext;
76,032✔
1974
  if (QUERY_NODE_FUNCTION != nodeType((*pNode))) {
76,032✔
1975
    return DEAL_RES_CONTINUE;
62,656✔
1976
  }
1977
  SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
13,376✔
1978
  if (!fmIsStreamPesudoColVal(pFuncNode->funcId)) {
13,376✔
1979
    return DEAL_RES_CONTINUE;
496✔
1980
  }
1981
  pData->code = fmSetStreamPseudoFuncParamVal(pFuncNode->funcId, pFuncNode->pParameterList, pData->pStreamRuntimeInfo);
12,880✔
1982
  if (pData->code != TSDB_CODE_SUCCESS) {
12,880✔
1983
    return DEAL_RES_ERROR;
×
1984
  }
1985
  SNode* pFirstParam = nodesListGetNode(pFuncNode->pParameterList, 0);
12,880✔
1986
  ((SValueNode*)pFirstParam)->translate = true;
12,880✔
1987
  SValueNode* res = NULL;
12,880✔
1988
  pData->code = nodesCloneNode(pFirstParam, (SNode**)&res);
12,880✔
1989
  if (NULL == res) {
12,880✔
1990
    return DEAL_RES_ERROR;
×
1991
  }
1992
  nodesDestroyNode(*pNode);
12,880✔
1993
  *pNode = (SNode*)res;
12,880✔
1994

1995
  return DEAL_RES_CONTINUE;
12,880✔
1996
}
1997

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

2004
  col_id_t colId = pColNode->colId;
20,664✔
2005
  void* _tmp = taosArrayPush(pColIdArray, &colId);
20,664✔
2006
}
20,664✔
2007

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

2021
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
10,332✔
2022
    extractTagColId((SOperatorNode*)pTagCond, *pTagColIds);
×
2023
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
10,332✔
2024
    SNode* pChild = NULL;
10,332✔
2025
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
30,996✔
2026
      extractTagColId((SOperatorNode*)pChild, *pTagColIds);
20,664✔
2027
    }
2028
  }
2029

2030
  taosArraySort(*pTagColIds, compareUint16Val);
10,332✔
2031

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

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

2057
static EDealRes canOptimizeTagCondFilter(SNode* pTagCond, void* pContext) {
96,104✔
2058
  if (NULL == pTagCond) {
96,104✔
2059
    *(bool*)pContext = false;
×
2060
    return DEAL_RES_END;
×
2061
  }
2062
  if (nodeType(pTagCond) == QUERY_NODE_VALUE ||
96,104✔
2063
    nodeType(pTagCond) == QUERY_NODE_COLUMN) {
63,796✔
2064
    return DEAL_RES_CONTINUE;
52,972✔
2065
  }
2066
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR &&
43,132✔
2067
    ((SOperatorNode*)pTagCond)->opType == OP_TYPE_EQUAL) {
21,156✔
2068
    return DEAL_RES_CONTINUE;
20,664✔
2069
  }
2070
  if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION &&
22,468✔
2071
    ((SLogicConditionNode*)pTagCond)->condType == LOGIC_COND_TYPE_AND) {
10,332✔
2072
    return DEAL_RES_CONTINUE;
10,332✔
2073
  }
2074
  if (nodeType(pTagCond) == QUERY_NODE_FUNCTION &&
23,780✔
2075
    fmIsStreamPesudoColVal(((SFunctionNode*)pTagCond)->funcId)) {
11,644✔
2076
    return DEAL_RES_CONTINUE;
11,644✔
2077
  }
2078
  *(bool*)pContext = false;
492✔
2079
  return DEAL_RES_END;
492✔
2080
}
2081

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

2089
  pListInfo->idInfo.suid = pScanNode->suid;
188,568,462✔
2090
  pListInfo->idInfo.tableType = pScanNode->tableType;
188,569,281✔
2091

2092
  SArray* pUidList = taosArrayInit(8, sizeof(uint64_t));
188,307,737✔
2093
  QUERY_CHECK_NULL(pUidList, code, lino, _error, terrno);
188,408,370✔
2094

2095
  char*   pTagCondKey = NULL;
188,408,370✔
2096
  int32_t tagCondKeyLen;
188,464,423✔
2097
  SArray* pTagColIds = NULL;
188,522,942✔
2098
  char*   pPayload = NULL;
188,531,558✔
2099
  qTrace("getTableList called, suid:%" PRIu64
188,531,558✔
2100
    ", tagCond:%p, tagIndexCond:%p, %d %d", pScanNode->suid, pTagCond,
2101
    pTagIndexCond, pScanNode->tableType, pScanNode->virtualStableScan);
2102
  if (pScanNode->tableType != TSDB_SUPER_TABLE && !pScanNode->virtualStableScan) {
188,534,148✔
2103
    pListInfo->idInfo.uid = pScanNode->uid;
76,127,265✔
2104
    if (pStorageAPI->metaFn.isTableExisted(pVnode, pScanNode->uid)) {
76,094,034✔
2105
      void* tmp = taosArrayPush(pUidList, &pScanNode->uid);
76,103,296✔
2106
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
76,101,709✔
2107
    }
2108
    code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, SFLT_NOT_INDEX, pStorageAPI, false, &listAdded, pStreamInfo);
76,147,149✔
2109
    QUERY_CHECK_CODE(code, lino, _end);
76,138,415✔
2110
  } else {
2111
    bool      isStream = (pStreamInfo != NULL);
112,488,488✔
2112
    bool      hasTagCond = (pTagCond != NULL);
112,488,488✔
2113
    bool      canCacheTagEqCondFilter = false;
112,488,488✔
2114
    T_MD5_CTX context = {0};
112,435,659✔
2115

2116
    qTrace("start to get table list by tag filter, suid:%" PRIu64
112,489,771✔
2117
      ",tsStableTagFilterCache:%d, tsTagFilterCache:%d", 
2118
      pScanNode->suid, tsStableTagFilterCache, tsTagFilterCache);
2119

2120
    bool acquired = false;
112,489,771✔
2121
    // first, check whether we can use stable tag filter cache
2122
    if (tsStableTagFilterCache && isStream && hasTagCond) {
112,443,588✔
2123
      canCacheTagEqCondFilter = true;
10,824✔
2124
      nodesWalkExpr(pTagCond, canOptimizeTagCondFilter,
10,824✔
2125
        (void*)&canCacheTagEqCondFilter);
2126
    }
2127
    if (canCacheTagEqCondFilter) {
112,343,910✔
2128
      qDebug("%s, stable tag filter condition can be optimized", idstr);
10,332✔
2129
      if (((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
10,332✔
2130
        SNode* tmp = NULL;
10,332✔
2131
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
10,332✔
2132
        QUERY_CHECK_CODE(code, lino, _error);
10,332✔
2133

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

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

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

2198
    if (!pTagCond) {  // no tag filter condition exists, let's fetch all tables of this super table
112,513,824✔
2199
      code = pStorageAPI->metaFn.getChildTableList(pVnode, pScanNode->suid, pUidList);
99,245,326✔
2200
      QUERY_CHECK_CODE(code, lino, _error);
99,211,509✔
2201
      qTrace("no tag filter, get all child tables, numOfTables:%d", (int32_t)taosArrayGetSize(pUidList));
99,211,509✔
2202
    } else {
2203
      SIdxFltStatus status = SFLT_NOT_INDEX;
13,268,498✔
2204
      if (pTagIndexCond) {
13,269,677✔
2205
        void* pIndex = pStorageAPI->metaFn.getInvertIndex(pVnode);
3,930,625✔
2206

2207
        SIndexMetaArg metaArg = {.metaEx = pVnode,
3,930,649✔
2208
                                 .idx = pStorageAPI->metaFn.storeGetIndexInfo(pVnode),
3,930,625✔
2209
                                 .ivtIdx = pIndex,
2210
                                 .suid = pScanNode->uid};
3,930,625✔
2211
        code = doFilterTag(pTagIndexCond, &metaArg, pUidList, &status, &pStorageAPI->metaFilter);
3,930,625✔
2212
        if (code != 0 || status == SFLT_NOT_INDEX) {  // temporarily disable it for performance sake
3,920,971✔
2213
          qDebug("failed to get tableIds from index, suid:%" PRIu64 ", uidListSize:%d", pScanNode->uid, (int32_t)taosArrayGetSize(pUidList));
1,054,245✔
2214
        } else {
2215
          qDebug("succ to get filter result, table num: %d", (int)taosArrayGetSize(pUidList));
2,866,726✔
2216
        }
2217
      }
2218
      qTrace("after index filter, pTagCond:%p uidListSize:%d", pTagCond, (int32_t)taosArrayGetSize(pUidList));
13,269,677✔
2219
      code = doFilterByTagCond(pListInfo, pUidList, pTagCond, pVnode, status,
13,270,606✔
2220
        pStorageAPI, tsTagFilterCache || tsStableTagFilterCache,
13,270,606✔
2221
        &listAdded, pStreamInfo);
2222
      QUERY_CHECK_CODE(code, lino, _error);
13,260,758✔
2223
    }
2224
    // let's add the filter results into meta-cache
2225
    numOfTables = taosArrayGetSize(pUidList);
112,479,573✔
2226

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

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

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

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

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

2278

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

2284
_end:
188,605,583✔
2285
  if (!listAdded) {
188,683,656✔
2286
    numOfTables = taosArrayGetSize(pUidList);
177,455,374✔
2287
    for (int i = 0; i < numOfTables; i++) {
686,429,832✔
2288
      void* tmp = taosArrayGet(pUidList, i);
508,968,229✔
2289
      QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
508,986,135✔
2290
      STableKeyInfo info = {.uid = *(uint64_t*)tmp, .groupId = 0};
508,986,135✔
2291

2292
      void* p = taosArrayPush(pListInfo->pTableList, &info);
509,009,814✔
2293
      if (p == NULL) {
509,067,169✔
2294
        taosArrayDestroy(pUidList);
×
2295
        return terrno;
×
2296
      }
2297

2298
      qTrace("tagfilter get uid:%" PRIu64 ", %s", info.uid, idstr);
509,067,169✔
2299
    }
2300
  }
2301

2302
  qDebug("%s, table list with %d uids built", idstr, (int32_t)numOfTables);
188,689,885✔
2303

2304
_error:
188,689,886✔
2305
  taosArrayDestroy(pUidList);
188,713,194✔
2306
  taosArrayDestroy(pTagColIds);
188,636,547✔
2307
  taosMemFreeClear(pTagCondKey);
188,647,942✔
2308
  taosMemFreeClear(pPayload);
188,647,942✔
2309
  if (code != TSDB_CODE_SUCCESS) {
188,647,942✔
2310
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
914✔
2311
  }
2312
  return code;
188,647,907✔
2313
}
2314

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

2437
  return TSDB_CODE_SUCCESS;
×
2438
}
2439

2440
SArray* makeColumnArrayFromList(SNodeList* pNodeList) {
3,206,428✔
2441
  if (!pNodeList) {
3,206,428✔
2442
    return NULL;
×
2443
  }
2444

2445
  size_t  numOfCols = LIST_LENGTH(pNodeList);
3,206,428✔
2446
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
3,204,868✔
2447
  if (pList == NULL) {
3,202,368✔
2448
    return NULL;
×
2449
  }
2450

2451
  for (int32_t i = 0; i < numOfCols; ++i) {
7,303,185✔
2452
    SColumnNode* pColNode = (SColumnNode*)nodesListGetNode(pNodeList, i);
4,105,257✔
2453
    if (!pColNode) {
4,104,544✔
2454
      taosArrayDestroy(pList);
×
2455
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2456
      return NULL;
×
2457
    }
2458

2459
    // todo extract method
2460
    SColumn c = {0};
4,104,544✔
2461
    c.slotId = pColNode->slotId;
4,104,967✔
2462
    c.colId = pColNode->colId;
4,105,340✔
2463
    c.type = pColNode->node.resType.type;
4,104,088✔
2464
    c.bytes = pColNode->node.resType.bytes;
4,102,599✔
2465
    c.precision = pColNode->node.resType.precision;
4,102,891✔
2466
    c.scale = pColNode->node.resType.scale;
4,102,829✔
2467

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

2476
  return pList;
3,197,928✔
2477
}
2478

2479
int32_t extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols,
253,668,373✔
2480
                            int32_t type, SColMatchInfo* pMatchInfo) {
2481
  size_t  numOfCols = LIST_LENGTH(pNodeList);
253,668,373✔
2482
  int32_t code = TSDB_CODE_SUCCESS;
253,668,268✔
2483
  int32_t lino = 0;
253,668,268✔
2484

2485
  pMatchInfo->matchType = type;
253,668,268✔
2486

2487
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchItem));
253,664,665✔
2488
  if (pList == NULL) {
253,594,233✔
2489
    code = terrno;
×
2490
    return code;
×
2491
  }
2492

2493
  SColMatchItem** infoBySlot = NULL;
253,594,233✔
2494
  if (numOfCols > 0) {
253,594,233✔
2495
    infoBySlot = taosMemoryCalloc(numOfCols, sizeof(*infoBySlot));
253,514,602✔
2496
    if (infoBySlot == NULL) {
253,553,667✔
2497
      code = terrno;
×
2498
      goto _end;
×
2499
    }
2500
  }
2501

2502
  SNode* node = NULL;
253,633,298✔
2503
  FOREACH(node, pNodeList) {
1,328,033,479✔
2504
    STargetNode* pNode = (STargetNode*)node;
1,074,231,141✔
2505
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,074,231,141✔
2506
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
1,074,231,141✔
2507
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
1,062,921,504✔
2508

2509
      SColMatchItem c = {.needOutput = true};
1,062,981,701✔
2510
      c.colId = pColNode->colId;
1,062,967,288✔
2511
      c.srcSlotId = pColNode->slotId;
1,062,974,076✔
2512
      c.dstSlotId = pNode->slotId;
1,063,049,844✔
2513
      c.isPk = pColNode->isPk;
1,063,087,988✔
2514
      c.dataType = pColNode->node.resType;
1,063,021,109✔
2515
      void* tmp = taosArrayPush(pList, &c);
1,063,122,824✔
2516
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,063,122,824✔
2517
      if (pNode->slotId >= 0 && pNode->slotId < numOfCols) {
1,063,122,824✔
2518
        infoBySlot[pNode->slotId] = (SColMatchItem*)taosArrayGet(pList, taosArrayGetSize(pList) - 1);
1,062,695,518✔
2519
      }
2520
    }
2521
  }
2522

2523
  // set the output flag for each column in SColMatchInfo, according to the
2524
  *numOfOutputCols = 0;
253,702,547✔
2525
  SNode* slotNode = NULL;
253,703,527✔
2526
  FOREACH(slotNode, pOutputNodeList->pSlots) {
1,524,281,012✔
2527
    SSlotDescNode* pNode = (SSlotDescNode*)slotNode;
1,270,565,317✔
2528
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,270,565,317✔
2529

2530
    // todo: add reserve flag check
2531
    // it is a column reserved for the arithmetic expression calculation
2532
    if (pNode->slotId >= numOfCols) {
1,270,565,317✔
2533
      (*numOfOutputCols) += 1;
195,850,871✔
2534
      continue;
195,852,780✔
2535
    }
2536

2537
    SColMatchItem* info = infoBySlot ? infoBySlot[pNode->slotId] : NULL;
1,074,822,213✔
2538

2539
    if (pNode->output) {
1,074,924,332✔
2540
      (*numOfOutputCols) += 1;
1,065,559,339✔
2541
    } else if (info != NULL) {
9,280,911✔
2542
      // select distinct tbname from stb where tbname='abc';
2543
      info->needOutput = false;
9,357,156✔
2544
    }
2545
  }
2546

2547
  pMatchInfo->pList = pList;
253,646,254✔
2548

2549
_end:
253,697,784✔
2550
  taosMemoryFree(infoBySlot);
253,697,784✔
2551
  if (code != TSDB_CODE_SUCCESS) {
253,588,960✔
2552
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2553
  }
2554
  return code;
253,598,617✔
2555
}
2556

2557
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
1,104,550,642✔
2558
                                  const char* name) {
2559
  SResSchema s = {0};
1,104,550,642✔
2560
  s.scale = scale;
1,104,642,732✔
2561
  s.type = type;
1,104,642,732✔
2562
  s.bytes = bytes;
1,104,642,732✔
2563
  s.slotId = slotId;
1,104,642,732✔
2564
  s.precision = precision;
1,104,642,732✔
2565
  tstrncpy(s.name, name, tListLen(s.name));
1,104,642,732✔
2566

2567
  return s;
1,104,642,732✔
2568
}
2569

2570
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType, EColumnType colType) {
1,078,186,338✔
2571
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
1,078,186,338✔
2572
  if (pCol == NULL) {
1,077,451,178✔
2573
    return NULL;
×
2574
  }
2575

2576
  pCol->slotId = slotId;
1,077,451,178✔
2577
  pCol->colId = colId;
1,077,480,572✔
2578
  pCol->bytes = pType->bytes;
1,077,528,069✔
2579
  pCol->type = pType->type;
1,077,701,311✔
2580
  pCol->scale = pType->scale;
1,077,723,119✔
2581
  pCol->precision = pType->precision;
1,078,067,357✔
2582
  pCol->dataBlockId = blockId;
1,077,965,010✔
2583
  pCol->colType = colType;
1,077,952,122✔
2584
  return pCol;
1,078,163,321✔
2585
}
2586

2587
int32_t createExprFromOneNode(SExprInfo* pExp, SNode* pNode, int16_t slotId) {
1,104,889,761✔
2588
  int32_t code = TSDB_CODE_SUCCESS;
1,104,889,761✔
2589
  int32_t lino = 0;
1,104,889,761✔
2590
  pExp->base.numOfParams = 0;
1,104,889,761✔
2591
  pExp->base.pParam = NULL;
1,105,070,199✔
2592
  pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
1,104,933,873✔
2593
  QUERY_CHECK_NULL(pExp->pExpr, code, lino, _end, terrno);
1,104,115,014✔
2594

2595
  pExp->pExpr->_function.num = 1;
1,104,228,798✔
2596
  pExp->pExpr->_function.functionId = -1;
1,104,370,781✔
2597

2598
  int32_t type = nodeType(pNode);
1,104,586,073✔
2599
  // it is a project query, or group by column
2600
  if (type == QUERY_NODE_COLUMN) {
1,104,784,438✔
2601
    pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
623,373,845✔
2602
    SColumnNode* pColNode = (SColumnNode*)pNode;
623,402,977✔
2603

2604
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
623,402,977✔
2605
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
623,330,906✔
2606

2607
    pExp->base.numOfParams = 1;
623,235,613✔
2608

2609
    SDataType* pType = &pColNode->node.resType;
623,407,604✔
2610
    pExp->base.resSchema =
2611
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pColNode->colName);
623,366,614✔
2612

2613
    pExp->base.pParam[0].pCol =
1,247,061,822✔
2614
        createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType, pColNode->colType);
1,246,912,311✔
2615
    QUERY_CHECK_NULL(pExp->base.pParam[0].pCol, code, lino, _end, terrno);
623,555,964✔
2616

2617
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
623,470,964✔
2618
  } else if (type == QUERY_NODE_VALUE) {
481,410,593✔
2619
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
16,113,847✔
2620
    SValueNode* pValNode = (SValueNode*)pNode;
16,112,902✔
2621

2622
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
16,112,902✔
2623
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
16,105,887✔
2624

2625
    pExp->base.numOfParams = 1;
16,101,028✔
2626

2627
    SDataType* pType = &pValNode->node.resType;
16,113,042✔
2628
    pExp->base.resSchema =
2629
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
16,108,828✔
2630
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
16,098,553✔
2631
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
16,108,500✔
2632
    QUERY_CHECK_CODE(code, lino, _end);
16,108,882✔
2633
  } else if (type == QUERY_NODE_REMOTE_VALUE) {
465,296,746✔
2634
    SRemoteValueNode* pRemote = (SRemoteValueNode*)pNode;
1,349,504✔
2635
    code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, pNode);
1,349,504✔
2636
    QUERY_CHECK_CODE(code, lino, _end);
1,351,584✔
2637

2638
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
951,704✔
2639
    SValueNode* pValNode = (SValueNode*)pNode;
951,704✔
2640

2641
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
951,704✔
2642
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
951,704✔
2643

2644
    pExp->base.numOfParams = 1;
951,704✔
2645

2646
    SDataType* pType = &pValNode->node.resType;
951,704✔
2647
    pExp->base.resSchema =
2648
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
951,704✔
2649
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
951,704✔
2650
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
951,704✔
2651
    QUERY_CHECK_CODE(code, lino, _end);
951,704✔
2652
  } else if (type == QUERY_NODE_FUNCTION) {
463,947,242✔
2653
    pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
441,680,597✔
2654
    SFunctionNode* pFuncNode = (SFunctionNode*)pNode;
441,694,966✔
2655

2656
    SDataType* pType = &pFuncNode->node.resType;
441,694,966✔
2657
    pExp->base.resSchema =
2658
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pFuncNode->node.aliasName);
441,714,745✔
2659
    tExprNode* pExprNode = pExp->pExpr;
441,677,130✔
2660

2661
    pExprNode->_function.functionId = pFuncNode->funcId;
441,620,883✔
2662
    pExprNode->_function.pFunctNode = pFuncNode;
441,719,436✔
2663
    pExprNode->_function.functionType = pFuncNode->funcType;
441,709,429✔
2664

2665
    tstrncpy(pExprNode->_function.functionName, pFuncNode->functionName, tListLen(pExprNode->_function.functionName));
441,776,979✔
2666

2667
    pExp->base.pParamList = pFuncNode->pParameterList;
441,801,866✔
2668
#if 1
2669
    // todo refactor: add the parameter for tbname function
2670
    const char* name = "tbname";
441,757,821✔
2671
    int32_t     len = strlen(name);
441,757,821✔
2672

2673
    if (!pFuncNode->pParameterList && (memcmp(pExprNode->_function.functionName, name, len) == 0) &&
441,757,821✔
2674
        pExprNode->_function.functionName[len] == 0) {
31,014,643✔
2675
      pFuncNode->pParameterList = NULL;
30,991,720✔
2676
      int32_t     code = nodesMakeList(&pFuncNode->pParameterList);
30,978,973✔
2677
      SValueNode* res = NULL;
31,020,392✔
2678
      if (TSDB_CODE_SUCCESS == code) {
31,015,500✔
2679
        code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
31,024,922✔
2680
      }
2681
      QUERY_CHECK_CODE(code, lino, _end);
31,029,744✔
2682
      res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
31,029,744✔
2683
      code = nodesListAppend(pFuncNode->pParameterList, (SNode*)res);
31,029,678✔
2684
      if (code != TSDB_CODE_SUCCESS) {
31,032,042✔
2685
        nodesDestroyNode((SNode*)res);
×
2686
        res = NULL;
×
2687
      }
2688
      QUERY_CHECK_CODE(code, lino, _end);
31,032,042✔
2689
    }
2690
#endif
2691

2692
    int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
441,765,957✔
2693

2694
    pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
441,690,880✔
2695
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
441,584,356✔
2696
    pExp->base.numOfParams = numOfParam;
441,665,674✔
2697

2698
    for (int32_t j = 0; j < numOfParam && TSDB_CODE_SUCCESS == code; ++j) {
1,071,915,438✔
2699
      SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
630,215,501✔
2700
      QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
630,323,452✔
2701
      if (p1->type == QUERY_NODE_COLUMN) {
630,323,452✔
2702
        SColumnNode* pcn = (SColumnNode*)p1;
454,707,242✔
2703

2704
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
454,707,242✔
2705
        pExp->base.pParam[j].pCol =
909,301,744✔
2706
            createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType, pcn->colType);
909,297,512✔
2707
        QUERY_CHECK_NULL(pExp->base.pParam[j].pCol, code, lino, _end, terrno);
454,674,905✔
2708
      } else if (p1->type == QUERY_NODE_VALUE) {
175,635,220✔
2709
        SValueNode* pvn = (SValueNode*)p1;
122,010,699✔
2710
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
122,010,699✔
2711
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
121,955,183✔
2712
        QUERY_CHECK_CODE(code, lino, _end);
121,965,376✔
2713
      } else if (p1->type == QUERY_NODE_REMOTE_VALUE) {
53,652,709✔
2714
        SRemoteValueNode* pRemote = (SRemoteValueNode*)p1;
69,160✔
2715
        code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, p1);
69,160✔
2716
        QUERY_CHECK_CODE(code, lino, _end);
69,680✔
2717

2718
        SValueNode* pvn = (SValueNode*)pRemote;
50,440✔
2719
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
50,440✔
2720
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
50,440✔
2721
        QUERY_CHECK_CODE(code, lino, _end);
97,189✔
2722
      }
2723
    }
2724
    pExp->pExpr->_function.bindExprID = ((SExprNode*)pNode)->bindExprID;
441,699,937✔
2725
  } else if (type == QUERY_NODE_OPERATOR) {
22,266,645✔
2726
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
21,091,941✔
2727
    SOperatorNode* pOpNode = (SOperatorNode*)pNode;
21,095,463✔
2728

2729
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
21,095,463✔
2730
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
21,069,782✔
2731
    pExp->base.numOfParams = 1;
21,066,785✔
2732

2733
    SDataType* pType = &pOpNode->node.resType;
21,087,908✔
2734
    pExp->base.resSchema =
2735
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pOpNode->node.aliasName);
21,078,887✔
2736
    pExp->pExpr->_optrRoot.pRootNode = pNode;
21,072,859✔
2737
  } else if (type == QUERY_NODE_CASE_WHEN) {
1,174,711✔
2738
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
1,161,281✔
2739
    SCaseWhenNode* pCaseNode = (SCaseWhenNode*)pNode;
1,161,281✔
2740

2741
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
1,161,281✔
2742
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
1,161,281✔
2743
    pExp->base.numOfParams = 1;
1,161,281✔
2744

2745
    SDataType* pType = &pCaseNode->node.resType;
1,161,281✔
2746
    pExp->base.resSchema =
2747
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCaseNode->node.aliasName);
1,161,281✔
2748
    pExp->pExpr->_optrRoot.pRootNode = pNode;
1,161,281✔
2749
  } else if (type == QUERY_NODE_LOGIC_CONDITION) {
13,430✔
2750
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
3,034✔
2751
    SLogicConditionNode* pCond = (SLogicConditionNode*)pNode;
3,034✔
2752
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
3,034✔
2753
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
3,034✔
2754
    pExp->base.numOfParams = 1;
3,034✔
2755
    SDataType* pType = &pCond->node.resType;
3,034✔
2756
    pExp->base.resSchema =
2757
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCond->node.aliasName);
3,034✔
2758
    pExp->pExpr->_optrRoot.pRootNode = pNode;
3,034✔
2759
  } else {
2760
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
10,396✔
2761
    QUERY_CHECK_CODE(code, lino, _end);
10,396✔
2762
  }
2763
  pExp->pExpr->relatedTo = ((SExprNode*)pNode)->relatedTo;
1,104,345,902✔
2764
_end:
1,104,850,918✔
2765
  if (code != TSDB_CODE_SUCCESS) {
1,104,850,918✔
2766
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
419,120✔
2767
  }
2768
  return code;
1,104,677,473✔
2769
}
2770

2771
int32_t createExprFromTargetNode(SExprInfo* pExp, STargetNode* pTargetNode) {
1,104,869,088✔
2772
  return createExprFromOneNode(pExp, pTargetNode->pExpr, pTargetNode->slotId);
1,104,869,088✔
2773
}
2774

2775
SExprInfo* createExpr(SNodeList* pNodeList, int32_t* numOfExprs) {
×
2776
  *numOfExprs = LIST_LENGTH(pNodeList);
×
2777
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
×
2778
  if (!pExprs) {
×
2779
    return NULL;
×
2780
  }
2781

2782
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
×
2783
    SExprInfo* pExp = &pExprs[i];
×
2784
    int32_t    code = createExprFromOneNode(pExp, nodesListGetNode(pNodeList, i), i + UD_TAG_COLUMN_INDEX);
×
2785
    if (code != TSDB_CODE_SUCCESS) {
×
2786
      taosMemoryFreeClear(pExprs);
×
2787
      terrno = code;
×
2788
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2789
      return NULL;
×
2790
    }
2791
  }
2792

2793
  return pExprs;
×
2794
}
2795

2796
int32_t createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, SExprInfo** pExprInfo, int32_t* numOfExprs) {
350,725,106✔
2797
  QRY_PARAM_CHECK(pExprInfo);
350,725,106✔
2798

2799
  int32_t code = 0;
350,775,270✔
2800
  int32_t lino = 0;
350,775,270✔
2801
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
350,775,270✔
2802
  int32_t numOfGroupKeys = 0;
350,643,235✔
2803
  if (pGroupKeys != NULL) {
350,643,235✔
2804
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
30,256,479✔
2805
  }
2806

2807
  *numOfExprs = numOfFuncs + numOfGroupKeys;
350,641,480✔
2808
  if (*numOfExprs == 0) {
350,693,482✔
2809
    return code;
35,099,298✔
2810
  }
2811

2812
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
315,613,083✔
2813
  QUERY_CHECK_NULL(pExprs, code, lino, _return, terrno);
315,396,735✔
2814

2815
  int32_t i = 0;
315,396,735✔
2816
  if (pNodeList != NULL) {
315,396,735✔
2817
    SNode* node = NULL;
288,986,933✔
2818
    FOREACH(node, pNodeList) {
1,331,745,054✔
2819
      STargetNode* pTargetNode = (STargetNode*)node;
1,043,397,319✔
2820
      SExprInfo*   pExp = &pExprs[i++];
1,043,397,319✔
2821
      code = createExprFromTargetNode(pExp, pTargetNode);
1,043,281,933✔
2822
      QUERY_CHECK_CODE(code, lino, _return);
1,043,177,241✔
2823
    }
2824
  }
2825

2826
  if (pGroupKeys != NULL) {
314,884,595✔
2827
    SNode* node = NULL;
30,250,084✔
2828
    FOREACH(node, pGroupKeys) {
91,601,673✔
2829
      STargetNode* pTargetNode = (STargetNode*)node;
61,361,956✔
2830
      SExprInfo*   pExp = &pExprs[i++];
61,361,956✔
2831
      code = createExprFromTargetNode(pExp, pTargetNode);
61,357,028✔
2832
      QUERY_CHECK_CODE(code, lino, _return);
61,351,589✔
2833
    }
2834
  }
2835

2836
  *pExprInfo = pExprs;
314,897,715✔
2837
  return code;
315,211,646✔
2838

2839
_return:
419,120✔
2840
  qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
419,120✔
2841
  destroyExprInfo(pExprs, *numOfExprs);
419,120✔
2842
  taosMemoryFreeClear(pExprs);
419,120✔
2843
  return code;
419,120✔
2844
}
2845

2846
static void deleteSubsidiareCtx(void* pData) {
×
2847
  SSubsidiaryResInfo* pCtx = (SSubsidiaryResInfo*)pData;
×
2848
  if (pCtx->pCtx) {
×
2849
    taosMemoryFreeClear(pCtx->pCtx);
×
2850
  }
2851
}
×
2852

2853
// set the output buffer for the selectivity + tag query
2854
static int32_t setSelectValueColumnInfo(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
340,945,568✔
2855
  int32_t num = 0;
340,945,568✔
2856
  int32_t code = TSDB_CODE_SUCCESS;
340,945,568✔
2857
  int32_t lino = 0;
340,945,568✔
2858

2859
  SArray* pValCtxArray = NULL;
340,945,568✔
2860
  for (int32_t i = numOfOutput - 1; i > 0; --i) {  // select Func is at the end of the list
1,121,923,102✔
2861
    int32_t funcIdx = pCtx[i].pExpr->pExpr->_function.bindExprID;
781,090,362✔
2862
    if (funcIdx > 0) {
781,052,637✔
2863
      if (pValCtxArray == NULL) {
1,477,097✔
2864
        // the end of the list is the select function of biggest index
2865
        pValCtxArray = taosArrayInit_s(sizeof(SSubsidiaryResInfo*), funcIdx);
1,058,499✔
2866
        if (pValCtxArray == NULL) {
1,059,303✔
2867
          return terrno;
×
2868
        }
2869
      }
2870
      if (funcIdx > pValCtxArray->size) {
1,477,901✔
2871
        qError("funcIdx:%d is out of range", funcIdx);
×
2872
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2873
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2874
      }
2875
      SSubsidiaryResInfo* pSubsidiary = &pCtx[i].subsidiaries;
1,477,499✔
2876
      pSubsidiary->pCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
1,477,097✔
2877
      if (pSubsidiary->pCtx == NULL) {
1,476,695✔
2878
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2879
        return terrno;
×
2880
      }
2881
      pSubsidiary->num = 0;
1,476,293✔
2882
      taosArraySet(pValCtxArray, funcIdx - 1, &pSubsidiary);
1,477,097✔
2883
    }
2884
  }
2885

2886
  SqlFunctionCtx*  p = NULL;
340,832,740✔
2887
  SqlFunctionCtx** pValCtx = NULL;
340,832,740✔
2888
  if (pValCtxArray == NULL) {
340,832,740✔
2889
    pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
339,799,847✔
2890
    if (pValCtx == NULL) {
339,740,040✔
2891
      QUERY_CHECK_CODE(terrno, lino, _end);
×
2892
    }
2893
  }
2894

2895
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,432,922,824✔
2896
    const char* pName = pCtx[i].pExpr->pExpr->_function.functionName;
1,092,097,186✔
2897
    if ((strcmp(pName, "_select_value") == 0)) {
1,092,151,305✔
2898
      if (pValCtxArray == NULL) {
7,785,594✔
2899
        pValCtx[num++] = &pCtx[i];
5,713,068✔
2900
      } else {
2901
        int32_t bindFuncIndex = pCtx[i].pExpr->pExpr->relatedTo;  // start from index 1;
2,072,570✔
2902
        if (bindFuncIndex > 0) {                                  // 0 is default index related to the select function
2,075,452✔
2903
          bindFuncIndex -= 1;
2,028,418✔
2904
        }
2905
        SSubsidiaryResInfo** pSubsidiary = taosArrayGet(pValCtxArray, bindFuncIndex);
2,075,452✔
2906
        if (pSubsidiary == NULL) {
2,075,050✔
2907
          QUERY_CHECK_CODE(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR, lino, _end);
×
2908
        }
2909
        (*pSubsidiary)->pCtx[(*pSubsidiary)->num] = &pCtx[i];
2,075,050✔
2910
        (*pSubsidiary)->num++;
2,075,854✔
2911
      }
2912
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
1,084,365,711✔
2913
      if (pValCtxArray == NULL) {
104,106,745✔
2914
        p = &pCtx[i];
102,304,998✔
2915
      }
2916
    }
2917
  }
2918

2919
  if (p != NULL) {
340,825,638✔
2920
    p->subsidiaries.pCtx = pValCtx;
30,809,370✔
2921
    p->subsidiaries.num = num;
30,804,157✔
2922
  } else {
2923
    taosMemoryFreeClear(pValCtx);
310,016,268✔
2924
  }
2925

2926
_end:
1,007,108✔
2927
  if (code != TSDB_CODE_SUCCESS) {
340,765,049✔
2928
    taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2929
    taosMemoryFreeClear(pValCtx);
×
2930
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2931
  } else {
2932
    taosArrayDestroy(pValCtxArray);
340,765,049✔
2933
  }
2934
  return code;
340,831,057✔
2935
}
2936

2937
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowEntryInfoOffset,
340,947,245✔
2938
                                     SFunctionStateStore* pStore) {
2939
  int32_t         code = TSDB_CODE_SUCCESS;
340,947,245✔
2940
  int32_t         lino = 0;
340,947,245✔
2941
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
340,947,245✔
2942
  if (pFuncCtx == NULL) {
340,588,385✔
2943
    return NULL;
×
2944
  }
2945

2946
  *rowEntryInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
340,588,385✔
2947
  if (*rowEntryInfoOffset == 0) {
340,947,423✔
2948
    taosMemoryFreeClear(pFuncCtx);
×
2949
    return NULL;
×
2950
  }
2951

2952
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,433,133,521✔
2953
    SExprInfo* pExpr = &pExprInfo[i];
1,092,280,527✔
2954

2955
    SExprBasicInfo* pFunct = &pExpr->base;
1,092,135,712✔
2956
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
1,092,185,327✔
2957

2958
    pCtx->functionId = -1;
1,092,203,101✔
2959
    pCtx->pExpr = pExpr;
1,092,260,094✔
2960

2961
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
1,092,234,514✔
2962
      SFuncExecEnv env = {0};
440,762,749✔
2963
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;
440,768,185✔
2964
      pCtx->isPseudoFunc = fmIsWindowPseudoColumnFunc(pCtx->functionId) || fmIsPlaceHolderFunc(pCtx->functionId);
440,804,536✔
2965
      pCtx->isNotNullFunc = fmIsNotNullOutputFunc(pCtx->functionId);
440,697,133✔
2966

2967
      bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
440,699,452✔
2968
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId)) {
728,555,232✔
2969
        if (!isUdaf) {
287,933,224✔
2970
          code = fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
287,834,413✔
2971
          QUERY_CHECK_CODE(code, lino, _end);
287,833,620✔
2972
        } else {
2973
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
98,811✔
2974
          pCtx->udfName = taosStrdup(udfName);
98,811✔
2975
          QUERY_CHECK_NULL(pCtx->udfName, code, lino, _end, terrno);
98,811✔
2976

2977
          code = fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
98,811✔
2978
          QUERY_CHECK_CODE(code, lino, _end);
98,811✔
2979
        }
2980
        bool tmp = pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
287,932,431✔
2981
        if (!tmp) {
287,864,044✔
2982
          code = terrno;
×
2983
          QUERY_CHECK_CODE(code, lino, _end);
122✔
2984
        }
2985
      } else {
2986
        if (fmIsPlaceHolderFunc(pCtx->functionId)) {
152,754,048✔
2987
          code = fmGetStreamPesudoFuncEnv(pCtx->functionId, pExpr->base.pParamList, &env);
5,294,663✔
2988
          QUERY_CHECK_CODE(code, lino, _end);
5,294,913✔
2989
        }      
2990
        
2991
        code = fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
152,784,405✔
2992
        if (code != TSDB_CODE_SUCCESS && isUdaf) {
152,762,721✔
2993
          code = TSDB_CODE_SUCCESS;
114,033✔
2994
        }
2995
        QUERY_CHECK_CODE(code, lino, _end);
152,762,721✔
2996

2997
        if (pCtx->sfp.getEnv != NULL) {
152,762,721✔
2998
          bool tmp = pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
23,424,456✔
2999
          if (!tmp) {
23,423,466✔
3000
            code = terrno;
×
3001
            QUERY_CHECK_CODE(code, lino, _end);
×
3002
          }
3003
        }
3004
      }
3005
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
440,663,491✔
3006
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
651,456,712✔
3007
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
17,039,168✔
3008
      // for simple column, the result buffer needs to hold at least one element.
3009
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
651,630,782✔
3010
    }
3011

3012
    pCtx->input.numOfInputCols = pFunct->numOfParams;
1,092,388,840✔
3013
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
1,092,245,827✔
3014
    QUERY_CHECK_NULL(pCtx->input.pData, code, lino, _end, terrno);
1,092,238,389✔
3015
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
1,092,273,153✔
3016
    QUERY_CHECK_NULL(pCtx->input.pColumnDataAgg, code, lino, _end, terrno);
1,092,234,185✔
3017

3018
    pCtx->pTsOutput = NULL;
1,092,307,684✔
3019
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
1,092,325,032✔
3020
    pCtx->resDataInfo.type = pFunct->resSchema.type;
1,092,379,073✔
3021
    pCtx->order = TSDB_ORDER_ASC;
1,092,276,786✔
3022
    pCtx->start.key = INT64_MIN;
1,092,206,876✔
3023
    pCtx->end.key = INT64_MIN;
1,092,360,448✔
3024
    pCtx->numOfParams = pExpr->base.numOfParams;
1,092,395,637✔
3025
    pCtx->param = pFunct->pParam;
1,092,319,234✔
3026
    pCtx->saveHandle.currentPage = -1;
1,092,570,174✔
3027
    pCtx->pStore = pStore;
1,092,422,701✔
3028
    pCtx->hasWindowOrGroup = false;
1,092,311,983✔
3029
    pCtx->needCleanup = false;
1,092,100,560✔
3030
    pCtx->skipDynDataCheck = false;
1,092,280,589✔
3031
  }
3032

3033
  for (int32_t i = 1; i < numOfOutput; ++i) {
1,122,175,010✔
3034
    (*rowEntryInfoOffset)[i] = (int32_t)((*rowEntryInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) +
1,562,427,468✔
3035
                                         pFuncCtx[i - 1].resDataInfo.interBufSize);
781,208,728✔
3036
  }
3037

3038
  code = setSelectValueColumnInfo(pFuncCtx, numOfOutput);
340,929,474✔
3039
  QUERY_CHECK_CODE(code, lino, _end);
340,825,092✔
3040

3041
_end:
340,825,092✔
3042
  if (code != TSDB_CODE_SUCCESS) {
340,701,090✔
3043
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3044
    for (int32_t i = 0; i < numOfOutput; ++i) {
×
3045
      taosMemoryFree(pFuncCtx[i].input.pData);
×
3046
      taosMemoryFree(pFuncCtx[i].input.pColumnDataAgg);
×
3047
    }
3048
    taosMemoryFreeClear(*rowEntryInfoOffset);
×
3049
    taosMemoryFreeClear(pFuncCtx);
×
3050

3051
    terrno = code;
×
3052
    return NULL;
×
3053
  }
3054
  return pFuncCtx;
340,701,090✔
3055
}
3056

3057
// NOTE: sources columns are more than the destination SSDatablock columns.
3058
// doFilter in table scan needs every column even its output is false
3059
int32_t relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray* pCols, bool outputEveryColumn) {
13,713,978✔
3060
  int32_t code = TSDB_CODE_SUCCESS;
13,713,978✔
3061
  size_t  numOfSrcCols = taosArrayGetSize(pCols);
13,713,978✔
3062

3063
  int32_t i = 0, j = 0;
13,714,553✔
3064
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
108,031,772✔
3065
    SColumnInfoData* p = taosArrayGet(pCols, i);
94,318,910✔
3066
    if (!p) {
94,319,984✔
3067
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3068
      return terrno;
×
3069
    }
3070
    SColMatchItem* pmInfo = taosArrayGet(pColMatchInfo, j);
94,319,984✔
3071
    if (!pmInfo) {
94,319,298✔
3072
      return terrno;
×
3073
    }
3074

3075
    if (p->info.colId == pmInfo->colId) {
94,319,298✔
3076
      SColumnInfoData* pDst = taosArrayGet(pBlock->pDataBlock, pmInfo->dstSlotId);
88,222,059✔
3077
      if (!pDst) {
88,221,794✔
3078
        return terrno;
×
3079
      }
3080
      code = colDataAssign(pDst, p, pBlock->info.rows, &pBlock->info);
88,221,794✔
3081
      if (code != TSDB_CODE_SUCCESS) {
88,219,294✔
3082
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3083
        return code;
×
3084
      }
3085
      i++;
88,219,294✔
3086
      j++;
88,219,294✔
3087
    } else if (p->info.colId < pmInfo->colId) {
6,097,925✔
3088
      i++;
6,097,925✔
3089
    } else {
3090
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3091
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3092
    }
3093
  }
3094
  return code;
13,715,667✔
3095
}
3096

3097
SInterval extractIntervalInfo(const STableScanPhysiNode* pTableScanNode) {
160,247,025✔
3098
  SInterval interval = {
320,498,782✔
3099
      .interval = pTableScanNode->interval,
160,159,300✔
3100
      .sliding = pTableScanNode->sliding,
160,155,530✔
3101
      .intervalUnit = pTableScanNode->intervalUnit,
160,296,189✔
3102
      .slidingUnit = pTableScanNode->slidingUnit,
160,124,622✔
3103
      .offset = pTableScanNode->offset,
160,082,857✔
3104
      .precision = pTableScanNode->scan.node.pOutputDataBlockDesc->precision,
160,079,235✔
3105
      .timeRange = pTableScanNode->scanRange,
3106
  };
3107
  calcIntervalAutoOffset(&interval);
160,291,961✔
3108

3109
  return interval;
160,040,198✔
3110
}
3111

3112
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
65,609,311✔
3113
  SColumn c = {0};
65,609,311✔
3114

3115
  c.slotId = pColNode->slotId;
65,609,311✔
3116
  c.colId = pColNode->colId;
65,608,299✔
3117
  c.type = pColNode->node.resType.type;
65,609,183✔
3118
  c.bytes = pColNode->node.resType.bytes;
65,614,571✔
3119
  c.scale = pColNode->node.resType.scale;
65,605,528✔
3120
  c.precision = pColNode->node.resType.precision;
65,600,885✔
3121
  return c;
65,612,069✔
3122
}
3123

3124

3125
/**
3126
 * @brief Determine the actual time range for reading data based on the RANGE clause and the WHERE conditions.
3127
 * @param[in] cond The range specified by WHERE condition.
3128
 * @param[in] range The range specified by RANGE clause.
3129
 * @param[out] twindow The range to be read in DESC order, and only one record is needed.
3130
 * @param[out] extTwindow The external range to read for only one record, which is used for FILL clause.
3131
 * @note `cond` and `twindow` may be the same address.
3132
 */
3133
static int32_t getQueryExtWindow(const STimeWindow* cond, const STimeWindow* range, STimeWindow* twindow,
3,069,901✔
3134
                                 STimeWindow* extTwindows) {
3135
  int32_t     code = TSDB_CODE_SUCCESS;
3,069,901✔
3136
  int32_t     lino = 0;
3,069,901✔
3137
  STimeWindow tempWindow;
3138

3139
  if (cond->skey > cond->ekey || range->skey > range->ekey) {
3,069,901✔
3140
    *twindow = extTwindows[0] = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
4,740✔
3141
    return code;
4,740✔
3142
  }
3143

3144
  if (range->ekey < cond->skey) {
3,065,161✔
3145
    extTwindows[1] = *cond;
469,019✔
3146
    *twindow = extTwindows[0] = TSWINDOW_DESC_INITIALIZER;
469,019✔
3147
    return code;
469,019✔
3148
  }
3149

3150
  if (cond->ekey < range->skey) {
2,596,142✔
3151
    extTwindows[0] = *cond;
341,635✔
3152
    *twindow = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
341,635✔
3153
    return code;
341,635✔
3154
  }
3155

3156
  // Only scan data in the time range intersecion.
3157
  extTwindows[0] = extTwindows[1] = *cond;
2,254,507✔
3158
  twindow->skey = TMAX(cond->skey, range->skey);
2,254,507✔
3159
  twindow->ekey = TMIN(cond->ekey, range->ekey);
2,254,507✔
3160
  extTwindows[0].ekey = twindow->skey - 1;
2,254,507✔
3161
  extTwindows[1].skey = twindow->ekey + 1;
2,254,507✔
3162

3163
  return code;
2,254,507✔
3164
}
3165

3166
static int32_t getPrimaryTimeRange(SNode** pPrimaryKeyCond, STimeWindow* pTimeRange, bool* isStrict) {
4,188✔
3167
  SNode*  pNew = NULL;
4,188✔
3168
  int32_t code = scalarCalculateRemoteConstants(*pPrimaryKeyCond, &pNew);
4,188✔
3169
  if (TSDB_CODE_SUCCESS == code) {
4,188✔
3170
    *pPrimaryKeyCond = pNew;
4,188✔
3171
    if (nodeType(pNew) != QUERY_NODE_VALUE) {
4,188✔
3172
      code = filterGetTimeRange(*pPrimaryKeyCond, pTimeRange, isStrict, NULL);
4,188✔
3173
    }
3174
  }
3175
  return code;
4,188✔
3176
}
3177

3178
int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, STableScanPhysiNode* pTableScanNode,
187,988,675✔
3179
                               const SReadHandle* readHandle, bool applyExtWin) {
3180
  int32_t code = 0;                             
187,988,675✔
3181
  pCond->order = pTableScanNode->scanSeq[0] > 0 ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
187,988,675✔
3182
  pCond->numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
187,863,261✔
3183

3184
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
187,889,888✔
3185
  if (!pCond->colList) {
187,940,355✔
3186
    return terrno;
×
3187
  }
3188
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
187,890,830✔
3189
  if (pCond->pSlotList == NULL) {
187,956,073✔
3190
    taosMemoryFreeClear(pCond->colList);
×
3191
    return terrno;
×
3192
  }
3193

3194
  // TODO: get it from stable scan node
3195
  pCond->twindows = pTableScanNode->scanRange;
187,820,229✔
3196
  pCond->suid = pTableScanNode->scan.suid;
187,999,869✔
3197
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
187,985,775✔
3198
  pCond->startVersion = -1;
187,884,235✔
3199
  pCond->endVersion = -1;
187,830,181✔
3200
  pCond->skipRollup = readHandle->skipRollup;
187,951,986✔
3201
  if (readHandle->winRangeValid) {
187,783,997✔
3202
    pCond->twindows = readHandle->winRange;
265,875✔
3203
  }
3204
  pCond->cacheSttStatis = readHandle->cacheSttStatis;
187,903,221✔
3205
  // allowed read stt file optimization mode
3206
  pCond->notLoadData = (pTableScanNode->dataRequired == FUNC_DATA_REQUIRED_NOT_LOAD) &&
376,005,524✔
3207
                       (pTableScanNode->scan.node.pConditions == NULL) && (pTableScanNode->interval == 0);
187,965,639✔
3208

3209
  int32_t j = 0;
188,017,159✔
3210
  SNode*  node = NULL;
188,017,159✔
3211
  FOREACH(node, pTableScanNode->scan.pScanCols) {
992,008,809✔
3212
    STargetNode* pNode = (STargetNode*)node;
803,760,288✔
3213
    if (!pNode) {
803,760,288✔
3214
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3215
      return terrno;
×
3216
    }
3217
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
803,760,288✔
3218
    if (pColNode->colType == COLUMN_TYPE_TAG) {
804,013,441✔
3219
      continue;
×
3220
    }
3221

3222
    pCond->colList[j].type = pColNode->node.resType.type;
804,015,351✔
3223
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
803,996,077✔
3224
    pCond->colList[j].colId = pColNode->colId;
803,901,121✔
3225
    pCond->colList[j].pk = pColNode->isPk;
804,013,238✔
3226

3227
    pCond->pSlotList[j] = pNode->slotId;
804,089,106✔
3228
    j += 1;
803,991,650✔
3229
  }
3230

3231
  pCond->numOfCols = j;
187,989,481✔
3232

3233
  if (applyExtWin) {
188,012,960✔
3234
    if (NULL != pTableScanNode->pExtScanRange) {
160,584,924✔
3235
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
3,005,437✔
3236
      code = getQueryExtWindow(&pCond->twindows, pTableScanNode->pExtScanRange, &pCond->twindows, pCond->extTwindows);
3,005,437✔
3237
    } else if (readHandle->extWinRangeValid) {
157,320,842✔
3238
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
64,464✔
3239
      code = getQueryExtWindow(&pCond->twindows, &readHandle->extWinRange, &pCond->twindows, pCond->extTwindows);
64,464✔
3240
    }
3241
  }
3242

3243
  if (pTableScanNode->pPrimaryCond) {
187,964,379✔
3244
    bool isStrict = false;
4,188✔
3245
    code = getPrimaryTimeRange((SNode**)&pTableScanNode->pPrimaryCond, &pCond->twindows, &isStrict);
4,188✔
3246
    if (code || !isStrict) {
4,188✔
3247
      code = nodesMergeNode((SNode**)&pTableScanNode->scan.node.pConditions, &pTableScanNode->pPrimaryCond);
1,396✔
3248
    }
3249
  }
3250

3251
  return code;
187,940,710✔
3252
}
3253

3254
int32_t initQueryTableDataCondWithColArray(SQueryTableDataCond* pCond, SQueryTableDataCond* pOrgCond,
40,371,998✔
3255
                                           const SReadHandle* readHandle, SArray* colArray) {
3256
  int32_t code = TSDB_CODE_SUCCESS;
40,371,998✔
3257
  int32_t lino = 0;
40,371,998✔
3258

3259
  pCond->order = TSDB_ORDER_ASC;
40,371,998✔
3260
  pCond->numOfCols = (int32_t)taosArrayGetSize(colArray);
40,375,531✔
3261

3262
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
40,368,339✔
3263
  QUERY_CHECK_NULL(pCond->colList, code, lino, _return, terrno);
40,364,501✔
3264

3265
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
40,362,991✔
3266
  QUERY_CHECK_NULL(pCond->pSlotList, code, lino, _return, terrno);
40,364,753✔
3267

3268
  pCond->twindows = pOrgCond->twindows;
40,360,590✔
3269
  pCond->order = pOrgCond->order;
40,368,349✔
3270
  pCond->type = pOrgCond->type;
40,367,847✔
3271
  pCond->startVersion = -1;
40,369,228✔
3272
  pCond->endVersion = -1;
40,331,928✔
3273
  pCond->skipRollup = true;
40,351,319✔
3274
  pCond->notLoadData = false;
40,343,263✔
3275

3276
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
196,510,110✔
3277
    SColIdPair* pColPair = taosArrayGet(colArray, i);
156,164,650✔
3278
    QUERY_CHECK_NULL(pColPair, code, lino, _return, terrno);
156,152,304✔
3279

3280
    bool find = false;
156,153,249✔
3281
    for (int32_t j = 0; j < pOrgCond->numOfCols; ++j) {
948,841,390✔
3282
      if (pOrgCond->colList[j].colId == pColPair->vtbColId) {
948,565,516✔
3283
        pCond->colList[i].type = pOrgCond->colList[j].type;
156,155,321✔
3284
        pCond->colList[i].bytes = pOrgCond->colList[j].bytes;
156,175,987✔
3285
        pCond->colList[i].colId = pColPair->orgColId;
156,177,634✔
3286
        pCond->colList[i].pk = pOrgCond->colList[j].pk;
156,180,037✔
3287
        pCond->pSlotList[i] = i;
156,189,350✔
3288
        find = true;
156,190,109✔
3289
        qDebug("%s mapped vtb colId:%d to org colId:%d", __func__, pColPair->vtbColId, pColPair->orgColId);
156,190,109✔
3290
        break;
156,162,685✔
3291
      }
3292
    }
3293
    QUERY_CHECK_CONDITION(find, code, lino, _return, TSDB_CODE_NOT_FOUND);
156,151,027✔
3294
  }
3295

3296
  return code;
40,376,536✔
3297
_return:
×
3298
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(terrno));
×
3299
  taosMemoryFreeClear(pCond->colList);
×
3300
  taosMemoryFreeClear(pCond->pSlotList);
×
3301
  return code;
×
3302
}
3303

3304
void cleanupQueryTableDataCond(SQueryTableDataCond* pCond) {
433,300,669✔
3305
  taosMemoryFreeClear(pCond->colList);
433,300,669✔
3306
  taosMemoryFreeClear(pCond->pSlotList);
433,297,760✔
3307
}
433,306,370✔
3308

3309
int32_t convertFillType(int32_t mode) {
3,626,488✔
3310
  int32_t type = TSDB_FILL_NONE;
3,626,488✔
3311
  switch (mode) {
3,626,488✔
3312
    case FILL_MODE_PREV:
170,063✔
3313
      type = TSDB_FILL_PREV;
170,063✔
3314
      break;
170,063✔
3315
    case FILL_MODE_NONE:
×
3316
      type = TSDB_FILL_NONE;
×
3317
      break;
×
3318
    case FILL_MODE_NULL:
170,327✔
3319
      type = TSDB_FILL_NULL;
170,327✔
3320
      break;
170,327✔
3321
    case FILL_MODE_NULL_F:
25,557✔
3322
      type = TSDB_FILL_NULL_F;
25,557✔
3323
      break;
25,557✔
3324
    case FILL_MODE_NEXT:
165,455✔
3325
      type = TSDB_FILL_NEXT;
165,455✔
3326
      break;
165,455✔
3327
    case FILL_MODE_VALUE:
154,060✔
3328
      type = TSDB_FILL_SET_VALUE;
154,060✔
3329
      break;
154,060✔
3330
    case FILL_MODE_VALUE_F:
10,948✔
3331
      type = TSDB_FILL_SET_VALUE_F;
10,948✔
3332
      break;
10,948✔
3333
    case FILL_MODE_LINEAR:
217,534✔
3334
      type = TSDB_FILL_LINEAR;
217,534✔
3335
      break;
217,534✔
3336
    case FILL_MODE_NEAR:
2,712,544✔
3337
      type = TSDB_FILL_NEAR;
2,712,544✔
3338
      break;
2,712,544✔
3339
    default:
×
3340
      type = TSDB_FILL_NONE;
×
3341
  }
3342

3343
  return type;
3,626,488✔
3344
}
3345

3346
void getInitialStartTimeWindow(SInterval* pInterval, TSKEY ts, STimeWindow* w, bool ascQuery) {
2,147,483,647✔
3347
  if (ascQuery) {
2,147,483,647✔
3348
    *w = getAlignQueryTimeWindow(pInterval, ts);
2,147,483,647✔
3349
  } else {
3350
    // the start position of the first time window in the endpoint that spreads beyond the queried last timestamp
3351
    *w = getAlignQueryTimeWindow(pInterval, ts);
22,019✔
3352

3353
    int64_t key = w->skey;
229,102✔
3354
    while (key < ts) {  // moving towards end
244,701✔
3355
      key = getNextTimeWindowStart(pInterval, key, TSDB_ORDER_ASC);
114,367✔
3356
      if (key > ts) {
114,367✔
3357
        break;
98,768✔
3358
      }
3359

3360
      w->skey = key;
15,599✔
3361
    }
3362
    w->ekey = taosTimeAdd(w->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
229,102✔
3363
  }
3364
}
2,147,483,647✔
3365

3366
static STimeWindow doCalculateTimeWindow(int64_t ts, SInterval* pInterval) {
68,167,762✔
3367
  STimeWindow w = {0};
68,167,762✔
3368

3369
  w.skey = taosTimeTruncate(ts, pInterval);
68,167,762✔
3370
  w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
68,163,793✔
3371
  return w;
68,169,472✔
3372
}
3373

3374
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order) {
1,784,875✔
3375
  STimeWindow win = *pWindow;
1,784,875✔
3376
  STimeWindow save = win;
1,784,875✔
3377
  while (win.skey <= ts && win.ekey >= ts) {
11,702,064✔
3378
    save = win;
9,917,189✔
3379
    // get previous time window
3380
    getNextTimeWindow(pInterval, &win, order == TSDB_ORDER_DESC ? TSDB_ORDER_ASC : TSDB_ORDER_DESC);
9,917,189✔
3381
  }
3382

3383
  return save;
1,784,875✔
3384
}
3385

3386
// get the correct time window according to the handled timestamp
3387
// todo refactor
3388
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
105,432,561✔
3389
                                int32_t order) {
3390
  STimeWindow w = {0};
105,432,561✔
3391
  if (pResultRowInfo->cur.pageId == -1) {  // the first window, from the previous stored value
105,443,239✔
3392
    getInitialStartTimeWindow(pInterval, ts, &w, (order != TSDB_ORDER_DESC));
2,652,943✔
3393
    return w;
2,657,517✔
3394
  }
3395

3396
  SResultRow* pRow = getResultRowByPos(pBuf, &pResultRowInfo->cur, false);
102,780,695✔
3397
  if (pRow) {
102,789,903✔
3398
    TAOS_SET_OBJ_ALIGNED(&w, pRow->win);
102,791,177✔
3399
  }
3400

3401
  // in case of typical time window, we can calculate time window directly.
3402
  if (w.skey > ts || w.ekey < ts) {
102,790,868✔
3403
    w = doCalculateTimeWindow(ts, pInterval);
68,169,541✔
3404
  }
3405

3406
  if (pInterval->interval != pInterval->sliding) {
102,790,799✔
3407
    // it is an sliding window query, in which sliding value is not equalled to
3408
    // interval value, and we need to find the first qualified time window.
3409
    w = getFirstQualifiedTimeWindow(ts, &w, pInterval, order);
1,784,875✔
3410
  }
3411

3412
  return w;
102,790,994✔
3413
}
3414

3415
TSKEY getNextTimeWindowStart(const SInterval* pInterval, TSKEY start, int32_t order) {
2,147,483,647✔
3416
  int32_t factor = GET_FORWARD_DIRECTION_FACTOR(order);
2,147,483,647✔
3417
  TSKEY   nextStart = taosTimeAdd(start, -1 * pInterval->offset, pInterval->offsetUnit, pInterval->precision, NULL);
2,147,483,647✔
3418
  nextStart = taosTimeAdd(nextStart, factor * pInterval->sliding, pInterval->slidingUnit, pInterval->precision, NULL);
2,147,483,647✔
3419
  nextStart = taosTimeAdd(nextStart, pInterval->offset, pInterval->offsetUnit, pInterval->precision, NULL);
2,147,483,647✔
3420
  return nextStart;
2,147,483,647✔
3421
}
3422

3423
void getNextTimeWindow(const SInterval* pInterval, STimeWindow* tw, int32_t order) {
2,147,483,647✔
3424
  tw->skey = getNextTimeWindowStart(pInterval, tw->skey, order);
2,147,483,647✔
3425
  tw->ekey = taosTimeAdd(tw->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
2,147,483,647✔
3426
}
2,147,483,647✔
3427

3428
bool hasLimitOffsetInfo(SLimitInfo* pLimitInfo) {
397,984,749✔
3429
  return (pLimitInfo->limit.limit != -1 || pLimitInfo->limit.offset != -1 || pLimitInfo->slimit.limit != -1 ||
791,722,013✔
3430
          pLimitInfo->slimit.offset != -1);
393,737,831✔
3431
}
3432

3433
bool hasSlimitOffsetInfo(SLimitInfo* pLimitInfo) {
×
3434
  return (pLimitInfo->slimit.limit != -1 || pLimitInfo->slimit.offset != -1);
×
3435
}
3436

3437
void initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo) {
480,698,631✔
3438
  SLimit limit = {.limit = getLimit(pLimit), .offset = getOffset(pLimit)};
480,698,631✔
3439
  SLimit slimit = {.limit = getLimit(pSLimit), .offset = getOffset(pSLimit)};
480,530,999✔
3440

3441
  pLimitInfo->limit = limit;
480,556,971✔
3442
  pLimitInfo->slimit = slimit;
480,559,717✔
3443
  pLimitInfo->remainOffset = limit.offset;
480,494,146✔
3444
  pLimitInfo->remainGroupOffset = slimit.offset;
480,532,571✔
3445
  pLimitInfo->numOfOutputRows = 0;
480,479,911✔
3446
  pLimitInfo->numOfOutputGroups = 0;
480,551,830✔
3447
  pLimitInfo->currentGroupId = 0;
480,488,393✔
3448
}
480,731,079✔
3449

3450
void resetLimitInfoForNextGroup(SLimitInfo* pLimitInfo) {
77,133,365✔
3451
  pLimitInfo->numOfOutputRows = 0;
77,133,365✔
3452
  pLimitInfo->remainOffset = pLimitInfo->limit.offset;
77,155,775✔
3453
}
77,118,484✔
3454

3455
int32_t tableListGetSize(const STableListInfo* pTableList, int32_t* pRes) {
510,241,036✔
3456
  if (taosArrayGetSize(pTableList->pTableList) != taosHashGetSize(pTableList->map)) {
510,241,036✔
3457
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3458
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3459
  }
3460
  (*pRes) = taosArrayGetSize(pTableList->pTableList);
510,212,634✔
3461
  return TSDB_CODE_SUCCESS;
510,245,315✔
3462
}
3463

3464
uint64_t tableListGetSuid(const STableListInfo* pTableList) { return pTableList->idInfo.suid; }
2,841,597✔
3465

3466
STableKeyInfo* tableListGetInfo(const STableListInfo* pTableList, int32_t index) {
201,528,947✔
3467
  if (taosArrayGetSize(pTableList->pTableList) == 0) {
201,528,947✔
3468
    return NULL;
2,018✔
3469
  }
3470

3471
  return taosArrayGet(pTableList->pTableList, index);
201,445,543✔
3472
}
3473

3474
int32_t tableListFind(const STableListInfo* pTableList, uint64_t uid, int32_t startIndex) {
46,152✔
3475
  int32_t numOfTables = taosArrayGetSize(pTableList->pTableList);
46,152✔
3476
  if (startIndex >= numOfTables) {
46,152✔
3477
    return -1;
×
3478
  }
3479

3480
  for (int32_t i = startIndex; i < numOfTables; ++i) {
477,039✔
3481
    STableKeyInfo* p = taosArrayGet(pTableList->pTableList, i);
477,039✔
3482
    if (!p) {
477,039✔
3483
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3484
      return -1;
×
3485
    }
3486
    if (p->uid == uid) {
477,039✔
3487
      return i;
46,152✔
3488
    }
3489
  }
3490
  return -1;
×
3491
}
3492

3493
void tableListGetSourceTableInfo(const STableListInfo* pTableList, uint64_t* psuid, uint64_t* uid, int32_t* type) {
376,413✔
3494
  *psuid = pTableList->idInfo.suid;
376,413✔
3495
  *uid = pTableList->idInfo.uid;
376,413✔
3496
  *type = pTableList->idInfo.tableType;
376,413✔
3497
}
376,413✔
3498

3499
uint64_t tableListGetTableGroupId(const STableListInfo* pTableList, uint64_t tableUid) {
697,319,735✔
3500
  int32_t* slot = taosHashGet(pTableList->map, &tableUid, sizeof(tableUid));
697,319,735✔
3501
  if (slot == NULL) {
697,413,899✔
3502
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3503
    return -1;
×
3504
  }
3505

3506
  STableKeyInfo* pKeyInfo = taosArrayGet(pTableList->pTableList, *slot);
697,413,899✔
3507
  if (pKeyInfo == NULL) {
697,560,016✔
3508
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3509
    return -1;
×
3510
  }
3511
  return pKeyInfo->groupId;
697,560,016✔
3512
}
3513

3514
// TODO handle the group offset info, fix it, the rule of group output will be broken by this function
3515
// int32_t tableListRemoveTableInfo(STableListInfo* pTableList, uint64_t uid) {
3516
//   int32_t code = TSDB_CODE_SUCCESS;
3517
//   int32_t lino = 0;
3518

3519
//   int32_t* slot = taosHashGet(pTableList->map, &uid, sizeof(uid));
3520
//   if (slot == NULL) {
3521
//     qDebug("table:%" PRIu64 " not found in table list", uid);
3522
//     return 0;
3523
//   }
3524

3525
//   taosArrayRemove(pTableList->pTableList, *slot);
3526
//   code = taosHashRemove(pTableList->map, &uid, sizeof(uid));
3527

3528
//   _end:
3529
//   if (code != TSDB_CODE_SUCCESS) {
3530
//     qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
3531
//   } else {
3532
//     qDebug("uid:%" PRIu64 ", remove from table list", uid);
3533
//   }
3534

3535
//   return code;
3536
// }
3537

3538
int32_t tableListAddTableInfo(STableListInfo* pTableList, uint64_t uid, uint64_t gid) {
1,021,991✔
3539
  int32_t code = TSDB_CODE_SUCCESS;
1,021,991✔
3540
  int32_t lino = 0;
1,021,991✔
3541
  if (pTableList->map == NULL) {
1,021,991✔
3542
    pTableList->map = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
×
3543
    QUERY_CHECK_NULL(pTableList->map, code, lino, _end, terrno);
×
3544
  }
3545

3546
  STableKeyInfo keyInfo = {.uid = uid, .groupId = gid};
1,024,042✔
3547
  void*         p = taosHashGet(pTableList->map, &uid, sizeof(uid));
1,023,456✔
3548
  if (p != NULL) {
1,021,991✔
3549
    qInfo("table:%" PRId64 " already in tableIdList, ignore it", uid);
634✔
3550
    goto _end;
634✔
3551
  }
3552

3553
  void* tmp = taosArrayPush(pTableList->pTableList, &keyInfo);
1,021,357✔
3554
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,022,529✔
3555

3556
  int32_t slot = (int32_t)taosArrayGetSize(pTableList->pTableList) - 1;
1,022,529✔
3557
  code = taosHashPut(pTableList->map, &uid, sizeof(uid), &slot, sizeof(slot));
1,022,822✔
3558
  if (code != TSDB_CODE_SUCCESS) {
1,026,338✔
3559
    // we have checked the existence of uid in hash map above
3560
    QUERY_CHECK_CONDITION((code != TSDB_CODE_DUP_KEY), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
3561
    taosArrayPopTailBatch(pTableList->pTableList, 1);  // let's pop the last element in the array list
×
3562
  }
3563

3564
_end:
1,026,972✔
3565
  if (code != TSDB_CODE_SUCCESS) {
1,025,214✔
3566
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3567
  } else {
3568
    qDebug("uid:%" PRIu64 ", groupId:%" PRIu64 " added into table list, slot:%d, total:%d", uid, gid, slot, slot + 1);
1,025,214✔
3569
  }
3570

3571
  return code;
1,028,144✔
3572
}
3573

3574
int32_t tableListGetGroupList(const STableListInfo* pTableList, int32_t ordinalGroupIndex, STableKeyInfo** pKeyInfo,
177,941,694✔
3575
                              int32_t* size) {
3576
  int32_t totalGroups = tableListGetOutputGroups(pTableList);
177,941,694✔
3577
  int32_t numOfTables = 0;
177,934,602✔
3578
  int32_t code = tableListGetSize(pTableList, &numOfTables);
177,953,731✔
3579
  if (code != TSDB_CODE_SUCCESS) {
177,939,382✔
3580
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3581
    return code;
×
3582
  }
3583

3584
  if (ordinalGroupIndex < 0 || ordinalGroupIndex >= totalGroups) {
177,939,382✔
3585
    return TSDB_CODE_INVALID_PARA;
×
3586
  }
3587

3588
  // here handle two special cases:
3589
  // 1. only one group exists, and 2. one table exists for each group.
3590
  if (totalGroups == 1) {
177,939,382✔
3591
    *size = numOfTables;
177,405,382✔
3592
    *pKeyInfo = (*size == 0) ? NULL : taosArrayGet(pTableList->pTableList, 0);
177,457,456✔
3593
    return TSDB_CODE_SUCCESS;
177,344,441✔
3594
  } else if (totalGroups == numOfTables) {
534,000✔
3595
    *size = 1;
454,865✔
3596
    *pKeyInfo = taosArrayGet(pTableList->pTableList, ordinalGroupIndex);
454,865✔
3597
    return TSDB_CODE_SUCCESS;
454,865✔
3598
  }
3599

3600
  int32_t offset = pTableList->groupOffset[ordinalGroupIndex];
79,135✔
3601
  if (ordinalGroupIndex < totalGroups - 1) {
54,679✔
3602
    *size = pTableList->groupOffset[ordinalGroupIndex + 1] - offset;
41,087✔
3603
  } else {
3604
    *size = numOfTables - offset;
13,592✔
3605
  }
3606

3607
  *pKeyInfo = taosArrayGet(pTableList->pTableList, offset);
54,679✔
3608
  return TSDB_CODE_SUCCESS;
54,679✔
3609
}
3610

3611
int32_t tableListGetOutputGroups(const STableListInfo* pTableList) { return pTableList->numOfOuputGroups; }
488,676,382✔
3612

3613
bool oneTableForEachGroup(const STableListInfo* pTableList) { return pTableList->oneTableForEachGroup; }
476,500✔
3614

3615
STableListInfo* tableListCreate() {
203,876,956✔
3616
  STableListInfo* pListInfo = taosMemoryCalloc(1, sizeof(STableListInfo));
203,876,956✔
3617
  if (pListInfo == NULL) {
203,800,104✔
3618
    return NULL;
×
3619
  }
3620

3621
  pListInfo->remainGroups = NULL;
203,800,104✔
3622
  pListInfo->pTableList = taosArrayInit(4, sizeof(STableKeyInfo));
203,829,366✔
3623
  if (pListInfo->pTableList == NULL) {
203,796,031✔
3624
    goto _error;
×
3625
  }
3626

3627
  pListInfo->map = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
203,858,800✔
3628
  if (pListInfo->map == NULL) {
203,979,273✔
3629
    goto _error;
×
3630
  }
3631

3632
  pListInfo->numOfOuputGroups = 1;
203,973,814✔
3633
  return pListInfo;
203,978,547✔
3634

3635
_error:
×
3636
  tableListDestroy(pListInfo);
×
3637
  return NULL;
×
3638
}
3639

3640
void tableListDestroy(STableListInfo* pTableListInfo) {
213,995,885✔
3641
  if (pTableListInfo == NULL) {
213,995,885✔
3642
    return;
10,097,805✔
3643
  }
3644

3645
  taosArrayDestroy(pTableListInfo->pTableList);
203,898,080✔
3646
  taosMemoryFreeClear(pTableListInfo->groupOffset);
203,851,904✔
3647

3648
  taosHashCleanup(pTableListInfo->map);
203,868,794✔
3649
  taosHashCleanup(pTableListInfo->remainGroups);
203,950,552✔
3650
  pTableListInfo->pTableList = NULL;
203,953,938✔
3651
  pTableListInfo->map = NULL;
203,953,462✔
3652
  taosMemoryFree(pTableListInfo);
203,944,284✔
3653
}
3654

3655
void tableListClear(STableListInfo* pTableListInfo) {
761,630✔
3656
  if (pTableListInfo == NULL) {
761,630✔
3657
    return;
×
3658
  }
3659

3660
  taosArrayClear(pTableListInfo->pTableList);
761,630✔
3661
  taosHashClear(pTableListInfo->map);
763,946✔
3662
  taosHashClear(pTableListInfo->remainGroups);
764,560✔
3663
  taosMemoryFree(pTableListInfo->groupOffset);
764,560✔
3664
  pTableListInfo->numOfOuputGroups = 1;
764,560✔
3665
  pTableListInfo->oneTableForEachGroup = false;
764,560✔
3666
}
3667

3668
static int32_t orderbyGroupIdComparFn(const void* p1, const void* p2) {
521,804,907✔
3669
  STableKeyInfo* pInfo1 = (STableKeyInfo*)p1;
521,804,907✔
3670
  STableKeyInfo* pInfo2 = (STableKeyInfo*)p2;
521,804,907✔
3671

3672
  if (pInfo1->groupId == pInfo2->groupId) {
521,804,907✔
3673
    return 0;
488,546,342✔
3674
  } else {
3675
    return pInfo1->groupId < pInfo2->groupId ? -1 : 1;
33,264,472✔
3676
  }
3677
}
3678

3679
int32_t sortTableGroup(STableListInfo* pTableListInfo) {
27,764,462✔
3680
  int32_t code = TSDB_CODE_SUCCESS;
27,764,462✔
3681
  taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
27,764,462✔
3682
  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
27,777,664✔
3683
  if (size == 0) {
27,767,724✔
3684
    pTableListInfo->numOfOuputGroups = 0;
×
3685
    return code;
×
3686
  }
3687

3688
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
27,767,724✔
3689
  if (!pList) {
27,774,216✔
3690
    code = terrno;
×
3691
    goto end;
×
3692
  }
3693

3694
  STableKeyInfo* pInfo = taosArrayGet(pTableListInfo->pTableList, 0);
27,774,216✔
3695
  if (pInfo == NULL) {
27,733,095✔
3696
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3697
    code = terrno;
×
3698
    goto end;
×
3699
  }
3700
  uint64_t gid = pInfo->groupId;
27,733,095✔
3701

3702
  int32_t start = 0;
27,757,998✔
3703
  void*   tmp = taosArrayPush(pList, &start);
27,762,454✔
3704
  if (tmp == NULL) {
27,762,454✔
3705
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3706
    code = terrno;
×
3707
    goto end;
×
3708
  }
3709

3710
  for (int32_t i = 1; i < size; ++i) {
147,692,369✔
3711
    pInfo = taosArrayGet(pTableListInfo->pTableList, i);
119,936,466✔
3712
    if (pInfo == NULL) {
119,935,515✔
3713
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3714
      code = terrno;
×
3715
      goto end;
×
3716
    }
3717
    if (pInfo->groupId != gid) {
119,935,515✔
3718
      tmp = taosArrayPush(pList, &i);
6,815,822✔
3719
      if (tmp == NULL) {
6,815,822✔
3720
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3721
        code = terrno;
×
3722
        goto end;
×
3723
      }
3724
      gid = pInfo->groupId;
6,815,822✔
3725
    }
3726
  }
3727

3728
  pTableListInfo->numOfOuputGroups = taosArrayGetSize(pList);
27,770,547✔
3729
  pTableListInfo->groupOffset = taosMemoryMalloc(sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
27,769,848✔
3730
  if (pTableListInfo->groupOffset == NULL) {
27,766,708✔
3731
    code = terrno;
×
3732
    goto end;
×
3733
  }
3734

3735
  memcpy(pTableListInfo->groupOffset, taosArrayGet(pList, 0), sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
27,742,577✔
3736

3737
end:
27,750,355✔
3738
  taosArrayDestroy(pList);
27,751,300✔
3739
  return code;
27,721,125✔
3740
}
3741

3742
int32_t buildGroupIdMapForAllTables(STableListInfo* pTableListInfo, SReadHandle* pHandle, SScanPhysiNode* pScanNode,
172,719,667✔
3743
                                    SNodeList* group, bool groupSort, uint8_t* digest, SStorageAPI* pAPI, SHashObj* groupIdMap) {
3744
  int32_t code = TSDB_CODE_SUCCESS;
172,719,667✔
3745

3746
  bool   groupByTbname = groupbyTbname(group);
172,719,667✔
3747
  size_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
172,729,060✔
3748
  if (!numOfTables) {
172,674,893✔
3749
    return code;
3,330✔
3750
  }
3751
  qDebug("numOfTables:%zu, groupByTbname:%d, group:%p", numOfTables, groupByTbname, group);
172,671,563✔
3752
  if (group == NULL || groupByTbname) {
172,619,590✔
3753
    if (tsCountAlwaysReturnValue && QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode) &&
168,655,675✔
3754
        ((STableScanPhysiNode*)pScanNode)->needCountEmptyTable) {
127,754,854✔
3755
      pTableListInfo->remainGroups =
14,377,472✔
3756
          taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
14,376,969✔
3757
      if (pTableListInfo->remainGroups == NULL) {
14,377,472✔
3758
        return terrno;
×
3759
      }
3760

3761
      for (int i = 0; i < numOfTables; i++) {
67,487,840✔
3762
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
53,110,536✔
3763
        if (!info) {
53,107,518✔
3764
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3765
          return terrno;
×
3766
        }
3767
        info->groupId = groupByTbname ? info->uid : 0;
53,107,518✔
3768
        int32_t tempRes = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId),
53,111,039✔
3769
                                      &(info->uid), sizeof(info->uid));
53,110,033✔
3770
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
53,110,368✔
3771
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3772
          return tempRes;
×
3773
        }
3774
      }
3775
    } else {
3776
      for (int32_t i = 0; i < numOfTables; i++) {
661,259,497✔
3777
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
507,004,991✔
3778
        if (!info) {
506,860,424✔
3779
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3780
          return terrno;
×
3781
        }
3782
        info->groupId = groupByTbname ? info->uid : 0;
506,860,424✔
3783
        
3784
      }
3785
    }
3786
    if (groupIdMap && group != NULL){
168,631,810✔
3787
      getColInfoResultForGroupbyForStream(pHandle->vnode, group, pTableListInfo, pAPI, groupIdMap);
66,922✔
3788
    }
3789

3790
    pTableListInfo->oneTableForEachGroup = groupByTbname;
168,631,810✔
3791
    if (numOfTables == 1 && pTableListInfo->idInfo.tableType == TSDB_CHILD_TABLE) {
168,750,015✔
3792
      pTableListInfo->oneTableForEachGroup = true;
36,450,515✔
3793
    }
3794

3795
    if (groupSort && groupByTbname) {
168,726,079✔
3796
      taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
2,189,234✔
3797
      pTableListInfo->numOfOuputGroups = numOfTables;
2,189,020✔
3798
    } else if (groupByTbname && pScanNode->groupOrderScan) {
166,536,845✔
3799
      pTableListInfo->numOfOuputGroups = numOfTables;
24,429✔
3800
    } else {
3801
      pTableListInfo->numOfOuputGroups = 1;
166,512,265✔
3802
    }
3803
    if (groupSort || pScanNode->groupOrderScan) {
168,739,569✔
3804
      code = sortTableGroup(pTableListInfo);
27,647,103✔
3805
    }
3806
  } else {
3807
    bool initRemainGroups = false;
3,963,915✔
3808
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode)) {
3,963,915✔
3809
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pScanNode;
3,874,684✔
3810
      if (tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable &&
3,874,684✔
3811
          !(groupSort || pScanNode->groupOrderScan)) {
1,947,934✔
3812
        initRemainGroups = true;
1,924,474✔
3813
      }
3814
    }
3815

3816
    code = getColInfoResultForGroupby(pHandle->vnode, group, pTableListInfo, digest, pAPI, initRemainGroups, groupIdMap);
3,963,915✔
3817
    if (code != TSDB_CODE_SUCCESS) {
3,963,016✔
3818
      return code;
×
3819
    }
3820

3821
    if (pScanNode->groupOrderScan) pTableListInfo->numOfOuputGroups = taosArrayGetSize(pTableListInfo->pTableList);
3,963,016✔
3822

3823
    if (groupSort || pScanNode->groupOrderScan) {
3,963,411✔
3824
      code = sortTableGroup(pTableListInfo);
134,830✔
3825
    }
3826
  }
3827

3828
  // add all table entry in the hash map
3829
  size_t size = taosArrayGetSize(pTableListInfo->pTableList);
172,646,204✔
3830
  for (int32_t i = 0; i < size; ++i) {
756,673,365✔
3831
    STableKeyInfo* p = taosArrayGet(pTableListInfo->pTableList, i);
583,988,504✔
3832
    if (!p) {
583,799,460✔
3833
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3834
      return terrno;
×
3835
    }
3836
    int32_t tempRes = taosHashPut(pTableListInfo->map, &p->uid, sizeof(uint64_t), &i, sizeof(int32_t));
583,799,460✔
3837
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
584,015,706✔
3838
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3839
      return tempRes;
×
3840
    }
3841
  }
3842

3843
  return code;
172,756,321✔
3844
}
3845

3846
int32_t createScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
188,561,461✔
3847
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
3848
                                SExecTaskInfo* pTaskInfo, SHashObj* groupIdMap) {
3849
  int64_t     st = taosGetTimestampUs();
188,534,947✔
3850
  const char* idStr = GET_TASKID(pTaskInfo);
188,534,947✔
3851

3852
  if (pHandle == NULL) {
188,275,392✔
3853
    qError("invalid handle, in creating operator tree, %s", idStr);
×
3854
    return TSDB_CODE_INVALID_PARA;
×
3855
  }
3856

3857
  if (pHandle->uid != 0) {
188,275,392✔
3858
    pScanNode->uid = pHandle->uid;
60,617✔
3859
    pScanNode->tableType = TSDB_CHILD_TABLE;
60,617✔
3860
  }
3861
  uint8_t digest[17] = {0};
188,535,752✔
3862
  int32_t code = getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, digest, idStr,
188,504,966✔
3863
                              &pTaskInfo->storageAPI, pTaskInfo->pStreamRuntimeInfo);
188,551,694✔
3864
  if (code != TSDB_CODE_SUCCESS) {
188,625,215✔
3865
    qError("failed to getTableList, code:%s", tstrerror(code));
914✔
3866
    return code;
914✔
3867
  }
3868

3869
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
188,624,301✔
3870

3871
  int64_t st1 = taosGetTimestampUs();
188,650,195✔
3872
  pTaskInfo->cost.extractListTime = (st1 - st) / 1000.0;
188,650,195✔
3873
  qDebug("extract queried table list completed, %d tables, elapsed time:%.2f ms %s", numOfTables,
188,694,551✔
3874
         pTaskInfo->cost.extractListTime, idStr);
3875

3876
  if (numOfTables == 0) {
188,611,403✔
3877
    qDebug("no table qualified for query, %s", idStr);
15,970,847✔
3878
    return TSDB_CODE_SUCCESS;
15,971,631✔
3879
  }
3880

3881
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, digest, &pTaskInfo->storageAPI, groupIdMap);
172,640,556✔
3882
  if (code != TSDB_CODE_SUCCESS) {
172,722,484✔
3883
    return code;
×
3884
  }
3885

3886
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
172,727,062✔
3887
  qDebug("generate group id map completed, elapsed time:%.2f ms %s", pTaskInfo->cost.groupIdMapTime, idStr);
172,721,245✔
3888

3889
  return TSDB_CODE_SUCCESS;
172,706,407✔
3890
}
3891

3892
char* getStreamOpName(uint16_t opType) {
5,578,406✔
3893
  switch (opType) {
5,578,406✔
3894
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
×
3895
      return "stream scan";
×
3896
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
5,412,517✔
3897
      return "project";
5,412,517✔
3898
    case QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW:
165,889✔
3899
      return "external window";
165,889✔
3900
  }
3901
  return "error name";
×
3902
}
3903

3904
void printDataBlock(SSDataBlock* pBlock, const char* flag, const char* taskIdStr, int64_t qId) {
628,359,859✔
3905
  if (qDebugFlag & DEBUG_TRACE) {
628,359,859✔
3906
    if (!pBlock) {
1,132,159✔
3907
      qDebug("%" PRIx64 " %s %s %s: Block is Null", qId, taskIdStr, flag, __func__);
3,655✔
3908
      return;
3,655✔
3909
    } else if (pBlock->info.rows == 0) {
1,128,504✔
3910
      qDebug("%" PRIx64 " %s %s %s: Block is Empty. block type %d", qId, taskIdStr, flag, __func__, pBlock->info.type);
739✔
3911
      return;
739✔
3912
    }
3913
    
3914
    char*   pBuf = NULL;
1,127,765✔
3915
    int32_t code = dumpBlockData(pBlock, flag, &pBuf, taskIdStr, qId);
1,127,765✔
3916
    if (code == 0) {
1,127,765✔
3917
      qDebugL("%" PRIx64 " %s %s", qId, __func__, pBuf);
1,127,765✔
3918
      taosMemoryFree(pBuf);
1,127,765✔
3919
    }
3920
  }
3921
}
3922

3923
void printSpecDataBlock(SSDataBlock* pBlock, const char* flag, const char* opStr, const char* taskIdStr) {
×
3924
  if (!pBlock) {
×
3925
    qDebug("%s===stream===%s %s: Block is Null", taskIdStr, flag, opStr);
×
3926
    return;
×
3927
  } else if (pBlock->info.rows == 0) {
×
3928
    qDebug("%s===stream===%s %s: Block is Empty. block type %d.skey:%" PRId64 ",ekey:%" PRId64 ",version%" PRId64,
×
3929
           taskIdStr, flag, opStr, pBlock->info.type, pBlock->info.window.skey, pBlock->info.window.ekey,
3930
           pBlock->info.version);
3931
    return;
×
3932
  }
3933
  if (qDebugFlag & DEBUG_TRACE) {
×
3934
    char* pBuf = NULL;
×
3935
    char  flagBuf[64];
×
3936
    snprintf(flagBuf, sizeof(flagBuf), "%s %s", flag, opStr);
×
3937
    int32_t code = dumpBlockData(pBlock, flagBuf, &pBuf, taskIdStr, 0);
×
3938
    if (code == 0) {
×
3939
      qDebug("%s", pBuf);
×
3940
      taosMemoryFree(pBuf);
×
3941
    }
3942
  }
3943
}
3944

3945
TSKEY getStartTsKey(STimeWindow* win, const TSKEY* tsCols) { return tsCols == NULL ? win->skey : tsCols[0]; }
11,819,890✔
3946

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

3950
  int64_t duration = pWin->ekey > pWin->skey ? pWin->ekey - pWin->skey + delta : pWin->skey - pWin->ekey + delta;
2,147,483,647✔
3951
  ts[2] = duration;            // set the duration
2,147,483,647✔
3952
  ts[3] = pWin->skey;          // window start key
2,147,483,647✔
3953
  ts[4] = pWin->ekey + delta;  // window end key
2,147,483,647✔
3954
}
2,147,483,647✔
3955

3956
int32_t compKeys(const SArray* pSortGroupCols, const char* oldkeyBuf, int32_t oldKeysLen, const SSDataBlock* pBlock,
2,147,483,647✔
3957
                 int32_t rowIndex) {
3958
  SColumnDataAgg* pColAgg = NULL;
2,147,483,647✔
3959
  const char*     isNull = oldkeyBuf;
2,147,483,647✔
3960
  const char*     p = oldkeyBuf + sizeof(int8_t) * pSortGroupCols->size;
2,147,483,647✔
3961

3962
  for (int32_t i = 0; i < pSortGroupCols->size; ++i) {
2,147,483,647✔
3963
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
2,147,483,647✔
3964
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
2,147,483,647✔
3965
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
2,147,483,647✔
3966

3967
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
2,147,483,647✔
3968
      if (isNull[i] != 1) return 1;
255,264,810✔
3969
    } else {
3970
      if (isNull[i] != 0) return 1;
2,147,483,647✔
3971
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,147,483,647✔
3972
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,147,483,647✔
3973
        int32_t len = getJsonValueLen(val);
×
3974
        if (memcmp(p, val, len) != 0) return 1;
×
3975
        p += len;
×
3976
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,147,483,647✔
3977
        if (IS_STR_DATA_BLOB(pCol->type)) {
1,178,181,279✔
3978
          if (memcmp(p, val, blobDataTLen(val)) != 0) return 1;
×
3979
          p += blobDataTLen(val);
×
3980
        } else {
3981
          if (memcmp(p, val, varDataTLen(val)) != 0) return 1;
1,178,614,783✔
3982
          p += varDataTLen(val);
1,177,910,776✔
3983
        }
3984
      } else {
3985
        if (0 != memcmp(p, val, pCol->bytes)) return 1;
2,147,483,647✔
3986
        p += pCol->bytes;
2,147,483,647✔
3987
      }
3988
    }
3989
  }
3990
  if ((int32_t)(p - oldkeyBuf) != oldKeysLen) return 1;
2,147,483,647✔
3991
  return 0;
2,147,483,647✔
3992
}
3993

3994
int32_t buildKeys(char* keyBuf, const SArray* pSortGroupCols, const SSDataBlock* pBlock, int32_t rowIndex) {
1,560,964✔
3995
  uint32_t        colNum = pSortGroupCols->size;
1,560,964✔
3996
  SColumnDataAgg* pColAgg = NULL;
1,560,964✔
3997
  char*           isNull = keyBuf;
1,560,964✔
3998
  char*           p = keyBuf + sizeof(int8_t) * colNum;
1,560,964✔
3999

4000
  for (int32_t i = 0; i < colNum; ++i) {
4,271,238✔
4001
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
2,710,274✔
4002
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
2,710,274✔
4003
    if (pCol->slotId > pBlock->pDataBlock->size) continue;
2,710,274✔
4004

4005
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
2,710,274✔
4006

4007
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
5,420,548✔
4008
      isNull[i] = 1;
131,100✔
4009
    } else {
4010
      isNull[i] = 0;
2,579,174✔
4011
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,579,174✔
4012
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,579,174✔
4013
        int32_t len = getJsonValueLen(val);
×
4014
        memcpy(p, val, len);
×
4015
        p += len;
×
4016
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,579,174✔
4017
        if (IS_STR_DATA_BLOB(pCol->type)) {
962,274✔
4018
          blobDataCopy(p, val);
×
4019
          p += blobDataTLen(val);
×
4020
        } else {
4021
          varDataCopy(p, val);
962,274✔
4022
          p += varDataTLen(val);
962,274✔
4023
        }
4024
      } else {
4025
        memcpy(p, val, pCol->bytes);
1,616,900✔
4026
        p += pCol->bytes;
1,616,900✔
4027
      }
4028
    }
4029
  }
4030
  return (int32_t)(p - keyBuf);
1,560,964✔
4031
}
4032

4033
uint64_t calcGroupId(char* pData, int32_t len) {
2,147,483,647✔
4034
  T_MD5_CTX context;
2,147,483,647✔
4035
  tMD5Init(&context);
2,147,483,647✔
4036
  tMD5Update(&context, (uint8_t*)pData, len);
2,147,483,647✔
4037
  tMD5Final(&context);
2,147,483,647✔
4038

4039
  // NOTE: only extract the initial 8 bytes of the final MD5 digest
4040
  uint64_t id = 0;
2,147,483,647✔
4041
  memcpy(&id, context.digest, sizeof(uint64_t));
2,147,483,647✔
4042
  if (0 == id) memcpy(&id, context.digest + 8, sizeof(uint64_t));
2,147,483,647✔
4043
  return id;
2,147,483,647✔
4044
}
4045

4046
SNodeList* makeColsNodeArrFromSortKeys(SNodeList* pSortKeys) {
103,132✔
4047
  SNode*     node;
4048
  SNodeList* ret = NULL;
103,132✔
4049
  FOREACH(node, pSortKeys) {
314,640✔
4050
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)node;
211,508✔
4051
    int32_t           code = nodesListMakeAppend(&ret, pSortKey->pExpr);
211,508✔
4052
    if (code != TSDB_CODE_SUCCESS) {
211,508✔
4053
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4054
      terrno = code;
×
4055
      return NULL;
×
4056
    }
4057
  }
4058
  return ret;
103,132✔
4059
}
4060

4061
int32_t extractKeysLen(const SArray* keys, int32_t* pLen) {
103,132✔
4062
  int32_t code = TSDB_CODE_SUCCESS;
103,132✔
4063
  int32_t lino = 0;
103,132✔
4064
  int32_t len = 0;
103,132✔
4065
  int32_t keyNum = taosArrayGetSize(keys);
103,132✔
4066
  for (int32_t i = 0; i < keyNum; ++i) {
262,200✔
4067
    SColumn* pCol = (SColumn*)taosArrayGet(keys, i);
159,068✔
4068
    QUERY_CHECK_NULL(pCol, code, lino, _end, terrno);
159,068✔
4069
    len += pCol->bytes;
159,068✔
4070
  }
4071
  len += sizeof(int8_t) * keyNum;  // null flag
103,132✔
4072
  *pLen = len;
103,132✔
4073

4074
_end:
103,132✔
4075
  if (code != TSDB_CODE_SUCCESS) {
103,132✔
4076
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4077
  }
4078
  return code;
103,132✔
4079
}
4080

4081
int32_t parseErrorMsgFromAnalyticServer(SJson* pJson, const char* pId) {
×
4082
  int32_t code = TSDB_CODE_ANA_ANODE_RETURN_ERROR;
×
4083
  if (pJson == NULL) {
×
4084
    return code;
×
4085
  }
4086

4087
  char    pMsg[1024] = {0};
×
4088
  int32_t ret = tjsonGetStringValue(pJson, "msg", pMsg);
×
4089

4090
  if (ret == 0) {
×
4091
    qError("%s failed to exec imputation, msg:%s", pId, pMsg);
×
4092
    if (strstr(pMsg, "white noise") != NULL) {
×
4093
      code = TSDB_CODE_ANA_WN_DATA;
×
4094
    } else if (strstr(pMsg, "white-noise") != NULL) {
×
4095
      code = TSDB_CODE_ANA_WN_DATA;
×
4096
    } else if (strstr(pMsg, "[Errno 111] Connection refused") != NULL) {
×
4097
      code = TSDB_CODE_ANA_ALGO_NOT_LOAD;
×
4098
    }
4099
  } else {
4100
    qError("%s failed to extract msg from server, unknown error", pId);
×
4101
  }
4102

4103
  return code;
×
4104
}
4105

4106

4107
int32_t createExprSubQResBlock(SSDataBlock** ppBlock, SDataType* pResType) {
1,924,146✔
4108
  int32_t code = 0;
1,924,146✔
4109
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
1,924,146✔
4110
  if (pBlock == NULL) {
1,922,066✔
4111
    return terrno;
×
4112
  }
4113

4114
  pBlock->pDataBlock = taosArrayInit(1, sizeof(SColumnInfoData));
1,922,066✔
4115
  if (pBlock->pDataBlock == NULL) {
1,923,626✔
4116
    code = terrno;
×
4117
    taosMemoryFree(pBlock);
×
4118
    return code;
×
4119
  }
4120

4121
  SColumnInfoData idata =
1,924,146✔
4122
      createColumnInfoData(pResType->type, pResType->bytes, 0);
1,924,146✔
4123
  idata.info.scale = pResType->scale;
1,924,146✔
4124
  idata.info.precision = pResType->precision;
1,924,146✔
4125

4126
  code = blockDataAppendColInfo(pBlock, &idata);
1,923,626✔
4127
  if (code != TSDB_CODE_SUCCESS) {
1,924,146✔
4128
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4129
    blockDataDestroy(pBlock);
×
4130
    *ppBlock = NULL;
×
4131
    return code;
×
4132
  }
4133

4134
  *ppBlock = pBlock;
1,924,146✔
4135

4136
  return code;
1,923,626✔
4137
}
4138

4139

4140
int32_t extractSingleRspBlock(SRetrieveTableRsp* pRetrieveRsp, SSDataBlock* pb) {
1,924,146✔
4141
  int32_t            code = TSDB_CODE_SUCCESS;
1,924,146✔
4142
  int32_t            lino = 0;
1,924,146✔
4143
  void*              decompBuf = NULL;
1,924,146✔
4144

4145
  char* pNextStart = pRetrieveRsp->data;
1,924,146✔
4146
  char* pStart = pNextStart;
1,923,626✔
4147

4148
  int32_t index = 0;
1,924,146✔
4149

4150
  if (pRetrieveRsp->compressed) {  // decompress the data
1,924,146✔
4151
    decompBuf = taosMemoryMalloc(pRetrieveRsp->payloadLen);
×
4152
    QUERY_CHECK_NULL(decompBuf, code, lino, _end, terrno);
×
4153
  }
4154

4155
  int32_t compLen = *(int32_t*)pStart;
1,924,146✔
4156
  pStart += sizeof(int32_t);
1,924,146✔
4157

4158
  int32_t rawLen = *(int32_t*)pStart;
1,924,666✔
4159
  pStart += sizeof(int32_t);
1,923,626✔
4160
  QUERY_CHECK_CONDITION((compLen <= rawLen && compLen != 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
1,923,626✔
4161

4162
  pNextStart = pStart + compLen;
1,923,626✔
4163
  if (pRetrieveRsp->compressed && (compLen < rawLen)) {
1,924,146✔
4164
    int32_t t = tsDecompressString(pStart, compLen, 1, decompBuf, rawLen, ONE_STAGE_COMP, NULL, 0);
×
4165
    QUERY_CHECK_CONDITION((t == rawLen), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
4166
    pStart = decompBuf;
×
4167
  }
4168

4169
  code = blockDecodeInternal(pb, pStart, (const char**)&pStart);
1,922,586✔
4170
  if (code != 0) {
1,924,146✔
4171
    taosMemoryFreeClear(pRetrieveRsp);
×
4172
    goto _end;
×
4173
  }
4174

4175
_end:
1,924,146✔
4176
  if (code != TSDB_CODE_SUCCESS) {
1,924,146✔
4177
    blockDataDestroy(pb);
×
4178
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4179
  }
4180
  return code;
1,924,146✔
4181
}
4182

4183
int32_t setValueFromResBlock(STaskSubJobCtx* ctx, SRemoteValueNode* pRes, SSDataBlock* pBlock) {
1,738,356✔
4184
  int32_t code = 0;
1,738,356✔
4185
  bool needFree = true;
1,738,356✔
4186
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
1,738,356✔
4187
  if (NULL == pBlock->pDataBlock || 1 != colNum || pBlock->info.rows > 1) {
1,736,796✔
4188
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d, rows:%" PRId64, 
1,560✔
4189
      ctx->idStr, pBlock->pDataBlock, colNum, pBlock->info.rows);
4190
    return TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
×
4191
  }
4192
  
4193
  pRes->val.node.type = QUERY_NODE_VALUE;
1,736,796✔
4194
  pRes->val.flag &= (~VALUE_FLAG_VAL_UNSET);
1,737,836✔
4195
  pRes->val.translate = true;
1,737,316✔
4196
  
4197
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
1,737,836✔
4198
  if (colDataIsNull_s(pCol, 0)) {
1,735,756✔
4199
    pRes->val.isNull = true;
107,640✔
4200
  } else {
4201
    code = nodesSetValueNodeValueExt(&pRes->val, colDataGetData(pCol, 0), &needFree);
1,628,116✔
4202
  }
4203

4204
  if (!needFree) {
1,737,836✔
4205
    pCol->pData = NULL;
×
4206
  }
4207

4208
  return code;
1,737,836✔
4209
}
4210

4211
void handleRemoteValueRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp) {
1,901,636✔
4212
  SSDataBlock* pResBlock = NULL;
1,901,636✔
4213

4214
  qDebug("%s scl fetch rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
1,901,636✔
4215

4216
  if (pRsp->numOfRows > 1 || pRsp->numOfBlocks > 1 || !pRsp->completed) {
1,901,636✔
4217
    qError("%s invalid scl fetch rsp received, subQIdx:%d, rows:%" PRId64 ", blocks:%d, completed:%d", 
2,600✔
4218
      ctx->idStr, pParam->subQIdx, pRsp->numOfRows, pRsp->numOfBlocks, pRsp->completed);
4219
    ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
2,600✔
4220

4221
    return;
×
4222
  }
4223

4224
  if (0 == pRsp->numOfRows) {
1,900,596✔
4225
    SRemoteValueNode* pRemote = (SRemoteValueNode*)pParam->pRes;
163,280✔
4226
    pRemote->val.node.type = QUERY_NODE_VALUE;
163,280✔
4227
    pRemote->val.isNull = true;
163,280✔
4228
    pRemote->val.translate = true;
163,280✔
4229
    pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
163,280✔
4230
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
163,280✔
4231

4232
    return;
163,280✔
4233
  }
4234
  
4235
  ctx->code = createExprSubQResBlock(&pResBlock, &((SRemoteValueNode*)pParam->pRes)->val.node.resType);
1,736,796✔
4236
  if (TSDB_CODE_SUCCESS == ctx->code) {
1,737,836✔
4237
    ctx->code = blockDataEnsureCapacity(pResBlock, 1);
1,737,836✔
4238
  }
4239
  if (TSDB_CODE_SUCCESS == ctx->code) {
1,737,836✔
4240
    ctx->code = extractSingleRspBlock(pRsp, pResBlock);
1,737,836✔
4241
  }
4242
  if (TSDB_CODE_SUCCESS == ctx->code) {
1,738,356✔
4243
    ctx->code = setValueFromResBlock(ctx, (SRemoteValueNode*)pParam->pRes, pResBlock);
1,737,316✔
4244
  }
4245
  if (TSDB_CODE_SUCCESS == ctx->code) {
1,737,836✔
4246
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
1,737,836✔
4247
  }
4248

4249
  blockDataDestroy(pResBlock);  
1,735,756✔
4250
}
4251

4252

4253
int32_t updateValueListFromResBlock(STaskSubJobCtx* ctx, SRemoteValueListNode* pRes, SSDataBlock* pBlock) {
185,790✔
4254
  int32_t code = 0, lino = 0;
185,790✔
4255
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
185,790✔
4256
  if (NULL == pBlock->pDataBlock || 1 != colNum) {
186,830✔
4257
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d", ctx->idStr, pBlock->pDataBlock, colNum);
1,040✔
4258
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4259
  }
4260

4261
  pRes->hasValue = true;
185,790✔
4262
  
4263
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
186,310✔
4264
  TAOS_CHECK_EXIT(scalarBuildRemoteListHash(pRes, pCol, pBlock->info.rows));
186,310✔
4265

4266
_exit:
185,441✔
4267

4268
  if (code) {
185,441✔
4269
    qError("%s %s failed with error: %s", ctx->idStr, __func__, tstrerror(code));
×
4270
  }
4271
  
4272
  return code;
185,441✔
4273
}
4274

4275

4276

4277
void handleRemoteValueListRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
185,790✔
4278
  SSDataBlock* pResBlock = NULL;
185,790✔
4279
  SRemoteValueListNode* pRemote = (SRemoteValueListNode*)pParam->pRes;
186,310✔
4280

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

4283
  if (pRsp->numOfRows > 0) {
185,790✔
4284
    ctx->code = createExprSubQResBlock(&pResBlock, &((SExprNode*)pParam->pRes)->resType);
185,790✔
4285
    if (TSDB_CODE_SUCCESS == ctx->code) {
185,790✔
4286
      ctx->code = blockDataEnsureCapacity(pResBlock, pRsp->numOfRows);
185,790✔
4287
    }
4288
    if (TSDB_CODE_SUCCESS == ctx->code) {
185,790✔
4289
      ctx->code = extractSingleRspBlock(pRsp, pResBlock);
186,830✔
4290
    }
4291
    if (TSDB_CODE_SUCCESS == ctx->code) {
185,790✔
4292
      ctx->code = updateValueListFromResBlock(ctx, pRemote, pResBlock);
186,830✔
4293
    }
4294
    if (TSDB_CODE_SUCCESS == ctx->code && pRsp->completed) {
185,441✔
4295
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
144,235✔
4296
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
144,235✔
4297
    }
4298

4299
    blockDataDestroy(pResBlock);  
186,310✔
4300
  } else if (0 == pRsp->numOfRows && pRsp->completed) {
×
4301
    if (!pRemote->hasValue) {
×
4302
      ctx->code = scalarBuildRemoteListHash(pRemote, NULL, 0);
×
4303
    }
4304
    if (TSDB_CODE_SUCCESS == ctx->code) {    
×
4305
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
×
4306
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
×
4307
    }
4308
  }
4309

4310
  *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
185,270✔
4311

4312
  if (!(*fetchDone)) {
184,750✔
4313
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes);
42,075✔
4314
    if (TSDB_CODE_SUCCESS != code) {
42,075✔
4315
      ctx->code = code;
×
4316
      *fetchDone = true;
×
4317
    }
4318
  }
4319
}
185,790✔
4320

4321

4322
int32_t remoteFetchCallBack(void* param, SDataBuf* pMsg, int32_t code) {
2,913,186✔
4323
  SScalarFetchParam* pParam = (SScalarFetchParam*)param;
2,913,186✔
4324
  STaskSubJobCtx* ctx = pParam->pSubJobCtx;
2,913,186✔
4325
  char idStr[64];
2,914,226✔
4326
  
4327
  taosMemoryFreeClear(pMsg->pEpSet);
2,914,226✔
4328

4329
  if (NULL == ctx) {
2,912,666✔
4330
    qWarn("scl fetch ctx not exists since it may have been released");
4,680✔
4331
    goto _exit;
4,680✔
4332
  }
4333

4334
  if (qDebugFlag & DEBUG_DEBUG) {
2,907,986✔
4335
    strncpy(idStr, ctx->idStr, sizeof(idStr));
92,706✔
4336
  }
4337
  
4338
  qDebug("%s subQIdx %d got rsp, blockIdx:%" PRId64 ", code:%d, rsp:%p", ctx->idStr, pParam->subQIdx, ctx->blockIdx, code, pMsg->pData);
2,907,986✔
4339

4340
  taosWLockLatch(&ctx->lock);
2,907,986✔
4341
  ctx->param = NULL;
2,909,546✔
4342
  taosWUnLockLatch(&ctx->lock);
2,909,026✔
4343

4344
  if (ctx->transporterId > 0) {
2,909,546✔
4345
    int32_t ret = asyncFreeConnById(ctx->rpcHandle, ctx->transporterId);
2,909,026✔
4346
    if (ret != 0) {
2,909,026✔
4347
      qDebug("%s failed to free subQ rpc handle, code:%s, subQIdx:%d", ctx->idStr, tstrerror(ret), pParam->subQIdx);
×
4348
    }
4349
    ctx->transporterId = -1;
2,909,026✔
4350
  }
4351

4352
  if (0 == code && NULL == pMsg->pData) {
2,909,546✔
4353
    qError("%s invalid rsp msg, msgType:%d, len:%d", ctx->idStr, pMsg->msgType, pMsg->len);
×
4354
    code = TSDB_CODE_QRY_INVALID_MSG;
×
4355
  }
4356

4357
  if (code == TSDB_CODE_SUCCESS) {
2,908,506✔
4358
    SRetrieveTableRsp* pRsp = pMsg->pData;
2,087,426✔
4359
    pRsp->numOfRows = htobe64(pRsp->numOfRows);
2,087,426✔
4360
    pRsp->compLen = htonl(pRsp->compLen);
2,087,426✔
4361
    pRsp->payloadLen = htonl(pRsp->payloadLen);
2,087,426✔
4362
    pRsp->numOfCols = htonl(pRsp->numOfCols);
2,087,426✔
4363
    pRsp->useconds = htobe64(pRsp->useconds);
2,087,426✔
4364
    pRsp->numOfBlocks = htonl(pRsp->numOfBlocks);
2,086,906✔
4365

4366
    qDebug("%s subQIdx %d blockIdx:%" PRIu64 " rsp detail, numOfBlocks:%d, numOfRows:%" PRId64 ", completed:%d", 
2,086,906✔
4367
      ctx->idStr, pParam->subQIdx, ctx->blockIdx, pRsp->numOfBlocks, pRsp->numOfRows, pRsp->completed);
4368

4369
    ctx->blockIdx++;
2,086,906✔
4370

4371
    switch (nodeType(pParam->pRes)) {
2,086,906✔
4372
      case QUERY_NODE_REMOTE_VALUE:
1,901,116✔
4373
        handleRemoteValueRes(pParam, ctx, pRsp);
1,901,116✔
4374
        break;
1,900,076✔
4375
      case QUERY_NODE_REMOTE_VALUE_LIST: {
186,830✔
4376
        bool fetchDone = false;
186,830✔
4377
        handleRemoteValueListRes(pParam, ctx, pRsp, &fetchDone);
186,830✔
4378
        qDebug("%s subQIdx %d handle remote value list finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
186,310✔
4379
        if (!fetchDone) {
185,270✔
4380
          goto _exit;
42,075✔
4381
        }
4382
        break;
143,195✔
4383
      }  
4384
      default:
×
4385
        qError("%s invalid scl fetch res node %d, subQIdx:%d", ctx->idStr, nodeType(pParam->pRes), pParam->subQIdx);
×
4386
        ctx->code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4387
        break;
×
4388
    }
4389
  } else {
4390
    ctx->code = rpcCvtErrCode(code);
821,080✔
4391
    if (ctx->code != code) {
821,080✔
4392
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s, cvted error: %s", ctx->idStr, pParam->subQIdx,
×
4393
             tstrerror(code), tstrerror(ctx->code));
4394
    } else {
4395
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s", ctx->idStr, pParam->subQIdx, tstrerror(code));
821,080✔
4396
    }
4397
  }
4398

4399
  qDebug("%s subQIdx %d sem_post subQ ready", ctx->idStr, pParam->subQIdx);
2,866,431✔
4400
  
4401
  code = tsem_post(&pParam->pSubJobCtx->ready);
2,866,431✔
4402
  if (code != TSDB_CODE_SUCCESS) {
2,867,471✔
4403
    qError("failed to invoke post when scl fetch rsp is ready, code:%s", tstrerror(code));
×
4404
  }
4405

4406
_exit:
2,912,666✔
4407

4408
  taosMemoryFree(pMsg->pData);
2,914,226✔
4409

4410
  return code;
2,913,706✔
4411
}
4412

4413

4414
int32_t sendFetchRemoteNodeReq(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes) {
2,911,626✔
4415
  int32_t          code = TSDB_CODE_SUCCESS;
2,911,626✔
4416
  int32_t          lino = 0;
2,911,626✔
4417
  SDownstreamSourceNode* pSource = (SDownstreamSourceNode*)taosArrayGetP(ctx->subEndPoints, subQIdx);
2,911,626✔
4418

4419
  SResFetchReq req = {0};
2,903,306✔
4420
  req.header.vgId = pSource->addr.nodeId;
2,902,266✔
4421
  req.sId = pSource->sId;
2,900,706✔
4422
  req.clientId = pSource->clientId;
2,904,866✔
4423
  req.taskId = pSource->taskId;
2,902,786✔
4424
  req.srcTaskId = ctx->taskId;
2,907,466✔
4425
  req.blockIdx = ctx->blockIdx;
2,903,306✔
4426
  req.queryId = ctx->queryId;
2,900,186✔
4427
  req.execId = pSource->execId;
2,903,826✔
4428

4429
  int32_t msgSize = tSerializeSResFetchReq(NULL, 0, &req, false);
2,900,186✔
4430
  if (msgSize < 0) {
2,911,106✔
4431
    return msgSize;
×
4432
  }
4433

4434
  void* msg = taosMemoryCalloc(1, msgSize);
2,911,106✔
4435
  if (NULL == msg) {
2,902,266✔
4436
    return terrno;
×
4437
  }
4438

4439
  msgSize = tSerializeSResFetchReq(msg, msgSize, &req, false);
2,902,266✔
4440
  if (msgSize < 0) {
2,910,586✔
4441
    taosMemoryFree(msg);
×
4442
    return msgSize;
×
4443
  }
4444

4445
  qDebug("%s scl build fetch msg and send to nodeId:%d, ep:%s, clientId:0x%" PRIx64 " taskId:0x%" PRIx64
2,910,586✔
4446
         ", execId:%d, blockIdx:%" PRId64,
4447
         ctx->idStr, pSource->addr.nodeId, pSource->addr.epSet.eps[0].fqdn, pSource->clientId,
4448
         pSource->taskId, pSource->execId, req.blockIdx);
4449

4450
  // send the fetch remote task result reques
4451
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
2,910,066✔
4452
  if (NULL == pMsgSendInfo) {
2,904,346✔
4453
    taosMemoryFreeClear(msg);
×
4454
    qError("%s prepare message %d failed", ctx->idStr, (int32_t)sizeof(SMsgSendInfo));
×
4455
    return terrno;
×
4456
  }
4457

4458
  SScalarFetchParam* param = taosMemoryMalloc(sizeof(SScalarFetchParam));
2,904,346✔
4459
  if (NULL == param) {
2,911,106✔
4460
    taosMemoryFreeClear(msg);
×
4461
    taosMemoryFreeClear(pMsgSendInfo);
×
4462
    qError("%s prepare param %d failed", ctx->idStr, (int32_t)sizeof(SScalarFetchParam));
×
4463
    return terrno;
×
4464
  }
4465

4466
  taosWLockLatch(&ctx->lock);
2,911,106✔
4467
  
4468
  if (ctx->code) {
2,908,506✔
4469
    qError("task has been killed, error:%s", tstrerror(ctx->code));
×
4470
    taosMemoryFree(param);
×
4471
    taosMemoryFreeClear(msg);
×
4472
    taosMemoryFreeClear(pMsgSendInfo);
×
4473
    code = ctx->code;
×
4474
    taosWUnLockLatch(&ctx->lock);
×
4475
    goto _end;
×
4476
  } else {
4477
    ctx->param = param;
2,907,466✔
4478
  }
4479
  
4480
  taosWUnLockLatch(&ctx->lock);
2,909,546✔
4481

4482
  param->subQIdx = subQIdx;
2,911,626✔
4483
  param->pRes = pRes;
2,911,626✔
4484
  param->pSubJobCtx = ctx;
2,908,506✔
4485

4486
  pMsgSendInfo->param = param;
2,912,146✔
4487
  pMsgSendInfo->paramFreeFp = taosAutoMemoryFree;
2,906,426✔
4488
  pMsgSendInfo->msgInfo.pData = msg;
2,902,266✔
4489
  pMsgSendInfo->msgInfo.len = msgSize;
2,899,146✔
4490
  pMsgSendInfo->msgType = pSource->fetchMsgType;
2,911,106✔
4491
  pMsgSendInfo->fp = remoteFetchCallBack;
2,897,586✔
4492
  pMsgSendInfo->requestId = ctx->queryId;
2,896,026✔
4493

4494
  code = asyncSendMsgToServer(ctx->rpcHandle, &pSource->addr.epSet, &ctx->transporterId, pMsgSendInfo);
2,907,986✔
4495
  QUERY_CHECK_CODE(code, lino, _end);
2,914,226✔
4496
      
4497
_end:
2,914,226✔
4498

4499
  if (code != TSDB_CODE_SUCCESS) {
2,914,226✔
4500
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
×
4501
  }
4502
  
4503
  return code;
2,914,226✔
4504
}
4505

4506
int32_t fetchRemoteNodeImpl(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes) {
2,870,071✔
4507
  int32_t          code = TSDB_CODE_SUCCESS;
2,870,071✔
4508
  int32_t          lino = 0;
2,870,071✔
4509

4510
  ctx->blockIdx = 0;
2,870,071✔
4511

4512
  code = sendFetchRemoteNodeReq(ctx, subQIdx, pRes);
2,871,111✔
4513
  QUERY_CHECK_CODE(code, lino, _end);
2,872,151✔
4514

4515
  code = qSemWait(ctx->pTaskInfo, &ctx->ready);
2,872,151✔
4516
  if (isTaskKilled(ctx->pTaskInfo)) {
2,872,671✔
4517
    code = getTaskCode(ctx->pTaskInfo);
10,920✔
4518
  } else {
4519
    code = ctx->code;
2,861,751✔
4520
  }
4521
      
4522
_end:
2,872,671✔
4523

4524
  taosWLockLatch(&ctx->lock);
2,872,671✔
4525
  ctx->param = NULL;
2,872,671✔
4526
  taosWUnLockLatch(&ctx->lock);
2,872,671✔
4527

4528
  if (code != TSDB_CODE_SUCCESS) {
2,872,671✔
4529
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
826,280✔
4530
  }
4531
  return code;
2,872,671✔
4532
}
4533

4534
int32_t remoteNodeCopy(SNode* pSrc, SNode* pDst) {
55,120✔
4535
  int32_t code = 0, lino = 0;
55,120✔
4536
  
4537
  switch (nodeType(pSrc)) {
55,120✔
4538
    case QUERY_NODE_VALUE:
55,120✔
4539
      TAOS_CHECK_EXIT(valueNodeCopy((SValueNode*)pSrc, &((SRemoteValueNode*)pDst)->val));
55,120✔
4540
      ((SRemoteValueNode*)pDst)->val.node.type = QUERY_NODE_VALUE;
55,120✔
4541
      break;
55,120✔
4542
    case QUERY_NODE_REMOTE_VALUE_LIST: {
×
4543
      SRemoteValueListNode* pDstNode = (SRemoteValueListNode*)pDst;
×
4544
      memcpy(pDst, pSrc, sizeof(SRemoteValueListNode));
×
4545
      pDstNode->hashAllocated = false;      
×
4546
      break;
×
4547
    } 
4548
    default:
×
4549
      break;
×
4550
  }
4551

4552
_exit:
55,120✔
4553

4554
  if (code) {
55,120✔
4555
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4556
  }
4557

4558
  return code;
55,120✔
4559
}
4560

4561
int32_t qFetchRemoteNode(void* pCtx, int32_t subQIdx, SNode* pRes) {
2,925,191✔
4562
  STaskSubJobCtx*  ctx = (STaskSubJobCtx*)pCtx;
2,925,191✔
4563
  int32_t code = 0, lino = 0;
2,925,191✔
4564
  int32_t       subEndPoinsNum = taosArrayGetSize(ctx->subEndPoints);
2,925,191✔
4565
  if (subQIdx >= subEndPoinsNum) {
2,924,671✔
4566
    qError("%s invalid subQIdx %d, subEndPointsNum:%d", ctx->idStr, subQIdx, subEndPoinsNum);
×
4567
    return TSDB_CODE_QRY_SUBQ_NOT_FOUND;
×
4568
  }
4569

4570
  SNode** ppRes = taosArrayGet(ctx->subResNodes, subQIdx);
2,924,671✔
4571
  if (NULL == *ppRes) {
2,918,431✔
4572
    TAOS_CHECK_EXIT(fetchRemoteNodeImpl(ctx, subQIdx, pRes));
2,866,951✔
4573
    *ppRes = pRes;
2,046,391✔
4574
  } else {
4575
    TAOS_CHECK_EXIT(remoteNodeCopy(*ppRes, pRes));
55,120✔
4576
  }
4577

4578
_exit:
55,120✔
4579

4580
  if (code) {
2,927,791✔
4581
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
826,280✔
4582
  } else {
4583
    qDebug("%s %s subQIdx %d succeed", ctx->idStr, __func__, subQIdx);
2,101,511✔
4584
  }
4585

4586
  return code;
2,927,791✔
4587
}
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