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

taosdata / TDengine / #4914

06 Jan 2026 01:30AM UTC coverage: 64.876% (-0.008%) from 64.884%
#4914

push

travis-ci

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

180 of 319 new or added lines in 14 files covered. (56.43%)

3475 existing lines in 124 files now uncovered.

194993 of 300563 relevant lines covered (64.88%)

116239151.85 hits per line

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

79.91
/source/dnode/vnode/src/tsdb/tsdbUtil.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 "tcompression.h"
17
#include "tdataformat.h"
18
#include "tsdb.h"
19
#include "tsdbDef.h"
20

21
int32_t tsdbGetColCmprAlgFromSet(SHashObj *set, int16_t colId, uint32_t *alg);
22

23
static int32_t tBlockDataCompressKeyPart(SBlockData *bData, SDiskDataHdr *hdr, SBuffer *buffer, SBuffer *assist,
24
                                         SColCompressInfo *pCompressExt);
25

26
// SMapData =======================================================================
27
void tMapDataReset(SMapData *pMapData) {
×
28
  pMapData->nItem = 0;
×
29
  pMapData->nData = 0;
×
30
}
×
31

32
void tMapDataClear(SMapData *pMapData) {
×
33
  tFree(pMapData->aOffset);
×
34
  tFree(pMapData->pData);
×
35
  pMapData->pData = NULL;
×
36
  pMapData->aOffset = NULL;
×
37
}
×
38

39
#ifdef BUILD_NO_CALL
40
int32_t tMapDataPutItem(SMapData *pMapData, void *pItem, int32_t (*tPutItemFn)(uint8_t *, void *)) {
41
  int32_t code = 0;
42
  int32_t offset = pMapData->nData;
43
  int32_t nItem = pMapData->nItem;
44

45
  pMapData->nItem++;
46
  pMapData->nData += tPutItemFn(NULL, pItem);
47

48
  // alloc
49
  code = tRealloc((uint8_t **)&pMapData->aOffset, sizeof(int32_t) * pMapData->nItem);
50
  if (code) goto _exit;
51
  code = tRealloc(&pMapData->pData, pMapData->nData);
52
  if (code) goto _exit;
53

54
  // put
55
  pMapData->aOffset[nItem] = offset;
56
  tPutItemFn(pMapData->pData + offset, pItem);
57

58
_exit:
59
  return code;
60
}
61

62
int32_t tMapDataCopy(SMapData *pFrom, SMapData *pTo) {
63
  int32_t code = 0;
64

65
  pTo->nItem = pFrom->nItem;
66
  pTo->nData = pFrom->nData;
67
  code = tRealloc((uint8_t **)&pTo->aOffset, sizeof(int32_t) * pFrom->nItem);
68
  if (code) goto _exit;
69
  code = tRealloc(&pTo->pData, pFrom->nData);
70
  if (code) goto _exit;
71
  memcpy(pTo->aOffset, pFrom->aOffset, sizeof(int32_t) * pFrom->nItem);
72
  memcpy(pTo->pData, pFrom->pData, pFrom->nData);
73

74
_exit:
75
  return code;
76
}
77

78
int32_t tMapDataSearch(SMapData *pMapData, void *pSearchItem, int32_t (*tGetItemFn)(uint8_t *, void *),
79
                       int32_t (*tItemCmprFn)(const void *, const void *), void *pItem) {
80
  int32_t code = 0;
81
  int32_t lidx = 0;
82
  int32_t ridx = pMapData->nItem - 1;
83
  int32_t midx;
84
  int32_t c;
85

86
  while (lidx <= ridx) {
87
    midx = (lidx + ridx) / 2;
88

89
    tMapDataGetItemByIdx(pMapData, midx, pItem, tGetItemFn);
90

91
    c = tItemCmprFn(pSearchItem, pItem);
92
    if (c == 0) {
93
      goto _exit;
94
    } else if (c < 0) {
95
      ridx = midx - 1;
96
    } else {
97
      lidx = midx + 1;
98
    }
99
  }
100

101
  code = TSDB_CODE_NOT_FOUND;
102

103
_exit:
104
  return code;
105
}
106
#endif
107

108
void tMapDataGetItemByIdx(SMapData *pMapData, int32_t idx, void *pItem, int32_t (*tGetItemFn)(uint8_t *, void *)) {
×
109
  int32_t r = tGetItemFn(pMapData->pData + pMapData->aOffset[idx], pItem);
×
110
}
×
111

112
#ifdef BUILD_NO_CALL
113
int32_t tMapDataToArray(SMapData *pMapData, int32_t itemSize, int32_t (*tGetItemFn)(uint8_t *, void *),
114
                        SArray **ppArray) {
115
  int32_t code = 0;
116

117
  SArray *pArray = taosArrayInit(pMapData->nItem, itemSize);
118
  if (pArray == NULL) {
119
    code = terrno;
120
    goto _exit;
121
  }
122

123
  for (int32_t i = 0; i < pMapData->nItem; i++) {
124
    tMapDataGetItemByIdx(pMapData, i, taosArrayReserve(pArray, 1), tGetItemFn);
125
  }
126

127
_exit:
128
  *ppArray = pArray;
129
  return code;
130
}
131

132
int32_t tPutMapData(uint8_t *p, SMapData *pMapData) {
133
  int32_t n = 0;
134

135
  n += tPutI32v(p ? p + n : p, pMapData->nItem);
136
  if (pMapData->nItem) {
137
    int32_t lOffset = 0;
138
    for (int32_t iItem = 0; iItem < pMapData->nItem; iItem++) {
139
      n += tPutI32v(p ? p + n : p, pMapData->aOffset[iItem] - lOffset);
140
      lOffset = pMapData->aOffset[iItem];
141
    }
142

143
    n += tPutI32v(p ? p + n : p, pMapData->nData);
144
    if (p) {
145
      memcpy(p + n, pMapData->pData, pMapData->nData);
146
    }
147
    n += pMapData->nData;
148
  }
149

150
  return n;
151
}
152
#endif
153

154
int32_t tGetMapData(uint8_t *p, SMapData *pMapData, int32_t *decodedSize) {
×
155
  int32_t n = 0;
×
156
  int32_t code;
157
  int32_t offset;
158

159
  tMapDataReset(pMapData);
×
160

161
  n += tGetI32v(p + n, &pMapData->nItem);
×
162
  if (pMapData->nItem) {
×
163
    code = tRealloc((uint8_t **)&pMapData->aOffset, sizeof(int32_t) * pMapData->nItem);
×
164
    if (code) {
×
165
      return code;
×
166
    }
167

168
    int32_t lOffset = 0;
×
169
    for (int32_t iItem = 0; iItem < pMapData->nItem; iItem++) {
×
170
      n += tGetI32v(p + n, &pMapData->aOffset[iItem]);
×
171
      pMapData->aOffset[iItem] += lOffset;
×
172
      lOffset = pMapData->aOffset[iItem];
×
173
    }
174

175
    n += tGetI32v(p + n, &pMapData->nData);
×
176
    code = tRealloc(&pMapData->pData, pMapData->nData);
×
177
    if (code) {
×
178
      return code;
×
179
    }
180
    memcpy(pMapData->pData, p + n, pMapData->nData);
×
181
    n += pMapData->nData;
×
182
  }
183

184
  if (decodedSize) {
×
185
    *decodedSize = n;
×
186
  }
187

188
  return 0;
×
189
}
190

191
#ifdef BUILD_NO_CALL
192
// TABLEID =======================================================================
193
int32_t tTABLEIDCmprFn(const void *p1, const void *p2) {
194
  TABLEID *pId1 = (TABLEID *)p1;
195
  TABLEID *pId2 = (TABLEID *)p2;
196

197
  if (pId1->suid < pId2->suid) {
198
    return -1;
199
  } else if (pId1->suid > pId2->suid) {
200
    return 1;
201
  }
202

203
  if (pId1->uid < pId2->uid) {
204
    return -1;
205
  } else if (pId1->uid > pId2->uid) {
206
    return 1;
207
  }
208

209
  return 0;
210
}
211

212
// SBlockIdx ======================================================
213
int32_t tPutBlockIdx(uint8_t *p, void *ph) {
214
  int32_t    n = 0;
215
  SBlockIdx *pBlockIdx = (SBlockIdx *)ph;
216

217
  n += tPutI64(p ? p + n : p, pBlockIdx->suid);
218
  n += tPutI64(p ? p + n : p, pBlockIdx->uid);
219
  n += tPutI64v(p ? p + n : p, pBlockIdx->offset);
220
  n += tPutI64v(p ? p + n : p, pBlockIdx->size);
221

222
  return n;
223
}
224
#endif
225

226
int32_t tGetBlockIdx(uint8_t *p, void *ph) {
×
227
  int32_t    n = 0;
×
228
  SBlockIdx *pBlockIdx = (SBlockIdx *)ph;
×
229

230
  n += tGetI64(p + n, &pBlockIdx->suid);
×
231
  n += tGetI64(p + n, &pBlockIdx->uid);
×
232
  n += tGetI64v(p + n, &pBlockIdx->offset);
×
233
  n += tGetI64v(p + n, &pBlockIdx->size);
×
234

235
  return n;
×
236
}
237

238
#ifdef BUILD_NO_CALL
239
int32_t tCmprBlockIdx(void const *lhs, void const *rhs) {
240
  SBlockIdx *lBlockIdx = (SBlockIdx *)lhs;
241
  SBlockIdx *rBlockIdx = (SBlockIdx *)rhs;
242

243
  if (lBlockIdx->suid < rBlockIdx->suid) {
244
    return -1;
245
  } else if (lBlockIdx->suid > rBlockIdx->suid) {
246
    return 1;
247
  }
248

249
  if (lBlockIdx->uid < rBlockIdx->uid) {
250
    return -1;
251
  } else if (lBlockIdx->uid > rBlockIdx->uid) {
252
    return 1;
253
  }
254

255
  return 0;
256
}
257

258
int32_t tCmprBlockL(void const *lhs, void const *rhs) {
259
  SBlockIdx *lBlockIdx = (SBlockIdx *)lhs;
260
  SSttBlk   *rBlockL = (SSttBlk *)rhs;
261

262
  if (lBlockIdx->suid < rBlockL->suid) {
263
    return -1;
264
  } else if (lBlockIdx->suid > rBlockL->suid) {
265
    return 1;
266
  }
267

268
  if (lBlockIdx->uid < rBlockL->minUid) {
269
    return -1;
270
  } else if (lBlockIdx->uid > rBlockL->maxUid) {
271
    return 1;
272
  }
273

274
  return 0;
275
}
276

277
// SDataBlk ======================================================
278
void tDataBlkReset(SDataBlk *pDataBlk) {
279
  *pDataBlk = (SDataBlk){.minKey = TSDBKEY_MAX, .maxKey = TSDBKEY_MIN, .minVer = VERSION_MAX, .maxVer = VERSION_MIN};
280
}
281

282
int32_t tPutDataBlk(uint8_t *p, void *ph) {
283
  int32_t   n = 0;
284
  SDataBlk *pDataBlk = (SDataBlk *)ph;
285

286
  n += tPutI64v(p ? p + n : p, pDataBlk->minKey.version);
287
  n += tPutI64v(p ? p + n : p, pDataBlk->minKey.ts);
288
  n += tPutI64v(p ? p + n : p, pDataBlk->maxKey.version);
289
  n += tPutI64v(p ? p + n : p, pDataBlk->maxKey.ts);
290
  n += tPutI64v(p ? p + n : p, pDataBlk->minVer);
291
  n += tPutI64v(p ? p + n : p, pDataBlk->maxVer);
292
  n += tPutI32v(p ? p + n : p, pDataBlk->nRow);
293
  n += tPutI8(p ? p + n : p, pDataBlk->hasDup);
294
  n += tPutI8(p ? p + n : p, pDataBlk->nSubBlock);
295
  for (int8_t iSubBlock = 0; iSubBlock < pDataBlk->nSubBlock; iSubBlock++) {
296
    n += tPutI64v(p ? p + n : p, pDataBlk->aSubBlock[iSubBlock].offset);
297
    n += tPutI32v(p ? p + n : p, pDataBlk->aSubBlock[iSubBlock].szBlock);
298
    n += tPutI32v(p ? p + n : p, pDataBlk->aSubBlock[iSubBlock].szKey);
299
  }
300
  if (pDataBlk->nSubBlock == 1 && !pDataBlk->hasDup) {
301
    n += tPutI64v(p ? p + n : p, pDataBlk->smaInfo.offset);
302
    n += tPutI32v(p ? p + n : p, pDataBlk->smaInfo.size);
303
  }
304

305
  return n;
306
}
307
#endif
308

