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

taosdata / TDengine / #3534

21 Nov 2024 07:36AM UTC coverage: 60.825% (+2.0%) from 58.848%
#3534

push

travis-ci

web-flow
Merge pull request #28810 from taosdata/ehn/add-sync-heartbeat-sent-time-to-log

ehn:add-sync-heartbeat-sent-time-to-log

120023 of 252376 branches covered (47.56%)

Branch coverage included in aggregate %.

43 of 47 new or added lines in 3 files covered. (91.49%)

2254 existing lines in 162 files now uncovered.

200876 of 275203 relevant lines covered (72.99%)

16110754.39 hits per line

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

52.55
/source/libs/executor/src/hashjoinoperator.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 "hashjoin.h"
29
#include "functionMgt.h"
30

31

32
bool hJoinBlkReachThreshold(SHJoinOperatorInfo* pInfo, int64_t blkRows) {
21,665✔
33
  if (INT64_MAX == pInfo->ctx.limit || pInfo->pFinFilter != NULL) {
21,665!
34
    return blkRows >= pInfo->blkThreshold;
21,665✔
35
  }
36
  
37
  return (pInfo->execInfo.resRows + blkRows) >= pInfo->ctx.limit;
×
38
}
39

40
int32_t hJoinHandleMidRemains(SHJoinOperatorInfo* pJoin, SHJoinCtx* pCtx) {
×
41
  TSWAP(pJoin->midBlk, pJoin->finBlk);
×
42

43
  pCtx->midRemains = false;
×
44

45
  return TSDB_CODE_SUCCESS;
×
46
}
47

48
int32_t hJoinCopyMergeMidBlk(SHJoinCtx* pCtx, SSDataBlock** ppMid, SSDataBlock** ppFin) {
×
49
  SSDataBlock* pLess = *ppMid;
×
50
  SSDataBlock* pMore = *ppFin;
×
51

52
/*
53
  if ((*ppMid)->info.rows < (*ppFin)->info.rows) {
54
    pLess = (*ppMid);
55
    pMore = (*ppFin);
56
  } else {
57
    pLess = (*ppFin);
58
    pMore = (*ppMid);
59
  }
60
*/
61

62
  int32_t totalRows = pMore->info.rows + pLess->info.rows;
×
63
  if (totalRows <= pMore->info.capacity) {
×
64
    HJ_ERR_RET(blockDataMerge(pMore, pLess));
×
65
    blockDataCleanup(pLess);
×
66
    pCtx->midRemains = false;
×
67
  } else {
68
    int32_t copyRows = pMore->info.capacity - pMore->info.rows;
×
69
    HJ_ERR_RET(blockDataMergeNRows(pMore, pLess, pLess->info.rows - copyRows, copyRows));
×
70
    blockDataShrinkNRows(pLess, copyRows);
×
71
    pCtx->midRemains = true;
×
72
  }
73

74
/*
75
  if (pMore != (*ppFin)) {
76
    TSWAP(*ppMid, *ppFin);
77
  }
78
*/
79

80
  return TSDB_CODE_SUCCESS;
×
81
}
82

83
int32_t hJoinSetImplFp(SHJoinOperatorInfo* pJoin) {
11,002✔
84
  switch (pJoin->joinType) {
11,002!
85
    case JOIN_TYPE_INNER:
11,002✔
86
      pJoin->joinFp = hInnerJoinDo;
11,002✔
87
      break;
11,002✔
88
    case JOIN_TYPE_LEFT:
×
89
    case JOIN_TYPE_RIGHT: {
90
      switch (pJoin->subType) {
×
91
        case JOIN_STYPE_OUTER:          
×
92
          pJoin->joinFp = hLeftJoinDo;
×
93
          break;
×
94
        default:
×
95
          break;
×
96
      }
97
      break;
×
98
    }      
99
    default:
×
100
      break;
×
101
  }
102

103
  return TSDB_CODE_SUCCESS;
11,002✔
104
}
105

106

107
int32_t hJoinLaunchPrimExpr(SSDataBlock* pBlock, SHJoinTableCtx* pTable, int32_t startIdx, int32_t endIdx) {
43,372✔
108
  if (NULL == pTable->primExpr) {
43,372!
109
    return TSDB_CODE_SUCCESS;
43,372✔
110
  }
111

112
  SHJoinPrimExprCtx* pCtx = &pTable->primCtx;
×
113
  SColumnInfoData* pPrimIn = taosArrayGet(pBlock->pDataBlock, pTable->primCol->srcSlot);
×
114
  SColumnInfoData* pPrimOut = taosArrayGet(pBlock->pDataBlock, pTable->primCtx.targetSlotId);
×
115
  if (0 != pCtx->timezoneUnit) {
×
116
    for (int32_t i = startIdx; i <= endIdx; ++i) {
×
117
      ((int64_t*)pPrimOut->pData)[i] = ((int64_t*)pPrimIn->pData)[i] - (((int64_t*)pPrimIn->pData)[i] - pCtx->timezoneUnit) % pCtx->truncateUnit;
×
118
    }
119
  } else {
120
    for (int32_t i = startIdx; i <= endIdx; ++i) {
×
121
      ((int64_t*)pPrimOut->pData)[i] = ((int64_t*)pPrimIn->pData)[i] / pCtx->truncateUnit * pCtx->truncateUnit;
×
122
    }
123
  }
124

125
  return TSDB_CODE_SUCCESS;
×
126
}
127

128

129
static int64_t hJoinGetSingleKeyRowsNum(SBufRowInfo* pRow) {
×
130
  int64_t rows = 0;
×
131
  while (pRow) {
×
132
    rows++;
×
133
    pRow = pRow->next;
×
134
  }
135
  return rows;
×
136
}
137

138
static int64_t hJoinGetRowsNumOfKeyHash(SSHashObj* pHash) {
×
139
  SGroupData* pGroup = NULL;
×
140
  int32_t iter = 0;
×
141
  int64_t rowsNum = 0;
×
142
  
143
  while (NULL != (pGroup = tSimpleHashIterate(pHash, pGroup, &iter))) {
×
144
    int32_t* pKey = tSimpleHashGetKey(pGroup, NULL);
×
145
    int64_t rows = hJoinGetSingleKeyRowsNum(pGroup->rows);
×
146
    //qTrace("build_key:%d, rows:%" PRId64, *pKey, rows);
147
    rowsNum += rows;
×
148
  }
149

150
  return rowsNum;
×
151
}
152

153
static int32_t hJoinInitKeyColsInfo(SHJoinTableCtx* pTable, SNodeList* pList) {
22,004✔
154
  pTable->keyNum = LIST_LENGTH(pList);
22,004!
155
  
156
  pTable->keyCols = taosMemoryMalloc(pTable->keyNum * sizeof(SHJoinColInfo));
22,004✔
157
  if (NULL == pTable->keyCols) {
22,004!
UNCOV
158
    return terrno;
×
159
  }
160

161
  int64_t bufSize = 0;
22,004✔
162
  int32_t i = 0;
22,004✔
163
  SNode* pNode = NULL;
22,004✔
164
  FOREACH(pNode, pList) {
44,032!
165
    SColumnNode* pColNode = (SColumnNode*)pNode;
22,028✔
166
    pTable->keyCols[i].srcSlot = pColNode->slotId;
22,028✔
167
    pTable->keyCols[i].vardata = IS_VAR_DATA_TYPE(pColNode->node.resType.type);
22,028!
168
    pTable->keyCols[i].bytes = pColNode->node.resType.bytes;
22,028✔
169
    bufSize += pColNode->node.resType.bytes;
22,028✔
170
    ++i;
22,028✔
171
  }  
172

173
  if (pTable->keyNum > 1) {
22,004✔
174
    pTable->keyBuf = taosMemoryMalloc(bufSize);
24✔
175
    if (NULL == pTable->keyBuf) {
24!
176
      return terrno;
×
177
    }
178
  }
179

180
  return TSDB_CODE_SUCCESS;
22,004✔
181
}
182

