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

taosdata / TDengine / #5044

06 May 2026 02:35AM UTC coverage: 73.169% (+0.06%) from 73.107%
#5044

push

travis-ci

web-flow
feat: [6659794715] cpu limit (#35153)

244 of 275 new or added lines in 23 files covered. (88.73%)

526 existing lines in 141 files now uncovered.

277745 of 379596 relevant lines covered (73.17%)

133740972.66 hits per line

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

74.35
/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 "ttypes.h"
35
#include "tutil.h"
36
#include "tjson.h"
37
#include "trpc.h"
38
#include "filter.h"
39
#include "operator.h"
40
#include "tref.h"
41
#include "streamMsg.h"
42

43
typedef struct tagFilterAssist {
44
  SHashObj* colHash;
45
  int32_t   index;
46
  SArray*   cInfoList;
47
  int32_t   code;
48
} tagFilterAssist;
49

50
typedef struct STransTagExprCtx {
51
  int32_t      code;
52
  SMetaReader* pReader;
53
} STransTagExprCtx;
54

55
typedef enum {
56
  FILTER_NO_LOGIC = 1,
57
  FILTER_AND,
58
  FILTER_OTHER,
59
} FilterCondType;
60

61
static int32_t optimizeTbnameInCondImpl(void* pVnode, SArray* list, SNode* pTagCond, SStorageAPI* pStoreAPI,
62
                                        uint64_t suid);
63

64
static int64_t getLimit(const SNode* pLimit) {
1,401,819,296✔
65
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->limit) ? -1 : ((SLimitNode*)pLimit)->limit->datum.i;
1,401,819,296✔
66
}
67
static int64_t getOffset(const SNode* pLimit) {
1,401,773,154✔
68
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->offset) ? -1 : ((SLimitNode*)pLimit)->offset->datum.i;
1,401,773,154✔
69
}
70
static void releaseColInfoData(void* pCol);
71
int32_t sendFetchRemoteNodeReq(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes, bool reset);
72

73
void initResultRowInfo(SResultRowInfo* pResultRowInfo) {
529,396,017✔
74
  pResultRowInfo->size = 0;
529,396,017✔
75
  pResultRowInfo->cur.pageId = -1;
529,443,387✔
76
}
529,469,236✔
77

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

80
void resetResultRow(SResultRow* pResultRow, size_t entrySize) {
1,001,282,675✔
81
  pResultRow->numOfRows = 0;
1,001,282,675✔
82
  pResultRow->nOrigRows = 0;
1,001,283,185✔
83
  pResultRow->closed = false;
1,001,282,309✔
84
  pResultRow->endInterp = false;
1,001,283,584✔
85
  pResultRow->startInterp = false;
1,001,284,277✔
86

87
  if (entrySize > 0) {
1,001,284,715✔
88
    memset(pResultRow->pEntryInfo, 0, entrySize);
1,001,285,918✔
89
  }
90
}
1,001,281,256✔
91

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

97
size_t getResultRowSize(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
316,068,020✔
98
  int32_t rowSize = (numOfOutput * sizeof(SResultRowEntryInfo)) + sizeof(SResultRow);
316,068,020✔
99

100
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,288,120,222✔
101
    rowSize += pCtx[i].resDataInfo.interBufSize;
972,127,928✔
102
  }
103

104
  return rowSize;
315,992,294✔
105
}
106

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

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

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

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

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

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

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

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

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

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

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

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

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

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

220
void cleanupGroupResInfo(SGroupResInfo* pGroupResInfo) {
145,366,150✔
221
  taosMemoryFreeClear(pGroupResInfo->pBuf);
145,366,150✔
222
  if (pGroupResInfo->freeItem) {
145,366,890✔
223
    //    taosArrayDestroy(pGroupResInfo->pRows);
224
    taosArrayDestroyEx(pGroupResInfo->pRows, freeEx);
×
225
    pGroupResInfo->freeItem = false;
×
226
    pGroupResInfo->pRows = NULL;
×
227
  } else {
228
    taosArrayDestroy(pGroupResInfo->pRows);
145,359,529✔
229
    pGroupResInfo->pRows = NULL;
145,353,331✔
230
  }
231
  pGroupResInfo->index = 0;
145,357,558✔
232
  pGroupResInfo->delIndex = 0;
145,359,364✔
233
}
145,332,985✔
234

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

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

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

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

255
int32_t initGroupedResultInfo(SGroupResInfo* pGroupResInfo, SSHashObj* pHashmap, int32_t order) {
110,351,765✔
256
  int32_t code = TSDB_CODE_SUCCESS;
110,351,765✔
257
  int32_t lino = 0;
110,351,765✔
258
  if (pGroupResInfo->pRows != NULL) {
110,351,765✔
259
    taosArrayDestroy(pGroupResInfo->pRows);
2,500,529✔
260
  }
261
  if (pGroupResInfo->pBuf) {
110,356,655✔
262
    taosMemoryFree(pGroupResInfo->pBuf);
2,499,879✔
263
    pGroupResInfo->pBuf = NULL;
2,499,879✔
264
  }
265

266
  // extract the result rows information from the hash map
267
  int32_t size = tSimpleHashGetSize(pHashmap);
110,349,538✔
268

269
  void* pData = NULL;
110,352,653✔
270
  pGroupResInfo->pRows = taosArrayInit(size, POINTER_BYTES);
110,352,653✔
271
  QUERY_CHECK_NULL(pGroupResInfo->pRows, code, lino, _end, terrno);
110,348,459✔
272

273
  size_t  keyLen = 0;
110,344,031✔
274
  int32_t iter = 0;
110,347,956✔
275
  int64_t bufLen = 0, offset = 0;
110,346,573✔
276

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

283
  pGroupResInfo->pBuf = taosMemoryMalloc(bufLen);
110,276,573✔
284
  QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
110,352,264✔
285

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

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

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

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

301
  if (order == TSDB_ORDER_ASC || order == TSDB_ORDER_DESC) {
110,036,369✔
302
    __compar_fn_t fn = (order == TSDB_ORDER_ASC) ? resultrowComparAsc : resultrowComparDesc;
6,840,724✔
303
    size = POINTER_BYTES;
6,840,724✔
304
    taosSort(pGroupResInfo->pRows->pData, taosArrayGetSize(pGroupResInfo->pRows), size, fn);
6,840,724✔
305
  }
306

307
  pGroupResInfo->index = 0;
110,036,232✔
308

309
_end:
110,346,348✔
310
  if (code != TSDB_CODE_SUCCESS) {
110,353,565✔
311
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
312
  }
313
  return code;
110,353,565✔
314
}
315

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

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

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

332
  return pGroupResInfo->index < taosArrayGetSize(pGroupResInfo->pRows);
436,388,294✔
333
}
334

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

340
  return (int32_t)taosArrayGetSize(pGroupResInfo->pRows);
233,845,508✔
341
}
342

343
SArray* createSortInfo(SNodeList* pNodeList) {
57,635,387✔
344
  size_t numOfCols = 0;
57,635,387✔
345

346
  if (pNodeList != NULL) {
57,635,387✔
347
    numOfCols = LIST_LENGTH(pNodeList);
57,395,139✔
348
  } else {
349
    numOfCols = 0;
240,248✔
350
  }
351

352
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
57,633,050✔
353
  if (pList == NULL) {
57,627,396✔
354
    return pList;
×
355
  }
356

357
  for (int32_t i = 0; i < numOfCols; ++i) {
131,062,420✔
358
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
73,426,491✔
359
    if (!pSortKey) {
73,434,083✔
360
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
361
      taosArrayDestroy(pList);
×
362
      pList = NULL;
×
363
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
364
      break;
×
365
    }
366
    SBlockOrderInfo bi = {0};
73,434,083✔
367
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
73,425,194✔
368
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);
73,423,688✔
369

370
    if (nodeType(pSortKey->pExpr) != QUERY_NODE_COLUMN) {
73,422,887✔
371
      qError("invalid order by expr type:%d", nodeType(pSortKey->pExpr));
×
372
      taosArrayDestroy(pList);
×
373
      pList = NULL;
×
374
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
375
      break;
×
376
    }
377
    
378
    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
73,408,985✔
379
    bi.slotId = pColNode->slotId;
73,425,378✔
380
    void* tmp = taosArrayPush(pList, &bi);
73,435,583✔
381
    if (!tmp) {
73,435,583✔
382
      taosArrayDestroy(pList);
×
383
      pList = NULL;
×
384
      break;
×
385
    }
386
  }
387

388
  return pList;
57,635,785✔
389
}
390

391
SSDataBlock* createDataBlockFromDescNode(void* p) {
837,166,731✔
392
  SDataBlockDescNode* pNode = (SDataBlockDescNode*)p;
837,166,731✔
393
  int32_t      numOfCols = LIST_LENGTH(pNode->pSlots);
837,166,731✔
394
  SSDataBlock* pBlock = NULL;
837,304,693✔
395
  int32_t      code = TSDB_CODE_SUCCESS;
837,284,578✔
396
  int32_t      lino = 0;
837,284,578✔
397

398
  code = createDataBlock(&pBlock);
837,284,578✔
399
  QUERY_CHECK_CODE(code, lino, _return);
837,140,378✔
400

401
  pBlock->info.id.blockId = pNode->dataBlockId;
837,140,378✔
402
  pBlock->info.calWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
837,209,873✔
403
  pBlock->info.watermark = INT64_MIN;
837,118,378✔
404

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

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

420
  if (pBlock != NULL && i != numOfCols) {
837,453,513✔
421
    code = TSDB_CODE_INVALID_PARA;
×
422
    QUERY_CHECK_CODE(code, lino, _return);
×
423
  }
424

425
  return pBlock;
837,453,513✔
426
_return:
50,730✔
427
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
428
  blockDataDestroy(pBlock);
×
429
  terrno = code;
×
430
  return NULL;
×
431
}
432

433
int32_t prepareDataBlockBuf(SSDataBlock* pDataBlock, SColMatchInfo* pMatchInfo) {
286,853,079✔
434
  SDataBlockInfo* pBlockInfo = &pDataBlock->info;
286,853,079✔
435

436
  for (int32_t i = 0; i < taosArrayGetSize(pMatchInfo->pList); ++i) {
1,277,679,777✔
437
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
1,013,547,005✔
438
    if (!pItem) {
1,013,438,107✔
439
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
440
      return terrno;
×
441
    }
442

443
    if (pItem->isPk) {
1,013,438,107✔
444
      SColumnInfoData* pInfoData = taosArrayGet(pDataBlock->pDataBlock, pItem->dstSlotId);
22,742,141✔
445
      if (!pInfoData) {
22,363,587✔
446
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
447
        return terrno;
×
448
      }
449
      pBlockInfo->pks[0].type = pInfoData->info.type;
22,363,587✔
450
      pBlockInfo->pks[1].type = pInfoData->info.type;
22,402,883✔
451

452
      // allocate enough buffer size, which is pInfoData->info.bytes
453
      if (IS_VAR_DATA_TYPE(pItem->dataType.type)) {
22,381,174✔
454
        pBlockInfo->pks[0].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
6,823,888✔
455
        if (pBlockInfo->pks[0].pData == NULL) {
6,808,614✔
456
          return terrno;
×
457
        }
458

459
        pBlockInfo->pks[1].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
6,817,647✔
460
        if (pBlockInfo->pks[1].pData == NULL) {
6,814,192✔
461
          taosMemoryFreeClear(pBlockInfo->pks[0].pData);
×
462
          return terrno;
×
463
        }
464

465
        pBlockInfo->pks[0].nData = pInfoData->info.bytes;
6,819,788✔
466
        pBlockInfo->pks[1].nData = pInfoData->info.bytes;
6,820,477✔
467
      }
468

469
      break;
22,407,782✔
470
    }
471
  }
472

473
  return TSDB_CODE_SUCCESS;
286,713,041✔
474
}
475

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

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

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

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

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

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

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

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

560
  return DEAL_RES_CONTINUE;
2,772,192✔
561
}
562

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

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

573
    return TSDB_CODE_SUCCESS;
×
574
  }
575

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

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

600
    return code;
×
601
  }
602

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

606
  nodesDestroyNode(pNew);
397,268✔
607
  return TSDB_CODE_SUCCESS;
397,268✔
608
}
609

610
static EDealRes getColumn(SNode** pNode, void* pContext) {
66,337,230✔
611
  tagFilterAssist* pData = (tagFilterAssist*)pContext;
66,337,230✔
612
  SColumnNode*     pSColumnNode = NULL;
66,337,230✔
613
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
66,344,444✔
614
    pSColumnNode = *(SColumnNode**)pNode;
22,922,293✔
615
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
43,424,747✔
616
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
1,332,228✔
617
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
1,332,228✔
618
      pData->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
833,987✔
619
      if (NULL == pSColumnNode) {
834,421✔
620
        return DEAL_RES_ERROR;
×
621
      }
622
      pSColumnNode->colId = -1;
834,421✔
623
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
834,421✔
624
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
834,421✔
625
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
834,421✔
626
      nodesDestroyNode(*pNode);
834,421✔
627
      *pNode = (SNode*)pSColumnNode;
833,789✔
628
    } else {
629
      return DEAL_RES_CONTINUE;
498,241✔
630
    }
631
  } else {
632
    return DEAL_RES_CONTINUE;
42,093,688✔
633
  }
634

635
  void* data = taosHashGet(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
23,759,608✔
636
  if (!data) {
23,751,033✔
637
    int32_t tempRes =
638
        taosHashPut(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
20,408,531✔
639
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
20,418,030✔
640
      return DEAL_RES_ERROR;
×
641
    }
642
    pSColumnNode->slotId = pData->index++;
20,418,030✔
643
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
20,418,030✔
644
                         .type = pSColumnNode->node.resType.type,
20,405,011✔
645
                         .bytes = pSColumnNode->node.resType.bytes,
20,412,934✔
646
                         .pk = pSColumnNode->isPk};
20,415,390✔
647
#if TAG_FILTER_DEBUG
648
    qDebug("tagfilter build column info, slotId:%d, colId:%d, type:%d", pSColumnNode->slotId, cInfo.colId, cInfo.type);
649
#endif
650
    void* tmp = taosArrayPush(pData->cInfoList, &cInfo);
20,415,139✔
651
    if (!tmp) {
20,416,978✔
652
      return DEAL_RES_ERROR;
×
653
    }
654
  } else {
655
    SColumnNode* col = *(SColumnNode**)data;
3,342,502✔
656
    pSColumnNode->slotId = col->slotId;
3,342,104✔
657
  }
658

659
  return DEAL_RES_CONTINUE;
23,757,540✔
660
}
661

662
static int32_t createResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
19,007,117✔
663
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
19,007,117✔
664
  if (pColumnData == NULL) {
19,006,618✔
665
    return terrno;
×
666
  }
667

668
  pColumnData->info.type = pType->type;
19,006,618✔
669
  pColumnData->info.bytes = pType->bytes;
19,002,983✔
670
  pColumnData->info.scale = pType->scale;
19,002,155✔
671
  pColumnData->info.precision = pType->precision;
18,996,630✔
672

673
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
19,000,472✔
674
  if (code != TSDB_CODE_SUCCESS) {
18,989,858✔
675
    terrno = code;
×
676
    releaseColInfoData(pColumnData);
×
677
    return terrno;
×
678
  }
679

680
  pParam->columnData = pColumnData;
18,989,858✔
681
  pParam->colAlloced = true;
19,002,322✔
682
  return TSDB_CODE_SUCCESS;
18,990,300✔
683
}
684

685
static void releaseColInfoData(void* pCol) {
5,374,020✔
686
  if (pCol) {
5,374,020✔
687
    SColumnInfoData* col = (SColumnInfoData*)pCol;
5,374,020✔
688
    colDataDestroy(col);
5,374,020✔
689
    taosMemoryFree(col);
5,373,884✔
690
  }
691
}
5,374,317✔
692

693
void freeItem(void* p) {
220,669,277✔
694
  STUidTagInfo* pInfo = p;
220,669,277✔
695
  if (pInfo->pTagVal != NULL) {
220,669,277✔
696
    taosMemoryFree(pInfo->pTagVal);
220,264,470✔
697
  }
698
}
220,654,893✔
699

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

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

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

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

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

775
  return TSDB_CODE_SUCCESS;
12,139✔
776
}
777

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

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

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

803
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
12,139✔
804
    extractTagDataEntry((SOperatorNode*)pTagCond, pIdWithVal);
×
805
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
12,139✔
806
    SNode* pChild = NULL;
12,139✔
807
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
36,417✔
808
      extractTagDataEntry((SOperatorNode*)pChild, pIdWithVal);
24,278✔
809
    }
810
  }
811

812
  taosArraySort(pIdWithVal, compareTagDataEntry);
12,139✔
813

814
  return TSDB_CODE_SUCCESS;
12,139✔
815
}
816

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

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

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

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

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

851
static int32_t genTagFilterDigest(const SNode* pTagCond, T_MD5_CTX* pContext) {
36,499✔
852
  if (pTagCond == NULL) {
36,499✔
853
    return TSDB_CODE_SUCCESS;
34,442✔
854
  }
855

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

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

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

874
  taosMemoryFree(payload);
2,057✔
875
  return TSDB_CODE_SUCCESS;
2,057✔
876
}
877

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

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

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

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

905
int32_t qGetColumnsFromNodeList(void* data, bool isList, SArray** pColList) {
18,521,812✔
906
  int32_t code = TSDB_CODE_SUCCESS;
18,521,812✔
907
  tagFilterAssist ctx = {0};
18,521,812✔
908
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
18,523,620✔
909
  if (ctx.colHash == NULL) {
18,523,081✔
910
    code = terrno;
×
911
    goto end;
×
912
  }
913

914
  ctx.index = 0;
18,523,081✔
915
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
18,523,081✔
916
  if (ctx.cInfoList == NULL) {
18,522,556✔
917
    code = terrno;
1,990✔
918
    goto end;
×
919
  }
920

921
  if (isList) {
18,520,566✔
922
    SNode* pNode = NULL;
4,887,205✔
923
    FOREACH(pNode, (SNodeList*)data) {
10,263,136✔
924
      nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
5,376,130✔
925
      if (TSDB_CODE_SUCCESS != ctx.code) {
5,375,437✔
926
        code = ctx.code;
×
927
        goto end;
×
928
      }
929
      REPLACE_NODE(pNode);
5,375,437✔
930
    }
931
  } else {
932
    SNode* pNode = (SNode*)data;
13,633,361✔
933
    nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
13,633,488✔
934
    if (TSDB_CODE_SUCCESS != ctx.code) {
13,634,782✔
935
      code = ctx.code;
×
936
      goto end;
×
937
    }
938
  }
939
  
940
  if (pColList != NULL) *pColList = ctx.cInfoList;
18,514,035✔
941
  ctx.cInfoList = NULL;
18,515,301✔
942

943
end:
18,523,054✔
944
  taosHashCleanup(ctx.colHash);
18,516,424✔
945
  taosArrayDestroy(ctx.cInfoList);
18,504,857✔
946
  return code;
18,506,526✔
947
}
948

949
static int32_t buildGroupInfo(SColumnInfoData* pValue, int32_t i, SArray* gInfo) {
724,608✔
950
  int32_t code = TSDB_CODE_SUCCESS;
724,608✔
951
  SStreamGroupValue* v = taosArrayReserve(gInfo, 1);
724,608✔
952
  if (v == NULL) {
725,272✔
953
    code = terrno;
×
954
    goto end;
×
955
  }
956
  if (colDataIsNull_s(pValue, i)) {
1,450,544✔
957
    v->isNull = true;
7,960✔
958
  } else {
959
    v->isNull = false;
717,312✔
960
    char* data = colDataGetData(pValue, i);
717,029✔
961
    if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
717,038✔
962
      if (tTagIsJson(data)) {
×
963
        code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
×
964
        goto end;
×
965
      }
966
      if (tTagIsJsonNull(data)) {
×
967
        v->isNull = true;
×
968
        goto end;
×
969
      }
970
      int32_t len = getJsonValueLen(data);
×
971
      v->data.type = pValue->info.type;
×
972
      v->data.nData = len;
×
973
      v->data.pData = taosMemoryCalloc(1, len + 1);
×
974
      if (v->data.pData == NULL) {
×
975
        code = terrno;
×
976
        goto end;
×
977
      }
978
      memcpy(v->data.pData, data, len);
×
979
      qDebug("buildGroupInfo:%d add json data len:%d, data:%s", i, len, (char*)v->data.pData);
×
980
    } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
716,630✔
981
      if (varDataTLen(data) > pValue->info.bytes) {
465,687✔
982
        code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
983
        goto end;
×
984
      }
985
      v->data.type = pValue->info.type;
464,854✔
986
      v->data.nData = varDataLen(data);
465,274✔
987
      v->data.pData = taosMemoryCalloc(1, varDataLen(data) + 1);
465,536✔
988
      if (v->data.pData == NULL) {
465,708✔
989
        code = terrno;
×
990
        goto end;
×
991
      }
992
      memcpy(v->data.pData, varDataVal(data), varDataLen(data));
465,274✔
993
      qDebug("buildGroupInfo:%d add var data type:%d, len:%d, data:%s", i, pValue->info.type, varDataLen(data), (char*)v->data.pData);
465,536✔
994
    } else if (pValue->info.type == TSDB_DATA_TYPE_DECIMAL) {  // reader todo decimal
251,776✔
995
      v->data.type = pValue->info.type;
×
996
      v->data.nData = pValue->info.bytes;
×
997
      v->data.pData = taosMemoryCalloc(1, pValue->info.bytes);
×
998
      if (v->data.pData == NULL) {
×
999
        code = terrno;
×
1000
        goto end;
×
1001
      }
1002
      memcpy(&v->data.pData, data, pValue->info.bytes);
×
1003
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
×
1004
    } else {  // reader todo decimal
1005
      v->data.type = pValue->info.type;
251,639✔
1006
      memcpy(&v->data.val, data, pValue->info.bytes);
251,776✔
1007
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
251,776✔
1008
    }
1009
  }
1010
end:
39,041✔
1011
  if (code != TSDB_CODE_SUCCESS) {
725,706✔
1012
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1013
    v->isNull = true;
×
1014
  }
1015
  return code;
725,706✔
1016
}
1017

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

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

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

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

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

1060
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
278,144✔
1061
    SColumnInfo* tmp = (SColumnInfo*)taosArrayGet(pColList, i);
182,469✔
1062
    if (tmp != NULL && tmp->colId == -1) {
182,469✔
1063
      tbNameIndex = i;
94,958✔
1064
    }
1065
  }
1066
  
1067
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
95,392✔
1068
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
95,392✔
1069
  taosArrayDestroy(pColList);
95,675✔
1070
  if (pResBlock == NULL) {
95,675✔
1071
    code = terrno;
293✔
1072
    goto end;
293✔
1073
  }
1074

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

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

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

1084
  SNode* pNode = NULL;
94,948✔
1085
  FOREACH(pNode, group) {
277,548✔
1086
    SScalarParam output = {0};
182,600✔
1087

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

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

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

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

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

1129
  for (int i = 0; i < rows; i++) {
426,557✔
1130
    gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
330,458✔
1131
    QUERY_CHECK_NULL(gInfo, code, lino, end, terrno);
330,497✔
1132

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

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

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

1161
end:
96,392✔
1162
  blockDataDestroy(pResBlock);
95,675✔
1163
  taosArrayDestroy(pBlockList);
95,675✔
1164
  taosArrayDestroyEx(pUidTagList, freeItem);
95,675✔
1165
  taosArrayDestroyP(groupData, releaseColInfoData);
95,241✔
1166
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
95,675✔
1167

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

1173
int32_t getColInfoResultForGroupby(SReadHandle* pHandle, SNodeList* group, STableListInfo* pTableListInfo, uint8_t* digest,
4,790,821✔
1174
                                   SStorageAPI* pAPI, bool initRemainGroups, SHashObj* groupIdMap, bool gIdFromBaseId) {
1175
  int32_t      code = TSDB_CODE_SUCCESS;
4,790,821✔
1176
  int32_t      lino = 0;
4,790,821✔
1177
  SArray*      pBlockList = NULL;
4,790,821✔
1178
  SSDataBlock* pResBlock = NULL;
4,790,821✔
1179
  void*        keyBuf = NULL;
4,791,530✔
1180
  SArray*      groupData = NULL;
4,791,530✔
1181
  SArray*      pUidTagList = NULL;
4,791,530✔
1182
  SArray*      tableList = NULL;
4,791,530✔
1183
  SArray*      gInfo = NULL;
4,791,530✔
1184
  void*        pVnode = pHandle->vnode;
4,791,530✔
1185

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

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

1202
    nodesFree(listNode);
×
1203

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

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

1217
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
4,791,530✔
1218
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
4,791,530✔
1219

1220
  for (int32_t i = 0; i < rows; ++i) {
30,650,167✔
1221
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
25,858,637✔
1222
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
25,858,637✔
1223
    STUidTagInfo info = {.uid = pkeyInfo->uid};
25,858,637✔
1224
    void*        tmp = taosArrayPush(pUidTagList, &info);
25,858,637✔
1225
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
25,858,637✔
1226
  }
1227

1228
  if (taosArrayGetSize(pUidTagList) > 0) {
4,791,530✔
1229
    code = pAPI->metaFn.getTableTagsByUidVersion(pVnode, pTableListInfo->idInfo.suid, pUidTagList, -1);
4,791,530✔
1230
  } else {
1231
    code = pAPI->metaFn.getTableTags(pVnode, pTableListInfo->idInfo.suid, pUidTagList);
×
1232
  }
1233
  if (code != TSDB_CODE_SUCCESS) {
4,791,530✔
1234
    goto end;
×
1235
  }
1236

1237
  SArray* pColList = NULL;
4,791,530✔
1238
  code = qGetColumnsFromNodeList(group, true, &pColList); 
4,791,530✔
1239
  if (code != TSDB_CODE_SUCCESS) {
4,790,926✔
1240
    goto end;
×
1241
  }
1242

1243
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
4,790,926✔
1244
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
4,790,221✔
1245
  taosArrayDestroy(pColList);
4,791,125✔
1246
  if (pResBlock == NULL) {
4,791,393✔
1247
    code = terrno;
×
1248
    goto end;
×
1249
  }
1250

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

1254
  pBlockList = taosArrayInit(2, POINTER_BYTES);
4,791,393✔
1255
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
4,790,851✔
1256

1257
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
4,790,988✔
1258
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
4,790,988✔
1259

1260
  groupData = taosArrayInit(2, POINTER_BYTES);
4,790,988✔
1261
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
4,791,530✔
1262

1263
  SNode* pNode = NULL;
4,790,988✔
1264
  FOREACH(pNode, group) {
9,982,409✔
1265
    SScalarParam output = {0};
5,192,533✔
1266

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

1292
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
5,191,419✔
1293
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
5,074,462✔
1294
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
5,074,462✔
1295
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
5,073,197✔
1296
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
5,073,197✔
1297
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
116,957✔
1298
      continue;
×
1299
    } else {
1300
      gTaskScalarExtra.pStreamInfo = NULL;
117,094✔
1301
      gTaskScalarExtra.pStreamRange = NULL;
117,094✔
1302
      code = scalarCalculate(pNode, pBlockList, &output, &gTaskScalarExtra);
117,094✔
1303
    }
1304

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

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

1314
  int32_t baseIdLen = gIdFromBaseId ? sizeof(int64_t) : 0;
4,790,960✔
1315
  int32_t keyLen = baseIdLen;
4,790,960✔
1316
  SNode*  node;
1317
  FOREACH(node, group) {
9,982,071✔
1318
    SExprNode* pExpr = (SExprNode*)node;
5,190,541✔
1319
    keyLen += pExpr->resType.bytes;
5,190,541✔
1320
  }
1321

1322
  int32_t nullFlagSize = sizeof(int8_t) * LIST_LENGTH(group);
4,788,557✔
1323
  keyLen += nullFlagSize;
4,789,127✔
1324

1325
  keyBuf = taosMemoryCalloc(1, keyLen);
4,789,127✔
1326
  if (keyBuf == NULL) {
4,789,723✔
1327
    code = terrno;
×
1328
    goto end;
×
1329
  }
1330

1331
  if (initRemainGroups) {
4,789,723✔
1332
    pTableListInfo->remainGroups =
2,188,235✔
1333
        taosHashInit(rows, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
2,187,095✔
1334
    if (pTableListInfo->remainGroups == NULL) {
2,188,235✔
1335
      code = terrno;
×
1336
      goto end;
×
1337
    }
1338
  }
1339

1340
  for (int i = 0; i < rows; i++) {
30,648,047✔
1341
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
25,857,766✔
1342
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
25,854,311✔
1343

1344
    if (groupIdMap != NULL){
25,854,311✔
1345
      gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
80,200✔
1346
    }
1347

1348
    if (baseIdLen > 0) {
25,854,311✔
1349
      *(uint64_t*)keyBuf = info->baseGId;
×
1350
    }
1351
    char* isNull = (char*)keyBuf + baseIdLen;
25,854,311✔
1352
    char* pStart = (char*)keyBuf + baseIdLen + sizeof(int8_t) * LIST_LENGTH(group);
25,853,332✔
1353
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
53,476,951✔
1354
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
27,622,478✔
1355

1356
      if (groupIdMap != NULL && gInfo != NULL) {
27,622,807✔
1357
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
127,511✔
1358
        if (ret != TSDB_CODE_SUCCESS) {
127,511✔
1359
          qError("buildGroupInfo failed at line %d since %s", __LINE__, tstrerror(ret));
×
1360
          taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1361
          gInfo = NULL;
×
1362
        }
1363
      }
1364
      
1365
      if (colDataIsNull_s(pValue, i)) {
55,246,789✔
1366
        isNull[j] = 1;
111,471✔
1367
      } else {
1368
        isNull[j] = 0;
27,512,511✔
1369
        char* data = colDataGetData(pValue, i);
27,511,340✔
1370
        if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
27,511,602✔
1371
          // if (tTagIsJson(data)) {
1372
          //   code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
1373
          //   goto end;
1374
          // }
1375
          if (tTagIsJsonNull(data)) {
85,412✔
1376
            isNull[j] = 1;
×
1377
            continue;
×
1378
          }
1379
          int32_t len = getJsonValueLen(data);
85,412✔
1380
          memcpy(pStart, data, len);
85,412✔
1381
          pStart += len;
85,412✔
1382
        } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
27,426,267✔
1383
          if (IS_STR_DATA_BLOB(pValue->info.type)) {
23,484,928✔
1384
            if (blobDataTLen(data) > TSDB_MAX_BLOB_LEN) {
967✔
1385
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1386
              goto end;
×
1387
            }
1388
            memcpy(pStart, data, blobDataTLen(data));
×
1389
            pStart += blobDataTLen(data);
×
1390
          } else {
1391
            if (varDataTLen(data) > pValue->info.bytes) {
23,483,443✔
1392
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1393
              goto end;
×
1394
            }
1395
            memcpy(pStart, data, varDataTLen(data));
23,483,147✔
1396
            pStart += varDataTLen(data);
23,484,136✔
1397
          }
1398
        } else {
1399
          memcpy(pStart, data, pValue->info.bytes);
3,941,534✔
1400
          pStart += pValue->info.bytes;
3,941,667✔
1401
        }
1402
      }
1403
    }
1404

1405
    int32_t len = (int32_t)(pStart - (char*)keyBuf);
25,848,844✔
1406
    info->groupId = calcGroupId(keyBuf, len);
25,848,844✔
1407
    if (groupIdMap != NULL && gInfo != NULL) {
25,854,808✔
1408
      int32_t ret = taosHashPut(groupIdMap, &info->groupId, sizeof(info->groupId), &gInfo, POINTER_BYTES);
80,462✔
1409
      if (ret != TSDB_CODE_SUCCESS) {
80,462✔
1410
        qError("put groupid to map failed at line %d since %s", __LINE__, tstrerror(ret));
×
1411
        taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1412
      }
1413
      qDebug("put groupid to map gid:%" PRIu64, info->groupId);
80,462✔
1414
      gInfo = NULL;
80,462✔
1415
    }
1416
    if (initRemainGroups) {
25,854,808✔
1417
      // groupId ~ table uid
1418
      code = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId), &(info->uid),
12,464,499✔
1419
                         sizeof(info->uid));
1420
      if (code == TSDB_CODE_DUP_KEY) {
12,466,613✔
1421
        code = TSDB_CODE_SUCCESS;
854,987✔
1422
      }
1423
      QUERY_CHECK_CODE(code, lino, end);
12,466,613✔
1424
    }