309
int32_t tGetDataBlk(uint8_t *p, void *ph) {
×
310
  int32_t   n = 0;
×
311
  SDataBlk *pDataBlk = (SDataBlk *)ph;
×
312

313
  n += tGetI64v(p + n, &pDataBlk->minKey.version);
×
314
  n += tGetI64v(p + n, &pDataBlk->minKey.ts);
×
315
  n += tGetI64v(p + n, &pDataBlk->maxKey.version);
×
316
  n += tGetI64v(p + n, &pDataBlk->maxKey.ts);
×
317
  n += tGetI64v(p + n, &pDataBlk->minVer);
×
318
  n += tGetI64v(p + n, &pDataBlk->maxVer);
×
319
  n += tGetI32v(p + n, &pDataBlk->nRow);
×
320
  n += tGetI8(p + n, &pDataBlk->hasDup);
×
321
  n += tGetI8(p + n, &pDataBlk->nSubBlock);
×
322
  for (int8_t iSubBlock = 0; iSubBlock < pDataBlk->nSubBlock; iSubBlock++) {
×
323
    n += tGetI64v(p + n, &pDataBlk->aSubBlock[iSubBlock].offset);
×
324
    n += tGetI32v(p + n, &pDataBlk->aSubBlock[iSubBlock].szBlock);
×
325
    n += tGetI32v(p + n, &pDataBlk->aSubBlock[iSubBlock].szKey);
×
326
  }
327
  if (pDataBlk->nSubBlock == 1 && !pDataBlk->hasDup) {
×
328
    n += tGetI64v(p + n, &pDataBlk->smaInfo.offset);
×
329
    n += tGetI32v(p + n, &pDataBlk->smaInfo.size);
×
330
  } else {
331
    pDataBlk->smaInfo.offset = 0;
×
332
    pDataBlk->smaInfo.size = 0;
×
333
  }
334

335
  return n;
×
336
}
337

338
#ifdef BUILD_NO_CALL
339
int32_t tDataBlkCmprFn(const void *p1, const void *p2) {
340
  SDataBlk *pBlock1 = (SDataBlk *)p1;
341
  SDataBlk *pBlock2 = (SDataBlk *)p2;
342

343
  if (tsdbKeyCmprFn(&pBlock1->maxKey, &pBlock2->minKey) < 0) {
344
    return -1;
345
  } else if (tsdbKeyCmprFn(&pBlock1->minKey, &pBlock2->maxKey) > 0) {
346
    return 1;
347
  }
348

349
  return 0;
350
}
351

352
bool tDataBlkHasSma(SDataBlk *pDataBlk) {
353
  if (pDataBlk->nSubBlock > 1) return false;
354
  if (pDataBlk->hasDup) return false;
355

356
  return pDataBlk->smaInfo.size > 0;
357
}
358

359
// SSttBlk ======================================================
360
int32_t tPutSttBlk(uint8_t *p, void *ph) {
361
  int32_t  n = 0;
362
  SSttBlk *pSttBlk = (SSttBlk *)ph;
363

364
  n += tPutI64(p ? p + n : p, pSttBlk->suid);
365
  n += tPutI64(p ? p + n : p, pSttBlk->minUid);
366
  n += tPutI64(p ? p + n : p, pSttBlk->maxUid);
367
  n += tPutI64v(p ? p + n : p, pSttBlk->minKey);
368
  n += tPutI64v(p ? p + n : p, pSttBlk->maxKey);
369
  n += tPutI64v(p ? p + n : p, pSttBlk->minVer);
370
  n += tPutI64v(p ? p + n : p, pSttBlk->maxVer);
371
  n += tPutI32v(p ? p + n : p, pSttBlk->nRow);
372
  n += tPutI64v(p ? p + n : p, pSttBlk->bInfo.offset);
373
  n += tPutI32v(p ? p + n : p, pSttBlk->bInfo.szBlock);
374
  n += tPutI32v(p ? p + n : p, pSttBlk->bInfo.szKey);
375

376
  return n;
377
}
378
#endif
379

380
int32_t tGetSttBlk(uint8_t *p, void *ph) {
×
381
  int32_t  n = 0;
×
382
  SSttBlk *pSttBlk = (SSttBlk *)ph;
×
383

384
  n += tGetI64(p + n, &pSttBlk->suid);
×
385
  n += tGetI64(p + n, &pSttBlk->minUid);
×
386
  n += tGetI64(p + n, &pSttBlk->maxUid);
×
387
  n += tGetI64v(p + n, &pSttBlk->minKey);
×
388
  n += tGetI64v(p + n, &pSttBlk->maxKey);
×
389
  n += tGetI64v(p + n, &pSttBlk->minVer);
×
390
  n += tGetI64v(p + n, &pSttBlk->maxVer);
×
391
  n += tGetI32v(p + n, &pSttBlk->nRow);
×
392
  n += tGetI64v(p + n, &pSttBlk->bInfo.offset);
×
393
  n += tGetI32v(p + n, &pSttBlk->bInfo.szBlock);
×
394
  n += tGetI32v(p + n, &pSttBlk->bInfo.szKey);
×
395

396
  return n;
×
397
}
398

399
// SBlockCol ======================================================
400

401
static const int32_t BLOCK_WITH_ALG_VER = 2;
402

403
int32_t tPutBlockCol(SBuffer *buffer, const SBlockCol *pBlockCol, int32_t ver, uint32_t defaultCmprAlg) {
391,277,035✔
404
  int32_t code;
405

406
  if ((code = tBufferPutI16v(buffer, pBlockCol->cid))) return code;
782,577,565✔
407
  if ((code = tBufferPutI8(buffer, pBlockCol->type))) return code;
782,573,584✔
408
  if ((code = tBufferPutI8(buffer, pBlockCol->cflag))) return code;
782,511,616✔
409
  if ((code = tBufferPutI8(buffer, pBlockCol->flag))) return code;
782,525,188✔
410
  if ((code = tBufferPutI32v(buffer, pBlockCol->szOrigin))) return code;
782,565,751✔
411

412
  if (pBlockCol->flag != HAS_NULL) {
391,279,125✔
413
    if (pBlockCol->flag != HAS_VALUE) {
375,051,888✔
414
      if ((code = tBufferPutI32v(buffer, pBlockCol->szBitmap))) return code;
52,522,548✔
415
    }
416

417
    if (IS_VAR_DATA_TYPE(pBlockCol->type)) {
375,046,228✔
418
      if ((code = tBufferPutI32v(buffer, pBlockCol->szOffset))) return code;
96,299,124✔
419
    }
420

421
    if (pBlockCol->flag != (HAS_NULL | HAS_NONE)) {
375,051,624✔
422
      if ((code = tBufferPutI32v(buffer, pBlockCol->szValue))) return code;
749,872,652✔
423
    }
424

425
    if ((code = tBufferPutI32v(buffer, pBlockCol->offset))) return code;
750,080,211✔
426
  }
427
  if (ver >= BLOCK_WITH_ALG_VER) {
391,251,722✔
428
    if ((code = tBufferPutU32(buffer, pBlockCol->alg))) return code;
782,490,270✔
429
  } else {
UNCOV
430
    if ((code = tBufferPutU32(buffer, defaultCmprAlg))) return code;
×
431
  }
432
  return 0;
391,238,548✔
433
}
434

435
int32_t tGetBlockCol(SBufferReader *br, SBlockCol *pBlockCol, int32_t ver, uint32_t defaultCmprAlg) {
1,197,388,219✔
436
  int32_t code;
437

438
  if ((code = tBufferGetI16v(br, &pBlockCol->cid))) return code;
1,197,388,219✔
439
  if ((code = tBufferGetI8(br, &pBlockCol->type))) return code;
1,197,594,472✔
440
  if ((code = tBufferGetI8(br, &pBlockCol->cflag))) return code;
1,197,666,607✔
441
  if ((code = tBufferGetI8(br, &pBlockCol->flag))) return code;
1,197,685,782✔
442
  if ((code = tBufferGetI32v(br, &pBlockCol->szOrigin))) return code;
1,197,696,212✔
443

444
  pBlockCol->szBitmap = 0;
1,197,689,042✔
445
  pBlockCol->szOffset = 0;
1,197,685,054✔
446
  pBlockCol->szValue = 0;
1,197,650,053✔
447
  pBlockCol->offset = 0;
1,197,612,477✔
448

449
  if (pBlockCol->flag != HAS_NULL) {
1,197,655,464✔
450
    if (pBlockCol->flag != HAS_VALUE) {
1,112,621,584✔
451
      if ((code = tBufferGetI32v(br, &pBlockCol->szBitmap))) return code;
240,108,466✔
452
    }
453

454
    if (IS_VAR_DATA_TYPE(pBlockCol->type)) {
1,112,616,952✔
455
      if ((code = tBufferGetI32v(br, &pBlockCol->szOffset))) return code;
166,166,858✔
456
    }
457

458
    if (pBlockCol->flag != (HAS_NULL | HAS_NONE)) {
1,112,602,531✔
459
      if ((code = tBufferGetI32v(br, &pBlockCol->szValue))) return code;
1,112,274,821✔
460
    }
461

462
    if ((code = tBufferGetI32v(br, &pBlockCol->offset))) return code;
1,112,650,771✔
463
  }
464

465
  if (ver >= BLOCK_WITH_ALG_VER) {
1,197,703,523✔
466
    if ((code = tBufferGetU32(br, &pBlockCol->alg))) return code;
1,197,703,523✔
467
  } else {
468
    pBlockCol->alg = defaultCmprAlg;
×
469
  }
470

471
  return 0;
1,197,582,825✔
472
}
473

474
#ifdef BUILD_NO_CALL
475
int32_t tBlockColCmprFn(const void *p1, const void *p2) {
476
  if (((SBlockCol *)p1)->cid < ((SBlockCol *)p2)->cid) {
477
    return -1;
478
  } else if (((SBlockCol *)p1)->cid > ((SBlockCol *)p2)->cid) {
479
    return 1;
480
  }
481

482
  return 0;
483
}
484

485
// SDelIdx ======================================================
486
int32_t tCmprDelIdx(void const *lhs, void const *rhs) {
487
  SDelIdx *lDelIdx = (SDelIdx *)lhs;
488
  SDelIdx *rDelIdx = (SDelIdx *)rhs;
489

490
  if (lDelIdx->suid < rDelIdx->suid) {
491
    return -1;
492
  } else if (lDelIdx->suid > rDelIdx->suid) {
493
    return 1;
494
  }
495

496
  if (lDelIdx->uid < rDelIdx->uid) {
497
    return -1;
498
  } else if (lDelIdx->uid > rDelIdx->uid) {
499
    return 1;
500
  }
501

502
  return 0;
503
}
504

505
int32_t tPutDelIdx(uint8_t *p, void *ph) {
506
  SDelIdx *pDelIdx = (SDelIdx *)ph;
507
  int32_t  n = 0;
508

509
  n += tPutI64(p ? p + n : p, pDelIdx->suid);
510
  n += tPutI64(p ? p + n : p, pDelIdx->uid);
511
  n += tPutI64v(p ? p + n : p, pDelIdx->offset);
512
  n += tPutI64v(p ? p + n : p, pDelIdx->size);
513

514
  return n;
515
}
516
#endif
517

518
int32_t tGetDelIdx(uint8_t *p, void *ph) {
×
519
  SDelIdx *pDelIdx = (SDelIdx *)ph;
×
520
  int32_t  n = 0;
×
521

522
  n += tGetI64(p + n, &pDelIdx->suid);
×
523
  n += tGetI64(p + n, &pDelIdx->uid);
×
524
  n += tGetI64v(p + n, &pDelIdx->offset);
×
525
  n += tGetI64v(p + n, &pDelIdx->size);
×
526

527
  return n;
×
528
}
529

530
#ifdef BUILD_NO_CALL
531
// SDelData ======================================================
532
int32_t tPutDelData(uint8_t *p, void *ph) {
533
  SDelData *pDelData = (SDelData *)ph;
534
  int32_t   n = 0;
535

536
  n += tPutI64v(p ? p + n : p, pDelData->version);
537
  n += tPutI64(p ? p + n : p, pDelData->sKey);
538
  n += tPutI64(p ? p + n : p, pDelData->eKey);
539

540
  return n;
541
}
542
#endif
543

544
int32_t tGetDelData(uint8_t *p, void *ph) {
×
545
  SDelData *pDelData = (SDelData *)ph;
×
546
  int32_t   n = 0;
×
547

548
  n += tGetI64v(p + n, &pDelData->version);
×
549
  n += tGetI64(p + n, &pDelData->sKey);
×
550
  n += tGetI64(p + n, &pDelData->eKey);
×
551

552
  return n;
×
553
}
554

555
int32_t tsdbKeyFid(TSKEY key, int32_t minutes, int8_t precision) {
152,051,474✔
556
  int64_t fid;
557
  if (key < 0) {
152,051,474✔
558
    fid = ((key + 1) / tsTickPerMin[precision] / minutes - 1);
1,107,332✔
559
    return (fid < INT32_MIN) ? INT32_MIN : (int32_t)fid;
1,107,332✔
560
  } else {
561
    fid = ((key / tsTickPerMin[precision] / minutes));
150,944,142✔
562
    return (fid > INT32_MAX) ? INT32_MAX : (int32_t)fid;
150,999,661✔
563
  }
564
}
565