183
static void hJoinGetValColsNum(SNodeList* pList, int32_t blkId, int32_t* colNum) {
22,004✔
184
  *colNum = 0;
22,004✔
185
  
186
  SNode* pNode = NULL;
22,004✔
187
  FOREACH(pNode, pList) {
110,020!
188
    STargetNode* pTarget = (STargetNode*)pNode;
88,016✔
189
    SColumnNode* pCol = (SColumnNode*)pTarget->pExpr;
88,016✔
190
    if (pCol->dataBlockId == blkId) {
88,016✔
191
      (*colNum)++;
44,008✔
192
    }
193
  }
194
}
22,004✔
195

196
static bool hJoinIsValColInKeyCols(int16_t slotId, int32_t keyNum, SHJoinColInfo* pKeys, int32_t* pKeyIdx) {
44,008✔
197
  for (int32_t i = 0; i < keyNum; ++i) {
88,064✔
198
    if (pKeys[i].srcSlot == slotId) {
44,056!
199
      *pKeyIdx = i;
×
200
      return true;
×
201
    }
202
  }
203

204
  return false;
44,008✔
205
}
206

207
static int32_t hJoinInitValColsInfo(SHJoinTableCtx* pTable, SNodeList* pList) {
22,004✔
208
  hJoinGetValColsNum(pList, pTable->blkId, &pTable->valNum);
22,004✔
209
  if (pTable->valNum == 0) {
22,004!
210
    return TSDB_CODE_SUCCESS;
×
211
  }
212
  
213
  pTable->valCols = taosMemoryMalloc(pTable->valNum * sizeof(SHJoinColInfo));
22,004✔
214
  if (NULL == pTable->valCols) {
22,004!
215
    return terrno;
×
216
  }
217

218
  int32_t i = 0;
22,004✔
219
  int32_t colNum = 0;
22,004✔
220
  SNode* pNode = NULL;
22,004✔
221
  FOREACH(pNode, pList) {
110,020!
222
    STargetNode* pTarget = (STargetNode*)pNode;
88,016✔
223
    SColumnNode* pColNode = (SColumnNode*)pTarget->pExpr;
88,016✔
224
    if (pColNode->dataBlockId == pTable->blkId) {
88,016✔
225
      if (hJoinIsValColInKeyCols(pColNode->slotId, pTable->keyNum, pTable->keyCols, &pTable->valCols[i].srcSlot)) {
44,008!
226
        pTable->valCols[i].keyCol = true;
×
227
      } else {
228
        pTable->valCols[i].keyCol = false;
44,008✔
229
        pTable->valCols[i].srcSlot = pColNode->slotId;
44,008✔
230
        pTable->valColExist = true;
44,008✔
231
        colNum++;
44,008✔
232
      }
233
      pTable->valCols[i].dstSlot = pTarget->slotId;
44,008✔
234
      pTable->valCols[i].vardata = IS_VAR_DATA_TYPE(pColNode->node.resType.type);
44,008!
235
      if (pTable->valCols[i].vardata && !pTable->valCols[i].keyCol) {
44,008!
236
        if (NULL == pTable->valVarCols) {
×
237
          pTable->valVarCols = taosArrayInit(pTable->valNum, sizeof(int32_t));
×
238
          if (NULL == pTable->valVarCols) {
×
239
            return terrno;
×
240
          }
241
        }
242
        if (NULL == taosArrayPush(pTable->valVarCols, &i)) {
×
243
          return terrno;
×
244
        }
245
      }
246
      pTable->valCols[i].bytes = pColNode->node.resType.bytes;
44,008✔
247
      if (!pTable->valCols[i].keyCol && !pTable->valCols[i].vardata) {
44,008!
248
        pTable->valBufSize += pColNode->node.resType.bytes;
44,008✔
249
      }
250
      i++;
44,008✔
251
    }
252
  }
253

254
  pTable->valBitMapSize = BitmapLen(colNum);
22,004✔
255
  pTable->valBufSize += pTable->valBitMapSize;
22,004✔
256

257
  return TSDB_CODE_SUCCESS;
22,004✔
258
}
259

260
static int32_t hJoinInitPrimKeyInfo(SHJoinTableCtx* pTable, int32_t slotId) {
22,004✔
261
  pTable->primCol = taosMemoryMalloc(sizeof(SHJoinColMap));
22,004✔
262
  if (NULL == pTable->primCol) {
22,004!
263
    return terrno;
×
264
  }
265

266
  pTable->primCol->srcSlot = slotId;
22,004✔
267

268
  return TSDB_CODE_SUCCESS;
22,004✔
269
}
270

271

272
static int32_t hJoinInitPrimExprCtx(SNode* pNode, SHJoinPrimExprCtx* pCtx, SHJoinTableCtx* pTable) {
22,004✔
273
  if (NULL == pNode) {
22,004!
274
    pCtx->targetSlotId = pTable->primCol->srcSlot;
22,004✔
275
    return TSDB_CODE_SUCCESS;
22,004✔
276
  }
277
  
278
  if (QUERY_NODE_TARGET != nodeType(pNode)) {
×
279
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
280
  }  
281

282
  STargetNode* pTarget = (STargetNode*)pNode;
×
283
  if (QUERY_NODE_FUNCTION != nodeType(pTarget->pExpr)) {
×
284
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
285
  }
286

287
  SFunctionNode* pFunc = (SFunctionNode*)pTarget->pExpr;
×
288
  if (FUNCTION_TYPE_TIMETRUNCATE != pFunc->funcType) {
×
289
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
290
  }
291

292
  if (4 != pFunc->pParameterList->length && 5 != pFunc->pParameterList->length) {
×
293
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
294
  }
295

296
  SValueNode* pUnit = (SValueNode*)nodesListGetNode(pFunc->pParameterList, 1);
×
297
  SValueNode* pCurrTz = (5 == pFunc->pParameterList->length) ? (SValueNode*)nodesListGetNode(pFunc->pParameterList, 2) : NULL;
×
298
  SValueNode* pTimeZone = (5 == pFunc->pParameterList->length) ? (SValueNode*)nodesListGetNode(pFunc->pParameterList, 4) : (SValueNode*)nodesListGetNode(pFunc->pParameterList, 3);
×
299

300
  pCtx->truncateUnit = pUnit->typeData;
×
301
  if ((NULL == pCurrTz || 1 == pCurrTz->typeData) && pCtx->truncateUnit >= (86400 * TSDB_TICK_PER_SECOND(pFunc->node.resType.precision))) {
×
302
    pCtx->timezoneUnit = offsetFromTz(varDataVal(pTimeZone->datum.p), TSDB_TICK_PER_SECOND(pFunc->node.resType.precision));
×
303
  }
304

305
  pCtx->targetSlotId = pTarget->slotId;
×
306

307
  return TSDB_CODE_SUCCESS;
×
308
}
309

310

311
static int32_t hJoinInitTableInfo(SHJoinOperatorInfo* pJoin, SHashJoinPhysiNode* pJoinNode, SOperatorInfo** pDownstream, int32_t idx, SQueryStat* pStat) {
22,004✔
312
  SNodeList* pKeyList = NULL;
22,004✔
313
  SHJoinTableCtx* pTable = &pJoin->tbs[idx];
22,004✔
314
  pTable->downStream = pDownstream[idx];
22,004✔
315
  pTable->blkId = pDownstream[idx]->resultDataBlockId;
22,004✔
316
  if (0 == idx) {
22,004✔
317
    pKeyList = pJoinNode->pOnLeft;
11,002✔
318
    pTable->hasTimeRange = pJoinNode->timeRangeTarget & 0x1;
11,002✔
319
  } else {
320
    pKeyList = pJoinNode->pOnRight;
11,002✔
321
    pTable->hasTimeRange = pJoinNode->timeRangeTarget & 0x2;
11,002✔
322
  }
323

324
  HJ_ERR_RET(hJoinInitPrimKeyInfo(pTable, (0 == idx) ? pJoinNode->leftPrimSlotId : pJoinNode->rightPrimSlotId));
22,004!
325
  
326
  int32_t code = hJoinInitKeyColsInfo(pTable, pKeyList);
22,004✔
327
  if (code) {
22,004!
UNCOV
328
    return code;
×
329
  }
330
  code = hJoinInitValColsInfo(pTable, pJoinNode->pTargets);
22,004✔
331
  if (code) {
22,004!
332
    return code;
×
333
  }
334

335
  TAOS_MEMCPY(&pTable->inputStat, pStat, sizeof(*pStat));
22,004✔
336

337
  HJ_ERR_RET(hJoinInitPrimExprCtx(pTable->primExpr, &pTable->primCtx, pTable));
22,004!
338

339
  return TSDB_CODE_SUCCESS;
22,004✔
340
}
341

