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

taosdata / TDengine / #5011

03 Apr 2026 03:59PM UTC coverage: 72.3% (+0.008%) from 72.292%
#5011

push

travis-ci

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

4053 of 5985 new or added lines in 68 files covered. (67.72%)

732 existing lines in 143 files now uncovered.

257430 of 356056 relevant lines covered (72.3%)

131834103.52 hits per line

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

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

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

30
#include "executil.h"
31
#include "executorInt.h"
32
#include "querytask.h"
33
#include "storageapi.h"
34
#include "ttypes.h"
35
#include "tutil.h"
36
#include "tjson.h"
37
#include "trpc.h"
38
#include "filter.h"
39
#include "operator.h"
40
#include "tref.h"
41
#include "streamMsg.h"
42

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

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

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

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

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

73
void initResultRowInfo(SResultRowInfo* pResultRowInfo) {
501,253,053✔
74
  pResultRowInfo->size = 0;
501,253,053✔
75
  pResultRowInfo->cur.pageId = -1;
501,279,519✔
76
}
501,305,298✔
77

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

80
void resetResultRow(SResultRow* pResultRow, size_t entrySize) {
948,381,848✔
81
  pResultRow->numOfRows = 0;
948,381,848✔
82
  pResultRow->nOrigRows = 0;
948,382,742✔
83
  pResultRow->closed = false;
948,382,503✔
84
  pResultRow->endInterp = false;
948,382,742✔
85
  pResultRow->startInterp = false;
948,382,503✔
86

87
  if (entrySize > 0) {
948,382,326✔
88
    memset(pResultRow->pEntryInfo, 0, entrySize);
948,382,742✔
89
  }
90
}
948,391,965✔
91

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

97
size_t getResultRowSize(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
298,959,887✔
98
  int32_t rowSize = (numOfOutput * sizeof(SResultRowEntryInfo)) + sizeof(SResultRow);
298,959,887✔
99

100
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,236,955,708✔
101
    rowSize += pCtx[i].resDataInfo.interBufSize;
938,044,563✔
102
  }
103

104
  return rowSize;
298,911,145✔
105
}
106

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

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

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

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

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

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

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

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

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

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

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

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

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

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

220
void cleanupGroupResInfo(SGroupResInfo* pGroupResInfo) {
137,408,758✔
221
  taosMemoryFreeClear(pGroupResInfo->pBuf);
137,408,758✔
222
  if (pGroupResInfo->freeItem) {
137,405,544✔
223
    //    taosArrayDestroy(pGroupResInfo->pRows);
224
    taosArrayDestroyEx(pGroupResInfo->pRows, freeEx);
×
225
    pGroupResInfo->freeItem = false;
×
226
    pGroupResInfo->pRows = NULL;
×
227
  } else {
228
    taosArrayDestroy(pGroupResInfo->pRows);
137,396,724✔
229
    pGroupResInfo->pRows = NULL;
137,386,946✔
230
  }
231
  pGroupResInfo->index = 0;
137,392,040✔
232
  pGroupResInfo->delIndex = 0;
137,387,154✔
233
}
137,400,860✔
234

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

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

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

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

255
int32_t initGroupedResultInfo(SGroupResInfo* pGroupResInfo, SSHashObj* pHashmap, int32_t order) {
104,203,442✔
256
  int32_t code = TSDB_CODE_SUCCESS;
104,203,442✔
257
  int32_t lino = 0;
104,203,442✔
258
  if (pGroupResInfo->pRows != NULL) {
104,203,442✔
259
    taosArrayDestroy(pGroupResInfo->pRows);
2,389,575✔
260
  }
261
  if (pGroupResInfo->pBuf) {
104,207,218✔
262
    taosMemoryFree(pGroupResInfo->pBuf);
2,389,575✔
263
    pGroupResInfo->pBuf = NULL;
2,389,575✔
264
  }
265

266
  // extract the result rows information from the hash map
267
  int32_t size = tSimpleHashGetSize(pHashmap);
104,203,596✔
268

269
  void* pData = NULL;
104,204,460✔
270
  pGroupResInfo->pRows = taosArrayInit(size, POINTER_BYTES);
104,204,460✔
271
  QUERY_CHECK_NULL(pGroupResInfo->pRows, code, lino, _end, terrno);
104,198,939✔
272

273
  size_t  keyLen = 0;
104,204,146✔
274
  int32_t iter = 0;
104,206,739✔
275
  int64_t bufLen = 0, offset = 0;
104,202,696✔
276

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

283
  pGroupResInfo->pBuf = taosMemoryMalloc(bufLen);
104,159,093✔
284
  QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
104,201,183✔
285

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

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

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

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

301
  if (order == TSDB_ORDER_ASC || order == TSDB_ORDER_DESC) {
104,194,244✔
302
    __compar_fn_t fn = (order == TSDB_ORDER_ASC) ? resultrowComparAsc : resultrowComparDesc;
6,398,976✔
303
    size = POINTER_BYTES;
6,398,976✔
304
    taosSort(pGroupResInfo->pRows->pData, taosArrayGetSize(pGroupResInfo->pRows), size, fn);
6,398,976✔
305
  }
306

307
  pGroupResInfo->index = 0;
104,193,996✔
308

309
_end:
104,199,857✔
310
  if (code != TSDB_CODE_SUCCESS) {
104,208,574✔
311
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
312
  }
313
  return code;
104,208,574✔
314
}
315

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

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

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

332
  return pGroupResInfo->index < taosArrayGetSize(pGroupResInfo->pRows);
413,446,366✔
333
}
334

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

340
  return (int32_t)taosArrayGetSize(pGroupResInfo->pRows);
220,992,532✔
341
}
342

343
SArray* createSortInfo(SNodeList* pNodeList) {
55,130,188✔
344
  size_t numOfCols = 0;
55,130,188✔
345

346
  if (pNodeList != NULL) {
55,130,188✔
347
    numOfCols = LIST_LENGTH(pNodeList);
54,909,301✔
348
  } else {
349
    numOfCols = 0;
220,887✔
350
  }
351

352
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
55,133,357✔
353
  if (pList == NULL) {
55,117,357✔
354
    return pList;
×
355
  }
356

357
  for (int32_t i = 0; i < numOfCols; ++i) {
125,228,534✔
358
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
70,102,462✔
359
    if (!pSortKey) {
70,106,757✔
360
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
361
      taosArrayDestroy(pList);
×
362
      pList = NULL;
×
363
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
364
      break;
×
365
    }
366
    SBlockOrderInfo bi = {0};
70,106,757✔
367
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
70,106,727✔
368
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);
70,100,547✔
369

370
    if (nodeType(pSortKey->pExpr) != QUERY_NODE_COLUMN) {
70,103,699✔
371
      qError("invalid order by expr type:%d", nodeType(pSortKey->pExpr));
×
372
      taosArrayDestroy(pList);
×
373
      pList = NULL;
×
374
      terrno = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
375
      break;
×
376
    }
377
    
378
    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
70,092,850✔
379
    bi.slotId = pColNode->slotId;
70,105,955✔
380
    void* tmp = taosArrayPush(pList, &bi);
70,109,666✔
381
    if (!tmp) {
70,109,666✔
382
      taosArrayDestroy(pList);
×
383
      pList = NULL;
×
384
      break;
×
385
    }
386
  }
387

388
  return pList;
55,125,482✔
389
}
390

391
SSDataBlock* createDataBlockFromDescNode(void* p) {
802,979,442✔
392
  SDataBlockDescNode* pNode = (SDataBlockDescNode*)p;
802,979,442✔
393
  int32_t      numOfCols = LIST_LENGTH(pNode->pSlots);
802,979,442✔
394
  SSDataBlock* pBlock = NULL;
803,109,024✔
395
  int32_t      code = TSDB_CODE_SUCCESS;
803,089,285✔
396
  int32_t      lino = 0;
803,089,285✔
397

398
  code = createDataBlock(&pBlock);
803,089,285✔
399
  QUERY_CHECK_CODE(code, lino, _return);
802,936,274✔
400

401
  pBlock->info.id.blockId = pNode->dataBlockId;
802,936,274✔
402
  pBlock->info.calWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
803,024,872✔
403
  pBlock->info.watermark = INT64_MIN;
803,043,289✔
404

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

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

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

425
  return pBlock;
803,248,044✔
426
_return:
54,234✔
427
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
428
  blockDataDestroy(pBlock);
×
429
  terrno = code;
×
430
  return NULL;
×
431
}
432

433
int32_t prepareDataBlockBuf(SSDataBlock* pDataBlock, SColMatchInfo* pMatchInfo) {
275,395,510✔
434
  SDataBlockInfo* pBlockInfo = &pDataBlock->info;
275,395,510✔
435

436
  for (int32_t i = 0; i < taosArrayGetSize(pMatchInfo->pList); ++i) {
1,230,100,989✔
437
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
975,958,534✔
438
    if (!pItem) {
975,838,675✔
UNCOV
439
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
UNCOV
440
      return terrno;
×
441
    }
442

443
    if (pItem->isPk) {
975,838,675✔
444
      SColumnInfoData* pInfoData = taosArrayGet(pDataBlock->pDataBlock, pItem->dstSlotId);
21,379,074✔
445
      if (!pInfoData) {
21,038,496✔
446
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
447
        return terrno;
×
448
      }
449
      pBlockInfo->pks[0].type = pInfoData->info.type;
21,038,496✔
450
      pBlockInfo->pks[1].type = pInfoData->info.type;
21,062,289✔
451

452
      // allocate enough buffer size, which is pInfoData->info.bytes
453
      if (IS_VAR_DATA_TYPE(pItem->dataType.type)) {
21,051,566✔
454
        pBlockInfo->pks[0].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
6,437,160✔
455
        if (pBlockInfo->pks[0].pData == NULL) {
6,422,530✔
456
          return terrno;
×
457
        }
458

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

465
        pBlockInfo->pks[0].nData = pInfoData->info.bytes;
6,428,491✔
466
        pBlockInfo->pks[1].nData = pInfoData->info.bytes;
6,428,455✔
467
      }
468

469
      break;
21,057,046✔
470
    }
471
  }
472

473
  return TSDB_CODE_SUCCESS;
275,232,962✔
474
}
475

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

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

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

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

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

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

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

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

560
  return DEAL_RES_CONTINUE;
2,568,699✔
561
}
562

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

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

573
    return TSDB_CODE_SUCCESS;
×
574
  }
575

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

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

600
    return code;
×
601
  }
602

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

606
  nodesDestroyNode(pNew);
366,957✔
607
  return TSDB_CODE_SUCCESS;
366,957✔
608
}
609

610
static EDealRes getColumn(SNode** pNode, void* pContext) {
61,262,320✔
611
  tagFilterAssist* pData = (tagFilterAssist*)pContext;
61,262,320✔
612
  SColumnNode*     pSColumnNode = NULL;
61,262,320✔
613
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
61,271,246✔
614
    pSColumnNode = *(SColumnNode**)pNode;
21,297,191✔
615
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
39,976,607✔
616
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
1,069,117✔
617
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
1,069,117✔
618
      pData->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
722,772✔
619
      if (NULL == pSColumnNode) {
722,772✔
620
        return DEAL_RES_ERROR;
×
621
      }
622
      pSColumnNode->colId = -1;
722,772✔
623
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
722,772✔
624
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
722,772✔
625
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
722,772✔
626
      nodesDestroyNode(*pNode);
722,772✔
627
      *pNode = (SNode*)pSColumnNode;
722,772✔
628
    } else {
629
      return DEAL_RES_CONTINUE;
346,345✔
630
    }
631
  } else {
632
    return DEAL_RES_CONTINUE;
38,901,109✔
633
  }
634

635
  void* data = taosHashGet(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
22,020,236✔
636
  if (!data) {
22,016,065✔
637
    int32_t tempRes =
638
        taosHashPut(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
19,044,076✔
639
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
19,049,884✔
640
      return DEAL_RES_ERROR;
×
641
    }
642
    pSColumnNode->slotId = pData->index++;
19,049,884✔
643
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
19,048,427✔
644
                         .type = pSColumnNode->node.resType.type,
19,042,758✔
645
                         .bytes = pSColumnNode->node.resType.bytes,
19,047,970✔
646
                         .pk = pSColumnNode->isPk};
19,042,038✔
647
#if TAG_FILTER_DEBUG
648
    qDebug("tagfilter build column info, slotId:%d, colId:%d, type:%d", pSColumnNode->slotId, cInfo.colId, cInfo.type);
649
#endif
650
    void* tmp = taosArrayPush(pData->cInfoList, &cInfo);
19,046,905✔
651
    if (!tmp) {
19,051,650✔
652
      return DEAL_RES_ERROR;
×
653
    }
654
  } else {
655
    SColumnNode* col = *(SColumnNode**)data;
2,971,989✔
656
    pSColumnNode->slotId = col->slotId;
2,971,465✔
657
  }
658

659
  return DEAL_RES_CONTINUE;
22,016,596✔
660
}
661

662
static int32_t createResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
17,756,367✔
663
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
17,756,367✔
664
  if (pColumnData == NULL) {
17,753,839✔
665
    return terrno;
×
666
  }
667

668
  pColumnData->info.type = pType->type;
17,753,839✔
669
  pColumnData->info.bytes = pType->bytes;
17,755,292✔
670
  pColumnData->info.scale = pType->scale;
17,761,100✔
671
  pColumnData->info.precision = pType->precision;
17,751,452✔
672

673
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
17,757,589✔
674
  if (code != TSDB_CODE_SUCCESS) {
17,746,459✔
675
    terrno = code;
×
676
    releaseColInfoData(pColumnData);
×
677
    return terrno;
×
678
  }
679

680
  pParam->columnData = pColumnData;
17,746,459✔
681
  pParam->colAlloced = true;
17,754,821✔
682
  return TSDB_CODE_SUCCESS;
17,748,360✔
683
}
684

685
static void releaseColInfoData(void* pCol) {
5,047,082✔
686
  if (pCol) {
5,047,082✔
687
    SColumnInfoData* col = (SColumnInfoData*)pCol;
5,047,740✔
688
    colDataDestroy(col);
5,047,740✔
689
    taosMemoryFree(col);
5,047,740✔
690
  }
691
}
5,046,539✔
692

693
void freeItem(void* p) {
211,220,856✔
694
  STUidTagInfo* pInfo = p;
211,220,856✔
695
  if (pInfo->pTagVal != NULL) {
211,220,856✔
696
    taosMemoryFree(pInfo->pTagVal);
210,837,972✔
697
  }
698
}
211,218,110✔
699

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

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

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

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

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

775
  return TSDB_CODE_SUCCESS;
10,856✔
776
}
777

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

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

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

803
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
10,856✔
804
    extractTagDataEntry((SOperatorNode*)pTagCond, pIdWithVal);
×
805
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
10,856✔
806
    SNode* pChild = NULL;
10,856✔
807
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
32,568✔
808
      extractTagDataEntry((SOperatorNode*)pChild, pIdWithVal);
21,712✔
809
    }
810
  }
811

812
  taosArraySort(pIdWithVal, compareTagDataEntry);
10,856✔
813

814
  return TSDB_CODE_SUCCESS;
10,856✔
815
}
816

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

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

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

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

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

851
static int32_t genTagFilterDigest(const SNode* pTagCond, T_MD5_CTX* pContext) {
33,948✔
852
  if (pTagCond == NULL) {
33,948✔
853
    return TSDB_CODE_SUCCESS;
32,016✔
854
  }
855

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

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

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

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

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

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

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

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

905
int32_t qGetColumnsFromNodeList(void* data, bool isList, SArray** pColList) {
17,322,980✔
906
  int32_t code = TSDB_CODE_SUCCESS;
17,322,980✔
907
  tagFilterAssist ctx = {0};
17,322,980✔
908
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
17,324,151✔
909
  if (ctx.colHash == NULL) {
17,321,707✔
910
    code = terrno;
×
911
    goto end;
×
912
  }
913

914
  ctx.index = 0;
17,321,707✔
915
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
17,321,707✔
916
  if (ctx.cInfoList == NULL) {
17,324,010✔
917
    code = terrno;
6,215✔
918
    goto end;
×
919
  }
920

921
  if (isList) {
17,317,795✔
922
    SNode* pNode = NULL;
4,609,402✔
923
    FOREACH(pNode, (SNodeList*)data) {
9,657,560✔
924
      nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
5,048,833✔
925
      if (TSDB_CODE_SUCCESS != ctx.code) {
5,048,833✔
926
        code = ctx.code;
×
927
        goto end;
×
928
      }
929
      REPLACE_NODE(pNode);
5,048,833✔
930
    }
931
  } else {
932
    SNode* pNode = (SNode*)data;
12,708,393✔
933
    nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
12,709,354✔
934
    if (TSDB_CODE_SUCCESS != ctx.code) {
12,713,443✔
935
      code = ctx.code;
×
936
      goto end;
×
937
    }
938
  }
939
  
940
  if (pColList != NULL) *pColList = ctx.cInfoList;
17,311,085✔
941
  ctx.cInfoList = NULL;
17,321,106✔
942

943
end:
17,333,044✔
944
  taosHashCleanup(ctx.colHash);
17,319,528✔
945
  taosArrayDestroy(ctx.cInfoList);
17,306,607✔
946
  return code;
17,307,548✔
947
}
948

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

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

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

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

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

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

1060
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
253,518✔
1061
    SColumnInfo* tmp = (SColumnInfo*)taosArrayGet(pColList, i);
167,782✔
1062
    if (tmp != NULL && tmp->colId == -1) {
167,782✔
1063
      tbNameIndex = i;
85,736✔
1064
    }
1065
  }
