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

taosdata / TDengine / #5005

26 Mar 2026 12:51PM UTC coverage: 72.152% (-0.2%) from 72.338%
#5005

push

travis-ci

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

512 of 851 new or added lines in 47 files covered. (60.16%)

6189 existing lines in 147 files now uncovered.

253282 of 351039 relevant lines covered (72.15%)

132156710.33 hits per line

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

76.39
/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

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

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

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

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

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

72
void initResultRowInfo(SResultRowInfo* pResultRowInfo) {
488,392,785✔
73
  pResultRowInfo->size = 0;
488,392,785✔
74
  pResultRowInfo->cur.pageId = -1;
488,425,296✔
75
}
488,444,039✔
76

77
void closeResultRow(SResultRow* pResultRow) { pResultRow->closed = true; }
5,624,011✔
78

79
void resetResultRow(SResultRow* pResultRow, size_t entrySize) {
934,860,023✔
80
  pResultRow->numOfRows = 0;
934,860,023✔
81
  pResultRow->nOrigRows = 0;
934,861,903✔
82
  pResultRow->closed = false;
934,860,728✔
83
  pResultRow->endInterp = false;
934,862,138✔
84
  pResultRow->startInterp = false;
934,861,668✔
85

86
  if (entrySize > 0) {
934,861,198✔
87
    memset(pResultRow->pEntryInfo, 0, entrySize);
934,861,903✔
88
  }
89
}
934,860,855✔
90

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

96
size_t getResultRowSize(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
292,500,712✔
97
  int32_t rowSize = (numOfOutput * sizeof(SResultRowEntryInfo)) + sizeof(SResultRow);
292,500,712✔
98

99
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,252,608,724✔
100
    rowSize += pCtx[i].resDataInfo.interBufSize;
960,133,272✔
101
  }
102

103
  return rowSize;
292,475,452✔
104
}
105

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

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

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

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

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

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

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

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

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

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

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

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

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

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

219
void cleanupGroupResInfo(SGroupResInfo* pGroupResInfo) {
136,747,691✔
220
  taosMemoryFreeClear(pGroupResInfo->pBuf);
136,747,691✔
221
  if (pGroupResInfo->freeItem) {
136,759,923✔
222
    //    taosArrayDestroy(pGroupResInfo->pRows);
223
    taosArrayDestroyEx(pGroupResInfo->pRows, freeEx);
×
224
    pGroupResInfo->freeItem = false;
×
225
    pGroupResInfo->pRows = NULL;
×
226
  } else {
227
    taosArrayDestroy(pGroupResInfo->pRows);
136,760,543✔
228
    pGroupResInfo->pRows = NULL;
136,753,929✔
229
  }
230
  pGroupResInfo->index = 0;
136,755,009✔
231
  pGroupResInfo->delIndex = 0;
136,742,256✔
232
}
136,762,395✔
233

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

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

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

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

254
int32_t initGroupedResultInfo(SGroupResInfo* pGroupResInfo, SSHashObj* pHashmap, int32_t order) {
104,905,512✔
255
  int32_t code = TSDB_CODE_SUCCESS;
104,905,512✔
256
  int32_t lino = 0;
104,905,512✔
257
  if (pGroupResInfo->pRows != NULL) {
104,905,512✔
258
    taosArrayDestroy(pGroupResInfo->pRows);
4,111,716✔
259
  }
260
  if (pGroupResInfo->pBuf) {
104,913,233✔
261
    taosMemoryFree(pGroupResInfo->pBuf);
4,112,478✔
262
    pGroupResInfo->pBuf = NULL;
4,110,573✔
263
  }
264

265
  // extract the result rows information from the hash map
266
  int32_t size = tSimpleHashGetSize(pHashmap);
104,902,140✔
267

268
  void* pData = NULL;
104,906,509✔
269
  pGroupResInfo->pRows = taosArrayInit(size, POINTER_BYTES);
104,906,509✔
270
  QUERY_CHECK_NULL(pGroupResInfo->pRows, code, lino, _end, terrno);
104,900,784✔
271

272
  size_t  keyLen = 0;
104,909,626✔
273
  int32_t iter = 0;
104,909,606✔
274
  int64_t bufLen = 0, offset = 0;
104,901,327✔
275

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

282
  pGroupResInfo->pBuf = taosMemoryMalloc(bufLen);
104,880,552✔
283
  QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
104,902,650✔
284

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

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

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

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

300
  if (order == TSDB_ORDER_ASC || order == TSDB_ORDER_DESC) {
104,843,031✔
301
    __compar_fn_t fn = (order == TSDB_ORDER_ASC) ? resultrowComparAsc : resultrowComparDesc;
6,046,212✔
302
    size = POINTER_BYTES;
6,046,212✔
303
    taosSort(pGroupResInfo->pRows->pData, taosArrayGetSize(pGroupResInfo->pRows), size, fn);
6,046,212✔
304
  }
305

306
  pGroupResInfo->index = 0;
104,843,151✔
307

308
_end:
104,905,464✔
309
  if (code != TSDB_CODE_SUCCESS) {
104,913,011✔
310
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
311
  }
312
  return code;
104,913,011✔
313
}
314

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

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

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

331
  return pGroupResInfo->index < taosArrayGetSize(pGroupResInfo->pRows);
415,348,799✔
332
}
333

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

339
  return (int32_t)taosArrayGetSize(pGroupResInfo->pRows);
220,130,243✔
340
}
341

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

345
  if (pNodeList != NULL) {
53,386,776✔
346
    numOfCols = LIST_LENGTH(pNodeList);
53,165,204✔
347
  } else {
348
    numOfCols = 0;
221,572✔
349
  }
350

351
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
53,387,051✔
352
  if (pList == NULL) {
53,369,523✔
353
    return pList;
×
354
  }
355

356
  for (int32_t i = 0; i < numOfCols; ++i) {
121,132,072✔
357
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
67,751,499✔
358
    if (!pSortKey) {
67,752,709✔
359
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
360
      taosArrayDestroy(pList);
×
361
      pList = NULL;
×
362
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
363
      break;
×
364
    }
365
    SBlockOrderInfo bi = {0};
67,752,709✔
366
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
67,756,328✔
367
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);
67,754,852✔
368

369
    if (nodeType(pSortKey->pExpr) != QUERY_NODE_COLUMN) {
67,750,118✔
370
      qError("invalid order by expr type:%d", nodeType(pSortKey->pExpr));
×
371
      taosArrayDestroy(pList);
×
372
      pList = NULL;
×
373
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
374
      break;
×
375
    }
376
    
377
    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
67,740,919✔
378
    bi.slotId = pColNode->slotId;
67,758,985✔
379
    void* tmp = taosArrayPush(pList, &bi);
67,762,549✔
380
    if (!tmp) {
67,762,549✔
381
      taosArrayDestroy(pList);
×
382
      pList = NULL;
×
383
      break;
×
384
    }
385
  }
386

387
  return pList;
53,380,573✔
388
}
389

390
SSDataBlock* createDataBlockFromDescNode(void* p) {
766,149,951✔
391
  SDataBlockDescNode* pNode = (SDataBlockDescNode*)p;
766,149,951✔
392
  int32_t      numOfCols = LIST_LENGTH(pNode->pSlots);
766,149,951✔
393
  SSDataBlock* pBlock = NULL;
766,292,838✔
394
  int32_t      code = TSDB_CODE_SUCCESS;
766,279,372✔
395
  int32_t      lino = 0;
766,279,372✔
396

397
  code = createDataBlock(&pBlock);
766,279,372✔
398
  QUERY_CHECK_CODE(code, lino, _return);
766,096,502✔
399

400
  pBlock->info.id.blockId = pNode->dataBlockId;
766,096,502✔
401
  pBlock->info.calWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
766,185,822✔
402
  pBlock->info.watermark = INT64_MIN;
766,154,330✔
403

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

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

419
  if (pBlock != NULL && i != numOfCols) {
766,438,521✔
420
    code = TSDB_CODE_INVALID_PARA;
×
421
    QUERY_CHECK_CODE(code, lino, _return);
×
422
  }
423

424
  return pBlock;
766,438,521✔
425
_return:
75,496✔
426
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
427
  blockDataDestroy(pBlock);
×
428
  terrno = code;
×
429
  return NULL;
×
430
}
431

432
int32_t prepareDataBlockBuf(SSDataBlock* pDataBlock, SColMatchInfo* pMatchInfo) {
259,914,843✔
433
  SDataBlockInfo* pBlockInfo = &pDataBlock->info;
259,914,843✔
434

435
  for (int32_t i = 0; i < taosArrayGetSize(pMatchInfo->pList); ++i) {
1,170,192,268✔
436
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
926,205,635✔
437
    if (!pItem) {
926,029,507✔
438
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
439
      return terrno;
×
440
    }
441

442
    if (pItem->isPk) {
926,029,507✔
443
      SColumnInfoData* pInfoData = taosArrayGet(pDataBlock->pDataBlock, pItem->dstSlotId);
16,089,752✔
444
      if (!pInfoData) {
15,658,420✔
445
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
446
        return terrno;
×
447
      }
448
      pBlockInfo->pks[0].type = pInfoData->info.type;
15,658,420✔
449
      pBlockInfo->pks[1].type = pInfoData->info.type;
15,695,947✔
450

451
      // allocate enough buffer size, which is pInfoData->info.bytes
452
      if (IS_VAR_DATA_TYPE(pItem->dataType.type)) {
15,676,242✔
453
        pBlockInfo->pks[0].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
4,932,266✔
454
        if (pBlockInfo->pks[0].pData == NULL) {
4,909,268✔
455
          return terrno;
×
456
        }
457

458
        pBlockInfo->pks[1].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
4,914,563✔
459
        if (pBlockInfo->pks[1].pData == NULL) {
4,912,525✔
460
          taosMemoryFreeClear(pBlockInfo->pks[0].pData);
×
461
          return terrno;
×
462
        }
463

464
        pBlockInfo->pks[0].nData = pInfoData->info.bytes;
4,919,145✔
465
        pBlockInfo->pks[1].nData = pInfoData->info.bytes;
4,917,810✔
466
      }
467

468
      break;
15,686,992✔
469
    }
470
  }
471

472
  return TSDB_CODE_SUCCESS;
259,738,620✔
473
}
474

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

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

498
    res->translate = true;
731,156✔
499
    res->node.resType = pSColumnNode->node.resType;
731,156✔
500

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

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

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

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

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

559
  return DEAL_RES_CONTINUE;
2,559,046✔
560
}
561

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

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

572
    return TSDB_CODE_SUCCESS;
×
573
  }
574

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

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

599
    return code;
×
600
  }
601

602
  SValueNode* pValue = (SValueNode*)pNew;
365,578✔
603
  *pQualified = pValue->datum.b;
365,578✔
604

605
  nodesDestroyNode(pNew);
365,578✔
606
  return TSDB_CODE_SUCCESS;
365,578✔
607
}
608

609
static EDealRes getColumn(SNode** pNode, void* pContext) {
62,100,663✔
610
  tagFilterAssist* pData = (tagFilterAssist*)pContext;
62,100,663✔
611
  SColumnNode*     pSColumnNode = NULL;
62,100,663✔
612
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
62,110,147✔
613
    pSColumnNode = *(SColumnNode**)pNode;
21,425,740✔
614
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
40,702,407✔
615
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
1,110,465✔
616
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
1,110,465✔
617
      pData->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
740,338✔
618
      if (NULL == pSColumnNode) {
741,520✔
619
        return DEAL_RES_ERROR;
×
620
      }
621
      pSColumnNode->colId = -1;
741,520✔
622
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
741,520✔
623
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
741,520✔
624
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
740,929✔
625
      nodesDestroyNode(*pNode);
741,520✔
626
      *pNode = (SNode*)pSColumnNode;
740,929✔
627
    } else {
628
      return DEAL_RES_CONTINUE;
368,945✔
629
    }
630
  } else {
631
    return DEAL_RES_CONTINUE;
39,581,856✔
632
  }
633

634
  void* data = taosHashGet(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
22,160,598✔
635
  if (!data) {
22,163,509✔
636
    int32_t tempRes =
637
        taosHashPut(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
19,132,818✔
638
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
19,127,134✔
639
      return DEAL_RES_ERROR;
×
640
    }
641
    pSColumnNode->slotId = pData->index++;
19,127,134✔
642
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
19,126,247✔
643
                         .type = pSColumnNode->node.resType.type,
19,126,325✔
644
                         .bytes = pSColumnNode->node.resType.bytes,
19,130,415✔
645
                         .pk = pSColumnNode->isPk};
19,121,292✔
646
#if TAG_FILTER_DEBUG
647
    qDebug("tagfilter build column info, slotId:%d, colId:%d, type:%d", pSColumnNode->slotId, cInfo.colId, cInfo.type);
648
#endif
649
    void* tmp = taosArrayPush(pData->cInfoList, &cInfo);
19,126,000✔
650
    if (!tmp) {
19,141,278✔
651
      return DEAL_RES_ERROR;
×
652
    }
653
  } else {
654
    SColumnNode* col = *(SColumnNode**)data;
3,030,691✔
655
    pSColumnNode->slotId = col->slotId;
3,030,189✔
656
  }
657

658
  return DEAL_RES_CONTINUE;
22,155,778✔
659
}
660

661
static int32_t createResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
17,794,202✔
662
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
17,794,202✔
663
  if (pColumnData == NULL) {
17,796,062✔
664
    return terrno;
×
665
  }
666

667
  pColumnData->info.type = pType->type;
17,796,062✔
668
  pColumnData->info.bytes = pType->bytes;
17,790,386✔
669
  pColumnData->info.scale = pType->scale;
17,788,369✔
670
  pColumnData->info.precision = pType->precision;
17,785,349✔
671

672
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
17,792,759✔
673
  if (code != TSDB_CODE_SUCCESS) {
17,783,954✔
674
    terrno = code;
×
675
    releaseColInfoData(pColumnData);
×
676
    return terrno;
×
677
  }
678

679
  pParam->columnData = pColumnData;
17,783,954✔
680
  pParam->colAlloced = true;
17,790,681✔
681
  return TSDB_CODE_SUCCESS;
17,781,021✔
682
}
683

684
static void releaseColInfoData(void* pCol) {
4,901,138✔
685
  if (pCol) {
4,901,138✔
686
    SColumnInfoData* col = (SColumnInfoData*)pCol;
4,901,712✔
687
    colDataDestroy(col);
4,901,712✔
688
    taosMemoryFree(col);
4,901,712✔
689
  }
690
}
4,900,006✔
691

692
void freeItem(void* p) {
209,686,204✔
693
  STUidTagInfo* pInfo = p;
209,686,204✔
694
  if (pInfo->pTagVal != NULL) {
209,686,204✔
695
    taosMemoryFree(pInfo->pTagVal);
209,243,773✔
696
  }
697
}
209,682,581✔
698

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

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

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

726
    (void)memcpy(pStart, &entry->colId, sizeof(col_id_t));
23,184✔
727
    pStart += sizeof(col_id_t);
23,184✔
728

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

774
  return TSDB_CODE_SUCCESS;
11,592✔
775
}
776

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

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

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

802
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
11,592✔
803
    extractTagDataEntry((SOperatorNode*)pTagCond, pIdWithVal);
×
804
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
11,592✔
805
    SNode* pChild = NULL;
11,592✔
806
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
34,776✔
807
      extractTagDataEntry((SOperatorNode*)pChild, pIdWithVal);
23,184✔
808
    }
809
  }
810

811
  taosArraySort(pIdWithVal, compareTagDataEntry);
11,592✔
812

813
  return TSDB_CODE_SUCCESS;
11,592✔
814
}
815

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

821
  char*   payload = NULL;
11,592✔
822
  int32_t len = 0;
11,592✔
823
  int32_t code = TSDB_CODE_SUCCESS;
11,592✔
824
  int32_t lino = 0;
11,592✔
825

826
  SArray* pIdWithVal = taosArrayInit(TARRAY_MIN_SIZE, sizeof(STagDataEntry));
11,592✔
827
  code = extractTagFilterTagDataEntries(pTagCond, pIdWithVal);
11,592✔
828
  QUERY_CHECK_CODE(code, lino, _end);
11,592✔
829
  for (int32_t i = 0; i < taosArrayGetSize(pIdWithVal); ++i) {
34,776✔
830
    STagDataEntry* pEntry = taosArrayGet(pIdWithVal, i);
23,184✔
831
    len += sizeof(col_id_t) + pEntry->bytes;
23,184✔
832
  }
833
  code = buildTagDataEntryKey(pIdWithVal, &payload, len);
11,592✔
834
  QUERY_CHECK_CODE(code, lino, _end);
11,592✔
835

836
  tMD5Init(pContext);
11,592✔
837
  tMD5Update(pContext, (uint8_t*)payload, (uint32_t)len);
11,592✔
838
  tMD5Final(pContext);
11,592✔
839

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

850
static int32_t genTagFilterDigest(const SNode* pTagCond, T_MD5_CTX* pContext) {
34,132✔
851
  if (pTagCond == NULL) {
34,132✔
852
    return TSDB_CODE_SUCCESS;
32,200✔
853
  }
854

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

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

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

873
  taosMemoryFree(payload);
1,932✔
874
  return TSDB_CODE_SUCCESS;
1,932✔
875
}
876

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

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

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

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

904
int32_t qGetColumnsFromNodeList(void* data, bool isList, SArray** pColList) {
17,381,039✔
905
  int32_t code = TSDB_CODE_SUCCESS;
17,381,039✔
906
  tagFilterAssist ctx = {0};
17,381,039✔
907
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
17,382,793✔
908
  if (ctx.colHash == NULL) {
17,380,547✔
909
    code = terrno;
×
910
    goto end;
×
911
  }
912

913
  ctx.index = 0;
17,380,547✔
914
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
17,380,547✔
915
  if (ctx.cInfoList == NULL) {
17,383,866✔
916
    code = terrno;
7,881✔
917
    goto end;
×
918
  }
919

920
  if (isList) {
17,375,985✔
921
    SNode* pNode = NULL;
4,483,703✔
922
    FOREACH(pNode, (SNodeList*)data) {
9,384,803✔
923
      nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
4,902,137✔
924
      if (TSDB_CODE_SUCCESS != ctx.code) {
4,900,853✔
925
        code = ctx.code;
×
926
        goto end;
×
927
      }
928
      REPLACE_NODE(pNode);
4,900,853✔
929
    }
930
  } else {
931
    SNode* pNode = (SNode*)data;
12,892,282✔
932
    nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
12,892,933✔
933
    if (TSDB_CODE_SUCCESS != ctx.code) {
12,896,793✔
934
      code = ctx.code;
×
935
      goto end;
×
936
    }
937
  }
938
  
939
  if (pColList != NULL) *pColList = ctx.cInfoList;
17,364,718✔
940
  ctx.cInfoList = NULL;
17,374,532✔
941

942
end:
17,390,600✔
943
  taosHashCleanup(ctx.colHash);
17,377,927✔
944
  taosArrayDestroy(ctx.cInfoList);
17,359,169✔
945
  return code;
17,363,288✔
946
}
947

948
static int32_t buildGroupInfo(SColumnInfoData* pValue, int32_t i, SArray* gInfo) {
694,868✔
949
  int32_t code = TSDB_CODE_SUCCESS;
694,868✔
950
  SStreamGroupValue* v = taosArrayReserve(gInfo, 1);
694,868✔
951
  if (v == NULL) {
694,598✔
952
    code = terrno;
×
953
    goto end;
×
954
  }
955
  if (colDataIsNull_s(pValue, i)) {
1,389,196✔
956
    v->isNull = true;
7,360✔
957
  } else {
958
    v->isNull = false;
687,238✔
959
    char* data = colDataGetData(pValue, i);
687,489✔
960
    if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
687,545✔
961
      if (tTagIsJson(data)) {
×
962
        code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
×
963
        goto end;
×
964
      }
965
      if (tTagIsJsonNull(data)) {
×
966
        v->isNull = true;
×
967
        goto end;
×
968
      }
969
      int32_t len = getJsonValueLen(data);
×
970
      v->data.type = pValue->info.type;
×
971
      v->data.nData = len;
×
972
      v->data.pData = taosMemoryCalloc(1, len + 1);
×
973
      if (v->data.pData == NULL) {
×
974
        code = terrno;
×
975
        goto end;
×
976
      }
977
      memcpy(v->data.pData, data, len);
×
978
      qDebug("buildGroupInfo:%d add json data len:%d, data:%s", i, len, (char*)v->data.pData);
×
979
    } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
687,066✔
980
      if (varDataTLen(data) > pValue->info.bytes) {
442,441✔
981
        code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
982
        goto end;
×
983
      }
984
      v->data.type = pValue->info.type;
442,648✔
985
      v->data.nData = varDataLen(data);
442,636✔
986
      v->data.pData = taosMemoryCalloc(1, varDataLen(data) + 1);
442,868✔
987
      if (v->data.pData == NULL) {
443,115✔
988
        code = terrno;
×
989
        goto end;
×
990
      }
991
      memcpy(v->data.pData, varDataVal(data), varDataLen(data));
442,880✔
992
      qDebug("buildGroupInfo:%d add var data type:%d, len:%d, data:%s", i, pValue->info.type, varDataLen(data), (char*)v->data.pData);
442,382✔
993
    } else if (pValue->info.type == TSDB_DATA_TYPE_DECIMAL) {  // reader todo decimal
244,430✔
994
      v->data.type = pValue->info.type;
×
995
      v->data.nData = pValue->info.bytes;
×
996
      v->data.pData = taosMemoryCalloc(1, pValue->info.bytes);
×
997
      if (v->data.pData == NULL) {
×
998
        code = terrno;
×
999
        goto end;
×
1000
      }
1001
      memcpy(&v->data.pData, data, pValue->info.bytes);
×
1002
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
×
1003
    } else {  // reader todo decimal
1004
      v->data.type = pValue->info.type;
244,625✔
1005
      memcpy(&v->data.val, data, pValue->info.bytes);
244,430✔
1006
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
244,070✔
1007
    }
1008
  }
1009
end:
36,167✔
1010
  if (code != TSDB_CODE_SUCCESS) {
695,100✔
1011
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1012
    v->isNull = true;
×
1013
  }
1014
  return code;
695,100✔
1015
}
1016

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

1028
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
84,300✔
1029
  if (rows == 0) {
84,300✔
1030
    return;
×
1031
  }
1032

1033
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
84,300✔
1034
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
84,300✔
1035

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

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

1059
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
248,943✔
1060
    SColumnInfo* tmp = (SColumnInfo*)taosArrayGet(pColList, i);
164,643✔
1061
    if (tmp != NULL && tmp->colId == -1) {
164,643✔
1062
      tbNameIndex = i;
84,300✔
1063
    }
1064
  }
1065
  
1066
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
84,092✔
1067
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
84,092✔
1068
  taosArrayDestroy(pColList);
84,065✔
1069
  if (pResBlock == NULL) {
84,300✔
1070
    code = terrno;
276✔
1071
    goto end;
276✔
1072
  }
1073

1074
  pBlockList = taosArrayInit(2, POINTER_BYTES);
84,024✔
1075
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
84,024✔
1076

1077
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
84,024✔
1078
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
84,024✔
1079

1080
  groupData = taosArrayInit(2, POINTER_BYTES);
84,024✔
1081
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
84,024✔
1082

1083
  SNode* pNode = NULL;
84,024✔
1084
  FOREACH(pNode, group) {
248,115✔
1085
    SScalarParam output = {0};
163,856✔
1086

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

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

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

1119
    if (code != TSDB_CODE_SUCCESS) {
163,859✔
1120
      releaseColInfoData(output.columnData);
×
1121
      goto end;
×
1122
    }
1123

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

1128
  for (int i = 0; i < rows; i++) {
403,130✔
1129
    gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
319,106✔
1130
    QUERY_CHECK_NULL(gInfo, code, lino, end, terrno);
319,106✔
1131

1132
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
319,106✔
1133
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
319,106✔
1134

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

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

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

1167
  if (code != TSDB_CODE_SUCCESS) {
84,300✔
1168
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
276✔
1169
  }
1170
}
1171

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

1184
  int32_t rows = taosArrayGetSize(pTableListInfo->pTableList);
4,399,403✔
1185
  if (rows == 0) {
4,399,403✔
1186
    return TSDB_CODE_SUCCESS;
×
1187
  } 
1188

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

1200
    nodesFree(listNode);
×
1201

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

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

1215
  pUidTagList = taosArrayInit(8, sizeof(STUidTagInfo));
4,399,403✔
1216
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
4,399,403✔
1217

1218
  for (int32_t i = 0; i < rows; ++i) {
28,636,369✔
1219
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
24,236,966✔
1220
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
24,236,966✔
1221
    STUidTagInfo info = {.uid = pkeyInfo->uid};
24,236,966✔
1222
    void*        tmp = taosArrayPush(pUidTagList, &info);
24,236,966✔
1223
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
24,236,966✔
1224
  }
1225

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

1235
  SArray* pColList = NULL;
4,399,403✔
1236
  code = qGetColumnsFromNodeList(group, true, &pColList); 
4,399,403✔
1237
  if (code != TSDB_CODE_SUCCESS) {
4,398,865✔
1238
    goto end;
×
1239
  }
1240

1241
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
4,398,865✔
1242
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
4,398,327✔
1243
  taosArrayDestroy(pColList);
4,399,403✔
1244
  if (pResBlock == NULL) {
4,399,403✔
1245
    code = terrno;
×
1246
    goto end;
×
1247
  }
1248

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

1252
  pBlockList = taosArrayInit(2, POINTER_BYTES);
4,399,403✔
1253
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
4,399,403✔
1254

1255
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
4,398,411✔
1256
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
4,398,411✔
1257

1258
  groupData = taosArrayInit(2, POINTER_BYTES);
4,398,411✔
1259
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
4,399,403✔
1260

1261
  SNode* pNode = NULL;
4,398,865✔
1262
  FOREACH(pNode, group) {
9,135,594✔
1263
    SScalarParam output = {0};
4,736,749✔
1264

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

1290
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
4,736,350✔
1291
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
4,640,063✔
1292
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
4,640,063✔
1293
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
4,638,747✔
1294
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
4,638,747✔
1295
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
96,945✔
1296
      continue;
×
1297
    } else {
1298
      gTaskScalarExtra.pStreamInfo = NULL;
97,678✔
1299
      gTaskScalarExtra.pStreamRange = NULL;
97,678✔
1300
      code = scalarCalculate(pNode, pBlockList, &output, &gTaskScalarExtra);
97,678✔
1301
    }
1302

1303
    if (code != TSDB_CODE_SUCCESS) {
4,736,294✔
1304
      releaseColInfoData(output.columnData);
×
1305
      goto end;
×
1306
    }
1307

1308
    void* tmp = taosArrayPush(groupData, &output.columnData);
4,737,741✔
1309
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
4,737,741✔
1310
  }
