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

taosdata / TDengine / #4800

16 Oct 2025 09:19AM UTC coverage: 53.935% (-7.1%) from 61.083%
#4800

push

travis-ci

web-flow
Merge b32e3a393 into a190048d5

134724 of 323629 branches covered (41.63%)

Branch coverage included in aggregate %.

184803 of 268802 relevant lines covered (68.75%)

69058627.2 hits per line

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

47.17
/source/libs/executor/src/groupcacheoperator.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 "executorInt.h"
17
#include "filter.h"
18
#include "function.h"
19
#include "operator.h"
20
#include "os.h"
21
#include "querynodes.h"
22
#include "querytask.h"
23
#include "tcompare.h"
24
#include "tdatablock.h"
25
#include "thash.h"
26
#include "tmsg.h"
27
#include "ttypes.h"
28
#include "groupcache.h"
29

30

31
static void removeGroupCacheFile(SGroupCacheFileInfo* pFileInfo) {
61,745✔
32
  if (pFileInfo->fd.fd) {
61,745!
33
    if (taosCloseFile(&pFileInfo->fd.fd) < 0) {
61,745!
34
      qError("close group cache file failed, fd:%p, error:%s", pFileInfo->fd.fd, tstrerror(terrno));
×
35
    }
36
    pFileInfo->fd.fd = NULL;
61,745✔
37
    (void)taosThreadMutexDestroy(&pFileInfo->fd.mutex);
61,745✔
38
  }
39
  pFileInfo->deleted = true;
61,745✔
40
}
61,745✔
41

42

43
static int32_t initGroupColsInfo(SGroupColsInfo* pCols, bool grpColsMayBeNull, SNodeList* pList) {
×
44
  pCols->colNum = LIST_LENGTH(pList);
×
45
  pCols->withNull = grpColsMayBeNull;  
×
46
  pCols->pColsInfo = taosMemoryMalloc(pCols->colNum * sizeof(SGroupColInfo));
×
47
  if (NULL == pCols->pColsInfo) {
×
48
    return terrno;
×
49
  }
50

51
  int32_t i = 0;
×
52
  SNode* pNode = NULL;
×
53
  FOREACH(pNode, pList) {
×
54
    SColumnNode* pColNode = (SColumnNode*)pNode;
×
55
    pCols->pColsInfo[i].slot = pColNode->slotId;
×
56
    pCols->pColsInfo[i].vardata = IS_VAR_DATA_TYPE(pColNode->node.resType.type);
×
57
    pCols->pColsInfo[i].bytes = pColNode->node.resType.bytes;
×
58
    pCols->bufSize += pColNode->node.resType.bytes;
×
59
    ++i;
×
60
  }  
61

62
  if (pCols->withNull) {
×
63
    pCols->bitMapSize = pCols->colNum / sizeof(int8_t) + ((pCols->colNum % sizeof(int8_t)) ? 1 : 0);
×
64
    pCols->bufSize += pCols->bitMapSize;
×
65
  }
66

67
  if (pCols->colNum > 1) {
×
68
    pCols->pBuf = taosMemoryMalloc(pCols->bufSize);
×
69
    if (NULL == pCols->pBuf) {
×
70
      return terrno;
×
71
    }
72
  }
73

74
  return TSDB_CODE_SUCCESS;
×
75
}
76

77
static void logGroupCacheExecInfo(SGroupCacheOperatorInfo* pGrpCacheOperator) {
33,099✔
78
  if (pGrpCacheOperator->downstreamNum <= 0 || NULL == pGrpCacheOperator->execInfo.pDownstreamBlkNum) {
33,099!
79
    return;
×
80
  }
81

82
  int32_t bufSize = pGrpCacheOperator->downstreamNum * 32 + 100;
33,099✔
83
  char* buf = taosMemoryMalloc(bufSize);
33,099!
84
  if (NULL == buf) {
33,099!
85
    return;
×
86
  }
87
  int32_t offset = tsnprintf(buf, bufSize, "groupCache exec info, downstreamBlkNum:");
33,099✔
88
  for (int32_t i = 0; i < pGrpCacheOperator->downstreamNum; ++i) {
99,297✔
89
    offset += tsnprintf(buf + offset, bufSize, " %" PRId64 , pGrpCacheOperator->execInfo.pDownstreamBlkNum[i]);
66,198✔
90
  }
91
  qDebug("%s", buf);
33,099!
92
  taosMemoryFree(buf);
33,099!
93
}
94

95
static void freeSGcSessionCtx(void* p) {
217,075✔
96
  SGcSessionCtx* pSession = p;
217,075✔
97
  if (pSession->semInit) {
217,075!
98
    if (tsem_destroy(&pSession->waitSem) < 0) {
×
99
      qError("tsem_destroy session waitSem failed, error:%s", tstrerror(terrno));
×
100
    }
101
  }
102
}
217,075✔
103

104
static void freeSGroupCacheFileInfo(void* p) {
61,745✔
105
  SGroupCacheFileInfo* pFileInfo = p;
61,745✔
106
  if (pFileInfo->deleted) {
61,745!
107
    return;
×
108
  }
109

110
  removeGroupCacheFile(pFileInfo);
61,745✔
111
}
112

113
static void freeSGcFileCacheCtx(SGcFileCacheCtx* pFileCtx) {
132,396✔
114
  taosHashCleanup(pFileCtx->pCacheFile);
132,396✔
115
  pFileCtx->pCacheFile = NULL;
132,396✔
116
}
132,396✔
117

118
static void freeSGcVgroupCtx(void* p) {
66,198✔
119
  SGcVgroupCtx* pVgCtx = p;
66,198✔
120
  taosArrayDestroy(pVgCtx->pTbList);
66,198✔
121
  freeSGcFileCacheCtx(&pVgCtx->fileCtx);
66,198✔
122
}
66,198✔
123

124
static void freeGcBlockInList(void* p) {
67,217✔
125
  SSDataBlock** ppBlock = p;
67,217✔
126
  if (*ppBlock) {
67,217!
127
    taosArrayDestroy((*ppBlock)->pDataBlock);
67,217✔
128
    taosMemoryFree(*ppBlock);
67,217!
129
  }
130
}
67,217✔
131

132
static void freeSGcDownstreamCtx(SGcDownstreamCtx* pCtx) {
66,198✔
133
  taosArrayDestroy(pCtx->pNewGrpList);
66,198✔
134
  taosHashCleanup(pCtx->pGrpHash);
66,198✔
135
  tSimpleHashCleanup(pCtx->pVgTbHash);
66,198✔
136

137
  taosArrayDestroyEx(pCtx->pFreeBlock, freeGcBlockInList);
66,198✔
138
  taosHashCleanup(pCtx->pSessions);
66,198✔
139
  taosHashCleanup(pCtx->pWaitSessions);
66,198✔
140
  freeSGcFileCacheCtx(&pCtx->fileCtx);
66,198✔
141
}
66,198✔
142

143
static void destroyGroupCacheDownstreamCtx(SGroupCacheOperatorInfo* pGrpCacheOperator) {
33,099✔
144
  if (NULL == pGrpCacheOperator->pDownstreams) {
33,099!
145
    return;
×
146
  }
147
  
148
  for (int32_t i = 0; i < pGrpCacheOperator->downstreamNum; ++i) {
99,297✔
149
    SGcDownstreamCtx* pCtx = &pGrpCacheOperator->pDownstreams[i];
66,198✔
150
    freeSGcDownstreamCtx(pCtx);
66,198✔
151
  }
152

153
  taosMemoryFree(pGrpCacheOperator->pDownstreams);
33,099!
154
}
155

156

157
void blockDataDeepCleanup(SSDataBlock* pDataBlock) {
144,404✔
158
  size_t numOfCols = taosArrayGetSize(pDataBlock->pDataBlock);
144,404✔
159
  for (int32_t i = 0; i < numOfCols; ++i) {
762,697✔
160
    SColumnInfoData* p = taosArrayGet(pDataBlock->pDataBlock, i);
618,293✔
161
    if (NULL == p) {
618,293!
162
      qError("fail to get %dth col in dataBlock, numOfCols:%d", i, (int32_t)numOfCols);
×
163
      continue;
×
164
    }
165
    taosMemoryFreeClear(p->pData);
618,293!
166
    if (IS_VAR_DATA_TYPE(p->info.type)) {
618,293!
167
      taosMemoryFreeClear(p->varmeta.offset);
93,922!
168
      p->varmeta.length = 0;
93,922✔
169
      p->varmeta.allocLen = 0;
93,922✔
170
    } else {
171
      taosMemoryFreeClear(p->nullbitmap);
524,371!
172
    }
173
  }
174
  pDataBlock->info.capacity = 0;
144,404✔
175
  pDataBlock->info.rows = 0;
144,404✔
176
}
144,404✔
177

178

179

180
static void destroySGcBlkCacheInfo(SGcBlkCacheInfo* pBlkCache) {
33,099✔
181
  taosHashCleanup(pBlkCache->pDirtyBlk);
33,099✔
182

183
  void* p = NULL;
33,099✔
184
  while (NULL != (p = taosHashIterate(pBlkCache->pReadBlk, p))) {
44,730✔
185
    blockDataDeepCleanup(*(SSDataBlock**)p);
11,631✔
186
    freeGcBlockInList(p);
11,631✔
187
  }
188

189
  taosHashCleanup(pBlkCache->pReadBlk);
33,099✔
190
}
33,099✔
191

192
static void destroyGroupCacheOperator(void* param) {
33,099✔
193
  SGroupCacheOperatorInfo* pGrpCacheOperator = (SGroupCacheOperatorInfo*)param;
33,099✔
194

195
  logGroupCacheExecInfo(pGrpCacheOperator);
33,099✔
196
  
197
  taosMemoryFree(pGrpCacheOperator->groupColsInfo.pColsInfo);
33,099!
198
  taosMemoryFree(pGrpCacheOperator->groupColsInfo.pBuf);
33,099!
199

200
  destroyGroupCacheDownstreamCtx(pGrpCacheOperator);
33,099✔
201
  destroySGcBlkCacheInfo(&pGrpCacheOperator->blkCache);
33,099✔
202
  taosHashCleanup(pGrpCacheOperator->pGrpHash);
33,099✔
203

204
  taosMemoryFree(pGrpCacheOperator->execInfo.pDownstreamBlkNum);
33,099!
205
  
206
  taosMemoryFreeClear(param);
33,099!
207
}
33,099✔
208

209
static FORCE_INLINE int32_t initOpenCacheFile(SGroupCacheFileFd* pFileFd, char* filename) {
210
  TdFilePtr newFd = taosOpenFile(filename, TD_FILE_CREATE|TD_FILE_READ|TD_FILE_WRITE|TD_FILE_AUTO_DEL);
61,745✔
211
  //TdFilePtr newFd = taosOpenFile(filename, TD_FILE_CREATE|TD_FILE_READ|TD_FILE_WRITE);
212
  if (NULL == newFd) {
61,745!
213
    QRY_ERR_RET(terrno);
×
214
  }
215
  pFileFd->fd = newFd;
61,745✔
216
  int32_t code = taosThreadMutexInit(&pFileFd->mutex, NULL);
61,745✔
217
  if (code) {
61,745!
218
    qError("taosThreadMutexInit failed, code:%x", code);
×
219
    QRY_ERR_RET(code);
×
220
  }
221

222
  qTrace("file path %s created", filename);
61,745!
223
  
224
  return TSDB_CODE_SUCCESS;
61,745✔
225
}
226