1066
  
1067
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
85,736✔
1068
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
85,736✔
1069
  taosArrayDestroy(pColList);
85,736✔
1070
  if (pResBlock == NULL) {
85,736✔
1071
    code = terrno;
275✔
1072
    goto end;
275✔
1073
  }
1074

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

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

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

1084
  SNode* pNode = NULL;
85,461✔
1085
  FOREACH(pNode, group) {
252,693✔
1086
    SScalarParam output = {0};
167,232✔
1087

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

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

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

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

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

1129
  for (int i = 0; i < rows; i++) {
408,611✔
1130
    gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
323,150✔
1131
    QUERY_CHECK_NULL(gInfo, code, lino, end, terrno);
323,150✔
1132

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

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

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

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

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

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

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

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

1202
    nodesFree(listNode);
×
1203

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

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

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

1220
  for (int32_t i = 0; i < rows; ++i) {
29,063,958✔
1221
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
24,539,434✔
1222
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
24,539,006✔
1223
    STUidTagInfo info = {.uid = pkeyInfo->uid};
24,539,006✔
1224
    void*        tmp = taosArrayPush(pUidTagList, &info);
24,539,873✔
1225
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
24,539,873✔
1226
  }
1227

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

1237
  SArray* pColList = NULL;
4,524,209✔
1238
  code = qGetColumnsFromNodeList(group, true, &pColList); 
4,524,209✔
1239
  if (code != TSDB_CODE_SUCCESS) {
4,523,476✔
1240
    goto end;
×
1241
  }
1242

1243
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
4,523,476✔
1244
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
4,522,275✔
1245
  taosArrayDestroy(pColList);
4,523,805✔
1246
  if (pResBlock == NULL) {
4,524,025✔
1247
    code = terrno;
×
1248
    goto end;
×
1249
  }
1250

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

1254
  pBlockList = taosArrayInit(2, POINTER_BYTES);
4,524,025✔
1255
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
4,524,209✔
1256

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

1260
  groupData = taosArrayInit(2, POINTER_BYTES);
4,524,019✔
1261
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
4,524,019✔
1262

1263
  SNode* pNode = NULL;
4,524,019✔
1264
  FOREACH(pNode, group) {
9,404,091✔
1265
    SScalarParam output = {0};
4,879,781✔
1266

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

1292
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
4,880,508✔
1293
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
4,776,992✔
1294
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
4,776,992✔
1295
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
4,776,265✔
1296
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
4,776,265✔
1297
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
104,059✔
1298
      continue;
×
1299
    } else {
1300
      gTaskScalarExtra.pStreamInfo = NULL;
104,059✔
1301
      gTaskScalarExtra.pStreamRange = NULL;
104,059✔
1302
      code = scalarCalculate(pNode, pBlockList, &output, &gTaskScalarExtra);
104,059✔
1303
    }
1304

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

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

1314
  int32_t baseIdLen = gIdFromBaseId ? sizeof(int64_t) : 0;
4,523,476✔
1315
  int32_t keyLen = baseIdLen;
4,523,476✔
1316
  SNode*  node;
1317
  FOREACH(node, group) {
9,401,849✔
1318
    SExprNode* pExpr = (SExprNode*)node;
4,879,561✔
1319
    keyLen += pExpr->resType.bytes;
4,879,561✔
1320
  }
1321

1322
  int32_t nullFlagSize = sizeof(int8_t) * LIST_LENGTH(group);
4,520,660✔
1323
  keyLen += nullFlagSize;
4,520,660✔
1324

1325
  keyBuf = taosMemoryCalloc(1, keyLen);
4,520,660✔
1326
  if (keyBuf == NULL) {
4,522,037✔
1327
    code = terrno;
×
1328
    goto end;
×
1329
  }
1330

1331
  if (initRemainGroups) {
4,522,037✔
1332
    pTableListInfo->remainGroups =
2,082,314✔
1333
        taosHashInit(rows, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
2,081,771✔
1334
    if (pTableListInfo->remainGroups == NULL) {
2,082,314✔
1335
      code = terrno;
×
1336
      goto end;
×
1337
    }
1338
  }
1339

1340
  for (int i = 0; i < rows; i++) {
29,059,166✔
1341
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
24,535,281✔
1342
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
24,533,355✔
1343

1344
    if (groupIdMap != NULL){
24,533,355✔
1345
      gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
74,081✔
1346
    }
1347

1348
    if (baseIdLen > 0) {
24,533,411✔
NEW
1349
      *(uint64_t*)keyBuf = info->baseGId;
×
1350
    }
1351
    char* isNull = (char*)keyBuf + baseIdLen;
24,533,411✔
1352
    char* pStart = (char*)keyBuf + baseIdLen + sizeof(int8_t) * LIST_LENGTH(group);
24,533,815✔
1353
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
50,712,863✔
1354
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
26,173,390✔
1355

1356
      if (groupIdMap != NULL && gInfo != NULL) {
26,173,390✔
1357
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
117,094✔
1358
        if (ret != TSDB_CODE_SUCCESS) {
117,284✔
1359
          qError("buildGroupInfo failed at line %d since %s", __LINE__, tstrerror(ret));
×
1360
          taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1361
          gInfo = NULL;
×
1362
        }
1363
      }
1364
      
1365
      if (colDataIsNull_s(pValue, i)) {
52,347,627✔
1366
        isNull[j] = 1;
105,708✔
1367
      } else {
1368
        isNull[j] = 0;
26,068,339✔
1369
        char* data = colDataGetData(pValue, i);
26,068,263✔
1370
        if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
26,069,349✔
1371
          // if (tTagIsJson(data)) {
1372
          //   code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
1373
          //   goto end;
1374
          // }
1375
          if (tTagIsJsonNull(data)) {
81,174✔
1376
            isNull[j] = 1;
×
1377
            continue;
×
1378
          }
1379
          int32_t len = getJsonValueLen(data);
81,174✔
1380
          memcpy(pStart, data, len);
81,174✔
1381
          pStart += len;
81,174✔
1382
        } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
25,988,718✔
1383
          if (IS_STR_DATA_BLOB(pValue->info.type)) {
22,291,146✔
UNCOV
1384
            if (blobDataTLen(data) > TSDB_MAX_BLOB_LEN) {
×
1385
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1386
              goto end;
×
1387
            }
1388
            memcpy(pStart, data, blobDataTLen(data));
×
1389
            pStart += blobDataTLen(data);
×
1390
          } else {
1391
            if (varDataTLen(data) > pValue->info.bytes) {
22,287,387✔
1392
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1393
              goto end;
×
1394
            }
1395
            memcpy(pStart, data, varDataTLen(data));
22,287,387✔
1396
            pStart += varDataTLen(data);
22,287,930✔
1397
          }
1398
        } else {
1399
          memcpy(pStart, data, pValue->info.bytes);
3,700,671✔
1400
          pStart += pValue->info.bytes;
3,701,874✔
1401
        }
1402
      }
1403
    }
1404

1405
    int32_t len = (int32_t)(pStart - (char*)keyBuf);
24,536,430✔
1406
    info->groupId = calcGroupId(keyBuf, len);
24,536,430✔
1407
    if (groupIdMap != NULL && gInfo != NULL) {
24,534,087✔
1408
      int32_t ret = taosHashPut(groupIdMap, &info->groupId, sizeof(info->groupId), &gInfo, POINTER_BYTES);
74,327✔
1409
      if (ret != TSDB_CODE_SUCCESS) {
74,327✔
1410
        qError("put groupid to map failed at line %d since %s", __LINE__, tstrerror(ret));
×
1411
        taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
×
1412
      }
1413
      qDebug("put groupid to map gid:%" PRIu64, info->groupId);
74,327✔
1414
      gInfo = NULL;
74,327✔
1415
    }
1416
    if (initRemainGroups) {
24,534,087✔
1417
      // groupId ~ table uid
1418
      code = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId), &(info->uid),
11,872,888✔
1419
                         sizeof(info->uid));
1420
      if (code == TSDB_CODE_DUP_KEY) {
11,875,387✔
1421
        code = TSDB_CODE_SUCCESS;
816,254✔
1422
      }
1423
      QUERY_CHECK_CODE(code, lino, end);
11,875,387✔
1424
    }
1425
  }
1426

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

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

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

1440
end:
4,523,885✔
1441
  taosMemoryFreeClear(keyBuf);
4,523,805✔
1442
  blockDataDestroy(pResBlock);
4,524,085✔
1443
  taosArrayDestroy(pBlockList);
4,524,209✔
1444
  taosArrayDestroyEx(pUidTagList, freeItem);
4,523,666✔
1445
  taosArrayDestroyP(groupData, releaseColInfoData);
4,523,551✔
1446
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
4,524,209✔
1447

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

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

1458
  int32_t ret = strcmp(pName1, pName2);
743,440✔
1459
  if (ret == 0) {
743,440✔
1460
    return 0;
15,057✔
1461
  } else {
1462
    return (ret > 0) ? 1 : -1;
728,383✔
1463
  }
1464
}
1465

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

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

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

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

1489
  size_t numOfTables = taosArrayGetSize(pTbList);
395,698✔
1490

1491
  // order the name
1492
  taosArraySort(pTbList, nameComparFn);
395,698✔
1493

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

1502
  for (int32_t i = 1; i < numOfTables; ++i) {
703,971✔
1503
    char** name = taosArrayGetLast(pNewList);
308,512✔
1504
    char** nameInOldList = taosArrayGet(pTbList, i);
308,273✔
1505
    QUERY_CHECK_NULL(nameInOldList, code, lino, _end, terrno);
308,273✔
1506
    if (strcmp(*name, *nameInOldList) == 0) {
308,273✔
1507
      continue;
7,868✔
1508
    }
1509

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

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

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

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

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

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

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

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

1545
static FilterCondType checkTagCond(SNode* cond) {
15,615,268✔
1546
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
15,615,268✔
1547
    return FILTER_NO_LOGIC;
11,332,752✔
1548
  }
1549
  if (nodeType(cond) == QUERY_NODE_LOGIC_CONDITION && ((SLogicConditionNode*)cond)->condType == LOGIC_COND_TYPE_AND) {
4,284,007✔
1550
    return FILTER_AND;
3,731,005✔
1551
  }
1552
  return FILTER_OTHER;
551,209✔
1553
}
1554

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

1560
  SLogicConditionNode* pNode = (SLogicConditionNode*)cond;
3,739,267✔
1561
  SNodeList*           pList = (SNodeList*)pNode->pParameterList;
3,739,267✔
1562

1563
  int32_t len = LIST_LENGTH(pList);
3,738,505✔
1564
  if (len <= 0) {
3,739,639✔
1565
    return -1;
×
1566
  }
1567

1568
  SListCell* cell = pList->pHead;
3,739,639✔
1569
  for (int i = 0; i < len; i++) {
11,613,959✔
1570
    if (cell == NULL) break;
7,881,008✔
1571
    if (nodeType(cell->pNode) == QUERY_NODE_OPERATOR && optimizeTbnameInCondImpl(pVnode, list, cell->pNode, pAPI, suid) == 0) {
7,881,008✔
1572
      return 0;
8,134✔
1573
    }
1574
    cell = cell->pNext;
7,874,239✔
1575
  }
1576
  return -1;
3,732,951✔
1577
}
1578

1579
static int32_t optimizeTbnameInCond(void* pVnode, int64_t suid, SArray* list, SNode* cond, SStorageAPI* pAPI) {
16,012,278✔
1580
  int32_t code = 0;
16,012,278✔
1581
  int32_t lino = 0;
16,012,278✔
1582
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
16,012,278✔
1583
    code = optimizeTbnameInCondImpl(pVnode, list, cond, pAPI, suid);
11,721,591✔
1584
  } else {
1585
    code = doInWithAnd(cond, pVnode, list, pAPI, suid);
4,292,703✔
1586
    QUERY_CHECK_CODE(code, lino, end);
4,291,104✔
1587
    code = pAPI->metaFn.getTableTagsByUid(pVnode, suid, list);
7,066✔
1588
  }
1589
  
1590
end:
16,011,597✔
1591
  return code;
16,011,597✔
1592
}
1593

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

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

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

1626
  for (int i = 0; i < numOfTables; i++) {
1,092,279✔
1627
    char* name = taosArrayGetP(pTbList, i);
696,581✔
1628
    uint64_t uid = 0, csuid = 0;
696,581✔
1629
    if (pStoreAPI->metaFn.getTableUidByName(pVnode, name, &uid) != 0) {
696,581✔
1630
      continue;
293,920✔
1631
    }
1632

1633
    ETableType tbType = TSDB_TABLE_MAX;
402,661✔
1634
    if (pStoreAPI->metaFn.getTableTypeSuidByName(pVnode, name, &tbType, &csuid) == 0 &&
402,661✔
1635
        (tbType == TSDB_CHILD_TABLE || tbType == TSDB_VIRTUAL_CHILD_TABLE) &&
402,661✔
1636
        csuid == suid) {
396,558✔
1637
      if (NULL == uHash || taosHashGet(uHash, &uid, sizeof(uid)) != NULL) {
394,390✔
1638
        STUidTagInfo s = {.uid = uid, .name = name, .pTagVal = NULL};
393,305✔
1639
        void*        tmp = taosArrayPush(pExistedUidList, &s);
393,305✔
1640
        if (!tmp) {
393,305✔
1641
          code = terrno;
×
1642
          goto end;
×
1643
        }
1644
      }
1645
    } 
1646
  }
1647
  taosArraySort(pExistedUidList, filterTableInfoCompare);
395,698✔
1648
  taosArrayRemoveDuplicate(pExistedUidList, filterTableInfoCompare, NULL);
395,698✔
1649
end:
395,698✔
1650
  taosArrayDestroy(pTbList);
395,698✔
1651
  taosHashCleanup(uHash);
395,698✔
1652
  return code;
395,698✔
1653
}
1654

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

1663
  if ((pNode->pLeft != NULL && ((nodeType(pNode->pLeft) == QUERY_NODE_FUNCTION &&
954,191✔
1664
                                 ((SFunctionNode*)pNode->pLeft)->funcType == FUNCTION_TYPE_TBNAME)) ||
397,922✔
1665
       (nodeType(pNode->pLeft) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pLeft)->colType == COLUMN_TYPE_TBNAME)) &&
555,360✔
1666
      (pNode->pRight != NULL && nodeType(pNode->pRight) == QUERY_NODE_NODE_LIST)) {
397,373✔
1667
    SNodeListNode* pList = (SNodeListNode*)pNode->pRight;
395,698✔
1668

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

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

1677
  return -1;
558,286✔
1678
}
1679

1680
SSDataBlock* createTagValBlockForFilter(SArray* pColList, int32_t numOfTables, SArray* pUidTagList, void* pVnode,
19,027,930✔
1681
                                        SStorageAPI* pStorageAPI) {
1682
  int32_t      code = TSDB_CODE_SUCCESS;
19,027,930✔
1683
  int32_t      lino = 0;
19,027,930✔
1684
  SSDataBlock* pResBlock = NULL;
19,027,930✔
1685
  code = createDataBlock(&pResBlock);
19,033,612✔
1686
  QUERY_CHECK_CODE(code, lino, _end);
19,026,621✔
1687

1688
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
39,797,355✔
1689
    SColumnInfoData colInfo = {0};
20,765,502✔
1690
    void*           tmp = taosArrayGet(pColList, i);
20,759,519✔
1691
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
20,746,456✔
1692
    colInfo.info = *(SColumnInfo*)tmp;
20,746,456✔
1693
    code = blockDataAppendColInfo(pResBlock, &colInfo);
20,745,876✔
1694
    QUERY_CHECK_CODE(code, lino, _end);
20,764,020✔
1695
  }
1696

1697
  code = blockDataEnsureCapacity(pResBlock, numOfTables);
19,028,688✔
1698
  if (code != TSDB_CODE_SUCCESS) {
19,026,049✔
1699
    terrno = code;
×
1700
    blockDataDestroy(pResBlock);
×
1701
    return NULL;
×
1702
  }
1703

1704
  pResBlock->info.rows = numOfTables;
19,026,049✔
1705

1706
  int32_t numOfCols = taosArrayGetSize(pResBlock->pDataBlock);
19,018,033✔
1707

1708
  for (int32_t i = 0; i < numOfTables; i++) {
232,400,281✔
1709
    STUidTagInfo* p1 = taosArrayGet(pUidTagList, i);
213,354,887✔
1710
    QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
213,356,913✔
1711

1712
    for (int32_t j = 0; j < numOfCols; j++) {
437,316,124✔
1713
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, j);
223,817,982✔
1714
      QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
223,809,355✔
1715

1716
      if (pColInfo->info.colId == -1) {  // tbname
223,809,355✔
1717
        char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
7,610,802✔
1718
        if (p1->name != NULL) {
7,611,069✔
1719
          STR_TO_VARSTR(str, p1->name);
5,965✔
1720
        } else {  // name is not retrieved during filter
1721
          code = pStorageAPI->metaFn.getTableNameByUid(pVnode, p1->uid, str);
7,603,453✔
1722
          QUERY_CHECK_CODE(code, lino, _end);
7,598,790✔
1723
        }
1724

1725
        code = colDataSetVal(pColInfo, i, str, false);
7,604,480✔
1726
        QUERY_CHECK_CODE(code, lino, _end);
7,610,403✔
1727
#if TAG_FILTER_DEBUG
1728
        qDebug("tagfilter uid:%ld, tbname:%s", *uid, str + 2);
1729
#endif
1730
      } else {
1731
        STagVal tagVal = {0};
216,176,892✔
1732
        tagVal.cid = pColInfo->info.colId;
216,227,128✔
1733
        if (p1->pTagVal == NULL) {
216,240,388✔
1734
          colDataSetNULL(pColInfo, i);
4,600✔
1735
        } else {
1736
          const char* p = pStorageAPI->metaFn.extractTagVal(p1->pTagVal, pColInfo->info.type, &tagVal);
216,231,314✔
1737

1738
          if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
216,322,755✔
1739
            colDataSetNULL(pColInfo, i);
2,263,162✔
1740
          } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
214,062,037✔
1741
            code = colDataSetVal(pColInfo, i, p, false);
641,709✔
1742
            QUERY_CHECK_CODE(code, lino, _end);
641,709✔
1743
          } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
352,837,228✔
1744
            if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
139,406,460✔
1745
              QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
1746
            }