1311

1312
  int32_t keyLen = 0;
4,398,271✔
1313
  SNode*  node;
1314
  FOREACH(node, group) {
9,132,592✔
1315
    SExprNode* pExpr = (SExprNode*)node;
4,736,966✔
1316
    keyLen += pExpr->resType.bytes;
4,736,966✔
1317
  }
1318

1319
  int32_t nullFlagSize = sizeof(int8_t) * LIST_LENGTH(group);
4,396,531✔
1320
  keyLen += nullFlagSize;
4,397,188✔
1321

1322
  keyBuf = taosMemoryCalloc(1, keyLen);
4,397,188✔
1323
  if (keyBuf == NULL) {
4,397,315✔
1324
    code = terrno;
×
1325
    goto end;
×
1326
  }
1327

1328
  if (initRemainGroups) {
4,397,315✔
1329
    pTableListInfo->remainGroups =
2,063,454✔
1330
        taosHashInit(rows, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
2,061,968✔
1331
    if (pTableListInfo->remainGroups == NULL) {
2,063,454✔
1332
      code = terrno;
×
1333
      goto end;
×
1334
    }
1335
  }
1336

1337
  for (int i = 0; i < rows; i++) {
28,628,411✔
1338
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
24,230,720✔
1339
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
24,229,184✔
1340

1341
    if (groupIdMap != NULL){
24,229,184✔
1342
      gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
73,973✔
1343
    }
1344
    
1345
    char* isNull = (char*)keyBuf;
24,228,221✔
1346
    char* pStart = (char*)keyBuf + sizeof(int8_t) * LIST_LENGTH(group);
24,228,221✔
1347
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
50,038,868✔
1348
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
25,809,373✔
1349

1350
      if (groupIdMap != NULL && gInfo != NULL) {
25,811,059✔
1351
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
116,001✔
1352
        if (ret != TSDB_CODE_SUCCESS) {
116,196✔
1353
          qError("buildGroupInfo failed at line %d since %s", __LINE__, tstrerror(ret));
×
1354
          taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1355
          gInfo = NULL;
×
1356
        }
1357
      }
1358
      
1359
      if (colDataIsNull_s(pValue, i)) {
51,622,816✔
1360
        isNull[j] = 1;
104,630✔
1361
      } else {
1362
        isNull[j] = 0;
25,706,932✔
1363
        char* data = colDataGetData(pValue, i);
25,707,057✔
1364
        if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
25,708,716✔
1365
          // if (tTagIsJson(data)) {
1366
          //   code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
1367
          //   goto end;
1368
          // }
1369
          if (tTagIsJsonNull(data)) {
80,196✔
1370
            isNull[j] = 1;
×
1371
            continue;
×
1372
          }
1373
          int32_t len = getJsonValueLen(data);
80,196✔
1374
          memcpy(pStart, data, len);
80,196✔
1375
          pStart += len;
80,196✔
1376
        } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
25,626,688✔
1377
          if (IS_STR_DATA_BLOB(pValue->info.type)) {
22,069,556✔
UNCOV
1378
            if (blobDataTLen(data) > TSDB_MAX_BLOB_LEN) {
×
1379
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1380
              goto end;
×
1381
            }
1382
            memcpy(pStart, data, blobDataTLen(data));
×
1383
            pStart += blobDataTLen(data);
×
1384
          } else {
1385
            if (varDataTLen(data) > pValue->info.bytes) {
22,071,709✔
1386
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1387
              goto end;
×
1388
            }
1389
            memcpy(pStart, data, varDataTLen(data));
22,072,593✔
1390
            pStart += varDataTLen(data);
22,073,669✔
1391
          }
1392
        } else {
1393
          memcpy(pStart, data, pValue->info.bytes);
3,556,910✔
1394
          pStart += pValue->info.bytes;
3,555,711✔
1395
        }
1396
      }
1397
    }
1398

1399
    int32_t len = (int32_t)(pStart - (char*)keyBuf);
24,225,622✔
1400
    info->groupId = calcGroupId(keyBuf, len);
24,225,622✔
1401
    if (groupIdMap != NULL && gInfo != NULL) {
24,233,923✔
1402
      int32_t ret = taosHashPut(groupIdMap, &info->groupId, sizeof(info->groupId), &gInfo, POINTER_BYTES);
74,220✔
1403
      if (ret != TSDB_CODE_SUCCESS) {
74,220✔
1404
        qError("put groupid to map failed at line %d since %s", __LINE__, tstrerror(ret));
×
1405
        taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1406
      }
1407
      qDebug("put groupid to map gid:%" PRIu64, info->groupId);
74,220✔
1408
      gInfo = NULL;
74,220✔
1409
    }
1410
    if (initRemainGroups) {
24,233,923✔
1411
      // groupId ~ table uid
1412
      code = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId), &(info->uid),
11,767,850✔
1413
                         sizeof(info->uid));
1414
      if (code == TSDB_CODE_DUP_KEY) {
11,763,593✔
1415
        code = TSDB_CODE_SUCCESS;
809,563✔
1416
      }
1417
      QUERY_CHECK_CODE(code, lino, end);
11,763,593✔
1418
    }
1419
  }
1420

1421
  if (tsTagFilterCache && groupIdMap == NULL) {
4,397,691✔
1422
    tableList = taosArrayDup(pTableListInfo->pTableList, NULL);
×
1423
    QUERY_CHECK_NULL(tableList, code, lino, end, terrno);
×
1424

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

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

1434
end:
4,397,691✔
1435
  taosMemoryFreeClear(keyBuf);
4,399,403✔
1436
  blockDataDestroy(pResBlock);
4,399,403✔
1437
  taosArrayDestroy(pBlockList);
4,398,689✔
1438
  taosArrayDestroyEx(pUidTagList, freeItem);
4,398,207✔
1439
  taosArrayDestroyP(groupData, releaseColInfoData);
4,399,403✔
1440
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
4,398,335✔
1441

1442
  if (code != TSDB_CODE_SUCCESS) {
4,398,929✔
1443
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1444
  }
1445
  return code;
4,398,355✔
1446
}
1447

1448
static int32_t nameComparFn(const void* p1, const void* p2) {
1,427,580✔
1449
  const char* pName1 = *(const char**)p1;
1,427,580✔
1450
  const char* pName2 = *(const char**)p2;
1,428,879✔
1451

1452
  int32_t ret = strcmp(pName1, pName2);
1,427,736✔
1453
  if (ret == 0) {
1,427,736✔
1454
    return 0;
14,916✔
1455
  } else {
1456
    return (ret > 0) ? 1 : -1;
1,412,820✔
1457
  }
1458
}
1459

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

1466
  SArray* pTbList = taosArrayInit(len, POINTER_BYTES);
742,345✔
1467
  QUERY_CHECK_NULL(pTbList, code, lino, _end, terrno);
742,726✔
1468

1469
  for (int i = 0; i < pList->pNodeList->length; i++) {
2,141,745✔
1470
    SValueNode* valueNode = (SValueNode*)cell->pNode;
1,398,257✔
1471
    if (!IS_VAR_DATA_TYPE(valueNode->node.resType.type)) {
1,398,257✔
1472
      terrno = TSDB_CODE_INVALID_PARA;
×
1473
      taosArrayDestroy(pTbList);
×
1474
      return NULL;
×
1475
    }
1476

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

1483
  size_t numOfTables = taosArrayGetSize(pTbList);
742,726✔
1484

1485
  // order the name
1486
  taosArraySort(pTbList, nameComparFn);
742,726✔
1487

1488
  // remove the duplicates
1489
  SArray* pNewList = taosArrayInit(taosArrayGetSize(pTbList), sizeof(void*));
741,808✔
1490
  QUERY_CHECK_NULL(pNewList, code, lino, _end, terrno);
742,726✔
1491
  void* tmpTbl = taosArrayGet(pTbList, 0);
742,726✔
1492
  QUERY_CHECK_NULL(tmpTbl, code, lino, _end, terrno);
742,189✔
1493
  void* tmp = taosArrayPush(pNewList, tmpTbl);
742,726✔
1494
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
742,726✔
1495

1496
  for (int32_t i = 1; i < numOfTables; ++i) {
1,399,019✔
1497
    char** name = taosArrayGetLast(pNewList);
655,375✔
1498
    char** nameInOldList = taosArrayGet(pTbList, i);
655,375✔
1499
    QUERY_CHECK_NULL(nameInOldList, code, lino, _end, terrno);
655,531✔
1500
    if (strcmp(*name, *nameInOldList) == 0) {
655,531✔
1501
      continue;
7,794✔
1502
    }
1503

1504
    tmp = taosArrayPush(pNewList, nameInOldList);
648,499✔
1505
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
648,499✔
1506
  }
1507

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

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

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

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

1528
static int32_t filterTableInfoCompare(const void* a, const void* b) {
144,426✔
1529
  STUidTagInfo* p1 = (STUidTagInfo*)a;
144,426✔
1530
  STUidTagInfo* p2 = (STUidTagInfo*)b;
144,426✔
1531

1532
  if (p1->uid == p2->uid) {
144,426✔
1533
    return 0;
×
1534
  }
1535

1536
  return p1->uid < p2->uid ? -1 : 1;
144,426✔
1537
}
1538

1539
static FilterCondType checkTagCond(SNode* cond) {
14,760,712✔
1540
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
14,760,712✔
1541
    return FILTER_NO_LOGIC;
10,826,035✔
1542
  }
1543
  if (nodeType(cond) == QUERY_NODE_LOGIC_CONDITION && ((SLogicConditionNode*)cond)->condType == LOGIC_COND_TYPE_AND) {
3,935,805✔
1544
    return FILTER_AND;
3,355,469✔
1545
  }
1546
  return FILTER_OTHER;
579,891✔
1547
}
1548

1549
static int32_t doInWithAnd(SNode* cond, void* pVnode, SArray* list, SStorageAPI* pAPI, uint64_t suid) {
3,942,455✔
1550
  if (nodeType(cond) != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
3,942,455✔
1551
    return -1;
579,749✔
1552
  }
1553

1554
  SLogicConditionNode* pNode = (SLogicConditionNode*)cond;
3,364,657✔
1555
  SNodeList*           pList = (SNodeList*)pNode->pParameterList;
3,364,657✔
1556

1557
  int32_t len = LIST_LENGTH(pList);
3,364,796✔
1558
  if (len <= 0) {
3,362,204✔
1559
    return -1;
×
1560
  }
1561

1562
  SListCell* cell = pList->pHead;
3,362,204✔
1563
  for (int i = 0; i < len; i++) {
10,488,298✔
1564
    if (cell == NULL) break;
7,133,076✔
1565
    if (nodeType(cell->pNode) == QUERY_NODE_OPERATOR && optimizeTbnameInCondImpl(pVnode, list, cell->pNode, pAPI, suid) == 0) {
7,133,076✔
1566
      return 0;
8,059✔
1567
    }
1568
    cell = cell->pNext;
7,124,797✔
1569
  }
1570
  return -1;
3,355,222✔
1571
}
1572

1573
static int32_t optimizeTbnameInCond(void* pVnode, int64_t suid, SArray* list, SNode* cond, SStorageAPI* pAPI) {
15,503,682✔
1574
  int32_t code = 0;
15,503,682✔
1575
  int32_t lino = 0;
15,503,682✔
1576
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
15,503,682✔
1577
    code = optimizeTbnameInCondImpl(pVnode, list, cond, pAPI, suid);
11,562,171✔
1578
  } else {
1579
    code = doInWithAnd(cond, pVnode, list, pAPI, suid);
3,943,320✔
1580
    QUERY_CHECK_CODE(code, lino, end);
3,943,023✔
1581
    code = pAPI->metaFn.getTableTagsByUid(pVnode, suid, list);
6,406✔
1582
  }
1583
  
1584
end:
15,504,089✔
1585
  return code;
15,504,089✔
1586
}
1587

1588
static int32_t getTableListInInOperator(void* pVnode, SArray* pExistedUidList, SNodeListNode* pList, SStorageAPI* pStoreAPI,
742,726✔
1589
                                        uint64_t suid) {
1590
  int32_t   code = 0;                                          
742,726✔
1591
  SArray*   pTbList = getTableNameList(pList);
742,726✔
1592
  int32_t   numOfTables = taosArrayGetSize(pTbList);
741,427✔
1593
  SHashObj* uHash = NULL;
741,427✔
1594

1595
  size_t numOfExisted = taosArrayGetSize(pExistedUidList);  // len > 0 means there already have uids
741,427✔
1596
  if (numOfExisted > 0) {
741,427✔
1597
    uHash = taosHashInit(numOfExisted / 0.7, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
2,150✔
1598
    if (!uHash) {
2,150✔
1599
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1600
      code = terrno;
×
1601
      goto end;
×
1602
    }
1603

1604
    for (int i = 0; i < numOfExisted; i++) {
2,096,201✔
1605
      STUidTagInfo* pTInfo = taosArrayGet(pExistedUidList, i);
2,094,051✔
1606
      if (!pTInfo) {
2,094,051✔
1607
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1608
        code = terrno;
×
1609
        goto end;
×
1610
      }
1611
      code = taosHashPut(uHash, &pTInfo->uid, sizeof(uint64_t), &i, sizeof(i));
2,094,051✔
1612
      if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_DUP_KEY) {
2,094,051✔
1613
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1614
        goto end;
×
1615
      }
1616
    }
1617
  }
1618
  taosArrayClear(pExistedUidList);
741,427✔
1619

1620
  for (int i = 0; i < numOfTables; i++) {
2,132,271✔
1621
    char* name = taosArrayGetP(pTbList, i);
1,389,164✔
1622
    uint64_t uid = 0, csuid = 0;
1,389,545✔
1623
    if (pStoreAPI->metaFn.getTableUidByName(pVnode, name, &uid) != 0) {
1,390,844✔
1624
      continue;
921,651✔
1625
    }
1626

1627
    ETableType tbType = TSDB_TABLE_MAX;
469,193✔
1628
    if (pStoreAPI->metaFn.getTableTypeSuidByName(pVnode, name, &tbType, &csuid) == 0 &&
469,193✔
1629
        (tbType == TSDB_CHILD_TABLE || tbType == TSDB_VIRTUAL_CHILD_TABLE) &&
469,193✔
1630
        csuid == suid) {
463,147✔
1631
      if (NULL == uHash || taosHashGet(uHash, &uid, sizeof(uid)) != NULL) {
460,999✔
1632
        STUidTagInfo s = {.uid = uid, .name = name, .pTagVal = NULL};
459,924✔
1633
        void*        tmp = taosArrayPush(pExistedUidList, &s);
459,924✔
1634
        if (!tmp) {
459,924✔
1635
          code = terrno;
×
1636
          goto end;
×
1637
        }
1638
      }
1639
    } 
1640
  }
1641
  taosArraySort(pExistedUidList, filterTableInfoCompare);
743,107✔
1642
  taosArrayRemoveDuplicate(pExistedUidList, filterTableInfoCompare, NULL);
742,726✔
1643
end:
742,726✔
1644
  taosArrayDestroy(pTbList);
742,726✔
1645
  taosHashCleanup(uHash);
742,726✔
1646
  return code;
742,726✔
1647
}
1648

1649
// only return uid that does not contained in pExistedUidList
1650
static int32_t optimizeTbnameInCondImpl(void* pVnode, SArray* pExistedUidList, SNode* pTagCond, SStorageAPI* pStoreAPI,
18,688,427✔
1651
                                        uint64_t suid) {
1652
  SOperatorNode* pNode = (SOperatorNode*)pTagCond;
18,688,427✔
1653
  if (pNode->opType != OP_TYPE_IN) {
18,688,427✔
1654
    return -1;
16,752,185✔
1655
  }
1656

1657
  if ((pNode->pLeft != NULL && ((nodeType(pNode->pLeft) == QUERY_NODE_FUNCTION &&
1,935,922✔
1658
                                 ((SFunctionNode*)pNode->pLeft)->funcType == FUNCTION_TYPE_TBNAME)) ||
744,553✔
1659
       (nodeType(pNode->pLeft) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pLeft)->colType == COLUMN_TYPE_TBNAME)) &&
1,192,252✔
1660
      (pNode->pRight != NULL && nodeType(pNode->pRight) == QUERY_NODE_NODE_LIST)) {
744,168✔
1661
    SNodeListNode* pList = (SNodeListNode*)pNode->pRight;
742,726✔
1662

1663
    int32_t len = LIST_LENGTH(pList->pNodeList);
742,726✔
1664
    if (len <= 0) {
742,345✔
1665
      return -1;
×
1666
    }
1667

1668
    return getTableListInInOperator(pVnode, pExistedUidList, pList, pStoreAPI, suid);
742,345✔
1669
  }
1670

1671
  return -1;
1,194,460✔
1672
}
1673

1674
SSDataBlock* createTagValBlockForFilter(SArray* pColList, int32_t numOfTables, SArray* pUidTagList, void* pVnode,
19,010,174✔
1675
                                        SStorageAPI* pStorageAPI) {
1676
  int32_t      code = TSDB_CODE_SUCCESS;
19,010,174✔
1677
  int32_t      lino = 0;
19,010,174✔
1678
  SSDataBlock* pResBlock = NULL;
19,010,174✔
1679
  code = createDataBlock(&pResBlock);
19,016,477✔
1680
  QUERY_CHECK_CODE(code, lino, _end);
19,011,946✔
1681

1682
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
39,785,203✔
1683
    SColumnInfoData colInfo = {0};
20,771,279✔
1684
    void*           tmp = taosArrayGet(pColList, i);
20,768,216✔
1685
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
20,759,559✔
1686
    colInfo.info = *(SColumnInfo*)tmp;
20,759,559✔
1687
    code = blockDataAppendColInfo(pResBlock, &colInfo);
20,755,679✔
1688
    QUERY_CHECK_CODE(code, lino, _end);
20,768,738✔
1689
  }
1690

1691
  code = blockDataEnsureCapacity(pResBlock, numOfTables);
19,013,786✔
1692
  if (code != TSDB_CODE_SUCCESS) {
19,015,990✔
1693
    terrno = code;
×
1694
    blockDataDestroy(pResBlock);
×
1695
    return NULL;
×
1696
  }
1697

1698
  pResBlock->info.rows = numOfTables;
19,015,990✔
1699

1700
  int32_t numOfCols = taosArrayGetSize(pResBlock->pDataBlock);
19,018,072✔
1701

1702
  for (int32_t i = 0; i < numOfTables; i++) {
230,615,520✔
1703
    STUidTagInfo* p1 = taosArrayGet(pUidTagList, i);
211,584,687✔
1704
    QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
211,606,525✔
1705

1706
    for (int32_t j = 0; j < numOfCols; j++) {
433,385,880✔
1707
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, j);
221,728,480✔
1708
      QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
221,686,758✔
1709

1710
      if (pColInfo->info.colId == -1) {  // tbname
221,686,772✔
1711
        char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
7,536,834✔
1712
        if (p1->name != NULL) {
7,537,410✔
1713
          STR_TO_VARSTR(str, p1->name);
5,910✔
1714
        } else {  // name is not retrieved during filter
1715
          code = pStorageAPI->metaFn.getTableNameByUid(pVnode, p1->uid, str);
7,530,583✔
1716
          QUERY_CHECK_CODE(code, lino, _end);
7,530,276✔
1717
        }
1718

1719
        code = colDataSetVal(pColInfo, i, str, false);
7,535,910✔
1720
        QUERY_CHECK_CODE(code, lino, _end);
7,537,033✔
1721
#if TAG_FILTER_DEBUG
1722
        qDebug("tagfilter uid:%ld, tbname:%s", *uid, str + 2);
1723
#endif
1724
      } else {
1725
        STagVal tagVal = {0};
214,146,334✔
1726
        tagVal.cid = pColInfo->info.colId;
214,156,848✔
1727
        if (p1->pTagVal == NULL) {
214,167,181✔
1728
          colDataSetNULL(pColInfo, i);
4,600✔
1729
        } else {
1730
          const char* p = pStorageAPI->metaFn.extractTagVal(p1->pTagVal, pColInfo->info.type, &tagVal);
213,862,443✔
1731

1732
          if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
214,249,879✔
1733
            colDataSetNULL(pColInfo, i);
2,425,312✔
1734
          } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
211,820,199✔
1735
            code = colDataSetVal(pColInfo, i, p, false);
633,999✔
1736
            QUERY_CHECK_CODE(code, lino, _end);
633,999✔
1737
          } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
349,519,437✔
1738
            if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
138,380,714✔
1739
              QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
1740
            }
1741
            char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
138,389,833✔
1742
            QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
138,361,425✔
1743
            varDataSetLen(tmp, tagVal.nData);
138,361,425✔
1744
            memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
138,363,350✔
1745
            code = colDataSetVal(pColInfo, i, tmp, false);
138,359,844✔
1746
#if TAG_FILTER_DEBUG
1747
            qDebug("tagfilter varch:%s", tmp + 2);
1748
#endif
1749
            taosMemoryFree(tmp);
138,386,354✔
1750
            QUERY_CHECK_CODE(code, lino, _end);
138,335,052✔
1751
          } else {
1752
            code = colDataSetVal(pColInfo, i, (const char*)&tagVal.i64, false);
72,800,944✔
1753
            QUERY_CHECK_CODE(code, lino, _end);
72,836,765✔
1754
#if TAG_FILTER_DEBUG
1755
            if (pColInfo->info.type == TSDB_DATA_TYPE_INT) {
1756
              qDebug("tagfilter int:%d", *(int*)(&tagVal.i64));
1757
            } else if (pColInfo->info.type == TSDB_DATA_TYPE_DOUBLE) {
1758
              qDebug("tagfilter double:%f", *(double*)(&tagVal.i64));
1759
            }
1760
#endif
1761
          }
1762
        }
1763
      }
1764
    }
1765
  }
1766

1767
_end:
19,028,959✔
1768
  if (code != TSDB_CODE_SUCCESS) {
19,031,611✔
1769
    blockDataDestroy(pResBlock);
5,781✔
1770
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
276✔
1771
    terrno = code;
276✔
1772
    return NULL;
276✔
1773
  }
1774
  return pResBlock;
19,025,830✔
1775
}
1776

1777
static int32_t doSetQualifiedUid(SArray* pUidList, const SArray* pUidTagList, SColumnInfoData* colData) {
15,476,307✔
1778
  taosArrayClear(pUidList);
15,476,307✔
1779

1780
  int32_t       numOfTables = taosArrayGetSize(pUidTagList);
15,475,534✔
1781
  bool* pResultList = colData == NULL ? NULL : (bool*)colData->pData;
15,482,933✔
1782
  for (int32_t i = 0; i < numOfTables; ++i) {
200,552,503✔
1783
    if (pResultList != NULL && !pResultList[i]) {
185,052,116✔
1784
       continue;
105,056,450✔
1785
    }
1786
    STUidTagInfo* tmpTag = (STUidTagInfo*)taosArrayGet(pUidTagList, i);
80,022,315✔
1787
    if (!tmpTag) {
80,009,614✔
1788
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1789
      return terrno;
×
1790
    }
1791
    uint64_t uid = tmpTag->uid;
80,009,614✔
1792
    qDebug("tagfilter get uid:%" PRId64, uid);
80,025,906✔
1793

1794
    void* tmp = taosArrayPush(pUidList, &uid);
80,029,622✔
1795
    if (tmp == NULL) {
80,029,622✔
1796
      return terrno;
×
1797
    }
1798
  }
1799

1800
  return TSDB_CODE_SUCCESS;
15,500,387✔
1801
}
1802

1803
static int32_t copyExistedUids(SArray* pUidTagList, const SArray* pUidList) {
15,505,484✔
1804
  int32_t code = TSDB_CODE_SUCCESS;
15,505,484✔
1805
  int32_t numOfExisted = taosArrayGetSize(pUidList);
15,505,484✔
1806
  if (numOfExisted == 0) {
15,506,792✔
1807
    return code;
12,761,015✔
1808
  }
1809

1810
  for (int32_t i = 0; i < numOfExisted; ++i) {
31,151,852✔
1811
    uint64_t* uid = taosArrayGet(pUidList, i);
28,406,075✔
1812
    if (!uid) {
28,405,693✔
1813
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1814
      return terrno;
×
1815
    }
1816
    STUidTagInfo info = {.uid = *uid};
28,405,693✔
1817
    void*        tmp = taosArrayPush(pUidTagList, &info);
28,406,075✔
1818
    if (!tmp) {
28,406,075✔
1819
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1820
      return code;
×
1821
    }
1822
  }
1823
  return code;
2,745,777✔
1824
}
1825

1826
static EDealRes getCids(SNode** pNode, void* pContext) {
6,678✔
1827
  SHashObj* colHash = (SHashObj*)pContext;
6,678✔
1828
  col_id_t colId = 0;
6,678✔
1829
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
6,678✔
1830
    colId = (*(SColumnNode**)pNode)->colId;
2,226✔
1831
  } else {
1832
    return DEAL_RES_CONTINUE;
4,452✔
1833
  }
1834

1835
  int32_t tempRes = taosHashPut(colHash, &colId, sizeof(colId), pNode, sizeof((*pNode)));
2,226✔
1836
  if (tempRes != TSDB_CODE_SUCCESS ) {
2,226✔
1837
    return DEAL_RES_ERROR;
×
1838
  }
1839

1840
  return DEAL_RES_CONTINUE;
2,226✔
1841
}
1842

1843
SNode* getTagCondNodeForQueryTmq(void* tinfo) {
689✔
1844
  SExecTaskInfo* pTaskInfo = (SExecTaskInfo*)tinfo;
689✔
1845
  const char*    id = GET_TASKID(pTaskInfo);
689✔
1846
  int32_t        code = 0;
689✔
1847

1848
  // traverse to the stream scanner node to add this table id
1849
  SOperatorInfo* pInfo = NULL;
689✔
1850
  code = extractOperatorInTree(pTaskInfo->pRoot, QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN, id, &pInfo);
689✔
1851
  if (code != 0 || pInfo == NULL) {
689✔
1852
    return NULL;
×
1853
  }
1854

1855
  STmqQueryScanInfo* pScanInfo = pInfo->info;
689✔
1856
  return pScanInfo->pTagCond;
689✔
1857
}
1858