227
static int32_t acquireFdFromFileCtx(SGcFileCacheCtx* pFileCtx, int32_t fileId, SGroupCacheFileFd** ppFd, bool* pDeleted) {
318,722✔
228
  int32_t code = TSDB_CODE_SUCCESS;
318,722✔
229
  if (NULL == pFileCtx->pCacheFile) {
318,722✔
230
    pFileCtx->pCacheFile = taosHashInit(10, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_ENTRY_LOCK);
61,745✔
231
    if (NULL == pFileCtx->pCacheFile) {
61,745!
232
      return terrno;
×
233
    }
234
    taosHashSetFreeFp(pFileCtx->pCacheFile, freeSGroupCacheFileInfo);
61,745✔
235
  }
236
  
237
  SGroupCacheFileInfo* pTmp = taosHashGet(pFileCtx->pCacheFile, &fileId, sizeof(fileId));
318,722✔
238
  if (NULL == pTmp) {
318,722✔
239
    (void)snprintf(&pFileCtx->baseFilename[pFileCtx->baseNameLen], sizeof(pFileCtx->baseFilename) - pFileCtx->baseNameLen, "_%d", fileId);
61,745✔
240

241
    SGroupCacheFileInfo newFile = {0};
61,745✔
242
    if (taosHashPut(pFileCtx->pCacheFile, &fileId, sizeof(fileId), &newFile, sizeof(newFile))) {
61,745!
243
      QRY_ERR_RET(terrno);
×
244
    }
245
    pTmp = taosHashGet(pFileCtx->pCacheFile, &fileId, sizeof(fileId));
61,745✔
246
    if (NULL == pTmp) {
61,745!
247
      qError("fail to get file %d from pCacheFile", fileId);
×
248
      QRY_ERR_RET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
249
    }
250
  }
251

252
  if (pTmp->deleted) {
318,722!
253
    *pDeleted = true;
×
254
    return TSDB_CODE_SUCCESS;
×
255
  }
256

257
  if (NULL == pTmp->fd.fd) {
318,722✔
258
    code = initOpenCacheFile(&pTmp->fd, pFileCtx->baseFilename);
61,745✔
259
    if (code) {
61,745!
260
      return code;
×
261
    }
262
  }
263

264
  (void)taosThreadMutexLock(&pTmp->fd.mutex);
318,722✔
265
  *ppFd = &pTmp->fd;
318,722✔
266
  
267
  return TSDB_CODE_SUCCESS;
318,722✔
268
}
269

270
static FORCE_INLINE void releaseFdToFileCtx(SGroupCacheFileFd* pFd) {
271
  if (NULL == pFd) {
318,722✔
272
    return;
×
273
  }
274
  (void)taosThreadMutexUnlock(&pFd->mutex);
318,722✔
275
}
276

277
static int32_t saveBlocksToDisk(SGroupCacheOperatorInfo* pGCache, SGcDownstreamCtx* pCtx, SGcBlkBufInfo* pHead) {
174,318✔
278
  int32_t code = TSDB_CODE_SUCCESS;
174,318✔
279
  SGroupCacheFileFd *pFd = NULL;
174,318✔
280
  SGcFileCacheCtx* pFileCtx = NULL;
174,318✔
281
  SHashObj* pGrpHash = pGCache->globalGrp ? pGCache->pGrpHash : pCtx->pGrpHash;
174,318!
282
  int64_t lastGroupId = 0;
174,318✔
283
  SGroupCacheData* pGroup = NULL;
174,318✔
284
  
285
  while (NULL != pHead) {
348,636✔
286
    pFd = NULL;
174,318✔
287
    
288
    if (pGCache->batchFetch) {
174,318!
289
      pFileCtx = &pHead->pCtx->fileCtx;
174,318✔
290
    } else {
291
      if (pHead->groupId != lastGroupId) {
×
292
        if (NULL != pGroup) {
×
293
          taosHashRelease(pGrpHash, pGroup);
×
294
        }
295
        pGroup = taosHashAcquire(pGrpHash, &pHead->groupId, sizeof(pHead->groupId));      
×
296
        lastGroupId = pHead->groupId;
×
297
      }
298
    
299
      if (NULL == pGroup) {
×
300
        qTrace("group %" PRIu64 " in downstream %d may already be deleted, skip write", pHead->groupId, pHead->pCtx->id);
×
301

302
        int64_t blkId = pHead->basic.blkId;
×
303
        pHead = pHead->next;
×
304
        code = taosHashRemove(pGCache->blkCache.pDirtyBlk, &blkId, sizeof(blkId));
×
305
        if (code) {
×
306
          qError("taosHashRemove blk %" PRId64 " from diryBlk failed, error:%s", blkId, tstrerror(code));
×
307
          goto _return;
×
308
        }
309
        continue;
×
310
      }
311
      
312
      pFileCtx = &pGroup->pVgCtx->fileCtx;
×
313
    }
314

315
    bool deleted = false;
174,318✔
316
    code = acquireFdFromFileCtx(pFileCtx, pHead->basic.fileId, &pFd, &deleted);
174,318✔
317
    if (code) {
174,318!
318
      goto _return;
×
319
    }
320

321
    if (deleted) {
174,318!
322
      releaseFdToFileCtx(pFd);
×
323

324
      qTrace("FileId:%d-%d-%d already be deleted, skip write", 
×
325
          pCtx->id, pGroup ? pGroup->vgId : GROUP_CACHE_DEFAULT_VGID, pHead->basic.fileId);
326
      
327
      int64_t blkId = pHead->basic.blkId;
×
328
      pHead = pHead->next;
×
329
      
330
      code = taosHashRemove(pGCache->blkCache.pDirtyBlk, &blkId, sizeof(blkId));
×
331
      if (code) {
×
332
        qError("taosHashRemove blk %" PRId64 " from diryBlk failed, error:%s", blkId, tstrerror(code));
×
333
        goto _return;
×
334
      }
335
      continue;
×
336
    }
337
    
338
    int64_t ret = taosLSeekFile(pFd->fd, pHead->basic.offset, SEEK_SET);
174,318✔
339
    if (ret < 0) {
174,318!
340
      releaseFdToFileCtx(pFd);
×
341
      code = terrno;
×
342
      goto _return;
×
343
    }
344
    
345
    ret = taosWriteFile(pFd->fd, pHead->pBuf, pHead->basic.bufSize);
174,318✔
346
    if (ret != pHead->basic.bufSize) {
174,318!
347
      releaseFdToFileCtx(pFd);
×
348
      code = terrno;
×
349
      goto _return;
×
350
    }
351
    
352
    releaseFdToFileCtx(pFd);
174,318!
353

354
    qTrace("FileId:%d-%d-%d blk %" PRIu64 " in group %" PRIu64 " size %" PRIu64 " written to offset %" PRIu64, 
174,318!
355
        pCtx->id, pGroup ? pGroup->vgId : GROUP_CACHE_DEFAULT_VGID, pHead->basic.fileId, pHead->basic.blkId, pHead->groupId, pHead->basic.bufSize, pHead->basic.offset);
356
    
357
    int64_t blkId = pHead->basic.blkId;
174,318✔
358
    pHead = pHead->next;
174,318✔
359

360
    code = taosHashRemove(pGCache->blkCache.pDirtyBlk, &blkId, sizeof(blkId));
174,318✔
361
    if (code) {
174,318!
362
      qError("taosHashRemove blk %" PRId64 " from diryBlk failed, error:%s", blkId, tstrerror(code));
×
363
      goto _return;
×
364
    }
365
  }
366

367
_return:
174,318✔
368

369
  if (NULL != pGroup) {
174,318!
370
    taosHashRelease(pGrpHash, pGroup);
×
371
  }
372

373
  (void)atomic_val_compare_exchange_32(&pGCache->blkCache.writeDownstreamId, pCtx->id, -1);
174,318✔
374

375
  return code;
174,318✔
376
}
377

378

379
void freeGcBlkBufInfo(void* ptr) {
174,318✔
380
  SGcBlkBufInfo* pBlk = (SGcBlkBufInfo*)ptr;
174,318✔
381
  taosMemoryFreeClear(pBlk->pBuf);
174,318!
382
}
174,318✔
383

384

385
static int32_t addBlkToDirtyBufList(SGroupCacheOperatorInfo* pGCache, SGcDownstreamCtx* pCtx, SGcBlkCacheInfo* pCache, SGcBlkBufInfo* pBufInfo) {
174,318✔
386
  if (0 != taosHashPut(pCache->pDirtyBlk, &pBufInfo->basic.blkId, sizeof(pBufInfo->basic.blkId), pBufInfo, sizeof(*pBufInfo))) {
174,318!
387
    freeGcBlkBufInfo(pBufInfo);
×
388
    return terrno;
×
389
  }
390
  pBufInfo = taosHashGet(pCache->pDirtyBlk, &pBufInfo->basic.blkId, sizeof(pBufInfo->basic.blkId));
174,318✔
391
  if (NULL == pBufInfo) {
174,318!
392
    qError("fail to get blk %" PRId64 " from pCache->pDirtyBlk", pBufInfo->basic.blkId);
×
393
    QRY_ERR_RET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
394
  }
395
  int32_t code = TSDB_CODE_SUCCESS;
174,318✔
396
  SGcBlkBufInfo* pWriteHead = NULL;
174,318✔
397
  
398
  taosWLockLatch(&pCache->dirtyLock);
174,318✔
399
  pCache->blkCacheSize += pBufInfo->basic.bufSize;
174,318✔
400
  qDebug("group cache total dirty block num:%d size:%" PRId64 , taosHashGetSize(pCache->pDirtyBlk), pCache->blkCacheSize);
174,318!
401

402
  if (NULL == pCache->pDirtyHead) {
174,318!
403
    pCache->pDirtyHead = pBufInfo;
174,318✔
404
  } else {
405
    pCache->pDirtyTail->next = pBufInfo;
×
406
  }
407
  pCache->pDirtyTail = pBufInfo;
174,318✔
408
    
409
  if (pGCache->maxCacheSize >= 0 && pCache->blkCacheSize > pGCache->maxCacheSize) {
174,318!
410
    if (-1 == atomic_val_compare_exchange_32(&pCache->writeDownstreamId, -1, pCtx->id)) {
174,318!
411
      pWriteHead = pCache->pDirtyHead;
174,318✔
412
      SGcBlkBufInfo* pTmp = pCache->pDirtyHead;
174,318✔
413
      while (NULL != pTmp) {
174,318!
414
        pCache->blkCacheSize -= pTmp->basic.bufSize;
174,318✔
415
        if (pCache->blkCacheSize <= pGCache->maxCacheSize) {
174,318!
416
          pCache->pDirtyHead = pTmp->next;
174,318✔
417
          pTmp->next = NULL;
174,318✔
418
          break;
174,318✔
419
        }
420
        pTmp = pTmp->next;
×
421
      }
422
    }
423
  }
424
  taosWUnLockLatch(&pCache->dirtyLock);
174,318✔
425

426
  if (NULL != pWriteHead) {
174,318!
427
    code = saveBlocksToDisk(pGCache, pCtx, pWriteHead);
174,318✔
428
  }
429

430
  return code;
174,318✔
431
}
432