342
static void hJoinSetBuildAndProbeTable(SHJoinOperatorInfo* pInfo, SHashJoinPhysiNode* pJoinNode) {
11,002✔
343
  int32_t buildIdx = 0;
11,002✔
344
  int32_t probeIdx = 1;
11,002✔
345

346
  pInfo->joinType = pJoinNode->joinType;
11,002✔
347
  pInfo->subType = pJoinNode->subType;
11,002✔
348
  
349
  switch (pInfo->joinType) {
11,002!
350
    case JOIN_TYPE_INNER:
11,002✔
351
    case JOIN_TYPE_FULL:
352
      if (pInfo->tbs[0].inputStat.inputRowNum <= pInfo->tbs[1].inputStat.inputRowNum) {
11,002!
353
        buildIdx = 0;
11,002✔
354
        probeIdx = 1;
11,002✔
355
      } else {
356
        buildIdx = 1;
×
357
        probeIdx = 0;
×
358
      }
359
      break;
11,002✔
360
    case JOIN_TYPE_LEFT:
×
361
      buildIdx = 1;
×
362
      probeIdx = 0;
×
363
      break;
×
364
    case JOIN_TYPE_RIGHT:
×
365
      buildIdx = 0;
×
366
      probeIdx = 1;
×
367
      break;
×
368
    default:
×
369
      break;
×
370
  } 
371
  
372
  pInfo->pBuild = &pInfo->tbs[buildIdx];
11,002✔
373
  pInfo->pProbe = &pInfo->tbs[probeIdx];
11,002✔
374
  
375
  pInfo->pBuild->downStreamIdx = buildIdx;
11,002✔
376
  pInfo->pProbe->downStreamIdx = probeIdx;
11,002✔
377

378
  if (0 == buildIdx) {
11,002!
379
    pInfo->pBuild->primExpr = pJoinNode->leftPrimExpr;
11,002✔
380
    pInfo->pProbe->primExpr = pJoinNode->rightPrimExpr;
11,002✔
381
  } else {
382
    pInfo->pBuild->primExpr = pJoinNode->rightPrimExpr;
×
383
    pInfo->pProbe->primExpr = pJoinNode->leftPrimExpr;
×
384
  }
385
}
11,002✔
386

387
static int32_t hJoinBuildResColsMap(SHJoinOperatorInfo* pInfo, SHashJoinPhysiNode* pJoinNode) {
11,002✔
388
  pInfo->pResColNum = pJoinNode->pTargets->length;
11,002✔
389
  pInfo->pResColMap = taosMemoryCalloc(pJoinNode->pTargets->length, sizeof(int8_t));
11,002✔
390
  if (NULL == pInfo->pResColMap) {
11,002!
391
    return terrno;
×
392
  }
393
  
394
  SNode* pNode = NULL;
11,002✔
395
  int32_t i = 0;
11,002✔
396
  FOREACH(pNode, pJoinNode->pTargets) {
55,010!
397
    STargetNode* pTarget = (STargetNode*)pNode;
44,008✔
398
    SColumnNode* pCol = (SColumnNode*)pTarget->pExpr;
44,008✔
399
    if (pCol->dataBlockId == pInfo->pBuild->blkId) {
44,008✔
400
      pInfo->pResColMap[i] = 1;
22,004✔
401
    }
402
    
403
    i++;
44,008✔
404
  }
405

406
  return TSDB_CODE_SUCCESS;
11,002✔
407
}
408

409

410
static FORCE_INLINE int32_t hJoinAddPageToBufs(SArray* pRowBufs) {
411
  SBufPageInfo page;
412
  page.pageSize = HASH_JOIN_DEFAULT_PAGE_SIZE;
11,002✔
413
  page.offset = 0;
11,002✔
414
  page.data = taosMemoryMalloc(page.pageSize);
×
415
  if (NULL == page.data) {
11,002!
416
    return terrno;
×
417
  }
418

419
  if (NULL == taosArrayPush(pRowBufs, &page)) {
11,002!
420
    return terrno;
×
421
  }
422
  return TSDB_CODE_SUCCESS;
11,002✔
423
}
424

425
static int32_t hJoinInitBufPages(SHJoinOperatorInfo* pInfo) {
11,002✔
426
  pInfo->pRowBufs = taosArrayInit(32, sizeof(SBufPageInfo));
11,002✔
427
  if (NULL == pInfo->pRowBufs) {
11,002!
428
    return terrno;
×
429
  }
430

431
  return hJoinAddPageToBufs(pInfo->pRowBufs);
22,004✔
432
}
433

434
static void hJoinFreeTableInfo(SHJoinTableCtx* pTable) {
22,004✔
435
  taosMemoryFreeClear(pTable->keyCols);
22,004!
436
  taosMemoryFreeClear(pTable->keyBuf);
22,004✔
437
  taosMemoryFreeClear(pTable->valCols);
22,004!
438
  taosArrayDestroy(pTable->valVarCols);
22,004✔
439
  taosMemoryFree(pTable->primCol);
22,004✔
440
}
22,004✔
441

442
static void hJoinFreeBufPage(void* param) {
11,002✔
443
  SBufPageInfo* pInfo = (SBufPageInfo*)param;
11,002✔
444
  taosMemoryFree(pInfo->data);
11,002✔
445
}
11,002✔
446

447
static void hJoinDestroyKeyHash(SSHashObj** ppHash) {
22,004✔
448
  if (NULL == ppHash || NULL == (*ppHash)) {
22,004!
449
    return;
11,002✔
450
  }
451

452
  void*   pIte = NULL;
11,002✔
453
  int32_t iter = 0;
11,002✔
454
  while ((pIte = tSimpleHashIterate(*ppHash, pIte, &iter)) != NULL) {
54,432✔
455
    SGroupData* pGroup = pIte;
43,430✔
456
    SBufRowInfo* pRow = pGroup->rows;
43,430✔
457
    SBufRowInfo* pNext = NULL;
43,430✔
458
    while (pRow) {
87,849✔
459
      pNext = pRow->next;
44,419✔
460
      taosMemoryFree(pRow);
44,419✔
461
      pRow = pNext;
44,419✔
462
    }
463
  }
464

465
  tSimpleHashCleanup(*ppHash);
11,002✔
466
  *ppHash = NULL;
11,002✔
467
}
468

469
static FORCE_INLINE int32_t hJoinRetrieveColDataFromRowBufs(SArray* pRowBufs, SBufRowInfo* pRow, char** ppData) {
470
  *ppData = NULL;
34,507✔
471
  
472
  if ((uint16_t)-1 == pRow->pageId) {
34,507✔
473
    return TSDB_CODE_SUCCESS;
×
474
  }
475
  SBufPageInfo *pPage = taosArrayGet(pRowBufs, pRow->pageId);
34,507✔
476
  if (NULL == pPage) {
34,507!
477
    qError("fail to get %d page, total:%d", pRow->pageId, (int32_t)taosArrayGetSize(pRowBufs));
×
478
    QRY_ERR_RET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
479
  }
480
  
481
  *ppData = pPage->data + pRow->offset;
34,507✔
482

483
  return TSDB_CODE_SUCCESS;
34,507✔
484
}
485