1859
SNode* getTagCondNodeForStableTmq(void* node) {
2,226✔
1860
  return node == NULL ? NULL : ((SSubplan*)node)->pTagCond;
2,226✔
1861
}
1862

1863
bool checkCidInTagCondition(SNode* pTagCond, SArray* cidList) {
2,915✔
1864
  if (pTagCond == NULL) { 
2,915✔
1865
    return false;
689✔
1866
  }
1867
  bool result = true;
2,226✔
1868
  SHashObj *colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), true, HASH_NO_LOCK);
2,226✔
1869
  if (colHash == NULL) {
2,226✔
1870
    goto end;
×
1871
  }
1872

1873
  nodesRewriteExprPostOrder(&pTagCond, getCids, (void*)colHash);
2,226✔
1874
  if (cidList != NULL) {
2,226✔
1875
    int32_t numOfCols = taosArrayGetSize(cidList);
×
1876
    int32_t i = 0;
×
1877
    for (; i < numOfCols; ++i) {
×
1878
      col_id_t* pCid = taosArrayGet(cidList, i);
×
1879
      void* data = taosHashGet(colHash, pCid, sizeof(*pCid));
×
1880
      if (data != NULL) break;
×
1881
    }
1882
    if (i == numOfCols) { // no tag column involved in condition
×
1883
      result = false;
×
1884
      goto end;
×
1885
    }
1886
  }
1887

1888
end:
2,226✔
1889
  taosHashCleanup(colHash);
2,226✔
1890
  return result;
2,226✔
1891
}
1892

1893
int32_t doFilterByTagCond(int64_t suid, SArray* pUidList, SNode* pTagCond, void* pVnode,
265,104,543✔
1894
                                 SIdxFltStatus status, SStorageAPI* pAPI, void* pStreamInfo) {
1895
  if (pTagCond == NULL) {
265,104,543✔
1896
    return TSDB_CODE_SUCCESS;
249,599,434✔
1897
  }
1898

1899
  terrno = TSDB_CODE_SUCCESS;
15,505,109✔
1900

1901
  int32_t      lino = 0;
15,507,090✔
1902
  int32_t      code = TSDB_CODE_SUCCESS;
15,507,090✔
1903
  SArray*      pBlockList = NULL;
15,507,090✔
1904
  SSDataBlock* pResBlock = NULL;
15,507,090✔
1905
  SScalarParam output = {0};
15,504,339✔
1906
  SArray*      pUidTagList = NULL;
15,506,260✔
1907

1908
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
15,506,260✔
1909

1910
  pUidTagList = taosArrayInit(10, sizeof(STUidTagInfo));
15,505,515✔
1911
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
15,502,014✔
1912

1913
  // pUidList size is 0 if tagIndex is NULL and in query condition
1914
  code = copyExistedUids(pUidTagList, pUidList);
15,502,014✔
1915
  QUERY_CHECK_CODE(code, lino, end);
15,502,446✔
1916

1917
  // Narrow down the scope of the tablelist set if there is tbname in condition and And Logical operator
1918
  code = optimizeTbnameInCond(pVnode, suid, pUidTagList, pTagCond, pAPI);
15,502,446✔
1919
  if (code == 0) {
15,503,940✔
1920
    if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
742,726✔
1921
      goto end;
734,667✔
1922
    }
1923
  } else {
1924
    qDebug("pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
14,761,214✔
1925

1926
    FilterCondType condType = checkTagCond(pTagCond);
14,761,214✔
1927
    if (((condType == FILTER_NO_LOGIC || condType == FILTER_AND) && status != SFLT_NOT_INDEX) || // (super table) use tagIndex and operator is and
14,761,739✔
1928
        (status == SFLT_NOT_INDEX && taosArrayGetSize(pUidTagList) > 0)) {                       // (child table with tagCond)
14,766,334✔
1929
      code = pAPI->metaFn.getTableTagsByUid(pVnode, suid, pUidTagList);
2,743,925✔
1930
    } else {
1931
      taosArrayClearEx(pUidTagList, freeItem);       // clear tablelist if using tagIndex and or condition
12,017,289✔
1932
      code = pAPI->metaFn.getTableTags(pVnode, suid, pUidTagList);
12,014,583✔
1933
    }
1934
    if (code != TSDB_CODE_SUCCESS) {
14,761,086✔
1935
      qError("failed to get table tags from meta, reason:%s, suid:%" PRIu64, tstrerror(code),suid);
×
1936
      terrno = code;
×
1937
      QUERY_CHECK_CODE(code, lino, end);
×
1938
    }
1939
  }
1940

1941
  qDebug("final pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
14,769,145✔
1942

1943
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
14,771,234✔
1944
  if (numOfTables == 0) {
14,773,505✔
1945
    goto end;
1,874,039✔
1946
  }
1947

1948
  SArray* pColList = NULL;
12,899,466✔
1949
  code = qGetColumnsFromNodeList(pTagCond, false, &pColList); 
12,900,062✔
1950
  if (code != TSDB_CODE_SUCCESS) {
12,883,356✔
1951
    goto end;
×
1952
  }
1953
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
12,883,356✔
1954
  taosArrayDestroy(pColList);
12,894,863✔
1955
  if (pResBlock == NULL) {
12,888,240✔
1956
    code = terrno;
×
1957
    QUERY_CHECK_CODE(code, lino, end);
×
1958
  }
1959

1960
  //fprintDataBlock(pResBlock, "tagFilter", "", 0);
1961

1962
  //  int64_t st1 = taosGetTimestampUs();
1963
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1964
  pBlockList = taosArrayInit(2, POINTER_BYTES);
12,888,240✔
1965
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
12,893,616✔
1966

1967
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
12,893,635✔
1968
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
12,893,635✔
1969

1970
  code = createResultData(&type, numOfTables, &output);
12,893,635✔
1971
  if (code != TSDB_CODE_SUCCESS) {
12,879,508✔
1972
    terrno = code;
×
1973
    QUERY_CHECK_CODE(code, lino, end);
×
1974
  }
1975

1976
  gTaskScalarExtra.pStreamInfo = pStreamInfo;
12,879,508✔
1977
  gTaskScalarExtra.pStreamRange = NULL;
12,879,508✔
1978
  code = scalarCalculate(pTagCond, pBlockList, &output, &gTaskScalarExtra);
12,883,956✔
1979
  if (code != TSDB_CODE_SUCCESS) {
12,867,960✔
1980
    qError("failed to calculate scalar, reason:%s", tstrerror(code));
984✔
1981
    terrno = code;
984✔
1982
    QUERY_CHECK_CODE(code, lino, end);
984✔
1983
  }
1984

1985
end:
15,476,666✔
1986
  if (code == 0) {
15,478,450✔
1987
    code = doSetQualifiedUid(pUidList, pUidTagList, output.columnData);
15,483,700✔
1988
    lino = __LINE__;
15,503,337✔
1989
  }
1990
  
1991
  if (code != 0) {
15,498,087✔
1992
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
984✔
1993
  }
1994
  blockDataDestroy(pResBlock);
15,498,087✔
1995
  taosArrayDestroy(pBlockList);
15,488,262✔
1996
  taosArrayDestroyEx(pUidTagList, freeItem);
15,489,774✔
1997

1998
  colDataDestroy(output.columnData);
15,501,095✔
1999
  taosMemoryFreeClear(output.columnData);
15,494,409✔
2000
  return code;
15,502,162✔
2001
}
2002

2003
typedef struct {
2004
  int32_t code;
2005
  SStreamRuntimeFuncInfo* pStreamRuntimeInfo;
2006
} PlaceHolderContext;
2007

2008
static EDealRes replacePlaceHolderColumn(SNode** pNode, void* pContext) {
85,284✔
2009
  PlaceHolderContext* pData = (PlaceHolderContext*)pContext;
85,284✔
2010
  if (QUERY_NODE_FUNCTION != nodeType((*pNode))) {
85,284✔
2011
    return DEAL_RES_CONTINUE;
70,288✔
2012
  }
2013
  SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
14,996✔
2014
  if (!fmIsStreamPesudoColVal(pFuncNode->funcId)) {
14,996✔
2015
    return DEAL_RES_CONTINUE;
552✔
2016
  }
2017
  pData->code = fmSetStreamPseudoFuncParamVal(pFuncNode->funcId, pFuncNode->pParameterList, pData->pStreamRuntimeInfo);
14,444✔
2018
  if (pData->code != TSDB_CODE_SUCCESS) {
14,444✔
2019
    return DEAL_RES_ERROR;
×
2020
  }
2021
  SNode* pFirstParam = nodesListGetNode(pFuncNode->pParameterList, 0);
14,444✔
2022
  ((SValueNode*)pFirstParam)->translate = true;
14,444✔
2023
  SValueNode* res = NULL;
14,444✔
2024
  pData->code = nodesCloneNode(pFirstParam, (SNode**)&res);
14,444✔
2025
  if (NULL == res) {
14,444✔
2026
    return DEAL_RES_ERROR;
×
2027
  }
2028
  nodesDestroyNode(*pNode);
14,444✔
2029
  *pNode = (SNode*)res;
14,444✔
2030

2031
  return DEAL_RES_CONTINUE;
14,444✔
2032
}
2033

2034
static void extractTagColId(SOperatorNode* pOpNode, SArray* pColIdArray) {
23,184✔
2035
  SNode* pLeft = pOpNode->pLeft;
23,184✔
2036
  SNode* pRight = pOpNode->pRight;
23,184✔
2037
  SColumnNode* pColNode = nodeType(pLeft) == QUERY_NODE_COLUMN ?
23,184✔
2038
    (SColumnNode*)pLeft : (SColumnNode*)pRight;
23,184✔
2039

2040
  col_id_t colId = pColNode->colId;
23,184✔
2041
  void* _tmp = taosArrayPush(pColIdArray, &colId);
23,184✔
2042
}
23,184✔
2043

2044
static int32_t buildTagCondKey(
11,592✔
2045
  const SNode* pTagCond, char** pTagCondKey,
2046
  int32_t* tagCondKeyLen, SArray** pTagColIds) {
2047
  if (NULL == pTagCond ||
11,592✔
2048
    (nodeType(pTagCond) != QUERY_NODE_OPERATOR &&
11,592✔
2049
      nodeType(pTagCond) != QUERY_NODE_LOGIC_CONDITION)) {
11,592✔
2050
    qError("invalid parameter to extract tag filter symbol");
×
2051
    return TSDB_CODE_INTERNAL_ERROR;
×
2052
  }
2053
  int32_t code = TSDB_CODE_SUCCESS;
11,592✔
2054
  int32_t lino = 0;
11,592✔
2055
  *pTagColIds = taosArrayInit(4, sizeof(col_id_t));
11,592✔
2056

2057
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
11,592✔
2058
    extractTagColId((SOperatorNode*)pTagCond, *pTagColIds);
×
2059
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
11,592✔
2060
    SNode* pChild = NULL;
11,592✔
2061
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
34,776✔
2062
      extractTagColId((SOperatorNode*)pChild, *pTagColIds);
23,184✔
2063
    }
2064
  }
2065

2066
  taosArraySort(*pTagColIds, compareUint16Val);
11,592✔
2067

2068
  // encode ordered colIds into key string, separated by ','
2069
  *tagCondKeyLen =
23,184✔
2070
    (int32_t)(taosArrayGetSize(*pTagColIds) * (sizeof(col_id_t) + 1) - 1);
11,592✔
2071
  *pTagCondKey = (char*)taosMemoryCalloc(1, *tagCondKeyLen);
11,592✔
2072
  TSDB_CHECK_NULL(*pTagCondKey, code, lino, _end, terrno);
11,592✔
2073
  char* pStart = *pTagCondKey;
11,592✔
2074
  for (int32_t i = 0; i < taosArrayGetSize(*pTagColIds); ++i) {
34,776✔
2075
    col_id_t* pColId = (col_id_t*)taosArrayGet(*pTagColIds, i);
23,184✔
2076
    TSDB_CHECK_NULL(pColId, code, lino, _end, terrno);
23,184✔
2077
    memcpy(pStart, pColId, sizeof(col_id_t));
23,184✔
2078
    pStart += sizeof(col_id_t);
23,184✔
2079
    if (i != taosArrayGetSize(*pTagColIds) - 1) {
23,184✔
2080
      *pStart = ',';
11,592✔
2081
      pStart += 1;
11,592✔
2082
    }
2083
  }
2084

2085
_end:
11,592✔
2086
  if (TSDB_CODE_SUCCESS != code) {
11,592✔
2087
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2088
    terrno = code;
×
2089
  }
2090
  return code;
11,592✔
2091
}
2092

2093
static EDealRes canOptimizeTagCondFilter(SNode* pTagCond, void* pContext) {
107,456✔
2094
  if (NULL == pTagCond) {
107,456✔
2095
    *(bool*)pContext = false;
×
2096
    return DEAL_RES_END;
×
2097
  }
2098
  if (nodeType(pTagCond) == QUERY_NODE_VALUE ||
107,456✔
2099
    nodeType(pTagCond) == QUERY_NODE_COLUMN) {
71,392✔
2100
    return DEAL_RES_CONTINUE;
59,432✔
2101
  }
2102
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR &&
48,024✔
2103
    ((SOperatorNode*)pTagCond)->opType == OP_TYPE_EQUAL) {
23,736✔
2104
    return DEAL_RES_CONTINUE;
23,184✔
2105
  }
2106
  if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION &&
25,208✔
2107
    ((SLogicConditionNode*)pTagCond)->condType == LOGIC_COND_TYPE_AND) {
11,592✔
2108
    return DEAL_RES_CONTINUE;
11,408✔
2109
  }
2110
  if (nodeType(pTagCond) == QUERY_NODE_FUNCTION &&
26,496✔
2111
    fmIsStreamPesudoColVal(((SFunctionNode*)pTagCond)->funcId)) {
12,880✔
2112
    return DEAL_RES_CONTINUE;
13,064✔
2113
  }
2114
  *(bool*)pContext = false;
736✔
2115
  return DEAL_RES_END;
552✔
2116
}
2117

2118
int32_t getTableList(void* pVnode, SScanPhysiNode* pScanNode, SNode* pTagCond, SNode* pTagIndexCond,
265,012,682✔
2119
                     STableListInfo* pListInfo, uint8_t* digest, const char* idstr, SStorageAPI* pStorageAPI, void* pStreamInfo) {
2120
  int32_t code = TSDB_CODE_SUCCESS;
265,012,682✔
2121
  int32_t lino = 0;
265,012,682✔
2122
  size_t  numOfTables = 0;
265,012,682✔
2123

2124
  pListInfo->idInfo.suid = pScanNode->suid;
265,012,682✔
2125
  pListInfo->idInfo.tableType = pScanNode->tableType;
265,020,852✔
2126

2127
  SArray* pUidList = taosArrayInit(8, sizeof(uint64_t));
264,963,777✔
2128
  QUERY_CHECK_NULL(pUidList, code, lino, end, terrno);
264,930,008✔
2129

2130
  char*   pTagCondKey = NULL;
264,930,008✔
2131
  int32_t tagCondKeyLen;
264,842,286✔
2132
  SArray* pTagColIds = NULL;
264,859,325✔
2133
  char*   pPayload = NULL;
264,931,029✔
2134
  SIdxFltStatus status = SFLT_NOT_INDEX;
264,931,029✔
2135

2136
  qTrace("getTableList called, suid:%" PRIu64
264,931,029✔
2137
    ", tagCond:%p, tagIndexCond:%p, %d %d", pScanNode->suid, pTagCond,
2138
    pTagIndexCond, pScanNode->tableType, pScanNode->virtualStableScan);
2139
  if (pScanNode->tableType != TSDB_SUPER_TABLE && !pScanNode->virtualStableScan) {
264,935,038✔
2140
    pListInfo->idInfo.uid = pScanNode->uid;
147,675,405✔
2141
    if (pStorageAPI->metaFn.isTableExisted(pVnode, pScanNode->uid)) {
147,697,110✔
2142
      void* tmp = taosArrayPush(pUidList, &pScanNode->uid);
147,665,027✔
2143
      QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
147,663,252✔
2144
    }
2145
    code = doFilterByTagCond(pScanNode->suid, pUidList, pTagCond, pVnode, status, pStorageAPI, pStreamInfo);
147,712,301✔
2146
    QUERY_CHECK_CODE(code, lino, end);
147,707,605✔
2147
  } else {
2148
    bool      isStream = (pStreamInfo != NULL);
117,383,862✔
2149
    bool      hasTagCond = (pTagCond != NULL);
117,383,862✔
2150
    bool      canCacheTagEqCondFilter = false;
117,383,862✔
2151
    T_MD5_CTX context = {0};
117,333,861✔
2152

2153
    qTrace("start to get table list by tag filter, suid:%" PRIu64
117,425,090✔
2154
      ",tsStableTagFilterCache:%d, tsTagFilterCache:%d", 
2155
      pScanNode->suid, tsStableTagFilterCache, tsTagFilterCache);
2156

2157
    bool acquired = false;
117,425,484✔
2158
    // first, check whether we can use stable tag filter cache
2159
    if (tsStableTagFilterCache && isStream && hasTagCond) {
117,331,155✔
2160
      canCacheTagEqCondFilter = true;
12,144✔
2161
      nodesWalkExpr(pTagCond, canOptimizeTagCondFilter,
12,144✔
2162
        (void*)&canCacheTagEqCondFilter);
2163
    }
2164
    if (canCacheTagEqCondFilter) {
117,032,407✔
2165
      qDebug("%s, stable tag filter condition can be optimized", idstr);
11,592✔
2166
      if (((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
11,592✔
2167
        SNode* tmp = NULL;
11,592✔
2168
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
11,592✔
2169
        QUERY_CHECK_CODE(code, lino, end);
11,592✔
2170

2171
        PlaceHolderContext ctx = {.code = TSDB_CODE_SUCCESS, .pStreamRuntimeInfo = (SStreamRuntimeFuncInfo*)pStreamInfo};
11,592✔
2172
        nodesRewriteExpr(&tmp, replacePlaceHolderColumn, (void*)&ctx);
11,592✔
2173
        if (TSDB_CODE_SUCCESS != ctx.code) {
11,592✔
2174
          nodesDestroyNode(tmp);
×
2175
          code = ctx.code;
×
2176
          goto end;
×
2177
        }
2178
        code = genStableTagFilterDigest(tmp, &context);
11,592✔
2179
        nodesDestroyNode(tmp);
11,592✔
2180
      } else {
2181
        code = genStableTagFilterDigest(pTagCond, &context);
×
2182
      }
2183
      QUERY_CHECK_CODE(code, lino, end);
11,592✔
2184

2185
      code = buildTagCondKey(
11,592✔
2186
        pTagCond, &pTagCondKey, &tagCondKeyLen, &pTagColIds);
2187
      QUERY_CHECK_CODE(code, lino, end);
11,592✔
2188
      code = pStorageAPI->metaFn.getStableCachedTableList(
11,592✔
2189
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
11,592✔
2190
        context.digest, tListLen(context.digest), pUidList, &acquired);
2191
      QUERY_CHECK_CODE(code, lino, end);
11,592✔
2192
    } else if (tsTagFilterCache) {
117,020,815✔
2193
      // second, try to use normal tag filter cache
2194
      qDebug("%s using normal tag filter cache", idstr);
34,132✔
2195
      if (pStreamInfo != NULL && ((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
35,512✔
2196
        SNode* tmp = NULL;
1,380✔
2197
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
1,380✔
2198
        QUERY_CHECK_CODE(code, lino, end);
1,380✔
2199

2200
        PlaceHolderContext ctx = {.code = TSDB_CODE_SUCCESS, .pStreamRuntimeInfo = (SStreamRuntimeFuncInfo*)pStreamInfo};
1,380✔
2201
        nodesRewriteExpr(&tmp, replacePlaceHolderColumn, (void*)&ctx);
1,380✔
2202
        if (TSDB_CODE_SUCCESS != ctx.code) {
1,380✔
2203
          nodesDestroyNode(tmp);
×
2204
          code = ctx.code;
×
2205
          goto end;
×
2206
        }
2207
        code = genTagFilterDigest(tmp, &context);
1,380✔
2208
        nodesDestroyNode(tmp);
1,380✔
2209
      } else {
2210
        code = genTagFilterDigest(pTagCond, &context);
32,752✔
2211
      }
2212
      // try to retrieve the result from meta cache
2213
      QUERY_CHECK_CODE(code, lino, end);      
34,132✔
2214
      code = pStorageAPI->metaFn.getCachedTableList(
34,132✔
2215
        pVnode, pScanNode->suid, context.digest,
34,132✔
2216
        tListLen(context.digest), pUidList, &acquired);
2217
      QUERY_CHECK_CODE(code, lino, end);
185,637✔
2218
    }
2219
    if (acquired) {
117,183,912✔
2220
      taosArrayDestroy(pTagColIds);
32,016✔
2221
      pTagColIds = NULL;
32,016✔
2222
      
2223
      if (digest != NULL) {
32,016✔
2224
        digest[0] = 1;
32,016✔
2225
        memcpy(digest + 1, context.digest, tListLen(context.digest));
32,016✔
2226
      }
2227
      
2228
      qDebug("suid:%" PRIu64 ", %s retrieve table uid list from cache, numOfTables:%d", 
32,016✔
2229
        pScanNode->suid, idstr, (int32_t)taosArrayGetSize(pUidList));
2230
      goto end;
32,016✔
2231
    } else {
2232
      qDebug("suid:%" PRIu64 
117,151,896✔
2233
        ", failed to get table uid list from cache", pScanNode->suid);
2234
    }
2235

2236
    if (!pTagCond) {  // no tag filter condition exists, let's fetch all tables of this super table
117,388,724✔
2237
      code = pStorageAPI->metaFn.getChildTableList(pVnode, pScanNode->suid, pUidList);
102,204,536✔
2238
      QUERY_CHECK_CODE(code, lino, end);
102,175,227✔
2239
      qTrace("no tag filter, get all child tables, numOfTables:%d", (int32_t)taosArrayGetSize(pUidList));
102,175,227✔
2240
    } else {
2241
      SIdxFltStatus status = SFLT_NOT_INDEX;
15,184,188✔
2242
      if (pTagIndexCond) {
15,184,571✔
2243
        void* pIndex = pStorageAPI->metaFn.getInvertIndex(pVnode);
4,255,074✔
2244

2245
        SIndexMetaArg metaArg = {.metaEx = pVnode,
4,255,102✔
2246
                                 .idx = pStorageAPI->metaFn.storeGetIndexInfo(pVnode),
4,255,074✔
2247
                                 .ivtIdx = pIndex,
2248
                                 .suid = pScanNode->uid};
4,255,074✔
2249
        code = doFilterTag(pTagIndexCond, &metaArg, pUidList, &status, &pStorageAPI->metaFilter);
4,254,954✔
2250
        if (code != 0 || status == SFLT_NOT_INDEX) {  // temporarily disable it for performance sake
4,244,901✔
2251
          qDebug("failed to get tableIds from index, suid:%" PRIu64 ", uidListSize:%d", pScanNode->uid, (int32_t)taosArrayGetSize(pUidList));
1,223,456✔
2252
        } else {
2253
          qDebug("succ to get filter result, table num: %d", (int)taosArrayGetSize(pUidList));
3,021,445✔
2254
        }
2255
      }
2256
    }
2257
    qTrace("after index filter, pTagCond:%p uidListSize:%d", pTagCond, (int32_t)taosArrayGetSize(pUidList));
117,360,550✔
2258
    code = doFilterByTagCond(pScanNode->suid, pUidList, pTagCond, pVnode, status, pStorageAPI, pStreamInfo);
117,360,550✔
2259
    QUERY_CHECK_CODE(code, lino, end);
117,379,162✔
2260

2261
    // let's add the filter results into meta-cache
2262
    numOfTables = taosArrayGetSize(pUidList);
117,378,178✔
2263

2264
    if (canCacheTagEqCondFilter) {
117,333,770✔
2265
      qInfo("%s, suid:%" PRIu64 ", add uid list to stable tag filter cache, "
5,152✔
2266
            "uidListSize:%d, origin key:%" PRIu64 ",%" PRIu64,
2267
            idstr, pScanNode->suid, (int32_t)numOfTables,
2268
            *(uint64_t*)context.digest, *(uint64_t*)(context.digest + 8));
2269

2270
      code = pStorageAPI->metaFn.putStableCachedTableList(
5,152✔
2271
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
2272
        context.digest, tListLen(context.digest),
2273
        pUidList, &pTagColIds);
2274
      QUERY_CHECK_CODE(code, lino, end);
5,152✔
2275

2276
      if (digest != NULL) {
5,152✔
2277
        digest[0] = 1;
5,152✔
2278
        memcpy(digest + 1, context.digest, tListLen(context.digest));
5,152✔
2279
      }
2280
    } else if (tsTagFilterCache) {
117,328,618✔
2281
      qInfo("%s, suid:%" PRIu64 ", add uid list to normal tag filter cache, "
8,556✔
2282
            "uidListSize:%d, origin key:%" PRIu64 ",%" PRIu64,
2283
            idstr, pScanNode->suid, (int32_t)numOfTables,
2284
            *(uint64_t*)context.digest, *(uint64_t*)(context.digest + 8));
2285
      size_t size = numOfTables * sizeof(uint64_t) + sizeof(int32_t);
8,556✔
2286
      pPayload = taosMemoryMalloc(size);
8,556✔
2287
      QUERY_CHECK_NULL(pPayload, code, lino, end, terrno);
8,556✔
2288

2289
      *(int32_t*)pPayload = (int32_t)numOfTables;
8,556✔
2290
      if (numOfTables > 0) {
8,556✔
2291
        void* tmp = taosArrayGet(pUidList, 0);
6,900✔
2292
        QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
6,900✔
2293
        memcpy(pPayload + sizeof(int32_t), tmp, numOfTables * sizeof(uint64_t));
6,900✔
2294
      }
2295

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

2317

2318
      if (digest != NULL) {
8,556✔
2319
        digest[0] = 1;
8,556✔
2320
        memcpy(digest + 1, context.digest, tListLen(context.digest));
8,556✔
2321
      }
2322
    }
2323
  }
2324

2325
  qDebug("%s, table list with %d uids built", idstr, (int32_t)numOfTables);
265,023,950✔
2326

2327
end:
265,111,628✔
2328
  if (code == 0) {
265,107,980✔
2329
     STableKeyInfo info = {.uid = 0, .groupId = 0};
265,106,996✔
2330
    for (int32_t i = 0; i < taosArrayGetSize(pUidList); ++i) {
966,183,322✔
2331
      int64_t* uid = (int64_t*)taosArrayGet(pUidList, i);
700,987,857✔
2332
      QUERY_CHECK_NULL(uid, code, lino, end, terrno);
701,060,514✔
2333

2334
      info.uid = *uid;
701,060,514✔
2335
        //qInfo("doSetQualifiedUid row:%d added to pTableList", i);
2336
      void* p = taosArrayPush(pListInfo->pTableList, &info);
701,026,390✔
2337
      QUERY_CHECK_NULL(p, code, lino, end, terrno);
701,068,131✔
2338
    }
2339
  } else {
2340
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
984✔
2341
  }
2342
  taosArrayDestroy(pUidList);
265,168,258✔
2343
  taosArrayDestroy(pTagColIds);
265,063,954✔
2344
  taosMemFreeClear(pTagCondKey);
265,088,031✔
2345
  taosMemFreeClear(pPayload);
265,088,031✔
2346
  return code;
265,088,031✔
2347
}
2348

2349
int32_t qGetTableList(int64_t suid, void* pVnode, void* node, SArray** tableList, void* pTaskInfo) {
23,650✔
2350
  int32_t        code = TSDB_CODE_SUCCESS;
23,650✔
2351
  int32_t        lino = 0;
23,650✔
2352
  SSubplan*      pSubplan = (SSubplan*)node;
23,650✔
2353
  SScanPhysiNode pNode = {0};
23,650✔
2354
  pNode.suid = suid;
23,650✔
2355
  pNode.uid = suid;
23,650✔
2356
  pNode.tableType = TSDB_SUPER_TABLE;
23,650✔
2357

2358
  STableListInfo* pTableListInfo = tableListCreate();
23,650✔
2359
  QUERY_CHECK_NULL(pTableListInfo, code, lino, _end, terrno);
23,650✔
2360
  code = getTableList(pVnode, &pNode, pSubplan ? pSubplan->pTagCond : NULL, pSubplan ? pSubplan->pTagIndexCond : NULL,
23,650✔
2361
                      pTableListInfo, NULL, "qGetTableList", &((SExecTaskInfo*)pTaskInfo)->storageAPI, NULL);
2362
  QUERY_CHECK_CODE(code, lino, _end);
23,650✔
2363
  *tableList = pTableListInfo->pTableList;
23,650✔
2364
  pTableListInfo->pTableList = NULL;
23,650✔
2365

2366
_end:
23,650✔
2367
  tableListDestroy(pTableListInfo);
23,650✔
2368
  if (code != TSDB_CODE_SUCCESS) {
23,650✔
2369
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2370
  }
2371
  return code;
23,650✔
2372
}
2373

2374
size_t getTableTagsBufLen(const SNodeList* pGroups) {
×
2375
  size_t keyLen = 0;
×
2376

2377
  SNode* node;
2378
  FOREACH(node, pGroups) {
×
2379
    SExprNode* pExpr = (SExprNode*)node;
×
2380
    keyLen += pExpr->resType.bytes;
×
2381
  }
2382

2383
  keyLen += sizeof(int8_t) * LIST_LENGTH(pGroups);
×
2384
  return keyLen;
×
2385
}
2386

2387
int32_t getGroupIdFromTagsVal(void* pVnode, uint64_t uid, SNodeList* pGroupNode, char* keyBuf, uint64_t* pGroupId,
×
2388
                              SStorageAPI* pAPI) {
2389
  SMetaReader mr = {0};
×
2390

2391
  pAPI->metaReaderFn.initReader(&mr, pVnode, META_READER_LOCK, &pAPI->metaFn);
×
2392
  if (pAPI->metaReaderFn.getEntryGetUidCache(&mr, uid) != 0) {  // table not exist
×
2393
    pAPI->metaReaderFn.clearReader(&mr);
×
2394
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
2395
  }
2396

2397
  SNodeList* groupNew = NULL;
×
2398
  int32_t    code = nodesCloneList(pGroupNode, &groupNew);
×
2399
  if (TSDB_CODE_SUCCESS != code) {
×
2400
    pAPI->metaReaderFn.clearReader(&mr);
×
2401
    return code;
×
2402
  }
2403

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

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

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

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

2464
  int32_t len = (int32_t)(pStart - (char*)keyBuf);
×
2465
  *pGroupId = calcGroupId(keyBuf, len);
×
2466

2467
  nodesDestroyList(groupNew);
×
2468
  pAPI->metaReaderFn.clearReader(&mr);
×
2469

2470
  return TSDB_CODE_SUCCESS;
×
2471
}
2472

2473
SArray* makeColumnArrayFromList(SNodeList* pNodeList) {
3,375,505✔
2474
  if (!pNodeList) {
3,375,505✔
2475
    return NULL;
×
2476
  }
2477

2478
  size_t  numOfCols = LIST_LENGTH(pNodeList);
3,375,505✔
2479
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
3,377,489✔
2480
  if (pList == NULL) {
3,373,450✔
2481
    return NULL;
×
2482
  }
2483

2484
  for (int32_t i = 0; i < numOfCols; ++i) {
7,721,310✔
2485
    SColumnNode* pColNode = (SColumnNode*)nodesListGetNode(pNodeList, i);
4,350,253✔
2486
    if (!pColNode) {
4,354,160✔
2487
      taosArrayDestroy(pList);
×
2488
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
2489
      return NULL;
×
2490
    }
2491

2492
    // todo extract method
2493
    SColumn c = {0};
4,354,160✔
2494
    c.slotId = pColNode->slotId;
4,352,946✔
2495
    c.colId = pColNode->colId;
4,352,158✔
2496
    c.type = pColNode->node.resType.type;
4,352,154✔
2497
    c.bytes = pColNode->node.resType.bytes;
4,351,048✔
2498
    c.precision = pColNode->node.resType.precision;
4,349,238✔
2499
    c.scale = pColNode->node.resType.scale;
4,351,230✔
2500

2501
    void* tmp = taosArrayPush(pList, &c);
4,350,962✔
2502
    if (!tmp) {
4,350,962✔
2503
      taosArrayDestroy(pList);
×
2504
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2505
      return NULL;
×
2506
    }
2507
  }
2508

2509
  return pList;
3,371,057✔
2510
}
2511

2512
int32_t extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols,
335,781,001✔
2513
                            int32_t type, SColMatchInfo* pMatchInfo) {
2514
  size_t  numOfCols = LIST_LENGTH(pNodeList);
335,781,001✔
2515
  int32_t code = TSDB_CODE_SUCCESS;
335,790,195✔
2516
  int32_t lino = 0;
335,790,195✔
2517
  bool    colIdOrdered = true;
335,790,195✔
2518
  bool    hasPrevColId = false;
335,790,195✔
2519
  int32_t prevColId = 0;
335,790,195✔
2520

2521
  pMatchInfo->matchType = type;
335,790,195✔
2522

2523
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchItem));
335,768,909✔
2524
  if (pList == NULL) {
335,698,845✔
2525
    code = terrno;
×
2526
    return code;
×
2527
  }
2528

2529
  SColMatchItem** infoBySlot = NULL;
335,698,845✔
2530
  if (numOfCols > 0) {
335,698,845✔
2531
    infoBySlot = taosMemoryCalloc(numOfCols, sizeof(*infoBySlot));
335,587,191✔
2532
    if (infoBySlot == NULL) {
335,608,394✔
2533
      code = terrno;
×
2534
      goto _end;
×
2535
    }
2536
  }
2537

2538
  SNode* node = NULL;
335,720,048✔
2539
  FOREACH(node, pNodeList) {
1,616,263,050✔
2540
    STargetNode* pNode = (STargetNode*)node;
1,280,383,391✔
2541
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,280,383,391✔
2542
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
1,280,383,391✔
2543
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
1,268,637,872✔
2544

2545
      SColMatchItem c = {.needOutput = true};
1,268,667,240✔
2546
      c.colId = pColNode->colId;
1,268,660,244✔
2547
      c.srcSlotId = pColNode->slotId;
1,268,624,945✔
2548
      c.dstSlotId = pNode->slotId;
1,268,630,993✔
2549
      c.isPk = pColNode->isPk;
1,268,692,023✔
2550
      c.dataType = pColNode->node.resType;
1,268,767,822✔
2551
      void* tmp = taosArrayPush(pList, &c);
1,268,833,489✔
2552
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,268,833,489✔
2553
      if (hasPrevColId) {
1,268,833,489✔
2554
        if (c.colId < prevColId) {
933,471,133✔
2555
          colIdOrdered = false;
14,356,748✔
2556
        }
2557
      } else {
2558
        hasPrevColId = true;
335,362,356✔
2559
      }
2560
      prevColId = c.colId;
1,268,833,489✔
2561
      if (pNode->slotId >= 0 && pNode->slotId < numOfCols) {
1,268,833,489✔
2562
        infoBySlot[pNode->slotId] = (SColMatchItem*)taosArrayGet(pList, taosArrayGetSize(pList) - 1);
1,268,286,619✔
2563
      }
2564
    }
2565
  }
2566
  // set the output flag for each column in SColMatchInfo, according to the
2567
  *numOfOutputCols = 0;
335,848,092✔
2568
  SNode* slotNode = NULL;
335,849,583✔
2569
  FOREACH(slotNode, pOutputNodeList->pSlots) {
1,797,952,990✔
2570
    SSlotDescNode* pNode = (SSlotDescNode*)slotNode;
1,462,248,357✔
2571
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,462,248,357✔
2572

2573
    // todo: add reserve flag check
2574
    // it is a column reserved for the arithmetic expression calculation
2575
    if (pNode->slotId >= numOfCols) {
1,462,248,357✔
2576
      (*numOfOutputCols) += 1;
182,788,082✔
2577
      continue;
182,802,139✔
2578
    }
2579

2580
    SColMatchItem* info = infoBySlot ? infoBySlot[pNode->slotId] : NULL;
1,279,324,257✔
2581

2582
    if (pNode->output) {
1,279,580,208✔
2583
      (*numOfOutputCols) += 1;
1,261,155,124✔
2584
    } else if (info != NULL) {
18,494,299✔
2585
      // select distinct tbname from stb where tbname='abc';
2586
      info->needOutput = false;
18,466,661✔
2587
    }
2588
  }
2589

2590
  pMatchInfo->pList = pList;
335,775,765✔
2591
  pMatchInfo->colIdOrdered = colIdOrdered;
335,785,565✔
2592

2593
_end:
335,740,649✔
2594
  taosMemoryFree(infoBySlot);
335,740,649✔
2595
  if (code != TSDB_CODE_SUCCESS) {
335,693,368✔
2596
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2597
  }
2598
  return code;
335,710,180✔
2599
}
2600