433
static FORCE_INLINE void chkRemoveVgroupCurrFile(SGcFileCacheCtx* pFileCtx, int32_t downstreamIdx, int32_t vgId) {
434
  SGroupCacheFileInfo* pFileInfo = taosHashGet(pFileCtx->pCacheFile, &pFileCtx->fileId, sizeof(pFileCtx->fileId));
×
435
  if (NULL == pFileInfo) {
×
436
    return;
×
437
  }
438
  
439
  if (0 == pFileInfo->groupNum) {
×
440
    removeGroupCacheFile(pFileInfo);
×
441

442
#if 0  
443
    /* debug only */
444
    snprintf(&pFileCtx->baseFilename[pFileCtx->baseNameLen], sizeof(pFileCtx->baseFilename) - pFileCtx->baseNameLen, "_%d", pFileCtx->fileId);
445
    taosRemoveFile(pFileCtx->baseFilename);
446
    /* debug only */
447
#endif
448

449
    qTrace("FileId:%d-%d-%d removed", downstreamIdx, vgId, pFileCtx->fileId);
×
450
    //taosHashRemove(pFileCtx->pCacheFile, &pGroup->fileId, sizeof(pGroup->fileId));
451
  }
452
}
453

454
static FORCE_INLINE void groupCacheSwitchNewFile(SGcFileCacheCtx* pFileCtx, int32_t downstreamIdx, int32_t vgId, bool removeCheck) {
455
  if (pFileCtx->fileSize < GROUP_CACHE_DEFAULT_MAX_FILE_SIZE) {
174,318!
456
    return;
174,318✔
457
  }
458

459
  if (removeCheck) {
×
460
    chkRemoveVgroupCurrFile(pFileCtx, downstreamIdx, vgId);
461
  }
462
      
463
  pFileCtx->fileId++;
×
464
  pFileCtx->fileSize = 0;
×
465
}
466

467

468
static int32_t addBlkToBufCache(struct SOperatorInfo* pOperator, SSDataBlock* pBlock, SGcDownstreamCtx* pCtx, SGroupCacheData* pGroup, SGcBlkBufInfo* pBufInfo) {
174,318✔
469
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
174,318✔
470
  int64_t bufSize = blockDataGetSize(pBlock) + sizeof(int32_t) + taosArrayGetSize(pBlock->pDataBlock) * sizeof(int32_t);
174,318✔
471
  pBufInfo->pBuf = taosMemoryMalloc(bufSize);
174,318!
472
  if (NULL == pBufInfo->pBuf) {
174,318!
473
    qError("group cache add block to cache failed, size:%" PRId64, bufSize);
×
474
    return terrno;
×
475
  }
476
  QRY_ERR_RET(blockDataToBuf(pBufInfo->pBuf, pBlock));
174,318!
477

478
  SGcFileCacheCtx* pFileCtx = pGCache->batchFetch ? &pCtx->fileCtx : &pGroup->pVgCtx->fileCtx;
174,318!
479

480
  pBufInfo->next = NULL;
174,318✔
481
  pBufInfo->basic.blkId = atomic_add_fetch_64(&pGCache->currentBlkId, 1);
174,318✔
482
  pBufInfo->basic.fileId = pGCache->batchFetch ? pFileCtx->fileId : pGroup->fileId;
174,318!
483
  pBufInfo->basic.bufSize = bufSize;
174,318✔
484
  pBufInfo->basic.offset = atomic_fetch_add_64(&pFileCtx->fileSize, bufSize);
174,318✔
485
  pBufInfo->pCtx = pCtx;
174,318✔
486
  pBufInfo->groupId = pBlock->info.id.groupId;
174,318✔
487

488
  if (pGCache->batchFetch) {    
174,318!
489
    groupCacheSwitchNewFile(pFileCtx, pCtx->id, pGroup->vgId, false);
174,318!
490
  }
491

492
  int32_t code = addBlkToDirtyBufList(pGCache, pCtx, &pGCache->blkCache, pBufInfo);
174,318✔
493

494
  return code;
174,318✔
495
}
496

497
void blockDataDeepClear(SSDataBlock* pDataBlock) {
67,217✔
498
  size_t numOfCols = taosArrayGetSize(pDataBlock->pDataBlock);
67,217✔
499
  for (int32_t i = 0; i < numOfCols; ++i) {
315,075✔
500
    SColumnInfoData* p = taosArrayGet(pDataBlock->pDataBlock, i);
247,858✔
501
    if (NULL == p) {
247,858!
502
      qError("fail to get %d col from pDataBlock, numOfCols:%d", i, (int32_t)numOfCols);
×
503
      continue;
×
504
    }
505
    p->pData = NULL;
247,858✔
506
    if (IS_VAR_DATA_TYPE(p->info.type)) {
247,858!
507
      p->varmeta.offset = NULL;
18,367✔
508
      p->varmeta.length = 0;
18,367✔
509
      p->varmeta.allocLen = 0;
18,367✔
510
    } else {
511
      p->nullbitmap = NULL;
229,491✔
512
    }
513
  }
514
  pDataBlock->info.capacity = 0;
67,217✔
515
  pDataBlock->info.rows = 0;
67,217✔
516
}
67,217✔
517

518
static int32_t buildGroupCacheBaseBlock(SSDataBlock** ppDst, SSDataBlock* pSrc) {
67,217✔
519
  *ppDst = taosMemoryMalloc(sizeof(*pSrc));
67,217!
520
  if (NULL == *ppDst) {
67,217!
521
    return terrno;
×
522
  }
523
  (*ppDst)->pBlockAgg = NULL;
67,217✔
524
  (*ppDst)->pDataBlock = taosArrayDup(pSrc->pDataBlock, NULL);
67,217✔
525
  if (NULL == (*ppDst)->pDataBlock) {
67,217!
526
    taosMemoryFree(*ppDst);
×
527
    return terrno;
×
528
  }
529
  TAOS_MEMCPY(&(*ppDst)->info, &pSrc->info, sizeof(pSrc->info));
67,217!
530
  blockDataDeepClear(*ppDst);
67,217✔
531
  
532
  return TSDB_CODE_SUCCESS;
67,217✔
533
}
534

535
static int32_t acquireBaseBlockFromList(SGcDownstreamCtx* pCtx, SSDataBlock** ppRes) {
144,404✔
536
  taosWLockLatch(&pCtx->blkLock);
144,404✔
537
  if (taosArrayGetSize(pCtx->pFreeBlock) <= 0) {
144,404✔
538
    taosWUnLockLatch(&pCtx->blkLock);
5,472✔
539
    return buildGroupCacheBaseBlock(ppRes, pCtx->pBaseBlock);
5,472✔
540
  }
541
  *ppRes = *(SSDataBlock**)taosArrayPop(pCtx->pFreeBlock);
138,932✔
542
  taosWUnLockLatch(&pCtx->blkLock);
138,932✔
543

544
  return TSDB_CODE_SUCCESS;  
138,932✔
545
}
546

547
static int32_t releaseBaseBlockToList(SGcDownstreamCtx* pCtx, SSDataBlock* pBlock) {
132,773✔
548
  int32_t code = TSDB_CODE_SUCCESS;
132,773✔
549
  
550
  blockDataDeepCleanup(pBlock);
132,773✔
551
  taosWLockLatch(&pCtx->blkLock);
132,773✔
552
  if (NULL == taosArrayPush(pCtx->pFreeBlock, &pBlock)) {
265,546!
553
    code = terrno;
×
554
  }
555
  taosWUnLockLatch(&pCtx->blkLock);
132,773✔
556

557
  return code;
132,773✔
558
}
559

560

561
static int32_t buildGroupCacheResultBlock(SGroupCacheOperatorInfo* pGCache, int32_t downstreamIdx, void* pBuf, SSDataBlock** ppRes) {
144,404✔
562
  int32_t code = acquireBaseBlockFromList(&pGCache->pDownstreams[downstreamIdx], ppRes);
144,404✔
563
  if (code) {
144,404!
564
    return code;
×
565
  }
566
  //TODO OPTIMIZE PERF
567
  return blockDataFromBuf(*ppRes, pBuf);
144,404✔
568
}
569

570
static int32_t readBlockFromDisk(SGroupCacheOperatorInfo* pGCache, SGroupCacheData* pGrp, SGcBlkBufBasic* pBasic, void** ppBuf) {
144,404✔
571
  SGroupCacheFileFd *pFileFd = NULL;
144,404✔
572
  SGcFileCacheCtx* pFileCtx = pGCache->batchFetch ? &pGCache->pDownstreams[pGrp->downstreamIdx].fileCtx : &pGrp->pVgCtx->fileCtx;
144,404!
573
  bool deleted = false;
144,404✔
574
  int32_t code = acquireFdFromFileCtx(pFileCtx, pBasic->fileId, &pFileFd, &deleted);
144,404✔
575
  if (code) {
144,404!
576
    return code;
×
577
  }
578
  if (deleted) {
144,404!
579
    qError("FileId:%d-%d-%d already be deleted, skip read", pGrp->downstreamIdx, pGrp->vgId, pBasic->fileId);
×
580
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
581
  }
582
  
583
  int64_t ret = taosLSeekFile(pFileFd->fd, pBasic->offset, SEEK_SET);
144,404✔
584
  if (ret < 0) {
144,404!
585
    code = terrno;
×
586
    goto _return;
×
587
  }
588

589
  *ppBuf = taosMemoryMalloc(pBasic->bufSize);
144,404!
590
  if (NULL == *ppBuf) {
144,404!
591
    code = terrno;
×
592
    goto _return;
×
593
  }
594
  
595
  ret = taosReadFile(pFileFd->fd, *ppBuf, pBasic->bufSize);
144,404✔
596
  if (ret != pBasic->bufSize) {
144,404!
597
    taosMemoryFreeClear(*ppBuf);
×
598
    code = terrno;
×
599
    goto _return;
×
600
  }
601

602
  qTrace("FileId:%d-%d-%d blk %" PRIu64 " size %" PRIu64 " read from offset %" PRIu64, 
144,404!
603
      pGrp->downstreamIdx, pGrp->vgId, pBasic->fileId, pBasic->blkId, pBasic->bufSize, pBasic->offset);
604

605
_return:
144,404✔
606

607
  releaseFdToFileCtx(pFileFd);
144,404!
608
  return code;
144,404✔
609
}
610

611
static int32_t retrieveBlkFromBufCache(SGroupCacheOperatorInfo* pGCache, SGroupCacheData* pGrp, int64_t sessionId, SGcBlkBufBasic* pBasic, SSDataBlock** ppRes) {
144,404✔
612
  int32_t code = TSDB_CODE_SUCCESS;
144,404✔
613
  SGcBlkCacheInfo* pCache = &pGCache->blkCache;
144,404✔
614
  void* pBuf = NULL;
144,404✔
615

616
  SGcBlkBufInfo* pBufInfo = taosHashAcquire(pCache->pDirtyBlk, &pBasic->blkId, sizeof(pBasic->blkId));
144,404✔
617
  if (NULL == pBufInfo) {
144,404!
618
    code = readBlockFromDisk(pGCache, pGrp, pBasic, &pBuf);
144,404✔
619
    if (code) {
144,404!
620
      return code;
×
621
    }
622
  } else {
623
    pBuf = pBufInfo->pBuf;
×
624
  }
625
  
626
  code = buildGroupCacheResultBlock(pGCache, pGrp->downstreamIdx, pBuf, ppRes);
144,404✔
627
  taosHashRelease(pCache->pDirtyBlk, pBufInfo);
144,404✔
628
  if (NULL == pBufInfo) {
144,404!
629
    taosMemoryFree(pBuf);
144,404!
630
  }
631
  
632
  if (code) {
144,404!
633
    return code;
×
634
  }
635

636
  QRY_ERR_RET(taosHashPut(pCache->pReadBlk, &sessionId, sizeof(sessionId), ppRes, POINTER_BYTES));
144,404!
637
  
638
  return TSDB_CODE_SUCCESS;
144,404✔
639
}
640