486
static int32_t hJoinCopyResRowsToBlock(SHJoinOperatorInfo* pJoin, int32_t rowNum, SBufRowInfo* pStart, SSDataBlock* pRes) {
1,668✔
487
  SHJoinTableCtx* pBuild = pJoin->pBuild;
1,668✔
488
  SHJoinTableCtx* pProbe = pJoin->pProbe;
1,668✔
489
  int32_t buildIdx = 0, buildValIdx = 0;
1,668✔
490
  int32_t probeIdx = 0;
1,668✔
491
  SBufRowInfo* pRow = pStart;
1,668✔
492
  int32_t code = 0;
1,668✔
493
  char* pData = NULL;
1,668✔
494

495
  for (int32_t r = 0; r < rowNum; ++r) {
36,175✔
496
    HJ_ERR_RET(hJoinRetrieveColDataFromRowBufs(pJoin->pRowBufs, pRow, &pData));
69,014!
497
    
498
    char* pValData = pData + pBuild->valBitMapSize;
34,507✔
499
    char* pKeyData = pProbe->keyData;
34,507✔
500
    buildIdx = buildValIdx = probeIdx = 0;
34,507✔
501
    for (int32_t i = 0; i < pJoin->pResColNum; ++i) {
172,535✔
502
      if (pJoin->pResColMap[i]) {
138,028✔
503
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pBuild->valCols[buildIdx].dstSlot);
69,014✔
504
        if (pBuild->valCols[buildIdx].keyCol) {
69,014!
505
          code = colDataSetVal(pDst, pRes->info.rows + r, pKeyData, false);
×
506
          if (code) {
×
507
            return code;
×
508
          }
509
          pKeyData += pBuild->valCols[buildIdx].vardata ? varDataTLen(pKeyData) : pBuild->valCols[buildIdx].bytes;
×
510
        } else {
511
          if (colDataIsNull_f(pData, buildValIdx)) {
69,014!
512
            code = colDataSetVal(pDst, pRes->info.rows + r, NULL, true);
×
513
            if (code) {
×
514
              return code;
×
515
            }
516
          } else {
517
            code = colDataSetVal(pDst, pRes->info.rows + r, pValData, false);
69,014✔
518
            if (code) {
69,014!
519
              return code;
×
520
            }
521
            pValData += pBuild->valCols[buildIdx].vardata ? varDataTLen(pValData) : pBuild->valCols[buildIdx].bytes;
69,014!
522
          }
523
          buildValIdx++;
69,014✔
524
        }
525
        buildIdx++;
69,014✔
526
      } else if (0 == r) {
69,014✔
527
        SColumnInfoData* pSrc = taosArrayGet(pJoin->ctx.pProbeData->pDataBlock, pProbe->valCols[probeIdx].srcSlot);
3,336✔
528
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pProbe->valCols[probeIdx].dstSlot);
3,336✔
529
    
530
        code = colDataCopyNItems(pDst, pRes->info.rows, colDataGetData(pSrc, pJoin->ctx.probeStartIdx), rowNum, colDataIsNull_s(pSrc, pJoin->ctx.probeStartIdx));
6,672!
531
        if (code) {
3,336!
532
          return code;
×
533
        }
534
        probeIdx++;
3,336✔
535
      }
536
    }
537
    pRow = pRow->next;
34,507✔
538
  }
539

540
  return TSDB_CODE_SUCCESS;
1,668✔
541
}
542

543
int32_t hJoinCopyNMatchRowsToBlock(SHJoinOperatorInfo* pJoin, SSDataBlock* pRes, int32_t startIdx, int32_t rows) {
×
544
  SHJoinTableCtx* pBuild = pJoin->pBuild;
×
545
  SHJoinTableCtx* pProbe = pJoin->pProbe;
×
546
  int32_t buildIdx = 0;
×
547
  int32_t probeIdx = 0;
×
548
  int32_t code = 0;
×
549

550
  for (int32_t i = 0; i < pJoin->pResColNum; ++i) {
×
551
    if (pJoin->pResColMap[i]) {
×
552
      SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pBuild->valCols[buildIdx].dstSlot);
×
553
      colDataSetNItemsNull(pDst, pRes->info.rows, rows);
×
554

555
      buildIdx++;
×
556
    } else {
557
      SColumnInfoData* pSrc = taosArrayGet(pJoin->ctx.pProbeData->pDataBlock, pProbe->valCols[probeIdx].srcSlot);
×
558
      SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pProbe->valCols[probeIdx].dstSlot);
×
559

560
      QRY_ERR_RET(colDataAssignNRows(pDst, pRes->info.rows, pSrc, startIdx, rows));
×
561

562
      probeIdx++;
×
563
    }
564
  }
565

566
  pRes->info.rows += rows;
×
567

568
  return TSDB_CODE_SUCCESS;
×
569
}
570

571

572

573
void hJoinAppendResToBlock(struct SOperatorInfo* pOperator, SSDataBlock* pRes, bool* allFetched) {
1,668✔
574
  SHJoinOperatorInfo* pJoin = pOperator->info;
1,668✔
575
  SHJoinCtx* pCtx = &pJoin->ctx;
1,668✔
576
  SBufRowInfo* pStart = pCtx->pBuildRow;
1,668✔
577
  int32_t rowNum = 0;
1,668✔
578
  int32_t resNum = pRes->info.rows;
1,668✔
579
  
580
  while (pCtx->pBuildRow && (resNum < pRes->info.capacity)) {
36,175!
581
    rowNum++;
34,507✔
582
    resNum++;
34,507✔
583
    pCtx->pBuildRow = pCtx->pBuildRow->next;
34,507✔
584
  }
585

586
  pJoin->execInfo.resRows += rowNum;
1,668✔
587

588
  int32_t code = hJoinCopyResRowsToBlock(pJoin, rowNum, pStart, pRes);
1,668✔
589
  if (code) {
1,668!
590
    pOperator->pTaskInfo->code = code;
×
591
    T_LONG_JMP(pOperator->pTaskInfo->env, code);
×
592
  }
593

594
  pRes->info.rows = resNum;
1,668✔
595
  *allFetched = pCtx->pBuildRow ? false : true;
1,668✔
596
}
1,668✔
597

598

599
bool hJoinCopyKeyColsDataToBuf(SHJoinTableCtx* pTable, int32_t rowIdx, size_t *pBufLen) {
67,596✔
600
  char *pData = NULL;
67,596✔
601
  size_t bufLen = 0;
67,596✔
602
  
603
  if (1 == pTable->keyNum) {
67,596✔
604
    if (colDataIsNull_s(pTable->keyCols[0].colData, rowIdx)) {
134,984✔
605
      return true;
25✔
606
    }
607
    if (pTable->keyCols[0].vardata) {
67,467✔
608
      pData = pTable->keyCols[0].data + pTable->keyCols[0].offset[rowIdx];
16,484✔
609
      bufLen = varDataTLen(pData);
16,484✔
610
    } else {
611
      pData = pTable->keyCols[0].data + pTable->keyCols[0].bytes * rowIdx;
50,983✔
612
      bufLen = pTable->keyCols[0].bytes;
50,983✔
613
    }
614
    pTable->keyData = pData;
67,467✔
615
  } else {
616
    for (int32_t i = 0; i < pTable->keyNum; ++i) {
312✔
617
      if (colDataIsNull_s(pTable->keyCols[i].colData, rowIdx)) {
416!
618
        return true;
×
619
      }
620
      if (pTable->keyCols[i].vardata) {
208✔
621
        pData = pTable->keyCols[i].data + pTable->keyCols[i].offset[rowIdx];
64✔
622
        TAOS_MEMCPY(pTable->keyBuf + bufLen, pData, varDataTLen(pData));
64✔
623
        bufLen += varDataTLen(pData);
64✔
624
      } else {
625
        pData = pTable->keyCols[i].data + pTable->keyCols[i].bytes * rowIdx;
144✔
626
        TAOS_MEMCPY(pTable->keyBuf + bufLen, pData, pTable->keyCols[i].bytes);
144✔
627
        bufLen += pTable->keyCols[i].bytes;
144✔
628
      }
629
    }
630
    pTable->keyData = pTable->keyBuf;
104✔
631
  }
632

633
  if (pBufLen) {
67,571!
634
    *pBufLen = bufLen;
67,571✔
635
  }
636

637
  return false;
67,571✔
638
}
639