566
void tsdbFidKeyRange(int32_t fid, int32_t minutes, int8_t precision, TSKEY *minKey, TSKEY *maxKey) {
423,310,651✔
567
  *minKey = tsTickPerMin[precision] * fid * minutes;
423,310,651✔
568
  *maxKey = *minKey + tsTickPerMin[precision] * minutes - 1;
423,323,984✔
569
}
423,343,050✔
570

571
int32_t tsdbFidLevel(int32_t fid, STsdbKeepCfg *pKeepCfg, int64_t nowSec) {
17,486,468✔
572
  int32_t aFid[3];
573
  TSKEY   key;
574

575
  if (pKeepCfg->precision == TSDB_TIME_PRECISION_MILLI) {
17,486,468✔
576
    nowSec = nowSec * 1000;
17,456,471✔
577
  } else if (pKeepCfg->precision == TSDB_TIME_PRECISION_MICRO) {
31,835✔
578
    nowSec = nowSec * 1000000l;
18,902✔
579
  } else if (pKeepCfg->precision == TSDB_TIME_PRECISION_NANO) {
18,375✔
580
    nowSec = nowSec * 1000000000l;
18,375✔
581
  } else {
582
    tsdbError("invalid time precision:%d", pKeepCfg->precision);
×
583
    return 0;
×
584
  }
585

586
  nowSec = nowSec - pKeepCfg->keepTimeOffset * tsTickPerHour[pKeepCfg->precision];
17,493,748✔
587

588
  key = nowSec - pKeepCfg->keep0 * tsTickPerMin[pKeepCfg->precision];
17,493,741✔
589
  aFid[0] = tsdbKeyFid(key, pKeepCfg->days, pKeepCfg->precision);
17,494,615✔
590
  key = nowSec - pKeepCfg->keep1 * tsTickPerMin[pKeepCfg->precision];
17,493,803✔
591
  aFid[1] = tsdbKeyFid(key, pKeepCfg->days, pKeepCfg->precision);
17,487,132✔
592
  key = nowSec - pKeepCfg->keep2 * tsTickPerMin[pKeepCfg->precision];
17,495,658✔
593
  aFid[2] = tsdbKeyFid(key, pKeepCfg->days, pKeepCfg->precision);
17,498,117✔
594

595
  if (fid >= aFid[0]) {
17,495,292✔
596
    return 0;
17,283,164✔
597
  } else if (fid >= aFid[1]) {
212,128✔
598
    return 1;
115,563✔
599
  } else if (fid >= aFid[2]) {
96,565✔
600
    return 2;
94,456✔
601
  } else {
602
    return -1;
2,109✔
603
  }
604
}
605

606
// TSDBROW ======================================================
607
void tsdbRowGetColVal(TSDBROW *pRow, STSchema *pTSchema, int32_t iCol, SColVal *pColVal) {
2,147,483,647✔
608
  int32_t   code = 0;
2,147,483,647✔
609
  STColumn *pTColumn = &pTSchema->columns[iCol];
2,147,483,647✔
610
  SValue    value;
611

612
  if (pRow->type == TSDBROW_ROW_FMT) {
2,147,483,647✔
613
    int32_t ret = tRowGet(pRow->pTSRow, pTSchema, iCol, pColVal);
2,147,483,647✔
614
    if (ret != 0) {
2,147,483,647✔
615
      tsdbError("failed to get column value, code:%d", ret);
×
616
    }
617
  } else if (pRow->type == TSDBROW_COL_FMT) {
2,147,483,647✔
618
    if (iCol == 0) {
2,147,483,647✔
619
      SValue val = {.type = TSDB_DATA_TYPE_TIMESTAMP};
×
620
      VALUE_SET_TRIVIAL_DATUM(&val, pRow->pBlockData->aTSKEY[pRow->iRow]);
×
621
      *pColVal = COL_VAL_VALUE(PRIMARYKEY_TIMESTAMP_COL_ID, val);
×
622
    } else {
623
      SColData *pColData = tBlockDataGetColData(pRow->pBlockData, pTColumn->colId);
2,147,483,647✔
624
      if (pColData) {
2,147,483,647✔
625
        if (tColDataGetValue(pColData, pRow->iRow, pColVal) != 0) {
2,147,483,647✔
626
          tsdbError("failed to tColDataGetValue");
×
627
        }
628
      } else {
629
        *pColVal = COL_VAL_NONE(pTColumn->colId, pTColumn->type);
2,147,483,647✔
630
      }
631
    }
632
  }
633
}
2,147,483,647✔
634

635
void tsdbRowGetKey(TSDBROW *row, STsdbRowKey *key) {
2,147,483,647✔
636
  if (row->type == TSDBROW_ROW_FMT) {
2,147,483,647✔
637
    key->version = row->version;
2,147,483,647✔
638
    tRowGetKey(row->pTSRow, &key->key);
2,147,483,647✔
639
  } else {
640
    key->version = row->pBlockData->aVersion[row->iRow];
2,147,483,647✔
641
    tColRowGetKey(row->pBlockData, row->iRow, &key->key);
2,147,483,647✔
642
  }
643
}
2,147,483,647✔
644

645
void tColRowGetPrimaryKey(SBlockData *pBlock, int32_t irow, SRowKey *key) {
2,147,483,647✔
646
  for (int32_t i = 0; i < pBlock->nColData; i++) {
2,147,483,647✔
647
    SColData *pColData = &pBlock->aColData[i];
2,147,483,647✔
648
    if (pColData->cflag & COL_IS_KEY) {
2,147,483,647✔
649
      SColVal cv;
124,920,042✔
650
      if (tColDataGetValue(pColData, irow, &cv) != 0) {
124,921,323✔
651
        break;
×
652
      }
653
      key->pks[key->numOfPKs] = cv.value;
124,921,550✔
654
      key->numOfPKs++;
124,923,060✔
655
    } else {
656
      break;
2,147,483,647✔
657
    }
658
  }
659
}
2,147,483,647✔
660

661
int32_t tsdbRowKeyCmpr(const STsdbRowKey *key1, const STsdbRowKey *key2) {
2,147,483,647✔
662
  int32_t c = tRowKeyCompare(&key1->key, &key2->key);
2,147,483,647✔
663

664
  if (c) {
2,147,483,647✔
665
    return c;
2,147,483,647✔
666
  }
667

668
  if (key1->version < key2->version) {
2,147,483,647✔
669
    return -1;
2,147,483,647✔
670
  } else if (key1->version > key2->version) {
2,147,483,647✔
671
    return 1;
2,147,483,647✔
672
  }
673
  return 0;
129,721✔
674
}
675

676
int32_t tsdbRowCompare(const void *p1, const void *p2) {
2,147,483,647✔
677
  STsdbRowKey key1, key2;
2,147,483,647✔
678

679
  tsdbRowGetKey((TSDBROW *)p1, &key1);
2,147,483,647✔
680
  tsdbRowGetKey((TSDBROW *)p2, &key2);
2,147,483,647✔
681
  return tsdbRowKeyCmpr(&key1, &key2);
2,147,483,647✔
682
}
683

684
int32_t tsdbRowCompareWithoutVersion(const void *p1, const void *p2) {
2,147,483,647✔
685
  STsdbRowKey key1, key2;
2,147,483,647✔
686

687
  tsdbRowGetKey((TSDBROW *)p1, &key1);
2,147,483,647✔
688
  tsdbRowGetKey((TSDBROW *)p2, &key2);
2,147,483,647✔
689
  return tRowKeyCompare(&key1.key, &key2.key);
2,147,483,647✔
690
}
691

692
// STSDBRowIter ======================================================
693
int32_t tsdbRowIterOpen(STSDBRowIter *pIter, TSDBROW *pRow, STSchema *pTSchema) {
4,868,265✔
694
  pIter->pRow = pRow;
4,868,265✔
695
  if (pRow->type == TSDBROW_ROW_FMT) {
4,868,265✔
696
    int32_t code = tRowIterOpen(pRow->pTSRow, pTSchema, &pIter->pIter);
1,826,485✔
697
    if (code) return code;
1,826,868✔
698
  } else if (pRow->type == TSDBROW_COL_FMT) {
3,041,780✔
699
    pIter->iColData = 0;
3,041,780✔
700
  }
701

702
  return 0;
4,868,648✔
703
}
704

705
void tsdbRowClose(STSDBRowIter *pIter) {
4,868,265✔
706
  if (pIter->pRow && pIter->pRow->type == TSDBROW_ROW_FMT) {
4,868,265✔
707
    tRowIterClose(&pIter->pIter);
1,826,485✔
708
  }
709
  pIter->pRow = NULL;
4,869,031✔
710
  pIter->pIter = NULL;
4,868,648✔
711
}
4,868,648✔
712

713
SColVal *tsdbRowIterNext(STSDBRowIter *pIter) {
21,707,637✔
714
  if (pIter->pRow->type == TSDBROW_ROW_FMT) {
21,707,637✔
715
    return tRowIterNext(pIter->pIter);
9,737,770✔
716
  } else if (pIter->pRow->type == TSDBROW_COL_FMT) {
12,166,817✔
717
    if (pIter->iColData == 0) {
12,166,514✔
718
      SValue val = {.type = TSDB_DATA_TYPE_TIMESTAMP};
3,041,780✔
719
      VALUE_SET_TRIVIAL_DATUM(&val, pIter->pRow->pBlockData->aTSKEY[pIter->pRow->iRow]);
3,041,780✔
720
      pIter->cv = COL_VAL_VALUE(PRIMARYKEY_TIMESTAMP_COL_ID, val);
3,041,780✔
721
      ++pIter->iColData;
3,041,780✔
722
      return &pIter->cv;
3,041,780✔
723
    }
724

725
    if (pIter->iColData <= pIter->pRow->pBlockData->nColData) {
9,125,340✔
726
      if (tColDataGetValue(&pIter->pRow->pBlockData->aColData[pIter->iColData - 1], pIter->pRow->iRow, &pIter->cv) !=
6,083,560✔
727
          0) {
728
        return NULL;
×
729
      }
730
      ++pIter->iColData;
6,083,257✔
731
      return &pIter->cv;
6,083,257✔
732
    } else {
733
      return NULL;
3,041,780✔
734
    }
735
  } else {
736
    tsdbError("invalid row type:%d", pIter->pRow->type);
×
737
    return NULL;
×
738
  }
739
}
740