641
static FORCE_INLINE void initGcVgroupCtx(SOperatorInfo* pOperator, SGcVgroupCtx* pVgCtx, int32_t downstreamId, int32_t vgId, SArray* pTbList) {
642
  pVgCtx->pTbList = pTbList;
66,198✔
643
  pVgCtx->id = vgId;
66,198✔
644
  (void)snprintf(pVgCtx->fileCtx.baseFilename, sizeof(pVgCtx->fileCtx.baseFilename) - 1, "%s/gc_%d_%" PRIx64 "_%" PRIu64 "_%p_%d_%d", 
132,396✔
645
     tsTempDir, taosGetPId(), pOperator->pTaskInfo->id.queryId, pOperator->pTaskInfo->id.taskId, pOperator, downstreamId, vgId);
132,396✔
646
  pVgCtx->fileCtx.baseFilename[sizeof(pVgCtx->fileCtx.baseFilename) - 1] = 0;
66,198✔
647

648
  pVgCtx->fileCtx.baseNameLen = strlen(pVgCtx->fileCtx.baseFilename);
66,198!
649
}
66,198✔
650

651
static int32_t addNewGroupToVgHash(SOperatorInfo* pOperator, SSHashObj* pHash, SGcNewGroupInfo* pNew) {
×
652
  SGcVgroupCtx* pVgCtx = pNew->pGroup->pVgCtx;
×
653
  if (NULL == pVgCtx) {
×
654
    SArray* pList = taosArrayInit(10, sizeof(*pNew));
×
655
    if (NULL == pList) {
×
656
      return terrno;
×
657
    }
658
    if (NULL == taosArrayPush(pList, pNew)) {
×
659
      QRY_ERR_RET(terrno);
×
660
    }
661
    
662
    SGcVgroupCtx vgCtx = {0};
×
663
    initGcVgroupCtx(pOperator, &vgCtx, pNew->pGroup->downstreamIdx, pNew->vgId, pList);
×
664
    QRY_ERR_RET(tSimpleHashPut(pHash, &pNew->vgId, sizeof(pNew->vgId), &vgCtx, sizeof(vgCtx)));
×
665
    
666
    pNew->pGroup->pVgCtx = tSimpleHashGet(pHash, &pNew->vgId, sizeof(pNew->vgId));
×
667
    if (NULL == pNew->pGroup->pVgCtx) {
×
668
      qError("fail to get vg %d ctx from vgHash", pNew->vgId);
×
669
      QRY_ERR_RET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
670
    }
671
    
672
    return TSDB_CODE_SUCCESS;
×
673
  }
674

675
  if (NULL == taosArrayPush(pVgCtx->pTbList, pNew)) {
×
676
    QRY_ERR_RET(terrno);
×
677
  }
678
  
679
  return TSDB_CODE_SUCCESS;
×
680
}
681

682
static FORCE_INLINE int32_t appendNewGroupToDownstream(struct SOperatorInfo* pOperator, int32_t downstreamIdx, SOperatorParam** ppParam) {
683
  int32_t code = TSDB_CODE_SUCCESS;
227,152✔
684
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
227,152✔
685
  SGcDownstreamCtx* pCtx = &pGCache->pDownstreams[downstreamIdx];  
227,152✔
686
  SOperatorParam* pDst = NULL;
227,152✔
687
  
688
  taosWLockLatch(&pCtx->grpLock);
227,152✔
689
  int32_t num = taosArrayGetSize(pCtx->pNewGrpList);
227,152✔
690
  if (num <= 0) {
227,152✔
691
    goto _return;
162,602✔
692
  }
693

694
  for (int32_t i = 0; i < num; ++i) {
129,100✔
695
    SGcNewGroupInfo* pNew = taosArrayGet(pCtx->pNewGrpList, i);
64,550✔
696
    if (NULL == pNew) {
64,550!
697
      qError("fail to get vg %d SGcNewGroupInfo from pNewGrpList", i);
×
698
      QRY_ERR_RET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
699
    }
700
    
701
    if (!pGCache->batchFetch) {
64,550!
702
      code = addNewGroupToVgHash(pOperator, pCtx->pVgTbHash, pNew);
×
703
      if (code) {
×
704
        goto _return;
×
705
      }
706
    }
707

708
    if (NULL == pDst) {
64,550!
709
      pDst = pNew->pParam;
64,550✔
710
    } else if (pNew->pParam) {
×
711
      code = mergeOperatorParams(pDst, pNew->pParam);
×
712
      if (code) {
×
713
        goto _return;
×
714
      }
715
    }
716
  }
717

718
  taosArrayClear(pCtx->pNewGrpList);
64,550✔
719
  
720
_return:
227,152✔
721

722
  taosWUnLockLatch(&pCtx->grpLock);
227,152✔
723
  *ppParam = pDst;
227,152✔
724
  
725
  return code;
227,152✔
726
}
727

728
static FORCE_INLINE int32_t getBlkFromDownstreamOperator(struct SOperatorInfo* pOperator, int32_t downstreamIdx,
729
                                                         SSDataBlock** ppRes) {
730
  int32_t                  code = TSDB_CODE_SUCCESS;
227,152✔
731
  SOperatorParam*          pDownstreamParam = NULL;
454,304✔
732
  SSDataBlock*             pBlock = NULL;
227,152✔
733
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
227,152✔
734

735
  code = appendNewGroupToDownstream(pOperator, downstreamIdx, &pDownstreamParam);
227,152✔
736
  if (code) {
227,152!
737
    return code;
×
738
  }
739

740
  SOperatorInfo* pDownstream = pOperator->pDownstream[downstreamIdx];
227,152✔
741
  if (pDownstreamParam) {
227,152✔
742
    code = pDownstream->fpSet.getNextExtFn(pDownstream, pDownstreamParam, &pBlock);
64,550✔
743
  } else {
744
    code = pDownstream->fpSet.getNextFn(pDownstream, &pBlock);
162,602✔
745
  }
746

747
  if (code) {
227,152!
748
    qError("failed to get block from downstream, code:%s %s", tstrerror(code), GET_TASKID(pOperator->pTaskInfo));
×
749
    return code;
×
750
  }
751

752
  if (pBlock) {
227,152✔
753
    qDebug("%s res block retrieved from group %" PRIu64, GET_TASKID(pOperator->pTaskInfo), pBlock->info.id.groupId);
174,318!
754

755
    pGCache->execInfo.pDownstreamBlkNum[downstreamIdx]++;
174,318✔
756
    if (NULL == pGCache->pDownstreams[downstreamIdx].pBaseBlock) {
174,318✔
757
      code = buildGroupCacheBaseBlock(&pGCache->pDownstreams[downstreamIdx].pBaseBlock, pBlock);
61,745✔
758
      if (code) {
61,745!
759
        return code;
×
760
      }
761

762
      if (NULL == taosArrayPush(pGCache->pDownstreams[downstreamIdx].pFreeBlock,
123,490!
763
                                &pGCache->pDownstreams[downstreamIdx].pBaseBlock)) {
61,745✔
764
        QRY_ERR_RET(terrno);
×
765
      }
766
    }
767
  }
768

769
  code = blockDataCheck(pBlock);
227,152✔
770

771
  *ppRes = pBlock;
227,152✔
772
  return code;
227,152✔
773
}
774

775
static int32_t notifyWaitingSessions(SArray* pWaitQueue) {
322,633✔
776
  if (NULL == pWaitQueue || taosArrayGetSize(pWaitQueue) <= 0) {
322,633!
777
    return TSDB_CODE_SUCCESS;
322,633✔
778
  }
779
  
780
  int32_t n = taosArrayGetSize(pWaitQueue);
×
781
  for (int32_t i = 0; i < n; ++i) {
×
782
    SGcSessionCtx* pSession = taosArrayGetP(pWaitQueue, i);
×
783
    if (NULL == pSession) {
×
784
      qError("fail to get %d SGcSessionCtx in pWaitQueue, total:%d", i, n);
×
785
      QRY_ERR_RET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
786
    }
787
    
788
    QRY_ERR_RET(tsem_post(&pSession->waitSem));
×
789
  }
790

791
  return TSDB_CODE_SUCCESS;
×
792
}
793

794
static FORCE_INLINE int32_t handleGroupFetchDone(SGroupCacheData* pGroup) {
795
  int32_t code = TSDB_CODE_SUCCESS;
148,315✔
796
  pGroup->pBlock = NULL;
148,315✔
797
  atomic_store_8((int8_t*)&pGroup->fetchDone, true);
148,315✔
798
  
799
  (void)taosThreadMutexLock(&pGroup->mutex);
148,315✔
800
  code = notifyWaitingSessions(pGroup->waitQueue);
148,315✔
801
  taosArrayClear(pGroup->waitQueue);
148,315✔
802
  (void)taosThreadMutexUnlock(&pGroup->mutex);
148,315✔
803

804
  return code;
148,315✔
805
}
806

807
static int32_t addFileRefTableNum(SGcFileCacheCtx* pFileCtx, int32_t fileId, int32_t downstreamId, int32_t vgId) {
×
808
  if (NULL == pFileCtx->pCacheFile) {
×
809
    pFileCtx->pCacheFile = taosHashInit(10, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_ENTRY_LOCK);
×
810
    if (NULL == pFileCtx->pCacheFile) {
×
811
      return terrno;
×
812
    }
813
    taosHashSetFreeFp(pFileCtx->pCacheFile, freeSGroupCacheFileInfo);
×
814
  }
815
  
816
  SGroupCacheFileInfo* pTmp = taosHashGet(pFileCtx->pCacheFile, &fileId, sizeof(fileId));
×
817
  if (NULL == pTmp) {
×
818
    (void)snprintf(&pFileCtx->baseFilename[pFileCtx->baseNameLen], sizeof(pFileCtx->baseFilename) - pFileCtx->baseNameLen, "_%u", fileId);
×
819

820
    SGroupCacheFileInfo newFile = {0};
×
821
    newFile.groupNum = 1;
×
822
    QRY_ERR_RET(taosHashPut(pFileCtx->pCacheFile, &fileId, sizeof(fileId), &newFile, sizeof(newFile)));
×
823
    pTmp = taosHashGet(pFileCtx->pCacheFile, &fileId, sizeof(fileId));
×
824
    if (NULL == pTmp) {
×
825
      qError("fail to get file %d in pCacheFile", fileId);
×
826
      QRY_ERR_RET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
827
    }
828
  } else {
829
    pTmp->groupNum++;
×
830
  }
831

832
  qTrace("FileId:%d-%d-%d add groupNum to %u", downstreamId, vgId, fileId, pTmp->groupNum);
×
833

834
  return TSDB_CODE_SUCCESS;
×
835
}
836