640
static int32_t hJoinSetKeyColsData(SSDataBlock* pBlock, SHJoinTableCtx* pTable) {
43,372✔
641
  for (int32_t i = 0; i < pTable->keyNum; ++i) {
86,792✔
642
    SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, pTable->keyCols[i].srcSlot);
43,420✔
643
    if (NULL == pCol) {
43,420!
644
      qError("fail to get %d col, total:%d", pTable->keyCols[i].srcSlot, (int32_t)taosArrayGetSize(pBlock->pDataBlock));
×
645
      QRY_ERR_RET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
646
    }
647
    if (pTable->keyCols[i].vardata != IS_VAR_DATA_TYPE(pCol->info.type))  {
43,420!
648
      qError("column type mismatch, idx:%d, slotId:%d, type:%d, vardata:%d", i, pTable->keyCols[i].srcSlot, pCol->info.type, pTable->keyCols[i].vardata);
×
649
      return TSDB_CODE_INVALID_PARA;
×
650
    }
651
    if (pTable->keyCols[i].bytes != pCol->info.bytes)  {
43,420!
652
      qError("column bytes mismatch, idx:%d, slotId:%d, bytes:%d, %d", i, pTable->keyCols[i].srcSlot, pCol->info.bytes, pTable->keyCols[i].bytes);
×
653
      return TSDB_CODE_INVALID_PARA;
×
654
    }
655
    pTable->keyCols[i].data = pCol->pData;
43,420✔
656
    if (pTable->keyCols[i].vardata) {
43,420✔
657
      pTable->keyCols[i].offset = pCol->varmeta.offset;
11,019✔
658
    }
659
    pTable->keyCols[i].colData = pCol;
43,420✔
660
  }
661

662
  return TSDB_CODE_SUCCESS;
43,372✔
663
}
664

665
static int32_t hJoinSetValColsData(SSDataBlock* pBlock, SHJoinTableCtx* pTable) {
66,084✔
666
  if (!pTable->valColExist) {
66,084!
667
    return TSDB_CODE_SUCCESS;
×
668
  }
669
  for (int32_t i = 0; i < pTable->valNum; ++i) {
198,252✔
670
    if (pTable->valCols[i].keyCol) {
132,168!
671
      continue;
×
672
    }
673
    SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, pTable->valCols[i].srcSlot);
132,168✔
674
    if (NULL == pCol) {
132,168!
675
      qError("fail to get %d col, total:%d", pTable->valCols[i].srcSlot, (int32_t)taosArrayGetSize(pBlock->pDataBlock));
×
676
      QRY_ERR_RET(TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
677
    }
678
    if (pTable->valCols[i].vardata != IS_VAR_DATA_TYPE(pCol->info.type))  {
132,168!
679
      qError("column type mismatch, idx:%d, slotId:%d, type:%d, vardata:%d", i, pTable->valCols[i].srcSlot, pCol->info.type, pTable->valCols[i].vardata);
×
680
      return TSDB_CODE_INVALID_PARA;
×
681
    }
682
    if (pTable->valCols[i].bytes != pCol->info.bytes)  {
132,168!
683
      qError("column bytes mismatch, idx:%d, slotId:%d, bytes:%d, %d", i, pTable->valCols[i].srcSlot, pCol->info.bytes, pTable->valCols[i].bytes);
×
684
      return TSDB_CODE_INVALID_PARA;
×
685
    }
686
    if (!pTable->valCols[i].vardata) {
132,168!
687
      pTable->valCols[i].bitMap = pCol->nullbitmap;
132,168✔
688
    }
689
    pTable->valCols[i].data = pCol->pData;
132,168✔
690
    if (pTable->valCols[i].vardata) {
132,168!
691
      pTable->valCols[i].offset = pCol->varmeta.offset;
×
692
    }
693
  }
694

695
  return TSDB_CODE_SUCCESS;
66,084✔
696
}
697

698

699

700
static FORCE_INLINE void hJoinCopyValColsDataToBuf(SHJoinTableCtx* pTable, int32_t rowIdx) {
701
  if (!pTable->valColExist) {
44,419!
702
    return;
×
703
  }
704

705
  char *pData = NULL;
44,419✔
706
  size_t bufLen = pTable->valBitMapSize;
44,419✔
707
  TAOS_MEMSET(pTable->valData, 0, pTable->valBitMapSize);
44,419✔
708
  for (int32_t i = 0, m = 0; i < pTable->valNum; ++i) {
133,257✔
709
    if (pTable->valCols[i].keyCol) {
88,838!
710
      continue;
×
711
    }
712
    if (pTable->valCols[i].vardata) {
88,838!
713
      if (-1 == pTable->valCols[i].offset[rowIdx]) {
×
714
        colDataSetNull_f(pTable->valData, m);
×
715
      } else {
716
        pData = pTable->valCols[i].data + pTable->valCols[i].offset[rowIdx];
×
717
        TAOS_MEMCPY(pTable->valData + bufLen, pData, varDataTLen(pData));
×
718
        bufLen += varDataTLen(pData);
×
719
      }
720
    } else {
721
      if (colDataIsNull_f(pTable->valCols[i].bitMap, rowIdx)) {
88,838!
722
        colDataSetNull_f(pTable->valData, m);
×
723
      } else {
724
        pData = pTable->valCols[i].data + pTable->valCols[i].bytes * rowIdx;
88,838✔
725
        TAOS_MEMCPY(pTable->valData + bufLen, pData, pTable->valCols[i].bytes);
88,838✔
726
        bufLen += pTable->valCols[i].bytes;
88,838✔
727
      }
728
    }
729
    m++;
88,838✔
730
  }
731
}
732

733

734
static FORCE_INLINE int32_t hJoinGetValBufFromPages(SArray* pPages, int32_t bufSize, char** pBuf, SBufRowInfo* pRow) {
735
  if (0 == bufSize) {
44,419✔
736
    pRow->pageId = -1;
×
737
    return TSDB_CODE_SUCCESS;
×
738
  }
739

740
  if (bufSize > HASH_JOIN_DEFAULT_PAGE_SIZE) {
44,419!
741
    qError("invalid join value buf size:%d", bufSize);
×
742
    return TSDB_CODE_INVALID_PARA;
×
743
  }
744
  
745
  do {
×
746
    SBufPageInfo* page = taosArrayGetLast(pPages);
44,419✔
747
    if ((page->pageSize - page->offset) >= bufSize) {
44,419!
748
      *pBuf = page->data + page->offset;
44,419✔
749
      pRow->pageId = taosArrayGetSize(pPages) - 1;
44,419✔
750
      pRow->offset = page->offset;
44,419✔
751
      page->offset += bufSize;
44,419✔
752
      return TSDB_CODE_SUCCESS;
44,419✔
753
    }
754

755
    int32_t code = hJoinAddPageToBufs(pPages);
×
756
    if (code) {
×
757
      return code;
×
758
    }
759
  } while (true);
760
}
761

762
static FORCE_INLINE int32_t hJoinGetValBufSize(SHJoinTableCtx* pTable, int32_t rowIdx) {
763
  if (NULL == pTable->valVarCols) {
44,419✔
764
    return pTable->valBufSize;
44,419✔
765
  }
766

767
  int32_t* varColIdx = NULL;
×
768
  int32_t bufLen = pTable->valBufSize;
×
769
  int32_t varColNum = taosArrayGetSize(pTable->valVarCols);
×
770
  for (int32_t i = 0; i < varColNum; ++i) {
×
771
    varColIdx = taosArrayGet(pTable->valVarCols, i);
×
772
    char* pData = pTable->valCols[*varColIdx].data + pTable->valCols[*varColIdx].offset[rowIdx];
×
773
    bufLen += varDataTLen(pData);
×
774
  }
775

776
  return bufLen;
×
777
}
778

779