1747
            char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
139,428,410✔
1748
            QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
139,427,027✔
1749
            varDataSetLen(tmp, tagVal.nData);
139,427,027✔
1750
            memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
139,428,675✔
1751
            code = colDataSetVal(pColInfo, i, tmp, false);
139,429,806✔
1752
#if TAG_FILTER_DEBUG
1753
            qDebug("tagfilter varch:%s", tmp + 2);
1754
#endif
1755
            taosMemoryFree(tmp);
139,447,310✔
1756
            QUERY_CHECK_CODE(code, lino, _end);
139,433,921✔
1757
          } else {
1758
            code = colDataSetVal(pColInfo, i, (const char*)&tagVal.i64, false);
73,981,659✔
1759
            QUERY_CHECK_CODE(code, lino, _end);
74,002,563✔
1760
#if TAG_FILTER_DEBUG
1761
            if (pColInfo->info.type == TSDB_DATA_TYPE_INT) {
1762
              qDebug("tagfilter int:%d", *(int*)(&tagVal.i64));
1763
            } else if (pColInfo->info.type == TSDB_DATA_TYPE_DOUBLE) {
1764
              qDebug("tagfilter double:%f", *(double*)(&tagVal.i64));
1765
            }
1766
#endif
1767
          }
1768
        }
1769
      }
1770
    }
1771
  }
1772

1773
_end:
19,038,789✔
1774
  if (code != TSDB_CODE_SUCCESS) {
19,044,607✔
1775
    blockDataDestroy(pResBlock);
4,400✔
1776
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
275✔
1777
    terrno = code;
275✔
1778
    return NULL;
275✔
1779
  }
1780
  return pResBlock;
19,040,207✔
1781
}
1782

1783
static int32_t doSetQualifiedUid(SArray* pUidList, const SArray* pUidTagList, SColumnInfoData* colData) {
15,992,008✔
1784
  taosArrayClear(pUidList);
15,992,008✔
1785

1786
  int32_t       numOfTables = taosArrayGetSize(pUidTagList);
15,983,559✔
1787
  bool* pResultList = colData == NULL ? NULL : (bool*)colData->pData;
15,985,343✔
1788
  for (int32_t i = 0; i < numOfTables; ++i) {
202,349,801✔
1789
    if (pResultList != NULL && !pResultList[i]) {
186,343,734✔
1790
       continue;
106,399,537✔
1791
    }
1792
    STUidTagInfo* tmpTag = (STUidTagInfo*)taosArrayGet(pUidTagList, i);
79,963,263✔
1793
    if (!tmpTag) {
79,944,796✔
1794
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1795
      return terrno;
×
1796
    }
1797
    uint64_t uid = tmpTag->uid;
79,944,796✔
1798
    qDebug("tagfilter get uid:%" PRId64, uid);
79,963,932✔
1799

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

1806
  return TSDB_CODE_SUCCESS;
16,006,067✔
1807
}
1808

1809
static int32_t copyExistedUids(SArray* pUidTagList, const SArray* pUidList) {
16,012,659✔
1810
  int32_t code = TSDB_CODE_SUCCESS;
16,012,659✔
1811
  int32_t numOfExisted = taosArrayGetSize(pUidList);
16,012,659✔
1812
  if (numOfExisted == 0) {
16,013,326✔
1813
    return code;
13,250,892✔
1814
  }
1815

1816
  for (int32_t i = 0; i < numOfExisted; ++i) {
31,173,955✔
1817
    uint64_t* uid = taosArrayGet(pUidList, i);
28,411,142✔
1818
    if (!uid) {
28,410,768✔
1819
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1820
      return terrno;
×
1821
    }
1822
    STUidTagInfo info = {.uid = *uid};
28,410,768✔
1823
    void*        tmp = taosArrayPush(pUidTagList, &info);
28,411,521✔
1824
    if (!tmp) {
28,411,521✔
1825
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1826
      return code;
×
1827
    }
1828
  }
1829
  return code;
2,762,813✔
1830
}
1831

1832
static EDealRes getCids(SNode** pNode, void* pContext) {
6,660✔
1833
  SHashObj* colHash = (SHashObj*)pContext;
6,660✔
1834
  col_id_t colId = 0;
6,660✔
1835
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
6,660✔
1836
    colId = (*(SColumnNode**)pNode)->colId;
2,220✔
1837
  } else {
1838
    return DEAL_RES_CONTINUE;
4,440✔
1839
  }
1840

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

1846
  return DEAL_RES_CONTINUE;
2,220✔
1847
}
1848

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

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

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

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

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

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

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

1899
int32_t doFilterByTagCond(int64_t suid, SArray* pUidList, SNode* pTagCond, void* pVnode,
279,494,213✔
1900
                                 SIdxFltStatus status, SStorageAPI* pAPI, void* pStreamInfo) {
1901
  if (pTagCond == NULL) {
279,494,213✔
1902
    return TSDB_CODE_SUCCESS;
263,479,699✔
1903
  }
1904

1905
  terrno = TSDB_CODE_SUCCESS;
16,014,514✔
1906

1907
  int32_t      lino = 0;
16,013,584✔
1908
  int32_t      code = TSDB_CODE_SUCCESS;
16,013,584✔
1909
  SArray*      pBlockList = NULL;
16,013,584✔
1910
  SSDataBlock* pResBlock = NULL;
16,013,584✔
1911
  SScalarParam output = {0};
16,013,231✔
1912
  SArray*      pUidTagList = NULL;
16,011,773✔
1913

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

1916
  pUidTagList = taosArrayInit(10, sizeof(STUidTagInfo));
16,006,056✔
1917
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
16,007,365✔
1918

1919
  // pUidList size is 0 if tagIndex is NULL and in query condition
1920
  code = copyExistedUids(pUidTagList, pUidList);
16,007,365✔
1921
  QUERY_CHECK_CODE(code, lino, end);
16,006,142✔
1922

1923
  // Narrow down the scope of the tablelist set if there is tbname in condition and And Logical operator
1924
  code = optimizeTbnameInCond(pVnode, suid, pUidTagList, pTagCond, pAPI);
16,006,142✔
1925
  if (code == 0) {
16,011,190✔
1926
    if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
395,698✔
1927
      goto end;
387,564✔
1928
    }
1929
  } else {
1930
    qDebug("pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
15,615,492✔
1931

1932
    FilterCondType condType = checkTagCond(pTagCond);
15,615,699✔
1933
    if (((condType == FILTER_NO_LOGIC || condType == FILTER_AND) && status != SFLT_NOT_INDEX) || // (super table) use tagIndex and operator is and
15,613,597✔
1934
        (status == SFLT_NOT_INDEX && taosArrayGetSize(pUidTagList) > 0)) {                       // (child table with tagCond)
15,616,635✔
1935
      code = pAPI->metaFn.getTableTagsByUid(pVnode, suid, pUidTagList);
2,759,454✔
1936
    } else {
1937
      taosArrayClearEx(pUidTagList, freeItem);       // clear tablelist if using tagIndex and or condition
12,855,341✔
1938
      code = pAPI->metaFn.getTableTags(pVnode, suid, pUidTagList);
12,854,865✔
1939
    }
1940
    if (code != TSDB_CODE_SUCCESS) {
15,612,680✔
1941
      qError("failed to get table tags from meta, reason:%s, suid:%" PRIu64, tstrerror(code),suid);
×
1942
      terrno = code;
×
1943
      QUERY_CHECK_CODE(code, lino, end);
×
1944
    }
1945
  }
1946

1947
  qDebug("final pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
15,620,814✔
1948

1949
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
15,623,684✔
1950
  if (numOfTables == 0) {
15,623,963✔
1951
    goto end;
2,911,160✔
1952
  }
1953

1954
  SArray* pColList = NULL;
12,712,803✔
1955
  code = qGetColumnsFromNodeList(pTagCond, false, &pColList); 
12,713,196✔
1956
  if (code != TSDB_CODE_SUCCESS) {
12,694,609✔
1957
    goto end;
×
1958
  }
1959
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
12,694,609✔
1960
  taosArrayDestroy(pColList);
12,708,937✔
1961
  if (pResBlock == NULL) {
12,708,421✔
1962
    code = terrno;
×
1963
    QUERY_CHECK_CODE(code, lino, end);
×
1964
  }
1965

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

1968
  //  int64_t st1 = taosGetTimestampUs();
1969
  //  qDebug("generate tag block rows:%d, cost:%ld us", rows, st1-st);
1970
  pBlockList = taosArrayInit(2, POINTER_BYTES);
12,708,421✔
1971
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
12,710,014✔
1972

1973
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
12,715,540✔
1974
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
12,715,540✔
1975

1976
  code = createResultData(&type, numOfTables, &output);
12,715,540✔
1977
  if (code != TSDB_CODE_SUCCESS) {
12,705,480✔
1978
    terrno = code;
×
1979
    QUERY_CHECK_CODE(code, lino, end);
×
1980
  }
1981

1982
  gTaskScalarExtra.pStreamInfo = pStreamInfo;
12,705,480✔
1983
  gTaskScalarExtra.pStreamRange = NULL;
12,705,480✔
1984
  code = scalarCalculate(pTagCond, pBlockList, &output, &gTaskScalarExtra);
12,693,116✔
1985
  if (code != TSDB_CODE_SUCCESS) {
12,689,926✔
1986
    qError("failed to calculate scalar, reason:%s", tstrerror(code));
996✔
1987
    terrno = code;
996✔
1988
    QUERY_CHECK_CODE(code, lino, end);
996✔
1989
  }
1990

1991
end:
15,988,650✔
1992
  if (code == 0) {
16,001,519✔
1993
    code = doSetQualifiedUid(pUidList, pUidTagList, output.columnData);
16,000,957✔
1994
    lino = __LINE__;
16,010,244✔
1995
  }
1996
  
1997
  if (code != 0) {
16,010,806✔
1998
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
996✔
1999
  }
2000
  blockDataDestroy(pResBlock);
16,010,806✔
2001
  taosArrayDestroy(pBlockList);
15,997,779✔
2002
  taosArrayDestroyEx(pUidTagList, freeItem);
15,990,271✔
2003

2004
  colDataDestroy(output.columnData);
16,005,452✔
2005
  taosMemoryFreeClear(output.columnData);
16,002,452✔
2006
  return code;
16,000,536✔
2007
}
2008

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

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

2037
  return DEAL_RES_CONTINUE;
13,708✔
2038
}
2039

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

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

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

2063
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
10,856✔
2064
    extractTagColId((SOperatorNode*)pTagCond, *pTagColIds);
×
2065
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
10,856✔
2066
    SNode* pChild = NULL;
10,856✔
2067
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
32,568✔
2068
      extractTagColId((SOperatorNode*)pChild, *pTagColIds);
21,712✔
2069
    }
2070
  }
2071

2072
  taosArraySort(*pTagColIds, compareUint16Val);
10,856✔
2073

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

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

2099
static EDealRes canOptimizeTagCondFilter(SNode* pTagCond, void* pContext) {
99,728✔
2100
  if (NULL == pTagCond) {
99,728✔
2101
    *(bool*)pContext = false;
×
2102
    return DEAL_RES_END;
×
2103
  }
2104
  if (nodeType(pTagCond) == QUERY_NODE_VALUE ||
99,728✔
2105
    nodeType(pTagCond) == QUERY_NODE_COLUMN) {
66,240✔
2106
    return DEAL_RES_CONTINUE;
55,016✔
2107
  }
2108
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR &&
44,712✔
2109
    ((SOperatorNode*)pTagCond)->opType == OP_TYPE_EQUAL) {
22,080✔
2110
    return DEAL_RES_CONTINUE;
21,528✔
2111
  }
2112
  if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION &&
23,368✔
2113
    ((SLogicConditionNode*)pTagCond)->condType == LOGIC_COND_TYPE_AND) {
10,856✔
2114
    return DEAL_RES_CONTINUE;
10,856✔
2115
  }
2116
  if (nodeType(pTagCond) == QUERY_NODE_FUNCTION &&
24,656✔
2117
    fmIsStreamPesudoColVal(((SFunctionNode*)pTagCond)->funcId)) {
11,960✔
2118
    return DEAL_RES_CONTINUE;
12,144✔
2119
  }
2120
  *(bool*)pContext = false;
552✔
2121
  return DEAL_RES_END;
552✔
2122
}
2123

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

2130
  pListInfo->idInfo.suid = pScanNode->suid;
279,422,065✔
2131
  pListInfo->idInfo.tableType = pScanNode->tableType;
279,431,347✔
2132

2133
  SArray* pUidList = taosArrayInit(8, sizeof(uint64_t));
279,217,748✔
2134
  QUERY_CHECK_NULL(pUidList, code, lino, end, terrno);
279,291,340✔
2135

2136
  char*   pTagCondKey = NULL;
279,291,340✔
2137
  int32_t tagCondKeyLen;
279,088,249✔
2138
  SArray* pTagColIds = NULL;
279,286,146✔
2139
  char*   pPayload = NULL;
279,301,628✔
2140
  SIdxFltStatus status = SFLT_NOT_INDEX;
279,301,628✔
2141

2142
  qTrace("getTableList called, suid:%" PRIu64
279,301,628✔
2143
    ", tagCond:%p, tagIndexCond:%p, %d %d", pScanNode->suid, pTagCond,
2144
    pTagIndexCond, pScanNode->tableType, pScanNode->virtualStableScan);
