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

taosdata / TDengine / #4732

12 Sep 2025 02:34AM UTC coverage: 59.006% (+0.2%) from 58.768%
#4732

push

travis-ci

web-flow
docs: optimize taosd config parameters doc better (#32964)

136080 of 292959 branches covered (46.45%)

Branch coverage included in aggregate %.

204691 of 284559 relevant lines covered (71.93%)

25330922.01 hits per line

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

69.56
/source/dnode/vnode/src/tsdb/tsdbRead2.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 "osDef.h"
17
#include "tsdb.h"
18
#include "tsdbDataFileRW.h"
19
#include "tsdbFS2.h"
20
#include "tsdbMerge.h"
21
#include "tsdbReadUtil.h"
22
#include "tsdbUtil2.h"
23
#include "tsimplehash.h"
24

25
#define ASCENDING_TRAVERSE(o)       (o == TSDB_ORDER_ASC)
26
#define getCurrentKeyInSttBlock(_r) (&((_r)->currentKey))
27
#define tColRowGetKeyDeepCopy(_pBlock, _irow, _slotId, _pKey)             \
28
  do {                                                                    \
29
    (_pKey)->ts = (_pBlock)->aTSKEY[(_irow)];                             \
30
    (_pKey)->numOfPKs = 0;                                                \
31
    if ((_slotId) != -1) {                                                \
32
      code = tColRowGetPriamyKeyDeepCopy(_pBlock, _irow, _slotId, _pKey); \
33
      TSDB_CHECK_CODE(code, lino, _end);                                  \
34
    }                                                                     \
35
  } while (0)
36

37
#define outOfTimeWindow(_ts, _window) (((_ts) > (_window)->ekey) || ((_ts) < (_window)->skey))
38

39
typedef struct {
40
  bool overlapWithNeighborBlock;
41
  bool hasDupTs;
42
  bool overlapWithDelInfo;
43
  bool overlapWithSttBlock;
44
  bool overlapWithKeyInBuf;
45
  bool partiallyRequired;
46
  bool moreThanCapcity;
47
} SDataBlockToLoadInfo;
48

49
static int32_t getCurrentBlockInfo(SDataBlockIter* pBlockIter, SFileDataBlockInfo** pInfo, const char* idStr);
50
static int32_t buildDataBlockFromBufImpl(STableBlockScanInfo* pBlockScanInfo, int64_t endKey, int32_t capacity,
51
                                         STsdbReader* pReader);
52
static int32_t getValidMemRow(SIterInfo* pIter, const SArray* pDelList, STsdbReader* pReader, TSDBROW** pRes);
53
static int32_t doMergeRowsInFileBlocks(SBlockData* pBlockData, STableBlockScanInfo* pScanInfo, SRowKey* pKey,
54
                                       STsdbReader* pReader);
55
static int32_t doMergeRowsInSttBlock(SSttBlockReader* pSttBlockReader, STableBlockScanInfo* pScanInfo,
56
                                     SRowMerger* pMerger, int32_t pkSrcSlot, SVersionRange* pVerRange, const char* id);
57
static int32_t doMergeRowsInBuf(SIterInfo* pIter, uint64_t uid, SRowKey* pCurKey, SArray* pDelList,
58
                                STsdbReader* pReader);
59
static int32_t doAppendRowFromTSRow(SSDataBlock* pBlock, STsdbReader* pReader, SRow* pTSRow,
60
                                    STableBlockScanInfo* pScanInfo);
61
static int32_t doAppendRowFromFileBlock(SSDataBlock* pResBlock, STsdbReader* pReader, SBlockData* pBlockData,
62
                                        int32_t rowIndex);
63
static void    setComposedBlockFlag(STsdbReader* pReader, bool composed);
64
static int32_t hasBeenDropped(const SArray* pDelList, int32_t* index, int64_t key, int64_t ver, int32_t order,
65
                              SVersionRange* pVerRange, bool hasPk, bool* dropped);
66

67
static int32_t doMergeMemTableMultiRows(TSDBROW* pRow, SRowKey* pKey, uint64_t uid, SIterInfo* pIter, SArray* pDelList,
68
                                        TSDBROW* pResRow, STsdbReader* pReader, bool* freeTSRow);
69
static int32_t doMergeMemIMemRows(TSDBROW* pRow, SRowKey* pRowKey, TSDBROW* piRow, SRowKey* piRowKey,
70
                                  STableBlockScanInfo* pBlockScanInfo, STsdbReader* pReader, SRow** pTSRow);
71
static int32_t mergeRowsInFileBlocks(SBlockData* pBlockData, STableBlockScanInfo* pBlockScanInfo, SRowKey* pKey,
72
                                     STsdbReader* pReader);
73
static int32_t mergeRowsInSttBlocks(SSttBlockReader* pSttBlockReader, STableBlockScanInfo* pScanInfo,
74
                                    STsdbReader* pReader);
75

76
static int32_t initDelSkylineIterator(STableBlockScanInfo* pBlockScanInfo, int32_t order, SReadCostSummary* pCost);
77
static void getTsdbByRetentions(SVnode* pVnode, SQueryTableDataCond* pCond, SRetention* retentions, const char* idstr,
78
                                int8_t* pLevel, STsdb** pTsdb);
79
static SVersionRange getQueryVerRange(SVnode* pVnode, SQueryTableDataCond* pCond, int8_t level);
80
static int32_t       doBuildDataBlock(STsdbReader* pReader);
81
static int32_t       getCurrentKeyInBuf(STableBlockScanInfo* pScanInfo, STsdbReader* pReader, TSDBKEY* key);
82
static bool          hasDataInFileBlock(const SBlockData* pBlockData, const SFileBlockDumpInfo* pDumpInfo);
83
static bool          hasDataInSttBlock(STableBlockScanInfo* pInfo);
84
static int32_t       initBlockDumpInfo(STsdbReader* pReader, SDataBlockIter* pBlockIter);
85
static int32_t       getInitialDelIndex(const SArray* pDelSkyline, int32_t order);
86
static int32_t       resetTableListIndex(SReaderStatus* pStatus, const char* id);
87
static void          getMemTableTimeRange(STsdbReader* pReader, int64_t* pMaxKey, int64_t* pMinKey);
88
static void          updateComposedBlockInfo(STsdbReader* pReader, double el, STableBlockScanInfo* pBlockScanInfo);
89
static int32_t       buildFromPreFilesetBuffer(STsdbReader* pReader);
90

91
static void resetPreFilesetMemTableListIndex(SReaderStatus* pStatus);
92

93
FORCE_INLINE int32_t pkCompEx(SRowKey* p1, SRowKey* p2) {
253✔
94
  if (p2 == NULL) {
605,765,226!
95
    return 1;
541,180,263✔
96
  }
97

98
  if (p1 == NULL) {
2,147,483,647!
99
    return -1;
×
100
  }
101

102
  if (p1->ts < p2->ts) {
2,147,483,647!
103
    return -1;
1,337,526,211✔
104
  } else if (p1->ts > p2->ts) {
975,650,261!
105
    return 1;
403,190,718✔
106
  }
107

108
  if (p1->numOfPKs == 0) {
572,459,543!
109
    return 0;
562,099,762✔
110
  } else {
111
    return tRowKeyCompare(p1, p2);
10,359,781✔
112
  }
113
}
114

115
static int32_t tColRowGetPriamyKeyDeepCopy(SBlockData* pBlock, int32_t irow, int32_t slotId, SRowKey* pKey) {
92,918,508✔
116
  int32_t   code = TSDB_CODE_SUCCESS;
92,918,508✔
117
  int32_t   lino = 0;
92,918,508✔
118
  SColData* pColData = NULL;
92,918,508✔
119
  SColVal   cv;
120

121
  TSDB_CHECK_CONDITION((pBlock != NULL) && (pBlock->aColData != NULL), code, lino, _end, TSDB_CODE_INVALID_PARA);
92,918,508!
122
  TSDB_CHECK_NULL(pKey, code, lino, _end, TSDB_CODE_INVALID_PARA);
92,918,508!
123

124
  pColData = &pBlock->aColData[slotId];
92,918,508✔
125

126
  code = tColDataGetValue(pColData, irow, &cv);
92,918,508✔
127
  TSDB_CHECK_CODE(code, lino, _end);
95,339,243!
128

129
  pKey->numOfPKs = 1;
95,339,243✔
130
  pKey->pks[0].type = cv.value.type;
95,339,243✔
131

132
  if (IS_NUMERIC_TYPE(cv.value.type)) {
95,339,243!
133
    valueCloneDatum(pKey->pks, &cv.value, cv.value.type);
93,953,730✔
134
  } else {
135
    pKey->pks[0].nData = cv.value.nData;
1,385,513✔
136
    TAOS_MEMCPY(pKey->pks[0].pData, cv.value.pData, cv.value.nData);
1,385,513✔
137
  }
138

139
_end:
94,213,386✔
140
  if (code != TSDB_CODE_SUCCESS) {
94,213,386!
141
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
142
  }
143
  return code;
94,231,141✔
144
}
145

146
// for test purpose, todo remove it
147
static int32_t tGetPrimaryKeyIndex(uint8_t* p, SPrimaryKeyIndex* index) {
30,110,080✔
148
  int32_t n = 0;
30,110,080✔
149
  n += tGetI8(p + n, &index->type);
30,110,080!
150
  n += tGetU32v(p + n, &index->offset);
30,110,080!
151
  return n;
30,110,080✔
152
}
153

154
static int32_t tRowGetPrimaryKeyDeepCopy(SRow* pRow, SRowKey* pKey) {
30,111,114✔
155
  int32_t          code = TSDB_CODE_SUCCESS;
30,111,114✔
156
  int32_t          lino = 0;
30,111,114✔
157
  SPrimaryKeyIndex indices[TD_MAX_PK_COLS];
158
  uint8_t*         data = NULL;
30,111,114✔
159
  int32_t          len = 0;
30,111,114✔
160

161
  TSDB_CHECK_NULL(pRow, code, lino, _end, TSDB_CODE_INVALID_PARA);
30,111,114!
162
  if (pRow->numOfPKs > 0) {
30,111,114!
163
    TSDB_CHECK_NULL(pRow->data, code, lino, _end, TSDB_CODE_INVALID_PARA);
30,111,422!
164
    TSDB_CHECK_NULL(pKey, code, lino, _end, TSDB_CODE_INVALID_PARA);
30,111,422!
165
  }
166

167
  data = pRow->data;
30,111,114✔
168
  for (int32_t i = 0; i < pRow->numOfPKs; i++) {
60,219,625✔
169
    data += tGetPrimaryKeyIndex(data, &indices[i]);
30,110,587✔
170
  }
171

172
  // primary keys
173
  for (int32_t i = 0; i < pRow->numOfPKs; i++) {
60,228,622✔
174
    pKey->pks[i].type = indices[i].type;
30,113,239✔
175

176
    uint8_t* tdata = data + indices[i].offset;
30,113,239✔
177
    if (pRow->flag >> 4) {
30,113,239✔
178
      tdata += tGetI16v(tdata, NULL);
2,744,854✔
179
    }
180

181
    if (IS_VAR_DATA_TYPE(indices[i].type)) {
30,113,239!
182
      tdata += tGetU32v(tdata, &pKey->pks[i].nData);
823,810✔
183
      TAOS_MEMCPY(pKey->pks[i].pData, tdata, pKey->pks[i].nData);
823,810✔
184
    } else {
185
      valueSetDatum(pKey->pks + i, indices[i].type, data + indices[i].offset, tDataTypes[pKey->pks[i].type].bytes);
29,289,429✔
186
    }
187
  }
188

189
_end:
30,115,383✔
190
  if (code != TSDB_CODE_SUCCESS) {
30,115,383!
191
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
192
  }
193
  return code;
30,117,635✔
194
}
195

196
static int32_t setColumnIdSlotList(SBlockLoadSuppInfo* pSupInfo, SColumnInfo* pCols, const int32_t* pSlotIdList,
8,515,688✔
197
                                   int32_t numOfCols) {
198
  int32_t code = TSDB_CODE_SUCCESS;
8,515,688✔
199
  int32_t lino = 0;
8,515,688✔
200

201
  TSDB_CHECK_NULL(pSupInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,515,688!
202
  if (numOfCols > 0) {
8,515,688!
203
    TSDB_CHECK_NULL(pSlotIdList, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,519,516!
204
    TSDB_CHECK_NULL(pCols, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,519,516!
205
  }
206

207
  pSupInfo->pk.pk = 0;
8,515,688✔
208
  pSupInfo->numOfPks = 0;
8,515,688✔
209
  pSupInfo->pkSrcSlot = -1;
8,515,688✔
210
  pSupInfo->pkDstSlot = -1;
8,515,688✔
211
  pSupInfo->smaValid = true;
8,515,688✔
212
  pSupInfo->numOfCols = numOfCols;
8,515,688✔
213

214
  pSupInfo->colId = taosMemoryCalloc(numOfCols, sizeof(int16_t) * 2 + POINTER_BYTES);
8,515,688!
215
  TSDB_CHECK_NULL(pSupInfo->colId, code, lino, _end, terrno);
8,530,524!
216

217
  pSupInfo->slotId = (int16_t*)((char*)pSupInfo->colId + (sizeof(int16_t) * numOfCols));
8,530,524✔
218
  pSupInfo->buildBuf = (char**)((char*)pSupInfo->slotId + (sizeof(int16_t) * numOfCols));
8,530,524✔
219
  for (int32_t i = 0; i < numOfCols; ++i) {
51,671,056✔
220
    pSupInfo->colId[i] = pCols[i].colId;
43,128,968✔
221
    pSupInfo->slotId[i] = pSlotIdList[i];
43,128,968✔
222

223
    if (pCols[i].pk) {
43,128,968✔
224
      pSupInfo->pk = pCols[i];
379,881✔
225
      pSupInfo->pkSrcSlot = i - 1;
379,881✔
226
      pSupInfo->pkDstSlot = pSlotIdList[i];
379,881✔
227
      pSupInfo->numOfPks += 1;
379,881✔
228
    }
229

230
    if (IS_VAR_DATA_TYPE(pCols[i].type)) {
43,128,968!
231
      pSupInfo->buildBuf[i] = taosMemoryMalloc(pCols[i].bytes);
5,241,191!
232
      if (pSupInfo->buildBuf[i] == NULL) {
5,252,910!
233
        tsdbError("failed to prepare memory for set columnId slot list, size:%d, code:%s", pCols[i].bytes,
×
234
                  tstrerror(terrno));
235
      }
236
      TSDB_CHECK_NULL(pSupInfo->buildBuf[i], code, lino, _end, terrno);
5,252,755!
237
    } else {
238
      pSupInfo->buildBuf[i] = NULL;
37,887,777✔
239
    }
240
  }
241

242
_end:
8,542,088✔
243
  if (code != TSDB_CODE_SUCCESS) {
8,542,088!
244
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
245
  }
246
  return code;
8,526,328✔
247
}
248

249
static int32_t updateBlockSMAInfo(STSchema* pSchema, SBlockLoadSuppInfo* pSupInfo) {
8,465,756✔
250
  int32_t   code = TSDB_CODE_SUCCESS;
8,465,756✔
251
  int32_t   lino = 0;
8,465,756✔
252
  int32_t   i = 0, j = 0;
8,465,756✔
253
  STColumn* pTCol = NULL;
8,465,756✔
254

255
  TSDB_CHECK_NULL(pSchema, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,465,756!
256
  TSDB_CHECK_NULL(pSupInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,465,756!
257

258
  while (i < pSchema->numOfCols && j < pSupInfo->numOfCols) {
106,496,286✔
259
    pTCol = &pSchema->columns[i];
98,039,964✔
260
    if (pTCol->colId == pSupInfo->colId[j]) {
98,039,964✔
261
      if (!IS_BSMA_ON(pTCol) && (PRIMARYKEY_TIMESTAMP_COL_ID != pTCol->colId)) {
42,856,248✔
262
        pSupInfo->smaValid = false;
9,424✔
263
        goto _end;
9,424✔
264
      }
265

266
      i += 1;
42,846,824✔
267
      j += 1;
42,846,824✔
268
    } else if (pTCol->colId < pSupInfo->colId[j]) {  // do nothing
55,183,716!
269
      i += 1;
55,194,867✔
270
    } else {
271
      code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
272
      TSDB_CHECK_CODE(code, lino, _end);
×
273
    }
274
  }
275

276
_end:
8,456,322✔
277
  if (code != TSDB_CODE_SUCCESS) {
8,465,756✔
278
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
10!
279
  }
280
  return code;
8,478,315✔
281
}
282

283
static bool isEmptyQueryTimeWindow(STimeWindow* pWindow) {
43,578,785✔
284
  return (pWindow == NULL) || (pWindow->skey > pWindow->ekey);
43,578,785!
285
}
286

287
// Update the query time window according to the data time to live(TTL) information, in order to avoid to return
288
// the expired data to client, even it is queried already.
289
static int32_t updateQueryTimeWindow(STsdb* pTsdb, STimeWindow* pWindow, STimeWindow* out) {
8,549,302✔
290
  int32_t code = TSDB_CODE_SUCCESS;
8,549,302✔
291
  int32_t lino = 0;
8,549,302✔
292
  int64_t earlyTs = 0;
8,549,302✔
293

294
  TSDB_CHECK_NULL(pTsdb, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,549,302!
295
  TSDB_CHECK_NULL(pWindow, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,549,302!
296
  TSDB_CHECK_NULL(out, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,549,302!
297

298
  earlyTs = tsdbGetEarliestTs(pTsdb);
8,549,302✔
299
  *out = *pWindow;
8,552,015✔
300
  if (out->skey < earlyTs) {
8,552,015✔
301
    out->skey = earlyTs;
4,287,782✔
302
  }
303

304
_end:
4,264,233✔
305
  if (code != TSDB_CODE_SUCCESS) {
8,552,015!
306
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
307
  }
308
  return code;
8,550,928✔
309
}
310

311
// init file iterator
312
static int32_t initFilesetIterator(SFilesetIter* pIter, TFileSetArray* pFileSetArray, STsdbReader* pReader) {
8,305,135✔
313
  int32_t             code = TSDB_CODE_SUCCESS;
8,305,135✔
314
  int32_t             lino = 0;
8,305,135✔
315
  SBlockLoadSuppInfo* pInfo = NULL;
8,305,135✔
316
  SSttBlockReader*    pSttReader = NULL;
8,305,135✔
317
  size_t              numOfFileset = 0;
8,305,135✔
318
  bool                asc = false;
8,305,135✔
319

320
  TSDB_CHECK_NULL(pIter, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,305,135!
321
  TSDB_CHECK_NULL(pFileSetArray, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,305,135!
322
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,305,135!
323

324
  pInfo = &pReader->suppInfo;
8,305,135✔
325
  numOfFileset = TARRAY2_SIZE(pFileSetArray);
8,305,135✔
326
  asc = ASCENDING_TRAVERSE(pReader->info.order);
8,305,135✔
327

328
  pIter->index = asc ? -1 : numOfFileset;
8,305,135✔
329
  pIter->order = pReader->info.order;
8,305,135✔
330
  pIter->pFilesetList = pFileSetArray;
8,305,135✔
331
  pIter->numOfFiles = numOfFileset;
8,305,135✔
332

333
  if (pIter->pSttBlockReader == NULL) {
8,305,135✔
334
    pIter->pSttBlockReader = taosMemoryCalloc(1, sizeof(struct SSttBlockReader));
8,287,398!
335
    if (pIter->pSttBlockReader == NULL) {
8,284,990!
336
      tsdbError("failed to prepare the last block iterator, since:%s %s", tstrerror(terrno), pReader->idStr);
×
337
    }
338
    TSDB_CHECK_NULL(pIter->pSttBlockReader, code, lino, _end, terrno);
8,286,028!
339
  }
340

341
  pSttReader = pIter->pSttBlockReader;
8,303,765✔
342
  pSttReader->order = pReader->info.order;
8,303,765✔
343
  pSttReader->window = pReader->info.window;
8,303,765✔
344
  pSttReader->verRange = pReader->info.verRange;
8,303,765✔
345
  pSttReader->numOfPks = pReader->suppInfo.numOfPks;
8,303,765✔
346
  pSttReader->uid = 0;
8,303,765✔
347

348
  tMergeTreeClose(&pSttReader->mergeTree);
8,303,765✔
349
  code = initRowKey(&pSttReader->currentKey, INT64_MIN, pInfo->numOfPks, pInfo->pk.type, pInfo->pk.bytes, asc);
8,305,928✔
350
  if (code != TSDB_CODE_SUCCESS) {
8,306,422!
351
    tsdbError("failed init row key, %s", pReader->idStr);
×
352
  } else {
353
    tsdbDebug("init fileset iterator, total files:%d %s", pIter->numOfFiles, pReader->idStr);
8,306,422✔
354
  }
355
  TSDB_CHECK_CODE(code, lino, _end);
8,306,077!
356

357
_end:
8,306,077✔
358
  if (code != TSDB_CODE_SUCCESS) {
8,306,077!
359
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
360
  }
361
  return code;
8,307,262✔
362
}
363

364
static int32_t filesetIteratorNext(SFilesetIter* pIter, STsdbReader* pReader, bool* hasNext) {
13,369,005✔
365
  int32_t           code = TSDB_CODE_SUCCESS;
13,369,005✔
366
  int32_t           lino = 0;
13,369,005✔
367
  bool              asc = false;
13,369,005✔
368
  int32_t           step = 0;
13,369,005✔
369
  SReadCostSummary* pCost = NULL;
13,369,005✔
370
  STFileObj**       pFileObj = NULL;
13,369,005✔
371

372
  TSDB_CHECK_NULL(pIter, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,369,005!
373
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,369,005!
374
  TSDB_CHECK_NULL(hasNext, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,369,005!
375

376
  asc = ASCENDING_TRAVERSE(pIter->order);
13,369,005✔
377
  step = asc ? 1 : -1;
13,369,005✔
378
  *hasNext = false;
13,369,005✔
379

380
  pIter->index += step;
13,369,005✔
381
  if ((asc && pIter->index >= pIter->numOfFiles) || ((!asc) && pIter->index < 0)) {
13,369,005✔
382
    *hasNext = false;
2,350,946✔
383
    goto _end;
2,350,946✔
384
  }
385

386
  pCost = &pReader->cost;
11,018,059✔
387

388
  TSDB_CHECK_NULL(pIter->pSttBlockReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
11,018,059!
389
  pIter->pSttBlockReader->uid = 0;
11,018,059✔
390
  tMergeTreeClose(&pIter->pSttBlockReader->mergeTree);
11,018,059✔
391
  destroySttBlockReader(pReader->status.pLDataIterArray, &pCost->sttCost);
11,020,029✔
392

393
  pReader->status.pLDataIterArray = taosArrayInit(4, POINTER_BYTES);
11,021,631✔
394
  TSDB_CHECK_NULL(pReader->status.pLDataIterArray, code, lino, _end, terrno);
11,022,067!
395

396
  // check file the time range of coverage
397
  STimeWindow win = {0};
11,022,067✔
398

399
  while (1) {
1,644,869✔
400
    if (pReader->pFileReader != NULL) {
12,666,936✔
401
      tsdbDataFileReaderClose(&pReader->pFileReader);
163,245✔
402
    }
403

404
    TSDB_CHECK_CONDITION(pIter->index < pIter->pFilesetList->size, code, lino, _end, TSDB_CODE_INTERNAL_ERROR);
12,665,030!
405
    pReader->status.pCurrentFileset = pIter->pFilesetList->data[pIter->index];
12,665,030✔
406

407
    pFileObj = pReader->status.pCurrentFileset->farr;
12,665,030✔
408
    if (pFileObj[0] != NULL || pFileObj[3] != NULL) {
12,665,030✔
409
      SDataFileReaderConfig conf = {.tsdb = pReader->pTsdb, .szPage = pReader->pTsdb->pVnode->config.tsdbPageSize};
203,735✔
410

411
      const char* filesName[4] = {0};
203,735✔
412

413
      if (pFileObj[0] != NULL) {
203,735✔
414
        conf.files[0].file = *pFileObj[0]->f;
196,327✔
415
        conf.files[0].exist = true;
196,327✔
416
        filesName[0] = pFileObj[0]->fname;
196,327✔
417

418
        conf.files[1].file = *pFileObj[1]->f;
196,327✔
419
        conf.files[1].exist = true;
196,327✔
420
        filesName[1] = pFileObj[1]->fname;
196,327✔
421

422
        conf.files[2].file = *pFileObj[2]->f;
196,327✔
423
        conf.files[2].exist = true;
196,327✔
424
        filesName[2] = pFileObj[2]->fname;
196,327✔
425
      }
426

427
      if (pFileObj[3] != NULL) {
203,735✔
428
        conf.files[3].exist = true;
44,182✔
429
        conf.files[3].file = *pFileObj[3]->f;
44,182✔
430
        filesName[3] = pFileObj[3]->fname;
44,182✔
431
      }
432

433
      code = tsdbDataFileReaderOpen(filesName, &conf, &pReader->pFileReader);
203,735✔
434
      TSDB_CHECK_CODE(code, lino, _end);
205,622!
435

436
      pReader->cost.headFileLoad += 1;
205,622✔
437
    }
438

439
    int32_t fid = pReader->status.pCurrentFileset->fid;
12,666,917✔
440
    tsdbFidKeyRange(fid, pReader->pTsdb->keepCfg.days, pReader->pTsdb->keepCfg.precision, &win.skey, &win.ekey);
12,666,917✔
441

442
    // current file are no longer overlapped with query time window, ignore remain files
443
    if ((asc && win.skey > pReader->info.window.ekey) || (!asc && win.ekey < pReader->info.window.skey)) {
12,663,715✔
444
      tsdbDebug("%p remain files are not qualified for qrange:%" PRId64 "-%" PRId64 ", ignore, %s", pReader,
992,400✔
445
                pReader->info.window.skey, pReader->info.window.ekey, pReader->idStr);
446
      *hasNext = false;
993,140✔
447
      break;
993,140✔
448
    }
449

450
    if ((asc && (win.ekey < pReader->info.window.skey)) || ((!asc) && (win.skey > pReader->info.window.ekey))) {
11,671,315✔
451
      pIter->index += step;
2,097,592✔
452
      if ((asc && pIter->index >= pIter->numOfFiles) || ((!asc) && pIter->index < 0)) {
2,097,592✔
453
        *hasNext = false;
452,723✔
454
        break;
452,723✔
455
      }
456
      continue;
1,644,869✔
457
    }
458

459
    tsdbDebug("%p file found fid:%d for qrange:%" PRId64 "-%" PRId64 ", %s", pReader, fid, pReader->info.window.skey,
9,573,723✔
460
              pReader->info.window.ekey, pReader->idStr);
461

462
    *hasNext = true;
9,574,699✔
463
    break;
9,574,699✔
464
  }
465

466
_end:
13,371,508✔
467
  if (code != TSDB_CODE_SUCCESS) {
13,371,508!
468
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
469
  }
470
  return code;
13,371,415✔
471
}
472

473
bool shouldFreePkBuf(SBlockLoadSuppInfo* pSupp) {
26,527,416✔
474
  return (pSupp != NULL) && (pSupp->numOfPks > 0) && IS_VAR_DATA_TYPE(pSupp->pk.type);
26,527,416!
475
}
476

477
int32_t resetDataBlockIterator(SDataBlockIter* pIter, int32_t order, bool needFree, const char* id) {
17,862,252✔
478
  int32_t code = TSDB_CODE_SUCCESS;
17,862,252✔
479
  int32_t lino = 0;
17,862,252✔
480

481
  TSDB_CHECK_NULL(pIter, code, lino, _end, TSDB_CODE_INVALID_PARA);
17,862,252!
482

483
  pIter->order = order;
17,862,252✔
484
  pIter->index = -1;
17,862,252✔
485
  pIter->numOfBlocks = 0;
17,862,252✔
486

487
  if (pIter->blockList == NULL) {
17,862,252✔
488
    pIter->blockList = taosArrayInit(4, sizeof(SFileDataBlockInfo));
8,285,316✔
489
    if (pIter->blockList == NULL) {
8,283,091!
490
      tsdbError("%s failed to reset block iter, func:%s at line:%d code:%s", id, __func__, __LINE__, tstrerror(terrno));
×
491
    }
492
    TSDB_CHECK_NULL(pIter->blockList, code, lino, _end, terrno);
8,283,056!
493
  } else {
494
    clearDataBlockIterator(pIter, needFree);
9,576,936✔
495
  }
496

497
_end:
17,863,139✔
498
  if (code != TSDB_CODE_SUCCESS) {
17,863,139!
499
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
500
  }
501
  return code;
17,860,683✔
502
}
503

504
static void initReaderStatus(SReaderStatus* pStatus) {
8,527,545✔
505
  if (pStatus == NULL) {
8,527,545!
506
    return;
×
507
  }
508
  pStatus->pTableIter = NULL;
8,527,545✔
509
  pStatus->loadFromFile = true;
8,527,545✔
510
}
511

512
static int32_t createResBlock(SQueryTableDataCond* pCond, int32_t capacity, SSDataBlock** pResBlock) {
2,533✔
513
  int32_t      code = TSDB_CODE_SUCCESS;
2,533✔
514
  int32_t      lino = 0;
2,533✔
515
  SSDataBlock* pBlock = NULL;
2,533✔
516

517
  TSDB_CHECK_NULL(pCond, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,533!
518
  TSDB_CHECK_CONDITION(capacity >= 0, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,533!
519
  TSDB_CHECK_NULL(pResBlock, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,533!
520

521
  code = createDataBlock(&pBlock);
2,533✔
522
  TSDB_CHECK_CODE(code, lino, _end);
2,532!
523

524
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
15,864✔
525
    SColumnInfoData colInfo = {0};
13,331✔
526
    colInfo.info = pCond->colList[i];
13,331✔
527
    code = blockDataAppendColInfo(pBlock, &colInfo);
13,331✔
528
    TSDB_CHECK_CODE(code, lino, _end);
13,332!
529
  }
530

531
  code = blockDataEnsureCapacity(pBlock, capacity);
2,533✔
532
  TSDB_CHECK_CODE(code, lino, _end);
2,533!
533

534
  *pResBlock = pBlock;
2,533✔
535
  pBlock = NULL;
2,533✔
536

537
_end:
2,533✔
538
  if (code != TSDB_CODE_SUCCESS) {
2,533!
539
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
540
  }
541
  if (pBlock) {
2,533!
542
    taosArrayDestroy(pBlock->pDataBlock);
×
543
    taosMemoryFreeClear(pBlock);
×
544
  }
545
  return code;
2,533✔
546
}
547

548
static int32_t tsdbInitReaderLock(STsdbReader* pReader) {
8,502,450✔
549
  int32_t code = TSDB_CODE_SUCCESS;
8,502,450✔
550
  int32_t lino = 0;
8,502,450✔
551

552
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,502,450!
553

554
  code = taosThreadMutexInit(&pReader->readerMutex, NULL);
8,502,450✔
555
  tsdbTrace("tsdb/read: %p, post-init read mutex: %p, code: %d", pReader, &pReader->readerMutex, code);
8,516,516✔
556
  TSDB_CHECK_CODE(code, lino, _end);
8,519,420!
557

558
_end:
8,519,420✔
559
  if (code != TSDB_CODE_SUCCESS) {
8,519,420!
560
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
561
  }
562
  return code;
8,516,257✔
563
}
564

565
static int32_t tsdbUninitReaderLock(STsdbReader* pReader) {
8,535,217✔
566
  int32_t code = TSDB_CODE_SUCCESS;
8,535,217✔
567
  int32_t lino = 0;
8,535,217✔
568

569
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,535,217!
570

571
  tsdbTrace("tsdb/read: %p, pre-uninit read mutex: %p, code: %d", pReader, &pReader->readerMutex, code);
8,535,217✔
572
  code = taosThreadMutexDestroy(&pReader->readerMutex);
8,535,217✔
573
  tsdbTrace("tsdb/read: %p, post-uninit read mutex: %p, code: %d", pReader, &pReader->readerMutex, code);
8,534,449✔
574
  TSDB_CHECK_CODE(code, lino, _end);
8,535,683✔
575

576
_end:
8,535,673✔
577
  if (code != TSDB_CODE_SUCCESS) {
8,535,683✔
578
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
10!
579
  }
580
  return code;
8,535,659✔
581
}
582

583
static int32_t tsdbAcquireReader(STsdbReader* pReader) {
43,624,254✔
584
  int32_t code = TSDB_CODE_SUCCESS;
43,624,254✔
585
  int32_t lino = 0;
43,624,254✔
586

587
  TSDB_CHECK_CONDITION((pReader != NULL) && (pReader->idStr != NULL), code, lino, _end, TSDB_CODE_INVALID_PARA);
43,624,254!
588

589
  tsdbTrace("tsdb/read: %s, pre-take read mutex: %p, code: %d", pReader->idStr, &pReader->readerMutex, code);
43,624,254✔
590
  code = taosThreadMutexLock(&pReader->readerMutex);
43,624,256✔
591
  if (code != TSDB_CODE_SUCCESS) {
43,638,717!
592
    tsdbError("tsdb/read:%p, failed to lock reader mutex, code:%s", pReader->idStr, tstrerror(code));
×
593
  } else {
594
    tsdbTrace("tsdb/read: %s, post-take read mutex: %p, code: %d", pReader->idStr, &pReader->readerMutex, code);
43,638,717✔
595
  }
596
  TSDB_CHECK_CODE(code, lino, _end);
43,636,927!
597

598
_end:
43,636,927✔
599
  if (code != TSDB_CODE_SUCCESS) {
43,636,927!
600
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
601
  }
602
  return code;
43,629,545✔
603
}
604

605
static int32_t tsdbTryAcquireReader(STsdbReader* pReader) {
×
606
  int32_t code = TSDB_CODE_SUCCESS;
×
607
  int32_t lino = 0;
×
608

609
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
610

611
  code = taosThreadMutexTryLock(&pReader->readerMutex);
×
612
  if (code != TSDB_CODE_SUCCESS) {
×
613
    // Failing to acquire the lock is reasonable, not an error
614
    tsdbWarn("tsdb/read: %p, post-trytake read mutex: %p, code: %d", pReader, &pReader->readerMutex, code);
×
615
  } else {
616
    tsdbTrace("tsdb/read: %p, post-trytask read mutex: %p", pReader, &pReader->readerMutex);
×
617
  }
618
  return code;
×
619

620
_end:
×
621
  if (code != TSDB_CODE_SUCCESS) {
×
622
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
623
  }
624
  return code;
×
625
}
626

627
static int32_t tsdbReleaseReader(STsdbReader* pReader) {
43,629,223✔
628
  int32_t code = TSDB_CODE_SUCCESS;
43,629,223✔
629
  int32_t lino = 0;
43,629,223✔
630

631
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
43,629,223!
632

633
  code = taosThreadMutexUnlock(&pReader->readerMutex);
43,629,223✔
634
  if (code != TSDB_CODE_SUCCESS) {
43,639,011!
635
    tsdbError("tsdb/read: %p post-untake read mutex:%p failed, code:%d", pReader, &pReader->readerMutex, code);
×
636
  } else {
637
    tsdbTrace("tsdb/read: %p, post-untake read mutex: %p", pReader, &pReader->readerMutex);
43,639,011✔
638
  }
639
  TSDB_CHECK_CODE(code, lino, _end);
43,636,751!
640

641
_end:
43,636,751✔
642
  if (code != TSDB_CODE_SUCCESS) {
43,636,751!
643
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
644
  }
645
  return code;
43,635,395✔
646
}
647

648
void tsdbReleaseDataBlock2(void* p) {
8,215,419✔
649
  if (p == NULL) return;
8,215,419!
650

651
  STsdbReader* pReader = (STsdbReader*)p;
8,215,419✔
652
  SReaderStatus* pStatus = &pReader->status;
8,215,419✔
653
  if (!pStatus->composedDataBlock) {
8,215,419✔
654
    (void)tsdbReleaseReader(pReader);
7,895,275✔
655
  }
656
}
657

658
static int32_t initResBlockInfo(SResultBlockInfo* pResBlockInfo, int64_t capacity, SSDataBlock* pResBlock,
8,514,300✔
659
                                SQueryTableDataCond* pCond, SBlockLoadSuppInfo* pSup) {
660
  int32_t      code = TSDB_CODE_SUCCESS;
8,514,300✔
661
  int32_t      lino = 0;
8,514,300✔
662
  SSDataBlock* p = NULL;
8,514,300✔
663

664
  TSDB_CHECK_NULL(pResBlockInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,514,300!
665
  TSDB_CHECK_CONDITION((pResBlockInfo->pResBlock != NULL) || (pSup != NULL), code, lino, _end, TSDB_CODE_INVALID_PARA);
8,514,300!
666

667
  pResBlockInfo->capacity = capacity;
8,514,300✔
668
  pResBlockInfo->pResBlock = pResBlock;
8,514,300✔
669

670
  if (pResBlockInfo->pResBlock == NULL) {
8,514,300✔
671
    pResBlockInfo->freeBlock = true;
2,533✔
672
    code = createResBlock(pCond, pResBlockInfo->capacity, &pResBlockInfo->pResBlock);
2,533✔
673
    TSDB_CHECK_CODE(code, lino, _end);
2,533!
674
  } else {
675
    pResBlockInfo->freeBlock = false;
8,511,767✔
676
  }
677

678
  if (pSup->numOfPks > 0) {
8,514,300✔
679
    p = pResBlockInfo->pResBlock;
379,856✔
680
    p->info.pks[0].type = pSup->pk.type;
379,856✔
681
    p->info.pks[1].type = pSup->pk.type;
379,856✔
682

683
    if (IS_VAR_DATA_TYPE(pSup->pk.type) && (p->info.pks[0].pData == NULL)) {
379,856!
684
      p->info.pks[0].pData = taosMemoryCalloc(1, pSup->pk.bytes);
×
685
      TSDB_CHECK_NULL(p->info.pks[0].pData, code, lino, _end, terrno);
×
686

687
      p->info.pks[1].pData = taosMemoryCalloc(1, pSup->pk.bytes);
×
688
      TSDB_CHECK_NULL(p->info.pks[0].pData, code, lino, _end, terrno);
×
689

690
      p->info.pks[0].nData = pSup->pk.bytes;
×
691
      p->info.pks[1].nData = pSup->pk.bytes;
×
692
    }
693
  }
694

695
_end:
8,514,300✔
696
  if (code != TSDB_CODE_SUCCESS) {
8,514,300!
697
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
698
  }
699
  return code;
8,512,781✔
700
}
701

702
static int32_t tsdbReaderCreate(SVnode* pVnode, SQueryTableDataCond* pCond, void** ppReader, int32_t capacity,
8,529,805✔
703
                                SSDataBlock* pResBlock, const char* idstr) {
704
  int32_t             code = TSDB_CODE_SUCCESS;
8,529,805✔
705
  int32_t             lino = 0;
8,529,805✔
706
  int8_t              level = 0;
8,529,805✔
707
  STsdbReader*        pReader = NULL;
8,529,805✔
708
  SBlockLoadSuppInfo* pSup = NULL;
8,529,805✔
709

710
  TSDB_CHECK_NULL(pVnode, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,529,805!
711
  TSDB_CHECK_NULL(pCond, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,529,805!
712
  TSDB_CHECK_NULL(ppReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,529,805!
713

714
  *ppReader = NULL;
8,529,805✔
715
  pReader = (STsdbReader*)taosMemoryCalloc(1, sizeof(*pReader));
8,529,805✔
716
  TSDB_CHECK_NULL(pReader, code, lino, _end, terrno);
8,524,299!
717

718
  if (VND_IS_TSMA(pVnode)) {
8,524,299!
719
    tsdbDebug("vgId:%d, tsma is selected to query, %s", TD_VID(pVnode), idstr);
×
720
  }
721

722
  initReaderStatus(&pReader->status);
8,524,299✔
723
  getTsdbByRetentions(pVnode, pCond, pVnode->config.tsdbCfg.retentions, idstr, &level, &pReader->pTsdb);
8,527,998✔
724

725
  pReader->info.suid = pCond->suid;
8,531,141✔
726
  pReader->info.order = pCond->order;
8,531,141✔
727
  pReader->info.verRange = getQueryVerRange(pVnode, pCond, level);
8,531,141✔
728
  code = updateQueryTimeWindow(pReader->pTsdb, &pCond->twindows, &pReader->info.window);
8,531,610✔
729
  TSDB_CHECK_CODE(code, lino, _end);
8,531,426!
730

731
  if (idstr == NULL) {
8,531,426✔
732
    idstr = "";
32,954✔
733
  }
734
  pReader->idStr = taosStrdup(idstr);
8,531,426!
735
  TSDB_CHECK_NULL(pReader->idStr, code, lino, _end, terrno);
8,526,907!
736

737
  pReader->type = pCond->type;
8,526,907✔
738
  pReader->bFilesetDelimited = false;
8,526,907✔
739
  pReader->blockInfoBuf.numPerBucket = 1000;  // 1000 tables per bucket
8,526,907✔
740

741
  if (pCond->numOfCols <= 0) {
8,526,907!
742
    tsdbError("vgId:%d, invalid column number %d in query cond, %s", TD_VID(pVnode), pCond->numOfCols, idstr);
×
743
    TSDB_CHECK_CONDITION(pCond->numOfCols > 0, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
744
  }
745

746
  // allocate buffer in order to load data blocks from file
747
  pSup = &pReader->suppInfo;
8,526,907✔
748
  pSup->tsColAgg.colId = PRIMARYKEY_TIMESTAMP_COL_ID;
8,526,907✔
749
  code = setColumnIdSlotList(pSup, pCond->colList, pCond->pSlotList, pCond->numOfCols);
8,526,907✔
750
  TSDB_CHECK_CODE(code, lino, _end);
8,518,757!
751

752
  code = initResBlockInfo(&pReader->resBlockInfo, capacity, pResBlock, pCond, pSup);
8,518,757✔
753
  TSDB_CHECK_CODE(code, lino, _end);
8,513,228!
754

755
  code = tBlockDataCreate(&pReader->status.fileBlockData);
8,513,228✔
756
  TSDB_CHECK_CODE(code, lino, _end);
8,509,494!
757

758
  if (pReader->suppInfo.colId[0] != PRIMARYKEY_TIMESTAMP_COL_ID) {
8,509,494!
759
    tsdbError("the first column isn't primary timestamp, %d, %s", pReader->suppInfo.colId[0], pReader->idStr);
×
760
    TSDB_CHECK_CONDITION(pReader->suppInfo.colId[0] == PRIMARYKEY_TIMESTAMP_COL_ID, code, lino, _end,
×
761
                         TSDB_CODE_INVALID_PARA);
762
  }
763

764
  pReader->status.pPrimaryTsCol = taosArrayGet(pReader->resBlockInfo.pResBlock->pDataBlock, pSup->slotId[0]);
8,509,494✔
765
  TSDB_CHECK_NULL(pReader->status.pPrimaryTsCol, code, lino, _end, terrno);
8,503,537!
766

767
  int32_t type = pReader->status.pPrimaryTsCol->info.type;
8,503,537✔
768
  if (type != TSDB_DATA_TYPE_TIMESTAMP) {
8,503,537!
769
    tsdbError("the first column isn't primary timestamp in result block, actual: %s, %s", tDataTypes[type].name,
×
770
              pReader->idStr);
771
    TSDB_CHECK_CONDITION(type == TSDB_DATA_TYPE_TIMESTAMP, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
772
  }
773

774
  code = tsdbInitReaderLock(pReader);
8,503,537✔
775
  TSDB_CHECK_CODE(code, lino, _end);
8,517,117!
776

777
  pSup->args = pReader->pTsdb->pVnode->pBse;
8,517,117✔
778

779
  code = tsem_init(&pReader->resumeAfterSuspend, 0, 0);
8,517,117✔
780
  TSDB_CHECK_CODE(code, lino, _end);
8,524,287!
781

782
  *ppReader = pReader;
8,524,287✔
783
  pReader = NULL;
8,524,287✔
784

785
_end:
8,524,287✔
786
  if (code != TSDB_CODE_SUCCESS) {
8,524,287!
787
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
788
  }
789
  if (pReader) {
8,522,780!
790
    tsdbReaderClose2(pReader);
×
791
  }
792
  return code;
8,524,050✔
793
}
794

795
static int32_t doLoadBlockIndex(STsdbReader* pReader, SDataFileReader* pFileReader, SArray* pIndexList) {
9,570,263✔
796
  int32_t              code = TSDB_CODE_SUCCESS;
9,570,263✔
797
  int32_t              lino = 0;
9,570,263✔
798
  int32_t              st = 0;
9,570,263✔
799
  int32_t              et1 = 0;
9,570,263✔
800
  int32_t              et2 = 0;
9,570,263✔
801
  int32_t              numOfTables = 0;
9,570,263✔
802
  const TBrinBlkArray* pBlkArray = NULL;
9,570,263✔
803
  STableUidList*       pList = NULL;
9,570,263✔
804
  SBrinBlk*            pBrinBlk = NULL;
9,570,263✔
805

806
  if (pFileReader == NULL) {
9,570,263✔
807
    goto _end;
9,396,669✔
808
  }
809

810
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
173,594!
811
  TSDB_CHECK_NULL(pIndexList, code, lino, _end, TSDB_CODE_INVALID_PARA);
173,594!
812

813
  st = taosGetTimestampUs();
175,340✔
814
  numOfTables = tSimpleHashGetSize(pReader->status.pTableMap);
175,340✔
815

816
  code = tsdbDataFileReadBrinBlk(pFileReader, &pBlkArray);
175,324✔
817
  TSDB_CHECK_CODE(code, lino, _end);
175,180!
818

819
#if 0
820
  LRUHandle* handle = NULL;
821

822
  int32_t    code = tsdbCacheGetBlockIdx(pFileReader->pTsdb->biCache, pFileReader, &handle);
823
  if (code != TSDB_CODE_SUCCESS || handle == NULL) {
824
    goto _end;
825
  }
826

827

828
  SArray* aBlockIdx = (SArray*)taosLRUCacheValue(pFileReader->pTsdb->biCache, handle);
829
  size_t  num = taosArrayGetSize(aBlockIdx);
830
  if (num == 0) {
831
    tsdbBICacheRelease(pFileReader->pTsdb->biCache, handle);
832
    return TSDB_CODE_SUCCESS;
833
  }
834
#endif
835

836
  // todo binary search to the start position
837
  et1 = taosGetTimestampUs();
175,082✔
838

839
  pList = &pReader->status.uidList;
175,082✔
840

841
  int32_t i = 0;
175,082✔
842
  int32_t j = 0;
175,082✔
843
  while (i < TARRAY2_SIZE(pBlkArray)) {
391,537✔
844
    pBrinBlk = &pBlkArray->data[i];
240,969✔
845
    if (pBrinBlk->maxTbid.suid < pReader->info.suid) {
240,969✔
846
      i += 1;
3,070✔
847
      continue;
3,070✔
848
    }
849

850
    if (pBrinBlk->minTbid.suid > pReader->info.suid) {  // not include the queried table/super table, quit the loop
237,899✔
851
      break;
3,556✔
852
    }
853

854
    TSDB_CHECK_CONDITION(
234,343!
855
        (pBrinBlk->minTbid.suid <= pReader->info.suid) && (pBrinBlk->maxTbid.suid >= pReader->info.suid), code, lino,
856
        _end, TSDB_CODE_INTERNAL_ERROR);
857

858
    if (pBrinBlk->maxTbid.suid == pReader->info.suid && pBrinBlk->maxTbid.uid < pList->tableUidList[j]) {
234,343✔
859
      i += 1;
26,247✔
860
      continue;
26,247✔
861
    }
862

863
    if (pBrinBlk->minTbid.suid == pReader->info.suid && pBrinBlk->minTbid.uid > pList->tableUidList[numOfTables - 1]) {
208,096✔
864
      break;
1,008✔
865
    }
866

867
    const void* p1 = taosArrayPush(pIndexList, pBrinBlk);
207,080✔
868
    TSDB_CHECK_NULL(p1, code, lino, _end, terrno);
207,080!
869

870
    i += 1;
207,080✔
871
    if (pBrinBlk->maxTbid.suid == pReader->info.suid) {
207,080✔
872
      while (j < numOfTables && pList->tableUidList[j] < pBrinBlk->maxTbid.uid) {
523,910✔
873
        j++;
328,971✔
874
      }
875
      if (j >= numOfTables) {
194,939✔
876
        break;
19,942✔
877
      }
878
    }
879
  }
880

881
  et2 = taosGetTimestampUs();
175,318✔
882
  tsdbDebug("load block index for %d/%d tables completed, elapsed time:%.2f ms, set BrinBlk:%.2f ms, size:%.2f Kb %s",
175,318✔
883
            numOfTables, (int32_t)pBlkArray->size, (et1 - st) / 1000.0, (et2 - et1) / 1000.0,
884
            pBlkArray->size * sizeof(SBrinBlk) / 1024.0, pReader->idStr);
885

886
  pReader->cost.headFileLoadTime += (et1 - st) / 1000.0;
175,356✔
887

888
_end:
9,572,025✔
889
  if (code != TSDB_CODE_SUCCESS) {
9,572,025!
890
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
891
  }
892
  return code;
9,571,211✔
893
}
894

895
static int32_t loadFileBlockBrinInfo(STsdbReader* pReader, SArray* pIndexList, SBlockNumber* pBlockNum,
9,567,678✔
896
                                     SArray* pTableScanInfoList) {
897
  int32_t         code = TSDB_CODE_SUCCESS;
9,567,678✔
898
  int32_t         lino = 0;
9,567,678✔
899
  int64_t         st = 0;
9,567,678✔
900
  bool            asc = false;
9,567,678✔
901
  STimeWindow     w = {0};
9,567,678✔
902
  SBrinRecordIter iter = {0};
9,567,678✔
903
  int32_t         numOfTables = 0;
9,567,678✔
904
  SBrinRecord*    pRecord = NULL;
9,567,678✔
905
  int32_t         k = 0;
9,567,678✔
906
  size_t          sizeInDisk = 0;
9,567,678✔
907

908
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
9,567,678!
909
  TSDB_CHECK_NULL(pBlockNum, code, lino, _end, TSDB_CODE_INVALID_PARA);
9,567,678!
910
  TSDB_CHECK_NULL(pTableScanInfoList, code, lino, _end, TSDB_CODE_INVALID_PARA);
9,567,678!
911

912
  st = taosGetTimestampUs();
9,568,667✔
913
  asc = ASCENDING_TRAVERSE(pReader->info.order);
9,568,667✔
914
  w = pReader->info.window;
9,568,667✔
915
  numOfTables = tSimpleHashGetSize(pReader->status.pTableMap);
9,568,667✔
916

917
  // clear info for the new file
918
  cleanupInfoForNextFileset(pReader->status.pTableMap);
9,569,404✔
919
  initBrinRecordIter(&iter, pReader->pFileReader, pIndexList);
9,573,288✔
920

921
  while (1) {
16,038,815✔
922
    code = getNextBrinRecord(&iter, &pRecord);
25,613,610✔
923
    TSDB_CHECK_CODE(code, lino, _end);
25,606,372!
924

925
    if (pRecord == NULL) {
25,606,372✔
926
      break;
9,539,009✔
927
    }
928

929
    if (pRecord->suid > pReader->info.suid) {
16,067,363✔
930
      break;
11,340✔
931
    }
932

933
    uint64_t uid = pReader->status.uidList.tableUidList[k];
16,056,023✔
934
    if (pRecord->suid < pReader->info.suid) {
16,056,023✔
935
      continue;
2,287,112✔
936
    }
937

938
    if (uid < pRecord->uid) {  // forward the table uid index
16,045,128✔
939
      while (k < numOfTables && pReader->status.uidList.tableUidList[k] < pRecord->uid) {
650,922✔
940
        k += 1;
329,592✔
941
      }
942

943
      if (k >= numOfTables) {
321,330✔
944
        break;
20,822✔
945
      }
946

947
      uid = pReader->status.uidList.tableUidList[k];
300,508✔
948
    }
949

950
    if (pRecord->uid < uid) {
16,024,306✔
951
      continue;
2,189,002✔
952
    }
953

954
    TSDB_CHECK_CONDITION((pRecord->suid == pReader->info.suid) && (uid == pRecord->uid), code, lino, _end,
13,835,304!
955
                         TSDB_CODE_INTERNAL_ERROR);
956

957
    STableBlockScanInfo* pScanInfo = NULL;
13,835,304✔
958
    code = getTableBlockScanInfo(pReader->status.pTableMap, uid, &pScanInfo, pReader->idStr);
13,835,304✔
959
    TSDB_CHECK_CODE(code, lino, _end);
13,836,547!
960

961
    // here we should find the first timestamp that is greater than the lastProcKey
962
    // the window is an open interval NOW.
963
    if (asc) {
13,836,547✔
964
      w.skey = pScanInfo->lastProcKey.ts;
12,929,476✔
965
    } else {
966
      w.ekey = pScanInfo->lastProcKey.ts;
907,071✔
967
    }
968

969
    // NOTE: specialized for open interval
970
    if (((w.skey < INT64_MAX) && ((w.skey + 1) > w.ekey)) || (w.skey == INT64_MAX)) {
13,836,547!
971
      k += 1;
×
972
      if (k >= numOfTables) {
×
973
        break;
×
974
      } else {
975
        continue;
×
976
      }
977
    }
978

979
    // 1. time range check
980
    if (pRecord->firstKey.key.ts > w.ekey || pRecord->lastKey.key.ts < w.skey) {
13,836,587✔
981
      continue;
62,209✔
982
    }
983
    // The data block's time range must intersect with the query time range
984
    if (pRecord->firstKey.key.ts > pReader->info.window.ekey || pRecord->lastKey.key.ts < pReader->info.window.skey) {
13,774,378✔
985
      continue;
24,969✔
986
    }
987

988
    if (asc) {
13,749,409✔
989
      if (pkCompEx(&pRecord->lastKey.key, &pScanInfo->lastProcKey) <= 0) {
25,836,492!
990
        continue;
×
991
      }
992
    } else {
993
      if (pkCompEx(&pRecord->firstKey.key, &pScanInfo->lastProcKey) >= 0) {
1,662,326!
994
        continue;
×
995
      }
996
    }
997

998
    // 2. version range check, version range is an CLOSED interval
999
    if (pRecord->minVer > pReader->info.verRange.maxVer || pRecord->maxVer < pReader->info.verRange.minVer) {
13,749,409!
1000
      continue;
77✔
1001
    }
1002

1003
    if (pScanInfo->pBlockList == NULL) {
13,749,332✔
1004
      pScanInfo->pBlockList = taosArrayInit(4, sizeof(SFileDataBlockInfo));
411,136✔
1005
      TSDB_CHECK_NULL(pScanInfo->pBlockList, code, lino, _end, terrno);
411,534!
1006
    }
1007

1008
    if (pScanInfo->pBlockIdxList == NULL) {
13,749,730✔
1009
      pScanInfo->pBlockIdxList = taosArrayInit(4, sizeof(STableDataBlockIdx));
95,406✔
1010
      TSDB_CHECK_NULL(pScanInfo->pBlockIdxList, code, lino, _end, terrno);
95,426!
1011
    }
1012

1013
    SFileDataBlockInfo blockInfo = {.tbBlockIdx = TARRAY_SIZE(pScanInfo->pBlockList)};
13,749,750✔
1014
    code = recordToBlockInfo(&blockInfo, pRecord);
13,749,750✔
1015
    TSDB_CHECK_CODE(code, lino, _end);
13,751,684!
1016
    sizeInDisk += blockInfo.blockSize;
13,751,684✔
1017

1018
    const void* p1 = taosArrayPush(pScanInfo->pBlockList, &blockInfo);
13,751,684✔
1019
    TSDB_CHECK_NULL(p1, code, lino, _end, terrno);
13,751,674!
1020

1021
    // todo: refactor to record the fileset skey/ekey
1022
    if (pScanInfo->filesetWindow.skey > pRecord->firstKey.key.ts) {
13,751,674✔
1023
      pScanInfo->filesetWindow.skey = pRecord->firstKey.key.ts;
411,317✔
1024
    }
1025

1026
    if (pScanInfo->filesetWindow.ekey < pRecord->lastKey.key.ts) {
13,751,674✔
1027
      pScanInfo->filesetWindow.ekey = pRecord->lastKey.key.ts;
13,751,656✔
1028
    }
1029

1030
    pBlockNum->numOfBlocks += 1;
13,751,674✔
1031
    if (taosArrayGetSize(pTableScanInfoList) == 0) {
13,751,674✔
1032
      p1 = taosArrayPush(pTableScanInfoList, &pScanInfo);
137,347✔
1033
      TSDB_CHECK_NULL(p1, code, lino, _end, terrno);
137,347!
1034
    } else {
1035
      STableBlockScanInfo** p = taosArrayGetLast(pTableScanInfoList);
13,614,285✔
1036
      TSDB_CHECK_NULL(p, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,614,378!
1037

1038
      if ((*p)->uid != uid) {
13,614,378✔
1039
        p1 = taosArrayPush(pTableScanInfoList, &pScanInfo);
274,182✔
1040
        TSDB_CHECK_NULL(p1, code, lino, _end, terrno);
274,182!
1041
      }
1042
    }
1043
  }
1044

1045
  pBlockNum->numOfSttFiles = pReader->status.pCurrentFileset->lvlArr->size;
9,571,171✔
1046
  int32_t total = pBlockNum->numOfSttFiles + pBlockNum->numOfBlocks;
9,571,171✔
1047

1048
  double el = (taosGetTimestampUs() - st) / 1000.0;
9,567,893✔
1049
  tsdbDebug(
9,567,893✔
1050
      "load block of %d tables completed, blocks:%d in %d tables, stt-files:%d, block-info-size:%.2f Kb, elapsed "
1051
      "time:%.2f ms %s",
1052
      numOfTables, pBlockNum->numOfBlocks, (int32_t)taosArrayGetSize(pTableScanInfoList), pBlockNum->numOfSttFiles,
1053
      sizeInDisk / 1000.0, el, pReader->idStr);
1054

1055
  pReader->cost.numOfBlocks += total;
9,568,046✔
1056
  pReader->cost.headFileLoadTime += el;
9,568,046✔
1057

1058
_end:
9,568,046✔
1059
  if (code != TSDB_CODE_SUCCESS) {
9,568,046!
1060
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1061
  }
1062
  clearBrinBlockIter(&iter);
9,568,046✔
1063
  return code;
9,570,582✔
1064
}
1065

1066
static void setBlockAllDumped(SFileBlockDumpInfo* pDumpInfo, int64_t maxKey, int32_t order) {
18,873,300✔
1067
  if (pDumpInfo != NULL) {
18,873,300!
1068
    pDumpInfo->allDumped = true;
18,873,414✔
1069
  }
1070
}
18,873,300✔
1071

1072
static int32_t updateLastKeyInfo(SRowKey* pKey, SFileDataBlockInfo* pBlockInfo, SDataBlockInfo* pInfo, int32_t numOfPks,
13,359,284✔
1073
                                 bool asc) {
1074
  int32_t code = TSDB_CODE_SUCCESS;
13,359,284✔
1075
  int32_t lino = 0;
13,359,284✔
1076

1077
  TSDB_CHECK_NULL(pKey, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,359,284!
1078
  TSDB_CHECK_NULL(pInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,359,284!
1079

1080
  pKey->ts = asc ? pInfo->window.ekey : pInfo->window.skey;
13,359,284✔
1081
  pKey->numOfPKs = numOfPks;
13,359,284✔
1082
  if (pKey->numOfPKs <= 0) {
13,359,284✔
1083
    goto _end;
13,196,340✔
1084
  }
1085

1086
  TSDB_CHECK_NULL(pBlockInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
162,944!
1087
  if (IS_NUMERIC_TYPE(pKey->pks[0].type)) {
162,944!
1088
    VALUE_SET_TRIVIAL_DATUM(pKey->pks, asc ? pBlockInfo->lastPk.val : pBlockInfo->firstPk.val);
162,943✔
1089
  } else {
1090
    uint8_t* p = asc ? pBlockInfo->lastPk.pData : pBlockInfo->firstPk.pData;
1!
1091
    pKey->pks[0].nData = asc ? varDataLen(pBlockInfo->lastPk.pData) : varDataLen(pBlockInfo->firstPk.pData);
1!
1092
    TAOS_MEMCPY(pKey->pks[0].pData, p, pKey->pks[0].nData);
1✔
1093
  }
1094

1095
_end:
13,359,284✔
1096
  if (code != TSDB_CODE_SUCCESS) {
13,359,284!
1097
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1098
  }
1099
  return code;
13,359,314✔
1100
}
1101

1102
static int32_t doGetValueFromBseBySeq(void* arg, uint8_t* pKey, int32_t keyLen, uint8_t** pValue, int32_t* len) {
×
1103
  int32_t  code = 0;
×
1104
  int32_t  lino = 0;
×
1105
  uint64_t seq = 0;
×
1106
  if (arg == NULL) {
×
1107
    tsdbError("failed to get value from bse by seq since %s", tstrerror(TSDB_CODE_INVALID_PARA));
×
1108
    return TSDB_CODE_INVALID_PARA;
×
1109
  }
1110

1111
  if (keyLen <= 0) {
×
1112
    *len = 0; 
×
1113
    return code;
×
1114
  } else {
1115
    int32_t unusedRet = tGetU64(pKey, &seq);
×
1116
  }
1117
 
1118
  if (seq == 0) {
×
1119
    tsdbError("failed to get value from bse by seq since seq is 0");
×
1120
    return TSDB_CODE_INVALID_PARA;
×
1121
  }
1122
  *len = 0;
×
1123
  code = bseGet((SBse*)arg, seq, pValue, len);
×
1124
  TSDB_CHECK_CODE(code, lino, _end);
×
1125

1126
_end:
×
1127
  if (code != 0) {
×
1128
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1129
  }
1130
  return code;
×
1131
}
1132
static int32_t doReallocBuf(SBlockLoadSuppInfo* pSup, int32_t colIndex, SColumnInfo* pInfo, int32_t len) {
×
1133
  int32_t code = 0;
×
1134
  int32_t bytes = pInfo->bytes;
×
1135
  int32_t extraSize = VARSTR_HEADER_SIZE;
×
1136

1137
  if (IS_STR_DATA_BLOB(pInfo->type)) {
×
1138
    if (len + BLOBSTR_HEADER_SIZE > TSDB_MAX_BLOB_LEN) {
×
1139
      tsdbWarn("column cid:%d actual data len %d is bigger than schema len %d", pSup->colId[colIndex], len, bytes);
×
1140
      return TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1141
    }
1142
    extraSize = BLOBSTR_HEADER_SIZE;
×
1143

1144
  } else {
1145
    if (len + VARSTR_HEADER_SIZE > pInfo->bytes) {
×
1146
      tsdbWarn("column cid:%d actual data len %d is bigger than schema len %d", pSup->colId[colIndex], len, bytes);
×
1147
      return TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1148
    }
1149
  }
1150

1151
  char* p = taosMemoryRealloc(pSup->buildBuf[colIndex], len + extraSize);
×
1152
  if (p == NULL) {
×
1153
    return terrno;
×
1154
  }
1155

1156
  pSup->buildBuf[colIndex] = p;
×
1157

1158
  if (IS_STR_DATA_BLOB(pInfo->type)) {
×
1159
    blobDataSetLen(pSup->buildBuf[colIndex], len);
×
1160
  } else {
1161
    varDataSetLen(pSup->buildBuf[colIndex], len);
×
1162
  }
1163

1164
  return code;
×
1165
}
1166
static int32_t doCopyColVal(SColumnInfoData* pColInfoData, int32_t rowIndex, int32_t colIndex, SColVal* pColVal,
2,147,483,647✔
1167
                            SBlockLoadSuppInfo* pSup) {
1168
  int32_t code = TSDB_CODE_SUCCESS;
2,147,483,647✔
1169
int32_t lino = 0;
2,147,483,647✔
1170

1171
  uint8_t* pValue = 0;
2,147,483,647✔
1172
  int32_t  len = 0;
2,147,483,647✔
1173
  TSDB_CHECK_NULL(pColInfoData, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,147,483,647!
1174
  TSDB_CHECK_NULL(pColVal, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,147,483,647!
1175

1176
  if (IS_VAR_DATA_TYPE(pColVal->value.type)) {
2,147,483,647!
1177
    if (!COL_VAL_IS_VALUE(pColVal)) {
984,063,629✔
1178
      colDataSetNULL(pColInfoData, rowIndex);
227,108,921✔
1179
    } else {
1180
      if (IS_STR_DATA_BLOB(pColVal->value.type)) {
756,954,708!
1181
        if (pColVal->value.nData != 0) {
×
1182
          code = doGetValueFromBseBySeq(pSup->args, pColVal->value.pData, pColVal->value.nData, &pValue, &len);
×
1183
          if (code != 0) {
×
1184
            colDataSetNULL(pColInfoData, rowIndex);
×
1185
            return 0;
×
1186
          } else {
1187
            TSDB_CHECK_CODE(code, lino, _end);
×
1188
          }
1189
        }
1190

1191
        if (IS_STR_DATA_BLOB(pColVal->value.type)) {
×
1192
          code = doReallocBuf(pSup, colIndex, &pColInfoData->info, len);
×
1193
        } else {
1194
          varDataSetLen(pSup->buildBuf[colIndex], len);
×
1195
          if ((len + VARSTR_HEADER_SIZE) > pColInfoData->info.bytes) {
×
1196
            tsdbWarn("column cid:%d actual data len %d is bigger than schema len %d", pColVal->cid, len,
×
1197
                     pColInfoData->info.bytes);
1198
            code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
×
1199
            TSDB_CHECK_CODE(code, lino, _end);
×
1200
          }
1201
        }
1202

1203
        if (len > 0) {
×
1204
          if (IS_STR_DATA_BLOB(pColVal->value.type)) {
×
1205
            (void)memcpy(blobDataVal(pSup->buildBuf[colIndex]), pValue, len);
×
1206
          } else {
1207
            (void)memcpy(varDataVal(pSup->buildBuf[colIndex]), pValue, len);
×
1208
          }
1209
        }
1210
        code = colDataSetVal(pColInfoData, rowIndex, pSup->buildBuf[colIndex], false);
×
1211
        taosMemFreeClear(pValue);
×
1212

1213
        TSDB_CHECK_CODE(code, lino, _end);
×
1214
      } else {
1215
        TSDB_CHECK_NULL(pSup, code, lino, _end, TSDB_CODE_INVALID_PARA);
967,905,683!
1216
        varDataSetLen(pSup->buildBuf[colIndex], pColVal->value.nData);
967,905,683✔
1217
        if ((pColVal->value.nData + VARSTR_HEADER_SIZE) > pColInfoData->info.bytes) {
967,905,683✔
1218
          tsdbWarn("column cid:%d actual data len %d is bigger than schema len %d", pColVal->cid, pColVal->value.nData,
4!
1219
                   pColInfoData->info.bytes);
1220
          code = TSDB_CODE_TDB_INVALID_TABLE_SCHEMA_VER;
4✔
1221
          TSDB_CHECK_CODE(code, lino, _end);
4!
1222
        }
1223

1224
        if (pColVal->value.nData > 0) {  // pData may be null, if nData is 0
967,905,679✔
1225
          (void)memcpy(varDataVal(pSup->buildBuf[colIndex]), pColVal->value.pData, pColVal->value.nData);
966,038,958✔
1226
        }
1227

1228
        code = colDataSetVal(pColInfoData, rowIndex, pSup->buildBuf[colIndex], false);
967,905,679✔
1229
        TSDB_CHECK_CODE(code, lino, _end);
950,296,318!
1230
      }
1231
    }
1232
  } else {
1233
    code = colDataSetVal(pColInfoData, rowIndex, VALUE_GET_DATUM(&pColVal->value, pColVal->value.type),
2,147,483,647!
1234
                         !COL_VAL_IS_VALUE(pColVal));
2,147,483,647✔
1235
    TSDB_CHECK_CODE(code, lino, _end);
2,147,483,647!
1236
  }
1237

1238
_end:
2,147,483,647✔
1239
  if (code != TSDB_CODE_SUCCESS) {
2,147,483,647✔
1240
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
4!
1241
  }
1242
  return code;
2,147,483,647✔
1243
}
1244

1245
static int32_t getCurrentBlockInfo(SDataBlockIter* pBlockIter, SFileDataBlockInfo** pInfo, const char* id) {
57,479,816✔
1246
  int32_t code = TSDB_CODE_SUCCESS;
57,479,816✔
1247
  int32_t lino = 0;
57,479,816✔
1248

1249
  TSDB_CHECK_NULL(pBlockIter, code, lino, _end, TSDB_CODE_INVALID_PARA);
57,479,816!
1250
  TSDB_CHECK_NULL(pInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
57,479,816!
1251
  TSDB_CHECK_NULL(pBlockIter->blockList, code, lino, _end, TSDB_CODE_INVALID_PARA);
57,479,816!
1252

1253
  *pInfo = NULL;
57,479,816✔
1254

1255
  size_t num = TARRAY_SIZE(pBlockIter->blockList);
57,479,816✔
1256
  if (num == 0) {
57,479,816✔
1257
    // Some callers would attempt to call this function. Filter out certain normal cases and return directly to avoid
1258
    // generating excessive unnecessary error logs.
1259
    TSDB_CHECK_CONDITION(num == pBlockIter->numOfBlocks, code, lino, _end, TSDB_CODE_INVALID_PARA);
9,420,494!
1260
    return TSDB_CODE_INVALID_PARA;
9,420,494✔
1261
  }
1262

1263
  *pInfo = taosArrayGet(pBlockIter->blockList, pBlockIter->index);
48,059,322✔
1264
  TSDB_CHECK_NULL(*pInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
48,062,487!
1265

1266
_end:
48,062,487✔
1267
  if (code != TSDB_CODE_SUCCESS) {
48,062,487!
1268
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1269
  }
1270
  return code;
48,062,452✔
1271
}
1272

1273
static int32_t doBinarySearchKey(const TSKEY* keyList, int num, int pos, TSKEY key, int order) {
2,362✔
1274
  // start end position
1275
  int s, e;
1276
  s = pos;
2,362✔
1277

1278
  // check
1279
  if (!(keyList != NULL && pos >= 0 && pos < num && num > 0)) {
2,362!
1280
    return -1;
×
1281
  }
1282
  if (order == TSDB_ORDER_ASC) {
2,363✔
1283
    // find the first position which is smaller than the key
1284
    e = num - 1;
2,069✔
1285
    if (key < keyList[pos]) return -1;
2,069!
1286
    while (1) {
24,188✔
1287
      // check can return
1288
      if (key >= keyList[e]) return e;
26,257!
1289
      if (key <= keyList[s]) return s;
26,257✔
1290
      if (e - s <= 1) return s;
26,252✔
1291

1292
      // change start or end position
1293
      int mid = s + (e - s + 1) / 2;
24,194✔
1294
      if (keyList[mid] > key)
24,194✔
1295
        e = mid;
15,566✔
1296
      else if (keyList[mid] < key)
8,628✔
1297
        s = mid;
8,622✔
1298
      else
1299
        return mid;
6✔
1300
    }
1301
  } else {  // DESC
1302
    // find the first position which is bigger than the key
1303
    e = 0;
294✔
1304
    if (key > keyList[pos]) return -1;
294!
1305
    while (1) {
2,919✔
1306
      // check can return
1307
      if (key <= keyList[e]) return e;
3,213!
1308
      if (key >= keyList[s]) return s;
3,213!
1309
      if (s - e <= 1) return s;
3,213✔
1310

1311
      // change start or end position
1312
      int mid = s - (s - e + 1) / 2;
2,927✔
1313
      if (keyList[mid] < key)
2,927✔
1314
        e = mid;
1,622✔
1315
      else if (keyList[mid] > key)
1,305✔
1316
        s = mid;
1,297✔
1317
      else
1318
        return mid;
8✔
1319
    }
1320
  }
1321
}
1322

1323
// handle the repeat ts cases.
1324
static int32_t findFirstPos(const int64_t* pTsList, int32_t num, int32_t startPos, bool asc) {
2,362✔
1325
  int32_t i = startPos;
2,362✔
1326
  int64_t startTs = pTsList[startPos];
2,362✔
1327
  if (asc) {
2,362✔
1328
    while (i < num && (pTsList[i] == startTs)) {
4,136!
1329
      i++;
2,068✔
1330
    }
1331
    return i - 1;
2,068✔
1332
  } else {
1333
    while (i >= 0 && (pTsList[i] == startTs)) {
588!
1334
      i--;
294✔
1335
    }
1336
    return i + 1;
294✔
1337
  }
1338
}
1339

1340
static int32_t getEndPosInDataBlock(STsdbReader* pReader, SBlockData* pBlockData, SBrinRecord* pRecord, int32_t pos) {
5,466,552✔
1341
  // NOTE: reverse the order to find the end position in data block
1342
  int32_t code = TSDB_CODE_SUCCESS;
5,466,552✔
1343
  int32_t lino = 0;
5,466,552✔
1344
  int32_t endPos = -1;
5,466,552✔
1345
  bool    asc = false;
5,466,552✔
1346

1347
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
5,466,552!
1348
  TSDB_CHECK_NULL(pRecord, code, lino, _end, TSDB_CODE_INVALID_PARA);
5,466,552!
1349

1350
  asc = ASCENDING_TRAVERSE(pReader->info.order);
5,466,552✔
1351

1352
  if (asc && pReader->info.window.ekey >= pRecord->lastKey.key.ts) {
5,466,552✔
1353
    endPos = pRecord->numRow - 1;
4,648,258✔
1354
  } else if (!asc && pReader->info.window.skey <= pRecord->firstKey.key.ts) {
818,294✔
1355
    endPos = 0;
817,252✔
1356
  } else {
1357
    int64_t key = asc ? pReader->info.window.ekey : pReader->info.window.skey;
1,042!
1358
    endPos = doBinarySearchKey(pBlockData->aTSKEY, pRecord->numRow, pos, key, pReader->info.order);
1,042✔
1359
    if (endPos == -1) {
1,177!
1360
      goto _end;
×
1361
    }
1362

1363
    endPos = findFirstPos(pBlockData->aTSKEY, pRecord->numRow, endPos, asc);
1,177✔
1364
  }
1365

1366
  if ((pReader->info.verRange.maxVer >= pRecord->minVer && pReader->info.verRange.maxVer < pRecord->maxVer) ||
5,466,781!
1367
      (pReader->info.verRange.minVer <= pRecord->maxVer && pReader->info.verRange.minVer > pRecord->minVer)) {
5,466,787!
1368
    int32_t i = endPos;
×
1369

1370
    if (asc) {
×
1371
      for (; i >= 0; --i) {
×
1372
        if (pBlockData->aVersion[i] <= pReader->info.verRange.maxVer) {
×
1373
          break;
×
1374
        }
1375
      }
1376
    } else {
1377
      for (; i < pRecord->numRow; ++i) {
×
1378
        if (pBlockData->aVersion[i] >= pReader->info.verRange.minVer) {
×
1379
          break;
×
1380
        }
1381
      }
1382
    }
1383

1384
    endPos = i;
×
1385
  }
1386

1387
_end:
5,466,781✔
1388
  if (code != TSDB_CODE_SUCCESS) {
5,466,781!
1389
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1390
  }
1391
  return endPos;
5,466,693✔
1392
}
1393

1394
static int32_t copyPrimaryTsCol(const SBlockData* pBlockData, SFileBlockDumpInfo* pDumpInfo, SColumnInfoData* pColData,
5,466,826✔
1395
                                int32_t dumpedRows, bool asc) {
1396
  int32_t code = TSDB_CODE_SUCCESS;
5,466,826✔
1397
  int32_t lino = 0;
5,466,826✔
1398

1399
  TSDB_CHECK_CONDITION((pBlockData != NULL) && (pBlockData->aTSKEY != NULL), code, lino, _end, TSDB_CODE_INVALID_PARA);
5,466,826!
1400
  TSDB_CHECK_NULL(pDumpInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
5,466,826!
1401
  TSDB_CHECK_NULL(pColData, code, lino, _end, TSDB_CODE_INVALID_PARA);
5,466,826!
1402
  TSDB_CHECK_CONDITION(dumpedRows >= 0, code, lino, _end, TSDB_CODE_INVALID_PARA);
5,466,826!
1403

1404
  if (asc) {
5,466,826✔
1405
    TAOS_MEMCPY(pColData->pData, &pBlockData->aTSKEY[pDumpInfo->rowIndex], dumpedRows * sizeof(int64_t));
4,649,601✔
1406
  } else {
1407
    int32_t startIndex = pDumpInfo->rowIndex - dumpedRows + 1;
817,225✔
1408
    TAOS_MEMCPY(pColData->pData, &pBlockData->aTSKEY[startIndex], dumpedRows * sizeof(int64_t));
817,225✔
1409

1410
    // todo: opt perf by extract the loop
1411
    // reverse the array list
1412
    int32_t  mid = dumpedRows >> 1u;
817,225✔
1413
    int64_t* pts = (int64_t*)pColData->pData;
817,225✔
1414
    for (int32_t j = 0; j < mid; ++j) {
1,664,934,440✔
1415
      int64_t t = pts[j];
1,664,117,215✔
1416
      pts[j] = pts[dumpedRows - j - 1];
1,664,117,215✔
1417
      pts[dumpedRows - j - 1] = t;
1,664,117,215✔
1418
    }
1419
  }
1420

1421
_end:
817,225✔
1422
  if (code != TSDB_CODE_SUCCESS) {
5,466,826!
1423
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1424
  }
1425
  return code;
5,466,753✔
1426
}
1427

1428
// a faster version of copy procedure.
1429
static int32_t copyNumericCols(const SColData* pData, SFileBlockDumpInfo* pDumpInfo, SColumnInfoData* pColData,
3,490,239✔
1430
                               int32_t dumpedRows, bool asc) {
1431
  int32_t  code = TSDB_CODE_SUCCESS;
3,490,239✔
1432
  int32_t  lino = 0;
3,490,239✔
1433
  uint8_t* p = NULL;
3,490,239✔
1434
  int32_t  step = asc ? 1 : -1;
3,490,239✔
1435

1436
  TSDB_CHECK_NULL(pData, code, lino, _end, TSDB_CODE_INVALID_PARA);
3,490,239!
1437
  TSDB_CHECK_NULL(pDumpInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
3,490,239!
1438
  TSDB_CHECK_NULL(pColData, code, lino, _end, TSDB_CODE_INVALID_PARA);
3,490,239!
1439
  TSDB_CHECK_CONDITION(dumpedRows >= 0, code, lino, _end, TSDB_CODE_INVALID_PARA);
3,490,239!
1440

1441
  TSDB_CHECK_CONDITION(pData->type < TSDB_DATA_TYPE_MAX, code, lino, _end, TSDB_CODE_INVALID_PARA);
3,490,239!
1442
  if (asc) {
3,490,239✔
1443
    p = pData->pData + tDataTypes[pData->type].bytes * pDumpInfo->rowIndex;
3,481,734✔
1444
  } else {
1445
    int32_t startIndex = pDumpInfo->rowIndex - dumpedRows + 1;
8,505✔
1446
    p = pData->pData + tDataTypes[pData->type].bytes * startIndex;
8,505✔
1447
  }
1448

1449
  // make sure it is aligned to 8bit, the allocated memory address is aligned to 256bit
1450

1451
  // 1. copy data in a batch model
1452
  TAOS_MEMCPY(pColData->pData, p, dumpedRows * tDataTypes[pData->type].bytes);
3,490,239✔
1453

1454
  // 2. reverse the array list in case of descending order scan data block
1455
  if (!asc) {
3,490,239✔
1456
    switch (pColData->info.type) {
8,639!
1457
      case TSDB_DATA_TYPE_TIMESTAMP:
6,110✔
1458
      case TSDB_DATA_TYPE_DOUBLE:
1459
      case TSDB_DATA_TYPE_BIGINT:
1460
      case TSDB_DATA_TYPE_DECIMAL64:
1461
      case TSDB_DATA_TYPE_UBIGINT: {
1462
        int32_t  mid = dumpedRows >> 1u;
6,110✔
1463
        int64_t* pts = (int64_t*)pColData->pData;
6,110✔
1464
        for (int32_t j = 0; j < mid; ++j) {
7,893,116✔
1465
          int64_t t = pts[j];
7,887,006✔
1466
          pts[j] = pts[dumpedRows - j - 1];
7,887,006✔
1467
          pts[dumpedRows - j - 1] = t;
7,887,006✔
1468
        }
1469
        break;
6,110✔
1470
      }
1471

1472
      case TSDB_DATA_TYPE_BOOL:
843✔
1473
      case TSDB_DATA_TYPE_TINYINT:
1474
      case TSDB_DATA_TYPE_UTINYINT: {
1475
        int32_t mid = dumpedRows >> 1u;
843✔
1476
        int8_t* pts = (int8_t*)pColData->pData;
843✔
1477
        for (int32_t j = 0; j < mid; ++j) {
1,209,495✔
1478
          int8_t t = pts[j];
1,208,652✔
1479
          pts[j] = pts[dumpedRows - j - 1];
1,208,652✔
1480
          pts[dumpedRows - j - 1] = t;
1,208,652✔
1481
        }
1482
        break;
843✔
1483
      }
1484

1485
      case TSDB_DATA_TYPE_SMALLINT:
562✔
1486
      case TSDB_DATA_TYPE_USMALLINT: {
1487
        int32_t  mid = dumpedRows >> 1u;
562✔
1488
        int16_t* pts = (int16_t*)pColData->pData;
562✔
1489
        for (int32_t j = 0; j < mid; ++j) {
806,281✔
1490
          int64_t t = pts[j];
805,719✔
1491
          pts[j] = pts[dumpedRows - j - 1];
805,719✔
1492
          pts[dumpedRows - j - 1] = t;
805,719✔
1493
        }
1494
        break;
562✔
1495
      }
1496

1497
      case TSDB_DATA_TYPE_FLOAT:
845✔
1498
      case TSDB_DATA_TYPE_INT:
1499
      case TSDB_DATA_TYPE_UINT: {
1500
        int32_t  mid = dumpedRows >> 1u;
845✔
1501
        int32_t* pts = (int32_t*)pColData->pData;
845✔
1502
        for (int32_t j = 0; j < mid; ++j) {
1,210,467✔
1503
          int32_t t = pts[j];
1,209,622✔
1504
          pts[j] = pts[dumpedRows - j - 1];
1,209,622✔
1505
          pts[dumpedRows - j - 1] = t;
1,209,622✔
1506
        }
1507
        break;
845✔
1508
      }
1509
      case TSDB_DATA_TYPE_DECIMAL: {
279✔
1510
        int32_t      mid = dumpedRows >> 1u;
279✔
1511
        DecimalWord* pDec = (DecimalWord*)pColData->pData;
279✔
1512
        DecimalWord  tmp[2] = {0};
279✔
1513
        for (int32_t j = 0; j < mid; ++j) {
400,415✔
1514
          tmp[0] = pDec[2 * j];
400,136✔
1515
          tmp[1] = pDec[2 * j + 1];
400,136✔
1516
          pDec[2 * j] = pDec[2 * (dumpedRows - j - 1)];
400,136✔
1517
          pDec[2 * j + 1] = pDec[2 * (dumpedRows - j - 1) + 1];
400,136✔
1518
          pDec[2 * (dumpedRows - j - 1)] = tmp[0];
400,136✔
1519
          pDec[2 * (dumpedRows - j - 1) + 1] = tmp[1];
400,136✔
1520
        }
1521
        break;
279✔
1522
      }
1523
    }
1524
  }
1525

1526
  // 3. if the  null value exists, check items one-by-one
1527
  if (pData->flag != HAS_VALUE) {
3,490,239✔
1528
    int32_t rowIndex = 0;
41,115✔
1529

1530
    for (int32_t j = pDumpInfo->rowIndex; rowIndex < dumpedRows; j += step, rowIndex++) {
132,793,998✔
1531
      uint8_t v = tColDataGetBitValue(pData, j);
132,762,233✔
1532
      if (v == 0 || v == 1) {
132,752,883!
1533
        colDataSetNull_f(pColData->nullbitmap, rowIndex);
19,419,373✔
1534
        pColData->hasNull = true;
19,419,373✔
1535
      }
1536
    }
1537
  }
1538

1539
_end:
3,480,889✔
1540
  if (code != TSDB_CODE_SUCCESS) {
3,480,889!
1541
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1542
  }
1543
  return code;
3,490,308✔
1544
}
1545

1546
static void blockInfoToRecord(SBrinRecord* record, SFileDataBlockInfo* pBlockInfo, SBlockLoadSuppInfo* pSupp) {
41,971,404✔
1547
  record->uid = pBlockInfo->uid;
41,971,404✔
1548
  record->firstKey = (STsdbRowKey){.key = {.ts = pBlockInfo->firstKey, .numOfPKs = pSupp->numOfPks}};
41,971,404✔
1549
  record->lastKey = (STsdbRowKey){.key = {.ts = pBlockInfo->lastKey, .numOfPKs = pSupp->numOfPks}};
41,971,404✔
1550

1551
  if (pSupp->numOfPks > 0) {
41,971,404✔
1552
    SValue* pFirst = &record->firstKey.key.pks[0];
350,722✔
1553
    SValue* pLast = &record->lastKey.key.pks[0];
350,722✔
1554

1555
    pFirst->type = pSupp->pk.type;
350,722✔
1556
    pLast->type = pSupp->pk.type;
350,722✔
1557

1558
    if (IS_VAR_DATA_TYPE(pFirst->type)) {
350,722!
1559
      if (IS_STR_DATA_BLOB(pFirst->type)) {
13!
1560
        pFirst->pData = (uint8_t*)blobDataVal(pBlockInfo->firstPk.pData);
×
1561
        pFirst->nData = blobDataLen(pBlockInfo->firstPk.pData);
×
1562

1563
        pLast->pData = (uint8_t*)blobDataVal(pBlockInfo->lastPk.pData);
×
1564
        pLast->nData = blobDataLen(pBlockInfo->lastPk.pData);
×
1565

1566
      } else {
1567
        pFirst->pData = (uint8_t*)varDataVal(pBlockInfo->firstPk.pData);
13✔
1568
        pFirst->nData = varDataLen(pBlockInfo->firstPk.pData);
13✔
1569

1570
        pLast->pData = (uint8_t*)varDataVal(pBlockInfo->lastPk.pData);
13✔
1571
        pLast->nData = varDataLen(pBlockInfo->lastPk.pData);
13✔
1572
      }
1573
    } else {
1574
      VALUE_SET_TRIVIAL_DATUM(pFirst, pBlockInfo->firstPk.val);
350,709✔
1575
      VALUE_SET_TRIVIAL_DATUM(pLast, pBlockInfo->lastPk.val);
350,709✔
1576
    }
1577
  }
1578

1579
  record->minVer = pBlockInfo->minVer;
41,971,404✔
1580
  record->maxVer = pBlockInfo->maxVer;
41,971,404✔
1581
  record->blockOffset = pBlockInfo->blockOffset;
41,971,404✔
1582
  record->smaOffset = pBlockInfo->smaOffset;
41,971,404✔
1583
  record->blockSize = pBlockInfo->blockSize;
41,971,404✔
1584
  record->blockKeySize = pBlockInfo->blockKeySize;
41,971,404✔
1585
  record->smaSize = pBlockInfo->smaSize;
41,971,404✔
1586
  record->numRow = pBlockInfo->numRow;
41,971,404✔
1587
  record->count = pBlockInfo->count;
41,971,404✔
1588
}
41,971,404✔
1589

1590
static int32_t copyBlockDataToSDataBlock(STsdbReader* pReader, SRowKey* pLastProcKey) {
5,466,697✔
1591
  int32_t             code = TSDB_CODE_SUCCESS;
5,466,697✔
1592
  int32_t             lino = 0;
5,466,697✔
1593
  SReaderStatus*      pStatus = NULL;
5,466,697✔
1594
  SDataBlockIter*     pBlockIter = NULL;
5,466,697✔
1595
  SBlockLoadSuppInfo* pSupInfo = NULL;
5,466,697✔
1596
  SFileBlockDumpInfo* pDumpInfo = NULL;
5,466,697✔
1597
  SBlockData*         pBlockData = NULL;
5,466,697✔
1598
  SFileDataBlockInfo* pBlockInfo = NULL;
5,466,697✔
1599
  SSDataBlock*        pResBlock = NULL;
5,466,697✔
1600
  int32_t             numOfOutputCols = 0;
5,466,697✔
1601
  int64_t             st = 0;
5,466,697✔
1602
  bool                asc = false;
5,466,697✔
1603
  int32_t             step = 0;
5,466,697✔
1604
  SColVal             cv = {0};
5,466,697✔
1605
  SBrinRecord         tmp;
1606
  SBrinRecord*        pRecord = NULL;
5,466,697✔
1607

1608
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
5,466,697!
1609

1610
  pStatus = &pReader->status;
5,466,697✔
1611
  pBlockIter = &pStatus->blockIter;
5,466,697✔
1612
  pSupInfo = &pReader->suppInfo;
5,466,697✔
1613

1614
  pSupInfo->args = pReader->pTsdb->pVnode->pBse;
5,466,697✔
1615
  pDumpInfo = &pReader->status.fBlockDumpInfo;
5,466,697✔
1616

1617
  pBlockData = &pStatus->fileBlockData;
5,466,697✔
1618
  pResBlock = pReader->resBlockInfo.pResBlock;
5,466,697✔
1619
  numOfOutputCols = pSupInfo->numOfCols;
5,466,697✔
1620
  st = taosGetTimestampUs();
5,466,978✔
1621
  asc = ASCENDING_TRAVERSE(pReader->info.order);
5,466,978✔
1622
  step = asc ? 1 : -1;
5,466,978✔
1623

1624
  code = getCurrentBlockInfo(pBlockIter, &pBlockInfo, pReader->idStr);
5,466,978✔
1625
  TSDB_CHECK_CODE(code, lino, _end);
5,466,733!
1626

1627
  blockInfoToRecord(&tmp, pBlockInfo, pSupInfo);
5,466,733✔
1628
  pRecord = &tmp;
5,466,846✔
1629

1630
  // no data exists, return directly.
1631
  if (pBlockData->nRow == 0 || pBlockData->aTSKEY == 0) {
5,466,846!
1632
    tsdbWarn("%p no need to copy since no data in blockData, table uid:%" PRIu64 " has been dropped, %s", pReader,
×
1633
             pBlockInfo->uid, pReader->idStr);
1634
    pResBlock->info.rows = 0;
×
1635
    goto _end;
×
1636
  }
1637

1638
  TSDB_CHECK_CONDITION((pDumpInfo->rowIndex >= 0) && (pDumpInfo->rowIndex < pRecord->numRow), code, lino, _end,
5,466,874!
1639
                       TSDB_CODE_INVALID_PARA);
1640

1641
  // row index of dump info remain the initial position, let's find the appropriate start position.
1642
  if (((pDumpInfo->rowIndex == 0) && asc) || ((pDumpInfo->rowIndex == (pRecord->numRow - 1)) && (!asc))) {
5,466,874!
1643
    if (asc && pReader->info.window.skey <= pRecord->firstKey.key.ts &&
5,466,726✔
1644
        pReader->info.verRange.minVer <= pRecord->minVer) {
4,649,193✔
1645
      // pDumpInfo->rowIndex = 0;
1646
    } else if (!asc && pReader->info.window.ekey >= pRecord->lastKey.key.ts &&
817,547✔
1647
               pReader->info.verRange.maxVer >= pRecord->maxVer) {
816,361!
1648
      // pDumpInfo->rowIndex = pRecord->numRow - 1;
1649
    } else {  // find the appropriate the start position in current block, and set it to be the current rowIndex
1650
      int32_t pos = asc ? pRecord->numRow - 1 : 0;
1,186✔
1651
      int32_t order = asc ? TSDB_ORDER_DESC : TSDB_ORDER_ASC;
1,186✔
1652
      int64_t key = asc ? pReader->info.window.skey : pReader->info.window.ekey;
1,186✔
1653
      pDumpInfo->rowIndex = doBinarySearchKey(pBlockData->aTSKEY, pRecord->numRow, pos, key, order);
1,186✔
1654

1655
      if (pDumpInfo->rowIndex < 0) {
1,186!
1656
        tsdbError(
×
1657
            "%p failed to locate the start position in current block, global index:%d, table index:%d, brange:%" PRId64
1658
            "-%" PRId64 ", minVer:%" PRId64 ", maxVer:%" PRId64 " %s",
1659
            pReader, pBlockIter->index, pBlockInfo->tbBlockIdx, pRecord->firstKey.key.ts, pRecord->lastKey.key.ts,
1660
            pRecord->minVer, pRecord->maxVer, pReader->idStr);
1661
        code = TSDB_CODE_INVALID_PARA;
×
1662
        TSDB_CHECK_CODE(code, lino, _end);
×
1663
      }
1664

1665
      pDumpInfo->rowIndex = findFirstPos(pBlockData->aTSKEY, pRecord->numRow, pDumpInfo->rowIndex, (!asc));
1,186✔
1666

1667
      TSDB_CHECK_CONDITION(
1,185!
1668
          (pReader->info.verRange.minVer <= pRecord->maxVer && pReader->info.verRange.maxVer >= pRecord->minVer), code,
1669
          lino, _end, TSDB_CODE_INVALID_PARA);
1670

1671
      // find the appropriate start position that satisfies the version requirement.
1672
      if ((pReader->info.verRange.maxVer >= pRecord->minVer && pReader->info.verRange.maxVer < pRecord->maxVer) ||
1,185!
1673
          (pReader->info.verRange.minVer <= pRecord->maxVer && pReader->info.verRange.minVer > pRecord->minVer)) {
1,185!
1674
        int32_t i = pDumpInfo->rowIndex;
×
1675
        if (asc) {
×
1676
          for (; i < pRecord->numRow; ++i) {
×
1677
            if (pBlockData->aVersion[i] >= pReader->info.verRange.minVer) {
×
1678
              break;
×
1679
            }
1680
          }
1681
        } else {
1682
          for (; i >= 0; --i) {
×
1683
            if (pBlockData->aVersion[i] <= pReader->info.verRange.maxVer) {
×
1684
              break;
×
1685
            }
1686
          }
1687
        }
1688

1689
        pDumpInfo->rowIndex = i;
×
1690
      }
1691
    }
1692
  }
1693

1694
  // time window check
1695
  int32_t endIndex = getEndPosInDataBlock(pReader, pBlockData, pRecord, pDumpInfo->rowIndex);
5,466,873✔
1696
  if (endIndex == -1) {
5,466,683!
1697
    setBlockAllDumped(pDumpInfo, pReader->info.window.ekey, pReader->info.order);
×
1698
    goto _end;
×
1699
  }
1700

1701
  endIndex += step;
5,466,777✔
1702
  int32_t dumpedRows = asc ? (endIndex - pDumpInfo->rowIndex) : (pDumpInfo->rowIndex - endIndex);
5,466,777✔
1703
  if (dumpedRows > pReader->resBlockInfo.capacity) {  // output buffer check
5,466,777!
1704
    dumpedRows = pReader->resBlockInfo.capacity;
×
1705
  } else if (dumpedRows <= 0) {  // no qualified rows in current data block, quit directly.
5,466,777!
1706
    setBlockAllDumped(pDumpInfo, pReader->info.window.ekey, pReader->info.order);
×
1707
    goto _end;
×
1708
  }
1709

1710
  int32_t i = 0;
5,466,777✔
1711
  int32_t rowIndex = 0;
5,466,777✔
1712

1713
  SColumnInfoData* pColData = taosArrayGet(pResBlock->pDataBlock, pSupInfo->slotId[i]);
5,466,777✔
1714
  TSDB_CHECK_NULL(pColData, code, lino, _end, TSDB_CODE_INVALID_PARA);
5,466,864!
1715

1716
  if (pSupInfo->colId[i] == PRIMARYKEY_TIMESTAMP_COL_ID) {
5,466,864!
1717
    code = copyPrimaryTsCol(pBlockData, pDumpInfo, pColData, dumpedRows, asc);
5,466,906✔
1718
    TSDB_CHECK_CODE(code, lino, _end);
5,466,750!
1719
    i += 1;
5,466,750✔
1720
  }
1721

1722
  int32_t colIndex = 0;
5,466,708✔
1723
  int32_t num = pBlockData->nColData;
5,466,708✔
1724
  while (i < numOfOutputCols && colIndex < num) {
9,248,661!
1725
    rowIndex = 0;
3,782,305✔
1726

1727
    SColData* pData = tBlockDataGetColDataByIdx(pBlockData, colIndex);
3,782,305✔
1728
    if (pData->cid < pSupInfo->colId[i]) {
3,782,305!
1729
      colIndex += 1;
×
1730
    } else if (pData->cid == pSupInfo->colId[i]) {
3,782,305!
1731
      pColData = taosArrayGet(pResBlock->pDataBlock, pSupInfo->slotId[i]);
3,782,390✔
1732
      TSDB_CHECK_NULL(pColData, code, lino, _end, TSDB_CODE_INVALID_PARA);
3,782,231!
1733

1734
      if (pData->flag == HAS_NONE || pData->flag == HAS_NULL || pData->flag == (HAS_NULL | HAS_NONE)) {
3,782,231!
1735
        colDataSetNNULL(pColData, 0, dumpedRows);
85,077!
1736
      } else {
1737
        if (IS_MATHABLE_TYPE(pColData->info.type)) {
3,697,154✔
1738
          code = copyNumericCols(pData, pDumpInfo, pColData, dumpedRows, asc);
3,490,086✔
1739
          TSDB_CHECK_CODE(code, lino, _end);
3,490,240!
1740
        } else {  // varchar/nchar type
1741
          for (int32_t j = pDumpInfo->rowIndex; rowIndex < dumpedRows; j += step) {
56,158,567✔
1742
            code = tColDataGetValue(pData, j, &cv);
55,951,931✔
1743
            TSDB_CHECK_CODE(code, lino, _end);
56,447,625!
1744
            code = doCopyColVal(pColData, rowIndex++, i, &cv, pSupInfo);
56,447,625✔
1745
            TSDB_CHECK_CODE(code, lino, _end);
55,951,499!
1746
          }
1747
        }
1748
      }
1749

1750
      colIndex += 1;
3,781,953✔
1751
      i += 1;
3,781,953✔
1752
    } else {  // the specified column does not exist in file block, fill with null data
1753
      pColData = taosArrayGet(pResBlock->pDataBlock, pSupInfo->slotId[i]);
×
1754
      TSDB_CHECK_NULL(pColData, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
1755

1756
      colDataSetNNULL(pColData, 0, dumpedRows);
×
1757
      i += 1;
×
1758
    }
1759
  }
1760

1761
  // fill the mis-matched columns with null value
1762
  while (i < numOfOutputCols) {
5,466,356!
1763
    pColData = taosArrayGet(pResBlock->pDataBlock, pSupInfo->slotId[i]);
×
1764
    TSDB_CHECK_NULL(pColData, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
1765

1766
    colDataSetNNULL(pColData, 0, dumpedRows);
×
1767
    i += 1;
×
1768
  }
1769

1770
  pResBlock->info.dataLoad = 1;
5,466,356✔
1771
  pResBlock->info.rows = dumpedRows;
5,466,356✔
1772
  pDumpInfo->rowIndex += step * dumpedRows;
5,466,356✔
1773

1774
  tColRowGetKeyDeepCopy(pBlockData, pDumpInfo->rowIndex - step, pSupInfo->pkSrcSlot, pLastProcKey);
5,466,356!
1775

1776
  // check if current block are all handled
1777
  if (pDumpInfo->rowIndex >= 0 && pDumpInfo->rowIndex < pRecord->numRow) {
5,467,532✔
1778
    int64_t ts = pBlockData->aTSKEY[pDumpInfo->rowIndex];
1,175✔
1779

1780
    // the remain data has out of query time window, ignore current block
1781
    if (outOfTimeWindow(ts, &pReader->info.window)) {
1,175!
1782
      setBlockAllDumped(pDumpInfo, ts, pReader->info.order);
1,175✔
1783
    }
1784
  } else {
1785
    int64_t ts = asc ? pRecord->lastKey.key.ts : pRecord->firstKey.key.ts;
5,465,181✔
1786
    setBlockAllDumped(pDumpInfo, ts, pReader->info.order);
5,465,181✔
1787
  }
1788

1789
  double elapsedTime = (taosGetTimestampUs() - st) / 1000.0;
5,466,770✔
1790
  pReader->cost.blockLoadTime += elapsedTime;
5,466,770✔
1791

1792
  int32_t unDumpedRows = asc ? pRecord->numRow - pDumpInfo->rowIndex : pDumpInfo->rowIndex + 1;
5,466,770✔
1793
  tsdbDebug("%p copy file block to sdatablock, global index:%d, table index:%d, brange:%" PRId64 "-%" PRId64
5,466,770✔
1794
            ", rows:%d, remain:%d, minVer:%" PRId64 ", maxVer:%" PRId64 ", uid:%" PRIu64 " elapsed time:%.2f ms, %s",
1795
            pReader, pBlockIter->index, pBlockInfo->tbBlockIdx, pRecord->firstKey.key.ts, pRecord->lastKey.key.ts,
1796
            dumpedRows, unDumpedRows, pRecord->minVer, pRecord->maxVer, pBlockInfo->uid, elapsedTime, pReader->idStr);
1797

1798
_end:
5,464,779✔
1799
  if (code != TSDB_CODE_SUCCESS) {
5,466,770!
1800
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1801
  }
1802
  return code;
5,466,791✔
1803
}
1804

1805
static FORCE_INLINE STSchema* getTableSchemaImpl(STsdbReader* pReader, uint64_t uid) {
1806
  int32_t code = TSDB_CODE_SUCCESS;
106✔
1807
  int32_t lino = 0;
106✔
1808

1809
  TSDB_CHECK_CONDITION((pReader != NULL) && (pReader->info.pSchema == NULL), code, lino, _end, TSDB_CODE_INVALID_PARA);
106!
1810

1811
  code = metaGetTbTSchemaEx(pReader->pTsdb->pVnode->pMeta, pReader->info.suid, uid, -1, &pReader->info.pSchema);
106✔
1812
  if (code != TSDB_CODE_SUCCESS || pReader->info.pSchema == NULL) {
106!
1813
    terrno = code;
6✔
1814
    tsdbError("failed to get table schema, uid:%" PRIu64 ", it may have been dropped, ver:-1, %s", uid, pReader->idStr);
6!
1815
  }
1816
  TSDB_CHECK_CODE(code, lino, _end);
106!
1817
  TSDB_CHECK_NULL(pReader->info.pSchema, code, lino, _end, TSDB_CODE_INTERNAL_ERROR);
100!
1818

1819
  code = tsdbRowMergerInit(&pReader->status.merger, pReader->info.pSchema);
100✔
1820
  TSDB_CHECK_CODE(code, lino, _end);
100!
1821

1822
_end:
100✔
1823
  if (code != TSDB_CODE_SUCCESS) {
106!
1824
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
6!
1825
    terrno = code;
6✔
1826
    return NULL;
6✔
1827
  }
1828
  return pReader->info.pSchema;
100✔
1829
}
1830

1831
static int32_t doLoadFileBlockData(STsdbReader* pReader, SDataBlockIter* pBlockIter, SBlockData* pBlockData,
5,513,329✔
1832
                                   uint64_t uid) {
1833
  int32_t             code = TSDB_CODE_SUCCESS;
5,513,329✔
1834
  int32_t             lino = 0;
5,513,329✔
1835
  STSchema*           pSchema = NULL;
5,513,329✔
1836
  SFileDataBlockInfo* pBlockInfo = NULL;
5,513,329✔
1837
  SBlockLoadSuppInfo* pSup = NULL;
5,513,329✔
1838
  SFileBlockDumpInfo* pDumpInfo = NULL;
5,513,329✔
1839
  int64_t             st = 0;
5,513,329✔
1840
  SBrinRecord         tmp;
1841
  SBrinRecord*        pRecord = NULL;
5,513,329✔
1842

1843
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
5,513,329!
1844
  TSDB_CHECK_NULL(pBlockData, code, lino, _end, TSDB_CODE_INVALID_PARA);
5,513,329!
1845

1846
  pSchema = pReader->info.pSchema;
5,513,329✔
1847
  st = taosGetTimestampUs();
5,513,508✔
1848
  pSup = &pReader->suppInfo;
5,513,508✔
1849

1850
  tBlockDataReset(pBlockData);
5,513,508✔
1851

1852
  if (pReader->info.pSchema == NULL) {
5,513,481!
1853
    pSchema = getTableSchemaImpl(pReader, uid);
×
1854
    if (pSchema == NULL) {
×
1855
      tsdbError("%p table uid:%" PRIu64 " failed to get tableschema, code:%s, %s", pReader, uid, tstrerror(code),
×
1856
                pReader->idStr);
1857
      TSDB_CHECK_NULL(pSchema, code, lino, _end, terrno);
×
1858
    }
1859
  }
1860

1861
  code = getCurrentBlockInfo(pBlockIter, &pBlockInfo, pReader->idStr);
5,513,481✔
1862
  TSDB_CHECK_CODE(code, lino, _end);
5,513,609!
1863

1864
  pDumpInfo = &pReader->status.fBlockDumpInfo;
5,513,609✔
1865

1866
  blockInfoToRecord(&tmp, pBlockInfo, pSup);
5,513,609✔
1867
  pRecord = &tmp;
5,513,750✔
1868
  code = tsdbDataFileReadBlockDataByColumn(pReader->pFileReader, pRecord, pBlockData, pSchema, &pSup->colId[1],
5,513,750✔
1869
                                           pSup->numOfCols - 1);
5,513,750✔
1870
  if (code != TSDB_CODE_SUCCESS) {
5,513,624!
1871
    tsdbError("%p error occurs in loading file block, global index:%d, table index:%d, brange:%" PRId64 "-%" PRId64
×
1872
              ", rows:%d, code:%s %s",
1873
              pReader, pBlockIter->index, pBlockInfo->tbBlockIdx, pBlockInfo->firstKey, pBlockInfo->lastKey,
1874
              pBlockInfo->numRow, tstrerror(code), pReader->idStr);
1875
    TSDB_CHECK_CODE(code, lino, _end);
×
1876
  }
1877

1878
  double elapsedTime = (taosGetTimestampUs() - st) / 1000.0;
5,513,622✔
1879

1880
  tsdbDebug("%p load file block into buffer, global index:%d, index in table block list:%d, brange:%" PRId64 "-%" PRId64
5,513,622✔
1881
            ", rows:%d, minVer:%" PRId64 ", maxVer:%" PRId64 ", elapsed time:%.2f ms, %s",
1882
            pReader, pBlockIter->index, pBlockInfo->tbBlockIdx, pRecord->firstKey.key.ts, pRecord->lastKey.key.ts,
1883
            pRecord->numRow, pRecord->minVer, pRecord->maxVer, elapsedTime, pReader->idStr);
1884

1885
  pReader->cost.blockLoadTime += elapsedTime;
5,513,685✔
1886
  pDumpInfo->allDumped = false;
5,513,685✔
1887

1888
_end:
5,513,685✔
1889
  if (code != TSDB_CODE_SUCCESS) {
5,513,685!
1890
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1891
  }
1892
  return code;
5,513,729✔
1893
}
1894

1895
/**
1896
 * This is an two rectangles overlap cases.
1897
 */
1898
static int32_t dataBlockPartiallyRequired(STimeWindow* pWindow, SVersionRange* pVerRange, SFileDataBlockInfo* pBlock) {
13,460,265✔
1899
  return (pWindow->ekey < pBlock->lastKey && pWindow->ekey >= pBlock->firstKey) ||
20,622!
1900
         (pWindow->skey > pBlock->firstKey && pWindow->skey <= pBlock->lastKey) ||
13,439,643!
1901
         (pVerRange->minVer > pBlock->minVer && pVerRange->minVer <= pBlock->maxVer) ||
40,359,052!
1902
         (pVerRange->maxVer < pBlock->maxVer && pVerRange->maxVer >= pBlock->minVer);
13,438,522!
1903
}
1904

1905
static int32_t getNeighborBlockOfTable(SDataBlockIter* pBlockIter, SFileDataBlockInfo* pBlockInfo,
13,507,268✔
1906
                                       STableBlockScanInfo* pScanInfo, int32_t* nextIndex, int32_t order,
1907
                                       SBrinRecord* pRecord, SBlockLoadSuppInfo* pSupInfo, bool* res) {
1908
  int32_t             code = TSDB_CODE_SUCCESS;
13,507,268✔
1909
  int32_t             lino = 0;
13,507,268✔
1910
  bool                asc = false;
13,507,268✔
1911
  int32_t             step = 0;
13,507,268✔
1912
  STableDataBlockIdx* pTableDataBlockIdx = NULL;
13,507,268✔
1913
  SFileDataBlockInfo* p = NULL;
13,507,268✔
1914

1915
  TSDB_CHECK_NULL(pBlockInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,507,268!
1916
  TSDB_CHECK_NULL(pScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,507,268!
1917
  TSDB_CHECK_NULL(res, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,507,268!
1918

1919
  *res = false;
13,507,268✔
1920
  asc = ASCENDING_TRAVERSE(order);
13,507,268✔
1921
  step = asc ? 1 : -1;
13,507,268✔
1922

1923
  if (asc && pBlockInfo->tbBlockIdx >= taosArrayGetSize(pScanInfo->pBlockIdxList) - 1) {
13,507,268✔
1924
    *res = false;
435,717✔
1925
  } else if (!asc && pBlockInfo->tbBlockIdx == 0) {
13,071,601✔
1926
    *res = false;
40,048✔
1927
  } else {
1928
    TSDB_CHECK_NULL(pBlockIter, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,031,553!
1929
    TSDB_CHECK_NULL(nextIndex, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,031,553!
1930
    TSDB_CHECK_NULL(pRecord, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,031,553!
1931
    TSDB_CHECK_NULL(pSupInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,031,553!
1932

1933
    pTableDataBlockIdx = taosArrayGet(pScanInfo->pBlockIdxList, pBlockInfo->tbBlockIdx + step);
13,031,553✔
1934
    TSDB_CHECK_NULL(pTableDataBlockIdx, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,031,728!
1935

1936
    p = taosArrayGet(pBlockIter->blockList, pTableDataBlockIdx->globalIndex);
13,031,728✔
1937
    TSDB_CHECK_NULL(p, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,031,728!
1938

1939
    blockInfoToRecord(pRecord, p, pSupInfo);
13,031,728✔
1940

1941
    *nextIndex = pBlockInfo->tbBlockIdx + step;
13,031,726✔
1942
    *res = true;
13,031,726✔
1943
  }
1944

1945
_end:
13,507,491✔
1946
  if (code != TSDB_CODE_SUCCESS) {
13,507,491!
1947
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1948
  }
1949
  return code;
13,507,375✔
1950
}
1951

1952
static int32_t setFileBlockActiveInBlockIter(STsdbReader* pReader, SDataBlockIter* pBlockIter, int32_t index,
5✔
1953
                                             int32_t step) {
1954
  int32_t              code = TSDB_CODE_SUCCESS;
5✔
1955
  int32_t              lino = 0;
5✔
1956
  const void*          p = NULL;
5✔
1957
  SFileDataBlockInfo   fblock;
1958
  SFileDataBlockInfo*  pBlockInfo = NULL;
5✔
1959
  STableBlockScanInfo* pBlockScanInfo = NULL;
5✔
1960
  STableDataBlockIdx*  pTableDataBlockIdx = NULL;
5✔
1961

1962
  TSDB_CHECK_NULL(pBlockIter, code, lino, _end, TSDB_CODE_INVALID_PARA);
5!
1963
  TSDB_CHECK_CONDITION((index >= 0) && (index < pBlockIter->numOfBlocks), code, lino, _end, TSDB_CODE_INVALID_PARA);
5!
1964

1965
  p = taosArrayGet(pBlockIter->blockList, index);
5✔
1966
  TSDB_CHECK_NULL(p, code, lino, _end, TSDB_CODE_INVALID_PARA);
5!
1967

1968
  fblock = *(SFileDataBlockInfo*)p;
5✔
1969
  pBlockIter->index += step;
5✔
1970

1971
  if (index != pBlockIter->index) {
5!
1972
    if (index > pBlockIter->index) {
×
1973
      for (int32_t i = index - 1; i >= pBlockIter->index; --i) {
×
1974
        pBlockInfo = taosArrayGet(pBlockIter->blockList, i);
×
1975
        TSDB_CHECK_NULL(pBlockInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
1976

1977
        code = getTableBlockScanInfo(pReader->status.pTableMap, pBlockInfo->uid, &pBlockScanInfo, pReader->idStr);
×
1978
        TSDB_CHECK_CODE(code, lino, _end);
×
1979

1980
        pTableDataBlockIdx = taosArrayGet(pBlockScanInfo->pBlockIdxList, pBlockInfo->tbBlockIdx);
×
1981
        TSDB_CHECK_NULL(pTableDataBlockIdx, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
1982

1983
        pTableDataBlockIdx->globalIndex = i + 1;
×
1984

1985
        taosArraySet(pBlockIter->blockList, i + 1, pBlockInfo);
×
1986
      }
1987
    } else if (index < pBlockIter->index) {
×
1988
      for (int32_t i = index + 1; i <= pBlockIter->index; ++i) {
×
1989
        pBlockInfo = taosArrayGet(pBlockIter->blockList, i);
×
1990
        TSDB_CHECK_NULL(pBlockInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
1991

1992
        code = getTableBlockScanInfo(pReader->status.pTableMap, pBlockInfo->uid, &pBlockScanInfo, pReader->idStr);
×
1993
        TSDB_CHECK_CODE(code, lino, _end);
×
1994

1995
        pTableDataBlockIdx = taosArrayGet(pBlockScanInfo->pBlockIdxList, pBlockInfo->tbBlockIdx);
×
1996
        TSDB_CHECK_NULL(pTableDataBlockIdx, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
1997

1998
        pTableDataBlockIdx->globalIndex = i - 1;
×
1999
        taosArraySet(pBlockIter->blockList, i - 1, pBlockInfo);
×
2000
      }
2001
    }
2002

2003
    taosArraySet(pBlockIter->blockList, pBlockIter->index, &fblock);
×
2004
    pBlockScanInfo = NULL;
×
2005
    code = getTableBlockScanInfo(pReader->status.pTableMap, fblock.uid, &pBlockScanInfo, pReader->idStr);
×
2006
    TSDB_CHECK_CODE(code, lino, _end);
×
2007

2008
    pTableDataBlockIdx = taosArrayGet(pBlockScanInfo->pBlockIdxList, fblock.tbBlockIdx);
×
2009
    TSDB_CHECK_NULL(pTableDataBlockIdx, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
2010

2011
    pTableDataBlockIdx->globalIndex = pBlockIter->index;
×
2012
  }
2013

2014
_end:
5✔
2015
  if (code != TSDB_CODE_SUCCESS) {
5!
2016
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2017
  }
2018
  return code;
5✔
2019
}
2020

2021
// todo: this attribute could be acquired during extractin the global ordered block list.
2022
static bool overlapWithNeighborBlock2(SFileDataBlockInfo* pBlock, SBrinRecord* pRec, int32_t order, int32_t pkType,
13,031,725✔
2023
                                      int32_t numOfPk) {
2024
  // it is the last block in current file, no chance to overlap with neighbor blocks.
2025
  if (ASCENDING_TRAVERSE(order)) {
13,031,725✔
2026
    if (pBlock->lastKey == pRec->firstKey.key.ts) {
12,211,669✔
2027
      if (numOfPk > 0) {
15✔
2028
        SValue v1 = {.type = pkType};
3✔
2029
        if (IS_VAR_DATA_TYPE(pkType)) {
3!
2030
          if (IS_STR_DATA_BLOB(pkType)) {
3!
2031
            v1.pData = (uint8_t*)blobDataVal(pBlock->lastPk.pData), v1.nData = blobDataLen(pBlock->lastPk.pData);
×
2032
          } else {
2033
            v1.pData = (uint8_t*)varDataVal(pBlock->lastPk.pData), v1.nData = varDataLen(pBlock->lastPk.pData);
3✔
2034
          }
2035
        } else {
2036
          VALUE_SET_TRIVIAL_DATUM(&v1, pBlock->lastPk.val);
×
2037
        }
2038
        return (tValueCompare(&v1, &pRec->firstKey.key.pks[0]) == 0);
3✔
2039
      } else {  // no pk
2040
        return true;
12✔
2041
      }
2042
    } else {
2043
      return false;
12,211,654✔
2044
    }
2045
  } else {
2046
    if (pBlock->firstKey == pRec->lastKey.key.ts) {
820,056!
2047
      if (numOfPk > 0) {
×
2048
        SValue v1 = {.type = pkType};
×
2049
        if (IS_VAR_DATA_TYPE(pkType)) {
×
2050
          if (IS_STR_DATA_BLOB(pkType)) {
×
2051
            v1.pData = (uint8_t*)blobDataVal(pBlock->firstPk.pData), v1.nData = blobDataLen(pBlock->firstPk.pData);
×
2052

2053
          } else {
2054
            v1.pData = (uint8_t*)varDataVal(pBlock->firstPk.pData), v1.nData = varDataLen(pBlock->firstPk.pData);
×
2055
          }
2056
        } else {
2057
          VALUE_SET_TRIVIAL_DATUM(&v1, pBlock->firstPk.val);
×
2058
        }
2059
        return (tValueCompare(&v1, &pRec->lastKey.key.pks[0]) == 0);
×
2060
      } else {  // no pk
2061
        return true;
×
2062
      }
2063
    } else {
2064
      return false;
820,056✔
2065
    }
2066
  }
2067
}
2068

2069
static int64_t getBoarderKeyInFiles(SFileDataBlockInfo* pBlock, STableBlockScanInfo* pScanInfo, int32_t order) {
13,360,427✔
2070
  bool ascScan = ASCENDING_TRAVERSE(order);
13,360,427✔
2071

2072
  int64_t key = 0;
13,360,427✔
2073
  if (pScanInfo->sttKeyInfo.status == STT_FILE_HAS_DATA) {
13,360,427✔
2074
    int64_t keyInStt = pScanInfo->sttKeyInfo.nextProcKey.ts;
6,512,834✔
2075
    key = ascScan ? TMIN(pBlock->firstKey, keyInStt) : TMAX(pBlock->lastKey, keyInStt);
6,512,834✔
2076
  } else {
2077
    key = ascScan ? pBlock->firstKey : pBlock->lastKey;
6,847,593✔
2078
  }
2079

2080
  return key;
13,360,427✔
2081
}
2082

2083
static bool bufferDataInFileBlockGap(TSDBKEY keyInBuf, SFileDataBlockInfo* pBlock, STableBlockScanInfo* pScanInfo,
13,360,220✔
2084
                                     int32_t order) {
2085
  bool    ascScan = ASCENDING_TRAVERSE(order);
13,360,220✔
2086
  int64_t key = getBoarderKeyInFiles(pBlock, pScanInfo, order);
13,360,220✔
2087

2088
  return (ascScan && (keyInBuf.ts != TSKEY_INITIAL_VAL && keyInBuf.ts < key)) ||
26,720,549✔
2089
         (!ascScan && (keyInBuf.ts != TSKEY_INITIAL_VAL && keyInBuf.ts > key));
13,360,236✔
2090
}
2091

2092
static bool keyOverlapFileBlock(TSDBKEY key, SFileDataBlockInfo* pBlock, SVersionRange* pVerRange) {
13,460,209✔
2093
  return (key.ts >= pBlock->firstKey && key.ts <= pBlock->lastKey) && (pBlock->maxVer >= pVerRange->minVer) &&
13,479,205!
2094
         (pBlock->minVer <= pVerRange->maxVer);
18,996!
2095
}
2096

2097
static int32_t getBlockToLoadInfo(SDataBlockToLoadInfo* pInfo, SFileDataBlockInfo* pBlockInfo,
13,460,177✔
2098
                                  STableBlockScanInfo* pScanInfo, TSDBKEY keyInBuf, STsdbReader* pReader) {
2099
  int32_t             code = TSDB_CODE_SUCCESS;
13,460,177✔
2100
  int32_t             lino = 0;
13,460,177✔
2101
  SBrinRecord         rec = {0};
13,460,177✔
2102
  int32_t             neighborIndex = 0;
13,460,177✔
2103
  int32_t             order = 0;
13,460,177✔
2104
  SBlockLoadSuppInfo* pSupInfo = NULL;
13,460,177✔
2105
  SBrinRecord         pRecord;
2106
  bool                hasNeighbor = false;
13,460,177✔
2107

2108
  TSDB_CHECK_NULL(pInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,460,177!
2109
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,460,177!
2110

2111
  order = pReader->info.order;
13,460,177✔
2112
  pSupInfo = &pReader->suppInfo;
13,460,177✔
2113

2114
  code = getNeighborBlockOfTable(&pReader->status.blockIter, pBlockInfo, pScanInfo, &neighborIndex, order, &rec,
13,460,177✔
2115
                                 pSupInfo, &hasNeighbor);
2116
  TSDB_CHECK_CODE(code, lino, _end);
13,460,436!
2117

2118
  // overlap with neighbor
2119
  if (hasNeighbor) {
13,460,436✔
2120
    pInfo->overlapWithNeighborBlock =
13,015,816✔
2121
        overlapWithNeighborBlock2(pBlockInfo, &rec, order, pSupInfo->pk.type, pSupInfo->numOfPks);
13,015,818✔
2122
  }
2123

2124
  blockInfoToRecord(&pRecord, pBlockInfo, pSupInfo);
13,460,434✔
2125

2126
  // has duplicated ts of different version in this block
2127
  pInfo->hasDupTs = (pBlockInfo->numRow > pBlockInfo->count) || (pBlockInfo->count <= 0);
13,460,379!
2128
  pInfo->overlapWithDelInfo = overlapWithDelSkyline(pScanInfo, &pRecord, order);
13,460,379✔
2129

2130
  // todo handle the primary key overlap case
2131
  if (pScanInfo->sttKeyInfo.status == STT_FILE_HAS_DATA) {
13,460,311✔
2132
    int64_t nextProcKeyInStt = pScanInfo->sttKeyInfo.nextProcKey.ts;
6,528,257✔
2133
    pInfo->overlapWithSttBlock = !(pBlockInfo->lastKey < nextProcKeyInStt || pBlockInfo->firstKey > nextProcKeyInStt);
6,528,257✔
2134
  }
2135

2136
  pInfo->moreThanCapcity = pBlockInfo->numRow > pReader->resBlockInfo.capacity;
13,460,311✔
2137
  pInfo->partiallyRequired = dataBlockPartiallyRequired(&pReader->info.window, &pReader->info.verRange, pBlockInfo);
13,460,311✔
2138
  pInfo->overlapWithKeyInBuf = keyOverlapFileBlock(keyInBuf, pBlockInfo, &pReader->info.verRange);
13,460,262✔
2139

2140
_end:
13,460,200✔
2141
  if (code != TSDB_CODE_SUCCESS) {
13,460,200!
2142
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2143
  }
2144
  return code;
13,460,235✔
2145
}
2146

2147
// 1. the version of all rows should be less than the endVersion
2148
// 2. current block should not overlap with next neighbor block
2149
// 3. current timestamp should not be overlap with each other
2150
// 4. output buffer should be large enough to hold all rows in current block
2151
// 5. delete info should not overlap with current block data
2152
// 6. current block should not contain the duplicated ts
2153
static int32_t fileBlockShouldLoad(STsdbReader* pReader, SFileDataBlockInfo* pBlockInfo, STableBlockScanInfo* pScanInfo,
13,409,559✔
2154
                                   TSDBKEY keyInBuf, bool* load) {
2155
  int32_t              code = TSDB_CODE_SUCCESS;
13,409,559✔
2156
  int32_t              lino = 0;
13,409,559✔
2157
  SDataBlockToLoadInfo info = {0};
13,409,559✔
2158

2159
  TSDB_CHECK_NULL(load, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,409,559!
2160

2161
  *load = false;
13,409,559✔
2162
  code = getBlockToLoadInfo(&info, pBlockInfo, pScanInfo, keyInBuf, pReader);
13,409,559✔
2163
  TSDB_CHECK_CODE(code, lino, _end);
13,409,637!
2164

2165
  *load = (info.overlapWithNeighborBlock || info.hasDupTs || info.partiallyRequired || info.overlapWithKeyInBuf ||
13,409,525!
2166
           info.moreThanCapcity || info.overlapWithDelInfo || info.overlapWithSttBlock);
26,819,162✔
2167

2168
  // log the reason why load the datablock for profile
2169
  if (*load) {
13,409,637✔
2170
    tsdbDebug("%p uid:%" PRIu64
49,359✔
2171
              " need to load the datablock, overlapneighbor:%d, hasDup:%d, partiallyRequired:%d, "
2172
              "overlapWithKey:%d, greaterThanBuf:%d, overlapWithDel:%d, overlapWithSttBlock:%d, %s",
2173
              pReader, pBlockInfo->uid, info.overlapWithNeighborBlock, info.hasDupTs, info.partiallyRequired,
2174
              info.overlapWithKeyInBuf, info.moreThanCapcity, info.overlapWithDelInfo, info.overlapWithSttBlock,
2175
              pReader->idStr);
2176
  }
2177

2178
_end:
13,409,205✔
2179
  if (code != TSDB_CODE_SUCCESS) {
13,409,637!
2180
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2181
  }
2182
  return code;
13,409,712✔
2183
}
2184

2185
static int32_t isCleanFileDataBlock(STsdbReader* pReader, SFileDataBlockInfo* pBlockInfo,
50,643✔
2186
                                    STableBlockScanInfo* pScanInfo, TSDBKEY keyInBuf, bool* res) {
2187
  int32_t              code = TSDB_CODE_SUCCESS;
50,643✔
2188
  int32_t              lino = 0;
50,643✔
2189
  SDataBlockToLoadInfo info = {0};
50,643✔
2190

2191
  TSDB_CHECK_NULL(res, code, lino, _end, TSDB_CODE_INVALID_PARA);
50,643!
2192

2193
  *res = false;
50,643✔
2194
  code = getBlockToLoadInfo(&info, pBlockInfo, pScanInfo, keyInBuf, pReader);
50,643✔
2195
  TSDB_CHECK_CODE(code, lino, _end);
50,642!
2196

2197
  *res = !(info.overlapWithNeighborBlock || info.hasDupTs || info.overlapWithKeyInBuf || info.overlapWithDelInfo ||
60,051!
2198
           info.overlapWithSttBlock);
9,409✔
2199

2200
_end:
50,642✔
2201
  if (code != TSDB_CODE_SUCCESS) {
50,642!
2202
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2203
  }
2204
  return code;
50,644✔
2205
}
2206

2207
static int32_t initRowMergeIfNeeded(STsdbReader* pReader, int64_t uid) {
972,416,453✔
2208
  int32_t     code = TSDB_CODE_SUCCESS;
972,416,453✔
2209
  int32_t     lino = 0;
972,416,453✔
2210
  SRowMerger* pMerger = NULL;
972,416,453✔
2211
  STSchema*   ps = NULL;
972,416,453✔
2212

2213
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
972,416,453!
2214

2215
  pMerger = &pReader->status.merger;
972,416,453✔
2216

2217
  if (pMerger->pArray == NULL) {
972,416,453✔
2218
    ps = getTableSchemaImpl(pReader, uid);
106!
2219
    TSDB_CHECK_NULL(ps, code, lino, _end, terrno);
106✔
2220
  }
2221

2222
_end:
972,416,447✔
2223
  if (code != TSDB_CODE_SUCCESS) {
972,416,453✔
2224
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
6!
2225
  }
2226
  return code;
972,115,045✔
2227
}
2228

2229
static int32_t buildDataBlockFromBuf(STsdbReader* pReader, STableBlockScanInfo* pBlockScanInfo, int64_t endKey) {
32,048,099✔
2230
  int32_t      code = TSDB_CODE_SUCCESS;
32,048,099✔
2231
  int32_t      lino = 0;
32,048,099✔
2232
  int64_t      st = 0;
32,048,099✔
2233
  SSDataBlock* pBlock = NULL;
32,048,099✔
2234

2235
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
32,048,099!
2236

2237
  if (!(pBlockScanInfo->iiter.hasVal || pBlockScanInfo->iter.hasVal)) {
32,048,099✔
2238
    goto _end;
23,212,028✔
2239
  }
2240

2241
  code = initRowMergeIfNeeded(pReader, pBlockScanInfo->uid);
8,836,071✔
2242
  TSDB_CHECK_CODE(code, lino, _end);
8,842,598✔
2243

2244
  st = taosGetTimestampUs();
8,843,920✔
2245
  pBlock = pReader->resBlockInfo.pResBlock;
8,843,920✔
2246
  code = buildDataBlockFromBufImpl(pBlockScanInfo, endKey, pReader->resBlockInfo.capacity, pReader);
8,843,920✔
2247
  TSDB_CHECK_CODE(code, lino, _end);
8,842,939✔
2248

2249
  double el = (taosGetTimestampUs() - st) / 1000.0;
8,843,688✔
2250
  updateComposedBlockInfo(pReader, el, pBlockScanInfo);
8,843,688✔
2251

2252
  tsdbDebug("%p build data block from cache completed, elapsed time:%.2f ms, numOfRows:%" PRId64 ", brange:%" PRId64
8,843,961✔
2253
            " - %" PRId64 ", uid:%" PRIu64 ",  %s",
2254
            pReader, el, pBlock->info.rows, pBlock->info.window.skey, pBlock->info.window.ekey, pBlockScanInfo->uid,
2255
            pReader->idStr);
2256

2257
  pReader->cost.buildmemBlock += el;
8,844,042✔
2258

2259
_end:
32,056,080✔
2260
  if (code != TSDB_CODE_SUCCESS) {
32,056,080✔
2261
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
10!
2262
  }
2263
  return code;
32,052,163✔
2264
}
2265

2266
static int32_t tryCopyDistinctRowFromFileBlock(STsdbReader* pReader, SBlockData* pBlockData, SRowKey* pKey,
41,312,586✔
2267
                                               SFileBlockDumpInfo* pDumpInfo, bool* copied) {
2268
  // opt version
2269
  // 1. it is not a border point
2270
  // 2. the direct next point is not an duplicated timestamp
2271
  int32_t code = TSDB_CODE_SUCCESS;
41,312,586✔
2272
  int32_t lino = 0;
41,312,586✔
2273
  bool    asc = false;
41,312,586✔
2274

2275
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
41,312,586!
2276
  TSDB_CHECK_NULL(pDumpInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
41,312,586!
2277
  TSDB_CHECK_NULL(copied, code, lino, _end, TSDB_CODE_INVALID_PARA);
41,312,586!
2278

2279
  *copied = false;
41,312,586✔
2280
  asc = (pReader->info.order == TSDB_ORDER_ASC);
41,312,586✔
2281
  if ((pDumpInfo->rowIndex < pDumpInfo->totalRows - 1 && asc) || (pDumpInfo->rowIndex > 0 && (!asc))) {
41,312,586!
2282
    int32_t step = ASCENDING_TRAVERSE(pReader->info.order) ? 1 : -1;
41,298,338✔
2283

2284
    SRowKey nextRowKey;
2285
    TSDB_CHECK_NULL(pBlockData, code, lino, _end, TSDB_CODE_INVALID_PARA);
41,298,338!
2286
    tColRowGetKey(pBlockData, pDumpInfo->rowIndex + step, &nextRowKey);
41,298,338✔
2287

2288
    if (pkCompEx(pKey, &nextRowKey) != 0) {  // merge is not needed
41,594,747✔
2289
      code = doAppendRowFromFileBlock(pReader->resBlockInfo.pResBlock, pReader, pBlockData, pDumpInfo->rowIndex);
41,592,573✔
2290
      TSDB_CHECK_CODE(code, lino, _end);
41,355,907!
2291
      pDumpInfo->rowIndex += step;
41,355,907✔
2292
      *copied = true;
41,355,907✔
2293
    }
2294
  }
2295

2296
_end:
14,248✔
2297
  if (code != TSDB_CODE_SUCCESS) {
41,372,329!
2298
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2299
  }
2300
  return code;
41,364,850✔
2301
}
2302

2303
static int32_t nextRowFromSttBlocks(SSttBlockReader* pSttBlockReader, STableBlockScanInfo* pScanInfo, int32_t pkSrcSlot,
620,525,452✔
2304
                                    SVersionRange* pVerRange) {
2305
  int32_t  code = TSDB_CODE_SUCCESS;
620,525,452✔
2306
  int32_t  lino = 0;
620,525,452✔
2307
  int32_t  order = 0;
620,525,452✔
2308
  int32_t  step = 0;
620,525,452✔
2309
  SRowKey* pNextProc = NULL;
620,525,452✔
2310

2311
  TSDB_CHECK_NULL(pSttBlockReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
620,525,452!
2312
  TSDB_CHECK_NULL(pScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
620,525,452!
2313

2314
  order = pSttBlockReader->order;
620,525,452✔
2315
  step = ASCENDING_TRAVERSE(order) ? 1 : -1;
620,525,452✔
2316
  pNextProc = &pScanInfo->sttKeyInfo.nextProcKey;
620,525,452✔
2317

2318
  while (1) {
5,644,921✔
2319
    bool hasVal = false;
626,170,373✔
2320
    code = tMergeTreeNext(&pSttBlockReader->mergeTree, &hasVal);
626,170,373✔
2321
    if (code) {
621,799,031!
2322
      tsdbError("failed to iter the next row in stt-file merge tree, code:%s, %s", tstrerror(code),
×
2323
                pSttBlockReader->mergeTree.idStr);
2324
      TSDB_CHECK_CODE(code, lino, _end);
613,052,653!
2325
    }
2326

2327
    if (!hasVal) {  // the next value will be the accessed key in stt
621,982,818✔
2328
      pScanInfo->sttKeyInfo.status = STT_FILE_NO_DATA;
20,499,123✔
2329

2330
      // next file, the timestamps in the next file must be greater than those in current
2331
      pNextProc->ts += step;
20,499,123✔
2332
      if (pSttBlockReader->numOfPks > 0) {
20,499,123✔
2333
        if (IS_NUMERIC_TYPE(pNextProc->pks[0].type)) {
1,335,514!
2334
          VALUE_SET_TRIVIAL_DATUM(pNextProc->pks, INT64_MAX);
992,695✔
2335
        } else {
2336
          memset(pNextProc->pks[0].pData, 0, pNextProc->pks[0].nData);
342,819✔
2337
        }
2338
      }
2339
      goto _end;
20,499,123✔
2340
    }
2341

2342
    TSDBROW* pRow = tMergeTreeGetRow(&pSttBlockReader->mergeTree);
601,483,695✔
2343
    int64_t  key = pRow->pBlockData->aTSKEY[pRow->iRow];
601,483,695✔
2344
    int64_t  ver = pRow->pBlockData->aVersion[pRow->iRow];
601,483,695✔
2345

2346
    if (pSttBlockReader->numOfPks == 0) {
601,483,695✔
2347
      pSttBlockReader->currentKey.ts = key;
553,323,859✔
2348
    } else {
2349
      tColRowGetKeyDeepCopy(pRow->pBlockData, pRow->iRow, pkSrcSlot, &pSttBlockReader->currentKey);
48,159,836!
2350
    }
2351

2352
    tColRowGetKeyDeepCopy(pRow->pBlockData, pRow->iRow, pkSrcSlot, pNextProc);
601,438,897!
2353

2354
    if (pScanInfo->delSkyline != NULL && TARRAY_SIZE(pScanInfo->delSkyline) > 0) {
607,499,268!
2355
      bool dropped = false;
47,982,583✔
2356
      code = hasBeenDropped(pScanInfo->delSkyline, &pScanInfo->sttBlockDelIndex, key, ver, order, pVerRange,
47,982,583✔
2357
                            pSttBlockReader->numOfPks > 0, &dropped);
47,982,583✔
2358
      TSDB_CHECK_CODE(code, lino, _end);
82,640,879!
2359
      if (!dropped) {
44,142,900✔
2360
        pScanInfo->sttKeyInfo.status = STT_FILE_HAS_DATA;
38,497,979✔
2361
        goto _end;
38,497,979✔
2362
      }
2363
    } else {
2364
      pScanInfo->sttKeyInfo.status = STT_FILE_HAS_DATA;
553,871,764✔
2365
      goto _end;
553,871,764✔
2366
    }
2367
  }
2368

2369
_end:
612,868,866✔
2370
  if (code != TSDB_CODE_SUCCESS) {
612,868,866!
2371
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2372
  }
2373
  return code;
612,494,878✔
2374
}
2375

2376
static void doPinSttBlock(SSttBlockReader* pSttBlockReader) { tMergeTreePinSttBlock(&pSttBlockReader->mergeTree); }
334,195,407✔
2377

2378
static void doUnpinSttBlock(SSttBlockReader* pSttBlockReader) { tMergeTreeUnpinSttBlock(&pSttBlockReader->mergeTree); }
329,347,904✔
2379

2380
static int32_t tryCopyDistinctRowFromSttBlock(TSDBROW* fRow, SSttBlockReader* pSttBlockReader,
335,031,130✔
2381
                                              STableBlockScanInfo* pScanInfo, SRowKey* pSttKey, STsdbReader* pReader,
2382
                                              bool* copied) {
2383
  int32_t  code = TSDB_CODE_SUCCESS;
335,031,130✔
2384
  int32_t  lino = 0;
335,031,130✔
2385
  SRowKey* pNext = NULL;
335,031,130✔
2386

2387
  TSDB_CHECK_NULL(pSttBlockReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
335,031,130!
2388
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
335,031,130!
2389
  TSDB_CHECK_NULL(copied, code, lino, _end, TSDB_CODE_INVALID_PARA);
335,031,130!
2390

2391
  *copied = false;
335,031,130✔
2392

2393
  // avoid the fetch next row replace the referenced stt block in buffer
2394
  doPinSttBlock(pSttBlockReader);
335,031,130✔
2395
  code = nextRowFromSttBlocks(pSttBlockReader, pScanInfo, pReader->suppInfo.pkSrcSlot, &pReader->info.verRange);
331,292,024✔
2396
  doUnpinSttBlock(pSttBlockReader);
329,686,331✔
2397
  TSDB_CHECK_CODE(code, lino, _end);
328,176,499!
2398

2399
  if (hasDataInSttBlock(pScanInfo)) {
328,176,499✔
2400
    pNext = getCurrentKeyInSttBlock(pSttBlockReader);
325,816,000!
2401
    if (pkCompEx(pSttKey, pNext) != 0) {
328,436,585✔
2402
      code = doAppendRowFromFileBlock(pReader->resBlockInfo.pResBlock, pReader, fRow->pBlockData, fRow->iRow);
323,588,511✔
2403
      *copied = (code == TSDB_CODE_SUCCESS);
331,079,368✔
2404
      TSDB_CHECK_CODE(code, lino, _end);
331,079,368!
2405
    }
2406
  } else {
2407
    code = doAppendRowFromFileBlock(pReader->resBlockInfo.pResBlock, pReader, fRow->pBlockData, fRow->iRow);
2,370,223✔
2408
    *copied = (code == TSDB_CODE_SUCCESS);
2,483,336✔
2409
    TSDB_CHECK_CODE(code, lino, _end);
2,483,336!
2410
  }
2411

2412
_end:
2,483,336✔
2413
  if (code != TSDB_CODE_SUCCESS) {
338,410,778!
2414
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2415
  }
2416
  return code;
334,723,078✔
2417
}
2418

2419
static FORCE_INLINE STSchema* doGetSchemaForTSRow(int32_t sversion, STsdbReader* pReader, uint64_t uid) {
2420
  int32_t   code = TSDB_CODE_SUCCESS;
2,147,483,647✔
2421
  int32_t   lino = 0;
2,147,483,647✔
2422
  STSchema* ps = NULL;
2,147,483,647✔
2423
  void**    p = NULL;
2,147,483,647✔
2424

2425
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
547,102,714!
2426

2427
  // always set the newest schema version in pReader->info.pSchema
2428
  if (pReader->info.pSchema == NULL) {
2,147,483,647!
2429
    ps = getTableSchemaImpl(pReader, uid);
×
2430
    TSDB_CHECK_NULL(ps, code, lino, _end, terrno);
×
2431
  }
2432

2433
  if (pReader->info.pSchema && sversion == pReader->info.pSchema->version) {
2,147,483,647!
2434
    ps = pReader->info.pSchema;
2,147,483,647✔
2435
    goto _end;
2,147,483,647✔
2436
  }
2437

2438
  p = tSimpleHashGet(pReader->pSchemaMap, &sversion, sizeof(sversion));
23,358,205✔
2439
  if (p != NULL) {
25,991,307!
2440
    ps = *(STSchema**)p;
25,913,061✔
2441
    goto _end;
25,913,061✔
2442
  }
2443

2444
  code = metaGetTbTSchemaEx(pReader->pTsdb->pVnode->pMeta, pReader->info.suid, uid, sversion, &ps);
78,246✔
2445
  TSDB_CHECK_CODE(code, lino, _end);
85,389!
2446

2447
  code = tSimpleHashPut(pReader->pSchemaMap, &sversion, sizeof(sversion), &ps, POINTER_BYTES);
85,389✔
2448
  TSDB_CHECK_CODE(code, lino, _end);
85,389!
2449

2450
_end:
85,389✔
2451
  if (code != TSDB_CODE_SUCCESS) {
2,147,483,647!
2452
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2453
    terrno = code;
×
2454
    return NULL;
×
2455
  }
2456
  return ps;
2,147,483,647✔
2457
}
2458

2459
// deep copy the primary key, to avoid ref invalid buffer if the primary key is varchar or something.
2460
static int32_t primaryKeyDup(SRowKey* pKey) {
68,136,730✔
2461
  if (pKey->numOfPKs > 0) {
68,136,730✔
2462
    if (!IS_NUMERIC_TYPE(pKey->pks[0].type)) {
12,229,163!
2463
      void* p = taosMemoryMalloc(pKey->pks[0].nData);
4,096!
2464
      if (p == NULL)  {
4,096!
2465
        tsdbError("%s failed at %s:%d since %s", __func__, __FILE__, __LINE__, tstrerror(terrno));
×
2466
        // memory allocation failed, set pData to NULL to avoid freeing original buffer
2467
        pKey->pks[0].pData = NULL;
×
2468
        return terrno;
×
2469
      }
2470

2471
      TAOS_MEMCPY(p, pKey->pks[0].pData, pKey->pks[0].nData);
4,096✔
2472
      pKey->pks[0].pData = p;
4,096✔
2473
    }
2474
  }
2475

2476
  return 0;
68,136,730✔
2477
}
2478

2479
static int32_t doMergeBufAndFileRows(STsdbReader* pReader, STableBlockScanInfo* pBlockScanInfo, TSDBROW* pRow,
540,771,972✔
2480
                                     SIterInfo* pIter, SSttBlockReader* pSttBlockReader) {
2481
  int32_t             code = TSDB_CODE_SUCCESS;
540,771,972✔
2482
  int32_t             lino = 0;
540,771,972✔
2483
  SRowMerger*         pMerger = NULL;
540,771,972✔
2484
  SRow*               pTSRow = NULL;
540,771,972✔
2485
  SBlockData*         pBlockData = NULL;
540,771,972✔
2486
  SFileBlockDumpInfo* pDumpInfo = NULL;
540,771,972✔
2487
  SRowKey*            pSttKey = NULL;
540,771,972✔
2488
  int32_t             pkSrcSlot = 0;
540,771,972✔
2489
  SRowKey             k = {0};
540,771,972✔
2490
  STSchema*           pSchema = NULL;
540,771,972✔
2491

2492
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
540,771,972!
2493
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
540,771,972!
2494
  TSDB_CHECK_NULL(pRow, code, lino, _end, TSDB_CODE_INVALID_PARA);
540,771,972!
2495

2496
  pMerger = &pReader->status.merger;
540,771,972✔
2497
  pMerger->arg = pReader->pTsdb->pVnode->pBse;
540,771,972✔
2498

2499
  pBlockData = &pReader->status.fileBlockData;
540,771,972✔
2500
  pDumpInfo = &pReader->status.fBlockDumpInfo;
540,771,972✔
2501
  pkSrcSlot = pReader->suppInfo.pkSrcSlot;
540,771,972✔
2502

2503
  if (hasDataInSttBlock(pBlockScanInfo) && (!pBlockScanInfo->cleanSttBlocks)) {
540,771,972!
2504
    pSttKey = getCurrentKeyInSttBlock(pSttBlockReader);
529,332,271✔
2505
  }
2506

2507
  tRowGetKeyEx(pRow, &k);
540,776,795!
2508

2509
  if (pRow->type == TSDBROW_ROW_FMT) {
540,952,183!
2510
    pSchema = doGetSchemaForTSRow(TSDBROW_SVERSION(pRow), pReader, pBlockScanInfo->uid);
541,381,858✔
2511
    TSDB_CHECK_NULL(pSchema, code, lino, _end, terrno);
541,107,649!
2512
  }
2513

2514
  SRowKey* pfKey = &(SRowKey){0};
540,677,974✔
2515
  if (hasDataInFileBlock(pBlockData, pDumpInfo)) {
540,677,974✔
2516
    tColRowGetKey(pBlockData, pDumpInfo->rowIndex, pfKey);
13,202,070✔
2517
    code = primaryKeyDup(pfKey);
13,203,077✔
2518
    TSDB_CHECK_CODE(code, lino, _end);
13,171,765!
2519
  } else {
2520
    pfKey = NULL;
527,744,488✔
2521
  }
2522

2523
  TSDBROW fRow = tsdbRowFromBlockData(pBlockData, pDumpInfo->rowIndex);
540,916,253✔
2524

2525
  // merge is not initialized yet, due to the fact that the pReader->info.pSchema is not initialized
2526
  code = initRowMergeIfNeeded(pReader, pBlockScanInfo->uid);
540,916,253✔
2527
  TSDB_CHECK_CODE(code, lino, _end);
540,716,671!
2528

2529
  SRowKey minKey = k;
540,716,671✔
2530
  if (pReader->info.order == TSDB_ORDER_ASC) {
540,716,671✔
2531
    if (pfKey != NULL && pkCompEx(pfKey, &minKey) < 0) {
486,364,662✔
2532
      minKey = *pfKey;
8,882,437✔
2533
    }
2534

2535
    if (pSttKey != NULL && pkCompEx(pSttKey, &minKey) < 0) {
935,984,282✔
2536
      minKey = *pSttKey;
227,038,753✔
2537
    }
2538
  } else {
2539
    if (pfKey != NULL && pkCompEx(pfKey, &minKey) > 0) {
67,532,017✔
2540
      minKey = *pfKey;
91,776✔
2541
    }
2542

2543
    if (pSttKey != NULL && pkCompEx(pSttKey, &minKey) > 0) {
134,471,502✔
2544
      minKey = *pSttKey;
26,343,921✔
2545
    }
2546
  }
2547

2548
  // copy the last key before the time of stt reader loading the next stt block, in which the underlying data block may
2549
  // be changed, resulting in the corresponding changing of the value of sttRowKey
2550
  tRowKeyAssign(&pBlockScanInfo->lastProcKey, &minKey);
541,103,373✔
2551

2552
  // file block ---> stt block -----> mem
2553
  if (pkCompEx(&pBlockScanInfo->lastProcKey, pfKey) == 0) {
1,081,215,176✔
2554
    code = tsdbRowMergerAdd(pMerger, &fRow, NULL);
13,023,955✔
2555
    TSDB_CHECK_CODE(code, lino, _end);
13,060,885!
2556

2557
    code = doMergeRowsInFileBlocks(pBlockData, pBlockScanInfo, pfKey, pReader);
13,060,885✔
2558
    TSDB_CHECK_CODE(code, lino, _end);
13,052,648!
2559
  }
2560

2561
  if (pkCompEx(&pBlockScanInfo->lastProcKey, pSttKey) == 0) {
1,081,907,627✔
2562
    TSDBROW* fRow1 = tMergeTreeGetRow(&pSttBlockReader->mergeTree);
257,369,833✔
2563
    code = tsdbRowMergerAdd(pMerger, fRow1, NULL);
257,369,833✔
2564
    TSDB_CHECK_CODE(code, lino, _end);
258,400,007!
2565
    code = doMergeRowsInSttBlock(pSttBlockReader, pBlockScanInfo, pMerger, pkSrcSlot, &pReader->info.verRange,
258,400,007✔
2566
                                 pReader->idStr);
258,400,007✔
2567
    TSDB_CHECK_CODE(code, lino, _end);
257,620,160!
2568
  }
2569

2570
  if (pkCompEx(&pBlockScanInfo->lastProcKey, &k) == 0) {
1,082,899,225!
2571
    code = tsdbRowMergerAdd(pMerger, pRow, pSchema);
278,847,616✔
2572
    TSDB_CHECK_CODE(code, lino, _end);
278,405,201!
2573

2574
    code = doMergeRowsInBuf(pIter, pBlockScanInfo->uid, &k, pBlockScanInfo->delSkyline, pReader);
278,405,201✔
2575
    TSDB_CHECK_CODE(code, lino, _end);
278,720,453!
2576
  }
2577

2578
  code = tsdbRowMergerGetRow(pMerger, &pTSRow);
541,248,792✔
2579
  TSDB_CHECK_CODE(code, lino, _end);
543,052,519!
2580

2581
  code = doAppendRowFromTSRow(pReader->resBlockInfo.pResBlock, pReader, pTSRow, pBlockScanInfo);
543,052,519✔
2582

2583
  taosMemoryFreeClear(pTSRow);
534,600,184!
2584
  tsdbRowMergerClear(pMerger);
542,552,230✔
2585
  TSDB_CHECK_CODE(code, lino, _end);
536,097,365!
2586

2587
_end:
536,097,365✔
2588
  clearRowKey(pfKey);
536,097,365✔
2589
  if (code != TSDB_CODE_SUCCESS) {
537,484,261!
2590
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2591
  }
2592
  return code;
537,496,470✔
2593
}
2594

2595
static int32_t mergeFileBlockAndSttBlock(STsdbReader* pReader, SSttBlockReader* pSttBlockReader, SRowKey* pKey,
378,982,303✔
2596
                                         STableBlockScanInfo* pBlockScanInfo, SBlockData* pBlockData) {
2597
  int32_t             code = TSDB_CODE_SUCCESS;
378,982,303✔
2598
  int32_t             lino = 0;
378,982,303✔
2599
  SFileBlockDumpInfo* pDumpInfo = NULL;
378,982,303✔
2600
  SRowMerger*         pMerger = NULL;
378,982,303✔
2601
  SRow*               pTSRow = NULL;
378,982,303✔
2602
  int32_t             pkSrcSlot = 0;
378,982,303✔
2603

2604
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
378,982,303!
2605
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
378,982,303!
2606
  TSDB_CHECK_NULL(pBlockData, code, lino, _end, TSDB_CODE_INVALID_PARA);
378,982,303!
2607

2608
  pDumpInfo = &pReader->status.fBlockDumpInfo;
378,982,303✔
2609
  pMerger = &pReader->status.merger;
378,982,303✔
2610
  pMerger->arg = pReader->pTsdb->pVnode->pBse;
378,982,303✔
2611

2612
  pkSrcSlot = pReader->suppInfo.pkSrcSlot;
378,982,303✔
2613

2614
  // merge is not initialized yet, due to the fact that the pReader->info.pSchema is not initialized
2615
  code = initRowMergeIfNeeded(pReader, pBlockScanInfo->uid);
378,982,303✔
2616
  TSDB_CHECK_CODE(code, lino, _end);
380,461,529!
2617

2618
  bool dataInDataFile = hasDataInFileBlock(pBlockData, pDumpInfo);
380,461,529✔
2619
  bool dataInSttFile = hasDataInSttBlock(pBlockScanInfo);
376,946,000✔
2620

2621
  if (dataInDataFile && (!dataInSttFile)) {
376,133,563✔
2622
    // no stt file block available, only data block exists
2623
    code = mergeRowsInFileBlocks(pBlockData, pBlockScanInfo, pKey, pReader);
22,056,630✔
2624
    TSDB_CHECK_CODE(code, lino, _end);
22,178,674!
2625
  } else if ((!dataInDataFile) && dataInSttFile) {
354,076,933!
2626
    // no data in data file exists
2627
    code = mergeRowsInSttBlocks(pSttBlockReader, pBlockScanInfo, pReader);
334,203,220✔
2628
    TSDB_CHECK_CODE(code, lino, _end);
334,448,783!
2629
  } else if (pBlockScanInfo->cleanSttBlocks && pReader->info.execMode == READER_EXEC_ROWS) {
19,873,713!
2630
    // opt model for count data in stt file, which is not overlap with data blocks in files.
2631
    code = mergeRowsInFileBlocks(pBlockData, pBlockScanInfo, pKey, pReader);
×
2632
    TSDB_CHECK_CODE(code, lino, _end);
×
2633
  } else {
2634
    TSDB_CHECK_NULL(pKey, code, lino, _end, TSDB_CODE_INVALID_PARA);
38,933,784!
2635
    // row in both stt file blocks and data file blocks
2636
    TSDBROW  fRow = tsdbRowFromBlockData(pBlockData, pDumpInfo->rowIndex);
19,873,713✔
2637
    SRowKey* pSttKey = getCurrentKeyInSttBlock(pSttBlockReader);
19,873,713!
2638

2639
    int32_t ret = pkCompEx(pKey, pSttKey);
19,878,948✔
2640

2641
    if (ASCENDING_TRAVERSE(pReader->info.order)) {
19,878,948✔
2642
      if (ret < 0) {  // asc
19,735,905✔
2643
        code = mergeRowsInFileBlocks(pBlockData, pBlockScanInfo, pKey, pReader);
19,187,588✔
2644
        TSDB_CHECK_CODE(code, lino, _end);
18,957,048!
2645
      } else if (ret > 0) {
548,317✔
2646
        code = mergeRowsInSttBlocks(pSttBlockReader, pBlockScanInfo, pReader);
25,688✔
2647
        TSDB_CHECK_CODE(code, lino, _end);
25,688!
2648
      }
2649
    } else {  // desc
2650
      if (ret > 0) {
143,043✔
2651
        code = mergeRowsInFileBlocks(pBlockData, pBlockScanInfo, pKey, pReader);
73,568✔
2652
        TSDB_CHECK_CODE(code, lino, _end);
73,559!
2653
      } else if (ret < 0) {
69,475✔
2654
        code = mergeRowsInSttBlocks(pSttBlockReader, pBlockScanInfo, pReader);
1,996✔
2655
        TSDB_CHECK_CODE(code, lino, _end);
1,998!
2656
      }
2657
    }
2658
    if (ret != 0) {
19,648,401✔
2659
      goto _end;
19,060,071✔
2660
    }
2661

2662
    // pKey == pSttKey
2663
    tRowKeyAssign(&pBlockScanInfo->lastProcKey, pKey);
588,330✔
2664

2665
    // the following for key == sttKey->key.ts
2666
    // file block ------> stt  block
2667

2668
    code = tsdbRowMergerAdd(pMerger, &fRow, NULL);
588,282✔
2669
    TSDB_CHECK_CODE(code, lino, _end);
589,923!
2670

2671
    code = doMergeRowsInFileBlocks(pBlockData, pBlockScanInfo, pKey, pReader);
589,923✔
2672
    TSDB_CHECK_CODE(code, lino, _end);
589,591!
2673

2674
    TSDBROW* pRow1 = tMergeTreeGetRow(&pSttBlockReader->mergeTree);
589,591✔
2675
    code = tsdbRowMergerAdd(pMerger, pRow1, NULL);
589,591✔
2676
    TSDB_CHECK_CODE(code, lino, _end);
585,517!
2677

2678
    // pSttKey will be changed when sttBlockReader iterates to the next row, so use pKey instead.
2679
    code = doMergeRowsInSttBlock(pSttBlockReader, pBlockScanInfo, pMerger, pkSrcSlot, &pReader->info.verRange,
585,517✔
2680
                                 pReader->idStr);
585,517✔
2681
    TSDB_CHECK_CODE(code, lino, _end);
588,570!
2682

2683
    code = tsdbRowMergerGetRow(pMerger, &pTSRow);
588,570✔
2684
    TSDB_CHECK_CODE(code, lino, _end);
591,544!
2685

2686
    code = doAppendRowFromTSRow(pReader->resBlockInfo.pResBlock, pReader, pTSRow, pBlockScanInfo);
591,544✔
2687

2688
    taosMemoryFreeClear(pTSRow);
590,687!
2689
    tsdbRowMergerClear(pMerger);
591,638✔
2690
    TSDB_CHECK_CODE(code, lino, _end);
586,789!
2691
  }
2692

2693
_end:
376,274,317✔
2694
  if (code != TSDB_CODE_SUCCESS) {
376,274,317!
2695
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2696
  }
2697
  return code;
376,144,627✔
2698
}
2699

2700
static int32_t doMergeMultiLevelRows(STsdbReader* pReader, STableBlockScanInfo* pBlockScanInfo, SBlockData* pBlockData,
2,130,727✔
2701
                                     SSttBlockReader* pSttBlockReader) {
2702
  int32_t             code = TSDB_CODE_SUCCESS;
2,130,727✔
2703
  int32_t             lino = 0;
2,130,727✔
2704
  SRowMerger*         pMerger = NULL;
2,130,727✔
2705
  SRow*               pTSRow = NULL;
2,130,727✔
2706
  SFileBlockDumpInfo* pDumpInfo = NULL;
2,130,727✔
2707
  SArray*             pDelList = NULL;
2,130,727✔
2708
  int32_t             pkSrcSlot = 0;
2,130,727✔
2709
  TSDBROW*            pRow = NULL;
2,130,727✔
2710
  TSDBROW*            piRow = NULL;
2,130,727✔
2711
  SRowKey*            pSttKey = NULL;
2,130,727✔
2712

2713
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,130,727!
2714
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,130,727!
2715
  TSDB_CHECK_NULL(pBlockData, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,130,727!
2716

2717
  pMerger = &pReader->status.merger;
2,130,727✔
2718
  pDumpInfo = &pReader->status.fBlockDumpInfo;
2,130,727✔
2719
  pDelList = pBlockScanInfo->delSkyline;
2,130,727✔
2720
  pkSrcSlot = pReader->suppInfo.pkSrcSlot;
2,130,727✔
2721

2722
  code = getValidMemRow(&pBlockScanInfo->iter, pDelList, pReader, &pRow);
4,261,454✔
2723
  TSDB_CHECK_CODE(code, lino, _end);
2,130,727!
2724
  code = getValidMemRow(&pBlockScanInfo->iiter, pDelList, pReader, &piRow);
4,261,454✔
2725
  TSDB_CHECK_CODE(code, lino, _end);
2,130,727!
2726

2727
  if (hasDataInSttBlock(pBlockScanInfo) && (!pBlockScanInfo->cleanSttBlocks)) {
2,130,727!
2728
    pSttKey = getCurrentKeyInSttBlock(pSttBlockReader);
2,130,727✔
2729
  }
2730

2731
  SRowKey* pfKey = &(SRowKey){0};
2,130,727✔
2732
  if (hasDataInFileBlock(pBlockData, pDumpInfo)) {
2,130,727!
2733
    tColRowGetKey(pBlockData, pDumpInfo->rowIndex, pfKey);
×
2734
  } else {
2735
    pfKey = NULL;
2,130,727✔
2736
  }
2737

2738
  SRowKey k = {0}, ik = {0};
2,130,727✔
2739
  tRowGetKeyEx(pRow, &k);
2,130,727!
2740
  tRowGetKeyEx(piRow, &ik);
2,130,727!
2741

2742
  STSchema* pSchema = NULL;
2,130,727✔
2743
  if (pRow->type == TSDBROW_ROW_FMT) {
2,130,727!
2744
    pSchema = doGetSchemaForTSRow(TSDBROW_SVERSION(pRow), pReader, pBlockScanInfo->uid);
2,130,727!
2745
    TSDB_CHECK_NULL(pSchema, code, lino, _end, terrno);
2,130,727!
2746
  }
2747

2748
  STSchema* piSchema = NULL;
2,130,727✔
2749
  if (piRow->type == TSDBROW_ROW_FMT) {
2,130,727!
2750
    piSchema = doGetSchemaForTSRow(TSDBROW_SVERSION(piRow), pReader, pBlockScanInfo->uid);
2,130,727!
2751
    TSDB_CHECK_NULL(pSchema, code, lino, _end, terrno);
2,130,727!
2752
  }
2753

2754
  // merge is not initialized yet, due to the fact that the pReader->info.pSchema is not initialized
2755
  code = initRowMergeIfNeeded(pReader, pBlockScanInfo->uid);
2,130,727✔
2756
  TSDB_CHECK_CODE(code, lino, _end);
2,130,727!
2757

2758
  SRowKey minKey = k;
2,130,727✔
2759
  if (ASCENDING_TRAVERSE(pReader->info.order)) {
2,130,727✔
2760
    if (pkCompEx(&ik, &minKey) < 0) {  // minKey > ik.key.ts) {
2,130,725✔
2761
      minKey = ik;
2,128,452✔
2762
    }
2763

2764
    if ((pfKey != NULL) && (pkCompEx(pfKey, &minKey) < 0)) {
2,130,725!
2765
      minKey = *pfKey;
×
2766
    }
2767

2768
    if ((pSttKey != NULL) && (pkCompEx(pSttKey, &minKey) < 0)) {
4,261,450!
2769
      minKey = *pSttKey;
2,124,976✔
2770
    }
2771
  } else {
2772
    if (pkCompEx(&ik, &minKey) > 0) {
2!
2773
      minKey = ik;
×
2774
    }
2775

2776
    if ((pfKey != NULL) && (pkCompEx(pfKey, &minKey) > 0)) {
2!
2777
      minKey = *pfKey;
×
2778
    }
2779

2780
    if ((pSttKey != NULL) && (pkCompEx(pSttKey, &minKey) > 0)) {
4!
2781
      minKey = *pSttKey;
×
2782
    }
2783
  }
2784

2785
  tRowKeyAssign(&pBlockScanInfo->lastProcKey, &minKey);
2,130,727✔
2786

2787
  // file block -----> stt block -----> imem -----> mem
2788
  if (pkCompEx(&pBlockScanInfo->lastProcKey, pfKey) == 0) {
4,261,454!
2789
    TSDBROW fRow = tsdbRowFromBlockData(pBlockData, pDumpInfo->rowIndex);
×
2790
    code = tsdbRowMergerAdd(pMerger, &fRow, NULL);
×
2791
    TSDB_CHECK_CODE(code, lino, _end);
×
2792

2793
    code = doMergeRowsInFileBlocks(pBlockData, pBlockScanInfo, pfKey, pReader);
×
2794
    TSDB_CHECK_CODE(code, lino, _end);
×
2795
  }
2796

2797
  if (pkCompEx(&pBlockScanInfo->lastProcKey, pSttKey) == 0) {
4,261,454!
2798
    TSDBROW* pRow1 = tMergeTreeGetRow(&pSttBlockReader->mergeTree);
2,130,587✔
2799
    code = tsdbRowMergerAdd(pMerger, pRow1, NULL);
2,130,587✔
2800
    TSDB_CHECK_CODE(code, lino, _end);
2,130,587!
2801

2802
    code = doMergeRowsInSttBlock(pSttBlockReader, pBlockScanInfo, pMerger, pkSrcSlot, &pReader->info.verRange,
2,130,587✔
2803
                                 pReader->idStr);
2,130,587✔
2804
    TSDB_CHECK_CODE(code, lino, _end);
2,130,587!
2805
  }
2806

2807
  if (pkCompEx(&pBlockScanInfo->lastProcKey, &ik) == 0) {
4,261,454!
2808
    code = tsdbRowMergerAdd(pMerger, piRow, piSchema);
5,750✔
2809
    TSDB_CHECK_CODE(code, lino, _end);
5,750!
2810

2811
    code = doMergeRowsInBuf(&pBlockScanInfo->iiter, pBlockScanInfo->uid, &ik, pBlockScanInfo->delSkyline, pReader);
5,750✔
2812
    TSDB_CHECK_CODE(code, lino, _end);
5,750!
2813
  }
2814

2815
  if (pkCompEx(&pBlockScanInfo->lastProcKey, &k) == 0) {
4,261,454!
2816
    code = tsdbRowMergerAdd(pMerger, pRow, pSchema);
2,275✔
2817
    TSDB_CHECK_CODE(code, lino, _end);
2,275!
2818

2819
    code = doMergeRowsInBuf(&pBlockScanInfo->iter, pBlockScanInfo->uid, &k, pBlockScanInfo->delSkyline, pReader);
2,275✔
2820
    TSDB_CHECK_CODE(code, lino, _end);
2,275!
2821
  }
2822

2823
  code = tsdbRowMergerGetRow(pMerger, &pTSRow);
2,130,727✔
2824
  TSDB_CHECK_CODE(code, lino, _end);
2,130,727!
2825

2826
  code = doAppendRowFromTSRow(pReader->resBlockInfo.pResBlock, pReader, pTSRow, pBlockScanInfo);
2,130,727✔
2827

2828
  taosMemoryFreeClear(pTSRow);
2,130,727!
2829
  tsdbRowMergerClear(pMerger);
2,130,727✔
2830
  TSDB_CHECK_CODE(code, lino, _end);
2,130,727!
2831

2832
_end:
2,130,727✔
2833
  if (code != TSDB_CODE_SUCCESS) {
2,130,727!
2834
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2835
  }
2836
  return code;
2,130,727✔
2837
}
2838

2839
int32_t doInitMemDataIter(STsdbReader* pReader, STbData** pData, STableBlockScanInfo* pBlockScanInfo, STsdbRowKey* pKey,
43,009,339✔
2840
                          SMemTable* pMem, SIterInfo* pIter, const char* type) {
2841
  int32_t code = TSDB_CODE_SUCCESS;
43,009,339✔
2842
  int32_t lino = 0;
43,009,339✔
2843
  int32_t backward = false;
43,009,339✔
2844

2845
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
43,009,339!
2846
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
43,009,339!
2847
  TSDB_CHECK_NULL(pIter, code, lino, _end, TSDB_CODE_INVALID_PARA);
43,009,339!
2848
  TSDB_CHECK_NULL(type, code, lino, _end, TSDB_CODE_INVALID_PARA);
43,009,339!
2849

2850
  backward = (!ASCENDING_TRAVERSE(pReader->info.order));
43,009,339✔
2851
  pIter->hasVal = false;
43,009,339✔
2852

2853
  if (pMem != NULL) {
43,009,339✔
2854
    TSDB_CHECK_NULL(pData, code, lino, _end, TSDB_CODE_INVALID_PARA);
20,028,491!
2855
    *pData = tsdbGetTbDataFromMemTable(pMem, pReader->info.suid, pBlockScanInfo->uid);
20,028,491✔
2856

2857
    if ((*pData) != NULL) {
20,039,956✔
2858
      code = tsdbTbDataIterCreate((*pData), pKey, backward, &pIter->iter);
7,021,719✔
2859
      if (code == TSDB_CODE_SUCCESS) {
7,019,418!
2860
        pIter->hasVal = (tsdbTbDataIterGet(pIter->iter) != NULL);
7,019,456!
2861

2862
        tsdbDebug("%p uid:%" PRIu64 ", check data in %s from skey:%" PRId64 ", order:%d, ts range in buf:%" PRId64
7,019,456✔
2863
                  "-%" PRId64 " %s",
2864
                  pReader, pBlockScanInfo->uid, type, pKey->key.ts, pReader->info.order, (*pData)->minKey,
2865
                  (*pData)->maxKey, pReader->idStr);
2866
      } else {
2867
        tsdbError("%p uid:%" PRIu64 ", failed to create iterator for %s, code:%s, %s", pReader, pBlockScanInfo->uid,
×
2868
                  type, tstrerror(code), pReader->idStr);
2869
        TSDB_CHECK_CODE(code, lino, _end);
×
2870
      }
2871
    }
2872
  } else {
2873
    tsdbDebug("%p uid:%" PRIu64 ", no data in %s, %s", pReader, pBlockScanInfo->uid, type, pReader->idStr);
22,980,848✔
2874
  }
2875

2876
_end:
22,894,696✔
2877
  if (code != TSDB_CODE_SUCCESS) {
43,018,544!
2878
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2879
  }
2880
  return code;
43,019,416✔
2881
}
2882

2883
static int32_t doForwardDataIter(SRowKey* pKey, SIterInfo* pIter, STableBlockScanInfo* pBlockScanInfo,
2,341,952✔
2884
                                 STsdbReader* pReader) {
2885
  int32_t  code = TSDB_CODE_SUCCESS;
2,341,952✔
2886
  int32_t  lino = 0;
2,341,952✔
2887
  SRowKey  rowKey = {0};
2,341,952✔
2888
  TSDBROW* pRow = NULL;
2,341,952✔
2889

2890
  while (1) {
243✔
2891
    code = getValidMemRow(pIter, pBlockScanInfo->delSkyline, pReader, &pRow);
4,684,106✔
2892
    TSDB_CHECK_CODE(code, lino, _end);
2,341,911!
2893
    if (!pIter->hasVal) {
2,341,911✔
2894
      break;
1,871,392✔
2895
    }
2896

2897
    tRowGetKeyEx(pRow, &rowKey);
470,519✔
2898
    int32_t ret = pkCompEx(pKey, &rowKey);
471,093✔
2899
    if (ret == 0) {
471,093!
2900
      pIter->hasVal = tsdbTbDataIterNext(pIter->iter);
×
2901
    } else {
2902
      break;
471,093✔
2903
    }
2904
  }
2905

2906
_end:
2,342,485✔
2907
  if (code != TSDB_CODE_SUCCESS) {
2,342,485!
2908
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2909
  }
2910
  return code;
2,341,827✔
2911
}
2912

2913
// handle the open interval issue. Find the first row key that is greater than the given one.
2914
static int32_t forwardDataIter(SRowKey* pKey, STableBlockScanInfo* pBlockScanInfo, STsdbReader* pReader) {
1,171,436✔
2915
  int32_t code = TSDB_CODE_SUCCESS;
1,171,436✔
2916
  int32_t lino = 0;
1,171,436✔
2917

2918
  TSDB_CHECK_NULL(pKey, code, lino, _end, TSDB_CODE_INVALID_PARA);
1,171,436!
2919
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
1,171,436!
2920

2921
  code = doForwardDataIter(pKey, &pBlockScanInfo->iter, pBlockScanInfo, pReader);
1,171,436✔
2922
  TSDB_CHECK_CODE(code, lino, _end);
1,171,017!
2923
  code = doForwardDataIter(pKey, &pBlockScanInfo->iiter, pBlockScanInfo, pReader);
1,171,017✔
2924
  TSDB_CHECK_CODE(code, lino, _end);
1,171,232!
2925

2926
_end:
1,171,232✔
2927
  if (code != TSDB_CODE_SUCCESS) {
1,171,232!
2928
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2929
  }
2930
  return code;
1,171,225✔
2931
}
2932

2933
static int32_t initMemDataIterator(STableBlockScanInfo* pBlockScanInfo, STsdbReader* pReader) {
52,575,913✔
2934
  int32_t        code = TSDB_CODE_SUCCESS;
52,575,913✔
2935
  int32_t        lino = 0;
52,575,913✔
2936
  STbData*       d = NULL;
52,575,913✔
2937
  STbData*       di = NULL;
52,575,913✔
2938
  bool           asc = false;
52,575,913✔
2939
  bool           forward = true;
52,575,913✔
2940
  STsdbReadSnap* pSnap = NULL;
52,575,913✔
2941
  STimeWindow*   pWindow = NULL;
52,575,913✔
2942
  STsdbRowKey    startKey;
2943

2944
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
52,575,913!
2945
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
52,575,913!
2946

2947
  asc = ASCENDING_TRAVERSE(pReader->info.order);
52,575,913✔
2948
  pSnap = pReader->pReadSnap;
52,575,913✔
2949
  pWindow = &pReader->info.window;
52,575,913✔
2950

2951
  if (pBlockScanInfo->iterInit) {
52,575,913✔
2952
    goto _end;
31,084,447✔
2953
  }
2954

2955
  startKey.key = pBlockScanInfo->lastProcKey;
21,491,466✔
2956
  startKey.version = asc ? pReader->info.verRange.minVer : pReader->info.verRange.maxVer;
21,491,466✔
2957
  if ((asc && (startKey.key.ts < pWindow->skey)) || ((!asc) && startKey.key.ts > pWindow->ekey)) {
21,491,466!
2958
    startKey.key.ts = asc ? pWindow->skey : pWindow->ekey;
20,337,190✔
2959
    forward = false;
20,337,190✔
2960
  }
2961

2962
  code = doInitMemDataIter(pReader, &d, pBlockScanInfo, &startKey, pSnap->pMem, &pBlockScanInfo->iter, "mem");
21,491,466✔
2963
  TSDB_CHECK_CODE(code, lino, _end);
21,514,432!
2964

2965
  code = doInitMemDataIter(pReader, &di, pBlockScanInfo, &startKey, pSnap->pIMem, &pBlockScanInfo->iiter, "imem");
21,514,432✔
2966
  TSDB_CHECK_CODE(code, lino, _end);
21,513,333!
2967

2968
  code = loadMemTombData(&pBlockScanInfo->pMemDelData, d, di, pReader->info.verRange.maxVer);
21,513,333✔
2969
  TSDB_CHECK_CODE(code, lino, _end);
21,513,684!
2970

2971
  if (forward) {
21,513,684✔
2972
    code = forwardDataIter(&startKey.key, pBlockScanInfo, pReader);
1,171,574✔
2973
    TSDB_CHECK_CODE(code, lino, _end);
1,171,257!
2974
  }
2975

2976
  pBlockScanInfo->iterInit = true;
21,513,367✔
2977

2978
_end:
52,597,814✔
2979
  if (code != TSDB_CODE_SUCCESS) {
52,597,814!
2980
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2981
  }
2982
  return code;
52,590,859✔
2983
}
2984

2985
static int32_t isValidFileBlockRow(SBlockData* pBlockData, int32_t rowIndex, STableBlockScanInfo* pBlockScanInfo,
105,050,553✔
2986
                                   bool asc, STsdbReaderInfo* pInfo, STsdbReader* pReader, bool* valid) {
2987
  int32_t code = TSDB_CODE_SUCCESS;
105,050,553✔
2988
  int32_t lino = 0;
105,050,553✔
2989

2990
  TSDB_CHECK_NULL(pBlockData, code, lino, _end, TSDB_CODE_INVALID_PARA);
105,050,553!
2991
  TSDB_CHECK_NULL(valid, code, lino, _end, TSDB_CODE_INVALID_PARA);
105,050,553!
2992

2993
  *valid = false;
105,050,553✔
2994
  // it is an multi-table data block
2995
  if (pBlockData->aUid != NULL) {
105,050,553!
2996
    uint64_t uid = pBlockData->aUid[rowIndex];
×
2997
    if (uid != pBlockScanInfo->uid) {  // move to next row
×
2998
      *valid = false;
×
2999
      goto _end;
×
3000
    }
3001
  }
3002

3003
  TSDB_CHECK_NULL(pInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
105,050,553!
3004
  // check for version and time range
3005
  int64_t ver = pBlockData->aVersion[rowIndex];
105,050,553✔
3006
  if (ver > pInfo->verRange.maxVer || ver < pInfo->verRange.minVer) {
105,050,553!
3007
    *valid = false;
×
3008
    goto _end;
×
3009
  }
3010

3011
  int64_t ts = pBlockData->aTSKEY[rowIndex];
105,085,741✔
3012
  if (ts > pInfo->window.ekey || ts < pInfo->window.skey) {
105,085,741✔
3013
    *valid = false;
21,373,593✔
3014
    goto _end;
21,373,593✔
3015
  }
3016

3017
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
83,712,148!
3018
  if ((asc && (ts < pBlockScanInfo->lastProcKey.ts)) || ((!asc) && (ts > pBlockScanInfo->lastProcKey.ts))) {
83,712,148!
3019
    *valid = false;
×
3020
    goto _end;
×
3021
  }
3022

3023
  if (ts == pBlockScanInfo->lastProcKey.ts) {  // todo opt perf
83,712,148✔
3024
    SRowKey nextRowKey;                        // lazy eval
3025
    tColRowGetKey(pBlockData, rowIndex, &nextRowKey);
26,711!
3026
    if (pkCompEx(&pBlockScanInfo->lastProcKey, &nextRowKey) == 0) {
53,422!
3027
      *valid = false;
×
3028
      goto _end;
×
3029
    }
3030
  }
3031

3032
  if (pBlockScanInfo->delSkyline != NULL && TARRAY_SIZE(pBlockScanInfo->delSkyline) > 0) {
83,712,148!
3033
    bool dropped = false;
52,030,730✔
3034
    code = hasBeenDropped(pBlockScanInfo->delSkyline, &pBlockScanInfo->fileDelIndex, ts, ver, pInfo->order,
52,030,730✔
3035
                          &pInfo->verRange, pReader->suppInfo.numOfPks > 0, &dropped);
52,030,730✔
3036
    TSDB_CHECK_CODE(code, lino, _end);
81,219,397!
3037
    if (dropped) {
52,019,182✔
3038
      *valid = false;
29,200,215✔
3039
      goto _end;
29,200,215✔
3040
    }
3041
  }
3042

3043
  *valid = true;
54,500,385✔
3044

3045
_end:
105,039,005✔
3046
  if (code != TSDB_CODE_SUCCESS) {
105,039,005!
3047
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3048
  }
3049
  return code;
105,047,037✔
3050
}
3051

3052
static void initSttBlockReader(SSttBlockReader* pSttBlockReader, STableBlockScanInfo* pScanInfo, STsdbReader* pReader) {
28,081,606✔
3053
  int32_t              code = TSDB_CODE_SUCCESS;
28,081,606✔
3054
  int32_t              lino = 0;
28,081,606✔
3055
  int32_t              order = 0;
28,081,606✔
3056
  bool                 asc = false;
28,081,606✔
3057
  int64_t              st = 0;
28,081,606✔
3058
  SSttDataInfoForTable info = (SSttDataInfoForTable){0};
28,081,606✔
3059

3060
  TSDB_CHECK_NULL(pSttBlockReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
28,081,606!
3061
  TSDB_CHECK_NULL(pScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
28,081,606!
3062
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
28,081,606!
3063

3064
  order = pReader->info.order;
28,081,606✔
3065
  asc = ASCENDING_TRAVERSE(order);
28,081,606✔
3066

3067
  // the stt block reader has been initialized for this table.
3068
  if (pSttBlockReader->uid == pScanInfo->uid) {
28,081,606✔
3069
    goto _end;
7,550,366✔
3070
  }
3071

3072
  if (pSttBlockReader->uid != 0) {
20,531,240✔
3073
    tMergeTreeClose(&pSttBlockReader->mergeTree);
10,989,001✔
3074
  }
3075

3076
  pSttBlockReader->uid = pScanInfo->uid;
20,536,427✔
3077

3078
  // second or third time init stt block reader
3079
  if (pScanInfo->cleanSttBlocks && (pReader->info.execMode == READER_EXEC_ROWS)) {
20,536,427!
3080
    // only allowed to retrieve clean stt blocks for count once
3081
    if (pScanInfo->sttBlockReturned) {
52!
3082
      pScanInfo->sttKeyInfo.status = STT_FILE_NO_DATA;
×
3083
      tsdbDebug("uid:%" PRIu64 " set no stt-file data after stt-block retrieved, %s", pScanInfo->uid, pReader->idStr);
×
3084
    }
3085
    goto _end;
52✔
3086
  }
3087

3088
  STimeWindow w = pSttBlockReader->window;
20,536,375✔
3089
  if (asc) {
20,536,375✔
3090
    w.skey = pScanInfo->sttKeyInfo.nextProcKey.ts;
17,691,242✔
3091
  } else {
3092
    w.ekey = pScanInfo->sttKeyInfo.nextProcKey.ts;
2,845,133✔
3093
  }
3094

3095
  st = taosGetTimestampUs();
20,540,647✔
3096
  tsdbDebug("init stt block reader, window:%" PRId64 "-%" PRId64 ", uid:%" PRIu64 ", %s", w.skey, w.ekey,
20,540,647✔
3097
            pScanInfo->uid, pReader->idStr);
3098

3099
  SMergeTreeConf conf = {
20,540,647✔
3100
      .uid = pScanInfo->uid,
20,540,647✔
3101
      .suid = pReader->info.suid,
20,540,647✔
3102
      .pTsdb = pReader->pTsdb,
20,540,647✔
3103
      .timewindow = w,
3104
      .verRange = pSttBlockReader->verRange,
3105
      .strictTimeRange = false,
3106
      .pSchema = pReader->info.pSchema,
20,540,647✔
3107
      .pCurrentFileset = pReader->status.pCurrentFileset,
20,540,647✔
3108
      .backward = (pSttBlockReader->order == TSDB_ORDER_DESC),
20,540,647✔
3109
      .pSttFileBlockIterArray = pReader->status.pLDataIterArray,
20,540,647✔
3110
      .pCols = pReader->suppInfo.colId,
20,540,647✔
3111
      .numOfCols = pReader->suppInfo.numOfCols,
20,540,647✔
3112
      .loadTombFn = loadSttTombDataForAll,
3113
      .pCurRowKey = &pScanInfo->sttKeyInfo.nextProcKey,
20,540,647✔
3114
      .pReader = pReader,
3115
      .idstr = pReader->idStr,
20,540,647✔
3116
      .rspRows = (pReader->info.execMode == READER_EXEC_ROWS),
20,540,647✔
3117
  };
3118

3119
  info.pKeyRangeList = taosArrayInit(4, sizeof(SSttKeyRange));
20,540,647✔
3120
  TSDB_CHECK_NULL(info.pKeyRangeList, code, lino, _end, terrno);
20,541,711!
3121

3122
  code = tMergeTreeOpen2(&pSttBlockReader->mergeTree, &conf, &info);
20,541,711✔
3123
  TSDB_CHECK_CODE(code, lino, _end);
20,537,229!
3124

3125
  code = initMemDataIterator(pScanInfo, pReader);
20,537,229✔
3126
  TSDB_CHECK_CODE(code, lino, _end);
20,541,142!
3127

3128
  code = initDelSkylineIterator(pScanInfo, pReader->info.order, &pReader->cost);
20,541,142✔
3129
  TSDB_CHECK_CODE(code, lino, _end);
20,540,518!
3130

3131
  if (conf.rspRows) {
20,540,518✔
3132
    pScanInfo->cleanSttBlocks = isCleanSttBlock(info.pKeyRangeList, &pReader->info.window, pScanInfo, order);
395,477✔
3133
    if (pScanInfo->cleanSttBlocks) {
395,439✔
3134
      pScanInfo->numOfRowsInStt = info.numOfRows;
36,463✔
3135

3136
      // calculate the time window for data in stt files
3137
      for (int32_t i = 0; i < taosArrayGetSize(info.pKeyRangeList); ++i) {
72,925✔
3138
        SSttKeyRange* pKeyRange = taosArrayGet(info.pKeyRangeList, i);
36,488✔
3139
        if (pKeyRange == NULL) {
36,484!
3140
          continue;
×
3141
        }
3142

3143
        if (pkCompEx(&pScanInfo->sttRange.skey, &pKeyRange->skey) > 0) {
72,966!
3144
          tRowKeyAssign(&pScanInfo->sttRange.skey, &pKeyRange->skey);
36,447✔
3145
        }
3146

3147
        if (pkCompEx(&pScanInfo->sttRange.ekey, &pKeyRange->ekey) < 0) {
72,943!
3148
          tRowKeyAssign(&pScanInfo->sttRange.ekey, &pKeyRange->ekey);
36,476✔
3149
        }
3150
      }
3151

3152
      pScanInfo->sttKeyInfo.status = taosArrayGetSize(info.pKeyRangeList) ? STT_FILE_HAS_DATA : STT_FILE_NO_DATA;
36,365!
3153

3154
      SRowKey* p = asc ? &pScanInfo->sttRange.skey : &pScanInfo->sttRange.ekey;
36,425✔
3155
      tRowKeyAssign(&pScanInfo->sttKeyInfo.nextProcKey, p);
36,425✔
3156
    } else {                                // not clean stt blocks
3157
      INIT_KEYRANGE(&pScanInfo->sttRange);  // reset the time window
358,976✔
3158
      code = nextRowFromSttBlocks(pSttBlockReader, pScanInfo, pReader->suppInfo.pkSrcSlot, &pReader->info.verRange);
358,976✔
3159
      if (code != TSDB_CODE_SUCCESS) {
358,992!
3160
        pScanInfo->sttBlockReturned = false;
×
3161
        TSDB_CHECK_CODE(code, lino, _end);
×
3162
      }
3163
    }
3164
  } else {
3165
    pScanInfo->cleanSttBlocks = false;
20,145,041✔
3166
    INIT_KEYRANGE(&pScanInfo->sttRange);  // reset the time window
20,145,041✔
3167
    code = nextRowFromSttBlocks(pSttBlockReader, pScanInfo, pReader->suppInfo.pkSrcSlot, &pReader->info.verRange);
20,145,041✔
3168
    if (code != TSDB_CODE_SUCCESS) {
20,141,382!
3169
      pScanInfo->sttBlockReturned = false;
×
3170
      TSDB_CHECK_CODE(code, lino, _end);
×
3171
    }
3172
  }
3173

3174
  pScanInfo->sttBlockReturned = false;
20,536,796✔
3175
  int64_t el = taosGetTimestampUs() - st;
20,538,724✔
3176
  pReader->cost.initSttBlockReader += (el / 1000.0);
20,538,724✔
3177

3178
  tsdbDebug("init stt block reader completed, elapsed time:%" PRId64 "us %s", el, pReader->idStr);
20,538,724✔
3179

3180
_end:
20,512,772✔
3181
  if (code != TSDB_CODE_SUCCESS) {
28,089,142!
3182
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3183
    if (pReader) {
×
3184
      pReader->code = code;
×
3185
    }
3186
  }
3187
  taosArrayDestroy(info.pKeyRangeList);
28,089,142✔
3188
}
28,090,362✔
3189

3190
static bool hasDataInSttBlock(STableBlockScanInfo* pInfo) { return pInfo->sttKeyInfo.status == STT_FILE_HAS_DATA; }
2,147,483,647✔
3191

3192
bool hasDataInFileBlock(const SBlockData* pBlockData, const SFileBlockDumpInfo* pDumpInfo) {
1,792,070,649✔
3193
  if ((pBlockData->nRow > 0) && (pBlockData->nRow != pDumpInfo->totalRows)) {
1,792,070,649!
3194
    return false;  // this is an invalid result.
×
3195
  }
3196
  return pBlockData->nRow > 0 && (!pDumpInfo->allDumped);
1,792,070,649!
3197
}
3198

3199
int32_t mergeRowsInFileBlocks(SBlockData* pBlockData, STableBlockScanInfo* pBlockScanInfo, SRowKey* pKey,
41,302,297✔
3200
                              STsdbReader* pReader) {
3201
  int32_t             code = TSDB_CODE_SUCCESS;
41,302,297✔
3202
  int32_t             lino = 0;
41,302,297✔
3203
  SRow*               pTSRow = NULL;
41,302,297✔
3204
  SRowMerger*         pMerger = NULL;
41,302,297✔
3205
  SFileBlockDumpInfo* pDumpInfo = NULL;
41,302,297✔
3206
  bool                copied = false;
41,302,297✔
3207

3208
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
41,302,297!
3209
  TSDB_CHECK_NULL(pKey, code, lino, _end, TSDB_CODE_INVALID_PARA);
41,302,297!
3210
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
41,302,297!
3211

3212
  pMerger = &pReader->status.merger;
41,302,297✔
3213
  pDumpInfo = &pReader->status.fBlockDumpInfo;
41,302,297✔
3214

3215
  code = tryCopyDistinctRowFromFileBlock(pReader, pBlockData, pKey, pDumpInfo, &copied);
41,302,297✔
3216
  TSDB_CHECK_CODE(code, lino, _end);
41,359,659!
3217

3218
  // merge is not initialized yet, due to the fact that the pReader->info.pSchema is not initialized
3219
  code = initRowMergeIfNeeded(pReader, pBlockScanInfo->uid);
41,359,659✔
3220
  TSDB_CHECK_CODE(code, lino, _end);
41,245,354!
3221

3222
  tRowKeyAssign(&pBlockScanInfo->lastProcKey, pKey);
41,245,354✔
3223

3224
  if (copied) {
41,199,704✔
3225
    goto _end;
41,190,520✔
3226
  }
3227

3228
  TSDBROW fRow = tsdbRowFromBlockData(pBlockData, pDumpInfo->rowIndex);
9,184✔
3229

3230
  code = tsdbRowMergerAdd(pMerger, &fRow, NULL);
9,184✔
3231
  TSDB_CHECK_CODE(code, lino, _end);
14,249!
3232

3233
  code = doMergeRowsInFileBlocks(pBlockData, pBlockScanInfo, pKey, pReader);
14,249✔
3234
  TSDB_CHECK_CODE(code, lino, _end);
14,249!
3235

3236
  code = tsdbRowMergerGetRow(pMerger, &pTSRow);
14,249✔
3237
  TSDB_CHECK_CODE(code, lino, _end);
14,249!
3238

3239
  code = doAppendRowFromTSRow(pReader->resBlockInfo.pResBlock, pReader, pTSRow, pBlockScanInfo);
14,249✔
3240

3241
  taosMemoryFreeClear(pTSRow);
14,249!
3242
  tsdbRowMergerClear(pMerger);
14,249✔
3243
  TSDB_CHECK_CODE(code, lino, _end);
14,248!
3244

3245
_end:
14,248✔
3246
  if (code != TSDB_CODE_SUCCESS) {
41,204,768!
3247
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3248
  }
3249
  return code;
41,214,825✔
3250
}
3251

3252
int32_t mergeRowsInSttBlocks(SSttBlockReader* pSttBlockReader, STableBlockScanInfo* pScanInfo, STsdbReader* pReader) {
333,827,392✔
3253
  int32_t     code = TSDB_CODE_SUCCESS;
333,827,392✔
3254
  int32_t     lino = 0;
333,827,392✔
3255
  SRow*       pTSRow = NULL;
333,827,392✔
3256
  int32_t     pkSrcSlot = 0;
333,827,392✔
3257
  SRowMerger* pMerger = NULL;
333,827,392✔
3258
  bool        copied = false;
333,827,392✔
3259

3260
  TSDB_CHECK_NULL(pSttBlockReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
333,827,392!
3261
  TSDB_CHECK_NULL(pScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
333,827,392!
3262
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
333,827,392!
3263

3264
  pkSrcSlot = pReader->suppInfo.pkSrcSlot;
333,827,392✔
3265
  pMerger = &pReader->status.merger;
333,827,392✔
3266

3267
  // let's record the last processed key
3268
  tRowKeyAssign(&pScanInfo->lastProcKey, getCurrentKeyInSttBlock(pSttBlockReader));
333,827,392✔
3269

3270
  TSDBROW* pRow = tMergeTreeGetRow(&pSttBlockReader->mergeTree);
333,743,985✔
3271
  TSDBROW  fRow = {.iRow = pRow->iRow, .type = TSDBROW_COL_FMT, .pBlockData = pRow->pBlockData};
333,743,985✔
3272

3273
  if (IS_VAR_DATA_TYPE(pScanInfo->lastProcKey.pks[0].type)) {
333,743,985!
3274
    tsdbTrace("fRow ptr:%p, %d, uid:%" PRIu64 ", ts:%" PRId64 " pk:%s %s", pRow->pBlockData, pRow->iRow,
155,778!
3275
              pSttBlockReader->uid, fRow.pBlockData->aTSKEY[fRow.iRow], pScanInfo->lastProcKey.pks[0].pData,
3276
              pReader->idStr);
3277
  }
3278

3279
  code = tryCopyDistinctRowFromSttBlock(&fRow, pSttBlockReader, pScanInfo, &pScanInfo->lastProcKey, pReader, &copied);
333,743,985✔
3280
  TSDB_CHECK_CODE(code, lino, _end);
334,458,923!
3281

3282
  if (copied) {
334,458,923✔
3283
    goto _end;
333,455,574✔
3284
  }
3285

3286
  code = tsdbRowMergerAdd(pMerger, &fRow, NULL);
1,003,349✔
3287
  TSDB_CHECK_CODE(code, lino, _end);
1,177,806!
3288

3289
  TSDBROW* pRow1 = tMergeTreeGetRow(&pSttBlockReader->mergeTree);
1,177,806✔
3290
  code = tsdbRowMergerAdd(pMerger, pRow1, NULL);
1,177,806✔
3291
  TSDB_CHECK_CODE(code, lino, _end);
1,150,214!
3292

3293
  code = doMergeRowsInSttBlock(pSttBlockReader, pScanInfo, pMerger, pkSrcSlot, &pReader->info.verRange, pReader->idStr);
1,150,214✔
3294
  TSDB_CHECK_CODE(code, lino, _end);
1,161,887!
3295

3296
  code = tsdbRowMergerGetRow(pMerger, &pTSRow);
1,161,887✔
3297
  TSDB_CHECK_CODE(code, lino, _end);
1,192,583!
3298

3299
  code = doAppendRowFromTSRow(pReader->resBlockInfo.pResBlock, pReader, pTSRow, pScanInfo);
1,192,583✔
3300

3301
  taosMemoryFreeClear(pTSRow);
1,188,820!
3302
  tsdbRowMergerClear(pMerger);
1,193,781✔
3303
  TSDB_CHECK_CODE(code, lino, _end);
1,159,069!
3304

3305
_end:
1,159,069✔
3306
  if (code != TSDB_CODE_SUCCESS) {
334,614,643!
3307
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3308
  }
3309
  return code;
334,750,649✔
3310
}
3311

3312
static int32_t buildComposedDataBlockImpl(STsdbReader* pReader, STableBlockScanInfo* pBlockScanInfo,
913,341,821✔
3313
                                          SBlockData* pBlockData, SSttBlockReader* pSttBlockReader) {
3314
  int32_t             code = TSDB_CODE_SUCCESS;
913,341,821✔
3315
  int32_t             lino = 0;
913,341,821✔
3316
  SFileBlockDumpInfo* pDumpInfo = NULL;
913,341,821✔
3317
  TSDBROW*            pRow = NULL;
913,341,821✔
3318
  TSDBROW*            piRow = NULL;
913,341,821✔
3319
  SRowKey*            pKey = &(SRowKey){0};
913,341,821✔
3320

3321
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
913,341,821!
3322
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
913,341,821!
3323
  TSDB_CHECK_NULL(pBlockData, code, lino, _end, TSDB_CODE_INVALID_PARA);
913,341,821!
3324

3325
  pDumpInfo = &pReader->status.fBlockDumpInfo;
913,341,821✔
3326

3327
  if (hasDataInFileBlock(pBlockData, pDumpInfo)) {
913,341,821✔
3328
    tColRowGetKey(pBlockData, pDumpInfo->rowIndex, pKey);
54,527,204✔
3329
    code = primaryKeyDup(pKey);
54,547,641✔
3330
    TSDB_CHECK_CODE(code, lino, _end);
55,017,424!
3331
    lino = __LINE__;
55,017,424✔
3332
  } else {
3333
    pKey = NULL;
860,273,695✔
3334
  }
3335

3336
  if (pBlockScanInfo->iter.hasVal) {
915,291,119✔
3337
    code = getValidMemRow(&pBlockScanInfo->iter, pBlockScanInfo->delSkyline, pReader, &pRow);
1,027,180,360✔
3338
    TSDB_CHECK_CODE(code, lino, _end);
513,801,337!
3339
  }
3340

3341
  if (pBlockScanInfo->iiter.hasVal) {
915,713,433✔
3342
    code = getValidMemRow(&pBlockScanInfo->iiter, pBlockScanInfo->delSkyline, pReader, &piRow);
62,841,734✔
3343
    TSDB_CHECK_CODE(code, lino, _end);
31,643,418!
3344
  }
3345

3346
  if (pRow != NULL && piRow != NULL) {
916,158,535✔
3347
    // imem + mem + file + stt block
3348
    // two levels of mem-table does contain the valid rows
3349
    code = doMergeMultiLevelRows(pReader, pBlockScanInfo, pBlockData, pSttBlockReader);
2,130,727✔
3350
    TSDB_CHECK_CODE(code, lino, _end);
2,130,727!
3351
  } else if (pBlockScanInfo->iiter.hasVal) {
914,027,808✔
3352
    // imem + file + stt block
3353
    code = doMergeBufAndFileRows(pReader, pBlockScanInfo, piRow, &pBlockScanInfo->iiter, pSttBlockReader);
29,519,594✔
3354
    TSDB_CHECK_CODE(code, lino, _end);
28,908,175!
3355
  } else if (pBlockScanInfo->iter.hasVal) {
884,508,214✔
3356
    // mem + file + stt block
3357
    code = doMergeBufAndFileRows(pReader, pBlockScanInfo, pRow, &pBlockScanInfo->iter, pSttBlockReader);
511,635,806✔
3358
    TSDB_CHECK_CODE(code, lino, _end);
509,064,032!
3359
  } else {
3360
    // files data blocks + stt block
3361
    code = mergeFileBlockAndSttBlock(pReader, pSttBlockReader, pKey, pBlockScanInfo, pBlockData);
372,872,408✔
3362
    TSDB_CHECK_CODE(code, lino, _end);
375,946,883!
3363
  }
3364

3365
_end:
375,946,883✔
3366
  clearRowKey(pKey);
916,049,817✔
3367
  if (code != TSDB_CODE_SUCCESS) {
913,331,145!
3368
    tsdbError("%s %s failed at line %d since %s", pReader->idStr, __func__, lino, tstrerror(code));
×
3369
  }
3370
  return code;
913,534,522✔
3371
}
3372

3373
static int32_t loadNeighborIfOverlap(SFileDataBlockInfo* pBlockInfo, STableBlockScanInfo* pBlockScanInfo,
46,950✔
3374
                                     STsdbReader* pReader, bool* loadNeighbor) {
3375
  int32_t             code = TSDB_CODE_SUCCESS;
46,950✔
3376
  int32_t             lino = 0;
46,950✔
3377
  int32_t             order = 0;
46,950✔
3378
  SDataBlockIter*     pIter = NULL;
46,950✔
3379
  SBlockLoadSuppInfo* pSupInfo = NULL;
46,950✔
3380
  int32_t             step = 0;
46,950✔
3381
  int32_t             nextIndex = -1;
46,950✔
3382
  SBrinRecord         rec = {0};
46,950✔
3383
  bool                hasNeighbor = false;
46,950✔
3384

3385
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
46,950!
3386
  TSDB_CHECK_NULL(loadNeighbor, code, lino, _end, TSDB_CODE_INVALID_PARA);
46,950!
3387

3388
  order = pReader->info.order;
46,950✔
3389
  pIter = &pReader->status.blockIter;
46,950✔
3390
  pSupInfo = &pReader->suppInfo;
46,950✔
3391
  step = ASCENDING_TRAVERSE(order) ? 1 : -1;
46,950✔
3392

3393
  *loadNeighbor = false;
46,950✔
3394
  code = getNeighborBlockOfTable(pIter, pBlockInfo, pBlockScanInfo, &nextIndex, order, &rec, pSupInfo, &hasNeighbor);
46,950✔
3395
  TSDB_CHECK_CODE(code, lino, _end);
46,950!
3396
  if (!hasNeighbor) {  // do nothing
46,950✔
3397
    goto _end;
31,041✔
3398
  }
3399

3400
  // load next block
3401
  if (overlapWithNeighborBlock2(pBlockInfo, &rec, order, pReader->suppInfo.pk.type, pReader->suppInfo.numOfPks)) {
15,909✔
3402
    SReaderStatus*  pStatus = &pReader->status;
5✔
3403
    SDataBlockIter* pBlockIter = &pStatus->blockIter;
5✔
3404

3405
    // 1. find the next neighbor block in the scan block list
3406
    STableDataBlockIdx* tableDataBlockIdx = taosArrayGet(pBlockScanInfo->pBlockIdxList, nextIndex);
5✔
3407
    TSDB_CHECK_NULL(tableDataBlockIdx, code, lino, _end, TSDB_CODE_INVALID_PARA);
5!
3408

3409
    // 2. remove it from the scan block list
3410
    int32_t neighborIndex = tableDataBlockIdx->globalIndex;
5✔
3411
    code = setFileBlockActiveInBlockIter(pReader, pBlockIter, neighborIndex, step);
5✔
3412
    TSDB_CHECK_CODE(code, lino, _end);
5!
3413

3414
    // 3. load the neighbor block, and set it to be the currently accessed file data block
3415
    code = doLoadFileBlockData(pReader, pBlockIter, &pStatus->fileBlockData, pBlockInfo->uid);
5✔
3416
    TSDB_CHECK_CODE(code, lino, _end);
5!
3417

3418
    // 4. check the data values
3419
    code = initBlockDumpInfo(pReader, pBlockIter);
5✔
3420
    TSDB_CHECK_CODE(code, lino, _end);
5!
3421
    *loadNeighbor = true;
5✔
3422
  }
3423

3424
_end:
15,904✔
3425
  if (code != TSDB_CODE_SUCCESS) {
46,950!
3426
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3427
  }
3428
  return code;
46,950✔
3429
}
3430

3431
void updateComposedBlockInfo(STsdbReader* pReader, double el, STableBlockScanInfo* pBlockScanInfo) {
16,450,105✔
3432
  SSDataBlock* pResBlock = pReader->resBlockInfo.pResBlock;
16,450,105✔
3433

3434
  pResBlock->info.id.uid = (pBlockScanInfo != NULL) ? pBlockScanInfo->uid : 0;
16,450,105!
3435
  pResBlock->info.dataLoad = 1;
16,450,105✔
3436
  pResBlock->info.version = pReader->info.verRange.maxVer;
16,450,105✔
3437

3438
  int32_t code = blockDataUpdateTsWindow(pResBlock, pReader->suppInfo.slotId[0]);
16,450,105✔
3439
  code = blockDataUpdatePkRange(pResBlock, pReader->suppInfo.pkDstSlot, ASCENDING_TRAVERSE(pReader->info.order));
16,452,312✔
3440
  setComposedBlockFlag(pReader, true);
16,451,685✔
3441

3442
  pReader->cost.composedBlocks += 1;
16,450,503✔
3443
  pReader->cost.buildComposedBlockTime += el;
16,450,503✔
3444
}
16,450,503✔
3445

3446
static int32_t buildComposedDataBlock(STsdbReader* pReader) {
50,645✔
3447
  int32_t              code = TSDB_CODE_SUCCESS;
50,645✔
3448
  int32_t              lino = 0;
50,645✔
3449
  bool                 asc = false;
50,645✔
3450
  int64_t              st = 0;
50,645✔
3451
  double               el = 0;
50,645✔
3452
  int32_t              step = 0;
50,645✔
3453
  SSDataBlock*         pResBlock = NULL;
50,645✔
3454
  SFileDataBlockInfo*  pBlockInfo = NULL;
50,645✔
3455
  SSttBlockReader*     pSttBlockReader = NULL;
50,645✔
3456
  SFileBlockDumpInfo*  pDumpInfo = NULL;
50,645✔
3457
  STableBlockScanInfo* pBlockScanInfo = NULL;
50,645✔
3458
  TSDBKEY              keyInBuf;
3459

3460
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
50,645!
3461

3462
  asc = ASCENDING_TRAVERSE(pReader->info.order);
50,645✔
3463
  st = taosGetTimestampUs();
50,645✔
3464
  step = asc ? 1 : -1;
50,645✔
3465
  pResBlock = pReader->resBlockInfo.pResBlock;
50,645✔
3466
  pSttBlockReader = pReader->status.fileIter.pSttBlockReader;
50,645✔
3467
  pDumpInfo = &pReader->status.fBlockDumpInfo;
50,645✔
3468

3469
  code = getCurrentBlockInfo(&pReader->status.blockIter, &pBlockInfo, pReader->idStr);
50,645✔
3470
  if (code != TSDB_CODE_SUCCESS) {
50,645!
3471
    return 0;
×
3472
  }
3473

3474
  if (pReader->pIgnoreTables && taosHashGet(*pReader->pIgnoreTables, &pBlockInfo->uid, sizeof(pBlockInfo->uid))) {
50,645!
3475
    setBlockAllDumped(pDumpInfo, pBlockInfo->lastKey, pReader->info.order);
×
3476
    return code;
×
3477
  }
3478

3479
  code = getTableBlockScanInfo(pReader->status.pTableMap, pBlockInfo->uid, &pBlockScanInfo, pReader->idStr);
50,645✔
3480
  TSDB_CHECK_CODE(code, lino, _end);
50,645!
3481

3482
  code = getCurrentKeyInBuf(pBlockScanInfo, pReader, &keyInBuf);
50,645✔
3483
  TSDB_CHECK_CODE(code, lino, _end);
50,644!
3484

3485
  // it is a clean block, load it directly
3486
  int64_t cap = pReader->resBlockInfo.capacity;
50,644✔
3487
  bool    isClean = false;
50,644✔
3488
  code = isCleanFileDataBlock(pReader, pBlockInfo, pBlockScanInfo, keyInBuf, &isClean);
50,644✔
3489
  TSDB_CHECK_CODE(code, lino, _end);
50,644!
3490
  bool directCopy = isClean && (pBlockInfo->numRow <= cap) && (pBlockScanInfo->sttKeyInfo.status == STT_FILE_NO_DATA) &&
53,059!
3491
                    ((asc && ((pBlockInfo->lastKey < keyInBuf.ts) || (keyInBuf.ts == INT64_MIN))) ||
1,465!
3492
                     (!asc && (pBlockInfo->lastKey > keyInBuf.ts)));
950!
3493
  if (directCopy) {
50,644✔
3494
    code = copyBlockDataToSDataBlock(pReader, &pBlockScanInfo->lastProcKey);
2,415✔
3495
    TSDB_CHECK_CODE(code, lino, _end);
2,415!
3496
    goto _end;
2,415✔
3497
  }
3498

3499
  SBlockData* pBlockData = &pReader->status.fileBlockData;
48,229✔
3500
  initSttBlockReader(pSttBlockReader, pBlockScanInfo, pReader);
48,229✔
3501
  code = pReader->code;
48,229✔
3502
  TSDB_CHECK_CODE(code, lino, _end);
48,229!
3503

3504
  while (1) {
54,510,528✔
3505
    bool hasBlockData = false;
54,558,757✔
3506
    {
3507
      while (pBlockData->nRow > 0 && pBlockData->uid == pBlockScanInfo->uid) {
105,080,380!
3508
        // find the first qualified row in data block
3509
        bool valid = false;
105,071,130✔
3510
        code =
3511
            isValidFileBlockRow(pBlockData, pDumpInfo->rowIndex, pBlockScanInfo, asc, &pReader->info, pReader, &valid);
105,071,130✔
3512
        TSDB_CHECK_CODE(code, lino, _end);
105,052,966!
3513
        if (valid) {
105,052,966✔
3514
          hasBlockData = true;
54,557,779✔
3515
          break;
54,586,154✔
3516
        }
3517

3518
        pDumpInfo->rowIndex += step;
50,495,187✔
3519

3520
        if (pDumpInfo->rowIndex >= pBlockData->nRow || pDumpInfo->rowIndex < 0) {
50,495,187!
3521
          // NOTE: get the new block info
3522
          code = getCurrentBlockInfo(&pReader->status.blockIter, &pBlockInfo, pReader->idStr);
×
3523
          TSDB_CHECK_CODE(code, lino, _end);
28,375!
3524

3525
          // continue check for the next file block if the last ts in the current block
3526
          // is overlapped with the next neighbor block
3527
          bool loadNeighbor = false;
28,375✔
3528
          code = loadNeighborIfOverlap(pBlockInfo, pBlockScanInfo, pReader, &loadNeighbor);
28,375✔
3529
          if ((!loadNeighbor) || (code != 0)) {
28,375!
3530
            lino = __LINE__;
28,375✔
3531
            setBlockAllDumped(pDumpInfo, pBlockInfo->lastKey, pReader->info.order);
28,375✔
3532
            break;
28,375✔
3533
          }
3534
        }
3535
      }
3536
    }
3537

3538
    // no data in last block and block, no need to proceed.
3539
    if (hasBlockData == false) {
54,595,404✔
3540
      break;
28,375✔
3541
    }
3542

3543
    code = buildComposedDataBlockImpl(pReader, pBlockScanInfo, pBlockData, pSttBlockReader);
54,567,029✔
3544
    TSDB_CHECK_CODE(code, lino, _end);
54,549,883!
3545

3546
    // currently loaded file data block is consumed
3547
    if ((pBlockData->nRow > 0) && (pDumpInfo->rowIndex >= pBlockData->nRow || pDumpInfo->rowIndex < 0)) {
54,549,883!
3548
      setBlockAllDumped(pDumpInfo, pBlockInfo->lastKey, pReader->info.order);
38,070✔
3549
      break;
18,569✔
3550
    }
3551

3552
    if (pResBlock->info.rows >= pReader->resBlockInfo.capacity) {
54,511,813✔
3553
      break;
1,285✔
3554
    }
3555
  }
3556

3557
_end:
50,644✔
3558
  if (pReader) {
50,644✔
3559
    el = (taosGetTimestampUs() - st) / 1000.0;
50,643✔
3560
    updateComposedBlockInfo(pReader, el, pBlockScanInfo);
50,643✔
3561
  }
3562

3563
  if (pResBlock && pResBlock->info.rows > 0) {
50,646✔
3564
    tsdbDebug("%p uid:%" PRIu64 ", composed data block created, brange:%" PRIu64 "-%" PRIu64 " rows:%" PRId64
23,013✔
3565
              ", elapsed time:%.2f ms %s",
3566
              pReader, pResBlock->info.id.uid, pResBlock->info.window.skey, pResBlock->info.window.ekey,
3567
              pResBlock->info.rows, el, pReader->idStr);
3568
  }
3569

3570
  if (code != TSDB_CODE_SUCCESS) {
50,646!
3571
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3572
  }
3573
  return code;
50,645✔
3574
}
3575

3576
void setComposedBlockFlag(STsdbReader* pReader, bool composed) { pReader->status.composedDataBlock = composed; }
29,845,808✔
3577

3578
int32_t getInitialDelIndex(const SArray* pDelSkyline, int32_t order) {
406,716✔
3579
  if (pDelSkyline == NULL) {
406,716!
3580
    return 0;
×
3581
  }
3582

3583
  return ASCENDING_TRAVERSE(order) ? 0 : taosArrayGetSize(pDelSkyline) - 1;
406,716✔
3584
}
3585

3586
int32_t initDelSkylineIterator(STableBlockScanInfo* pBlockScanInfo, int32_t order, SReadCostSummary* pCost) {
52,586,440✔
3587
  int32_t code = TSDB_CODE_SUCCESS;
52,586,440✔
3588
  int32_t lino = 0;
52,586,440✔
3589
  int32_t newDelDataInFile = 0;
52,586,440✔
3590
  int64_t st = 0;
52,586,440✔
3591
  SArray* pSource = NULL;
52,586,440✔
3592

3593
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
52,586,440!
3594

3595
  newDelDataInFile = taosArrayGetSize(pBlockScanInfo->pFileDelData);
52,586,440✔
3596
  if (newDelDataInFile == 0 &&
52,583,203✔
3597
      ((pBlockScanInfo->delSkyline != NULL) || (TARRAY_SIZE(pBlockScanInfo->pMemDelData) == 0))) {
52,469,294✔
3598
    goto _end;
52,179,573✔
3599
  }
3600

3601
  st = taosGetTimestampUs();
406,716✔
3602

3603
  if (pBlockScanInfo->delSkyline != NULL) {
406,716✔
3604
    taosArrayClear(pBlockScanInfo->delSkyline);
71,221✔
3605
  } else {
3606
    pBlockScanInfo->delSkyline = taosArrayInit(4, sizeof(TSDBKEY));
335,495✔
3607
    TSDB_CHECK_NULL(pBlockScanInfo->delSkyline, code, lino, _end, terrno);
335,498!
3608
  }
3609

3610
  pSource = pBlockScanInfo->pFileDelData;
406,720✔
3611
  if (pSource == NULL) {
406,720✔
3612
    pSource = pBlockScanInfo->pMemDelData;
277,556✔
3613
  } else {
3614
    const void* p1 = taosArrayAddAll(pSource, pBlockScanInfo->pMemDelData);
129,164✔
3615
    TSDB_CHECK_NULL(p1, code, lino, _end, terrno);
129,160!
3616
  }
3617

3618
  code = tsdbBuildDeleteSkyline(pSource, 0, taosArrayGetSize(pSource) - 1, pBlockScanInfo->delSkyline);
406,716✔
3619

3620
  taosArrayClear(pBlockScanInfo->pFileDelData);
406,718✔
3621
  int32_t index = getInitialDelIndex(pBlockScanInfo->delSkyline, order);
406,718✔
3622

3623
  pBlockScanInfo->iter.index = index;
406,715✔
3624
  pBlockScanInfo->iiter.index = index;
406,715✔
3625
  pBlockScanInfo->fileDelIndex = index;
406,715✔
3626
  pBlockScanInfo->sttBlockDelIndex = index;
406,715✔
3627
  TSDB_CHECK_CODE(code, lino, _end);
406,715!
3628

3629
  TSDB_CHECK_NULL(pCost, code, lino, _end, TSDB_CODE_INVALID_PARA);
406,715!
3630
  double el = taosGetTimestampUs() - st;
406,716✔
3631
  pCost->createSkylineIterTime = el / 1000.0;
406,716✔
3632

3633
_end:
52,586,289✔
3634
  if (code != TSDB_CODE_SUCCESS) {
52,586,289!
3635
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3636
  }
3637
  return code;
52,584,322✔
3638
}
3639

3640
int32_t getCurrentKeyInBuf(STableBlockScanInfo* pScanInfo, STsdbReader* pReader, TSDBKEY* pkey) {
13,460,335✔
3641
  int32_t  code = TSDB_CODE_SUCCESS;
13,460,335✔
3642
  int32_t  lino = 0;
13,460,335✔
3643
  bool     asc = false;
13,460,335✔
3644
  bool     hasKey = false;
13,460,335✔
3645
  bool     hasIKey = false;
13,460,335✔
3646
  TSDBROW* pRow = NULL;
13,460,335✔
3647
  TSDBROW* pIRow = NULL;
13,460,335✔
3648
  TSDBKEY  key = {.ts = TSKEY_INITIAL_VAL};
13,460,335✔
3649
  TSDBKEY  ikey = {.ts = TSKEY_INITIAL_VAL};
13,460,335✔
3650

3651
  TSDB_CHECK_NULL(pScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,460,335!
3652
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,460,335!
3653
  TSDB_CHECK_NULL(pkey, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,460,335!
3654

3655
  asc = ASCENDING_TRAVERSE(pReader->info.order);
13,460,335✔
3656

3657
  code = getValidMemRow(&pScanInfo->iter, pScanInfo->delSkyline, pReader, &pRow);
26,920,559✔
3658
  TSDB_CHECK_CODE(code, lino, _end);
13,460,224!
3659
  if (pRow != NULL) {
13,460,224✔
3660
    hasKey = true;
5,083,222✔
3661
    key = TSDBROW_KEY(pRow);
5,083,222!
3662
  }
3663

3664
  code = getValidMemRow(&pScanInfo->iiter, pScanInfo->delSkyline, pReader, &pIRow);
26,920,451✔
3665
  TSDB_CHECK_CODE(code, lino, _end);
13,460,227!
3666
  if (pIRow != NULL) {
13,460,227✔
3667
    hasIKey = true;
409,601✔
3668
    ikey = TSDBROW_KEY(pIRow);
409,601!
3669
  }
3670

3671
  if (hasKey) {
13,460,227✔
3672
    if (hasIKey) {  // has data in mem & imem
5,083,222✔
3673
      if (asc) {
113,264!
3674
        *pkey = key.ts <= ikey.ts ? key : ikey;
113,264!
3675
      } else {
3676
        *pkey = key.ts <= ikey.ts ? ikey : key;
×
3677
      }
3678
    } else {  // no data in imem
3679
      *pkey = key;
4,969,958✔
3680
    }
3681
  } else {
3682
    // no data in mem & imem, return the initial value
3683
    // only imem has data, return ikey
3684
    *pkey = ikey;
8,377,005✔
3685
  }
3686

3687
_end:
13,460,227✔
3688
  if (code != TSDB_CODE_SUCCESS) {
13,460,227!
3689
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3690
  }
3691
  return code;
13,460,204✔
3692
}
3693

3694
static void prepareDurationForNextFileSet(STsdbReader* pReader) {
4,671,067✔
3695
  if (pReader->status.bProcMemFirstFileset) {
4,671,067✔
3696
    pReader->status.prevFilesetStartKey = INT64_MIN;
1,522,906✔
3697
    pReader->status.prevFilesetEndKey = INT64_MAX;
1,522,906✔
3698
    pReader->status.bProcMemFirstFileset = false;
1,522,906✔
3699
  }
3700

3701
  int32_t     fid = pReader->status.pCurrentFileset->fid;
4,671,067✔
3702
  STimeWindow winFid = {0};
4,671,067✔
3703
  tsdbFidKeyRange(fid, pReader->pTsdb->keepCfg.days, pReader->pTsdb->keepCfg.precision, &winFid.skey, &winFid.ekey);
4,671,067✔
3704

3705
  if (ASCENDING_TRAVERSE(pReader->info.order)) {
4,672,357✔
3706
    pReader->status.bProcMemPreFileset = !(pReader->status.memTableMaxKey < pReader->status.prevFilesetStartKey ||
7,068,445✔
3707
                                           (winFid.skey - 1) < pReader->status.memTableMinKey);
3,014,574✔
3708
  } else {
3709
    pReader->status.bProcMemPreFileset = !(pReader->status.memTableMaxKey < (winFid.ekey + 1) ||
1,052,184✔
3710
                                           pReader->status.prevFilesetEndKey < pReader->status.memTableMinKey);
433,698!
3711
  }
3712

3713
  if (pReader->status.bProcMemPreFileset) {
4,672,357✔
3714
    tsdbDebug("will start pre-fileset %d buffer processing. %s", fid, pReader->idStr);
2,114,299!
3715
    pReader->status.procMemUidList.tableUidList = pReader->status.uidList.tableUidList;
2,114,299✔
3716
    resetPreFilesetMemTableListIndex(&pReader->status);
2,114,299✔
3717
  }
3718

3719
  if (!pReader->status.bProcMemPreFileset) {
4,672,394✔
3720
    if (pReader->notifyFn) {
2,557,110!
3721
      STsdReaderNotifyInfo info = {0};
2,557,539✔
3722
      info.duration.filesetId = fid;
2,557,539✔
3723
      pReader->notifyFn(TSD_READER_NOTIFY_DURATION_START, &info, pReader->notifyParam);
2,557,539✔
3724
      tsdbDebug("new duration %d start notification when no buffer preceeding fileset, %s", fid, pReader->idStr);
2,557,992✔
3725
    }
3726
  }
3727

3728
  pReader->status.prevFilesetStartKey = winFid.skey;
4,673,008✔
3729
  pReader->status.prevFilesetEndKey = winFid.ekey;
4,673,008✔
3730
}
4,673,008✔
3731

3732
static int32_t moveToNextFile(STsdbReader* pReader, SBlockNumber* pBlockNum, SArray* pTableList) {
13,358,557✔
3733
  int32_t        code = TSDB_CODE_SUCCESS;
13,358,557✔
3734
  int32_t        lino = 0;
13,358,557✔
3735
  SReaderStatus* pStatus = NULL;
13,358,557✔
3736
  SArray*        pIndexList = NULL;
13,358,557✔
3737
  size_t         numOfTables = 0;
13,358,557✔
3738

3739
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,358,557!
3740
  TSDB_CHECK_NULL(pBlockNum, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,358,557!
3741

3742
  pStatus = &pReader->status;
13,358,557✔
3743
  pBlockNum->numOfBlocks = 0;
13,358,557✔
3744
  pBlockNum->numOfSttFiles = 0;
13,358,557✔
3745

3746
  numOfTables = tSimpleHashGetSize(pReader->status.pTableMap);
13,358,557✔
3747
  pIndexList = taosArrayInit(numOfTables, sizeof(SBrinBlk));
13,360,600✔
3748
  TSDB_CHECK_NULL(pIndexList, code, lino, _end, terrno);
13,359,205!
3749

3750
  while (1) {
10,086✔
3751
    // only check here, since the iterate data in memory is very fast.
3752
    if (pReader->code != TSDB_CODE_SUCCESS) {
13,369,291!
3753
      tsdbWarn("tsdb reader is stopped ASAP, code:%s, %s", tstrerror(pReader->code), pReader->idStr);
×
3754
      code = pReader->code;
×
3755
      TSDB_CHECK_CODE(code, lino, _end);
×
3756
    }
3757

3758
    bool hasNext = false;
13,369,291✔
3759
    code = filesetIteratorNext(&pStatus->fileIter, pReader, &hasNext);
13,369,291✔
3760
    TSDB_CHECK_CODE(code, lino, _end);
13,370,012!
3761

3762
    if (!hasNext) {  // no data files on disk
13,370,012✔
3763
      break;
3,796,417✔
3764
    }
3765

3766
    taosArrayClear(pIndexList);
9,573,595✔
3767
    code = doLoadBlockIndex(pReader, pReader->pFileReader, pIndexList);
9,571,718✔
3768
    TSDB_CHECK_CODE(code, lino, _end);
9,570,800!
3769

3770
    if (taosArrayGetSize(pIndexList) > 0 || pReader->status.pCurrentFileset->lvlArr->size > 0) {
9,570,800✔
3771
      code = loadFileBlockBrinInfo(pReader, pIndexList, pBlockNum, pTableList);
9,567,655✔
3772
      TSDB_CHECK_CODE(code, lino, _end);
9,570,553!
3773

3774
      if (pBlockNum->numOfBlocks + pBlockNum->numOfSttFiles > 0) {
9,570,553✔
3775
        if (pReader->bFilesetDelimited) {
9,563,668✔
3776
          prepareDurationForNextFileSet(pReader);
4,673,217✔
3777
        }
3778
        break;
9,562,606✔
3779
      }
3780
    }
3781

3782
    // no blocks in current file, try next files
3783
  }
3784

3785
  code = loadDataFileTombDataForAll(pReader);
13,359,023✔
3786
  TSDB_CHECK_CODE(code, lino, _end);
13,355,783!
3787

3788
_end:
13,355,783✔
3789
  if (code != TSDB_CODE_SUCCESS) {
13,355,783!
3790
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3791
  }
3792
  if (pIndexList) {
13,356,788!
3793
    taosArrayDestroy(pIndexList);
13,356,788✔
3794
  }
3795
  return code;
13,360,105✔
3796
}
3797

3798
// pTableIter can be NULL, no need to handle the return value
3799
static int32_t resetTableListIndex(SReaderStatus* pStatus, const char* id) {
23,958,899✔
3800
  int32_t        code = TSDB_CODE_SUCCESS;
23,958,899✔
3801
  int32_t        lino = 0;
23,958,899✔
3802
  STableUidList* pList = NULL;
23,958,899✔
3803

3804
  TSDB_CHECK_NULL(pStatus, code, lino, _end, TSDB_CODE_INVALID_PARA);
23,958,899!
3805

3806
  pList = &pStatus->uidList;
23,958,899✔
3807

3808
  pList->currentIndex = 0;
23,958,899✔
3809
  uint64_t uid = pList->tableUidList[0];
23,958,899✔
3810
  pStatus->pTableIter = tSimpleHashGet(pStatus->pTableMap, &uid, sizeof(uid));
23,958,899✔
3811
  if (pStatus->pTableIter == NULL) {
23,971,818!
3812
    tsdbError("%s failed to load tableBlockScanInfo for uid:%" PRId64 ", code: internal error", id, uid);
×
3813
    TSDB_CHECK_NULL(pStatus->pTableIter, code, lino, _end, TSDB_CODE_INTERNAL_ERROR);
×
3814
  }
3815

3816
_end:
23,971,818✔
3817
  if (code != TSDB_CODE_SUCCESS) {
23,971,818!
3818
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3819
  }
3820
  return code;
23,968,536✔
3821
}
3822

3823
static void resetPreFilesetMemTableListIndex(SReaderStatus* pStatus) {
2,114,218✔
3824
  STableUidList* pList = &pStatus->procMemUidList;
2,114,218✔
3825

3826
  pList->currentIndex = 0;
2,114,218✔
3827
  uint64_t uid = pList->tableUidList[0];
2,114,218✔
3828
  pStatus->pProcMemTableIter = tSimpleHashGet(pStatus->pTableMap, &uid, sizeof(uid));
2,114,218✔
3829
}
2,114,339✔
3830

3831
static bool moveToNextTable(STableUidList* pOrderedCheckInfo, SReaderStatus* pStatus) {
41,506,611✔
3832
  pOrderedCheckInfo->currentIndex += 1;
41,506,611✔
3833
  if (pOrderedCheckInfo->currentIndex >= tSimpleHashGetSize(pStatus->pTableMap)) {
41,506,611✔
3834
    pStatus->pTableIter = NULL;
17,512,310✔
3835
    return false;
17,512,310✔
3836
  }
3837

3838
  uint64_t uid = pOrderedCheckInfo->tableUidList[pOrderedCheckInfo->currentIndex];
23,994,819✔
3839
  pStatus->pTableIter = tSimpleHashGet(pStatus->pTableMap, &uid, sizeof(uid));
23,994,819✔
3840
  return (pStatus->pTableIter != NULL);
24,005,176✔
3841
}
3842

3843
static bool moveToNextTableForPreFileSetMem(SReaderStatus* pStatus) {
5,188,830✔
3844
  STableUidList* pUidList = &pStatus->procMemUidList;
5,188,830✔
3845
  pUidList->currentIndex += 1;
5,188,830✔
3846
  if (pUidList->currentIndex >= tSimpleHashGetSize(pStatus->pTableMap)) {
5,188,830✔
3847
    pStatus->pProcMemTableIter = NULL;
2,114,377✔
3848
    return false;
2,114,377✔
3849
  }
3850

3851
  uint64_t uid = pUidList->tableUidList[pUidList->currentIndex];
3,074,455✔
3852
  pStatus->pProcMemTableIter = tSimpleHashGet(pStatus->pTableMap, &uid, sizeof(uid));
3,074,455✔
3853
  return (pStatus->pProcMemTableIter != NULL);
3,074,596✔
3854
}
3855

3856
static int32_t buildCleanBlockFromSttFiles(STsdbReader* pReader, STableBlockScanInfo* pScanInfo) {
36,432✔
3857
  int32_t          code = TSDB_CODE_SUCCESS;
36,432✔
3858
  int32_t          lino = 0;
36,432✔
3859
  SReaderStatus*   pStatus = NULL;
36,432✔
3860
  SSttBlockReader* pSttBlockReader = NULL;
36,432✔
3861
  SSDataBlock*     pResBlock = NULL;
36,432✔
3862
  SDataBlockInfo*  pInfo = NULL;
36,432✔
3863
  bool             asc = false;
36,432✔
3864

3865
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
36,432!
3866
  TSDB_CHECK_NULL(pScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
36,432!
3867

3868
  pStatus = &pReader->status;
36,432✔
3869
  pSttBlockReader = pStatus->fileIter.pSttBlockReader;
36,432✔
3870
  pResBlock = pReader->resBlockInfo.pResBlock;
36,432✔
3871

3872
  asc = ASCENDING_TRAVERSE(pReader->info.order);
36,432✔
3873
  code = blockDataEnsureCapacity(pResBlock, pScanInfo->numOfRowsInStt);
36,432✔
3874
  TSDB_CHECK_CODE(code, lino, _end);
36,424!
3875

3876
  pInfo = &pResBlock->info;
36,424✔
3877
  pInfo->rows = pScanInfo->numOfRowsInStt;
36,424✔
3878
  pInfo->id.uid = pScanInfo->uid;
36,424✔
3879
  pInfo->dataLoad = 1;
36,424✔
3880
  pInfo->window.skey = pScanInfo->sttRange.skey.ts;
36,424✔
3881
  pInfo->window.ekey = pScanInfo->sttRange.ekey.ts;
36,424✔
3882

3883
  setComposedBlockFlag(pReader, true);
36,424✔
3884

3885
  pScanInfo->sttKeyInfo.nextProcKey.ts = asc ? pScanInfo->sttRange.ekey.ts + 1 : pScanInfo->sttRange.skey.ts - 1;
36,412!
3886
  pScanInfo->sttKeyInfo.status = STT_FILE_NO_DATA;
36,412✔
3887

3888
  if (asc) {
36,412!
3889
    tRowKeyAssign(&pScanInfo->lastProcKey, &pScanInfo->sttRange.ekey);
36,421✔
3890
  } else {
3891
    tRowKeyAssign(&pScanInfo->lastProcKey, &pScanInfo->sttRange.skey);
×
3892
  }
3893

3894
  pScanInfo->sttBlockReturned = true;
36,424✔
3895
  pSttBlockReader->mergeTree.pIter = NULL;
36,424✔
3896

3897
  tsdbDebug("%p uid:%" PRId64 " return clean stt block as one, brange:%" PRId64 "-%" PRId64 " rows:%" PRId64 " %s",
36,424✔
3898
            pReader, pResBlock->info.id.uid, pResBlock->info.window.skey, pResBlock->info.window.ekey,
3899
            pResBlock->info.rows, pReader->idStr);
3900

3901
_end:
36,422✔
3902
  if (code != TSDB_CODE_SUCCESS) {
36,424!
3903
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3904
  }
3905
  return code;
36,425✔
3906
}
3907

3908
static int32_t buildCleanBlockFromDataFiles(STsdbReader* pReader, STableBlockScanInfo* pScanInfo,
13,359,396✔
3909
                                            SFileDataBlockInfo* pBlockInfo, int32_t blockIndex) {
3910
  int32_t         code = TSDB_CODE_SUCCESS;
13,359,396✔
3911
  int32_t         lino = 0;
13,359,396✔
3912
  SReaderStatus*  pStatus = NULL;
13,359,396✔
3913
  SDataBlockInfo* pInfo = NULL;
13,359,396✔
3914
  bool            asc = false;
13,359,396✔
3915

3916
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,359,396!
3917
  TSDB_CHECK_NULL(pScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,359,396!
3918
  TSDB_CHECK_NULL(pBlockInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,359,396!
3919

3920
  // whole data block is required, return it directly
3921
  pStatus = &pReader->status;
13,359,396✔
3922
  pInfo = &pReader->resBlockInfo.pResBlock->info;
13,359,396✔
3923
  asc = ASCENDING_TRAVERSE(pReader->info.order);
13,359,396✔
3924

3925
  pInfo->rows = pBlockInfo->numRow;
13,359,396✔
3926
  pInfo->id.uid = pScanInfo->uid;
13,359,396✔
3927
  pInfo->dataLoad = 0;
13,359,396✔
3928
  pInfo->version = pReader->info.verRange.maxVer;
13,359,396✔
3929
  pInfo->window = (STimeWindow){.skey = pBlockInfo->firstKey, .ekey = pBlockInfo->lastKey};
13,359,396✔
3930

3931
  if (pReader->suppInfo.numOfPks > 0) {
13,359,396✔
3932
    if (IS_NUMERIC_TYPE(pReader->suppInfo.pk.type)) {
162,986!
3933
      VALUE_SET_TRIVIAL_DATUM(pInfo->pks, pBlockInfo->firstPk.val);
162,985✔
3934
      VALUE_SET_TRIVIAL_DATUM(pInfo->pks + 1, pBlockInfo->lastPk.val);
162,985✔
3935
    } else {
3936
      (void)memcpy(pInfo->pks[0].pData, varDataVal(pBlockInfo->firstPk.pData), varDataLen(pBlockInfo->firstPk.pData));
1✔
3937
      (void)memcpy(pInfo->pks[1].pData, varDataVal(pBlockInfo->lastPk.pData), varDataLen(pBlockInfo->lastPk.pData));
1✔
3938

3939
      pInfo->pks[0].nData = varDataLen(pBlockInfo->firstPk.pData);
1✔
3940
      pInfo->pks[1].nData = varDataLen(pBlockInfo->lastPk.pData);
1✔
3941
    }
3942
  }
3943

3944
  // update the last key for the corresponding table
3945
  setComposedBlockFlag(pReader, false);
13,359,396✔
3946
  setBlockAllDumped(&pStatus->fBlockDumpInfo, pBlockInfo->lastKey, pReader->info.order);
13,359,240✔
3947
  code = updateLastKeyInfo(&pScanInfo->lastProcKey, pBlockInfo, pInfo, pReader->suppInfo.numOfPks, asc);
13,359,315✔
3948
  TSDB_CHECK_CODE(code, lino, _end);
13,359,294!
3949

3950
  tsdbDebug("%p uid:%" PRIu64
13,359,294✔
3951
            " clean file block retrieved from file, global index:%d, "
3952
            "table index:%d, rows:%d, brange:%" PRId64 "-%" PRId64 ", %s",
3953
            pReader, pScanInfo->uid, blockIndex, pBlockInfo->tbBlockIdx, pBlockInfo->numRow, pBlockInfo->firstKey,
3954
            pBlockInfo->lastKey, pReader->idStr);
3955

3956
_end:
13,357,695✔
3957
  if (code != TSDB_CODE_SUCCESS) {
13,359,294!
3958
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3959
  }
3960
  return code;
13,359,320✔
3961
}
3962

3963
static int32_t doLoadSttBlockSequentially(STsdbReader* pReader) {
17,029,063✔
3964
  int32_t              code = TSDB_CODE_SUCCESS;
17,029,063✔
3965
  int32_t              lino = 0;
17,029,063✔
3966
  SReaderStatus*       pStatus = NULL;
17,029,063✔
3967
  SSttBlockReader*     pSttBlockReader = NULL;
17,029,063✔
3968
  STableUidList*       pUidList = NULL;
17,029,063✔
3969
  SSDataBlock*         pResBlock = NULL;
17,029,063✔
3970
  STableBlockScanInfo* pScanInfo = NULL;
17,029,063✔
3971

3972
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
17,029,063!
3973

3974
  pStatus = &pReader->status;
17,029,063✔
3975
  pSttBlockReader = pStatus->fileIter.pSttBlockReader;
17,029,063✔
3976
  pUidList = &pStatus->uidList;
17,029,063✔
3977

3978
  if (tSimpleHashGetSize(pStatus->pTableMap) == 0) {
17,029,063!
3979
    goto _end;
×
3980
  }
3981

3982
  pResBlock = pReader->resBlockInfo.pResBlock;
17,030,156✔
3983

3984
  while (1) {
10,792,986✔
3985
    if (pReader->code != TSDB_CODE_SUCCESS) {
27,823,142!
3986
      tsdbWarn("tsdb reader is stopped ASAP, code:%s, %s", tstrerror(pReader->code), pReader->idStr);
×
3987
      code = pReader->code;
×
3988
      TSDB_CHECK_CODE(code, lino, _end);
×
3989
    }
3990

3991
    // load the last data block of current table
3992
    if (pStatus->pTableIter == NULL) {
27,819,456!
3993
      tsdbError("table Iter is null, invalid pScanInfo, try next table %s", pReader->idStr);
×
3994
      bool hasNexTable = moveToNextTable(pUidList, pStatus);
×
3995
      if (!hasNexTable) {
×
3996
        break;
×
3997
      }
3998

3999
      continue;
×
4000
    } else {
4001
      pScanInfo = *(STableBlockScanInfo**)pStatus->pTableIter;
27,819,456✔
4002
    }
4003

4004
    if (pReader->pIgnoreTables && taosHashGet(*pReader->pIgnoreTables, &pScanInfo->uid, sizeof(pScanInfo->uid))) {
27,819,456✔
4005
      // reset the index in last block when handing a new file
4006
      bool hasNexTable = moveToNextTable(pUidList, pStatus);
197,007✔
4007
      if (!hasNexTable) {
197,023✔
4008
        break;
2,245✔
4009
      }
4010

4011
      continue;
194,778✔
4012
    }
4013

4014
    initSttBlockReader(pSttBlockReader, pScanInfo, pReader);
27,620,971✔
4015
    code = pReader->code;
27,629,241✔
4016
    TSDB_CHECK_CODE(code, lino, _end);
27,629,241!
4017

4018
    if (!hasDataInSttBlock(pScanInfo)) {
27,629,241✔
4019
      bool hasNexTable = moveToNextTable(pUidList, pStatus);
20,039,939✔
4020
      if (!hasNexTable) {
20,039,039✔
4021
        break;
9,440,831✔
4022
      }
4023

4024
      continue;
10,598,208✔
4025
    }
4026

4027
    // if only require the total rows, no need to load data from stt file if it is clean stt blocks
4028
    if (pReader->info.execMode == READER_EXEC_ROWS && pScanInfo->cleanSttBlocks) {
7,589,560✔
4029
      code = buildCleanBlockFromSttFiles(pReader, pScanInfo);
36,436✔
4030
      TSDB_CHECK_CODE(code, lino, _end);
36,423!
4031
      break;
36,423✔
4032
    }
4033

4034
    int64_t st = taosGetTimestampUs();
7,558,192✔
4035
    while (1) {
4036
      // no data in stt block and block, no need to proceed.
4037
      if (!hasDataInSttBlock(pScanInfo)) {
865,083,741✔
4038
        break;
7,493,948✔
4039
      }
4040

4041
      code = buildComposedDataBlockImpl(pReader, pScanInfo, &pReader->status.fileBlockData, pSttBlockReader);
856,882,626✔
4042
      TSDB_CHECK_CODE(code, lino, _end);
857,589,138!
4043

4044
      if (pResBlock->info.rows >= pReader->resBlockInfo.capacity) {
857,589,138✔
4045
        break;
63,589✔
4046
      }
4047
    }
4048

4049
    double el = (taosGetTimestampUs() - st) / 1000.0;
7,557,740✔
4050
    updateComposedBlockInfo(pReader, el, pScanInfo);
7,557,740✔
4051

4052
    if (pResBlock->info.rows > 0) {
7,557,401!
4053
      tsdbDebug("%p uid:%" PRIu64 ", composed data block created, brange:%" PRIu64 "-%" PRIu64 " rows:%" PRId64
7,557,777✔
4054
                ", elapsed time:%.2f ms %s",
4055
                pReader, pResBlock->info.id.uid, pResBlock->info.window.skey, pResBlock->info.window.ekey,
4056
                pResBlock->info.rows, el, pReader->idStr);
4057
      break;
7,557,871✔
4058
    }
4059

4060
    // current table is exhausted, let's try next table
4061
    bool hasNexTable = moveToNextTable(pUidList, pStatus);
×
4062
    if (!hasNexTable) {
×
4063
      break;
×
4064
    }
4065
  }
4066

4067
_end:
17,037,370✔
4068
  if (code != TSDB_CODE_SUCCESS) {
17,037,370!
4069
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4070
  }
4071
  return code;
17,034,677✔
4072
}
4073

4074
// current active data block not overlap with the stt-files/stt-blocks
4075
static bool notOverlapWithFiles(SFileDataBlockInfo* pBlockInfo, STableBlockScanInfo* pScanInfo, bool asc) {
13,359,904✔
4076
  if ((!hasDataInSttBlock(pScanInfo)) || (pScanInfo->cleanSttBlocks == true)) {
13,359,904✔
4077
    return true;
8,737,868✔
4078
  } else {
4079
    int64_t keyInStt = pScanInfo->sttKeyInfo.nextProcKey.ts;
4,622,052✔
4080
    return (asc && pBlockInfo->lastKey < keyInStt) || (!asc && pBlockInfo->firstKey > keyInStt);
4,622,052✔
4081
  }
4082
}
4083

4084
static int32_t doBuildDataBlock(STsdbReader* pReader) {
13,410,352✔
4085
  int32_t              code = TSDB_CODE_SUCCESS;
13,410,352✔
4086
  int32_t              lino = 0;
13,410,352✔
4087
  SReaderStatus*       pStatus = NULL;
13,410,352✔
4088
  SDataBlockIter*      pBlockIter = NULL;
13,410,352✔
4089
  STableBlockScanInfo* pScanInfo = NULL;
13,410,352✔
4090
  SFileDataBlockInfo*  pBlockInfo = NULL;
13,410,352✔
4091
  SSttBlockReader*     pSttBlockReader = NULL;
13,410,352✔
4092
  TSDBKEY              keyInBuf;
4093
  bool                 asc = false;
13,410,352✔
4094

4095
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,410,352!
4096

4097
  pStatus = &pReader->status;
13,410,352✔
4098
  pBlockIter = &pStatus->blockIter;
13,410,352✔
4099
  pSttBlockReader = pReader->status.fileIter.pSttBlockReader;
13,410,352✔
4100
  asc = ASCENDING_TRAVERSE(pReader->info.order);
13,410,352✔
4101

4102
  code = getCurrentBlockInfo(pBlockIter, &pBlockInfo, pReader->idStr);
13,410,352✔
4103
  TSDB_CHECK_CODE(code, lino, _end);
13,410,427!
4104

4105
  if (pReader->pIgnoreTables && taosHashGet(*pReader->pIgnoreTables, &pBlockInfo->uid, sizeof(pBlockInfo->uid))) {
13,410,427✔
4106
    setBlockAllDumped(&pStatus->fBlockDumpInfo, pBlockInfo->lastKey, pReader->info.order);
697✔
4107
    goto _end;
709✔
4108
  }
4109

4110
  code = pReader->code;
13,409,616✔
4111
  TSDB_CHECK_CODE(code, lino, _end);
13,409,616!
4112

4113
  code = getTableBlockScanInfo(pReader->status.pTableMap, pBlockInfo->uid, &pScanInfo, pReader->idStr);
13,409,616✔
4114
  TSDB_CHECK_CODE(code, lino, _end);
13,409,806!
4115

4116
  if (pScanInfo->sttKeyInfo.status == STT_FILE_READER_UNINIT) {
13,409,806✔
4117
    initSttBlockReader(pSttBlockReader, pScanInfo, pReader);
411,366✔
4118
    code = pReader->code;
411,354✔
4119
    TSDB_CHECK_CODE(code, lino, _end);
411,354!
4120
  }
4121

4122
  code = getCurrentKeyInBuf(pScanInfo, pReader, &keyInBuf);
13,409,794✔
4123
  TSDB_CHECK_CODE(code, lino, _end);
13,409,634!
4124
  bool load = false;
13,409,634✔
4125
  code = fileBlockShouldLoad(pReader, pBlockInfo, pScanInfo, keyInBuf, &load);
13,409,634✔
4126
  TSDB_CHECK_CODE(code, lino, _end);
13,409,693!
4127
  if (load) {
13,409,693✔
4128
    code = doLoadFileBlockData(pReader, pBlockIter, &pStatus->fileBlockData, pScanInfo->uid);
49,355✔
4129
    TSDB_CHECK_CODE(code, lino, _end);
49,360!
4130

4131
    // build composed data block
4132
    code = buildComposedDataBlock(pReader);
49,360✔
4133
    TSDB_CHECK_CODE(code, lino, _end);
49,359!
4134
  } else if (bufferDataInFileBlockGap(keyInBuf, pBlockInfo, pScanInfo, pReader->info.order)) {
13,360,338✔
4135
    // data in memory that are earlier than current file block and stt blocks
4136
    // rows in buffer should be less than the file block in asc, greater than file block in desc
4137
    int64_t endKey = getBoarderKeyInFiles(pBlockInfo, pScanInfo, pReader->info.order);
242✔
4138
    code = buildDataBlockFromBuf(pReader, pScanInfo, endKey);
242✔
4139
    TSDB_CHECK_CODE(code, lino, _end);
242!
4140
  } else {
4141
    if (notOverlapWithFiles(pBlockInfo, pScanInfo, asc)) {
13,360,095✔
4142
      int64_t keyInStt = pScanInfo->sttKeyInfo.nextProcKey.ts;
13,359,472✔
4143

4144
      if ((!hasDataInSttBlock(pScanInfo)) || (asc && pBlockInfo->lastKey < keyInStt) ||
13,359,472✔
4145
          (!asc && pBlockInfo->firstKey > keyInStt)) {
73!
4146
        // the stt blocks may located in the gap of different data block, but the whole sttRange may overlap with the
4147
        // data block, so the overlap check is invalid actually.
4148
        code = buildCleanBlockFromDataFiles(pReader, pScanInfo, pBlockInfo, pBlockIter->index);
13,359,541✔
4149
        TSDB_CHECK_CODE(code, lino, _end);
13,359,312!
4150
      } else {  // clean stt block
4151
        TSDB_CHECK_CONDITION((pReader->info.execMode == READER_EXEC_ROWS) && (pSttBlockReader->mergeTree.pIter == NULL),
1!
4152
                             code, lino, _end, TSDB_CODE_INTERNAL_ERROR);
4153
        code = buildCleanBlockFromSttFiles(pReader, pScanInfo);
1✔
4154
        TSDB_CHECK_CODE(code, lino, _end);
1!
4155
      }
4156
    } else {
4157
      SBlockData* pBData = &pReader->status.fileBlockData;
511✔
4158
      tBlockDataReset(pBData);
511✔
4159

4160
      SSDataBlock* pResBlock = pReader->resBlockInfo.pResBlock;
536✔
4161

4162
      tsdbDebug("load data in stt block firstly %s", pReader->idStr);
536!
4163
      int64_t st = taosGetTimestampUs();
536✔
4164

4165
      // let's load data from stt files, make sure clear the cleanStt block flag before load the data from stt files
4166
      initSttBlockReader(pSttBlockReader, pScanInfo, pReader);
536✔
4167
      code = pReader->code;
536✔
4168
      TSDB_CHECK_CODE(code, lino, _end);
536!
4169

4170
      // no data in stt block, no need to proceed.
4171
      while (hasDataInSttBlock(pScanInfo)) {
1,637,670✔
4172
        TSDB_CHECK_CONDITION(pScanInfo->sttKeyInfo.status == STT_FILE_HAS_DATA, code, lino, _end,
1,637,347!
4173
                             TSDB_CODE_INTERNAL_ERROR);
4174

4175
        code = buildComposedDataBlockImpl(pReader, pScanInfo, &pReader->status.fileBlockData, pSttBlockReader);
1,637,347✔
4176
        TSDB_CHECK_CODE(code, lino, _end);
1,637,526!
4177

4178
        if (pResBlock->info.rows >= pReader->resBlockInfo.capacity) {
1,637,526✔
4179
          break;
380✔
4180
        }
4181

4182
        // data in stt now overlaps with current active file data block, need to composed with file data block.
4183
        SRowKey* pSttKey = getCurrentKeyInSttBlock(pSttBlockReader);
1,637,146✔
4184
        if ((pSttKey->ts >= pBlockInfo->firstKey && asc) || (pSttKey->ts <= pBlockInfo->lastKey && (!asc))) {
1,637,146✔
4185
          tsdbDebug("%p lastKeyInStt:%" PRId64 ", overlap with file block, brange:%" PRId64 "-%" PRId64 " %s", pReader,
12!
4186
                    pSttKey->ts, pBlockInfo->firstKey, pBlockInfo->lastKey, pReader->idStr);
4187
          break;
13✔
4188
        }
4189
      }
4190

4191
      double el = (taosGetTimestampUs() - st) / 1000.0;
536✔
4192
      updateComposedBlockInfo(pReader, el, pScanInfo);
536✔
4193

4194
      if (pResBlock->info.rows > 0) {
536!
4195
        tsdbDebug("%p uid:%" PRIu64 ", composed data block created, brange:%" PRIu64 "-%" PRIu64 " rows:%" PRId64
536!
4196
                  ", elapsed time:%.2f ms %s",
4197
                  pReader, pResBlock->info.id.uid, pResBlock->info.window.skey, pResBlock->info.window.ekey,
4198
                  pResBlock->info.rows, el, pReader->idStr);
4199
      }
4200
    }
4201
  }
4202

4203
  code = pReader->code;
13,409,413✔
4204
  TSDB_CHECK_CODE(code, lino, _end);
13,409,413!
4205

4206
_end:
13,409,413✔
4207
  if (code != TSDB_CODE_SUCCESS) {
13,410,122!
4208
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4209
  }
4210
  return code;
13,410,157✔
4211
}
4212

4213
static int32_t buildBlockFromBufferSeqForPreFileset(STsdbReader* pReader, int64_t endKey) {
2,567,893✔
4214
  int32_t        code = TSDB_CODE_SUCCESS;
2,567,893✔
4215
  int32_t        lino = 0;
2,567,893✔
4216
  SReaderStatus* pStatus = NULL;
2,567,893✔
4217

4218
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,567,893!
4219

4220
  pStatus = &pReader->status;
2,567,893✔
4221

4222
  tsdbDebug("seq load data blocks from cache that preceeds fileset %d, %s", pReader->status.pCurrentFileset->fid,
2,567,893!
4223
            pReader->idStr);
4224

4225
  while (1) {
3,074,533✔
4226
    if (pReader->code != TSDB_CODE_SUCCESS) {
5,642,474!
4227
      tsdbWarn("tsdb reader is stopped ASAP, code:%s, %s", tstrerror(pReader->code), pReader->idStr);
×
4228
      code = pReader->code;
×
4229
      TSDB_CHECK_CODE(code, lino, _end);
×
4230
    }
4231

4232
    STableBlockScanInfo** pBlockScanInfo = pStatus->pProcMemTableIter;
5,642,474✔
4233
    if (pReader->pIgnoreTables &&
11,284,923✔
4234
        taosHashGet(*pReader->pIgnoreTables, &(*pBlockScanInfo)->uid, sizeof((*pBlockScanInfo)->uid))) {
5,642,463✔
4235
      bool hasNexTable = moveToNextTableForPreFileSetMem(pStatus);
114,708✔
4236
      if (!hasNexTable) {
114,711✔
4237
        break;
48✔
4238
      }
4239
      continue;
114,663✔
4240
    }
4241

4242
    code = initMemDataIterator(*pBlockScanInfo, pReader);
5,527,752✔
4243
    TSDB_CHECK_CODE(code, lino, _end);
5,527,729!
4244

4245
    code = initDelSkylineIterator(*pBlockScanInfo, pReader->info.order, &pReader->cost);
5,527,729✔
4246
    TSDB_CHECK_CODE(code, lino, _end);
5,527,652!
4247

4248
    code = buildDataBlockFromBuf(pReader, *pBlockScanInfo, endKey);
5,527,652✔
4249
    TSDB_CHECK_CODE(code, lino, _end);
5,527,767!
4250

4251
    if (pReader->resBlockInfo.pResBlock->info.rows > 0) {
5,527,767✔
4252
      break;
453,657✔
4253
    }
4254

4255
    // current table is exhausted, let's try next table
4256
    bool hasNexTable = moveToNextTableForPreFileSetMem(pStatus);
5,074,110✔
4257
    if (!hasNexTable) {
5,074,199✔
4258
      break;
2,114,329✔
4259
    }
4260
  }
4261

4262
_end:
2,568,034✔
4263
  if (code != TSDB_CODE_SUCCESS) {
2,568,034!
4264
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4265
  }
4266
  return code;
2,568,026✔
4267
}
4268

4269
static int32_t buildBlockFromBufferSequentially(STsdbReader* pReader, int64_t endKey) {
13,408,622✔
4270
  int32_t        code = TSDB_CODE_SUCCESS;
13,408,622✔
4271
  int32_t        lino = 0;
13,408,622✔
4272
  SReaderStatus* pStatus = NULL;
13,408,622✔
4273
  STableUidList* pUidList = NULL;
13,408,622✔
4274

4275
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,408,622!
4276

4277
  pStatus = &pReader->status;
13,408,622✔
4278
  pUidList = &pStatus->uidList;
13,408,622✔
4279

4280
  tsdbDebug("seq load data blocks from cache, %s", pReader->idStr);
13,408,622✔
4281

4282
  while (1) {
13,205,928✔
4283
    if (pReader->code != TSDB_CODE_SUCCESS) {
26,618,728!
4284
      tsdbWarn("tsdb reader is stopped ASAP, code:%s, %s", tstrerror(pReader->code), pReader->idStr);
×
4285
      code = pReader->code;
×
4286
      TSDB_CHECK_CODE(code, lino, _end);
×
4287
    }
4288

4289
    STableBlockScanInfo** pBlockScanInfo = pStatus->pTableIter;
26,618,313✔
4290
    if (pBlockScanInfo == NULL || *pBlockScanInfo == NULL) {
26,618,313!
4291
      break;
4292
    }
4293

4294
    if (pReader->pIgnoreTables &&
53,168,440✔
4295
        taosHashGet(*pReader->pIgnoreTables, &(*pBlockScanInfo)->uid, sizeof((*pBlockScanInfo)->uid))) {
26,553,535✔
4296
      bool hasNexTable = moveToNextTable(pUidList, pStatus);
96,493✔
4297
      if (!hasNexTable) {
96,496✔
4298
        break;
1,861✔
4299
      }
4300
      continue;
94,635✔
4301
    }
4302

4303
    code = initMemDataIterator(*pBlockScanInfo, pReader);
26,518,412✔
4304
    TSDB_CHECK_CODE(code, lino, _end);
26,526,946!
4305

4306
    code = initDelSkylineIterator(*pBlockScanInfo, pReader->info.order, &pReader->cost);
26,526,946✔
4307
    TSDB_CHECK_CODE(code, lino, _end);
26,523,504!
4308

4309
    code = buildDataBlockFromBuf(pReader, *pBlockScanInfo, endKey);
26,523,504✔
4310
    TSDB_CHECK_CODE(code, lino, _end);
26,525,095✔
4311

4312
    if (pReader->resBlockInfo.pResBlock->info.rows > 0) {
26,525,085✔
4313
      break;
5,349,711✔
4314
    }
4315

4316
    // current table is exhausted, let's try next table
4317
    bool hasNexTable = moveToNextTable(pUidList, pStatus);
21,175,374✔
4318
    if (!hasNexTable) {
21,178,907✔
4319
      break;
8,067,614✔
4320
    }
4321
  }
4322

4323
_end:
×
4324
  if (code != TSDB_CODE_SUCCESS) {
13,417,363✔
4325
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
10!
4326
  }
4327
  return code;
13,418,621✔
4328
}
4329

4330
// set the correct start position in case of the first/last file block, according to the query time window
4331
static int32_t initBlockDumpInfo(STsdbReader* pReader, SDataBlockIter* pBlockIter) {
22,828,929✔
4332
  int32_t             code = TSDB_CODE_SUCCESS;
22,828,929✔
4333
  int32_t             lino = 0;
22,828,929✔
4334
  SFileDataBlockInfo* pBlockInfo = NULL;
22,828,929✔
4335
  SReaderStatus*      pStatus = NULL;
22,828,929✔
4336
  SFileBlockDumpInfo* pDumpInfo = NULL;
22,828,929✔
4337

4338
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
22,828,929!
4339

4340
  pStatus = &pReader->status;
22,828,929✔
4341
  pDumpInfo = &pStatus->fBlockDumpInfo;
22,828,929✔
4342

4343
  code = getCurrentBlockInfo(pBlockIter, &pBlockInfo, pReader->idStr);
22,828,929✔
4344
  if (code == TSDB_CODE_SUCCESS) {
22,830,428✔
4345
    pDumpInfo->totalRows = pBlockInfo->numRow;
13,409,706✔
4346
    pDumpInfo->rowIndex = ASCENDING_TRAVERSE(pReader->info.order) ? 0 : pBlockInfo->numRow - 1;
13,409,706✔
4347
  } else {
4348
    pDumpInfo->totalRows = 0;
9,420,722✔
4349
    pDumpInfo->rowIndex = 0;
9,420,722✔
4350
    code = TSDB_CODE_SUCCESS;
9,420,722✔
4351
  }
4352

4353
  pDumpInfo->allDumped = false;
22,830,428✔
4354

4355
_end:
22,830,428✔
4356
  if (code != TSDB_CODE_SUCCESS) {
22,830,428!
4357
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4358
  }
4359
  return code;
22,828,860✔
4360
}
4361

4362
static int32_t initForFirstBlockInFile(STsdbReader* pReader, SDataBlockIter* pBlockIter) {
13,357,031✔
4363
  int32_t      code = TSDB_CODE_SUCCESS;
13,357,031✔
4364
  int32_t      lino = 0;
13,357,031✔
4365
  SBlockNumber num = {0};
13,357,031✔
4366
  SArray*      pTableList = NULL;
13,357,031✔
4367

4368
  pTableList = taosArrayInit(40, POINTER_BYTES);
13,357,031✔
4369
  TSDB_CHECK_NULL(pTableList, code, lino, _end, terrno);
13,359,793!
4370

4371
  code = moveToNextFile(pReader, &num, pTableList);
13,359,793✔
4372
  TSDB_CHECK_CODE(code, lino, _end);
13,359,717!
4373

4374
  // all data files are consumed, try data in buffer
4375
  if (num.numOfBlocks + num.numOfSttFiles == 0) {
13,359,717✔
4376
    pReader->status.loadFromFile = false;
3,796,908✔
4377
    goto _end;
3,796,908✔
4378
  }
4379

4380
  // initialize the block iterator for a new fileset
4381
  if (num.numOfBlocks > 0) {
9,562,809✔
4382
    code = initBlockIterator(pReader, pBlockIter, num.numOfBlocks, pTableList);
137,350✔
4383
    TSDB_CHECK_CODE(code, lino, _end);
137,345!
4384
  } else {  // no block data, only last block exists
4385
    tBlockDataReset(&pReader->status.fileBlockData);
9,425,459✔
4386
    code = resetDataBlockIterator(pBlockIter, pReader->info.order, shouldFreePkBuf(&pReader->suppInfo), pReader->idStr);
9,418,990✔
4387
    TSDB_CHECK_CODE(code, lino, _end);
9,423,568!
4388

4389
    code = resetTableListIndex(&pReader->status, pReader->idStr);
9,423,568✔
4390
    TSDB_CHECK_CODE(code, lino, _end);
9,425,698!
4391
  }
4392

4393
  code = initBlockDumpInfo(pReader, pBlockIter);
9,563,043✔
4394
  TSDB_CHECK_CODE(code, lino, _end);
9,557,335!
4395

4396
_end:
9,557,335✔
4397
  if (code != TSDB_CODE_SUCCESS) {
13,354,243!
4398
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4399
  }
4400
  if (pTableList) {
13,355,005!
4401
    taosArrayDestroy(pTableList);
13,355,005✔
4402
  }
4403
  return code;
13,361,494✔
4404
}
4405

4406
static bool fileBlockPartiallyRead(SFileBlockDumpInfo* pDumpInfo, bool asc) {
13,427,885✔
4407
  return (!pDumpInfo->allDumped) &&
13,446,166✔
4408
         ((pDumpInfo->rowIndex > 0 && asc) || (pDumpInfo->rowIndex < (pDumpInfo->totalRows - 1) && (!asc)));
18,281✔
4409
}
4410

4411
typedef enum {
4412
  TSDB_READ_RETURN = 0x1,
4413
  TSDB_READ_CONTINUE = 0x2,
4414
} ERetrieveType;
4415

4416
static int32_t doReadDataFromSttFiles(STsdbReader* pReader, ERetrieveType* pReturnType) {
10,834,209✔
4417
  int32_t         code = TSDB_CODE_SUCCESS;
10,834,209✔
4418
  int32_t         lino = 0;
10,834,209✔
4419
  SSDataBlock*    pResBlock = NULL;
10,834,209✔
4420
  SDataBlockIter* pBlockIter = NULL;
10,834,209✔
4421

4422
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
10,834,209!
4423
  TSDB_CHECK_NULL(pReturnType, code, lino, _end, TSDB_CODE_INVALID_PARA);
10,834,209!
4424

4425
  pResBlock = pReader->resBlockInfo.pResBlock;
10,834,209✔
4426
  pBlockIter = &pReader->status.blockIter;
10,834,209✔
4427

4428
  *pReturnType = TSDB_READ_RETURN;
10,834,209✔
4429

4430
  tsdbDebug("seq load data blocks from stt files %s", pReader->idStr);
10,834,209✔
4431

4432
  while (1) {
4433
    code = doLoadSttBlockSequentially(pReader);
17,033,382✔
4434
    TSDB_CHECK_CODE(code, lino, _end);
17,033,776!
4435

4436
    if (pResBlock->info.rows > 0) {
17,033,776✔
4437
      goto _end;
7,594,089✔
4438
    }
4439

4440
    // all data blocks are checked in this stt file, now let's try the next file set
4441
    TSDB_CHECK_CONDITION(pReader->status.pTableIter == NULL, code, lino, _end, TSDB_CODE_INTERNAL_ERROR);
9,439,687!
4442

4443
    code = initForFirstBlockInFile(pReader, pBlockIter);
9,439,687✔
4444
    TSDB_CHECK_CODE(code, lino, _end);
9,442,990!
4445

4446
    // all the data files are completely checked
4447
    if (pReader->status.loadFromFile == false) {
9,442,990✔
4448
      goto _end;
2,830,430✔
4449
    }
4450

4451
    if (pReader->status.bProcMemPreFileset) {
6,612,560✔
4452
      code = buildFromPreFilesetBuffer(pReader);
1,975,311✔
4453
      TSDB_CHECK_CODE(code, lino, _end);
1,975,296!
4454
      if (pResBlock->info.rows > 0) {
1,975,296✔
4455
        pReader->status.processingMemPreFileSet = true;
293,196✔
4456
        goto _end;
293,196✔
4457
      }
4458
    }
4459

4460
    if (pBlockIter->numOfBlocks > 0) {  // there are data blocks existed.
6,319,349✔
4461
      *pReturnType = TSDB_READ_CONTINUE;
121,017✔
4462
      goto _end;
121,017✔
4463
    } else {  // all blocks in data file are checked, let's check the data in stt-files
4464
      code = resetTableListIndex(&pReader->status, pReader->idStr);
6,198,332✔
4465
      TSDB_CHECK_CODE(code, lino, _end);
6,198,441!
4466
    }
4467
  }
4468

4469
_end:
10,838,732✔
4470
  if (code != TSDB_CODE_SUCCESS) {
10,838,732!
4471
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4472
  }
4473
  return code;
10,837,577✔
4474
}
4475

4476
static int32_t buildBlockFromFiles(STsdbReader* pReader) {
24,096,506✔
4477
  int32_t             code = TSDB_CODE_SUCCESS;
24,096,506✔
4478
  int32_t             lino = 0;
24,096,506✔
4479
  SDataBlockIter*     pBlockIter = NULL;
24,096,506✔
4480
  SSDataBlock*        pResBlock = NULL;
24,096,506✔
4481
  SFileBlockDumpInfo* pDumpInfo = NULL;
24,096,506✔
4482
  SBlockData*         pBlockData = NULL;
24,096,506✔
4483
  const char*         id = NULL;
24,096,506✔
4484
  bool                asc = false;
24,096,506✔
4485

4486
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
24,096,506!
4487

4488
  asc = ASCENDING_TRAVERSE(pReader->info.order);
24,096,506✔
4489

4490
  pBlockIter = &pReader->status.blockIter;
24,096,506✔
4491
  pResBlock = pReader->resBlockInfo.pResBlock;
24,096,506✔
4492
  pDumpInfo = &pReader->status.fBlockDumpInfo;
24,096,506✔
4493
  pBlockData = &pReader->status.fileBlockData;
24,096,506✔
4494
  id = pReader->idStr;
24,096,506✔
4495

4496
  if (pBlockIter->numOfBlocks == 0) {
24,096,506✔
4497
    // let's try to extract data from stt files.
4498
    ERetrieveType type = 0;
10,698,745✔
4499
    code = doReadDataFromSttFiles(pReader, &type);
10,698,745✔
4500
    TSDB_CHECK_CODE(code, lino, _end);
21,398,994!
4501
    if (type == TSDB_READ_RETURN) {
10,699,732✔
4502
      goto _end;
10,698,768✔
4503
    }
4504

4505
    code = doBuildDataBlock(pReader);
964✔
4506
    TSDB_CHECK_CODE(code, lino, _end);
950!
4507
    if (pResBlock->info.rows > 0) {
950✔
4508
      goto _end;
494✔
4509
    }
4510
  }
4511

4512
  while (1) {
4513
    if (fileBlockPartiallyRead(pDumpInfo, asc)) {  // file data block is partially loaded
13,426,007✔
4514
      code = buildComposedDataBlock(pReader);
1,285✔
4515
      TSDB_CHECK_CODE(code, lino, _end);
1,285!
4516
    } else {
4517
      // current block are exhausted, try the next file block
4518
      if (pDumpInfo->allDumped) {
13,426,605✔
4519
        // try next data block in current file
4520
        bool hasNext = blockIteratorNext(&pReader->status.blockIter);
13,409,610✔
4521
        if (hasNext) {  // check for the next block in the block accessed order list
13,409,671✔
4522
          code = initBlockDumpInfo(pReader, pBlockIter);
13,272,535✔
4523
          TSDB_CHECK_CODE(code, lino, _end);
13,272,529!
4524
        } else {
4525
          // all data blocks in files are checked, let's check the data in last files.
4526
          // data blocks in current file are exhausted, let's try the next file now
4527
          if (pBlockData->uid != 0) {
137,136✔
4528
            tBlockDataClear(pBlockData);
136,194✔
4529
          }
4530

4531
          tBlockDataReset(pBlockData);
137,132✔
4532
          code = resetDataBlockIterator(pBlockIter, pReader->info.order, shouldFreePkBuf(&pReader->suppInfo), id);
137,281✔
4533
          TSDB_CHECK_CODE(code, lino, _end);
137,239!
4534

4535
          code = resetTableListIndex(&pReader->status, id);
137,239✔
4536
          TSDB_CHECK_CODE(code, lino, _end);
137,248!
4537

4538
          ERetrieveType type = 0;
137,248✔
4539
          code = doReadDataFromSttFiles(pReader, &type);
137,248✔
4540
          TSDB_CHECK_CODE(code, lino, _end);
137,173!
4541
          if (type == TSDB_READ_RETURN) {
137,173✔
4542
            break;
17,108✔
4543
          }
4544
        }
4545
      }
4546

4547
      code = doBuildDataBlock(pReader);
13,409,589✔
4548
      TSDB_CHECK_CODE(code, lino, _end);
13,409,199!
4549
    }
4550

4551
    if (pResBlock->info.rows > 0) {
13,410,484✔
4552
      break;
13,382,694✔
4553
    }
4554
  }
4555

4556
_end:
24,099,064✔
4557
  if (code != TSDB_CODE_SUCCESS) {
24,099,064!
4558
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4559
  }
4560
  return code;
24,098,762✔
4561
}
4562

4563
static void getTsdbByRetentions(SVnode* pVnode, SQueryTableDataCond* pCond, SRetention* retentions, const char* idStr,
8,528,464✔
4564
                                int8_t* pLevel, STsdb** pTsdb) {
4565
  if (pTsdb == NULL) {
8,528,464!
4566
    return;
×
4567
  }
4568

4569
  *pTsdb = NULL;
8,528,464✔
4570
  if (VND_IS_RSMA(pVnode) && !pCond->skipRollup) {
8,528,464!
4571
    int8_t  level = 0;
×
4572
    int8_t  precision = pVnode->config.tsdbCfg.precision;
×
4573
    int64_t now = taosGetTimestamp(precision);
×
4574
    int64_t offset = tsQueryRsmaTolerance * ((precision == TSDB_TIME_PRECISION_MILLI)   ? 1L
×
4575
                                             : (precision == TSDB_TIME_PRECISION_MICRO) ? 1000L
×
4576
                                                                                        : 1000000L);
×
4577

4578
    for (int32_t i = 0; i < TSDB_RETENTION_MAX; ++i) {
×
4579
      SRetention* pRetention = retentions + level;
×
4580
      if (pRetention->keep <= 0) {
×
4581
        if (level > 0) {
×
4582
          --level;
×
4583
        }
4584
        break;
×
4585
      }
4586
      if ((now - pRetention->keep) <= (pCond->twindows.skey + offset)) {
×
4587
        break;
×
4588
      }
4589
      ++level;
×
4590
    }
4591

4592
    const char* str = (idStr != NULL) ? idStr : "";
×
4593

4594
    if (level == TSDB_RETENTION_L0) {
×
4595
      *pLevel = TSDB_RETENTION_L0;
×
4596
      tsdbDebug("vgId:%d, rsma level %d is selected to query %s", TD_VID(pVnode), TSDB_RETENTION_L0, str);
×
4597
      *pTsdb = VND_RSMA0(pVnode);
×
4598
      return;
×
4599
    } else if (level == TSDB_RETENTION_L1) {
×
4600
      *pLevel = TSDB_RETENTION_L1;
×
4601
      tsdbDebug("vgId:%d, rsma level %d is selected to query %s", TD_VID(pVnode), TSDB_RETENTION_L1, str);
×
4602
      *pTsdb = VND_RSMA1(pVnode);
×
4603
      return;
×
4604
    } else {
4605
      *pLevel = TSDB_RETENTION_L2;
×
4606
      tsdbDebug("vgId:%d, rsma level %d is selected to query %s", TD_VID(pVnode), TSDB_RETENTION_L2, str);
×
4607
      *pTsdb = VND_RSMA2(pVnode);
×
4608
      return;
×
4609
    }
4610
  }
4611

4612
  *pTsdb = VND_TSDB(pVnode);
8,528,464✔
4613
}
4614

4615
SVersionRange getQueryVerRange(SVnode* pVnode, SQueryTableDataCond* pCond, int8_t level) {
8,529,946✔
4616
  int64_t startVer = (pCond->startVersion == -1) ? 0 : pCond->startVersion;
8,529,946!
4617

4618
  int64_t endVer = 0;
8,529,946✔
4619
  if (pCond->endVersion == -1) {
8,529,946✔
4620
    // user not specified end version, set current maximum version of vnode as the endVersion
4621
    endVer = pVnode->state.applied;
8,497,562✔
4622
  } else {
4623
    endVer = (pCond->endVersion > pVnode->state.applied) ? pVnode->state.applied : pCond->endVersion;
32,384✔
4624
  }
4625

4626
  return (SVersionRange){.minVer = startVer, .maxVer = endVer};
8,529,946✔
4627
}
4628

4629
static int32_t reverseSearchStartPos(const SArray* pDelList, int32_t index, int64_t key, bool asc, int32_t* start) {
125,445,508✔
4630
  int32_t code = TSDB_CODE_SUCCESS;
125,445,508✔
4631
  int32_t lino = 0;
125,445,508✔
4632
  size_t  num = 0;
125,445,508✔
4633

4634
  num = taosArrayGetSize(pDelList);
125,445,508✔
4635
  *start = index;
124,367,126✔
4636

4637
  if (asc) {
124,367,126✔
4638
    if (*start >= num - 1) {
95,695,141✔
4639
      *start = num - 1;
27,440,122✔
4640
    }
4641

4642
    TSDBKEY* p = taosArrayGet(pDelList, *start);
95,695,141✔
4643
    TSDB_CHECK_NULL(p, code, lino, _end, TSDB_CODE_INVALID_PARA);
95,474,642!
4644

4645
    while (p->ts >= key && *start > 0) {
95,582,692✔
4646
      *start -= 1;
108,050✔
4647
    }
4648
  } else {
4649
    if (index <= 0) {
28,671,985!
4650
      *start = 0;
×
4651
    }
4652

4653
    TSDBKEY* p = taosArrayGet(pDelList, *start);
28,671,985✔
4654
    TSDB_CHECK_NULL(p, code, lino, _end, TSDB_CODE_INVALID_PARA);
28,709,789!
4655

4656
    while (p->ts <= key && *start < num - 1) {
28,714,184✔
4657
      *start += 1;
4,395✔
4658
    }
4659
  }
4660

4661
_end:
28,709,789✔
4662
  if (code != TSDB_CODE_SUCCESS) {
124,184,431!
4663
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4664
  }
4665
  return code;
124,307,399✔
4666
}
4667

4668
int32_t hasBeenDropped(const SArray* pDelList, int32_t* index, int64_t key, int64_t ver, int32_t order,
147,378,950✔
4669
                       SVersionRange* pVerRange, bool hasPk, bool* dropped) {
4670
  int32_t code = TSDB_CODE_SUCCESS;
147,378,950✔
4671
  int32_t lino = 0;
147,378,950✔
4672
  size_t  num = 0;
147,378,950✔
4673
  int32_t step = 0;
147,378,950✔
4674
  bool    asc = false;
147,378,950✔
4675

4676
  *dropped = false;
147,378,950✔
4677

4678
  if (pDelList == NULL || (TARRAY_SIZE(pDelList) == 0)) {
147,378,950!
4679
    goto _end;
×
4680
  }
4681

4682
  num = taosArrayGetSize(pDelList);
147,684,162✔
4683
  asc = ASCENDING_TRAVERSE(order);
147,003,557✔
4684
  step = asc ? 1 : -1;
147,003,557✔
4685

4686
  if (hasPk) {  // handle the case where duplicated timestamps existed.
147,003,557✔
4687
    code = reverseSearchStartPos(pDelList, *index, key, asc, index);
125,570,224✔
4688
    TSDB_CHECK_CODE(code, lino, _end);
124,070,880!
4689
  }
4690

4691
  if (asc) {
145,504,213✔
4692
    if (*index >= num - 1) {
110,195,764✔
4693
      TSDBKEY* last = taosArrayGetLast(pDelList);
32,304,319✔
4694
      if (last == NULL) {
32,266,201!
4695
        goto _end;
×
4696
      }
4697

4698
      if (key > last->ts) {
32,266,201!
4699
        goto _end;
32,270,213✔
4700
      } else if (key == last->ts) {
×
4701
        TSDBKEY* prev = taosArrayGet(pDelList, num - 2);
4,249✔
4702
        if (prev == NULL) {
4,249!
4703
          goto _end;
×
4704
        }
4705

4706
        *dropped = (prev->version >= ver && prev->version <= pVerRange->maxVer && prev->version >= pVerRange->minVer);
4,249!
4707
      }
4708
    } else {
4709
      TSDBKEY* pCurrent = taosArrayGet(pDelList, *index);
77,891,445✔
4710
      TSDBKEY* pNext = taosArrayGet(pDelList, (*index) + 1);
82,176,195✔
4711
      if (pCurrent == NULL || pNext == NULL) {
81,867,373!
4712
        goto _end;
×
4713
      }
4714

4715
      if (key < pCurrent->ts) {
81,867,373✔
4716
        goto _end;
16,305,538✔
4717
      }
4718

4719
      if (pCurrent->ts <= key && pNext->ts >= key && pCurrent->version >= ver &&
65,561,835!
4720
          pVerRange->maxVer >= pCurrent->version) {
43,635,305!
4721
        *dropped = true;
43,635,630✔
4722
        goto _end;
43,635,630✔
4723
      }
4724

4725
      while (pNext->ts <= key && (*index) < num - 1) {
23,250,139✔
4726
        (*index) += 1;
1,593,331✔
4727

4728
        if ((*index) < num - 1) {
1,593,331✔
4729
          pCurrent = taosArrayGet(pDelList, *index);
1,501,925✔
4730
          pNext = taosArrayGet(pDelList, (*index) + 1);
1,501,927✔
4731
          if (pCurrent == NULL || pNext == NULL) {
4,657,075!
4732
            break;
4733
          }
4734

4735
          // it is not a consecutive deletion range, ignore it
4736
          if (pCurrent->version == 0 && pNext->version > 0) {
1,501,927!
4737
            continue;
628,479✔
4738
          }
4739

4740
          if (pCurrent->ts <= key && pNext->ts >= key && pCurrent->version >= ver &&
873,448!
4741
              pVerRange->maxVer >= pCurrent->version) {
269,399!
4742
            *dropped = true;
269,399✔
4743
            break;
269,399✔
4744
          }
4745
        }
4746
      }
4747
    }
4748
  } else {
4749
    if (*index <= 0) {
35,308,449!
4750
      TSDBKEY* pFirst = taosArrayGet(pDelList, 0);
×
4751
      if (pFirst == NULL) {
×
4752
        goto _end;
×
4753
      }
4754

4755
      if (key < pFirst->ts) {
×
4756
        goto _end;
×
4757
      } else if (key == pFirst->ts) {
×
4758
        *dropped = pFirst->version >= ver;
×
4759
      } else {
4760
        tsdbError("unexpected error, key:%" PRId64 ", first:%" PRId64, key, pFirst->ts);
×
4761
      }
4762
    } else {
4763
      TSDBKEY* pCurrent = taosArrayGet(pDelList, *index);
35,308,449✔
4764
      TSDBKEY* pPrev = taosArrayGet(pDelList, (*index) - 1);
35,195,030✔
4765
      if (pCurrent == NULL || pPrev == NULL) {
35,170,932!
4766
        goto _end;
×
4767
      }
4768

4769
      if (key > pCurrent->ts) {
35,170,932✔
4770
        goto _end;
19,336,156✔
4771
      }
4772

4773
      if (pPrev->ts <= key && pCurrent->ts >= key && pPrev->version >= ver) {
15,834,776!
4774
        *dropped = true;
1,628,286✔
4775
        goto _end;
1,628,286✔
4776
      }
4777

4778
      while (pPrev->ts >= key && (*index) > 1) {
14,572,489✔
4779
        (*index) += step;
466,353✔
4780

4781
        if ((*index) >= 1) {
466,353✔
4782
          pCurrent = taosArrayGet(pDelList, *index);
176,112✔
4783
          pPrev = taosArrayGet(pDelList, (*index) - 1);
176,112✔
4784
          if (pCurrent == NULL || pPrev == NULL) {
176,112!
4785
            break;
4786
          }
4787

4788
          // it is not a consecutive deletion range, ignore it
4789
          if (pCurrent->version > 0 && pPrev->version == 0) {
176,112✔
4790
            continue;
29,600✔
4791
          }
4792

4793
          if (pPrev->ts <= key && pCurrent->ts >= key && pPrev->version >= ver) {
146,512!
4794
            *dropped = true;
100,354✔
4795
            break;
100,354✔
4796
          }
4797
        }
4798
      }
4799
    }
4800
  }
4801

4802
_end:
14,106,136✔
4803
  if (code != TSDB_CODE_SUCCESS) {
152,154,444!
4804
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4805
  }
4806
  return code;
143,799,237✔
4807
}
4808

4809
FORCE_INLINE int32_t getValidMemRow(SIterInfo* pIter, const SArray* pDelList, STsdbReader* pReader, TSDBROW** pRes) {
4810
  int32_t  code = TSDB_CODE_SUCCESS;
2,147,483,647✔
4811
  int32_t  lino = 0;
2,147,483,647✔
4812
  int32_t  order = 0;
2,147,483,647✔
4813
  TSDBROW* pRow = NULL;
2,147,483,647✔
4814
  TSDBKEY  key;
4815

4816
  TSDB_CHECK_NULL(pIter, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,118,997,905!
4817
  TSDB_CHECK_NULL(pRes, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,147,483,647!
4818

4819
  *pRes = NULL;
2,147,483,647✔
4820

4821
  if (!pIter->hasVal) {
2,147,483,647!
4822
    goto _end;
1,577,628,511✔
4823
  }
4824

4825
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,147,483,647!
4826
  order = pReader->info.order;
2,147,483,647✔
4827
  pRow = tsdbTbDataIterGet(pIter->iter);
2,147,483,647!
4828

4829
  TSDBROW_INIT_KEY(pRow, key);
2,147,483,647!
4830
  if (outOfTimeWindow(key.ts, &pReader->info.window)) {
2,147,483,647!
4831
    pIter->hasVal = false;
×
4832
    goto _end;
×
4833
  }
4834

4835
  // it is a valid data version
4836
  if (key.version <= pReader->info.verRange.maxVer && key.version >= pReader->info.verRange.minVer) {
2,147,483,647!
4837
    if (pDelList == NULL || TARRAY_SIZE(pDelList) == 0) {
2,147,483,647!
4838
      *pRes = pRow;
2,147,483,647✔
4839
      goto _end;
2,147,483,647✔
4840
    } else {
4841
      bool dropped = false;
37,720,656✔
4842
      code = hasBeenDropped(pDelList, &pIter->index, key.ts, key.version, order, &pReader->info.verRange,
37,720,656✔
4843
                            pReader->suppInfo.numOfPks > 0, &dropped);
37,720,656✔
4844
      TSDB_CHECK_CODE(code, lino, _end);
75,125,605!
4845
      if (!dropped) {
37,707,722!
4846
        *pRes = pRow;
37,417,883✔
4847
        goto _end;
37,417,883✔
4848
      }
4849
    }
4850
  }
4851

4852
  while (1) {
4853
    pIter->hasVal = tsdbTbDataIterNext(pIter->iter);
8,659,585✔
4854
    if (!pIter->hasVal) {
10,105,948!
4855
      goto _end;
10,293✔
4856
    }
4857

4858
    pRow = tsdbTbDataIterGet(pIter->iter);
10,095,655!
4859

4860
    TSDBROW_INIT_KEY(pRow, key);
10,095,655!
4861
    if (outOfTimeWindow(key.ts, &pReader->info.window)) {
10,095,655!
4862
      pIter->hasVal = false;
105,826✔
4863
      goto _end;
105,826✔
4864
    }
4865

4866
    if (key.version <= pReader->info.verRange.maxVer && key.version >= pReader->info.verRange.minVer) {
9,989,829!
4867
      if (pDelList == NULL || TARRAY_SIZE(pDelList) == 0) {
9,846,675!
4868
        *pRes = pRow;
×
4869
        goto _end;
×
4870
      } else {
4871
        bool dropped = false;
9,846,752✔
4872
        code = hasBeenDropped(pDelList, &pIter->index, key.ts, key.version, order, &pReader->info.verRange,
9,846,752✔
4873
                              pReader->suppInfo.numOfPks > 0, &dropped);
9,846,752✔
4874
        TSDB_CHECK_CODE(code, lino, _end);
10,014,547!
4875
        if (!dropped) {
9,841,875!
4876
          *pRes = pRow;
172,672✔
4877
          goto _end;
172,672✔
4878
        }
4879
      }
4880
    }
4881
  }
4882

4883
_end:
2,147,483,647✔
4884
  if (code != TSDB_CODE_SUCCESS) {
2,147,483,647!
4885
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4886
  }
4887
  return code;
2,147,483,647✔
4888
}
4889

4890
int32_t doMergeRowsInBuf(SIterInfo* pIter, uint64_t uid, SRowKey* pCurKey, SArray* pDelList, STsdbReader* pReader) {
283,648,511✔
4891
  int32_t     code = TSDB_CODE_SUCCESS;
283,648,511✔
4892
  int32_t     lino = 0;
283,648,511✔
4893
  SRowMerger* pMerger = NULL;
283,648,511✔
4894

4895
  TSDB_CHECK_NULL(pIter, code, lino, _end, TSDB_CODE_INVALID_PARA);
283,648,511!
4896
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
283,648,511!
4897

4898
  pMerger = &pReader->status.merger;
283,648,511✔
4899

4900
  while (1) {
1,321,327✔
4901
    pIter->hasVal = tsdbTbDataIterNext(pIter->iter);
284,969,838✔
4902
    if (!pIter->hasVal) {
284,962,326✔
4903
      break;
284,015,489✔
4904
    }
4905

4906
    // data exists but not valid
4907
    TSDBROW* pRow = NULL;
284,525,433!
4908
    code = getValidMemRow(pIter, pDelList, pReader, &pRow);
284,909,410✔
4909
    TSDB_CHECK_CODE(code, lino, _end);
284,909,410!
4910
    if (pRow == NULL) {
284,909,410✔
4911
      break;
32,150✔
4912
    }
4913

4914
    // ts is not identical, quit
4915
    if (TSDBROW_TS(pRow) != pCurKey->ts) {
284,877,260!
4916
      break;
283,546,014✔
4917
    }
4918

4919
    if (pCurKey->numOfPKs > 0) {
1,331,246✔
4920
      SRowKey nextKey = {0};
726✔
4921
      tRowGetKeyEx(pRow, &nextKey);
726!
4922
      if (pkCompEx(pCurKey, &nextKey) != 0) {
726✔
4923
        break;
432✔
4924
      }
4925
    }
4926

4927
    STSchema* pTSchema = NULL;
1,330,814✔
4928
    if (pRow->type == TSDBROW_ROW_FMT) {
1,330,814✔
4929
      pTSchema = doGetSchemaForTSRow(TSDBROW_SVERSION(pRow), pReader, uid);
1,320,752!
4930
      TSDB_CHECK_NULL(pTSchema, code, lino, _end, terrno);
1,320,818!
4931
    }
4932

4933
    code = tsdbRowMergerAdd(pMerger, pRow, pTSchema);
1,330,880✔
4934
    TSDB_CHECK_CODE(code, lino, _end);
1,321,327!
4935
  }
4936

4937
_end:
284,015,489✔
4938
  if (code != TSDB_CODE_SUCCESS) {
284,015,489!
4939
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4940
  }
4941
  return code;
283,984,244✔
4942
}
4943

4944
static int32_t doMergeRowsInFileBlockImpl(SBlockData* pBlockData, int32_t* rowIndex, SRowKey* pKey, SRowMerger* pMerger,
13,641,165✔
4945
                                          SVersionRange* pVerRange, int32_t step) {
4946
  int32_t code = TSDB_CODE_SUCCESS;
13,641,165✔
4947
  int32_t lino = 0;
13,641,165✔
4948

4949
  TSDB_CHECK_NULL(pBlockData, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,641,165!
4950
  TSDB_CHECK_NULL(rowIndex, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,641,165!
4951

4952
  while ((*rowIndex) < pBlockData->nRow && (*rowIndex) >= 0) {
13,640,921!
4953
    SRowKey cur;
4954
    tColRowGetKey(pBlockData, (*rowIndex), &cur);
13,640,586✔
4955
    if (pkCompEx(&cur, pKey) != 0) {
13,637,942!
4956
      break;
13,638,186✔
4957
    }
4958

4959
    if (pBlockData->aVersion[(*rowIndex)] > pVerRange->maxVer ||
×
4960
        pBlockData->aVersion[(*rowIndex)] < pVerRange->minVer) {
5!
4961
      (*rowIndex) += step;
×
4962
      continue;
×
4963
    }
4964

4965
    TSDBROW fRow = tsdbRowFromBlockData(pBlockData, (*rowIndex));
5✔
4966
    code = tsdbRowMergerAdd(pMerger, &fRow, NULL);
5✔
4967
    TSDB_CHECK_CODE(code, lino, _end);
5!
4968
    (*rowIndex) += step;
5✔
4969
  }
4970

4971
_end:
335✔
4972
  if (code != TSDB_CODE_SUCCESS) {
13,638,521!
4973
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4974
  }
4975
  return code;
13,638,722✔
4976
}
4977

4978
typedef enum {
4979
  CHECK_FILEBLOCK_CONT = 0x1,
4980
  CHECK_FILEBLOCK_QUIT = 0x2,
4981
} CHECK_FILEBLOCK_STATE;
4982

4983
static int32_t checkForNeighborFileBlock(STsdbReader* pReader, STableBlockScanInfo* pScanInfo,
18,575✔
4984
                                         SFileDataBlockInfo* pFBlock, SRowMerger* pMerger, SRowKey* pKey,
4985
                                         CHECK_FILEBLOCK_STATE* state) {
4986
  int32_t             code = TSDB_CODE_SUCCESS;
18,575✔
4987
  int32_t             lino = 0;
18,575✔
4988
  SFileBlockDumpInfo* pDumpInfo = NULL;
18,575✔
4989
  SBlockData*         pBlockData = NULL;
18,575✔
4990
  bool                asc = false;
18,575✔
4991
  SVersionRange*      pVerRange = NULL;
18,575✔
4992
  bool                loadNeighbor = true;
18,575✔
4993
  int32_t             step = 0;
18,575✔
4994

4995
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
18,575!
4996
  TSDB_CHECK_NULL(state, code, lino, _end, TSDB_CODE_INVALID_PARA);
18,575!
4997

4998
  pDumpInfo = &pReader->status.fBlockDumpInfo;
18,575✔
4999
  pBlockData = &pReader->status.fileBlockData;
18,575✔
5000
  asc = ASCENDING_TRAVERSE(pReader->info.order);
18,575✔
5001
  pVerRange = &pReader->info.verRange;
18,575✔
5002
  step = ASCENDING_TRAVERSE(pReader->info.order) ? 1 : -1;
18,575✔
5003

5004
  *state = CHECK_FILEBLOCK_QUIT;
18,575✔
5005
  code = loadNeighborIfOverlap(pFBlock, pScanInfo, pReader, &loadNeighbor);
18,575✔
5006
  TSDB_CHECK_CODE(code, lino, _end);
18,575!
5007

5008
  if (loadNeighbor) {
18,575✔
5009
    code = doMergeRowsInFileBlockImpl(pBlockData, &pDumpInfo->rowIndex, pKey, pMerger, pVerRange, step);
5✔
5010
    TSDB_CHECK_CODE(code, lino, _end);
5!
5011
    if ((pDumpInfo->rowIndex >= pDumpInfo->totalRows && asc) || (pDumpInfo->rowIndex < 0 && !asc)) {
5!
5012
      *state = CHECK_FILEBLOCK_CONT;
×
5013
    }
5014
  }
5015

5016
_end:
18,575✔
5017
  if (code != TSDB_CODE_SUCCESS) {
18,575!
5018
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5019
  }
5020
  return code;
18,575✔
5021
}
5022

5023
int32_t doMergeRowsInFileBlocks(SBlockData* pBlockData, STableBlockScanInfo* pScanInfo, SRowKey* pKey,
13,662,046✔
5024
                                STsdbReader* pReader) {
5025
  int32_t             code = TSDB_CODE_SUCCESS;
13,662,046✔
5026
  int32_t             lino = 0;
13,662,046✔
5027
  SFileBlockDumpInfo* pDumpInfo = NULL;
13,662,046✔
5028
  SRowMerger*         pMerger = NULL;
13,662,046✔
5029
  bool                asc = false;
13,662,046✔
5030
  int32_t             step = 0;
13,662,046✔
5031
  SVersionRange*      pRange = NULL;
13,662,046✔
5032

5033
  TSDB_CHECK_NULL(pBlockData, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,662,046!
5034
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
13,662,046!
5035

5036
  pDumpInfo = &pReader->status.fBlockDumpInfo;
13,662,046✔
5037
  pMerger = &pReader->status.merger;
13,662,046✔
5038
  asc = ASCENDING_TRAVERSE(pReader->info.order);
13,662,046✔
5039
  step = asc ? 1 : -1;
13,662,046✔
5040
  pRange = &pReader->info.verRange;
13,662,046✔
5041

5042
  pDumpInfo->rowIndex += step;
13,662,046✔
5043
  if ((pDumpInfo->rowIndex <= pBlockData->nRow - 1 && asc) || (pDumpInfo->rowIndex >= 0 && !asc)) {
13,662,046✔
5044
    code = doMergeRowsInFileBlockImpl(pBlockData, &pDumpInfo->rowIndex, pKey, pMerger, pRange, step);
13,642,135✔
5045
    TSDB_CHECK_CODE(code, lino, _end);
13,632,394!
5046
  }
5047

5048
  // all rows are consumed, let's try next file block
5049
  if ((pDumpInfo->rowIndex >= pBlockData->nRow && asc) || (pDumpInfo->rowIndex < 0 && !asc)) {
13,652,305!
5050
    while (1) {
×
5051
      SFileDataBlockInfo* pFileBlockInfo = NULL;
18,575✔
5052
      code = getCurrentBlockInfo(&pReader->status.blockIter, &pFileBlockInfo, pReader->idStr);
18,575✔
5053
      TSDB_CHECK_CODE(code, lino, _end);
18,575!
5054

5055
      if (pFileBlockInfo == NULL) {
18,575!
5056
        break;
18,575✔
5057
      }
5058

5059
      CHECK_FILEBLOCK_STATE st = CHECK_FILEBLOCK_QUIT;
18,575✔
5060
      code = checkForNeighborFileBlock(pReader, pScanInfo, pFileBlockInfo, pMerger, pKey, &st);
18,575✔
5061
      TSDB_CHECK_CODE(code, lino, _end);
18,575!
5062
      if (st == CHECK_FILEBLOCK_QUIT) {
18,575!
5063
        break;
18,575✔
5064
      }
5065
    }
5066
  }
5067

5068
_end:
13,633,730✔
5069
  if (code != TSDB_CODE_SUCCESS) {
13,652,305!
5070
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5071
  }
5072
  return code;
13,654,988✔
5073
}
5074

5075
int32_t doMergeRowsInSttBlock(SSttBlockReader* pSttBlockReader, STableBlockScanInfo* pScanInfo, SRowMerger* pMerger,
262,240,432✔
5076
                              int32_t pkSrcSlot, SVersionRange* pVerRange, const char* idStr) {
5077
  int32_t  code = TSDB_CODE_SUCCESS;
262,240,432✔
5078
  int32_t  lino = 0;
262,240,432✔
5079
  SRowKey* pRowKey = NULL;
262,240,432✔
5080
  SRowKey* pNextKey = NULL;
262,240,432✔
5081

5082
  TSDB_CHECK_NULL(pScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
262,240,432!
5083

5084
  pRowKey = &pScanInfo->lastProcKey;
262,240,432✔
5085

5086
  while (1) {
1,283,392✔
5087
    code = nextRowFromSttBlocks(pSttBlockReader, pScanInfo, pkSrcSlot, pVerRange);
263,523,824✔
5088
    TSDB_CHECK_CODE(code, lino, _end);
262,800,800!
5089
    if (!hasDataInSttBlock(pScanInfo)) {
262,800,800✔
5090
      break;
5,011,275✔
5091
    }
5092

5093
    pNextKey = getCurrentKeyInSttBlock(pSttBlockReader);
257,762,677!
5094

5095
    int32_t ret = pkCompEx(pRowKey, pNextKey);
257,851,821✔
5096
    if (ret == 0) {
257,851,821✔
5097
      TSDBROW* pRow1 = tMergeTreeGetRow(&pSttBlockReader->mergeTree);
1,284,642✔
5098
      code = tsdbRowMergerAdd(pMerger, pRow1, NULL);
1,284,642✔
5099
      TSDB_CHECK_CODE(code, lino, _end);
1,283,392!
5100
    } else {
5101
      tsdbTrace("uid:%" PRIu64 " last del index:%d, del range:%d, lastKeyInStt:%" PRId64 ", %s", pScanInfo->uid,
256,567,179✔
5102
                pScanInfo->sttBlockDelIndex, (int32_t)taosArrayGetSize(pScanInfo->delSkyline),
5103
                pScanInfo->sttKeyInfo.nextProcKey.ts, idStr);
5104
      break;
256,496,403✔
5105
    }
5106
  }
5107

5108
_end:
261,507,678✔
5109
  if (code != TSDB_CODE_SUCCESS) {
261,507,678!
5110
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5111
  }
5112
  return code;
261,493,362✔
5113
}
5114

5115
int32_t doMergeMemTableMultiRows(TSDBROW* pRow, SRowKey* pKey, uint64_t uid, SIterInfo* pIter, SArray* pDelList,
1,548,333,867✔
5116
                                 TSDBROW* pResRow, STsdbReader* pReader, bool* freeTSRow) {
5117
  int32_t     code = TSDB_CODE_SUCCESS;
1,548,333,867✔
5118
  int32_t     lino = 0;
1,548,333,867✔
5119
  SRowMerger* pMerger = NULL;
1,548,333,867✔
5120
  TSDBROW*    pNextRow = NULL;
1,548,333,867✔
5121
  STSchema*   pTSchema = NULL;
1,548,333,867✔
5122
  STSchema*   pTSchema1 = NULL;
1,548,333,867✔
5123
  TSDBROW     current;
5124

5125
  TSDB_CHECK_NULL(pRow, code, lino, _end, TSDB_CODE_INVALID_PARA);
1,548,333,867!
5126
  TSDB_CHECK_NULL(pIter, code, lino, _end, TSDB_CODE_INVALID_PARA);
1,548,333,867!
5127
  TSDB_CHECK_NULL(pResRow, code, lino, _end, TSDB_CODE_INVALID_PARA);
1,548,333,867!
5128
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
1,548,333,867!
5129
  TSDB_CHECK_NULL(freeTSRow, code, lino, _end, TSDB_CODE_INVALID_PARA);
1,548,333,867!
5130

5131
  pMerger = &pReader->status.merger;
1,548,333,867✔
5132
  pMerger->arg = pReader->pTsdb->pVnode->pBse;
1,548,333,867✔
5133

5134
  current = *pRow;
1,548,333,867✔
5135

5136
  {  // if the timestamp of the next valid row has a different ts, return current row directly
5137
    pIter->hasVal = tsdbTbDataIterNext(pIter->iter);
1,548,333,867✔
5138

5139
    if (!pIter->hasVal) {
1,544,534,150✔
5140
      *pResRow = *pRow;
3,637,792✔
5141
      *freeTSRow = false;
3,637,792✔
5142
      goto _end;
3,637,792✔
5143
    } else {  // has next point in mem/imem
5144
      code = getValidMemRow(pIter, pDelList, pReader, &pNextRow);
1,543,413,495✔
5145
      TSDB_CHECK_CODE(code, lino, _end);
1,543,413,495!
5146
      if (pNextRow == NULL) {
1,543,413,495✔
5147
        *pResRow = current;
1,428,840✔
5148
        *freeTSRow = false;
1,428,840✔
5149
        goto _end;
1,428,840✔
5150
      }
5151

5152
      if (TSDBROW_TS(&current) != TSDBROW_TS(pNextRow)) {
1,541,984,655✔
5153
        *pResRow = current;
1,533,344,298✔
5154
        *freeTSRow = false;
1,533,344,298✔
5155
        goto _end;
1,533,344,298✔
5156
      }
5157

5158
      if (pKey->numOfPKs > 0) {
8,640,357✔
5159
        SRowKey nextRowKey = {0};
3,220,360✔
5160
        tRowGetKeyEx(pNextRow, &nextRowKey);
3,220,360!
5161
        if (pkCompEx(pKey, &nextRowKey) != 0) {
3,217,576✔
5162
          *pResRow = current;
3,216,690✔
5163
          *freeTSRow = false;
3,216,690✔
5164
          goto _end;
3,216,690✔
5165
        }
5166
      }
5167
    }
5168
  }
5169

5170
  // start to merge duplicated rows
5171
  if (current.type == TSDBROW_ROW_FMT) {  // get the correct schema for row-wise data in memory
5,420,883✔
5172
    pTSchema = doGetSchemaForTSRow(current.pTSRow->sver, pReader, uid);
5,117,449!
5173
    TSDB_CHECK_NULL(pTSchema, code, lino, _end, terrno);
5,117,379!
5174
  }
5175

5176
  code = tsdbRowMergerAdd(pMerger, &current, pTSchema);
5,420,813✔
5177
  TSDB_CHECK_CODE(code, lino, _end);
5,115,818!
5178

5179
  if (pNextRow->type == TSDBROW_ROW_FMT) {  // get the correct schema for row-wise data in memory
5,115,818!
5180
    pTSchema1 = doGetSchemaForTSRow(pNextRow->pTSRow->sver, pReader, uid);
5,115,872!
5181
    TSDB_CHECK_NULL(pTSchema1, code, lino, _end, terrno);
5,115,668!
5182
  }
5183

5184
  code = tsdbRowMergerAdd(pMerger, pNextRow, pTSchema1);
5,115,614✔
5185
  TSDB_CHECK_CODE(code, lino, _end);
5,117,057!
5186

5187
  code = doMergeRowsInBuf(pIter, uid, pKey, pDelList, pReader);
5,117,057✔
5188
  TSDB_CHECK_CODE(code, lino, _end);
5,116,226!
5189

5190
  code = tsdbRowMergerGetRow(pMerger, &pResRow->pTSRow);
5,116,226✔
5191
  TSDB_CHECK_CODE(code, lino, _end);
5,112,769!
5192

5193
  pResRow->type = TSDBROW_ROW_FMT;
5,112,769✔
5194
  tsdbRowMergerClear(pMerger);
5,112,769✔
5195
  *freeTSRow = true;
5,104,792✔
5196

5197
_end:
1,546,732,412✔
5198
  if (code != TSDB_CODE_SUCCESS) {
1,546,732,412!
5199
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5200
  }
5201
  return code;
1,544,547,162✔
5202
}
5203

5204
int32_t doMergeMemIMemRows(TSDBROW* pRow, SRowKey* pRowKey, TSDBROW* piRow, SRowKey* piRowKey,
69,325✔
5205
                           STableBlockScanInfo* pBlockScanInfo, STsdbReader* pReader, SRow** pTSRow) {
5206
  int32_t     code = TSDB_CODE_SUCCESS;
69,325✔
5207
  int32_t     lino = 0;
69,325✔
5208
  SRowMerger* pMerger = NULL;
69,325✔
5209
  STSchema*   pSchema = NULL;
69,325✔
5210
  STSchema*   piSchema = NULL;
69,325✔
5211

5212
  TSDB_CHECK_NULL(pRow, code, lino, _end, TSDB_CODE_INVALID_PARA);
69,325!
5213
  TSDB_CHECK_NULL(piRow, code, lino, _end, TSDB_CODE_INVALID_PARA);
69,325!
5214
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
69,325!
5215
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
69,325!
5216

5217
  pMerger = &pReader->status.merger;
69,325✔
5218
  pMerger->arg = pReader->pTsdb->pVnode->pBse;
69,325✔
5219

5220
  if (pRow->type == TSDBROW_ROW_FMT) {
69,325!
5221
    pSchema = doGetSchemaForTSRow(TSDBROW_SVERSION(pRow), pReader, pBlockScanInfo->uid);
69,325!
5222
    TSDB_CHECK_NULL(pSchema, code, lino, _end, terrno);
69,325!
5223
  }
5224

5225
  if (piRow->type == TSDBROW_ROW_FMT) {
69,325!
5226
    piSchema = doGetSchemaForTSRow(TSDBROW_SVERSION(piRow), pReader, pBlockScanInfo->uid);
69,325!
5227
    TSDB_CHECK_NULL(piSchema, code, lino, _end, terrno);
69,325!
5228
  }
5229

5230
  if (ASCENDING_TRAVERSE(pReader->info.order)) {  // ascending order imem --> mem
69,325✔
5231
    code = tsdbRowMergerAdd(pMerger, piRow, piSchema);
69,324✔
5232
    TSDB_CHECK_CODE(code, lino, _end);
69,324!
5233

5234
    code = doMergeRowsInBuf(&pBlockScanInfo->iiter, pBlockScanInfo->uid, piRowKey, pBlockScanInfo->delSkyline, pReader);
69,324✔
5235
    TSDB_CHECK_CODE(code, lino, _end);
69,324!
5236

5237
    code = tsdbRowMergerAdd(pMerger, pRow, pSchema);
69,324✔
5238
    TSDB_CHECK_CODE(code, lino, _end);
69,324!
5239

5240
    code = doMergeRowsInBuf(&pBlockScanInfo->iter, pBlockScanInfo->uid, pRowKey, pBlockScanInfo->delSkyline, pReader);
69,324✔
5241
    TSDB_CHECK_CODE(code, lino, _end);
69,324!
5242
  } else {
5243
    code = tsdbRowMergerAdd(pMerger, pRow, pSchema);
1✔
5244
    TSDB_CHECK_CODE(code, lino, _end);
1!
5245

5246
    code = doMergeRowsInBuf(&pBlockScanInfo->iter, pBlockScanInfo->uid, pRowKey, pBlockScanInfo->delSkyline, pReader);
1✔
5247
    TSDB_CHECK_CODE(code, lino, _end);
1!
5248

5249
    code = tsdbRowMergerAdd(pMerger, piRow, piSchema);
1✔
5250
    TSDB_CHECK_CODE(code, lino, _end);
1!
5251

5252
    code = doMergeRowsInBuf(&pBlockScanInfo->iiter, pBlockScanInfo->uid, piRowKey, pBlockScanInfo->delSkyline, pReader);
1✔
5253
    TSDB_CHECK_CODE(code, lino, _end);
1!
5254
  }
5255

5256
  tRowKeyAssign(&pBlockScanInfo->lastProcKey, pRowKey);
69,325✔
5257

5258
  code = tsdbRowMergerGetRow(pMerger, pTSRow);
69,325✔
5259
  tsdbRowMergerClear(pMerger);
69,325✔
5260
  TSDB_CHECK_CODE(code, lino, _end);
69,325!
5261

5262
_end:
69,325✔
5263
  if (code != TSDB_CODE_SUCCESS) {
69,325!
5264
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5265
  }
5266
  return code;
69,325✔
5267
}
5268

5269
static int32_t tsdbGetNextRowInMem(STableBlockScanInfo* pBlockScanInfo, STsdbReader* pReader, TSDBROW* pResRow,
1,506,343,857✔
5270
                                   int64_t endKey, bool* freeTSRow) {
5271
  int32_t    code = TSDB_CODE_SUCCESS;
1,506,343,857✔
5272
  int32_t    lino = 0;
1,506,343,857✔
5273
  TSDBROW*   pRow = NULL;
1,506,343,857✔
5274
  TSDBROW*   piRow = NULL;
1,506,343,857✔
5275
  SArray*    pDelList = NULL;
1,506,343,857✔
5276
  uint64_t   uid = 0;
1,506,343,857✔
5277
  SIterInfo* piter = NULL;
1,506,343,857✔
5278
  SIterInfo* piiter = NULL;
1,506,343,857✔
5279
  SRowKey    rowKey = {0};
1,506,343,857✔
5280
  SRowKey    irowKey = {0};
1,506,343,857✔
5281
  bool       asc = false;
1,506,343,857✔
5282

5283
  TSDB_CHECK_NULL(pBlockScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
1,506,343,857!
5284
  TSDB_CHECK_NULL(pResRow, code, lino, _end, TSDB_CODE_INVALID_PARA);
1,506,343,857!
5285
  TSDB_CHECK_NULL(freeTSRow, code, lino, _end, TSDB_CODE_INVALID_PARA);
1,506,343,857!
5286

5287
  code = getValidMemRow(&pBlockScanInfo->iter, pBlockScanInfo->delSkyline, pReader, &pRow);
1,506,343,857!
5288
  TSDB_CHECK_CODE(code, lino, _end);
1,562,770,682!
5289
  code = getValidMemRow(&pBlockScanInfo->iiter, pBlockScanInfo->delSkyline, pReader, &piRow);
1,562,770,682!
5290
  TSDB_CHECK_CODE(code, lino, _end);
1,558,157,381!
5291

5292
  pDelList = pBlockScanInfo->delSkyline;
1,558,157,381✔
5293
  uid = pBlockScanInfo->uid;
1,558,157,381✔
5294
  piter = &pBlockScanInfo->iter;
1,558,157,381✔
5295
  piiter = &pBlockScanInfo->iiter;
1,558,157,381✔
5296

5297
  // todo refactor
5298
  asc = ASCENDING_TRAVERSE(pReader->info.order);
1,558,157,381✔
5299
  if (piter->hasVal) {
1,558,157,381✔
5300
    tRowGetKeyEx(pRow, &rowKey);
1,528,970,175✔
5301
    if ((rowKey.ts >= endKey && asc) || (rowKey.ts <= endKey && !asc)) {
1,529,478,873✔
5302
      pRow = NULL;
2,629,696✔
5303
    }
5304
  }
5305

5306
  if (piiter->hasVal) {
1,558,666,079✔
5307
    tRowGetKeyEx(piRow, &irowKey);
38,598,618!
5308
    if ((irowKey.ts >= endKey && asc) || (irowKey.ts <= endKey && !asc)) {
35,148,939!
5309
      piRow = NULL;
2✔
5310
    }
5311
  }
5312

5313
  if (pRow != NULL && piRow != NULL) {
1,555,216,400✔
5314
    if (rowKey.numOfPKs == 0) {
8,955,341✔
5315
      if ((rowKey.ts > irowKey.ts && asc) || (rowKey.ts < irowKey.ts && (!asc))) {  // ik.ts < k.ts
8,901,839!
5316
        code = doMergeMemTableMultiRows(piRow, &irowKey, uid, piiter, pDelList, pResRow, pReader, freeTSRow);
8,829,958✔
5317
        TSDB_CHECK_CODE(code, lino, _end);
8,827,235!
5318
      } else if ((rowKey.ts < irowKey.ts && asc) || (rowKey.ts > irowKey.ts && (!asc))) {
71,881!
5319
        code = doMergeMemTableMultiRows(pRow, &rowKey, uid, piter, pDelList, pResRow, pReader, freeTSRow);
2,953✔
5320
        TSDB_CHECK_CODE(code, lino, _end);
2,953!
5321
      } else {  // ik.ts == k.ts
5322
        pResRow->type = TSDBROW_ROW_FMT;
68,928✔
5323
        code = doMergeMemIMemRows(pRow, &rowKey, piRow, &irowKey, pBlockScanInfo, pReader, &pResRow->pTSRow);
68,928✔
5324
        TSDB_CHECK_CODE(code, lino, _end);
69,325!
5325
        *freeTSRow = true;
69,325✔
5326
      }
5327
    } else {
5328
      int32_t ret = pkCompEx(&rowKey, &irowKey);
53,585✔
5329
      if (ret != 0) {
53,585!
5330
        if ((ret > 0 && asc) || (ret < 0 && (!asc))) {  // ik.ts < k.ts
53,585✔
5331
          code = doMergeMemTableMultiRows(piRow, &irowKey, uid, piiter, pDelList, pResRow, pReader, freeTSRow);
33,121✔
5332
          TSDB_CHECK_CODE(code, lino, _end);
33,120!
5333
        } else if ((ret < 0 && asc) || (ret > 0 && (!asc))) {
20,464!
5334
          code = doMergeMemTableMultiRows(pRow, &rowKey, uid, piter, pDelList, pResRow, pReader, freeTSRow);
20,467✔
5335
          TSDB_CHECK_CODE(code, lino, _end);
20,464!
5336
        }
5337
      } else {  // ik.ts == k.ts
5338
        pResRow->type = TSDBROW_ROW_FMT;
×
5339
        code = doMergeMemIMemRows(pRow, &rowKey, piRow, &irowKey, pBlockScanInfo, pReader, &pResRow->pTSRow);
×
5340
        TSDB_CHECK_CODE(code, lino, _end);
×
5341
        *freeTSRow = true;
×
5342
      }
5343
    }
5344
  } else if (piter->hasVal && pRow != NULL) {
1,546,261,059✔
5345
    code = doMergeMemTableMultiRows(pRow, &rowKey, uid, piter, pDelList, pResRow, pReader, freeTSRow);
1,512,966,153✔
5346
    TSDB_CHECK_CODE(code, lino, _end);
1,503,957,944!
5347
  } else if (piiter->hasVal && piRow != NULL) {
33,294,906!
5348
    code = doMergeMemTableMultiRows(piRow, &irowKey, uid, piiter, pDelList, pResRow, pReader, freeTSRow);
29,807,349✔
5349
    TSDB_CHECK_CODE(code, lino, _end);
29,558,226!
5350
  }
5351

5352
_end:
33,045,783✔
5353
  if (code != TSDB_CODE_SUCCESS) {
1,545,956,821!
5354
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5355
  }
5356
  return code;
1,544,015,682✔
5357
}
5358

5359
int32_t doAppendRowFromTSRow(SSDataBlock* pBlock, STsdbReader* pReader, SRow* pTSRow, STableBlockScanInfo* pScanInfo) {
2,081,497,480✔
5360
  int32_t             code = TSDB_CODE_SUCCESS;
2,081,497,480✔
5361
  int32_t             lino = 0;
2,081,497,480✔
5362
  int32_t             outputRowIndex = 0;
2,081,497,480✔
5363
  int64_t             uid = 0;
2,081,497,480✔
5364
  SBlockLoadSuppInfo* pSupInfo = NULL;
2,081,497,480✔
5365
  STSchema*           pSchema = NULL;
2,081,497,480✔
5366
  SColVal             colVal = {0};
2,081,497,480✔
5367
  SColVal             dstVal = {0};
2,081,497,480✔
5368
  int32_t             i = 0, j = 0;
2,081,497,480✔
5369

5370
  TSDB_CHECK_NULL(pBlock, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,081,497,480!
5371
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,081,497,480!
5372
  TSDB_CHECK_NULL(pTSRow, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,081,497,480!
5373
  TSDB_CHECK_NULL(pScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,081,497,480!
5374

5375
  outputRowIndex = pBlock->info.rows;
2,081,497,480✔
5376
  uid = pScanInfo->uid;
2,081,497,480✔
5377

5378
  pSupInfo = &pReader->suppInfo;
2,081,497,480✔
5379
  pSupInfo->args = pReader->pTsdb->pVnode->pBse;
2,081,497,480✔
5380

5381
  pSchema = doGetSchemaForTSRow(pTSRow->sver, pReader, uid);
2,081,497,480!
5382
  TSDB_CHECK_NULL(pSchema, code, lino, _end, terrno);
2,084,412,142!
5383

5384
  if (pSupInfo->colId[i] == PRIMARYKEY_TIMESTAMP_COL_ID) {
2,084,412,142✔
5385
    SColumnInfoData* pColData = taosArrayGet(pBlock->pDataBlock, pSupInfo->slotId[i]);
2,077,174,001✔
5386
    TSDB_CHECK_NULL(pColData, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,041,928,671!
5387

5388
    ((int64_t*)pColData->pData)[outputRowIndex] = pTSRow->ts;
2,041,928,671✔
5389
    i += 1;
2,041,928,671✔
5390
  }
5391

5392
  while (i < pSupInfo->numOfCols && j < pSchema->numOfCols) {
2,147,483,647!
5393
    col_id_t colId = pSupInfo->colId[i];
2,147,483,647✔
5394

5395
    if (colId == pSchema->columns[j].colId) {
2,147,483,647✔
5396
      SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, pSupInfo->slotId[i]);
2,147,483,647✔
5397
      TSDB_CHECK_NULL(pColInfoData, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,147,483,647!
5398

5399
      code = tRowGet(pTSRow, pSchema, j, &colVal);
2,147,483,647✔
5400
      TSDB_CHECK_CODE(code, lino, _end);
2,147,483,647!
5401
      code = doCopyColVal(pColInfoData, outputRowIndex, i, &colVal, pSupInfo);
2,147,483,647✔
5402
      TSDB_CHECK_CODE(code, lino, _end);
2,147,483,647✔
5403
      i += 1;
2,147,483,647✔
5404
      j += 1;
2,147,483,647✔
5405
    } else if (colId < pSchema->columns[j].colId) {
2,147,483,647✔
5406
      SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, pSupInfo->slotId[i]);
30,000✔
5407
      TSDB_CHECK_NULL(pColInfoData, code, lino, _end, TSDB_CODE_INVALID_PARA);
30,000!
5408

5409
      colDataSetNULL(pColInfoData, outputRowIndex);
30,000!
5410
      i += 1;
30,000✔
5411
    } else if (colId > pSchema->columns[j].colId) {
2,147,483,647!
5412
      j += 1;
2,147,483,647✔
5413
    }
5414
  }
5415

5416
  // set null value since current column does not exist in the "pSchema"
5417
  while (i < pSupInfo->numOfCols) {
2,058,878,601✔
5418
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, pSupInfo->slotId[i]);
41,346✔
5419
    TSDB_CHECK_NULL(pColInfoData, code, lino, _end, TSDB_CODE_INVALID_PARA);
41,346!
5420

5421
    colDataSetNULL(pColInfoData, outputRowIndex);
41,346✔
5422
    i += 1;
41,346✔
5423
  }
5424

5425
  pBlock->info.dataLoad = 1;
2,058,837,255✔
5426
  pBlock->info.rows += 1;
2,058,837,255✔
5427

5428
_end:
2,058,837,259✔
5429
  if (code != TSDB_CODE_SUCCESS) {
2,058,837,259✔
5430
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
4!
5431
  }
5432
  return code;
1,995,585,823✔
5433
}
5434
int32_t doAppendRowFromFileBlock(SSDataBlock* pResBlock, STsdbReader* pReader, SBlockData* pBlockData,
368,282,873✔
5435
                                 int32_t rowIndex) {
5436
  int32_t             code = TSDB_CODE_SUCCESS;
368,282,873✔
5437
  int32_t             lino = 0;
368,282,873✔
5438
  int32_t             i = 0, j = 0;
368,282,873✔
5439
  int32_t             outputRowIndex = 0;
368,282,873✔
5440
  SBlockLoadSuppInfo* pSupInfo = NULL;
368,282,873✔
5441
  SColVal             cv = {0};
368,282,873✔
5442
  int32_t             numOfInputCols = 0;
368,282,873✔
5443
  int32_t             numOfOutputCols = 0;
368,282,873✔
5444

5445
  TSDB_CHECK_NULL(pResBlock, code, lino, _end, TSDB_CODE_INVALID_PARA);
368,282,873!
5446
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
368,282,873!
5447
  TSDB_CHECK_NULL(pBlockData, code, lino, _end, TSDB_CODE_INVALID_PARA);
368,282,873!
5448

5449
  outputRowIndex = pResBlock->info.rows;
368,282,873✔
5450

5451
  pSupInfo = &pReader->suppInfo;
368,282,873✔
5452
  pSupInfo->args = pReader->pTsdb->pVnode->pBse;
368,282,873✔
5453

5454
  ((int64_t*)pReader->status.pPrimaryTsCol->pData)[outputRowIndex] = pBlockData->aTSKEY[rowIndex];
368,282,873✔
5455
  i += 1;
368,282,873✔
5456

5457
  numOfInputCols = pBlockData->nColData;
368,282,873✔
5458
  numOfOutputCols = pSupInfo->numOfCols;
368,282,873✔
5459

5460
  while (i < numOfOutputCols && j < numOfInputCols) {
844,405,302✔
5461
    SColData* pData = tBlockDataGetColDataByIdx(pBlockData, j);
470,364,189✔
5462
    if (pData->cid < pSupInfo->colId[i]) {
470,364,189✔
5463
      j += 1;
635,450✔
5464
      continue;
635,450✔
5465
    }
5466
    SColumnInfoData* pCol = TARRAY_GET_ELEM(pResBlock->pDataBlock, pSupInfo->slotId[i]);
469,728,739✔
5467
    if (pData->cid == pSupInfo->colId[i]) {
469,728,739!
5468
      code = tColDataGetValue(pData, rowIndex, &cv);
470,052,977✔
5469
      TSDB_CHECK_CODE(code, lino, _end);
475,880,213!
5470

5471
      code = doCopyColVal(pCol, outputRowIndex, i, &cv, pSupInfo);
475,880,213✔
5472
      TSDB_CHECK_CODE(code, lino, _end);
475,811,217!
5473
      j += 1;
475,811,217✔
5474
    } else if (pData->cid > pCol->info.colId) {
×
5475
      // the specified column does not exist in file block, fill with null data
5476
      colDataSetNULL(pCol, outputRowIndex);
×
5477
    }
5478

5479
    i += 1;
475,486,979✔
5480
  }
5481

5482
  while (i < numOfOutputCols) {
374,045,991✔
5483
    SColumnInfoData* pCol = taosArrayGet(pResBlock->pDataBlock, pSupInfo->slotId[i]);
4,887✔
5484
    TSDB_CHECK_NULL(pCol, code, lino, _end, TSDB_CODE_INVALID_PARA);
4,878!
5485

5486
    colDataSetNULL(pCol, outputRowIndex);
4,878✔
5487
    i += 1;
4,878✔
5488
  }
5489

5490
  pResBlock->info.dataLoad = 1;
374,041,104✔
5491
  pResBlock->info.rows += 1;
374,041,104✔
5492

5493
_end:
374,041,104✔
5494
  if (code != TSDB_CODE_SUCCESS) {
374,041,104!
5495
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5496
  }
5497
  return code;
376,125,472✔
5498
}
5499

5500
int32_t buildDataBlockFromBufImpl(STableBlockScanInfo* pBlockScanInfo, int64_t endKey, int32_t capacity,
8,841,841✔
5501
                                  STsdbReader* pReader) {
5502
  int32_t      code = TSDB_CODE_SUCCESS;
8,841,841✔
5503
  int32_t      lino = 0;
8,841,841✔
5504
  SSDataBlock* pBlock = NULL;
8,841,841✔
5505

5506
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,841,841!
5507

5508
  pBlock = pReader->resBlockInfo.pResBlock;
8,841,841✔
5509

5510
  do {
1,483,570,766✔
5511
    TSDBROW row = {.type = -1};
1,492,412,607✔
5512
    bool    freeTSRow = false;
1,492,412,607✔
5513
    code = tsdbGetNextRowInMem(pBlockScanInfo, pReader, &row, endKey, &freeTSRow);
1,492,412,607✔
5514
    TSDB_CHECK_CODE(code, lino, _end);
1,542,299,532!
5515

5516
    if (row.type == -1) {
1,542,299,528✔
5517
      break;
8,843,366✔
5518
    }
5519

5520
    if (row.type == TSDBROW_ROW_FMT) {
1,538,806,800!
5521
      code = doAppendRowFromTSRow(pBlock, pReader, row.pTSRow, pBlockScanInfo);
1,538,954,034✔
5522
      if (code != TSDB_CODE_SUCCESS) {
1,488,345,509✔
5523
        if (freeTSRow) {
4!
5524
          taosMemoryFreeClear(row.pTSRow);
×
5525
        }
5526
        TSDB_CHECK_CODE(code, lino, _end);
4!
5527
      }
5528
      pBlockScanInfo->lastProcKey.ts = row.pTSRow->ts;
1,488,345,505✔
5529
      pBlockScanInfo->lastProcKey.numOfPKs = row.pTSRow->numOfPKs;
1,488,345,505✔
5530
      if (row.pTSRow->numOfPKs > 0) {
1,488,345,505✔
5531
        code = tRowGetPrimaryKeyDeepCopy(row.pTSRow, &pBlockScanInfo->lastProcKey);
30,111,923✔
5532
        if (code != TSDB_CODE_SUCCESS) {
30,116,774!
5533
          if (freeTSRow) {
×
5534
            taosMemoryFreeClear(row.pTSRow);
×
5535
          }
5536
          TSDB_CHECK_CODE(code, lino, _end);
×
5537
        }
5538
      }
5539

5540
      if (freeTSRow) {
1,488,350,356✔
5541
        taosMemoryFreeClear(row.pTSRow);
5,180,613!
5542
      }
5543
    } else {
5544
      code = doAppendRowFromFileBlock(pBlock, pReader, row.pBlockData, row.iRow);
×
5545
      TSDB_CHECK_CODE(code, lino, _end);
562,556!
5546

5547
      tColRowGetKeyDeepCopy(row.pBlockData, row.iRow, pReader->suppInfo.pkSrcSlot, &pBlockScanInfo->lastProcKey);
562,556!
5548
    }
5549

5550
    // no data in buffer, return immediately
5551
    if (!(pBlockScanInfo->iter.hasVal || pBlockScanInfo->iiter.hasVal)) {
1,488,921,404✔
5552
      break;
5,218,375✔
5553
    }
5554

5555
    if (pBlock->info.rows >= capacity) {
1,483,703,029✔
5556
      break;
132,263✔
5557
    }
5558
  } while (1);
5559

5560
_end:
8,843,370✔
5561
  if (code != TSDB_CODE_SUCCESS) {
8,843,370✔
5562
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
4!
5563
  }
5564
  return code;
8,842,760✔
5565
}
5566

5567
// TODO refactor: with createDataBlockScanInfo
5568
int32_t tsdbSetTableList2(void* ptr, const void* pTableList, int32_t num) {
19,076✔
5569
  int32_t               code = TSDB_CODE_SUCCESS;
19,076✔
5570
  int32_t               lino = 0;
19,076✔
5571
  int32_t               size = 0;
19,076✔
5572
  STableBlockScanInfo** p = NULL;
19,076✔
5573
  STableUidList*        pUidList = NULL;
19,076✔
5574
  int32_t               iter = 0;
19,076✔
5575
  bool                  acquired = false;
19,076✔
5576

5577
  STsdbReader* pReader = (STsdbReader*)ptr;
19,076✔
5578
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
19,076!
5579
  size = tSimpleHashGetSize(pReader->status.pTableMap);
19,076✔
5580

5581
  code = tsdbAcquireReader(pReader);
19,079✔
5582
  TSDB_CHECK_CODE(code, lino, _end);
19,082!
5583
  acquired = true;
19,082✔
5584

5585
  while ((p = tSimpleHashIterate(pReader->status.pTableMap, p, &iter)) != NULL) {
38,170✔
5586
    clearBlockScanInfo(*p);
19,087✔
5587
  }
5588

5589
  tSimpleHashClear(pReader->status.pTableMap);
19,083✔
5590

5591
  if (size < num) {
19,078✔
5592
    code = ensureBlockScanInfoBuf(&pReader->blockInfoBuf, num);
15✔
5593
    TSDB_CHECK_CODE(code, lino, _end);
8!
5594

5595
    char* p1 = taosMemoryRealloc(pReader->status.uidList.tableUidList, sizeof(uint64_t) * num);
8!
5596
    TSDB_CHECK_NULL(p1, code, lino, _end, terrno);
8!
5597

5598
    pReader->status.uidList.tableUidList = (uint64_t*)p1;
8✔
5599
  }
5600

5601
  pUidList = &pReader->status.uidList;
19,071✔
5602
  pUidList->currentIndex = 0;
19,071✔
5603

5604
  STableKeyInfo* pList = (STableKeyInfo*)pTableList;
19,071✔
5605
  for (int32_t i = 0; i < num; ++i) {
38,146✔
5606
    pUidList->tableUidList[i] = pList[i].uid;
19,080✔
5607

5608
    STableBlockScanInfo* pInfo = NULL;
19,080✔
5609
    code = getPosInBlockInfoBuf(&pReader->blockInfoBuf, i, &pInfo);
19,080✔
5610
    TSDB_CHECK_CODE(code, lino, _end);
19,089!
5611

5612
    code = initTableBlockScanInfo(pInfo, pList[i].uid, pReader->status.pTableMap, pReader);
19,089✔
5613
    TSDB_CHECK_CODE(code, lino, _end);
19,075!
5614
  }
5615

5616
_end:
19,066✔
5617
  if (code != TSDB_CODE_SUCCESS) {
19,066!
5618
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5619
  }
5620
  if (acquired) {
19,063!
5621
    (void)tsdbReleaseReader(pReader);
19,063✔
5622
  }
5623
  return code;
19,081✔
5624
}
5625

5626
uint64_t tsdbGetReaderMaxVersion2(STsdbReader* pReader) { return pReader->info.verRange.maxVer; }
×
5627

5628
static int32_t doOpenReaderImpl(STsdbReader* pReader) {
8,282,170✔
5629
  int32_t         code = TSDB_CODE_SUCCESS;
8,282,170✔
5630
  int32_t         lino = 0;
8,282,170✔
5631
  SReaderStatus*  pStatus = NULL;
8,282,170✔
5632
  SDataBlockIter* pBlockIter = NULL;
8,282,170✔
5633

5634
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,282,170!
5635

5636
  pStatus = &pReader->status;
8,282,170✔
5637
  pBlockIter = &pStatus->blockIter;
8,282,170✔
5638

5639
  if (pReader->bFilesetDelimited) {
8,282,170✔
5640
    getMemTableTimeRange(pReader, &pReader->status.memTableMaxKey, &pReader->status.memTableMinKey);
3,778,790✔
5641
    pReader->status.bProcMemFirstFileset = true;
3,781,540✔
5642
  }
5643

5644
  code = initFilesetIterator(&pStatus->fileIter, pReader->pReadSnap->pfSetArray, pReader);
8,284,920✔
5645
  TSDB_CHECK_CODE(code, lino, _end);
8,286,163!
5646

5647
  code = resetDataBlockIterator(&pStatus->blockIter, pReader->info.order, shouldFreePkBuf(&pReader->suppInfo),
8,284,088✔
5648
                                pReader->idStr);
8,286,163✔
5649
  TSDB_CHECK_CODE(code, lino, _end);
8,283,756!
5650

5651
  if (pStatus->fileIter.numOfFiles == 0) {
8,283,756✔
5652
    pStatus->loadFromFile = false;
4,396,061✔
5653
  } else {
5654
    code = initForFirstBlockInFile(pReader, pBlockIter);
3,887,695✔
5655
  }
5656

5657
  if (!pStatus->loadFromFile) {
8,285,396✔
5658
    code = resetTableListIndex(pStatus, pReader->idStr);
5,355,988✔
5659
    TSDB_CHECK_CODE(code, lino, _end);
5,356,831!
5660
  }
5661

5662
_end:
8,286,239✔
5663
  if (code != TSDB_CODE_SUCCESS) {
8,286,239!
5664
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5665
  }
5666
  return code;
8,285,004✔
5667
}
5668

5669
static void freeSchemaFunc(void* param) {
85,390✔
5670
  void** p = (void**)param;
85,390✔
5671
  taosMemoryFreeClear(*p);
85,390!
5672
}
85,392✔
5673

5674
static void clearSharedPtr(STsdbReader* p) {
23,714✔
5675
  if (p) {
23,714!
5676
    p->status.pTableMap = NULL;
23,714✔
5677
    p->status.uidList.tableUidList = NULL;
23,714✔
5678
    p->info.pSchema = NULL;
23,714✔
5679
    p->pReadSnap = NULL;
23,714✔
5680
    p->pSchemaMap = NULL;
23,714✔
5681
  }
5682
}
23,714✔
5683

5684
static int32_t setSharedPtr(STsdbReader* pDst, const STsdbReader* pSrc) {
17,260✔
5685
  pDst->status.pTableMap = pSrc->status.pTableMap;
17,260✔
5686
  pDst->status.uidList = pSrc->status.uidList;
17,260✔
5687
  pDst->info.pSchema = pSrc->info.pSchema;
17,260✔
5688
  pDst->pSchemaMap = pSrc->pSchemaMap;
17,260✔
5689
  pDst->pReadSnap = pSrc->pReadSnap;
17,260✔
5690
  pDst->pReadSnap->pfSetArray = pSrc->pReadSnap->pfSetArray;
17,260✔
5691

5692
  if (pDst->info.pSchema) {
17,260!
5693
    return tsdbRowMergerInit(&pDst->status.merger, pDst->info.pSchema);
17,260✔
5694
  }
5695

5696
  return TSDB_CODE_SUCCESS;
×
5697
}
5698

5699
// ====================================== EXPOSED APIs ======================================
5700
int32_t tsdbReaderOpen2(void* pVnode, SQueryTableDataCond* pCond, void* pTableList, int32_t numOfTables,
8,480,752✔
5701
                        SSDataBlock* pResBlock, void** ppReader, const char* idstr, SHashObj** pIgnoreTables) {
5702
  int32_t      code = TSDB_CODE_SUCCESS;
8,480,752✔
5703
  int32_t      lino = 0;
8,480,752✔
5704
  STimeWindow  window = {0};
8,480,752✔
5705
  SVnodeCfg*   pConf = NULL;
8,480,752✔
5706
  STsdbReader* pReader = NULL;
8,480,752✔
5707
  int32_t      capacity = 0;
8,480,752✔
5708

5709
  window = pCond->twindows;
8,480,752✔
5710
  pConf = &(((SVnode*)pVnode)->config);
8,480,752✔
5711

5712
  capacity = pConf->tsdbCfg.maxRows;
8,480,752✔
5713
  if (pResBlock != NULL) {
8,480,752!
5714
    code = blockDataEnsureCapacity(pResBlock, capacity);
8,504,458✔
5715
    TSDB_CHECK_CODE(code, lino, _end);
8,507,635!
5716
  }
5717

5718
  code = tsdbReaderCreate(pVnode, pCond, ppReader, capacity, pResBlock, idstr);
8,483,929✔
5719
  TSDB_CHECK_CODE(code, lino, _end);
8,492,616!
5720

5721
  // check for query time window
5722
  pReader = *ppReader;
8,492,616✔
5723
  if (isEmptyQueryTimeWindow(&pReader->info.window) && pCond->type == TIMEWINDOW_RANGE_CONTAINED) {
8,492,616✔
5724
    tsdbDebug("%p query window not overlaps with the data set, no result returned, %s", pReader, pReader->idStr);
25,887✔
5725
    goto _end;
25,879✔
5726
  }
5727

5728
  if (pCond->type == TIMEWINDOW_RANGE_EXTERNAL) {
8,469,309✔
5729
    // update the SQueryTableDataCond to create inner reader
5730
    int32_t order = pCond->order;
11,857✔
5731
    if (order == TSDB_ORDER_ASC) {
11,857!
5732
      pCond->twindows = pCond->extTwindows[0];
11,857✔
5733
      pCond->order = TSDB_ORDER_DESC;
11,857✔
5734
    } else {
5735
      pCond->twindows = pCond->extTwindows[1];
×
5736
      pCond->order = TSDB_ORDER_ASC;
×
5737
    }
5738

5739
    // here we only need one more row, so the capacity is set to be ONE.
5740
    code = tsdbReaderCreate(pVnode, pCond, (void**)&((STsdbReader*)pReader)->innerReader[0], 1, pResBlock, idstr);
11,857✔
5741
    TSDB_CHECK_CODE(code, lino, _end);
11,857!
5742

5743
    if (order == TSDB_ORDER_ASC) {
11,857!
5744
      pCond->twindows = pCond->extTwindows[1];
11,857✔
5745
    } else {
5746
      pCond->twindows = pCond->extTwindows[0];
×
5747
    }
5748
    pCond->order = order;
11,857✔
5749

5750
    code = tsdbReaderCreate(pVnode, pCond, (void**)&((STsdbReader*)pReader)->innerReader[1], 1, pResBlock, idstr);
11,857✔
5751
    TSDB_CHECK_CODE(code, lino, _end);
11,857!
5752
  }
5753

5754
  // NOTE: the endVersion in pCond is the data version not schema version, so pCond->endVersion is not correct here.
5755
  //  no valid error code set in metaGetTbTSchema, so let's set the error code here.
5756
  //  we should proceed in case of tmq processing.
5757
  if (pCond->suid != 0) {
8,469,309✔
5758
    code = metaGetTbTSchemaMaybeNull(pReader->pTsdb->pVnode->pMeta, pReader->info.suid, -1, 1, &pReader->info.pSchema);
8,092,155✔
5759
    TSDB_CHECK_CODE(code, lino, _end);
8,110,284!
5760
    if (pReader->info.pSchema == NULL) {
8,110,284✔
5761
      tsdbWarn("failed to get table schema, suid:%" PRIu64 ", ver:-1, %s", pReader->info.suid, pReader->idStr);
100!
5762
    }
5763
  } else if (numOfTables > 0) {
377,154✔
5764
    STableKeyInfo* pKey = pTableList;
365,446✔
5765
    code = metaGetTbTSchemaMaybeNull(pReader->pTsdb->pVnode->pMeta, pKey->uid, -1, 1, &pReader->info.pSchema);
365,446✔
5766
    TSDB_CHECK_CODE(code, lino, _end);
365,447!
5767
    if (pReader->info.pSchema == NULL) {
365,447✔
5768
      tsdbWarn("failed to get table schema, uid:%" PRIu64 ", ver:-1, %s", pKey->uid, pReader->idStr);
6!
5769
    }
5770
  }
5771

5772
  if (pReader->info.pSchema != NULL) {
8,487,439✔
5773
    code = tsdbRowMergerInit(&pReader->status.merger, pReader->info.pSchema);
8,460,624✔
5774
    TSDB_CHECK_CODE(code, lino, _end);
8,467,956!
5775
  }
5776

5777
  pReader->pSchemaMap = tSimpleHashInit(8, taosFastHash);
8,494,771✔
5778
  if (pReader->pSchemaMap == NULL) {
8,476,402!
5779
    tsdbError("failed init schema hash for reader %s", pReader->idStr);
×
5780
    TSDB_CHECK_NULL(pReader->pSchemaMap, code, lino, _end, terrno);
×
5781
  }
5782

5783
  tSimpleHashSetFreeFp(pReader->pSchemaMap, freeSchemaFunc);
8,476,402✔
5784
  if (pReader->info.pSchema != NULL) {
8,466,685!
5785
    code = updateBlockSMAInfo(pReader->info.pSchema, &pReader->suppInfo);
8,466,773✔
5786
    TSDB_CHECK_CODE(code, lino, _end);
8,480,104✔
5787
  }
5788

5789
  STsdbReader* p = (pReader->innerReader[0] != NULL) ? pReader->innerReader[0] : pReader;
8,480,006✔
5790

5791
  code = createDataBlockScanInfo(p, &pReader->blockInfoBuf, pTableList, &pReader->status.uidList, numOfTables,
8,480,006✔
5792
                                 &pReader->status.pTableMap);
5793
  TSDB_CHECK_CODE(code, lino, _end);
8,479,007!
5794

5795
  pReader->status.pLDataIterArray = taosArrayInit(4, POINTER_BYTES);
8,479,007✔
5796
  TSDB_CHECK_NULL(pReader->status.pLDataIterArray, code, lino, _end, terrno);
8,481,508!
5797

5798
  pReader->flag = READER_STATUS_SUSPEND;
8,481,601✔
5799
  pReader->info.execMode = pCond->notLoadData ? READER_EXEC_ROWS : READER_EXEC_DATA;
8,481,601✔
5800

5801
  pReader->pIgnoreTables = pIgnoreTables;
8,481,601✔
5802
  tsdbDebug("%p total numOfTable:%d, window:%" PRId64 " - %" PRId64 ", verRange:%" PRId64 " - %" PRId64
8,481,601✔
5803
            " in this query %s",
5804
            pReader, numOfTables, pReader->info.window.skey, pReader->info.window.ekey, pReader->info.verRange.minVer,
5805
            pReader->info.verRange.maxVer, pReader->idStr);
5806

5807
_end:
8,449,230✔
5808
  if (code != TSDB_CODE_SUCCESS) {
8,506,924✔
5809
    tsdbError("%s failed at line %d since %s, %s", __func__, lino, tstrerror(code), idstr);
10!
5810
    tsdbReaderClose2(*ppReader);
10✔
5811
    *ppReader = NULL;  // reset the pointer value.
10✔
5812
  }
5813
  return code;
8,506,924✔
5814
}
5815

5816
void tsdbReaderClose2(void* ptr) {
16,511,725✔
5817
  if (ptr == NULL) {
16,511,725✔
5818
    return;
7,977,170✔
5819
  }
5820
  STsdbReader* pReader = (STsdbReader*)ptr;
8,534,555✔
5821
  int32_t code = tsdbAcquireReader(pReader);
8,534,555✔
5822
  if (code) {
8,536,329!
5823
    return;
×
5824
  }
5825

5826
  {
5827
    if (pReader->innerReader[0] != NULL || pReader->innerReader[1] != NULL) {
8,536,329!
5828
      STsdbReader* p = pReader->innerReader[0];
12,314✔
5829
      clearSharedPtr(p);
12,314✔
5830

5831
      p = pReader->innerReader[1];
11,857✔
5832
      clearSharedPtr(p);
11,857✔
5833

5834
      tsdbReaderClose2(pReader->innerReader[0]);
11,857✔
5835
      tsdbReaderClose2(pReader->innerReader[1]);
11,857✔
5836
    }
5837
  }
5838

5839
  SBlockLoadSuppInfo* pSupInfo = &pReader->suppInfo;
8,535,864✔
5840
  TARRAY2_DESTROY(&pSupInfo->colAggArray, NULL);
8,535,864!
5841

5842
  if (pSupInfo->buildBuf) {
8,535,864✔
5843
    for (int32_t i = 0; i < pSupInfo->numOfCols; ++i) {
51,719,252✔
5844
      if (pSupInfo->buildBuf[i] != NULL) {
43,183,733✔
5845
        taosMemoryFreeClear(pSupInfo->buildBuf[i]);
5,257,089!
5846
      }
5847
    }
5848
  }
5849

5850
  if (pReader->resBlockInfo.freeBlock) {
8,535,783✔
5851
    blockDataDestroy(pReader->resBlockInfo.pResBlock);
2,533✔
5852
    pReader->resBlockInfo.pResBlock = NULL;
2,533✔
5853
  }
5854

5855
  taosMemoryFree(pSupInfo->colId);
8,535,783!
5856
  tBlockDataDestroy(&pReader->status.fileBlockData);
8,536,131✔
5857
  cleanupDataBlockIterator(&pReader->status.blockIter, shouldFreePkBuf(&pReader->suppInfo));
8,534,309✔
5858

5859
  size_t numOfTables = tSimpleHashGetSize(pReader->status.pTableMap);
8,536,076✔
5860
  if (pReader->status.pTableMap != NULL) {
8,535,805✔
5861
    destroyAllBlockScanInfo(&pReader->status.pTableMap);
8,486,337✔
5862
  }
5863
  clearBlockScanInfoBuf(&pReader->blockInfoBuf);
8,536,275✔
5864

5865
  if (pReader->pFileReader != NULL) {
8,536,286✔
5866
    tsdbDataFileReaderClose(&pReader->pFileReader);
42,271✔
5867
  }
5868

5869
  SReadCostSummary* pCost = &pReader->cost;
8,536,286✔
5870
  SFilesetIter*     pFilesetIter = &pReader->status.fileIter;
8,536,286✔
5871
  if (pFilesetIter->pSttBlockReader != NULL) {
8,536,286✔
5872
    SSttBlockReader* pSttBlockReader = pFilesetIter->pSttBlockReader;
8,291,735✔
5873
    tMergeTreeClose(&pSttBlockReader->mergeTree);
8,291,735✔
5874

5875
    clearRowKey(&pSttBlockReader->currentKey);
8,291,498✔
5876
    taosMemoryFree(pSttBlockReader);
8,291,775!
5877
  }
5878

5879
  destroySttBlockReader(pReader->status.pLDataIterArray, &pCost->sttCost);
8,536,919✔
5880
  pReader->status.pLDataIterArray = NULL;
8,535,983✔
5881
  taosMemoryFreeClear(pReader->status.uidList.tableUidList);
8,535,983!
5882

5883
  tsdbTrace("tsdb/reader-close: %p, untake snapshot", pReader);
8,536,236✔
5884
  void* p = pReader->pReadSnap;
8,536,236✔
5885
  if ((p == atomic_val_compare_exchange_ptr((void**)&pReader->pReadSnap, p, NULL)) && (p != NULL)) {
8,536,236!
5886
    tsdbUntakeReadSnap2(pReader, p, true);
8,275,113✔
5887
    pReader->pReadSnap = NULL;
8,274,970✔
5888
  }
5889

5890
  code = tsem_destroy(&pReader->resumeAfterSuspend);
8,536,302✔
5891
  if (code != 0) {
8,535,289!
5892
    tsdbWarn("faild to destroy tsem"); 
×
5893
  }
5894
  (void)tsdbReleaseReader(pReader);
8,535,289✔
5895
  (void)tsdbUninitReaderLock(pReader);
8,535,738✔
5896

5897
  tsdbDebug(
8,535,544✔
5898
      "%p :io-cost summary: head-file:%" PRIu64 ", head-file time:%.2f ms, SMA:%" PRId64
5899
      " SMA-time:%.2f ms, fileBlocks:%" PRId64
5900
      ", fileBlocks-load-time:%.2f ms, "
5901
      "build in-memory-block-time:%.2f ms, sttBlocks:%" PRId64 ", sttBlocks-time:%.2f ms, sttStatisBlock:%" PRId64
5902
      ", stt-statis-Block-time:%.2f ms, composed-blocks:%" PRId64
5903
      ", composed-blocks-time:%.2fms, STableBlockScanInfo size:%.2f Kb, createTime:%.2f ms,createSkylineIterTime:%.2f "
5904
      "ms, initSttBlockReader:%.2fms, %s",
5905
      pReader, pCost->headFileLoad, pCost->headFileLoadTime, pCost->smaDataLoad, pCost->smaLoadTime, pCost->numOfBlocks,
5906
      pCost->blockLoadTime, pCost->buildmemBlock, pCost->sttCost.loadBlocks, pCost->sttCost.blockElapsedTime,
5907
      pCost->sttCost.loadStatisBlocks, pCost->sttCost.statisElapsedTime, pCost->composedBlocks,
5908
      pCost->buildComposedBlockTime, numOfTables * sizeof(STableBlockScanInfo) / 1000.0, pCost->createScanInfoList,
5909
      pCost->createSkylineIterTime, pCost->initSttBlockReader, pReader->idStr);
5910

5911
  taosMemoryFree(pReader->idStr);
8,535,544✔
5912

5913
  tsdbRowMergerCleanup(&pReader->status.merger);
8,536,301✔
5914
  taosMemoryFree(pReader->info.pSchema);
8,535,004!
5915

5916
  tSimpleHashCleanup(pReader->pSchemaMap);
8,536,440✔
5917
  taosMemoryFreeClear(pReader);
8,536,401!
5918
}
5919

5920
static int32_t doSuspendCurrentReader(STsdbReader* pCurrentReader) {
×
5921
  int32_t               code = TSDB_CODE_SUCCESS;
×
5922
  int32_t               lino = 0;
×
5923
  SReaderStatus*        pStatus = NULL;
×
5924
  STableBlockScanInfo** p = NULL;
×
5925

5926
  TSDB_CHECK_NULL(pCurrentReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
5927

5928
  pStatus = &pCurrentReader->status;
×
5929

5930
  if (pStatus->loadFromFile) {
×
5931
    tsdbDataFileReaderClose(&pCurrentReader->pFileReader);
×
5932

5933
    SReadCostSummary* pCost = &pCurrentReader->cost;
×
5934
    destroySttBlockReader(pStatus->pLDataIterArray, &pCost->sttCost);
×
5935
    pStatus->pLDataIterArray = taosArrayInit(4, POINTER_BYTES);
×
5936
    TSDB_CHECK_NULL(pStatus->pLDataIterArray, code, lino, _end, terrno);
×
5937
  }
5938

5939
  // resetDataBlockScanInfo excluding lastKey
5940
  int32_t step = ASCENDING_TRAVERSE(pCurrentReader->info.order) ? 1 : -1;
×
5941
  int32_t iter = 0;
×
5942
  while ((p = tSimpleHashIterate(pStatus->pTableMap, p, &iter)) != NULL) {
×
5943
    STableBlockScanInfo* pInfo = *(STableBlockScanInfo**)p;
×
5944
    clearBlockScanInfo(pInfo);
×
5945
    //    pInfo->sttKeyInfo.nextProcKey = pInfo->lastProcKey.ts + step;
5946
    //    pInfo->sttKeyInfo.nextProcKey = pInfo->lastProcKey + step;
5947
  }
5948

5949
  pStatus->uidList.currentIndex = 0;
×
5950
  initReaderStatus(pStatus);
×
5951

5952
_end:
×
5953
  if (code != TSDB_CODE_SUCCESS) {
×
5954
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5955
  }
5956
  return code;
×
5957
}
5958

5959
int32_t tsdbReaderSuspend2(STsdbReader* pReader) {
×
5960
  int32_t code = TSDB_CODE_SUCCESS;
×
5961
  int32_t lino = 0;
×
5962

5963
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
5964

5965
  // save reader's base state & reset top state to be reconstructed from base state
5966
  pReader->status.suspendInvoked = true;  // record the suspend status
×
5967

5968
  if (pReader->type == TIMEWINDOW_RANGE_EXTERNAL) {
×
5969
    if (pReader->step == EXTERNAL_ROWS_PREV) {
×
5970
      code = doSuspendCurrentReader(pReader->innerReader[0]);
×
5971
    } else if (pReader->step == EXTERNAL_ROWS_MAIN) {
×
5972
      code = doSuspendCurrentReader(pReader);
×
5973
    } else {
5974
      code = doSuspendCurrentReader(pReader->innerReader[1]);
×
5975
    }
5976
  } else {
5977
    code = doSuspendCurrentReader(pReader);
×
5978
  }
5979

5980
  // make sure only release once
5981
  void* p = pReader->pReadSnap;
×
5982
  if ((p == atomic_val_compare_exchange_ptr((void**)&pReader->pReadSnap, p, NULL)) && (p != NULL)) {
×
5983
    tsdbUntakeReadSnap2(pReader, p, false);
×
5984
    pReader->pReadSnap = NULL;
×
5985
  }
5986

5987
  if (pReader->bFilesetDelimited) {
×
5988
    pReader->status.memTableMinKey = INT64_MAX;
×
5989
    pReader->status.memTableMaxKey = INT64_MIN;
×
5990
  }
5991
  pReader->flag = READER_STATUS_SUSPEND;
×
5992

5993
  if (pReader->type == TIMEWINDOW_RANGE_EXTERNAL) {
×
5994
    clearSharedPtr(pReader->innerReader[0]);
×
5995
    clearSharedPtr(pReader->innerReader[1]);
×
5996
  }
5997

5998
#if SUSPEND_RESUME_TEST
5999
  tsem_post(&pReader->resumeAfterSuspend);
6000
#endif
6001

6002
  tsdbDebug("reader: %p suspended in this query %s, step:%d", pReader, pReader->idStr, pReader->step);
×
6003

6004
_end:
×
6005
  if (code != TSDB_CODE_SUCCESS) {
×
6006
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6007
  }
6008
  return code;
×
6009
}
6010

6011
static int32_t tsdbSetQueryReseek(void* pQHandle) {
×
6012
  int32_t      code = 0;
×
6013
  STsdbReader* pReader = pQHandle;
×
6014

6015
  code = tsdbTryAcquireReader(pReader);
×
6016
  if (code == 0) {
×
6017
    if (pReader->flag == READER_STATUS_SUSPEND) {
×
6018
      code = tsdbReleaseReader(pReader);
×
6019
      return code;
×
6020
    }
6021

6022
    code = tsdbReaderSuspend2(pReader);
×
6023
    (void)tsdbReleaseReader(pReader);
×
6024
    return code;
×
6025
  } else if (code == EBUSY) {
×
6026
    return TSDB_CODE_VND_QUERY_BUSY;
×
6027
  } else {
6028
    terrno = TAOS_SYSTEM_ERROR(code);
×
6029
    return TSDB_CODE_FAILED;
×
6030
  }
6031
}
6032

6033
int32_t tsdbReaderResume2(STsdbReader* pReader) {
8,473,768✔
6034
  int32_t               code = TSDB_CODE_SUCCESS;
8,473,768✔
6035
  int32_t               lino = 0;
8,473,768✔
6036
  STableBlockScanInfo** pBlockScanInfo = NULL;
8,473,768✔
6037
  int32_t               numOfTables = 0;
8,473,768✔
6038

6039
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
8,473,768!
6040

6041
  pBlockScanInfo = pReader->status.pTableIter;
8,473,768✔
6042

6043
  //  restore reader's state, task snapshot
6044
  numOfTables = tSimpleHashGetSize(pReader->status.pTableMap);
8,473,768✔
6045
  if (numOfTables > 0) {
8,475,077✔
6046
    tsdbTrace("tsdb/reader: %p, take snapshot", pReader);
8,268,289✔
6047
    code = tsdbTakeReadSnap2(pReader, tsdbSetQueryReseek, &pReader->pReadSnap, pReader->idStr);
8,268,289✔
6048
    TSDB_CHECK_CODE(code, lino, _end);
8,270,089!
6049

6050
    // open reader failure may cause the flag still to be READER_STATUS_SUSPEND, which may cause suspend reader failure.
6051
    // So we need to set it A.S.A.P
6052
    pReader->flag = READER_STATUS_NORMAL;
8,270,089✔
6053

6054
    if (pReader->type == TIMEWINDOW_RANGE_CONTAINED) {
8,270,089✔
6055
      code = doOpenReaderImpl(pReader);
8,262,941✔
6056
      TSDB_CHECK_CODE(code, lino, _end);
8,257,150!
6057
    } else {
6058
      STsdbReader* pPrevReader = pReader->innerReader[0];
7,148✔
6059
      STsdbReader* pNextReader = pReader->innerReader[1];
7,148✔
6060

6061
      // we need only one row
6062
      pPrevReader->resBlockInfo.capacity = 1;
7,148✔
6063
      code = setSharedPtr(pPrevReader, pReader);
7,148✔
6064
      TSDB_CHECK_CODE(code, lino, _end);
8,630!
6065

6066
      pNextReader->resBlockInfo.capacity = 1;
8,630✔
6067
      code = setSharedPtr(pNextReader, pReader);
8,630✔
6068
      TSDB_CHECK_CODE(code, lino, _end);
8,630!
6069

6070
      if (pReader->step == 0 || pReader->step == EXTERNAL_ROWS_PREV) {
8,630!
6071
        code = doOpenReaderImpl(pPrevReader);
8,630✔
6072
        TSDB_CHECK_CODE(code, lino, _end);
8,630!
6073
      } else if (pReader->step == EXTERNAL_ROWS_MAIN) {
×
6074
        code = doOpenReaderImpl(pReader);
×
6075
        TSDB_CHECK_CODE(code, lino, _end);
×
6076
      } else {
6077
        code = doOpenReaderImpl(pNextReader);
×
6078
        TSDB_CHECK_CODE(code, lino, _end);
×
6079
      }
6080
    }
6081
  }
6082

6083
  tsdbDebug("reader: %p resumed uid %" PRIu64 ", numOfTable:%" PRId32 ", in this query %s", pReader,
8,472,568!
6084
            pBlockScanInfo ? (*pBlockScanInfo)->uid : 0, numOfTables, pReader->idStr);
6085

6086
_end:
8,443,394✔
6087
  if (code != TSDB_CODE_SUCCESS) {
8,472,568!
6088
    tsdbError("%s failed at line %d since %s, %s", __func__, lino, tstrerror(code),
×
6089
              (pReader && pReader->idStr) ? pReader->idStr : "");
6090
  }
6091
  return code;
8,472,741✔
6092
}
6093

6094
static int32_t buildFromPreFilesetBuffer(STsdbReader* pReader) {
2,567,960✔
6095
  int32_t        code = TSDB_CODE_SUCCESS;
2,567,960✔
6096
  int32_t        lino = 0;
2,567,960✔
6097
  SReaderStatus* pStatus = NULL;
2,567,960✔
6098
  SSDataBlock*   pBlock = NULL;
2,567,960✔
6099
  int32_t        fid = 0;
2,567,960✔
6100
  STimeWindow    win = {0};
2,567,960✔
6101

6102
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
2,567,960!
6103

6104
  pStatus = &pReader->status;
2,567,960✔
6105
  pBlock = pReader->resBlockInfo.pResBlock;
2,567,960✔
6106
  fid = pReader->status.pCurrentFileset->fid;
2,567,960✔
6107
  tsdbFidKeyRange(fid, pReader->pTsdb->keepCfg.days, pReader->pTsdb->keepCfg.precision, &win.skey, &win.ekey);
2,567,960✔
6108

6109
  int64_t endKey = (ASCENDING_TRAVERSE(pReader->info.order)) ? win.skey : win.ekey;
2,567,975✔
6110
  code = buildBlockFromBufferSeqForPreFileset(pReader, endKey);
2,567,975✔
6111
  TSDB_CHECK_CODE(code, lino, _end);
2,568,014!
6112
  if (pBlock->info.rows <= 0) {
2,568,014✔
6113
    tsdbDebug("finished pre-fileset %d buffer processing. %s", fid, pReader->idStr);
2,114,377!
6114
    pStatus->bProcMemPreFileset = false;
2,114,381✔
6115
    pStatus->processingMemPreFileSet = false;
2,114,381✔
6116
    if (pReader->notifyFn) {
2,114,381✔
6117
      STsdReaderNotifyInfo info = {0};
2,114,361✔
6118
      info.duration.filesetId = fid;
2,114,361✔
6119
      pReader->notifyFn(TSD_READER_NOTIFY_DURATION_START, &info, pReader->notifyParam);
2,114,361✔
6120
      tsdbDebug("new duration %d start notification when buffer pre-fileset, %s", fid, pReader->idStr);
2,114,355!
6121
    }
6122
  }
6123

6124
_end:
453,657✔
6125
  if (code != TSDB_CODE_SUCCESS) {
2,568,012!
6126
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6127
  }
6128
  return code;
2,568,016✔
6129
}
6130

6131
static int32_t doTsdbNextDataBlockFilesetDelimited(STsdbReader* pReader) {
12,792,590✔
6132
  int32_t        code = TSDB_CODE_SUCCESS;
12,792,590✔
6133
  int32_t        lino = 0;
12,792,590✔
6134
  SReaderStatus* pStatus = NULL;
12,792,590✔
6135
  SSDataBlock*   pBlock = NULL;
12,792,590✔
6136

6137
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
12,792,590!
6138

6139
  pStatus = &pReader->status;
12,792,590✔
6140
  pBlock = pReader->resBlockInfo.pResBlock;
12,792,590✔
6141

6142
  if (pStatus->loadFromFile) {
12,792,590✔
6143
    if (pStatus->bProcMemPreFileset) {
8,195,563✔
6144
      code = buildFromPreFilesetBuffer(pReader);
592,714✔
6145
      TSDB_CHECK_CODE(code, lino, _end);
592,739!
6146
      if (pBlock->info.rows > 0) {
592,739✔
6147
        goto _end;
160,460✔
6148
      }
6149
    }
6150

6151
    code = buildBlockFromFiles(pReader);
8,035,128✔
6152
    TSDB_CHECK_CODE(code, lino, _end);
8,036,291!
6153

6154
    tsdbTrace("block from file rows: %" PRId64 ", will process pre-file set buffer: %d. %s", pBlock->info.rows,
8,036,291✔
6155
              pStatus->bProcMemFirstFileset, pReader->idStr);
6156
    if (pStatus->bProcMemPreFileset) {
8,036,291✔
6157
      if (pBlock->info.rows > 0) {
293,196✔
6158
        if (pReader->notifyFn && !pReader->status.processingMemPreFileSet) {
293,195!
6159
          int32_t              fid = pReader->status.pCurrentFileset->fid;
×
6160
          STsdReaderNotifyInfo info = {0};
×
6161
          info.duration.filesetId = fid;
×
6162
          pReader->notifyFn(TSD_READER_NOTIFY_NEXT_DURATION_BLOCK, &info, pReader->notifyParam);
×
6163
        }
6164
      } else {
6165
        pStatus->bProcMemPreFileset = false;
1✔
6166
      }
6167
    }
6168

6169
    if (pBlock->info.rows <= 0) {
8,036,291✔
6170
      code = resetTableListIndex(&pReader->status, pReader->idStr);
1,523,791✔
6171
      TSDB_CHECK_CODE(code, lino, _end);
1,523,803!
6172

6173
      int64_t endKey = (ASCENDING_TRAVERSE(pReader->info.order)) ? INT64_MAX : INT64_MIN;
1,523,803✔
6174
      code = buildBlockFromBufferSequentially(pReader, endKey);
1,523,803✔
6175
      TSDB_CHECK_CODE(code, lino, _end);
1,523,805!
6176
    }
6177
  } else {  // no data in files, let's try the buffer
6178
    int64_t endKey = (ASCENDING_TRAVERSE(pReader->info.order)) ? INT64_MAX : INT64_MIN;
4,597,027✔
6179
    code = buildBlockFromBufferSequentially(pReader, endKey);
4,597,027✔
6180
    TSDB_CHECK_CODE(code, lino, _end);
4,600,816!
6181
  }
6182

6183
_end:
4,600,816✔
6184
  if (code != TSDB_CODE_SUCCESS) {
12,797,581!
6185
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6186
  }
6187
  return code;
12,797,036✔
6188
}
6189

6190
static int32_t doTsdbNextDataBlockFilesFirst(STsdbReader* pReader) {
22,046,245✔
6191
  int32_t        code = TSDB_CODE_SUCCESS;
22,046,245✔
6192
  int32_t        lino = 0;
22,046,245✔
6193
  SReaderStatus* pStatus = NULL;
22,046,245✔
6194
  SSDataBlock*   pBlock = NULL;
22,046,245✔
6195

6196
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
22,046,245!
6197

6198
  pStatus = &pReader->status;
22,046,245✔
6199
  pBlock = pReader->resBlockInfo.pResBlock;
22,046,245✔
6200

6201
  if (pStatus->loadFromFile) {
22,046,245✔
6202
    code = buildBlockFromFiles(pReader);
16,063,870✔
6203
    TSDB_CHECK_CODE(code, lino, _end);
16,062,357!
6204

6205
    if (pBlock->info.rows <= 0) {
16,062,357✔
6206
      code = resetTableListIndex(&pReader->status, pReader->idStr);
1,306,620✔
6207
      TSDB_CHECK_CODE(code, lino, _end);
1,306,631!
6208

6209
      int64_t endKey = (ASCENDING_TRAVERSE(pReader->info.order)) ? INT64_MAX : INT64_MIN;
1,306,631✔
6210
      code = buildBlockFromBufferSequentially(pReader, endKey);
1,306,631✔
6211
      TSDB_CHECK_CODE(code, lino, _end);
1,306,625✔
6212
    }
6213
  } else {  // no data in files, let's try the buffer
6214
    int64_t endKey = (ASCENDING_TRAVERSE(pReader->info.order)) ? INT64_MAX : INT64_MIN;
5,982,375✔
6215
    code = buildBlockFromBufferSequentially(pReader, endKey);
5,982,375✔
6216
    TSDB_CHECK_CODE(code, lino, _end);
5,987,830✔
6217
  }
6218

6219
_end:
5,987,826✔
6220
  if (code != TSDB_CODE_SUCCESS) {
22,050,192✔
6221
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
10!
6222
  }
6223
  return code;
22,049,777✔
6224
}
6225

6226
static int32_t doTsdbNextDataBlock2(STsdbReader* pReader, bool* hasNext) {
35,042,153✔
6227
  int32_t      code = TSDB_CODE_SUCCESS;
35,042,153✔
6228
  int32_t      lino = 0;
35,042,153✔
6229
  SSDataBlock* pBlock = NULL;
35,042,153✔
6230

6231
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
35,042,153!
6232

6233
  // cleanup the data that belongs to the previous data block
6234
  pBlock = pReader->resBlockInfo.pResBlock;
35,042,153✔
6235
  blockDataCleanup(pBlock);
35,042,153✔
6236

6237
  *hasNext = false;
35,053,486✔
6238

6239
  SReaderStatus* pStatus = &pReader->status;
35,053,486✔
6240
  if (tSimpleHashGetSize(pStatus->pTableMap) == 0) {
35,053,486✔
6241
    goto _end;
207,552✔
6242
  }
6243

6244
  if (!pReader->bFilesetDelimited) {
34,847,478✔
6245
    code = doTsdbNextDataBlockFilesFirst(pReader);
22,051,548✔
6246
  } else {
6247
    code = doTsdbNextDataBlockFilesetDelimited(pReader);
12,795,930✔
6248
  }
6249

6250
  *hasNext = pBlock->info.rows > 0;
34,846,139✔
6251
  TSDB_CHECK_CODE(code, lino, _end);
34,846,139✔
6252

6253
_end:
34,846,129✔
6254
  if (code != TSDB_CODE_SUCCESS) {
35,053,691✔
6255
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
10!
6256
  }
6257
  return code;
35,054,356✔
6258
}
6259

6260
int32_t tsdbNextDataBlock2(void* p, bool* hasNext) {
35,073,970✔
6261
  int32_t        code = TSDB_CODE_SUCCESS;
35,073,970✔
6262
  int32_t        lino = 0;
35,073,970✔
6263
  SReaderStatus* pStatus = NULL;
35,073,970✔
6264
  bool           acquired = false;
35,073,970✔
6265

6266
  STsdbReader* pReader = (STsdbReader*)p;
35,073,970✔
6267
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
35,073,970!
6268
  TSDB_CHECK_NULL(hasNext, code, lino, _end, TSDB_CODE_INVALID_PARA);
35,073,970!
6269

6270
  *hasNext = false;
35,073,970✔
6271

6272
  code = pReader->code;
35,073,970✔
6273
  TSDB_CHECK_CODE(code, lino, _end);
35,073,970!
6274

6275
  if (isEmptyQueryTimeWindow(&pReader->info.window) || pReader->step == EXTERNAL_ROWS_NEXT) {
35,073,970✔
6276
    goto _end;
33,924✔
6277
  }
6278

6279
  pStatus = &pReader->status;
35,045,676✔
6280

6281
  // NOTE: the following codes is used to perform test for suspend/resume for tsdbReader when it blocks the commit
6282
  // the data should be ingested in round-robin and all the child tables should be createted before ingesting data
6283
  // the version range of query will be used to identify the correctness of suspend/resume functions.
6284
  // this function will be blocked before loading the SECOND block from vnode-buffer, and restart itself from sst-files
6285
#if SUSPEND_RESUME_TEST
6286
  if (!pReader->status.suspendInvoked && !pReader->status.loadFromFile) {
6287
    tsem_wait(&pReader->resumeAfterSuspend);
6288
  }
6289
#endif
6290

6291
  code = tsdbAcquireReader(pReader);
35,045,676✔
6292
  TSDB_CHECK_CODE(code, lino, _end);
35,042,928!
6293
  acquired = true;
35,042,928✔
6294

6295
  tsdbTrace("tsdb/read: %p, take read mutex, code: %d", pReader, code);
35,042,928✔
6296

6297
  if (pReader->flag == READER_STATUS_SUSPEND) {
35,042,928✔
6298
    code = tsdbReaderResume2(pReader);
8,471,157✔
6299
    TSDB_CHECK_CODE(code, lino, _end);
8,464,142!
6300
  }
6301

6302
  if (pReader->innerReader[0] != NULL && pReader->step == 0) {
35,035,913✔
6303
    code = doTsdbNextDataBlock2(pReader->innerReader[0], hasNext);
8,630✔
6304
    TSDB_CHECK_CODE(code, lino, _end);
8,630!
6305

6306
    pReader->step = EXTERNAL_ROWS_PREV;
8,630✔
6307
    if (*hasNext) {
8,630✔
6308
      pStatus = &pReader->innerReader[0]->status;
6,875✔
6309
      if (pStatus->composedDataBlock) {
6,875!
6310
        tsdbTrace("tsdb/read: %p, unlock read mutex", pReader);
6,875✔
6311
        code = tsdbReleaseReader(pReader);
6,875✔
6312
        acquired = false;
6,875✔
6313
        TSDB_CHECK_CODE(code, lino, _end);
6,875!
6314
      }
6315

6316
      return code;
6,875✔
6317
    }
6318
  }
6319

6320
  if (pReader->step == EXTERNAL_ROWS_PREV) {
35,029,038✔
6321
    // prepare for the main scan
6322
    if (tSimpleHashGetSize(pReader->status.pTableMap) > 0) {
8,630!
6323
      code = doOpenReaderImpl(pReader);
8,630✔
6324
    }
6325

6326
    int32_t step = 1;
8,630✔
6327
    resetAllDataBlockScanInfo(pReader->status.pTableMap, pReader->innerReader[0]->info.window.ekey, step);
8,630✔
6328
    TSDB_CHECK_CODE(code, lino, _end);
8,630!
6329

6330
    pReader->step = EXTERNAL_ROWS_MAIN;
8,630✔
6331
  }
6332

6333
  code = doTsdbNextDataBlock2(pReader, hasNext);
35,029,038✔
6334
  TSDB_CHECK_CODE(code, lino, _end);
35,036,015✔
6335

6336
  if (*hasNext) {
35,036,005✔
6337
    if (pStatus->composedDataBlock) {
26,766,008✔
6338
      tsdbTrace("tsdb/read: %p, unlock read mutex", pReader);
13,406,695✔
6339
      code = tsdbReleaseReader(pReader);
13,406,695✔
6340
      acquired = false;
13,408,424✔
6341
      TSDB_CHECK_CODE(code, lino, _end);
13,408,424!
6342
    }
6343
    return code;
26,767,737✔
6344
  }
6345

6346
  if (pReader->step == EXTERNAL_ROWS_MAIN && pReader->innerReader[1] != NULL) {
8,269,997!
6347
    // prepare for the next row scan
6348
    if (tSimpleHashGetSize(pReader->status.pTableMap) > 0) {
8,630!
6349
      code = doOpenReaderImpl(pReader->innerReader[1]);
8,630✔
6350
    }
6351

6352
    int32_t step = -1;
8,630✔
6353
    resetAllDataBlockScanInfo(pReader->innerReader[1]->status.pTableMap, pReader->info.window.ekey, step);
8,630✔
6354
    TSDB_CHECK_CODE(code, lino, _end);
8,630!
6355

6356
    code = doTsdbNextDataBlock2(pReader->innerReader[1], hasNext);
8,630✔
6357
    TSDB_CHECK_CODE(code, lino, _end);
8,630!
6358

6359
    pReader->step = EXTERNAL_ROWS_NEXT;
8,630✔
6360
    if (*hasNext) {
8,630✔
6361
      pStatus = &pReader->innerReader[1]->status;
6,837✔
6362
      if (pStatus->composedDataBlock) {
6,837!
6363
        tsdbTrace("tsdb/read: %p, unlock read mutex", pReader);
6,837✔
6364
        code = tsdbReleaseReader(pReader);
6,837✔
6365
        acquired = false;
6,837✔
6366
        TSDB_CHECK_CODE(code, lino, _end);
6,837!
6367
      }
6368

6369
      return code;
6,837✔
6370
    }
6371
  }
6372

6373
  tsdbTrace("tsdb/read: %p, unlock read mutex", pReader);
8,263,160✔
6374
  code = tsdbReleaseReader(pReader);
8,263,160✔
6375
  acquired = false;
8,267,004✔
6376
  TSDB_CHECK_CODE(code, lino, _end);
8,267,004!
6377

6378
_end:
8,267,004✔
6379
  if (code != TSDB_CODE_SUCCESS) {
8,300,938✔
6380
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
10!
6381
  }
6382
  if (acquired) {
8,302,876✔
6383
    tsdbTrace("tsdb/read: %p, unlock read mutex", pReader);
10✔
6384
    (void)tsdbReleaseReader(pReader);
10✔
6385
  }
6386
  return code;
8,302,983✔
6387
}
6388

6389
static int32_t doFillNullColSMA(SBlockLoadSuppInfo* pSup, int32_t numOfRows, int32_t numOfCols,
4,500,055✔
6390
                                SColumnDataAgg* pTsAgg) {
6391
  // do fill all null column value SMA info
6392
  int32_t code = TSDB_CODE_SUCCESS;
4,500,055✔
6393
  int32_t lino = 0;
4,500,055✔
6394
  int32_t i = 0, j = 0;
4,500,055✔
6395
  int32_t size = 0;
4,500,055✔
6396

6397
  TSDB_CHECK_NULL(pSup, code, lino, _end, TSDB_CODE_INVALID_PARA);
4,500,055!
6398

6399
  size = (int32_t)TARRAY2_SIZE(&pSup->colAggArray);
4,500,055✔
6400
  code = TARRAY2_INSERT_PTR(&pSup->colAggArray, 0, pTsAgg);
4,500,055!
6401
  TSDB_CHECK_CODE(code, lino, _end);
4,500,055!
6402

6403
  size++;
4,500,055✔
6404

6405
  while (j < numOfCols && i < size) {
13,625,663!
6406
    SColumnDataAgg* pAgg = &pSup->colAggArray.data[i];
9,125,608✔
6407
    if (pAgg->colId == pSup->colId[j]) {
9,125,608✔
6408
      i += 1;
9,059,828✔
6409
      j += 1;
9,059,828✔
6410
    } else if (pAgg->colId < pSup->colId[j]) {
65,780!
6411
      i += 1;
65,780✔
6412
    } else if (pSup->colId[j] < pAgg->colId) {
×
6413
      if (pSup->colId[j] != PRIMARYKEY_TIMESTAMP_COL_ID) {
×
6414
        SColumnDataAgg nullColAgg = {.colId = pSup->colId[j], .numOfNull = numOfRows};
×
6415
        code = TARRAY2_INSERT_PTR(&pSup->colAggArray, i, &nullColAgg);
×
6416
        TSDB_CHECK_CODE(code, lino, _end);
×
6417

6418
        i += 1;
×
6419
        size++;
×
6420
      }
6421
      j += 1;
×
6422
    }
6423
  }
6424

6425
  while (j < numOfCols) {
4,500,056✔
6426
    if (pSup->colId[j] != PRIMARYKEY_TIMESTAMP_COL_ID) {
1!
6427
      SColumnDataAgg nullColAgg = {.colId = pSup->colId[j], .numOfNull = numOfRows};
1✔
6428
      code = TARRAY2_INSERT_PTR(&pSup->colAggArray, i, &nullColAgg);
1!
6429
      TSDB_CHECK_CODE(code, lino, _end);
1!
6430

6431
      i += 1;
1✔
6432
    }
6433
    j++;
1✔
6434
  }
6435

6436
_end:
4,500,055✔
6437
  if (code != TSDB_CODE_SUCCESS) {
4,500,055!
6438
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6439
  }
6440
  return code;
4,500,049✔
6441
}
6442

6443
int32_t tsdbRetrieveDatablockSMA2(STsdbReader* pReader, SSDataBlock* pDataBlock, bool* allHave, bool* hasNullSMA) {
12,036,273✔
6444
  int32_t             code = TSDB_CODE_SUCCESS;
12,036,273✔
6445
  int32_t             lino = 0;
12,036,273✔
6446
  SColumnDataAgg**    pBlockSMA = NULL;
12,036,273✔
6447
  SFileDataBlockInfo* pBlockInfo = NULL;
12,036,273✔
6448

6449
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
12,036,273!
6450
  TSDB_CHECK_NULL(pDataBlock, code, lino, _end, TSDB_CODE_INVALID_PARA);
12,036,273!
6451
  TSDB_CHECK_NULL(allHave, code, lino, _end, TSDB_CODE_INVALID_PARA);
12,036,273!
6452

6453
  pBlockSMA = &pDataBlock->pBlockAgg;
12,036,273✔
6454
  *allHave = false;
12,036,273✔
6455
  *pBlockSMA = NULL;
12,036,273✔
6456

6457
  if (pReader->type == TIMEWINDOW_RANGE_EXTERNAL) {
12,036,273✔
6458
    goto _end;
25✔
6459
  }
6460

6461
  // there is no statistics data for composed block
6462
  if (pReader->status.composedDataBlock || (!pReader->suppInfo.smaValid)) {
12,036,248!
6463
    goto _end;
7,536,189✔
6464
  }
6465

6466
  code = getCurrentBlockInfo(&pReader->status.blockIter, &pBlockInfo, pReader->idStr);
4,500,059✔
6467
  TSDB_CHECK_CODE(code, lino, _end);
4,500,059!
6468

6469
  SBlockLoadSuppInfo* pSup = &pReader->suppInfo;
4,500,059✔
6470

6471
  SSDataBlock* pResBlock = pReader->resBlockInfo.pResBlock;
4,500,059✔
6472
  if (pResBlock->info.id.uid != pBlockInfo->uid) {
4,500,059!
6473
    goto _end;
×
6474
  }
6475

6476
  //  int64_t st = taosGetTimestampUs();
6477
  TARRAY2_CLEAR(&pSup->colAggArray, 0);
4,500,059✔
6478

6479
  SBrinRecord pRecord;
6480
  blockInfoToRecord(&pRecord, pBlockInfo, pSup);
4,500,059✔
6481
  code = tsdbDataFileReadBlockSma(pReader->pFileReader, &pRecord, &pSup->colAggArray);
4,500,059✔
6482
  if (code != TSDB_CODE_SUCCESS) {
4,500,052!
6483
    tsdbDebug("vgId:%d, failed to load block SMA for uid %" PRIu64 ", code:%s, %s", 0, pBlockInfo->uid, tstrerror(code),
×
6484
              pReader->idStr);
6485
    TSDB_CHECK_CODE(code, lino, _end);
×
6486
  }
6487

6488
  if (pSup->colAggArray.size > 0) {
4,500,051!
6489
    *allHave = true;
4,500,051✔
6490
  } else {
6491
    *pBlockSMA = NULL;
×
6492
    goto _end;
×
6493
  }
6494

6495
  // always load the first primary timestamp column data
6496
  SColumnDataAgg* pTsAgg = &pSup->tsColAgg;
4,500,051✔
6497

6498
  pTsAgg->numOfNull = 0;
4,500,051✔
6499
  pTsAgg->colId = PRIMARYKEY_TIMESTAMP_COL_ID;
4,500,051✔
6500
  pTsAgg->min = pResBlock->info.window.skey;
4,500,051✔
6501
  pTsAgg->max = pResBlock->info.window.ekey;
4,500,051✔
6502

6503
  // update the number of NULL data rows
6504
  size_t numOfCols = pSup->numOfCols;
4,500,051✔
6505

6506
  if (pResBlock->pBlockAgg == NULL) {
4,500,051!
6507
    size_t num = taosArrayGetSize(pResBlock->pDataBlock);
4,500,051✔
6508
    pResBlock->pBlockAgg = taosMemoryCalloc(num, sizeof(SColumnDataAgg));
4,500,051!
6509
    TSDB_CHECK_NULL(pResBlock->pBlockAgg, code, lino, _end, terrno);
4,500,056!
6510
    for (int i = 0; i < num; ++i) {
13,572,182✔
6511
      pResBlock->pBlockAgg[i].colId = -1;
9,072,126✔
6512
    }
6513
  }
6514

6515
  // do fill all null column value SMA info
6516
  code = doFillNullColSMA(pSup, pBlockInfo->numRow, numOfCols, pTsAgg);
4,500,056✔
6517
  TSDB_CHECK_CODE(code, lino, _end);
4,500,048!
6518

6519
  size_t size = pSup->colAggArray.size;
4,500,048✔
6520

6521
  int32_t i = 0, j = 0;
4,500,048✔
6522
  while (j < numOfCols && i < size) {
13,625,561!
6523
    SColumnDataAgg* pAgg = &pSup->colAggArray.data[i];
9,125,513✔
6524
    if (pAgg->colId == pSup->colId[j]) {
9,125,513✔
6525
      pResBlock->pBlockAgg[pSup->slotId[j]] = *pAgg;
9,059,739✔
6526
      i += 1;
9,059,739✔
6527
      j += 1;
9,059,739✔
6528
    } else if (pAgg->colId < pSup->colId[j]) {
65,774!
6529
      i += 1;
65,774✔
6530
    } else if (pSup->colId[j] < pAgg->colId) {
×
6531
      pResBlock->pBlockAgg[pSup->slotId[j]].colId = -1;
×
6532
      *allHave = false;
×
6533
      j += 1;
×
6534
    }
6535
  }
6536

6537
  *pBlockSMA = pResBlock->pBlockAgg;
4,500,048✔
6538
  pReader->cost.smaDataLoad += 1;
4,500,048✔
6539

6540
  //  double elapsedTime = (taosGetTimestampUs() - st) / 1000.0;
6541
  pReader->cost.smaLoadTime += 0;  // elapsedTime;
4,500,048✔
6542

6543
  tsdbDebug("vgId:%d, succeed to load block SMA for uid %" PRIu64 ", %s", 0, pBlockInfo->uid, pReader->idStr);
4,500,048!
6544

6545
_end:
4,500,048✔
6546
  if (code != TSDB_CODE_SUCCESS) {
12,036,262!
6547
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6548
  }
6549
  return code;
12,036,285✔
6550
}
6551

6552
static int32_t doRetrieveDataBlock(STsdbReader* pReader, SSDataBlock** pBlock) {
5,463,939✔
6553
  int32_t              code = TSDB_CODE_SUCCESS;
5,463,939✔
6554
  int32_t              lino = 0;
5,463,939✔
6555
  SReaderStatus*       pStatus = NULL;
5,463,939✔
6556
  SFileDataBlockInfo*  pBlockInfo = NULL;
5,463,939✔
6557
  STableBlockScanInfo* pBlockScanInfo = NULL;
5,463,939✔
6558
  bool                 reset = false;
5,463,939✔
6559

6560
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
5,463,939!
6561
  TSDB_CHECK_NULL(pBlock, code, lino, _end, TSDB_CODE_INVALID_PARA);
5,463,939!
6562

6563
  pStatus = &pReader->status;
5,463,939✔
6564
  *pBlock = NULL;
5,463,939✔
6565

6566
  code = getCurrentBlockInfo(&pStatus->blockIter, &pBlockInfo, pReader->idStr);
5,463,939✔
6567
  TSDB_CHECK_CODE(code, lino, _end);
5,464,094!
6568

6569
  code = pReader->code;
5,464,094✔
6570
  TSDB_CHECK_CODE(code, lino, _end);
5,464,094!
6571

6572
  code = getTableBlockScanInfo(pStatus->pTableMap, pBlockInfo->uid, &pBlockScanInfo, pReader->idStr);
5,464,094✔
6573
  TSDB_CHECK_CODE(code, lino, _end);
5,464,119!
6574

6575
  reset = true;
5,464,119✔
6576
  code = doLoadFileBlockData(pReader, &pStatus->blockIter, &pStatus->fileBlockData, pBlockScanInfo->uid);
5,464,119✔
6577
  TSDB_CHECK_CODE(code, lino, _end);
5,464,285!
6578

6579
  code = copyBlockDataToSDataBlock(pReader, &pBlockScanInfo->lastProcKey);
5,464,285✔
6580
  TSDB_CHECK_CODE(code, lino, _end);
5,464,368!
6581

6582
  *pBlock = pReader->resBlockInfo.pResBlock;
5,464,368✔
6583

6584
_end:
5,464,368✔
6585
  if (code != TSDB_CODE_SUCCESS) {
5,464,368!
6586
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6587
    if (reset) {
×
6588
      tBlockDataReset(&pStatus->fileBlockData);
×
6589
    }
6590
  }
6591
  return code;
5,464,201✔
6592
}
6593

6594
int32_t tsdbRetrieveDataBlock2(void* p, SSDataBlock** pBlock, SArray* pIdList) {
18,564,176✔
6595
  int32_t      code = TSDB_CODE_SUCCESS;
18,564,176✔
6596
  int32_t      lino = 0;
18,564,176✔
6597
  STsdbReader* pTReader = NULL;
18,564,176✔
6598

6599
  STsdbReader* pReader = (STsdbReader*)p;
18,564,176✔
6600
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
18,564,176!
6601
  TSDB_CHECK_NULL(pBlock, code, lino, _end, TSDB_CODE_INVALID_PARA);
18,564,176!
6602

6603
  *pBlock = NULL;
18,564,176✔
6604

6605
  pTReader = pReader;
18,564,176✔
6606
  if (pReader->type == TIMEWINDOW_RANGE_EXTERNAL) {
18,564,176✔
6607
    if (pReader->step == EXTERNAL_ROWS_PREV) {
19,616✔
6608
      pTReader = pReader->innerReader[0];
6,875✔
6609
    } else if (pReader->step == EXTERNAL_ROWS_NEXT) {
12,741✔
6610
      pTReader = pReader->innerReader[1];
6,837✔
6611
    }
6612
  }
6613

6614
  SReaderStatus* pStatus = &pTReader->status;
18,564,176✔
6615
  if (pStatus->composedDataBlock || pReader->info.execMode == READER_EXEC_ROWS) {
18,564,176!
6616
    //    tsdbReaderSuspend2(pReader);
6617
    //    tsdbReaderResume2(pReader);
6618
    *pBlock = pTReader->resBlockInfo.pResBlock;
13,100,092✔
6619
    goto _end;
13,100,092✔
6620
  }
6621

6622
  code = doRetrieveDataBlock(pTReader, pBlock);
5,464,084✔
6623

6624
  tsdbTrace("tsdb/read-retrieve: %p, unlock read mutex", pReader);
5,464,168✔
6625
  (void)tsdbReleaseReader(pReader);
5,464,168✔
6626
  TSDB_CHECK_CODE(code, lino, _end);
5,464,276!
6627

6628
  //  tsdbReaderSuspend2(pReader);
6629
  //  tsdbReaderResume2(pReader);
6630
_end:
5,464,276✔
6631
  if (code != TSDB_CODE_SUCCESS) {
18,564,368!
6632
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6633
  }
6634
  return code;
18,564,145✔
6635
}
6636

6637
int32_t tsdbReaderReset2(void* p, SQueryTableDataCond* pCond) {
20,194✔
6638
  int32_t code = TSDB_CODE_SUCCESS;
20,194✔
6639
  int32_t lino = 0;
20,194✔
6640
  bool    acquired = false;
20,194✔
6641

6642
  STsdbReader* pReader = (STsdbReader*)p;
20,194✔
6643
  tsdbTrace("tsdb/reader-reset: %p, take read mutex", pReader);
20,194✔
6644
  code = tsdbAcquireReader(pReader);
20,194✔
6645
  TSDB_CHECK_CODE(code, lino, _end);
20,196!
6646
  acquired = true;
20,196✔
6647

6648
  if (pReader->flag == READER_STATUS_SUSPEND) {
20,196✔
6649
    code = tsdbReaderResume2(pReader);
307✔
6650
    TSDB_CHECK_CODE(code, lino, _end);
307!
6651
  }
6652

6653
  if (isEmptyQueryTimeWindow(&pReader->info.window) || pReader->pReadSnap == NULL) {
20,196✔
6654
    tsdbDebug("tsdb reader reset return %p, %s", pReader->pReadSnap, pReader->idStr);
64!
6655
    code = tsdbReleaseReader(pReader);
64✔
6656
    acquired = false;
63✔
6657
    TSDB_CHECK_CODE(code, lino, _end);
63!
6658
    goto _end;
63✔
6659
  }
6660

6661
  SReaderStatus*  pStatus = &pReader->status;
20,132✔
6662
  SDataBlockIter* pBlockIter = &pStatus->blockIter;
20,132✔
6663

6664
  TSDB_CHECK_NULL(pCond, code, lino, _end, TSDB_CODE_INVALID_PARA);
20,132!
6665
  pReader->info.order = pCond->order;
20,132✔
6666
  pReader->type = TIMEWINDOW_RANGE_CONTAINED;
20,132✔
6667
  code = updateQueryTimeWindow(pReader->pTsdb, &pCond->twindows, &pReader->info.window);
20,132✔
6668
  TSDB_CHECK_CODE(code, lino, _end);
20,133!
6669
  pStatus->loadFromFile = true;
20,133✔
6670
  pStatus->pTableIter = NULL;
20,133✔
6671

6672
  // allocate buffer in order to load data blocks from file
6673
  memset(&pReader->suppInfo.tsColAgg, 0, sizeof(SColumnDataAgg));
20,133✔
6674

6675
  pReader->suppInfo.tsColAgg.colId = PRIMARYKEY_TIMESTAMP_COL_ID;
20,133✔
6676
  tsdbDataFileReaderClose(&pReader->pFileReader);
20,133✔
6677

6678
  int32_t numOfTables = tSimpleHashGetSize(pStatus->pTableMap);
20,130✔
6679

6680
  code = initFilesetIterator(&pStatus->fileIter, pReader->pReadSnap->pfSetArray, pReader);
20,130✔
6681
  TSDB_CHECK_CODE(code, lino, _end);
20,131!
6682

6683
  code = resetDataBlockIterator(pBlockIter, pReader->info.order, shouldFreePkBuf(&pReader->suppInfo), pReader->idStr);
20,131✔
6684
  TSDB_CHECK_CODE(code, lino, _end);
20,131!
6685

6686
  code = resetTableListIndex(&pReader->status, pReader->idStr);
20,131✔
6687
  TSDB_CHECK_CODE(code, lino, _end);
20,133!
6688

6689
  bool    asc = ASCENDING_TRAVERSE(pReader->info.order);
20,133✔
6690
  int32_t step = asc ? 1 : -1;
20,133✔
6691
  int64_t ts = 0;
20,133✔
6692
  if (asc) {
20,133✔
6693
    ts = (pReader->info.window.skey > INT64_MIN) ? pReader->info.window.skey - 1 : pReader->info.window.skey;
19,635!
6694
  } else {
6695
    ts = (pReader->info.window.ekey < INT64_MAX) ? pReader->info.window.ekey + 1 : pReader->info.window.ekey;
498✔
6696
  }
6697
  resetAllDataBlockScanInfo(pStatus->pTableMap, ts, step);
20,133✔
6698

6699
  // no data in files, let's try buffer in memory
6700
  if (pStatus->fileIter.numOfFiles == 0) {
20,132✔
6701
    pStatus->loadFromFile = false;
2,981✔
6702
    code = resetTableListIndex(pStatus, pReader->idStr);
2,981✔
6703
    TSDB_CHECK_CODE(code, lino, _end);
2,982!
6704
  } else {
6705
    code = initForFirstBlockInFile(pReader, pBlockIter);
17,151✔
6706
    if (code != TSDB_CODE_SUCCESS) {
17,142!
6707
      tsdbError("%p reset reader failed, numOfTables:%d, query range:%" PRId64 " - %" PRId64 " in query %s", pReader,
×
6708
                numOfTables, pReader->info.window.skey, pReader->info.window.ekey, pReader->idStr);
6709
      TSDB_CHECK_CODE(code, lino, _end);
×
6710
    }
6711
  }
6712

6713
  tsdbDebug("%p reset reader, suid:%" PRIu64 ", numOfTables:%d, skey:%" PRId64 ", query range:%" PRId64 " - %" PRId64
20,124✔
6714
            " in query %s",
6715
            pReader, pReader->info.suid, numOfTables, pCond->twindows.skey, pReader->info.window.skey,
6716
            pReader->info.window.ekey, pReader->idStr);
6717

6718
  code = tsdbReleaseReader(pReader);
20,124✔
6719
  acquired = false;
20,122✔
6720
  TSDB_CHECK_CODE(code, lino, _end);
20,122!
6721

6722
_end:
20,122✔
6723
  if (code != TSDB_CODE_SUCCESS) {
20,185!
6724
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6725
  }
6726
  if (acquired) {
20,186!
6727
    (void)tsdbReleaseReader(pReader);
×
6728
  }
6729
  return code;
20,190✔
6730
}
6731

6732
static int32_t getBucketIndex(int32_t startRow, int32_t bucketRange, int32_t numOfRows, int32_t numOfBucket) {
202,700✔
6733
  if (numOfRows < startRow) {
202,700!
6734
    return 0;
×
6735
  }
6736
  int32_t bucketIndex = ((numOfRows - startRow) / bucketRange);
202,700✔
6737
  if (bucketIndex == numOfBucket) {
202,700!
6738
    bucketIndex -= 1;
×
6739
  }
6740
  return bucketIndex;
202,700✔
6741
}
6742

6743
int32_t tsdbGetFileBlocksDistInfo2(STsdbReader* pReader, STableBlockDistInfo* pTableBlockInfo) {
6,494✔
6744
  int32_t       code = TSDB_CODE_SUCCESS;
6,494✔
6745
  int32_t       lino = 0;
6,494✔
6746
  const int32_t numOfBuckets = 20.0;
6,494✔
6747
  bool          acquired = false;
6,494✔
6748

6749
  TSDB_CHECK_NULL(pTableBlockInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
6,494!
6750

6751
  pTableBlockInfo->totalSize = 0;
6,494✔
6752
  pTableBlockInfo->totalRows = 0;
6,494✔
6753
  pTableBlockInfo->numOfVgroups = 1;
6,494✔
6754

6755
  // find the start data block in file
6756
  code = tsdbAcquireReader(pReader);
6,494✔
6757
  TSDB_CHECK_CODE(code, lino, _end);
6,513!
6758
  acquired = true;
6,513✔
6759

6760
  if (pReader->flag == READER_STATUS_SUSPEND) {
6,513!
6761
    code = tsdbReaderResume2(pReader);
6,518✔
6762
    TSDB_CHECK_CODE(code, lino, _end);
6,507!
6763
  }
6764

6765
  SMergeTreeConf conf = {
6,502✔
6766
      .pReader = pReader,
6767
      .pSchema = pReader->info.pSchema,
6,502✔
6768
      .pCols = pReader->suppInfo.colId,
6,502✔
6769
      .numOfCols = pReader->suppInfo.numOfCols,
6,502✔
6770
      .suid = pReader->info.suid,
6,502✔
6771
  };
6772

6773
  SReaderStatus* pStatus = &pReader->status;
6,502✔
6774
  if (pStatus->pCurrentFileset != NULL) {
6,502✔
6775
    pTableBlockInfo->numOfSttRows += tsdbGetRowsInSttFiles(pStatus->pCurrentFileset, pStatus->pLDataIterArray,
2,448✔
6776
                                                           pReader->pTsdb, &conf, pReader->idStr);
2,447✔
6777
  }
6778

6779
  STsdbCfg* pc = &pReader->pTsdb->pVnode->config.tsdbCfg;
6,503✔
6780
  pTableBlockInfo->defMinRows = pc->minRows;
6,503✔
6781
  pTableBlockInfo->defMaxRows = pc->maxRows;
6,503✔
6782

6783
  int32_t bucketRange = ceil(((double)(pc->maxRows - pc->minRows)) / numOfBuckets);
6,503✔
6784

6785
  pTableBlockInfo->numOfFiles += 1;
6,503✔
6786

6787
  int32_t numOfTables = (int32_t)tSimpleHashGetSize(pStatus->pTableMap);
6,503✔
6788

6789
  SDataBlockIter* pBlockIter = &pStatus->blockIter;
6,508✔
6790
  pTableBlockInfo->numOfFiles += pStatus->fileIter.numOfFiles;
6,508✔
6791

6792
  if (pBlockIter->numOfBlocks > 0) {
6,508✔
6793
    pTableBlockInfo->numOfBlocks += pBlockIter->numOfBlocks;
6✔
6794
  }
6795

6796
  pTableBlockInfo->numOfTables = numOfTables;
6,508✔
6797
  bool hasNext = (pBlockIter->numOfBlocks > 0);
6,508✔
6798

6799
  while (true) {
6800
    if (hasNext) {
214,092✔
6801
      SFileDataBlockInfo* pBlockInfo = NULL;
202,700✔
6802
      code = getCurrentBlockInfo(pBlockIter, &pBlockInfo, pReader->idStr);
202,700✔
6803
      TSDB_CHECK_CODE(code, lino, _end);
202,700!
6804

6805
      int32_t numOfRows = pBlockInfo->numRow;
202,700✔
6806

6807
      pTableBlockInfo->totalRows += numOfRows;
202,700✔
6808

6809
      if (numOfRows > pTableBlockInfo->maxRows) {
202,700✔
6810
        pTableBlockInfo->maxRows = numOfRows;
9✔
6811
      }
6812

6813
      if (numOfRows < pTableBlockInfo->minRows) {
202,700✔
6814
        pTableBlockInfo->minRows = numOfRows;
12✔
6815
      }
6816

6817
      pTableBlockInfo->totalSize += pBlockInfo->blockSize;
202,700✔
6818

6819
      int32_t bucketIndex = getBucketIndex(pTableBlockInfo->defMinRows, bucketRange, numOfRows, numOfBuckets);
202,700✔
6820
      pTableBlockInfo->blockRowsHisto[bucketIndex]++;
202,700✔
6821

6822
      hasNext = blockIteratorNext(&pStatus->blockIter);
202,700✔
6823
    } else {
6824
      code = initForFirstBlockInFile(pReader, pBlockIter);
11,392✔
6825
      TSDB_CHECK_CODE(code, lino, _end);
11,403!
6826
      if (pStatus->loadFromFile == false) {
11,403✔
6827
        break;
6,519✔
6828
      }
6829

6830
      // add the data in stt files of new fileset
6831
      if (pStatus->pCurrentFileset != NULL) {
4,884!
6832
        pTableBlockInfo->numOfSttRows += tsdbGetRowsInSttFiles(pStatus->pCurrentFileset, pStatus->pLDataIterArray,
4,893✔
6833
                                                               pReader->pTsdb, &conf, pReader->idStr);
4,886✔
6834
      }
6835

6836
      pTableBlockInfo->numOfBlocks += pBlockIter->numOfBlocks;
4,891✔
6837
      hasNext = (pBlockIter->numOfBlocks > 0);
4,891✔
6838
    }
6839
  }
6840

6841
  // record the data in stt files
6842
_end:
6,519✔
6843
  if (code != TSDB_CODE_SUCCESS) {
6,519!
6844
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6845
  }
6846
  if (acquired) {
6,512!
6847
    (void)tsdbReleaseReader(pReader);
6,512✔
6848
  }
6849
  return code;
6,502✔
6850
}
6851

6852
static void getMemTableTimeRange(STsdbReader* pReader, int64_t* pMaxKey, int64_t* pMinKey) {
3,777,020✔
6853
  SReaderStatus* pStatus = &pReader->status;
3,777,020✔
6854

6855
  int32_t iter = 0;
3,777,020✔
6856
  int64_t maxKey = INT64_MIN;
3,777,020✔
6857
  int64_t minKey = INT64_MAX;
3,777,020✔
6858

6859
  void* pHashIter = tSimpleHashIterate(pStatus->pTableMap, NULL, &iter);
3,777,020✔
6860
  while (pHashIter != NULL) {
13,622,371✔
6861
    STableBlockScanInfo* pBlockScanInfo = *(STableBlockScanInfo**)pHashIter;
9,840,822✔
6862

6863
    STbData* d = NULL;
9,840,822✔
6864
    if (pReader->pReadSnap->pMem != NULL) {
9,840,822✔
6865
      d = tsdbGetTbDataFromMemTable(pReader->pReadSnap->pMem, pReader->info.suid, pBlockScanInfo->uid);
9,225,935✔
6866
      if (d != NULL) {
9,230,663✔
6867
        if (d->maxKey > maxKey) {
3,098,347✔
6868
          maxKey = d->maxKey;
2,599,637✔
6869
        }
6870
        if (d->minKey < minKey) {
3,098,347✔
6871
          minKey = d->minKey;
2,590,543✔
6872
        }
6873
      }
6874
    }
6875

6876
    STbData* di = NULL;
9,845,550✔
6877
    if (pReader->pReadSnap->pIMem != NULL) {
9,845,550✔
6878
      di = tsdbGetTbDataFromMemTable(pReader->pReadSnap->pIMem, pReader->info.suid, pBlockScanInfo->uid);
13✔
6879
      if (di != NULL) {
13!
6880
        if (di->maxKey > maxKey) {
13✔
6881
          maxKey = di->maxKey;
11✔
6882
        }
6883
        if (di->minKey < minKey) {
13✔
6884
          minKey = di->minKey;
5✔
6885
        }
6886
      }
6887
    }
6888

6889
    // current table is exhausted, let's try the next table
6890
    pHashIter = tSimpleHashIterate(pStatus->pTableMap, pHashIter, &iter);
9,845,550✔
6891
  }
6892

6893
  *pMaxKey = maxKey;
3,781,549✔
6894
  *pMinKey = minKey;
3,781,549✔
6895
}
3,781,549✔
6896

6897
int64_t tsdbGetNumOfRowsInMemTable2(STsdbReader* pReader, uint32_t* rows) {
6,497✔
6898
  int32_t        code = TSDB_CODE_SUCCESS;
6,497✔
6899
  int32_t        lino = 0;
6,497✔
6900
  SReaderStatus* pStatus = NULL;
6,497✔
6901
  bool           acquired = false;
6,497✔
6902

6903
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
6,497!
6904
  TSDB_CHECK_NULL(rows, code, lino, _end, TSDB_CODE_INVALID_PARA);
6,497!
6905

6906
  *rows = 0;
6,497✔
6907
  pStatus = &pReader->status;
6,497✔
6908
  code = tsdbAcquireReader(pReader);
6,497✔
6909
  TSDB_CHECK_CODE(code, lino, _end);
6,510!
6910
  acquired = true;
6,510✔
6911

6912
  if (pReader->flag == READER_STATUS_SUSPEND) {
6,510✔
6913
    code = tsdbReaderResume2(pReader);
1,093✔
6914
    TSDB_CHECK_CODE(code, lino, _end);
1,093!
6915
  }
6916

6917
  int32_t iter = 0;
6,510✔
6918
  pStatus->pTableIter = tSimpleHashIterate(pStatus->pTableMap, NULL, &iter);
6,510✔
6919

6920
  while (pStatus->pTableIter != NULL) {
20,858✔
6921
    STableBlockScanInfo* pBlockScanInfo = *(STableBlockScanInfo**)pStatus->pTableIter;
14,330✔
6922

6923
    STbData* d = NULL;
14,330✔
6924
    if (pReader->pReadSnap->pMem != NULL) {
14,330✔
6925
      d = tsdbGetTbDataFromMemTable(pReader->pReadSnap->pMem, pReader->info.suid, pBlockScanInfo->uid);
14,035✔
6926
      if (d != NULL) {
14,043✔
6927
        *rows += tsdbGetNRowsInTbData(d);
3,252✔
6928
      }
6929
    }
6930

6931
    STbData* di = NULL;
14,338✔
6932
    if (pReader->pReadSnap->pIMem != NULL) {
14,338✔
6933
      di = tsdbGetTbDataFromMemTable(pReader->pReadSnap->pIMem, pReader->info.suid, pBlockScanInfo->uid);
2✔
6934
      if (di != NULL) {
2!
6935
        *rows += tsdbGetNRowsInTbData(di);
2✔
6936
      }
6937
    }
6938

6939
    // current table is exhausted, let's try the next table
6940
    pStatus->pTableIter = tSimpleHashIterate(pStatus->pTableMap, pStatus->pTableIter, &iter);
14,338✔
6941
  }
6942

6943
_end:
6,528✔
6944
  if (code != TSDB_CODE_SUCCESS) {
6,528!
6945
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6946
  }
6947
  if (acquired) {
6,520!
6948
    (void)tsdbReleaseReader(pReader);
6,520✔
6949
  }
6950
  return code;
6,520✔
6951
}
6952

6953
int32_t tsdbGetTableSchema(SMeta* pMeta, int64_t uid, STSchema** pSchema, int64_t* suid, SSchemaWrapper** pTagSchema) {
440✔
6954
  SMetaReader mr = {0};
440✔
6955
  metaReaderDoInit(&mr, pMeta, META_READER_LOCK);
440✔
6956
  int32_t code = metaReaderGetTableEntryByUidCache(&mr, uid);
440✔
6957
  if (code != TSDB_CODE_SUCCESS) {
440!
6958
    code = TSDB_CODE_TDB_INVALID_TABLE_ID;
×
6959
    metaReaderClear(&mr);
×
6960
    return code;
×
6961
  }
6962

6963
  *suid = 0;
440✔
6964

6965
  if (mr.me.type == TSDB_CHILD_TABLE) {
440✔
6966
    tDecoderClear(&mr.coder);
330✔
6967
    *suid = mr.me.ctbEntry.suid;
330✔
6968
    code = metaReaderGetTableEntryByUidCache(&mr, *suid);
330✔
6969
    if (code != TSDB_CODE_SUCCESS) {
330!
6970
      code = TSDB_CODE_TDB_INVALID_TABLE_ID;
×
6971
      metaReaderClear(&mr);
×
6972
      return code;
×
6973
    }
6974
  } else if (mr.me.type == TSDB_NORMAL_TABLE) {  // do nothing
110✔
6975
  } else if (mr.me.type == TSDB_SUPER_TABLE) {
4!
6976
    tDecoderClear(&mr.coder);
4✔
6977
    *suid = uid;
4✔
6978
    code = metaReaderGetTableEntryByUidCache(&mr, *suid);
4✔
6979
    if (code != TSDB_CODE_SUCCESS) {
4!
6980
      code = TSDB_CODE_TDB_INVALID_TABLE_ID;
×
6981
      metaReaderClear(&mr);
×
6982
      return code;
×
6983
    }
6984
    *pTagSchema = tCloneSSchemaWrapper(&mr.me.stbEntry.schemaTag);
4✔
6985
  } else {
6986
    code = TSDB_CODE_INVALID_PARA;
×
6987
    tsdbError("invalid mr.me.type:%d, code:%s", mr.me.type, tstrerror(code));
×
6988
    metaReaderClear(&mr);
×
6989
    return code;
×
6990
  }
6991

6992
  metaReaderClear(&mr);
440✔
6993

6994
  // get the newest table schema version
6995
  code = metaGetTbTSchemaEx(pMeta, *suid, uid, -1, pSchema);
440✔
6996
  return code;
440✔
6997
}
6998

6999
int32_t tsdbTakeReadSnap2(STsdbReader* pReader, _query_reseek_func_t reseek, STsdbReadSnap** ppSnap, const char* id) {
8,273,278✔
7000
  int32_t        code = 0;
8,273,278✔
7001
  STsdb*         pTsdb = pReader->pTsdb;
8,273,278✔
7002
  SVersionRange* pRange = &pReader->info.verRange;
8,273,278✔
7003
  int32_t        lino = 0;
8,273,278✔
7004
  *ppSnap = NULL;
8,273,278✔
7005

7006
  // lock
7007
  code = taosThreadMutexLock(&pTsdb->mutex);
8,273,278✔
7008
  if (code != TSDB_CODE_SUCCESS) {
8,285,203!
7009
    tsdbError("%s failed to lock tsdb, code:%s", id, tstrerror(code));
×
7010
    return code;
×
7011
  }
7012

7013
  // alloc
7014
  STsdbReadSnap* pSnap = (STsdbReadSnap*)taosMemoryCalloc(1, sizeof(STsdbReadSnap));
8,285,203✔
7015
  if (pSnap == NULL) {
8,283,109!
7016
    (void)taosThreadMutexUnlock(&pTsdb->mutex);
×
7017
    TSDB_CHECK_NULL(pSnap, code, lino, _exit, terrno);
×
7018
  }
7019

7020
  // take snapshot
7021
  if (pTsdb->mem && (pRange->minVer <= pTsdb->mem->maxVer && pRange->maxVer >= pTsdb->mem->minVer)) {
8,283,109!
7022
    pSnap->pMem = pTsdb->mem;
7,512,717✔
7023
    pSnap->pNode = taosMemoryMalloc(sizeof(*pSnap->pNode));
7,512,717!
7024
    if (pSnap->pNode == NULL) {
7,510,595!
7025
      (void)taosThreadMutexUnlock(&pTsdb->mutex);
×
7026
      TSDB_CHECK_NULL(pSnap->pNode, code, lino, _exit, terrno);
×
7027
    }
7028

7029
    pSnap->pNode->pQHandle = pReader;
7,510,595✔
7030
    pSnap->pNode->reseek = reseek;
7,510,595✔
7031

7032
    code = tsdbRefMemTable(pTsdb->mem, pSnap->pNode);
7,510,595✔
7033
    TSDB_CHECK_CODE(code, lino, _exit);
7,511,020!
7034
  }
7035

7036
  if (pTsdb->imem && (pRange->minVer <= pTsdb->imem->maxVer && pRange->maxVer >= pTsdb->imem->minVer)) {
8,281,412!
7037
    pSnap->pIMem = pTsdb->imem;
25,339✔
7038
    pSnap->pINode = taosMemoryMalloc(sizeof(*pSnap->pINode));
25,339!
7039
    if (pSnap->pINode == NULL) {
25,338!
7040
      code = terrno;
×
7041

7042
      if (pTsdb->mem && pSnap->pNode) {
×
7043
        tsdbUnrefMemTable(pTsdb->mem, pSnap->pNode, true);  // unref the previous refed mem
×
7044
      }
7045

7046
      (void)taosThreadMutexUnlock(&pTsdb->mutex);
×
7047
      goto _exit;
×
7048
    }
7049

7050
    pSnap->pINode->pQHandle = pReader;
25,338✔
7051
    pSnap->pINode->reseek = reseek;
25,338✔
7052

7053
    code = tsdbRefMemTable(pTsdb->imem, pSnap->pINode);
25,338✔
7054
    TSDB_CHECK_CODE(code, lino, _exit);
25,338!
7055
  }
7056

7057
  // fs
7058
  code = tsdbFSCreateRefSnapshotWithoutLock(pTsdb->pFS, &pSnap->pfSetArray);
8,281,411✔
7059
  if (code) {
8,278,604!
7060
    if (pSnap->pNode) {
×
7061
      tsdbUnrefMemTable(pTsdb->mem, pSnap->pNode, true);  // unref the previous refed mem
×
7062
    }
7063

7064
    if (pSnap->pINode) {
×
7065
      tsdbUnrefMemTable(pTsdb->imem, pSnap->pINode, true);
×
7066
    }
7067

7068
    (void)taosThreadMutexUnlock(&pTsdb->mutex);
×
7069
    TSDB_CHECK_CODE(code, lino, _exit);
×
7070
  }
7071

7072
  // unlock
7073
  (void)taosThreadMutexUnlock(&pTsdb->mutex);
8,278,604✔
7074
  *ppSnap = pSnap;
8,282,640✔
7075

7076
  tsdbTrace("%s vgId:%d, take read snapshot", id, TD_VID(pTsdb->pVnode));
8,282,640✔
7077
  return code;
8,280,949✔
7078

7079
_exit:
×
7080
  tsdbError("%s vgId:%d take read snapshot failed, line:%d code:%s", id, TD_VID(pTsdb->pVnode), lino, tstrerror(code));
×
7081

7082
  if (pSnap) {
×
7083
    if (pSnap->pNode) taosMemoryFree(pSnap->pNode);
×
7084
    if (pSnap->pINode) taosMemoryFree(pSnap->pINode);
×
7085
    taosMemoryFree(pSnap);
×
7086
  }
7087
  return code;
×
7088
}
7089

7090
void tsdbUntakeReadSnap2(STsdbReader* pReader, STsdbReadSnap* pSnap, bool proactive) {
8,285,158✔
7091
  STsdb* pTsdb = pReader->pTsdb;
8,285,158✔
7092

7093
  if (pSnap) {
8,285,158!
7094
    if (pSnap->pMem) {
8,285,622✔
7095
      tsdbUnrefMemTable(pSnap->pMem, pSnap->pNode, proactive);
7,513,825✔
7096
    }
7097

7098
    if (pSnap->pIMem) {
8,285,672✔
7099
      tsdbUnrefMemTable(pSnap->pIMem, pSnap->pINode, proactive);
25,340✔
7100
    }
7101

7102
    if (pSnap->pNode) taosMemoryFree(pSnap->pNode);
8,285,672✔
7103
    if (pSnap->pINode) taosMemoryFree(pSnap->pINode);
8,285,568!
7104

7105
    tsdbFSDestroyRefSnapshot(&pSnap->pfSetArray);
8,285,568✔
7106

7107
    taosMemoryFree(pSnap);
8,285,322!
7108
  }
7109
  tsdbTrace("vgId:%d, untake read snapshot", TD_VID(pTsdb->pVnode));
8,285,476✔
7110
}
8,285,476✔
7111

7112
// if failed, do nothing
7113
int32_t tsdbReaderSetId(void* p, const char* idstr) {
740✔
7114
  int32_t      code = TSDB_CODE_SUCCESS;
740✔
7115
  int32_t      lino = 0;
740✔
7116
  STsdbReader* pReader = (STsdbReader*)p;
740✔
7117

7118
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
740!
7119

7120
  taosMemoryFreeClear(pReader->idStr);
740!
7121
  pReader->idStr = taosStrdup(idstr);
740!
7122
  TSDB_CHECK_NULL(pReader->idStr, code, lino, _end, terrno);
739!
7123

7124
  pReader->status.fileIter.pSttBlockReader->mergeTree.idStr = pReader->idStr;
739✔
7125

7126
_end:
739✔
7127
  if (code != TSDB_CODE_SUCCESS) {
739!
7128
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7129
  }
7130
  return code;
739✔
7131
}
7132

7133
void tsdbReaderSetCloseFlag(STsdbReader* pReader) { /*pReader->code = TSDB_CODE_TSC_QUERY_CANCELLED;*/
1,018✔
7134
}
1,018✔
7135

7136
void tsdbSetFilesetDelimited(STsdbReader* pReader) { pReader->bFilesetDelimited = true; }
3,781,702✔
7137

7138
void tsdbReaderSetNotifyCb(STsdbReader* pReader, TsdReaderNotifyCbFn notifyFn, void* param) {
3,782,668✔
7139
  pReader->notifyFn = notifyFn;
3,782,668✔
7140
  pReader->notifyParam = param;
3,782,668✔
7141
}
3,782,668✔
7142

7143
#if 0
7144
static int32_t tsdbEncodeLastProcKeys(const SSHashObj* pTableMap, void** buf) {
7145
  void* p = NULL;
7146
  int32_t iter = 0;
7147
  int32_t len = 0;
7148

7149
  len += taosEncodeFixedI32(buf, tSimpleHashGetSize(pTableMap)); // number of tables
7150

7151
  while ((p = tSimpleHashIterate(pTableMap, p, &iter)) != NULL) {
7152
    uint64_t uid = *(uint64_t*)tSimpleHashGetKey(p, NULL);
7153
    SRowKey* pKey = &(*(STableBlockScanInfo**)p)->lastProcKey;
7154

7155
    len += taosEncodeFixedU64(buf, uid);            // table uid
7156
    len += taosEncodeFixedI64(buf, pKey->ts);       // last read ts
7157
    len += taosEncodeFixedU8(buf, pKey->numOfPKs);  // number of pks
7158

7159
    if (pKey->numOfPKs > 0) {
7160
      int8_t type = pKey->pks[0].type;
7161
      len += taosEncodeFixedI8(buf, type);  // pk data type
7162

7163
      uint8_t* pData = 0;
7164
      uint32_t nData = 0;
7165
      if (IS_VAR_DATA_TYPE(type)) {
7166
        pData = pKey->pks[0].pData;
7167
        nData = pKey->pks[0].nData;
7168
        len += taosEncodeFixedU32(buf, nData);  // data length
7169
      } else {
7170
        pData = (uint8_t*)&pKey->pks[0].val;
7171
        nData = tDataTypes[pKey->pks[0].type].bytes;
7172
      }
7173
      len += taosEncodeBinary(buf, pData, nData); // data value
7174
    }
7175
  }
7176

7177
  return len;
7178
}
7179

7180
static int32_t tsdbDecodeLastProcKeys(const void* buf, int32_t len, SSHashObj* pTableMap) {
7181
  int32_t code = TSDB_CODE_SUCCESS;
7182
  int32_t lino = 0;
7183
  const void *begin = buf;
7184

7185
  TSDB_CHECK_NULL(buf, code, lino, _end, TSDB_CODE_INVALID_PARA);
7186
  TSDB_CHECK_NULL(pTableMap, code, lino, _end, TSDB_CODE_INVALID_PARA);
7187

7188
  int32_t numOfTables = 0;
7189
  buf = taosDecodeFixedI32(buf, &numOfTables);
7190

7191
  TSDB_CHECK_CONDITION(numOfTables == tSimpleHashGetSize(pTableMap), code, lino, _end, TSDB_CODE_INVALID_PARA);
7192

7193
  for (int32_t i = 0; i < numOfTables; ++i) {
7194
    uint64_t uid = 0;
7195
    buf = taosDecodeFixedU64(buf, &uid);
7196

7197
    STableBlockScanInfo** pScanInfo = tSimpleHashGet(pTableMap, &uid, sizeof(uid));
7198
    if (pScanInfo == NULL) {
7199
      tsdbError("failed to find scan info of uid: %" PRIu64 " when decoding last proc keys", uid);
7200
      TSDB_CHECK_NULL(pScanInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
7201
    }
7202
    SRowKey* pKey = &(*pScanInfo)->lastProcKey;
7203

7204
    buf = taosDecodeFixedI64(buf, &pKey->ts);
7205
    uint8_t numOfPks = 0;
7206
    buf = taosDecodeFixedU8(buf, &numOfPks);
7207
    if (numOfPks != pKey->numOfPKs) {
7208
      tsdbError("numOfPks of uid %" PRIu64 " mismatch: [%u/%u]", uid, numOfPks, pKey->numOfPKs);
7209
      TSDB_CHECK_CONDITION(numOfPks == pKey->numOfPKs, code, lino, _end, TSDB_CODE_INVALID_PARA);
7210
    }
7211

7212
    if (numOfPks > 0) {
7213
      int8_t type = 0;
7214
      buf = taosDecodeFixedI8(buf, &type);
7215
      if (type != pKey->pks[0].type) {
7216
        tsdbError("pk type of uid %" PRIu64 " mismatch: [%d/%d]", uid, type, pKey->pks[0].type);
7217
        TSDB_CHECK_CONDITION(type == pKey->pks[0].type, code, lino, _end, TSDB_CODE_INVALID_PARA);
7218
      }
7219

7220
      uint8_t *pData = 0;
7221
      uint32_t nData = 0;
7222
      if (IS_VAR_DATA_TYPE(type)) {
7223
        pData = pKey->pks[0].pData;
7224
        buf = taosDecodeFixedU32(buf, &nData);
7225
        pKey->pks[0].nData = nData;
7226
      } else {
7227
        pKey->pks[0].val = 0;
7228
        pData = (uint8_t*)&pKey->pks[0].val;
7229
        nData = tDataTypes[type].bytes;
7230
      }
7231
      buf = taosDecodeBinary(buf, (void **)&pData, nData);
7232
    }
7233
  }
7234

7235
  TSDB_CHECK_CONDITION(buf == begin + len , code, lino, _end, TSDB_CODE_INVALID_PARA);
7236

7237
_end:
7238
  if (code != TSDB_CODE_SUCCESS) {
7239
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
7240
  }
7241
  return code;
7242
}
7243
#endif
7244

7245
int32_t tsdbReaderGetProgress(const STsdbReader* pReader, void** pBuf, uint64_t* pLen) {
×
7246
  int32_t code = TSDB_CODE_SUCCESS;
×
7247
#if 0
7248
  int32_t              lino = 0;
7249
  void*                buf = NULL;
7250
  int32_t len = 0;
7251

7252
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
7253
  TSDB_CHECK_NULL(pBuf, code, lino, _end, TSDB_CODE_INVALID_PARA);
7254
  TSDB_CHECK_NULL(pLen, code, lino, _end, TSDB_CODE_INVALID_PARA);
7255

7256
  *pBuf = NULL;
7257
  *pLen = tsdbEncodeLastProcKeys(pReader->status.pTableMap, NULL);
7258

7259
  buf = taosMemoryMalloc(*pLen);
7260
  TSDB_CHECK_NULL(buf, code, lino, _end, terrno);
7261
  *pBuf = buf;
7262
  len = tsdbEncodeLastProcKeys(pReader->status.pTableMap, &buf);
7263
  TSDB_CHECK_CONDITION(len == *pLen, code, lino, _end, TSDB_CODE_INTERNAL_ERROR);
7264
  buf = NULL;
7265

7266
_end:
7267
  if (code != TSDB_CODE_SUCCESS) {
7268
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
7269
  }
7270
  if (buf != NULL) {
7271
    taosMemoryFreeClear(buf);
7272
  }
7273
#endif
7274
  return code;
×
7275
}
7276

7277
int32_t tsdbReaderSetProgress(STsdbReader* pReader, const void* buf, uint64_t len) {
×
7278
  int32_t code = TSDB_CODE_SUCCESS;
×
7279
#if 0
7280
  int32_t lino = 0;
7281

7282
  if (len == 0) {
7283
    goto _end;
7284
  }
7285

7286
  TSDB_CHECK_NULL(pReader, code, lino, _end, TSDB_CODE_INVALID_PARA);
7287
  TSDB_CHECK_NULL(buf, code, lino, _end, TSDB_CODE_INVALID_PARA);
7288

7289
  code = tsdbDecodeLastProcKeys(buf, len, pReader->status.pTableMap);
7290
  TSDB_CHECK_CODE(code, lino, _end);
7291

7292
_end:
7293
  if (code != TSDB_CODE_SUCCESS) {
7294
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
7295
  }
7296
#endif
7297
  return code;
×
7298
}
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