780
static int32_t hJoinAddRowToHashImpl(SHJoinOperatorInfo* pJoin, SGroupData* pGroup, SHJoinTableCtx* pTable, size_t keyLen, int32_t rowIdx) {
44,419✔
781
  SGroupData group = {0};
44,419✔
782
  SBufRowInfo* pRow = NULL;
44,419✔
783

784
  if (NULL == pGroup) {
44,419✔
785
    group.rows = taosMemoryMalloc(sizeof(SBufRowInfo));
43,430✔
786
    if (NULL == group.rows) {
43,430!
787
      return terrno;
×
788
    }
789
    pRow = group.rows;
43,430✔
790
  } else {
791
    pRow = taosMemoryMalloc(sizeof(SBufRowInfo));
989✔
792
    if (NULL == pRow) {
989!
793
      return terrno;
×
794
    }
795
  }
796

797
  int32_t code = hJoinGetValBufFromPages(pJoin->pRowBufs, hJoinGetValBufSize(pTable, rowIdx), &pTable->valData, pRow);
88,838!
798
  if (code) {
44,419!
799
    taosMemoryFree(pRow);
×
800
    return code;
×
801
  }
802

803
  if (NULL == pGroup) {
44,419✔
804
    pRow->next = NULL;
43,430✔
805
    if (tSimpleHashPut(pJoin->pKeyHash, pTable->keyData, keyLen, &group, sizeof(group))) {
43,430!
806
      taosMemoryFree(pRow);
×
807
      return TSDB_CODE_OUT_OF_MEMORY;
×
808
    }
809
  } else {
810
    pRow->next = pGroup->rows;
989✔
811
    pGroup->rows = pRow;
989✔
812
  }
813

814
  return TSDB_CODE_SUCCESS;
44,419✔
815
}
816

817
static int32_t hJoinAddRowToHash(SHJoinOperatorInfo* pJoin, SSDataBlock* pBlock, size_t keyLen, int32_t rowIdx) {
44,419✔
818
  SHJoinTableCtx* pBuild = pJoin->pBuild;
44,419✔
819
  int32_t code = hJoinSetValColsData(pBlock, pBuild);
44,419✔
820
  if (code) {
44,419!
821
    return code;
×
822
  }
823

824
  SGroupData* pGroup = tSimpleHashGet(pJoin->pKeyHash, pBuild->keyData, keyLen);
44,419✔
825
  code = hJoinAddRowToHashImpl(pJoin, pGroup, pBuild, keyLen, rowIdx);
44,419✔
826
  if (code) {
44,419!
827
    return code;
×
828
  }
829
  
830
  hJoinCopyValColsDataToBuf(pBuild, rowIdx);
831

832
  return TSDB_CODE_SUCCESS;
44,419✔
833
}
834

835
static bool hJoinFilterTimeRange(SSDataBlock* pBlock, STimeWindow* pRange, int32_t primSlot, int32_t* startIdx, int32_t* endIdx) {
×
836
  SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, primSlot);
×
837
  if (NULL == pCol) {
×
838
    qError("hash join can't get prim col, slot:%d, slotNum:%d", primSlot, (int32_t)taosArrayGetSize(pBlock->pDataBlock));
×
839
    return false;
×
840
  }
841

842
  TSKEY skey = *(TSKEY*)colDataGetData(pCol, 0);
×
843
  TSKEY ekey = *(TSKEY*)colDataGetData(pCol, (pBlock->info.rows - 1));
×
844

845
  if (ekey < pRange->skey || skey > pRange->ekey) {
×
846
    return false;
×
847
  }
848

849
  if (skey >= pRange->skey && ekey <= pRange->ekey) {
×
850
    *startIdx = 0;
×
851
    *endIdx = pBlock->info.rows - 1;
×
852
    return true;
×
853
  }
854

855
  if (skey < pRange->skey && ekey > pRange->ekey) {
×
856
    TSKEY *pStart = (TSKEY*)taosbsearch(&pRange->skey, pCol->pData, pBlock->info.rows, sizeof(TSKEY), compareInt64Val, TD_GE);
×
857
    TSKEY *pEnd = (TSKEY*)taosbsearch(&pRange->ekey, pCol->pData, pBlock->info.rows, sizeof(TSKEY), compareInt64Val, TD_LE);
×
858
    *startIdx = ((uint64_t)pStart - (uint64_t)pCol->pData) / sizeof(int64_t);
×
859
    *endIdx = ((uint64_t)pEnd - (uint64_t)pCol->pData) / sizeof(int64_t);
×
860
    return true;
×
861
  }
862

863
  if (skey >= pRange->skey) {
×
864
    TSKEY *pEnd = (TSKEY*)taosbsearch(&pRange->ekey, pCol->pData, pBlock->info.rows, sizeof(TSKEY), compareInt64Val, TD_LE);
×
865
    *startIdx = 0;
×
866
    *endIdx = ((uint64_t)pEnd - (uint64_t)pCol->pData) / sizeof(int64_t);
×
867
    return true;
×
868
  }
869

870
  TSKEY *pStart = (TSKEY*)taosbsearch(&pRange->skey, pCol->pData, pBlock->info.rows, sizeof(TSKEY), compareInt64Val, TD_GE);
×
871
  *startIdx = ((uint64_t)pStart - (uint64_t)pCol->pData) / sizeof(int64_t);
×
872
  *endIdx = pBlock->info.rows - 1;
×
873
  
874
  return true;
×
875
}
876

877
static int32_t hJoinAddBlockRowsToHash(SSDataBlock* pBlock, SHJoinOperatorInfo* pJoin) {
21,707✔
878
  SHJoinTableCtx* pBuild = pJoin->pBuild;
21,707✔
879
  int32_t startIdx = 0, endIdx = pBlock->info.rows - 1;
21,707✔
880
  if (pBuild->hasTimeRange && !hJoinFilterTimeRange(pBlock, &pJoin->tblTimeRange, pBuild->primCol->srcSlot, &startIdx, &endIdx)) {
21,707!
881
    return TSDB_CODE_SUCCESS;
×
882
  }
883

884
  HJ_ERR_RET(hJoinLaunchPrimExpr(pBlock, pBuild, startIdx, endIdx));
21,707!
885

886
  int32_t code = hJoinSetKeyColsData(pBlock, pBuild);
21,707✔
887
  if (code) {
21,707!
888
    return code;
×
889
  }
890

891
  size_t bufLen = 0;
21,707✔
892
  for (int32_t i = startIdx; i <= endIdx; ++i) {
66,141✔
893
    if (hJoinCopyKeyColsDataToBuf(pBuild, i, &bufLen)) {
44,434✔
894
      continue;
15✔
895
    }
896
    code = hJoinAddRowToHash(pJoin, pBlock, bufLen, i);
44,419✔
897
    if (code) {
44,419!
898
      return code;
×
899
    }
900
  }
901

902
  return code;
21,707✔
903
}
904

905
static int32_t hJoinBuildHash(struct SOperatorInfo* pOperator, bool* queryDone) {
11,002✔
906
  SHJoinOperatorInfo* pJoin = pOperator->info;
11,002✔
907
  SSDataBlock* pBlock = NULL;
11,002✔
908
  int32_t code = TSDB_CODE_SUCCESS;
11,002✔
909
  
910
  while (true) {
911
    pBlock = getNextBlockFromDownstream(pOperator, pJoin->pBuild->downStreamIdx);
32,709✔
912
    if (NULL == pBlock) {
32,709✔
913
      break;
11,002✔
914
    }
915

916
    pJoin->execInfo.buildBlkNum++;
21,707✔
917
    pJoin->execInfo.buildBlkRows += pBlock->info.rows;
21,707✔
918

919
    code = hJoinAddBlockRowsToHash(pBlock, pJoin);
21,707✔
920
    if (code) {
21,707!
921
      return code;
×
922
    }
923
  }
924

925
  if (IS_INNER_NONE_JOIN(pJoin->joinType, pJoin->subType) && tSimpleHashGetSize(pJoin->pKeyHash) <= 0) {
11,002!
926
    hJoinSetDone(pOperator);
128✔
927
    *queryDone = true;
128✔
928
  }
929
  
930
  //qTrace("build table rows:%" PRId64, hJoinGetRowsNumOfKeyHash(pJoin->pKeyHash));
931

932
  return TSDB_CODE_SUCCESS;
11,002✔
933
}
934

