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

taosdata / TDengine / #5083

17 May 2026 01:15AM UTC coverage: 73.377% (+0.05%) from 73.328%
#5083

push

travis-ci

web-flow
feat (TDgpt): Dynamic Model Synchronization Enhancements (#35344)

* refactor: do some internal refactor.

* fix: fix multiprocess sync issue.

* feat: add dynamic anomaly detection and forecasting services

* fix: log error message for undeploying model in exception handling

* Potential fix for pull request finding

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

* Potential fix for pull request finding

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

* Potential fix for pull request finding

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

* Potential fix for pull request finding

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

* fix: handle undeploy when model exists only on disk

Agent-Logs-Url: https://github.com/taosdata/TDengine/sessions/286aafa0-c3ce-4c27-b803-2707571e9dc1

Co-authored-by: hjxilinx <8252296+hjxilinx@users.noreply.github.com>

* fix: guard dynamic registry concurrent access

Agent-Logs-Url: https://github.com/taosdata/TDengine/sessions/5e4db858-6458-40f4-ac28-d1b1b7f97c18

Co-authored-by: hjxilinx <8252296+hjxilinx@users.noreply.github.com>

* fix: tighten service list locking scope

Agent-Logs-Url: https://github.com/taosdata/TDengine/sessions/5e4db858-6458-40f4-ac28-d1b1b7f97c18

Co-authored-by: hjxilinx <8252296+hjxilinx@users.noreply.github.com>

* fix: restore prophet support and update tests per review feedback

Agent-Logs-Url: https://github.com/taosdata/TDengine/sessions/92298ae1-7da6-4d07-b20e-101c7cd0b26b

Co-authored-by: hjxilinx <8252296+hjxilinx@users.noreply.github.com>

* fix: improve test name and move copy inside lock scope

Agent-Logs-Url: https://github.com/taosdata/TDengine/sessions/92298ae1-7da6-4d07-b20e-101c7cd0b26b

Co-authored-by: hjxilinx <8252296+hjxilinx@users.noreply.github.com>

* Potential fix for pull request finding

Co-au... (continued)

281619 of 383795 relevant lines covered (73.38%)

134936217.89 hits per line

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

74.26
/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,475,507,348✔
65
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->limit) ? -1 : ((SLimitNode*)pLimit)->limit->datum.i;
1,475,507,348✔
66
}
67
static int64_t getOffset(const SNode* pLimit) {
1,475,378,549✔
68
  return (NULL == pLimit || NULL == ((SLimitNode*)pLimit)->offset) ? -1 : ((SLimitNode*)pLimit)->offset->datum.i;
1,475,378,549✔
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) {
554,825,770✔
74
  pResultRowInfo->size = 0;
554,825,770✔
75
  pResultRowInfo->cur.pageId = -1;
554,851,427✔
76
}
554,826,672✔
77

78
void closeResultRow(SResultRow* pResultRow) { pResultRow->closed = true; }
6,233,318✔
79

80
void resetResultRow(SResultRow* pResultRow, size_t entrySize) {
1,069,363,321✔
81
  pResultRow->numOfRows = 0;
1,069,363,321✔
82
  pResultRow->nOrigRows = 0;
1,069,363,321✔
83
  pResultRow->closed = false;
1,069,362,212✔
84
  pResultRow->endInterp = false;
1,069,362,696✔
85
  pResultRow->startInterp = false;
1,069,361,750✔
86

87
  if (entrySize > 0) {
1,069,363,413✔
88
    memset(pResultRow->pEntryInfo, 0, entrySize);
1,069,366,000✔
89
  }
90
}
1,069,364,523✔
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) {
332,440,835✔
98
  int32_t rowSize = (numOfOutput * sizeof(SResultRowEntryInfo)) + sizeof(SResultRow);
332,440,835✔
99

100
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,418,945,273✔
101
    rowSize += pCtx[i].resDataInfo.interBufSize;
1,086,586,696✔
102
  }
103

104
  return rowSize;
332,358,577✔
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) {
155,140,652✔
221
  taosMemoryFreeClear(pGroupResInfo->pBuf);
155,140,652✔
222
  if (pGroupResInfo->freeItem) {
155,139,354✔
223
    //    taosArrayDestroy(pGroupResInfo->pRows);
224
    taosArrayDestroyEx(pGroupResInfo->pRows, freeEx);
×
225
    pGroupResInfo->freeItem = false;
×
226
    pGroupResInfo->pRows = NULL;
×
227
  } else {
228
    taosArrayDestroy(pGroupResInfo->pRows);
155,138,278✔
229
    pGroupResInfo->pRows = NULL;
155,132,716✔
230
  }
231
  pGroupResInfo->index = 0;
155,139,045✔
232
  pGroupResInfo->delIndex = 0;
155,139,604✔
233
}
155,125,779✔
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) {
117,957,172✔
256
  int32_t code = TSDB_CODE_SUCCESS;
117,957,172✔
257
  int32_t lino = 0;
117,957,172✔
258
  if (pGroupResInfo->pRows != NULL) {
117,957,172✔
259
    taosArrayDestroy(pGroupResInfo->pRows);
2,618,805✔
260
  }
261
  if (pGroupResInfo->pBuf) {
117,960,829✔
262
    taosMemoryFree(pGroupResInfo->pBuf);
2,618,124✔
263
    pGroupResInfo->pBuf = NULL;
2,618,124✔
264
  }
265

266
  // extract the result rows information from the hash map
267
  int32_t size = tSimpleHashGetSize(pHashmap);
117,953,919✔
268

269
  void* pData = NULL;
117,958,106✔
270
  pGroupResInfo->pRows = taosArrayInit(size, POINTER_BYTES);
117,958,106✔
271
  QUERY_CHECK_NULL(pGroupResInfo->pRows, code, lino, _end, terrno);
117,960,639✔
272

273
  size_t  keyLen = 0;
117,953,249✔
274
  int32_t iter = 0;
117,953,678✔
275
  int64_t bufLen = 0, offset = 0;
117,954,830✔
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);
117,908,478✔
284
  QUERY_CHECK_NULL(pGroupResInfo->pBuf, code, lino, _end, terrno);
117,959,855✔
285

286
  iter = 0;
117,955,708✔
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) {
117,933,423✔
302
    __compar_fn_t fn = (order == TSDB_ORDER_ASC) ? resultrowComparAsc : resultrowComparDesc;
7,137,270✔
303
    size = POINTER_BYTES;
7,137,270✔
304
    taosSort(pGroupResInfo->pRows->pData, taosArrayGetSize(pGroupResInfo->pRows), size, fn);
7,137,270✔
305
  }
306

307
  pGroupResInfo->index = 0;
117,933,260✔
308

309
_end:
117,939,767✔
310
  if (code != TSDB_CODE_SUCCESS) {
117,961,102✔
311
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
312
  }
313
  return code;
117,961,102✔
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) {
468,437,132✔
328
  if (pGroupResInfo->pRows == NULL) {
468,437,132✔
329
    return false;
×
330
  }
331

332
  return pGroupResInfo->index < taosArrayGetSize(pGroupResInfo->pRows);
468,442,746✔
333
}
334

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

340
  return (int32_t)taosArrayGetSize(pGroupResInfo->pRows);
249,533,523✔
341
}
342

343
SArray* createSortInfo(SNodeList* pNodeList) {
60,039,037✔
344
  size_t numOfCols = 0;
60,039,037✔
345

346
  if (pNodeList != NULL) {
60,039,037✔
347
    numOfCols = LIST_LENGTH(pNodeList);
59,789,556✔
348
  } else {
349
    numOfCols = 0;
249,481✔
350
  }
351

352
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
60,045,711✔
353
  if (pList == NULL) {
60,034,432✔
354
    return pList;
×
355
  }
356

357
  for (int32_t i = 0; i < numOfCols; ++i) {
136,134,696✔
358
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
76,093,624✔
359
    if (!pSortKey) {
76,097,797✔
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};
76,097,797✔
367
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
76,094,334✔
368
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);
76,092,360✔
369

370
    if (nodeType(pSortKey->pExpr) != QUERY_NODE_COLUMN) {
76,093,346✔
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;
76,085,278✔
379
    bi.slotId = pColNode->slotId;
76,086,628✔
380
    void* tmp = taosArrayPush(pList, &bi);
76,100,425✔
381
    if (!tmp) {
76,100,425✔
382
      taosArrayDestroy(pList);
×
383
      pList = NULL;
×
384
      break;
×
385
    }
386
  }
387

388
  return pList;
60,040,651✔
389
}
390

391
SSDataBlock* createDataBlockFromDescNode(void* p) {
882,356,456✔
392
  SDataBlockDescNode* pNode = (SDataBlockDescNode*)p;
882,356,456✔
393
  int32_t      numOfCols = LIST_LENGTH(pNode->pSlots);
882,356,456✔
394
  SSDataBlock* pBlock = NULL;
882,264,411✔
395
  int32_t      code = TSDB_CODE_SUCCESS;
882,261,843✔
396
  int32_t      lino = 0;
882,261,843✔
397

398
  code = createDataBlock(&pBlock);
882,261,843✔
399
  QUERY_CHECK_CODE(code, lino, _return);
882,267,648✔
400

401
  pBlock->info.id.blockId = pNode->dataBlockId;
882,267,648✔
402
  pBlock->info.calWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
882,275,922✔
403
  pBlock->info.watermark = INT64_MIN;
882,349,893✔
404

405
  int32_t i = 0;
882,304,429✔
406
  SNode*  node = NULL;
882,304,429✔
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) {
882,632,767✔
421
    code = TSDB_CODE_INVALID_PARA;
×
422
    QUERY_CHECK_CODE(code, lino, _return);
×
423
  }
424

425
  return pBlock;
882,632,767✔
426
_return:
×
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) {
303,678,576✔
434
  SDataBlockInfo* pBlockInfo = &pDataBlock->info;
303,678,576✔
435

436
  for (int32_t i = 0; i < taosArrayGetSize(pMatchInfo->pList); ++i) {
1,346,742,964✔
437
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
1,067,682,877✔
438
    if (!pItem) {
1,067,600,979✔
439
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
440
      return terrno;
×
441
    }
442

443
    if (pItem->isPk) {
1,067,600,979✔
444
      SColumnInfoData* pInfoData = taosArrayGet(pDataBlock->pDataBlock, pItem->dstSlotId);
24,636,927✔
445
      if (!pInfoData) {
24,551,298✔
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;
24,551,298✔
450
      pBlockInfo->pks[1].type = pInfoData->info.type;
24,564,991✔
451

452
      // allocate enough buffer size, which is pInfoData->info.bytes
453
      if (IS_VAR_DATA_TYPE(pItem->dataType.type)) {
24,560,016✔
454
        pBlockInfo->pks[0].pData = taosMemoryCalloc(1, pInfoData->info.bytes);
7,522,069✔
455
        if (pBlockInfo->pks[0].pData == NULL) {
7,530,095✔
456
          return terrno;
×
457
        }
458

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

465
        pBlockInfo->pks[0].nData = pInfoData->info.bytes;
7,535,186✔
466
        pBlockInfo->pks[1].nData = pInfoData->info.bytes;
7,536,654✔
467
      }
468

469
      break;
24,559,413✔
470
    }
471
  }
472

473
  return TSDB_CODE_SUCCESS;
303,552,337✔
474
}
475

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

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

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

502
    STagVal tagVal = {0};
848,187✔
503
    tagVal.cid = pSColumnNode->colId;
848,187✔
504
    const char* p = mr->pAPI->extractTagVal(mr->me.ctbEntry.pTags, pSColumnNode->node.resType.type, &tagVal);
848,187✔
505
    if (p == NULL) {
848,187✔
506
      res->node.resType.type = TSDB_DATA_TYPE_NULL;
357✔
507
    } else if (pSColumnNode->node.resType.type == TSDB_DATA_TYPE_JSON) {
847,830✔
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)) {
847,830✔
515
      if (IS_STR_DATA_BLOB(pSColumnNode->node.resType.type)) {
846,402✔
516
        return TSDB_CODE_BLOB_NOT_SUPPORT_TAG;
×
517
      }
518

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

524
      if (IS_STR_DATA_BLOB(pSColumnNode->node.resType.type)) {
846,402✔
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);
846,402✔
529
        varDataSetLen(res->datum.p, tagVal.nData);
846,402✔
530
      }
531
    } else {
532
      int32_t code = nodesSetValueNodeValue(res, &(tagVal.i64));
1,428✔
533
      if (code != TSDB_CODE_SUCCESS) {
1,428✔
534
        return DEAL_RES_ERROR;
×
535
      }
536
    }
537
    nodesDestroyNode(*pNode);
848,187✔
538
    *pNode = (SNode*)res;
848,187✔
539
  } else if (isTbname) {
2,119,932✔
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,968,119✔
561
}
562

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

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

573
    return TSDB_CODE_SUCCESS;
×
574
  }
575

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

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

600
    return code;
×
601
  }
602

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

606
  nodesDestroyNode(pNew);
425,343✔
607
  return TSDB_CODE_SUCCESS;
425,343✔
608
}
609

610
static EDealRes getColumn(SNode** pNode, void* pContext) {
69,507,743✔
611
  tagFilterAssist* pData = (tagFilterAssist*)pContext;
69,507,743✔
612
  SColumnNode*     pSColumnNode = NULL;
69,507,743✔
613
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
69,514,884✔
614
    pSColumnNode = *(SColumnNode**)pNode;
24,065,979✔
615
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
45,468,786✔
616
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
1,353,298✔
617
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
1,353,960✔
618
      pData->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
864,867✔
619
      if (NULL == pSColumnNode) {
866,191✔
620
        return DEAL_RES_ERROR;
×
621
      }
622
      pSColumnNode->colId = -1;
866,191✔
623
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
866,191✔
624
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
866,191✔
625
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
865,762✔
626
      nodesDestroyNode(*pNode);
866,191✔
627
      *pNode = (SNode*)pSColumnNode;
866,191✔
628
    } else {
629
      return DEAL_RES_CONTINUE;
488,431✔
630
    }
631
  } else {
632
    return DEAL_RES_CONTINUE;
44,104,590✔
633
  }
634

635
  void* data = taosHashGet(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
24,931,154✔
636
  if (!data) {
24,928,142✔
637
    int32_t tempRes =
638
        taosHashPut(pData->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
21,499,994✔
639
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
21,507,490✔
640
      return DEAL_RES_ERROR;
×
641
    }
642
    pSColumnNode->slotId = pData->index++;
21,507,490✔
643
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
21,503,733✔
644
                         .type = pSColumnNode->node.resType.type,
21,489,467✔
645
                         .bytes = pSColumnNode->node.resType.bytes,
21,494,642✔
646
                         .pk = pSColumnNode->isPk};
21,485,765✔
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);
21,499,568✔
651
    if (!tmp) {
21,498,186✔
652
      return DEAL_RES_ERROR;
×
653
    }
654
  } else {
655
    SColumnNode* col = *(SColumnNode**)data;
3,428,148✔
656
    pSColumnNode->slotId = col->slotId;
3,427,298✔
657
  }
658

659
  return DEAL_RES_CONTINUE;
24,917,193✔
660
}
661

662
static int32_t createResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
20,045,684✔
663
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
20,045,684✔
664
  if (pColumnData == NULL) {
20,043,093✔
665
    return terrno;
×
666
  }
667

668
  pColumnData->info.type = pType->type;
20,043,093✔
669
  pColumnData->info.bytes = pType->bytes;
20,042,379✔
670
  pColumnData->info.scale = pType->scale;
20,040,881✔
671
  pColumnData->info.precision = pType->precision;
20,044,931✔
672

673
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
20,034,679✔
674
  if (code != TSDB_CODE_SUCCESS) {
20,032,952✔
675
    terrno = code;
×
676
    releaseColInfoData(pColumnData);
×
677
    return terrno;
×
678
  }
679

680
  pParam->columnData = pColumnData;
20,032,952✔
681
  pParam->colAlloced = true;
20,035,851✔
682
  return TSDB_CODE_SUCCESS;
20,035,425✔
683
}
684

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

693
void freeItem(void* p) {
233,264,357✔
694
  STUidTagInfo* pInfo = p;
233,264,357✔
695
  if (pInfo->pTagVal != NULL) {
233,264,357✔
696
    taosMemoryFree(pInfo->pTagVal);
232,835,378✔
697
  }
698
}
233,258,955✔
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) {
25,830✔
707
  STagDataEntry* p1 = (STagDataEntry*)a;
25,830✔
708
  STagDataEntry* p2 = (STagDataEntry*)b;
25,830✔
709
  return compareInt16Val(&p1->colId, &p2->colId);
25,830✔
710
}
711

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

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

730
    if (!pValueNode->isNull) {
25,830✔
731
      switch (pValueNode->node.resType.type) {
20,705✔
732
        case TSDB_DATA_TYPE_BOOL:
2,665✔
733
          (void)memcpy(
2,665✔
734
            pStart, &pValueNode->datum.b, bytes);
2,665✔
735
          pStart += bytes;
2,665✔
736
          break;
2,665✔
737
        case TSDB_DATA_TYPE_TINYINT:
9,225✔
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,225✔
743
            pStart, &pValueNode->datum.i, bytes);
9,225✔
744
          pStart += bytes;
9,225✔
745
          break;
9,225✔
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:
3,075✔
755
        case TSDB_DATA_TYPE_DOUBLE:
756
          (void)memcpy(
3,075✔
757
            pStart, &pValueNode->datum.d, bytes);
3,075✔
758
          pStart += bytes;
3,075✔
759
          break;
3,075✔
760
        case TSDB_DATA_TYPE_VARCHAR:
5,740✔
761
        case TSDB_DATA_TYPE_VARBINARY:
762
        case TSDB_DATA_TYPE_NCHAR:
763
          (void)memcpy(pStart,
5,740✔
764
            varDataVal(pValueNode->datum.p), varDataLen(pValueNode->datum.p));
5,740✔
765
          pStart += varDataLen(pValueNode->datum.p);
5,740✔
766
          break;
5,740✔
767
        default:
×
768
          qError("unsupported tag data type %d in tag filter optimization",
×
769
            pValueNode->node.resType.type);
770
          return TSDB_CODE_STREAM_INTERNAL_ERROR;
×
771
      }
772
    }
773
  }
774

775
  return TSDB_CODE_SUCCESS;
12,915✔
776
}
777

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

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

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

803
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
12,915✔
804
    extractTagDataEntry((SOperatorNode*)pTagCond, pIdWithVal);
×
805
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
12,915✔
806
    SNode* pChild = NULL;
12,915✔
807
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
38,745✔
808
      extractTagDataEntry((SOperatorNode*)pChild, pIdWithVal);
25,830✔
809
    }
810
  }
811

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

814
  return TSDB_CODE_SUCCESS;
12,915✔
815
}
816

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

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

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

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

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