837
static int32_t handleVgroupTableFetchDone(SGcDownstreamCtx* pCtx, SGroupCacheData* pGroup, uint64_t uid) {
×
838
  if (pCtx->lastBlkUid == uid || pGroup->pVgCtx->lastBlkUid == uid) {
×
839
    return TSDB_CODE_SUCCESS;
×
840
  }
841
  
842
  pCtx->lastBlkUid = uid;
×
843
  pGroup->pVgCtx->lastBlkUid = uid;
×
844
  
845
  int32_t i = 0;
×
846
  while (true) {
×
847
    SGcNewGroupInfo* pNew = taosArrayGet(pGroup->pVgCtx->pTbList, i++);
×
848
    if (NULL == pNew || pNew->uid == uid) {
×
849
      break;
850
    }
851
    QRY_ERR_RET(handleGroupFetchDone(pNew->pGroup));
×
852
  }
853

854
  groupCacheSwitchNewFile(&pGroup->pVgCtx->fileCtx, pGroup->downstreamIdx, pGroup->vgId, true);
×
855

856
  pGroup->fileId = pGroup->pVgCtx->fileCtx.fileId;
×
857
  pGroup->startOffset = pGroup->pVgCtx->fileCtx.fileSize;
×
858

859
  qTrace("FileId:%d-%d-%d add groupNum for group %" PRIu64, pGroup->downstreamIdx, pGroup->vgId, pGroup->pVgCtx->fileCtx.fileId, uid);
×
860

861
  if (pGroup->needCache) {
×
862
    return addFileRefTableNum(&pGroup->pVgCtx->fileCtx, pGroup->pVgCtx->fileCtx.fileId, pGroup->downstreamIdx, pGroup->vgId);
×
863
  }
864

865
  return TSDB_CODE_SUCCESS;
×
866
}
867

868

869
static FORCE_INLINE int32_t initNewGroupData(SGcDownstreamCtx* pCtx, SGroupCacheData* pGroup, int32_t downstreamIdx, int32_t vgId, bool batchFetch, bool needCache) {
870
  QRY_ERR_RET(taosThreadMutexInit(&pGroup->mutex, NULL));
177,921!
871
  
872
  pGroup->downstreamIdx = downstreamIdx;
177,921✔
873
  pGroup->vgId = vgId;
177,921✔
874
  pGroup->fileId = -1;
177,921✔
875
  pGroup->blkList.pList = taosArrayInit(10, sizeof(SGcBlkBufBasic));
177,921✔
876
  if (NULL == pGroup->blkList.pList) {
177,921!
877
    QRY_ERR_RET(terrno);
×
878
  }
879
  pGroup->startOffset = -1;
177,921✔
880
  pGroup->needCache = needCache;
177,921✔
881
  pGroup->pVgCtx = tSimpleHashGet(pCtx->pVgTbHash, &pGroup->vgId, sizeof(pGroup->vgId));
177,921✔
882

883
  return TSDB_CODE_SUCCESS;
177,921✔
884
}
885

886
static int32_t addNewGroupData(struct SOperatorInfo* pOperator, SOperatorParam* pParam, SGroupCacheData** ppGrp, int32_t vgId, int64_t uid) {
177,921✔
887
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
177,921✔
888
  SGcDownstreamCtx* pCtx = &pGCache->pDownstreams[pParam->downstreamIdx];
177,921✔
889
  SGcOperatorParam* pGcParam = pParam->value;  
177,921✔
890
  SHashObj* pGrpHash = pGCache->globalGrp ? pGCache->pGrpHash : pCtx->pGrpHash;
177,921!
891
  SGroupCacheData grpData = {0};
177,921✔
892
  
893
  while (true) {
894
    if (0 != taosHashPut(pGrpHash, &uid, sizeof(uid), &grpData, sizeof(grpData))) {
177,921!
895
      if (terrno == TSDB_CODE_DUP_KEY) {
×
896
        *ppGrp = taosHashGet(pGrpHash, &uid, sizeof(uid));
×
897
        if (*ppGrp) {
×
898
          break;
×
899
        }
900
      } else {
901
        return terrno;
×
902
      }
903
    }
904

905
    break;
177,921✔
906
  }
907

908
  *ppGrp = taosHashGet(pGrpHash, &uid, sizeof(uid));
177,921✔
909
  if (NULL == *ppGrp) {
177,921!
910
    return terrno;
×
911
  }
912
  QRY_ERR_RET(initNewGroupData(pCtx, *ppGrp, pParam->downstreamIdx, vgId, pGCache->batchFetch, pGcParam->needCache));
355,842!
913

914
  qDebug("new group %" PRIu64 " initialized, downstreamIdx:%d, vgId:%d, needCache:%d", uid, pParam->downstreamIdx, vgId, pGcParam->needCache);
177,921!
915

916
  if (pParam->pChildren) {
177,921✔
917
    SGcNewGroupInfo newGroup;
64,550✔
918
    newGroup.pGroup = *ppGrp;
64,550✔
919
    newGroup.vgId = vgId;
64,550✔
920
    newGroup.uid = uid;
64,550✔
921
    newGroup.pParam = taosArrayGetP(pParam->pChildren, 0);
64,550✔
922
    
923
    taosWLockLatch(&pCtx->grpLock);
64,550✔
924
    if (NULL == taosArrayPush(pCtx->pNewGrpList, &newGroup)) {
129,100!
925
      taosWUnLockLatch(&pCtx->grpLock);
×
926
      return terrno;
×
927
    }
928
    taosWUnLockLatch(&pCtx->grpLock);
64,550✔
929
    
930
    taosArrayDestroy(pParam->pChildren);
64,550✔
931
    pParam->pChildren = NULL;
64,550✔
932
    pCtx->fetchDone = false;
64,550✔
933
  }
934

935
  return TSDB_CODE_SUCCESS;
177,921✔
936
}
937

938
static int32_t addBlkToGroupCache(bool batchFetch, SGroupCacheData* pGroup, SGcBlkBufInfo* pNewBlk, int64_t* pIdx) {
174,318✔
939
  taosWLockLatch(&pGroup->blkList.lock);
174,318✔
940
  if (NULL == taosArrayPush(pGroup->blkList.pList, &pNewBlk->basic)) {
348,636!
941
    QRY_ERR_RET(terrno);
×
942
  }
943
  *pIdx = taosArrayGetSize(pGroup->blkList.pList) - 1;
174,318✔
944
  taosWUnLockLatch(&pGroup->blkList.lock);
174,318✔
945

946
  qDebug("block added to group cache, total block num:%" PRId64, *pIdx + 1);
174,318!
947
  
948
  return TSDB_CODE_SUCCESS;
174,318✔
949
}
950

951
static int32_t handleGroupCacheRetrievedBlk(struct SOperatorInfo* pOperator, SSDataBlock* pBlock, SGcSessionCtx* pSession, bool* continueFetch) {
174,318✔
952
  int32_t code = TSDB_CODE_SUCCESS;
174,318✔
953
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
174,318✔
954
  SGcDownstreamCtx* pCtx = &pGCache->pDownstreams[pSession->downstreamIdx];
174,318✔
955
  SHashObj* pGrpHash = pGCache->globalGrp ? pGCache->pGrpHash : pCtx->pGrpHash;
174,318!
956
  int64_t newBlkIdx = 0;
174,318✔
957

958
  SGroupCacheData* pGroup = taosHashGet(pGrpHash, &pBlock->info.id.groupId, sizeof(pBlock->info.id.groupId));
174,318✔
959
  if (NULL == pGroup) {
174,318✔
960
    if (pGCache->batchFetch) {
105,250!
961
      SGcOperatorParam fakeGcParam = {0};
105,250✔
962
      SOperatorParam fakeParam = {0};
105,250✔
963
      fakeGcParam.needCache = true;
105,250✔
964
      fakeParam.downstreamIdx = pSession->downstreamIdx;
105,250✔
965
      fakeParam.value = &fakeGcParam;
105,250✔
966
      fakeParam.reUse = false;
105,250✔
967
      code = addNewGroupData(pOperator, &fakeParam, &pGroup, GROUP_CACHE_DEFAULT_VGID, pBlock->info.id.groupId);
105,250✔
968
      if (TSDB_CODE_SUCCESS != code) {
105,250!
969
        return code;
×
970
      }
971
    } else {
972
      qError("group %" PRIu64 " not found in group hash", pBlock->info.id.groupId);
×
973
      return TSDB_CODE_INVALID_PARA;
×
974
    }
975
  }
976

977
  if (!pGCache->batchFetch) {
174,318!
978
    code = handleVgroupTableFetchDone(pCtx, pGroup, pBlock->info.id.groupId);
×
979
    if (TSDB_CODE_SUCCESS != code) {
×
980
      return code;
×
981
    }
982
  }
983

984
  if (pGroup->needCache) {
174,318!
985
    qDebug("add block to group cache");
174,318!
986
    
987
    SGcBlkBufInfo newBlkBuf;    
174,318✔
988
    code = addBlkToBufCache(pOperator, pBlock, pCtx, pGroup, &newBlkBuf);
174,318✔
989
    if (code) {
174,318!
990
      return code;
×
991
    }
992

993
    code = addBlkToGroupCache(pGCache->batchFetch, pGroup, &newBlkBuf, &newBlkIdx);
174,318!
994
    if (code) {
174,318!
995
      return code;
×
996
    }
997
  } else {
998
    qDebug("no need to add block to group cache");
×
999
    
1000
    pGroup->pBlock = pBlock;
×
1001
  }
1002

1003
  QRY_ERR_RET(notifyWaitingSessions(pGroup->waitQueue));
174,318!
1004
  if (pGroup == pSession->pGroupData) {
174,318✔
1005
    if (pGroup->needCache) {
69,068!
1006
      pSession->lastBlkId = newBlkIdx;
69,068✔
1007
    }
1008
    
1009
    *continueFetch = false;
69,068✔
1010
  }
1011

1012
  return TSDB_CODE_SUCCESS;
174,318✔
1013
}
1014

1015
static int32_t handleDownstreamFetchDone(struct SOperatorInfo* pOperator, SGcSessionCtx* pSession) {
52,834✔
1016
  int32_t code = TSDB_CODE_SUCCESS;
52,834✔
1017
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
52,834✔
1018
  SGcDownstreamCtx* pCtx = &pGCache->pDownstreams[pSession->downstreamIdx];  
52,834✔
1019
  if (pGCache->batchFetch) {
52,834!
1020
    SHashObj* pGrpHash = pGCache->globalGrp ? pGCache->pGrpHash : pCtx->pGrpHash;
52,834!
1021
    SGroupCacheData* pGroup = NULL;
52,834✔
1022
    while (NULL != (pGroup = taosHashIterate(pGrpHash, pGroup))) {
201,149✔
1023
      QRY_ERR_RET(handleGroupFetchDone(pGroup));
148,315!
1024
    }
1025
    pCtx->fetchDone = true;
52,834✔
1026
  } else {
1027
    int32_t uidNum = 0;
×
1028
    SGcVgroupCtx* pVgCtx = NULL;
×
1029
    int32_t iter = 0;
×
1030
    while (NULL != (pVgCtx = tSimpleHashIterate(pCtx->pVgTbHash, pVgCtx, &iter))) {
×
1031
      uidNum = taosArrayGetSize(pVgCtx->pTbList);
×
1032
      for (int32_t i = 0; i < uidNum; ++i) {
×
1033
        SGcNewGroupInfo* pNew = taosArrayGet(pVgCtx->pTbList, i);
×
1034
        QRY_ERR_RET(handleGroupFetchDone(pNew->pGroup));
×
1035
      }
1036
      taosArrayClear(pVgCtx->pTbList);
×
1037
    }    
1038
  }
1039

1040
  taosHashClear(pCtx->pWaitSessions);
52,834✔
1041

1042
  return TSDB_CODE_SUCCESS;
52,834✔
1043
}
1044