2145
  if (pScanNode->tableType != TSDB_SUPER_TABLE && !pScanNode->virtualStableScan) {
279,306,463✔
2146
    pListInfo->idInfo.uid = pScanNode->uid;
149,330,263✔
2147
    if (pStorageAPI->metaFn.isTableExisted(pVnode, pScanNode->uid)) {
149,313,663✔
2148
      void* tmp = taosArrayPush(pUidList, &pScanNode->uid);
149,302,806✔
2149
      QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
149,302,802✔
2150
    }
2151
    code = doFilterByTagCond(pScanNode->suid, pUidList, pTagCond, pVnode, status, pStorageAPI, pStreamInfo);
149,351,809✔
2152
    QUERY_CHECK_CODE(code, lino, end);
149,345,164✔
2153
  } else {
2154
    bool      isStream = (pStreamInfo != NULL);
130,129,032✔
2155
    bool      hasTagCond = (pTagCond != NULL);
130,129,032✔
2156
    bool      canCacheTagEqCondFilter = false;
130,129,032✔
2157
    T_MD5_CTX context = {0};
130,093,257✔
2158

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

2163
    bool acquired = false;
130,107,211✔
2164
    // first, check whether we can use stable tag filter cache
2165
    if (tsStableTagFilterCache && isStream && hasTagCond) {
130,083,562✔
2166
      canCacheTagEqCondFilter = true;
11,408✔
2167
      nodesWalkExpr(pTagCond, canOptimizeTagCondFilter,
11,408✔
2168
        (void*)&canCacheTagEqCondFilter);
2169
    }
2170
    if (canCacheTagEqCondFilter) {
129,994,641✔
2171
      qDebug("%s, stable tag filter condition can be optimized", idstr);
10,672✔
2172
      if (((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
10,856✔
2173
        SNode* tmp = NULL;
10,856✔
2174
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
10,856✔
2175
        QUERY_CHECK_CODE(code, lino, end);
10,856✔
2176

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

2191
      code = buildTagCondKey(
10,856✔
2192
        pTagCond, &pTagCondKey, &tagCondKeyLen, &pTagColIds);
2193
      QUERY_CHECK_CODE(code, lino, end);
10,856✔
2194
      code = pStorageAPI->metaFn.getStableCachedTableList(
10,856✔
2195
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
10,856✔
2196
        context.digest, tListLen(context.digest), pUidList, &acquired);
2197
      QUERY_CHECK_CODE(code, lino, end);
10,856✔
2198
    } else if (tsTagFilterCache) {
129,983,969✔
2199
      // second, try to use normal tag filter cache
2200
      qDebug("%s using normal tag filter cache", idstr);
33,948✔
2201
      if (pStreamInfo != NULL && ((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
35,328✔
2202
        SNode* tmp = NULL;
1,380✔
2203
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
1,380✔
2204
        QUERY_CHECK_CODE(code, lino, end);
1,380✔
2205

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

2242
    if (!pTagCond) {  // no tag filter condition exists, let's fetch all tables of this super table
130,142,065✔
2243
      code = pStorageAPI->metaFn.getChildTableList(pVnode, pScanNode->suid, pUidList);
114,454,034✔
2244
      QUERY_CHECK_CODE(code, lino, end);
114,405,252✔
2245
      qTrace("no tag filter, get all child tables, numOfTables:%d", (int32_t)taosArrayGetSize(pUidList));
114,405,252✔
2246
    } else {
2247
      SIdxFltStatus status = SFLT_NOT_INDEX;
15,688,031✔
2248
      if (pTagIndexCond) {
15,689,068✔
2249
        void* pIndex = pStorageAPI->metaFn.getInvertIndex(pVnode);
4,288,614✔
2250

2251
        SIndexMetaArg metaArg = {.metaEx = pVnode,
4,288,640✔
2252
                                 .idx = pStorageAPI->metaFn.storeGetIndexInfo(pVnode),
4,288,614✔
2253
                                 .ivtIdx = pIndex,
2254
                                 .suid = pScanNode->uid};
4,288,614✔
2255
        code = doFilterTag(pTagIndexCond, &metaArg, pUidList, &status, &pStorageAPI->metaFilter);
4,288,614✔
2256
        if (code != 0 || status == SFLT_NOT_INDEX) {  // temporarily disable it for performance sake
4,278,494✔
2257
          qDebug("failed to get tableIds from index, suid:%" PRIu64 ", uidListSize:%d", pScanNode->uid, (int32_t)taosArrayGetSize(pUidList));
1,239,518✔
2258
        } else {
2259
          qDebug("succ to get filter result, table num: %d", (int)taosArrayGetSize(pUidList));
3,038,976✔
2260
        }
2261
      }
2262
    }
2263
    qTrace("after index filter, pTagCond:%p uidListSize:%d", pTagCond, (int32_t)taosArrayGetSize(pUidList));
130,113,018✔
2264
    code = doFilterByTagCond(pScanNode->suid, pUidList, pTagCond, pVnode, status, pStorageAPI, pStreamInfo);
130,113,018✔
2265
    QUERY_CHECK_CODE(code, lino, end);
130,097,804✔
2266

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

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

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

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

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

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

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

2330
  qDebug("%s, table list with %d uids built", idstr, (int32_t)numOfTables);
279,408,109✔
2331

2332
end:
279,497,482✔
2333
  if (code == 0) {
279,508,932✔
2334
     STableKeyInfo info = {.uid = 0, .groupId = 0};
279,507,936✔
2335
    for (int32_t i = 0; i < taosArrayGetSize(pUidList); ++i) {
973,115,183✔
2336
      int64_t* uid = (int64_t*)taosArrayGet(pUidList, i);
693,536,670✔
2337
      QUERY_CHECK_NULL(uid, code, lino, end, terrno);
693,588,690✔
2338

2339
      info.uid = *uid;
693,588,690✔
2340
        //qInfo("doSetQualifiedUid row:%d added to pTableList", i);
2341
      void* p = taosArrayPush(pListInfo->pTableList, &info);
693,576,893✔
2342
      QUERY_CHECK_NULL(p, code, lino, end, terrno);
693,608,700✔
2343
    }
2344
  } else {
2345
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
996✔
2346
  }
2347
  taosArrayDestroy(pUidList);
279,547,240✔
2348
  taosArrayDestroy(pTagColIds);
279,446,441✔
2349
  taosMemFreeClear(pTagCondKey);
279,473,426✔
2350
  taosMemFreeClear(pPayload);
279,473,426✔
2351
  return code;
279,473,426✔
2352
}
2353

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

2475
  return TSDB_CODE_SUCCESS;
×
2476
}
2477

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

2483
  size_t  numOfCols = LIST_LENGTH(pNodeList);
3,136,697✔
2484
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
3,137,244✔
2485
  if (pList == NULL) {
3,134,369✔
2486
    return NULL;
×
2487
  }
2488

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

2497
    // todo extract method
2498
    SColumn c = {0};
4,090,664✔
2499
    c.slotId = pColNode->slotId;
4,091,298✔
2500
    c.colId = pColNode->colId;
4,091,139✔
2501
    c.type = pColNode->node.resType.type;
4,088,343✔
2502
    c.bytes = pColNode->node.resType.bytes;
4,092,112✔
2503
    c.precision = pColNode->node.resType.precision;
4,091,539✔
2504
    c.scale = pColNode->node.resType.scale;
4,090,030✔
2505

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

2514
  return pList;
3,130,027✔
2515
}
2516

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

2526
  pMatchInfo->matchType = type;
359,156,765✔
2527

2528
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchItem));
359,081,435✔
2529
  if (pList == NULL) {
359,012,383✔
2530
    code = terrno;
×
2531
    return code;
×
2532
  }
2533

2534
  SColMatchItem** infoBySlot = NULL;
359,012,383✔
2535
  if (numOfCols > 0) {
359,012,383✔
2536
    infoBySlot = taosMemoryCalloc(numOfCols, sizeof(*infoBySlot));
358,910,983✔
2537
    if (infoBySlot == NULL) {
358,952,492✔
2538
      code = terrno;
×
2539
      goto _end;
×
2540
    }
2541
  }
2542

2543
  SNode* node = NULL;
359,053,892✔
2544
  FOREACH(node, pNodeList) {
1,752,972,629✔
2545
    STargetNode* pNode = (STargetNode*)node;
1,393,744,198✔
2546
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,393,744,198✔
2547
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
1,393,744,198✔
2548
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
1,382,868,916✔
2549

2550
      SColMatchItem c = {.needOutput = true};
1,382,905,194✔
2551
      c.colId = pColNode->colId;
1,382,891,064✔
2552
      c.srcSlotId = pColNode->slotId;
1,383,029,741✔
2553
      c.dstSlotId = pNode->slotId;
1,383,041,668✔
2554
      c.isPk = pColNode->isPk;
1,383,048,241✔
2555
      c.dataType = pColNode->node.resType;
1,383,001,671✔
2556
      void* tmp = taosArrayPush(pList, &c);
1,382,767,187✔
2557
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,382,767,187✔
2558
      if (hasPrevColId) {
1,382,767,187✔
2559
        if (c.colId < prevColId) {
1,024,329,104✔
2560
          colIdOrdered = false;
15,705,950✔
2561
        }
2562
      } else {
2563
        hasPrevColId = true;
358,438,083✔
2564
      }
2565
      prevColId = c.colId;
1,382,767,187✔
2566
      if (pNode->slotId >= 0 && pNode->slotId < numOfCols) {
1,382,767,187✔
2567
        infoBySlot[pNode->slotId] = (SColMatchItem*)taosArrayGet(pList, taosArrayGetSize(pList) - 1);
1,382,551,063✔
2568
      }
2569
    }
2570
  }
2571
  // set the output flag for each column in SColMatchInfo, according to the
2572
  *numOfOutputCols = 0;
359,150,666✔
2573
  SNode* slotNode = NULL;
359,153,511✔
2574
  FOREACH(slotNode, pOutputNodeList->pSlots) {
1,930,560,543✔
2575
    SSlotDescNode* pNode = (SSlotDescNode*)slotNode;
1,571,322,174✔
2576
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,571,322,174✔
2577

2578
    // todo: add reserve flag check
2579
    // it is a column reserved for the arithmetic expression calculation
2580
    if (pNode->slotId >= numOfCols) {
1,571,322,174✔
2581
      (*numOfOutputCols) += 1;
177,948,368✔
2582
      continue;
177,958,429✔
2583
    }
2584

2585
    SColMatchItem* info = infoBySlot ? infoBySlot[pNode->slotId] : NULL;
1,393,617,676✔
2586

2587
    if (pNode->output) {
1,393,595,900✔
2588
      (*numOfOutputCols) += 1;
1,372,226,977✔
2589
    } else if (info != NULL) {
21,422,874✔
2590
      // select distinct tbname from stb where tbname='abc';
2591
      info->needOutput = false;
21,397,081✔
2592
    }
2593
  }
2594

2595
  pMatchInfo->pList = pList;
359,051,329✔
2596
  pMatchInfo->colIdOrdered = colIdOrdered;
359,054,960✔
2597

2598
_end:
358,945,090✔
2599
  taosMemoryFree(infoBySlot);
358,945,090✔
2600
  if (code != TSDB_CODE_SUCCESS) {
359,023,297✔
2601
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2602
  }
2603
  return code;
359,012,387✔
2604
}
2605

2606
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
1,190,668,997✔
2607
                                  const char* name) {
2608
  SResSchema s = {0};
1,190,668,997✔
2609
  s.scale = scale;
1,190,735,632✔
2610
  s.type = type;
1,190,735,632✔
2611
  s.bytes = bytes;
1,190,735,632✔
2612
  s.slotId = slotId;
1,190,735,632✔
2613
  s.precision = precision;
1,190,735,632✔
2614
  tstrncpy(s.name, name, tListLen(s.name));
1,190,735,632✔
2615

2616
  return s;
1,190,543,886✔
2617
}
2618

2619
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType, EColumnType colType) {
1,112,033,390✔
2620
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
1,112,033,390✔
2621
  if (pCol == NULL) {
1,111,631,050✔
2622
    return NULL;
×
2623
  }
2624

2625
  pCol->slotId = slotId;
1,111,631,050✔
2626
  pCol->colId = colId;
1,111,658,091✔
2627
  pCol->bytes = pType->bytes;
1,111,684,519✔
2628
  pCol->type = pType->type;
1,111,859,548✔
2629
  pCol->scale = pType->scale;
1,111,883,077✔
2630
  pCol->precision = pType->precision;
1,112,052,312✔
2631
  pCol->dataBlockId = blockId;
1,112,020,621✔
2632
  pCol->colType = colType;
1,111,955,627✔
2633
  return pCol;
1,112,093,169✔
2634
}
2635

2636
int32_t createExprFromOneNode(SExprInfo* pExp, SNode* pNode, int16_t slotId) {
1,191,076,386✔
2637
  int32_t code = TSDB_CODE_SUCCESS;
1,191,076,386✔
2638
  int32_t lino = 0;
1,191,076,386✔
2639
  pExp->base.numOfParams = 0;
1,191,076,386✔
2640
  pExp->base.pParam = NULL;
1,191,160,655✔
2641
  pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
1,191,014,198✔
2642
  QUERY_CHECK_NULL(pExp->pExpr, code, lino, _end, terrno);
1,190,575,588✔
2643

2644
  pExp->pExpr->_function.num = 1;
1,190,644,711✔
2645
  pExp->pExpr->_function.functionId = -1;
1,190,763,743✔
2646

2647
  int32_t type = nodeType(pNode);
1,190,882,396✔
2648
  // it is a project query, or group by column
2649
  if (type == QUERY_NODE_COLUMN) {
1,190,998,135✔
2650
    pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
693,160,361✔
2651
    SColumnNode* pColNode = (SColumnNode*)pNode;
693,188,416✔
2652

2653
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
693,188,416✔
2654
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
693,138,795✔
2655

2656
    pExp->base.numOfParams = 1;
693,041,233✔
2657

2658
    SDataType* pType = &pColNode->node.resType;
693,184,958✔
2659
    pExp->base.resSchema =
2660
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pColNode->colName);
693,161,909✔
2661

2662
    pExp->base.pParam[0].pCol =
1,386,519,371✔
2663
        createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType, pColNode->colType);
1,385,162,388✔
2664
    QUERY_CHECK_NULL(pExp->base.pParam[0].pCol, code, lino, _end, terrno);
693,304,703✔
2665

2666
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
693,222,838✔
2667
  } else if (type == QUERY_NODE_VALUE) {
497,837,774✔
2668
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
18,556,530✔
2669
    SValueNode* pValNode = (SValueNode*)pNode;
18,557,122✔
2670

2671
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
18,557,122✔
2672
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
18,550,579✔
2673

2674
    pExp->base.numOfParams = 1;
18,548,358✔
2675

2676
    SDataType* pType = &pValNode->node.resType;
18,557,091✔
2677
    pExp->base.resSchema =
2678
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
18,554,137✔
2679
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
18,545,085✔
2680
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
18,556,561✔
2681
    QUERY_CHECK_CODE(code, lino, _end);
18,551,567✔
2682
  } else if (type == QUERY_NODE_REMOTE_VALUE) {
479,281,244✔
2683
    SRemoteValueNode* pRemote = (SRemoteValueNode*)pNode;
1,462,948✔
2684
    code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, pNode);
1,462,948✔
2685
    QUERY_CHECK_CODE(code, lino, _end);
1,464,628✔
2686

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

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

2693
    pExp->base.numOfParams = 1;
1,031,188✔
2694

2695
    SDataType* pType = &pValNode->node.resType;
1,031,188✔
2696
    pExp->base.resSchema =
2697
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
1,031,188✔
2698
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
1,031,188✔
2699
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
1,031,188✔
2700
    QUERY_CHECK_CODE(code, lino, _end);
1,031,218✔
2701
  } else if (type == QUERY_NODE_FUNCTION) {
477,818,296✔
2702
    pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
436,410,818✔
2703
    SFunctionNode* pFuncNode = (SFunctionNode*)pNode;
436,415,649✔
2704

2705
    SDataType* pType = &pFuncNode->node.resType;
436,415,649✔
2706
    pExp->base.resSchema =
2707
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pFuncNode->node.aliasName);
436,424,722✔
2708
    tExprNode* pExprNode = pExp->pExpr;
436,378,330✔
2709

2710
    pExprNode->_function.functionId = pFuncNode->funcId;
436,411,515✔
2711
    pExprNode->_function.pFunctNode = pFuncNode;
436,427,568✔
2712
    pExprNode->_function.functionType = pFuncNode->funcType;
436,420,194✔
2713

2714
    tstrncpy(pExprNode->_function.functionName, pFuncNode->functionName, tListLen(pExprNode->_function.functionName));
436,462,128✔
2715

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

2722
    if (!pFuncNode->pParameterList && (memcmp(pExprNode->_function.functionName, name, len) == 0) &&
436,430,440✔
2723
        pExprNode->_function.functionName[len] == 0) {
29,432,763✔
2724
      pFuncNode->pParameterList = NULL;
29,431,193✔
2725
      int32_t     code = nodesMakeList(&pFuncNode->pParameterList);
29,435,850✔
2726
      SValueNode* res = NULL;
29,436,370✔
2727
      if (TSDB_CODE_SUCCESS == code) {
29,440,813✔
2728
        code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
29,444,042✔
2729
      }
2730
      QUERY_CHECK_CODE(code, lino, _end);
29,447,555✔
2731
      res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
29,447,555✔
2732
      code = nodesListAppend(pFuncNode->pParameterList, (SNode*)res);
29,443,692✔
2733
      if (code != TSDB_CODE_SUCCESS) {
29,440,894✔
2734
        nodesDestroyNode((SNode*)res);
×
2735
        res = NULL;
×
2736
      }
2737
      QUERY_CHECK_CODE(code, lino, _end);
29,440,894✔
2738
    }
2739
#endif
2740

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

2743
    pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
436,430,345✔
2744
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
436,355,935✔
2745
    pExp->base.numOfParams = numOfParam;
436,374,778✔
2746

2747
    for (int32_t j = 0; j < numOfParam && TSDB_CODE_SUCCESS == code; ++j) {
1,025,449,120✔
2748
      SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
589,099,001✔
2749
      QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
589,124,000✔
2750
      if (p1->type == QUERY_NODE_COLUMN) {
589,124,000✔
2751
        SColumnNode* pcn = (SColumnNode*)p1;
418,820,424✔
2752

2753
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
418,820,424✔
2754
        pExp->base.pParam[j].pCol =
837,650,761✔
2755
            createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType, pcn->colType);
837,587,890✔
2756
        QUERY_CHECK_NULL(pExp->base.pParam[j].pCol, code, lino, _end, terrno);
418,853,127✔
2757
      } else if (p1->type == QUERY_NODE_VALUE) {
170,289,173✔
2758
        SValueNode* pvn = (SValueNode*)p1;
112,384,493✔
2759
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
112,384,493✔
2760
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
112,340,446✔
2761
        QUERY_CHECK_CODE(code, lino, _end);
112,369,093✔
2762
      } else if (p1->type == QUERY_NODE_REMOTE_VALUE) {
57,955,635✔
2763
        SRemoteValueNode* pRemote = (SRemoteValueNode*)p1;
74,480✔
2764
        code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, p1);
74,480✔
2765
        QUERY_CHECK_CODE(code, lino, _end);
74,480✔
2766

2767
        SValueNode* pvn = (SValueNode*)pRemote;
54,320✔
2768
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
54,320✔
2769
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
54,320✔
2770
        QUERY_CHECK_CODE(code, lino, _end);
694✔
2771
      }
2772
    }
2773
    pExp->pExpr->_function.bindExprID = ((SExprNode*)pNode)->bindExprID;
436,350,119✔
2774
  } else if (type == QUERY_NODE_OPERATOR) {
41,407,478✔
2775
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
39,924,457✔
2776
    SOperatorNode* pOpNode = (SOperatorNode*)pNode;
39,922,800✔
2777

2778
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
39,922,800✔
2779
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
39,915,738✔
2780
    pExp->base.numOfParams = 1;
39,907,548✔
2781

2782
    SDataType* pType = &pOpNode->node.resType;
39,921,203✔
2783
    pExp->base.resSchema =
2784
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pOpNode->node.aliasName);
39,918,488✔
2785
    pExp->pExpr->_optrRoot.pRootNode = pNode;
39,915,915✔
2786
  } else if (type == QUERY_NODE_CASE_WHEN) {
1,483,047✔
2787
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
1,368,239✔
2788
    SCaseWhenNode* pCaseNode = (SCaseWhenNode*)pNode;
1,368,239✔
2789

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

2794
    SDataType* pType = &pCaseNode->node.resType;
1,368,239✔
2795
    pExp->base.resSchema =
2796
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCaseNode->node.aliasName);
1,368,239✔
2797
    pExp->pExpr->_optrRoot.pRootNode = pNode;