741
// SRowMerger ======================================================
742
int32_t tsdbRowMergerAdd(SRowMerger *pMerger, TSDBROW *pRow, STSchema *pTSchema) {
2,147,483,647✔
743
  int32_t   code = 0;
2,147,483,647✔
744
  TSDBKEY   key = TSDBROW_KEY(pRow);
2,147,483,647✔
745
  SColVal  *pColVal = &(SColVal){0};
2,147,483,647✔
746
  STColumn *pTColumn;
747
  int32_t   iCol, jCol = 1;
2,147,483,647✔
748
  pRow->arg = pMerger->arg;
2,147,483,647✔
749

750
  if (NULL == pTSchema) {
2,147,483,647✔
751
    pTSchema = pMerger->pTSchema;
2,147,483,647✔
752
  }
753

754
  if (taosArrayGetSize(pMerger->pArray) == 0) {
2,147,483,647✔
755
    // ts
756
    jCol = 0;
2,147,483,647✔
757
    pTColumn = &pTSchema->columns[jCol++];
2,147,483,647✔
758
    SValue val = {.type = pTColumn->type};
2,147,483,647✔
759
    VALUE_SET_TRIVIAL_DATUM(&val, key.ts);
2,147,483,647✔
760
    *pColVal = COL_VAL_VALUE(pTColumn->colId, val);
2,147,483,647✔
761
    if (taosArrayPush(pMerger->pArray, pColVal) == NULL) {
2,147,483,647✔
762
      code = terrno;
×
763
      return code;
×
764
    }
765

766
    // other
767
    for (iCol = 1; jCol < pTSchema->numOfCols && iCol < pMerger->pTSchema->numOfCols; ++iCol) {
2,147,483,647✔
768
      pTColumn = &pMerger->pTSchema->columns[iCol];
2,147,483,647✔
769
      if (pTSchema->columns[jCol].colId < pTColumn->colId) {
2,147,483,647✔
770
        ++jCol;
×
771
        --iCol;
×
772
        continue;
×
773
      } else if (pTSchema->columns[jCol].colId > pTColumn->colId) {
2,147,483,647✔
774
        if (taosArrayPush(pMerger->pArray, &COL_VAL_NONE(pTColumn->colId, pTColumn->type)) == NULL) {
×
775
          return terrno;
×
776
        }
777
        continue;
×
778
      }
779

780
      tsdbRowGetColVal(pRow, pTSchema, jCol++, pColVal);
2,147,483,647✔
781
      bool usepData = IS_VAR_DATA_TYPE(pColVal->value.type) || pColVal->value.type == TSDB_DATA_TYPE_DECIMAL;
2,147,483,647✔
782
      if ((!COL_VAL_IS_NONE(pColVal)) && (!COL_VAL_IS_NULL(pColVal)) && usepData) {
2,147,483,647✔
783
        uint8_t *pVal = pColVal->value.pData;
2,147,483,647✔
784

785
        pColVal->value.pData = NULL;
2,147,483,647✔
786
        code = tRealloc(&pColVal->value.pData, pColVal->value.nData);
2,147,483,647✔
787
        if (code) {
2,147,483,647✔
788
          return code;
×
789
        }
790

791
        if (pColVal->value.nData) {
2,147,483,647✔
792
          memcpy(pColVal->value.pData, pVal, pColVal->value.nData);
2,147,483,647✔
793
        }
794
      }
795

796
      if (taosArrayPush(pMerger->pArray, pColVal) == NULL) {
2,147,483,647✔
797
        return terrno;
×
798
      }
799
    }
800

801
    for (; iCol < pMerger->pTSchema->numOfCols; ++iCol) {
2,147,483,647✔
802
      pTColumn = &pMerger->pTSchema->columns[iCol];
4,803✔
803
      if (taosArrayPush(pMerger->pArray, &COL_VAL_NONE(pTColumn->colId, pTColumn->type)) == NULL) {
9,606✔
804
        return terrno;
×
805
      }
806
    }
807

808
    pMerger->version = key.version;
2,147,483,647✔
809
    return 0;
2,147,483,647✔
810
  } else {
811
    for (iCol = 1; iCol < pMerger->pTSchema->numOfCols && jCol < pTSchema->numOfCols; ++iCol) {
2,147,483,647✔
812
      pTColumn = &pMerger->pTSchema->columns[iCol];
2,147,483,647✔
813
      if (pTSchema->columns[jCol].colId < pTColumn->colId) {
2,147,483,647✔
814
        ++jCol;
×
815
        --iCol;
×
816
        continue;
×
817
      } else if (pTSchema->columns[jCol].colId > pTColumn->colId) {
2,147,483,647✔
818
        continue;
×
819
      }
820

821
      tsdbRowGetColVal(pRow, pTSchema, jCol++, pColVal);
2,147,483,647✔
822

823
      if (key.version > pMerger->version) {
2,147,483,647✔
824
        if (!COL_VAL_IS_NONE(pColVal)) {
2,147,483,647✔
825
          if (IS_VAR_DATA_TYPE(pColVal->value.type) || pColVal->value.type == TSDB_DATA_TYPE_DECIMAL) {
2,147,483,647✔
826
            SColVal *pTColVal = taosArrayGet(pMerger->pArray, iCol);
2,147,483,647✔
827
            if (!pTColVal) return terrno;
2,147,483,647✔
828
            if (!COL_VAL_IS_NULL(pColVal)) {
2,147,483,647✔
829
              code = tRealloc(&pTColVal->value.pData, pColVal->value.nData);
2,147,483,647✔
830
              if (code) return code;
2,147,483,647✔
831

832
              pTColVal->value.nData = pColVal->value.nData;
2,147,483,647✔
833
              if (pTColVal->value.nData) {
2,147,483,647✔
834
                memcpy(pTColVal->value.pData, pColVal->value.pData, pTColVal->value.nData);
2,147,483,647✔
835
              }
836
              pTColVal->flag = 0;
2,147,483,647✔
837
            } else {
838
              tFree(pTColVal->value.pData);
4,863,746✔
839
              taosArraySet(pMerger->pArray, iCol, pColVal);
4,864,286✔
840
            }
841
          } else {
842
            taosArraySet(pMerger->pArray, iCol, pColVal);
2,147,483,647✔
843
          }
844
        }
845
      } else if (key.version < pMerger->version) {
1,461,005✔
846
        SColVal *tColVal = (SColVal *)taosArrayGet(pMerger->pArray, iCol);
1,461,005✔
847
        if (COL_VAL_IS_NONE(tColVal) && !COL_VAL_IS_NONE(pColVal)) {
1,461,005✔
848
          bool usepData = IS_VAR_DATA_TYPE(pColVal->value.type) || pColVal->value.type == TSDB_DATA_TYPE_DECIMAL;
155,232✔
849
          if ((!COL_VAL_IS_NULL(pColVal)) && usepData) {
155,232✔
850
            code = tRealloc(&tColVal->value.pData, pColVal->value.nData);
51,744✔
851
            if (code) return code;
51,744✔
852

853
            tColVal->value.nData = pColVal->value.nData;
51,744✔
854
            if (pColVal->value.nData) {
51,744✔
855
              memcpy(tColVal->value.pData, pColVal->value.pData, pColVal->value.nData);
51,744✔
856
            }
857
            tColVal->flag = 0;
51,744✔
858
          } else {
859
            taosArraySet(pMerger->pArray, iCol, pColVal);
103,488✔
860
          }
861
        }
862
      } else {
863
        return TSDB_CODE_SUCCESS;  // same version, no need to merge
×
864
      }
865
    }
866

867
    pMerger->version = key.version;
2,147,483,647✔
868
    return code;
2,147,483,647✔
869
  }
870
}
871

872
int32_t tsdbRowMergerInit(SRowMerger *pMerger, STSchema *pSchema) {
259,362,916✔
873
  pMerger->pTSchema = pSchema;
259,362,916✔
874
  pMerger->pArray = taosArrayInit(pSchema->numOfCols, sizeof(SColVal));
259,523,598✔
875
  if (pMerger->pArray == NULL) {
258,935,130✔
876
    return terrno;
×
877
  } else {
878
    return TSDB_CODE_SUCCESS;
259,134,653✔
879
  }
880
}
881

882
void tsdbRowMergerClear(SRowMerger *pMerger) {
2,147,483,647✔
883
  for (int32_t iCol = 1; iCol < pMerger->pTSchema->numOfCols; iCol++) {
2,147,483,647✔
884
    SColVal *pTColVal = taosArrayGet(pMerger->pArray, iCol);
2,147,483,647✔
885
    if (IS_VAR_DATA_TYPE(pTColVal->value.type) || pTColVal->value.type == TSDB_DATA_TYPE_DECIMAL) {
2,147,483,647✔
886
      tFree(pTColVal->value.pData);
2,147,483,647✔
887
    }
888
  }
889

890
  taosArrayClear(pMerger->pArray);
2,147,483,647✔
891
}
2,147,483,647✔
892

893
void tsdbRowMergerCleanup(SRowMerger *pMerger) {
261,528,776✔
894
  int32_t numOfCols = taosArrayGetSize(pMerger->pArray);
261,528,776✔
895
  for (int32_t iCol = 1; iCol < numOfCols; iCol++) {
261,562,800✔
896
    SColVal *pTColVal = taosArrayGet(pMerger->pArray, iCol);
×
897
    if (IS_VAR_DATA_TYPE(pTColVal->value.type) || pTColVal->value.type == TSDB_DATA_TYPE_DECIMAL) {
×
898
      tFree(pTColVal->value.pData);
×
899
    }
900
  }
901

902
  taosArrayDestroy(pMerger->pArray);
261,562,800✔
903
}
261,279,443✔
904

905
int32_t tsdbRowMergerGetRow(SRowMerger *pMerger, SRow **ppRow) {
2,147,483,647✔
906
  SRowBuildScanInfo scanInfo = {.hasBlob = 0};
2,147,483,647✔
907
  return tRowBuild(pMerger->pArray, pMerger->pTSchema, ppRow, &scanInfo);
2,147,483,647✔
908
}
909

910
// delete skyline ======================================================
911
static void tsdbMergeSkyline(SArray *pSkyline1, SArray *pSkyline2, SArray *pSkyline) {
50,742,399✔
912
  int32_t  i1 = 0;
50,742,399✔
913
  int32_t  n1 = taosArrayGetSize(pSkyline1);
50,742,399✔
914
  int32_t  i2 = 0;
50,742,399✔
915
  int32_t  n2 = taosArrayGetSize(pSkyline2);
50,742,399✔
916
  TSDBKEY *pKey1;
917
  TSDBKEY *pKey2;
918
  int64_t  version1 = 0;
50,742,399✔
919
  int64_t  version2 = 0;
50,742,399✔
920

921
  taosArrayClear(pSkyline);
50,742,399✔
922
  TSDBKEY **pItem = TARRAY_GET_ELEM(pSkyline, 0);
50,742,399✔
923

924
  while (i1 < n1 && i2 < n2) {
219,280,965✔
925
    pKey1 = (TSDBKEY *)taosArrayGetP(pSkyline1, i1);
168,538,773✔
926
    pKey2 = (TSDBKEY *)taosArrayGetP(pSkyline2, i2);
168,538,359✔
927

928
    if (pKey1->ts < pKey2->ts) {
168,538,152✔
929
      version1 = pKey1->version;
52,899,688✔
930
      *pItem = pKey1;
52,899,688✔
931
      i1++;
52,899,688✔
932
    } else if (pKey1->ts > pKey2->ts) {
115,638,464✔
933
      version2 = pKey2->version;
32,564,878✔
934
      *pItem = pKey2;
32,564,878✔
935
      i2++;
32,564,878✔
936
    } else {
937
      version1 = pKey1->version;
83,073,586✔
938
      version2 = pKey2->version;
83,073,586✔
939
      *pItem = pKey1;
83,073,586✔
940
      i1++;
83,073,586✔
941
      i2++;
83,073,586✔
942
    }
943

944
    (*pItem)->version = TMAX(version1, version2);
168,538,152✔
945
    pItem++;
168,538,152✔
946
  }
947

948
  while (i1 < n1) {
70,867,537✔
949
    pKey1 = (TSDBKEY *)taosArrayGetP(pSkyline1, i1);
20,125,345✔
950
    *pItem = pKey1;
20,125,345✔
951
    pItem++;
20,125,345✔
952
    i1++;
20,125,345✔
953
  }
954

955
  while (i2 < n2) {
73,868,406✔
956
    pKey2 = (TSDBKEY *)taosArrayGetP(pSkyline2, i2);
23,126,007✔
957
    *pItem = pKey2;
23,126,007✔
958
    pItem++;
23,126,007✔
959
    i2++;
23,126,214✔
960
  }
961

962
  pSkyline->size = TARRAY_ELEM_IDX(pSkyline, pItem);
50,742,399✔
963
}
50,742,399✔
964

965
int32_t tsdbBuildDeleteSkylineImpl(SArray *aSkyline, int32_t sidx, int32_t eidx, SArray *pSkyline) {
107,626,751✔
966
  int32_t   code = 0;
107,626,751✔
967
  SDelData *pDelData;
968
  int32_t   midx;
969

970
  taosArrayClear(pSkyline);
107,626,751✔
971
  if (sidx == eidx) {
107,626,751✔
972
    TSDBKEY *pItem1 = taosArrayGet(aSkyline, sidx * 2);
56,884,352✔
973
    TSDBKEY *pItem2 = taosArrayGet(aSkyline, sidx * 2 + 1);
56,884,352✔
974
    if (taosArrayPush(pSkyline, &pItem1) == NULL) {
56,884,352✔
975
      return terrno;
×
976
    }
977

978
    if (taosArrayPush(pSkyline, &pItem2) == NULL) {
56,884,571✔
979
      return terrno;
×
980
    }
981
  } else {
982
    SArray *pSkyline1 = NULL;
50,742,399✔
983
    SArray *pSkyline2 = NULL;
50,742,399✔
984
    midx = (sidx + eidx) / 2;
50,742,399✔
985

986
    pSkyline1 = taosArrayInit((midx - sidx + 1) * 2, POINTER_BYTES);
50,742,399✔
987
    if (pSkyline1 == NULL) {
50,742,399✔
988
      return terrno;
×
989
    }
990
    pSkyline2 = taosArrayInit((eidx - midx) * 2, POINTER_BYTES);
50,742,399✔
991
    if (pSkyline2 == NULL) {
50,742,399✔
992
      taosArrayDestroy(pSkyline1);
×
993
      return terrno;
×
994
    }
995

996
    code = tsdbBuildDeleteSkylineImpl(aSkyline, sidx, midx, pSkyline1);
50,742,399✔
997
    if (code) goto _clear;
50,742,399✔
998

999
    code = tsdbBuildDeleteSkylineImpl(aSkyline, midx + 1, eidx, pSkyline2);
50,742,399✔
1000
    if (code) goto _clear;
50,742,399✔
1001

1002
    tsdbMergeSkyline(pSkyline1, pSkyline2, pSkyline);
50,742,399✔
1003

1004
  _clear:
50,742,399✔
1005
    taosArrayDestroy(pSkyline1);
50,742,399✔
1006
    taosArrayDestroy(pSkyline2);
50,742,192✔
1007
  }
1008

1009
  return code;
107,626,323✔
1010
}
1011