1425
  }
1426

1427
  if (tsTagFilterCache && groupIdMap == NULL && digest) {
4,790,281✔
1428
    tableList = taosArrayDup(pTableListInfo->pTableList, NULL);
×
1429
    QUERY_CHECK_NULL(tableList, code, lino, end, terrno);
×
1430

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

1437
  //  int64_t st2 = taosGetTimestampUs();
1438
  //  qDebug("calculate tag block rows:%d, cost:%ld us", rows, st2-st1);
1439

1440
end:
4,790,281✔
1441
  taosMemoryFreeClear(keyBuf);
4,790,988✔
1442
  blockDataDestroy(pResBlock);
4,790,555✔
1443
  taosArrayDestroy(pBlockList);
4,790,988✔
1444
  taosArrayDestroyEx(pUidTagList, freeItem);
4,790,988✔
1445
  taosArrayDestroyP(groupData, releaseColInfoData);
4,790,988✔
1446
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
4,790,418✔
1447

1448
  if (code != TSDB_CODE_SUCCESS) {
4,790,988✔
1449
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1450
  }
1451
  return code;
4,790,988✔
1452
}
1453

1454
static int32_t nameComparFn(const void* p1, const void* p2) {
776,132✔
1455
  const char* pName1 = *(const char**)p1;
776,132✔
1456
  const char* pName2 = *(const char**)p2;
776,132✔
1457

1458
  int32_t ret = strcmp(pName1, pName2);
776,132✔
1459
  if (ret == 0) {
776,132✔
1460
    return 0;
15,795✔
1461
  } else {
1462
    return (ret > 0) ? 1 : -1;
760,337✔
1463
  }
1464
}
1465

1466
static SArray* getTableNameList(const SNodeListNode* pList) {
417,326✔
1467
  int32_t    code = TSDB_CODE_SUCCESS;
417,326✔
1468
  int32_t    lino = 0;
417,326✔
1469
  int32_t    len = LIST_LENGTH(pList->pNodeList);
417,326✔
1470
  SListCell* cell = pList->pNodeList->pHead;
417,326✔
1471

1472
  SArray* pTbList = taosArrayInit(len, POINTER_BYTES);
417,326✔
1473
  QUERY_CHECK_NULL(pTbList, code, lino, _end, terrno);
417,326✔
1474

1475
  for (int i = 0; i < pList->pNodeList->length; i++) {
1,160,586✔
1476
    SValueNode* valueNode = (SValueNode*)cell->pNode;
743,260✔
1477
    if (!IS_VAR_DATA_TYPE(valueNode->node.resType.type)) {
743,260✔
1478
      terrno = TSDB_CODE_INVALID_PARA;
×
1479
      taosArrayDestroy(pTbList);
×
1480
      return NULL;
×
1481
    }
1482

1483
    char* name = varDataVal(valueNode->datum.p);
743,260✔
1484
    void* tmp = taosArrayPush(pTbList, &name);
743,260✔
1485
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
743,260✔
1486
    cell = cell->pNext;
743,260✔
1487
  }
1488

1489
  size_t numOfTables = taosArrayGetSize(pTbList);
417,326✔
1490

1491
  // order the name
1492
  taosArraySort(pTbList, nameComparFn);
417,326✔
1493

1494
  // remove the duplicates
1495
  SArray* pNewList = taosArrayInit(taosArrayGetSize(pTbList), sizeof(void*));
417,326✔
1496
  QUERY_CHECK_NULL(pNewList, code, lino, _end, terrno);
417,326✔
1497
  void* tmpTbl = taosArrayGet(pTbList, 0);
417,326✔
1498
  QUERY_CHECK_NULL(tmpTbl, code, lino, _end, terrno);
417,326✔
1499
  void* tmp = taosArrayPush(pNewList, tmpTbl);
417,326✔
1500
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
417,326✔
1501

1502
  for (int32_t i = 1; i < numOfTables; ++i) {
743,260✔
1503
    char** name = taosArrayGetLast(pNewList);
325,934✔
1504
    char** nameInOldList = taosArrayGet(pTbList, i);
325,934✔
1505
    QUERY_CHECK_NULL(nameInOldList, code, lino, _end, terrno);
325,934✔
1506
    if (strcmp(*name, *nameInOldList) == 0) {
325,934✔
1507
      continue;
8,252✔
1508
    }
1509

1510
    tmp = taosArrayPush(pNewList, nameInOldList);
317,682✔
1511
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
317,682✔
1512
  }
1513

1514
_end:
417,326✔
1515
  taosArrayDestroy(pTbList);
417,326✔
1516
  if (code != TSDB_CODE_SUCCESS) {
417,326✔
1517
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1518
    return NULL;
×
1519
  }
1520
  return pNewList;
417,326✔
1521
}
1522

1523
static int tableUidCompare(const void* a, const void* b) {
×
1524
  uint64_t u1 = *(uint64_t*)a;
×
1525
  uint64_t u2 = *(uint64_t*)b;
×
1526

1527
  if (u1 == u2) {
×
1528
    return 0;
×
1529
  }
1530

1531
  return u1 < u2 ? -1 : 1;
×
1532
}
1533

1534
static int32_t filterTableInfoCompare(const void* a, const void* b) {
41,613✔
1535
  STUidTagInfo* p1 = (STUidTagInfo*)a;
41,613✔
1536
  STUidTagInfo* p2 = (STUidTagInfo*)b;
41,613✔
1537

1538
  if (p1->uid == p2->uid) {
41,613✔
1539
    return 0;
×
1540
  }
1541

1542
  return p1->uid < p2->uid ? -1 : 1;
41,613✔
1543
}
1544

1545
static FilterCondType checkTagCond(SNode* cond) {
16,233,575✔
1546
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
16,233,575✔
1547
    return FILTER_NO_LOGIC;
11,539,582✔
1548
  }
1549
  if (nodeType(cond) == QUERY_NODE_LOGIC_CONDITION && ((SLogicConditionNode*)cond)->condType == LOGIC_COND_TYPE_AND) {
4,694,995✔
1550
    return FILTER_AND;
4,073,482✔
1551
  }
1552
  return FILTER_OTHER;
620,701✔
1553
}
1554

1555
static int32_t doInWithAnd(SNode* cond, void* pVnode, SArray* list, SStorageAPI* pAPI, uint64_t suid) {
4,702,561✔
1556
  if (nodeType(cond) != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
4,702,561✔
1557
    return -1;
620,701✔
1558
  }
1559

1560
  SLogicConditionNode* pNode = (SLogicConditionNode*)cond;
4,082,258✔
1561
  SNodeList*           pList = (SNodeList*)pNode->pParameterList;
4,082,258✔
1562

1563
  int32_t len = LIST_LENGTH(pList);
4,079,939✔
1564
  if (len <= 0) {
4,081,469✔
1565
    return -1;
×
1566
  }
1567

1568
  SListCell* cell = pList->pHead;
4,081,469✔
1569
  for (int i = 0; i < len; i++) {
12,660,875✔
1570
    if (cell == NULL) break;
8,588,755✔
1571
    if (nodeType(cell->pNode) == QUERY_NODE_OPERATOR && optimizeTbnameInCondImpl(pVnode, list, cell->pNode, pAPI, suid) == 0) {
8,588,755✔
1572
      return 0;
8,539✔
1573
    }
1574
    cell = cell->pNext;
8,577,924✔
1575
  }
1576
  return -1;
4,072,120✔
1577
}
1578

1579
static int32_t optimizeTbnameInCond(void* pVnode, int64_t suid, SArray* list, SNode* cond, SStorageAPI* pAPI, int64_t version) {
16,649,644✔
1580
  int32_t code = 0;
16,649,644✔
1581
  int32_t lino = 0;
16,649,644✔
1582
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
16,649,644✔
1583
    code = optimizeTbnameInCondImpl(pVnode, list, cond, pAPI, suid);
11,948,022✔
1584
  } else {
1585
    code = doInWithAnd(cond, pVnode, list, pAPI, suid);
4,704,024✔
1586
    QUERY_CHECK_CODE(code, lino, end);
4,701,604✔
1587
    code = pAPI->metaFn.getTableTagsByUidVersion(pVnode, suid, list, version);
8,700✔
1588
  }
1589
  
1590
end:
16,645,501✔
1591
  return code;
16,645,501✔
1592
}
1593

1594
static int32_t getTableListInInOperator(void* pVnode, SArray* pExistedUidList, SNodeListNode* pList, SStorageAPI* pStoreAPI,
417,326✔
1595
                                        uint64_t suid) {
1596
  int32_t   code = 0;                                          
417,326✔
1597
  SArray*   pTbList = getTableNameList(pList);
417,326✔
1598
  int32_t   numOfTables = taosArrayGetSize(pTbList);
417,326✔
1599
  SHashObj* uHash = NULL;
417,326✔
1600

1601
  size_t numOfExisted = taosArrayGetSize(pExistedUidList);  // len > 0 means there already have uids
417,326✔
1602
  if (numOfExisted > 0) {
417,326✔
1603
    uHash = taosHashInit(numOfExisted / 0.7, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
2,278✔
1604
    if (!uHash) {
2,278✔
1605
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1606
      code = terrno;
×
1607
      goto end;
×
1608
    }
1609

1610
    for (int i = 0; i < numOfExisted; i++) {
2,276,861✔
1611
      STUidTagInfo* pTInfo = taosArrayGet(pExistedUidList, i);
2,274,583✔
1612
      if (!pTInfo) {
2,274,583✔
1613
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1614
        code = terrno;
×
1615
        goto end;
×
1616
      }
1617
      code = taosHashPut(uHash, &pTInfo->uid, sizeof(uint64_t), &i, sizeof(i));
2,274,583✔
1618
      if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_DUP_KEY) {
2,274,583✔
1619
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1620
        goto end;
×
1621
      }
1622
    }
1623
  }
1624
  taosArrayClear(pExistedUidList);
417,326✔
1625

1626
  for (int i = 0; i < numOfTables; i++) {
1,152,334✔
1627
    char* name = taosArrayGetP(pTbList, i);
735,008✔
1628
    uint64_t uid = 0, csuid = 0;
735,008✔
1629
    if (pStoreAPI->metaFn.getTableUidByName(pVnode, name, &uid) != 0) {
735,008✔
1630
      continue;
310,396✔
1631
    }
1632

1633
    ETableType tbType = TSDB_TABLE_MAX;
424,612✔
1634
    if (pStoreAPI->metaFn.getTableTypeSuidByName(pVnode, name, &tbType, &csuid) == 0 &&
424,612✔
1635
        (tbType == TSDB_CHILD_TABLE || tbType == TSDB_VIRTUAL_CHILD_TABLE) &&
424,612✔
1636
        csuid == suid) {
418,209✔
1637
      if (NULL == uHash || taosHashGet(uHash, &uid, sizeof(uid)) != NULL) {
415,933✔
1638
        STUidTagInfo s = {.uid = uid, .name = name, .pTagVal = NULL};
414,794✔
1639
        void*        tmp = taosArrayPush(pExistedUidList, &s);
414,794✔
1640
        if (!tmp) {
414,794✔
1641
          code = terrno;
×
1642
          goto end;
×
1643
        }
1644
      }
1645
    } 
1646
  }
1647
  taosArraySort(pExistedUidList, filterTableInfoCompare);
417,326✔
1648
  taosArrayRemoveDuplicate(pExistedUidList, filterTableInfoCompare, NULL);
417,326✔
1649
end:
417,326✔
1650
  taosArrayDestroy(pTbList);
417,326✔
1651
  taosHashCleanup(uHash);
417,326✔
1652
  return code;
417,326✔
1653
}
1654

1655
// only return uid that does not contained in pExistedUidList
1656
static int32_t optimizeTbnameInCondImpl(void* pVnode, SArray* pExistedUidList, SNode* pTagCond, SStorageAPI* pStoreAPI,
20,532,913✔
1657
                                        uint64_t suid) {
1658
  SOperatorNode* pNode = (SOperatorNode*)pTagCond;
20,532,913✔
1659
  if (pNode->opType != OP_TYPE_IN) {
20,532,913✔
1660
    return -1;
19,556,674✔
1661
  }
1662

1663
  if ((pNode->pLeft != NULL && ((nodeType(pNode->pLeft) == QUERY_NODE_FUNCTION &&
976,798✔
1664
                                 ((SFunctionNode*)pNode->pLeft)->funcType == FUNCTION_TYPE_TBNAME)) ||
419,666✔
1665
       (nodeType(pNode->pLeft) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pLeft)->colType == COLUMN_TYPE_TBNAME)) &&
556,175✔
1666
      (pNode->pRight != NULL && nodeType(pNode->pRight) == QUERY_NODE_NODE_LIST)) {
418,870✔
1667
    SNodeListNode* pList = (SNodeListNode*)pNode->pRight;
417,326✔
1668

1669
    int32_t len = LIST_LENGTH(pList->pNodeList);
417,326✔
1670
    if (len <= 0) {
417,326✔
1671
      return -1;
×
1672
    }
1673

1674
    return getTableListInInOperator(pVnode, pExistedUidList, pList, pStoreAPI, suid);
417,326✔
1675
  }
1676

1677
  return -1;
559,472✔
1678
}
1679

1680
SSDataBlock* createTagValBlockForFilter(SArray* pColList, int32_t numOfTables, SArray* pUidTagList, void* pVnode,
20,292,728✔
1681
                                        SStorageAPI* pStorageAPI) {
1682
  int32_t      code = TSDB_CODE_SUCCESS;
20,292,728✔
1683
  int32_t      lino = 0;
20,292,728✔
1684
  SSDataBlock* pResBlock = NULL;
20,292,728✔
1685
  code = createDataBlock(&pResBlock);
20,294,454✔
1686
  QUERY_CHECK_CODE(code, lino, _end);
20,289,430✔
1687

1688
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
42,483,508✔
1689
    SColumnInfoData colInfo = {0};
22,191,990✔
1690
    void*           tmp = taosArrayGet(pColList, i);
22,189,672✔
1691
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
22,188,630✔
1692
    colInfo.info = *(SColumnInfo*)tmp;
22,188,630✔
1693
    code = blockDataAppendColInfo(pResBlock, &colInfo);
22,175,864✔
1694
    QUERY_CHECK_CODE(code, lino, _end);
22,191,183✔
1695
  }
1696

1697
  code = blockDataEnsureCapacity(pResBlock, numOfTables);
20,291,899✔
1698
  if (code != TSDB_CODE_SUCCESS) {
20,287,871✔
1699
    terrno = code;
×
1700
    blockDataDestroy(pResBlock);
×
1701
    return NULL;
13✔
1702
  }
1703

1704
  pResBlock->info.rows = numOfTables;
20,287,871✔
1705

1706
  int32_t numOfCols = taosArrayGetSize(pResBlock->pDataBlock);
20,285,437✔
1707

1708
  for (int32_t i = 0; i < numOfTables; i++) {
243,193,869✔
1709
    STUidTagInfo* p1 = taosArrayGet(pUidTagList, i);
222,893,556✔
1710
    QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
222,911,173✔
1711

1712
    for (int32_t j = 0; j < numOfCols; j++) {
456,661,655✔
1713
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, j);
233,691,863✔
1714
      QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
233,686,946✔
1715

1716
      if (pColInfo->info.colId == -1) {  // tbname
233,686,946✔
1717
        char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
7,966,935✔
1718
        if (p1->name != NULL) {
7,969,149✔
1719
          STR_TO_VARSTR(str, p1->name);
6,262✔
1720
        } else {  // name is not retrieved during filter
1721
          code = pStorageAPI->metaFn.getTableNameByUid(pVnode, p1->uid, str);
7,963,543✔
1722
          QUERY_CHECK_CODE(code, lino, _end);
7,961,575✔
1723
        }
1724

1725
        code = colDataSetVal(pColInfo, i, str, false);
7,967,544✔
1726
        QUERY_CHECK_CODE(code, lino, _end);
7,969,055✔
1727
#if TAG_FILTER_DEBUG
1728
        qDebug("tagfilter uid:%ld, tbname:%s", *uid, str + 2);
1729
#endif
1730
      } else {
1731
        STagVal tagVal = {0};
225,708,449✔
1732
        tagVal.cid = pColInfo->info.colId;
225,726,260✔
1733
        if (p1->pTagVal == NULL) {
225,733,241✔
1734
          colDataSetNULL(pColInfo, i);
4,975✔
1735
        } else {
1736
          const char* p = pStorageAPI->metaFn.extractTagVal(p1->pTagVal, pColInfo->info.type, &tagVal);
225,719,815✔
1737

1738
          if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
225,795,043✔
1739
            colDataSetNULL(pColInfo, i);
2,364,042✔
1740
          } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
223,423,563✔
1741
            code = colDataSetVal(pColInfo, i, p, false);
675,185✔
1742
            QUERY_CHECK_CODE(code, lino, _end);
675,185✔
1743
          } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
369,011,115✔
1744
            if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
146,325,499✔
1745
              QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
1746
            }
1747
            char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
146,280,188✔
1748
            QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
146,276,080✔
1749
            varDataSetLen(tmp, tagVal.nData);
146,276,080✔
1750
            memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
146,275,113✔
1751
            code = colDataSetVal(pColInfo, i, tmp, false);
146,266,366✔
1752
#if TAG_FILTER_DEBUG
1753
            qDebug("tagfilter varch:%s", tmp + 2);
1754
#endif
1755
            taosMemoryFree(tmp);
146,279,427✔
1756
            QUERY_CHECK_CODE(code, lino, _end);
146,266,597✔
1757
          } else {
1758
            code = colDataSetVal(pColInfo, i, (const char*)&tagVal.i64, false);
76,443,666✔
1759
            QUERY_CHECK_CODE(code, lino, _end);
76,468,184✔
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:
20,298,483✔
1774
  if (code != TSDB_CODE_SUCCESS) {
20,301,970✔
1775
    blockDataDestroy(pResBlock);
1,267✔
1776
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
293✔
1777
    terrno = code;
293✔
1778
    return NULL;
293✔
1779
  }
1780
  return pResBlock;
20,300,730✔
1781
}
1782

1783
static int32_t doSetQualifiedUid(SArray* pUidList, const SArray* pUidTagList, SColumnInfoData* colData) {
16,632,450✔
1784
  taosArrayClear(pUidList);
16,632,450✔
1785

1786
  int32_t       numOfTables = taosArrayGetSize(pUidTagList);
16,629,884✔
1787
  bool* pResultList = colData == NULL ? NULL : (bool*)colData->pData;
16,633,006✔
1788
  for (int32_t i = 0; i < numOfTables; ++i) {
211,129,982✔
1789
    if (pResultList != NULL && !pResultList[i]) {
194,481,249✔
1790
       continue;
111,140,088✔
1791
    }
1792
    STUidTagInfo* tmpTag = (STUidTagInfo*)taosArrayGet(pUidTagList, i);
83,354,909✔
1793
    if (!tmpTag) {
83,331,056✔
1794
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1795
      return terrno;
×
1796
    }
1797
    uint64_t uid = tmpTag->uid;
83,331,056✔
1798
    qDebug("tagfilter get uid:%" PRId64, uid);
83,353,856✔
1799

1800
    void* tmp = taosArrayPush(pUidList, &uid);
83,360,018✔
1801
    if (tmp == NULL) {
83,360,018✔
1802
      return terrno;
×
1803
    }
1804
  }
1805

1806
  return TSDB_CODE_SUCCESS;
16,648,733✔
1807
}
1808

1809
static int32_t copyExistedUids(SArray* pUidTagList, const SArray* pUidList) {
16,652,462✔
1810
  int32_t code = TSDB_CODE_SUCCESS;
16,652,462✔
1811
  int32_t numOfExisted = taosArrayGetSize(pUidList);
16,652,462✔
1812
  if (numOfExisted == 0) {
16,650,185✔
1813
    return code;
13,739,860✔
1814
  }
1815

1816
  for (int32_t i = 0; i < numOfExisted; ++i) {
32,710,274✔
1817
    uint64_t* uid = taosArrayGet(pUidList, i);
29,800,365✔
1818
    if (!uid) {
29,800,365✔
1819
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1820
      return terrno;
×
1821
    }
1822
    STUidTagInfo info = {.uid = *uid};
29,800,365✔
1823
    void*        tmp = taosArrayPush(pUidTagList, &info);
29,799,949✔
1824
    if (!tmp) {
29,799,949✔
1825
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1826
      return code;
×
1827
    }
1828
  }
1829
  return code;
2,909,909✔
1830
}
1831

1832
static EDealRes getCids(SNode** pNode, void* pContext) {
17,135✔
1833
  SHashObj* colHash = (SHashObj*)pContext;
17,135✔
1834
  col_id_t colId = 0;
17,135✔
1835
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
17,135✔
1836
    colId = (*(SColumnNode**)pNode)->colId;
5,489✔
1837
  } else {
1838
    return DEAL_RES_CONTINUE;
11,646✔
1839
  }
1840

1841
  int32_t tempRes = taosHashPut(colHash, &colId, sizeof(colId), pNode, sizeof((*pNode)));
5,489✔
1842
  if (tempRes != TSDB_CODE_SUCCESS ) {
5,489✔
1843
    return DEAL_RES_ERROR;
×
1844
  }
1845

1846
  return DEAL_RES_CONTINUE;
5,489✔
1847
}
1848

1849
SNode* getTagCondNodeForQueryTmq(void* tinfo) {
3,403✔
1850
  SExecTaskInfo* pTaskInfo = (SExecTaskInfo*)tinfo;
3,403✔
1851
  const char*    id = GET_TASKID(pTaskInfo);
3,403✔
1852
  int32_t        code = 0;
3,403✔
1853

1854
  // traverse to the stream scanner node to add this table id
1855
  SOperatorInfo* pInfo = NULL;
3,403✔
1856
  code = extractOperatorInTree(pTaskInfo->pRoot, QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN, id, &pInfo);
3,403✔
1857
  if (code != 0 || pInfo == NULL) {
3,403✔
1858
    return NULL;
×
1859
  }
1860

1861
  STmqQueryScanInfo* pScanInfo = pInfo->info;
3,403✔
1862
  return pScanInfo->pTagCond;
3,403✔
1863
}
1864

1865
SNode* getTagCondNodeForStableTmq(void* node) {
8,433✔
1866
  return node == NULL ? NULL : ((SSubplan*)node)->pTagCond;
8,433✔
1867
}
1868

1869
bool checkCidInTagCondition(SNode* pTagCond, SArray* cidList) {
9,824✔
1870
  if (pTagCond == NULL) { 
9,824✔
1871
    return false;
4,335✔
1872
  }
1873
  bool result = true;
5,489✔
1874
  SHashObj *colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), true, HASH_NO_LOCK);
5,489✔
1875
  if (colHash == NULL) {
5,489✔
1876
    goto end;
×
1877
  }
1878

1879
  nodesRewriteExprPostOrder(&pTagCond, getCids, (void*)colHash);
5,489✔
1880
  if (cidList != NULL) {
5,489✔
1881
    int32_t numOfCols = taosArrayGetSize(cidList);
620✔
1882
    int32_t i = 0;
620✔
1883
    for (; i < numOfCols; ++i) {
620✔
1884
      col_id_t* pCid = taosArrayGet(cidList, i);
620✔
1885
      void* data = taosHashGet(colHash, pCid, sizeof(*pCid));
620✔
1886
      if (data != NULL) break;
620✔
1887
    }
1888
    if (i == numOfCols) { // no tag column involved in condition
620✔
1889
      result = false;
×
1890
      goto end;
×
1891
    }
1892
  }
1893

1894
end:
5,489✔
1895
  taosHashCleanup(colHash);
5,489✔
1896
  return result;
5,489✔
1897
}
1898