2601
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
1,226,086,378✔
2602
                                  const char* name) {
2603
  SResSchema s = {0};
1,226,086,378✔
2604
  s.scale = scale;
1,226,188,769✔
2605
  s.type = type;
1,226,188,769✔
2606
  s.bytes = bytes;
1,226,188,769✔
2607
  s.slotId = slotId;
1,226,188,769✔
2608
  s.precision = precision;
1,226,188,769✔
2609
  tstrncpy(s.name, name, tListLen(s.name));
1,226,188,769✔
2610

2611
  return s;
1,226,243,933✔
2612
}
2613

2614
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType, EColumnType colType) {
1,165,934,920✔
2615
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
1,165,934,920✔
2616
  if (pCol == NULL) {
1,165,300,298✔
2617
    return NULL;
×
2618
  }
2619

2620
  pCol->slotId = slotId;
1,165,300,298✔
2621
  pCol->colId = colId;
1,165,335,451✔
2622
  pCol->bytes = pType->bytes;
1,165,377,218✔
2623
  pCol->type = pType->type;
1,165,528,614✔
2624
  pCol->scale = pType->scale;
1,165,731,000✔
2625
  pCol->precision = pType->precision;
1,165,715,000✔
2626
  pCol->dataBlockId = blockId;
1,165,953,104✔
2627
  pCol->colType = colType;
1,165,909,608✔
2628
  return pCol;
1,165,926,860✔
2629
}
2630

2631
int32_t createExprFromOneNode(SExprInfo* pExp, SNode* pNode, int16_t slotId) {
1,226,515,803✔
2632
  int32_t code = TSDB_CODE_SUCCESS;
1,226,515,803✔
2633
  int32_t lino = 0;
1,226,515,803✔
2634
  pExp->base.numOfParams = 0;
1,226,515,803✔
2635
  pExp->base.pParam = NULL;
1,226,609,237✔
2636
  pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
1,226,586,506✔
2637
  QUERY_CHECK_NULL(pExp->pExpr, code, lino, _end, terrno);
1,225,900,326✔
2638

2639
  pExp->pExpr->_function.num = 1;
1,225,927,696✔
2640
  pExp->pExpr->_function.functionId = -1;
1,226,084,988✔
2641

2642
  int32_t type = nodeType(pNode);
1,226,222,523✔
2643
  // it is a project query, or group by column
2644
  if (type == QUERY_NODE_COLUMN) {
1,226,405,669✔
2645
    pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
688,270,269✔
2646
    SColumnNode* pColNode = (SColumnNode*)pNode;
688,326,086✔
2647

2648
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
688,326,086✔
2649
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
688,215,353✔
2650

2651
    pExp->base.numOfParams = 1;
688,216,516✔
2652

2653
    SDataType* pType = &pColNode->node.resType;
688,228,563✔
2654
    pExp->base.resSchema =
2655
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pColNode->colName);
688,249,763✔
2656

2657
    pExp->base.pParam[0].pCol =
1,376,656,238✔
2658
        createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType, pColNode->colType);
1,376,575,381✔
2659
    QUERY_CHECK_NULL(pExp->base.pParam[0].pCol, code, lino, _end, terrno);
688,328,330✔
2660

2661
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
688,189,784✔
2662
  } else if (type == QUERY_NODE_VALUE) {
538,135,400✔
2663
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
17,845,463✔
2664
    SValueNode* pValNode = (SValueNode*)pNode;
17,842,736✔
2665

2666
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
17,842,736✔
2667
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
17,838,164✔
2668

2669
    pExp->base.numOfParams = 1;
17,842,351✔
2670

2671
    SDataType* pType = &pValNode->node.resType;
17,836,613✔
2672
    pExp->base.resSchema =
2673
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
17,835,264✔
2674
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
17,836,895✔
2675
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
17,836,539✔
2676
    QUERY_CHECK_CODE(code, lino, _end);
17,840,710✔
2677
  } else if (type == QUERY_NODE_REMOTE_VALUE) {
520,289,937✔
2678
    SRemoteValueNode* pRemote = (SRemoteValueNode*)pNode;
1,447,554✔
2679
    code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, pNode);
1,447,554✔
2680
    QUERY_CHECK_CODE(code, lino, _end);
1,448,109✔
2681

2682
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
1,017,984✔
2683
    SValueNode* pValNode = (SValueNode*)pNode;
1,017,984✔
2684

2685
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
1,017,984✔
2686
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
1,017,984✔
2687

2688
    pExp->base.numOfParams = 1;
1,017,984✔
2689

2690
    SDataType* pType = &pValNode->node.resType;
1,017,984✔
2691
    pExp->base.resSchema =
2692
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
1,017,984✔
2693
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
1,017,984✔
2694
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
1,017,984✔
2695
    QUERY_CHECK_CODE(code, lino, _end);
1,017,984✔
2696
  } else if (type == QUERY_NODE_FUNCTION) {
518,842,383✔
2697
    pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
478,243,531✔
2698
    SFunctionNode* pFuncNode = (SFunctionNode*)pNode;
478,259,364✔
2699

2700
    SDataType* pType = &pFuncNode->node.resType;
478,259,364✔
2701
    pExp->base.resSchema =
2702
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pFuncNode->node.aliasName);
478,264,789✔
2703
    tExprNode* pExprNode = pExp->pExpr;
478,256,950✔
2704

2705
    pExprNode->_function.functionId = pFuncNode->funcId;
478,266,922✔
2706
    pExprNode->_function.pFunctNode = pFuncNode;
478,273,433✔
2707
    pExprNode->_function.functionType = pFuncNode->funcType;
478,332,570✔
2708

2709
    tstrncpy(pExprNode->_function.functionName, pFuncNode->functionName, tListLen(pExprNode->_function.functionName));
478,272,418✔
2710

2711
    pExp->base.pParamList = pFuncNode->pParameterList;
478,263,904✔
2712
#if 1
2713
    // todo refactor: add the parameter for tbname function
2714
    const char* name = "tbname";
478,315,872✔
2715
    int32_t     len = strlen(name);
478,315,872✔
2716

2717
    if (!pFuncNode->pParameterList && (memcmp(pExprNode->_function.functionName, name, len) == 0) &&
478,315,872✔
2718
        pExprNode->_function.functionName[len] == 0) {
31,176,716✔
2719
      pFuncNode->pParameterList = NULL;
31,165,655✔
2720
      int32_t     code = nodesMakeList(&pFuncNode->pParameterList);
31,175,463✔
2721
      SValueNode* res = NULL;
31,174,614✔
2722
      if (TSDB_CODE_SUCCESS == code) {
31,174,614✔
2723
        code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
31,175,288✔
2724
      }
2725
      QUERY_CHECK_CODE(code, lino, _end);
31,176,539✔
2726
      res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
31,176,539✔
2727
      code = nodesListAppend(pFuncNode->pParameterList, (SNode*)res);
31,173,090✔
2728
      if (code != TSDB_CODE_SUCCESS) {
31,178,799✔
2729
        nodesDestroyNode((SNode*)res);
×
2730
        res = NULL;
×
2731
      }
2732
      QUERY_CHECK_CODE(code, lino, _end);
31,178,799✔
2733
    }
2734
#endif
2735

2736
    int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
478,300,356✔
2737

2738
    pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
478,301,779✔
2739
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
478,157,383✔
2740
    pExp->base.numOfParams = numOfParam;
478,149,888✔
2741

2742
    for (int32_t j = 0; j < numOfParam && TSDB_CODE_SUCCESS == code; ++j) {
1,134,017,238✔
2743
      SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
655,812,726✔
2744
      QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
655,871,348✔
2745
      if (p1->type == QUERY_NODE_COLUMN) {
655,871,348✔
2746
        SColumnNode* pcn = (SColumnNode*)p1;
477,614,027✔
2747

2748
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
477,614,027✔
2749
        pExp->base.pParam[j].pCol =
955,139,692✔
2750
            createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType, pcn->colType);
955,161,534✔
2751
        QUERY_CHECK_NULL(pExp->base.pParam[j].pCol, code, lino, _end, terrno);
477,564,073✔
2752
      } else if (p1->type == QUERY_NODE_VALUE) {
178,287,520✔
2753
        SValueNode* pvn = (SValueNode*)p1;
120,460,885✔
2754
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
120,460,885✔
2755
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
120,446,302✔
2756
        QUERY_CHECK_CODE(code, lino, _end);
120,425,902✔
2757
      } else if (p1->type == QUERY_NODE_REMOTE_VALUE) {
57,902,492✔
2758
        SRemoteValueNode* pRemote = (SRemoteValueNode*)p1;
74,370✔
2759
        code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, p1);
74,370✔
2760
        QUERY_CHECK_CODE(code, lino, _end);
74,370✔
2761

2762
        SValueNode* pvn = (SValueNode*)pRemote;
53,835✔
2763
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
53,835✔
2764
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
53,835✔
2765
        QUERY_CHECK_CODE(code, lino, _end);
32✔
2766
      }
2767
    }
2768
    pExp->pExpr->_function.bindExprID = ((SExprNode*)pNode)->bindExprID;
478,204,512✔
2769
  } else if (type == QUERY_NODE_OPERATOR) {
40,598,852✔
2770
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
39,018,639✔
2771
    SOperatorNode* pOpNode = (SOperatorNode*)pNode;
39,018,106✔
2772

2773
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
39,018,106✔
2774
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
39,006,870✔
2775
    pExp->base.numOfParams = 1;
39,009,014✔
2776

2777
    SDataType* pType = &pOpNode->node.resType;
39,011,694✔
2778
    pExp->base.resSchema =
2779
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pOpNode->node.aliasName);
39,013,797✔
2780
    pExp->pExpr->_optrRoot.pRootNode = pNode;
39,015,466✔
2781
  } else if (type == QUERY_NODE_CASE_WHEN) {
1,581,258✔
2782
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
1,495,784✔
2783
    SCaseWhenNode* pCaseNode = (SCaseWhenNode*)pNode;
1,495,784✔
2784

2785
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
1,495,784✔
2786
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
1,495,784✔
2787
    pExp->base.numOfParams = 1;
1,495,784✔
2788

2789
    SDataType* pType = &pCaseNode->node.resType;
1,495,784✔
2790
    pExp->base.resSchema =
2791
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCaseNode->node.aliasName);
1,495,784✔
2792
    pExp->pExpr->_optrRoot.pRootNode = pNode;
1,495,784✔
2793
  } else if (type == QUERY_NODE_LOGIC_CONDITION) {
85,622✔
2794
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
105,284✔
2795
    SLogicConditionNode* pCond = (SLogicConditionNode*)pNode;
105,284✔
2796
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
105,284✔
2797
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
105,827✔
2798
    pExp->base.numOfParams = 1;
105,284✔
2799
    SDataType* pType = &pCond->node.resType;
105,284✔
2800
    pExp->base.resSchema =
2801
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCond->node.aliasName);
105,827✔
2802
    pExp->pExpr->_optrRoot.pRootNode = pNode;
105,284✔
2803
  } else {
2804
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2805
    QUERY_CHECK_CODE(code, lino, _end);
×
2806
  }
2807
  pExp->pExpr->relatedTo = ((SExprNode*)pNode)->relatedTo;
1,226,005,719✔
2808
_end:
1,226,442,351✔
2809
  if (code != TSDB_CODE_SUCCESS) {
1,226,442,351✔
2810
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
450,660✔
2811
  }
2812
  return code;
1,226,475,789✔
2813
}
2814

2815
int32_t createExprFromTargetNode(SExprInfo* pExp, STargetNode* pTargetNode) {
1,226,423,381✔
2816
  return createExprFromOneNode(pExp, pTargetNode->pExpr, pTargetNode->slotId);
1,226,423,381✔
2817
}
2818

2819
SExprInfo* createExpr(SNodeList* pNodeList, int32_t* numOfExprs) {
×
2820
  *numOfExprs = LIST_LENGTH(pNodeList);
×
2821
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
×
2822
  if (!pExprs) {
×
2823
    return NULL;
×
2824
  }
2825

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

2837
  return pExprs;
×
2838
}
2839

2840
int32_t createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, SExprInfo** pExprInfo, int32_t* numOfExprs) {
450,972,221✔
2841
  QRY_PARAM_CHECK(pExprInfo);
450,972,221✔
2842

2843
  int32_t code = 0;
451,031,961✔
2844
  int32_t lino = 0;
451,031,961✔
2845
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
451,031,961✔
2846
  int32_t numOfGroupKeys = 0;
450,931,203✔
2847
  if (pGroupKeys != NULL) {
450,931,203✔
2848
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
31,434,023✔
2849
  }
2850

2851
  *numOfExprs = numOfFuncs + numOfGroupKeys;
450,931,132✔
2852
  if (*numOfExprs == 0) {
450,999,175✔
2853
    return code;
39,283,361✔
2854
  }
2855

2856
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
411,749,413✔
2857
  QUERY_CHECK_NULL(pExprs, code, lino, _return, terrno);
411,496,175✔
2858

2859
  int32_t i = 0;
411,496,175✔
2860
  if (pNodeList != NULL) {
411,496,175✔
2861
    SNode* node = NULL;
383,913,492✔
2862
    FOREACH(node, pNodeList) {
1,554,498,753✔
2863
      STargetNode* pTargetNode = (STargetNode*)node;
1,171,047,840✔
2864
      SExprInfo*   pExp = &pExprs[i++];
1,171,047,840✔
2865
      code = createExprFromTargetNode(pExp, pTargetNode);
1,170,911,986✔
2866
      QUERY_CHECK_CODE(code, lino, _return);
1,171,035,921✔
2867
    }
2868
  }
2869

2870
  if (pGroupKeys != NULL) {
411,320,809✔
2871
    SNode* node = NULL;
31,428,798✔
2872
    FOREACH(node, pGroupKeys) {
86,708,901✔
2873
      STargetNode* pTargetNode = (STargetNode*)node;
55,283,315✔
2874
      SExprInfo*   pExp = &pExprs[i++];
55,283,315✔
2875
      code = createExprFromTargetNode(pExp, pTargetNode);
55,278,790✔
2876
      QUERY_CHECK_CODE(code, lino, _return);
55,280,103✔
2877
    }
2878
  }
2879

2880
  *pExprInfo = pExprs;
411,320,320✔
2881
  return code;
411,245,767✔
2882

2883
_return:
450,660✔
2884
  qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
450,660✔
2885
  destroyExprInfo(pExprs, *numOfExprs);
450,660✔
2886
  taosMemoryFreeClear(pExprs);
450,660✔
2887
  return code;
450,660✔
2888
}
2889

2890
static void deleteSubsidiareCtx(void* pData) {
×
2891
  SSubsidiaryResInfo* pCtx = (SSubsidiaryResInfo*)pData;
×
2892
  if (pCtx->pCtx) {
×
2893
    taosMemoryFreeClear(pCtx->pCtx);
×
2894
  }
2895
}
×
2896