1012
int32_t tsdbBuildDeleteSkyline(SArray *aDelData, int32_t sidx, int32_t eidx, SArray *aSkyline) {
6,142,172✔
1013
  SDelData *pDelData;
1014
  int32_t   code = 0;
6,142,172✔
1015
  int32_t   dataNum = eidx - sidx + 1;
6,142,172✔
1016
  SArray   *aTmpSkyline = taosArrayInit(dataNum * 2, sizeof(TSDBKEY));
6,142,172✔
1017
  if (aTmpSkyline == NULL) {
6,142,172✔
1018
    return terrno;
×
1019
  }
1020

1021
  SArray *pSkyline = taosArrayInit(dataNum * 2, POINTER_BYTES);
6,142,172✔
1022
  if (pSkyline == NULL) {
6,141,953✔
1023
    taosArrayDestroy(aTmpSkyline);
×
1024
    return terrno;
×
1025
  }
1026

1027
  taosArrayClear(aSkyline);
6,141,953✔
1028
  for (int32_t i = sidx; i <= eidx; ++i) {
63,026,305✔
1029
    pDelData = (SDelData *)taosArrayGet(aDelData, i);
56,884,352✔
1030
    if (taosArrayPush(aTmpSkyline, &(TSDBKEY){.ts = pDelData->sKey, .version = pDelData->version}) == NULL) {
113,768,497✔
1031
      code = terrno;
×
1032
      goto _clear;
×
1033
    }
1034

1035
    if (taosArrayPush(aTmpSkyline, &(TSDBKEY){.ts = pDelData->eKey, .version = 0}) == NULL) {
113,768,704✔
1036
      code = terrno;
×
1037
      goto _clear;
×
1038
    }
1039
  }
1040

1041
  code = tsdbBuildDeleteSkylineImpl(aTmpSkyline, sidx, eidx, pSkyline);
6,141,953✔
1042
  if (code) goto _clear;
6,142,172✔
1043

1044
  int32_t skylineNum = taosArrayGetSize(pSkyline);
6,142,172✔
1045
  for (int32_t i = 0; i < skylineNum; ++i) {
36,837,314✔
1046
    TSDBKEY *p = taosArrayGetP(pSkyline, i);
30,694,651✔
1047
    if (taosArrayPush(aSkyline, p) == NULL) {
30,695,142✔
1048
      code = terrno;
×
1049
      goto _clear;
×
1050
    }
1051
  }
1052

1053
_clear:
6,142,663✔
1054
  taosArrayDestroy(aTmpSkyline);
6,142,456✔
1055
  taosArrayDestroy(pSkyline);
6,142,172✔
1056

1057
  return code;
6,142,172✔
1058
}
1059

1060
/*
1061
int32_t tsdbBuildDeleteSkyline2(SArray *aDelData, int32_t sidx, int32_t eidx, SArray *aSkyline) {
1062
  int32_t   code = 0;
1063
  SDelData *pDelData;
1064
  int32_t   midx;
1065

1066
  taosArrayClear(aSkyline);
1067
  if (sidx == eidx) {
1068
    pDelData = (SDelData *)taosArrayGet(aDelData, sidx);
1069
    taosArrayPush(aSkyline, &(TSDBKEY){.ts = pDelData->sKey, .version = pDelData->version});
1070
    taosArrayPush(aSkyline, &(TSDBKEY){.ts = pDelData->eKey, .version = 0});
1071
  } else {
1072
    SArray *aSkyline1 = NULL;
1073
    SArray *aSkyline2 = NULL;
1074

1075
    aSkyline1 = taosArrayInit(0, sizeof(TSDBKEY));
1076
    aSkyline2 = taosArrayInit(0, sizeof(TSDBKEY));
1077
    if (aSkyline1 == NULL || aSkyline2 == NULL) {
1078
      code = TSDB_CODE_OUT_OF_MEMORY;
1079
      goto _clear;
1080
    }
1081
    midx = (sidx + eidx) / 2;
1082

1083
    code = tsdbBuildDeleteSkyline(aDelData, sidx, midx, aSkyline1);
1084
    if (code) goto _clear;
1085

1086
    code = tsdbBuildDeleteSkyline(aDelData, midx + 1, eidx, aSkyline2);
1087
    if (code) goto _clear;
1088

1089
    code = tsdbMergeSkyline(aSkyline1, aSkyline2, aSkyline);
1090

1091
  _clear:
1092
    taosArrayDestroy(aSkyline1);
1093
    taosArrayDestroy(aSkyline2);
1094
  }
1095

1096
  return code;
1097
}
1098
*/
1099

1100
// SBlockData ======================================================
1101
int32_t tBlockDataCreate(SBlockData *pBlockData) {
855,532,809✔
1102
  pBlockData->suid = 0;
855,532,809✔
1103
  pBlockData->uid = 0;
855,797,704✔
1104
  pBlockData->nRow = 0;
855,512,616✔
1105
  pBlockData->aUid = NULL;
855,446,161✔
1106
  pBlockData->aVersion = NULL;
855,237,181✔
1107
  pBlockData->aTSKEY = NULL;
855,315,721✔
1108
  pBlockData->nColData = 0;
855,077,602✔
1109
  pBlockData->aColData = NULL;
854,958,221✔
1110
  return 0;
855,309,073✔
1111
}
1112

1113
void tBlockDataDestroy(SBlockData *pBlockData) {
908,720,592✔
1114
  tFree(pBlockData->aUid);
908,720,592✔
1115
  tFree(pBlockData->aVersion);
908,979,762✔
1116
  tFree(pBlockData->aTSKEY);
908,919,078✔
1117

1118
  for (int32_t i = 0; i < pBlockData->nColData; i++) {
1,440,620,890✔
1119
    tColDataDestroy(&pBlockData->aColData[i]);
531,814,046✔
1120
  }
1121

1122
  if (pBlockData->aColData) {
908,596,969✔
1123
    taosMemoryFree(pBlockData->aColData);
137,145,487✔
1124
    pBlockData->aColData = NULL;
137,111,082✔
1125
  }
1126
}
908,892,152✔
1127

1128
static int32_t tBlockDataAdjustColData(SBlockData *pBlockData, int32_t nColData) {
98,395,614✔
1129
  int32_t code = 0;
98,395,614✔
1130

1131
  if (pBlockData->nColData > nColData) {
98,395,614✔
1132
    for (int32_t i = nColData; i < pBlockData->nColData; i++) {
7,352,011✔
1133
      tColDataDestroy(&pBlockData->aColData[i]);
7,165,510✔
1134
    }
1135
  } else if (pBlockData->nColData < nColData) {
98,243,702✔
1136
    SColData *aColData = taosMemoryRealloc(pBlockData->aColData, sizeof(SBlockData) * nColData);
72,019,050✔
1137
    if (aColData == NULL) {
71,969,274✔
1138
      code = terrno;
×
1139
      goto _exit;
×
1140
    }
1141

1142
    pBlockData->aColData = aColData;
71,969,274✔
1143
    memset(&pBlockData->aColData[pBlockData->nColData], 0, sizeof(SBlockData) * (nColData - pBlockData->nColData));
71,970,527✔
1144
  }
1145
  pBlockData->nColData = nColData;
98,385,469✔
1146

1147
_exit:
98,459,517✔
1148
  return code;
98,459,517✔
1149
}
1150
int32_t tBlockDataInit(SBlockData *pBlockData, TABLEID *pId, STSchema *pTSchema, int16_t *aCid, int32_t nCid) {
98,398,215✔
1151
  int32_t code = 0;
98,398,215✔
1152

1153
  if (!pId->suid && !pId->uid) {
98,398,215✔
1154
    return TSDB_CODE_INVALID_PARA;
×
1155
  }
1156

1157
  pBlockData->suid = pId->suid;
98,440,322✔
1158
  pBlockData->uid = pId->uid;
98,448,276✔
1159
  pBlockData->nRow = 0;
98,448,087✔
1160

1161
  if (aCid) {
98,415,859✔
1162
    code = tBlockDataAdjustColData(pBlockData, nCid);
×
1163
    if (code) goto _exit;
×
1164

1165
    int32_t   iColumn = 1;
×
1166
    STColumn *pTColumn = &pTSchema->columns[iColumn];
×
1167
    for (int32_t iCid = 0; iCid < nCid; iCid++) {
×
1168
      // aCid array (from taos client catalog) contains columns that does not exist in the pTSchema. the pTSchema is
1169
      // newer
1170
      if (pTColumn == NULL) {
×
1171
        continue;
×
1172
      }
1173

1174
      while (pTColumn->colId < aCid[iCid]) {
×
1175
        iColumn++;
×
1176
        if (!(iColumn < pTSchema->numOfCols)) {
×
1177
          return TSDB_CODE_INVALID_PARA;
×
1178
        }
1179
        pTColumn = &pTSchema->columns[iColumn];
×
1180
      }
1181

1182
      if (pTColumn->colId != aCid[iCid]) {
×
1183
        continue;
×
1184
      }
1185

1186
      tColDataInit(&pBlockData->aColData[iCid], pTColumn->colId, pTColumn->type, pTColumn->flags);
×
1187

1188
      iColumn++;
×
1189
      pTColumn = (iColumn < pTSchema->numOfCols) ? &pTSchema->columns[iColumn] : NULL;
×
1190
    }
1191
  } else {
1192
    code = tBlockDataAdjustColData(pBlockData, pTSchema->numOfCols - 1);
98,415,859✔
1193
    if (code) goto _exit;
98,446,479✔
1194

1195
    for (int32_t iColData = 0; iColData < pBlockData->nColData; iColData++) {
691,292,330✔
1196
      STColumn *pTColumn = &pTSchema->columns[iColData + 1];
592,833,347✔
1197
      tColDataInit(&pBlockData->aColData[iColData], pTColumn->colId, pTColumn->type, pTColumn->flags);
592,818,837✔
1198
    }
1199
  }
1200

1201
_exit:
98,469,559✔
1202
  return code;
98,469,559✔
1203
}
1204

1205
void tBlockDataReset(SBlockData *pBlockData) {
604,547,494✔
1206
  pBlockData->suid = 0;
604,547,494✔
1207
  pBlockData->uid = 0;
604,697,957✔
1208
  pBlockData->nRow = 0;
604,802,858✔
1209
  for (int32_t i = 0; i < pBlockData->nColData; i++) {
1,116,782,655✔
1210
    tColDataDestroy(&pBlockData->aColData[i]);
511,949,792✔
1211
  }
1212
  pBlockData->nColData = 0;
604,577,973✔
1213
  taosMemoryFreeClear(pBlockData->aColData);
604,804,130✔
1214
}
604,793,693✔
1215

1216
void tBlockDataClear(SBlockData *pBlockData) {
99,236,821✔
1217
  pBlockData->nRow = 0;
99,236,821✔
1218
  for (int32_t iColData = 0; iColData < pBlockData->nColData; iColData++) {
597,758,556✔
1219
    tColDataClear(tBlockDataGetColDataByIdx(pBlockData, iColData));
498,514,188✔
1220
  }
1221
}
99,277,599✔
1222

1223
int32_t tBlockDataAddColData(SBlockData *pBlockData, int16_t cid, int8_t type, int8_t cflag, SColData **ppColData) {
566,617,396✔
1224
  if (pBlockData->nColData != 0 && pBlockData->aColData[pBlockData->nColData - 1].cid >= cid) {
566,617,396✔
1225
    return TSDB_CODE_INVALID_PARA;
×
1226
  }
1227

1228
  SColData *newColData = taosMemoryRealloc(pBlockData->aColData, sizeof(SColData) * (pBlockData->nColData + 1));
566,709,120✔
1229
  if (newColData == NULL) {
566,597,396✔
1230
    return terrno;
×
1231
  }
1232

1233
  pBlockData->aColData = newColData;
566,597,396✔
1234
  pBlockData->nColData++;
566,641,213✔
1235

1236
  *ppColData = &pBlockData->aColData[pBlockData->nColData - 1];
566,672,010✔
1237
  memset(*ppColData, 0, sizeof(SColData));
566,602,391✔
1238
  tColDataInit(*ppColData, cid, type, cflag);
566,663,816✔
1239

1240
  return 0;
566,618,892✔
1241
}
1242

1243
/* flag > 0: forward update
1244
 * flag == 0: insert
1245
 * flag < 0: backward update
1246
 */