1,368,239✔
2798
  } else if (type == QUERY_NODE_LOGIC_CONDITION) {
114,808✔
2799
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
106,623✔
2800
    SLogicConditionNode* pCond = (SLogicConditionNode*)pNode;
105,529✔
2801
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
105,529✔
2802
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
106,623✔
2803
    pExp->base.numOfParams = 1;
106,076✔
2804
    SDataType* pType = &pCond->node.resType;
106,623✔
2805
    pExp->base.resSchema =
2806
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCond->node.aliasName);
106,623✔
2807
    pExp->pExpr->_optrRoot.pRootNode = pNode;
106,076✔
2808
  } else {
2809
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
8,239✔
2810
    QUERY_CHECK_CODE(code, lino, _end);
8,239✔
2811
  }
2812
  pExp->pExpr->relatedTo = ((SExprNode*)pNode)->relatedTo;
1,190,554,610✔
2813
_end:
1,191,101,290✔
2814
  if (code != TSDB_CODE_SUCCESS) {
1,191,101,290✔
2815
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
453,600✔
2816
  }
2817
  return code;
1,191,002,363✔
2818
}
2819

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

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

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

2842
  return pExprs;
×
2843
}
2844

2845
int32_t createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, SExprInfo** pExprInfo, int32_t* numOfExprs) {
458,847,174✔
2846
  QRY_PARAM_CHECK(pExprInfo);
458,847,174✔
2847

2848
  int32_t code = 0;
458,894,975✔
2849
  int32_t lino = 0;
458,894,975✔
2850
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
458,894,975✔
2851
  int32_t numOfGroupKeys = 0;
458,819,592✔
2852
  if (pGroupKeys != NULL) {
458,819,592✔
2853
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
31,466,833✔
2854
  }
2855

2856
  *numOfExprs = numOfFuncs + numOfGroupKeys;
458,821,161✔
2857
  if (*numOfExprs == 0) {
458,854,809✔
2858
    return code;
40,480,432✔
2859
  }
2860

2861
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
418,353,179✔
2862
  QUERY_CHECK_NULL(pExprs, code, lino, _return, terrno);
418,118,231✔
2863

2864
  int32_t i = 0;
418,118,231✔
2865
  if (pNodeList != NULL) {
418,118,231✔
2866
    SNode* node = NULL;
390,570,891✔
2867
    FOREACH(node, pNodeList) {
1,532,219,326✔
2868
      STargetNode* pTargetNode = (STargetNode*)node;
1,142,189,918✔
2869
      SExprInfo*   pExp = &pExprs[i++];
1,142,189,918✔
2870
      code = createExprFromTargetNode(pExp, pTargetNode);
1,142,119,084✔
2871
      QUERY_CHECK_CODE(code, lino, _return);
1,142,102,035✔
2872
    }
2873
  }
2874

2875
  if (pGroupKeys != NULL) {
417,676,157✔
2876
    SNode* node = NULL;
31,458,302✔
2877
    FOREACH(node, pGroupKeys) {
80,158,465✔
2878
      STargetNode* pTargetNode = (STargetNode*)node;
48,708,583✔
2879
      SExprInfo*   pExp = &pExprs[i++];
48,708,583✔
2880
      code = createExprFromTargetNode(pExp, pTargetNode);
48,706,211✔
2881
      QUERY_CHECK_CODE(code, lino, _return);
48,700,163✔
2882
    }
2883
  }
2884

2885
  *pExprInfo = pExprs;
417,675,026✔
2886
  return code;
417,935,975✔
2887

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

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

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

2908
  SArray* pValCtxArray = NULL;
446,708,550✔
2909
  for (int32_t i = numOfOutput - 1; i > 0; --i) {  // select Func is at the end of the list
1,211,854,983✔
2910
    int32_t funcIdx = pCtx[i].pExpr->pExpr->_function.bindExprID;
765,238,858✔
2911
    if (funcIdx > 0) {
765,226,598✔
2912
      if (pValCtxArray == NULL) {
1,624,260✔
2913
        // the end of the list is the select function of biggest index
2914
        pValCtxArray = taosArrayInit_s(sizeof(SSubsidiaryResInfo*), funcIdx);
1,165,382✔
2915
        if (pValCtxArray == NULL) {
1,165,382✔
2916
          return terrno;
×
2917
        }
2918
      }
2919
      if (funcIdx > pValCtxArray->size) {
1,624,260✔
2920
        qError("funcIdx:%d is out of range", funcIdx);
×
2921
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2922
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
2923
      }
2924
      SSubsidiaryResInfo* pSubsidiary = &pCtx[i].subsidiaries;
1,623,818✔
2925
      pSubsidiary->pCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
1,623,818✔
2926
      if (pSubsidiary->pCtx == NULL) {
1,624,702✔
2927
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2928
        return terrno;
×
2929
      }
2930
      pSubsidiary->num = 0;
1,624,702✔
2931
      taosArraySet(pValCtxArray, funcIdx - 1, &pSubsidiary);
1,624,702✔
2932
    }
2933
  }
2934

2935
  SqlFunctionCtx*  p = NULL;
446,616,125✔
2936
  SqlFunctionCtx** pValCtx = NULL;
446,616,125✔
2937
  if (pValCtxArray == NULL) {
446,616,125✔
2938
    pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
445,474,598✔
2939
    if (pValCtx == NULL) {
445,465,708✔
2940
      QUERY_CHECK_CODE(terrno, lino, _end);
×
2941
    }
2942
  }
2943

2944
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,625,203,678✔
2945
    const char* pName = pCtx[i].pExpr->pExpr->_function.functionName;
1,178,572,485✔
2946
    if ((strcmp(pName, "_select_value") == 0)) {
1,178,691,336✔
2947
      if (pValCtxArray == NULL) {
9,007,103✔
2948
        pValCtx[num++] = &pCtx[i];
6,725,233✔
2949
      } else {
2950
        int32_t bindFuncIndex = pCtx[i].pExpr->pExpr->relatedTo;  // start from index 1;
2,281,897✔
2951
        if (bindFuncIndex > 0) {                                  // 0 is default index related to the select function
2,282,979✔
2952
          bindFuncIndex -= 1;
2,230,381✔
2953
        }
2954
        SSubsidiaryResInfo** pSubsidiary = taosArrayGet(pValCtxArray, bindFuncIndex);
2,282,979✔
2955
        if (pSubsidiary == NULL) {
2,282,537✔
2956
          QUERY_CHECK_CODE(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR, lino, _end);
×
2957
        }
2958
        (*pSubsidiary)->pCtx[(*pSubsidiary)->num] = &pCtx[i];
2,282,537✔
2959
        (*pSubsidiary)->num++;
2,282,537✔
2960
      }
2961
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
1,169,684,233✔
2962
      if (pValCtxArray == NULL) {
97,290,024✔
2963
        p = &pCtx[i];
95,308,939✔
2964
      }
2965
    }
2966
  }
2967

2968
  if (p != NULL) {
446,631,193✔
2969
    p->subsidiaries.pCtx = pValCtx;
49,948,937✔
2970
    p->subsidiaries.num = num;
49,942,960✔
2971
  } else {
2972
    taosMemoryFreeClear(pValCtx);
396,682,256✔
2973
  }
2974

2975
_end:
1,164,997✔
2976
  if (code != TSDB_CODE_SUCCESS) {
446,579,022✔
2977
    taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2978
    taosMemoryFreeClear(pValCtx);
×
2979
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2980
  } else {
2981
    taosArrayDestroy(pValCtxArray);
446,579,022✔
2982
  }
2983
  return code;
446,628,903✔
2984
}
2985

2986
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowEntryInfoOffset,
446,693,594✔
2987
                                     SFunctionStateStore* pStore) {
2988
  int32_t         code = TSDB_CODE_SUCCESS;
446,693,594✔
2989
  int32_t         lino = 0;
446,693,594✔
2990
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
446,693,594✔
2991
  if (pFuncCtx == NULL) {
446,463,148✔
2992
    return NULL;
×
2993
  }
2994

2995
  *rowEntryInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
446,463,148✔
2996
  if (*rowEntryInfoOffset == 0) {
446,672,626✔
2997
    taosMemoryFreeClear(pFuncCtx);
×
2998
    return NULL;
×
2999
  }
3000

3001
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,625,484,827✔
3002
    SExprInfo* pExpr = &pExprInfo[i];
1,178,830,358✔
3003

3004
    SExprBasicInfo* pFunct = &pExpr->base;
1,178,705,196✔
3005
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
1,178,762,234✔
3006

3007
    pCtx->functionId = -1;
1,178,756,552✔
3008
    pCtx->pExpr = pExpr;
1,178,789,619✔
3009

3010
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
1,178,716,032✔
3011
      SFuncExecEnv env = {0};
435,074,838✔
3012
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;
435,095,686✔
3013
      pCtx->isPseudoFunc = fmIsWindowPseudoColumnFunc(pCtx->functionId) || fmIsPlaceHolderFunc(pCtx->functionId);
435,094,985✔
3014
      pCtx->isNotNullFunc = fmIsNotNullOutputFunc(pCtx->functionId);
435,099,253✔
3015

3016
      bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
435,075,292✔
3017
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId) ||
581,993,422✔
3018
          fmIsProcessByRowFunc(pCtx->functionId)) {
435,053,268✔
3019
        if (!isUdaf) {
288,037,197✔
3020
          code = fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
288,000,234✔
3021
          QUERY_CHECK_CODE(code, lino, _end);
287,989,088✔
3022
        } else {
3023
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
36,963✔
3024
          pCtx->udfName = taosStrdup(udfName);
36,963✔
3025
          QUERY_CHECK_NULL(pCtx->udfName, code, lino, _end, terrno);
36,963✔
3026

3027
          code = fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
36,963✔
3028
          QUERY_CHECK_CODE(code, lino, _end);
36,963✔
3029
        }
3030
        bool tmp = pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
288,026,051✔
3031
        if (!tmp) {
287,998,798✔
3032
          code = terrno;
×
3033
          QUERY_CHECK_CODE(code, lino, _end);
1,894✔
3034
        }
3035
      } else {
3036
        if (fmIsPlaceHolderFunc(pCtx->functionId)) {
147,003,658✔
3037
          code = fmGetStreamPesudoFuncEnv(pCtx->functionId, pExpr->base.pParamList, &env);
7,312,002✔
3038
          QUERY_CHECK_CODE(code, lino, _end);
7,313,265✔
3039
        }      
3040
        
3041
        code = fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
147,039,151✔
3042
        if (code != TSDB_CODE_SUCCESS && isUdaf) {
147,036,986✔
3043
          code = TSDB_CODE_SUCCESS;
62,748✔
3044
        }
3045
        QUERY_CHECK_CODE(code, lino, _end);
147,036,986✔
3046

3047
        if (pCtx->sfp.getEnv != NULL) {
147,036,986✔
3048
          bool tmp = pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
29,176,548✔
3049
          if (!tmp) {
29,176,818✔
3050
            code = terrno;
×
3051
            QUERY_CHECK_CODE(code, lino, _end);
×
3052
          }
3053
        }
3054
      }
3055
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
435,047,892✔
3056
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
743,626,728✔
3057
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
19,563,573✔
3058
      // for simple column, the result buffer needs to hold at least one element.
3059
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
743,768,901✔
3060
    }
3061

3062
    pCtx->input.numOfInputCols = pFunct->numOfParams;
1,178,881,502✔
3063
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
1,178,818,738✔
3064
    QUERY_CHECK_NULL(pCtx->input.pData, code, lino, _end, terrno);
1,178,741,735✔
3065
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
1,178,811,706✔
3066
    QUERY_CHECK_NULL(pCtx->input.pColumnDataAgg, code, lino, _end, terrno);
1,178,752,033✔
3067

3068
    pCtx->pTsOutput = NULL;
1,178,834,970✔
3069
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
1,178,827,535✔
3070
    pCtx->resDataInfo.type = pFunct->resSchema.type;
1,178,858,475✔
3071
    pCtx->order = TSDB_ORDER_ASC;
1,178,772,868✔
3072
    pCtx->start.key = INT64_MIN;
1,178,689,217✔
3073
    pCtx->end.key = INT64_MIN;
1,178,850,213✔
3074
    pCtx->numOfParams = pExpr->base.numOfParams;
1,178,881,884✔
3075
    pCtx->param = pFunct->pParam;
1,178,806,925✔
3076
    pCtx->saveHandle.currentPage = -1;
1,179,017,350✔
3077
    pCtx->pStore = pStore;
1,178,942,664✔
3078
    pCtx->hasWindowOrGroup = false;
1,178,841,044✔
3079
    pCtx->needCleanup = false;
1,178,630,603✔
3080
    pCtx->skipDynDataCheck = false;
1,178,804,597✔
3081
  }
3082

3083
  for (int32_t i = 1; i < numOfOutput; ++i) {
1,212,060,767✔
3084
    (*rowEntryInfoOffset)[i] = (int32_t)((*rowEntryInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) +
1,529,819,682✔
3085
                                         pFuncCtx[i - 1].resDataInfo.interBufSize);
765,319,930✔
3086
  }
3087

3088
  code = setSelectValueColumnInfo(pFuncCtx, numOfOutput);
446,709,096✔
3089
  QUERY_CHECK_CODE(code, lino, _end);
446,624,422✔
3090

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

3101
    terrno = code;
×
3102
    return NULL;
×
3103
  }
3104
  return pFuncCtx;
446,565,248✔
3105
}
3106

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

3113
  int32_t i = 0, j = 0;
15,602,248✔
3114
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
125,230,024✔
3115
    SColumnInfoData* p = taosArrayGet(pCols, i);
109,632,507✔
3116
    if (!p) {
109,627,834✔
3117
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3118
      return terrno;
×
3119
    }
3120
    SColMatchItem* pmInfo = taosArrayGet(pColMatchInfo, j);
109,627,834✔
3121
    if (!pmInfo) {
109,627,395✔
3122
      return terrno;
×
3123
    }
3124

3125
    if (p->info.colId == pmInfo->colId) {
109,627,395✔
3126
      SColumnInfoData* pDst = taosArrayGet(pBlock->pDataBlock, pmInfo->dstSlotId);
102,371,891✔
3127
      if (!pDst) {
102,370,338✔
3128
        return terrno;
×
3129
      }
3130
      code = colDataAssign(pDst, p, pBlock->info.rows, &pBlock->info);
102,370,338✔
3131
      if (code != TSDB_CODE_SUCCESS) {
102,370,350✔
3132
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3133
        return code;
×
3134
      }
3135
      i++;
102,370,350✔
3136
      j++;
102,370,350✔
3137
    } else if (p->info.colId < pmInfo->colId) {
7,257,426✔
3138
      i++;
7,257,426✔
3139
    } else {
3140
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3141
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3142
    }
3143
  }
3144
  return code;
15,603,103✔
3145
}
3146

3147
SInterval extractIntervalInfo(const STableScanPhysiNode* pTableScanNode) {
247,173,343✔
3148
  SInterval interval = {
494,159,868✔
3149
      .interval = pTableScanNode->interval,
247,012,565✔
3150
      .sliding = pTableScanNode->sliding,
247,014,180✔
3151
      .intervalUnit = pTableScanNode->intervalUnit,
247,189,880✔
3152
      .slidingUnit = pTableScanNode->slidingUnit,
247,108,264✔
3153
      .offset = pTableScanNode->offset,
247,205,125✔
3154
      .precision = pTableScanNode->scan.node.pOutputDataBlockDesc->precision,
247,195,321✔
3155
      .timeRange = pTableScanNode->scanRange,
3156
  };
3157
  calcIntervalAutoOffset(&interval);
246,983,106✔
3158

3159
  return interval;
247,096,385✔
3160
}
3161

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

3165
  c.slotId = pColNode->slotId;
53,457,496✔
3166
  c.colId = pColNode->colId;
53,454,192✔
3167
  c.type = pColNode->node.resType.type;
53,459,979✔
3168
  c.bytes = pColNode->node.resType.bytes;
53,456,349✔
3169
  c.scale = pColNode->node.resType.scale;
53,457,636✔
3170
  c.precision = pColNode->node.resType.precision;
53,460,101✔
3171
  return c;
53,454,481✔
3172
}
3173

3174

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

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

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

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

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

3213
  return code;
2,457,108✔
3214
}
3215

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

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

3234
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
278,791,970✔
3235
  if (!pCond->colList) {
278,703,973✔
3236
    return terrno;
×
3237
  }
3238
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
278,596,398✔
3239
  if (pCond->pSlotList == NULL) {
278,721,689✔
3240
    taosMemoryFreeClear(pCond->colList);
×
3241
    return terrno;
×
3242
  }
3243

3244
  // TODO: get it from stable scan node
3245
  pCond->twindows = pTableScanNode->scanRange;
278,487,703✔
3246
  pCond->suid = pTableScanNode->scan.suid;
278,762,952✔
3247
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
278,584,808✔
3248
  pCond->startVersion = -1;
278,738,514✔
3249
  pCond->endVersion = -1;
278,763,048✔
3250
  pCond->skipRollup = readHandle->skipRollup;
278,619,062✔
3251
  if (readHandle->winRangeValid) {
278,723,199✔
3252
    pCond->twindows = readHandle->winRange;
647,085✔
3253
  }
3254
  pCond->cacheSttStatis = readHandle->cacheSttStatis;
278,784,139✔
3255
  // allowed read stt file optimization mode
3256
  pCond->notLoadData = (pTableScanNode->dataRequired == FUNC_DATA_REQUIRED_NOT_LOAD) &&