2897
// set the output buffer for the selectivity + tag query
2898
static int32_t setSelectValueColumnInfo(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
438,782,476✔
2899
  int32_t num = 0;
438,782,476✔
2900
  int32_t code = TSDB_CODE_SUCCESS;
438,782,476✔
2901
  int32_t lino = 0;
438,782,476✔
2902

2903
  SArray* pValCtxArray = NULL;
438,782,476✔
2904
  for (int32_t i = numOfOutput - 1; i > 0; --i) {  // select Func is at the end of the list
1,245,112,005✔
2905
    int32_t funcIdx = pCtx[i].pExpr->pExpr->_function.bindExprID;
806,471,689✔
2906
    if (funcIdx > 0) {
806,464,399✔
2907
      if (pValCtxArray == NULL) {
1,606,298✔
2908
        // the end of the list is the select function of biggest index
2909
        pValCtxArray = taosArrayInit_s(sizeof(SSubsidiaryResInfo*), funcIdx);
1,152,625✔
2910
        if (pValCtxArray == NULL) {
1,151,751✔
2911
          return terrno;
×
2912
        }
2913
      }
2914
      if (funcIdx > pValCtxArray->size) {
1,605,424✔
2915
        qError("funcIdx:%d is out of range", funcIdx);
×
2916
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2917
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2918
      }
2919
      SSubsidiaryResInfo* pSubsidiary = &pCtx[i].subsidiaries;
1,605,861✔
2920
      pSubsidiary->pCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
1,604,113✔
2921
      if (pSubsidiary->pCtx == NULL) {
1,605,861✔
2922
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2923
        return terrno;
×
2924
      }
2925
      pSubsidiary->num = 0;
1,604,987✔
2926
      taosArraySet(pValCtxArray, funcIdx - 1, &pSubsidiary);
1,603,239✔
2927
    }
2928
  }
2929

2930
  SqlFunctionCtx*  p = NULL;
438,640,316✔
2931
  SqlFunctionCtx** pValCtx = NULL;
438,640,316✔
2932
  if (pValCtxArray == NULL) {
438,640,316✔
2933
    pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
437,543,505✔
2934
    if (pValCtx == NULL) {
437,544,399✔
2935
      QUERY_CHECK_CODE(terrno, lino, _end);
×
2936
    }
2937
  }
2938

2939
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,651,632,924✔
2940
    const char* pName = pCtx[i].pExpr->pExpr->_function.functionName;
1,212,946,422✔
2941
    if ((strcmp(pName, "_select_value") == 0)) {
1,213,178,400✔
2942
      if (pValCtxArray == NULL) {
8,079,321✔
2943
        pValCtx[num++] = &pCtx[i];
5,822,204✔
2944
      } else {
2945
        int32_t bindFuncIndex = pCtx[i].pExpr->pExpr->relatedTo;  // start from index 1;
2,257,117✔
2946
        if (bindFuncIndex > 0) {                                  // 0 is default index related to the select function
2,255,806✔
2947
          bindFuncIndex -= 1;
2,202,929✔
2948
        }
2949
        SSubsidiaryResInfo** pSubsidiary = taosArrayGet(pValCtxArray, bindFuncIndex);
2,255,806✔
2950
        if (pSubsidiary == NULL) {
2,255,369✔
2951
          QUERY_CHECK_CODE(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR, lino, _end);
×
2952
        }
2953
        (*pSubsidiary)->pCtx[(*pSubsidiary)->num] = &pCtx[i];
2,255,369✔
2954
        (*pSubsidiary)->num++;
2,255,806✔
2955
      }
2956
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
1,205,099,079✔
2957
      if (pValCtxArray == NULL) {
120,207,911✔
2958
        p = &pCtx[i];
118,248,402✔
2959
      }
2960
    }
2961
  }
2962

2963
  if (p != NULL) {
438,686,502✔
2964
    p->subsidiaries.pCtx = pValCtx;
52,035,768✔
2965
    p->subsidiaries.num = num;
52,037,114✔
2966
  } else {
2967
    taosMemoryFreeClear(pValCtx);
386,650,734✔
2968
  }
2969

2970
_end:
1,205,093✔
2971
  if (code != TSDB_CODE_SUCCESS) {
438,617,258✔
2972
    taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2973
    taosMemoryFreeClear(pValCtx);
×
2974
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2975
  } else {
2976
    taosArrayDestroy(pValCtxArray);
438,617,258✔
2977
  }
2978
  return code;
438,729,452✔
2979
}
2980

2981
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowEntryInfoOffset,
438,789,932✔
2982
                                     SFunctionStateStore* pStore) {
2983
  int32_t         code = TSDB_CODE_SUCCESS;
438,789,932✔
2984
  int32_t         lino = 0;
438,789,932✔
2985
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
438,789,932✔
2986
  if (pFuncCtx == NULL) {
438,539,885✔
2987
    return NULL;
×
2988
  }
2989

2990
  *rowEntryInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
438,539,885✔
2991
  if (*rowEntryInfoOffset == 0) {
438,734,077✔
2992
    taosMemoryFreeClear(pFuncCtx);
×
2993
    return NULL;
×
2994
  }
2995

2996
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,651,864,488✔
2997
    SExprInfo* pExpr = &pExprInfo[i];
1,213,183,918✔
2998

2999
    SExprBasicInfo* pFunct = &pExpr->base;
1,213,095,536✔
3000
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
1,213,151,131✔
3001

3002
    pCtx->functionId = -1;
1,213,156,875✔
3003
    pCtx->pExpr = pExpr;
1,213,186,610✔
3004

3005
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
1,213,113,882✔
3006
      SFuncExecEnv env = {0};
476,919,078✔
3007
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;
476,935,866✔
3008
      pCtx->isPseudoFunc = fmIsWindowPseudoColumnFunc(pCtx->functionId) || fmIsPlaceHolderFunc(pCtx->functionId);
476,947,509✔
3009
      pCtx->isNotNullFunc = fmIsNotNullOutputFunc(pCtx->functionId);
476,896,060✔
3010

3011
      bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
476,930,020✔
3012
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId)) {
799,418,150✔
3013
        if (!isUdaf) {
322,701,499✔
3014
          code = fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
322,664,957✔
3015
          QUERY_CHECK_CODE(code, lino, _end);
322,633,503✔
3016
        } else {
3017
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
36,542✔
3018
          pCtx->udfName = taosStrdup(udfName);
36,542✔
3019
          QUERY_CHECK_NULL(pCtx->udfName, code, lino, _end, terrno);
36,542✔
3020

3021
          code = fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
36,542✔
3022
          QUERY_CHECK_CODE(code, lino, _end);
36,542✔
3023
        }
3024
        bool tmp = pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
322,670,045✔
3025
        if (!tmp) {
322,603,877✔
3026
          code = terrno;
×
3027
          QUERY_CHECK_CODE(code, lino, _end);
111✔
3028
        }
3029
      } else {
3030
        if (fmIsPlaceHolderFunc(pCtx->functionId)) {
154,177,245✔
3031
          code = fmGetStreamPesudoFuncEnv(pCtx->functionId, pExpr->base.pParamList, &env);
7,083,588✔
3032
          QUERY_CHECK_CODE(code, lino, _end);
7,083,169✔
3033
        }      
3034
        
3035
        code = fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
154,209,462✔
3036
        if (code != TSDB_CODE_SUCCESS && isUdaf) {
154,193,302✔
3037
          code = TSDB_CODE_SUCCESS;
61,704✔
3038
        }
3039
        QUERY_CHECK_CODE(code, lino, _end);
154,193,302✔
3040

3041
        if (pCtx->sfp.getEnv != NULL) {
154,193,302✔
3042
          bool tmp = pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
27,458,148✔
3043
          if (!tmp) {
27,462,513✔
3044
            code = terrno;
×
3045
            QUERY_CHECK_CODE(code, lino, _end);
×
3046
          }
3047
        }
3048
      }
3049
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
476,809,346✔
3050
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
736,220,397✔
3051
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
18,848,521✔
3052
      // for simple column, the result buffer needs to hold at least one element.
3053
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
736,341,319✔
3054
    }
3055

3056
    pCtx->input.numOfInputCols = pFunct->numOfParams;
1,213,249,355✔
3057
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
1,213,153,498✔
3058
    QUERY_CHECK_NULL(pCtx->input.pData, code, lino, _end, terrno);
1,213,301,533✔
3059
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
1,213,217,116✔
3060
    QUERY_CHECK_NULL(pCtx->input.pColumnDataAgg, code, lino, _end, terrno);
1,213,277,520✔
3061

3062
    pCtx->pTsOutput = NULL;
1,213,173,830✔
3063
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
1,213,271,449✔
3064
    pCtx->resDataInfo.type = pFunct->resSchema.type;
1,213,161,964✔
3065
    pCtx->order = TSDB_ORDER_ASC;
1,213,256,556✔
3066
    pCtx->start.key = INT64_MIN;
1,213,257,642✔
3067
    pCtx->end.key = INT64_MIN;
1,213,253,319✔
3068
    pCtx->numOfParams = pExpr->base.numOfParams;
1,213,186,967✔
3069
    pCtx->param = pFunct->pParam;
1,213,354,464✔
3070
    pCtx->saveHandle.currentPage = -1;
1,213,227,932✔
3071
    pCtx->pStore = pStore;
1,213,317,045✔
3072
    pCtx->hasWindowOrGroup = false;
1,213,315,809✔
3073
    pCtx->needCleanup = false;
1,213,171,634✔
3074
    pCtx->skipDynDataCheck = false;
1,213,119,208✔
3075
  }
3076

3077
  for (int32_t i = 1; i < numOfOutput; ++i) {
1,245,365,503✔
3078
    (*rowEntryInfoOffset)[i] = (int32_t)((*rowEntryInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) +
1,613,068,158✔
3079
                                         pFuncCtx[i - 1].resDataInfo.interBufSize);
806,563,654✔
3080
  }
3081

3082
  code = setSelectValueColumnInfo(pFuncCtx, numOfOutput);
438,780,644✔
3083
  QUERY_CHECK_CODE(code, lino, _end);
438,737,394✔
3084

3085
_end:
438,737,394✔
3086
  if (code != TSDB_CODE_SUCCESS) {
438,690,327✔
3087
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3088
    for (int32_t i = 0; i < numOfOutput; ++i) {
×
3089
      taosMemoryFree(pFuncCtx[i].input.pData);
×
3090
      taosMemoryFree(pFuncCtx[i].input.pColumnDataAgg);
×
3091
    }
3092
    taosMemoryFreeClear(*rowEntryInfoOffset);
×
3093
    taosMemoryFreeClear(pFuncCtx);
×
3094

3095
    terrno = code;
×
3096
    return NULL;
×
3097
  }
3098
  return pFuncCtx;
438,690,327✔
3099
}
3100

3101
// NOTE: sources columns are more than the destination SSDatablock columns.
3102
// doFilter in table scan needs every column even its output is false
3103
int32_t relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray* pCols, bool outputEveryColumn) {
15,314,751✔
3104
  int32_t code = TSDB_CODE_SUCCESS;
15,314,751✔
3105
  size_t  numOfSrcCols = taosArrayGetSize(pCols);
15,314,751✔
3106

3107
  int32_t i = 0, j = 0;
15,315,526✔
3108
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
123,320,459✔
3109
    SColumnInfoData* p = taosArrayGet(pCols, i);
108,006,295✔
3110
    if (!p) {
108,005,609✔
3111
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3112
      return terrno;
×
3113
    }
3114
    SColMatchItem* pmInfo = taosArrayGet(pColMatchInfo, j);
108,005,609✔
3115
    if (!pmInfo) {
108,004,558✔
3116
      return terrno;
×
3117
    }
3118

3119
    if (p->info.colId == pmInfo->colId) {
108,004,558✔
3120
      SColumnInfoData* pDst = taosArrayGet(pBlock->pDataBlock, pmInfo->dstSlotId);
100,936,850✔
3121
      if (!pDst) {
100,935,721✔
3122
        return terrno;
×
3123
      }
3124
      code = colDataAssign(pDst, p, pBlock->info.rows, &pBlock->info);
100,935,721✔
3125
      if (code != TSDB_CODE_SUCCESS) {
100,935,081✔
3126
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3127
        return code;
×
3128
      }
3129
      i++;
100,935,081✔
3130
      j++;
100,935,081✔
3131
    } else if (p->info.colId < pmInfo->colId) {
7,069,852✔
3132
      i++;
7,069,852✔
3133
    } else {
3134
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3135
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3136
    }
3137
  }
3138
  return code;
15,315,613✔
3139
}
3140

3141
SInterval extractIntervalInfo(const STableScanPhysiNode* pTableScanNode) {
234,583,237✔
3142
  SInterval interval = {
468,996,326✔
3143
      .interval = pTableScanNode->interval,
234,458,025✔
3144
      .sliding = pTableScanNode->sliding,
234,443,934✔
3145
      .intervalUnit = pTableScanNode->intervalUnit,
234,611,836✔
3146
      .slidingUnit = pTableScanNode->slidingUnit,
234,521,524✔
3147
      .offset = pTableScanNode->offset,
234,608,821✔
3148
      .precision = pTableScanNode->scan.node.pOutputDataBlockDesc->precision,
234,615,607✔
3149
      .timeRange = pTableScanNode->scanRange,
3150
  };
3151
  calcIntervalAutoOffset(&interval);
234,396,575✔
3152

3153
  return interval;
234,477,048✔
3154
}
3155

3156
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
60,019,310✔
3157
  SColumn c = {0};
60,019,310✔
3158

3159
  c.slotId = pColNode->slotId;
60,019,310✔
3160
  c.colId = pColNode->colId;
60,015,604✔
3161
  c.type = pColNode->node.resType.type;
60,016,207✔
3162
  c.bytes = pColNode->node.resType.bytes;
60,012,574✔
3163
  c.scale = pColNode->node.resType.scale;
60,020,134✔
3164
  c.precision = pColNode->node.resType.precision;
60,020,607✔
3165
  return c;
60,009,775✔
3166
}
3167

3168

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

3183
  if (cond->skey > cond->ekey || range->skey > range->ekey) {
3,301,048✔
3184
    *twindow = extTwindows[0] = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
5,467✔
3185
    return code;
5,040✔
3186
  }
3187

3188
  if (range->ekey < cond->skey) {
3,295,581✔
3189
    extTwindows[1] = *cond;
516,802✔
3190
    *twindow = extTwindows[0] = TSWINDOW_DESC_INITIALIZER;
516,802✔
3191
    return code;
516,802✔
3192
  }
3193

3194
  if (cond->ekey < range->skey) {
2,779,206✔
3195
    extTwindows[0] = *cond;
363,415✔
3196
    *twindow = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
363,415✔
3197
    return code;
363,415✔
3198
  }
3199

3200
  // Only scan data in the time range intersecion.
3201
  extTwindows[0] = extTwindows[1] = *cond;
2,415,364✔
3202
  twindow->skey = TMAX(cond->skey, range->skey);
2,415,791✔
3203
  twindow->ekey = TMIN(cond->ekey, range->ekey);
2,415,364✔
3204
  extTwindows[0].ekey = twindow->skey - 1;
2,415,791✔
3205
  extTwindows[1].skey = twindow->ekey + 1;
2,415,364✔
3206

3207
  return code;
2,415,791✔
3208
}
3209

3210
static int32_t getPrimaryTimeRange(SNode** pPrimaryKeyCond, STimeWindow* pTimeRange, bool* isStrict) {
634,500✔
3211
  SNode*  pNew = NULL;
634,500✔
3212
  int32_t code = scalarCalculateRemoteConstants(*pPrimaryKeyCond, &pNew);
634,500✔
3213
  if (TSDB_CODE_SUCCESS == code) {
634,500✔
3214
    *pPrimaryKeyCond = pNew;
634,500✔
3215
    if (nodeType(pNew) != QUERY_NODE_VALUE) {
634,500✔
3216
      code = filterGetTimeRange(*pPrimaryKeyCond, pTimeRange, isStrict, NULL);
634,500✔
3217
    }
3218
  }
3219
  return code;
634,500✔
3220
}
3221

3222
int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, STableScanPhysiNode* pTableScanNode,
263,168,241✔
3223
                               const SReadHandle* readHandle, bool applyExtWin) {
3224
  int32_t code = 0;                             
263,168,241✔
3225
  pCond->order = pTableScanNode->scanSeq[0] > 0 ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
263,168,241✔
3226
  pCond->numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
263,167,235✔
3227

3228
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
263,207,724✔
3229
  if (!pCond->colList) {
263,117,643✔
3230
    return terrno;
×
3231
  }
3232
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
263,039,538✔
3233
  if (pCond->pSlotList == NULL) {
263,158,318✔
3234
    taosMemoryFreeClear(pCond->colList);
×
3235
    return terrno;
×
3236
  }
3237

3238
  // TODO: get it from stable scan node
3239
  pCond->twindows = pTableScanNode->scanRange;
262,960,348✔
3240
  pCond->suid = pTableScanNode->scan.suid;
263,207,754✔
3241
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
263,047,684✔
3242
  pCond->startVersion = -1;
263,158,326✔
3243
  pCond->endVersion = -1;
263,186,452✔
3244
  pCond->skipRollup = readHandle->skipRollup;
263,135,519✔
3245
  if (readHandle->winRangeValid) {
263,134,140✔
3246
    pCond->twindows = readHandle->winRange;
469,842✔
3247
  }
3248
  pCond->cacheSttStatis = readHandle->cacheSttStatis;
263,210,458✔
3249
  // allowed read stt file optimization mode
3250
  pCond->notLoadData = (pTableScanNode->dataRequired == FUNC_DATA_REQUIRED_NOT_LOAD) &&
526,374,178✔
3251
                       (pTableScanNode->scan.node.pConditions == NULL) && (pTableScanNode->interval == 0);
263,149,290✔
3252

3253
  int32_t j = 0;
263,213,618✔
3254
  SNode*  node = NULL;
263,213,618✔
3255
  FOREACH(node, pTableScanNode->scan.pScanCols) {
1,233,117,753✔
3256
    STargetNode* pNode = (STargetNode*)node;
969,606,806✔
3257
    if (!pNode) {
969,606,806✔
3258
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3259
      return terrno;
×
3260
    }
3261
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
969,606,806✔
3262
    if (pColNode->colType == COLUMN_TYPE_TAG) {
969,851,549✔
3263
      continue;
×
3264
    }
3265

3266
    pCond->colList[j].type = pColNode->node.resType.type;
969,781,553✔
3267
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
969,763,698✔
3268
    pCond->colList[j].colId = pColNode->colId;
969,861,242✔
3269
    pCond->colList[j].pk = pColNode->isPk;
970,048,561✔
3270

3271
    pCond->pSlotList[j] = pNode->slotId;
969,749,208✔
3272
    j += 1;
969,904,135✔
3273
  }
3274

3275
  pCond->numOfCols = j;
263,165,362✔
3276

3277
  if (applyExtWin) {
263,170,440✔
3278
    if (NULL != pTableScanNode->pExtScanRange) {
235,173,266✔
3279
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
3,229,480✔
3280
      code = getQueryExtWindow(&pCond->twindows, pTableScanNode->pExtScanRange, &pCond->twindows, pCond->extTwindows);
3,229,480✔
3281
    } else if (readHandle->extWinRangeValid) {
231,940,780✔
3282
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
71,568✔
3283
      code = getQueryExtWindow(&pCond->twindows, &readHandle->extWinRange, &pCond->twindows, pCond->extTwindows);
71,568✔
3284
    }
3285
  }
3286

3287
  if (pTableScanNode->pPrimaryCond) {
263,156,210✔
3288
    bool isStrict = false;
634,500✔
3289
    code = getPrimaryTimeRange((SNode**)&pTableScanNode->pPrimaryCond, &pCond->twindows, &isStrict);
634,500✔
3290
    if (code || !isStrict) {
634,500✔
3291
      code = nodesMergeNode((SNode**)&pTableScanNode->scan.node.pConditions, &pTableScanNode->pPrimaryCond);
622,464✔
3292
    }
3293
  }
3294

3295
  return code;
263,148,133✔
3296
}
3297

3298
int32_t initQueryTableDataCondWithColArray(SQueryTableDataCond* pCond, SQueryTableDataCond* pOrgCond,
104,418,545✔
3299
                                           const SReadHandle* readHandle, SArray* colArray) {
3300
  int32_t code = TSDB_CODE_SUCCESS;
104,418,545✔
3301
  int32_t lino = 0;
104,418,545✔
3302

3303
  pCond->order = TSDB_ORDER_ASC;
104,418,545✔
3304
  pCond->numOfCols = (int32_t)taosArrayGetSize(colArray);
104,451,949✔
3305

3306
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
104,462,167✔
3307
  QUERY_CHECK_NULL(pCond->colList, code, lino, _return, terrno);
104,423,887✔
3308

3309
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
104,422,936✔
3310
  QUERY_CHECK_NULL(pCond->pSlotList, code, lino, _return, terrno);
104,439,073✔
3311

3312
  pCond->twindows = pOrgCond->twindows;
104,437,325✔
3313
  pCond->order = pOrgCond->order;
104,457,890✔
3314
  pCond->type = pOrgCond->type;
104,474,926✔
3315
  pCond->startVersion = -1;
104,475,707✔
3316
  pCond->endVersion = -1;
104,473,501✔
3317
  pCond->skipRollup = true;
104,471,085✔
3318
  pCond->notLoadData = false;
104,462,214✔
3319

3320
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
469,975,527✔
3321
    SColIdPair* pColPair = taosArrayGet(colArray, i);
365,490,558✔
3322
    QUERY_CHECK_NULL(pColPair, code, lino, _return, terrno);
365,535,943✔
3323

3324
    bool find = false;
365,530,951✔
3325
    for (int32_t j = 0; j < pOrgCond->numOfCols; ++j) {
1,980,437,881✔
3326
      if (pOrgCond->colList[j].colId == pColPair->vtbColId) {
1,980,133,471✔
3327
        pCond->colList[i].type = pOrgCond->colList[j].type;
365,526,734✔
3328
        // For variable-length types (nchar/binary/varchar/varbinary), use the source table's bytes
3329
        // to avoid tsdb reader doCopyColVal length check failure when source data is longer
3330
        // than the virtual table's defined length.
3331
        if (IS_VAR_DATA_TYPE(pOrgCond->colList[j].type) && pColPair->type.bytes > 0) {
365,540,520✔
3332
          pCond->colList[i].bytes = TMAX(pOrgCond->colList[j].bytes, pColPair->type.bytes);
46,447,562✔
3333
        } else {
3334
          pCond->colList[i].bytes = pOrgCond->colList[j].bytes;
319,127,891✔
3335
        }
3336
        pCond->colList[i].colId = pColPair->orgColId;
365,545,443✔
3337
        pCond->colList[i].pk = pOrgCond->colList[j].pk;
365,549,744✔
3338
        pCond->pSlotList[i] = i;
365,542,049✔
3339
        find = true;
365,560,545✔
3340
        qDebug("%s mapped vtb colId:%d to org colId:%d, type:%d, bytes:%d", __func__, pColPair->vtbColId,
365,560,545✔
3341
               pColPair->orgColId, pCond->colList[i].type, pCond->colList[i].bytes);
3342
        break;
365,479,402✔
3343
      }
3344
    }
3345
    QUERY_CHECK_CONDITION(find, code, lino, _return, TSDB_CODE_NOT_FOUND);
365,506,516✔
3346
  }
3347

3348
  return code;
104,483,808✔
3349
_return:
×
3350
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(terrno));
×
3351
  taosMemoryFreeClear(pCond->colList);
×
3352
  taosMemoryFreeClear(pCond->pSlotList);
×
3353
  return code;
×
3354
}
3355

3356
void cleanupQueryTableDataCond(SQueryTableDataCond* pCond) {
652,472,902✔
3357
  taosMemoryFreeClear(pCond->colList);
652,472,902✔
3358
  taosMemoryFreeClear(pCond->pSlotList);
652,594,207✔
3359
}
652,521,122✔
3360

3361
int32_t convertFillType(int32_t mode) {
3,974,941✔
3362
  int32_t type = TSDB_FILL_NONE;
3,974,941✔
3363
  switch (mode) {
3,974,941✔
3364
    case FILL_MODE_PREV:
200,758✔
3365
      type = TSDB_FILL_PREV;
200,758✔
3366
      break;
200,758✔
3367
    case FILL_MODE_NONE:
×
3368
      type = TSDB_FILL_NONE;
×
3369
      break;
×
3370
    case FILL_MODE_NULL:
215,990✔
3371
      type = TSDB_FILL_NULL;
215,990✔
3372
      break;
215,990✔
3373
    case FILL_MODE_NULL_F:
33,763✔
3374
      type = TSDB_FILL_NULL_F;
33,763✔
3375
      break;
33,763✔
3376
    case FILL_MODE_NEXT:
196,119✔
3377
      type = TSDB_FILL_NEXT;
196,119✔
3378
      break;
196,119✔
3379
    case FILL_MODE_VALUE:
168,695✔
3380
      type = TSDB_FILL_SET_VALUE;
168,695✔
3381
      break;
168,695✔
3382
    case FILL_MODE_VALUE_F:
11,826✔
3383
      type = TSDB_FILL_SET_VALUE_F;
11,826✔
3384
      break;
11,826✔
3385
    case FILL_MODE_LINEAR:
242,001✔
3386
      type = TSDB_FILL_LINEAR;
242,001✔
3387
      break;
242,001✔
3388
    case FILL_MODE_NEAR:
2,905,789✔
3389
      type = TSDB_FILL_NEAR;
2,905,789✔
3390
      break;
2,905,789✔
UNCOV
3391
    default:
×
UNCOV
3392
      type = TSDB_FILL_NONE;
×
3393
  }
3394

3395
  return type;
3,974,941✔
3396
}
3397

3398
void getInitialStartTimeWindow(SInterval* pInterval, TSKEY ts, STimeWindow* w, bool ascQuery) {
2,147,483,647✔
3399
  if (ascQuery) {
2,147,483,647✔
3400
    *w = getAlignQueryTimeWindow(pInterval, ts);
2,147,483,647✔
3401
  } else {
3402
    // the start position of the first time window in the endpoint that spreads beyond the queried last timestamp
3403
    *w = getAlignQueryTimeWindow(pInterval, ts);
22,474✔
3404

3405
    int64_t key = w->skey;
260,788✔
3406
    while (key < ts) {  // moving towards end
277,672✔
3407
      key = getNextTimeWindowStart(pInterval, key, TSDB_ORDER_ASC);
141,722✔
3408
      if (key > ts) {
141,722✔
3409
        break;
124,838✔
3410
      }
3411

3412
      w->skey = key;
16,884✔
3413
    }
3414
    w->ekey = taosTimeAdd(w->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
260,788✔
3415
  }
3416
}
2,147,483,647✔
3417

3418
static STimeWindow doCalculateTimeWindow(int64_t ts, SInterval* pInterval) {
74,129,252✔
3419
  STimeWindow w = {0};
74,129,252✔
3420

3421
  w.skey = taosTimeTruncate(ts, pInterval);
74,129,252✔
3422
  w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
74,126,680✔
3423
  return w;
74,139,942✔
3424
}
3425

3426
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order) {
1,540,087✔
3427
  STimeWindow win = *pWindow;
1,540,087✔
3428
  STimeWindow save = win;
1,540,087✔
3429
  while (win.skey <= ts && win.ekey >= ts) {
8,579,370✔
3430
    save = win;
7,039,283✔
3431
    // get previous time window
3432
    getNextTimeWindow(pInterval, &win, order == TSDB_ORDER_DESC ? TSDB_ORDER_ASC : TSDB_ORDER_DESC);
7,039,283✔
3433
  }
3434

3435
  return save;
1,540,087✔
3436
}
3437