1899
int32_t doFilterByTagCond(int64_t suid, SArray* pUidList, int64_t version, SNode* pTagCond, void* pVnode,
289,930,547✔
1900
                                 SIdxFltStatus status, SStorageAPI* pAPI, void* pStreamInfo) {
1901
  if (pTagCond == NULL) {
289,930,547✔
1902
    return TSDB_CODE_SUCCESS;
273,290,889✔
1903
  }
1904

1905
  terrno = TSDB_CODE_SUCCESS;
16,639,658✔
1906

1907
  int32_t      lino = 0;
16,651,351✔
1908
  int32_t      code = TSDB_CODE_SUCCESS;
16,651,351✔
1909
  SArray*      pBlockList = NULL;
16,651,351✔
1910
  SSDataBlock* pResBlock = NULL;
16,651,351✔
1911
  SScalarParam output = {0};
16,651,496✔
1912
  SArray*      pUidTagList = NULL;
16,651,894✔
1913

1914
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
16,651,894✔
1915

1916
  pUidTagList = taosArrayInit(10, sizeof(STUidTagInfo));
16,645,128✔
1917
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
16,644,384✔
1918

1919
  // pUidList size is 0 if tagIndex is NULL and in query condition
1920
  code = copyExistedUids(pUidTagList, pUidList);
16,644,384✔
1921
  QUERY_CHECK_CODE(code, lino, end);
16,645,491✔
1922

1923
  // Narrow down the scope of the tablelist set if there is tbname in condition and And Logical operator
1924
  code = optimizeTbnameInCond(pVnode, suid, pUidTagList, pTagCond, pAPI, version);
16,645,491✔
1925
  if (code == 0) {
16,643,430✔
1926
    if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
417,326✔
1927
      goto end;
408,787✔
1928
    }
1929
  } else {
1930
    qDebug("pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
16,226,104✔
1931

1932
    FilterCondType condType = checkTagCond(pTagCond);
16,226,894✔
1933
    if (((condType == FILTER_NO_LOGIC || condType == FILTER_AND) && status != SFLT_NOT_INDEX) || // (super table) use tagIndex and operator is and
16,231,274✔
1934
        (status == SFLT_NOT_INDEX && taosArrayGetSize(pUidTagList) > 0)) {                       // (child table with tagCond)
16,230,429✔
1935
      code = pAPI->metaFn.getTableTagsByUidVersion(pVnode, suid, pUidTagList, version);
2,909,386✔
1936
    } else {
1937
      taosArrayClearEx(pUidTagList, freeItem);       // clear tablelist if using tagIndex and or condition
13,322,094✔
1938
      code = pAPI->metaFn.getTableTags(pVnode, suid, pUidTagList);
13,319,504✔
1939
    }
1940
    if (code != TSDB_CODE_SUCCESS) {
16,232,981✔
1941
      qError("failed to get table tags from meta, reason:%s, suid:%" PRIu64, tstrerror(code),suid);
×
1942
      terrno = code;
×
1943
      QUERY_CHECK_CODE(code, lino, end);
×
1944
    }
1945
  }
1946

1947
  qDebug("final pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
16,241,520✔
1948

1949
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
16,243,074✔
1950
  if (numOfTables == 0) {
16,244,903✔
1951
    goto end;
2,609,816✔
1952
  }
1953

1954
  SArray* pColList = NULL;
13,635,087✔
1955
  code = qGetColumnsFromNodeList(pTagCond, false, &pColList); 
13,636,281✔
1956
  if (code != TSDB_CODE_SUCCESS) {
13,620,099✔
1957
    goto end;
×
1958
  }
1959
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
13,620,099✔
1960
  taosArrayDestroy(pColList);
13,631,640✔
1961
  if (pResBlock == NULL) {
13,629,808✔
1962
    code = terrno;
×
1963
    QUERY_CHECK_CODE(code, lino, end);
×
1964
  }
1965

1966
  //fprintDataBlock(pResBlock, "tagFilter", "", 0);
1967

1968
  //  int64_t st1 = taosGetTimestampUs();
1969
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1970
  pBlockList = taosArrayInit(2, POINTER_BYTES);
13,629,808✔
1971
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
13,633,152✔
1972

1973
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
13,635,883✔
1974
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
13,635,883✔
1975

1976
  code = createResultData(&type, numOfTables, &output);
13,635,883✔
1977
  if (code != TSDB_CODE_SUCCESS) {
13,626,730✔
1978
    terrno = code;
×
1979
    QUERY_CHECK_CODE(code, lino, end);
×
1980
  }
1981

1982
  gTaskScalarExtra.pStreamInfo = pStreamInfo;
13,626,730✔
1983
  gTaskScalarExtra.pStreamRange = NULL;
13,626,730✔
1984
  code = scalarCalculate(pTagCond, pBlockList, &output, &gTaskScalarExtra);
13,614,147✔
1985
  if (code != TSDB_CODE_SUCCESS) {
13,613,008✔
1986
    qError("failed to calculate scalar, reason:%s", tstrerror(code));
1,048✔
1987
    terrno = code;
1,048✔
1988
    QUERY_CHECK_CODE(code, lino, end);
1,048✔
1989
  }
1990

1991
end:
16,631,611✔
1992
  if (code == 0) {
16,642,302✔
1993
    code = doSetQualifiedUid(pUidList, pUidTagList, output.columnData);
16,645,313✔
1994
    lino = __LINE__;
16,647,886✔
1995
  }
1996
  
1997
  if (code != 0) {
16,644,875✔
1998
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1,048✔
1999
  }
2000
  blockDataDestroy(pResBlock);
16,644,875✔
2001
  taosArrayDestroy(pBlockList);
16,640,704✔
2002
  taosArrayDestroyEx(pUidTagList, freeItem);
16,630,514✔
2003

2004
  colDataDestroy(output.columnData);
16,648,211✔
2005
  taosMemoryFreeClear(output.columnData);
16,642,595✔
2006
  return code;
16,644,117✔
2007
}
2008

2009
typedef struct {
2010
  int32_t code;
2011
  SStreamRuntimeFuncInfo* pStreamRuntimeInfo;
2012
} PlaceHolderContext;
2013

2014
static EDealRes replacePlaceHolderColumn(SNode** pNode, void* pContext) {
89,392✔
2015
  PlaceHolderContext* pData = (PlaceHolderContext*)pContext;
89,392✔
2016
  if (QUERY_NODE_FUNCTION != nodeType((*pNode))) {
89,392✔
2017
    return DEAL_RES_CONTINUE;
73,604✔
2018
  }
2019
  SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
15,788✔
2020
  if (!fmIsStreamPesudoColVal(pFuncNode->funcId)) {
15,788✔
2021
    return DEAL_RES_CONTINUE;
584✔
2022
  }
2023
  pData->code = fmSetStreamPseudoFuncParamVal(pFuncNode->funcId, pFuncNode->pParameterList, pData->pStreamRuntimeInfo);
15,204✔
2024
  if (pData->code != TSDB_CODE_SUCCESS) {
15,204✔
2025
    return DEAL_RES_ERROR;
×
2026
  }
2027
  SNode* pFirstParam = nodesListGetNode(pFuncNode->pParameterList, 0);
15,204✔
2028
  ((SValueNode*)pFirstParam)->translate = true;
15,204✔
2029
  SValueNode* res = NULL;
15,204✔
2030
  pData->code = nodesCloneNode(pFirstParam, (SNode**)&res);
15,204✔
2031
  if (NULL == res) {
15,204✔
2032
    return DEAL_RES_ERROR;
×
2033
  }
2034
  nodesDestroyNode(*pNode);
15,204✔
2035
  *pNode = (SNode*)res;
15,204✔
2036

2037
  return DEAL_RES_CONTINUE;
15,204✔
2038
}
2039

2040
static void extractTagColId(SOperatorNode* pOpNode, SArray* pColIdArray) {
24,278✔
2041
  SNode* pLeft = pOpNode->pLeft;
24,278✔
2042
  SNode* pRight = pOpNode->pRight;
24,278✔
2043
  SColumnNode* pColNode = nodeType(pLeft) == QUERY_NODE_COLUMN ?
24,278✔
2044
    (SColumnNode*)pLeft : (SColumnNode*)pRight;
24,278✔
2045

2046
  col_id_t colId = pColNode->colId;
24,278✔
2047
  void* _tmp = taosArrayPush(pColIdArray, &colId);
24,278✔
2048
}
24,278✔
2049

2050
static int32_t buildTagCondKey(
12,139✔
2051
  const SNode* pTagCond, char** pTagCondKey,
2052
  int32_t* tagCondKeyLen, SArray** pTagColIds) {
2053
  if (NULL == pTagCond ||
12,139✔
2054
    (nodeType(pTagCond) != QUERY_NODE_OPERATOR &&
12,139✔
2055
      nodeType(pTagCond) != QUERY_NODE_LOGIC_CONDITION)) {
12,139✔
2056
    qError("invalid parameter to extract tag filter symbol");
×
2057
    return TSDB_CODE_INTERNAL_ERROR;
×
2058
  }
2059
  int32_t code = TSDB_CODE_SUCCESS;
12,139✔
2060
  int32_t lino = 0;
12,139✔
2061
  *pTagColIds = taosArrayInit(4, sizeof(col_id_t));
12,139✔
2062

2063
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
12,139✔
2064
    extractTagColId((SOperatorNode*)pTagCond, *pTagColIds);
×
2065
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
12,139✔
2066
    SNode* pChild = NULL;
12,139✔
2067
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
36,417✔
2068
      extractTagColId((SOperatorNode*)pChild, *pTagColIds);
24,278✔
2069
    }
2070
  }
2071

2072
  taosArraySort(*pTagColIds, compareUint16Val);
12,139✔
2073

2074
  // encode ordered colIds into key string, separated by ','
2075
  *tagCondKeyLen =
24,278✔
2076
    (int32_t)(taosArrayGetSize(*pTagColIds) * (sizeof(col_id_t) + 1) - 1);
12,139✔
2077
  *pTagCondKey = (char*)taosMemoryCalloc(1, *tagCondKeyLen);
12,139✔
2078
  TSDB_CHECK_NULL(*pTagCondKey, code, lino, _end, terrno);
12,139✔
2079
  char* pStart = *pTagCondKey;
12,139✔
2080
  for (int32_t i = 0; i < taosArrayGetSize(*pTagColIds); ++i) {
36,417✔
2081
    col_id_t* pColId = (col_id_t*)taosArrayGet(*pTagColIds, i);
24,278✔
2082
    TSDB_CHECK_NULL(pColId, code, lino, _end, terrno);
24,278✔
2083
    memcpy(pStart, pColId, sizeof(col_id_t));
24,278✔
2084
    pStart += sizeof(col_id_t);
24,278✔
2085
    if (i != taosArrayGetSize(*pTagColIds) - 1) {
24,278✔
2086
      *pStart = ',';
12,139✔
2087
      pStart += 1;
12,139✔
2088
    }
2089
  }
2090

2091
_end:
12,139✔
2092
  if (TSDB_CODE_SUCCESS != code) {
12,139✔
2093
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2094
    terrno = code;
×
2095
  }
2096
  return code;
12,139✔
2097
}
2098

2099
static EDealRes canOptimizeTagCondFilter(SNode* pTagCond, void* pContext) {
113,032✔
2100
  if (NULL == pTagCond) {
113,032✔
2101
    *(bool*)pContext = false;
×
2102
    return DEAL_RES_END;
×
2103
  }
2104
  if (nodeType(pTagCond) == QUERY_NODE_VALUE ||
113,032✔
2105
    nodeType(pTagCond) == QUERY_NODE_COLUMN) {
75,023✔
2106
    return DEAL_RES_CONTINUE;
62,287✔
2107
  }
2108
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR &&
50,745✔
2109
    ((SOperatorNode*)pTagCond)->opType == OP_TYPE_EQUAL) {
24,875✔
2110
    return DEAL_RES_CONTINUE;
24,278✔
2111
  }
2112
  if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION &&
26,467✔
2113
    ((SLogicConditionNode*)pTagCond)->condType == LOGIC_COND_TYPE_AND) {
12,139✔
2114
    return DEAL_RES_CONTINUE;
12,139✔
2115
  }
2116
  if (nodeType(pTagCond) == QUERY_NODE_FUNCTION &&
28,059✔
2117
    fmIsStreamPesudoColVal(((SFunctionNode*)pTagCond)->funcId)) {
13,731✔
2118
    return DEAL_RES_CONTINUE;
13,731✔
2119
  }
2120
  *(bool*)pContext = false;
597✔
2121
  return DEAL_RES_END;
597✔
2122
}
2123

2124
int32_t getTableList(void* pVnode, SScanPhysiNode* pScanNode, SNode* pTagCond, SNode* pTagIndexCond,
289,784,151✔
2125
                     STableListInfo* pListInfo, uint8_t* digest, const char* idstr, SStorageAPI* pStorageAPI, void* pStreamInfo) {
2126
  int32_t code = TSDB_CODE_SUCCESS;
289,784,151✔
2127
  int32_t lino = 0;
289,784,151✔
2128
  size_t  numOfTables = 0;
289,784,151✔
2129

2130
  pListInfo->idInfo.suid = pScanNode->suid;
289,784,151✔
2131
  pListInfo->idInfo.tableType = pScanNode->tableType;
289,799,928✔
2132

2133
  SArray* pUidList = taosArrayInit(8, sizeof(uint64_t));
289,683,082✔
2134
  QUERY_CHECK_NULL(pUidList, code, lino, end, terrno);
289,635,663✔
2135

2136
  char*   pTagCondKey = NULL;
289,635,663✔
2137
  int32_t tagCondKeyLen;
289,684,945✔
2138
  SArray* pTagColIds = NULL;
289,689,665✔
2139
  char*   pPayload = NULL;
289,707,621✔
2140
  SIdxFltStatus status = SFLT_NOT_INDEX;
289,707,621✔
2141

2142
  qTrace("getTableList called, suid:%" PRIu64
289,707,621✔
2143
    ", tagCond:%p, tagIndexCond:%p, %d %d", pScanNode->suid, pTagCond,
2144
    pTagIndexCond, pScanNode->tableType, pScanNode->virtualStableScan);
2145
  if (pScanNode->tableType != TSDB_SUPER_TABLE && !pScanNode->virtualStableScan) {
289,710,143✔
2146
    pListInfo->idInfo.uid = pScanNode->uid;
157,476,166✔
2147
    if (pStorageAPI->metaFn.isTableExisted(pVnode, pScanNode->uid)) {
157,466,168✔
2148
      void* tmp = taosArrayPush(pUidList, &pScanNode->uid);
157,449,504✔
2149
      QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
157,449,336✔
2150
    }
2151
    code = doFilterByTagCond(pScanNode->suid, pUidList, -1, pTagCond, pVnode, status, pStorageAPI, pStreamInfo);
157,501,554✔
2152
    QUERY_CHECK_CODE(code, lino, end);
157,485,086✔
2153
  } else {
2154
    bool      isStream = (pStreamInfo != NULL);
132,366,245✔
2155
    bool      hasTagCond = (pTagCond != NULL);
132,366,245✔
2156
    bool      canCacheTagEqCondFilter = false;
132,366,245✔
2157
    T_MD5_CTX context = {0};
132,311,000✔
2158

2159
    qTrace("start to get table list by tag filter, suid:%" PRIu64
132,355,239✔
2160
      ",tsStableTagFilterCache:%d, tsTagFilterCache:%d", 
2161
      pScanNode->suid, tsStableTagFilterCache, tsTagFilterCache);
2162

2163
    bool acquired = false;
132,355,239✔
2164
    // first, check whether we can use stable tag filter cache
2165
    if (tsStableTagFilterCache && isStream && hasTagCond) {
132,285,453✔
2166
      canCacheTagEqCondFilter = true;
12,736✔
2167
      nodesWalkExpr(pTagCond, canOptimizeTagCondFilter,
12,736✔
2168
        (void*)&canCacheTagEqCondFilter);
2169
    }
2170
    if (canCacheTagEqCondFilter) {
132,234,991✔
2171
      qDebug("%s, stable tag filter condition can be optimized", idstr);
12,139✔
2172
      if (((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
12,139✔
2173
        SNode* tmp = NULL;
12,139✔
2174
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
12,139✔
2175
        QUERY_CHECK_CODE(code, lino, end);
12,139✔
2176

2177
        PlaceHolderContext ctx = {.code = TSDB_CODE_SUCCESS, .pStreamRuntimeInfo = (SStreamRuntimeFuncInfo*)pStreamInfo};
12,139✔
2178
        nodesRewriteExpr(&tmp, replacePlaceHolderColumn, (void*)&ctx);
12,139✔
2179
        if (TSDB_CODE_SUCCESS != ctx.code) {
12,139✔
2180
          nodesDestroyNode(tmp);
×
2181
          code = ctx.code;
×
2182
          goto end;
×
2183
        }
2184
        code = genStableTagFilterDigest(tmp, &context);
12,139✔
2185
        nodesDestroyNode(tmp);
12,139✔
2186
      } else {
2187
        code = genStableTagFilterDigest(pTagCond, &context);
×
2188
      }
2189
      QUERY_CHECK_CODE(code, lino, end);
12,139✔
2190

2191
      code = buildTagCondKey(
12,139✔
2192
        pTagCond, &pTagCondKey, &tagCondKeyLen, &pTagColIds);
2193
      QUERY_CHECK_CODE(code, lino, end);
12,139✔
2194
      code = pStorageAPI->metaFn.getStableCachedTableList(
12,139✔
2195
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
12,139✔
2196
        context.digest, tListLen(context.digest), pUidList, &acquired);
2197
      QUERY_CHECK_CODE(code, lino, end);
11,940✔
2198
    } else if (tsTagFilterCache) {
132,222,852✔
2199
      // second, try to use normal tag filter cache
2200
      qDebug("%s using normal tag filter cache", idstr);
36,499✔
2201
      if (pStreamInfo != NULL && ((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
38,171✔
2202
        SNode* tmp = NULL;
1,473✔
2203
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
1,473✔
2204
        QUERY_CHECK_CODE(code, lino, end);
1,473✔
2205

2206
        PlaceHolderContext ctx = {.code = TSDB_CODE_SUCCESS, .pStreamRuntimeInfo = (SStreamRuntimeFuncInfo*)pStreamInfo};
1,473✔
2207
        nodesRewriteExpr(&tmp, replacePlaceHolderColumn, (void*)&ctx);
1,473✔
2208
        if (TSDB_CODE_SUCCESS != ctx.code) {
1,473✔
2209
          nodesDestroyNode(tmp);
×
2210
          code = ctx.code;
×
2211
          goto end;
×
2212
        }
2213
        code = genTagFilterDigest(tmp, &context);
1,473✔
2214
        nodesDestroyNode(tmp);
1,473✔
2215
      } else {
2216
        code = genTagFilterDigest(pTagCond, &context);
35,225✔
2217
      }
2218
      // try to retrieve the result from meta cache
2219
      QUERY_CHECK_CODE(code, lino, end);      
36,499✔
2220
      code = pStorageAPI->metaFn.getCachedTableList(
36,499✔
2221
        pVnode, pScanNode->suid, context.digest,
36,499✔
2222
        tListLen(context.digest), pUidList, &acquired);
2223
      QUERY_CHECK_CODE(code, lino, end);
30,665✔
2224
    }
2225
    if (acquired) {
132,225,146✔
2226
      taosArrayDestroy(pTagColIds);
33,672✔
2227
      pTagColIds = NULL;
33,672✔
2228
      
2229
      if (digest != NULL) {
33,672✔
2230
        digest[0] = 1;
33,672✔
2231
        memcpy(digest + 1, context.digest, tListLen(context.digest));
33,672✔
2232
      }
2233
      
2234
      qDebug("suid:%" PRIu64 ", %s retrieve table uid list from cache, numOfTables:%d", 
33,672✔
2235
        pScanNode->suid, idstr, (int32_t)taosArrayGetSize(pUidList));
2236
      goto end;
33,871✔
2237
    } else {
2238
      qDebug("suid:%" PRIu64 
132,191,474✔
2239
        ", failed to get table uid list from cache", pScanNode->suid);
2240
    }
2241

2242
    if (!pTagCond) {  // no tag filter condition exists, let's fetch all tables of this super table
132,383,089✔
2243
      code = pStorageAPI->metaFn.getChildTableList(pVnode, pScanNode->suid, pUidList);
116,089,236✔
2244
      QUERY_CHECK_CODE(code, lino, end);
116,018,605✔
2245
      qTrace("no tag filter, get all child tables, numOfTables:%d", (int32_t)taosArrayGetSize(pUidList));
116,018,605✔
2246
    } else {
2247
      SIdxFltStatus status = SFLT_NOT_INDEX;
16,293,853✔
2248
      if (pTagIndexCond) {
16,293,689✔
2249
        void* pIndex = pStorageAPI->metaFn.getInvertIndex(pVnode);
4,756,864✔
2250

2251
        SIndexMetaArg metaArg = {.metaEx = pVnode,
4,756,890✔
2252
                                 .idx = pStorageAPI->metaFn.storeGetIndexInfo(pVnode),
4,756,864✔
2253
                                 .ivtIdx = pIndex,
2254
                                 .suid = pScanNode->uid};
4,756,864✔
2255
        code = doFilterTag(pTagIndexCond, &metaArg, pUidList, &status, &pStorageAPI->metaFilter);
4,756,864✔
2256
        if (code != 0 || status == SFLT_NOT_INDEX) {  // temporarily disable it for performance sake
4,742,125✔
2257
          qDebug("failed to get tableIds from index, suid:%" PRIu64 ", uidListSize:%d", pScanNode->uid, (int32_t)taosArrayGetSize(pUidList));
1,316,198✔
2258
        } else {
2259
          qDebug("succ to get filter result, table num: %d", (int)taosArrayGetSize(pUidList));
3,425,927✔
2260
        }
2261
      }
2262
    }
2263
    qTrace("after index filter, pTagCond:%p uidListSize:%d", pTagCond, (int32_t)taosArrayGetSize(pUidList));
132,344,299✔
2264
    code = doFilterByTagCond(pScanNode->suid, pUidList, -1, pTagCond, pVnode, status, pStorageAPI, pStreamInfo);
132,344,718✔
2265
    QUERY_CHECK_CODE(code, lino, end);
132,305,202✔
2266

2267
    // let's add the filter results into meta-cache
2268
    numOfTables = taosArrayGetSize(pUidList);
132,304,154✔
2269

2270
    if (canCacheTagEqCondFilter) {
132,330,873✔
2271
      qInfo("%s, suid:%" PRIu64 ", add uid list to stable tag filter cache, "
5,572✔
2272
            "uidListSize:%d, origin key:%" PRIu64 ",%" PRIu64,
2273
            idstr, pScanNode->suid, (int32_t)numOfTables,
2274
            *(uint64_t*)context.digest, *(uint64_t*)(context.digest + 8));
2275

2276
      code = pStorageAPI->metaFn.putStableCachedTableList(
5,572✔
2277
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
2278
        context.digest, tListLen(context.digest),
2279
        pUidList, &pTagColIds);
2280
      QUERY_CHECK_CODE(code, lino, end);
5,572✔
2281

2282
      if (digest != NULL) {
5,572✔
2283
        digest[0] = 1;
5,572✔
2284
        memcpy(digest + 1, context.digest, tListLen(context.digest));
5,572✔
2285
      }
2286
    } else if (tsTagFilterCache) {
132,325,301✔
2287
      qInfo("%s, suid:%" PRIu64 ", add uid list to normal tag filter cache, "
9,195✔
2288
            "uidListSize:%d, origin key:%" PRIu64 ",%" PRIu64,
2289
            idstr, pScanNode->suid, (int32_t)numOfTables,
2290
            *(uint64_t*)context.digest, *(uint64_t*)(context.digest + 8));
2291
      size_t size = numOfTables * sizeof(uint64_t) + sizeof(int32_t);
9,195✔
2292
      pPayload = taosMemoryMalloc(size);
9,195✔
2293
      QUERY_CHECK_NULL(pPayload, code, lino, end, terrno);
9,195✔
2294

2295
      *(int32_t*)pPayload = (int32_t)numOfTables;
9,195✔
2296
      if (numOfTables > 0) {
9,195✔
2297
        void* tmp = taosArrayGet(pUidList, 0);
7,616✔
2298
        QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
7,616✔
2299
        memcpy(pPayload + sizeof(int32_t), tmp, numOfTables * sizeof(uint64_t));
7,616✔
2300
      }
2301

2302
      code = pStorageAPI->metaFn.putCachedTableList(pVnode, pScanNode->suid,
9,195✔
2303
                                                    context.digest,
2304
                                                    tListLen(context.digest),
2305
                                                    pPayload, size, 1);
2306
      if (TSDB_CODE_SUCCESS == code) {
9,195✔
2307
        /*
2308
          data referenced by pPayload is used in lru cache,
2309
          reset pPayload to NULL to avoid being freed in _error block
2310
        */
2311
        pPayload = NULL;
8,996✔
2312
      } else {
2313
        if (TSDB_CODE_DUP_KEY == code) {
199✔
2314
          /*
2315
            another thread has already put the same key into cache,
2316
            we can just ignore this error
2317
          */
2318
          code = TSDB_CODE_SUCCESS;
199✔
2319
        }
2320
        QUERY_CHECK_CODE(code, lino, end);
199✔
2321
      }
2322

2323
      if (digest != NULL) {
9,195✔
2324
        digest[0] = 1;
9,195✔
2325
        memcpy(digest + 1, context.digest, tListLen(context.digest));
9,195✔
2326
      }
2327
    }
2328
  }
2329

2330
  qDebug("%s, table list with %d uids built", idstr, (int32_t)numOfTables);
289,786,768✔
2331

2332
end:
289,891,753✔
2333
  if (code == 0) {
289,894,259✔
2334
     STableKeyInfo info = {.uid = 0, .groupId = 0};
289,893,211✔
2335
    for (int32_t i = 0; i < taosArrayGetSize(pUidList); ++i) {
1,020,374,691✔
2336
      int64_t* uid = (int64_t*)taosArrayGet(pUidList, i);
730,403,135✔
2337
      QUERY_CHECK_NULL(uid, code, lino, end, terrno);
730,433,819✔
2338

2339
      info.uid = *uid;
730,433,819✔
2340
        //qInfo("doSetQualifiedUid row:%d added to pTableList", i);
2341
      void* p = taosArrayPush(pListInfo->pTableList, &info);
730,439,304✔
2342
      QUERY_CHECK_NULL(p, code, lino, end, terrno);
730,468,874✔
2343
    }
2344
  } else {
2345
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1,048✔
2346
  }
2347
  taosArrayDestroy(pUidList);
289,944,092✔
2348
  taosArrayDestroy(pTagColIds);
289,832,348✔
2349
  taosMemFreeClear(pTagCondKey);
289,856,992✔
2350
  taosMemFreeClear(pPayload);
289,856,992✔
2351
  return code;
289,856,992✔
2352
}
2353

2354
int32_t qGetTableList(int64_t suid, void* pVnode, void* node, SArray** tableList, void* pTaskInfo) {
26,729✔
2355
  int32_t        code = TSDB_CODE_SUCCESS;
26,729✔
2356
  int32_t        lino = 0;
26,729✔
2357
  SSubplan*      pSubplan = (SSubplan*)node;
26,729✔
2358
  SScanPhysiNode pNode = {0};
26,729✔
2359
  pNode.suid = suid;
26,729✔
2360
  pNode.uid = suid;
26,729✔
2361
  pNode.tableType = TSDB_SUPER_TABLE;
26,729✔
2362

2363
  STableListInfo* pTableListInfo = tableListCreate();
26,729✔
2364
  QUERY_CHECK_NULL(pTableListInfo, code, lino, _end, terrno);
26,729✔
2365
  code = getTableList(pVnode, &pNode, pSubplan ? pSubplan->pTagCond : NULL, pSubplan ? pSubplan->pTagIndexCond : NULL,
26,729✔
2366
                      pTableListInfo, NULL, "qGetTableList", &((SExecTaskInfo*)pTaskInfo)->storageAPI, NULL);
2367
  QUERY_CHECK_CODE(code, lino, _end);
26,729✔
2368
  *tableList = pTableListInfo->pTableList;
26,729✔
2369
  pTableListInfo->pTableList = NULL;
26,729✔
2370

2371
_end:
26,729✔
2372
  tableListDestroy(pTableListInfo);
26,729✔
2373
  if (code != TSDB_CODE_SUCCESS) {
26,729✔
2374
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2375
  }
2376
  return code;
26,729✔
2377
}
2378

2379
size_t getTableTagsBufLen(const SNodeList* pGroups) {
×
2380
  size_t keyLen = 0;
×
2381

2382
  SNode* node;
2383
  FOREACH(node, pGroups) {
×
2384
    SExprNode* pExpr = (SExprNode*)node;
×
2385
    keyLen += pExpr->resType.bytes;
×
2386
  }
2387

2388
  keyLen += sizeof(int8_t) * LIST_LENGTH(pGroups);
×
2389
  return keyLen;
×
2390
}
2391

2392
int32_t getGroupIdFromTagsVal(void* pVnode, uint64_t uid, SNodeList* pGroupNode, char* keyBuf, uint64_t* pGroupId,
×
2393
                              SStorageAPI* pAPI) {
2394
  SMetaReader mr = {0};
×
2395

2396
  pAPI->metaReaderFn.initReader(&mr, pVnode, META_READER_LOCK, &pAPI->metaFn);
×
2397
  if (pAPI->metaReaderFn.getEntryGetUidCache(&mr, uid) != 0) {  // table not exist
×
2398
    pAPI->metaReaderFn.clearReader(&mr);
×
2399
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
2400
  }
2401

2402
  SNodeList* groupNew = NULL;
×
2403
  int32_t    code = nodesCloneList(pGroupNode, &groupNew);
×
2404
  if (TSDB_CODE_SUCCESS != code) {
×
2405
    pAPI->metaReaderFn.clearReader(&mr);
×
2406
    return code;
×
2407
  }
2408

2409
  STransTagExprCtx ctx = {.code = 0, .pReader = &mr};
×
2410
  nodesRewriteExprsPostOrder(groupNew, doTranslateTagExpr, &ctx);
×
2411
  if (TSDB_CODE_SUCCESS != ctx.code) {
×
2412
    nodesDestroyList(groupNew);
×
2413
    pAPI->metaReaderFn.clearReader(&mr);
×
2414
    return code;
×
2415
  }
2416
  char* isNull = (char*)keyBuf;
×
2417
  char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(pGroupNode);
×
2418

2419
  SNode*  pNode;
2420
  int32_t index = 0;
×
2421
  FOREACH(pNode, groupNew) {
×
2422
    SNode*  pNew = NULL;
×
2423
    int32_t code = scalarCalculateConstants(pNode, &pNew);
×
2424
    if (TSDB_CODE_SUCCESS == code) {
×
2425
      REPLACE_NODE(pNew);
×
2426
    } else {
2427
      nodesDestroyList(groupNew);
×
2428
      pAPI->metaReaderFn.clearReader(&mr);
×
2429
      return code;
×
2430
    }
2431

2432
    if (nodeType(pNew) != QUERY_NODE_VALUE) {
×
2433
      nodesDestroyList(groupNew);
×
2434
      pAPI->metaReaderFn.clearReader(&mr);
×
2435
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2436
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2437
    }
2438
    SValueNode* pValue = (SValueNode*)pNew;
×
2439

2440
    if (pValue->node.resType.type == TSDB_DATA_TYPE_NULL || pValue->isNull) {
×
2441
      isNull[index++] = 1;
×
2442
      continue;
×
2443
    } else {
2444
      isNull[index++] = 0;
×
2445
      char* data = nodesGetValueFromNode(pValue);
×
2446
      if (pValue->node.resType.type == TSDB_DATA_TYPE_JSON) {
×
2447
        if (tTagIsJson(data)) {
×
2448
          terrno = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
×
2449
          nodesDestroyList(groupNew);
×
2450
          pAPI->metaReaderFn.clearReader(&mr);
×
2451
          return terrno;
×
2452
        }
2453
        int32_t len = getJsonValueLen(data);
×
2454
        memcpy(pStart, data, len);
×
2455
        pStart += len;
×
2456
      } else if (IS_VAR_DATA_TYPE(pValue->node.resType.type)) {
×
2457
        if (IS_STR_DATA_BLOB(pValue->node.resType.type)) {
×
2458
          return TSDB_CODE_BLOB_NOT_SUPPORT_TAG;
×
2459
        }
2460
        memcpy(pStart, data, varDataTLen(data));
×
2461
        pStart += varDataTLen(data);
×
2462
      } else {
2463
        memcpy(pStart, data, pValue->node.resType.bytes);
×
2464
        pStart += pValue->node.resType.bytes;
×
2465
      }
2466
    }
2467
  }
2468

2469
  int32_t len = (int32_t)(pStart - (char*)keyBuf);
×
2470
  *pGroupId = calcGroupId(keyBuf, len);
×
2471

2472
  nodesDestroyList(groupNew);
×
2473
  pAPI->metaReaderFn.clearReader(&mr);
×
2474

2475
  return TSDB_CODE_SUCCESS;
×
2476
}
2477

2478
SArray* makeColumnArrayFromList(SNodeList* pNodeList) {
3,292,001✔
2479
  if (!pNodeList) {
3,292,001✔
2480
    return NULL;
×
2481
  }
2482

2483
  size_t  numOfCols = LIST_LENGTH(pNodeList);
3,292,001✔
2484
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
3,294,311✔
2485
  if (pList == NULL) {
3,287,900✔
2486
    return NULL;
×
2487
  }
2488

2489
  for (int32_t i = 0; i < numOfCols; ++i) {
7,620,044✔
2490
    SColumnNode* pColNode = (SColumnNode*)nodesListGetNode(pNodeList, i);
4,334,005✔
2491
    if (!pColNode) {
4,333,328✔
2492
      taosArrayDestroy(pList);
×
2493
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2494
      return NULL;
×
2495
    }
2496

2497
    // todo extract method
2498
    SColumn c = {0};
4,333,328✔
2499
    c.slotId = pColNode->slotId;
4,334,158✔
2500
    c.colId = pColNode->colId;
4,333,578✔
2501
    c.type = pColNode->node.resType.type;
4,332,972✔
2502
    c.bytes = pColNode->node.resType.bytes;
4,334,279✔
2503
    c.precision = pColNode->node.resType.precision;
4,334,153✔
2504
    c.scale = pColNode->node.resType.scale;
4,330,858✔
2505

2506
    void* tmp = taosArrayPush(pList, &c);
4,334,857✔
2507
    if (!tmp) {
4,334,857✔
2508
      taosArrayDestroy(pList);
×
2509
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2510
      return NULL;
×
2511
    }
2512
  }
2513

2514
  return pList;
3,286,039✔
2515
}
2516

2517
int32_t extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols,
371,396,692✔
2518
                            int32_t type, SColMatchInfo* pMatchInfo) {
2519
  size_t  numOfCols = LIST_LENGTH(pNodeList);
371,396,692✔
2520
  int32_t code = TSDB_CODE_SUCCESS;
371,445,394✔
2521
  int32_t lino = 0;
371,445,394✔
2522
  bool    colIdOrdered = true;
371,445,394✔
2523
  bool    hasPrevColId = false;
371,445,394✔
2524
  int32_t prevColId = 0;
371,445,394✔
2525

2526
  pMatchInfo->matchType = type;
371,445,394✔
2527

2528
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchItem));
371,393,206✔
2529
  if (pList == NULL) {
371,268,434✔
2530
    code = terrno;
×
2531
    return code;
×
2532
  }
2533

2534
  SColMatchItem** infoBySlot = NULL;
371,268,434✔
2535
  if (numOfCols > 0) {
371,268,434✔
2536
    infoBySlot = taosMemoryCalloc(numOfCols, sizeof(*infoBySlot));
371,186,239✔
2537
    if (infoBySlot == NULL) {
371,275,504✔
2538
      code = terrno;
×
2539
      goto _end;
×
2540
    }
2541
  }
2542

2543
  SNode* node = NULL;
371,357,699✔
2544
  FOREACH(node, pNodeList) {
1,796,626,615✔
2545
    STargetNode* pNode = (STargetNode*)node;
1,425,039,133✔
2546
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,425,039,133✔
2547
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
1,425,039,133✔
2548
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
1,413,532,462✔
2549

2550
      SColMatchItem c = {.needOutput = true};
1,413,577,878✔
2551
      c.colId = pColNode->colId;
1,413,530,640✔
2552
      c.srcSlotId = pColNode->slotId;
1,413,584,940✔
2553
      c.dstSlotId = pNode->slotId;
1,413,562,903✔
2554
      c.isPk = pColNode->isPk;
1,413,652,979✔
2555
      c.dataType = pColNode->node.resType;
1,413,596,947✔
2556
      void* tmp = taosArrayPush(pList, &c);
1,413,667,350✔
2557
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,413,667,350✔
2558
      if (hasPrevColId) {
1,413,667,350✔
2559
        if (c.colId < prevColId) {
1,042,641,081✔
2560
          colIdOrdered = false;
17,085,483✔
2561
        }
2562
      } else {
2563
        hasPrevColId = true;
371,026,269✔
2564
      }
2565
      prevColId = c.colId;
1,413,667,350✔
2566
      if (pNode->slotId >= 0 && pNode->slotId < numOfCols) {
1,413,667,350✔
2567
        infoBySlot[pNode->slotId] = (SColMatchItem*)taosArrayGet(pList, taosArrayGetSize(pList) - 1);
1,413,045,031✔
2568
      }
2569
    }
2570
  }
2571
  // set the output flag for each column in SColMatchInfo, according to the
2572
  *numOfOutputCols = 0;
371,461,741✔
2573
  SNode* slotNode = NULL;
371,460,619✔
2574
  FOREACH(slotNode, pOutputNodeList->pSlots) {
1,975,875,676✔
2575
    SSlotDescNode* pNode = (SSlotDescNode*)slotNode;
1,604,383,445✔
2576
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,604,383,445✔
2577

2578
    // todo: add reserve flag check
2579
    // it is a column reserved for the arithmetic expression calculation
2580
    if (pNode->slotId >= numOfCols) {
1,604,383,445✔
2581
      (*numOfOutputCols) += 1;
179,595,787✔
2582
      continue;
179,602,122✔
2583
    }
2584

2585
    SColMatchItem* info = infoBySlot ? infoBySlot[pNode->slotId] : NULL;
1,424,952,404✔
2586

2587
    if (pNode->output) {
1,424,996,971✔
2588
      (*numOfOutputCols) += 1;
1,403,751,890✔
2589
    } else if (info != NULL) {
21,255,161✔
2590
      // select distinct tbname from stb where tbname='abc';
2591
      info->needOutput = false;
21,202,506✔
2592
    }
2593
  }
2594

2595
  pMatchInfo->pList = pList;
371,353,034✔
2596
  pMatchInfo->colIdOrdered = colIdOrdered;
371,356,855✔
2597

2598
_end:
371,276,598✔
2599
  taosMemoryFree(infoBySlot);
371,276,598✔
2600
  if (code != TSDB_CODE_SUCCESS) {
371,325,704✔
2601
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2602
  }
2603
  return code;
371,303,585✔
2604
}
2605