557,513,996✔
3257
                       (pTableScanNode->scan.node.pConditions == NULL) && (pTableScanNode->interval == 0);
278,741,692✔
3258

3259
  int32_t j = 0;
278,768,522✔
3260
  SNode*  node = NULL;
278,768,522✔
3261
  FOREACH(node, pTableScanNode->scan.pScanCols) {
1,305,435,430✔
3262
    STargetNode* pNode = (STargetNode*)node;
1,026,326,687✔
3263
    if (!pNode) {
1,026,326,687✔
3264
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3265
      return terrno;
×
3266
    }
3267
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
1,026,326,687✔
3268
    if (pColNode->colType == COLUMN_TYPE_TAG) {
1,026,637,840✔
3269
      continue;
×
3270
    }
3271

3272
    pCond->colList[j].type = pColNode->node.resType.type;
1,026,605,977✔
3273
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
1,026,576,998✔
3274
    pCond->colList[j].colId = pColNode->colId;
1,026,638,279✔
3275
    pCond->colList[j].pk = pColNode->isPk;
1,026,749,495✔
3276

3277
    pCond->pSlotList[j] = pNode->slotId;
1,026,512,697✔
3278
    j += 1;
1,026,666,908✔
3279
  }
3280

3281
  pCond->numOfCols = j;
278,731,532✔
3282

3283
  if (applyExtWin) {
278,736,131✔
3284
    if (NULL != pTableScanNode->pExtScanRange) {
247,982,478✔
3285
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
3,246,969✔
3286
      code = getQueryExtWindow(&pCond->twindows, pTableScanNode->pExtScanRange, &pCond->twindows, pCond->extTwindows);
3,246,969✔
3287
    } else if (readHandle->extWinRangeValid) {
244,725,272✔
3288
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
72,065✔
3289
      code = getQueryExtWindow(&pCond->twindows, &readHandle->extWinRange, &pCond->twindows, pCond->extTwindows);
72,065✔
3290
    }
3291
  }
3292

3293
  if (pTableScanNode->pPrimaryCond) {
278,715,994✔
3294
    bool isStrict = false;
639,600✔
3295
    code = getPrimaryTimeRange((SNode**)&pTableScanNode->pPrimaryCond, &pCond->twindows, &isStrict);
639,600✔
3296
    if (code || !isStrict) {
639,600✔
3297
      code = nodesMergeNode((SNode**)&pTableScanNode->scan.node.pConditions, &pTableScanNode->pPrimaryCond);
627,472✔
3298
    }
3299
  }
3300

3301
  return code;
278,725,632✔
3302
}
3303

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

3309
  pCond->order = TSDB_ORDER_ASC;
116,855,549✔
3310
  pCond->numOfCols = (int32_t)taosArrayGetSize(colArray);
116,884,783✔
3311

3312
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
116,889,906✔
3313
  QUERY_CHECK_NULL(pCond->colList, code, lino, _return, terrno);
116,858,203✔
3314

3315
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
116,870,634✔
3316
  QUERY_CHECK_NULL(pCond->pSlotList, code, lino, _return, terrno);
116,877,493✔
3317

3318
  pCond->twindows = pOrgCond->twindows;
116,891,930✔
3319
  pCond->order = pOrgCond->order;
116,892,699✔
3320
  pCond->type = pOrgCond->type;
116,901,258✔
3321
  pCond->startVersion = -1;
116,903,320✔
3322
  pCond->endVersion = -1;
116,907,871✔
3323
  pCond->skipRollup = true;
116,900,331✔
3324
  pCond->notLoadData = false;
116,878,421✔
3325

3326
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
475,014,538✔
3327
    SColIdPair* pColPair = taosArrayGet(colArray, i);
358,123,549✔
3328
    QUERY_CHECK_NULL(pColPair, code, lino, _return, terrno);
358,115,891✔
3329

3330
    bool find = false;
358,113,927✔
3331
    for (int32_t j = 0; j < pOrgCond->numOfCols; ++j) {
1,822,397,044✔
3332
      if (pOrgCond->colList[j].colId == pColPair->vtbColId) {
1,822,213,034✔
3333
        pCond->colList[i].type = pOrgCond->colList[j].type;
358,121,588✔
3334
        // For variable-length types (nchar/binary/varchar/varbinary), use the source table's bytes
3335
        // to avoid tsdb reader doCopyColVal length check failure when source data is longer
3336
        // than the virtual table's defined length.
3337
        if (IS_VAR_DATA_TYPE(pOrgCond->colList[j].type) && pColPair->type.bytes > 0) {
358,146,044✔
3338
          pCond->colList[i].bytes = TMAX(pOrgCond->colList[j].bytes, pColPair->type.bytes);
41,435,824✔
3339
        } else {
3340
          pCond->colList[i].bytes = pOrgCond->colList[j].bytes;
316,710,950✔
3341
        }
3342
        pCond->colList[i].colId = pColPair->orgColId;
358,159,809✔
3343
        pCond->colList[i].pk = pOrgCond->colList[j].pk;
358,153,071✔
3344
        pCond->pSlotList[i] = i;
358,143,145✔
3345
        find = true;
358,146,714✔
3346
        qDebug("%s mapped vtb colId:%d to org colId:%d, type:%d, bytes:%d", __func__, pColPair->vtbColId,
358,146,714✔
3347
               pColPair->orgColId, pCond->colList[i].type, pCond->colList[i].bytes);
3348
        break;
358,094,511✔
3349
      }
3350
    }
3351
    QUERY_CHECK_CONDITION(find, code, lino, _return, TSDB_CODE_NOT_FOUND);
358,113,413✔
3352
  }
3353

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

3362
void cleanupQueryTableDataCond(SQueryTableDataCond* pCond) {
696,240,951✔
3363
  taosMemoryFreeClear(pCond->colList);
696,240,951✔
3364
  taosMemoryFreeClear(pCond->pSlotList);
696,272,830✔
3365
}
696,216,435✔
3366

3367
int32_t convertFillType(int32_t mode) {
3,954,449✔
3368
  int32_t type = TSDB_FILL_NONE;
3,954,449✔
3369
  switch (mode) {
3,954,449✔
3370
    case FILL_MODE_PREV:
195,169✔
3371
      type = TSDB_FILL_PREV;
195,169✔
3372
      break;
195,169✔
3373
    case FILL_MODE_NONE:
×
3374
      type = TSDB_FILL_NONE;
×
3375
      break;
×
3376
    case FILL_MODE_NULL:
190,593✔
3377
      type = TSDB_FILL_NULL;
190,593✔
3378
      break;
190,593✔
3379
    case FILL_MODE_NULL_F:
37,449✔
3380
      type = TSDB_FILL_NULL_F;
37,449✔
3381
      break;
37,449✔
3382
    case FILL_MODE_NEXT:
188,982✔
3383
      type = TSDB_FILL_NEXT;
188,982✔
3384
      break;
188,982✔
3385
    case FILL_MODE_VALUE:
169,482✔
3386
      type = TSDB_FILL_SET_VALUE;
169,482✔
3387
      break;
169,482✔
3388
    case FILL_MODE_VALUE_F:
12,260✔
3389
      type = TSDB_FILL_SET_VALUE_F;
12,260✔
3390
      break;
12,260✔
3391
    case FILL_MODE_LINEAR:
239,678✔
3392
      type = TSDB_FILL_LINEAR;
239,678✔
3393
      break;
239,678✔
3394
    case FILL_MODE_NEAR:
2,920,836✔
3395
      type = TSDB_FILL_NEAR;
2,920,836✔
3396
      break;
2,920,836✔
3397
    default:
×
3398
      type = TSDB_FILL_NONE;
×
3399
  }
3400

3401
  return type;
3,954,449✔
3402
}
3403

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

3411
    int64_t key = w->skey;
330,999✔
3412
    while (key < ts) {  // moving towards end
347,986✔
3413
      key = getNextTimeWindowStart(pInterval, key, TSDB_ORDER_ASC);
176,680✔
3414
      if (key > ts) {
177,320✔
3415
        break;
160,333✔
3416
      }
3417

3418
      w->skey = key;
16,987✔
3419
    }
3420
    w->ekey = taosTimeAdd(w->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
331,639✔
3421
  }
3422
}
2,147,483,647✔
3423

3424
static STimeWindow doCalculateTimeWindow(int64_t ts, SInterval* pInterval) {
75,441,067✔
3425
  STimeWindow w = {0};
75,441,067✔
3426

3427
  w.skey = taosTimeTruncate(ts, pInterval);
75,441,067✔
3428
  w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
75,434,321✔
3429
  return w;
75,440,561✔
3430
}
3431

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

3441
  return save;
1,819,799✔
3442
}
3443

3444
// get the correct time window according to the handled timestamp
3445
// todo refactor
3446
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
116,667,471✔
3447
                                int32_t order) {
3448
  STimeWindow w = {0};
116,667,471✔
3449
  if (pResultRowInfo->cur.pageId == -1) {  // the first window, from the previous stored value
116,674,016✔
3450
    getInitialStartTimeWindow(pInterval, ts, &w, (order != TSDB_ORDER_DESC));
3,448,720✔
3451
    return w;
3,446,501✔
3452
  }
3453

3454
  SResultRow* pRow = getResultRowByPos(pBuf, &pResultRowInfo->cur, false);
113,223,693✔
3455
  if (pRow) {
113,223,138✔
3456
    TAOS_SET_OBJ_ALIGNED(&w, pRow->win);
113,224,622✔
3457
  }
3458

3459
  // in case of typical time window, we can calculate time window directly.
3460
  if (w.skey > ts || w.ekey < ts) {
113,225,971✔
3461
    w = doCalculateTimeWindow(ts, pInterval);
75,439,993✔
3462
  }
3463

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

3470
  return w;
113,221,989✔
3471
}
3472

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

3478
bool hasLimitOffsetInfo(SLimitInfo* pLimitInfo) {
548,081,582✔
3479
  return (pLimitInfo->limit.limit != -1 || pLimitInfo->limit.offset != -1 || pLimitInfo->slimit.limit != -1 ||
1,091,933,828✔
3480
          pLimitInfo->slimit.offset != -1);
543,851,868✔
3481
}
3482

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

3487
void initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo) {
673,240,408✔
3488
  SLimit limit = {.limit = getLimit(pLimit), .offset = getOffset(pLimit)};
673,240,408✔
3489
  SLimit slimit = {.limit = getLimit(pSLimit), .offset = getOffset(pSLimit)};
673,050,072✔
3490

3491
  pLimitInfo->limit = limit;
673,038,381✔
3492
  pLimitInfo->slimit = slimit;
673,065,500✔
3493
  pLimitInfo->remainOffset = limit.offset;
673,079,948✔
3494
  pLimitInfo->remainGroupOffset = slimit.offset;
673,072,403✔
3495
  pLimitInfo->numOfOutputRows = 0;
673,118,558✔
3496
  pLimitInfo->numOfOutputGroups = 0;
673,255,134✔
3497
  pLimitInfo->currentGroupId = 0;
673,233,107✔
3498
}
673,224,488✔
3499

3500
void resetLimitInfoForNextGroup(SLimitInfo* pLimitInfo) {
99,478,191✔
3501
  pLimitInfo->numOfOutputRows = 0;
99,478,191✔
3502
  pLimitInfo->remainOffset = pLimitInfo->limit.offset;
99,500,768✔
3503
}
99,471,255✔
3504

3505
int32_t tableListGetSize(const STableListInfo* pTableList, int32_t* pRes) {
703,650,801✔
3506
  if (taosArrayGetSize(pTableList->pTableList) != taosHashGetSize(pTableList->map)) {
703,650,801✔
3507
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR));
×
3508
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
3509
  }
3510
  (*pRes) = taosArrayGetSize(pTableList->pTableList);
703,603,176✔
3511
  return TSDB_CODE_SUCCESS;
703,607,077✔
3512
}
3513

3514
uint64_t tableListGetSuid(const STableListInfo* pTableList) { return pTableList->idInfo.suid; }
6,074,882✔
3515

3516
STableKeyInfo* tableListGetInfo(const STableListInfo* pTableList, int32_t index) {
216,581,447✔
3517
  if (taosArrayGetSize(pTableList->pTableList) == 0) {
216,581,447✔
3518
    return NULL;
2,207✔
3519
  }
3520

3521
  return taosArrayGet(pTableList->pTableList, index);
216,525,046✔
3522
}
3523

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

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

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

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

3556
  STableKeyInfo* pKeyInfo = taosArrayGet(pTableList->pTableList, *slot);
853,462,056✔
3557
  if (pKeyInfo == NULL) {
853,846,427✔
3558
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
13✔
NEW
3559
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
3560
  }
3561

3562
  *gid = pKeyInfo->groupId;
853,846,414✔
3563
  *baseGid = pKeyInfo->baseGId;
853,850,890✔
3564
  
3565
  return 0;
853,815,389✔
3566
}
3567

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

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

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

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

3589
//   return code;
3590
// }
3591

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

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

3607
  void* tmp = taosArrayPush(pTableList->pTableList, &keyInfo);
1,354,219✔
3608
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,355,507✔
3609

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

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

3625
  return code;
1,355,844✔
3626
}
3627

3628
int32_t tableListGetGroupList(const STableListInfo* pTableList, int32_t ordinalGroupIndex, STableKeyInfo** pKeyInfo,
267,363,172✔
3629
                              int32_t* size) {
3630
  int32_t totalGroups = tableListGetOutputGroups(pTableList);
267,363,172✔
3631
  int32_t numOfTables = 0;
267,348,370✔
3632
  int32_t code = tableListGetSize(pTableList, &numOfTables);
267,371,313✔
3633
  if (code != TSDB_CODE_SUCCESS) {
267,332,696✔
3634
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3635
    return code;
×
3636
  }
3637

3638
  if (ordinalGroupIndex < 0 || ordinalGroupIndex >= totalGroups) {
267,332,696✔
3639
    return TSDB_CODE_INVALID_PARA;
×
3640
  }
3641

3642
  // here handle two special cases:
3643
  // 1. only one group exists, and 2. one table exists for each group.
3644
  if (totalGroups == 1) {
267,332,696✔
3645
    *size = numOfTables;
266,779,012✔
3646
    *pKeyInfo = (*size == 0) ? NULL : taosArrayGet(pTableList->pTableList, 0);
266,813,681✔
3647
    return TSDB_CODE_SUCCESS;
266,715,642✔
3648
  } else if (totalGroups == numOfTables) {
553,684✔
3649
    *size = 1;
485,689✔
3650
    *pKeyInfo = taosArrayGet(pTableList->pTableList, ordinalGroupIndex);
485,689✔
3651
    return TSDB_CODE_SUCCESS;
485,689✔
3652
  }
3653

3654
  int32_t offset = pTableList->groupOffset[ordinalGroupIndex];
67,995✔
3655
  if (ordinalGroupIndex < totalGroups - 1) {
59,598✔
3656
    *size = pTableList->groupOffset[ordinalGroupIndex + 1] - offset;
44,791✔
3657
  } else {
3658
    *size = numOfTables - offset;
14,807✔
3659
  }
3660

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

3665
int32_t tableListGetOutputGroups(const STableListInfo* pTableList) { return pTableList->numOfOuputGroups; }
744,008,662✔
3666

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

3669
STableListInfo* tableListCreate() {
303,850,877✔
3670
  STableListInfo* pListInfo = taosMemoryCalloc(1, sizeof(STableListInfo));
303,850,877✔
3671
  if (pListInfo == NULL) {
303,780,906✔
3672
    return NULL;
×
3673
  }
3674

3675
  pListInfo->remainGroups = NULL;
303,780,906✔
3676
  pListInfo->pTableList = taosArrayInit(4, sizeof(STableKeyInfo));
303,813,630✔
3677
  if (pListInfo->pTableList == NULL) {
303,797,676✔
3678
    goto _error;
×
3679
  }
3680

3681
  pListInfo->map = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
303,824,157✔
3682
  if (pListInfo->map == NULL) {
303,958,206✔
3683
    goto _error;
×
3684
  }
3685

3686
  pListInfo->numOfOuputGroups = 1;
303,952,323✔
3687
  return pListInfo;
303,958,280✔
3688

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

3694
void tableListDestroy(STableListInfo* pTableListInfo) {
315,591,776✔
3695
  if (pTableListInfo == NULL) {
315,591,776✔
3696
    return;
11,706,577✔
3697
  }
3698

3699
  taosArrayDestroy(pTableListInfo->pTableList);
303,885,199✔
3700
  taosMemoryFreeClear(pTableListInfo->groupOffset);
303,817,559✔
3701

3702
  taosHashCleanup(pTableListInfo->map);
303,849,959✔
3703
  taosHashCleanup(pTableListInfo->remainGroups);
303,914,925✔
3704
  pTableListInfo->pTableList = NULL;
303,925,663✔
3705
  pTableListInfo->map = NULL;
303,919,738✔
3706
  taosMemoryFree(pTableListInfo);
303,910,841✔
3707
}
3708

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

3714
  taosArrayClear(pTableListInfo->pTableList);
821,693✔
3715
  taosHashClear(pTableListInfo->map);
821,344✔
3716
  taosHashClear(pTableListInfo->remainGroups);
822,343✔
3717
  taosMemoryFree(pTableListInfo->groupOffset);
822,343✔
3718
  pTableListInfo->numOfOuputGroups = 1;
822,343✔
3719
  pTableListInfo->oneTableForEachGroup = false;
822,343✔
3720
}
3721

3722
static int32_t orderbyGroupIdComparFn(const void* p1, const void* p2) {
551,425,299✔
3723
  STableKeyInfo* pInfo1 = (STableKeyInfo*)p1;
551,425,299✔
3724
  STableKeyInfo* pInfo2 = (STableKeyInfo*)p2;
551,425,299✔
3725

3726
  if (pInfo1->groupId == pInfo2->groupId) {
551,425,299✔
3727
    return 0;
515,059,958✔
3728
  } else {
3729
    return pInfo1->groupId < pInfo2->groupId ? -1 : 1;
36,372,806✔
3730
  }
3731
}
3732

3733
int32_t sortTableGroup(STableListInfo* pTableListInfo) {
31,659,751✔
3734
  int32_t code = TSDB_CODE_SUCCESS;
31,659,751✔
3735
  taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
31,659,751✔
3736
  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
31,666,171✔
3737
  if (size == 0) {
31,663,070✔
3738
    pTableListInfo->numOfOuputGroups = 0;
×
3739
    return code;
×
3740
  }
3741

3742
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
31,663,070✔
3743
  if (!pList) {
31,662,691✔
3744
    code = terrno;
×
3745
    goto end;
×
3746
  }
3747

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

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

3764
  for (int32_t i = 1; i < size; ++i) {
160,686,145✔
3765
    pInfo = taosArrayGet(pTableListInfo->pTableList, i);
129,029,946✔
3766
    if (pInfo == NULL) {
129,029,866✔
3767
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3768
      code = terrno;
×
3769
      goto end;
×
3770
    }
3771
    if (pInfo->groupId != gid) {
129,029,866✔
3772
      tmp = taosArrayPush(pList, &i);
7,452,834✔
3773
      if (tmp == NULL) {
7,452,834✔
3774
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3775
        code = terrno;
×
3776
        goto end;
×
3777
      }
3778
      gid = pInfo->groupId;
7,452,834✔
3779
    }
3780
  }
3781

3782
  pTableListInfo->numOfOuputGroups = taosArrayGetSize(pList);
31,659,525✔
3783
  pTableListInfo->groupOffset = taosMemoryMalloc(sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
31,658,830✔
3784
  if (pTableListInfo->groupOffset == NULL) {
31,660,290✔
3785
    code = terrno;
×
3786
    goto end;
×
3787
  }
3788

3789
  memcpy(pTableListInfo->groupOffset, taosArrayGet(pList, 0), sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
31,631,088✔
3790

3791
end:
31,643,770✔
3792
  taosArrayDestroy(pList);
31,645,598✔
3793
  return code;
31,612,490✔
3794
}
3795

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

3800
  bool   groupByTbname = groupbyTbname(group);
253,910,578✔
3801
  size_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
253,878,651✔
3802
  if (!numOfTables) {
253,890,815✔
3803
    return code;
×
3804
  }
3805
  qDebug("numOfTables:%zu, groupByTbname:%d, group:%p", numOfTables, groupByTbname, group);
253,890,815✔
3806
  if (group == NULL || groupByTbname) {
253,862,701✔
3807
    if (tsCountAlwaysReturnValue && QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode) &&
249,338,492✔
3808
        ((STableScanPhysiNode*)pScanNode)->needCountEmptyTable) {
205,813,670✔
3809
      pTableListInfo->remainGroups =
13,875,496✔
3810
          taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
13,874,981✔
3811
      if (pTableListInfo->remainGroups == NULL) {
13,875,496✔
3812
        return terrno;
×
3813
      }
3814

3815
      for (int i = 0; i < numOfTables; i++) {
71,653,717✔
3816
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
57,778,475✔
3817
        if (!info) {
57,777,805✔
3818
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3819
          return terrno;
×
3820
        }
3821
        info->groupId = groupByTbname ? info->uid : 0;
57,777,805✔
3822
        int32_t tempRes = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId),
57,777,805✔
3823
                                      &(info->uid), sizeof(info->uid));
57,777,135✔
3824
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
57,778,221✔
3825
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3826
          return tempRes;
×
3827
        }
3828
      }
3829
    } else {
3830
      for (int32_t i = 0; i < numOfTables; i++) {
846,477,928✔
3831
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
611,098,349✔
3832
        if (!info) {
611,080,965✔
3833
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3834
          return terrno;
×
3835
        }
3836
        info->groupId = groupByTbname ? info->uid : 0;
611,080,965✔
3837
        
3838
      }
3839
    }