851
static int32_t genTagFilterDigest(const SNode* pTagCond, T_MD5_CTX* pContext) {
40,258✔
852
  if (pTagCond == NULL) {
40,258✔
853
    return TSDB_CODE_SUCCESS;
38,058✔
854
  }
855

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

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

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

874
  taosMemoryFree(payload);
2,200✔
875
  return TSDB_CODE_SUCCESS;
2,200✔
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) {
19,502,659✔
906
  int32_t code = TSDB_CODE_SUCCESS;
19,502,659✔
907
  tagFilterAssist ctx = {0};
19,502,659✔
908
  ctx.colHash = taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
19,504,501✔
909
  if (ctx.colHash == NULL) {
19,503,788✔
910
    code = terrno;
×
911
    goto end;
×
912
  }
913

914
  ctx.index = 0;
19,503,788✔
915
  ctx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
19,503,788✔
916
  if (ctx.cInfoList == NULL) {
19,505,093✔
917
    code = terrno;
5,186✔
918
    goto end;
×
919
  }
920

921
  if (isList) {
19,499,907✔
922
    SNode* pNode = NULL;
5,139,851✔
923
    FOREACH(pNode, (SNodeList*)data) {
10,828,315✔
924
      nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
5,688,567✔
925
      if (TSDB_CODE_SUCCESS != ctx.code) {
5,689,060✔
926
        code = ctx.code;
×
927
        goto end;
×
928
      }
929
      REPLACE_NODE(pNode);
5,689,060✔
930
    }
931
  } else {
932
    SNode* pNode = (SNode*)data;
14,360,056✔
933
    nodesRewriteExprPostOrder(&pNode, getColumn, (void*)&ctx);
14,361,653✔
934
    if (TSDB_CODE_SUCCESS != ctx.code) {
14,363,284✔
935
      code = ctx.code;
×
936
      goto end;
×
937
    }
938
  }
939
  
940
  if (pColList != NULL) *pColList = ctx.cInfoList;
19,496,444✔
941
  ctx.cInfoList = NULL;
19,487,014✔
942

943
end:
19,492,616✔
944
  taosHashCleanup(ctx.colHash);
19,495,659✔
945
  taosArrayDestroy(ctx.cInfoList);
19,472,494✔
946
  return code;
19,477,373✔
947
}
948

949
static int32_t buildGroupInfo(SColumnInfoData* pValue, int32_t i, SArray* gInfo) {
808,643✔
950
  int32_t code = TSDB_CODE_SUCCESS;
808,643✔
951
  SStreamGroupValue* v = taosArrayReserve(gInfo, 1);
808,643✔
952
  if (v == NULL) {
808,643✔
953
    code = terrno;
×
954
    goto end;
×
955
  }
956
  if (colDataIsNull_s(pValue, i)) {
1,617,286✔
957
    v->isNull = true;
10,250✔
958
  } else {
959
    v->isNull = false;
798,393✔
960
    char* data = colDataGetData(pValue, i);
798,134✔
961
    if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
798,130✔
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)) {
797,646✔
981
      if (varDataTLen(data) > pValue->info.bytes) {
504,839✔
982
        code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
983
        goto end;
×
984
      }
985
      v->data.type = pValue->info.type;
505,068✔
986
      v->data.nData = varDataLen(data);
505,327✔
987
      v->data.pData = taosMemoryCalloc(1, varDataLen(data) + 1);
505,068✔
988
      if (v->data.pData == NULL) {
505,327✔
989
        code = terrno;
×
990
        goto end;
×
991
      }
992
      memcpy(v->data.pData, varDataVal(data), varDataLen(data));
505,327✔
993
      qDebug("buildGroupInfo:%d add var data type:%d, len:%d, data:%s", i, pValue->info.type, varDataLen(data), (char*)v->data.pData);
505,064✔
994
    } else if (pValue->info.type == TSDB_DATA_TYPE_DECIMAL) {  // reader todo decimal
293,066✔
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;
293,066✔
1006
      memcpy(&v->data.val, data, pValue->info.bytes);
293,066✔
1007
      qDebug("buildGroupInfo:%d add data type:%d, data:%"PRId64, i, pValue->info.type, v->data.val);
293,066✔
1008
    }
1009
  }
1010
end:
42,513✔
1011
  if (code != TSDB_CODE_SUCCESS) {
808,643✔
1012
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1013
    v->isNull = true;
×
1014
  }
1015
  return code;
808,643✔
1016
}
1017

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

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

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

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

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

1060
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
306,233✔
1061
    SColumnInfo* tmp = (SColumnInfo*)taosArrayGet(pColList, i);
202,903✔
1062
    if (tmp != NULL && tmp->colId == -1) {
202,903✔
1063
      tbNameIndex = i;
103,330✔
1064
    }
1065
  }
1066
  
1067
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
103,330✔
1068
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
103,330✔
1069
  taosArrayDestroy(pColList);
103,053✔
1070
  if (pResBlock == NULL) {
103,053✔
1071
    code = terrno;
318✔
1072
    goto end;
318✔
1073
  }
1074

1075
  pBlockList = taosArrayInit(2, POINTER_BYTES);
102,735✔
1076
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
103,012✔
1077

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

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

1084
  SNode* pNode = NULL;
103,012✔
1085
  FOREACH(pNode, group) {
305,116✔
1086
    SScalarParam output = {0};
202,104✔
1087

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

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

1107
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
202,004✔
1108
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
202,267✔
1109
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
202,267✔
1110
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
202,004✔
1111
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
202,004✔
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) {
202,004✔
1121
      releaseColInfoData(output.columnData);
×
1122
      goto end;
×
1123
    }
1124

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

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

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

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

1152
    int32_t ret = taosHashPut(groupIdMap, &info->uid, sizeof(info->uid), &gInfo, POINTER_BYTES);
354,892✔
1153
    if (ret != TSDB_CODE_SUCCESS) {
355,152✔
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);
355,152✔
1158
    gInfo = NULL;
355,152✔
1159
  }
1160

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

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

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

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

1191
  T_MD5_CTX context = {0};
5,036,521✔
1192
  if (tsTagFilterCache && groupIdMap == NULL && digest) {
5,036,521✔
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));
5,036,521✔
1218
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
5,035,925✔
1219

1220
  for (int32_t i = 0; i < rows; ++i) {
32,142,042✔
1221
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableListInfo->pTableList, i);
27,106,117✔
1222
    QUERY_CHECK_NULL(pkeyInfo, code, lino, end, terrno);
27,105,521✔
1223
    STUidTagInfo info = {.uid = pkeyInfo->uid};
27,105,521✔
1224
    void*        tmp = taosArrayPush(pUidTagList, &info);
27,105,028✔
1225
    QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
27,105,028✔
1226
  }
1227

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

1237
  SArray* pColList = NULL;
5,036,521✔
1238
  code = qGetColumnsFromNodeList(group, true, &pColList); 
5,036,521✔
1239
  if (code != TSDB_CODE_SUCCESS) {
5,035,127✔
1240
    goto end;
×
1241
  }
1242

1243
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
5,035,127✔
1244
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
5,035,127✔
1245
  taosArrayDestroy(pColList);
5,036,521✔
1246
  if (pResBlock == NULL) {
5,036,521✔
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);
5,036,521✔
1255
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
5,035,925✔
1256

1257
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
5,036,521✔
1258
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
5,036,521✔
1259

1260
  groupData = taosArrayInit(2, POINTER_BYTES);
5,036,521✔
1261
  QUERY_CHECK_NULL(groupData, code, lino, end, terrno);
5,036,521✔
1262

1263
  SNode* pNode = NULL;
5,036,521✔
1264
  FOREACH(pNode, group) {
10,521,617✔
1265
    SScalarParam output = {0};
5,486,094✔
1266

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

1292
    if (nodeType(pNode) == QUERY_NODE_COLUMN) {
5,486,018✔
1293
      SColumnNode*     pSColumnNode = (SColumnNode*)pNode;
5,343,696✔
1294
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, pSColumnNode->slotId);
5,343,696✔
1295
      QUERY_CHECK_NULL(pColInfo, code, lino, end, terrno);
5,342,719✔
1296
      code = colDataAssign(output.columnData, pColInfo, rows, NULL);
5,342,719✔
1297
    } else if (nodeType(pNode) == QUERY_NODE_VALUE) {
142,322✔
1298
      continue;
×
1299
    } else {
1300
      gTaskScalarExtra.pStreamInfo = NULL;
142,322✔
1301
      gTaskScalarExtra.pStreamRange = NULL;
142,322✔
1302
      code = scalarCalculate(pNode, pBlockList, &output, &gTaskScalarExtra);
142,322✔
1303
    }
1304

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

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

1314
  int32_t baseIdLen = gIdFromBaseId ? sizeof(int64_t) : 0;
5,035,027✔
1315
  int32_t keyLen = baseIdLen;
5,035,027✔
1316
  SNode*  node;
1317
  FOREACH(node, group) {
10,517,027✔
1318
    SExprNode* pExpr = (SExprNode*)node;
5,484,500✔
1319
    keyLen += pExpr->resType.bytes;
5,484,500✔
1320
  }
1321

1322
  int32_t nullFlagSize = sizeof(int8_t) * LIST_LENGTH(group);
5,035,166✔
1323
  keyLen += nullFlagSize;
5,035,762✔
1324

1325
  keyBuf = taosMemoryCalloc(1, keyLen);
5,035,762✔
1326
  if (keyBuf == NULL) {
5,034,407✔
1327
    code = terrno;
×
1328
    goto end;
×
1329
  }
1330

1331
  if (initRemainGroups) {
5,034,407✔
1332
    pTableListInfo->remainGroups =
2,284,879✔
1333
        taosHashInit(rows, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
2,282,356✔
1334
    if (pTableListInfo->remainGroups == NULL) {
2,284,879✔
1335
      code = terrno;
×
1336
      goto end;
×
1337
    }
1338
  }
1339

1340
  for (int i = 0; i < rows; i++) {
32,136,278✔
1341
    STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
27,101,689✔
1342
    QUERY_CHECK_NULL(info, code, lino, end, terrno);
27,099,059✔
1343

1344
    if (groupIdMap != NULL){
27,099,059✔
1345
      gInfo = taosArrayInit(taosArrayGetSize(groupData), sizeof(SStreamGroupValue));
99,209✔
1346
    }
1347

1348
    if (baseIdLen > 0) {
27,099,059✔
1349
      *(uint64_t*)keyBuf = info->baseGId;
×
1350
    }
1351
    char* isNull = (char*)keyBuf + baseIdLen;
27,099,059✔
1352
    char* pStart = (char*)keyBuf + baseIdLen + sizeof(int8_t) * LIST_LENGTH(group);
27,097,826✔
1353
    for (int j = 0; j < taosArrayGetSize(groupData); j++) {
56,120,645✔
1354
      SColumnInfoData* pValue = (SColumnInfoData*)taosArrayGetP(groupData, j);
29,018,974✔
1355

1356
      if (groupIdMap != NULL && gInfo != NULL) {
29,020,645✔
1357
        int32_t ret = buildGroupInfo(pValue, i, gInfo);
153,553✔
1358
        if (ret != TSDB_CODE_SUCCESS) {
153,553✔
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)) {
58,039,318✔
1366
        isNull[j] = 1;
118,671✔
1367
      } else {
1368
        isNull[j] = 0;
28,900,002✔
1369
        char* data = colDataGetData(pValue, i);
28,900,709✔
1370
        if (pValue->info.type == TSDB_DATA_TYPE_JSON) {
28,902,516✔
1371
          // if (tTagIsJson(data)) {
1372
          //   code = TSDB_CODE_QRY_JSON_IN_GROUP_ERROR;
1373
          //   goto end;
1374
          // }
1375
          if (tTagIsJsonNull(data)) {
89,161✔
1376
            isNull[j] = 1;
×
1377
            continue;
×
1378
          }
1379
          int32_t len = getJsonValueLen(data);
89,161✔
1380
          memcpy(pStart, data, len);
89,161✔
1381
          pStart += len;
89,161✔
1382
        } else if (IS_VAR_DATA_TYPE(pValue->info.type)) {
28,813,469✔
1383
          if (IS_STR_DATA_BLOB(pValue->info.type)) {
24,597,305✔
1384
            if (blobDataTLen(data) > TSDB_MAX_BLOB_LEN) {
2,114✔
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) {
24,598,680✔
1392
              code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1393
              goto end;
×
1394
            }
1395
            memcpy(pStart, data, varDataTLen(data));
24,598,843✔
1396
            pStart += varDataTLen(data);
24,598,843✔
1397
          }
1398
        } else {
1399
          memcpy(pStart, data, pValue->info.bytes);
4,214,623✔
1400
          pStart += pValue->info.bytes;
4,213,561✔
1401
        }
1402
      }
1403
    }
1404

1405
    int32_t len = (int32_t)(pStart - (char*)keyBuf);
27,100,978✔
1406
    info->groupId = calcGroupId(keyBuf, len);
27,100,978✔
1407
    if (groupIdMap != NULL && gInfo != NULL) {
27,098,214✔
1408
      int32_t ret = taosHashPut(groupIdMap, &info->groupId, sizeof(info->groupId), &gInfo, POINTER_BYTES);
99,209✔
1409
      if (ret != TSDB_CODE_SUCCESS) {
99,209✔
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);
99,209✔
1414
      gInfo = NULL;
99,209✔
1415
    }
1416
    if (initRemainGroups) {
27,098,214✔
1417
      // groupId ~ table uid
1418
      code = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId), &(info->uid),
13,023,638✔
1419
                         sizeof(info->uid));
1420
      if (code == TSDB_CODE_DUP_KEY) {
13,025,837✔
1421
        code = TSDB_CODE_SUCCESS;
887,712✔
1422
      }
1423
      QUERY_CHECK_CODE(code, lino, end);
13,025,837✔
1424
    }
1425
  }
1426

1427
  if (tsTagFilterCache && groupIdMap == NULL && digest) {
5,034,589✔
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:
5,034,589✔
1441
  taosMemoryFreeClear(keyBuf);
5,035,418✔
1442
  blockDataDestroy(pResBlock);
5,036,074✔
1443
  taosArrayDestroy(pBlockList);
5,035,036✔
1444
  taosArrayDestroyEx(pUidTagList, freeItem);
5,035,498✔
1445
  taosArrayDestroyP(groupData, releaseColInfoData);
5,035,478✔
1446
  taosArrayDestroyEx(gInfo, tDestroySStreamGroupValue);
5,035,395✔
1447

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

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

1458
  int32_t ret = strcmp(pName1, pName2);
814,192✔
1459
  if (ret == 0) {
814,192✔
1460
    return 0;
16,509✔
1461
  } else {
1462
    return (ret > 0) ? 1 : -1;
797,683✔
1463
  }
1464
}
1465

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

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

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

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

1489
  size_t numOfTables = taosArrayGetSize(pTbList);
439,068✔
1490

1491
  // order the name
1492
  taosArraySort(pTbList, nameComparFn);
439,068✔
1493

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

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

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

1514
_end:
439,068✔
1515
  taosArrayDestroy(pTbList);
439,068✔
1516
  if (code != TSDB_CODE_SUCCESS) {
439,068✔
1517
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1518
    return NULL;
×
1519
  }
1520
  return pNewList;
439,068✔
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) {
43,680✔
1535
  STUidTagInfo* p1 = (STUidTagInfo*)a;
43,680✔
1536
  STUidTagInfo* p2 = (STUidTagInfo*)b;
43,680✔
1537

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

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

1545
static FilterCondType checkTagCond(SNode* cond) {
17,069,145✔
1546
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
17,069,145✔
1547
    return FILTER_NO_LOGIC;
12,200,924✔
1548
  }
1549
  if (nodeType(cond) == QUERY_NODE_LOGIC_CONDITION && ((SLogicConditionNode*)cond)->condType == LOGIC_COND_TYPE_AND) {
4,870,954✔
1550
    return FILTER_AND;
4,237,648✔
1551
  }
1552
  return FILTER_OTHER;
629,994✔
1553
}
1554

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

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

1563
  int32_t len = LIST_LENGTH(pList);
4,246,428✔
1564
  if (len <= 0) {
4,247,345✔
1565
    return -1;
×
1566
  }
1567

1568
  SListCell* cell = pList->pHead;
4,247,345✔
1569
  for (int i = 0; i < len; i++) {
13,177,547✔
1570
    if (cell == NULL) break;
8,938,825✔
1571
    if (nodeType(cell->pNode) == QUERY_NODE_OPERATOR && optimizeTbnameInCondImpl(pVnode, list, cell->pNode, pAPI, suid) == 0) {
8,938,825✔
1572
      return 0;
8,940✔
1573
    }
1574
    cell = cell->pNext;
8,925,487✔
1575
  }
1576
  return -1;
4,238,722✔
1577
}
1578

1579
static int32_t optimizeTbnameInCond(void* pVnode, int64_t suid, SArray* list, SNode* cond, SStorageAPI* pAPI, int64_t version) {
17,514,382✔
1580
  int32_t code = 0;
17,514,382✔
1581
  int32_t lino = 0;
17,514,382✔
1582
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
17,514,382✔
1583
    code = optimizeTbnameInCondImpl(pVnode, list, cond, pAPI, suid);
12,630,998✔
1584
  } else {
1585
    code = doInWithAnd(cond, pVnode, list, pAPI, suid);
4,884,947✔
1586
    QUERY_CHECK_CODE(code, lino, end);
4,876,788✔
1587
    code = pAPI->metaFn.getTableTagsByUidVersion(pVnode, suid, list, version);
7,506✔
1588
  }
1589
  
1590
end:
17,506,802✔
1591
  return code;
17,506,802✔
1592
}
1593

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

1601
  size_t numOfExisted = taosArrayGetSize(pExistedUidList);  // len > 0 means there already have uids