1045
static int32_t getCacheBlkFromDownstreamOperator(struct SOperatorInfo* pOperator, SGcDownstreamCtx* pCtx, int64_t sessionId, SGcSessionCtx* pSession, SSDataBlock** ppRes) {
121,902✔
1046
  bool continueFetch = true;
121,902✔
1047
  int32_t code = TSDB_CODE_SUCCESS;
121,902✔
1048
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
121,902✔
1049

1050
  while (continueFetch && TSDB_CODE_SUCCESS == code) {
296,220!
1051
    QRY_ERR_RET(getBlkFromDownstreamOperator(pOperator, pSession->downstreamIdx, ppRes));
454,304!
1052
    
1053
    if (NULL == *ppRes) {
227,152✔
1054
      QRY_ERR_RET(handleDownstreamFetchDone(pOperator, pSession));
52,834!
1055
      break;
52,834✔
1056
    } else {
1057
      QRY_ERR_RET(handleGroupCacheRetrievedBlk(pOperator, *ppRes, pSession, &continueFetch));
174,318!
1058
    }
1059
  }
1060

1061
  if (!continueFetch) {
121,902!
1062
    SGcSessionCtx** ppWaitCtx = taosHashIterate(pCtx->pWaitSessions, NULL);
69,068✔
1063
    if (ppWaitCtx) {
69,068!
1064
      taosHashCancelIterate(pCtx->pWaitSessions, ppWaitCtx);
×
1065
      int64_t* pSessionId = taosHashGetKey(ppWaitCtx, NULL);
×
1066
      if (sessionId != atomic_val_compare_exchange_64(&pCtx->fetchSessionId, sessionId, *pSessionId)) {
×
1067
        qError("wrong fetch sessionId: %" PRIu64 " expected: %" PRIu64 , pCtx->fetchSessionId, sessionId);
×
1068
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1069
      }
1070
      SGcSessionCtx* pWaitCtx = *ppWaitCtx;
×
1071
      pWaitCtx->newFetch = true;
×
1072
      code = taosHashRemove(pCtx->pWaitSessions, pSessionId, sizeof(*pSessionId));
×
1073
      if (code) {
×
1074
        qError("taosHashRemove session %" PRId64 " from waitSession failed, error: %s", *pSessionId, tstrerror(code));
×
1075
        return code;
×
1076
      }
1077
      QRY_ERR_RET(tsem_post(&pWaitCtx->waitSem));
×
1078

1079
      return code;
×
1080
    }
1081
  }
1082

1083
  if (sessionId != atomic_val_compare_exchange_64(&pCtx->fetchSessionId, sessionId, -1)) {
121,902!
1084
    qError("wrong fetch sessionId: %" PRIu64 " expected: %" PRIu64 , pCtx->fetchSessionId, sessionId);
×
1085
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1086
  }
1087
  
1088
  return code;
121,902✔
1089
}
1090

1091
static int32_t getBlkFromSessionCacheImpl(struct SOperatorInfo* pOperator, int64_t sessionId, SGcSessionCtx* pSession, SSDataBlock** ppRes, bool* got) {
399,079✔
1092
  int32_t code = TSDB_CODE_SUCCESS;
399,079✔
1093
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
399,079✔
1094
  *got = true;
399,079✔
1095

1096
  if (NULL != pSession->pGroupData) {
399,079!
1097
    if (pSession->pGroupData->needCache) {
399,079!
1098
      SGcBlkList* pBlkList = &pSession->pGroupData->blkList;
399,079✔
1099
      taosRLockLatch(&pBlkList->lock);
399,079✔
1100
      int64_t blkNum = taosArrayGetSize(pBlkList->pList);
399,079✔
1101
      if (pSession->lastBlkId < 0) {
399,079✔
1102
        if (blkNum > 0) {
217,075✔
1103
          SGcBlkBufBasic* pBasic = taosArrayGet(pBlkList->pList, 0);
144,404✔
1104
          taosRUnLockLatch(&pBlkList->lock);
144,404✔
1105
          code = retrieveBlkFromBufCache(pGCache, pSession->pGroupData, sessionId, pBasic, ppRes);
144,404✔
1106
          pSession->lastBlkId = 0;
144,404✔
1107
          return code;
144,404✔
1108
        }
1109
      } else if ((pSession->lastBlkId + 1) < blkNum) {
182,004!
1110
        SGcBlkBufBasic* pBasic = taosArrayGet(pBlkList->pList, pSession->lastBlkId + 1);
×
1111
        taosRUnLockLatch(&pBlkList->lock);
×
1112
        code = retrieveBlkFromBufCache(pGCache, pSession->pGroupData, sessionId, pBasic, ppRes);
×
1113
        pSession->lastBlkId++;
×
1114
        return code;
×
1115
      }
1116
      taosRUnLockLatch(&pBlkList->lock);
254,675✔
1117
    } else if (pSession->pGroupData->pBlock) {
×
1118
      *ppRes = pSession->pGroupData->pBlock;
×
1119
      pSession->pGroupData->pBlock = NULL;
×
1120
      return TSDB_CODE_SUCCESS;
×
1121
    }
1122

1123
    if (atomic_load_8((int8_t*)&pSession->pGroupData->fetchDone)) {
254,675✔
1124
      *ppRes = NULL;
132,773✔
1125
      qDebug("sessionId: %" PRIu64 " fetch done", sessionId);
132,773!
1126
      return code;
132,773✔
1127
    }
1128
  } else {
1129
    *ppRes = NULL;
×
1130
    qDebug("sessionId: %" PRIu64 " fetch done since downstream fetch done", sessionId);
×
1131
    return code;
×
1132
  }
1133

1134
  *got = false;
121,902✔
1135
  return code;
121,902✔
1136
}
1137

1138

1139
static int32_t groupCacheSessionWait(struct SOperatorInfo* pOperator, SGcDownstreamCtx* pCtx, int64_t sessionId, SGcSessionCtx* pSession, SSDataBlock** ppRes) {
×
1140
  // FOR NOW, IT'S ERROR TO REACH HERE
1141
#if 1
1142
  qError("should not enter session wait");
×
1143
  return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1144
#else
1145
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
1146
  SGroupCacheData* pGroup = pSession->pGroupData;
1147
  int32_t code = TSDB_CODE_SUCCESS;
1148
  bool inLock = true;
1149
  if (NULL == pGroup->waitQueue) {
1150
    pGroup->waitQueue = taosArrayInit(1, POINTER_BYTES);
1151
    if (NULL == pGroup->waitQueue) {
1152
      QRY_ERR_JRET(terrno);
1153
    }
1154
  }
1155
  
1156
  if (NULL == taosArrayPush(pGroup->waitQueue, &pSession)) {
1157
    QRY_ERR_JRET(terrno);
1158
  }
1159

1160
  if (!pSession->semInit) {
1161
    QRY_ERR_JRET(tsem_init(&pSession->waitSem, 0, 0));
1162
    pSession->semInit = true;
1163
  }
1164

1165
  (void)taosThreadMutexUnlock(&pSession->pGroupData->mutex);
1166
  inLock = false;
1167

1168
  QRY_ERR_JRET(taosHashPut(pCtx->pWaitSessions, &sessionId, sizeof(sessionId), &pSession, POINTER_BYTES));
1169

1170
  code = tsem_wait(&pSession->waitSem);
1171
  if (code) {
1172
    qError("tsem_wait failed, error:%s", tstrerror(code));
1173
    QRY_ERR_JRET(code);
1174
  }
1175

1176
  if (pSession->newFetch) {
1177
    pSession->newFetch = false;
1178
    return getCacheBlkFromDownstreamOperator(pOperator, pCtx, sessionId, pSession, ppRes);
1179
  }
1180

1181
  code = taosHashRemove(pCtx->pWaitSessions, &sessionId, sizeof(sessionId));
1182
  if (code) {
1183
    qError("taosHashRemove session %" PRId64 " from waitSession failed, error: %s", sessionId, tstrerror(code));
1184
    QRY_ERR_JRET(code);
1185
  }
1186

1187
  bool got = false;
1188
  return getBlkFromSessionCacheImpl(pOperator, sessionId, pSession, ppRes, &got);
1189

1190
_return:
1191

1192
  if (inLock) {
1193
    (void)taosThreadMutexUnlock(&pSession->pGroupData->mutex);
1194
  }
1195

1196
  return code;
1197
#endif
1198
}
1199

1200

1201
static int32_t getBlkFromSessionCache(struct SOperatorInfo* pOperator, int64_t sessionId, SGcSessionCtx* pSession, SSDataBlock** ppRes) {
399,079✔
1202
  int32_t code = TSDB_CODE_SUCCESS;
399,079✔
1203
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
399,079✔
1204
  bool locked = false;
399,079✔
1205
  SGcDownstreamCtx* pCtx = &pGCache->pDownstreams[pSession->downstreamIdx];
399,079✔
1206
  
1207
  while (true) {
×
1208
    bool got = false;
399,079✔
1209
    code = getBlkFromSessionCacheImpl(pOperator, sessionId, pSession, ppRes, &got);
399,079✔
1210
    if (TSDB_CODE_SUCCESS != code || got) {
399,079!
1211
      goto _return;
277,177✔
1212
    }
1213
    
1214
    if ((atomic_load_64(&pCtx->fetchSessionId) == sessionId)
121,902!
1215
      || (-1 == atomic_val_compare_exchange_64(&pCtx->fetchSessionId, -1, sessionId))) {
121,902!
1216
      if (locked) {
121,902!
1217
        (void)taosThreadMutexUnlock(&pSession->pGroupData->mutex);
×
1218
        locked = false;
×
1219
      }
1220
      
1221
      code = getCacheBlkFromDownstreamOperator(pOperator, pCtx, sessionId, pSession, ppRes);
121,902✔
1222
      goto _return;
121,902✔
1223
    } else {
1224
      // FOR NOW, SHOULD NOT REACH HERE
1225
      qError("Invalid fetchSessionId:%" PRId64 ", currentSessionId:%" PRId64, pCtx->fetchSessionId, sessionId);
×
1226
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1227
    }
1228

1229
    if (locked) {
1230
      code = groupCacheSessionWait(pOperator, pCtx, sessionId, pSession, ppRes);
1231
      locked = false;
1232
      if (TSDB_CODE_SUCCESS != code) {
1233
        goto _return;
1234
      }
1235
      
1236
      break;
1237
    }
1238
    
1239
    (void)taosThreadMutexLock(&pSession->pGroupData->mutex);
1240
    locked = true;
1241
  };
1242

1243

1244
_return:
399,079✔
1245

1246
  if (locked) {
399,079!
1247
    (void)taosThreadMutexUnlock(&pSession->pGroupData->mutex);
×
1248
  }
1249

1250
  return code;
399,079✔
1251
}
1252

1253