3840
    if (groupIdMap && group != NULL){
249,254,821✔
3841
      getColInfoResultForGroupbyForStream(pHandle->vnode, group, pTableListInfo, pAPI, groupIdMap);
85,736✔
3842
    }
3843

3844
    pTableListInfo->oneTableForEachGroup = groupByTbname;
249,254,821✔
3845
    if (numOfTables == 1 && pTableListInfo->idInfo.tableType == TSDB_CHILD_TABLE) {
249,330,938✔
3846
      pTableListInfo->oneTableForEachGroup = true;
105,130,508✔
3847
    }
3848

3849
    if (groupSort && groupByTbname) {
249,334,080✔
3850
      taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
2,447,902✔
3851
      pTableListInfo->numOfOuputGroups = numOfTables;
2,449,945✔
3852
    } else if (groupByTbname && pScanNode->groupOrderScan) {
246,886,178✔
3853
      pTableListInfo->numOfOuputGroups = numOfTables;
25,396✔
3854
    } else {
3855
      pTableListInfo->numOfOuputGroups = 1;
246,861,160✔
3856
    }
3857
    if (groupSort || pScanNode->groupOrderScan) {
249,388,069✔
3858
      code = sortTableGroup(pTableListInfo);
31,530,927✔
3859
    }
3860
  } else {
3861
    bool initRemainGroups = false;
4,524,209✔
3862
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode)) {
4,524,209✔
3863
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pScanNode;
4,415,683✔
3864
      if (tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable &&
4,415,683✔
3865
          !(groupSort || pScanNode->groupOrderScan)) {
2,107,556✔
3866
        initRemainGroups = true;
2,082,314✔
3867
      }
3868
    }
3869

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

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

3877
    if (groupSort || pScanNode->groupOrderScan) {
4,524,209✔
3878
      code = sortTableGroup(pTableListInfo);
134,578✔
3879
    }
3880
  }
3881

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

3888
  // add all table entry in the hash map
3889
  size_t size = taosArrayGetSize(pTableListInfo->pTableList);
253,873,688✔
3890
  for (int32_t i = 0; i < size; ++i) {
947,306,163✔
3891
    STableKeyInfo* p = taosArrayGet(pTableListInfo->pTableList, i);
693,404,573✔
3892
    if (!p) {
693,217,502✔
3893
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3894
      return terrno;
×
3895
    }
3896
    int32_t tempRes = taosHashPut(pTableListInfo->map, &p->uid, sizeof(uint64_t), &i, sizeof(int32_t));
693,217,502✔
3897
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
693,441,859✔
3898
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3899
      return tempRes;
×
3900
    }
3901
  }
3902

3903
  return code;
253,951,518✔
3904
}
3905

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

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

3917
  if (pHandle->uid != 0) {
279,243,600✔
3918
    pScanNode->uid = pHandle->uid;
129,406✔
3919
    pScanNode->tableType = TSDB_CHILD_TABLE;
129,406✔
3920
  }
3921
  uint8_t digest[17] = {0};
279,394,542✔
3922
  int32_t code = getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, digest, idStr,
279,372,446✔
3923
                              &pTaskInfo->storageAPI, pTaskInfo->pStreamRuntimeInfo);
279,396,894✔
3924
  if (code != TSDB_CODE_SUCCESS) {
279,470,669✔
3925
    qError("failed to getTableList, code:%s", tstrerror(code));
996✔
3926
    return code;
996✔
3927
  }
3928

3929
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
279,469,673✔
3930

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

3936
  if (numOfTables == 0) {
279,447,053✔
3937
    qDebug("no table qualified for query, %s", idStr);
25,599,722✔
3938
    return TSDB_CODE_SUCCESS;
25,599,279✔
3939
  }
3940

3941
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, digest, &pTaskInfo->storageAPI, groupIdMap, false);
253,847,331✔
3942
  if (code != TSDB_CODE_SUCCESS) {
253,912,481✔
3943
    return code;
×
3944
  }
3945

3946
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
253,915,036✔
3947
  qDebug("generate group id map completed, elapsed time:%.2f ms %s", pTaskInfo->cost.groupIdMapTime, idStr);
253,913,293✔
3948

3949
  return TSDB_CODE_SUCCESS;
253,900,555✔
3950
}
3951

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

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

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

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

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

NEW
3978
_exit:
×
3979

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

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

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

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

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

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

NEW
4041
_exit:
×
4042

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

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

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

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

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

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

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

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

NEW
4090
_exit:
×
4091

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

4099

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

4144
  return TSDB_CODE_SUCCESS;
281,919,403✔
4145
}
4146

4147

4148
char* getStreamOpName(uint16_t opType) {
8,464,309✔
4149
  switch (opType) {
8,464,309✔
4150
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
×
4151
      return "stream scan";
×
4152
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
7,917,661✔
4153
      return "project";
7,917,661✔
4154
    case QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW:
546,648✔
4155
      return "external window";
546,648✔
4156
  }
4157
  return "error name";
×
4158
}
4159

4160
void printDataBlock(SSDataBlock* pBlock, const char* flag, const char* taskIdStr, int64_t qId) {
633,254,691✔
4161
  if (qDebugFlag & DEBUG_TRACE) {
633,254,691✔
4162
    if (!pBlock) {
3,262,633✔
4163
      qDebug("%" PRIx64 " %s %s %s: Block is Null", qId, taskIdStr, flag, __func__);
212,456✔
4164
      return;
374,081✔
4165
    } else if (pBlock->info.rows == 0) {
3,050,177✔
4166
      qDebug("%" PRIx64 " %s %s %s: Block is Empty.", qId, taskIdStr, flag, __func__);
169,995✔
4167
      return;
169,995✔
4168
    }
4169
    
4170
    char*   pBuf = NULL;
2,880,182✔
4171
    int32_t code = dumpBlockData(pBlock, flag, &pBuf, taskIdStr, qId);
2,880,182✔
4172
    if (code == 0) {
2,879,909✔
4173
      qDebugL("%" PRIx64 " %s %s", qId, __func__, pBuf);
2,880,059✔
4174
      taosMemoryFree(pBuf);
2,880,119✔
4175
    }
4176
  }
4177
}
4178

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

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

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

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

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

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

4223
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
2,147,483,647✔
4224
      if (isNull[i] != 1) return 1;
277,492,625✔
4225
    } else {
4226
      if (isNull[i] != 0) return 1;
2,147,483,647✔
4227
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,147,483,647✔
4228
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,147,483,647✔
4229
        int32_t len = getJsonValueLen(val);
×
4230
        if (memcmp(p, val, len) != 0) return 1;
×
4231
        p += len;
×
4232
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,147,483,647✔
4233
        if (IS_STR_DATA_BLOB(pCol->type)) {
1,284,774,300✔
4234
          if (memcmp(p, val, blobDataTLen(val)) != 0) return 1;
×
4235
          p += blobDataTLen(val);
×
4236
        } else {
4237
          if (memcmp(p, val, varDataTLen(val)) != 0) return 1;
1,283,597,725✔
4238
          p += varDataTLen(val);
1,282,969,300✔
4239
        }
4240
      } else {
4241
        if (0 != memcmp(p, val, pCol->bytes)) return 1;
2,147,483,647✔
4242
        p += pCol->bytes;
2,147,483,647✔
4243
      }
4244
    }
4245
  }
4246
  if ((int32_t)(p - oldkeyBuf) != oldKeysLen) return 1;
2,147,483,647✔
4247
  return 0;
2,147,483,647✔
4248
}
4249

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

4256
  for (int32_t i = 0; i < colNum; ++i) {
4,634,100✔
4257
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
2,941,675✔
4258
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
2,941,675✔
4259
    if (pCol->slotId > pBlock->pDataBlock->size) continue;
2,941,675✔
4260

4261
    if (pBlock->pBlockAgg) pColAgg = &pBlock->pBlockAgg[pCol->slotId];
2,941,675✔
4262

4263
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
5,883,350✔
4264
      isNull[i] = 1;
142,500✔
4265
    } else {
4266
      isNull[i] = 0;
2,799,175✔
4267
      const char* val = colDataGetData(pColInfoData, rowIndex);
2,799,175✔
4268
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
2,799,175✔
4269
        int32_t len = getJsonValueLen(val);
×
4270
        memcpy(p, val, len);
×
4271
        p += len;
×
4272
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
2,799,175✔
4273
        if (IS_STR_DATA_BLOB(pCol->type)) {
1,046,425✔
4274
          blobDataCopy(p, val);
×
4275
          p += blobDataTLen(val);
×
4276
        } else {
4277
          varDataCopy(p, val);
1,046,425✔
4278
          p += varDataTLen(val);
1,046,425✔
4279
        }
4280
      } else {
4281
        memcpy(p, val, pCol->bytes);
1,752,750✔
4282
        p += pCol->bytes;
1,752,750✔
4283
      }
4284
    }
4285
  }
4286
  return (int32_t)(p - keyBuf);
1,692,425✔
4287
}
4288

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

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

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

4317
int32_t extractKeysLen(const SArray* keys, int32_t* pLen) {
112,100✔
4318
  int32_t code = TSDB_CODE_SUCCESS;
112,100✔
4319
  int32_t lino = 0;
112,100✔
4320
  int32_t len = 0;
112,100✔
4321
  int32_t keyNum = taosArrayGetSize(keys);
112,100✔
4322
  for (int32_t i = 0; i < keyNum; ++i) {
284,525✔
4323
    SColumn* pCol = (SColumn*)taosArrayGet(keys, i);
172,425✔
4324
    QUERY_CHECK_NULL(pCol, code, lino, _end, terrno);
171,475✔
4325
    len += pCol->bytes;
171,475✔
4326
  }
4327
  len += sizeof(int8_t) * keyNum;  // null flag
112,100✔
4328
  *pLen = len;
112,100✔
4329

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

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

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

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

4364

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

4372
  pBlock->pDataBlock = taosArrayInit(cols, sizeof(SColumnInfoData));
29,597,233✔
4373
  if (pBlock->pDataBlock == NULL) {
29,596,686✔
4374
    code = terrno;
×
4375
    taosMemoryFree(pBlock);
×
4376
    return code;
×
4377
  }
4378

4379
  for (int32_t i = 0; i < cols; ++i) {
60,182,103✔
4380
    SColumnInfoData idata = {0};
30,584,870✔
4381
    if (pResType) {
30,584,870✔
4382
      idata = createColumnInfoData(pResType->type, pResType->bytes, 0);
29,084,818✔
4383
      idata.info.scale = pResType->scale;
29,085,365✔
4384
      idata.info.precision = pResType->precision;
29,085,365✔
4385
    }
4386
    
4387
    code = blockDataAppendColInfo(pBlock, &idata);
30,584,870✔
4388
    if (code != TSDB_CODE_SUCCESS) {
30,585,164✔
NEW
4389
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
NEW
4390
      blockDataDestroy(pBlock);
×
NEW
4391
      *ppBlock = NULL;
×
NEW
4392
      return code;
×
4393
    }
4394
  }
4395

4396
  *ppBlock = pBlock;
29,597,233✔
4397

4398
  return code;
29,597,233✔
4399
}
4400

4401

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

4407
  char* pNextStart = pRetrieveRsp->data;
29,597,233✔
4408
  char* pStart = pNextStart;
29,597,233✔
4409

4410
  int32_t index = 0;
29,597,233✔
4411

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

4417
  int32_t compLen = *(int32_t*)pStart;
29,596,829✔
4418
  pStart += sizeof(int32_t);
29,596,829✔
4419

4420
  int32_t rawLen = *(int32_t*)pStart;
29,597,233✔
4421
  pStart += sizeof(int32_t);
29,596,829✔
4422
  QUERY_CHECK_CONDITION((compLen <= rawLen && compLen != 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
29,596,282✔
4423

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

4431
  code = blockDecodeInternal(pb, pStart, (const char**)&pStart);
29,596,829✔
4432
  if (code != 0) {
29,596,829✔
4433
    taosMemoryFreeClear(pRetrieveRsp);
×
4434
    goto _end;
×
4435
  }
4436

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

4445
int32_t setValueFromResBlock(STaskSubJobCtx* ctx, SValueNode* pRes, SSDataBlock* pBlock) {
18,993,641✔
4446
  int32_t code = 0;
18,993,641✔
4447
  bool needFree = true;
18,993,641✔
4448
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
18,993,641✔
4449
  if (NULL == pBlock->pDataBlock || 1 > colNum || pBlock->info.rows > 1) {
18,993,641✔
4450
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d, rows:%" PRId64, 
547✔
4451
      ctx->idStr, pBlock->pDataBlock, colNum, pBlock->info.rows);
4452
    return TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
×
4453
  }
4454
  
4455
  pRes->flag &= (~VALUE_FLAG_VAL_UNSET);
18,993,094✔
4456
  pRes->translate = true;
18,993,641✔
4457
  
4458
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
18,993,094✔
4459
  if (colDataIsNull_s(pCol, 0)) {
18,993,641✔
4460
    pRes->isNull = true;
5,120,820✔
4461
  } else {
4462
    code = nodesSetValueNodeValueExt(pRes, colDataGetData(pCol, 0), &needFree);
13,872,821✔
4463
  }
4464

4465
  if (!needFree) {
18,993,641✔
4466
    pCol->pData = NULL;
1,881,360✔
4467
  }
4468

4469
  return code;
18,993,641✔
4470
}
4471

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

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

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

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

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

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

4519
    return;
115,100✔
4520
  }