3438
// get the correct time window according to the handled timestamp
3439
// todo refactor
3440
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
113,996,165✔
3441
                                int32_t order) {
3442
  STimeWindow w = {0};
113,996,165✔
3443
  if (pResultRowInfo->cur.pageId == -1) {  // the first window, from the previous stored value
114,005,970✔
3444
    getInitialStartTimeWindow(pInterval, ts, &w, (order != TSDB_ORDER_DESC));
3,225,124✔
3445
    return w;
3,224,209✔
3446
  }
3447

3448
  SResultRow* pRow = getResultRowByPos(pBuf, &pResultRowInfo->cur, false);
110,780,546✔
3449
  if (pRow) {
110,783,189✔
3450
    TAOS_SET_OBJ_ALIGNED(&w, pRow->win);
110,787,216✔
3451
  }
3452

3453
  // in case of typical time window, we can calculate time window directly.
3454
  if (w.skey > ts || w.ekey < ts) {
110,785,780✔
3455
    w = doCalculateTimeWindow(ts, pInterval);
74,133,601✔
3456
  }
3457

3458
  if (pInterval->interval != pInterval->sliding) {
110,792,121✔
3459
    // it is an sliding window query, in which sliding value is not equalled to
3460
    // interval value, and we need to find the first qualified time window.
3461
    w = getFirstQualifiedTimeWindow(ts, &w, pInterval, order);
1,540,087✔
3462
  }
3463

3464
  return w;
110,780,047✔
3465
}
3466

3467
void getNextTimeWindow(const SInterval* pInterval, STimeWindow* tw, int32_t order) {
2,147,483,647✔
3468
  tw->skey = getNextTimeWindowStart(pInterval, tw->skey, order);
2,147,483,647✔
3469
  tw->ekey = taosTimeGetIntervalEnd(tw->skey, pInterval);
2,147,483,647✔
3470
}
2,147,483,647✔
3471

3472
bool hasLimitOffsetInfo(SLimitInfo* pLimitInfo) {
517,195,820✔
3473
  return (pLimitInfo->limit.limit != -1 || pLimitInfo->limit.offset != -1 || pLimitInfo->slimit.limit != -1 ||
1,030,239,080✔
3474
          pLimitInfo->slimit.offset != -1);
513,043,762✔
3475
}
3476

3477
bool hasSlimitOffsetInfo(SLimitInfo* pLimitInfo) {
×
3478
  return (pLimitInfo->slimit.limit != -1 || pLimitInfo->slimit.offset != -1);
×
3479
}
3480

3481
void initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo) {
638,866,236✔
3482
  SLimit limit = {.limit = getLimit(pLimit), .offset = getOffset(pLimit)};
638,866,236✔
3483
  SLimit slimit = {.limit = getLimit(pSLimit), .offset = getOffset(pSLimit)};
638,679,269✔
3484

3485
  pLimitInfo->limit = limit;
638,648,422✔
3486
  pLimitInfo->slimit = slimit;
638,671,525✔
3487
  pLimitInfo->remainOffset = limit.offset;
638,693,963✔
3488
  pLimitInfo->remainGroupOffset = slimit.offset;
638,669,270✔
3489
  pLimitInfo->numOfOutputRows = 0;
638,732,170✔
3490
  pLimitInfo->numOfOutputGroups = 0;
638,872,123✔
3491
  pLimitInfo->currentGroupId = 0;
638,867,805✔
3492
}
638,823,231✔
3493

3494
void resetLimitInfoForNextGroup(SLimitInfo* pLimitInfo) {
81,458,408✔
3495
  pLimitInfo->numOfOutputRows = 0;
81,458,408✔
3496
  pLimitInfo->remainOffset = pLimitInfo->limit.offset;
81,474,123✔
3497
}
81,436,653✔
3498

3499
int32_t tableListGetSize(const STableListInfo* pTableList, int32_t* pRes) {
669,214,030✔
3500
  if (taosArrayGetSize(pTableList->pTableList) != taosHashGetSize(pTableList->map)) {
669,214,030✔
3501
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3502
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3503
  }
3504
  (*pRes) = taosArrayGetSize(pTableList->pTableList);
669,178,702✔
3505
  return TSDB_CODE_SUCCESS;
669,174,907✔
3506
}
3507

3508
uint64_t tableListGetSuid(const STableListInfo* pTableList) { return pTableList->idInfo.suid; }
4,053,588✔
3509

3510
STableKeyInfo* tableListGetInfo(const STableListInfo* pTableList, int32_t index) {
214,912,649✔
3511
  if (taosArrayGetSize(pTableList->pTableList) == 0) {
214,912,649✔
3512
    return NULL;
2,203✔
3513
  }
3514

3515
  return taosArrayGet(pTableList->pTableList, index);
214,861,081✔
3516
}
3517

3518
int32_t tableListFind(const STableListInfo* pTableList, uint64_t uid, int32_t startIndex) {
48,263✔
3519
  int32_t numOfTables = taosArrayGetSize(pTableList->pTableList);
48,263✔
3520
  if (startIndex >= numOfTables) {
48,263✔
3521
    return -1;
×
3522
  }
3523

3524
  for (int32_t i = startIndex; i < numOfTables; ++i) {
386,426✔
3525
    STableKeyInfo* p = taosArrayGet(pTableList->pTableList, i);
386,426✔
3526
    if (!p) {
386,426✔
3527
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3528
      return -1;
×
3529
    }
3530
    if (p->uid == uid) {
386,426✔
3531
      return i;
48,263✔
3532
    }
3533
  }
3534
  return -1;
×
3535
}
3536

3537
void tableListGetSourceTableInfo(const STableListInfo* pTableList, uint64_t* psuid, uint64_t* uid, int32_t* type) {
403,108✔
3538
  *psuid = pTableList->idInfo.suid;
403,108✔
3539
  *uid = pTableList->idInfo.uid;
403,108✔
3540
  *type = pTableList->idInfo.tableType;
403,108✔
3541
}
403,108✔
3542

3543
uint64_t tableListGetTableGroupId(const STableListInfo* pTableList, uint64_t tableUid) {
836,856,815✔
3544
  int32_t* slot = taosHashGet(pTableList->map, &tableUid, sizeof(tableUid));
836,856,815✔
3545
  if (slot == NULL) {
837,193,438✔
3546
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3547
    return -1;
×
3548
  }
3549

3550
  STableKeyInfo* pKeyInfo = taosArrayGet(pTableList->pTableList, *slot);
837,193,438✔
3551
  if (pKeyInfo == NULL) {
837,328,455✔
3552
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3553
    return -1;
×
3554
  }
3555
  return pKeyInfo->groupId;
837,328,455✔
3556
}
3557

3558
// TODO handle the group offset info, fix it, the rule of group output will be broken by this function
3559
// int32_t tableListRemoveTableInfo(STableListInfo* pTableList, uint64_t uid) {
3560
//   int32_t code = TSDB_CODE_SUCCESS;
3561
//   int32_t lino = 0;
3562

3563
//   int32_t* slot = taosHashGet(pTableList->map, &uid, sizeof(uid));
3564
//   if (slot == NULL) {
3565
//     qDebug("table:%" PRIu64 " not found in table list", uid);
3566
//     return 0;
3567
//   }
3568

3569
//   taosArrayRemove(pTableList->pTableList, *slot);
3570
//   code = taosHashRemove(pTableList->map, &uid, sizeof(uid));
3571

3572
//   _end:
3573
//   if (code != TSDB_CODE_SUCCESS) {
3574
//     qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
3575
//   } else {
3576
//     qDebug("uid:%" PRIu64 ", remove from table list", uid);
3577
//   }
3578

3579
//   return code;
3580
// }
3581

3582
int32_t tableListAddTableInfo(STableListInfo* pTableList, uint64_t uid, uint64_t gid) {
1,069,592✔
3583
  int32_t code = TSDB_CODE_SUCCESS;
1,069,592✔
3584
  int32_t lino = 0;
1,069,592✔
3585
  if (pTableList->map == NULL) {
1,069,592✔
3586
    pTableList->map = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
×
3587
    QUERY_CHECK_NULL(pTableList->map, code, lino, _end, terrno);
×
3588
  }
3589

3590
  STableKeyInfo keyInfo = {.uid = uid, .groupId = gid};
1,069,911✔
3591
  void*         p = taosHashGet(pTableList->map, &uid, sizeof(uid));
1,070,868✔
3592
  if (p != NULL) {
1,069,911✔
3593
    qInfo("table:%" PRId64 " already in tableIdList, ignore it", uid);
×
3594
    goto _end;
×
3595
  }
3596

3597
  void* tmp = taosArrayPush(pTableList->pTableList, &keyInfo);
1,069,911✔
3598
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,069,911✔
3599

3600
  int32_t slot = (int32_t)taosArrayGetSize(pTableList->pTableList) - 1;
1,069,911✔
3601
  code = taosHashPut(pTableList->map, &uid, sizeof(uid), &slot, sizeof(slot));
1,069,592✔
3602
  if (code != TSDB_CODE_SUCCESS) {
1,070,868✔
3603
    // we have checked the existence of uid in hash map above
3604
    QUERY_CHECK_CONDITION((code != TSDB_CODE_DUP_KEY), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
3605
    taosArrayPopTailBatch(pTableList->pTableList, 1);  // let's pop the last element in the array list
×
3606
  }
3607

3608
_end:
1,070,868✔
3609
  if (code != TSDB_CODE_SUCCESS) {
1,070,230✔
3610
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3611
  } else {
3612
    qDebug("uid:%" PRIu64 ", groupId:%" PRIu64 " added into table list, slot:%d, total:%d", uid, gid, slot, slot + 1);
1,070,230✔
3613
  }
3614

3615
  return code;
1,070,549✔
3616
}
3617

3618
int32_t tableListGetGroupList(const STableListInfo* pTableList, int32_t ordinalGroupIndex, STableKeyInfo** pKeyInfo,
252,096,374✔
3619
                              int32_t* size) {
3620
  int32_t totalGroups = tableListGetOutputGroups(pTableList);
252,096,374✔
3621
  int32_t numOfTables = 0;
252,110,060✔
3622
  int32_t code = tableListGetSize(pTableList, &numOfTables);
252,133,225✔
3623
  if (code != TSDB_CODE_SUCCESS) {
252,078,856✔
3624
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3625
    return code;
×
3626
  }
3627

3628
  if (ordinalGroupIndex < 0 || ordinalGroupIndex >= totalGroups) {
252,078,856✔
3629
    return TSDB_CODE_INVALID_PARA;
×
3630
  }
3631

3632
  // here handle two special cases:
3633
  // 1. only one group exists, and 2. one table exists for each group.
3634
  if (totalGroups == 1) {
252,078,856✔
3635
    *size = numOfTables;
251,550,860✔
3636
    *pKeyInfo = (*size == 0) ? NULL : taosArrayGet(pTableList->pTableList, 0);
251,534,090✔
3637
    return TSDB_CODE_SUCCESS;
251,556,676✔
3638
  } else if (totalGroups == numOfTables) {
528,053✔
3639
    *size = 1;
490,519✔
3640
    *pKeyInfo = taosArrayGet(pTableList->pTableList, ordinalGroupIndex);
490,519✔
3641
    return TSDB_CODE_SUCCESS;
490,947✔
3642
  }
3643

3644
  int32_t offset = pTableList->groupOffset[ordinalGroupIndex];
39,064✔
3645
  if (ordinalGroupIndex < totalGroups - 1) {
59,598✔
3646
    *size = pTableList->groupOffset[ordinalGroupIndex + 1] - offset;
44,830✔
3647
  } else {
3648
    *size = numOfTables - offset;
14,768✔
3649
  }
3650

3651
  *pKeyInfo = taosArrayGet(pTableList->pTableList, offset);
59,598✔
3652
  return TSDB_CODE_SUCCESS;
59,598✔
3653
}
3654

3655
int32_t tableListGetOutputGroups(const STableListInfo* pTableList) { return pTableList->numOfOuputGroups; }
708,019,363✔
3656

3657
bool oneTableForEachGroup(const STableListInfo* pTableList) { return pTableList->oneTableForEachGroup; }
873,170✔
3658

3659
STableListInfo* tableListCreate() {
282,428,907✔
3660
  STableListInfo* pListInfo = taosMemoryCalloc(1, sizeof(STableListInfo));
282,428,907✔
3661
  if (pListInfo == NULL) {
282,285,115✔
3662
    return NULL;
×
3663
  }
3664

3665
  pListInfo->remainGroups = NULL;
282,285,115✔
3666
  pListInfo->pTableList = taosArrayInit(4, sizeof(STableKeyInfo));
282,313,067✔
3667
  if (pListInfo->pTableList == NULL) {
282,319,558✔
3668
    goto _error;
×
3669
  }
3670

3671
  pListInfo->map = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
282,353,354✔
3672
  if (pListInfo->map == NULL) {
282,540,272✔
3673
    goto _error;
×
3674
  }
3675

3676
  pListInfo->numOfOuputGroups = 1;
282,538,815✔
3677
  return pListInfo;
282,540,228✔
3678

3679
_error:
×
3680
  tableListDestroy(pListInfo);
×
3681
  return NULL;
×
3682
}
3683

3684
void tableListDestroy(STableListInfo* pTableListInfo) {
293,787,371✔
3685
  if (pTableListInfo == NULL) {
293,787,371✔
3686
    return;
11,317,483✔
3687
  }
3688

3689
  taosArrayDestroy(pTableListInfo->pTableList);
282,469,888✔
3690
  taosMemoryFreeClear(pTableListInfo->groupOffset);
282,439,147✔
3691

3692
  taosHashCleanup(pTableListInfo->map);
282,433,354✔
3693
  taosHashCleanup(pTableListInfo->remainGroups);
282,506,112✔
3694
  pTableListInfo->pTableList = NULL;
282,499,959✔
3695
  pTableListInfo->map = NULL;
282,520,853✔
3696
  taosMemoryFree(pTableListInfo);
282,514,151✔
3697
}
3698

3699
void tableListClear(STableListInfo* pTableListInfo) {
808,465✔
3700
  if (pTableListInfo == NULL) {
808,465✔
3701
    return;
×
3702
  }
3703

3704
  taosArrayClear(pTableListInfo->pTableList);
808,465✔
3705
  taosHashClear(pTableListInfo->map);
809,087✔
3706
  taosHashClear(pTableListInfo->remainGroups);
809,422✔
3707
  taosMemoryFree(pTableListInfo->groupOffset);
809,422✔
3708
  pTableListInfo->numOfOuputGroups = 1;
809,422✔
3709
  pTableListInfo->oneTableForEachGroup = false;
809,103✔
3710
}
3711

3712
static int32_t orderbyGroupIdComparFn(const void* p1, const void* p2) {
604,937,924✔
3713
  STableKeyInfo* pInfo1 = (STableKeyInfo*)p1;
604,937,924✔
3714
  STableKeyInfo* pInfo2 = (STableKeyInfo*)p2;
604,937,924✔
3715

3716
  if (pInfo1->groupId == pInfo2->groupId) {
604,937,924✔
3717
    return 0;
568,836,734✔
3718
  } else {
3719
    return pInfo1->groupId < pInfo2->groupId ? -1 : 1;
36,105,212✔
3720
  }
3721
}
3722

3723
int32_t sortTableGroup(STableListInfo* pTableListInfo) {
28,168,317✔
3724
  int32_t code = TSDB_CODE_SUCCESS;
28,168,317✔
3725
  taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
28,168,317✔
3726
  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
28,179,265✔
3727
  if (size == 0) {
28,163,167✔
3728
    pTableListInfo->numOfOuputGroups = 0;
×
3729
    return code;
×
3730
  }
3731

3732
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
28,163,167✔
3733
  if (!pList) {
28,165,025✔
3734
    code = terrno;
×
3735
    goto end;
×
3736
  }
3737

3738
  STableKeyInfo* pInfo = taosArrayGet(pTableListInfo->pTableList, 0);
28,165,025✔
3739
  if (pInfo == NULL) {
28,159,087✔
3740
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3741
    code = terrno;
×
3742
    goto end;
×
3743
  }
3744
  uint64_t gid = pInfo->groupId;
28,159,087✔
3745

3746
  int32_t start = 0;
28,168,706✔
3747
  void*   tmp = taosArrayPush(pList, &start);
28,171,606✔
3748
  if (tmp == NULL) {
28,171,606✔
3749
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3750
    code = terrno;
×
3751
    goto end;
×
3752
  }
3753

3754
  for (int32_t i = 1; i < size; ++i) {
158,911,578✔
3755
    pInfo = taosArrayGet(pTableListInfo->pTableList, i);
130,750,311✔
3756
    if (pInfo == NULL) {
130,736,435✔
3757
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3758
      code = terrno;
×
3759
      goto end;
×
3760
    }
3761
    if (pInfo->groupId != gid) {
130,736,435✔
3762
      tmp = taosArrayPush(pList, &i);
7,399,101✔
3763
      if (tmp == NULL) {
7,399,101✔
3764
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3765
        code = terrno;
×
3766
        goto end;
×
3767
      }
3768
      gid = pInfo->groupId;
7,399,101✔
3769
    }
3770
  }
3771

3772
  pTableListInfo->numOfOuputGroups = taosArrayGetSize(pList);
28,173,488✔
3773
  pTableListInfo->groupOffset = taosMemoryMalloc(sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
28,177,483✔
3774
  if (pTableListInfo->groupOffset == NULL) {
28,165,214✔
3775
    code = terrno;
×
3776
    goto end;
×
3777
  }
3778

3779
  memcpy(pTableListInfo->groupOffset, taosArrayGet(pList, 0), sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
28,160,750✔
3780

3781
end:
28,147,369✔
3782
  taosArrayDestroy(pList);
28,154,567✔
3783
  return code;
28,134,102✔
3784
}
3785

3786
int32_t buildGroupIdMapForAllTables(STableListInfo* pTableListInfo, SReadHandle* pHandle, SScanPhysiNode* pScanNode,
248,328,276✔
3787
                                    SNodeList* group, bool groupSort, uint8_t* digest, SStorageAPI* pAPI, SHashObj* groupIdMap) {
3788
  int32_t code = TSDB_CODE_SUCCESS;
248,328,276✔
3789

3790
  bool   groupByTbname = groupbyTbname(group);
248,328,276✔
3791
  size_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
248,342,358✔
3792
  if (!numOfTables) {
248,343,205✔
3793
    return code;
×
3794
  }
3795
  qDebug("numOfTables:%zu, groupByTbname:%d, group:%p", numOfTables, groupByTbname, group);
248,343,205✔
3796
  if (group == NULL || groupByTbname) {
248,279,285✔
3797
    if (tsCountAlwaysReturnValue && QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode) &&
243,879,882✔
3798
        ((STableScanPhysiNode*)pScanNode)->needCountEmptyTable) {
200,581,250✔
3799
      pTableListInfo->remainGroups =
14,301,624✔
3800
          taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
14,301,624✔
3801
      if (pTableListInfo->remainGroups == NULL) {
14,301,624✔
3802
        return terrno;
×
3803
      }
3804

3805
      for (int i = 0; i < numOfTables; i++) {
71,317,403✔
3806
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
57,013,209✔
3807
        if (!info) {
57,014,567✔
3808
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3809
          return terrno;
×
3810
        }
3811
        info->groupId = groupByTbname ? info->uid : 0;
57,014,567✔
3812
        int32_t tempRes = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId),
57,015,105✔
3813
                                      &(info->uid), sizeof(info->uid));
57,015,105✔
3814
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
57,015,779✔
3815
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3816
          return tempRes;
×
3817
        }
3818
      }
3819
    } else {
3820
      for (int32_t i = 0; i < numOfTables; i++) {
849,108,298✔
3821
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
619,629,338✔
3822
        if (!info) {
619,594,610✔
3823
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3824
          return terrno;
×
3825
        }
3826
        info->groupId = groupByTbname ? info->uid : 0;
619,594,610✔
3827
        
3828
      }
3829
    }
3830
    if (groupIdMap && group != NULL){
243,783,154✔
3831
      getColInfoResultForGroupbyForStream(pHandle->vnode, group, pTableListInfo, pAPI, groupIdMap);
84,300✔
3832
    }
3833

3834
    pTableListInfo->oneTableForEachGroup = groupByTbname;
243,783,154✔
3835
    if (numOfTables == 1 && pTableListInfo->idInfo.tableType == TSDB_CHILD_TABLE) {
243,895,102✔
3836
      pTableListInfo->oneTableForEachGroup = true;
104,096,521✔
3837
    }
3838

3839
    if (groupSort && groupByTbname) {
243,900,796✔
3840
      taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
2,378,921✔
3841
      pTableListInfo->numOfOuputGroups = numOfTables;
2,378,689✔
3842
    } else if (groupByTbname && pScanNode->groupOrderScan) {
241,521,875✔
3843
      pTableListInfo->numOfOuputGroups = numOfTables;
31,403✔
3844
    } else {
3845
      pTableListInfo->numOfOuputGroups = 1;
241,491,607✔
3846
    }
3847
    if (groupSort || pScanNode->groupOrderScan) {
243,956,897✔
3848
      code = sortTableGroup(pTableListInfo);
28,037,096✔
3849
    }
3850
  } else {
3851
    bool initRemainGroups = false;
4,399,403✔
3852
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode)) {
4,399,403✔
3853
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pScanNode;
4,293,230✔
3854
      if (tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable &&
4,293,230✔
3855
          !(groupSort || pScanNode->groupOrderScan)) {
2,088,478✔
3856
        initRemainGroups = true;
2,063,454✔
3857
      }
3858
    }
3859

3860
    code = getColInfoResultForGroupby(pHandle->vnode, group, pTableListInfo, digest, pAPI, initRemainGroups, groupIdMap);
4,399,403✔
3861
    if (code != TSDB_CODE_SUCCESS) {
4,398,829✔
3862
      return code;
×
3863
    }
3864

3865
    if (pScanNode->groupOrderScan) pTableListInfo->numOfOuputGroups = taosArrayGetSize(pTableListInfo->pTableList);
4,398,829✔
3866

3867
    if (groupSort || pScanNode->groupOrderScan) {
4,398,844✔
3868
      code = sortTableGroup(pTableListInfo);
155,604✔
3869
    }
3870
  }
3871

3872
  // add all table entry in the hash map
3873
  size_t size = taosArrayGetSize(pTableListInfo->pTableList);
248,264,271✔
3874
  for (int32_t i = 0; i < size; ++i) {
949,154,650✔
3875
    STableKeyInfo* p = taosArrayGet(pTableListInfo->pTableList, i);
700,832,382✔
3876
    if (!p) {
700,657,441✔
3877
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3878
      return terrno;
×
3879
    }
3880
    int32_t tempRes = taosHashPut(pTableListInfo->map, &p->uid, sizeof(uint64_t), &i, sizeof(int32_t));
700,657,441✔
3881
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
700,868,837✔
3882
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3883
      return tempRes;
×
3884
    }
3885
  }
3886

3887
  return code;
248,373,321✔
3888
}
3889

3890
int32_t createScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
265,006,952✔
3891
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
3892
                                SExecTaskInfo* pTaskInfo, SHashObj* groupIdMap) {
3893
  int64_t     st = taosGetTimestampUs();
265,043,023✔
3894
  const char* idStr = GET_TASKID(pTaskInfo);
265,043,023✔
3895

3896
  if (pHandle == NULL) {
264,933,264✔
3897
    qError("invalid handle, in creating operator tree, %s", idStr);
×
3898
    return TSDB_CODE_INVALID_PARA;
×
3899
  }
3900

3901
  if (pHandle->uid != 0) {
264,933,264✔
3902
    pScanNode->uid = pHandle->uid;
126,784✔
3903
    pScanNode->tableType = TSDB_CHILD_TABLE;
126,784✔
3904
  }
3905
  uint8_t digest[17] = {0};
265,001,760✔
3906
  int32_t code = getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, digest, idStr,
264,959,696✔
3907
                              &pTaskInfo->storageAPI, pTaskInfo->pStreamRuntimeInfo);
265,000,235✔
3908
  if (code != TSDB_CODE_SUCCESS) {
265,082,390✔
3909
    qError("failed to getTableList, code:%s", tstrerror(code));
984✔
3910
    return code;
984✔
3911
  }
3912

3913
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
265,081,406✔
3914

3915
  int64_t st1 = taosGetTimestampUs();
265,112,405✔
3916
  pTaskInfo->cost.extractListTime = (st1 - st) / 1000.0;
265,112,405✔
3917
  qDebug("extract queried table list completed, %d tables, elapsed time:%.2f ms %s", numOfTables,
265,039,867✔
3918
         pTaskInfo->cost.extractListTime, idStr);
3919

3920
  if (numOfTables == 0) {
265,034,295✔
3921
    qDebug("no table qualified for query, %s", idStr);
16,794,524✔
3922
    return TSDB_CODE_SUCCESS;
16,794,522✔
3923
  }
3924

3925
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, digest, &pTaskInfo->storageAPI, groupIdMap);
248,239,771✔
3926
  if (code != TSDB_CODE_SUCCESS) {
248,339,055✔
3927
    return code;
×
3928
  }
3929

3930
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
248,339,620✔
3931
  qDebug("generate group id map completed, elapsed time:%.2f ms %s", pTaskInfo->cost.groupIdMapTime, idStr);
248,326,507✔
3932

3933
  return TSDB_CODE_SUCCESS;
248,334,240✔
3934
}
3935

3936
char* getStreamOpName(uint16_t opType) {
8,123,375✔
3937
  switch (opType) {
8,123,375✔
3938
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
×
3939
      return "stream scan";
×
3940
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
7,761,108✔
3941
      return "project";
7,761,108✔
3942
    case QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW:
362,267✔
3943
      return "external window";
362,267✔
3944
  }
3945
  return "error name";
×
3946
}
3947