1247
static int32_t tBlockDataUpsertBlockRow(SBlockData *pBlockData, SBlockData *pBlockDataFrom, int32_t iRow,
2,147,483,647✔
1248
                                        int32_t flag) {
1249
  int32_t code = 0;
2,147,483,647✔
1250

1251
  SColVal   cv = {0};
2,147,483,647✔
1252
  int32_t   iColDataFrom = 0;
2,147,483,647✔
1253
  SColData *pColDataFrom = (iColDataFrom < pBlockDataFrom->nColData) ? &pBlockDataFrom->aColData[iColDataFrom] : NULL;
2,147,483,647✔
1254

1255
  for (int32_t iColDataTo = 0; iColDataTo < pBlockData->nColData; iColDataTo++) {
2,147,483,647✔
1256
    SColData *pColDataTo = &pBlockData->aColData[iColDataTo];
2,147,483,647✔
1257

1258
    while (pColDataFrom && pColDataFrom->cid < pColDataTo->cid) {
2,147,483,647✔
1259
      pColDataFrom = (++iColDataFrom < pBlockDataFrom->nColData) ? &pBlockDataFrom->aColData[iColDataFrom] : NULL;
13,583,958✔
1260
    }
1261

1262
    if (pColDataFrom == NULL || pColDataFrom->cid > pColDataTo->cid) {
2,147,483,647✔
1263
      cv = COL_VAL_NONE(pColDataTo->cid, pColDataTo->type);
2,147,483,647✔
1264
      if (flag == 0 && (code = tColDataAppendValue(pColDataTo, &cv))) goto _exit;
2,147,483,647✔
1265
    } else {
1266
      code = tColDataGetValue(pColDataFrom, iRow, &cv);
2,147,483,647✔
1267
      if (code) goto _exit;
2,147,483,647✔
1268

1269
      if (flag) {
2,147,483,647✔
1270
        code = tColDataUpdateValue(pColDataTo, &cv, flag > 0);
2,147,483,647✔
1271
      } else {
1272
        code = tColDataAppendValue(pColDataTo, &cv);
2,147,483,647✔
1273
      }
1274
      if (code) goto _exit;
2,147,483,647✔
1275

1276
      pColDataFrom = (++iColDataFrom < pBlockDataFrom->nColData) ? &pBlockDataFrom->aColData[iColDataFrom] : NULL;
2,147,483,647✔
1277
    }
1278
  }
1279

1280
_exit:
2,147,483,647✔
1281
  return code;
2,147,483,647✔
1282
}
1283

1284
int32_t tBlockDataAppendRow(SBlockData *pBlockData, TSDBROW *pRow, STSchema *pTSchema, int64_t uid) {
2,147,483,647✔
1285
  int32_t code = 0;
2,147,483,647✔
1286

1287
  if (!(pBlockData->suid || pBlockData->uid)) {
2,147,483,647✔
1288
    return TSDB_CODE_INVALID_PARA;
×
1289
  }
1290

1291
  // uid
1292
  if (pBlockData->uid == 0) {
2,147,483,647✔
1293
    if (!uid) {
2,147,483,647✔
1294
      return TSDB_CODE_INVALID_PARA;
×
1295
    }
1296
    code = tRealloc((uint8_t **)&pBlockData->aUid, sizeof(int64_t) * (pBlockData->nRow + 1));
2,147,483,647✔
1297
    if (code) goto _exit;
2,147,483,647✔
1298
    pBlockData->aUid[pBlockData->nRow] = uid;
2,147,483,647✔
1299
  }
1300
  // version
1301
  code = tRealloc((uint8_t **)&pBlockData->aVersion, sizeof(int64_t) * (pBlockData->nRow + 1));
2,147,483,647✔
1302
  if (code) goto _exit;
2,147,483,647✔
1303
  pBlockData->aVersion[pBlockData->nRow] = TSDBROW_VERSION(pRow);
2,147,483,647✔
1304
  // timestamp
1305
  code = tRealloc((uint8_t **)&pBlockData->aTSKEY, sizeof(TSKEY) * (pBlockData->nRow + 1));
2,147,483,647✔
1306
  if (code) goto _exit;
2,147,483,647✔
1307
  pBlockData->aTSKEY[pBlockData->nRow] = TSDBROW_TS(pRow);
2,147,483,647✔
1308

1309
  if (pRow->type == TSDBROW_ROW_FMT) {
2,147,483,647✔
1310
    code = tRowUpsertColData(pRow->pTSRow, pTSchema, pBlockData->aColData, pBlockData->nColData, 0 /* append */);
2,147,483,647✔
1311
    if (code) goto _exit;
2,147,483,647✔
1312
  } else if (pRow->type == TSDBROW_COL_FMT) {
2,147,483,647✔
1313
    code = tBlockDataUpsertBlockRow(pBlockData, pRow->pBlockData, pRow->iRow, 0 /* append */);
2,147,483,647✔
1314
    if (code) goto _exit;
2,147,483,647✔
1315
  } else {
1316
    return TSDB_CODE_INVALID_PARA;
1,833✔
1317
  }
1318
  pBlockData->nRow++;
2,147,483,647✔
1319

1320
_exit:
2,147,483,647✔
1321
  return code;
2,147,483,647✔
1322
}
1323
int32_t tBlockDataUpdateRow(SBlockData *pBlockData, TSDBROW *pRow, STSchema *pTSchema) {
2,147,483,647✔
1324
  int32_t code = 0;
2,147,483,647✔
1325

1326
  // version
1327
  int64_t lversion = pBlockData->aVersion[pBlockData->nRow - 1];
2,147,483,647✔
1328
  int64_t rversion = TSDBROW_VERSION(pRow);
2,147,483,647✔
1329
  if (lversion == rversion) {
2,147,483,647✔
1330
    return TSDB_CODE_SUCCESS;
×
1331
  }
1332
  if (rversion > lversion) {
2,147,483,647✔
1333
    pBlockData->aVersion[pBlockData->nRow - 1] = rversion;
2,147,483,647✔
1334
  }
1335

1336
  // update other rows
1337
  if (pRow->type == TSDBROW_ROW_FMT) {
2,147,483,647✔
1338
    code = tRowUpsertColData(pRow->pTSRow, pTSchema, pBlockData->aColData, pBlockData->nColData,
64,636,446✔
1339
                             (rversion > lversion) ? 1 : -1 /* update */);
1340
    if (code) goto _exit;
64,656,583✔
1341
  } else if (pRow->type == TSDBROW_COL_FMT) {
2,147,483,647✔
1342
    code = tBlockDataUpsertBlockRow(pBlockData, pRow->pBlockData, pRow->iRow, (rversion > lversion) ? 1 : -1);
2,147,483,647✔
1343
    if (code) goto _exit;
2,147,483,647✔
1344
  } else {
1345
    code = TSDB_CODE_INVALID_PARA;
×
1346
    goto _exit;
×
1347
  }
1348

1349
_exit:
2,147,483,647✔
1350
  return code;
2,147,483,647✔
1351
}
1352

1353
#ifdef BUILD_NO_CALL
1354
int32_t tBlockDataTryUpsertRow(SBlockData *pBlockData, TSDBROW *pRow, int64_t uid) {
1355
  if (pBlockData->nRow == 0) {
1356
    return 1;
1357
  } else if (pBlockData->aTSKEY[pBlockData->nRow - 1] == TSDBROW_TS(pRow)) {
1358
    return pBlockData->nRow;
1359
  } else {
1360
    return pBlockData->nRow + 1;
1361
  }
1362
}
1363

1364
int32_t tBlockDataUpsertRow(SBlockData *pBlockData, TSDBROW *pRow, STSchema *pTSchema, int64_t uid) {
1365
  if (pBlockData->nRow > 0 && pBlockData->aTSKEY[pBlockData->nRow - 1] == TSDBROW_TS(pRow)) {
1366
    return tBlockDataUpdateRow(pBlockData, pRow, pTSchema);
1367
  } else {
1368
    return tBlockDataAppendRow(pBlockData, pRow, pTSchema, uid);
1369
  }
1370
}
1371
#endif
1372

1373
SColData *tBlockDataGetColData(SBlockData *pBlockData, int16_t cid) {
2,147,483,647✔
1374
  int32_t lidx = 0;
2,147,483,647✔
1375
  int32_t ridx = pBlockData->nColData - 1;
2,147,483,647✔
1376

1377
  while (lidx <= ridx) {
2,147,483,647✔
1378
    int32_t   midx = (lidx + ridx) >> 1;
2,147,483,647✔
1379
    SColData *pColData = tBlockDataGetColDataByIdx(pBlockData, midx);
2,147,483,647✔
1380
    int32_t   c = (pColData->cid == cid) ? 0 : ((pColData->cid > cid) ? 1 : -1);
2,147,483,647✔
1381

1382
    if (c == 0) {
2,147,483,647✔
1383
      return pColData;
2,147,483,647✔
1384
    } else if (c < 0) {
2,147,483,647✔
1385
      lidx = midx + 1;
2,147,483,647✔
1386
    } else {
1387
      ridx = midx - 1;
2,147,483,647✔
1388
    }
1389
  }
1390

1391
  return NULL;
2,147,483,647✔
1392
}
1393

1394
/* buffers[0]: SDiskDataHdr
1395
 * buffers[1]: key part: uid + version + ts + primary keys
1396
 * buffers[2]: SBlockCol part
1397
 * buffers[3]: regular column part
1398
 */
1399
int32_t tBlockDataCompress(SBlockData *bData, void *pCompr, SBuffer *buffers, SBuffer *assist) {
72,018,032✔
1400
  int32_t code = 0;
72,018,032✔
1401
  int32_t lino = 0;
72,018,032✔
1402

1403
  SColCompressInfo *pInfo = pCompr;
72,018,032✔
1404
  code = tsdbGetColCmprAlgFromSet(pInfo->pColCmpr, 1, &pInfo->defaultCmprAlg);
72,018,032✔
1405
  TAOS_UNUSED(code);
1406

1407
  SDiskDataHdr hdr = {
72,070,989✔
1408
      .delimiter = TSDB_FILE_DLMT,
1409
      .fmtVer = 2,
1410
      .suid = bData->suid,
72,073,930✔
1411
      .uid = bData->uid,
72,072,085✔
1412
      .szUid = 0,     // filled by compress key
1413
      .szVer = 0,     // filled by compress key
1414
      .szKey = 0,     // filled by compress key
1415
      .szBlkCol = 0,  // filled by this func
1416
      .nRow = bData->nRow,
72,069,775✔
1417
      .cmprAlg = pInfo->defaultCmprAlg,
72,068,720✔
1418
      .numOfPKs = 0,  // filled by compress key
1419
  };
1420
  // Key part
1421

1422
  tBufferClear(&buffers[1]);
71,996,555✔
1423
  code = tBlockDataCompressKeyPart(bData, &hdr, &buffers[1], assist, (SColCompressInfo *)pInfo);
72,062,775✔
1424
  TSDB_CHECK_CODE(code, lino, _exit);
72,072,091✔
1425

1426
  // Regulart column part
1427
  tBufferClear(&buffers[2]);
72,072,091✔
1428
  tBufferClear(&buffers[3]);
72,071,027✔
1429
  for (int i = 0; i < bData->nColData; i++) {
476,572,942✔
1430
    SColData *colData = tBlockDataGetColDataByIdx(bData, i);
404,506,278✔
1431

1432
    if (colData->cflag & COL_IS_KEY) {
404,500,823✔
1433
      continue;
472,742✔
1434
    }
1435
    if (colData->flag == HAS_NONE) {
404,027,069✔
1436
      continue;
13,231,483✔
1437
    }
1438

1439
    SColDataCompressInfo cinfo = {
390,801,755✔
1440
        .cmprAlg = pInfo->defaultCmprAlg,
390,799,942✔
1441
    };
1442
    code = tsdbGetColCmprAlgFromSet(pInfo->pColCmpr, colData->cid, &cinfo.cmprAlg);
390,813,470✔
1443
    if (code < 0) {
1444
      //
1445
    }
1446

1447
    int32_t offset = buffers[3].size;
390,771,197✔
1448
    code = tColDataCompress(colData, &cinfo, &buffers[3], assist);
390,783,402✔
1449
    TSDB_CHECK_CODE(code, lino, _exit);
390,741,083✔
1450

1451
    SBlockCol blockCol = (SBlockCol){.cid = cinfo.columnId,
390,741,083✔
1452
                                     .type = cinfo.dataType,
390,796,677✔
1453
                                     .cflag = cinfo.columnFlag,
390,796,677✔
1454
                                     .flag = cinfo.flag,
390,796,677✔
1455
                                     .szOrigin = cinfo.dataOriginalSize,
390,796,677✔
1456
                                     .szBitmap = cinfo.bitmapCompressedSize,
390,796,677✔
1457
                                     .szOffset = cinfo.offsetCompressedSize,
390,796,677✔
1458
                                     .szValue = cinfo.dataCompressedSize,
390,796,677✔
1459
                                     .offset = offset,
1460
                                     .alg = cinfo.cmprAlg};
390,796,677✔
1461

1462
    code = tPutBlockCol(&buffers[2], &blockCol, hdr.fmtVer, hdr.cmprAlg);
390,796,677✔
1463
    TSDB_CHECK_CODE(code, lino, _exit);
390,668,221✔
1464
  }
1465
  hdr.szBlkCol = buffers[2].size;
72,070,406✔
1466

1467
  // SDiskDataHdr part
1468
  tBufferClear(&buffers[0]);
1469
  code = tPutDiskDataHdr(&buffers[0], &hdr);
72,075,113✔
1470
  TSDB_CHECK_CODE(code, lino, _exit);
71,994,422✔
1471

1472
_exit:
71,979,263✔
1473
  if (code) {
72,032,195✔
1474
    tsdbError("tBlockDataCompress failed: code=%d, line=%d", code, lino);
×
1475
  }
1476
  return code;
72,032,195✔
1477
}
1478