4521

4522
  *fetchDone = true;
3,331,692✔
4523

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

4529
    return;
×
4530
  }
4531

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

4536
    return;
×
4537
  }
4538

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

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

4564
  blockDataDestroy(pResBlock);  
3,153,104✔
4565
}
4566

4567

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

4576
  pRes->hasValue = true;
10,091,471✔
4577
  
4578
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
10,091,724✔
4579
  TAOS_CHECK_EXIT(scalarBuildRemoteListHash(ctx->idStr, pRes, pCol, pBlock->info.rows));
10,091,218✔
4580

4581
_exit:
9,362,094✔
4582

4583
  if (code) {
10,091,724✔
4584
    qError("%s %s failed with error: %s", ctx->idStr, __func__, tstrerror(code));
729,630✔
4585
  }
4586
  
4587
  return code;
10,091,724✔
4588
}
4589

4590

4591

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

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

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

4607
    ctx->code = createExprSubQResBlock(&pResBlock, &((SExprNode*)pParam->pRes)->resType, 1);
10,091,724✔
4608
    if (TSDB_CODE_SUCCESS == ctx->code) {
10,091,471✔
4609
      ctx->code = blockDataEnsureCapacity(pResBlock, pRsp->numOfRows);
10,091,724✔
4610
    }
4611
    if (TSDB_CODE_SUCCESS == ctx->code) {
10,091,724✔
4612
      ctx->code = extractSingleRspBlock(pRsp, pResBlock);
10,091,724✔
4613
    }
4614
    if (TSDB_CODE_SUCCESS == ctx->code) {
10,091,471✔
4615
      ctx->code = setValueListFromResBlock(ctx, pRemote, pResBlock);
10,091,724✔
4616
    }
4617
    if (TSDB_CODE_SUCCESS == ctx->code && pRsp->completed) {
10,091,724✔
4618
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
8,003,435✔
4619
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
8,003,435✔
4620
    }
4621

4622
    blockDataDestroy(pResBlock);  
10,091,724✔
4623
  } else if (0 == pRsp->numOfRows && pRsp->completed) {
2,979,794✔
4624
    if (!pRemote->hasValue) {
2,979,794✔
4625
      ctx->code = scalarBuildRemoteListHash(ctx->idStr, pRemote, NULL, 0);
2,961,205✔
4626
    }
4627
    if (TSDB_CODE_SUCCESS == ctx->code) {    
2,979,794✔
4628
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
2,732,564✔
4629
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
2,732,564✔
4630
    }
4631
  }
4632

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

4642
    pRsp->completed = true;
19,064✔
4643
  }
4644

4645
  *fetchDone = (TSDB_CODE_SUCCESS != ctx->code || pRsp->completed) ? true : false;
13,071,518✔
4646

4647
  if (!(*fetchDone)) {
13,070,958✔
4648
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
1,358,659✔
4649
    if (TSDB_CODE_SUCCESS != code) {
1,358,659✔
4650
      ctx->code = code;
×
4651
      *fetchDone = true;
×
4652
    }
4653
  }
4654
}
4655

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

4672
  return code;
15,782,440✔
4673
}
4674

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

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

4681
  SRemoteRowNode* pRemote = (SRemoteRowNode*)pParam->pRes;
21,391,393✔
4682

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

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

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

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

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

4740
    return;
1,012✔
4741
  }
4742

4743
  *fetchDone = true;
21,390,381✔
4744

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

4750
    return;
×
4751
  }
4752

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

4762
    return;
5,607,900✔
4763
  }
4764

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

4769
    return;
×
4770
  }
4771

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

4797
  blockDataDestroy(pResBlock);
15,780,840✔
4798
}
4799

4800

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

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

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

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

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

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

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

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

4846

4847

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

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

4858
_exit:
511,464✔
4859

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

4865
  *ppBlock = NULL;
511,464✔
4866
  
4867
  return code;
511,464✔
4868
}
4869

4870

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

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

4877
  if (pRsp->numOfRows > 0) {
514,292✔
4878
    ctx->code = createExprSubQResBlock(&pResBlock, NULL, pRemote->resCols);
511,868✔
4879
    if (TSDB_CODE_SUCCESS == ctx->code) {
511,868✔
4880
      ctx->code = extractSingleRspBlock(pRsp, pResBlock);
511,868✔
4881
    }
4882
    if (TSDB_CODE_SUCCESS == ctx->code) {
511,868✔
4883
      ctx->code = setTableFromResBlock(ctx, pRemote, &pResBlock);
511,464✔
4884
    }
4885
    if (TSDB_CODE_SUCCESS == ctx->code && pRsp->completed) {
511,464✔
4886
      pRemote->flag &= (~REMOTE_TABLE_FLAG_VAL_UNSET);
396,728✔
4887
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
397,132✔
4888
    }
4889

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

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

4898
  if (!(*fetchDone)) {
513,888✔
4899
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, 1);
114,736✔
4900
    if (TSDB_CODE_SUCCESS != code) {
114,736✔
NEW
4901
      ctx->code = code;
×
NEW
4902
      *fetchDone = true;
×
4903
    }
4904
  }
4905
}
513,888✔
4906

4907

4908
int32_t remoteFetchCallBack(void* param, SDataBuf* pMsg, int32_t code) {
39,595,665✔
4909
  taosMemoryFreeClear(pMsg->pEpSet);
39,595,665✔
4910

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

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

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

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

4939
  if (code == TSDB_CODE_SUCCESS) {
39,592,318✔
4940
    SRetrieveTableRsp* pRsp = pMsg->pData;
38,703,598✔
4941
    pRsp->numOfRows = htobe64(pRsp->numOfRows);
38,703,598✔
4942
    pRsp->compLen = htonl(pRsp->compLen);
38,703,598✔
4943
    pRsp->payloadLen = htonl(pRsp->payloadLen);
38,704,145✔
4944
    pRsp->numOfCols = htonl(pRsp->numOfCols);
38,704,145✔
4945
    pRsp->useconds = htobe64(pRsp->useconds);
38,704,145✔
4946
    pRsp->numOfBlocks = htonl(pRsp->numOfBlocks);
38,704,145✔
4947

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

4951
    ctx->blockIdx++;
38,704,145✔
4952

4953
    switch (nodeType(pParam->pRes)) {
38,704,145✔
4954
      case QUERY_NODE_REMOTE_VALUE: {
3,446,792✔
4955
        bool fetchDone = false;
3,446,792✔
4956
        handleRemoteValueRes(pParam, ctx, pRsp, &fetchDone);
3,446,792✔
4957
        qDebug("%s subQIdx %d handle remote value, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
3,446,792✔
4958
        if (!fetchDone) {
3,446,792✔
4959
          goto _exit;
57,550✔
4960
        }
4961
        break;
3,389,242✔
4962
      }
4963
      case QUERY_NODE_REMOTE_VALUE_LIST: {
13,071,518✔
4964
        bool fetchDone = false;
13,071,518✔
4965
        handleRemoteValueListRes(pParam, ctx, pRsp, &fetchDone);
13,071,518✔
4966
        qDebug("%s subQIdx %d handle remote value list finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
13,071,518✔
4967
        if (!fetchDone) {
13,071,518✔
4968
          goto _exit;
1,358,659✔
4969
        }
4970
        break;
11,712,859✔
4971
      }  
4972
      case QUERY_NODE_REMOTE_ROW: {
21,391,393✔
4973
        bool fetchDone = false;
21,391,393✔
4974
        handleRemoteRowRes(pParam, ctx, pRsp, &fetchDone);
21,391,393✔
4975
        qDebug("%s subQIdx %d handle remote row finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
21,391,393✔
4976
        if (!fetchDone) {
21,390,846✔
4977
          goto _exit;
506✔
4978
        }
4979
        break;
21,390,340✔
4980
      }
4981
      case QUERY_NODE_REMOTE_ZERO_ROWS: {
280,150✔
4982
        bool fetchDone = false;
280,150✔
4983
        handleRemoteZeroRowsRes(pParam, ctx, pRsp, &fetchDone);
280,150✔
4984
        qDebug("%s subQIdx %d handle remote zeroRows finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
280,150✔
4985
        if (!fetchDone) {
280,150✔
4986
          goto _exit;
×
4987
        }
4988
        break;
280,150✔
4989
      }
4990
      case QUERY_NODE_REMOTE_TABLE: {
514,292✔
4991
        bool fetchDone = false;
514,292✔
4992
        handleRemoteTableRes(pParam, ctx, pRsp, &fetchDone);
514,292✔
4993
        qDebug("%s subQIdx %d handle remote table finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
514,292✔
4994
        if (!fetchDone) {
514,292✔
4995
          goto _exit;
114,736✔
4996
        }
4997
        break;
399,556✔
4998
      }
4999
      default:
×
5000
        qError("%s invalid scl fetch res node %d, subQIdx:%d", ctx->idStr, nodeType(pParam->pRes), pParam->subQIdx);
×
5001
        ctx->code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
5002
        break;
×
5003
    }
5004
  } else {
5005
    ctx->code = rpcCvtErrCode(code);
888,720✔
5006
    if (ctx->code != code) {
888,720✔
5007
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s, cvted error: %s", ctx->idStr, pParam->subQIdx,
×
5008
             tstrerror(code), tstrerror(ctx->code));
5009
    } else {
5010
      qError("%s scl fetch rsp received, subQIdx:%d, error:%s", ctx->idStr, pParam->subQIdx, tstrerror(code));
888,720✔
5011
    }
5012
  }
5013

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

5021
_exit:
39,592,318✔
5022

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

5028
  taosMemoryFree(pMsg->pData);
39,592,865✔
5029

5030
  return code;
39,592,865✔
5031
}
5032

5033
#define STREAM_TASK_REPLICA_NUM 5
5034
#define STREAM_TASK_DEPLOY_NUM 3
5035

5036
int32_t sendFetchRemoteNodeReq(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes, bool reset) {
39,596,225✔
5037
  int32_t          code = TSDB_CODE_SUCCESS;
39,596,225✔
5038
  int32_t          lino = 0;
39,596,225✔
5039
  int32_t          innerIdx = 0;
39,596,225✔
5040
  SExecTaskInfo*   pTaskInfo = ctx->pTaskInfo;
39,596,225✔
5041
  bool             needStreamPesudoFuncVals = false;
39,593,477✔
5042

5043
  if (IS_STREAM_MODE(pTaskInfo)) {
39,593,477✔
5044
    innerIdx = pTaskInfo->pStreamRuntimeInfo->execId / STREAM_TASK_REPLICA_NUM;
1,206,245✔
5045
    subQIdx = subQIdx * STREAM_TASK_DEPLOY_NUM + innerIdx;
1,206,245✔
5046
  }
5047

5048
  SDownstreamSourceNode* pSource = (SDownstreamSourceNode*)taosArrayGetP(ctx->subEndPoints, subQIdx);
39,588,984✔
5049

5050
  SResFetchReq req = {0};
39,587,877✔
5051
  req.header.vgId = pSource->addr.nodeId;
39,587,877✔
5052
  req.sId = pSource->sId;
39,591,198✔
5053
  req.clientId = pSource->clientId;
39,596,238✔
5054
  req.taskId = pSource->taskId;
39,594,024✔
5055
  req.srcTaskId = ctx->taskId;
39,587,864✔
5056
  req.blockIdx = ctx->blockIdx;
39,588,437✔
5057
  req.queryId = ctx->queryId;
39,596,785✔
5058
  req.execId = pSource->execId;
39,587,278✔
5059

5060
  if (IS_STREAM_MODE(pTaskInfo)) {
39,583,918✔
5061
    req.queryId = pSource->clientId;
1,206,245✔
5062
    req.execId = pTaskInfo->pStreamRuntimeInfo->execId;
1,206,245✔
5063
    req.pStRtFuncInfo = &pTaskInfo->pStreamRuntimeInfo->funcInfo;
1,206,245✔
5064
    req.reset = reset;
1,206,245✔
5065

5066
    needStreamPesudoFuncVals = true;
1,206,245✔
5067
    subQIdx = (subQIdx - innerIdx) / STREAM_TASK_DEPLOY_NUM;
1,206,245✔
5068
  }
5069

5070
  int32_t msgSize = tSerializeSResFetchReq(NULL, 0, &req, needStreamPesudoFuncVals, false);
39,588,411✔
5071
  if (msgSize < 0) {
39,595,118✔
5072
    return msgSize;
×
5073
  }
5074

5075
  void* msg = taosMemoryCalloc(1, msgSize);
39,595,118✔
5076
  if (NULL == msg) {
39,582,798✔
5077
    return terrno;
×
5078
  }
5079

5080
  msgSize = tSerializeSResFetchReq(msg, msgSize, &req, needStreamPesudoFuncVals, false);
39,582,798✔
5081
  if (msgSize < 0) {
39,586,744✔
5082
    taosMemoryFree(msg);
×
5083
    return msgSize;
×
5084
  }
5085

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

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

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

5107
  if (ctx->code) {
39,587,877✔
5108
    qError("task has been killed, error:%s", tstrerror(ctx->code));
×
5109
    taosMemoryFree(param);
×
5110
    taosMemoryFreeClear(msg);
×
5111
    taosMemoryFreeClear(pMsgSendInfo);
×
5112
    code = ctx->code;
×
5113
    goto _end;
×
5114
  }
5115
  
5116
  param->subQIdx = subQIdx;
39,590,130✔
5117
  param->pRes = pRes;
39,593,451✔
5118
  param->subJobRefId = ctx->subJobRefId;
39,592,895✔
5119

5120
  pMsgSendInfo->param = param;
39,595,131✔
5121
  pMsgSendInfo->paramFreeFp = taosAutoMemoryFree;
39,592,331✔
5122
  pMsgSendInfo->msgInfo.pData = msg;
39,587,890✔
5123
  pMsgSendInfo->msgInfo.len = msgSize;
39,594,571✔
5124
  pMsgSendInfo->msgType = pSource->fetchMsgType;
39,595,131✔
5125
  pMsgSendInfo->fp = remoteFetchCallBack;
39,582,828✔
5126
  pMsgSendInfo->requestId = ctx->queryId;
39,585,624✔
5127

5128
  code = asyncSendMsgToServer(ctx->rpcHandle, &pSource->addr.epSet, &ctx->transporterId, pMsgSendInfo);
39,592,357✔
5129
  QUERY_CHECK_CODE(code, lino, _end);
39,597,905✔
5130
      
5131
_end:
39,597,905✔
5132

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

5140
int32_t fetchRemoteNodeImpl(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes) {
38,064,774✔
5141
  int32_t          code = TSDB_CODE_SUCCESS;
38,064,774✔
5142
  int32_t          lino = 0;
38,064,774✔
5143

5144
  ctx->blockIdx = 0;
38,064,774✔
5145

5146
  code = sendFetchRemoteNodeReq(ctx, subQIdx, pRes, true);
38,064,774✔
5147
  QUERY_CHECK_CODE(code, lino, _end);
38,067,014✔
5148

5149
  code = qSemWait(ctx->pTaskInfo, &ctx->ready);
38,067,014✔
5150
  if (isTaskKilled(ctx->pTaskInfo)) {
38,067,014✔
5151
    code = getTaskCode(ctx->pTaskInfo);
20,160✔
5152
  } else {
5153
    code = ctx->code;
38,046,854✔
5154
  }
5155
      
5156
_end:
38,067,014✔
5157

5158
  if (code != TSDB_CODE_SUCCESS) {
38,067,014✔
5159
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
1,871,180✔
5160
  }
5161
  return code;
38,067,014✔
5162
}
5163

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

5203
_exit:
119,530✔
5204

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

5209
  return code;
119,530✔
5210
}
5211

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

5221
  SNode** ppRes = taosArrayGet(ctx->subResNodes, subQIdx);
38,172,596✔
5222
  if (NULL == *ppRes) {
38,177,610✔
5223
    TAOS_CHECK_EXIT(fetchRemoteNodeImpl(ctx, subQIdx, pRes));
38,062,534✔
5224
    *ppRes = pRes;
36,195,834✔
5225
  } else {
5226
    TAOS_CHECK_EXIT(remoteNodeCopy(*ppRes, pRes));
119,530✔
5227
  }
5228

5229
_exit:
119,530✔
5230

5231
  if (code) {
38,186,544✔
5232
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
1,871,180✔
5233
  } else {
5234
    qDebug("%s %s subQIdx %d succeed", ctx->idStr, __func__, subQIdx);
36,315,364✔
5235
  }
5236

5237
  return code;
38,186,544✔
5238
}
5239

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

5245
  int32_t numOfCols = (int32_t)taosArrayGetSize(pBlock->pDataBlock);
224,802,596✔
5246
  if (slotId >= 0 && slotId < numOfCols) {
224,803,126✔
5247
    SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, slotId);
224,803,126✔
5248
    if (NULL != pCol && pCol->info.slotId == slotId) {
224,803,126✔
5249
      return slotId;
224,803,126✔
5250
    }
5251
  }
5252

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

NEW
5260
  return -1;
×
5261
}
5262

5263
SColumnInfoData* getDataBlockColBySlotId(const SSDataBlock* pBlock, int32_t slotId, int32_t* pIndex) {
224,803,126✔
5264
  int32_t index = findDataBlockColIndexBySlotId(pBlock, slotId);
224,803,126✔
5265
  if (NULL != pIndex) {
224,803,126✔
NEW
5266
    *pIndex = index;
×
5267
  }
5268
  return (index >= 0) ? taosArrayGet(pBlock->pDataBlock, index) : NULL;
224,803,126✔
5269
}
STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2026 Coveralls, Inc