2606
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
1,228,195,218✔
2607
                                  const char* name) {
2608
  SResSchema s = {0};
1,228,195,218✔
2609
  s.scale = scale;
1,228,260,272✔
2610
  s.type = type;
1,228,260,272✔
2611
  s.bytes = bytes;
1,228,260,272✔
2612
  s.slotId = slotId;
1,228,260,272✔
2613
  s.precision = precision;
1,228,260,272✔
2614
  tstrncpy(s.name, name, tListLen(s.name));
1,228,260,272✔
2615

2616
  return s;
1,228,070,983✔
2617
}
2618

2619
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType, EColumnType colType) {
1,142,294,423✔
2620
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
1,142,294,423✔
2621
  if (pCol == NULL) {
1,141,760,992✔
2622
    return NULL;
×
2623
  }
2624

2625
  pCol->slotId = slotId;
1,141,760,992✔
2626
  pCol->colId = colId;
1,141,794,766✔
2627
  pCol->bytes = pType->bytes;
1,141,840,161✔
2628
  pCol->type = pType->type;
1,141,954,627✔
2629
  pCol->scale = pType->scale;
1,141,968,544✔
2630
  pCol->precision = pType->precision;
1,142,201,809✔
2631
  pCol->dataBlockId = blockId;
1,142,160,212✔
2632
  pCol->colType = colType;
1,142,150,561✔
2633
  return pCol;
1,142,221,392✔
2634
}
2635

2636
int32_t createExprFromOneNode(SExprInfo* pExp, SNode* pNode, int16_t slotId) {
1,228,583,384✔
2637
  int32_t code = TSDB_CODE_SUCCESS;
1,228,583,384✔
2638
  int32_t lino = 0;
1,228,583,384✔
2639
  pExp->base.numOfParams = 0;
1,228,583,384✔
2640
  pExp->base.pParam = NULL;
1,228,649,277✔
2641
  pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
1,228,523,087✔
2642
  QUERY_CHECK_NULL(pExp->pExpr, code, lino, _end, terrno);
1,228,094,015✔
2643

2644
  pExp->pExpr->_function.num = 1;
1,228,167,607✔
2645
  pExp->pExpr->_function.functionId = -1;
1,228,198,850✔
2646

2647
  int32_t type = nodeType(pNode);
1,228,414,382✔
2648
  // it is a project query, or group by column
2649
  if (type == QUERY_NODE_COLUMN) {
1,228,504,945✔
2650
    pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
715,435,165✔
2651
    SColumnNode* pColNode = (SColumnNode*)pNode;
715,476,372✔
2652

2653
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
715,476,372✔
2654
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
715,448,668✔
2655

2656
    pExp->base.numOfParams = 1;
715,396,711✔
2657

2658
    SDataType* pType = &pColNode->node.resType;
715,498,713✔
2659
    pExp->base.resSchema =
2660
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pColNode->colName);
715,478,132✔
2661

2662
    pExp->base.pParam[0].pCol =
1,431,067,332✔
2663
        createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType, pColNode->colType);
1,431,014,898✔
2664
    QUERY_CHECK_NULL(pExp->base.pParam[0].pCol, code, lino, _end, terrno);
715,508,368✔
2665

2666
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
715,436,507✔
2667
  } else if (type == QUERY_NODE_VALUE) {
513,069,780✔
2668
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
18,282,478✔
2669
    SValueNode* pValNode = (SValueNode*)pNode;
18,282,437✔
2670

2671
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
18,282,437✔
2672
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
18,275,661✔
2673

2674
    pExp->base.numOfParams = 1;
18,274,602✔
2675

2676
    SDataType* pType = &pValNode->node.resType;
18,284,211✔
2677
    pExp->base.resSchema =
2678
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
18,272,597✔
2679
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
18,272,967✔
2680
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
18,278,258✔
2681
    QUERY_CHECK_CODE(code, lino, _end);
18,283,336✔
2682
  } else if (type == QUERY_NODE_REMOTE_VALUE) {
494,787,302✔
2683
    SRemoteValueNode* pRemote = (SRemoteValueNode*)pNode;
1,514,112✔
2684
    code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, pNode);
1,514,112✔
2685
    QUERY_CHECK_CODE(code, lino, _end);
1,515,272✔
2686

2687
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
1,068,092✔
2688
    SValueNode* pValNode = (SValueNode*)pNode;
1,068,092✔
2689

2690
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
1,068,092✔
2691
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
1,068,092✔
2692

2693
    pExp->base.numOfParams = 1;
1,068,092✔
2694

2695
    SDataType* pType = &pValNode->node.resType;
1,068,092✔
2696
    pExp->base.resSchema =
2697
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
1,068,092✔
2698
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
1,068,092✔
2699
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
1,068,092✔
2700
    QUERY_CHECK_CODE(code, lino, _end);
1,068,113✔
2701
  } else if (type == QUERY_NODE_FUNCTION) {
493,273,190✔
2702
    pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
449,506,429✔
2703
    SFunctionNode* pFuncNode = (SFunctionNode*)pNode;
449,520,317✔
2704

2705
    SDataType* pType = &pFuncNode->node.resType;
449,520,317✔
2706
    pExp->base.resSchema =
2707
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pFuncNode->node.aliasName);
449,544,810✔
2708
    tExprNode* pExprNode = pExp->pExpr;
449,476,210✔
2709

2710
    pExprNode->_function.functionId = pFuncNode->funcId;
449,496,154✔
2711
    pExprNode->_function.pFunctNode = pFuncNode;
449,545,779✔
2712
    pExprNode->_function.functionType = pFuncNode->funcType;
449,524,270✔
2713

2714
    tstrncpy(pExprNode->_function.functionName, pFuncNode->functionName, tListLen(pExprNode->_function.functionName));
449,572,112✔
2715

2716
    pExp->base.pParamList = pFuncNode->pParameterList;
449,586,250✔
2717
#if 1
2718
    // todo refactor: add the parameter for tbname function
2719
    const char* name = "tbname";
449,522,479✔
2720
    int32_t     len = strlen(name);
449,522,479✔
2721

2722
    if (!pFuncNode->pParameterList && (memcmp(pExprNode->_function.functionName, name, len) == 0) &&
449,522,479✔
2723
        pExprNode->_function.functionName[len] == 0) {
30,350,053✔
2724
      pFuncNode->pParameterList = NULL;
30,353,254✔
2725
      int32_t     code = nodesMakeList(&pFuncNode->pParameterList);
30,356,084✔
2726
      SValueNode* res = NULL;
30,352,522✔
2727
      if (TSDB_CODE_SUCCESS == code) {
30,355,660✔
2728
        code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
30,358,153✔
2729
      }
2730
      QUERY_CHECK_CODE(code, lino, _end);
30,362,881✔
2731
      res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
30,362,881✔
2732
      code = nodesListAppend(pFuncNode->pParameterList, (SNode*)res);
30,362,640✔
2733
      if (code != TSDB_CODE_SUCCESS) {
30,360,378✔
2734
        nodesDestroyNode((SNode*)res);
×
2735
        res = NULL;
×
2736
      }
2737
      QUERY_CHECK_CODE(code, lino, _end);
30,360,378✔
2738
    }
2739
#endif
2740

2741
    int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
449,565,593✔
2742

2743
    pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
449,558,925✔
2744
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
449,448,340✔
2745
    pExp->base.numOfParams = numOfParam;
449,475,941✔
2746

2747
    for (int32_t j = 0; j < numOfParam && TSDB_CODE_SUCCESS == code; ++j) {
1,055,789,445✔
2748
      SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
606,497,896✔
2749
      QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
606,485,115✔
2750
      if (p1->type == QUERY_NODE_COLUMN) {
606,485,102✔
2751
        SColumnNode* pcn = (SColumnNode*)p1;
426,760,380✔
2752

2753
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
426,760,380✔
2754
        pExp->base.pParam[j].pCol =
853,477,618✔
2755
            createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType, pcn->colType);
853,430,346✔
2756
        QUERY_CHECK_NULL(pExp->base.pParam[j].pCol, code, lino, _end, terrno);
426,775,314✔
2757
      } else if (p1->type == QUERY_NODE_VALUE) {
179,721,190✔
2758
        SValueNode* pvn = (SValueNode*)p1;
117,512,305✔
2759
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
117,512,305✔
2760
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
117,471,284✔
2761
        QUERY_CHECK_CODE(code, lino, _end);
117,482,511✔
2762
      } else if (p1->type == QUERY_NODE_REMOTE_VALUE) {
62,278,948✔
2763
        SRemoteValueNode* pRemote = (SRemoteValueNode*)p1;
77,720✔
2764
        code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, p1);
77,720✔
2765
        QUERY_CHECK_CODE(code, lino, _end);
77,720✔
2766

2767
        SValueNode* pvn = (SValueNode*)pRemote;
56,260✔
2768
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
56,260✔
2769
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
56,260✔
2770
        QUERY_CHECK_CODE(code, lino, _end);
471✔
2771
      }
2772
    }
2773
    pExp->pExpr->_function.bindExprID = ((SExprNode*)pNode)->bindExprID;
449,291,549✔
2774
  } else if (type == QUERY_NODE_OPERATOR) {
43,766,761✔
2775
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
42,117,963✔
2776
    SOperatorNode* pOpNode = (SOperatorNode*)pNode;
42,109,378✔
2777

2778
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
42,109,378✔
2779
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
42,116,263✔
2780
    pExp->base.numOfParams = 1;
42,090,603✔
2781

2782
    SDataType* pType = &pOpNode->node.resType;
42,115,643✔
2783
    pExp->base.resSchema =
2784
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pOpNode->node.aliasName);
42,113,933✔
2785
    pExp->pExpr->_optrRoot.pRootNode = pNode;
42,112,211✔
2786
  } else if (type == QUERY_NODE_CASE_WHEN) {
1,648,798✔
2787
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
1,529,002✔
2788
    SCaseWhenNode* pCaseNode = (SCaseWhenNode*)pNode;
1,529,002✔
2789

2790
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
1,529,002✔
2791
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
1,529,002✔
2792
    pExp->base.numOfParams = 1;
1,529,002✔
2793

2794
    SDataType* pType = &pCaseNode->node.resType;
1,529,002✔
2795
    pExp->base.resSchema =
2796
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCaseNode->node.aliasName);
1,529,002✔
2797
    pExp->pExpr->_optrRoot.pRootNode = pNode;
1,529,002✔
2798
  } else if (type == QUERY_NODE_LOGIC_CONDITION) {
119,796✔
2799
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
112,077✔
2800
    SLogicConditionNode* pCond = (SLogicConditionNode*)pNode;
112,077✔
2801
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
112,077✔
2802
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
112,077✔
2803
    pExp->base.numOfParams = 1;
112,077✔
2804
    SDataType* pType = &pCond->node.resType;
112,077✔
2805
    pExp->base.resSchema =
2806
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCond->node.aliasName);
112,077✔
2807
    pExp->pExpr->_optrRoot.pRootNode = pNode;
112,077✔
2808
  } else {
2809
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
7,719✔
2810
    QUERY_CHECK_CODE(code, lino, _end);
7,719✔
2811
  }
2812
  pExp->pExpr->relatedTo = ((SExprNode*)pNode)->relatedTo;
1,228,012,842✔
2813
_end:
1,228,575,115✔
2814
  if (code != TSDB_CODE_SUCCESS) {
1,228,575,115✔
2815
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
468,640✔
2816
  }
2817
  return code;
1,228,469,272✔
2818
}
2819

2820
int32_t createExprFromTargetNode(SExprInfo* pExp, STargetNode* pTargetNode) {
1,228,521,463✔
2821
  return createExprFromOneNode(pExp, pTargetNode->pExpr, pTargetNode->slotId);
1,228,521,463✔
2822
}
2823

2824
SExprInfo* createExpr(SNodeList* pNodeList, int32_t* numOfExprs) {
×
2825
  *numOfExprs = LIST_LENGTH(pNodeList);
×
2826
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
×
2827
  if (!pExprs) {
×
2828
    return NULL;
×
2829
  }
2830

2831
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
×
2832
    SExprInfo* pExp = &pExprs[i];
×
2833
    int32_t    code = createExprFromOneNode(pExp, nodesListGetNode(pNodeList, i), i + UD_TAG_COLUMN_INDEX);
×
2834
    if (code != TSDB_CODE_SUCCESS) {
×
2835
      taosMemoryFreeClear(pExprs);
×
2836
      terrno = code;
×
2837
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2838
      return NULL;
×
2839
    }
2840
  }
2841

2842
  return pExprs;
×
2843
}
2844

2845
int32_t createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, SExprInfo** pExprInfo, int32_t* numOfExprs) {
482,677,292✔
2846
  QRY_PARAM_CHECK(pExprInfo);
482,677,292✔
2847

2848
  int32_t code = 0;
482,697,899✔
2849
  int32_t lino = 0;
482,697,899✔
2850
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
482,697,899✔
2851
  int32_t numOfGroupKeys = 0;
482,600,305✔
2852
  if (pGroupKeys != NULL) {
482,600,305✔
2853
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
33,014,560✔
2854
  }
2855

2856
  *numOfExprs = numOfFuncs + numOfGroupKeys;
482,600,321✔
2857
  if (*numOfExprs == 0) {
482,684,531✔
2858
    return code;
42,615,289✔
2859
  }
2860

2861
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
440,059,712✔
2862
  QUERY_CHECK_NULL(pExprs, code, lino, _return, terrno);
439,812,387✔
2863

2864
  int32_t i = 0;
439,812,387✔
2865
  if (pNodeList != NULL) {
439,812,387✔
2866
    SNode* node = NULL;
410,854,476✔
2867
    FOREACH(node, pNodeList) {
1,590,381,884✔
2868
      STargetNode* pTargetNode = (STargetNode*)node;
1,180,100,527✔
2869
      SExprInfo*   pExp = &pExprs[i++];
1,180,100,527✔
2870
      code = createExprFromTargetNode(pExp, pTargetNode);
1,180,025,215✔
2871
      QUERY_CHECK_CODE(code, lino, _return);
1,179,996,048✔
2872
    }
2873
  }
2874

2875
  if (pGroupKeys != NULL) {
439,326,111✔
2876
    SNode* node = NULL;
33,001,096✔
2877
    FOREACH(node, pGroupKeys) {
81,262,339✔
2878
      STargetNode* pTargetNode = (STargetNode*)node;
48,268,530✔
2879
      SExprInfo*   pExp = &pExprs[i++];
48,268,530✔
2880
      code = createExprFromTargetNode(pExp, pTargetNode);
48,260,717✔
2881
      QUERY_CHECK_CODE(code, lino, _return);
48,261,243✔
2882
    }
2883
  }
2884

2885
  *pExprInfo = pExprs;
439,340,275✔
2886
  return code;
439,618,619✔
2887

2888
_return:
468,640✔
2889
  qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
468,640✔
2890
  destroyExprInfo(pExprs, *numOfExprs);
468,640✔
2891
  taosMemoryFreeClear(pExprs);
468,640✔
2892
  return code;
468,640✔
2893
}
2894

2895
static void deleteSubsidiareCtx(void* pData) {
×
2896
  SSubsidiaryResInfo* pCtx = (SSubsidiaryResInfo*)pData;
×
2897
  if (pCtx->pCtx) {
×
2898
    taosMemoryFreeClear(pCtx->pCtx);
×
2899
  }
2900
}
×
2901

2902
// set the output buffer for the selectivity + tag query
2903
static int32_t setSelectValueColumnInfo(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
469,821,461✔
2904
  int32_t num = 0;
469,821,461✔
2905
  int32_t code = TSDB_CODE_SUCCESS;
469,821,461✔
2906
  int32_t lino = 0;
469,821,461✔
2907

2908
  SArray* pValCtxArray = NULL;
469,821,461✔
2909
  for (int32_t i = numOfOutput - 1; i > 0; --i) {  // select Func is at the end of the list
1,250,418,899✔
2910
    int32_t funcIdx = pCtx[i].pExpr->pExpr->_function.bindExprID;
780,711,992✔
2911
    if (funcIdx > 0) {
780,692,268✔
2912
      if (pValCtxArray == NULL) {
1,719,463✔
2913
        // the end of the list is the select function of biggest index
2914
        pValCtxArray = taosArrayInit_s(sizeof(SSubsidiaryResInfo*), funcIdx);
1,233,723✔
2915
        if (pValCtxArray == NULL) {
1,233,255✔
2916
          return terrno;
×
2917
        }
2918
      }
2919
      if (funcIdx > pValCtxArray->size) {
1,718,995✔
2920
        qError("funcIdx:%d is out of range", funcIdx);
×
2921
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2922
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2923
      }
2924
      SSubsidiaryResInfo* pSubsidiary = &pCtx[i].subsidiaries;
1,718,059✔
2925
      pSubsidiary->pCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
1,719,463✔
2926
      if (pSubsidiary->pCtx == NULL) {
1,719,931✔
2927
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2928
        return terrno;
×
2929
      }
2930
      pSubsidiary->num = 0;
1,718,059✔
2931
      taosArraySet(pValCtxArray, funcIdx - 1, &pSubsidiary);
1,718,059✔
2932
    }
2933
  }
2934

2935
  SqlFunctionCtx*  p = NULL;
469,706,907✔
2936
  SqlFunctionCtx** pValCtx = NULL;
469,706,907✔
2937
  if (pValCtxArray == NULL) {
469,706,907✔
2938
    pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
468,512,184✔
2939
    if (pValCtx == NULL) {
468,494,972✔
2940
      QUERY_CHECK_CODE(terrno, lino, _end);
×
2941
    }
2942
  }
2943

2944
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,685,204,076✔
2945
    const char* pName = pCtx[i].pExpr->pExpr->_function.functionName;
1,215,509,282✔
2946
    if ((strcmp(pName, "_select_value") == 0)) {
1,215,490,568✔
2947
      if (pValCtxArray == NULL) {
10,348,842✔
2948
        pValCtx[num++] = &pCtx[i];
7,936,978✔
2949
      } else {
2950
        int32_t bindFuncIndex = pCtx[i].pExpr->pExpr->relatedTo;  // start from index 1;
2,412,047✔
2951
        if (bindFuncIndex > 0) {                                  // 0 is default index related to the select function
2,416,791✔
2952
          bindFuncIndex -= 1;
2,361,099✔
2953
        }
2954
        SSubsidiaryResInfo** pSubsidiary = taosArrayGet(pValCtxArray, bindFuncIndex);
2,416,791✔
2955
        if (pSubsidiary == NULL) {
2,416,323✔
2956
          QUERY_CHECK_CODE(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR, lino, _end);
×
2957
        }
2958
        (*pSubsidiary)->pCtx[(*pSubsidiary)->num] = &pCtx[i];
2,416,323✔
2959
        (*pSubsidiary)->num++;
2,415,855✔
2960
      }
2961
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
1,205,141,726✔
2962
      if (pValCtxArray == NULL) {
98,498,024✔
2963
        p = &pCtx[i];
96,400,714✔
2964
      }
2965
    }
2966
  }
2967

2968
  if (p != NULL) {
469,694,794✔
2969
    p->subsidiaries.pCtx = pValCtx;
52,543,068✔
2970
    p->subsidiaries.num = num;
52,536,863✔
2971
  } else {
2972
    taosMemoryFreeClear(pValCtx);
417,151,726✔
2973
  }
2974

2975
_end:
1,186,923✔
2976
  if (code != TSDB_CODE_SUCCESS) {
469,681,574✔
2977
    taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2978
    taosMemoryFreeClear(pValCtx);
×
2979
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2980
  } else {
2981
    taosArrayDestroy(pValCtxArray);
469,681,574✔
2982
  }
2983
  return code;
469,726,557✔
2984
}
2985

2986
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowEntryInfoOffset,
469,817,852✔
2987
                                     SFunctionStateStore* pStore) {
2988
  int32_t         code = TSDB_CODE_SUCCESS;
469,817,852✔
2989
  int32_t         lino = 0;
469,817,852✔
2990
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
469,817,852✔
2991
  if (pFuncCtx == NULL) {
469,576,136✔
2992
    return NULL;
×
2993
  }
2994

2995
  *rowEntryInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
469,576,136✔
2996
  if (*rowEntryInfoOffset == 0) {
469,777,300✔
2997
    taosMemoryFreeClear(pFuncCtx);
×
2998
    return NULL;
×
2999
  }
3000

3001
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,685,433,558✔
3002
    SExprInfo* pExpr = &pExprInfo[i];
1,215,686,020✔
3003

3004
    SExprBasicInfo* pFunct = &pExpr->base;
1,215,535,447✔
3005
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
1,215,586,116✔
3006

3007
    pCtx->functionId = -1;
1,215,614,482✔
3008
    pCtx->pExpr = pExpr;
1,215,640,453✔
3009

3010
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
1,215,579,993✔
3011
      SFuncExecEnv env = {0};
448,000,361✔
3012
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;
448,039,775✔
3013
      pCtx->isPseudoFunc = fmIsWindowPseudoColumnFunc(pCtx->functionId) || fmIsPlaceHolderFunc(pCtx->functionId);
448,038,412✔
3014
      pCtx->isNotNullFunc = fmIsNotNullOutputFunc(pCtx->functionId);
448,046,319✔
3015

3016
      bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
448,015,461✔
3017
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId) ||
601,048,411✔
3018
          fmIsProcessByRowFunc(pCtx->functionId)) {
447,960,542✔
3019
        if (!isUdaf) {
294,853,876✔
3020
          code = fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
294,814,869✔
3021
          QUERY_CHECK_CODE(code, lino, _end);
294,782,791✔
3022
        } else {
3023
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
39,007✔
3024
          pCtx->udfName = taosStrdup(udfName);
39,007✔
3025
          QUERY_CHECK_NULL(pCtx->udfName, code, lino, _end, terrno);
39,007✔
3026

3027
          code = fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
39,007✔
3028
          QUERY_CHECK_CODE(code, lino, _end);
39,007✔
3029
        }
3030
        bool tmp = pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
294,821,798✔
3031
        if (!tmp) {
294,800,048✔
3032
          code = terrno;
×
3033
          QUERY_CHECK_CODE(code, lino, _end);
4,752✔
3034
        }
3035
      } else {
3036
        if (fmIsPlaceHolderFunc(pCtx->functionId)) {
153,124,628✔
3037
          code = fmGetStreamPesudoFuncEnv(pCtx->functionId, pExpr->base.pParamList, &env);
7,740,615✔
3038
          QUERY_CHECK_CODE(code, lino, _end);
7,740,398✔
3039
        }      
3040
        
3041
        code = fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
153,139,385✔
3042
        if (code != TSDB_CODE_SUCCESS && isUdaf) {
153,144,055✔
3043
          code = TSDB_CODE_SUCCESS;
65,340✔
3044
        }
3045
        QUERY_CHECK_CODE(code, lino, _end);
153,144,055✔
3046

3047
        if (pCtx->sfp.getEnv != NULL) {
153,144,055✔
3048
          bool tmp = pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
30,806,815✔
3049
          if (!tmp) {
30,811,423✔
3050
            code = terrno;
×
3051
            QUERY_CHECK_CODE(code, lino, _end);
×
3052
          }
3053
        }
3054
      }