935
static int32_t hJoinPrepareStart(struct SOperatorInfo* pOperator, SSDataBlock* pBlock) {
21,665✔
936
  SHJoinOperatorInfo* pJoin = pOperator->info;
21,665✔
937
  SHJoinTableCtx* pProbe = pJoin->pProbe;
21,665✔
938
  int32_t startIdx = 0, endIdx = pBlock->info.rows - 1;
21,665✔
939
  if (pProbe->hasTimeRange && !hJoinFilterTimeRange(pBlock, &pJoin->tblTimeRange, pProbe->primCol->srcSlot, &startIdx, &endIdx)) {
21,665!
940
    if (!IS_INNER_NONE_JOIN(pJoin->joinType, pJoin->subType)) {
×
941
      pJoin->ctx.probeEndIdx = -1;
×
942
      pJoin->ctx.probePostIdx = 0;
×
943
      pJoin->ctx.pProbeData = pBlock;
×
944
      pJoin->ctx.rowRemains = true;
×
945
      pJoin->ctx.probePhase = E_JOIN_PHASE_POST;
×
946
      
947
      HJ_ERR_RET((*pJoin->joinFp)(pOperator));
×
948
    }
949
    
950
    return TSDB_CODE_SUCCESS;
×
951
  }
952

953
  HJ_ERR_RET(hJoinLaunchPrimExpr(pBlock, pProbe, startIdx, endIdx));
21,665!
954

955
  int32_t code = hJoinSetKeyColsData(pBlock, pProbe);
21,665✔
956
  if (code) {
21,665!
957
    return code;
×
958
  }
959
  code = hJoinSetValColsData(pBlock, pProbe);
21,665✔
960
  if (code) {
21,665!
961
    return code;
×
962
  }
963

964
  pJoin->ctx.probeStartIdx = startIdx;
21,665✔
965
  pJoin->ctx.probeEndIdx = endIdx;
21,665✔
966
  pJoin->ctx.pBuildRow = NULL;
21,665✔
967
  pJoin->ctx.pProbeData = pBlock;
21,665✔
968
  pJoin->ctx.rowRemains = true;
21,665✔
969
  pJoin->ctx.probePreIdx = 0;
21,665✔
970
  pJoin->ctx.probePostIdx = endIdx + 1;
21,665✔
971

972
  if (!IS_INNER_NONE_JOIN(pJoin->joinType, pJoin->subType) && startIdx > 0) {
21,665!
973
    pJoin->ctx.probePhase = E_JOIN_PHASE_PRE;
×
974
  } else {
975
    pJoin->ctx.probePhase = E_JOIN_PHASE_CUR;
21,665✔
976
  }
977

978
  HJ_ERR_RET((*pJoin->joinFp)(pOperator));
21,665!
979

980
  return TSDB_CODE_SUCCESS;
21,665✔
981
}
982

983
void hJoinSetDone(struct SOperatorInfo* pOperator) {
11,002✔
984
  setOperatorCompleted(pOperator);
11,002✔
985

986
  SHJoinOperatorInfo* pInfo = pOperator->info;
11,002✔
987
  hJoinDestroyKeyHash(&pInfo->pKeyHash);
11,002✔
988

989
  qDebug("hash Join done");  
11,002✔
990
}
11,002✔
991

992
static int32_t hJoinMainProcess(struct SOperatorInfo* pOperator, SSDataBlock** pResBlock) {
11,314✔
993
  SHJoinOperatorInfo* pJoin = pOperator->info;
11,314✔
994
  SExecTaskInfo*      pTaskInfo = pOperator->pTaskInfo;
11,314✔
995
  int32_t             code = TSDB_CODE_SUCCESS;
11,314✔
996
  int32_t             lino = 0;
11,314✔
997
  SSDataBlock*        pRes = pJoin->finBlk;
11,314✔
998
  int64_t             st = 0;
11,314✔
999

1000
  QRY_PARAM_CHECK(pResBlock);
11,314!
1001
  if (pOperator->cost.openCost == 0) {
11,314✔
1002
    st = taosGetTimestampUs();
11,002✔
1003
  }
1004

1005
  if (pOperator->status == OP_EXEC_DONE) {
11,314✔
1006
    pRes->info.rows = 0;
312✔
1007
    goto _end;
312✔
1008
  }
1009

1010
  if (!pJoin->keyHashBuilt) {
11,002!
1011
    pJoin->keyHashBuilt = true;
11,002✔
1012

1013
    bool queryDone = false;
11,002✔
1014
    code = hJoinBuildHash(pOperator, &queryDone);
11,002✔
1015
    QUERY_CHECK_CODE(code, lino, _end);
11,130!
1016

1017
    if (queryDone) {
11,002✔
1018
      goto _end;
128✔
1019
    }
1020
  }
1021

1022
  blockDataCleanup(pRes);
10,874✔
1023

1024
  if (pJoin->ctx.rowRemains) {
10,874!
1025
    code = (*pJoin->joinFp)(pOperator);
×
1026
    QUERY_CHECK_CODE(code, lino, _end);
×
1027

1028
    if (pRes->info.rows > 0 && pJoin->pFinFilter != NULL) {
×
1029
      code = doFilter(pRes, pJoin->pFinFilter, NULL);
×
1030
      QUERY_CHECK_CODE(code, lino, _end);
×
1031
    }
1032

1033
    if (pRes->info.rows > 0) {
×
1034
      *pResBlock = pRes;
×
1035
      return code;
×
1036
    }
1037
  }
1038

1039
  while (true) {
21,665✔
1040
    SSDataBlock* pBlock = getNextBlockFromDownstream(pOperator, pJoin->pProbe->downStreamIdx);
32,539✔
1041
    if (NULL == pBlock) {
32,539✔
1042
      hJoinSetDone(pOperator);
10,874✔
1043
      break;
10,874✔
1044
    }
1045

1046
    pJoin->execInfo.probeBlkNum++;
21,665✔
1047
    pJoin->execInfo.probeBlkRows += pBlock->info.rows;
21,665✔
1048

1049
    code = hJoinPrepareStart(pOperator, pBlock);
21,665✔
1050
    QUERY_CHECK_CODE(code, lino, _end);
21,665!
1051

1052
    if (!hJoinBlkReachThreshold(pJoin, pRes->info.rows)) {
21,665!
1053
      continue;
21,665✔
1054
    }
1055

1056
    if (pRes->info.rows > 0 && pJoin->pFinFilter != NULL) {
×
1057
      code = doFilter(pRes, pJoin->pFinFilter, NULL);
×
1058
      QUERY_CHECK_CODE(code, lino, _end);
×
1059
    }
1060

1061
    if (pRes->info.rows > 0) {
×
1062
      break;
×
1063
    }
1064
  }
1065

1066
_end:
11,314✔
1067
  if (pOperator->cost.openCost == 0) {
11,314✔
1068
    pOperator->cost.openCost = (taosGetTimestampUs() - st) / 1000.0;
11,002✔
1069
  }
1070
  if (code != TSDB_CODE_SUCCESS) {
11,314!
1071
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1072
    pTaskInfo->code = code;
×
1073
    T_LONG_JMP(pTaskInfo->env, code);
×
1074
  }
1075
  if (pRes->info.rows > 0) {
11,314✔
1076
    *pResBlock = pRes;
312✔
1077
  }
1078

1079
  return code;
11,314✔
1080
}
1081

1082
static void destroyHashJoinOperator(void* param) {
11,002✔
1083
  SHJoinOperatorInfo* pJoinOperator = (SHJoinOperatorInfo*)param;
11,002✔
1084
  qDebug("hashJoin exec info, buildBlk:%" PRId64 ", buildRows:%" PRId64 ", probeBlk:%" PRId64 ", probeRows:%" PRId64 ", resRows:%" PRId64, 
11,002✔
1085
         pJoinOperator->execInfo.buildBlkNum, pJoinOperator->execInfo.buildBlkRows, pJoinOperator->execInfo.probeBlkNum, 
1086
         pJoinOperator->execInfo.probeBlkRows, pJoinOperator->execInfo.resRows);
1087

1088
  hJoinDestroyKeyHash(&pJoinOperator->pKeyHash);
11,002✔
1089

1090
  hJoinFreeTableInfo(&pJoinOperator->tbs[0]);
11,002✔
1091
  hJoinFreeTableInfo(&pJoinOperator->tbs[1]);
11,002✔
1092
  blockDataDestroy(pJoinOperator->finBlk);
11,002✔
1093
  pJoinOperator->finBlk = NULL;
11,002✔
1094
  taosMemoryFreeClear(pJoinOperator->pResColMap);
11,002!
1095
  taosArrayDestroyEx(pJoinOperator->pRowBufs, hJoinFreeBufPage);
11,002✔
1096

1097
  taosMemoryFreeClear(param);
11,002!
1098
}
11,002✔
1099