439,068✔
1602
  if (numOfExisted > 0) {
439,068✔
1603
    uHash = taosHashInit(numOfExisted / 0.7, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
2,384✔
1604
    if (!uHash) {
2,384✔
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,382,808✔
1611
      STUidTagInfo* pTInfo = taosArrayGet(pExistedUidList, i);
2,380,424✔
1612
      if (!pTInfo) {
2,380,424✔
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,380,424✔
1618
      if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_DUP_KEY) {
2,380,424✔
1619
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
1620
        goto end;
×
1621
      }
1622
    }
1623
  }
1624
  taosArrayClear(pExistedUidList);
439,068✔
1625

1626
  for (int i = 0; i < numOfTables; i++) {
1,212,663✔
1627
    char* name = taosArrayGetP(pTbList, i);
773,595✔
1628
    uint64_t uid = 0, csuid = 0;
773,595✔
1629
    if (pStoreAPI->metaFn.getTableUidByName(pVnode, name, &uid) != 0) {
773,595✔
1630
      continue;
326,155✔
1631
    }
1632

1633
    ETableType tbType = TSDB_TABLE_MAX;
447,440✔
1634
    if (pStoreAPI->metaFn.getTableTypeSuidByName(pVnode, name, &tbType, &csuid) == 0 &&
447,440✔
1635
        (tbType == TSDB_CHILD_TABLE || tbType == TSDB_VIRTUAL_CHILD_TABLE) &&
447,440✔
1636
        csuid == suid) {
439,845✔
1637
      if (NULL == uHash || taosHashGet(uHash, &uid, sizeof(uid)) != NULL) {
437,341✔
1638
        STUidTagInfo s = {.uid = uid, .name = name, .pTagVal = NULL};
436,149✔
1639
        void*        tmp = taosArrayPush(pExistedUidList, &s);
436,149✔
1640
        if (!tmp) {
436,149✔
1641
          code = terrno;
×
1642
          goto end;
×
1643
        }
1644
      }
1645
    } 
1646
  }
1647
  taosArraySort(pExistedUidList, filterTableInfoCompare);
439,068✔
1648
  taosArrayRemoveDuplicate(pExistedUidList, filterTableInfoCompare, NULL);
439,068✔
1649
end:
439,068✔
1650
  taosArrayDestroy(pTbList);
439,068✔
1651
  taosHashCleanup(uHash);
439,068✔
1652
  return code;
439,068✔
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,
21,567,915✔
1657
                                        uint64_t suid) {
1658
  SOperatorNode* pNode = (SOperatorNode*)pTagCond;
21,567,915✔
1659
  if (pNode->opType != OP_TYPE_IN) {
21,567,915✔
1660
    return -1;
20,524,824✔
1661
  }
1662

1663
  if ((pNode->pLeft != NULL && ((nodeType(pNode->pLeft) == QUERY_NODE_FUNCTION &&
1,041,371✔
1664
                                 ((SFunctionNode*)pNode->pLeft)->funcType == FUNCTION_TYPE_TBNAME)) ||
441,524✔
1665
       (nodeType(pNode->pLeft) == QUERY_NODE_COLUMN && ((SColumnNode*)pNode->pLeft)->colType == COLUMN_TYPE_TBNAME)) &&
599,263✔
1666
      (pNode->pRight != NULL && nodeType(pNode->pRight) == QUERY_NODE_NODE_LIST)) {
442,692✔
1667
    SNodeListNode* pList = (SNodeListNode*)pNode->pRight;
439,068✔
1668

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

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

1677
  return -1;
602,887✔
1678
}
1679

1680
SSDataBlock* createTagValBlockForFilter(SArray* pColList, int32_t numOfTables, SArray* pUidTagList, void* pVnode,
21,455,742✔
1681
                                        SStorageAPI* pStorageAPI) {
1682
  int32_t      code = TSDB_CODE_SUCCESS;
21,455,742✔
1683
  int32_t      lino = 0;
21,455,742✔
1684
  SSDataBlock* pResBlock = NULL;
21,455,742✔
1685
  code = createDataBlock(&pResBlock);
21,456,901✔
1686
  QUERY_CHECK_CODE(code, lino, _end);
21,446,662✔
1687

1688
  for (int32_t i = 0; i < taosArrayGetSize(pColList); ++i) {
44,900,266✔
1689
    SColumnInfoData colInfo = {0};
23,464,690✔
1690
    void*           tmp = taosArrayGet(pColList, i);
23,463,685✔
1691
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
23,438,869✔
1692
    colInfo.info = *(SColumnInfo*)tmp;
23,438,869✔
1693
    code = blockDataAppendColInfo(pResBlock, &colInfo);
23,449,667✔
1694
    QUERY_CHECK_CODE(code, lino, _end);
23,456,090✔
1695
  }
1696

1697
  code = blockDataEnsureCapacity(pResBlock, numOfTables);
21,443,717✔
1698
  if (code != TSDB_CODE_SUCCESS) {
21,449,575✔
1699
    terrno = code;
×
1700
    blockDataDestroy(pResBlock);
×
1701
    return NULL;
×
1702
  }
1703

1704
  pResBlock->info.rows = numOfTables;
21,449,575✔
1705

1706
  int32_t numOfCols = taosArrayGetSize(pResBlock->pDataBlock);
21,442,110✔
1707

1708
  for (int32_t i = 0; i < numOfTables; i++) {
257,183,223✔
1709
    STUidTagInfo* p1 = taosArrayGet(pUidTagList, i);
235,716,762✔
1710
    QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
235,724,362✔
1711

1712
    for (int32_t j = 0; j < numOfCols; j++) {
483,292,392✔
1713
      SColumnInfoData* pColInfo = (SColumnInfoData*)taosArrayGet(pResBlock->pDataBlock, j);
247,545,171✔
1714
      QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
247,499,173✔
1715

1716
      if (pColInfo->info.colId == -1) {  // tbname
247,499,173✔
1717
        char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
8,379,135✔
1718
        if (p1->name != NULL) {
8,382,138✔
1719
          STR_TO_VARSTR(str, p1->name);
6,556✔
1720
        } else {  // name is not retrieved during filter
1721
          code = pStorageAPI->metaFn.getTableNameByUid(pVnode, p1->uid, str);
8,373,447✔
1722
          QUERY_CHECK_CODE(code, lino, _end);
8,368,815✔
1723
        }
1724

1725
        code = colDataSetVal(pColInfo, i, str, false);
8,375,053✔
1726
        QUERY_CHECK_CODE(code, lino, _end);
8,382,328✔
1727
#if TAG_FILTER_DEBUG
1728
        qDebug("tagfilter uid:%ld, tbname:%s", *uid, str + 2);
1729
#endif
1730
      } else {
1731
        STagVal tagVal = {0};
239,117,753✔
1732
        tagVal.cid = pColInfo->info.colId;
239,131,570✔
1733
        if (p1->pTagVal == NULL) {
239,128,771✔
1734
          colDataSetNULL(pColInfo, i);
5,125✔
1735
        } else {
1736
          const char* p = pStorageAPI->metaFn.extractTagVal(p1->pTagVal, pColInfo->info.type, &tagVal);
239,108,781✔
1737

1738
          if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
239,153,470✔
1739
            colDataSetNULL(pColInfo, i);
2,481,043✔
1740
          } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
236,674,901✔
1741
            code = colDataSetVal(pColInfo, i, p, false);
704,903✔
1742
            QUERY_CHECK_CODE(code, lino, _end);
704,903✔
1743
          } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
390,192,720✔
1744
            if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
154,241,829✔
1745
              QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
1746
            }
1747
            char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
154,245,048✔
1748
            QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
154,260,338✔
1749
            varDataSetLen(tmp, tagVal.nData);
154,260,338✔
1750
            memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
154,254,533✔
1751
            code = colDataSetVal(pColInfo, i, tmp, false);
154,257,127✔
1752
#if TAG_FILTER_DEBUG
1753
            qDebug("tagfilter varch:%s", tmp + 2);
1754
#endif
1755
            taosMemoryFree(tmp);
154,241,290✔
1756
            QUERY_CHECK_CODE(code, lino, _end);
154,264,597✔
1757
          } else {
1758
            code = colDataSetVal(pColInfo, i, (const char*)&tagVal.i64, false);
81,693,987✔
1759
            QUERY_CHECK_CODE(code, lino, _end);
81,719,823✔
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:
21,470,483✔
1774
  if (code != TSDB_CODE_SUCCESS) {
21,467,363✔
1775
    blockDataDestroy(pResBlock);
6,581✔
1776
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
318✔
1777
    terrno = code;
318✔
1778
    return NULL;
318✔
1779
  }
1780
  return pResBlock;
21,460,782✔
1781
}
1782

1783
static int32_t doSetQualifiedUid(SArray* pUidList, const SArray* pUidTagList, SColumnInfoData* colData) {
17,482,380✔
1784
  taosArrayClear(pUidList);
17,482,380✔
1785

1786
  int32_t       numOfTables = taosArrayGetSize(pUidTagList);
17,475,058✔
1787
  bool* pResultList = colData == NULL ? NULL : (bool*)colData->pData;
17,480,296✔
1788
  for (int32_t i = 0; i < numOfTables; ++i) {
223,269,918✔
1789
    if (pResultList != NULL && !pResultList[i]) {
205,758,616✔
1790
       continue;
117,482,495✔
1791
    }
1792
    STUidTagInfo* tmpTag = (STUidTagInfo*)taosArrayGet(pUidTagList, i);
88,303,674✔
1793
    if (!tmpTag) {
88,287,599✔
1794
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1795
      return terrno;
×
1796
    }
1797
    uint64_t uid = tmpTag->uid;
88,287,599✔
1798
    qDebug("tagfilter get uid:%" PRId64, uid);
88,294,282✔
1799

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

1806
  return TSDB_CODE_SUCCESS;
17,511,302✔
1807
}
1808

1809
static int32_t copyExistedUids(SArray* pUidTagList, const SArray* pUidList) {
17,513,961✔
1810
  int32_t code = TSDB_CODE_SUCCESS;
17,513,961✔
1811
  int32_t numOfExisted = taosArrayGetSize(pUidList);
17,513,961✔
1812
  if (numOfExisted == 0) {
17,514,837✔
1813
    return code;
14,453,754✔
1814
  }
1815

1816
  for (int32_t i = 0; i < numOfExisted; ++i) {
34,577,616✔
1817
    uint64_t* uid = taosArrayGet(pUidList, i);
31,516,111✔
1818
    if (!uid) {
31,516,111✔
1819
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1820
      return terrno;
×
1821
    }
1822
    STUidTagInfo info = {.uid = *uid};
31,516,111✔
1823
    void*        tmp = taosArrayPush(pUidTagList, &info);
31,516,533✔
1824
    if (!tmp) {
31,516,533✔
1825
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1826
      return code;
×
1827
    }
1828
  }
1829
  return code;
3,061,505✔
1830
}
1831

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

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

1846
  return DEAL_RES_CONTINUE;
5,942✔
1847
}
1848

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

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

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

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

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

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

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

1899
int32_t doFilterByTagCond(int64_t suid, SArray* pUidList, int64_t version, SNode* pTagCond, void* pVnode,
306,783,920✔
1900
                                 SIdxFltStatus status, SStorageAPI* pAPI, void* pStreamInfo) {
1901
  if (pTagCond == NULL) {
306,783,920✔
1902
    return TSDB_CODE_SUCCESS;
289,282,323✔
1903
  }
1904

1905
  terrno = TSDB_CODE_SUCCESS;
17,501,597✔
1906

1907
  int32_t      lino = 0;
17,516,542✔
1908
  int32_t      code = TSDB_CODE_SUCCESS;
17,516,542✔
1909
  SArray*      pBlockList = NULL;
17,516,542✔
1910
  SSDataBlock* pResBlock = NULL;
17,516,542✔
1911
  SScalarParam output = {0};
17,509,962✔
1912
  SArray*      pUidTagList = NULL;
17,510,967✔
1913

1914
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
17,510,967✔
1915

1916
  pUidTagList = taosArrayInit(10, sizeof(STUidTagInfo));
17,513,251✔
1917
  QUERY_CHECK_NULL(pUidTagList, code, lino, end, terrno);
17,505,216✔
1918

1919
  // pUidList size is 0 if tagIndex is NULL and in query condition
1920
  code = copyExistedUids(pUidTagList, pUidList);
17,505,216✔
1921
  QUERY_CHECK_CODE(code, lino, end);
17,507,577✔
1922

1923
  // Narrow down the scope of the tablelist set if there is tbname in condition and And Logical operator
1924
  code = optimizeTbnameInCond(pVnode, suid, pUidTagList, pTagCond, pAPI, version);
17,507,577✔
1925
  if (code == 0) {
17,506,802✔
1926
    if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
439,068✔
1927
      goto end;
430,128✔
1928
    }
1929
  } else {
1930
    qDebug("pUidTagList size:%d", (int32_t)taosArrayGetSize(pUidTagList));
17,067,734✔
1931

1932
    FilterCondType condType = checkTagCond(pTagCond);
17,067,734✔
1933
    if (((condType == FILTER_NO_LOGIC || condType == FILTER_AND) && status != SFLT_NOT_INDEX) || // (super table) use tagIndex and operator is and
17,064,614✔
1934
        (status == SFLT_NOT_INDEX && taosArrayGetSize(pUidTagList) > 0)) {                       // (child table with tagCond)
17,070,338✔
1935
      code = pAPI->metaFn.getTableTagsByUidVersion(pVnode, suid, pUidTagList, version);
3,059,392✔
1936
    } else {
1937
      taosArrayClearEx(pUidTagList, freeItem);       // clear tablelist if using tagIndex and or condition
14,005,961✔
1938
      code = pAPI->metaFn.getTableTags(pVnode, suid, pUidTagList);
14,013,094✔
1939
    }
1940
    if (code != TSDB_CODE_SUCCESS) {
17,071,977✔
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));
17,080,917✔
1948

1949
  int32_t numOfTables = taosArrayGetSize(pUidTagList);
17,082,908✔
1950
  if (numOfTables == 0) {
17,083,909✔
1951
    goto end;
2,721,331✔
1952
  }
1953

1954
  SArray* pColList = NULL;
14,362,578✔
1955
  code = qGetColumnsFromNodeList(pTagCond, false, &pColList); 
14,364,658✔
1956
  if (code != TSDB_CODE_SUCCESS) {
14,344,066✔
1957
    goto end;
×
1958
  }
1959
  pResBlock = createTagValBlockForFilter(pColList, numOfTables, pUidTagList, pVnode, pAPI);
14,344,066✔
1960
  taosArrayDestroy(pColList);
14,356,898✔
1961
  if (pResBlock == NULL) {
14,357,136✔
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);
14,357,136✔
1971
  QUERY_CHECK_NULL(pBlockList, code, lino, end, terrno);
14,360,545✔
1972

1973
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
14,358,305✔
1974
  QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
14,358,305✔
1975

1976
  code = createResultData(&type, numOfTables, &output);
14,358,305✔
1977
  if (code != TSDB_CODE_SUCCESS) {
14,345,916✔
1978
    terrno = code;
×
1979
    QUERY_CHECK_CODE(code, lino, end);
×
1980
  }
1981

1982
  gTaskScalarExtra.pStreamInfo = pStreamInfo;
14,345,916✔
1983
  gTaskScalarExtra.pStreamRange = NULL;
14,345,916✔
1984
  code = scalarCalculate(pTagCond, pBlockList, &output, &gTaskScalarExtra);
14,348,076✔
1985
  if (code != TSDB_CODE_SUCCESS) {
14,332,546✔
1986
    qError("failed to calculate scalar, reason:%s", tstrerror(code));
1,094✔
1987
    terrno = code;
1,094✔
1988
    QUERY_CHECK_CODE(code, lino, end);
1,094✔
1989
  }
1990

1991
end:
17,484,005✔
1992
  if (code == 0) {
17,495,002✔
1993
    code = doSetQualifiedUid(pUidList, pUidTagList, output.columnData);
17,495,212✔
1994
    lino = __LINE__;
17,509,312✔
1995
  }
1996
  
1997
  if (code != 0) {
17,509,102✔
1998
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1,094✔
1999
  }
2000
  blockDataDestroy(pResBlock);
17,509,102✔
2001
  taosArrayDestroy(pBlockList);
17,499,421✔
2002
  taosArrayDestroyEx(pUidTagList, freeItem);
17,493,696✔
2003

2004
  colDataDestroy(output.columnData);
17,507,411✔
2005
  taosMemoryFreeClear(output.columnData);
17,509,611✔
2006
  return code;
17,507,122✔
2007
}
2008

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

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

2037
  return DEAL_RES_CONTINUE;
16,121✔
2038
}
2039

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

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

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

2063
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR) {
12,915✔
2064
    extractTagColId((SOperatorNode*)pTagCond, *pTagColIds);
×
2065
  } else if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION) {
12,915✔
2066
    SNode* pChild = NULL;
12,915✔
2067
    FOREACH(pChild, ((SLogicConditionNode*)pTagCond)->pParameterList) {
38,745✔
2068
      extractTagColId((SOperatorNode*)pChild, *pTagColIds);
25,830✔
2069
    }
2070
  }
2071

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

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

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

2099
static EDealRes canOptimizeTagCondFilter(SNode* pTagCond, void* pContext) {
120,130✔
2100
  if (NULL == pTagCond) {
120,130✔
2101
    *(bool*)pContext = false;
×
2102
    return DEAL_RES_END;
×
2103
  }
2104
  if (nodeType(pTagCond) == QUERY_NODE_VALUE ||
120,130✔
2105
    nodeType(pTagCond) == QUERY_NODE_COLUMN) {
79,745✔
2106
    return DEAL_RES_CONTINUE;
66,215✔
2107
  }
2108
  if (nodeType(pTagCond) == QUERY_NODE_OPERATOR &&
53,915✔
2109
    ((SOperatorNode*)pTagCond)->opType == OP_TYPE_EQUAL) {
26,445✔
2110
    return DEAL_RES_CONTINUE;
25,830✔
2111
  }
2112
  if (nodeType(pTagCond) == QUERY_NODE_LOGIC_CONDITION &&
28,085✔
2113
    ((SLogicConditionNode*)pTagCond)->condType == LOGIC_COND_TYPE_AND) {
12,915✔
2114
    return DEAL_RES_CONTINUE;
12,915✔
2115
  }
2116
  if (nodeType(pTagCond) == QUERY_NODE_FUNCTION &&
29,725✔
2117
    fmIsStreamPesudoColVal(((SFunctionNode*)pTagCond)->funcId)) {
14,555✔
2118
    return DEAL_RES_CONTINUE;
14,555✔
2119
  }
2120
  *(bool*)pContext = false;
615✔
2121
  return DEAL_RES_END;
615✔
2122
}
2123

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

2130
  pListInfo->idInfo.suid = pScanNode->suid;
306,647,909✔
2131
  pListInfo->idInfo.tableType = pScanNode->tableType;
306,570,845✔
2132

2133
  SArray* pUidList = taosArrayInit(8, sizeof(uint64_t));
306,488,477✔
2134
  QUERY_CHECK_NULL(pUidList, code, lino, end, terrno);
306,570,360✔
2135

2136
  char*   pTagCondKey = NULL;
306,570,360✔
2137
  int32_t tagCondKeyLen;
306,576,211✔
2138
  SArray* pTagColIds = NULL;
306,556,014✔
2139
  char*   pPayload = NULL;
306,563,296✔
2140
  SIdxFltStatus status = SFLT_NOT_INDEX;
306,563,296✔
2141