3055
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
447,941,935✔
3056
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
767,560,637✔
3057
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
19,330,282✔
3058
      // for simple column, the result buffer needs to hold at least one element.
3059
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
767,670,586✔
3060
    }
3061

3062
    pCtx->input.numOfInputCols = pFunct->numOfParams;
1,215,753,974✔
3063
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
1,215,669,020✔
3064
    QUERY_CHECK_NULL(pCtx->input.pData, code, lino, _end, terrno);
1,215,635,098✔
3065
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
1,215,700,902✔
3066
    QUERY_CHECK_NULL(pCtx->input.pColumnDataAgg, code, lino, _end, terrno);
1,215,632,138✔
3067

3068
    pCtx->pTsOutput = NULL;
1,215,695,095✔
3069
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
1,215,738,255✔
3070
    pCtx->resDataInfo.type = pFunct->resSchema.type;
1,215,761,044✔
3071
    pCtx->order = TSDB_ORDER_ASC;
1,215,719,265✔
3072
    pCtx->start.key = INT64_MIN;
1,215,599,676✔
3073
    pCtx->end.key = INT64_MIN;
1,215,740,099✔
3074
    pCtx->numOfParams = pExpr->base.numOfParams;
1,215,760,193✔
3075
    pCtx->param = pFunct->pParam;
1,215,657,783✔
3076
    pCtx->saveHandle.currentPage = -1;
1,215,857,114✔
3077
    pCtx->pStore = pStore;
1,215,652,797✔
3078
    pCtx->hasWindowOrGroup = false;
1,215,726,269✔
3079
    pCtx->needCleanup = false;
1,215,445,866✔
3080
    pCtx->skipDynDataCheck = false;
1,215,629,094✔
3081
  }
3082

3083
  for (int32_t i = 1; i < numOfOutput; ++i) {
1,250,604,617✔
3084
    (*rowEntryInfoOffset)[i] = (int32_t)((*rowEntryInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) +
1,561,571,747✔
3085
                                         pFuncCtx[i - 1].resDataInfo.interBufSize);
780,782,427✔
3086
  }
3087

3088
  code = setSelectValueColumnInfo(pFuncCtx, numOfOutput);
469,803,167✔
3089
  QUERY_CHECK_CODE(code, lino, _end);
469,729,422✔
3090

3091
_end:
469,729,422✔
3092
  if (code != TSDB_CODE_SUCCESS) {
469,669,631✔
3093
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3094
    for (int32_t i = 0; i < numOfOutput; ++i) {
×
3095
      taosMemoryFree(pFuncCtx[i].input.pData);
×
3096
      taosMemoryFree(pFuncCtx[i].input.pColumnDataAgg);
×
3097
    }
3098
    taosMemoryFreeClear(*rowEntryInfoOffset);
×
3099
    taosMemoryFreeClear(pFuncCtx);
×
3100

3101
    terrno = code;
×
3102
    return NULL;
×
3103
  }
3104
  return pFuncCtx;
469,669,631✔
3105
}
3106

3107
// NOTE: sources columns are more than the destination SSDatablock columns.
3108
// doFilter in table scan needs every column even its output is false
3109
int32_t relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray* pCols, bool outputEveryColumn) {
16,122,265✔
3110
  int32_t code = TSDB_CODE_SUCCESS;
16,122,265✔
3111
  size_t  numOfSrcCols = taosArrayGetSize(pCols);
16,122,265✔
3112

3113
  int32_t i = 0, j = 0;
16,122,897✔
3114
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
130,404,989✔
3115
    SColumnInfoData* p = taosArrayGet(pCols, i);
114,282,717✔
3116
    if (!p) {
114,282,595✔
3117
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3118
      return terrno;
×
3119
    }
3120
    SColMatchItem* pmInfo = taosArrayGet(pColMatchInfo, j);
114,282,595✔
3121
    if (!pmInfo) {
114,275,621✔
3122
      return terrno;
×
3123
    }
3124

3125
    if (p->info.colId == pmInfo->colId) {
114,275,621✔
3126
      SColumnInfoData* pDst = taosArrayGet(pBlock->pDataBlock, pmInfo->dstSlotId);
106,154,251✔
3127
      if (!pDst) {
106,153,247✔
3128
        return terrno;
×
3129
      }
3130
      code = colDataAssign(pDst, p, pBlock->info.rows, &pBlock->info);
106,153,247✔
3131
      if (code != TSDB_CODE_SUCCESS) {
106,152,767✔
3132
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3133
        return code;
×
3134
      }
3135
      i++;
106,152,767✔
3136
      j++;
106,152,767✔
3137
    } else if (p->info.colId < pmInfo->colId) {
8,129,505✔
3138
      i++;
8,129,325✔
3139
    } else {
3140
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3141
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3142
    }
3143
  }
3144
  return code;
16,124,191✔
3145
}
3146

3147
SInterval extractIntervalInfo(const STableScanPhysiNode* pTableScanNode) {
259,399,359✔
3148
  SInterval interval = {
518,630,929✔
3149
      .interval = pTableScanNode->interval,
259,233,265✔
3150
      .sliding = pTableScanNode->sliding,
259,216,961✔
3151
      .intervalUnit = pTableScanNode->intervalUnit,
259,431,407✔
3152
      .slidingUnit = pTableScanNode->slidingUnit,
259,314,970✔
3153
      .offset = pTableScanNode->offset,
259,423,093✔
3154
      .precision = pTableScanNode->scan.node.pOutputDataBlockDesc->precision,
259,431,712✔
3155
      .timeRange = pTableScanNode->scanRange,
3156
  };
3157
  calcIntervalAutoOffset(&interval);
259,220,718✔
3158

3159
  return interval;
259,256,287✔
3160
}
3161

3162
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
53,543,656✔
3163
  SColumn c = {0};
53,543,656✔
3164

3165
  c.slotId = pColNode->slotId;
53,543,656✔
3166
  c.colId = pColNode->colId;
53,541,969✔
3167
  c.type = pColNode->node.resType.type;
53,545,292✔
3168
  c.bytes = pColNode->node.resType.bytes;
53,541,304✔
3169
  c.scale = pColNode->node.resType.scale;
53,545,611✔
3170
  c.precision = pColNode->node.resType.precision;
53,548,543✔
3171
  return c;
53,536,618✔
3172
}
3173

3174

3175
/**
3176
 * @brief Determine the actual time range for reading data based on the RANGE clause and the WHERE conditions.
3177
 * @param[in] cond The range specified by WHERE condition.
3178
 * @param[in] range The range specified by RANGE clause.
3179
 * @param[out] twindow The range to be read in DESC order, and only one record is needed.
3180
 * @param[out] extTwindow The external range to read for only one record, which is used for FILL clause.
3181
 * @note `cond` and `twindow` may be the same address.
3182
 */
3183
static int32_t getQueryExtWindow(const STimeWindow* cond, const STimeWindow* range, STimeWindow* twindow,
3,496,995✔
3184
                                 STimeWindow* extTwindows) {
3185
  int32_t     code = TSDB_CODE_SUCCESS;
3,496,995✔
3186
  int32_t     lino = 0;
3,496,995✔
3187
  STimeWindow tempWindow;
3188

3189
  if (cond->skey > cond->ekey || range->skey > range->ekey) {
3,496,995✔
3190
    *twindow = extTwindows[0] = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
5,335✔
3191
    return code;
5,335✔
3192
  }
3193

3194
  if (range->ekey < cond->skey) {
3,491,660✔
3195
    extTwindows[1] = *cond;
541,983✔
3196
    *twindow = extTwindows[0] = TSWINDOW_DESC_INITIALIZER;
541,983✔
3197
    return code;
541,983✔
3198
  }
3199

3200
  if (cond->ekey < range->skey) {
2,949,677✔
3201
    extTwindows[0] = *cond;
390,071✔
3202
    *twindow = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
390,071✔
3203
    return code;
390,071✔
3204
  }
3205

3206
  // Only scan data in the time range intersecion.
3207
  extTwindows[0] = extTwindows[1] = *cond;
2,559,606✔
3208
  twindow->skey = TMAX(cond->skey, range->skey);
2,559,606✔
3209
  twindow->ekey = TMIN(cond->ekey, range->ekey);
2,559,606✔
3210
  extTwindows[0].ekey = twindow->skey - 1;
2,559,606✔
3211
  extTwindows[1].skey = twindow->ekey + 1;
2,559,606✔
3212

3213
  return code;
2,559,606✔
3214
}
3215

3216
static int32_t getPrimaryTimeRange(SNode** pPrimaryKeyCond, STimeWindow* pTimeRange, bool* isStrict) {
675,228✔
3217
  SNode*  pNew = NULL;
675,228✔
3218
  int32_t code = scalarCalculateRemoteConstants(*pPrimaryKeyCond, &pNew);
675,228✔
3219
  if (TSDB_CODE_SUCCESS == code) {
675,228✔
3220
    *pPrimaryKeyCond = pNew;
675,228✔
3221
    if (nodeType(pNew) != QUERY_NODE_VALUE) {
675,228✔
3222
      code = filterGetTimeRange(*pPrimaryKeyCond, pTimeRange, isStrict, NULL);
675,228✔
3223
    }
3224
  }
3225
  return code;
675,228✔
3226
}
3227

3228
int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, STableScanPhysiNode* pTableScanNode,
290,228,453✔
3229
                               const SReadHandle* readHandle, bool applyExtWin) {
3230
  int32_t code = 0;                             
290,228,453✔
3231
  pCond->order = pTableScanNode->scanSeq[0] > 0 ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
290,228,453✔
3232
  pCond->numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
290,201,190✔
3233

3234
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
290,267,408✔
3235
  if (!pCond->colList) {
290,173,299✔
3236
    return terrno;
×
3237
  }
3238
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
290,107,090✔
3239
  if (pCond->pSlotList == NULL) {
290,211,780✔
3240
    taosMemoryFreeClear(pCond->colList);
×
3241
    return terrno;
×
3242
  }
3243

3244
  // TODO: get it from stable scan node
3245
  pCond->twindows = pTableScanNode->scanRange;
289,995,613✔
3246
  pCond->suid = pTableScanNode->scan.suid;
290,233,393✔
3247
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
290,082,511✔
3248
  pCond->startVersion = -1;
290,203,199✔
3249
  pCond->endVersion = -1;
290,221,206✔
3250
  pCond->skipRollup = readHandle->skipRollup;
290,103,748✔
3251
  if (readHandle->winRangeValid) {
290,161,286✔
3252
    pCond->twindows = readHandle->winRange;
533,035✔
3253
  }
3254
  pCond->cacheSttStatis = readHandle->cacheSttStatis;
290,207,881✔
3255
  // allowed read stt file optimization mode
3256
  pCond->notLoadData = (pTableScanNode->dataRequired == FUNC_DATA_REQUIRED_NOT_LOAD) &&
580,404,265✔
3257
                       (pTableScanNode->scan.node.pConditions == NULL) && (pTableScanNode->interval == 0);
290,198,874✔
3258

3259
  int32_t j = 0;
290,274,317✔
3260
  SNode*  node = NULL;
290,274,317✔
3261
  FOREACH(node, pTableScanNode->scan.pScanCols) {
1,354,369,118✔
3262
    STargetNode* pNode = (STargetNode*)node;
1,063,691,879✔
3263
    if (!pNode) {
1,063,691,879✔
3264
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3265
      return terrno;
×
3266
    }
3267
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
1,063,691,879✔
3268
    if (pColNode->colType == COLUMN_TYPE_TAG) {
1,064,060,364✔
3269
      continue;
×
3270
    }
3271

3272
    pCond->colList[j].type = pColNode->node.resType.type;
1,064,029,121✔
3273
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
1,064,055,342✔
3274
    pCond->colList[j].colId = pColNode->colId;
1,064,121,664✔
3275
    pCond->colList[j].pk = pColNode->isPk;
1,064,265,465✔
3276

3277
    pCond->pSlotList[j] = pNode->slotId;
1,063,957,189✔
3278
    j += 1;
1,064,094,801✔
3279
  }
3280

3281
  pCond->numOfCols = j;
290,233,260✔
3282

3283
  if (applyExtWin) {
290,242,372✔
3284
    if (NULL != pTableScanNode->pExtScanRange) {
260,026,962✔
3285
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
3,421,238✔
3286
      code = getQueryExtWindow(&pCond->twindows, pTableScanNode->pExtScanRange, &pCond->twindows, pCond->extTwindows);
3,421,238✔
3287
    } else if (readHandle->extWinRangeValid) {
256,534,724✔
3288
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
75,757✔
3289
      code = getQueryExtWindow(&pCond->twindows, &readHandle->extWinRange, &pCond->twindows, pCond->extTwindows);
75,757✔
3290
    }
3291
  }
3292

3293
  if (pTableScanNode->pPrimaryCond) {
290,220,660✔
3294
    bool isStrict = false;
675,228✔
3295
    code = getPrimaryTimeRange((SNode**)&pTableScanNode->pPrimaryCond, &pCond->twindows, &isStrict);
675,228✔
3296
    if (code || !isStrict) {
675,228✔
3297
      code = nodesMergeNode((SNode**)&pTableScanNode->scan.node.pConditions, &pTableScanNode->pPrimaryCond);
662,532✔
3298
    }
3299
  }
3300

3301
  return code;
290,193,037✔
3302
}
3303

3304
int32_t initQueryTableDataCondWithColArray(SQueryTableDataCond* pCond, SQueryTableDataCond* pOrgCond,
117,045,413✔
3305
                                           const SReadHandle* readHandle, SArray* colArray) {
3306
  int32_t code = TSDB_CODE_SUCCESS;
117,045,413✔
3307
  int32_t lino = 0;
117,045,413✔
3308

3309
  pCond->order = TSDB_ORDER_ASC;
117,045,413✔
3310
  pCond->numOfCols = (int32_t)taosArrayGetSize(colArray);
117,094,803✔
3311

3312
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
117,104,658✔
3313
  QUERY_CHECK_NULL(pCond->colList, code, lino, _return, terrno);
117,067,582✔
3314

3315
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
117,081,357✔
3316
  QUERY_CHECK_NULL(pCond->pSlotList, code, lino, _return, terrno);
117,082,460✔
3317

3318
  pCond->twindows = pOrgCond->twindows;
117,088,314✔
3319
  pCond->order = pOrgCond->order;
117,098,036✔
3320
  pCond->type = pOrgCond->type;
117,109,786✔
3321
  pCond->startVersion = -1;
117,103,864✔
3322
  pCond->endVersion = -1;
117,105,130✔
3323
  pCond->skipRollup = true;
117,095,959✔
3324
  pCond->notLoadData = false;
117,103,406✔
3325

3326
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
480,597,846✔
3327
    SColIdPair* pColPair = taosArrayGet(colArray, i);
363,512,321✔
3328
    QUERY_CHECK_NULL(pColPair, code, lino, _return, terrno);
363,529,829✔
3329

3330
    bool find = false;
363,527,841✔
3331
    for (int32_t j = 0; j < pOrgCond->numOfCols; ++j) {
1,860,431,632✔
3332
      if (pOrgCond->colList[j].colId == pColPair->vtbColId) {
1,860,212,761✔
3333
        pCond->colList[i].type = pOrgCond->colList[j].type;
363,514,630✔
3334
        // For variable-length types (nchar/binary/varchar/varbinary), use the source table's bytes
3335
        // to avoid tsdb reader doCopyColVal length check failure when source data is longer
3336
        // than the virtual table's defined length.
3337
        if (IS_VAR_DATA_TYPE(pOrgCond->colList[j].type) && pColPair->type.bytes > 0) {
363,532,889✔
3338
          pCond->colList[i].bytes = TMAX(pOrgCond->colList[j].bytes, pColPair->type.bytes);
42,356,489✔
3339
        } else {
3340
          pCond->colList[i].bytes = pOrgCond->colList[j].bytes;
321,186,921✔
3341
        }
3342
        pCond->colList[i].colId = pColPair->orgColId;
363,542,372✔
3343
        pCond->colList[i].pk = pOrgCond->colList[j].pk;
363,542,210✔
3344
        pCond->pSlotList[i] = i;
363,539,675✔
3345
        find = true;
363,525,070✔
3346
        qDebug("%s mapped vtb colId:%d to org colId:%d, type:%d, bytes:%d", __func__, pColPair->vtbColId,
363,525,070✔
3347
               pColPair->orgColId, pCond->colList[i].type, pCond->colList[i].bytes);
3348
        break;
363,480,258✔
3349
      }
3350
    }
3351
    QUERY_CHECK_CONDITION(find, code, lino, _return, TSDB_CODE_NOT_FOUND);
363,488,036✔
3352
  }
3353

3354
  return code;
117,116,238✔
3355
_return:
×
3356
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(terrno));
×
3357
  taosMemoryFreeClear(pCond->colList);
×
3358
  taosMemoryFreeClear(pCond->pSlotList);
×
3359
  return code;
×
3360
}
3361

3362
void cleanupQueryTableDataCond(SQueryTableDataCond* pCond) {
720,296,984✔
3363
  taosMemoryFreeClear(pCond->colList);
720,296,984✔
3364
  taosMemoryFreeClear(pCond->pSlotList);
720,331,309✔
3365
}
720,278,725✔
3366

3367
int32_t convertFillType(int32_t mode) {
4,273,014✔
3368
  int32_t type = TSDB_FILL_NONE;
4,273,014✔
3369
  switch (mode) {
4,273,014✔
3370
    case FILL_MODE_PREV:
207,059✔
3371
      type = TSDB_FILL_PREV;
207,059✔
3372
      break;
207,059✔
3373
    case FILL_MODE_NONE:
×
3374
      type = TSDB_FILL_NONE;
×
3375
      break;
×
3376
    case FILL_MODE_NULL:
263,226✔
3377
      type = TSDB_FILL_NULL;
263,226✔
3378
      break;
263,226✔
3379
    case FILL_MODE_NULL_F:
51,513✔
3380
      type = TSDB_FILL_NULL_F;
51,513✔
3381
      break;
51,513✔
3382
    case FILL_MODE_NEXT:
206,615✔
3383
      type = TSDB_FILL_NEXT;
206,615✔
3384
      break;
206,615✔
3385
    case FILL_MODE_VALUE:
191,295✔
3386
      type = TSDB_FILL_SET_VALUE;
191,295✔
3387
      break;
191,295✔
3388
    case FILL_MODE_VALUE_F:
20,208✔
3389
      type = TSDB_FILL_SET_VALUE_F;
20,208✔
3390
      break;
20,208✔
3391
    case FILL_MODE_LINEAR:
256,727✔
3392
      type = TSDB_FILL_LINEAR;
256,727✔
3393
      break;
256,727✔
3394
    case FILL_MODE_NEAR:
3,076,371✔
3395
      type = TSDB_FILL_NEAR;
3,076,371✔
3396
      break;
3,076,371✔
3397
    default:
×
3398
      type = TSDB_FILL_NONE;
×
3399
  }
3400

3401
  return type;
4,273,014✔
3402
}
3403

3404
void getInitialStartTimeWindow(SInterval* pInterval, TSKEY ts, STimeWindow* w, bool ascQuery) {
2,147,483,647✔
3405
  if (ascQuery) {
2,147,483,647✔
3406
    *w = getAlignQueryTimeWindow(pInterval, ts);
2,147,483,647✔
3407
  } else {
3408
    // the start position of the first time window in the endpoint that spreads beyond the queried last timestamp
3409
    *w = getAlignQueryTimeWindow(pInterval, ts);
36,296✔
3410

3411
    int64_t key = w->skey;
358,171✔
3412
    while (key < ts) {  // moving towards end
376,121✔
3413
      key = getNextTimeWindowStart(pInterval, key, TSDB_ORDER_ASC);
192,903✔
3414
      if (key > ts) {
192,903✔
3415
        break;
174,953✔
3416
      }
3417

3418
      w->skey = key;
17,950✔
3419
    }
3420
    w->ekey = taosTimeAdd(w->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
358,171✔
3421
  }
3422
}
2,147,483,647✔
3423

3424
static STimeWindow doCalculateTimeWindow(int64_t ts, SInterval* pInterval) {
78,353,486✔
3425
  STimeWindow w = {0};
78,353,486✔
3426

3427
  w.skey = taosTimeTruncate(ts, pInterval);
78,353,486✔
3428
  w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
78,350,474✔
3429
  return w;
78,349,972✔
3430
}
3431

3432
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order) {
1,628,091✔
3433
  STimeWindow win = *pWindow;
1,628,091✔
3434
  STimeWindow save = win;
1,628,091✔
3435
  while (win.skey <= ts && win.ekey >= ts) {
10,456,888✔
3436
    save = win;
8,828,797✔
3437
    // get previous time window
3438
    getNextTimeWindow(pInterval, &win, order == TSDB_ORDER_DESC ? TSDB_ORDER_ASC : TSDB_ORDER_DESC);
8,828,797✔
3439
  }
3440

3441
  return save;
1,628,091✔
3442
}
3443

3444
// get the correct time window according to the handled timestamp
3445
// todo refactor
3446
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
121,278,179✔
3447
                                int32_t order) {
3448
  STimeWindow w = {0};
121,278,179✔
3449
  if (pResultRowInfo->cur.pageId == -1) {  // the first window, from the previous stored value
121,279,978✔
3450
    getInitialStartTimeWindow(pInterval, ts, &w, (order != TSDB_ORDER_DESC));
3,951,893✔
3451
    return w;
3,956,690✔
3452
  }
3453

3454
  SResultRow* pRow = getResultRowByPos(pBuf, &pResultRowInfo->cur, false);
117,322,147✔
3455
  if (pRow) {
117,318,982✔
3456
    TAOS_SET_OBJ_ALIGNED(&w, pRow->win);
117,318,481✔
3457
  }
3458

3459
  // in case of typical time window, we can calculate time window directly.
3460
  if (w.skey > ts || w.ekey < ts) {
117,322,556✔
3461
    w = doCalculateTimeWindow(ts, pInterval);
78,353,987✔
3462
  }
3463

3464
  if (pInterval->interval != pInterval->sliding) {
117,318,541✔
3465
    // it is an sliding window query, in which sliding value is not equalled to
3466
    // interval value, and we need to find the first qualified time window.
3467
    w = getFirstQualifiedTimeWindow(ts, &w, pInterval, order);
1,627,508✔
3468
  }
3469

3470
  return w;
117,319,074✔
3471
}
3472

3473
void getNextTimeWindow(const SInterval* pInterval, STimeWindow* tw, int32_t order) {
2,147,483,647✔
3474
  tw->skey = getNextTimeWindowStart(pInterval, tw->skey, order);
2,147,483,647✔
3475
  tw->ekey = taosTimeGetIntervalEnd(tw->skey, pInterval);
2,147,483,647✔
3476
}
2,147,483,647✔
3477

3478
bool hasLimitOffsetInfo(SLimitInfo* pLimitInfo) {
553,837,764✔
3479
  return (pLimitInfo->limit.limit != -1 || pLimitInfo->limit.offset != -1 || pLimitInfo->slimit.limit != -1 ||
1,102,858,893✔
3480
          pLimitInfo->slimit.offset != -1);
549,020,549✔
3481
}
3482

3483
bool hasSlimitOffsetInfo(SLimitInfo* pLimitInfo) {
×
3484
  return (pLimitInfo->slimit.limit != -1 || pLimitInfo->slimit.offset != -1);
×
3485
}
3486

3487
void initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo) {
701,077,713✔
3488
  SLimit limit = {.limit = getLimit(pLimit), .offset = getOffset(pLimit)};
701,077,713✔
3489
  SLimit slimit = {.limit = getLimit(pSLimit), .offset = getOffset(pSLimit)};
700,924,576✔
3490

3491
  pLimitInfo->limit = limit;
700,887,592✔
3492
  pLimitInfo->slimit = slimit;
700,914,744✔
3493
  pLimitInfo->remainOffset = limit.offset;
700,933,202✔
3494
  pLimitInfo->remainGroupOffset = slimit.offset;
700,905,560✔
3495
  pLimitInfo->numOfOutputRows = 0;
700,966,034✔
3496
  pLimitInfo->numOfOutputGroups = 0;
701,107,654✔
3497
  pLimitInfo->currentGroupId = 0;
701,044,363✔
3498
}
700,981,709✔
3499

3500
void resetLimitInfoForNextGroup(SLimitInfo* pLimitInfo) {
88,259,472✔
3501
  pLimitInfo->numOfOutputRows = 0;
88,259,472✔
3502
  pLimitInfo->remainOffset = pLimitInfo->limit.offset;
88,278,497✔
3503
}
88,219,674✔
3504

3505
int32_t tableListGetSize(const STableListInfo* pTableList, int32_t* pRes) {
733,729,125✔
3506
  if (taosArrayGetSize(pTableList->pTableList) != taosHashGetSize(pTableList->map)) {
733,729,125✔
3507
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3508
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3509
  }
3510
  (*pRes) = taosArrayGetSize(pTableList->pTableList);
733,677,955✔
3511
  return TSDB_CODE_SUCCESS;
733,686,441✔
3512
}
3513

3514
uint64_t tableListGetSuid(const STableListInfo* pTableList) { return pTableList->idInfo.suid; }
4,709,695✔
3515

3516
STableKeyInfo* tableListGetInfo(const STableListInfo* pTableList, int32_t index) {
225,340,243✔
3517
  if (taosArrayGetSize(pTableList->pTableList) == 0) {
225,340,243✔
3518
    return NULL;
2,367✔
3519
  }
3520

3521
  return taosArrayGet(pTableList->pTableList, index);
225,270,911✔
3522
}
3523

3524
int32_t tableListFind(const STableListInfo* pTableList, uint64_t uid, int32_t startIndex) {
51,418✔
3525
  int32_t numOfTables = taosArrayGetSize(pTableList->pTableList);
51,418✔
3526
  if (startIndex >= numOfTables) {
51,418✔
3527
    return -1;
×
3528
  }
3529

3530
  for (int32_t i = startIndex; i < numOfTables; ++i) {
419,948✔
3531
    STableKeyInfo* p = taosArrayGet(pTableList->pTableList, i);
419,948✔
3532
    if (!p) {
419,948✔
3533
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3534
      return -1;
×
3535
    }
3536
    if (p->uid == uid) {
419,948✔
3537
      return i;
51,418✔
3538
    }
3539
  }
3540
  return -1;
×
3541
}
3542

3543
void tableListGetSourceTableInfo(const STableListInfo* pTableList, uint64_t* psuid, uint64_t* uid, int32_t* type) {
466,837✔
3544
  *psuid = pTableList->idInfo.suid;
466,837✔
3545
  *uid = pTableList->idInfo.uid;
466,837✔
3546
  *type = pTableList->idInfo.tableType;
466,837✔
3547
}
466,837✔
3548

3549
int32_t tableListGetTableGroupId(const STableListInfo* pTableList, uint64_t tableUid, uint64_t* gid, uint64_t* baseGid) {
892,626,782✔
3550
  int32_t* slot = taosHashGet(pTableList->map, &tableUid, sizeof(tableUid));
892,626,782✔
3551
  if (slot == NULL) {
892,773,362✔
3552
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3553
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
3554
  }
3555

3556
  STableKeyInfo* pKeyInfo = taosArrayGet(pTableList->pTableList, *slot);
892,773,362✔
3557
  if (pKeyInfo == NULL) {
892,961,160✔
3558
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3559
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
3560
  }
3561

3562
  *gid = pKeyInfo->groupId;
892,961,173✔
3563
  *baseGid = pKeyInfo->baseGId;
892,973,019✔
3564
  
3565
  return 0;
892,939,761✔
3566
}
3567

3568
// TODO handle the group offset info, fix it, the rule of group output will be broken by this function
3569
// int32_t tableListRemoveTableInfo(STableListInfo* pTableList, uint64_t uid) {
3570
//   int32_t code = TSDB_CODE_SUCCESS;
3571
//   int32_t lino = 0;
3572

3573
//   int32_t* slot = taosHashGet(pTableList->map, &uid, sizeof(uid));
3574
//   if (slot == NULL) {
3575
//     qDebug("table:%" PRIu64 " not found in table list", uid);
3576
//     return 0;
3577
//   }
3578

3579
//   taosArrayRemove(pTableList->pTableList, *slot);
3580
//   code = taosHashRemove(pTableList->map, &uid, sizeof(uid));
3581

3582
//   _end:
3583
//   if (code != TSDB_CODE_SUCCESS) {
3584
//     qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
3585
//   } else {
3586
//     qDebug("uid:%" PRIu64 ", remove from table list", uid);
3587
//   }
3588

3589
//   return code;
3590
// }
3591