1254
static int32_t initGroupCacheBlockCache(SGroupCacheOperatorInfo* pInfo) {
33,099✔
1255
  SGcBlkCacheInfo* pCache = &pInfo->blkCache;
33,099✔
1256
  pCache->pDirtyBlk = taosHashInit(10, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
33,099✔
1257
  if (NULL == pCache->pDirtyBlk) {
33,099!
1258
    return terrno;
×
1259
  }
1260
  taosHashSetFreeFp(pCache->pDirtyBlk, freeGcBlkBufInfo);
33,099✔
1261
  pCache->pReadBlk = taosHashInit(10, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
33,099✔
1262
  if (NULL == pCache->pReadBlk) {
33,099!
1263
    return terrno;
×
1264
  }
1265
  pCache->writeDownstreamId = -1;
33,099✔
1266

1267
  return TSDB_CODE_SUCCESS;
33,099✔
1268
}
1269

1270
static FORCE_INLINE void initGroupCacheSessionCtx(SGcSessionCtx* pSession, SGcOperatorParam* pGcParam, SGroupCacheData* pGroup) {
1271
  pSession->pParam = pGcParam;
217,075✔
1272
  pSession->downstreamIdx = pGcParam->downstreamIdx;
217,075✔
1273
  pSession->pGroupData = pGroup;
217,075✔
1274
  pSession->lastBlkId = -1;
217,075✔
1275
}
217,075✔
1276

1277
static int32_t initGroupCacheSession(struct SOperatorInfo* pOperator, SOperatorParam* pParam, SGcSessionCtx** ppSession) {
227,433✔
1278
  int32_t code = TSDB_CODE_SUCCESS;
227,433✔
1279
  SGcSessionCtx ctx = {0};
227,433✔
1280
  SGcOperatorParam* pGcParam = pParam->value;  
227,433✔
1281
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
227,433✔
1282
  SGcDownstreamCtx* pCtx = &pGCache->pDownstreams[pParam->downstreamIdx];
227,433✔
1283
  SHashObj* pGrpHash = pGCache->globalGrp ? pGCache->pGrpHash : pCtx->pGrpHash;
227,433!
1284

1285
  SGroupCacheData* pGroup = taosHashGet(pGrpHash, &pGcParam->tbUid, sizeof(pGcParam->tbUid));
227,433✔
1286
  if (NULL == pGroup && (NULL != pParam->pChildren || !pCtx->fetchDone)) {
227,433!
1287
    code = addNewGroupData(pOperator, pParam, &pGroup, pGCache->batchFetch ? GROUP_CACHE_DEFAULT_VGID : pGcParam->vgId, pGcParam->tbUid);
72,671!
1288
    if (TSDB_CODE_SUCCESS != code) {
72,671!
1289
      return code;
×
1290
    }
1291
  }
1292

1293
  if (NULL == pGroup) {
227,433✔
1294
    return TSDB_CODE_SUCCESS;
10,358✔
1295
  }
1296

1297
  initGroupCacheSessionCtx(&ctx, pGcParam, pGroup);
217,075✔
1298

1299
  code = taosHashPut(pCtx->pSessions, &pGcParam->sessionId, sizeof(pGcParam->sessionId), &ctx, sizeof(ctx));
217,075✔
1300
  if (TSDB_CODE_SUCCESS != code) {
217,075!
1301
    return code;
×
1302
  }
1303

1304
  *ppSession = taosHashGet(pCtx->pSessions, &pGcParam->sessionId, sizeof(pGcParam->sessionId));
217,075✔
1305
  if (NULL == *ppSession) {
217,075!
1306
    qError("fail to get session %" PRId64 " from pSessions", pGcParam->sessionId);
×
1307
    QRY_ERR_RET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
1308
  }
1309

1310
  qDebug("session:%" PRId64 " initialized, downstreamIdx:%d, vgId:%d, tbUid:%" PRId64 ", needCache:%d", 
217,075!
1311
    pGcParam->sessionId, pGcParam->downstreamIdx, pGcParam->vgId, pGcParam->tbUid, pGcParam->needCache);
1312
  
1313
  return TSDB_CODE_SUCCESS;
217,075✔
1314
}
1315

1316
static int32_t getBlkFromGroupCache(struct SOperatorInfo* pOperator, SSDataBlock** ppRes, SOperatorParam* pParam) {
409,437✔
1317
  int32_t code = TSDB_CODE_SUCCESS;
409,437✔
1318
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
409,437✔
1319
  SGcOperatorParam* pGcParam = pParam->value;
409,437✔
1320
  SGcDownstreamCtx* pCtx = &pGCache->pDownstreams[pParam->downstreamIdx];
409,437✔
1321
  SGcSessionCtx* pSession = taosHashGet(pCtx->pSessions, &pGcParam->sessionId, sizeof(pGcParam->sessionId));
409,437✔
1322
  if (NULL == pSession) {
409,437✔
1323
    int32_t code = initGroupCacheSession(pOperator, pParam, &pSession);
227,433✔
1324
    if (TSDB_CODE_SUCCESS != code) {
227,433!
1325
      return code;
×
1326
    }
1327
    if (NULL == pSession) {
227,433✔
1328
      qDebug("session %" PRId64 " in downstream %d total got 0 rows since downtream fetch done", pGcParam->sessionId, pCtx->id);
10,358!
1329
      return TSDB_CODE_SUCCESS;
10,358✔
1330
    }
1331
  } else if (pSession->pGroupData->needCache) {
182,004!
1332
    SSDataBlock** ppBlock = taosHashGet(pGCache->blkCache.pReadBlk, &pGcParam->sessionId, sizeof(pGcParam->sessionId));
182,004✔
1333
    if (ppBlock) {
182,004✔
1334
      QRY_ERR_RET(releaseBaseBlockToList(pCtx, *ppBlock));
132,773!
1335
      code = taosHashRemove(pGCache->blkCache.pReadBlk, &pGcParam->sessionId, sizeof(pGcParam->sessionId));
132,773✔
1336
      if (code) {
132,773!
1337
        qError("taosHashRemove session %" PRId64 " from pReadBlk failed, error: %s", pGcParam->sessionId, tstrerror(code));
×
1338
        QRY_ERR_RET(code);
×
1339
      }
1340
    }
1341
  }
1342
  
1343
  QRY_ERR_RET(getBlkFromSessionCache(pOperator, pGcParam->sessionId, pSession, ppRes));
399,079!
1344
  if (NULL == *ppRes) {
399,079✔
1345
    qDebug("session %" PRId64 " in downstream %d total got %" PRId64 " rows", pGcParam->sessionId, pCtx->id, pSession->resRows);
185,607!
1346
    code = taosHashRemove(pCtx->pSessions, &pGcParam->sessionId, sizeof(pGcParam->sessionId));
185,607✔
1347
    if (code) {
185,607!
1348
      qError("taosHashRemove session %" PRId64 " from pSessions failed, error: %s", pGcParam->sessionId, tstrerror(code));
×
1349
      QRY_ERR_RET(code);
×
1350
    }
1351
  } else {
1352
    pSession->resRows += (*ppRes)->info.rows;
213,472✔
1353
    qDebug("session %" PRId64 " in downstream %d got %" PRId64 " rows in one block", pGcParam->sessionId, pCtx->id, (*ppRes)->info.rows);
213,472!
1354
  }
1355

1356
  return code;
399,079✔
1357
}
1358

1359
static int32_t initGroupCacheExecInfo(SOperatorInfo*        pOperator) {
33,099✔
1360
  SGroupCacheOperatorInfo* pInfo = pOperator->info;
33,099✔
1361
  pInfo->execInfo.pDownstreamBlkNum = taosMemoryCalloc(pOperator->numOfDownstream, sizeof(int64_t));
33,099!
1362
  if (NULL == pInfo->execInfo.pDownstreamBlkNum) {
33,099!
1363
    return terrno;
×
1364
  }
1365
  return TSDB_CODE_SUCCESS;
33,099✔
1366
}
1367

1368
static void freeRemoveGroupCacheData(void* p) {
177,921✔
1369
  SGroupCacheData* pGroup = p;
177,921✔
1370
  if (pGroup->vgId > 0 && pGroup->needCache) {
177,921!
1371
    SGcFileCacheCtx* pFileCtx = &pGroup->pVgCtx->fileCtx;
×
1372
    if (pGroup->fileId >= 0) {
×
1373
      SGroupCacheFileInfo* pFileInfo = taosHashGet(pFileCtx->pCacheFile, &pGroup->fileId, sizeof(pGroup->fileId));
×
1374
      uint32_t remainNum = atomic_sub_fetch_32(&pFileInfo->groupNum, 1);
×
1375

1376
      qTrace("FileId:%d-%d-%d sub group num to %u", pGroup->downstreamIdx, pGroup->vgId, pFileCtx->fileId, remainNum);
×
1377

1378
      if (0 == remainNum && pGroup->fileId != pFileCtx->fileId) {
×
1379
        removeGroupCacheFile(pFileInfo);
×
1380

1381
#if 0
1382
        /* debug only */
1383
        snprintf(&pFileCtx->baseFilename[pFileCtx->baseNameLen], sizeof(pFileCtx->baseFilename) - pFileCtx->baseNameLen, "_%d", pGroup->fileId);
1384
        taosRemoveFile(pFileCtx->baseFilename);
1385
        /* debug only */
1386
#endif
1387

1388
        qTrace("FileId:%d-%d-%d removed", pGroup->downstreamIdx, pGroup->vgId, pFileCtx->fileId);
×
1389
        //taosHashRemove(pFileCtx->pCacheFile, &pGroup->fileId, sizeof(pGroup->fileId));
1390
      }
1391
    }
1392
  }
1393

1394
  taosArrayDestroy(pGroup->waitQueue);
177,921✔
1395
  taosArrayDestroy(pGroup->blkList.pList);
177,921✔
1396
  (void)taosThreadMutexDestroy(&pGroup->mutex);
177,921✔
1397

1398
  qTrace("group removed");
177,921!
1399
}
177,921✔
1400

1401

1402

1403
static int32_t initGroupCacheDownstreamCtx(SOperatorInfo*          pOperator) {
33,099✔
1404
  SGroupCacheOperatorInfo* pInfo = pOperator->info;
33,099✔
1405
  pInfo->pDownstreams = taosMemoryCalloc(pOperator->numOfDownstream, sizeof(*pInfo->pDownstreams));
33,099!
1406
  if (NULL == pInfo->pDownstreams) {
33,099!
1407
    return terrno;
×
1408
  }
1409
  pInfo->downstreamNum = pOperator->numOfDownstream;
33,099✔
1410

1411
  for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
99,297✔
1412
    SGcDownstreamCtx* pCtx = &pInfo->pDownstreams[i];
66,198✔
1413
    pCtx->id = i;
66,198✔
1414
    pCtx->fetchSessionId = -1;
66,198✔
1415
    pCtx->lastBlkUid = 0;
66,198✔
1416
    pCtx->pVgTbHash = tSimpleHashInit(10, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT));
66,198✔
1417
    if (NULL == pCtx->pVgTbHash) {
66,198!
1418
      return terrno;
×
1419
    }
1420
    tSimpleHashSetFreeFp(pCtx->pVgTbHash, freeSGcVgroupCtx);      
66,198✔
1421

1422
    if (pInfo->batchFetch) {
66,198!
1423
      int32_t defaultVg = 0;
66,198✔
1424
      SGcVgroupCtx vgCtx = {0};
66,198✔
1425
      initGcVgroupCtx(pOperator, &vgCtx, pCtx->id, defaultVg, NULL);      
66,198✔
1426
      QRY_ERR_RET(tSimpleHashPut(pCtx->pVgTbHash, &defaultVg, sizeof(defaultVg), &vgCtx, sizeof(vgCtx)));
66,198!
1427
    }
1428
    
1429
    pCtx->pNewGrpList = taosArrayInit(10, sizeof(SGcNewGroupInfo));
66,198✔
1430
    if (NULL == pCtx->pNewGrpList) {
66,198!
1431
      return terrno;
×
1432
    }
1433
    if (!pInfo->globalGrp) {
66,198!
1434
      pCtx->pGrpHash = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
66,198✔
1435
      if (pCtx->pGrpHash == NULL) {
66,198!
1436
        return terrno;
×
1437
      }
1438
      taosHashSetFreeFp(pCtx->pGrpHash, freeRemoveGroupCacheData);      
66,198✔
1439
    }
1440

1441
    pCtx->pSessions = taosHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
66,198✔
1442
    if (pCtx->pSessions == NULL) {
66,198!
1443
      return terrno;
×
1444
    }
1445
    taosHashSetFreeFp(pCtx->pSessions, freeSGcSessionCtx);
66,198✔
1446
  
1447
    pCtx->pFreeBlock = taosArrayInit(10, POINTER_BYTES);
66,198✔
1448
    if (NULL == pCtx->pFreeBlock) {
66,198!
1449
      return terrno;
×
1450
    }
1451
    
1452
    pCtx->pWaitSessions = taosHashInit(20, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
66,198✔
1453
    if (pCtx->pWaitSessions == NULL) {
66,198!
1454
      return terrno;
×
1455
    }
1456

1457
    (void)snprintf(pCtx->fileCtx.baseFilename, sizeof(pCtx->fileCtx.baseFilename) - 1, "%s/gc_%d_%" PRIx64 "_%" PRIu64 "_%p_%d", 
198,594✔
1458
      tsTempDir, taosGetPId(), pOperator->pTaskInfo->id.queryId, pOperator->pTaskInfo->id.taskId, pOperator, pCtx->id);
132,396✔
1459
    pCtx->fileCtx.baseFilename[sizeof(pCtx->fileCtx.baseFilename) - 1] = 0;
66,198✔
1460
    pCtx->fileCtx.baseNameLen = strlen(pCtx->fileCtx.baseFilename);
66,198!
1461
  }
1462

1463
  return TSDB_CODE_SUCCESS;
33,099✔
1464
}
1465

1466
static int32_t groupCacheGetNext(struct SOperatorInfo* pOperator, SOperatorParam* pParam, SSDataBlock** pRes) {
409,437✔
1467
  *pRes = NULL;
409,437✔
1468

1469
  SSDataBlock* pBlock = NULL;
409,437✔
1470
  int64_t      st = 0;
409,437✔
1471
  int32_t      code = 0;
409,437✔
1472

1473
  if (pOperator->cost.openCost == 0) {
409,437✔
1474
    st = taosGetTimestampUs();
32,275✔
1475
  }
1476

1477
  code = getBlkFromGroupCache(pOperator, &pBlock, pParam);
409,437✔
1478
  if (TSDB_CODE_SUCCESS != code) {
409,437!
1479
    pOperator->pTaskInfo->code = code;
×
1480
    T_LONG_JMP(pOperator->pTaskInfo->env, pOperator->pTaskInfo->code);
×
1481
  }
1482

1483
  if (pOperator->cost.openCost == 0) {
409,437✔
1484
    pOperator->cost.openCost = (taosGetTimestampUs() - st) / 1000.0;
32,275✔
1485
  }
1486

1487
  *pRes = pBlock;
409,437✔
1488
  return code;
409,437✔
1489
}
1490

1491
static int32_t groupCacheTableCacheEnd(SOperatorInfo* pOperator, SOperatorParam* pParam) {
×
1492
  SGcNotifyOperatorParam* pGcParam = pParam->value;
×
1493
  SGroupCacheOperatorInfo* pGCache = pOperator->info;
×
1494
  SGcDownstreamCtx* pCtx = &pGCache->pDownstreams[pGcParam->downstreamIdx];
×
1495
  SHashObj* pGrpHash = pGCache->globalGrp ? pGCache->pGrpHash : pCtx->pGrpHash;
×
1496

1497
  qTrace("try to remove group %" PRIu64, pGcParam->tbUid);
×
1498
  if (taosHashRemove(pGrpHash, &pGcParam->tbUid, sizeof(pGcParam->tbUid))) {
×
1499
    qError("failed to remove group %" PRIu64 " in vgId %d downstreamIdx %d", pGcParam->tbUid, pGcParam->vgId, pGcParam->downstreamIdx);
×
1500
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1501
  }
1502

1503
  return TSDB_CODE_SUCCESS;
×
1504
}
1505

1506
static void resetGroupCacheBlockCache(SGcBlkCacheInfo* pCache) {
×
1507
  taosHashClear(pCache->pDirtyBlk);
×
1508

1509
  void* p = NULL;
×
1510
  while (NULL != (p = taosHashIterate(pCache->pReadBlk, p))) {
×
1511
    blockDataDeepCleanup(*(SSDataBlock**)p);
×
1512
    freeGcBlockInList(p);
×
1513
  }
1514

1515
  taosHashClear(pCache->pReadBlk);
×
1516

1517
  pCache->dirtyLock = 0;
×
1518
  pCache->pDirtyHead = NULL;
×
1519
  pCache->pDirtyTail = NULL;
×
1520
  pCache->blkCacheSize = 0;
×
1521
  pCache->writeDownstreamId = -1;
×
1522

1523
  return;
×
1524
}
1525

1526
static int32_t resetGroupCacheDownstreamCtx(SOperatorInfo* pOper) {
×
1527
  SGroupCacheOperatorInfo* pInfo = pOper->info;
×
1528
  if (NULL == pInfo->pDownstreams) {
×
1529
    return TSDB_CODE_SUCCESS;
×
1530
  }
1531
  
1532
  for (int32_t i = 0; i < pInfo->downstreamNum; ++i) {
×
1533
    SGcDownstreamCtx* pCtx = &pInfo->pDownstreams[i];
×
1534
    taosArrayClear(pCtx->pNewGrpList);
×
1535
    taosHashClear(pCtx->pGrpHash);
×
1536

1537
    tSimpleHashClear(pCtx->pVgTbHash);
×
1538
    if (pInfo->batchFetch) {
×
1539
      int32_t defaultVg = 0;
×
1540
      SGcVgroupCtx vgCtx = {0};
×
1541
      initGcVgroupCtx(pOper, &vgCtx, pCtx->id, defaultVg, NULL);      
×
1542
      tSimpleHashPut(pCtx->pVgTbHash, &defaultVg, sizeof(defaultVg), &vgCtx, sizeof(vgCtx));
×
1543
    }
1544
    
1545
    taosArrayClearEx(pCtx->pFreeBlock, freeGcBlockInList);
×
1546
    taosHashClear(pCtx->pSessions);
×
1547
    taosHashClear(pCtx->pWaitSessions);
×
1548
    freeSGcFileCacheCtx(&pCtx->fileCtx);
×
1549

1550
    pCtx->grpLock = 0;
×
1551
    pCtx->fetchSessionId = -1;
×
1552
    pCtx->blkLock = 0;
×
1553
    pCtx->lastBlkUid = 0;
×
1554
    pCtx->pBaseBlock = NULL;
×
1555
    pCtx->fetchDone = false;
×
1556
  }
1557

1558
  return TSDB_CODE_SUCCESS;
×
1559
}
1560

1561
static int32_t resetGroupCacheOperState(SOperatorInfo* pOper) {
×
1562
  int32_t code = 0, lino = 0;
×
1563
  SGroupCacheOperatorInfo* pInfo = pOper->info;
×
1564

1565
  pOper->status = OP_NOT_OPENED;
×
1566

1567
  resetGroupCacheBlockCache(&pInfo->blkCache);
×
1568

1569
  taosHashClear(pInfo->pGrpHash);
×
1570

1571
  resetGroupCacheDownstreamCtx(pOper);
×
1572

1573
  memset(pInfo->execInfo.pDownstreamBlkNum, 0, pOper->numOfDownstream * sizeof(int64_t));
×
1574
  
1575
_exit:
×
1576

1577
  if (code) {
×
1578
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1579
  }
1580

1581
  return code;
×
1582
}
1583

1584
int32_t createGroupCacheOperatorInfo(SOperatorInfo** pDownstream, int32_t numOfDownstream,
33,099✔
1585
                                     SGroupCachePhysiNode* pPhyciNode, SExecTaskInfo* pTaskInfo,
1586
                                     SOperatorInfo** pOptrInfo) {
1587
  QRY_PARAM_CHECK(pOptrInfo);
33,099!
1588
  int32_t code = TSDB_CODE_SUCCESS;
33,099✔
1589

1590
  SGroupCacheOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SGroupCacheOperatorInfo));