2142
  qTrace("getTableList called, suid:%" PRIu64
306,563,296✔
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) {
306,571,340✔
2146
    pListInfo->idInfo.uid = pScanNode->uid;
165,098,472✔
2147
    if (pStorageAPI->metaFn.isTableExisted(pVnode, pScanNode->uid)) {
165,127,374✔
2148
      void* tmp = taosArrayPush(pUidList, &pScanNode->uid);
165,115,438✔
2149
      QUERY_CHECK_NULL(tmp, code, lino, end, terrno);
165,118,656✔
2150
    }
2151
    code = doFilterByTagCond(pScanNode->suid, pUidList, -1, pTagCond, pVnode, status, pStorageAPI, pStreamInfo);
165,174,579✔
2152
    QUERY_CHECK_CODE(code, lino, end);
165,165,479✔
2153
  } else {
2154
    bool      isStream = (pStreamInfo != NULL);
141,618,153✔
2155
    bool      hasTagCond = (pTagCond != NULL);
141,618,153✔
2156
    bool      canCacheTagEqCondFilter = false;
141,618,153✔
2157
    T_MD5_CTX context = {0};
141,513,971✔
2158

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

2163
    bool acquired = false;
141,450,691✔
2164
    // first, check whether we can use stable tag filter cache
2165
    if (tsStableTagFilterCache && isStream && hasTagCond) {
141,498,508✔
2166
      canCacheTagEqCondFilter = true;
13,530✔
2167
      nodesWalkExpr(pTagCond, canOptimizeTagCondFilter,
13,530✔
2168
        (void*)&canCacheTagEqCondFilter);
2169
    }
2170
    if (canCacheTagEqCondFilter) {
141,243,474✔
2171
      qDebug("%s, stable tag filter condition can be optimized", idstr);
12,915✔
2172
      if (((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
12,915✔
2173
        SNode* tmp = NULL;
12,915✔
2174
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
12,915✔
2175
        QUERY_CHECK_CODE(code, lino, end);
12,915✔
2176

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

2191
      code = buildTagCondKey(
12,915✔
2192
        pTagCond, &pTagCondKey, &tagCondKeyLen, &pTagColIds);
2193
      QUERY_CHECK_CODE(code, lino, end);
12,915✔
2194
      code = pStorageAPI->metaFn.getStableCachedTableList(
12,915✔
2195
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
12,915✔
2196
        context.digest, tListLen(context.digest), pUidList, &acquired);
2197
      QUERY_CHECK_CODE(code, lino, end);
12,915✔
2198
    } else if (tsTagFilterCache) {
141,230,559✔
2199
      // second, try to use normal tag filter cache
2200
      qDebug("%s using normal tag filter cache", idstr);
40,258✔
2201
      if (pStreamInfo != NULL && ((SStreamRuntimeFuncInfo*)pStreamInfo)->hasPlaceHolder) {
41,824✔
2202
        SNode* tmp = NULL;
1,566✔
2203
        code = nodesCloneNode((SNode*)pTagCond, &tmp);
1,566✔
2204
        QUERY_CHECK_CODE(code, lino, end);
1,566✔
2205

2206
        PlaceHolderContext ctx = {.code = TSDB_CODE_SUCCESS, .pStreamRuntimeInfo = (SStreamRuntimeFuncInfo*)pStreamInfo};
1,566✔
2207
        nodesRewriteExpr(&tmp, replacePlaceHolderColumn, (void*)&ctx);
1,566✔
2208
        if (TSDB_CODE_SUCCESS != ctx.code) {
1,566✔
2209
          nodesDestroyNode(tmp);
×
2210
          code = ctx.code;
×
2211
          goto end;
×
2212
        }
2213
        code = genTagFilterDigest(tmp, &context);
1,566✔
2214
        nodesDestroyNode(tmp);
1,566✔
2215
      } else {
2216
        code = genTagFilterDigest(pTagCond, &context);
38,692✔
2217
      }
2218
      // try to retrieve the result from meta cache
2219
      QUERY_CHECK_CODE(code, lino, end);      
40,258✔
2220
      code = pStorageAPI->metaFn.getCachedTableList(
40,258✔
2221
        pVnode, pScanNode->suid, context.digest,
40,258✔
2222
        tListLen(context.digest), pUidList, &acquired);
2223
      QUERY_CHECK_CODE(code, lino, end);
278,435✔
2224
    }
2225
    if (acquired) {
141,481,651✔
2226
      taosArrayDestroy(pTagColIds);
37,815✔
2227
      pTagColIds = NULL;
37,815✔
2228
      
2229
      if (digest != NULL) {
37,815✔
2230
        digest[0] = 1;
37,815✔
2231
        memcpy(digest + 1, context.digest, tListLen(context.digest));
37,815✔
2232
      }
2233
      
2234
      qDebug("suid:%" PRIu64 ", %s retrieve table uid list from cache, numOfTables:%d", 
37,815✔
2235
        pScanNode->suid, idstr, (int32_t)taosArrayGetSize(pUidList));
2236
      goto end;
37,815✔
2237
    } else {
2238
      qDebug("suid:%" PRIu64 
141,443,836✔
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
141,548,739✔
2243
      code = pStorageAPI->metaFn.getChildTableList(pVnode, pScanNode->suid, pUidList);
124,411,531✔
2244
      QUERY_CHECK_CODE(code, lino, end);
124,366,992✔
2245
      qTrace("no tag filter, get all child tables, numOfTables:%d", (int32_t)taosArrayGetSize(pUidList));
124,366,992✔
2246
    } else {
2247
      SIdxFltStatus status = SFLT_NOT_INDEX;
17,137,208✔
2248
      if (pTagIndexCond) {
17,138,471✔
2249
        void* pIndex = pStorageAPI->metaFn.getInvertIndex(pVnode);
5,058,932✔
2250

2251
        SIndexMetaArg metaArg = {.metaEx = pVnode,
5,058,958✔
2252
                                 .idx = pStorageAPI->metaFn.storeGetIndexInfo(pVnode),
5,058,932✔
2253
                                 .ivtIdx = pIndex,
2254
                                 .suid = pScanNode->uid};
5,058,932✔
2255
        code = doFilterTag(pTagIndexCond, &metaArg, pUidList, &status, &pStorageAPI->metaFilter);
5,058,932✔
2256
        if (code != 0 || status == SFLT_NOT_INDEX) {  // temporarily disable it for performance sake
5,050,902✔
2257
          qDebug("failed to get tableIds from index, suid:%" PRIu64 ", uidListSize:%d", pScanNode->uid, (int32_t)taosArrayGetSize(pUidList));
1,381,858✔
2258
        } else {
2259
          qDebug("succ to get filter result, table num: %d", (int)taosArrayGetSize(pUidList));
3,669,044✔
2260
        }
2261
      }
2262
    }
2263
    qTrace("after index filter, pTagCond:%p uidListSize:%d", pTagCond, (int32_t)taosArrayGetSize(pUidList));
141,522,246✔
2264
    code = doFilterByTagCond(pScanNode->suid, pUidList, -1, pTagCond, pVnode, status, pStorageAPI, pStreamInfo);
141,522,246✔
2265
    QUERY_CHECK_CODE(code, lino, end);
141,520,597✔
2266

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

2270
    if (canCacheTagEqCondFilter) {
141,490,847✔
2271
      qInfo("%s, suid:%" PRIu64 ", add uid list to stable tag filter cache, "
5,740✔
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,740✔
2277
        pVnode, pScanNode->suid, pTagCondKey, tagCondKeyLen,
2278
        context.digest, tListLen(context.digest),
2279
        pUidList, &pTagColIds);
2280
      QUERY_CHECK_CODE(code, lino, end);
5,740✔
2281

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

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

2302
      code = pStorageAPI->metaFn.putCachedTableList(pVnode, pScanNode->suid,
9,618✔
2303
                                                    context.digest,
2304
                                                    tListLen(context.digest),
2305
                                                    pPayload, size, 1);
2306
      if (TSDB_CODE_SUCCESS == code) {
9,618✔
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;
9,618✔
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) {
9,618✔
2324
        digest[0] = 1;
9,618✔
2325
        memcpy(digest + 1, context.digest, tListLen(context.digest));
9,618✔
2326
      }
2327
    }
2328
  }
2329

2330
  qDebug("%s, table list with %d uids built", idstr, (int32_t)numOfTables);
306,612,384✔
2331

2332
end:
306,716,809✔
2333
  if (code == 0) {
306,709,835✔
2334
     STableKeyInfo info = {.uid = 0, .groupId = 0};
306,708,741✔
2335
    for (int32_t i = 0; i < taosArrayGetSize(pUidList); ++i) {
1,072,286,984✔
2336
      int64_t* uid = (int64_t*)taosArrayGet(pUidList, i);
765,474,133✔
2337
      QUERY_CHECK_NULL(uid, code, lino, end, terrno);
765,580,647✔
2338

2339
      info.uid = *uid;
765,580,647✔
2340
        //qInfo("doSetQualifiedUid row:%d added to pTableList", i);
2341
      void* p = taosArrayPush(pListInfo->pTableList, &info);
765,564,793✔
2342
      QUERY_CHECK_NULL(p, code, lino, end, terrno);
765,605,595✔
2343
    }
2344
  } else {
2345
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1,094✔
2346
  }
2347
  taosArrayDestroy(pUidList);
306,780,103✔
2348
  taosArrayDestroy(pTagColIds);
306,718,410✔
2349
  taosMemFreeClear(pTagCondKey);
306,722,810✔
2350
  taosMemFreeClear(pPayload);
306,722,810✔
2351
  return code;
306,722,810✔
2352
}
2353

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

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

2371
_end:
28,540✔
2372
  tableListDestroy(pTableListInfo);
28,540✔
2373
  if (code != TSDB_CODE_SUCCESS) {
28,540✔
2374
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2375
  }
2376
  return code;
28,540✔
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,457,988✔
2479
  if (!pNodeList) {
3,457,988✔
2480
    return NULL;
×
2481
  }
2482

2483
  size_t  numOfCols = LIST_LENGTH(pNodeList);
3,457,988✔
2484
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
3,457,980✔
2485
  if (pList == NULL) {
3,453,218✔
2486
    return NULL;
×
2487
  }
2488

2489
  for (int32_t i = 0; i < numOfCols; ++i) {
8,001,447✔
2490
    SColumnNode* pColNode = (SColumnNode*)nodesListGetNode(pNodeList, i);
4,548,992✔
2491
    if (!pColNode) {
4,553,326✔
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,553,326✔
2499
    c.slotId = pColNode->slotId;
4,552,892✔
2500
    c.colId = pColNode->colId;
4,550,488✔
2501
    c.type = pColNode->node.resType.type;
4,550,337✔
2502
    c.bytes = pColNode->node.resType.bytes;
4,551,289✔
2503
    c.precision = pColNode->node.resType.precision;
4,546,360✔
2504
    c.scale = pColNode->node.resType.scale;
4,547,438✔
2505

2506
    void* tmp = taosArrayPush(pList, &c);
4,550,829✔
2507
    if (!tmp) {
4,550,829✔
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,452,455✔
2515
}
2516

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

2526
  pMatchInfo->matchType = type;
392,302,646✔
2527

2528
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchItem));
392,298,905✔
2529
  if (pList == NULL) {
392,208,265✔
2530
    code = terrno;
×
2531
    return code;
×
2532
  }
2533

2534
  SColMatchItem** infoBySlot = NULL;
392,208,265✔
2535
  if (numOfCols > 0) {
392,208,265✔
2536
    infoBySlot = taosMemoryCalloc(numOfCols, sizeof(*infoBySlot));
392,102,057✔
2537
    if (infoBySlot == NULL) {
392,148,436✔
2538
      code = terrno;
×
2539
      goto _end;
×
2540
    }
2541
  }
2542

2543
  SNode* node = NULL;
392,254,644✔
2544
  FOREACH(node, pNodeList) {
1,905,679,490✔
2545
    STargetNode* pNode = (STargetNode*)node;
1,513,270,355✔
2546
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,513,270,355✔
2547
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
1,513,270,355✔
2548
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
1,501,071,069✔
2549

2550
      SColMatchItem c = {.needOutput = true};
1,501,100,947✔
2551
      c.colId = pColNode->colId;
1,501,096,922✔
2552
      c.srcSlotId = pColNode->slotId;
1,501,086,614✔
2553
      c.dstSlotId = pNode->slotId;
1,501,089,062✔
2554
      c.isPk = pColNode->isPk;
1,501,150,465✔
2555
      c.dataType = pColNode->node.resType;
1,501,090,679✔
2556
      void* tmp = taosArrayPush(pList, &c);
1,501,242,167✔
2557
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,501,242,167✔
2558
      if (hasPrevColId) {
1,501,242,167✔
2559
        if (c.colId < prevColId) {
1,109,330,786✔
2560
          colIdOrdered = false;
16,941,648✔
2561
        }
2562
      } else {
2563
        hasPrevColId = true;
391,911,381✔
2564
      }
2565
      prevColId = c.colId;
1,501,242,167✔
2566
      if (pNode->slotId >= 0 && pNode->slotId < numOfCols) {
1,501,242,167✔
2567
        infoBySlot[pNode->slotId] = (SColMatchItem*)taosArrayGet(pList, taosArrayGetSize(pList) - 1);
1,500,589,331✔
2568
      }
2569
    }
2570
  }
2571
  // set the output flag for each column in SColMatchInfo, according to the
2572
  *numOfOutputCols = 0;
392,370,972✔
2573
  SNode* slotNode = NULL;
392,369,783✔
2574
  FOREACH(slotNode, pOutputNodeList->pSlots) {
2,126,330,807✔
2575
    SSlotDescNode* pNode = (SSlotDescNode*)slotNode;
1,734,070,087✔
2576
    QUERY_CHECK_NULL(pNode, code, lino, _end, terrno);
1,734,070,087✔
2577

2578
    // todo: add reserve flag check
2579
    // it is a column reserved for the arithmetic expression calculation
2580
    if (pNode->slotId >= numOfCols) {
1,734,070,087✔
2581
      (*numOfOutputCols) += 1;
220,903,617✔
2582
      continue;
220,913,758✔
2583
    }
2584

2585
    SColMatchItem* info = infoBySlot ? infoBySlot[pNode->slotId] : NULL;
1,513,282,100✔
2586

2587
    if (pNode->output) {
1,513,246,819✔
2588
      (*numOfOutputCols) += 1;
1,491,157,893✔
2589
    } else if (info != NULL) {
22,119,782✔
2590
      // select distinct tbname from stb where tbname='abc';
2591
      info->needOutput = false;
22,127,332✔
2592
    }
2593
  }
2594

2595
  pMatchInfo->pList = pList;
392,297,042✔
2596
  pMatchInfo->colIdOrdered = colIdOrdered;
392,302,709✔
2597

2598
_end:
392,309,643✔
2599
  taosMemoryFree(infoBySlot);
392,309,643✔
2600
  if (code != TSDB_CODE_SUCCESS) {
392,259,360✔
2601
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2602
  }
2603
  return code;
392,146,947✔
2604
}
2605

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

2616
  return s;
1,400,550,643✔
2617
}
2618

2619
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType, EColumnType colType) {
1,328,518,464✔
2620
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
1,328,518,464✔
2621
  if (pCol == NULL) {
1,328,059,944✔
2622
    return NULL;
×
2623
  }
2624

2625
  pCol->slotId = slotId;
1,328,059,944✔
2626
  pCol->colId = colId;
1,328,094,758✔
2627
  pCol->bytes = pType->bytes;
1,328,114,487✔
2628
  pCol->type = pType->type;
1,328,171,300✔
2629
  pCol->scale = pType->scale;
1,328,250,516✔
2630
  pCol->precision = pType->precision;
1,328,339,968✔
2631
  pCol->dataBlockId = blockId;
1,328,477,626✔
2632
  pCol->colType = colType;
1,328,495,968✔
2633
  return pCol;
1,328,498,920✔
2634
}
2635

2636
int32_t createExprFromOneNode(SExprInfo* pExp, SNode* pNode, int16_t slotId) {
1,400,608,667✔
2637
  int32_t code = TSDB_CODE_SUCCESS;
1,400,608,667✔
2638
  int32_t lino = 0;
1,400,608,667✔
2639
  pExp->base.numOfParams = 0;
1,400,608,667✔
2640
  pExp->base.pParam = NULL;
1,400,844,316✔
2641
  pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
1,400,772,532✔
2642
  QUERY_CHECK_NULL(pExp->pExpr, code, lino, _end, terrno);
1,400,347,314✔
2643

2644
  pExp->pExpr->_function.num = 1;
1,400,416,943✔
2645
  pExp->pExpr->_function.functionId = -1;
1,400,431,551✔
2646

2647
  int32_t type = nodeType(pNode);
1,400,552,427✔
2648
  // it is a project query, or group by column
2649
  if (type == QUERY_NODE_COLUMN) {
1,400,802,449✔
2650
    pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
786,203,587✔
2651
    SColumnNode* pColNode = (SColumnNode*)pNode;
786,228,196✔
2652

2653
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
786,228,196✔
2654
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
786,213,854✔
2655

2656
    pExp->base.numOfParams = 1;
786,199,614✔
2657

2658
    SDataType* pType = &pColNode->node.resType;
786,152,939✔
2659
    pExp->base.resSchema =
2660
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pColNode->colName);
786,196,461✔
2661

2662
    pExp->base.pParam[0].pCol =
1,572,499,766✔
2663
        createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType, pColNode->colType);
1,572,549,542✔
2664
    QUERY_CHECK_NULL(pExp->base.pParam[0].pCol, code, lino, _end, terrno);
786,172,735✔
2665

2666
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
786,190,136✔
2667
  } else if (type == QUERY_NODE_VALUE) {
614,598,862✔
2668
    pExp->pExpr->nodeType = QUERY_NODE_VALUE;
21,106,577✔
2669
    SValueNode* pValNode = (SValueNode*)pNode;
21,097,240✔
2670

2671
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
21,097,240✔
2672
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
21,094,582✔
2673

2674
    pExp->base.numOfParams = 1;
21,091,901✔
2675

2676
    SDataType* pType = &pValNode->node.resType;
21,096,679✔
2677
    pExp->base.resSchema =
2678
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
21,091,704✔
2679
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
21,090,868✔
2680
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
21,095,046✔
2681
    QUERY_CHECK_CODE(code, lino, _end);
21,094,621✔
2682
  } else if (type == QUERY_NODE_REMOTE_VALUE) {
593,492,285✔
2683
    SRemoteValueNode* pRemote = (SRemoteValueNode*)pNode;
1,578,630✔
2684
    code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, pNode);
1,578,630✔
2685
    QUERY_CHECK_CODE(code, lino, _end);
1,578,630✔
2686

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

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

2693
    pExp->base.numOfParams = 1;
1,112,342✔
2694

2695
    SDataType* pType = &pValNode->node.resType;
1,112,342✔
2696
    pExp->base.resSchema =
2697
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pValNode->node.aliasName);
1,112,342✔
2698
    pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
1,111,738✔
2699
    code = nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
1,112,342✔
2700
    QUERY_CHECK_CODE(code, lino, _end);
1,112,342✔
2701
  } else if (type == QUERY_NODE_FUNCTION) {
591,913,655✔
2702
    pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
545,936,932✔
2703
    SFunctionNode* pFuncNode = (SFunctionNode*)pNode;
545,977,836✔
2704

2705
    SDataType* pType = &pFuncNode->node.resType;
545,977,836✔
2706
    pExp->base.resSchema =
2707
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pFuncNode->node.aliasName);
545,988,045✔
2708
    tExprNode* pExprNode = pExp->pExpr;
545,978,090✔
2709

2710
    pExprNode->_function.functionId = pFuncNode->funcId;
545,970,742✔
2711
    pExprNode->_function.pFunctNode = pFuncNode;
546,016,034✔
2712
    pExprNode->_function.functionType = pFuncNode->funcType;
546,002,561✔
2713

2714
    tstrncpy(pExprNode->_function.functionName, pFuncNode->functionName, tListLen(pExprNode->_function.functionName));
545,990,216✔
2715

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

2722
    if (!pFuncNode->pParameterList && (memcmp(pExprNode->_function.functionName, name, len) == 0) &&
546,014,791✔
2723
        pExprNode->_function.functionName[len] == 0) {
35,611,881✔
2724
      pFuncNode->pParameterList = NULL;
35,601,744✔
2725
      int32_t     code = nodesMakeList(&pFuncNode->pParameterList);
35,605,252✔
2726
      SValueNode* res = NULL;
35,616,616✔
2727
      if (TSDB_CODE_SUCCESS == code) {
35,617,662✔
2728
        code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&res);
35,617,066✔
2729
      }
2730
      QUERY_CHECK_CODE(code, lino, _end);
35,622,907✔
2731
      res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
35,622,907✔
2732
      code = nodesListAppend(pFuncNode->pParameterList, (SNode*)res);
35,614,498✔
2733
      if (code != TSDB_CODE_SUCCESS) {
35,623,047✔
2734
        nodesDestroyNode((SNode*)res);
×
2735
        res = NULL;
×
2736
      }
2737
      QUERY_CHECK_CODE(code, lino, _end);
35,623,047✔
2738
    }
2739
#endif
2740

2741
    int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
546,026,206✔
2742

2743
    pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
546,040,630✔
2744
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
545,898,810✔
2745
    pExp->base.numOfParams = numOfParam;
545,897,323✔
2746

2747
    for (int32_t j = 0; j < numOfParam && TSDB_CODE_SUCCESS == code; ++j) {
1,294,465,534✔
2748
      SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
748,519,638✔
2749
      QUERY_CHECK_NULL(p1, code, lino, _end, terrno);
748,561,576✔
2750
      if (p1->type == QUERY_NODE_COLUMN) {
748,561,576✔
2751
        SColumnNode* pcn = (SColumnNode*)p1;
542,261,400✔
2752

2753
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
542,261,400✔
2754
        pExp->base.pParam[j].pCol =
1,084,343,691✔
2755
            createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType, pcn->colType);
1,084,487,546✔
2756
        QUERY_CHECK_NULL(pExp->base.pParam[j].pCol, code, lino, _end, terrno);
542,204,223✔
2757
      } else if (p1->type == QUERY_NODE_VALUE) {
206,266,984✔
2758
        SValueNode* pvn = (SValueNode*)p1;
141,187,480✔
2759
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
141,187,480✔
2760
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
141,170,752✔
2761
        QUERY_CHECK_CODE(code, lino, _end);
141,156,638✔
2762
      } else if (p1->type == QUERY_NODE_REMOTE_VALUE) {
65,210,209✔
2763
        SRemoteValueNode* pRemote = (SRemoteValueNode*)p1;
80,936✔
2764
        code = qFetchRemoteNode(gTaskScalarExtra.pSubJobCtx, pRemote->subQIdx, p1);
80,936✔
2765
        QUERY_CHECK_CODE(code, lino, _end);
80,936✔
2766

2767
        SValueNode* pvn = (SValueNode*)pRemote;
58,588✔
2768
        pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
58,588✔
2769
        code = nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
58,588✔
2770
        QUERY_CHECK_CODE(code, lino, _end);
233✔
2771
      }