3592
int32_t tableListAddTableInfo(STableListInfo* pTableList, uint64_t uid, uint64_t gid) {
1,184,253✔
3593
  int32_t code = TSDB_CODE_SUCCESS;
1,184,253✔
3594
  int32_t lino = 0;
1,184,253✔
3595
  if (pTableList->map == NULL) {
1,184,253✔
3596
    pTableList->map = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
×
3597
    QUERY_CHECK_NULL(pTableList->map, code, lino, _end, terrno);
×
3598
  }
3599

3600
  STableKeyInfo keyInfo = {.uid = uid, .groupId = gid};
1,186,612✔
3601
  void*         p = taosHashGet(pTableList->map, &uid, sizeof(uid));
1,186,275✔
3602
  if (p != NULL) {
1,184,580✔
3603
    qInfo("table:%" PRId64 " already in tableIdList, ignore it", uid);
×
3604
    goto _end;
×
3605
  }
3606

3607
  void* tmp = taosArrayPush(pTableList->pTableList, &keyInfo);
1,184,580✔
3608
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,187,613✔
3609

3610
  int32_t slot = (int32_t)taosArrayGetSize(pTableList->pTableList) - 1;
1,187,613✔
3611
  code = taosHashPut(pTableList->map, &uid, sizeof(uid), &slot, sizeof(slot));
1,187,960✔
3612
  if (code != TSDB_CODE_SUCCESS) {
1,187,960✔
3613
    // we have checked the existence of uid in hash map above
3614
    QUERY_CHECK_CONDITION((code != TSDB_CODE_DUP_KEY), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
3615
    taosArrayPopTailBatch(pTableList->pTableList, 1);  // let's pop the last element in the array list
×
3616
  }
3617

3618
_end:
1,187,960✔
3619
  if (code != TSDB_CODE_SUCCESS) {
1,185,601✔
3620
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3621
  } else {
3622
    qDebug("uid:%" PRIu64 ", groupId:%" PRIu64 " added into table list, slot:%d, total:%d", uid, gid, slot, slot + 1);
1,185,601✔
3623
  }
3624

3625
  return code;
1,186,949✔
3626
}
3627

3628
int32_t tableListGetGroupList(const STableListInfo* pTableList, int32_t ordinalGroupIndex, STableKeyInfo** pKeyInfo,
278,718,666✔
3629
                              int32_t* size) {
3630
  int32_t totalGroups = tableListGetOutputGroups(pTableList);
278,718,666✔
3631
  int32_t numOfTables = 0;
278,716,419✔
3632
  int32_t code = tableListGetSize(pTableList, &numOfTables);
278,730,718✔
3633
  if (code != TSDB_CODE_SUCCESS) {
278,705,945✔
3634
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3635
    return code;
×
3636
  }
3637

3638
  if (ordinalGroupIndex < 0 || ordinalGroupIndex >= totalGroups) {
278,705,945✔
3639
    return TSDB_CODE_INVALID_PARA;
×
3640
  }
3641

3642
  // here handle two special cases:
3643
  // 1. only one group exists, and 2. one table exists for each group.
3644
  if (totalGroups == 1) {
278,705,945✔
3645
    *size = numOfTables;
278,111,295✔
3646
    *pKeyInfo = (*size == 0) ? NULL : taosArrayGet(pTableList->pTableList, 0);
278,156,985✔
3647
    return TSDB_CODE_SUCCESS;
278,050,979✔
3648
  } else if (totalGroups == numOfTables) {
594,650✔
3649
    *size = 1;
515,594✔
3650
    *pKeyInfo = taosArrayGet(pTableList->pTableList, ordinalGroupIndex);
515,594✔
3651
    return TSDB_CODE_SUCCESS;
515,189✔
3652
  }
3653

3654
  int32_t offset = pTableList->groupOffset[ordinalGroupIndex];
79,056✔
3655
  if (ordinalGroupIndex < totalGroups - 1) {
63,171✔
3656
    *size = pTableList->groupOffset[ordinalGroupIndex + 1] - offset;
47,515✔
3657
  } else {
3658
    *size = numOfTables - offset;
15,656✔
3659
  }
3660

3661
  *pKeyInfo = taosArrayGet(pTableList->pTableList, offset);
63,171✔
3662
  return TSDB_CODE_SUCCESS;
63,171✔
3663
}
3664

3665
int32_t tableListGetOutputGroups(const STableListInfo* pTableList) { return pTableList->numOfOuputGroups; }
780,047,471✔
3666

3667
bool oneTableForEachGroup(const STableListInfo* pTableList) { return pTableList->oneTableForEachGroup; }
1,135,444✔
3668

3669
STableListInfo* tableListCreate() {
313,376,472✔
3670
  STableListInfo* pListInfo = taosMemoryCalloc(1, sizeof(STableListInfo));
313,376,472✔
3671
  if (pListInfo == NULL) {
313,290,230✔
3672
    return NULL;
×
3673
  }
3674

3675
  pListInfo->remainGroups = NULL;
313,290,230✔
3676
  pListInfo->pTableList = taosArrayInit(4, sizeof(STableKeyInfo));
313,316,793✔
3677
  if (pListInfo->pTableList == NULL) {
313,276,888✔
3678
    goto _error;
×
3679
  }
3680

3681
  pListInfo->map = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
313,324,635✔
3682
  if (pListInfo->map == NULL) {
313,485,071✔
3683
    goto _error;
×
3684
  }
3685

3686
  pListInfo->numOfOuputGroups = 1;
313,477,386✔
3687
  return pListInfo;
313,481,814✔
3688

3689
_error:
×
3690
  tableListDestroy(pListInfo);
×
3691
  return NULL;
×
3692
}
3693

3694
void tableListDestroy(STableListInfo* pTableListInfo) {
325,863,087✔
3695
  if (pTableListInfo == NULL) {
325,863,087✔
3696
    return;
12,455,615✔
3697
  }
3698

3699
  taosArrayDestroy(pTableListInfo->pTableList);
313,407,472✔
3700
  taosMemoryFreeClear(pTableListInfo->groupOffset);
313,357,363✔
3701

3702
  taosHashCleanup(pTableListInfo->map);
313,384,410✔
3703
  taosHashCleanup(pTableListInfo->remainGroups);
313,445,109✔
3704
  pTableListInfo->pTableList = NULL;
313,453,133✔
3705
  pTableListInfo->map = NULL;
313,450,236✔
3706
  taosMemoryFree(pTableListInfo);
313,442,799✔
3707
}
3708

3709
void tableListClear(STableListInfo* pTableListInfo) {
879,725✔
3710
  if (pTableListInfo == NULL) {
879,725✔
3711
    return;
×
3712
  }
3713

3714
  taosArrayClear(pTableListInfo->pTableList);
879,725✔
3715
  taosHashClear(pTableListInfo->map);
880,726✔
3716
  taosHashClear(pTableListInfo->remainGroups);
881,073✔
3717
  taosMemoryFree(pTableListInfo->groupOffset);
880,726✔
3718
  pTableListInfo->numOfOuputGroups = 1;
881,073✔
3719
  pTableListInfo->oneTableForEachGroup = false;
881,073✔
3720
}
3721

3722
static int32_t orderbyGroupIdComparFn(const void* p1, const void* p2) {
620,845,460✔
3723
  STableKeyInfo* pInfo1 = (STableKeyInfo*)p1;
620,845,460✔
3724
  STableKeyInfo* pInfo2 = (STableKeyInfo*)p2;
620,845,460✔
3725

3726
  if (pInfo1->groupId == pInfo2->groupId) {
620,845,460✔
3727
    return 0;
582,572,012✔
3728
  } else {
3729
    return pInfo1->groupId < pInfo2->groupId ? -1 : 1;
38,288,648✔
3730
  }
3731
}
3732

3733
int32_t sortTableGroup(STableListInfo* pTableListInfo) {
30,783,920✔
3734
  int32_t code = TSDB_CODE_SUCCESS;
30,783,920✔
3735
  taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
30,783,920✔
3736
  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
30,801,435✔
3737
  if (size == 0) {
30,792,128✔
3738
    pTableListInfo->numOfOuputGroups = 0;
×
3739
    return code;
×
3740
  }
3741

3742
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
30,792,128✔
3743
  if (!pList) {
30,797,443✔
3744
    code = terrno;
×
3745
    goto end;
×
3746
  }
3747

3748
  STableKeyInfo* pInfo = taosArrayGet(pTableListInfo->pTableList, 0);
30,797,443✔
3749
  if (pInfo == NULL) {
30,767,424✔
3750
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3751
    code = terrno;
×
3752
    goto end;
×
3753
  }
3754
  uint64_t gid = pInfo->groupId;
30,767,424✔
3755

3756
  int32_t start = 0;
30,784,090✔
3757
  void*   tmp = taosArrayPush(pList, &start);
30,791,818✔
3758
  if (tmp == NULL) {
30,791,818✔
3759
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3760
    code = terrno;
×
3761
    goto end;
×
3762
  }
3763

3764
  for (int32_t i = 1; i < size; ++i) {
169,147,994✔
3765
    pInfo = taosArrayGet(pTableListInfo->pTableList, i);
138,361,735✔
3766
    if (pInfo == NULL) {
138,361,201✔
3767
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3768
      code = terrno;
×
3769
      goto end;
×
3770
    }
3771
    if (pInfo->groupId != gid) {
138,361,201✔
3772
      tmp = taosArrayPush(pList, &i);
7,856,097✔
3773
      if (tmp == NULL) {
7,856,097✔
3774
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3775
        code = terrno;
×
3776
        goto end;
×
3777
      }
3778
      gid = pInfo->groupId;
7,856,097✔
3779
    }
3780
  }
3781

3782
  pTableListInfo->numOfOuputGroups = taosArrayGetSize(pList);
30,797,479✔
3783
  pTableListInfo->groupOffset = taosMemoryMalloc(sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
30,791,671✔
3784
  if (pTableListInfo->groupOffset == NULL) {
30,788,668✔
3785
    code = terrno;
×
3786
    goto end;
×
3787
  }
3788

3789
  memcpy(pTableListInfo->groupOffset, taosArrayGet(pList, 0), sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
30,762,112✔
3790

3791
end:
30,760,606✔
3792
  taosArrayDestroy(pList);
30,769,521✔
3793
  return code;
30,747,799✔
3794
}
3795

3796
int32_t buildGroupIdMapForAllTables(STableListInfo* pTableListInfo, SReadHandle* pHandle, SScanPhysiNode* pScanNode,
264,665,927✔
3797
                                    SNodeList* group, bool groupSort, uint8_t* digest, SStorageAPI* pAPI, SHashObj* groupIdMap, bool gIdFromBaseId) {
3798
  int32_t code = TSDB_CODE_SUCCESS;
264,665,927✔
3799

3800
  bool   groupByTbname = groupbyTbname(group);
264,665,927✔
3801
  size_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
264,602,075✔
3802
  if (!numOfTables) {
264,580,163✔
3803
    return code;
×
3804
  }
3805
  qDebug("numOfTables:%zu, groupByTbname:%d, group:%p", numOfTables, groupByTbname, group);
264,580,163✔
3806
  if (group == NULL || groupByTbname) {
264,620,854✔
3807
    if (tsCountAlwaysReturnValue && QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode) &&
259,829,324✔
3808
        ((STableScanPhysiNode*)pScanNode)->needCountEmptyTable) {
216,955,539✔
3809
      pTableListInfo->remainGroups =
13,971,704✔
3810
          taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
13,972,274✔
3811
      if (pTableListInfo->remainGroups == NULL) {
13,972,274✔
3812
        return terrno;
×
3813
      }
3814

3815
      for (int i = 0; i < numOfTables; i++) {
74,535,614✔
3816
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
60,561,630✔
3817
        if (!info) {
60,560,037✔
3818
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3819
          return terrno;
×
3820
        }
3821
        info->groupId = groupByTbname ? info->uid : 0;
60,560,037✔
3822
        int32_t tempRes = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId),
60,564,597✔
3823
                                      &(info->uid), sizeof(info->uid));
60,564,027✔
3824
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
60,563,340✔
3825
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3826
          return tempRes;
×
3827
        }
3828
      }
3829
    } else {
3830
      for (int32_t i = 0; i < numOfTables; i++) {
889,575,285✔
3831
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
643,768,684✔
3832
        if (!info) {
643,781,264✔
3833
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3834
          return terrno;
×
3835
        }
3836
        info->groupId = groupByTbname ? info->uid : 0;
643,781,264✔
3837
        
3838
      }
3839
    }
3840
    if (groupIdMap && group != NULL){
259,780,585✔
3841
      getColInfoResultForGroupbyForStream(pHandle->vnode, group, pTableListInfo, pAPI, groupIdMap);
95,675✔
3842
    }
3843

3844
    pTableListInfo->oneTableForEachGroup = groupByTbname;
259,780,151✔
3845
    if (numOfTables == 1 && pTableListInfo->idInfo.tableType == TSDB_CHILD_TABLE) {
259,826,158✔
3846
      pTableListInfo->oneTableForEachGroup = true;
110,723,349✔
3847
    }
3848

3849
    if (groupSort && groupByTbname) {
259,831,486✔
3850
      taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
2,572,429✔
3851
      pTableListInfo->numOfOuputGroups = numOfTables;
2,572,993✔
3852
    } else if (groupByTbname && pScanNode->groupOrderScan) {
257,259,057✔
3853
      pTableListInfo->numOfOuputGroups = numOfTables;
31,166✔
3854
    } else {
3855
      pTableListInfo->numOfOuputGroups = 1;
257,227,716✔
3856
    }
3857
    if (groupSort || pScanNode->groupOrderScan) {
259,881,577✔
3858
      code = sortTableGroup(pTableListInfo);
30,674,125✔
3859
    }
3860
  } else {
3861
    bool initRemainGroups = false;
4,791,530✔
3862
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode)) {
4,791,530✔
3863
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pScanNode;
4,664,265✔
3864
      if (tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable &&
4,664,265✔
3865
          !(groupSort || pScanNode->groupOrderScan)) {
2,214,511✔
3866
        initRemainGroups = true;
2,188,235✔
3867
      }
3868
    }
3869

3870
    code = getColInfoResultForGroupby(pHandle, group, pTableListInfo, digest, pAPI, initRemainGroups, groupIdMap, gIdFromBaseId);
4,791,530✔
3871
    if (code != TSDB_CODE_SUCCESS) {
4,790,988✔
3872
      return code;
×
3873
    }
3874

3875
    if (pScanNode->groupOrderScan) pTableListInfo->numOfOuputGroups = taosArrayGetSize(pTableListInfo->pTableList);
4,790,988✔
3876

3877
    if (groupSort || pScanNode->groupOrderScan) {
4,791,125✔
3878
      code = sortTableGroup(pTableListInfo);
153,947✔
3879
    }
3880
  }
3881

3882
  qTrace("EXEC_GROUP_TRACE build_group_map done scanType:%d group:%p groupByTbname:%d groupSort:%d groupOrderScan:%d "
264,634,584✔
3883
         "oneTableEachGroup:%d numTables:%zu numOutputGroups:%d code:%d",
3884
         nodeType((SNode*)pScanNode), group, groupByTbname, groupSort, pScanNode->groupOrderScan,
3885
         pTableListInfo->oneTableForEachGroup, taosArrayGetSize(pTableListInfo->pTableList),
3886
         pTableListInfo->numOfOuputGroups, code);
3887

3888
  // add all table entry in the hash map
3889
  size_t size = taosArrayGetSize(pTableListInfo->pTableList);
264,637,106✔
3890
  for (int32_t i = 0; i < size; ++i) {
994,825,896✔
3891
    STableKeyInfo* p = taosArrayGet(pTableListInfo->pTableList, i);
730,184,288✔
3892
    if (!p) {
730,035,978✔
3893
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3894
      return terrno;
×
3895
    }
3896
    int32_t tempRes = taosHashPut(pTableListInfo->map, &p->uid, sizeof(uint64_t), &i, sizeof(int32_t));
730,035,978✔
3897
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
730,212,539✔
3898
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3899
      return tempRes;
×
3900
    }
3901
  }
3902

3903
  return code;
264,720,161✔
3904
}
3905

3906
int32_t createNonStreamScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
289,769,068✔
3907
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
3908
                                SExecTaskInfo* pTaskInfo, SHashObj* groupIdMap) {
3909
  int64_t     st = taosGetTimestampUs();
289,674,397✔
3910
  const char* idStr = GET_TASKID(pTaskInfo);
289,674,397✔
3911

3912
  if (pHandle == NULL) {
289,661,350✔
3913
    qError("invalid handle, in creating operator tree, %s", idStr);
×
3914
    return TSDB_CODE_INVALID_PARA;
×
3915
  }
3916

3917
  if (pHandle->uid != 0) {
289,661,350✔
3918
    pScanNode->uid = pHandle->uid;
120,691✔
3919
    pScanNode->tableType = TSDB_CHILD_TABLE;
120,691✔
3920
  }
3921
  uint8_t digest[17] = {0};
289,744,209✔
3922
  int32_t code = getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, digest, idStr,
289,726,162✔
3923
                              &pTaskInfo->storageAPI, pTaskInfo->pStreamRuntimeInfo);
289,760,142✔
3924
  if (code != TSDB_CODE_SUCCESS) {
289,863,142✔
3925
    qError("failed to getTableList, code:%s", tstrerror(code));
1,048✔
3926
    return code;
1,048✔
3927
  }
3928

3929
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
289,862,094✔
3930

3931
  int64_t st1 = taosGetTimestampUs();
289,878,827✔
3932
  pTaskInfo->cost.extractListTime = (st1 - st) / 1000.0;
289,878,827✔
3933
  qDebug("extract queried table list completed, %d tables, elapsed time:%.2f ms %s", numOfTables,
289,836,632✔
3934
         pTaskInfo->cost.extractListTime, idStr);
3935

3936
  if (numOfTables == 0) {
289,844,252✔
3937
    qDebug("no table qualified for query, %s", idStr);
25,225,542✔
3938
    return TSDB_CODE_SUCCESS;
25,227,996✔
3939
  }
3940

3941
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, digest, &pTaskInfo->storageAPI, groupIdMap, false);
264,618,710✔
3942
  if (code != TSDB_CODE_SUCCESS) {
264,674,710✔
3943
    return code;
×
3944
  }
3945

3946
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
264,684,246✔
3947
  qDebug("generate group id map completed, elapsed time:%.2f ms %s", pTaskInfo->cost.groupIdMapTime, idStr);
264,666,891✔
3948

3949
  return TSDB_CODE_SUCCESS;
264,664,138✔
3950
}
3951

3952
int32_t createStreamGrpTableListFromCond(SScanPhysiNode* pScanNode, SReadHandle* pHandle, SExecTaskInfo* pTaskInfo, SStreamRuntimeFuncInfo* pStream, STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond, int32_t grpNum, bool hasCalcGrp) {
×
3953
  int32_t code = 0, lino = 0;
×
3954

3955
  for (int32_t i = 0; i < grpNum; ++i) {
×
3956
    SSTriggerGroupReadInfo* pGrpRead = taosArrayGet(pStream->curGrpRead, i);
×
3957
    SSTriggerGroupCalcInfo* pGrpCalc = tSimpleHashGet(pStream->pGroupCalcInfos, &pGrpRead->gid, sizeof(pGrpRead->gid));
×
3958

3959
    pStream->pStreamPartColVals = pGrpCalc->pGroupColVals;
×
3960

3961
    int32_t oldSize = taosArrayGetSize(pTableListInfo->pTableList);
×
3962
    TAOS_CHECK_EXIT(getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, NULL, GET_TASKID(pTaskInfo),
×
3963
                                &pTaskInfo->storageAPI, pTaskInfo->pStreamRuntimeInfo));
3964
    int32_t newSize = taosArrayGetSize(pTableListInfo->pTableList);
×
3965
    for (int32_t i = oldSize; i < newSize; ++i) {
×
3966
      STableKeyInfo* pKey = taosArrayGet(pTableListInfo->pTableList, i);
×
3967
      pKey->baseGId = pGrpRead->gid;
×
3968
      if (!hasCalcGrp) {
×
3969
        pKey->groupId = pGrpRead->gid;
×
3970
      }
3971
    }
3972
    
3973
    qDebug("%s get %d tables for tgrp %" PRIu64, GET_TASKID(pTaskInfo), newSize - oldSize, pGrpRead->gid);
×
3974
  }
3975

3976
  qDebug("%s get total %d tables in table list from cond", GET_TASKID(pTaskInfo), (int32_t)taosArrayGetSize(pTableListInfo->pTableList));
×
3977

3978
_exit:
×
3979

3980
  if (code) {
×
3981
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, __LINE__, tstrerror(code));
×
3982
  }
3983
  
3984
  return code;}
×
3985

3986
int32_t createStreamGrpTableListFromTrig(SExecTaskInfo* pTaskInfo, SStreamRuntimeFuncInfo* pStream, STableListInfo* pTableListInfo, int32_t grpNum, bool hasCalcGrp) {
×
3987
  STableKeyInfo tableKey = {0};
×
3988
  int32_t tblNum = 0, idx = 0;
×
3989
  int32_t code = 0, lino = 0;
×
3990

3991
  if (!hasCalcGrp) {
×
3992
    pTableListInfo->numOfOuputGroups = grpNum;
×
3993
  }
3994
  
3995
  for (int32_t i = 0; i < grpNum; ++i) {
×
3996
    SSTriggerGroupReadInfo* pGrp = taosArrayGet(pStream->curGrpRead, i);
×
3997
    tableKey.baseGId = pGrp->gid;
×
3998
    tblNum = taosArrayGetSize(pGrp->pTables);
×
3999

4000
    if (pStream->stbPartByTbname) {
×
4001
      tableKey.uid = pGrp->gid;
×
4002
      if (!hasCalcGrp) {
×
4003
        tableKey.groupId = pGrp->gid;
×
4004
        pTableListInfo->oneTableForEachGroup = true;
×
4005
      }
4006
      
4007
      TSDB_CHECK_NULL(taosArrayPush(pTableListInfo->pTableList, &tableKey), code, lino, _exit, terrno);
×
4008
      if (!hasCalcGrp) {
×
4009
        int32_t tempRes = taosHashPut(pTableListInfo->map, &tableKey.uid, sizeof(uint64_t), &idx, sizeof(int32_t));
×
4010
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
×
4011
          TAOS_CHECK_EXIT(tempRes);
×
4012
        }      
4013
      }      
4014
      
4015
      qDebug("%s gid:%" PRIu64 " table:%" PRIu64 " added to stream table list, partByTbname", GET_TASKID(pTaskInfo), tableKey.groupId, tableKey.uid);
×
4016
      idx++;
×
4017
      continue;
×
4018
    }
4019
    
4020
    for (int32_t n = 0; n < tblNum; ++n, ++idx) {
×
4021
      uint64_t* pUid = taosArrayGet(pGrp->pTables, n);
×
4022
      tableKey.uid = *pUid;
×
4023
      if (!hasCalcGrp) {
×
4024
        tableKey.groupId = pGrp->gid;
×
4025
      }
4026
      
4027
      TSDB_CHECK_NULL(taosArrayPush(pTableListInfo->pTableList, &tableKey), code, lino, _exit, terrno);
×
4028
      if (!hasCalcGrp) {
×
4029
        int32_t tempRes = taosHashPut(pTableListInfo->map, &tableKey.uid, sizeof(uint64_t), &idx, sizeof(int32_t));
×
4030
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
×
4031
          TAOS_CHECK_EXIT(tempRes);
×
4032
        }
4033
      }
4034

4035
      qDebug("%s gid:%" PRIu64 " table:%" PRIu64 " added to stream table list", GET_TASKID(pTaskInfo), tableKey.groupId, tableKey.uid);
×
4036
    }
4037
  }
4038

4039
  qDebug("%s get total %d tables in table list from trig", GET_TASKID(pTaskInfo), (int32_t)taosArrayGetSize(pTableListInfo->pTableList));
×
4040

4041
_exit:
×
4042

4043
  if (code) {
×
4044
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, __LINE__, tstrerror(code));
×
4045
  }
4046
  
4047
  return code;
×
4048
}
4049

4050
int32_t createStreamMultiGrpTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
×
4051
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
4052
                                SExecTaskInfo* pTaskInfo, SHashObj* groupIdMap) {
4053
  int32_t code = 0, lino = 0;
×
4054
  const char* idStr = GET_TASKID(pTaskInfo);
×
4055
  SStreamRuntimeFuncInfo* pStream = &pTaskInfo->pStreamRuntimeInfo->funcInfo;
×
4056
  bool hasCalcGrp = (pGroupTags && pGroupTags->length > 0);
×
4057
  int32_t grpNum = taosArrayGetSize(pStream->curGrpRead);
×
4058
  SSTriggerGroupReadInfo* pGrp = taosArrayGet(pStream->curGrpRead, 0);
×
4059
  int64_t     st = taosGetTimestampUs();
×
4060

4061
  qDebug("%s start to create table list, hasCalcGrp:%d, tgrpNum:%d", idStr, hasCalcGrp, grpNum);
×
4062
  
4063
  if (taosArrayGetSize(pGrp->pTables) <= 0) {
×
4064
    TAOS_CHECK_EXIT(createStreamGrpTableListFromCond(pScanNode, pHandle, pTaskInfo, pStream, pTableListInfo, pTagCond, pTagIndexCond, grpNum, hasCalcGrp));
×
4065
  } else {
4066
    TAOS_CHECK_EXIT(createStreamGrpTableListFromTrig(pTaskInfo, pStream, pTableListInfo, grpNum, hasCalcGrp));
×
4067
  }
4068
  
4069
  pTableListInfo->idInfo.suid = pScanNode->suid;
×
4070
  pTableListInfo->idInfo.tableType = pScanNode->tableType;
×
4071

4072
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
×
4073

4074
  int64_t st1 = taosGetTimestampUs();
×
4075
  qDebug("%s build multi tgrp table list completed, elapsed time:%.2f ms", idStr, (st1 - st) / 1000.0);
×
4076

4077
  if (numOfTables == 0) {
×
4078
    qDebug("%s no table qualified for stream multi tgrp", idStr);
×
4079
    return TSDB_CODE_SUCCESS;
×
4080
  }
4081

4082
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, NULL, &pTaskInfo->storageAPI, groupIdMap, true);
×
4083
  if (code != TSDB_CODE_SUCCESS) {
×
4084
    return code;
×
4085
  }
4086

4087
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
×
4088
  qDebug("%s generate stream group id map completed, elapsed time:%.2f ms", idStr, pTaskInfo->cost.groupIdMapTime);
×
4089

4090
_exit:
×
4091

4092
  if (code) {
×
4093
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, lino, tstrerror(code));
×
4094
  }
4095
  
4096
  return code;
×
4097
}
4098

4099

4100
int32_t createScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
292,271,504✔
4101
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
4102
                                SExecTaskInfo* pTaskInfo, SHashObj* groupIdMap) {
4103
  int32_t code = 0;
292,271,504✔
4104
  // Ensure curGrpRead is prepared from pre-initialized group calc infos for non-stream external-window initialization.
4105
  if (pTaskInfo->pStreamRuntimeInfo) {
292,271,504✔
4106
    SStreamRuntimeFuncInfo* pStream = &pTaskInfo->pStreamRuntimeInfo->funcInfo;
4,967,581✔
4107
    if (pStream->isMultiGroupCalc && pStream->curGrpRead == NULL && pStream->pGroupCalcInfos != NULL) {
4,967,581✔
4108
      int32_t n = tSimpleHashGetSize(pStream->pGroupCalcInfos);
×
4109
      if (n < 0) n = 0;
×
4110
      pStream->curGrpRead = taosArrayInit_s(sizeof(SSTriggerGroupReadInfo), n > 0 ? n : 4);
×
4111
      if (pStream->curGrpRead == NULL) {
×
4112
        return terrno;
×
4113
      }
4114
      int32_t iter = 0;
×
4115
      SSTriggerGroupCalcInfo* pGrp = tSimpleHashIterate(pStream->pGroupCalcInfos, NULL, &iter);
×
4116
      while (pGrp != NULL) {
×
4117
        uint64_t gid = *(uint64_t*)tSimpleHashGetKey(pGrp, NULL);
×
4118
        SSTriggerGroupReadInfo readInfo = {0};
×
4119
        readInfo.gid = gid;
×
4120
        readInfo.pTables = NULL;  // tables decided by getTableList when needed
×
4121
        if (taosArrayPush(pStream->curGrpRead, &readInfo) == NULL) {
×
4122
          return terrno;
×
4123
        }
4124
        pGrp = tSimpleHashIterate(pStream->pGroupCalcInfos, pGrp, &iter);
×
4125
      }
4126
    }
4127
  }
4128
  if (pTaskInfo->pStreamRuntimeInfo && pTaskInfo->pStreamRuntimeInfo->funcInfo.isMultiGroupCalc /*&&got table list*/) {
292,298,884✔
4129
    code = createStreamMultiGrpTableListInfo(pScanNode, pGroupTags, groupSort,
×
4130
                                    pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo, groupIdMap);
4131
    if (code) {
×
4132
      qError("%s failed to createStreamMultiGrpTableListInfo, code:%s", __func__, tstrerror(code));
×
4133
      return code;
×
4134
    }
4135
  } else if (!pScanNode->node.dynamicOp) {
292,324,670✔
4136
    code = createNonStreamScanTableListInfo(pScanNode, pGroupTags, groupSort,
289,739,746✔
4137
                                    pHandle, pTableListInfo, pTagCond, pTagIndexCond, pTaskInfo, groupIdMap);
4138
    if (code) {
289,888,778✔
4139
      qError("%s failed to createScanTableListInfo, code:%s", __func__, tstrerror(code));
47,080✔
4140
      return code;
1,048✔
4141
    }
4142
  }
4143

4144
  return TSDB_CODE_SUCCESS;
292,372,275✔
4145
}
4146

4147