3948
void printDataBlock(SSDataBlock* pBlock, const char* flag, const char* taskIdStr, int64_t qId) {
761,670,280✔
3949
  if (qDebugFlag & DEBUG_TRACE) {
761,670,280✔
3950
    if (!pBlock) {
1,495,419✔
3951
      qDebug("%" PRIx64 " %s %s %s: Block is Null", qId, taskIdStr, flag, __func__);
51,368✔
3952
      return;
52,684✔
3953
    } else if (pBlock->info.rows == 0) {
1,444,051✔
3954
      qDebug("%" PRIx64 " %s %s %s: Block is Empty.", qId, taskIdStr, flag, __func__);
9,583✔
3955
      return;
9,583✔
3956
    }
3957
    
3958
    char*   pBuf = NULL;
1,434,468✔
3959
    int32_t code = dumpBlockData(pBlock, flag, &pBuf, taskIdStr, qId);
1,434,468✔
3960
    if (code == 0) {
1,434,468✔
3961
      qDebugL("%" PRIx64 " %s %s", qId, __func__, pBuf);
1,434,468✔
3962
      taosMemoryFree(pBuf);
1,434,468✔
3963
    }
3964
  }
3965
}
3966

3967
void printSpecDataBlock(SSDataBlock* pBlock, const char* flag, const char* opStr, const char* taskIdStr) {
×
3968
  if (!pBlock) {
×
3969
    qDebug("%s===stream===%s %s: Block is Null", taskIdStr, flag, opStr);
×
3970
    return;
×
3971
  } else if (pBlock->info.rows == 0) {
×
3972
    qDebug("%s===stream===%s %s: Block is Empty.skey:%" PRId64 ",ekey:%" PRId64 ",version%" PRId64,
×
3973
           taskIdStr, flag, opStr, pBlock->info.window.skey, pBlock->info.window.ekey,
3974
           pBlock->info.version);
3975
    return;
×
3976
  }
3977
  if (qDebugFlag & DEBUG_TRACE) {
×
3978
    char* pBuf = NULL;
×
3979
    char  flagBuf[64];
×
3980
    snprintf(flagBuf, sizeof(flagBuf), "%s %s", flag, opStr);
×
3981
    int32_t code = dumpBlockData(pBlock, flagBuf, &pBuf, taskIdStr, 0);
×
3982
    if (code == 0) {
×
3983
      qDebug("%s", pBuf);
×
3984
      taosMemoryFree(pBuf);
×
3985
    }
3986
  }
3987
}
3988

3989
TSKEY getStartTsKey(STimeWindow* win, const TSKEY* tsCols) { return tsCols == NULL ? win->skey : tsCols[0]; }
12,395,744✔
3990

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

3994
  int64_t duration = pWin->ekey > pWin->skey ? pWin->ekey - pWin->skey + delta : pWin->skey - pWin->ekey + delta;
2,147,483,647✔
3995
  ts[2] = duration;            // set the duration
2,147,483,647✔
3996
  ts[3] = pWin->skey;          // window start key
2,147,483,647✔
3997
  ts[4] = pWin->ekey + delta;  // window end key
2,147,483,647✔
3998
}
2,147,483,647✔
3999

4000
int32_t compKeys(const SArray* pSortGroupCols, const char* oldkeyBuf, int32_t oldKeysLen, const SSDataBlock* pBlock,
2,147,483,647✔
4001
                 int32_t rowIndex) {
4002
  SColumnDataAgg* pColAgg = NULL;
2,147,483,647✔
4003
  const char*     isNull = oldkeyBuf;
2,147,483,647✔
4004
  const char*     p = oldkeyBuf + sizeof(int8_t) * pSortGroupCols->size;
2,147,483,647✔
4005

4006
  for (int32_t i = 0; i < pSortGroupCols->size; ++i) {
2,147,483,647✔
4007
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
2,147,483,647✔
4008
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
2,147,483,647✔
4009
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
2,147,483,647✔
4010

4011
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
2,147,483,647✔
4012
      if (isNull[i] != 1) return 1;
276,874,302✔
4013
    } else {
4014
      if (isNull[i] != 0) return 1;
2,147,483,647✔
4015
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,147,483,647✔
4016
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,147,483,647✔
4017
        int32_t len = getJsonValueLen(val);
×
4018
        if (memcmp(p, val, len) != 0) return 1;
×
4019
        p += len;
×
4020
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,147,483,647✔
4021
        if (IS_STR_DATA_BLOB(pCol->type)) {
1,286,050,164✔
4022
          if (memcmp(p, val, blobDataTLen(val)) != 0) return 1;
×
4023
          p += blobDataTLen(val);
×
4024
        } else {
4025
          if (memcmp(p, val, varDataTLen(val)) != 0) return 1;
1,285,450,080✔
4026
          p += varDataTLen(val);
1,284,870,378✔
4027
        }
4028
      } else {
4029
        if (0 != memcmp(p, val, pCol->bytes)) return 1;
2,147,483,647✔
4030
        p += pCol->bytes;
2,147,483,647✔
4031
      }
4032
    }
4033
  }
4034
  if ((int32_t)(p - oldkeyBuf) != oldKeysLen) return 1;
2,147,483,647✔
4035
  return 0;
2,147,483,647✔
4036
}
4037

4038
int32_t buildKeys(char* keyBuf, const SArray* pSortGroupCols, const SSDataBlock* pBlock, int32_t rowIndex) {
1,692,180✔
4039
  uint32_t        colNum = pSortGroupCols->size;
1,692,180✔
4040
  SColumnDataAgg* pColAgg = NULL;
1,692,180✔
4041
  char*           isNull = keyBuf;
1,692,180✔
4042
  char*           p = keyBuf + sizeof(int8_t) * colNum;
1,692,180✔
4043

4044
  for (int32_t i = 0; i < colNum; ++i) {
4,630,980✔
4045
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
2,938,800✔
4046
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
2,938,800✔
4047
    if (pCol->slotId > pBlock->pDataBlock->size) continue;
2,938,800✔
4048

4049
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
2,938,800✔
4050

4051
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
5,877,600✔
4052
      isNull[i] = 1;
142,200✔
4053
    } else {
4054
      isNull[i] = 0;
2,796,600✔
4055
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,796,600✔
4056
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,796,600✔
4057
        int32_t len = getJsonValueLen(val);
×
4058
        memcpy(p, val, len);
×
4059
        p += len;
×
4060
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,796,600✔
4061
        if (IS_STR_DATA_BLOB(pCol->type)) {
1,045,644✔
4062
          blobDataCopy(p, val);
×
4063
          p += blobDataTLen(val);
×
4064
        } else {
4065
          varDataCopy(p, val);
1,045,644✔
4066
          p += varDataTLen(val);
1,045,644✔
4067
        }
4068
      } else {
4069
        memcpy(p, val, pCol->bytes);
1,750,956✔
4070
        p += pCol->bytes;
1,750,956✔
4071
      }
4072
    }
4073
  }
4074
  return (int32_t)(p - keyBuf);
1,692,180✔
4075
}
4076

4077
uint64_t calcGroupId(char* pData, int32_t len) {
2,147,483,647✔
4078
  T_MD5_CTX context;
2,147,483,647✔
4079
  tMD5Init(&context);
2,147,483,647✔
4080
  tMD5Update(&context, (uint8_t*)pData, len);
2,147,483,647✔
4081
  tMD5Final(&context);
2,147,483,647✔
4082

4083
  // NOTE: only extract the initial 8 bytes of the final MD5 digest
4084
  uint64_t id = 0;
2,147,483,647✔
4085
  memcpy(&id, context.digest, sizeof(uint64_t));
2,147,483,647✔
4086
  if (0 == id) memcpy(&id, context.digest + 8, sizeof(uint64_t));
2,147,483,647✔
4087
  return id;
2,147,483,647✔
4088
}
4089

4090
SNodeList* makeColsNodeArrFromSortKeys(SNodeList* pSortKeys) {
111,864✔
4091
  SNode*     node;
4092
  SNodeList* ret = NULL;
111,864✔
4093
  FOREACH(node, pSortKeys) {
341,280✔
4094
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)node;
229,416✔
4095
    int32_t           code = nodesListMakeAppend(&ret, pSortKey->pExpr);
229,416✔
4096
    if (code != TSDB_CODE_SUCCESS) {
229,416✔
4097
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4098
      terrno = code;
×
4099
      return NULL;
×
4100
    }
4101
  }
4102
  return ret;
111,864✔
4103
}
4104

4105
int32_t extractKeysLen(const SArray* keys, int32_t* pLen) {
111,864✔
4106
  int32_t code = TSDB_CODE_SUCCESS;
111,864✔
4107
  int32_t lino = 0;
111,864✔
4108
  int32_t len = 0;
111,864✔
4109
  int32_t keyNum = taosArrayGetSize(keys);
111,864✔
4110
  for (int32_t i = 0; i < keyNum; ++i) {
284,400✔
4111
    SColumn* pCol = (SColumn*)taosArrayGet(keys, i);
172,536✔
4112
    QUERY_CHECK_NULL(pCol, code, lino, _end, terrno);
172,536✔
4113
    len += pCol->bytes;
172,536✔
4114
  }
4115
  len += sizeof(int8_t) * keyNum;  // null flag
111,864✔
4116
  *pLen = len;
111,864✔
4117

4118
_end:
111,864✔
4119
  if (code != TSDB_CODE_SUCCESS) {
111,864✔
4120
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4121
  }
4122
  return code;
111,864✔
4123
}
4124

4125
int32_t parseErrorMsgFromAnalyticServer(SJson* pJson, const char* typeStr, const char* pId) {
×
4126
  int32_t code = TSDB_CODE_ANA_ANODE_RETURN_ERROR;
×
4127
  char*   pMsg = NULL;
×
4128
  if (pJson == NULL) {
×
4129
    return code;
×
4130
  }
4131

4132
  int32_t ret = tjsonDupStringValue(pJson, "msg", &pMsg);
×
4133
  if (ret == 0 && pMsg != NULL) {
×
4134
    qError("%s failed to exec %s operation, msg:%s", pId, typeStr, pMsg);
×
4135
    if (strstr(pMsg, "white noise") != NULL) {
×
4136
      code = TSDB_CODE_ANA_WN_DATA;
×
4137
    } else if (strstr(pMsg, "white-noise") != NULL) {
×
4138
      code = TSDB_CODE_ANA_WN_DATA;
×
4139
    } else if (strstr(pMsg, "[Errno 111] Connection refused") != NULL) {
×
4140
      code = TSDB_CODE_ANA_ALGO_NOT_LOAD;
×
4141
    } else if (strstr(pMsg, "failed to load model") != NULL) {
×
4142
      code = TSDB_CODE_ANA_ALGO_NOT_LOAD;
×
4143
    }
4144
  } else {
4145
    qError("%s failed to extract msg from server, unknown error", pId);
×
4146
  }
4147

4148
  taosMemoryFreeClear(pMsg);
×
4149
  return code;
×
4150
}
4151

4152

4153
int32_t createExprSubQResBlock(SSDataBlock** ppBlock, SDataType* pResType) {
28,740,206✔
4154
  int32_t code = 0;
28,740,206✔
4155
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
28,740,206✔
4156
  if (pBlock == NULL) {
28,740,194✔
4157
    return terrno;
×
4158
  }
4159

4160
  pBlock->pDataBlock = taosArrayInit(1, sizeof(SColumnInfoData));
28,740,194✔
4161
  if (pBlock->pDataBlock == NULL) {
28,739,096✔
4162
    code = terrno;
×
4163
    taosMemoryFree(pBlock);
×
4164
    return code;
×
4165
  }
4166

4167
  SColumnInfoData idata =
28,740,194✔
4168
      createColumnInfoData(pResType->type, pResType->bytes, 0);
28,740,749✔
4169
  idata.info.scale = pResType->scale;
28,740,749✔
4170
  idata.info.precision = pResType->precision;
28,740,749✔
4171

4172
  code = blockDataAppendColInfo(pBlock, &idata);
28,740,749✔
4173
  if (code != TSDB_CODE_SUCCESS) {
28,740,206✔
4174
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4175
    blockDataDestroy(pBlock);
×
4176
    *ppBlock = NULL;
×
4177
    return code;
×
4178
  }
4179

4180
  *ppBlock = pBlock;
28,740,206✔
4181

4182
  return code;
28,740,206✔
4183
}
4184

4185