2772
    }
2773
    pExp->pExpr->_function.bindExprID = ((SExprNode*)pNode)->bindExprID;
545,945,896✔
2774
  } else if (type == QUERY_NODE_OPERATOR) {
45,976,723✔
2775
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
44,219,072✔
2776
    SOperatorNode* pOpNode = (SOperatorNode*)pNode;
44,213,695✔
2777

2778
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
44,213,695✔
2779
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
44,208,937✔
2780
    pExp->base.numOfParams = 1;
44,198,803✔
2781

2782
    SDataType* pType = &pOpNode->node.resType;
44,192,927✔
2783
    pExp->base.resSchema =
2784
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pOpNode->node.aliasName);
44,194,731✔
2785
    pExp->pExpr->_optrRoot.pRootNode = pNode;
44,211,294✔
2786
  } else if (type == QUERY_NODE_CASE_WHEN) {
1,757,651✔
2787
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
1,611,787✔
2788
    SCaseWhenNode* pCaseNode = (SCaseWhenNode*)pNode;
1,611,787✔
2789

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

2794
    SDataType* pType = &pCaseNode->node.resType;
1,611,788✔
2795
    pExp->base.resSchema =
2796
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCaseNode->node.aliasName);
1,611,788✔
2797
    pExp->pExpr->_optrRoot.pRootNode = pNode;
1,611,789✔
2798
  } else if (type == QUERY_NODE_LOGIC_CONDITION) {
145,864✔
2799
    pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
116,554✔
2800
    SLogicConditionNode* pCond = (SLogicConditionNode*)pNode;
116,554✔
2801
    pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
116,554✔
2802
    QUERY_CHECK_NULL(pExp->base.pParam, code, lino, _end, terrno);
116,554✔
2803
    pExp->base.numOfParams = 1;
115,953✔
2804
    SDataType* pType = &pCond->node.resType;
115,953✔
2805
    pExp->base.resSchema =
2806
        createResSchema(pType->type, pType->bytes, slotId, pType->scale, pType->precision, pCond->node.aliasName);
115,953✔
2807
    pExp->pExpr->_optrRoot.pRootNode = pNode;
116,554✔
2808
  } else {
2809
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
29,310✔
2810
    QUERY_CHECK_CODE(code, lino, _end);
29,310✔
2811
  }
2812
  pExp->pExpr->relatedTo = ((SExprNode*)pNode)->relatedTo;
1,400,268,145✔
2813
_end:
1,400,619,538✔
2814
  if (code != TSDB_CODE_SUCCESS) {
1,400,619,538✔
2815
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
488,636✔
2816
  }
2817
  return code;
1,400,505,740✔
2818
}
2819

2820
int32_t createExprFromTargetNode(SExprInfo* pExp, STargetNode* pTargetNode) {
1,400,607,242✔
2821
  return createExprFromOneNode(pExp, pTargetNode->pExpr, pTargetNode->slotId);
1,400,607,242✔
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) {
514,932,945✔
2846
  QRY_PARAM_CHECK(pExprInfo);
514,932,945✔
2847

2848
  int32_t code = 0;
514,973,459✔
2849
  int32_t lino = 0;
514,973,459✔
2850
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
514,973,459✔
2851
  int32_t numOfGroupKeys = 0;
514,918,459✔
2852
  if (pGroupKeys != NULL) {
514,918,459✔
2853
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
35,297,960✔
2854
  }
2855

2856
  *numOfExprs = numOfFuncs + numOfGroupKeys;
514,917,826✔
2857
  if (*numOfExprs == 0) {
514,794,977✔
2858
    return code;
44,435,342✔
2859
  }
2860

2861
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
470,433,994✔
2862
  QUERY_CHECK_NULL(pExprs, code, lino, _return, terrno);
470,243,640✔
2863

2864
  int32_t i = 0;
470,243,640✔
2865
  if (pNodeList != NULL) {
470,243,640✔
2866
    SNode* node = NULL;
439,424,879✔
2867
    FOREACH(node, pNodeList) {
1,772,098,086✔
2868
      STargetNode* pTargetNode = (STargetNode*)node;
1,333,330,582✔
2869
      SExprInfo*   pExp = &pExprs[i++];
1,333,330,582✔
2870
      code = createExprFromTargetNode(pExp, pTargetNode);
1,333,255,035✔
2871
      QUERY_CHECK_CODE(code, lino, _return);
1,333,161,843✔
2872
    }
2873
  }
2874

2875
  if (pGroupKeys != NULL) {
469,814,948✔
2876
    SNode* node = NULL;
35,286,059✔
2877
    FOREACH(node, pGroupKeys) {
102,447,362✔
2878
      STargetNode* pTargetNode = (STargetNode*)node;
67,161,783✔
2879
      SExprInfo*   pExp = &pExprs[i++];
67,161,783✔
2880
      code = createExprFromTargetNode(pExp, pTargetNode);
67,157,919✔
2881
      QUERY_CHECK_CODE(code, lino, _return);
67,161,303✔
2882
    }
2883
  }
2884

2885
  *pExprInfo = pExprs;
469,814,265✔
2886
  return code;
469,889,418✔
2887

2888
_return:
488,636✔
2889
  qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
488,636✔
2890
  destroyExprInfo(pExprs, *numOfExprs);
488,636✔
2891
  taosMemoryFreeClear(pExprs);
488,636✔
2892
  return code;
488,636✔
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) {
500,985,767✔
2904
  int32_t num = 0;
500,985,767✔
2905
  int32_t code = TSDB_CODE_SUCCESS;
500,985,767✔
2906
  int32_t lino = 0;
500,985,767✔
2907

2908
  SArray* pValCtxArray = NULL;
500,985,767✔
2909
  for (int32_t i = numOfOutput - 1; i > 0; --i) {  // select Func is at the end of the list
1,423,111,012✔
2910
    int32_t funcIdx = pCtx[i].pExpr->pExpr->_function.bindExprID;
922,172,798✔
2911
    if (funcIdx > 0) {
922,261,885✔
2912
      if (pValCtxArray == NULL) {
1,792,084✔
2913
        // the end of the list is the select function of biggest index
2914
        pValCtxArray = taosArrayInit_s(sizeof(SSubsidiaryResInfo*), funcIdx);
1,285,558✔
2915
        if (pValCtxArray == NULL) {
1,286,046✔
2916
          return terrno;
×
2917
        }
2918
      }
2919
      if (funcIdx > pValCtxArray->size) {
1,792,572✔
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,792,084✔
2925
      pSubsidiary->pCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
1,793,548✔
2926
      if (pSubsidiary->pCtx == NULL) {
1,794,036✔
2927
        taosArrayDestroyP(pValCtxArray, deleteSubsidiareCtx);
×
2928
        return terrno;
×
2929
      }
2930
      pSubsidiary->num = 0;
1,791,596✔
2931
      taosArraySet(pValCtxArray, funcIdx - 1, &pSubsidiary);
1,792,084✔
2932
    }
2933
  }
2934

2935
  SqlFunctionCtx*  p = NULL;
500,938,214✔
2936
  SqlFunctionCtx** pValCtx = NULL;
500,938,214✔
2937
  if (pValCtxArray == NULL) {
500,938,214✔
2938
    pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
499,696,420✔
2939
    if (pValCtx == NULL) {
499,800,485✔
2940
      QUERY_CHECK_CODE(terrno, lino, _end);
×
2941
    }
2942
  }
2943

2944
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,888,067,641✔
2945
    const char* pName = pCtx[i].pExpr->pExpr->_function.functionName;
1,386,932,801✔
2946
    if ((strcmp(pName, "_select_value") == 0)) {
1,387,088,670✔
2947
      if (pValCtxArray == NULL) {
10,023,133✔
2948
        pValCtx[num++] = &pCtx[i];
7,504,904✔
2949
      } else {
2950
        int32_t bindFuncIndex = pCtx[i].pExpr->pExpr->relatedTo;  // start from index 1;
2,518,305✔
2951
        if (bindFuncIndex > 0) {                                  // 0 is default index related to the select function
2,519,697✔
2952
          bindFuncIndex -= 1;
2,460,649✔
2953
        }
2954
        SSubsidiaryResInfo** pSubsidiary = taosArrayGet(pValCtxArray, bindFuncIndex);
2,519,697✔
2955
        if (pSubsidiary == NULL) {
2,518,233✔
2956
          QUERY_CHECK_CODE(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR, lino, _end);
×
2957
        }
2958
        (*pSubsidiary)->pCtx[(*pSubsidiary)->num] = &pCtx[i];
2,518,233✔
2959
        (*pSubsidiary)->num++;
2,518,721✔
2960
      }
2961
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
1,377,065,537✔
2962
      if (pValCtxArray == NULL) {
124,600,778✔
2963
        p = &pCtx[i];
122,413,703✔
2964
      }
2965
    }
2966
  }
2967

2968
  if (p != NULL) {
501,134,840✔
2969
    p->subsidiaries.pCtx = pValCtx;
59,016,561✔
2970
    p->subsidiaries.num = num;
59,015,818✔
2971
  } else {
2972
    taosMemoryFreeClear(pValCtx);
442,118,279✔
2973
  }
2974

2975
_end:
1,405,258✔
2976
  if (code != TSDB_CODE_SUCCESS) {
500,996,383✔
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);
500,996,383✔
2982
  }
2983
  return code;
501,008,805✔
2984
}
2985

2986
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowEntryInfoOffset,
501,183,433✔
2987
                                     SFunctionStateStore* pStore) {
2988
  int32_t         code = TSDB_CODE_SUCCESS;
501,183,433✔
2989
  int32_t         lino = 0;
501,183,433✔
2990
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
501,183,433✔
2991
  if (pFuncCtx == NULL) {
500,963,456✔
2992
    return NULL;
×
2993
  }
2994

2995
  *rowEntryInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
500,963,456✔
2996
  if (*rowEntryInfoOffset == 0) {
501,220,200✔
2997
    taosMemoryFreeClear(pFuncCtx);
×
2998
    return NULL;
×
2999
  }
3000

3001
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,888,226,229✔
3002
    SExprInfo* pExpr = &pExprInfo[i];
1,387,123,938✔
3003

3004
    SExprBasicInfo* pFunct = &pExpr->base;
1,387,044,175✔
3005
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
1,387,085,501✔
3006

3007
    pCtx->functionId = -1;
1,387,113,516✔
3008
    pCtx->pExpr = pExpr;
1,387,106,483✔
3009

3010
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
1,387,038,403✔
3011
      SFuncExecEnv env = {0};
544,309,067✔
3012
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;
544,304,300✔
3013
      pCtx->isPseudoFunc = fmIsWindowPseudoColumnFunc(pCtx->functionId) || fmIsPlaceHolderFunc(pCtx->functionId);
544,276,792✔
3014
      pCtx->isNotNullFunc = fmIsNotNullOutputFunc(pCtx->functionId);
544,340,907✔
3015

3016
      bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
544,322,040✔
3017
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId) ||
724,136,869✔
3018
          fmIsProcessByRowFunc(pCtx->functionId)) {
544,266,499✔
3019
        if (!isUdaf) {
364,362,700✔
3020
          code = fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
364,309,492✔
3021
          QUERY_CHECK_CODE(code, lino, _end);
364,224,106✔
3022
        } else {
3023
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
53,208✔
3024
          pCtx->udfName = taosStrdup(udfName);
53,208✔
3025
          QUERY_CHECK_NULL(pCtx->udfName, code, lino, _end, terrno);
53,208✔
3026

3027
          code = fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
53,208✔
3028
          QUERY_CHECK_CODE(code, lino, _end);
53,208✔
3029
        }
3030
        bool tmp = pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
364,277,314✔
3031
        if (!tmp) {
364,300,600✔
3032
          code = terrno;
×
3033
          QUERY_CHECK_CODE(code, lino, _end);
7,845✔
3034
        }
3035
      } else {
3036
        if (fmIsPlaceHolderFunc(pCtx->functionId)) {
179,923,539✔
3037
          code = fmGetStreamPesudoFuncEnv(pCtx->functionId, pExpr->base.pParamList, &env);
8,094,421✔
3038
          QUERY_CHECK_CODE(code, lino, _end);
8,094,468✔
3039
        }      
3040
        
3041
        code = fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
179,933,867✔
3042
        if (code != TSDB_CODE_SUCCESS && isUdaf) {
179,949,530✔
3043
          code = TSDB_CODE_SUCCESS;
69,511✔
3044
        }
3045
        QUERY_CHECK_CODE(code, lino, _end);
179,949,530✔
3046

3047
        if (pCtx->sfp.getEnv != NULL) {
179,949,530✔
3048
          bool tmp = pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
32,247,085✔
3049
          if (!tmp) {
32,245,712✔
3050
            code = terrno;
×
3051
            QUERY_CHECK_CODE(code, lino, _end);
×
3052
          }
3053
        }
3054
      }
3055
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
544,262,593✔
3056
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
842,680,602✔
3057
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
22,186,758✔
3058
      // for simple column, the result buffer needs to hold at least one element.
3059
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
842,760,131✔
3060
    }
3061

3062
    pCtx->input.numOfInputCols = pFunct->numOfParams;
1,386,961,075✔
3063
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
1,387,072,374✔
3064
    QUERY_CHECK_NULL(pCtx->input.pData, code, lino, _end, terrno);
1,387,195,896✔
3065
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
1,387,097,889✔
3066
    QUERY_CHECK_NULL(pCtx->input.pColumnDataAgg, code, lino, _end, terrno);
1,387,202,980✔
3067

3068
    pCtx->pTsOutput = NULL;
1,387,137,420✔
3069
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
1,387,147,247✔
3070
    pCtx->resDataInfo.type = pFunct->resSchema.type;
1,387,046,153✔
3071
    pCtx->order = TSDB_ORDER_ASC;
1,386,986,909✔
3072
    pCtx->start.key = INT64_MIN;
1,387,144,214✔
3073
    pCtx->end.key = INT64_MIN;
1,387,008,499✔
3074
    pCtx->numOfParams = pExpr->base.numOfParams;
1,387,171,746✔
3075
    pCtx->param = pFunct->pParam;
1,387,164,853✔
3076
    pCtx->saveHandle.currentPage = -1;
1,387,198,212✔
3077
    pCtx->pStore = pStore;
1,387,083,484✔
3078
    pCtx->hasWindowOrGroup = false;
1,387,214,239✔
3079
    pCtx->needCleanup = false;
1,387,086,898✔
3080
    pCtx->skipDynDataCheck = false;
1,386,959,959✔
3081
  }
3082

3083
  for (int32_t i = 1; i < numOfOutput; ++i) {
1,423,376,696✔
3084
    (*rowEntryInfoOffset)[i] = (int32_t)((*rowEntryInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) +
1,844,604,227✔
3085
                                         pFuncCtx[i - 1].resDataInfo.interBufSize);
922,317,721✔
3086
  }
3087

3088
  code = setSelectValueColumnInfo(pFuncCtx, numOfOutput);
501,032,263✔
3089
  QUERY_CHECK_CODE(code, lino, _end);
500,991,531✔
3090

3091
_end:
500,991,531✔
3092
  if (code != TSDB_CODE_SUCCESS) {
501,008,074✔
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;
501,008,074✔
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) {
17,423,134✔
3110
  int32_t code = TSDB_CODE_SUCCESS;
17,423,134✔
3111
  size_t  numOfSrcCols = taosArrayGetSize(pCols);
17,423,134✔
3112

3113
  int32_t i = 0, j = 0;
17,423,468✔
3114
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
142,359,763✔
3115
    SColumnInfoData* p = taosArrayGet(pCols, i);
124,936,337✔
3116
    if (!p) {
124,937,359✔
3117
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3118
      return terrno;
×
3119
    }
3120
    SColMatchItem* pmInfo = taosArrayGet(pColMatchInfo, j);
124,937,359✔
3121
    if (!pmInfo) {
124,937,609✔
3122
      return terrno;
×
3123
    }
3124

3125
    if (p->info.colId == pmInfo->colId) {
124,937,609✔
3126
      SColumnInfoData* pDst = taosArrayGet(pBlock->pDataBlock, pmInfo->dstSlotId);
116,008,728✔
3127
      if (!pDst) {
116,010,608✔
3128
        return terrno;
×
3129
      }
3130
      code = colDataAssign(pDst, p, pBlock->info.rows, &pBlock->info);
116,010,608✔
3131
      if (code != TSDB_CODE_SUCCESS) {
116,007,485✔
3132
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3133
        return code;
×
3134
      }
3135
      i++;
116,007,485✔
3136
      j++;
116,007,485✔
3137
    } else if (p->info.colId < pmInfo->colId) {
8,929,195✔
3138
      i++;
8,928,810✔
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;
17,424,943✔
3145
}
3146

3147
SInterval extractIntervalInfo(const STableScanPhysiNode* pTableScanNode) {
272,142,121✔
3148
  SInterval interval = {
544,244,693✔
3149
      .interval = pTableScanNode->interval,
272,201,428✔
3150
      .sliding = pTableScanNode->sliding,
272,188,340✔
3151
      .intervalUnit = pTableScanNode->intervalUnit,
272,168,142✔
3152
      .slidingUnit = pTableScanNode->slidingUnit,
272,169,838✔
3153
      .offset = pTableScanNode->offset,
272,155,022✔
3154
      .precision = pTableScanNode->scan.node.pOutputDataBlockDesc->precision,
272,188,548✔
3155
      .timeRange = pTableScanNode->scanRange,
3156
  };
3157
  calcIntervalAutoOffset(&interval);
272,181,965✔
3158

3159
  return interval;
272,191,617✔
3160
}
3161

3162
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
72,722,555✔
3163
  SColumn c = {0};
72,722,555✔
3164

3165
  c.slotId = pColNode->slotId;
72,722,555✔
3166
  c.colId = pColNode->colId;
72,723,738✔
3167
  c.type = pColNode->node.resType.type;
72,717,011✔
3168
  c.bytes = pColNode->node.resType.bytes;
72,724,580✔
3169
  c.scale = pColNode->node.resType.scale;
72,718,608✔
3170
  c.precision = pColNode->node.resType.precision;
72,719,198✔
3171
  return c;
72,708,520✔
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,660,660✔
3184
                                 STimeWindow* extTwindows) {
3185
  int32_t     code = TSDB_CODE_SUCCESS;
3,660,660✔
3186
  int32_t     lino = 0;
3,660,660✔
3187
  STimeWindow tempWindow;
3188

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

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

3200
  if (cond->ekey < range->skey) {
3,077,596✔
3201
    extTwindows[0] = *cond;
407,515✔
3202
    *twindow = extTwindows[1] = TSWINDOW_DESC_INITIALIZER;
407,515✔
3203
    return code;
407,515✔
3204
  }
3205

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

3213
  return code;
2,670,081✔
3214
}
3215

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

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

3234
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
307,056,021✔
3235
  if (!pCond->colList) {
307,014,859✔
3236
    return terrno;
×
3237
  }
3238
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
307,008,029✔
3239
  if (pCond->pSlotList == NULL) {
307,048,102✔
3240
    taosMemoryFreeClear(pCond->colList);
×
3241
    return terrno;
×
3242
  }
3243

3244
  // TODO: get it from stable scan node
3245
  pCond->twindows = pTableScanNode->scanRange;
306,975,645✔
3246
  pCond->suid = pTableScanNode->scan.suid;