4148
char* getStreamOpName(uint16_t opType) {
7,969,596✔
4149
  switch (opType) {
7,969,596✔
4150
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
×
4151
      return "stream scan";
×
4152
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
7,536,109✔
4153
      return "project";
7,536,109✔
4154
    case QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW:
433,487✔
4155
      return "external window";
433,487✔
4156
  }
4157
  return "error name";
×
4158
}
4159

4160
void printDataBlock(SSDataBlock* pBlock, const char* flag, const char* taskIdStr, int64_t qId) {
661,529,504✔
4161
  if (qDebugFlag & DEBUG_TRACE) {
661,529,504✔
4162
    if (!pBlock) {
1,588,302✔
4163
      qDebug("%" PRIx64 " %s %s %s: Block is Null", qId, taskIdStr, flag, __func__);
54,573✔
4164
      return;
55,915✔
4165
    } else if (pBlock->info.rows == 0) {
1,533,729✔
4166
      qDebug("%" PRIx64 " %s %s %s: Block is Empty.", qId, taskIdStr, flag, __func__);
10,233✔
4167
      return;
10,233✔
4168
    }
4169
    
4170
    char*   pBuf = NULL;
1,523,496✔
4171
    int32_t code = dumpBlockData(pBlock, flag, &pBuf, taskIdStr, qId);
1,523,496✔
4172
    if (code == 0) {
1,523,496✔
4173
      qDebugL("%" PRIx64 " %s %s", qId, __func__, pBuf);
1,523,496✔
4174
      taosMemoryFree(pBuf);
1,523,496✔
4175
    }
4176
  }
4177
}
4178

4179
void printSpecDataBlock(SSDataBlock* pBlock, const char* flag, const char* opStr, const char* taskIdStr) {
×
4180
  if (!pBlock) {
×
4181
    qDebug("%s===stream===%s %s: Block is Null", taskIdStr, flag, opStr);
×
4182
    return;
×
4183
  } else if (pBlock->info.rows == 0) {
×
4184
    qDebug("%s===stream===%s %s: Block is Empty.skey:%" PRId64 ",ekey:%" PRId64 ",version%" PRId64,
×
4185
           taskIdStr, flag, opStr, pBlock->info.window.skey, pBlock->info.window.ekey,
4186
           pBlock->info.version);
4187
    return;
×
4188
  }
4189
  if (qDebugFlag & DEBUG_TRACE) {
×
4190
    char* pBuf = NULL;
×
4191
    char  flagBuf[64];
×
4192
    snprintf(flagBuf, sizeof(flagBuf), "%s %s", flag, opStr);
×
4193
    int32_t code = dumpBlockData(pBlock, flagBuf, &pBuf, taskIdStr, 0);
×
4194
    if (code == 0) {
×
4195
      qDebug("%s", pBuf);
×
4196
      taosMemoryFree(pBuf);
×
4197
    }
4198
  }
4199
}
4200

4201
TSKEY getStartTsKey(STimeWindow* win, const TSKEY* tsCols) { return tsCols == NULL ? win->skey : tsCols[0]; }
13,523,788✔
4202

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

4206
  int64_t duration = pWin->ekey > pWin->skey ? pWin->ekey - pWin->skey + delta : pWin->skey - pWin->ekey + delta;
2,147,483,647✔
4207
  ts[2] = duration;            // set the duration
2,147,483,647✔
4208
  ts[3] = pWin->skey;          // window start key
2,147,483,647✔
4209
  ts[4] = pWin->ekey + delta;  // window end key
2,147,483,647✔
4210
}
2,147,483,647✔
4211

4212
int32_t compKeys(const SArray* pSortGroupCols, const char* oldkeyBuf, int32_t oldKeysLen, const SSDataBlock* pBlock,
2,147,483,647✔
4213
                 int32_t rowIndex) {
4214
  SColumnDataAgg* pColAgg = NULL;
2,147,483,647✔
4215
  const char*     isNull = oldkeyBuf;
2,147,483,647✔
4216
  const char*     p = oldkeyBuf + sizeof(int8_t) * pSortGroupCols->size;
2,147,483,647✔
4217

4218
  for (int32_t i = 0; i < pSortGroupCols->size; ++i) {
2,147,483,647✔
4219
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
2,147,483,647✔
4220
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
2,147,483,647✔
4221
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
2,147,483,647✔
4222

4223
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
2,147,483,647✔
4224
      if (isNull[i] != 1) return 1;
295,354,210✔
4225
    } else {
4226
      if (isNull[i] != 0) return 1;
2,147,483,647✔
4227
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,147,483,647✔
4228
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,147,483,647✔
4229
        int32_t len = getJsonValueLen(val);
×
4230
        if (memcmp(p, val, len) != 0) return 1;
×
4231
        p += len;
×
4232
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,147,483,647✔
4233
        if (IS_STR_DATA_BLOB(pCol->type)) {
1,359,314,094✔
4234
          if (memcmp(p, val, blobDataTLen(val)) != 0) return 1;
×
4235
          p += blobDataTLen(val);
×
4236
        } else {
4237
          if (memcmp(p, val, varDataTLen(val)) != 0) return 1;
1,358,853,760✔
4238
          p += varDataTLen(val);
1,358,068,632✔
4239
        }
4240
      } else {
4241
        if (0 != memcmp(p, val, pCol->bytes)) return 1;
2,147,483,647✔
4242
        p += pCol->bytes;
2,147,483,647✔
4243
      }
4244
    }
4245
  }
4246
  if ((int32_t)(p - oldkeyBuf) != oldKeysLen) return 1;
2,147,483,647✔
4247
  return 0;
2,147,483,647✔
4248
}
4249

4250
int32_t buildKeys(char* keyBuf, const SArray* pSortGroupCols, const SSDataBlock* pBlock, int32_t rowIndex) {
1,792,140✔
4251
  uint32_t        colNum = pSortGroupCols->size;
1,792,140✔
4252
  SColumnDataAgg* pColAgg = NULL;
1,792,140✔
4253
  char*           isNull = keyBuf;
1,792,140✔
4254
  char*           p = keyBuf + sizeof(int8_t) * colNum;
1,792,140✔
4255

4256
  for (int32_t i = 0; i < colNum; ++i) {
4,904,540✔
4257
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
3,112,400✔
4258
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
3,112,400✔
4259
    if (pCol->slotId > pBlock->pDataBlock->size) continue;
3,112,400✔
4260

4261
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
3,111,898✔
4262

4263
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
6,224,298✔
4264
      isNull[i] = 1;
150,600✔
4265
    } else {
4266
      isNull[i] = 0;
2,961,298✔
4267
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,961,800✔
4268
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,961,298✔
4269
        int32_t len = getJsonValueLen(val);
×
4270
        memcpy(p, val, len);
×
4271
        p += len;
×
4272
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,961,800✔
4273
        if (IS_STR_DATA_BLOB(pCol->type)) {
1,107,914✔
4274
          blobDataCopy(p, val);
×
4275
          p += blobDataTLen(val);
×
4276
        } else {
4277
          varDataCopy(p, val);
1,106,910✔
4278
          p += varDataTLen(val);
1,106,910✔
4279
        }
4280
      } else {
4281
        memcpy(p, val, pCol->bytes);
1,854,890✔
4282
        p += pCol->bytes;
1,854,890✔
4283
      }
4284
    }
4285
  }
4286
  return (int32_t)(p - keyBuf);
1,792,140✔
4287
}
4288

4289
uint64_t calcGroupId(char* pData, int32_t len) {
2,147,483,647✔
4290
  T_MD5_CTX context;
2,147,483,647✔
4291
  tMD5Init(&context);
2,147,483,647✔
4292
  tMD5Update(&context, (uint8_t*)pData, len);
2,147,483,647✔
4293
  tMD5Final(&context);
2,147,483,647✔
4294

4295
  // NOTE: only extract the initial 8 bytes of the final MD5 digest
4296
  uint64_t id = 0;
2,147,483,647✔
4297
  memcpy(&id, context.digest, sizeof(uint64_t));
2,147,483,647✔
4298
  if (0 == id) memcpy(&id, context.digest + 8, sizeof(uint64_t));
2,147,483,647✔
4299
  return id;
2,147,483,647✔
4300
}
4301

4302
SNodeList* makeColsNodeArrFromSortKeys(SNodeList* pSortKeys) {
118,472✔
4303
  SNode*     node;
4304
  SNodeList* ret = NULL;
118,472✔
4305
  FOREACH(node, pSortKeys) {
361,440✔
4306
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)node;
242,968✔
4307
    int32_t           code = nodesListMakeAppend(&ret, pSortKey->pExpr);
242,968✔
4308
    if (code != TSDB_CODE_SUCCESS) {
242,968✔
4309
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4310
      terrno = code;
×
4311
      return NULL;
×
4312
    }
4313
  }
4314
  return ret;
118,472✔
4315
}
4316

4317
int32_t extractKeysLen(const SArray* keys, int32_t* pLen) {
118,472✔
4318
  int32_t code = TSDB_CODE_SUCCESS;
118,472✔
4319
  int32_t lino = 0;
118,472✔
4320
  int32_t len = 0;
118,472✔
4321
  int32_t keyNum = taosArrayGetSize(keys);
118,472✔
4322
  for (int32_t i = 0; i < keyNum; ++i) {
301,200✔
4323
    SColumn* pCol = (SColumn*)taosArrayGet(keys, i);
182,728✔
4324
    QUERY_CHECK_NULL(pCol, code, lino, _end, terrno);
182,728✔
4325
    len += pCol->bytes;
182,728✔
4326
  }
4327
  len += sizeof(int8_t) * keyNum;  // null flag
118,472✔
4328
  *pLen = len;
118,472✔
4329

4330
_end:
118,472✔
4331
  if (code != TSDB_CODE_SUCCESS) {
118,472✔
4332
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4333
  }
4334
  return code;
118,472✔
4335
}
4336

4337
int32_t parseErrorMsgFromAnalyticServer(SJson* pJson, const char* typeStr, const char* pId) {
×
4338
  int32_t code = TSDB_CODE_ANA_ANODE_RETURN_ERROR;
×
4339
  char*   pMsg = NULL;
×
4340
  if (pJson == NULL) {
×
4341
    return code;
×
4342
  }
4343

4344
  int32_t ret = tjsonDupStringValue(pJson, "msg", &pMsg);
×
4345
  if (ret == 0 && pMsg != NULL) {
×
4346
    qError("%s failed to exec %s operation, msg:%s", pId, typeStr, pMsg);
×
4347
    if (strstr(pMsg, "white noise") != NULL) {
×
4348
      code = TSDB_CODE_ANA_WN_DATA;
×
4349
    } else if (strstr(pMsg, "white-noise") != NULL) {
×
4350
      code = TSDB_CODE_ANA_WN_DATA;
×
4351
    } else if (strstr(pMsg, "[Errno 111] Connection refused") != NULL) {
×
4352
      code = TSDB_CODE_ANA_ALGO_NOT_LOAD;
×
4353
    } else if (strstr(pMsg, "failed to load model") != NULL) {
×
4354
      code = TSDB_CODE_ANA_ALGO_NOT_LOAD;
×
4355
    }
4356
  } else {
4357
    qError("%s failed to extract msg from server, unknown error", pId);
×
4358
  }
4359

4360
  taosMemoryFreeClear(pMsg);
×
4361
  return code;
×
4362
}
4363

4364

4365
int32_t createExprSubQResBlock(SSDataBlock** ppBlock, SDataType* pResType, int32_t cols) {
30,477,337✔
4366
  int32_t code = 0;
30,477,337✔
4367
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
30,477,337✔
4368
  if (pBlock == NULL) {
30,477,337✔
4369
    return terrno;
×
4370
  }
4371

4372
  pBlock->pDataBlock = taosArrayInit(cols, sizeof(SColumnInfoData));
30,477,337✔
4373
  if (pBlock->pDataBlock == NULL) {
30,476,182✔
4374
    code = terrno;
×
4375
    taosMemoryFree(pBlock);
×
4376
    return code;
×
4377
  }
4378

4379
  for (int32_t i = 0; i < cols; ++i) {
62,227,018✔
4380
    SColumnInfoData idata = {0};
31,749,681✔
4381
    if (pResType) {
31,747,371✔
4382
      idata = createColumnInfoData(pResType->type, pResType->bytes, 0);
29,820,009✔
4383
      idata.info.scale = pResType->scale;
29,821,739✔
4384
      idata.info.precision = pResType->precision;
29,822,319✔
4385
    }
4386
    
4387
    code = blockDataAppendColInfo(pBlock, &idata);
31,747,946✔
4388
    if (code != TSDB_CODE_SUCCESS) {
31,749,681✔
4389
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
580✔
4390
      blockDataDestroy(pBlock);
580✔
4391
      *ppBlock = NULL;
×
4392
      return code;
×
4393
    }
4394
  }
4395

4396
  *ppBlock = pBlock;
30,477,337✔
4397

4398
  return code;
30,476,762✔
4399
}
4400

4401