1479
int32_t tBlockDataDecompress(SBufferReader *br, SBlockData *blockData, SBuffer *assist) {
17,470,626✔
1480
  int32_t       code = 0;
17,470,626✔
1481
  int32_t       lino = 0;
17,470,626✔
1482
  SDiskDataHdr  hdr = {0};
17,470,626✔
1483
  SCompressInfo cinfo;
1484

1485
  // SDiskDataHdr
1486
  code = tGetDiskDataHdr(br, &hdr);
17,470,626✔
1487
  TSDB_CHECK_CODE(code, lino, _exit);
17,470,021✔
1488

1489
  tBlockDataReset(blockData);
17,470,021✔
1490
  blockData->suid = hdr.suid;
17,470,019✔
1491
  blockData->uid = hdr.uid;
17,470,019✔
1492
  blockData->nRow = hdr.nRow;
17,470,019✔
1493

1494
  // Key part
1495
  code = tBlockDataDecompressKeyPart(&hdr, br, blockData, assist);
17,468,882✔
1496
  TSDB_CHECK_CODE(code, lino, _exit);
17,468,789✔
1497

1498
  // Column part
1499
  SBufferReader br2 = *br;
17,468,789✔
1500
  br->offset += hdr.szBlkCol;
17,468,789✔
1501
  for (uint32_t startOffset = br2.offset; br2.offset - startOffset < hdr.szBlkCol;) {
84,645,492✔
1502
    SBlockCol blockCol;
67,173,935✔
1503

1504
    code = tGetBlockCol(&br2, &blockCol, hdr.fmtVer, hdr.cmprAlg);
67,176,628✔
1505
    TSDB_CHECK_CODE(code, lino, _exit);
67,176,487✔
1506
    if (blockCol.alg == 0) blockCol.alg = hdr.cmprAlg;
67,176,487✔
1507
    code = tBlockDataDecompressColData(&hdr, &blockCol, br, blockData, assist);
67,176,487✔
1508
    TSDB_CHECK_CODE(code, lino, _exit);
67,174,910✔
1509
  }
1510

1511
_exit:
17,470,074✔
1512
  return code;
17,470,074✔
1513
}
1514

1515
// SDiskDataHdr ==============================
1516
int32_t tPutDiskDataHdr(SBuffer *buffer, const SDiskDataHdr *pHdr) {
72,000,307✔
1517
  int32_t code;
1518

1519
  if ((code = tBufferPutU32(buffer, pHdr->delimiter))) return code;
144,073,188✔
1520
  if ((code = tBufferPutU32v(buffer, pHdr->fmtVer))) return code;
144,136,525✔
1521
  if ((code = tBufferPutI64(buffer, pHdr->suid))) return code;
144,112,004✔
1522
  if ((code = tBufferPutI64(buffer, pHdr->uid))) return code;
144,098,259✔
1523
  if ((code = tBufferPutI32v(buffer, pHdr->szUid))) return code;
144,119,616✔
1524
  if ((code = tBufferPutI32v(buffer, pHdr->szVer))) return code;
144,100,774✔
1525
  if ((code = tBufferPutI32v(buffer, pHdr->szKey))) return code;
144,063,020✔
1526
  if ((code = tBufferPutI32v(buffer, pHdr->szBlkCol))) return code;
144,096,729✔
1527
  if ((code = tBufferPutI32v(buffer, pHdr->nRow))) return code;
144,129,540✔
1528
  if (pHdr->fmtVer < 2) {
72,064,774✔
1529
    if ((code = tBufferPutI8(buffer, pHdr->cmprAlg))) return code;
×
1530
  } else if (pHdr->fmtVer == 2) {
71,983,956✔
1531
    if ((code = tBufferPutU32(buffer, pHdr->cmprAlg))) return code;
144,061,701✔
1532
  } else {
1533
    // more data fmt ver
1534
  }
1535
  if (pHdr->fmtVer >= 1) {
72,059,937✔
1536
    if ((code = tBufferPutI8(buffer, pHdr->numOfPKs))) return code;
144,080,245✔
1537
    for (int i = 0; i < pHdr->numOfPKs; i++) {
72,505,208✔
1538
      if ((code = tPutBlockCol(buffer, &pHdr->primaryBlockCols[i], pHdr->fmtVer, pHdr->cmprAlg))) return code;
471,983✔
1539
    }
1540
  }
1541

1542
  return 0;
71,982,965✔
1543
}
1544

1545
int32_t tGetDiskDataHdr(SBufferReader *br, SDiskDataHdr *pHdr) {
214,063,750✔
1546
  int32_t code;
1547

1548
  if ((code = tBufferGetU32(br, &pHdr->delimiter))) return code;
214,063,750✔
1549
  if ((code = tBufferGetU32v(br, &pHdr->fmtVer))) return code;
214,080,996✔
1550
  if ((code = tBufferGetI64(br, &pHdr->suid))) return code;
214,060,676✔
1551
  if ((code = tBufferGetI64(br, &pHdr->uid))) return code;
214,024,462✔
1552
  if ((code = tBufferGetI32v(br, &pHdr->szUid))) return code;
214,134,500✔
1553
  if ((code = tBufferGetI32v(br, &pHdr->szVer))) return code;
214,115,999✔
1554
  if ((code = tBufferGetI32v(br, &pHdr->szKey))) return code;
214,186,642✔
1555
  if ((code = tBufferGetI32v(br, &pHdr->szBlkCol))) return code;
214,188,061✔
1556
  if ((code = tBufferGetI32v(br, &pHdr->nRow))) return code;
214,189,814✔
1557
  if (pHdr->fmtVer < 2) {
214,190,721✔
1558
    int8_t cmprAlg = 0;
×
1559
    if ((code = tBufferGetI8(br, &cmprAlg))) return code;
×
1560
    pHdr->cmprAlg = cmprAlg;
×
1561
  } else if (pHdr->fmtVer == 2) {
214,088,451✔
1562
    if ((code = tBufferGetU32(br, &pHdr->cmprAlg))) return code;
214,097,094✔
1563
  } else {
1564
    // more data fmt ver
1565
  }
1566
  if (pHdr->fmtVer >= 1) {
214,144,911✔
1567
    if ((code = tBufferGetI8(br, &pHdr->numOfPKs))) return code;
214,143,400✔
1568
    for (int i = 0; i < pHdr->numOfPKs; i++) {
232,527,059✔
1569
      if ((code = tGetBlockCol(br, &pHdr->primaryBlockCols[i], pHdr->fmtVer, pHdr->cmprAlg))) {
18,511,065✔
1570
        return code;
×
1571
      }
1572
    }
1573
  } else {
1574
    pHdr->numOfPKs = 0;
×
1575
  }
1576

1577
  return 0;
214,030,609✔
1578
}
1579

1580
// ALGORITHM ==============================
1581
int32_t tPutColumnDataAgg(SBuffer *buffer, SColumnDataAgg *pColAgg) {
123,536,126✔
1582
  int32_t code;
1583

1584
  if (pColAgg->colId & DECIMAL_AGG_FLAG) {
123,536,126✔
1585
    if ((code = tBufferPutI32v(buffer, pColAgg->colId))) return code;
222,352✔
1586
    if ((code = tBufferPutI16v(buffer, pColAgg->numOfNull))) return code;
222,352✔
1587
    if ((code = tBufferPutU64(buffer, pColAgg->decimal128Sum[0]))) return code;
222,352✔
1588
    if ((code = tBufferPutU64(buffer, pColAgg->decimal128Sum[1]))) return code;
222,352✔
1589
    if ((code = tBufferPutU64(buffer, pColAgg->decimal128Max[0]))) return code;
222,352✔
1590
    if ((code = tBufferPutU64(buffer, pColAgg->decimal128Max[1]))) return code;
222,352✔
1591
    if ((code = tBufferPutU64(buffer, pColAgg->decimal128Min[0]))) return code;
222,352✔
1592
    if ((code = tBufferPutU64(buffer, pColAgg->decimal128Min[1]))) return code;
222,352✔
1593
    if ((code = tBufferPutU8(buffer, pColAgg->overflow))) return code;
222,352✔
1594
  } else {
1595
    if ((code = tBufferPutI32v(buffer, pColAgg->colId))) return code;
246,850,649✔
1596
    if ((code = tBufferPutI16v(buffer, pColAgg->numOfNull))) return code;
246,847,582✔
1597
    if ((code = tBufferPutI64(buffer, pColAgg->sum))) return code;
246,844,069✔
1598
    if ((code = tBufferPutI64(buffer, pColAgg->max))) return code;
246,841,671✔
1599
    if ((code = tBufferPutI64(buffer, pColAgg->min))) return code;
246,842,466✔
1600
  }
1601

1602
  return 0;
123,534,157✔
1603
}
1604

1605
int32_t tGetColumnDataAgg(SBufferReader *br, SColumnDataAgg *pColAgg) {
118,968,750✔
1606
  int32_t code;
1607

1608
  if ((code = tBufferGetI32v(br, &pColAgg->colId))) return code;
118,968,750✔
1609
  if ((code = tBufferGetI16v(br, &pColAgg->numOfNull))) return code;
118,973,195✔
1610
  if (pColAgg->colId & DECIMAL_AGG_FLAG) {
118,978,113✔
1611
    pColAgg->colId &= 0xFFFF;
9,739,980✔
1612
    if ((code = tBufferGetU64(br, &pColAgg->decimal128Sum[0]))) return code;
9,740,683✔
1613
    if ((code = tBufferGetU64(br, &pColAgg->decimal128Sum[1]))) return code;
9,739,277✔
1614
    if ((code = tBufferGetU64(br, &pColAgg->decimal128Max[0]))) return code;
9,741,386✔
1615
    if ((code = tBufferGetU64(br, &pColAgg->decimal128Max[1]))) return code;
9,741,386✔
1616
    if ((code = tBufferGetU64(br, &pColAgg->decimal128Min[0]))) return code;
9,740,683✔
1617
    if ((code = tBufferGetU64(br, &pColAgg->decimal128Min[1]))) return code;
9,741,386✔
1618
    if ((code = tBufferGetU8(br, &pColAgg->overflow))) return code;
9,741,386✔
1619
  } else {
1620
    if ((code = tBufferGetI64(br, &pColAgg->sum))) return code;
109,237,883✔
1621
    if ((code = tBufferGetI64(br, &pColAgg->max))) return code;
109,231,313✔
1622
    if ((code = tBufferGetI64(br, &pColAgg->min))) return code;
109,238,141✔
1623
  }
1624

1625
  return 0;
118,971,951✔
1626
}
1627