33,099!
1591
  SOperatorInfo*           pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
33,099!
1592
  if (pOperator == NULL || pInfo == NULL) {
33,099!
1593
    code = terrno;
×
1594
    goto _error;
×
1595
  }
1596

1597
  pOperator->transparent = true;
33,099✔
1598
  pOperator->pPhyNode = pPhyciNode;
33,099✔
1599
  
1600
  setOperatorInfo(pOperator, "GroupCacheOperator", QUERY_NODE_PHYSICAL_PLAN_GROUP_CACHE, false, OP_NOT_OPENED, pInfo, pTaskInfo);
33,099✔
1601

1602
  pInfo->maxCacheSize = 0;
33,099✔
1603
  pInfo->grpByUid = pPhyciNode->grpByUid;
33,099!
1604
  pInfo->globalGrp = pPhyciNode->globalGrp;
33,099!
1605
  pInfo->batchFetch = pPhyciNode->batchFetch;
33,099!
1606
  
1607
  if (!pInfo->grpByUid) {
33,099!
1608
    qError("only group cache by uid is supported now");
×
1609
    code = TSDB_CODE_INVALID_PARA;
×
1610
    goto _error;
×
1611
  }
1612
  
1613
  if (pPhyciNode->pGroupCols) {
33,099!
1614
    code = initGroupColsInfo(&pInfo->groupColsInfo, pPhyciNode->grpColsMayBeNull, pPhyciNode->pGroupCols);
×
1615
    if (code) {
×
1616
      goto _error;
×
1617
    }
1618
  }
1619

1620
  code = initGroupCacheBlockCache(pInfo);
33,099✔
1621
  if (code) {
33,099!
1622
    goto _error;
×
1623
  }
1624

1625
  if (pInfo->globalGrp) {
33,099!
1626
    pInfo->pGrpHash = taosHashInit(32, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_ENTRY_LOCK);
×
1627
    if (pInfo->pGrpHash == NULL) {
×
1628
      code = terrno;
×
1629
      goto _error;
×
1630
    }
1631
    taosHashSetFreeFp(pInfo->pGrpHash, freeRemoveGroupCacheData);
×
1632
  }
1633

1634
  code = appendDownstream(pOperator, pDownstream, numOfDownstream);
33,099✔
1635
  if (TSDB_CODE_SUCCESS != code) {
33,099!
1636
    goto _error;
×
1637
  }
1638

1639
  code = initGroupCacheDownstreamCtx(pOperator);
33,099✔
1640
  if (TSDB_CODE_SUCCESS != code) {
33,099!
1641
    goto _error;
×
1642
  }
1643

1644
  code = initGroupCacheExecInfo(pOperator);
33,099✔
1645
  if (TSDB_CODE_SUCCESS != code) {
33,099!
1646
    goto _error;
×
1647
  }
1648

1649
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, NULL, NULL, destroyGroupCacheOperator, optrDefaultBufFn, NULL, groupCacheGetNext, groupCacheTableCacheEnd);
33,099✔
1650

1651
  setOperatorResetStateFn(pOperator, resetGroupCacheOperState);
33,099✔
1652

1653
  qTrace("new group cache operator, maxCacheSize:%" PRId64 ", globalGrp:%d, batchFetch:%d", pInfo->maxCacheSize, pInfo->globalGrp, pInfo->batchFetch);
33,099!
1654

1655
  *pOptrInfo = pOperator;
33,099✔
1656
  return TSDB_CODE_SUCCESS;
33,099✔
1657

1658
_error:
×
1659
  if (pInfo != NULL) {
×
1660
    destroyGroupCacheOperator(pInfo);
×
1661
  }
1662

1663
  destroyOperatorAndDownstreams(pOperator, pDownstream, numOfDownstream);
×
1664
  pTaskInfo->code = code;
×
1665
  return code;
×
1666
}
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