4402
int32_t extractSingleRspBlock(SRetrieveTableRsp* pRetrieveRsp, SSDataBlock* pb) {
30,477,917✔
4403
  int32_t            code = TSDB_CODE_SUCCESS;
30,477,917✔
4404
  int32_t            lino = 0;
30,477,917✔
4405
  void*              decompBuf = NULL;
30,477,917✔
4406

4407
  char* pNextStart = pRetrieveRsp->data;
30,477,917✔
4408
  char* pStart = pNextStart;
30,476,762✔
4409

4410
  int32_t index = 0;
30,476,762✔
4411

4412
  if (pRetrieveRsp->compressed) {  // decompress the data
30,476,762✔
4413
    decompBuf = taosMemoryMalloc(pRetrieveRsp->payloadLen);
×
4414
    QUERY_CHECK_NULL(decompBuf, code, lino, _end, terrno);
×
4415
  }
4416

4417
  int32_t compLen = *(int32_t*)pStart;
30,477,337✔
4418
  pStart += sizeof(int32_t);
30,477,917✔
4419

4420
  int32_t rawLen = *(int32_t*)pStart;
30,476,762✔
4421
  pStart += sizeof(int32_t);
30,476,762✔
4422
  QUERY_CHECK_CONDITION((compLen <= rawLen && compLen != 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
30,477,337✔
4423

4424
  pNextStart = pStart + compLen;
30,477,337✔
4425
  if (pRetrieveRsp->compressed && (compLen < rawLen)) {
30,477,337✔
4426
    int32_t t = tsDecompressString(pStart, compLen, 1, decompBuf, rawLen, ONE_STAGE_COMP, NULL, 0);
×
4427
    QUERY_CHECK_CONDITION((t == rawLen), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
4428
    pStart = decompBuf;
×
4429
  }
4430

4431
  code = blockDecodeInternal(pb, pStart, (const char**)&pStart);
30,476,762✔
4432
  if (code != 0) {
30,477,917✔
4433
    taosMemoryFreeClear(pRetrieveRsp);
×
4434
    goto _end;
×
4435
  }
4436

4437
_end:
30,477,917✔
4438
  if (code != TSDB_CODE_SUCCESS) {
30,477,917✔
4439
    blockDataDestroy(pb);
×
4440
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4441
  }
4442
  return code;
30,477,917✔
4443
}
4444

4445
int32_t setValueFromResBlock(STaskSubJobCtx* ctx, SValueNode* pRes, SSDataBlock* pBlock) {
19,959,440✔
4446
  int32_t code = 0;
19,959,440✔
4447
  bool needFree = true;
19,959,440✔
4448
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
19,959,440✔
4449
  if (NULL == pBlock->pDataBlock || 1 > colNum || pBlock->info.rows > 1) {
19,959,440✔
UNCOV
4450
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d, rows:%" PRId64, 
×
4451
      ctx->idStr, pBlock->pDataBlock, colNum, pBlock->info.rows);
4452
    return TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
×
4453
  }
4454
  
4455
  pRes->flag &= (~VALUE_FLAG_VAL_UNSET);
19,958,860✔
4456
  pRes->translate = true;
19,958,860✔
4457
  
4458
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
19,958,280✔
4459
  if (colDataIsNull_s(pCol, 0)) {
19,959,440✔
4460
    pRes->isNull = true;
5,393,880✔
4461
  } else {
4462
    code = nodesSetValueNodeValueExt(pRes, colDataGetData(pCol, 0), &needFree);
14,565,560✔
4463
  }
4464

4465
  if (!needFree) {
19,958,860✔
4466
    pCol->pData = NULL;
1,982,448✔
4467
  }
4468

4469
  return code;
19,958,860✔
4470
}
4471

4472
void handleRemoteValueRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
3,553,870✔
4473
  SSDataBlock*      pResBlock = NULL;
3,553,870✔
4474
  SExecTaskInfo*    pTaskInfo = ctx->pTaskInfo;
3,553,870✔
4475
  SRemoteValueNode* pRemote = (SRemoteValueNode*)pParam->pRes;
3,553,290✔
4476

4477
  qDebug("%s scl fetch value rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
3,553,870✔
4478

4479
  if (IS_STREAM_MODE(pTaskInfo)) {
3,553,870✔
4480
    SNode** ppRes = taosArrayGet(ctx->subResNodes, pParam->subQIdx);
80,094✔
4481
    if (NULL == *ppRes && 0 == pRsp->numOfRows) {
80,094✔
4482
      pRemote->val.node.type = QUERY_NODE_VALUE;
×
4483
      pRemote->val.isNull = true;
×
4484
      pRemote->val.translate = true;
×
4485
      pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
×
4486
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
×
4487
    } else if (pRsp->numOfRows > 0 || pRsp->numOfBlocks > 0) {
80,094✔
4488
      ctx->code = createExprSubQResBlock(&pResBlock, &((SRemoteValueNode*)pParam->pRes)->val.node.resType, 1);
40,047✔
4489
      if (TSDB_CODE_SUCCESS == ctx->code) {
40,047✔
4490
        ctx->code = blockDataEnsureCapacity(pResBlock, 1);
40,047✔
4491
      }
4492
      if (TSDB_CODE_SUCCESS == ctx->code) {
40,047✔
4493
        ctx->code = extractSingleRspBlock(pRsp, pResBlock);
40,047✔
4494
      }
4495
      if (TSDB_CODE_SUCCESS == ctx->code) {
40,047✔
4496
        ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
40,047✔
4497
        pRemote->val.node.type = QUERY_NODE_REMOTE_VALUE;
40,047✔
4498
      }
4499
      if (TSDB_CODE_SUCCESS == ctx->code) {
40,047✔
4500
        taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
40,047✔
4501
      }
4502

4503
      blockDataDestroy(pResBlock);
40,047✔
4504
    } else if (NULL != *ppRes && 0 == pRsp->numOfRows) {
40,047✔
4505
      pRemote->val.node.type = QUERY_NODE_VALUE;
40,047✔
4506
      pRsp->completed = true;
40,047✔
4507
    }
4508

4509
    *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
80,094✔
4510

4511
    if (!(*fetchDone)) {
80,094✔
4512
      int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
40,047✔
4513
      if (TSDB_CODE_SUCCESS != code) {
40,047✔
4514
        ctx->code = code;
×
4515
        *fetchDone = true;
×
4516
      }
4517
    }
4518

4519
    return;
80,094✔
4520
  }
4521

4522
  *fetchDone = true;
3,472,616✔
4523

4524
  if (pRsp->numOfRows > 1 || pRsp->numOfBlocks > 1 || !pRsp->completed) {
3,472,616✔
4525
    qError("%s invalid scl value fetch rsp received, subQIdx:%d, rows:%" PRId64 ", blocks:%d, completed:%d", 
×
4526
      ctx->idStr, pParam->subQIdx, pRsp->numOfRows, pRsp->numOfBlocks, pRsp->completed);
4527
    ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
×
4528

4529
    return;
×
4530
  }
4531

4532
  if (1 != pRsp->numOfCols && pRsp->numOfRows > 0) {
3,473,196✔
4533
    qError("%s invalid scl value fetch rsp received, subQIdx:%d, cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfCols);
×
4534
    ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_COLS;
×
4535

4536
    return;
×
4537
  }
4538

4539
  if (0 == pRsp->numOfRows) {
3,473,196✔
4540
    pRemote->val.node.type = QUERY_NODE_VALUE;
185,000✔
4541
    pRemote->val.isNull = true;
185,000✔
4542
    pRemote->val.translate = true;
185,000✔
4543
    pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
185,000✔
4544
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
185,000✔
4545

4546
    return;
185,000✔
4547
  }
4548
  
4549
  ctx->code = createExprSubQResBlock(&pResBlock, &((SRemoteValueNode*)pParam->pRes)->val.node.resType, 1);
3,288,196✔
4550
  if (TSDB_CODE_SUCCESS == ctx->code) {
3,288,776✔
4551
    ctx->code = blockDataEnsureCapacity(pResBlock, 1);
3,288,776✔
4552
  }
4553
  if (TSDB_CODE_SUCCESS == ctx->code) {
3,288,776✔
4554
    ctx->code = extractSingleRspBlock(pRsp, pResBlock);
3,288,776✔
4555
  }
4556
  if (TSDB_CODE_SUCCESS == ctx->code) {
3,288,776✔
4557
    ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
3,287,616✔
4558
  }
4559
  if (TSDB_CODE_SUCCESS == ctx->code) {
3,289,356✔
4560
    pRemote->val.node.type = QUERY_NODE_VALUE;
3,288,776✔
4561
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
3,287,616✔
4562
  }
4563

4564
  blockDataDestroy(pResBlock);  
3,287,616✔
4565
}
4566

4567

4568
int32_t setValueListFromResBlock(STaskSubJobCtx* ctx, SRemoteValueListNode* pRes, SSDataBlock* pBlock) {
9,862,879✔
4569
  int32_t code = 0, lino = 0;
9,862,879✔
4570
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
9,862,879✔
4571
  if (NULL == pBlock->pDataBlock || 1 != colNum) {
9,861,729✔
UNCOV
4572
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d", ctx->idStr, pBlock->pDataBlock, colNum);
×
4573
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4574
  }
4575

4576
  pRes->hasValue = true;
9,862,304✔
4577
  
4578
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
9,862,879✔
4579
  TAOS_CHECK_EXIT(scalarBuildRemoteListHash(ctx->idStr, pRes, pCol, pBlock->info.rows));
9,861,729✔
4580

4581
_exit:
9,093,470✔
4582

4583
  if (code) {
9,862,304✔
4584
    qError("%s %s failed with error: %s", ctx->idStr, __func__, tstrerror(code));
768,834✔
4585
  }
4586
  
4587
  return code;
9,862,879✔
4588
}
4589

4590

4591

4592
void handleRemoteValueListRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
12,995,715✔
4593
  SSDataBlock* pResBlock = NULL;
12,995,715✔
4594
  SRemoteValueListNode* pRemote = (SRemoteValueListNode*)pParam->pRes;
12,995,715✔
4595
  SExecTaskInfo* pTaskInfo = ctx->pTaskInfo;
12,995,715✔
4596

4597
  qDebug("%s scl fetch valueList rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
12,995,715✔
4598

4599
  if (pRsp->numOfRows > 0) {
12,995,715✔
4600
    if (1 != pRsp->numOfCols) {
9,862,879✔
4601
      qError("%s invalid scl valueList fetch rsp received, subQIdx:%d, cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfCols);
×
4602
      ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_COLS;
×
4603
      *fetchDone = true;
×
4604
      return;
×
4605
    }
4606

4607
    ctx->code = createExprSubQResBlock(&pResBlock, &((SExprNode*)pParam->pRes)->resType, 1);
9,861,729✔
4608
    if (TSDB_CODE_SUCCESS == ctx->code) {
9,862,879✔
4609
      ctx->code = blockDataEnsureCapacity(pResBlock, pRsp->numOfRows);
9,862,304✔
4610
    }
4611
    if (TSDB_CODE_SUCCESS == ctx->code) {
9,862,304✔
4612
      ctx->code = extractSingleRspBlock(pRsp, pResBlock);
9,861,729✔
4613
    }
4614
    if (TSDB_CODE_SUCCESS == ctx->code) {
9,863,454✔
4615
      ctx->code = setValueListFromResBlock(ctx, pRemote, pResBlock);
9,862,879✔
4616
    }
4617
    if (TSDB_CODE_SUCCESS == ctx->code && pRsp->completed) {
9,862,879✔
4618
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
8,430,948✔
4619
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
8,430,373✔
4620
    }
4621

4622
    blockDataDestroy(pResBlock);  
9,862,879✔
4623
  } else if (0 == pRsp->numOfRows && pRsp->completed) {
3,132,836✔
4624
    if (!pRemote->hasValue) {
3,132,836✔
4625
      ctx->code = scalarBuildRemoteListHash(ctx->idStr, pRemote, NULL, 0);
3,120,084✔
4626
    }
4627
    if (TSDB_CODE_SUCCESS == ctx->code) {    
3,132,836✔
4628
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
2,872,322✔
4629
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
2,872,322✔
4630
    }
4631
  }
4632

4633
  if (IS_STREAM_MODE(pTaskInfo) && 0 == pRsp->numOfRows) {
12,995,715✔
4634
    if (!pRemote->hasValue) {
13,022✔
4635
      ctx->code = scalarBuildRemoteListHash(ctx->idStr, pRemote, NULL, 0);
270✔
4636
    }
4637
    if (TSDB_CODE_SUCCESS == ctx->code) {
13,022✔
4638
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
13,022✔
4639
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
13,022✔
4640
    }
4641

4642
    pRsp->completed = true;
13,022✔
4643
  }
4644

4645
  *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
12,995,715✔
4646

4647
  if (!(*fetchDone)) {
12,995,140✔
4648
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
663,097✔
4649
    if (TSDB_CODE_SUCCESS != code) {
663,097✔
4650
      ctx->code = code;
×
4651
      *fetchDone = true;
×
4652
    }
4653
  }
4654
}
4655

4656
int32_t setRowHasNullFromResBlock(STaskSubJobCtx* ctx, bool* hasNull, SSDataBlock* pBlock) {
16,630,617✔
4657
  int32_t code = 0;
16,630,617✔
4658
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
16,630,617✔
4659
  if (2 != colNum) {
16,630,617✔
4660
    qError("%s invalid scl fetch res block, colNum:%d", ctx->idStr, colNum);
×
4661
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4662
  }
4663
  
4664
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 1);
16,630,617✔
4665
  if (colDataIsNull_s(pCol, 0)) {
16,630,617✔
4666
    qError("%s invalid has_null res since it's null", ctx->idStr);
×
4667
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4668
  } else {
4669
    *hasNull = *(bool*)colDataGetData(pCol, 0);
16,630,617✔
4670
  }
4671

4672
  return code;
16,630,617✔
4673
}
4674

4675
void handleRemoteRowRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
22,540,377✔
4676
  SSDataBlock* pResBlock = NULL;
22,540,377✔
4677
  SExecTaskInfo* pTaskInfo = ctx->pTaskInfo;
22,540,377✔
4678

4679
  qDebug("%s scl fetch row rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
22,540,377✔
4680

4681
  SRemoteRowNode* pRemote = (SRemoteRowNode*)pParam->pRes;
22,540,377✔
4682

4683
  if (IS_STREAM_MODE(pTaskInfo)) {
22,540,377✔
4684
    SNode** ppRes = taosArrayGet(ctx->subResNodes, pParam->subQIdx);
1,080✔
4685
    if (NULL == *ppRes && 0 == pRsp->numOfRows) {
1,080✔
4686
      pRemote->valSet = true;
×
4687
      pRemote->hasValue = false;
×
4688
      pRemote->hasNull = false;
×
4689
      pRemote->val.isNull = true;
×
4690
      pRemote->val.translate = true;
×
4691
      pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
×
4692
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
×
4693
    } else if (pRsp->numOfRows > 0 || pRsp->numOfBlocks > 0) {
1,080✔
4694
      if (2 != pRsp->numOfCols) {
540✔
4695
        qError("%s invalid scl fetch row rsp received, subQIdx:%d, cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfCols);
×
4696
        ctx->code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4697
        *fetchDone = true;
×
4698
        return;
×
4699
      }
4700

4701
      ctx->code = createExprSubQResBlock(&pResBlock, &pRemote->val.node.resType, 1);
540✔
4702
      if (TSDB_CODE_SUCCESS == ctx->code) {
540✔
4703
        SColumnInfoData idata = createColumnInfoData(TSDB_DATA_TYPE_BOOL, tDataTypes[TSDB_DATA_TYPE_BOOL].bytes, 0);
540✔
4704
        ctx->code = blockDataAppendColInfo(pResBlock, &idata);
540✔
4705
      }
4706
      if (TSDB_CODE_SUCCESS == ctx->code) {
540✔
4707
        ctx->code = blockDataEnsureCapacity(pResBlock, 1);
540✔
4708
      }
4709
      if (TSDB_CODE_SUCCESS == ctx->code) {
540✔
4710
        ctx->code = extractSingleRspBlock(pRsp, pResBlock);
540✔
4711
      }
4712
      if (TSDB_CODE_SUCCESS == ctx->code) {
540✔
4713
        ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
540✔
4714
      }
4715
      if (TSDB_CODE_SUCCESS == ctx->code) {
540✔
4716
        ctx->code = setRowHasNullFromResBlock(ctx, &pRemote->hasNull, pResBlock);
540✔
4717
      }
4718
      if (TSDB_CODE_SUCCESS == ctx->code) {
540✔
4719
        pRemote->valSet = true;
540✔
4720
        pRemote->hasValue = true;
540✔
4721
        pRemote->val.node.type = QUERY_NODE_REMOTE_ROW;
540✔
4722
        taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
540✔
4723
      }
4724

4725
      blockDataDestroy(pResBlock);
540✔
4726
    } else if (NULL != *ppRes && 0 == pRsp->numOfRows) {
540✔
4727
      pRemote->val.node.type = QUERY_NODE_VALUE;
540✔
4728
      pRsp->completed = true;
540✔
4729
    }
4730

4731
    *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
1,080✔
4732

4733
    if (!(*fetchDone)) {
1,080✔
4734
      ctx->code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
540✔
4735
      if (TSDB_CODE_SUCCESS != ctx->code) {
540✔
4736
        *fetchDone = true;
×
4737
      }
4738
    }
4739

4740
    return;
1,080✔
4741
  }
4742

4743
  *fetchDone = true;
22,539,297✔
4744

4745
  if (pRsp->numOfRows > 1 || pRsp->numOfBlocks > 1 || !pRsp->completed) {
22,539,297✔
4746
    qError("%s invalid scl fetch row rsp received, subQIdx:%d, rows:%" PRId64 ", blocks:%d, completed:%d",
×
4747
      ctx->idStr, pParam->subQIdx, pRsp->numOfRows, pRsp->numOfBlocks, pRsp->completed);
4748
    ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
×
4749

4750
    return;
×
4751
  }
4752

4753
  if (0 == pRsp->numOfRows) {
22,539,297✔
4754
    pRemote->valSet = true;
5,909,220✔
4755
    pRemote->hasValue = false;
5,909,220✔
4756
    pRemote->hasNull = false;
5,909,220✔
4757
    pRemote->val.isNull = true;
5,909,220✔
4758
    pRemote->val.translate = true;
5,909,220✔
4759
    pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
5,909,220✔
4760
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
5,909,220✔
4761

4762
    return;
5,909,220✔
4763
  }
4764

4765
  if (2 != pRsp->numOfCols) {
16,630,077✔
4766
    qError("%s invalid scl fetch row rsp received, subQIdx:%d, cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfCols);
×
4767
    ctx->code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4768

4769
    return;
×
4770
  }
4771

4772
  ctx->code = createExprSubQResBlock(&pResBlock, &pRemote->val.node.resType, 1);
16,630,077✔
4773
  if (TSDB_CODE_SUCCESS == ctx->code) {
16,630,077✔
4774
    SColumnInfoData idata = createColumnInfoData(TSDB_DATA_TYPE_BOOL, tDataTypes[TSDB_DATA_TYPE_BOOL].bytes, 0);
16,630,077✔
4775
    ctx->code = blockDataAppendColInfo(pResBlock, &idata);
16,630,077✔
4776
  }
4777
  if (TSDB_CODE_SUCCESS == ctx->code) {
16,630,077✔
4778
    ctx->code = blockDataEnsureCapacity(pResBlock, 1);
16,630,077✔
4779
  }
4780
  if (TSDB_CODE_SUCCESS == ctx->code) {
16,630,077✔
4781
    ctx->code = extractSingleRspBlock(pRsp, pResBlock);
16,630,077✔
4782
  }
4783
  if (TSDB_CODE_SUCCESS == ctx->code) {
16,630,077✔
4784
    ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
16,630,077✔
4785
  }
4786
  if (TSDB_CODE_SUCCESS == ctx->code) {
16,630,077✔
4787
    ctx->code = setRowHasNullFromResBlock(ctx, &pRemote->hasNull, pResBlock);
16,630,077✔
4788
  }
4789
  if (TSDB_CODE_SUCCESS == ctx->code) {
16,630,077✔
4790
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
16,630,077✔
4791
  }
4792
  if (TSDB_CODE_SUCCESS == ctx->code) {
16,630,077✔
4793
    pRemote->valSet = true;
16,630,077✔
4794
    pRemote->hasValue = true;
16,630,077✔
4795
  }
4796

4797
  blockDataDestroy(pResBlock);
16,630,077✔
4798
}
4799

4800

4801
int32_t setZeroRowsResValue(STaskSubJobCtx* ctx, SValueNode* pRes, int32_t rows) {
294,774✔
4802
  pRes->node.type = QUERY_NODE_VALUE;
294,774✔
4803
  pRes->flag &= (~VALUE_FLAG_VAL_UNSET);
294,774✔
4804
  pRes->translate = true;
294,774✔
4805
  
4806
  return nodesSetValueNodeValue(pRes, &rows);
294,774✔
4807
}
4808

4809
void handleRemoteZeroRowsRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
294,504✔
4810
  SRemoteZeroRowsNode* pRemote = (SRemoteZeroRowsNode*)pParam->pRes;
294,504✔
4811
  SExecTaskInfo* pTaskInfo = ctx->pTaskInfo;
294,504✔
4812

4813
  qDebug("%s scl fetch zeroRows rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
294,504✔
4814

4815
  int32_t resRows = (pRsp->numOfRows > 0) ? 1 : 0;
294,504✔
4816
  if (resRows > 0 || pRsp->completed) {
294,504✔
4817
    ctx->code = setZeroRowsResValue(ctx, &pRemote->val, resRows);
294,504✔
4818
    if (TSDB_CODE_SUCCESS == ctx->code) {
294,504✔
4819
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
294,504✔
4820
    }
4821

4822
    *fetchDone = true;
294,504✔
4823
  } else {
4824
    *fetchDone = false;
×
4825
  }
4826

4827
  if (IS_STREAM_MODE(pTaskInfo) && 0 == pRsp->numOfRows) {
294,504✔
4828
    ctx->code = setZeroRowsResValue(ctx, &pRemote->val, 0);
270✔
4829
    if (TSDB_CODE_SUCCESS == ctx->code) {
270✔
4830
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
270✔
4831
    }
4832

4833
    pRsp->completed = true;
270✔
4834
    *fetchDone = true;
270✔
4835
  }
4836

4837
  if (!(*fetchDone)) {
294,504✔
4838
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
×
4839
    if (TSDB_CODE_SUCCESS != code) {
×
4840
      ctx->code = code;
×
4841
      *fetchDone = true;
×
4842
    }
4843
  }
4844
}
294,504✔
4845

4846

4847

4848
int32_t setTableFromResBlock(STaskSubJobCtx* ctx, SRemoteTableNode* pRes, SSDataBlock** ppBlock) {
655,598✔
4849
  int32_t code = 0, lino = 0;
655,598✔
4850
  
4851
  if (NULL == pRes->pResBlks) {
655,598✔
4852
    pRes->pResBlks = taosArrayInit(10, POINTER_BYTES);
535,182✔
4853
    TSDB_CHECK_NULL(pRes->pResBlks, code, lino, _exit, terrno);
535,182✔
4854
  }
4855

4856
  TSDB_CHECK_NULL(taosArrayPush(pRes->pResBlks, ppBlock), code, lino, _exit, terrno);
1,311,196✔
4857

4858
_exit:
655,598✔
4859

4860
  if (code) {
655,598✔
4861
    blockDataDestroy(*ppBlock);    
×
4862
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4863
  }
4864

4865
  *ppBlock = NULL;
655,598✔
4866
  
4867
  return code;
655,598✔
4868
}
4869

4870

4871
void handleRemoteTableRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
658,142✔
4872
  SSDataBlock* pResBlock = NULL;
658,142✔
4873
  SRemoteTableNode* pRemote = (SRemoteTableNode*)pParam->pRes;
658,142✔
4874

4875
  qDebug("%s scl fetch remoteTable rsp received, subQIdx:%d, rows:%" PRId64 ", cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfRows, pRsp->numOfCols);
658,142✔
4876

4877
  if (pRsp->numOfRows > 0) {
658,142✔
4878
    ctx->code = createExprSubQResBlock(&pResBlock, NULL, pRemote->resCols);
655,598✔
4879
    if (TSDB_CODE_SUCCESS == ctx->code) {
655,598✔
4880
      ctx->code = extractSingleRspBlock(pRsp, pResBlock);
655,598✔
4881
    }
4882
    if (TSDB_CODE_SUCCESS == ctx->code) {
655,598✔
4883
      ctx->code = setTableFromResBlock(ctx, pRemote, &pResBlock);
655,598✔
4884
    }
4885
    if (TSDB_CODE_SUCCESS == ctx->code && pRsp->completed) {
655,598✔
4886
      pRemote->flag &= (~REMOTE_TABLE_FLAG_VAL_UNSET);
535,182✔
4887
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
535,182✔
4888
    }
4889

4890
    blockDataDestroy(pResBlock);  
655,598✔
4891
  } else if (0 == pRsp->numOfRows && pRsp->completed) {
2,544✔
4892
    pRemote->flag &= (~REMOTE_TABLE_FLAG_VAL_UNSET);
2,544✔
4893
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
2,544✔
4894
  }
4895

4896
  *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
658,142✔
4897

4898
  if (!(*fetchDone)) {
658,142✔
4899
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, 1);
120,416✔
4900
    if (TSDB_CODE_SUCCESS != code) {
120,416✔
4901
      ctx->code = code;
×
4902
      *fetchDone = true;
×
4903
    }
4904
  }
4905
}
658,142✔
4906

4907

4908
int32_t remoteFetchCallBack(void* param, SDataBuf* pMsg, int32_t code) {
40,965,393✔
4909
  taosMemoryFreeClear(pMsg->pEpSet);
40,965,393✔
4910

4911
  SScalarFetchParam* pParam = (SScalarFetchParam*)param;
40,964,813✔
4912
  STaskSubJobCtx* ctx = taosAcquireRef(fetchObjRefPool, pParam->subJobRefId);
40,964,813✔
4913
  if (ctx == NULL) {
40,964,808✔
4914
    qWarn("failed to acquire subJobCtx, since it may have been released, refId:%" PRIu64, pParam->subJobRefId);
6,380✔
4915
    taosMemoryFree(pMsg->pData);
6,380✔
4916
    return TSDB_CODE_SUCCESS;
6,380✔
4917
  }
4918

4919
  char idStr[64];
40,958,428✔
4920
  if (qDebugFlag & DEBUG_DEBUG) {
40,958,428✔
4921
    tstrncpy(idStr, ctx->idStr, sizeof(idStr));
1,391,460✔
4922
  }
4923
  
4924
  qDebug("%s subQIdx %d got rsp, blockIdx:%" PRId64 ", code:%d, rsp:%p", ctx->idStr, pParam->subQIdx, ctx->blockIdx, code, pMsg->pData);
40,958,428✔
4925

4926
  if (ctx->transporterId > 0) {
40,958,428✔
4927
    int32_t ret = asyncFreeConnById(ctx->rpcHandle, ctx->transporterId);
40,959,013✔
4928
    if (ret != 0) {
40,959,588✔
4929
      qDebug("%s failed to free subQ rpc handle, code:%s, subQIdx:%d", ctx->idStr, tstrerror(ret), pParam->subQIdx);
×
4930
    }
4931
    ctx->transporterId = -1;
40,959,588✔
4932
  }
4933

4934
  if (0 == code && NULL == pMsg->pData) {
40,959,588✔
4935
    qError("%s invalid rsp msg, msgType:%d, len:%d", ctx->idStr, pMsg->msgType, pMsg->len);
×
4936
    code = TSDB_CODE_QRY_INVALID_MSG;
×
4937
  }
4938

4939
  if (code == TSDB_CODE_SUCCESS) {
40,959,588✔
4940
    SRetrieveTableRsp* pRsp = pMsg->pData;
40,042,608✔
4941
    pRsp->numOfRows = htobe64(pRsp->numOfRows);
40,042,608✔
4942
    pRsp->compLen = htonl(pRsp->compLen);
40,042,028✔
4943
    pRsp->payloadLen = htonl(pRsp->payloadLen);
40,042,608✔
4944
    pRsp->numOfCols = htonl(pRsp->numOfCols);
40,042,033✔
4945
    pRsp->useconds = htobe64(pRsp->useconds);
40,042,608✔
4946
    pRsp->numOfBlocks = htonl(pRsp->numOfBlocks);
40,042,028✔
4947

4948
    qDebug("%s subQIdx %d blockIdx:%" PRIu64 " rsp detail, numOfBlocks:%d, numOfRows:%" PRId64 ", numOfCols:%" PRId64 ", completed:%d", 
40,042,028✔
4949
      ctx->idStr, pParam->subQIdx, ctx->blockIdx, pRsp->numOfBlocks, pRsp->numOfRows, pRsp->numOfCols, pRsp->completed);
4950

4951
    ctx->blockIdx++;
40,042,028✔
4952

4953
    switch (nodeType(pParam->pRes)) {
40,042,033✔
4954
      case QUERY_NODE_REMOTE_VALUE: {
3,553,870✔
4955
        bool fetchDone = false;
3,553,870✔
4956
        handleRemoteValueRes(pParam, ctx, pRsp, &fetchDone);
3,553,870✔
4957
        qDebug("%s subQIdx %d handle remote value, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
3,553,290✔
4958
        if (!fetchDone) {
3,553,870✔
4959
          goto _exit;
40,047✔
4960
        }
4961
        break;
3,513,823✔
4962
      }
4963
      case QUERY_NODE_REMOTE_VALUE_LIST: {
12,995,140✔
4964
        bool fetchDone = false;
12,995,140✔
4965
        handleRemoteValueListRes(pParam, ctx, pRsp, &fetchDone);
12,995,715✔
4966
        qDebug("%s subQIdx %d handle remote value list finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
12,995,715✔
4967
        if (!fetchDone) {
12,995,715✔
4968
          goto _exit;
663,097✔
4969
        }
4970
        break;
12,332,618✔
4971
      }  
4972
      case QUERY_NODE_REMOTE_ROW: {
22,540,377✔
4973
        bool fetchDone = false;
22,540,377✔
4974
        handleRemoteRowRes(pParam, ctx, pRsp, &fetchDone);
22,540,377✔
4975
        qDebug("%s subQIdx %d handle remote row finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
22,540,377✔
4976
        if (!fetchDone) {
22,540,377✔
4977
          goto _exit;
540✔
4978
        }
4979
        break;
22,539,837✔
4980
      }
4981
      case QUERY_NODE_REMOTE_ZERO_ROWS: {
294,504✔
4982
        bool fetchDone = false;
294,504✔
4983
        handleRemoteZeroRowsRes(pParam, ctx, pRsp, &fetchDone);
294,504✔
4984
        qDebug("%s subQIdx %d handle remote zeroRows finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
294,504✔
4985
        if (!fetchDone) {
294,504✔
4986
          goto _exit;
×
4987
        }
4988
        break;
294,504✔
4989
      }
4990
      case QUERY_NODE_REMOTE_TABLE: {
658,142✔
4991
        bool fetchDone = false;
658,142✔
4992
        handleRemoteTableRes(pParam, ctx, pRsp, &fetchDone);
658,142✔
4993
        qDebug("%s subQIdx %d handle remote table finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
658,142✔
4994
        if (!fetchDone) {
658,142✔
4995
          goto _exit;
120,416✔
4996
        }
4997
        break;
537,726✔
4998
      }
4999
      default:
×
5000
        qError("%s invalid scl fetch res node %d, subQIdx:%d", ctx->idStr, nodeType(pParam->pRes), pParam->subQIdx);
×
5001
        ctx->code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
5002
        break;
×
5003
    }
5004
  } else {
5005
    ctx->code = rpcCvtErrCode(code);
916,980✔
5006
    if (ctx->code != code) {
916,980✔
5007
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s, cvted error: %s", ctx->idStr, pParam->subQIdx,
×
5008
             tstrerror(code), tstrerror(ctx->code));
5009
    } else {
5010
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s", ctx->idStr, pParam->subQIdx, tstrerror(code));
916,400✔
5011
    }
5012
  }
5013

5014
  qDebug("%s subQIdx %d sem_post subQ ready", ctx->idStr, pParam->subQIdx);
40,135,488✔
5015
  
5016
  code = tsem_post(&ctx->ready);
40,135,488✔
5017
  if (code != TSDB_CODE_SUCCESS) {
40,135,488✔
5018
    qError("failed to invoke post when scl fetch rsp is ready, code:%s", tstrerror(code));
×
5019
  }
5020

5021
_exit:
40,959,588✔
5022

5023
  code = taosReleaseRef(fetchObjRefPool, pParam->subJobRefId);
40,960,168✔
5024
  if (code != TSDB_CODE_SUCCESS) {
40,960,168✔
5025
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
5026
  }
5027

5028
  taosMemoryFree(pMsg->pData);
40,960,168✔
5029

5030
  return code;
40,959,588✔
5031
}
5032

5033
#define STREAM_TASK_REPLICA_NUM 5
5034
#define STREAM_TASK_DEPLOY_NUM 3
5035

5036
int32_t sendFetchRemoteNodeReq(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes, bool reset) {
40,965,978✔
5037
  int32_t          code = TSDB_CODE_SUCCESS;
40,965,978✔
5038
  int32_t          lino = 0;
40,965,978✔
5039
  int32_t          innerIdx = 0;
40,965,978✔
5040
  SExecTaskInfo*   pTaskInfo = ctx->pTaskInfo;
40,965,978✔
5041
  bool             needStreamPesudoFuncVals = false;
40,964,813✔
5042

5043
  if (IS_STREAM_MODE(pTaskInfo)) {
40,964,813✔
5044
    innerIdx = pTaskInfo->pStreamRuntimeInfo->execId / STREAM_TASK_REPLICA_NUM;
455,473✔
5045
    subQIdx = subQIdx * STREAM_TASK_DEPLOY_NUM + innerIdx;
455,473✔
5046
  }
5047

5048
  SDownstreamSourceNode* pSource = (SDownstreamSourceNode*)taosArrayGetP(ctx->subEndPoints, subQIdx);
40,958,443✔
5049

5050
  SResFetchReq req = {0};
40,964,233✔
5051
  req.header.vgId = pSource->addr.nodeId;
40,964,233✔
5052
  req.sId = pSource->sId;
40,965,973✔
5053
  req.clientId = pSource->clientId;
40,968,868✔
5054
  req.taskId = pSource->taskId;
40,961,328✔
5055
  req.srcTaskId = ctx->taskId;
40,955,543✔
5056
  req.blockIdx = ctx->blockIdx;
40,954,958✔
5057
  req.queryId = ctx->queryId;
40,967,133✔
5058
  req.execId = pSource->execId;
40,957,288✔
5059

5060
  if (IS_STREAM_MODE(pTaskInfo)) {
40,956,708✔
5061
    req.queryId = pSource->clientId;
455,473✔
5062
    req.execId = pTaskInfo->pStreamRuntimeInfo->execId;
455,473✔
5063
    req.pStRtFuncInfo = &pTaskInfo->pStreamRuntimeInfo->funcInfo;
455,473✔
5064
    req.reset = reset;
455,473✔
5065

5066
    needStreamPesudoFuncVals = true;
455,473✔
5067
    subQIdx = (subQIdx - innerIdx) / STREAM_TASK_DEPLOY_NUM;
455,473✔
5068
  }
5069

5070
  int32_t msgSize = tSerializeSResFetchReq(NULL, 0, &req, needStreamPesudoFuncVals, false);
40,954,973✔
5071
  if (msgSize < 0) {
40,964,148✔
5072
    return msgSize;
×
5073
  }
5074

5075
  void* msg = taosMemoryCalloc(1, msgSize);
40,964,148✔
5076
  if (NULL == msg) {
40,955,039✔
5077
    return terrno;
×
5078
  }
5079

5080
  msgSize = tSerializeSResFetchReq(msg, msgSize, &req, needStreamPesudoFuncVals, false);
40,955,039✔
5081
  if (msgSize < 0) {
40,956,864✔
5082
    taosMemoryFree(msg);
×
5083
    return msgSize;
×
5084
  }
5085

5086
  qDebug("%s scl build fetch msg and send to nodeId:%d, ep:%s, clientId:0x%" PRIx64 " taskId:0x%" PRIx64
40,956,864✔
5087
         ", execId:%d, blockIdx:%" PRId64,
5088
         ctx->idStr, pSource->addr.nodeId, pSource->addr.epSet.eps[0].fqdn, pSource->clientId,
5089
         pSource->taskId, pSource->execId, req.blockIdx);
5090

5091
  // send the fetch remote task result reques
5092
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
40,963,563✔
5093
  if (NULL == pMsgSendInfo) {
40,956,703✔
5094
    taosMemoryFreeClear(msg);
×
5095
    qError("%s prepare message %d failed", ctx->idStr, (int32_t)sizeof(SMsgSendInfo));
×
5096
    return terrno;
×
5097
  }
5098

5099
  SScalarFetchParam* param = taosMemoryMalloc(sizeof(SScalarFetchParam));
40,956,703✔
5100
  if (NULL == param) {
40,957,198✔
5101
    taosMemoryFreeClear(msg);
×
5102
    taosMemoryFreeClear(pMsgSendInfo);
×
5103
    qError("%s prepare param %d failed", ctx->idStr, (int32_t)sizeof(SScalarFetchParam));
×
5104
    return terrno;
×
5105
  }
5106

5107
  if (ctx->code) {
40,957,198✔
5108
    qError("task has been killed, error:%s", tstrerror(ctx->code));
×
5109
    taosMemoryFree(param);
×
5110
    taosMemoryFreeClear(msg);
×
5111
    taosMemoryFreeClear(pMsgSendInfo);
×
5112
    code = ctx->code;
×
5113
    goto _end;
×
5114
  }
5115
  
5116
  param->subQIdx = subQIdx;
40,952,088✔
5117
  param->pRes = pRes;
40,959,028✔
5118
  param->subJobRefId = ctx->subJobRefId;
40,960,753✔
5119

5120
  pMsgSendInfo->param = param;
40,962,398✔
5121
  pMsgSendInfo->paramFreeFp = taosAutoMemoryFree;
40,954,403✔
5122
  pMsgSendInfo->msgInfo.pData = msg;
40,951,963✔
5123
  pMsgSendInfo->msgInfo.len = msgSize;
40,963,083✔
5124
  pMsgSendInfo->msgType = pSource->fetchMsgType;
40,961,918✔
5125
  pMsgSendInfo->fp = remoteFetchCallBack;
40,953,813✔
5126
  pMsgSendInfo->requestId = ctx->queryId;
40,955,543✔
5127

5128
  code = asyncSendMsgToServer(ctx->rpcHandle, &pSource->addr.epSet, &ctx->transporterId, pMsgSendInfo);
40,956,123✔
5129
  QUERY_CHECK_CODE(code, lino, _end);
40,965,393✔
5130
      
5131
_end:
40,965,393✔
5132

5133
  if (code != TSDB_CODE_SUCCESS) {
40,965,393✔
5134
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
×
5135
  }
5136
  
5137
  return code;
40,968,868✔
5138
}
5139

5140
int32_t fetchRemoteNodeImpl(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes) {
40,141,878✔
5141
  int32_t          code = TSDB_CODE_SUCCESS;
40,141,878✔
5142
  int32_t          lino = 0;
40,141,878✔
5143

5144
  ctx->blockIdx = 0;
40,141,878✔
5145

5146
  code = sendFetchRemoteNodeReq(ctx, subQIdx, pRes, true);
40,140,718✔
5147
  QUERY_CHECK_CODE(code, lino, _end);
40,145,348✔
5148

5149
  code = qSemWait(ctx->pTaskInfo, &ctx->ready);
40,145,348✔
5150
  if (isTaskKilled(ctx->pTaskInfo)) {
40,145,928✔
5151
    code = getTaskCode(ctx->pTaskInfo);
19,140✔
5152
  } else {
5153
    code = ctx->code;
40,126,213✔
5154
  }
5155
      
5156
_end:
40,145,353✔
5157

5158
  if (code != TSDB_CODE_SUCCESS) {
40,145,353✔
5159
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
1,956,768✔
5160
  }
5161
  return code;
40,145,353✔
5162
}
5163

5164
int32_t remoteNodeCopy(SNode* pSrc, SNode* pDst) {
124,730✔
5165
  int32_t code = 0, lino = 0;
124,730✔
5166
  
5167
  switch (nodeType(pSrc)) {
124,730✔
5168
    case QUERY_NODE_VALUE:
89,080✔
5169
      TAOS_CHECK_EXIT(valueNodeCopy((SValueNode*)pSrc, &((SRemoteValueNode*)pDst)->val));
89,080✔
5170
      ((SRemoteValueNode*)pDst)->val.node.type = QUERY_NODE_VALUE;
89,080✔
5171
      break;
89,080✔
5172
    case QUERY_NODE_REMOTE_VALUE_LIST: {
26,450✔
5173
      SRemoteValueListNode* pDstNode = (SRemoteValueListNode*)pDst;
26,450✔
5174
      memcpy(pDst, pSrc, sizeof(SRemoteValueListNode));
26,450✔
5175
      pDstNode->hashAllocated = false;      
26,450✔
5176
      break;
26,450✔
5177
    } 
5178
    case QUERY_NODE_REMOTE_ROW: {
9,200✔
5179
      SRemoteRowNode* pRemote = (SRemoteRowNode*)pDst;
9,200✔
5180
      TAOS_CHECK_EXIT(valueNodeCopy((SValueNode*)pSrc, &pRemote->val));
9,200✔
5181
      pRemote->valSet = true;
9,200✔
5182
      pRemote->hasValue = ((SRemoteRowNode*)pSrc)->hasValue;
9,200✔
5183
      pRemote->hasNull = ((SRemoteRowNode*)pSrc)->hasNull;
9,200✔
5184
      break;
9,200✔
5185
    }
5186
    case QUERY_NODE_REMOTE_TABLE: {
×
5187
      SRemoteTableNode* pDstNode = (SRemoteTableNode*)pDst;
×
5188
      SRemoteTableNode* pSrcNode = (SRemoteTableNode*)pSrc;
×
5189
      pDstNode->pResBlks = pSrcNode->pResBlks;
×
5190
      pDstNode->flag = 0;      
×
5191
      break;
×
5192
    }
5193
    case QUERY_NODE_REMOTE_ZERO_ROWS: {
×
5194
      SRemoteZeroRowsNode* pRemote = (SRemoteZeroRowsNode*)pDst;
×
5195
      TAOS_CHECK_EXIT(valueNodeCopy((SValueNode*)pSrc, &pRemote->val));
×
5196
      pRemote->val.node.type = QUERY_NODE_VALUE;
×
5197
      break;
×
5198
    }
5199
    default:
×
5200
      break;
×
5201
  }
5202

5203
_exit:
124,730✔
5204

5205
  if (code) {
124,730✔
5206
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5207
  }
5208

5209
  return code;
124,730✔
5210
}
5211

5212
int32_t qFetchRemoteNode(void* pCtx, int32_t subQIdx, SNode* pRes) {
40,266,033✔
5213
  STaskSubJobCtx*  ctx = (STaskSubJobCtx*)pCtx;
40,266,033✔
5214
  int32_t code = 0, lino = 0;
40,266,033✔
5215
  int32_t       subEndPoinsNum = taosArrayGetSize(ctx->subEndPoints);
40,266,033✔
5216
  if (subQIdx >= subEndPoinsNum) {
40,261,388✔
5217
    qError("%s invalid subQIdx %d, subEndPointsNum:%d", ctx->idStr, subQIdx, subEndPoinsNum);
×
5218
    return TSDB_CODE_QRY_SUBQ_NOT_FOUND;
×
5219
  }
5220

5221
  SNode** ppRes = taosArrayGet(ctx->subResNodes, subQIdx);
40,261,388✔
5222
  if (NULL == *ppRes) {
40,264,283✔
5223
    TAOS_CHECK_EXIT(fetchRemoteNodeImpl(ctx, subQIdx, pRes));
40,139,553✔
5224
    *ppRes = pRes;
38,188,585✔
5225
  } else {
5226
    TAOS_CHECK_EXIT(remoteNodeCopy(*ppRes, pRes));
124,730✔
5227
  }
5228

5229
_exit:
124,730✔
5230

5231
  if (code) {
40,270,658✔
5232
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
1,956,768✔
5233
  } else {
5234
    qDebug("%s %s subQIdx %d succeed", ctx->idStr, __func__, subQIdx);
38,313,890✔
5235
  }
5236

5237
  return code;
40,270,658✔
5238
}
5239

5240
int32_t findDataBlockColIndexBySlotId(const SSDataBlock* pBlock, int32_t slotId) {
211,319,244✔
5241
  if (NULL == pBlock || NULL == pBlock->pDataBlock) {
211,319,244✔
5242
    return -1;
×
5243
  }
5244

5245
  int32_t numOfCols = (int32_t)taosArrayGetSize(pBlock->pDataBlock);
211,318,846✔
5246
  if (slotId >= 0 && slotId < numOfCols) {
211,319,244✔
5247
    SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, slotId);
211,319,244✔
5248
    if (NULL != pCol && pCol->info.slotId == slotId) {
211,319,244✔
5249
      return slotId;
211,319,642✔
5250
    }
5251
  }
5252

5253
  for (int32_t i = 0; i < numOfCols; ++i) {
×
5254
    SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, i);
×
5255
    if (NULL != pCol && pCol->info.slotId == slotId) {
×
5256
      return i;
×
5257
    }
5258
  }
5259

5260
  return -1;
×
5261
}
5262

5263
SColumnInfoData* getDataBlockColBySlotId(const SSDataBlock* pBlock, int32_t slotId, int32_t* pIndex) {
211,319,244✔
5264
  int32_t index = findDataBlockColIndexBySlotId(pBlock, slotId);
211,319,244✔
5265
  if (NULL != pIndex) {
211,319,642✔
5266
    *pIndex = index;
×
5267
  }
5268
  return (index >= 0) ? taosArrayGet(pBlock->pDataBlock, index) : NULL;
211,319,642✔
5269
}
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