1628
static int32_t tBlockDataCompressKeyPart(SBlockData *bData, SDiskDataHdr *hdr, SBuffer *buffer, SBuffer *assist,
72,024,496✔
1629
                                         SColCompressInfo *compressInfo) {
1630
  int32_t       code = 0;
72,024,496✔
1631
  int32_t       lino = 0;
72,024,496✔
1632
  SCompressInfo cinfo;
72,022,958✔
1633

1634
  // uid
1635
  if (bData->uid == 0) {
72,044,538✔
1636
    cinfo = (SCompressInfo){
31,642,079✔
1637
        .cmprAlg = hdr->cmprAlg,
31,640,769✔
1638
        .dataType = TSDB_DATA_TYPE_BIGINT,
1639
        .originalSize = sizeof(int64_t) * bData->nRow,
31,641,773✔
1640
    };
1641
    code = tCompressDataToBuffer(bData->aUid, &cinfo, buffer, assist);
31,642,079✔
1642
    TSDB_CHECK_CODE(code, lino, _exit);
31,639,741✔
1643
    hdr->szUid = cinfo.compressedSize;
31,639,741✔
1644
  }
1645

1646
  // version
1647
  cinfo = (SCompressInfo){
72,056,012✔
1648
      .cmprAlg = hdr->cmprAlg,
72,061,618✔
1649
      .dataType = TSDB_DATA_TYPE_BIGINT,
1650
      .originalSize = sizeof(int64_t) * bData->nRow,
72,046,598✔
1651
  };
1652
  code = tCompressDataToBuffer((uint8_t *)bData->aVersion, &cinfo, buffer, assist);
72,056,012✔
1653
  TSDB_CHECK_CODE(code, lino, _exit);
72,044,088✔
1654
  hdr->szVer = cinfo.compressedSize;
72,044,088✔
1655

1656
  // ts
1657
  cinfo = (SCompressInfo){
72,055,423✔
1658
      .cmprAlg = hdr->cmprAlg,
72,047,015✔
1659
      .dataType = TSDB_DATA_TYPE_TIMESTAMP,
1660
      .originalSize = sizeof(TSKEY) * bData->nRow,
72,062,533✔
1661
  };
1662

1663
  code = tCompressDataToBuffer((uint8_t *)bData->aTSKEY, &cinfo, buffer, assist);
72,055,423✔
1664
  TSDB_CHECK_CODE(code, lino, _exit);
72,079,788✔
1665
  hdr->szKey = cinfo.compressedSize;
72,079,788✔
1666

1667
  // primary keys
1668
  for (hdr->numOfPKs = 0; hdr->numOfPKs < bData->nColData; hdr->numOfPKs++) {
72,552,386✔
1669
    if (!(hdr->numOfPKs <= TD_MAX_PK_COLS)) {
72,541,467✔
1670
      return TSDB_CODE_INVALID_PARA;
×
1671
    }
1672

1673
    SBlockCol *blockCol = &hdr->primaryBlockCols[hdr->numOfPKs];
72,540,614✔
1674
    SColData  *colData = tBlockDataGetColDataByIdx(bData, hdr->numOfPKs);
72,533,057✔
1675

1676
    if ((colData->cflag & COL_IS_KEY) == 0) {
72,540,035✔
1677
      break;
72,074,502✔
1678
    }
1679

1680
    SColDataCompressInfo info = {
472,599✔
1681
        .cmprAlg = hdr->cmprAlg,
472,599✔
1682
    };
1683
    code = tsdbGetColCmprAlgFromSet(compressInfo->pColCmpr, colData->cid, &info.cmprAlg);
472,599✔
1684
    if (code < 0) {
1685
      // do nothing
1686
    } else {
1687
    }
1688

1689
    code = tColDataCompress(colData, &info, buffer, assist);
472,599✔
1690
    TSDB_CHECK_CODE(code, lino, _exit);
472,599✔
1691

1692
    *blockCol = (SBlockCol){
472,599✔
1693
        .cid = info.columnId,
472,599✔
1694
        .type = info.dataType,
472,599✔
1695
        .cflag = info.columnFlag,
472,599✔
1696
        .flag = info.flag,
472,599✔
1697
        .szOrigin = info.dataOriginalSize,
472,599✔
1698
        .szBitmap = info.bitmapCompressedSize,
472,599✔
1699
        .szOffset = info.offsetCompressedSize,
472,599✔
1700
        .szValue = info.dataCompressedSize,
472,599✔
1701
        .offset = 0,
1702
        .alg = info.cmprAlg,
472,599✔
1703
    };
1704
  }
1705

1706
_exit:
72,073,250✔
1707
  return code;
72,073,324✔
1708
}
1709

1710
int32_t tBlockDataDecompressColData(const SDiskDataHdr *hdr, const SBlockCol *blockCol, SBufferReader *br,
566,633,030✔
1711
                                    SBlockData *blockData, SBuffer *assist) {
1712
  int32_t code = 0;
566,633,030✔
1713
  int32_t lino = 0;
566,633,030✔
1714

1715
  SColData *colData;
566,628,825✔
1716

1717
  code = tBlockDataAddColData(blockData, blockCol->cid, blockCol->type, blockCol->cflag, &colData);
566,611,145✔
1718
  TSDB_CHECK_CODE(code, lino, _exit);
566,623,527✔
1719

1720
  SColDataCompressInfo info = {
1,022,246,318✔
1721
      .cmprAlg = blockCol->alg,
566,591,560✔
1722
      .columnFlag = blockCol->cflag,
566,647,282✔
1723
      .flag = blockCol->flag,
566,671,036✔
1724
      .dataType = blockCol->type,
566,607,837✔
1725
      .columnId = blockCol->cid,
566,631,784✔
1726
      .numOfData = hdr->nRow,
566,607,104✔
1727
      .bitmapOriginalSize = 0,
1728
      .bitmapCompressedSize = blockCol->szBitmap,
566,640,324✔
1729
      .offsetOriginalSize = blockCol->szOffset ? sizeof(int32_t) * hdr->nRow : 0,
566,635,517✔
1730
      .offsetCompressedSize = blockCol->szOffset,
566,586,410✔
1731
      .dataOriginalSize = blockCol->szOrigin,
566,628,847✔
1732
      .dataCompressedSize = blockCol->szValue,
566,605,238✔
1733
  };
1734

1735
  switch (blockCol->flag) {
566,633,551✔
1736
    case (HAS_NONE | HAS_NULL | HAS_VALUE):
297,391✔
1737
      info.bitmapOriginalSize = BIT2_SIZE(hdr->nRow);
297,391✔
1738
      break;
297,391✔
1739
    case (HAS_NONE | HAS_NULL):
88,143,407✔
1740
    case (HAS_NONE | HAS_VALUE):
1741
    case (HAS_NULL | HAS_VALUE):
1742
      info.bitmapOriginalSize = BIT1_SIZE(hdr->nRow);
88,143,407✔
1743
      break;
88,145,330✔
1744
  }
1745

1746
  code = tColDataDecompress(BR_PTR(br), &info, colData, assist);
566,615,692✔
1747
  TSDB_CHECK_CODE(code, lino, _exit);
566,555,899✔
1748
  br->offset += blockCol->szBitmap + blockCol->szOffset + blockCol->szValue;
566,555,899✔
1749

1750
_exit:
566,641,864✔
1751
  return code;
566,671,096✔
1752
}
1753

1754
int32_t tBlockDataDecompressKeyPart(const SDiskDataHdr *hdr, SBufferReader *br, SBlockData *blockData,
214,016,839✔
1755
                                    SBuffer *assist) {
1756
  int32_t       code = 0;
214,016,839✔
1757
  int32_t       lino = 0;
214,016,839✔
1758
  SCompressInfo cinfo;
214,014,896✔
1759

1760
  // uid
1761
  if (hdr->szUid > 0) {
214,059,799✔
1762
    cinfo = (SCompressInfo){
319,028,806✔
1763
        .cmprAlg = hdr->cmprAlg,
159,532,820✔
1764
        .dataType = TSDB_DATA_TYPE_BIGINT,
1765
        .compressedSize = hdr->szUid,
159,527,801✔
1766
        .originalSize = sizeof(int64_t) * hdr->nRow,
159,520,932✔
1767
    };
1768

1769
    code = tRealloc((uint8_t **)&blockData->aUid, cinfo.originalSize);
159,520,402✔
1770
    TSDB_CHECK_CODE(code, lino, _exit);
159,535,602✔
1771
    code = tDecompressData(BR_PTR(br), &cinfo, blockData->aUid, cinfo.originalSize, assist);
159,535,602✔
1772
    TSDB_CHECK_CODE(code, lino, _exit);
159,484,552✔
1773
    br->offset += cinfo.compressedSize;
159,484,552✔
1774
  }
1775

1776
  // version
1777
  cinfo = (SCompressInfo){
428,120,517✔
1778
      .cmprAlg = hdr->cmprAlg,
214,063,686✔
1779
      .dataType = TSDB_DATA_TYPE_BIGINT,
1780
      .compressedSize = hdr->szVer,
214,095,624✔
1781
      .originalSize = sizeof(int64_t) * hdr->nRow,
214,069,196✔
1782
  };
1783
  code = tRealloc((uint8_t **)&blockData->aVersion, cinfo.originalSize);
214,097,826✔
1784
  TSDB_CHECK_CODE(code, lino, _exit);
214,079,165✔
1785
  code = tDecompressData(BR_PTR(br), &cinfo, blockData->aVersion, cinfo.originalSize, assist);
214,079,165✔
1786
  TSDB_CHECK_CODE(code, lino, _exit);
214,139,424✔
1787
  br->offset += cinfo.compressedSize;
214,139,424✔
1788

1789
  // ts
1790
  cinfo = (SCompressInfo){
428,133,257✔
1791
      .cmprAlg = hdr->cmprAlg,
214,114,186✔
1792
      .dataType = TSDB_DATA_TYPE_TIMESTAMP,
1793
      .compressedSize = hdr->szKey,
214,131,280✔
1794
      .originalSize = sizeof(TSKEY) * hdr->nRow,
214,125,791✔
1795
  };
1796
  code = tRealloc((uint8_t **)&blockData->aTSKEY, cinfo.originalSize);
214,071,328✔
1797
  TSDB_CHECK_CODE(code, lino, _exit);
214,123,905✔
1798
  code = tDecompressData(BR_PTR(br), &cinfo, blockData->aTSKEY, cinfo.originalSize, assist);
214,123,905✔
1799
  TSDB_CHECK_CODE(code, lino, _exit);
214,174,304✔
1800
  br->offset += cinfo.compressedSize;
214,174,304✔
1801

1802
  // primary keys
1803
  for (int i = 0; i < hdr->numOfPKs; i++) {
232,642,319✔
1804
    const SBlockCol *blockCol = &hdr->primaryBlockCols[i];
18,518,391✔
1805

1806
    if (!(blockCol->flag == HAS_VALUE)) {
18,518,385✔
1807
      TSDB_CHECK_CODE(code = TSDB_CODE_FILE_CORRUPTED, lino, _exit);
×
1808
    }
1809
    if (!(blockCol->cflag & COL_IS_KEY)) {
18,508,003✔
1810
      TSDB_CHECK_CODE(code = TSDB_CODE_FILE_CORRUPTED, lino, _exit);
×
1811
    }
1812

1813
    code = tBlockDataDecompressColData(hdr, blockCol, br, blockData, assist);
18,515,329✔
1814
    TSDB_CHECK_CODE(code, lino, _exit);
18,504,319✔
1815
  }
1816

1817
_exit:
214,060,755✔
1818
  return code;
214,095,482✔
1819
}
1820

1821
int32_t tsdbGetColCmprAlgFromSet(SHashObj *set, int16_t colId, uint32_t *alg) {
463,312,071✔
1822
  if (set == NULL) return -1;
463,312,071✔
1823

1824
  uint32_t *ret = taosHashGet(set, &colId, sizeof(colId));
463,085,331✔
1825
  if (ret == NULL) {
463,076,920✔
1826
    return TSDB_CODE_NOT_FOUND;
1,358✔
1827
  }
1828

1829
  *alg = *ret;
463,075,562✔
1830
  return 0;
463,067,276✔
1831
}
1832
uint32_t tsdbCvtTimestampAlg(uint32_t alg) {
×
1833
  DEFINE_VAR(alg)
×
1834

1835
  return 0;
×
1836
}
1837

1838
int32_t tsdbAllocateDisk(STsdb *tsdb, const char *label, int32_t expLevel, SDiskID *diskId) {
19,168,039✔
1839
  int32_t code = 0;
19,168,039✔
1840
  int32_t lino = 0;
19,168,039✔
1841
  SDiskID did = {0};
19,168,039✔
1842
  STfs   *tfs = tsdb->pVnode->pTfs;
19,176,281✔
1843

1844
  code = tfsAllocDisk(tfs, expLevel, label, &did);
19,183,142✔
1845
  if (code) {
19,197,012✔
1846
    tsdbError("vgId:%d %s failed at %s:%d since %s", TD_VID(tsdb->pVnode), __func__, __FILE__, __LINE__,
×
1847
              tstrerror(code));
1848
    return code;
×
1849
  }
1850

1851
  if (tfsMkdirRecurAt(tfs, tsdb->path, did) != 0) {
19,197,012✔
1852
    tsdbError("vgId:%d %s failed at %s:%d since %s", TD_VID(tsdb->pVnode), __func__, __FILE__, __LINE__,
×
1853
              tstrerror(code));
1854
  }
1855

1856
  if (diskId) {
19,173,424✔
1857
    *diskId = did;
19,177,997✔
1858
  }
1859
  return code;
19,181,126✔
1860
}
1861

1862
int32_t tsdbAllocateDiskAtLevel(STsdb *tsdb, int32_t level, const char *label, SDiskID *diskId) {
66,002✔
1863
  int32_t code = 0;
66,002✔
1864
  SDiskID did = {0};
66,002✔
1865
  STfs   *tfs = tsdb->pVnode->pTfs;
66,002✔
1866

1867
  code = tfsAllocDiskAtLevel(tfs, level, label, &did);
66,002✔
1868
  if (code) {
66,002✔
1869
    return code;
×
1870
  }
1871

1872
  if (tfsMkdirRecurAt(tfs, tsdb->path, did) != 0) {
66,002✔
1873
    tsdbError("vgId:%d %s failed at %s:%d since %s", TD_VID(tsdb->pVnode), __func__, __FILE__, __LINE__,
×
1874
              tstrerror(code));
1875
  }
1876

1877
  if (diskId) {
66,002✔
1878
    *diskId = did;
66,002✔
1879
  }
1880
  return 0;
66,002✔
1881
}
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