307,174,653✔
3247
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
306,996,724✔
3248
  pCond->startVersion = -1;
306,981,834✔
3249
  pCond->endVersion = -1;
307,012,762✔
3250
  pCond->skipRollup = readHandle->skipRollup;
306,858,731✔
3251
  if (readHandle->winRangeValid) {
306,875,487✔
3252
    pCond->twindows = readHandle->winRange;
570,008✔
3253
  }
3254
  pCond->cacheSttStatis = readHandle->cacheSttStatis;
307,036,846✔
3255
  // allowed read stt file optimization mode
3256
  pCond->notLoadData = (pTableScanNode->dataRequired == FUNC_DATA_REQUIRED_NOT_LOAD) &&
614,077,987✔
3257
                       (pTableScanNode->scan.node.pConditions == NULL) && (pTableScanNode->interval == 0);
306,912,636✔
3258

3259
  int32_t j = 0;
307,036,824✔
3260
  SNode*  node = NULL;
307,036,824✔
3261
  FOREACH(node, pTableScanNode->scan.pScanCols) {
1,439,345,906✔
3262
    STargetNode* pNode = (STargetNode*)node;
1,132,025,093✔
3263
    if (!pNode) {
1,132,025,093✔
3264
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3265
      return terrno;
×
3266
    }
3267
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
1,132,025,093✔
3268
    if (pColNode->colType == COLUMN_TYPE_TAG) {
1,132,095,665✔
3269
      continue;
×
3270
    }
3271

3272
    pCond->colList[j].type = pColNode->node.resType.type;
1,132,106,236✔
3273
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
1,132,205,451✔
3274
    pCond->colList[j].colId = pColNode->colId;
1,132,227,643✔
3275
    pCond->colList[j].pk = pColNode->isPk;
1,132,369,350✔
3276

3277
    pCond->pSlotList[j] = pNode->slotId;
1,132,295,565✔
3278
    j += 1;
1,132,309,082✔
3279
  }
3280

3281
  pCond->numOfCols = j;
307,153,580✔
3282

3283
  if (applyExtWin) {
307,150,361✔
3284
    if (NULL != pTableScanNode->pExtScanRange) {
272,988,441✔
3285
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
3,581,282✔
3286
      code = getQueryExtWindow(&pCond->twindows, pTableScanNode->pExtScanRange, &pCond->twindows, pCond->extTwindows);
3,581,282✔
3287
    } else if (readHandle->extWinRangeValid) {
269,188,344✔
3288
      pCond->type = TIMEWINDOW_RANGE_EXTERNAL;
79,378✔
3289
      code = getQueryExtWindow(&pCond->twindows, &readHandle->extWinRange, &pCond->twindows, pCond->extTwindows);
79,378✔
3290
    }
3291
  }
3292

3293
  if (pTableScanNode->pPrimaryCond) {
306,941,542✔
3294
    bool isStrict = false;
708,372✔
3295
    code = getPrimaryTimeRange((SNode**)&pTableScanNode->pPrimaryCond, &pCond->twindows, &isStrict);
708,372✔
3296
    if (code || !isStrict) {
708,372✔
3297
      code = nodesMergeNode((SNode**)&pTableScanNode->scan.node.pConditions, &pTableScanNode->pPrimaryCond);
695,096✔
3298
    }
3299
  }
3300

3301
  return code;
306,882,039✔
3302
}
3303

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

3309
  pCond->order = TSDB_ORDER_ASC;
122,437,135✔
3310
  pCond->numOfCols = (int32_t)taosArrayGetSize(colArray);
122,461,582✔
3311

3312
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
122,468,800✔
3313
  QUERY_CHECK_NULL(pCond->colList, code, lino, _return, terrno);
122,428,022✔
3314

3315
  pCond->pSlotList = taosMemoryMalloc(sizeof(int32_t) * pCond->numOfCols);
122,435,342✔
3316
  QUERY_CHECK_NULL(pCond->pSlotList, code, lino, _return, terrno);
122,451,521✔
3317

3318
  pCond->twindows = pOrgCond->twindows;
122,444,396✔
3319
  pCond->order = pOrgCond->order;
122,459,698✔
3320
  pCond->type = pOrgCond->type;
122,480,888✔
3321
  pCond->startVersion = -1;
122,481,594✔
3322
  pCond->endVersion = -1;
122,478,992✔
3323
  pCond->skipRollup = true;
122,468,209✔
3324
  pCond->notLoadData = false;
122,479,682✔
3325

3326
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
502,572,569✔
3327
    SColIdPair* pColPair = taosArrayGet(colArray, i);
380,128,293✔
3328
    QUERY_CHECK_NULL(pColPair, code, lino, _return, terrno);
380,153,256✔
3329

3330
    bool find = false;
380,127,635✔
3331
    for (int32_t j = 0; j < pOrgCond->numOfCols; ++j) {
1,945,399,350✔
3332
      if (pOrgCond->colList[j].colId == pColPair->vtbColId) {
1,945,134,619✔
3333
        pCond->colList[i].type = pOrgCond->colList[j].type;
380,148,583✔
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) {
380,153,744✔
3338
          pCond->colList[i].bytes = TMAX(pOrgCond->colList[j].bytes, pColPair->type.bytes);
44,293,282✔
3339
        } else {
3340
          pCond->colList[i].bytes = pOrgCond->colList[j].bytes;
335,855,523✔
3341
        }
3342
        pCond->colList[i].colId = pColPair->orgColId;
380,173,286✔
3343
        pCond->colList[i].pk = pOrgCond->colList[j].pk;
380,152,811✔
3344
        pCond->pSlotList[i] = i;
380,162,109✔
3345
        find = true;
380,163,775✔
3346
        qDebug("%s mapped vtb colId:%d to org colId:%d, type:%d, bytes:%d", __func__, pColPair->vtbColId,
380,163,775✔
3347
               pColPair->orgColId, pCond->colList[i].type, pCond->colList[i].bytes);
3348
        break;
380,095,698✔
3349
      }
3350
    }
3351
    QUERY_CHECK_CONDITION(find, code, lino, _return, TSDB_CODE_NOT_FOUND);
380,108,546✔
3352
  }
3353

3354
  return code;
122,490,040✔
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) {
762,514,632✔
3363
  taosMemoryFreeClear(pCond->colList);
762,514,632✔
3364
  taosMemoryFreeClear(pCond->pSlotList);
762,553,346✔
3365
}
762,488,600✔
3366

3367
int32_t convertFillType(int32_t mode) {
4,479,017✔
3368
  int32_t type = TSDB_FILL_NONE;
4,479,017✔
3369
  switch (mode) {
4,479,017✔
3370
    case FILL_MODE_PREV:
222,185✔
3371
      type = TSDB_FILL_PREV;
222,185✔
3372
      break;
222,185✔
3373
    case FILL_MODE_NONE:
×
3374
      type = TSDB_FILL_NONE;
×
3375
      break;
×
3376
    case FILL_MODE_NULL:
280,501✔
3377
      type = TSDB_FILL_NULL;
280,501✔
3378
      break;
280,501✔
3379
    case FILL_MODE_NULL_F:
44,822✔
3380
      type = TSDB_FILL_NULL_F;
44,822✔
3381
      break;
44,822✔
3382
    case FILL_MODE_NEXT:
219,585✔
3383
      type = TSDB_FILL_NEXT;
219,585✔
3384
      break;
219,585✔
3385
    case FILL_MODE_VALUE:
199,996✔
3386
      type = TSDB_FILL_SET_VALUE;
199,996✔
3387
      break;
199,996✔
3388
    case FILL_MODE_VALUE_F:
21,029✔
3389
      type = TSDB_FILL_SET_VALUE_F;
21,029✔
3390
      break;
21,029✔
3391
    case FILL_MODE_LINEAR:
266,717✔
3392
      type = TSDB_FILL_LINEAR;
266,717✔
3393
      break;
266,717✔
3394
    case FILL_MODE_NEAR:
3,224,182✔
3395
      type = TSDB_FILL_NEAR;
3,224,182✔
3396
      break;
3,224,182✔
3397
    default:
×
3398
      type = TSDB_FILL_NONE;
×
3399
  }
3400

3401
  return type;
4,479,017✔
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);
30,825✔
3410

3411
    int64_t key = w->skey;
344,980✔
3412
    while (key < ts) {  // moving towards end
363,765✔
3413
      key = getNextTimeWindowStart(pInterval, key, TSDB_ORDER_ASC);
186,611✔
3414
      if (key > ts) {
186,611✔
3415
        break;
167,826✔
3416
      }
3417

3418
      w->skey = key;
18,785✔
3419
    }
3420
    w->ekey = taosTimeAdd(w->skey, pInterval->interval, pInterval->intervalUnit, pInterval->precision, NULL) - 1;
344,980✔
3421
  }
3422
}
2,147,483,647✔
3423

3424
static STimeWindow doCalculateTimeWindow(int64_t ts, SInterval* pInterval) {
82,701,693✔
3425
  STimeWindow w = {0};
82,701,693✔
3426

3427
  w.skey = taosTimeTruncate(ts, pInterval);
82,701,693✔
3428
  w.ekey = taosTimeGetIntervalEnd(w.skey, pInterval);
82,692,020✔
3429
  return w;
82,712,160✔
3430
}
3431

3432
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order) {
1,724,924✔
3433
  STimeWindow win = *pWindow;
1,724,924✔
3434
  STimeWindow save = win;
1,724,315✔
3435
  while (win.skey <= ts && win.ekey >= ts) {
9,937,334✔
3436
    save = win;
8,212,410✔
3437
    // get previous time window
3438
    getNextTimeWindow(pInterval, &win, order == TSDB_ORDER_DESC ? TSDB_ORDER_ASC : TSDB_ORDER_DESC);
8,212,410✔
3439
  }
3440

3441
  return save;
1,724,924✔
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,
127,871,700✔
3447
                                int32_t order) {
3448
  STimeWindow w = {0};
127,871,700✔
3449
  if (pResultRowInfo->cur.pageId == -1) {  // the first window, from the previous stored value
127,880,015✔
3450
    getInitialStartTimeWindow(pInterval, ts, &w, (order != TSDB_ORDER_DESC));
4,165,318✔
3451
    return w;
4,168,678✔
3452
  }
3453

3454
  SResultRow* pRow = getResultRowByPos(pBuf, &pResultRowInfo->cur, false);
123,710,191✔
3455
  if (pRow) {
123,718,293✔
3456
    TAOS_SET_OBJ_ALIGNED(&w, pRow->win);
123,721,381✔
3457
  }
3458

3459
  // in case of typical time window, we can calculate time window directly.
3460
  if (w.skey > ts || w.ekey < ts) {
123,720,912✔
3461
    w = doCalculateTimeWindow(ts, pInterval);
82,710,693✔
3462
  }
3463

3464
  if (pInterval->interval != pInterval->sliding) {
123,723,439✔
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,724,924✔
3468
  }
3469

3470
  return w;
123,717,079✔
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) {
589,358,083✔
3479
  return (pLimitInfo->limit.limit != -1 || pLimitInfo->limit.offset != -1 || pLimitInfo->slimit.limit != -1 ||
1,173,937,173✔
3480
          pLimitInfo->slimit.offset != -1);
584,579,925✔
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) {
737,925,606✔
3488
  SLimit limit = {.limit = getLimit(pLimit), .offset = getOffset(pLimit)};
737,925,606✔
3489
  SLimit slimit = {.limit = getLimit(pSLimit), .offset = getOffset(pSLimit)};
737,703,232✔
3490

3491
  pLimitInfo->limit = limit;
737,717,508✔
3492
  pLimitInfo->slimit = slimit;
737,731,624✔
3493
  pLimitInfo->remainOffset = limit.offset;
737,712,936✔
3494
  pLimitInfo->remainGroupOffset = slimit.offset;
737,674,998✔
3495
  pLimitInfo->numOfOutputRows = 0;
737,769,223✔
3496
  pLimitInfo->numOfOutputGroups = 0;
737,780,100✔
3497
  pLimitInfo->currentGroupId = 0;
737,772,524✔
3498
}
737,831,799✔
3499

3500
void resetLimitInfoForNextGroup(SLimitInfo* pLimitInfo) {
94,916,710✔
3501
  pLimitInfo->numOfOutputRows = 0;
94,916,710✔
3502
  pLimitInfo->remainOffset = pLimitInfo->limit.offset;
94,953,570✔
3503
}
94,964,291✔
3504

3505
int32_t tableListGetSize(const STableListInfo* pTableList, int32_t* pRes) {
778,712,328✔
3506
  if (taosArrayGetSize(pTableList->pTableList) != taosHashGetSize(pTableList->map)) {
778,712,328✔
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);
778,638,272✔
3511
  return TSDB_CODE_SUCCESS;
778,659,849✔
3512
}
3513

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

3516
STableKeyInfo* tableListGetInfo(const STableListInfo* pTableList, int32_t index) {
237,933,232✔
3517
  if (taosArrayGetSize(pTableList->pTableList) == 0) {
237,933,232✔
3518
    return NULL;
2,562✔
3519
  }
3520

3521
  return taosArrayGet(pTableList->pTableList, index);
237,877,520✔
3522
}
3523

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

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

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

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

3556
  STableKeyInfo* pKeyInfo = taosArrayGet(pTableList->pTableList, *slot);
874,613,021✔
3557
  if (pKeyInfo == NULL) {
874,660,552✔
3558
    qDebug("table:%" PRIu64 " not found in table list", tableUid);
×
3559
    return TSDB_CODE_PAR_TABLE_NOT_EXIST;
×
3560
  }
3561

3562
  *gid = pKeyInfo->groupId;
874,660,552✔
3563
  *baseGid = pKeyInfo->baseGId;
874,667,579✔
3564
  
3565
  return 0;
874,671,381✔
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,530,354✔
3593
  int32_t code = TSDB_CODE_SUCCESS;
1,530,354✔
3594
  int32_t lino = 0;