1100
int32_t hJoinHandleConds(SHJoinOperatorInfo* pJoin, SHashJoinPhysiNode* pJoinNode) {
11,002✔
1101
  switch (pJoin->joinType) {
11,002!
1102
    case JOIN_TYPE_INNER: {
11,002✔
1103
      SNode* pCond = NULL;
11,002✔
1104
      if (pJoinNode->pFullOnCond != NULL) {
11,002!
1105
        if (pJoinNode->node.pConditions != NULL) {
×
1106
          HJ_ERR_RET(mergeJoinConds(&pJoinNode->pFullOnCond, &pJoinNode->node.pConditions));
×
1107
        }
1108
        pCond = pJoinNode->pFullOnCond;
×
1109
      } else if (pJoinNode->node.pConditions != NULL) {
11,002!
1110
        pCond = pJoinNode->node.pConditions;
×
1111
      }
1112
      
1113
      HJ_ERR_RET(filterInitFromNode(pCond, &pJoin->pFinFilter, 0));
11,002!
1114
      break;
11,002✔
1115
    }
1116
    case JOIN_TYPE_LEFT:
×
1117
    case JOIN_TYPE_RIGHT:
1118
    case JOIN_TYPE_FULL:
1119
      if (pJoinNode->pFullOnCond != NULL) {
×
1120
        HJ_ERR_RET(filterInitFromNode(pJoinNode->pFullOnCond, &pJoin->pPreFilter, 0));
×
1121
      }
1122
      if (pJoinNode->node.pConditions != NULL) {
×
1123
        HJ_ERR_RET(filterInitFromNode(pJoinNode->node.pConditions, &pJoin->pFinFilter, 0));
×
1124
      }
1125
      break;
×
1126
    default:
×
1127
      break;
×
1128
  }
1129

1130
  return TSDB_CODE_SUCCESS;
11,002✔
1131
}
1132

1133
static uint32_t hJoinGetFinBlkCapacity(SHJoinOperatorInfo* pJoin, SHashJoinPhysiNode* pJoinNode) {
11,002✔
1134
  uint32_t maxRows = TMAX(HJOIN_DEFAULT_BLK_ROWS_NUM, HJOIN_BLK_SIZE_LIMIT/pJoinNode->node.pOutputDataBlockDesc->totalRowSize);
11,002✔
1135
  if (INT64_MAX != pJoin->ctx.limit && NULL == pJoin->pFinFilter) {
11,002!
1136
    uint32_t limitMaxRows = pJoin->ctx.limit / HJOIN_BLK_THRESHOLD_RATIO + 1;
×
1137
    return (maxRows > limitMaxRows) ? limitMaxRows : maxRows;
×
1138
  }
1139

1140
  return maxRows;
11,002✔
1141
}
1142

1143

1144
int32_t hJoinInitResBlocks(SHJoinOperatorInfo* pJoin, SHashJoinPhysiNode* pJoinNode) {
11,002✔
1145
  pJoin->finBlk = createDataBlockFromDescNode(pJoinNode->node.pOutputDataBlockDesc);
11,002✔
1146
  if (NULL == pJoin->finBlk) {
11,002!
1147
    QRY_ERR_RET(terrno);
×
1148
  }
1149

1150
  int32_t code = blockDataEnsureCapacity(pJoin->finBlk, hJoinGetFinBlkCapacity(pJoin, pJoinNode));
11,002✔
1151
  if (TSDB_CODE_SUCCESS != code) {
11,002!
1152
    QRY_ERR_RET(code);
×
1153
  }
1154
  
1155
  if (NULL != pJoin->pPreFilter) {
11,002!
1156
    pJoin->midBlk = NULL;
×
1157
    code = createOneDataBlock(pJoin->finBlk, false, &pJoin->midBlk);
×
1158
    if (code) {
×
1159
      QRY_ERR_RET(code);
×
1160
    }
1161

1162
    code = blockDataEnsureCapacity(pJoin->midBlk, pJoin->finBlk->info.capacity);
×
1163
    if (TSDB_CODE_SUCCESS != code) {
×
1164
      QRY_ERR_RET(code);
×
1165
    }
1166
  }
1167

1168
  pJoin->blkThreshold = pJoin->finBlk->info.capacity * HJOIN_BLK_THRESHOLD_RATIO;
11,002✔
1169
  return TSDB_CODE_SUCCESS;
11,002✔
1170
}
1171

1172

1173
int32_t createHashJoinOperatorInfo(SOperatorInfo** pDownstream, int32_t numOfDownstream,
11,002✔
1174
                                           SHashJoinPhysiNode* pJoinNode, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
1175
  QRY_PARAM_CHECK(pOptrInfo);
11,002!
1176

1177
  int32_t             code = TSDB_CODE_SUCCESS;
11,002✔
1178
  SHJoinOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SHJoinOperatorInfo));
11,002✔
1179
  SOperatorInfo*      pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
11,002✔
1180
  if (pOperator == NULL || pInfo == NULL) {
11,002!
UNCOV
1181
    code = terrno;
×
UNCOV
1182
    goto _return;
×
1183
  }
1184

1185
  pInfo->tblTimeRange.skey = pJoinNode->timeRange.skey;
11,002✔
1186
  pInfo->tblTimeRange.ekey = pJoinNode->timeRange.ekey;
11,002✔
1187
  
1188
  pInfo->ctx.limit = pJoinNode->node.pLimit ? ((SLimitNode*)pJoinNode->node.pLimit)->limit : INT64_MAX;
11,002!
1189

1190
  setOperatorInfo(pOperator, "HashJoinOperator", QUERY_NODE_PHYSICAL_PLAN_HASH_JOIN, false, OP_NOT_OPENED, pInfo, pTaskInfo);
11,002✔
1191

1192
  HJ_ERR_JRET(hJoinInitTableInfo(pInfo, pJoinNode, pDownstream, 0, &pJoinNode->inputStat[0]));
11,002!
1193
  HJ_ERR_JRET(hJoinInitTableInfo(pInfo, pJoinNode, pDownstream, 1, &pJoinNode->inputStat[1]));
11,002!
1194

1195
  hJoinSetBuildAndProbeTable(pInfo, pJoinNode);
11,002✔
1196
  
1197
  HJ_ERR_JRET(hJoinBuildResColsMap(pInfo, pJoinNode));
11,002!
1198

1199
  HJ_ERR_JRET(hJoinInitBufPages(pInfo));
11,002!
1200

1201
  size_t hashCap = pInfo->pBuild->inputStat.inputRowNum > 0 ? (pInfo->pBuild->inputStat.inputRowNum * 1.5) : 1024;
11,002!
1202
  pInfo->pKeyHash = tSimpleHashInit(hashCap, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY));
11,002✔
1203
  if (pInfo->pKeyHash == NULL) {
11,002!
UNCOV
1204
    code = TSDB_CODE_OUT_OF_MEMORY;
×
UNCOV
1205
    goto _return;
×
1206
  }
1207

1208
  HJ_ERR_JRET(hJoinHandleConds(pInfo, pJoinNode));
11,002!
1209

1210
  HJ_ERR_JRET(hJoinInitResBlocks(pInfo, pJoinNode));
11,002!
1211

1212
  HJ_ERR_JRET(hJoinSetImplFp(pInfo));
11,002!
1213

1214
  HJ_ERR_JRET(appendDownstream(pOperator, pDownstream, numOfDownstream));
11,002!
1215

1216
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, hJoinMainProcess, NULL, destroyHashJoinOperator, optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
11,002✔
1217

1218
  qDebug("create hash Join operator done");
11,002✔
1219

1220
  *pOptrInfo = pOperator;
11,002✔
1221
  return TSDB_CODE_SUCCESS;
11,002✔
1222

UNCOV
1223
_return:
×
1224

UNCOV
1225
  if (pInfo != NULL) {
×
UNCOV
1226
    destroyHashJoinOperator(pInfo);
×
1227
  }
UNCOV
1228
  destroyOperatorAndDownstreams(pOperator, pDownstream, numOfDownstream);
×
UNCOV
1229
  pTaskInfo->code = code;
×
UNCOV
1230
  return code;
×
1231
}
1232

1233

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