4186
int32_t extractSingleRspBlock(SRetrieveTableRsp* pRetrieveRsp, SSDataBlock* pb) {
28,739,957✔
4187
  int32_t            code = TSDB_CODE_SUCCESS;
28,739,957✔
4188
  int32_t            lino = 0;
28,739,957✔
4189
  void*              decompBuf = NULL;
28,739,957✔
4190

4191
  char* pNextStart = pRetrieveRsp->data;
28,739,957✔
4192
  char* pStart = pNextStart;
28,740,749✔
4193

4194
  int32_t index = 0;
28,740,749✔
4195

4196
  if (pRetrieveRsp->compressed) {  // decompress the data
28,740,749✔
4197
    decompBuf = taosMemoryMalloc(pRetrieveRsp->payloadLen);
×
4198
    QUERY_CHECK_NULL(decompBuf, code, lino, _end, terrno);
×
4199
  }
4200

4201
  int32_t compLen = *(int32_t*)pStart;
28,740,194✔
4202
  pStart += sizeof(int32_t);
28,739,957✔
4203

4204
  int32_t rawLen = *(int32_t*)pStart;
28,740,500✔
4205
  pStart += sizeof(int32_t);
28,740,749✔
4206
  QUERY_CHECK_CONDITION((compLen <= rawLen && compLen != 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
28,740,500✔
4207

4208
  pNextStart = pStart + compLen;
28,740,500✔
4209
  if (pRetrieveRsp->compressed && (compLen < rawLen)) {
28,739,945✔
4210
    int32_t t = tsDecompressString(pStart, compLen, 1, decompBuf, rawLen, ONE_STAGE_COMP, NULL, 0);
×
4211
    QUERY_CHECK_CONDITION((t == rawLen), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
4212
    pStart = decompBuf;
×
4213
  }
4214

4215
  code = blockDecodeInternal(pb, pStart, (const char**)&pStart);
28,740,194✔
4216
  if (code != 0) {
28,739,957✔
4217
    taosMemoryFreeClear(pRetrieveRsp);
×
4218
    goto _end;
×
4219
  }
4220

4221
_end:
28,739,957✔
4222
  if (code != TSDB_CODE_SUCCESS) {
28,739,957✔
4223
    blockDataDestroy(pb);
×
4224
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4225
  }
4226
  return code;
28,739,957✔
4227
}
4228

4229
int32_t setValueFromResBlock(STaskSubJobCtx* ctx, SValueNode* pRes, SSDataBlock* pBlock) {
18,786,459✔
4230
  int32_t code = 0;
18,786,459✔
4231
  bool needFree = true;
18,786,459✔
4232
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
18,786,459✔
4233
  if (NULL == pBlock->pDataBlock || 1 > colNum || pBlock->info.rows > 1) {
18,786,459✔
UNCOV
4234
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d, rows:%" PRId64, 
×
4235
      ctx->idStr, pBlock->pDataBlock, colNum, pBlock->info.rows);
4236
    return TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
×
4237
  }
4238
  
4239
  pRes->flag &= (~VALUE_FLAG_VAL_UNSET);
18,785,904✔
4240
  pRes->translate = true;
18,785,349✔
4241
  
4242
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
18,786,459✔
4243
  if (colDataIsNull_s(pCol, 0)) {
18,786,459✔
4244
    pRes->isNull = true;
5,065,005✔
4245
  } else {
4246
    code = nodesSetValueNodeValueExt(pRes, colDataGetData(pCol, 0), &needFree);
13,721,454✔
4247
  }
4248

4249
  if (!needFree) {
18,786,459✔
4250
    pCol->pData = NULL;
1,860,768✔
4251
  }
4252

4253
  return code;
18,786,459✔
4254
}
4255

4256
void handleRemoteValueRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
3,409,760✔
4257
  SSDataBlock*      pResBlock = NULL;
3,409,760✔
4258
  SExecTaskInfo*    pTaskInfo = ctx->pTaskInfo;
3,409,760✔
4259
  SRemoteValueNode* pRemote = (SRemoteValueNode*)pParam->pRes;
3,409,760✔
4260

4261
  qDebug("%s scl fetch value rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
3,408,650✔
4262

4263
  if (IS_STREAM_MODE(pTaskInfo)) {
3,408,650✔
4264
    SNode** ppRes = taosArrayGet(ctx->subResNodes, pParam->subQIdx);
114,294✔
4265
    if (NULL == *ppRes && 0 == pRsp->numOfRows) {
114,294✔
4266
      pRemote->val.node.type = QUERY_NODE_VALUE;
×
4267
      pRemote->val.isNull = true;
×
4268
      pRemote->val.translate = true;
×
4269
      pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
×
4270
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
×
4271
    } else if (pRsp->numOfRows > 0 || pRsp->numOfBlocks > 0) {
114,294✔
4272
      ctx->code = createExprSubQResBlock(&pResBlock, &((SRemoteValueNode*)pParam->pRes)->val.node.resType);
57,147✔
4273
      if (TSDB_CODE_SUCCESS == ctx->code) {
57,147✔
4274
        ctx->code = blockDataEnsureCapacity(pResBlock, 1);
57,147✔
4275
      }
4276
      if (TSDB_CODE_SUCCESS == ctx->code) {
57,147✔
4277
        ctx->code = extractSingleRspBlock(pRsp, pResBlock);
57,147✔
4278
      }
4279
      if (TSDB_CODE_SUCCESS == ctx->code) {
57,147✔
4280
        ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
57,147✔
4281
        pRemote->val.node.type = QUERY_NODE_REMOTE_VALUE;
57,147✔
4282
      }
4283
      if (TSDB_CODE_SUCCESS == ctx->code) {
57,147✔
4284
        taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
57,147✔
4285
      }
4286

4287
      blockDataDestroy(pResBlock);
57,147✔
4288
    } else if (NULL != *ppRes && 0 == pRsp->numOfRows) {
57,147✔
4289
      pRemote->val.node.type = QUERY_NODE_VALUE;
57,147✔
4290
      pRsp->completed = true;
57,147✔
4291
    }
4292

4293
    *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
114,294✔
4294

4295
    if (!(*fetchDone)) {
114,294✔
4296
      int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
57,147✔
4297
      if (TSDB_CODE_SUCCESS != code) {
57,147✔
4298
        ctx->code = code;
×
4299
        *fetchDone = true;
×
4300
      }
4301
    }
4302

4303
    return;
114,294✔
4304
  }
4305

4306
  *fetchDone = true;
3,294,911✔
4307

4308
  if (pRsp->numOfRows > 1 || pRsp->numOfBlocks > 1 || !pRsp->completed) {
3,294,911✔
4309
    qError("%s invalid scl value fetch rsp received, subQIdx:%d, rows:%" PRId64 ", blocks:%d, completed:%d", 
555✔
4310
      ctx->idStr, pParam->subQIdx, pRsp->numOfRows, pRsp->numOfBlocks, pRsp->completed);
4311
    ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
555✔
4312

4313
    return;
×
4314
  }
4315

4316
  if (1 != pRsp->numOfCols && pRsp->numOfRows > 0) {
3,294,911✔
4317
    qError("%s invalid scl value fetch rsp received, subQIdx:%d, cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfCols);
×
4318
    ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_COLS;
×
4319

4320
    return;
×
4321
  }
4322

4323
  if (0 == pRsp->numOfRows) {
3,294,911✔
4324
    pRemote->val.node.type = QUERY_NODE_VALUE;
176,997✔
4325
    pRemote->val.isNull = true;
176,997✔
4326
    pRemote->val.translate = true;
176,997✔
4327
    pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
176,997✔
4328
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
176,997✔
4329

4330
    return;
176,997✔
4331
  }
4332
  
4333
  ctx->code = createExprSubQResBlock(&pResBlock, &((SRemoteValueNode*)pParam->pRes)->val.node.resType);
3,118,469✔
4334
  if (TSDB_CODE_SUCCESS == ctx->code) {
3,117,914✔
4335
    ctx->code = blockDataEnsureCapacity(pResBlock, 1);
3,118,469✔
4336
  }
4337
  if (TSDB_CODE_SUCCESS == ctx->code) {
3,118,469✔
4338
    ctx->code = extractSingleRspBlock(pRsp, pResBlock);
3,118,469✔
4339
  }
4340
  if (TSDB_CODE_SUCCESS == ctx->code) {
3,117,359✔
4341
    ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
3,117,914✔
4342
  }
4343
  if (TSDB_CODE_SUCCESS == ctx->code) {
3,117,914✔
4344
    pRemote->val.node.type = QUERY_NODE_VALUE;
3,118,469✔
4345
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
3,118,469✔
4346
  }
4347

4348
  blockDataDestroy(pResBlock);  
3,117,359✔
4349
}
4350

4351

4352
int32_t updateValueListFromResBlock(STaskSubJobCtx* ctx, SRemoteValueListNode* pRes, SSDataBlock* pBlock) {
9,953,498✔
4353
  int32_t code = 0, lino = 0;
9,953,498✔
4354
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
9,953,498✔
4355
  if (NULL == pBlock->pDataBlock || 1 != colNum) {
9,954,290✔
4356
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d", ctx->idStr, pBlock->pDataBlock, colNum);
543✔
4357
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4358
  }
4359

4360
  pRes->hasValue = true;
9,953,498✔
4361
  
4362
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
9,953,747✔
4363
  TAOS_CHECK_EXIT(scalarBuildRemoteListHash(ctx->idStr, pRes, pCol, pBlock->info.rows));
9,954,041✔
4364

4365
_exit:
9,232,103✔
4366

4367
  if (code) {
9,953,747✔
4368
    qError("%s %s failed with error: %s", ctx->idStr, __func__, tstrerror(code));
721,644✔
4369
  }
4370
  
4371
  return code;
9,953,747✔
4372
}
4373

4374

4375

4376
void handleRemoteValueListRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
12,901,877✔
4377
  SSDataBlock* pResBlock = NULL;
12,901,877✔
4378
  SRemoteValueListNode* pRemote = (SRemoteValueListNode*)pParam->pRes;
12,901,877✔
4379
  SExecTaskInfo* pTaskInfo = ctx->pTaskInfo;
12,901,334✔
4380

4381
  qDebug("%s scl fetch valueList rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
12,901,334✔
4382

4383
  if (pRsp->numOfRows > 0) {
12,901,334✔
4384
    if (1 != pRsp->numOfCols) {
9,953,747✔
4385
      qError("%s invalid scl valueList fetch rsp received, subQIdx:%d, cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfCols);
×
4386
      ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_COLS;
×
4387
      *fetchDone = true;
×
4388
      return;
×
4389
    }
4390

4391
    ctx->code = createExprSubQResBlock(&pResBlock, &((SExprNode*)pParam->pRes)->resType);
9,953,747✔
4392
    if (TSDB_CODE_SUCCESS == ctx->code) {
9,954,290✔
4393
      ctx->code = blockDataEnsureCapacity(pResBlock, pRsp->numOfRows);
9,954,290✔
4394
    }
4395
    if (TSDB_CODE_SUCCESS == ctx->code) {
9,953,747✔
4396
      ctx->code = extractSingleRspBlock(pRsp, pResBlock);
9,953,747✔
4397
    }
4398
    if (TSDB_CODE_SUCCESS == ctx->code) {
9,953,498✔
4399
      ctx->code = updateValueListFromResBlock(ctx, pRemote, pResBlock);
9,954,290✔
4400
    }
4401
    if (TSDB_CODE_SUCCESS == ctx->code && pRsp->completed) {
9,953,204✔
4402
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
7,917,365✔
4403
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
7,917,365✔
4404
    }
4405

4406
    blockDataDestroy(pResBlock);  
9,954,290✔
4407
  } else if (0 == pRsp->numOfRows && pRsp->completed) {
2,947,587✔
4408
    if (!pRemote->hasValue) {
2,947,587✔
4409
      ctx->code = scalarBuildRemoteListHash(ctx->idStr, pRemote, NULL, 0);
2,928,833✔
4410
    }
4411
    if (TSDB_CODE_SUCCESS == ctx->code) {    
2,947,587✔
4412
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
2,703,063✔
4413
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
2,703,063✔
4414
    }
4415
  }
4416

4417
  if (IS_STREAM_MODE(pTaskInfo) && 0 == pRsp->numOfRows) {
12,900,791✔
4418
    if (!pRemote->hasValue) {
19,263✔
4419
      ctx->code = scalarBuildRemoteListHash(ctx->idStr, pRemote, NULL, 0);
509✔
4420
    }
4421
    if (TSDB_CODE_SUCCESS == ctx->code) {
19,263✔
4422
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
19,263✔
4423
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
19,263✔
4424
    }
4425

4426
    pRsp->completed = true;
19,263✔
4427
  }
4428

4429
  *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
12,901,334✔
4430

4431
  if (!(*fetchDone)) {
12,900,248✔
4432
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
1,313,652✔
4433
    if (TSDB_CODE_SUCCESS != code) {
1,315,281✔
4434
      ctx->code = code;
×
4435
      *fetchDone = true;
×
4436
    }
4437
  }
4438
}
4439

4440
int32_t setRowHasNullFromResBlock(STaskSubJobCtx* ctx, bool* hasNull, SSDataBlock* pBlock) {
15,610,843✔
4441
  int32_t code = 0;
15,610,843✔
4442
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
15,610,843✔
4443
  if (2 != colNum) {
15,610,843✔
4444
    qError("%s invalid scl fetch res block, colNum:%d", ctx->idStr, colNum);
×
4445
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4446
  }
4447
  
4448
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 1);
15,610,843✔
4449
  if (colDataIsNull_s(pCol, 0)) {
15,610,843✔
4450
    qError("%s invalid has_null res since it's null", ctx->idStr);
×
4451
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4452
  } else {
4453
    *hasNull = *(bool*)colDataGetData(pCol, 0);
15,610,843✔
4454
  }
4455

4456
  return code;
15,610,843✔
4457
}
4458

4459
void handleRemoteRowRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
21,157,865✔
4460
  SSDataBlock* pResBlock = NULL;
21,157,865✔
4461
  SExecTaskInfo* pTaskInfo = ctx->pTaskInfo;
21,157,865✔
4462

4463
  qDebug("%s scl fetch row rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
21,157,865✔
4464

4465
  SRemoteRowNode* pRemote = (SRemoteRowNode*)pParam->pRes;
21,157,865✔
4466

4467
  if (IS_STREAM_MODE(pTaskInfo)) {
21,157,865✔
4468
    SNode** ppRes = taosArrayGet(ctx->subResNodes, pParam->subQIdx);
1,004✔
4469
    if (NULL == *ppRes && 0 == pRsp->numOfRows) {
1,004✔
4470
      pRemote->valSet = true;
×
4471
      pRemote->hasValue = false;
×
4472
      pRemote->hasNull = false;
×
4473
      pRemote->val.isNull = true;
×
4474
      pRemote->val.translate = true;
×
4475
      pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
×
4476
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
×
4477
    } else if (pRsp->numOfRows > 0 || pRsp->numOfBlocks > 0) {
1,004✔
4478
      if (2 != pRsp->numOfCols) {
502✔
4479
        qError("%s invalid scl fetch row rsp received, subQIdx:%d, cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfCols);
×
4480
        ctx->code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4481
        *fetchDone = true;
×
4482
        return;
×
4483
      }
4484

4485
      ctx->code = createExprSubQResBlock(&pResBlock, &pRemote->val.node.resType);
502✔
4486
      if (TSDB_CODE_SUCCESS == ctx->code) {
502✔
4487
        SColumnInfoData idata = createColumnInfoData(TSDB_DATA_TYPE_BOOL, tDataTypes[TSDB_DATA_TYPE_BOOL].bytes, 0);
502✔
4488
        ctx->code = blockDataAppendColInfo(pResBlock, &idata);
502✔
4489
      }
4490
      if (TSDB_CODE_SUCCESS == ctx->code) {
502✔
4491
        ctx->code = blockDataEnsureCapacity(pResBlock, 1);
502✔
4492
      }
4493
      if (TSDB_CODE_SUCCESS == ctx->code) {
502✔
4494
        ctx->code = extractSingleRspBlock(pRsp, pResBlock);
502✔
4495
      }
4496
      if (TSDB_CODE_SUCCESS == ctx->code) {
502✔
4497
        ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
502✔
4498
      }
4499
      if (TSDB_CODE_SUCCESS == ctx->code) {
502✔
4500
        ctx->code = setRowHasNullFromResBlock(ctx, &pRemote->hasNull, pResBlock);
502✔
4501
      }
4502
      if (TSDB_CODE_SUCCESS == ctx->code) {
502✔
4503
        pRemote->valSet = true;
502✔
4504
        pRemote->hasValue = true;
502✔
4505
        pRemote->val.node.type = QUERY_NODE_REMOTE_ROW;
502✔
4506
        taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
502✔
4507
      }
4508

4509
      blockDataDestroy(pResBlock);
502✔
4510
    } else if (NULL != *ppRes && 0 == pRsp->numOfRows) {
502✔
4511
      pRemote->val.node.type = QUERY_NODE_VALUE;
502✔
4512
      pRsp->completed = true;
502✔
4513
    }
4514

4515
    *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
1,004✔
4516

4517
    if (!(*fetchDone)) {
1,004✔
4518
      ctx->code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
502✔
4519
      if (TSDB_CODE_SUCCESS != ctx->code) {
502✔
4520
        *fetchDone = true;
×
4521
      }
4522
    }
4523

4524
    return;
1,004✔
4525
  }
4526

4527
  *fetchDone = true;
21,156,861✔
4528

4529
  if (pRsp->numOfRows > 1 || pRsp->numOfBlocks > 1 || !pRsp->completed) {
21,156,861✔
4530
    qError("%s invalid scl fetch row rsp received, subQIdx:%d, rows:%" PRId64 ", blocks:%d, completed:%d",
×
4531
      ctx->idStr, pParam->subQIdx, pRsp->numOfRows, pRsp->numOfBlocks, pRsp->completed);
4532
    ctx->code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
×
4533

4534
    return;
×
4535
  }
4536

4537
  if (0 == pRsp->numOfRows) {
21,156,861✔
4538
    pRemote->valSet = true;
5,546,520✔
4539
    pRemote->hasValue = false;
5,546,520✔
4540
    pRemote->hasNull = false;
5,546,520✔
4541
    pRemote->val.isNull = true;
5,546,520✔
4542
    pRemote->val.translate = true;
5,546,520✔
4543
    pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
5,546,520✔
4544
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
5,546,520✔
4545

4546
    return;
5,546,520✔
4547
  }
4548

4549
  if (2 != pRsp->numOfCols) {
15,610,341✔
4550
    qError("%s invalid scl fetch row rsp received, subQIdx:%d, cols:%" PRId64, ctx->idStr, pParam->subQIdx, pRsp->numOfCols);
×
4551
    ctx->code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4552

4553
    return;
×
4554
  }
4555

4556
  ctx->code = createExprSubQResBlock(&pResBlock, &pRemote->val.node.resType);
15,610,341✔
4557
  if (TSDB_CODE_SUCCESS == ctx->code) {
15,610,341✔
4558
    SColumnInfoData idata = createColumnInfoData(TSDB_DATA_TYPE_BOOL, tDataTypes[TSDB_DATA_TYPE_BOOL].bytes, 0);
15,610,341✔
4559
    ctx->code = blockDataAppendColInfo(pResBlock, &idata);
15,610,341✔
4560
  }
4561
  if (TSDB_CODE_SUCCESS == ctx->code) {
15,610,341✔
4562
    ctx->code = blockDataEnsureCapacity(pResBlock, 1);
15,610,341✔
4563
  }
4564
  if (TSDB_CODE_SUCCESS == ctx->code) {
15,610,341✔
4565
    ctx->code = extractSingleRspBlock(pRsp, pResBlock);
15,610,341✔
4566
  }
4567
  if (TSDB_CODE_SUCCESS == ctx->code) {
15,610,341✔
4568
    ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
15,610,341✔
4569
  }
4570
  if (TSDB_CODE_SUCCESS == ctx->code) {
15,610,341✔
4571
    ctx->code = setRowHasNullFromResBlock(ctx, &pRemote->hasNull, pResBlock);
15,610,341✔
4572
  }
4573
  if (TSDB_CODE_SUCCESS == ctx->code) {
15,610,341✔
4574
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
15,610,341✔
4575
  }
4576
  if (TSDB_CODE_SUCCESS == ctx->code) {
15,610,341✔
4577
    pRemote->valSet = true;
15,610,341✔
4578
    pRemote->hasValue = true;
15,610,341✔
4579
  }
4580

4581
  blockDataDestroy(pResBlock);
15,610,341✔
4582
}
4583

4584

4585
int32_t setZeroRowsResValue(STaskSubJobCtx* ctx, SValueNode* pRes, int32_t rows) {
278,309✔
4586
  pRes->node.type = QUERY_NODE_VALUE;
278,309✔
4587
  pRes->flag &= (~VALUE_FLAG_VAL_UNSET);
278,309✔
4588
  pRes->translate = true;
278,309✔
4589
  
4590
  return nodesSetValueNodeValue(pRes, &rows);
278,309✔
4591
}
4592

4593
void handleRemoteZeroRowsRes(SScalarFetchParam* pParam, STaskSubJobCtx* ctx, SRetrieveTableRsp* pRsp, bool* fetchDone) {
278,058✔
4594
  SRemoteZeroRowsNode* pRemote = (SRemoteZeroRowsNode*)pParam->pRes;
278,058✔
4595
  SExecTaskInfo* pTaskInfo = ctx->pTaskInfo;
278,058✔
4596

4597
  qDebug("%s scl fetch zeroRows rsp received, subQIdx:%d, rows:%" PRId64 , ctx->idStr, pParam->subQIdx, pRsp->numOfRows);
278,058✔
4598

4599
  int32_t resRows = (pRsp->numOfRows > 0) ? 1 : 0;
278,058✔
4600
  if (resRows > 0 || pRsp->completed) {
278,058✔
4601
    ctx->code = setZeroRowsResValue(ctx, &pRemote->val, resRows);
278,058✔
4602
    if (TSDB_CODE_SUCCESS == ctx->code) {
278,058✔
4603
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
278,058✔
4604
    }
4605

4606
    *fetchDone = true;
278,058✔
4607
  } else {
4608
    *fetchDone = false;
×
4609
  }
4610

4611
  if (IS_STREAM_MODE(pTaskInfo) && 0 == pRsp->numOfRows) {
278,058✔
4612
    ctx->code = setZeroRowsResValue(ctx, &pRemote->val, 0);
251✔
4613
    if (TSDB_CODE_SUCCESS == ctx->code) {
251✔
4614
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
251✔
4615
    }
4616

4617
    pRsp->completed = true;
251✔
4618
    *fetchDone = true;
251✔
4619
  }
4620

4621
  if (!(*fetchDone)) {
278,058✔
4622
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
×
4623
    if (TSDB_CODE_SUCCESS != code) {
×
4624
      ctx->code = code;
×
4625
      *fetchDone = true;
×
4626
    }
4627
  }
4628
}
278,058✔
4629

4630

4631
int32_t remoteFetchCallBack(void* param, SDataBuf* pMsg, int32_t code) {
38,631,189✔
4632
  taosMemoryFreeClear(pMsg->pEpSet);
38,631,189✔
4633

4634
  SScalarFetchParam* pParam = (SScalarFetchParam*)param;
38,632,230✔
4635
  STaskSubJobCtx* ctx = taosAcquireRef(fetchObjRefPool, pParam->subJobRefId);
38,632,230✔
4636
  if (ctx == NULL) {
38,632,230✔
4637
    qWarn("failed to acquire subJobCtx, since it may have been released, refId:%" PRIu64, pParam->subJobRefId);
4,440✔
4638
    taosMemoryFree(pMsg->pData);
4,440✔
4639
    return TSDB_CODE_SUCCESS;
4,440✔
4640
  }
4641

4642
  char idStr[64];
38,627,790✔
4643
  if (qDebugFlag & DEBUG_DEBUG) {
38,627,541✔
4644
    tstrncpy(idStr, ctx->idStr, sizeof(idStr));
1,422,829✔
4645
  }
4646
  
4647
  qDebug("%s subQIdx %d got rsp, blockIdx:%" PRId64 ", code:%d, rsp:%p", ctx->idStr, pParam->subQIdx, ctx->blockIdx, code, pMsg->pData);
38,627,541✔
4648

4649
  if (ctx->transporterId > 0) {
38,627,541✔
4650
    int32_t ret = asyncFreeConnById(ctx->rpcHandle, ctx->transporterId);
38,626,680✔
4651
    if (ret != 0) {
38,627,247✔
4652
      qDebug("%s failed to free subQ rpc handle, code:%s, subQIdx:%d", ctx->idStr, tstrerror(ret), pParam->subQIdx);
249✔
4653
    }
4654
    ctx->transporterId = -1;
38,627,247✔
4655
  }
4656

4657
  if (0 == code && NULL == pMsg->pData) {
38,628,357✔
4658
    qError("%s invalid rsp msg, msgType:%d, len:%d", ctx->idStr, pMsg->msgType, pMsg->len);
×
4659
    code = TSDB_CODE_QRY_INVALID_MSG;
×
4660
  }
4661

4662
  if (code == TSDB_CODE_SUCCESS) {
38,627,247✔
4663
    SRetrieveTableRsp* pRsp = pMsg->pData;
37,747,017✔
4664
    pRsp->numOfRows = htobe64(pRsp->numOfRows);
37,747,560✔
4665
    pRsp->compLen = htonl(pRsp->compLen);
37,747,560✔
4666
    pRsp->payloadLen = htonl(pRsp->payloadLen);
37,747,017✔
4667
    pRsp->numOfCols = htonl(pRsp->numOfCols);
37,746,462✔
4668
    pRsp->useconds = htobe64(pRsp->useconds);
37,746,462✔
4669
    pRsp->numOfBlocks = htonl(pRsp->numOfBlocks);
37,747,560✔
4670

4671
    qDebug("%s subQIdx %d blockIdx:%" PRIu64 " rsp detail, numOfBlocks:%d, numOfRows:%" PRId64 ", numOfCols:%" PRId64 ", completed:%d", 
37,747,005✔
4672
      ctx->idStr, pParam->subQIdx, ctx->blockIdx, pRsp->numOfBlocks, pRsp->numOfRows, pRsp->numOfCols, pRsp->completed);
4673

4674
    ctx->blockIdx++;
37,747,005✔
4675

4676
    switch (nodeType(pParam->pRes)) {
37,747,560✔
4677
      case QUERY_NODE_REMOTE_VALUE: {
3,409,205✔
4678
        bool fetchDone = false;
3,409,205✔
4679
        handleRemoteValueRes(pParam, ctx, pRsp, &fetchDone);
3,409,205✔
4680
        qDebug("%s subQIdx %d handle remote value, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
3,409,760✔
4681
        if (!fetchDone) {
3,409,760✔
4682
          goto _exit;
57,147✔
4683
        }
4684
        break;
3,352,613✔
4685
      }
4686
      case QUERY_NODE_REMOTE_VALUE_LIST: {
12,901,334✔
4687
        bool fetchDone = false;
12,901,334✔
4688
        handleRemoteValueListRes(pParam, ctx, pRsp, &fetchDone);
12,901,334✔
4689
        qDebug("%s subQIdx %d handle remote value list finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
12,901,877✔
4690
        if (!fetchDone) {
12,901,877✔
4691
          goto _exit;
1,315,281✔
4692
        }
4693
        break;
11,586,596✔
4694
      }  
4695
      case QUERY_NODE_REMOTE_ROW: {
21,157,865✔
4696
        bool fetchDone = false;
21,157,865✔
4697
        handleRemoteRowRes(pParam, ctx, pRsp, &fetchDone);
21,157,865✔
4698
        qDebug("%s subQIdx %d handle remote row finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
21,157,865✔
4699
        if (!fetchDone) {
21,157,865✔
4700
          goto _exit;
502✔
4701
        }
4702
        break;
21,157,363✔
4703
      }
4704
      case QUERY_NODE_REMOTE_ZERO_ROWS: {
278,058✔
4705
        bool fetchDone = false;
278,058✔
4706
        handleRemoteZeroRowsRes(pParam, ctx, pRsp, &fetchDone);
278,058✔
4707
        qDebug("%s subQIdx %d handle remote zeroRows finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
278,058✔
4708
        if (!fetchDone) {
278,058✔
4709
          goto _exit;
×
4710
        }
4711
        break;
278,058✔
4712
      }
4713
      default:
×
4714
        qError("%s invalid scl fetch res node %d, subQIdx:%d", ctx->idStr, nodeType(pParam->pRes), pParam->subQIdx);
×
4715
        ctx->code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4716
        break;
×
4717
    }
4718
  } else {
4719
    ctx->code = rpcCvtErrCode(code);
880,230✔
4720
    if (ctx->code != code) {
880,230✔
4721
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s, cvted error: %s", ctx->idStr, pParam->subQIdx,
×
4722
             tstrerror(code), tstrerror(ctx->code));
4723
    } else {
4724
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s", ctx->idStr, pParam->subQIdx, tstrerror(code));
880,230✔
4725
    }
4726
  }
4727

4728
  qDebug("%s subQIdx %d sem_post subQ ready", ctx->idStr, pParam->subQIdx);
37,254,860✔
4729
  
4730
  code = tsem_post(&ctx->ready);
37,254,860✔
4731
  if (code != TSDB_CODE_SUCCESS) {
37,254,860✔
4732
    qError("failed to invoke post when scl fetch rsp is ready, code:%s", tstrerror(code));
×
4733
  }
4734

4735
_exit:
38,627,790✔
4736

4737
  code = taosReleaseRef(fetchObjRefPool, pParam->subJobRefId);
38,627,235✔
4738
  if (code != TSDB_CODE_SUCCESS) {
38,627,790✔
4739
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4740
  }
4741

4742
  taosMemoryFree(pMsg->pData);
38,627,790✔
4743

4744
  return code;
38,626,692✔
4745
}
4746

4747
#define STREAM_TASK_REPLICA_NUM 5
4748
#define STREAM_TASK_DEPLOY_NUM 3
4749

4750
int32_t sendFetchRemoteNodeReq(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes, bool reset) {
38,632,833✔
4751
  int32_t          code = TSDB_CODE_SUCCESS;
38,632,833✔
4752
  int32_t          lino = 0;
38,632,833✔
4753
  int32_t          innerIdx = 0;
38,632,833✔
4754
  SExecTaskInfo*   pTaskInfo = ctx->pTaskInfo;
38,632,833✔
4755
  bool             needStreamPesudoFuncVals = false;
38,630,010✔
4756

4757
  if (IS_STREAM_MODE(pTaskInfo)) {
38,630,010✔
4758
    innerIdx = pTaskInfo->pStreamRuntimeInfo->execId / STREAM_TASK_REPLICA_NUM;
1,164,508✔
4759
    subQIdx = subQIdx * STREAM_TASK_DEPLOY_NUM + innerIdx;
1,164,508✔
4760
  }
4761

4762
  SDownstreamSourceNode* pSource = (SDownstreamSourceNode*)taosArrayGetP(ctx->subEndPoints, subQIdx);
38,630,205✔
4763

4764
  SResFetchReq req = {0};
38,629,515✔
4765
  req.header.vgId = pSource->addr.nodeId;
38,629,515✔
4766
  req.sId = pSource->sId;
38,626,740✔
4767
  req.clientId = pSource->clientId;
38,629,479✔
4768
  req.taskId = pSource->taskId;
38,630,034✔
4769
  req.srcTaskId = ctx->taskId;
38,634,450✔
4770
  req.blockIdx = ctx->blockIdx;
38,627,814✔
4771
  req.queryId = ctx->queryId;
38,623,941✔
4772
  req.execId = pSource->execId;
38,626,716✔
4773

4774
  if (IS_STREAM_MODE(pTaskInfo)) {
38,633,376✔
4775
    req.queryId = pSource->clientId;
1,164,508✔
4776
    req.execId = pTaskInfo->pStreamRuntimeInfo->execId;
1,164,508✔
4777
    req.pStRtFuncInfo = &pTaskInfo->pStreamRuntimeInfo->funcInfo;
1,164,508✔
4778
    req.reset = reset;
1,164,508✔
4779

4780
    needStreamPesudoFuncVals = true;
1,164,508✔
4781
    subQIdx = (subQIdx - innerIdx) / STREAM_TASK_DEPLOY_NUM;
1,164,508✔
4782
  }
4783

4784
  int32_t msgSize = tSerializeSResFetchReq(NULL, 0, &req, needStreamPesudoFuncVals);
38,629,467✔
4785
  if (msgSize < 0) {
38,632,290✔
4786
    return msgSize;
×
4787
  }
4788

4789
  void* msg = taosMemoryCalloc(1, msgSize);
38,632,290✔
4790
  if (NULL == msg) {
38,623,422✔
4791
    return terrno;
×
4792
  }
4793

4794
  msgSize = tSerializeSResFetchReq(msg, msgSize, &req, needStreamPesudoFuncVals);
38,623,422✔
4795
  if (msgSize < 0) {
38,632,584✔
4796
    taosMemoryFree(msg);
×
4797
    return msgSize;
×
4798
  }
4799

4800
  qDebug("%s scl build fetch msg and send to nodeId:%d, ep:%s, clientId:0x%" PRIx64 " taskId:0x%" PRIx64
38,632,584✔
4801
         ", execId:%d, blockIdx:%" PRId64,
4802
         ctx->idStr, pSource->addr.nodeId, pSource->addr.epSet.eps[0].fqdn, pSource->clientId,
4803
         pSource->taskId, pSource->execId, req.blockIdx);
4804

4805
  // send the fetch remote task result reques
4806
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
38,632,833✔
4807
  if (NULL == pMsgSendInfo) {
38,626,149✔
4808
    taosMemoryFreeClear(msg);
×
4809
    qError("%s prepare message %d failed", ctx->idStr, (int32_t)sizeof(SMsgSendInfo));
×
4810
    return terrno;
×
4811
  }
4812

4813
  SScalarFetchParam* param = taosMemoryMalloc(sizeof(SScalarFetchParam));
38,626,149✔
4814
  if (NULL == param) {
38,628,924✔
4815
    taosMemoryFreeClear(msg);
×
4816
    taosMemoryFreeClear(pMsgSendInfo);
×
4817
    qError("%s prepare param %d failed", ctx->idStr, (int32_t)sizeof(SScalarFetchParam));
×
4818
    return terrno;
×
4819
  }
4820

4821
  if (ctx->code) {
38,628,924✔
4822
    qError("task has been killed, error:%s", tstrerror(ctx->code));
555✔
4823
    taosMemoryFree(param);
555✔
4824
    taosMemoryFreeClear(msg);
555✔
4825
    taosMemoryFreeClear(pMsgSendInfo);
555✔
4826
    code = ctx->code;
555✔
4827
    goto _end;
555✔
4828
  }
4829
  
4830
  param->subQIdx = subQIdx;
38,627,838✔
4831
  param->pRes = pRes;
38,631,711✔
4832
  param->subJobRefId = ctx->subJobRefId;
38,631,675✔
4833

4834
  pMsgSendInfo->param = param;
38,623,941✔
4835
  pMsgSendInfo->paramFreeFp = taosAutoMemoryFree;
38,632,278✔
4836
  pMsgSendInfo->msgInfo.pData = msg;
38,628,924✔
4837
  pMsgSendInfo->msgInfo.len = msgSize;
38,626,692✔
4838
  pMsgSendInfo->msgType = pSource->fetchMsgType;
38,626,185✔
4839
  pMsgSendInfo->fp = remoteFetchCallBack;
38,632,797✔
4840
  pMsgSendInfo->requestId = ctx->queryId;
38,621,757✔
4841

4842
  code = asyncSendMsgToServer(ctx->rpcHandle, &pSource->addr.epSet, &ctx->transporterId, pMsgSendInfo);
38,619,513✔
4843
  QUERY_CHECK_CODE(code, lino, _end);
38,635,005✔
4844
      
4845
_end:
38,635,560✔
4846

4847
  if (code != TSDB_CODE_SUCCESS) {
38,634,450✔
4848
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
555✔
4849
  }
4850
  
4851
  return code;
38,633,907✔
4852
}
4853

4854
int32_t fetchRemoteNodeImpl(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes) {
37,259,324✔
4855
  int32_t          code = TSDB_CODE_SUCCESS;
37,259,324✔
4856
  int32_t          lino = 0;
37,259,324✔
4857

4858
  ctx->blockIdx = 0;
37,259,324✔
4859

4860
  code = sendFetchRemoteNodeReq(ctx, subQIdx, pRes, true);
37,260,422✔
4861
  QUERY_CHECK_CODE(code, lino, _end);
37,260,422✔
4862

4863
  code = qSemWait(ctx->pTaskInfo, &ctx->ready);
37,259,867✔
4864
  if (isTaskKilled(ctx->pTaskInfo)) {
37,262,630✔
4865
    code = getTaskCode(ctx->pTaskInfo);
17,205✔
4866
  } else {
4867
    code = ctx->code;
37,245,425✔
4868
  }
4869
      
4870
_end:
37,263,185✔
4871

4872
  if (code != TSDB_CODE_SUCCESS) {
37,263,185✔
4873
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
1,854,723✔
4874
  }
4875
  return code;
37,263,185✔
4876
}
4877

4878
int32_t remoteNodeCopy(SNode* pSrc, SNode* pDst) {
118,560✔
4879
  int32_t code = 0, lino = 0;
118,560✔
4880
  
4881
  switch (nodeType(pSrc)) {
118,560✔
4882
    case QUERY_NODE_VALUE:
84,894✔
4883
      TAOS_CHECK_EXIT(valueNodeCopy((SValueNode*)pSrc, &((SRemoteValueNode*)pDst)->val));
84,894✔
4884
      ((SRemoteValueNode*)pDst)->val.node.type = QUERY_NODE_VALUE;
84,894✔
4885
      break;
84,894✔
4886
    case QUERY_NODE_REMOTE_VALUE_LIST: {
24,978✔
4887
      SRemoteValueListNode* pDstNode = (SRemoteValueListNode*)pDst;
24,978✔
4888
      memcpy(pDst, pSrc, sizeof(SRemoteValueListNode));
24,978✔
4889
      pDstNode->hashAllocated = false;      
24,978✔
4890
      break;
24,978✔
4891
    } 
4892
    case QUERY_NODE_REMOTE_ROW: {
8,688✔
4893
      SRemoteRowNode* pRemote = (SRemoteRowNode*)pDst;
8,688✔
4894
      TAOS_CHECK_EXIT(valueNodeCopy((SValueNode*)pSrc, &pRemote->val));
8,688✔
4895
      pRemote->valSet = true;
8,688✔
4896
      pRemote->hasValue = ((SRemoteRowNode*)pSrc)->hasValue;
8,688✔
4897
      pRemote->hasNull = ((SRemoteRowNode*)pSrc)->hasNull;
8,688✔
4898
      break;
8,688✔
4899
    }
4900
    case QUERY_NODE_REMOTE_ZERO_ROWS: {
×
4901
      SRemoteZeroRowsNode* pRemote = (SRemoteZeroRowsNode*)pDst;
×
4902
      TAOS_CHECK_EXIT(valueNodeCopy((SValueNode*)pSrc, &pRemote->val));
×
4903
      pRemote->val.node.type = QUERY_NODE_VALUE;
×
4904
      break;
×
4905
    }
4906
    default:
×
4907
      break;
×
4908
  }
4909

4910
_exit:
118,560✔
4911

4912
  if (code) {
118,560✔
4913
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4914
  }
4915

4916
  return code;
118,560✔
4917
}
4918

4919
int32_t qFetchRemoteNode(void* pCtx, int32_t subQIdx, SNode* pRes) {
37,379,549✔
4920
  STaskSubJobCtx*  ctx = (STaskSubJobCtx*)pCtx;
37,379,549✔
4921
  int32_t code = 0, lino = 0;
37,379,549✔
4922
  int32_t       subEndPoinsNum = taosArrayGetSize(ctx->subEndPoints);
37,379,549✔
4923
  if (subQIdx >= subEndPoinsNum) {
37,370,705✔
4924
    qError("%s invalid subQIdx %d, subEndPointsNum:%d", ctx->idStr, subQIdx, subEndPoinsNum);
×
4925
    return TSDB_CODE_QRY_SUBQ_NOT_FOUND;
×
4926
  }
4927

4928
  SNode** ppRes = taosArrayGet(ctx->subResNodes, subQIdx);
37,370,705✔
4929
  if (NULL == *ppRes) {
37,376,219✔
4930
    TAOS_CHECK_EXIT(fetchRemoteNodeImpl(ctx, subQIdx, pRes));
37,260,410✔
4931
    *ppRes = pRes;
35,407,907✔
4932
  } else {
4933
    TAOS_CHECK_EXIT(remoteNodeCopy(*ppRes, pRes));
118,560✔
4934
  }
4935

4936
_exit:
118,560✔
4937

4938
  if (code) {
37,381,745✔
4939
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
1,854,723✔
4940
  } else {
4941
    qDebug("%s %s subQIdx %d succeed", ctx->idStr, __func__, subQIdx);
35,527,022✔
4942
  }
4943

4944
  return code;
37,381,745✔
4945
}
STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE TRIAL · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2026 Coveralls, Inc