1,530,354✔
3595
  if (pTableList->map == NULL) {
1,530,354✔
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,531,810✔
3601
  void*         p = taosHashGet(pTableList->map, &uid, sizeof(uid));
1,532,174✔
3602
  if (p != NULL) {
1,530,354✔
3603
    qInfo("table:%" PRId64 " already in tableIdList, ignore it", uid);
×
3604
    goto _end;
×
3605
  }
3606

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

3610
  int32_t slot = (int32_t)taosArrayGetSize(pTableList->pTableList) - 1;
1,532,538✔
3611
  code = taosHashPut(pTableList->map, &uid, sizeof(uid), &slot, sizeof(slot));
1,532,174✔
3612
  if (code != TSDB_CODE_SUCCESS) {
1,532,538✔
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,532,538✔
3619
  if (code != TSDB_CODE_SUCCESS) {
1,531,446✔
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,531,446✔
3623
  }
3624

3625
  return code;
1,532,538✔
3626
}
3627

3628
int32_t tableListGetGroupList(const STableListInfo* pTableList, int32_t ordinalGroupIndex, STableKeyInfo** pKeyInfo,
292,449,219✔
3629
                              int32_t* size) {
3630
  int32_t totalGroups = tableListGetOutputGroups(pTableList);
292,449,219✔
3631
  int32_t numOfTables = 0;
292,419,829✔
3632
  int32_t code = tableListGetSize(pTableList, &numOfTables);
292,423,867✔
3633
  if (code != TSDB_CODE_SUCCESS) {
292,481,829✔
3634
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3635
    return code;
×
3636
  }
3637

3638
  if (ordinalGroupIndex < 0 || ordinalGroupIndex >= totalGroups) {
292,481,829✔
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) {
292,481,829✔
3645
    *size = numOfTables;
291,898,599✔
3646
    *pKeyInfo = (*size == 0) ? NULL : taosArrayGet(pTableList->pTableList, 0);
291,885,656✔
3647
    return TSDB_CODE_SUCCESS;
291,858,577✔
3648
  } else if (totalGroups == numOfTables) {
583,230✔
3649
    *size = 1;
538,569✔
3650
    *pKeyInfo = taosArrayGet(pTableList->pTableList, ordinalGroupIndex);
538,569✔
3651
    return TSDB_CODE_SUCCESS;
538,608✔
3652
  }
3653

3654
  int32_t offset = pTableList->groupOffset[ordinalGroupIndex];
46,128✔
3655
  if (ordinalGroupIndex < totalGroups - 1) {
66,437✔
3656
    *size = pTableList->groupOffset[ordinalGroupIndex + 1] - offset;
50,002✔
3657
  } else {
3658
    *size = numOfTables - offset;
16,435✔
3659
  }
3660

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

3665
int32_t tableListGetOutputGroups(const STableListInfo* pTableList) { return pTableList->numOfOuputGroups; }
818,727,553✔
3666

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

3669
STableListInfo* tableListCreate() {
331,365,353✔
3670
  STableListInfo* pListInfo = taosMemoryCalloc(1, sizeof(STableListInfo));
331,365,353✔
3671
  if (pListInfo == NULL) {
331,223,343✔
3672
    return NULL;
×
3673
  }
3674

3675
  pListInfo->remainGroups = NULL;
331,223,343✔
3676
  pListInfo->pTableList = taosArrayInit(4, sizeof(STableKeyInfo));
331,223,940✔
3677
  if (pListInfo->pTableList == NULL) {
331,253,335✔
3678
    goto _error;
×
3679
  }
3680

3681
  pListInfo->map = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
331,300,349✔
3682
  if (pListInfo->map == NULL) {
331,469,253✔
3683
    goto _error;
×
3684
  }
3685

3686
  pListInfo->numOfOuputGroups = 1;
331,470,260✔
3687
  return pListInfo;
331,467,788✔
3688

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

3694
void tableListDestroy(STableListInfo* pTableListInfo) {
344,941,851✔
3695
  if (pTableListInfo == NULL) {
344,941,851✔
3696
    return;
13,543,321✔
3697
  }
3698

3699
  taosArrayDestroy(pTableListInfo->pTableList);
331,398,530✔
3700
  taosMemoryFreeClear(pTableListInfo->groupOffset);
331,386,325✔
3701

3702
  taosHashCleanup(pTableListInfo->map);
331,376,138✔
3703
  taosHashCleanup(pTableListInfo->remainGroups);
331,427,173✔
3704
  pTableListInfo->pTableList = NULL;
331,436,749✔
3705
  pTableListInfo->map = NULL;
331,427,392✔
3706
  taosMemoryFree(pTableListInfo);
331,422,672✔
3707
}
3708

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

3714
  taosArrayClear(pTableListInfo->pTableList);
950,564✔
3715
  taosHashClear(pTableListInfo->map);
953,067✔
3716
  taosHashClear(pTableListInfo->remainGroups);
953,476✔
3717
  taosMemoryFree(pTableListInfo->groupOffset);
953,476✔
3718
  pTableListInfo->numOfOuputGroups = 1;
953,476✔
3719
  pTableListInfo->oneTableForEachGroup = false;
953,112✔
3720
}
3721

3722
static int32_t orderbyGroupIdComparFn(const void* p1, const void* p2) {
607,456,993✔
3723
  STableKeyInfo* pInfo1 = (STableKeyInfo*)p1;
607,456,993✔
3724
  STableKeyInfo* pInfo2 = (STableKeyInfo*)p2;
607,456,993✔
3725

3726
  if (pInfo1->groupId == pInfo2->groupId) {
607,456,993✔
3727
    return 0;
567,318,725✔
3728
  } else {
3729
    return pInfo1->groupId < pInfo2->groupId ? -1 : 1;
40,137,707✔
3730
  }
3731
}
3732

3733
int32_t sortTableGroup(STableListInfo* pTableListInfo) {
34,384,391✔
3734
  int32_t code = TSDB_CODE_SUCCESS;
34,384,391✔
3735
  taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
34,384,391✔
3736
  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
34,436,432✔
3737
  if (size == 0) {
34,414,212✔
3738
    pTableListInfo->numOfOuputGroups = 0;
×
3739
    return code;
×
3740
  }
3741

3742
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
34,414,212✔
3743
  if (!pList) {
34,433,075✔
3744
    code = terrno;
×
3745
    goto end;
×
3746
  }
3747

3748
  STableKeyInfo* pInfo = taosArrayGet(pTableListInfo->pTableList, 0);
34,433,075✔
3749
  if (pInfo == NULL) {
34,399,298✔
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;
34,399,298✔
3755

3756
  int32_t start = 0;
34,408,533✔
3757
  void*   tmp = taosArrayPush(pList, &start);
34,433,636✔
3758
  if (tmp == NULL) {
34,433,636✔
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) {
175,330,857✔
3765
    pInfo = taosArrayGet(pTableListInfo->pTableList, i);
140,900,192✔
3766
    if (pInfo == NULL) {
140,891,068✔
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) {
140,891,068✔
3772
      tmp = taosArrayPush(pList, &i);
8,238,106✔
3773
      if (tmp == NULL) {
8,238,106✔
3774
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3775
        code = terrno;
×
3776
        goto end;
×
3777
      }
3778
      gid = pInfo->groupId;
8,238,106✔
3779
    }
3780
  }
3781

3782
  pTableListInfo->numOfOuputGroups = taosArrayGetSize(pList);
34,422,600✔
3783
  pTableListInfo->groupOffset = taosMemoryMalloc(sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
34,423,147✔
3784
  if (pTableListInfo->groupOffset == NULL) {
34,406,338✔
3785
    code = terrno;
×
3786
    goto end;
×
3787
  }
3788

3789
  memcpy(pTableListInfo->groupOffset, taosArrayGet(pList, 0), sizeof(int32_t) * pTableListInfo->numOfOuputGroups);
34,396,375✔
3790

3791
end:
34,412,433✔
3792
  taosArrayDestroy(pList);
34,416,953✔
3793
  return code;
34,409,341✔
3794
}
3795

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

3800
  bool   groupByTbname = groupbyTbname(group);
281,216,751✔
3801
  size_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
281,230,479✔
3802
  if (!numOfTables) {
281,189,612✔
3803
    return code;
×
3804
  }
3805
  qDebug("numOfTables:%zu, groupByTbname:%d, group:%p", numOfTables, groupByTbname, group);
281,189,612✔
3806
  if (group == NULL || groupByTbname) {
281,132,746✔
3807
    if (tsCountAlwaysReturnValue && QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode) &&
276,096,225✔
3808
        ((STableScanPhysiNode*)pScanNode)->needCountEmptyTable) {
228,844,679✔
3809
      pTableListInfo->remainGroups =
17,321,381✔
3810
          taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
17,321,381✔
3811
      if (pTableListInfo->remainGroups == NULL) {
17,321,381✔
3812
        return terrno;
×
3813
      }
3814

3815
      for (int i = 0; i < numOfTables; i++) {
80,525,124✔
3816
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
63,204,604✔
3817
        if (!info) {
63,204,604✔
3818
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3819
          return terrno;
×
3820
        }
3821
        info->groupId = groupByTbname ? info->uid : 0;
63,204,604✔
3822
        int32_t tempRes = taosHashPut(pTableListInfo->remainGroups, &(info->groupId), sizeof(info->groupId),
63,203,412✔
3823
                                      &(info->uid), sizeof(info->uid));
63,204,008✔
3824
        if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
63,203,743✔
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++) {
933,819,238✔
3831
        STableKeyInfo* info = taosArrayGet(pTableListInfo->pTableList, i);
674,995,554✔
3832
        if (!info) {
674,939,023✔
3833
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3834
          return terrno;
×
3835
        }
3836
        info->groupId = groupByTbname ? info->uid : 0;
674,939,023✔
3837
        
3838
      }
3839
    }
3840
    if (groupIdMap && group != NULL){
276,144,204✔
3841
      getColInfoResultForGroupbyForStream(pHandle->vnode, group, pTableListInfo, pAPI, groupIdMap);
103,330✔
3842
    }
3843

3844
    pTableListInfo->oneTableForEachGroup = groupByTbname;
276,144,204✔
3845
    if (numOfTables == 1 && pTableListInfo->idInfo.tableType == TSDB_CHILD_TABLE) {
276,118,924✔
3846
      pTableListInfo->oneTableForEachGroup = true;
116,311,359✔
3847
    }
3848

3849
    if (groupSort && groupByTbname) {
276,097,679✔
3850
      taosArraySort(pTableListInfo->pTableList, orderbyGroupIdComparFn);
2,689,444✔
3851
      pTableListInfo->numOfOuputGroups = numOfTables;
2,690,186✔
3852
    } else if (groupByTbname && pScanNode->groupOrderScan) {
273,408,235✔
3853
      pTableListInfo->numOfOuputGroups = numOfTables;
32,028✔
3854
    } else {
3855
      pTableListInfo->numOfOuputGroups = 1;
273,377,995✔
3856
    }
3857
    if (groupSort || pScanNode->groupOrderScan) {
276,153,434✔
3858
      code = sortTableGroup(pTableListInfo);
34,237,828✔
3859
    }
3860
  } else {
3861
    bool initRemainGroups = false;
5,036,521✔
3862
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pScanNode)) {
5,036,521✔
3863
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pScanNode;
4,901,146✔
3864
      if (tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable &&
4,901,146✔
3865
          !(groupSort || pScanNode->groupOrderScan)) {
2,312,341✔
3866
        initRemainGroups = true;
2,284,879✔
3867
      }
3868
    }
3869

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

3875
    if (pScanNode->groupOrderScan) pTableListInfo->numOfOuputGroups = taosArrayGetSize(pTableListInfo->pTableList);
5,036,358✔
3876

3877
    if (groupSort || pScanNode->groupOrderScan) {
5,036,358✔
3878
      code = sortTableGroup(pTableListInfo);
228,495✔
3879
    }
3880
  }
3881

3882
  qTrace("EXEC_GROUP_TRACE build_group_map done scanType:%d group:%p groupByTbname:%d groupSort:%d groupOrderScan:%d "
281,128,347✔
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);
281,136,415✔
3890
  for (int32_t i = 0; i < size; ++i) {
1,046,373,162✔
3891
    STableKeyInfo* p = taosArrayGet(pTableListInfo->pTableList, i);
765,132,878✔
3892
    if (!p) {
765,023,191✔
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));
765,023,191✔
3897
    if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
765,278,933✔
3898
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
3899
      return tempRes;
×
3900
    }
3901
  }
3902

3903
  return code;
281,269,597✔
3904
}
3905

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

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

3917
  if (pHandle->uid != 0) {
306,367,490✔
3918
    pScanNode->uid = pHandle->uid;
122,467✔
3919
    pScanNode->tableType = TSDB_CHILD_TABLE;
122,467✔
3920
  }
3921
  uint8_t digest[17] = {0};
306,637,404✔
3922
  int32_t code = getTableList(pHandle->vnode, pScanNode, pTagCond, pTagIndexCond, pTableListInfo, digest, idStr,
306,480,294✔
3923
                              &pTaskInfo->storageAPI, pTaskInfo->pStreamRuntimeInfo);
306,578,309✔
3924
  if (code != TSDB_CODE_SUCCESS) {
306,670,080✔
3925
    qError("failed to getTableList, code:%s", tstrerror(code));
1,094✔
3926
    return code;
1,094✔
3927
  }
3928

3929
  int32_t numOfTables = taosArrayGetSize(pTableListInfo->pTableList);
306,668,986✔
3930

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

3936
  if (numOfTables == 0) {
306,722,976✔
3937
    qDebug("no table qualified for query, %s", idStr);
25,517,472✔
3938
    return TSDB_CODE_SUCCESS;
25,514,887✔
3939
  }
3940

3941
  code = buildGroupIdMapForAllTables(pTableListInfo, pHandle, pScanNode, pGroupTags, groupSort, digest, &pTaskInfo->storageAPI, groupIdMap, false);
281,205,504✔
3942
  if (code != TSDB_CODE_SUCCESS) {
281,240,673✔
3943
    return code;
×
3944
  }
3945

3946
  pTaskInfo->cost.groupIdMapTime = (taosGetTimestampUs() - st1) / 1000.0;
281,240,092✔
3947
  qDebug("generate group id map completed, elapsed time:%.2f ms %s", pTaskInfo->cost.groupIdMapTime, idStr);
281,224,974✔
3948

3949
  return TSDB_CODE_SUCCESS;
281,227,511✔
3950
}
3951

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

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

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

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

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

3978
_exit:
×
3979

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

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

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

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

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

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

4041
_exit:
×
4042

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

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

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

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

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

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

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

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

4090
_exit:
×
4091

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

4099

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

4144
  return TSDB_CODE_SUCCESS;
309,349,244✔
4145
}
4146

4147

4148
char* getStreamOpName(uint16_t opType) {
8,766,833✔
4149
  switch (opType) {
8,766,833✔
4150
    case QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN:
×
4151
      return "stream scan";
×
4152
    case QUERY_NODE_PHYSICAL_PLAN_PROJECT:
8,284,581✔
4153
      return "project";
8,284,581✔
4154
    case QUERY_NODE_PHYSICAL_PLAN_EXTERNAL_WINDOW:
482,252✔
4155
      return "external window";
482,252✔
4156
  }
4157
  return "error name";
×
4158
}
4159

4160
void printDataBlock(SSDataBlock* pBlock, const char* flag, const char* taskIdStr, int64_t qId) {
664,546,948✔
4161
  if (qDebugFlag & DEBUG_TRACE) {
664,546,948✔
4162
    if (!pBlock) {
1,685,780✔
4163
      qDebug("%" PRIx64 " %s %s %s: Block is Null", qId, taskIdStr, flag, __func__);
60,090✔
4164
      return;
61,513✔
4165
    } else if (pBlock->info.rows == 0) {
1,625,690✔
4166
      qDebug("%" PRIx64 " %s %s %s: Block is Empty.", qId, taskIdStr, flag, __func__);
10,999✔
4167
      return;
10,999✔
4168
    }
4169
    
4170
    char*   pBuf = NULL;
1,614,691✔
4171
    int32_t code = dumpBlockData(pBlock, flag, &pBuf, taskIdStr, qId);
1,614,691✔
4172
    if (code == 0) {
1,614,691✔
4173
      qDebugL("%" PRIx64 " %s %s", qId, __func__, pBuf);
1,614,691✔
4174
      taosMemoryFree(pBuf);
1,614,691✔
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,167,390✔
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;
309,479,720✔
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,433,378,110✔
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,434,202,260✔
4238
          p += varDataTLen(val);
1,433,545,590✔
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,889,450✔
4251
  uint32_t        colNum = pSortGroupCols->size;
1,889,450✔
4252
  SColumnDataAgg* pColAgg = NULL;
1,889,450✔
4253
  char*           isNull = keyBuf;
1,889,450✔
4254
  char*           p = keyBuf + sizeof(int8_t) * colNum;
1,889,450✔
4255

4256
  for (int32_t i = 0; i < colNum; ++i) {
5,172,800✔
4257
    const SColumn*         pCol = (SColumn*)TARRAY_GET_ELEM(pSortGroupCols, i);
3,283,350✔
4258
    const SColumnInfoData* pColInfoData = TARRAY_GET_ELEM(pBlock->pDataBlock, pCol->slotId);
3,283,350✔
4259
    if (pCol->slotId > pBlock->pDataBlock->size) continue;
3,283,350✔
4260

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

4263
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
6,566,700✔
4264
      isNull[i] = 1;
159,000✔
4265
    } else {
4266
      isNull[i] = 0;
3,124,350✔
4267
      const char* val = colDataGetData(pColInfoData, rowIndex);
3,124,350✔
4268
      if (pCol->type == TSDB_DATA_TYPE_JSON) {
3,124,350✔
4269
        int32_t len = getJsonValueLen(val);
×
4270
        memcpy(p, val, len);
×
4271
        p += len;
×
4272
      } else if (IS_VAR_DATA_TYPE(pCol->type)) {
3,124,350✔
4273
        if (IS_STR_DATA_BLOB(pCol->type)) {
1,168,650✔
4274
          blobDataCopy(p, val);
×
4275
          p += blobDataTLen(val);
×
4276
        } else {
4277
          varDataCopy(p, val);
1,168,650✔
4278
          p += varDataTLen(val);
1,168,650✔
4279
        }
4280
      } else {
4281
        memcpy(p, val, pCol->bytes);
1,955,700✔
4282
        p += pCol->bytes;
1,955,700✔
4283
      }
4284
    }
4285
  }
4286
  return (int32_t)(p - keyBuf);
1,889,450✔
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) {
125,080✔
4303
  SNode*     node;
4304
  SNodeList* ret = NULL;
125,080✔
4305
  FOREACH(node, pSortKeys) {
381,070✔
4306
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)node;
255,990✔
4307
    int32_t           code = nodesListMakeAppend(&ret, pSortKey->pExpr);
255,990✔
4308
    if (code != TSDB_CODE_SUCCESS) {
255,990✔
4309
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4310
      terrno = code;
×
4311
      return NULL;
×
4312
    }
4313
  }
4314
  return ret;
124,550✔
4315
}
4316

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

4330
_end:
124,550✔
4331
  if (code != TSDB_CODE_SUCCESS) {
124,550✔
4332
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4333
  }
4334
  return code;
125,080✔
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) {
32,018,291✔
4366
  int32_t code = 0;
32,018,291✔
4367
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
32,018,291✔
4368
  if (pBlock == NULL) {
32,018,288✔
4369
    return terrno;
×
4370
  }
4371

4372
  pBlock->pDataBlock = taosArrayInit(cols, sizeof(SColumnInfoData));
32,018,288✔
4373
  if (pBlock->pDataBlock == NULL) {
32,017,684✔
4374
    code = terrno;
×
4375
    taosMemoryFree(pBlock);
×
4376
    return code;
×
4377
  }
4378

4379
  for (int32_t i = 0; i < cols; ++i) {
65,377,875✔
4380
    SColumnInfoData idata = {0};
33,358,983✔
4381
    if (pResType) {
33,358,983✔
4382
      idata = createColumnInfoData(pResType->type, pResType->bytes, 0);
31,327,759✔
4383
      idata.info.scale = pResType->scale;
31,327,155✔
4384
      idata.info.precision = pResType->precision;
31,327,155✔
4385
    }
4386
    
4387
    code = blockDataAppendColInfo(pBlock, &idata);
33,358,379✔
4388
    if (code != TSDB_CODE_SUCCESS) {
33,360,191✔
4389
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
1,812✔
4390
      blockDataDestroy(pBlock);
1,812✔
4391
      *ppBlock = NULL;
×
4392
      return code;
×
4393
    }
4394
  }
4395

4396
  *ppBlock = pBlock;
32,018,892✔
4397

4398
  return code;
32,018,288✔
4399
}
4400

4401

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

4407
  char* pNextStart = pRetrieveRsp->data;
32,018,288✔
4408
  char* pStart = pNextStart;
32,018,892✔
4409

4410
  int32_t index = 0;
32,018,892✔
4411

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

4417
  int32_t compLen = *(int32_t*)pStart;
32,018,288✔
4418
  pStart += sizeof(int32_t);
32,017,080✔
4419

4420
  int32_t rawLen = *(int32_t*)pStart;
32,018,288✔
4421
  pStart += sizeof(int32_t);
32,019,496✔
4422
  QUERY_CHECK_CONDITION((compLen <= rawLen && compLen != 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
32,018,892✔
4423

4424
  pNextStart = pStart + compLen;
32,018,892✔
4425
  if (pRetrieveRsp->compressed && (compLen < rawLen)) {
32,017,684✔
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);
32,017,080✔
4432
  if (code != 0) {
32,018,895✔
4433
    taosMemoryFreeClear(pRetrieveRsp);
×
4434
    goto _end;
×
4435
  }
4436

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

4445
int32_t setValueFromResBlock(STaskSubJobCtx* ctx, SValueNode* pRes, SSDataBlock* pBlock) {
20,939,984✔
4446
  int32_t code = 0;
20,939,984✔
4447
  bool needFree = true;
20,939,984✔
4448
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
20,939,984✔
4449
  if (NULL == pBlock->pDataBlock || 1 > colNum || pBlock->info.rows > 1) {
20,939,984✔
4450
    qError("%s invalid scl fetch res block, pDataBlock:%p, colNum:%d, rows:%" PRId64, 
×
4451
      ctx->idStr, pBlock->pDataBlock, colNum, pBlock->info.rows);
4452
    return TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;
×
4453
  }
4454
  
4455
  pRes->flag &= (~VALUE_FLAG_VAL_UNSET);
20,939,984✔
4456
  pRes->translate = true;
20,940,588✔
4457
  
4458
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
20,941,192✔
4459
  if (colDataIsNull_s(pCol, 0)) {
20,939,380✔
4460
    pRes->isNull = true;
5,662,788✔
4461
  } else {
4462
    code = nodesSetValueNodeValueExt(pRes, colDataGetData(pCol, 0), &needFree);
15,276,592✔
4463
  }
4464

4465
  if (!needFree) {
20,938,776✔
4466
    pCol->pData = NULL;
2,081,664✔
4467
  }
4468

4469
  return code;
20,938,776✔
4470
}
4471

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

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

4479
  if (IS_STREAM_MODE(pTaskInfo)) {
3,714,289✔
4480
    SNode** ppRes = taosArrayGet(ctx->subResNodes, pParam->subQIdx);
86,292✔
4481
    if (NULL == *ppRes && 0 == pRsp->numOfRows) {
86,292✔
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) {
86,292✔
4488
      ctx->code = createExprSubQResBlock(&pResBlock, &((SRemoteValueNode*)pParam->pRes)->val.node.resType, 1);
43,146✔
4489
      if (TSDB_CODE_SUCCESS == ctx->code) {
43,146✔
4490
        ctx->code = blockDataEnsureCapacity(pResBlock, 1);
43,146✔
4491
      }
4492
      if (TSDB_CODE_SUCCESS == ctx->code) {
43,146✔
4493
        ctx->code = extractSingleRspBlock(pRsp, pResBlock);
43,146✔
4494
      }
4495
      if (TSDB_CODE_SUCCESS == ctx->code) {
43,146✔
4496
        ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
43,146✔
4497
        pRemote->val.node.type = QUERY_NODE_REMOTE_VALUE;
43,146✔
4498
      }
4499
      if (TSDB_CODE_SUCCESS == ctx->code) {
43,146✔
4500
        taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
43,146✔
4501
      }
4502

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

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

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

4519
    return;
86,292✔
4520
  }
4521

4522
  *fetchDone = true;
3,627,997✔
4523

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

4529
    return;
×
4530
  }
4531

4532
  if (1 != pRsp->numOfCols && pRsp->numOfRows > 0) {
3,627,393✔
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,627,997✔
4540
    pRemote->val.node.type = QUERY_NODE_VALUE;
192,664✔
4541
    pRemote->val.isNull = true;
192,664✔
4542
    pRemote->val.translate = true;
192,664✔
4543
    pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
192,664✔
4544
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
192,664✔
4545

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

4564
  blockDataDestroy(pResBlock);  
3,435,333✔
4565
}
4566

4567

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

4576
  pRes->hasValue = true;
10,386,570✔
4577
  
4578
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, 0);
10,386,570✔
4579
  TAOS_CHECK_EXIT(scalarBuildRemoteListHash(ctx->idStr, pRes, pCol, pBlock->info.rows));
10,387,171✔
4580

4581
_exit:
9,579,146✔
4582

4583
  if (code) {
10,386,458✔
4584
    qError("%s %s failed with error: %s", ctx->idStr, __func__, tstrerror(code));
807,312✔
4585
  }
4586
  
4587
  return code;
10,386,570✔
4588
}
4589

4590

4591

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

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

4599
  if (pRsp->numOfRows > 0) {
13,677,540✔
4600
    if (1 != pRsp->numOfCols) {
10,387,171✔
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,385,969✔
4608
    if (TSDB_CODE_SUCCESS == ctx->code) {
10,387,171✔
4609
      ctx->code = blockDataEnsureCapacity(pResBlock, pRsp->numOfRows);
10,387,171✔
4610
    }
4611
    if (TSDB_CODE_SUCCESS == ctx->code) {
10,387,171✔
4612
      ctx->code = extractSingleRspBlock(pRsp, pResBlock);
10,387,171✔
4613
    }
4614
    if (TSDB_CODE_SUCCESS == ctx->code) {
10,387,171✔
4615
      ctx->code = setValueListFromResBlock(ctx, pRemote, pResBlock);
10,387,171✔
4616
    }
4617
    if (TSDB_CODE_SUCCESS == ctx->code && pRsp->completed) {
10,387,171✔
4618
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
8,849,984✔
4619
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
8,849,984✔
4620
    }
4621

4622
    blockDataDestroy(pResBlock);  
10,387,171✔
4623
  } else if (0 == pRsp->numOfRows && pRsp->completed) {
3,290,369✔
4624
    if (!pRemote->hasValue) {
3,290,369✔
4625
      ctx->code = scalarBuildRemoteListHash(ctx->idStr, pRemote, NULL, 0);
3,276,541✔
4626
    }
4627
    if (TSDB_CODE_SUCCESS == ctx->code) {    
3,290,369✔
4628
      pRemote->flag &= (~VALUELIST_FLAG_VAL_UNSET);
3,016,817✔
4629
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
3,016,817✔
4630
    }
4631
  }
4632

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

4642
    pRsp->completed = true;
14,417✔
4643
  }
4644

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

4647
  if (!(*fetchDone)) {
13,677,540✔
4648
    int32_t code = sendFetchRemoteNodeReq(ctx, pParam->subQIdx, pParam->pRes, false);
729,274✔
4649
    if (TSDB_CODE_SUCCESS != code) {
729,274✔
4650
      ctx->code = code;
×
4651
      *fetchDone = true;
×
4652
    }
4653
  }
4654
}
4655

4656
int32_t setRowHasNullFromResBlock(STaskSubJobCtx* ctx, bool* hasNull, SSDataBlock* pBlock) {
17,462,109✔
4657
  int32_t code = 0;
17,462,109✔
4658
  int32_t colNum = taosArrayGetSize(pBlock->pDataBlock);
17,462,109✔
4659
  if (2 != colNum) {
17,462,109✔
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);
17,462,109✔
4665
  if (colDataIsNull_s(pCol, 0)) {
17,462,109✔
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);
17,462,109✔
4670
  }
4671

4672
  return code;
17,462,109✔
4673
}
4674

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

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

4681
  SRemoteRowNode* pRemote = (SRemoteRowNode*)pParam->pRes;
23,667,655✔
4682

4683
  if (IS_STREAM_MODE(pTaskInfo)) {
23,667,655✔
4684
    SNode** ppRes = taosArrayGet(ctx->subResNodes, pParam->subQIdx);
1,172✔
4685
    if (NULL == *ppRes && 0 == pRsp->numOfRows) {
1,172✔
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,172✔
4694
      if (2 != pRsp->numOfCols) {
586✔
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);
586✔
4702
      if (TSDB_CODE_SUCCESS == ctx->code) {
586✔
4703
        SColumnInfoData idata = createColumnInfoData(TSDB_DATA_TYPE_BOOL, tDataTypes[TSDB_DATA_TYPE_BOOL].bytes, 0);
586✔
4704
        ctx->code = blockDataAppendColInfo(pResBlock, &idata);
586✔
4705
      }
4706
      if (TSDB_CODE_SUCCESS == ctx->code) {
586✔
4707
        ctx->code = blockDataEnsureCapacity(pResBlock, 1);
586✔
4708
      }
4709
      if (TSDB_CODE_SUCCESS == ctx->code) {
586✔
4710
        ctx->code = extractSingleRspBlock(pRsp, pResBlock);
586✔
4711
      }
4712
      if (TSDB_CODE_SUCCESS == ctx->code) {
586✔
4713
        ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
586✔
4714
      }
4715
      if (TSDB_CODE_SUCCESS == ctx->code) {
586✔
4716
        ctx->code = setRowHasNullFromResBlock(ctx, &pRemote->hasNull, pResBlock);
586✔
4717
      }
4718
      if (TSDB_CODE_SUCCESS == ctx->code) {
586✔
4719
        pRemote->valSet = true;
586✔
4720
        pRemote->hasValue = true;
586✔
4721
        pRemote->val.node.type = QUERY_NODE_REMOTE_ROW;
586✔
4722
        taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
586✔
4723
      }
4724

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

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

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

4740
    return;
1,172✔
4741
  }
4742

4743
  *fetchDone = true;
23,666,483✔
4744

4745
  if (pRsp->numOfRows > 1 || pRsp->numOfBlocks > 1 || !pRsp->completed) {
23,666,483✔
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) {
23,666,483✔
4754
    pRemote->valSet = true;
6,204,960✔
4755
    pRemote->hasValue = false;
6,204,960✔
4756
    pRemote->hasNull = false;
6,204,960✔
4757
    pRemote->val.isNull = true;
6,204,960✔
4758
    pRemote->val.translate = true;
6,204,960✔
4759
    pRemote->val.flag &= (~VALUE_FLAG_VAL_UNSET);
6,204,960✔
4760
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
6,204,960✔
4761

4762
    return;
6,204,960✔
4763
  }
4764

4765
  if (2 != pRsp->numOfCols) {
17,461,523✔
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);
17,461,523✔
4773
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,461,523✔
4774
    SColumnInfoData idata = createColumnInfoData(TSDB_DATA_TYPE_BOOL, tDataTypes[TSDB_DATA_TYPE_BOOL].bytes, 0);
17,461,523✔
4775
    ctx->code = blockDataAppendColInfo(pResBlock, &idata);
17,461,523✔
4776
  }
4777
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,461,523✔
4778
    ctx->code = blockDataEnsureCapacity(pResBlock, 1);
17,461,523✔
4779
  }
4780
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,461,523✔
4781
    ctx->code = extractSingleRspBlock(pRsp, pResBlock);
17,461,523✔
4782
  }
4783
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,461,523✔
4784
    ctx->code = setValueFromResBlock(ctx, &pRemote->val, pResBlock);
17,461,523✔
4785
  }
4786
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,461,523✔
4787
    ctx->code = setRowHasNullFromResBlock(ctx, &pRemote->hasNull, pResBlock);
17,461,523✔
4788
  }
4789
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,461,523✔
4790
    taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
17,461,523✔
4791
  }
4792
  if (TSDB_CODE_SUCCESS == ctx->code) {
17,461,523✔
4793
    pRemote->valSet = true;
17,461,523✔
4794
    pRemote->hasValue = true;
17,461,523✔
4795
  }
4796

4797
  blockDataDestroy(pResBlock);
17,461,523✔
4798
}
4799

4800

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

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

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

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

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

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

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

4837
  if (!(*fetchDone)) {
307,878✔
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
}
307,277✔
4845

4846

4847

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

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

4858
_exit:
691,133✔
4859

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

4865
  *ppBlock = NULL;
691,133✔
4866
  
4867
  return code;
691,133✔
4868
}
4869

4870

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

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

4877
  if (pRsp->numOfRows > 0) {
693,815✔
4878
    ctx->code = createExprSubQResBlock(&pResBlock, NULL, pRemote->resCols);
691,133✔
4879
    if (TSDB_CODE_SUCCESS == ctx->code) {
691,133✔
4880
      ctx->code = extractSingleRspBlock(pRsp, pResBlock);
691,133✔
4881
    }
4882
    if (TSDB_CODE_SUCCESS == ctx->code) {
691,133✔
4883
      ctx->code = setTableFromResBlock(ctx, pRemote, &pResBlock);
691,133✔
4884
    }
4885
    if (TSDB_CODE_SUCCESS == ctx->code && pRsp->completed) {
691,133✔
4886
      pRemote->flag &= (~REMOTE_TABLE_FLAG_VAL_UNSET);
564,185✔
4887
      taosArraySet(ctx->subResNodes, pParam->subQIdx, &pParam->pRes);
564,185✔
4888
    }
4889

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

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

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

4907

4908
int32_t remoteFetchCallBack(void* param, SDataBuf* pMsg, int32_t code) {
43,022,147✔
4909
  taosMemoryFreeClear(pMsg->pEpSet);
43,022,147✔
4910

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

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

4926
  if (ctx->transporterId > 0) {
43,016,711✔
4927
    int32_t ret = asyncFreeConnById(ctx->rpcHandle, ctx->transporterId);
43,019,121✔
4928
    if (ret != 0) {
43,017,916✔
4929
      qDebug("%s failed to free subQ rpc handle, code:%s, subQIdx:%d", ctx->idStr, tstrerror(ret), pParam->subQIdx);
×
4930
    }
4931
    ctx->transporterId = -1;
43,017,916✔
4932
  }
4933

4934
  if (0 == code && NULL == pMsg->pData) {
43,016,711✔
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) {
43,016,711✔
4940
    SRetrieveTableRsp* pRsp = pMsg->pData;
42,059,975✔
4941
    pRsp->numOfRows = htobe64(pRsp->numOfRows);
42,060,576✔
4942
    pRsp->compLen = htonl(pRsp->compLen);
42,061,781✔
4943
    pRsp->payloadLen = htonl(pRsp->payloadLen);
42,061,784✔
4944
    pRsp->numOfCols = htonl(pRsp->numOfCols);
42,059,975✔
4945
    pRsp->useconds = htobe64(pRsp->useconds);
42,060,576✔
4946
    pRsp->numOfBlocks = htonl(pRsp->numOfBlocks);
42,061,177✔
4947

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

4951
    ctx->blockIdx++;
42,059,972✔
4952

4953
    switch (nodeType(pParam->pRes)) {
42,061,177✔
4954
      case QUERY_NODE_REMOTE_VALUE: {
3,714,893✔
4955
        bool fetchDone = false;
3,714,893✔
4956
        handleRemoteValueRes(pParam, ctx, pRsp, &fetchDone);
3,714,893✔
4957
        qDebug("%s subQIdx %d handle remote value, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
3,713,685✔
4958
        if (!fetchDone) {
3,713,685✔
4959
          goto _exit;
43,146✔
4960
        }
4961
        break;
3,670,539✔
4962
      }
4963
      case QUERY_NODE_REMOTE_VALUE_LIST: {
13,676,939✔
4964
        bool fetchDone = false;
13,676,939✔
4965
        handleRemoteValueListRes(pParam, ctx, pRsp, &fetchDone);
13,676,939✔
4966
        qDebug("%s subQIdx %d handle remote value list finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
13,676,939✔
4967
        if (!fetchDone) {
13,677,540✔
4968
          goto _exit;
729,274✔
4969
        }
4970
        break;
12,948,266✔
4971
      }  
4972
      case QUERY_NODE_REMOTE_ROW: {
23,667,655✔
4973
        bool fetchDone = false;
23,667,655✔
4974
        handleRemoteRowRes(pParam, ctx, pRsp, &fetchDone);
23,667,655✔
4975
        qDebug("%s subQIdx %d handle remote row finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
23,667,054✔
4976
        if (!fetchDone) {
23,667,054✔
4977
          goto _exit;
586✔
4978
        }
4979
        break;
23,666,468✔
4980
      }
4981
      case QUERY_NODE_REMOTE_ZERO_ROWS: {
307,277✔
4982
        bool fetchDone = false;
307,277✔
4983
        handleRemoteZeroRowsRes(pParam, ctx, pRsp, &fetchDone);
307,277✔
4984
        qDebug("%s subQIdx %d handle remote zeroRows finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
307,277✔
4985
        if (!fetchDone) {
307,277✔
4986
          goto _exit;
×
4987
        }
4988
        break;
307,277✔
4989
      }
4990
      case QUERY_NODE_REMOTE_TABLE: {
693,815✔
4991
        bool fetchDone = false;
693,815✔
4992
        handleRemoteTableRes(pParam, ctx, pRsp, &fetchDone);
693,815✔
4993
        qDebug("%s subQIdx %d handle remote table finished, fetchDone:%d", idStr, pParam->subQIdx, fetchDone);
693,815✔
4994
        if (!fetchDone) {
693,815✔
4995
          goto _exit;
126,948✔
4996
        }
4997
        break;
566,867✔
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);
956,736✔
5006
    if (ctx->code != code) {
956,736✔
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));
956,736✔
5011
    }
5012
  }
5013

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

5021
_exit:
43,019,728✔
5022

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

5028
  taosMemoryFree(pMsg->pData);
43,019,121✔
5029

5030
  return code;
43,018,520✔
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) {
43,017,940✔
5037
  int32_t          code = TSDB_CODE_SUCCESS;
43,017,940✔
5038
  int32_t          lino = 0;
43,017,940✔
5039
  int32_t          innerIdx = 0;
43,017,940✔
5040
  SExecTaskInfo*   pTaskInfo = ctx->pTaskInfo;
43,017,940✔
5041
  bool             needStreamPesudoFuncVals = false;
43,020,945✔
5042

5043
  if (IS_STREAM_MODE(pTaskInfo)) {
43,020,945✔
5044
    innerIdx = pTaskInfo->pStreamRuntimeInfo->execId / STREAM_TASK_REPLICA_NUM;
515,463✔
5045
    subQIdx = subQIdx * STREAM_TASK_DEPLOY_NUM + innerIdx;
515,463✔
5046
  }
5047

5048
  SDownstreamSourceNode* pSource = (SDownstreamSourceNode*)taosArrayGetP(ctx->subEndPoints, subQIdx);
43,010,852✔
5049

5050
  SResFetchReq req = {0};
43,016,732✔
5051
  req.header.vgId = pSource->addr.nodeId;
43,017,333✔
5052
  req.sId = pSource->sId;
43,011,900✔
5053
  req.clientId = pSource->clientId;
43,023,352✔
5054
  req.taskId = pSource->taskId;
43,023,959✔
5055
  req.srcTaskId = ctx->taskId;
43,020,338✔
5056
  req.blockIdx = ctx->blockIdx;
43,011,897✔
5057
  req.queryId = ctx->queryId;
43,015,512✔
5058
  req.execId = pSource->execId;
43,011,293✔
5059

5060
  if (IS_STREAM_MODE(pTaskInfo)) {
43,015,518✔
5061
    req.queryId = pSource->clientId;
515,463✔
5062
    req.execId = pTaskInfo->pStreamRuntimeInfo->execId;
515,463✔
5063
    req.pStRtFuncInfo = &pTaskInfo->pStreamRuntimeInfo->funcInfo;
515,463✔
5064
    req.reset = reset;
515,463✔
5065

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

5070
  int32_t msgSize = tSerializeSResFetchReq(NULL, 0, &req, needStreamPesudoFuncVals, false);
43,008,279✔
5071
  if (msgSize < 0) {
43,015,524✔
5072
    return msgSize;
×
5073
  }
5074

5075
  void* msg = taosMemoryCalloc(1, msgSize);
43,015,524✔
5076
  if (NULL == msg) {
43,014,304✔
5077
    return terrno;
×
5078
  }
5079

5080
  msgSize = tSerializeSResFetchReq(msg, msgSize, &req, needStreamPesudoFuncVals, false);
43,014,304✔
5081
  if (msgSize < 0) {
43,023,352✔
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
43,023,352✔
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));
43,025,167✔
5093
  if (NULL == pMsgSendInfo) {
43,015,515✔
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));
43,015,515✔
5100
  if (NULL == param) {
43,020,338✔
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) {
43,020,338✔
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;
43,019,130✔
5117
  param->pRes = pRes;
43,016,729✔
5118
  param->subJobRefId = ctx->subJobRefId;
43,016,726✔
5119

5120
  pMsgSendInfo->param = param;
43,015,515✔
5121
  pMsgSendInfo->paramFreeFp = taosAutoMemoryFree;
43,011,897✔
5122
  pMsgSendInfo->msgInfo.pData = msg;
43,014,307✔
5123
  pMsgSendInfo->msgInfo.len = msgSize;
43,014,899✔
5124
  pMsgSendInfo->msgType = pSource->fetchMsgType;
43,013,709✔
5125
  pMsgSendInfo->fp = remoteFetchCallBack;
43,005,857✔
5126
  pMsgSendInfo->requestId = ctx->queryId;
43,020,945✔
5127

5128
  code = asyncSendMsgToServer(ctx->rpcHandle, &pSource->addr.epSet, &ctx->transporterId, pMsgSendInfo);
43,012,498✔
5129
  QUERY_CHECK_CODE(code, lino, _end);
43,026,973✔
5130
      
5131
_end:
43,026,973✔
5132

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

5140
int32_t fetchRemoteNodeImpl(STaskSubJobCtx* ctx, int32_t subQIdx, SNode* pRes) {
42,125,213✔
5141
  int32_t          code = TSDB_CODE_SUCCESS;
42,125,213✔
5142
  int32_t          lino = 0;
42,125,213✔
5143

5144
  ctx->blockIdx = 0;
42,125,213✔
5145

5146
  code = sendFetchRemoteNodeReq(ctx, subQIdx, pRes, true);
42,125,213✔
5147
  QUERY_CHECK_CODE(code, lino, _end);
42,125,811✔
5148

5149
  code = qSemWait(ctx->pTaskInfo, &ctx->ready);
42,125,811✔
5150
  if (isTaskKilled(ctx->pTaskInfo)) {
42,127,623✔
5151
    code = getTaskCode(ctx->pTaskInfo);
20,536✔
5152
  } else {
5153
    code = ctx->code;
42,106,486✔
5154
  }
5155
      
5156
_end:
42,127,022✔
5157

5158
  if (code != TSDB_CODE_SUCCESS) {
42,127,022✔
5159
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
2,046,056✔
5160
  }
5161
  return code;
42,127,022✔
5162
}
5163

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

5203
_exit:
130,134✔
5204

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

5209
  return code;
130,134✔
5210
}
5211

5212
int32_t qFetchRemoteNode(void* pCtx, int32_t subQIdx, SNode* pRes) {
42,255,951✔
5213
  STaskSubJobCtx*  ctx = (STaskSubJobCtx*)pCtx;
42,255,951✔
5214
  int32_t code = 0, lino = 0;
42,255,951✔
5215
  int32_t       subEndPoinsNum = taosArrayGetSize(ctx->subEndPoints);
42,255,951✔
5216
  if (subQIdx >= subEndPoinsNum) {
42,245,094✔
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);
42,245,094✔
5222
  if (NULL == *ppRes) {
42,250,530✔
5223
    TAOS_CHECK_EXIT(fetchRemoteNodeImpl(ctx, subQIdx, pRes));
42,120,393✔
5224
    *ppRes = pRes;
40,080,362✔
5225
  } else {
5226
    TAOS_CHECK_EXIT(remoteNodeCopy(*ppRes, pRes));
130,134✔
5227
  }
5228

5229
_exit:
129,533✔
5230

5231
  if (code) {
42,256,552✔
5232
    qError("%s %s failed at line %d since %s", ctx->idStr, __func__, lino, tstrerror(code));
2,046,056✔
5233
  } else {
5234
    qDebug("%s %s subQIdx %d succeed", ctx->idStr, __func__, subQIdx);
40,210,496✔
5235
  }
5236

5237
  return code;
42,257,156✔
5238
}
5239

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

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

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

5260
  return -1;
×
5261
}
5262

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