tsdbSnapshot.c 53.7 KB
Newer Older
H
Hongze Cheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "tsdb.h"

H
Hongze Cheng 已提交
18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85
extern int32_t tsdbReadDataBlockEx(SDataFReader* pReader, SDataBlk* pDataBlk, SBlockData* pBlockData);

// STsdbDataIter2 ========================================
#define TSDB_MEM_TABLE_DATA_ITER 0
#define TSDB_DATA_FILE_DATA_ITER 1
#define TSDB_STT_FILE_DATA_ITER  2

typedef struct STsdbDataIter2 STsdbDataIter2;
struct STsdbDataIter2 {
  STsdbDataIter2* next;
  SRBTreeNode     rbtn;

  int32_t  type;
  SRowInfo rowInfo;
  union {
    // TSDB_MEM_TABLE_DATA_ITER
    struct {
      SMemTable* pMemTable;
    } mIter;

    // TSDB_DATA_FILE_DATA_ITER
    struct {
      SDataFReader* pReader;
      SArray*       aBlockIdx;  // SArray<SBlockIdx>
      SMapData      mDataBlk;
      SBlockData    bData;
      int32_t       iBlockIdx;
      int32_t       iDataBlk;
      int32_t       iRow;

    } dIter;

    // TSDB_STT_FILE_DATA_ITER
    struct {
      SDataFReader* pReader;
      int32_t       iStt;
      SArray*       aSttBlk;
      SBlockData    bData;
      int32_t       iSttBlk;
      int32_t       iRow;
    } sIter;
  };
};

#define TSDB_RBTN_TO_DATA_ITER(pNode) ((STsdbDataIter2*)(((char*)pNode) - offsetof(STsdbDataIter2, rbtn)))

/* open */
static int32_t tsdbOpenDataFileDataIter(SDataFReader* pReader, STsdbDataIter2** ppIter) {
  int32_t code = 0;
  int32_t lino = 0;

  // create handle
  STsdbDataIter2* pIter = (STsdbDataIter2*)taosMemoryCalloc(1, sizeof(*pIter));
  if (pIter == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    TSDB_CHECK_CODE(code, lino, _exit);
  }

  pIter->type = TSDB_DATA_FILE_DATA_ITER;
  pIter->dIter.pReader = pReader;
  if ((pIter->dIter.aBlockIdx = taosArrayInit(0, sizeof(SBlockIdx))) == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    TSDB_CHECK_CODE(code, lino, _exit);
  }

  code = tBlockDataCreate(&pIter->dIter.bData);
  TSDB_CHECK_CODE(code, lino, _exit);

H
Hongze Cheng 已提交
86 87 88
  pIter->dIter.iBlockIdx = 0;
  pIter->dIter.iDataBlk = 0;
  pIter->dIter.iRow = 0;
H
Hongze Cheng 已提交
89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132

  // read data
  code = tsdbReadBlockIdx(pReader, pIter->dIter.aBlockIdx);
  TSDB_CHECK_CODE(code, lino, _exit);

  if (taosArrayGetSize(pIter->dIter.aBlockIdx) == 0) goto _clear;

_exit:
  if (code) {
    if (pIter) {
    _clear:
      tBlockDataDestroy(&pIter->dIter.bData, 1);
      taosArrayDestroy(pIter->dIter.aBlockIdx);
      taosMemoryFree(pIter);
      pIter = NULL;
    }
  }
  *ppIter = pIter;
  return code;
}

static int32_t tsdbOpenSttFileDataIter(SDataFReader* pReader, int32_t iStt, STsdbDataIter2** ppIter) {
  int32_t code = 0;
  int32_t lino = 0;

  // create handle
  STsdbDataIter2* pIter = (STsdbDataIter2*)taosMemoryCalloc(1, sizeof(*pIter));
  if (pIter == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    TSDB_CHECK_CODE(code, lino, _exit);
  }

  pIter->type = TSDB_STT_FILE_DATA_ITER;
  pIter->sIter.pReader = pReader;
  pIter->sIter.iStt = iStt;
  pIter->sIter.aSttBlk = taosArrayInit(0, sizeof(SSttBlk));
  if (pIter->sIter.aSttBlk == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    TSDB_CHECK_CODE(code, lino, _exit);
  }

  code = tBlockDataCreate(&pIter->sIter.bData);
  TSDB_CHECK_CODE(code, lino, _exit);

H
Hongze Cheng 已提交
133 134
  pIter->sIter.iSttBlk = 0;
  pIter->sIter.iRow = 0;
H
Hongze Cheng 已提交
135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196

  // read data
  code = tsdbReadSttBlk(pReader, iStt, pIter->sIter.aSttBlk);
  TSDB_CHECK_CODE(code, lino, _exit);

  if (taosArrayGetSize(pIter->sIter.aSttBlk) == 0) goto _clear;

_exit:
  if (code) {
    if (pIter) {
    _clear:
      taosArrayDestroy(pIter->sIter.aSttBlk);
      tBlockDataDestroy(&pIter->sIter.bData, 1);
      taosMemoryFree(pIter);
      pIter = NULL;
    }
  }
  *ppIter = pIter;
  return code;
}

/* close */
static void tsdbCloseDataFileDataIter(STsdbDataIter2* pIter) {
  tBlockDataDestroy(&pIter->dIter.bData, 1);
  tMapDataClear(&pIter->dIter.mDataBlk);
  taosArrayDestroy(pIter->dIter.aBlockIdx);
  taosMemoryFree(pIter);
}

static void tsdbCloseSttFileDataIter(STsdbDataIter2* pIter) {
  tBlockDataDestroy(&pIter->sIter.bData, 1);
  taosArrayDestroy(pIter->sIter.aSttBlk);
  taosMemoryFree(pIter);
}

static void tsdbCloseDataIter2(STsdbDataIter2* pIter) {
  if (pIter->type == TSDB_MEM_TABLE_DATA_ITER) {
    ASSERT(0);
  } else if (pIter->type == TSDB_DATA_FILE_DATA_ITER) {
    tsdbCloseDataFileDataIter(pIter);
  } else if (pIter->type == TSDB_STT_FILE_DATA_ITER) {
    tsdbCloseSttFileDataIter(pIter);
  } else {
    ASSERT(0);
  }
}

/* cmpr */
static int32_t tsdbDataIterCmprFn(const SRBTreeNode* pNode1, const SRBTreeNode* pNode2) {
  STsdbDataIter2* pIter1 = TSDB_RBTN_TO_DATA_ITER(pNode1);
  STsdbDataIter2* pIter2 = TSDB_RBTN_TO_DATA_ITER(pNode2);
  return tRowInfoCmprFn(&pIter1->rowInfo, &pIter2->rowInfo);
}

/* seek */

/* iter next */
static int32_t tsdbDataFileDataIterNext(STsdbDataIter2* pIter) {
  int32_t code = 0;
  int32_t lino = 0;

  for (;;) {
H
Hongze Cheng 已提交
197
    if (pIter->dIter.iRow < pIter->dIter.bData.nRow) {
H
Hongze Cheng 已提交
198 199 200
      pIter->rowInfo.suid = pIter->dIter.bData.suid;
      pIter->rowInfo.uid = pIter->dIter.bData.uid;
      pIter->rowInfo.row = tsdbRowFromBlockData(&pIter->dIter.bData, pIter->dIter.iRow);
H
Hongze Cheng 已提交
201
      pIter->dIter.iRow++;
H
Hongze Cheng 已提交
202 203 204 205
      break;
    }

    for (;;) {
H
Hongze Cheng 已提交
206
      if (pIter->dIter.iDataBlk < pIter->dIter.mDataBlk.nItem) {
H
Hongze Cheng 已提交
207 208 209 210 211 212
        SDataBlk dataBlk;
        tMapDataGetItemByIdx(&pIter->dIter.mDataBlk, pIter->dIter.iDataBlk, &dataBlk, tGetDataBlk);

        code = tsdbReadDataBlockEx(pIter->dIter.pReader, &dataBlk, &pIter->dIter.bData);
        TSDB_CHECK_CODE(code, lino, _exit);

H
Hongze Cheng 已提交
213 214
        pIter->dIter.iDataBlk++;
        pIter->dIter.iRow = 0;
H
Hongze Cheng 已提交
215 216 217 218 219

        break;
      }

      for (;;) {
H
Hongze Cheng 已提交
220
        if (pIter->dIter.iBlockIdx < taosArrayGetSize(pIter->dIter.aBlockIdx)) {
H
Hongze Cheng 已提交
221 222 223 224 225
          SBlockIdx* pBlockIdx = taosArrayGet(pIter->dIter.aBlockIdx, pIter->dIter.iBlockIdx);

          code = tsdbReadDataBlk(pIter->dIter.pReader, pBlockIdx, &pIter->dIter.mDataBlk);
          TSDB_CHECK_CODE(code, lino, _exit);

H
Hongze Cheng 已提交
226 227
          pIter->dIter.iBlockIdx++;
          pIter->dIter.iDataBlk = 0;
H
Hongze Cheng 已提交
228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249

          break;
        } else {
          pIter->rowInfo = (SRowInfo){0};
          goto _exit;
        }
      }
    }
  }

_exit:
  if (code) {
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
  }
  return code;
}

static int32_t tsdbSttFileDataIterNext(STsdbDataIter2* pIter) {
  int32_t code = 0;
  int32_t lino = 0;

  for (;;) {
H
Hongze Cheng 已提交
250
    if (pIter->sIter.iRow < pIter->sIter.bData.nRow) {
H
Hongze Cheng 已提交
251 252 253
      pIter->rowInfo.suid = pIter->sIter.bData.suid;
      pIter->rowInfo.uid = pIter->sIter.bData.uid ? pIter->sIter.bData.uid : pIter->sIter.bData.aUid[pIter->sIter.iRow];
      pIter->rowInfo.row = tsdbRowFromBlockData(&pIter->sIter.bData, pIter->sIter.iRow);
H
Hongze Cheng 已提交
254
      pIter->sIter.iRow++;
H
Hongze Cheng 已提交
255 256 257
      break;
    }

H
Hongze Cheng 已提交
258
    if (pIter->sIter.iSttBlk < taosArrayGetSize(pIter->sIter.aSttBlk)) {
H
Hongze Cheng 已提交
259 260 261 262 263
      SSttBlk* pSttBlk = taosArrayGet(pIter->sIter.aSttBlk, pIter->sIter.iSttBlk);

      code = tsdbReadSttBlockEx(pIter->sIter.pReader, pIter->sIter.iStt, pSttBlk, &pIter->sIter.bData);
      TSDB_CHECK_CODE(code, lino, _exit);

H
Hongze Cheng 已提交
264 265 266
      pIter->sIter.iSttBlk++;

      pIter->sIter.iRow = 0;
H
Hongze Cheng 已提交
267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297
    } else {
      pIter->rowInfo = (SRowInfo){0};
      break;
    }
  }

_exit:
  if (code) {
    tsdbError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
  }
  return code;
}

static int32_t tsdbDataIterNext2(STsdbDataIter2* pIter) {
  int32_t code = 0;

  if (pIter->type == TSDB_MEM_TABLE_DATA_ITER) {
    ASSERT(0);
    return code;
  } else if (pIter->type == TSDB_DATA_FILE_DATA_ITER) {
    return tsdbDataFileDataIterNext(pIter);
  } else if (pIter->type == TSDB_STT_FILE_DATA_ITER) {
    return tsdbSttFileDataIterNext(pIter);
  } else {
    ASSERT(0);
    return code;
  }
}

/* get */

H
Hongze Cheng 已提交
298
// STsdbSnapReader ========================================
H
Hongze Cheng 已提交
299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321
typedef enum { SNAP_DATA_FILE_ITER = 0, SNAP_STT_FILE_ITER } EFIterT;
typedef struct {
  SRBTreeNode n;
  SRowInfo    rInfo;
  EFIterT     type;
  union {
    struct {
      SArray*    aBlockIdx;
      int32_t    iBlockIdx;
      SBlockIdx* pBlockIdx;
      SMapData   mBlock;
      int32_t    iBlock;
    };  // .data file
    struct {
      int32_t iStt;
      SArray* aSttBlk;
      int32_t iSttBlk;
    };  // .stt file
  };
  SBlockData bData;
  int32_t    iRow;
} SFDataIter;

H
Hongze Cheng 已提交
322
struct STsdbSnapReader {
H
more  
Hongze Cheng 已提交
323 324 325
  STsdb*  pTsdb;
  int64_t sver;
  int64_t ever;
H
Hongze Cheng 已提交
326
  STsdbFS fs;
C
Cary Xu 已提交
327
  int8_t  type;
H
more  
Hongze Cheng 已提交
328
  // for data file
H
Hongze Cheng 已提交
329 330
  int8_t        dataDone;
  int32_t       fid;
H
more  
Hongze Cheng 已提交
331
  SDataFReader* pDataFReader;
H
Hongze Cheng 已提交
332 333
  SFDataIter*   pIter;
  SRBTree       rbt;
H
Hongze Cheng 已提交
334
  SFDataIter    aFDataIter[TSDB_MAX_STT_TRIGGER + 1];
H
Hongze Cheng 已提交
335 336
  SBlockData    bData;
  SSkmInfo      skmTable;
H
more  
Hongze Cheng 已提交
337
  // for del file
H
Hongze Cheng 已提交
338
  int8_t       delDone;
H
more  
Hongze Cheng 已提交
339
  SDelFReader* pDelFReader;
H
Hongze Cheng 已提交
340
  SArray*      aDelIdx;  // SArray<SDelIdx>
H
Hongze Cheng 已提交
341 342
  int32_t      iDelIdx;
  SArray*      aDelData;  // SArray<SDelData>
H
Hongze Cheng 已提交
343
  uint8_t*     aBuf[5];
H
Hongze Cheng 已提交
344 345
};

H
Hongze Cheng 已提交
346 347
extern int32_t tsdbUpdateTableSchema(SMeta* pMeta, int64_t suid, int64_t uid, SSkmInfo* pSkmInfo);

H
Hongze Cheng 已提交
348 349 350 351 352 353 354
static int32_t tFDataIterCmprFn(const SRBTreeNode* pNode1, const SRBTreeNode* pNode2) {
  SFDataIter* pIter1 = (SFDataIter*)(((uint8_t*)pNode1) - offsetof(SFDataIter, n));
  SFDataIter* pIter2 = (SFDataIter*)(((uint8_t*)pNode2) - offsetof(SFDataIter, n));

  return tRowInfoCmprFn(&pIter1->rInfo, &pIter2->rInfo);
}

H
Hongze Cheng 已提交
355
static int32_t tsdbSnapReadOpenFile(STsdbSnapReader* pReader) {
H
Hongze Cheng 已提交
356
  int32_t code = 0;
H
add log  
Hongze Cheng 已提交
357
  int32_t lino = 0;
H
Hongze Cheng 已提交
358

H
Hongze Cheng 已提交
359 360 361
  SDFileSet  dFileSet = {.fid = pReader->fid};
  SDFileSet* pSet = taosArraySearch(pReader->fs.aDFileSet, &dFileSet, tDFileSetCmprFn, TD_GT);
  if (pSet == NULL) return code;
H
Hongze Cheng 已提交
362

H
Hongze Cheng 已提交
363 364
  pReader->fid = pSet->fid;
  code = tsdbDataFReaderOpen(&pReader->pDataFReader, pReader->pTsdb, pSet);
H
add log  
Hongze Cheng 已提交
365
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
366 367

  pReader->pIter = NULL;
H
Hongze Cheng 已提交
368
  tRBTreeCreate(&pReader->rbt, tFDataIterCmprFn);
H
Hongze Cheng 已提交
369 370 371 372 373 374

  // .data file
  SFDataIter* pIter = &pReader->aFDataIter[0];
  pIter->type = SNAP_DATA_FILE_ITER;

  code = tsdbReadBlockIdx(pReader->pDataFReader, pIter->aBlockIdx);
H
add log  
Hongze Cheng 已提交
375
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
376 377 378 379

  for (pIter->iBlockIdx = 0; pIter->iBlockIdx < taosArrayGetSize(pIter->aBlockIdx); pIter->iBlockIdx++) {
    pIter->pBlockIdx = (SBlockIdx*)taosArrayGet(pIter->aBlockIdx, pIter->iBlockIdx);

H
Hongze Cheng 已提交
380
    code = tsdbReadDataBlk(pReader->pDataFReader, pIter->pBlockIdx, &pIter->mBlock);
H
add log  
Hongze Cheng 已提交
381
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
382 383 384 385 386 387

    for (pIter->iBlock = 0; pIter->iBlock < pIter->mBlock.nItem; pIter->iBlock++) {
      SDataBlk dataBlk;
      tMapDataGetItemByIdx(&pIter->mBlock, pIter->iBlock, &dataBlk, tGetDataBlk);

      if (dataBlk.minVer > pReader->ever || dataBlk.maxVer < pReader->sver) continue;
H
Hongze Cheng 已提交
388

H
Hongze Cheng 已提交
389
      code = tsdbReadDataBlockEx(pReader->pDataFReader, &dataBlk, &pIter->bData);
H
add log  
Hongze Cheng 已提交
390
      TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
391

392 393
      ASSERT(pIter->pBlockIdx->suid == pIter->bData.suid);
      ASSERT(pIter->pBlockIdx->uid == pIter->bData.uid);
H
Hongze Cheng 已提交
394

H
Hongze Cheng 已提交
395 396
      for (pIter->iRow = 0; pIter->iRow < pIter->bData.nRow; pIter->iRow++) {
        int64_t rowVer = pIter->bData.aVersion[pIter->iRow];
H
Hongze Cheng 已提交
397

H
Hongze Cheng 已提交
398 399 400 401 402 403
        if (rowVer >= pReader->sver && rowVer <= pReader->ever) {
          pIter->rInfo.suid = pIter->pBlockIdx->suid;
          pIter->rInfo.uid = pIter->pBlockIdx->uid;
          pIter->rInfo.row = tsdbRowFromBlockData(&pIter->bData, pIter->iRow);
          goto _add_iter_and_break;
        }
H
Hongze Cheng 已提交
404
      }
H
Hongze Cheng 已提交
405
    }
H
Hongze Cheng 已提交
406

H
Hongze Cheng 已提交
407
    continue;
H
Hongze Cheng 已提交
408

H
Hongze Cheng 已提交
409 410 411 412
  _add_iter_and_break:
    tRBTreePut(&pReader->rbt, (SRBTreeNode*)pIter);
    break;
  }
H
Hongze Cheng 已提交
413

H
Hongze Cheng 已提交
414 415 416 417 418
  // .stt file
  pIter = &pReader->aFDataIter[1];
  for (int32_t iStt = 0; iStt < pSet->nSttF; iStt++) {
    pIter->type = SNAP_STT_FILE_ITER;
    pIter->iStt = iStt;
H
Hongze Cheng 已提交
419

H
Hongze Cheng 已提交
420
    code = tsdbReadSttBlk(pReader->pDataFReader, iStt, pIter->aSttBlk);
H
add log  
Hongze Cheng 已提交
421
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
422 423 424 425 426 427 428

    for (pIter->iSttBlk = 0; pIter->iSttBlk < taosArrayGetSize(pIter->aSttBlk); pIter->iSttBlk++) {
      SSttBlk* pSttBlk = (SSttBlk*)taosArrayGet(pIter->aSttBlk, pIter->iSttBlk);

      if (pSttBlk->minVer > pReader->ever) continue;
      if (pSttBlk->maxVer < pReader->sver) continue;

H
Hongze Cheng 已提交
429
      code = tsdbReadSttBlockEx(pReader->pDataFReader, iStt, pSttBlk, &pIter->bData);
H
add log  
Hongze Cheng 已提交
430
      TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
431 432 433 434 435 436

      for (pIter->iRow = 0; pIter->iRow < pIter->bData.nRow; pIter->iRow++) {
        int64_t rowVer = pIter->bData.aVersion[pIter->iRow];

        if (rowVer >= pReader->sver && rowVer <= pReader->ever) {
          pIter->rInfo.suid = pIter->bData.suid;
H
Hongze Cheng 已提交
437
          pIter->rInfo.uid = pIter->bData.uid ? pIter->bData.uid : pIter->bData.aUid[pIter->iRow];
H
Hongze Cheng 已提交
438 439 440 441
          pIter->rInfo.row = tsdbRowFromBlockData(&pIter->bData, pIter->iRow);
          goto _add_iter;
        }
      }
H
Hongze Cheng 已提交
442
    }
H
Hongze Cheng 已提交
443

H
Hongze Cheng 已提交
444
    continue;
H
Hongze Cheng 已提交
445

H
Hongze Cheng 已提交
446 447 448 449
  _add_iter:
    tRBTreePut(&pReader->rbt, (SRBTreeNode*)pIter);
    pIter++;
  }
H
Hongze Cheng 已提交
450

H
add log  
Hongze Cheng 已提交
451 452 453 454 455 456 457
_exit:
  if (code) {
    tsdbError("vgId:%d, %s failed since %s", TD_VID(pReader->pTsdb->pVnode), __func__, tstrerror(code));
  } else {
    tsdbInfo("vgId:%d, %s done, path:%s, fid:%d", TD_VID(pReader->pTsdb->pVnode), __func__, pReader->pTsdb->path,
             pReader->fid);
  }
H
Hongze Cheng 已提交
458 459 460 461 462 463 464
  return code;
}

static int32_t tsdbSnapNextRow(STsdbSnapReader* pReader) {
  int32_t code = 0;

  if (pReader->pIter) {
H
Hongze Cheng 已提交
465
    SFDataIter* pIter = NULL;
H
Hongze Cheng 已提交
466 467
    while (true) {
    _find_row:
H
Hongze Cheng 已提交
468
      pIter = pReader->pIter;
H
Hongze Cheng 已提交
469 470 471 472
      for (pIter->iRow++; pIter->iRow < pIter->bData.nRow; pIter->iRow++) {
        int64_t rowVer = pIter->bData.aVersion[pIter->iRow];

        if (rowVer >= pReader->sver && rowVer <= pReader->ever) {
H
Hongze Cheng 已提交
473
          pIter->rInfo.suid = pIter->bData.suid;
H
Hongze Cheng 已提交
474 475 476
          pIter->rInfo.uid = pIter->bData.uid ? pIter->bData.uid : pIter->bData.aUid[pIter->iRow];
          pIter->rInfo.row = tsdbRowFromBlockData(&pIter->bData, pIter->iRow);
          goto _out;
H
Hongze Cheng 已提交
477
        }
H
Hongze Cheng 已提交
478
      }
H
Hongze Cheng 已提交
479

H
Hongze Cheng 已提交
480 481 482 483 484
      if (pIter->type == SNAP_DATA_FILE_ITER) {
        while (true) {
          for (pIter->iBlock++; pIter->iBlock < pIter->mBlock.nItem; pIter->iBlock++) {
            SDataBlk dataBlk;
            tMapDataGetItemByIdx(&pIter->mBlock, pIter->iBlock, &dataBlk, tGetDataBlk);
H
Hongze Cheng 已提交
485

H
Hongze Cheng 已提交
486 487 488
            if (dataBlk.minVer > pReader->ever || dataBlk.maxVer < pReader->sver) continue;

            code = tsdbReadDataBlockEx(pReader->pDataFReader, &dataBlk, &pIter->bData);
H
Hongze Cheng 已提交
489 490
            if (code) goto _err;

H
Hongze Cheng 已提交
491 492
            pIter->iRow = -1;
            goto _find_row;
H
Hongze Cheng 已提交
493
          }
H
Hongze Cheng 已提交
494 495 496 497 498

          pIter->iBlockIdx++;
          if (pIter->iBlockIdx >= taosArrayGetSize(pIter->aBlockIdx)) break;

          pIter->pBlockIdx = (SBlockIdx*)taosArrayGet(pIter->aBlockIdx, pIter->iBlockIdx);
H
Hongze Cheng 已提交
499
          code = tsdbReadDataBlk(pReader->pDataFReader, pIter->pBlockIdx, &pIter->mBlock);
H
Hongze Cheng 已提交
500 501
          if (code) goto _err;
          pIter->iBlock = -1;
H
Hongze Cheng 已提交
502
        }
H
Hongze Cheng 已提交
503

H
Hongze Cheng 已提交
504
        pReader->pIter = NULL;
H
Hongze Cheng 已提交
505
        break;
H
Hongze Cheng 已提交
506 507 508
      } else if (pIter->type == SNAP_STT_FILE_ITER) {
        for (pIter->iSttBlk++; pIter->iSttBlk < taosArrayGetSize(pIter->aSttBlk); pIter->iSttBlk++) {
          SSttBlk* pSttBlk = (SSttBlk*)taosArrayGet(pIter->aSttBlk, pIter->iSttBlk);
H
Hongze Cheng 已提交
509

H
Hongze Cheng 已提交
510
          if (pSttBlk->minVer > pReader->ever || pSttBlk->maxVer < pReader->sver) continue;
H
Hongze Cheng 已提交
511

H
Hongze Cheng 已提交
512
          code = tsdbReadSttBlockEx(pReader->pDataFReader, pIter->iStt, pSttBlk, &pIter->bData);
H
Hongze Cheng 已提交
513 514 515 516
          if (code) goto _err;

          pIter->iRow = -1;
          goto _find_row;
H
Hongze Cheng 已提交
517
        }
H
Hongze Cheng 已提交
518 519

        pReader->pIter = NULL;
H
Hongze Cheng 已提交
520
        break;
H
Hongze Cheng 已提交
521
      } else {
522
        ASSERT(0);
H
Hongze Cheng 已提交
523 524 525 526 527 528 529 530 531 532 533
      }
    }

  _out:
    pIter = (SFDataIter*)tRBTreeMin(&pReader->rbt);
    if (pReader->pIter && pIter) {
      int32_t c = tRowInfoCmprFn(&pReader->pIter->rInfo, &pIter->rInfo);
      if (c > 0) {
        tRBTreePut(&pReader->rbt, (SRBTreeNode*)pReader->pIter);
        pReader->pIter = NULL;
      } else {
534
        ASSERT(c);
H
Hongze Cheng 已提交
535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551
      }
    }
  }

  if (pReader->pIter == NULL) {
    pReader->pIter = (SFDataIter*)tRBTreeMin(&pReader->rbt);
    if (pReader->pIter) {
      tRBTreeDrop(&pReader->rbt, (SRBTreeNode*)pReader->pIter);
    }
  }

  return code;

_err:
  return code;
}

H
Hongze Cheng 已提交
552 553 554 555 556 557 558 559 560 561 562 563 564 565
static SRowInfo* tsdbSnapGetRow(STsdbSnapReader* pReader) {
  if (pReader->pIter) {
    return &pReader->pIter->rInfo;
  } else {
    tsdbSnapNextRow(pReader);

    if (pReader->pIter) {
      return &pReader->pIter->rInfo;
    } else {
      return NULL;
    }
  }
}

H
Hongze Cheng 已提交
566 567 568
static int32_t tsdbSnapCmprData(STsdbSnapReader* pReader, uint8_t** ppData) {
  int32_t code = 0;

569
  ASSERT(pReader->bData.nRow);
H
Hongze Cheng 已提交
570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600

  int32_t aBufN[5] = {0};
  code = tCmprBlockData(&pReader->bData, TWO_STAGE_COMP, NULL, NULL, pReader->aBuf, aBufN);
  if (code) goto _exit;

  int32_t size = aBufN[0] + aBufN[1] + aBufN[2] + aBufN[3];
  *ppData = taosMemoryMalloc(sizeof(SSnapDataHdr) + size);
  if (*ppData == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    goto _exit;
  }

  SSnapDataHdr* pHdr = (SSnapDataHdr*)*ppData;
  pHdr->type = SNAP_DATA_TSDB;
  pHdr->size = size;

  memcpy(pHdr->data, pReader->aBuf[3], aBufN[3]);
  memcpy(pHdr->data + aBufN[3], pReader->aBuf[2], aBufN[2]);
  if (aBufN[1]) {
    memcpy(pHdr->data + aBufN[3] + aBufN[2], pReader->aBuf[1], aBufN[1]);
  }
  if (aBufN[0]) {
    memcpy(pHdr->data + aBufN[3] + aBufN[2] + aBufN[1], pReader->aBuf[0], aBufN[0]);
  }

_exit:
  return code;
}

static int32_t tsdbSnapReadData(STsdbSnapReader* pReader, uint8_t** ppData) {
  int32_t code = 0;
H
add log  
Hongze Cheng 已提交
601 602 603
  int32_t lino = 0;

  STsdb* pTsdb = pReader->pTsdb;
H
Hongze Cheng 已提交
604 605 606 607

  while (true) {
    if (pReader->pDataFReader == NULL) {
      code = tsdbSnapReadOpenFile(pReader);
H
add log  
Hongze Cheng 已提交
608
      TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
609 610 611 612 613 614 615 616 617 618 619 620 621 622
    }

    if (pReader->pDataFReader == NULL) break;

    SRowInfo* pRowInfo = tsdbSnapGetRow(pReader);
    if (pRowInfo == NULL) {
      tsdbDataFReaderClose(&pReader->pDataFReader);
      continue;
    }

    TABLEID     id = {.suid = pRowInfo->suid, .uid = pRowInfo->uid};
    SBlockData* pBlockData = &pReader->bData;

    code = tsdbUpdateTableSchema(pTsdb->pVnode->pMeta, id.suid, id.uid, &pReader->skmTable);
H
add log  
Hongze Cheng 已提交
623
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
624

H
Hongze Cheng 已提交
625
    code = tBlockDataInit(pBlockData, &id, pReader->skmTable.pTSchema, NULL, 0);
H
add log  
Hongze Cheng 已提交
626
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
627 628 629

    while (pRowInfo->suid == id.suid && pRowInfo->uid == id.uid) {
      code = tBlockDataAppendRow(pBlockData, &pRowInfo->row, NULL, pRowInfo->uid);
H
add log  
Hongze Cheng 已提交
630
      TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
631 632

      code = tsdbSnapNextRow(pReader);
H
add log  
Hongze Cheng 已提交
633
      TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
634 635 636

      pRowInfo = tsdbSnapGetRow(pReader);
      if (pRowInfo == NULL) {
H
Hongze Cheng 已提交
637 638
        tsdbDataFReaderClose(&pReader->pDataFReader);
        break;
H
Hongze Cheng 已提交
639
      }
H
Hongze Cheng 已提交
640 641

      if (pBlockData->nRow >= 4096) break;
H
Hongze Cheng 已提交
642
    }
H
Hongze Cheng 已提交
643 644

    code = tsdbSnapCmprData(pReader, ppData);
H
add log  
Hongze Cheng 已提交
645
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
646 647

    break;
H
Hongze Cheng 已提交
648 649
  }

H
add log  
Hongze Cheng 已提交
650 651 652 653
_exit:
  if (code) {
    tsdbError("vgId:%d, %s failed since %s, path:%s", TD_VID(pTsdb->pVnode), __func__, tstrerror(code), pTsdb->path);
  }
H
Hongze Cheng 已提交
654 655 656 657
  return code;
}

static int32_t tsdbSnapReadDel(STsdbSnapReader* pReader, uint8_t** ppData) {
H
add log  
Hongze Cheng 已提交
658 659 660
  int32_t code = 0;
  int32_t lino = 0;

H
Hongze Cheng 已提交
661
  STsdb*    pTsdb = pReader->pTsdb;
H
Hongze Cheng 已提交
662
  SDelFile* pDelFile = pReader->fs.pDelFile;
H
Hongze Cheng 已提交
663 664 665 666 667 668 669

  if (pReader->pDelFReader == NULL) {
    if (pDelFile == NULL) {
      goto _exit;
    }

    // open
H
Hongze Cheng 已提交
670
    code = tsdbDelFReaderOpen(&pReader->pDelFReader, pDelFile, pTsdb);
H
add log  
Hongze Cheng 已提交
671
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
672

H
Hongze Cheng 已提交
673
    // read index
H
Hongze Cheng 已提交
674
    code = tsdbReadDelIdx(pReader->pDelFReader, pReader->aDelIdx);
H
add log  
Hongze Cheng 已提交
675
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
676 677 678 679

    pReader->iDelIdx = 0;
  }

H
Hongze Cheng 已提交
680 681 682 683 684 685
  while (true) {
    if (pReader->iDelIdx >= taosArrayGetSize(pReader->aDelIdx)) {
      tsdbDelFReaderClose(&pReader->pDelFReader);
      break;
    }

H
Hongze Cheng 已提交
686 687
    SDelIdx* pDelIdx = (SDelIdx*)taosArrayGet(pReader->aDelIdx, pReader->iDelIdx);

H
Hongze Cheng 已提交
688 689
    pReader->iDelIdx++;

H
Hongze Cheng 已提交
690
    code = tsdbReadDelData(pReader->pDelFReader, pDelIdx, pReader->aDelData);
H
add log  
Hongze Cheng 已提交
691
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
692

H
Hongze Cheng 已提交
693
    int32_t size = 0;
H
Hongze Cheng 已提交
694 695 696 697
    for (int32_t iDelData = 0; iDelData < taosArrayGetSize(pReader->aDelData); iDelData++) {
      SDelData* pDelData = (SDelData*)taosArrayGet(pReader->aDelData, iDelData);

      if (pDelData->version >= pReader->sver && pDelData->version <= pReader->ever) {
H
Hongze Cheng 已提交
698
        size += tPutDelData(NULL, pDelData);
H
Hongze Cheng 已提交
699 700
      }
    }
H
Hongze Cheng 已提交
701
    if (size == 0) continue;
H
Hongze Cheng 已提交
702

H
Hongze Cheng 已提交
703 704 705 706 707
    // org data
    size = sizeof(TABLEID) + size;
    *ppData = taosMemoryMalloc(sizeof(SSnapDataHdr) + size);
    if (*ppData == NULL) {
      code = TSDB_CODE_OUT_OF_MEMORY;
H
add log  
Hongze Cheng 已提交
708
      TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
709
    }
H
Hongze Cheng 已提交
710

H
Hongze Cheng 已提交
711
    SSnapDataHdr* pHdr = (SSnapDataHdr*)(*ppData);
C
Cary Xu 已提交
712
    pHdr->type = SNAP_DATA_DEL;
H
Hongze Cheng 已提交
713
    pHdr->size = size;
H
Hongze Cheng 已提交
714

H
Hongze Cheng 已提交
715 716 717 718 719 720
    TABLEID* pId = (TABLEID*)(&pHdr[1]);
    pId->suid = pDelIdx->suid;
    pId->uid = pDelIdx->uid;
    int32_t n = sizeof(SSnapDataHdr) + sizeof(TABLEID);
    for (int32_t iDelData = 0; iDelData < taosArrayGetSize(pReader->aDelData); iDelData++) {
      SDelData* pDelData = (SDelData*)taosArrayGet(pReader->aDelData, iDelData);
H
Hongze Cheng 已提交
721

H
Hongze Cheng 已提交
722 723
      if (pDelData->version < pReader->sver) continue;
      if (pDelData->version > pReader->ever) continue;
H
Hongze Cheng 已提交
724

H
Hongze Cheng 已提交
725
      n += tPutDelData((*ppData) + n, pDelData);
H
Hongze Cheng 已提交
726 727
    }

H
Hongze Cheng 已提交
728
    tsdbInfo("vgId:%d, vnode snapshot tsdb read del data for %s, suid:%" PRId64 " uid:%" PRId64 " size:%d",
C
Cary Xu 已提交
729
             TD_VID(pTsdb->pVnode), pTsdb->path, pDelIdx->suid, pDelIdx->uid, size);
H
Hongze Cheng 已提交
730 731 732

    break;
  }
H
Hongze Cheng 已提交
733 734

_exit:
H
add log  
Hongze Cheng 已提交
735 736 737
  if (code) {
    tsdbError("vgId:%d, %s failed since %s, path:%s", TD_VID(pTsdb->pVnode), __func__, tstrerror(code), pTsdb->path);
  }
H
Hongze Cheng 已提交
738 739
  return code;
}
H
more  
Hongze Cheng 已提交
740

C
Cary Xu 已提交
741
int32_t tsdbSnapReaderOpen(STsdb* pTsdb, int64_t sver, int64_t ever, int8_t type, STsdbSnapReader** ppReader) {
H
Hongze Cheng 已提交
742
  int32_t          code = 0;
H
Hongze Cheng 已提交
743
  int32_t          lino = 0;
H
Hongze Cheng 已提交
744
  STsdbSnapReader* pReader = NULL;
H
Hongze Cheng 已提交
745

H
more  
Hongze Cheng 已提交
746
  // alloc
H
Hongze Cheng 已提交
747
  pReader = (STsdbSnapReader*)taosMemoryCalloc(1, sizeof(*pReader));
H
more  
Hongze Cheng 已提交
748 749
  if (pReader == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
750
    TSDB_CHECK_CODE(code, lino, _exit);
H
more  
Hongze Cheng 已提交
751 752 753 754
  }
  pReader->pTsdb = pTsdb;
  pReader->sver = sver;
  pReader->ever = ever;
C
Cary Xu 已提交
755
  pReader->type = type;
H
more  
Hongze Cheng 已提交
756

H
Hongze Cheng 已提交
757 758 759
  code = taosThreadRwlockRdlock(&pTsdb->rwLock);
  if (code) {
    code = TAOS_SYSTEM_ERROR(code);
H
Hongze Cheng 已提交
760
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
761 762 763 764 765
  }

  code = tsdbFSRef(pTsdb, &pReader->fs);
  if (code) {
    taosThreadRwlockUnlock(&pTsdb->rwLock);
H
Hongze Cheng 已提交
766
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
767 768 769 770 771
  }

  code = taosThreadRwlockUnlock(&pTsdb->rwLock);
  if (code) {
    code = TAOS_SYSTEM_ERROR(code);
H
Hongze Cheng 已提交
772
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
773 774
  }

H
Hongze Cheng 已提交
775
  // data
H
Hongze Cheng 已提交
776
  pReader->fid = INT32_MIN;
H
Hongze Cheng 已提交
777 778 779 780 781 782 783
  for (int32_t iIter = 0; iIter < sizeof(pReader->aFDataIter) / sizeof(pReader->aFDataIter[0]); iIter++) {
    SFDataIter* pIter = &pReader->aFDataIter[iIter];

    if (iIter == 0) {
      pIter->aBlockIdx = taosArrayInit(0, sizeof(SBlockIdx));
      if (pIter->aBlockIdx == NULL) {
        code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
784
        TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
785 786 787 788 789
      }
    } else {
      pIter->aSttBlk = taosArrayInit(0, sizeof(SSttBlk));
      if (pIter->aSttBlk == NULL) {
        code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
790
        TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
791 792 793 794
      }
    }

    code = tBlockDataCreate(&pIter->bData);
H
Hongze Cheng 已提交
795
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
796
  }
H
Hongze Cheng 已提交
797 798

  code = tBlockDataCreate(&pReader->bData);
H
Hongze Cheng 已提交
799
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
800

H
Hongze Cheng 已提交
801
  // del
H
Hongze Cheng 已提交
802 803 804
  pReader->aDelIdx = taosArrayInit(0, sizeof(SDelIdx));
  if (pReader->aDelIdx == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
805
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
806 807 808 809
  }
  pReader->aDelData = taosArrayInit(0, sizeof(SDelData));
  if (pReader->aDelData == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
810
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
811 812
  }

H
Hongze Cheng 已提交
813 814
_exit:
  if (code) {
S
Shengliang Guan 已提交
815
    tsdbError("vgId:%d, %s failed at line %d since %s, TSDB path: %s", TD_VID(pTsdb->pVnode), __func__, lino,
H
Hongze Cheng 已提交
816
              tstrerror(code), pTsdb->path);
H
Hongze Cheng 已提交
817 818 819 820 821 822 823 824 825 826 827 828 829
    *ppReader = NULL;

    if (pReader) {
      taosArrayDestroy(pReader->aDelData);
      taosArrayDestroy(pReader->aDelIdx);
      tBlockDataDestroy(&pReader->bData, 1);
      tsdbFSDestroy(&pReader->fs);
      taosMemoryFree(pReader);
    }
  } else {
    *ppReader = pReader;
    tsdbInfo("vgId:%d, vnode snapshot tsdb reader opened for %s", TD_VID(pTsdb->pVnode), pTsdb->path);
  }
H
more  
Hongze Cheng 已提交
830
  return code;
H
Hongze Cheng 已提交
831 832
}

H
Hongze Cheng 已提交
833
int32_t tsdbSnapReaderClose(STsdbSnapReader** ppReader) {
H
Hongze Cheng 已提交
834
  int32_t          code = 0;
H
Hongze Cheng 已提交
835 836
  STsdbSnapReader* pReader = *ppReader;

H
Hongze Cheng 已提交
837 838 839 840 841 842 843 844 845 846 847 848 849
  // data
  if (pReader->pDataFReader) tsdbDataFReaderClose(&pReader->pDataFReader);
  for (int32_t iIter = 0; iIter < sizeof(pReader->aFDataIter) / sizeof(pReader->aFDataIter[0]); iIter++) {
    SFDataIter* pIter = &pReader->aFDataIter[iIter];

    if (iIter == 0) {
      taosArrayDestroy(pIter->aBlockIdx);
      tMapDataClear(&pIter->mBlock);
    } else {
      taosArrayDestroy(pIter->aSttBlk);
    }

    tBlockDataDestroy(&pIter->bData, 1);
H
Hongze Cheng 已提交
850
  }
H
Hongze Cheng 已提交
851 852

  tBlockDataDestroy(&pReader->bData, 1);
853
  tDestroyTSchema(pReader->skmTable.pTSchema);
H
Hongze Cheng 已提交
854 855 856

  // del
  if (pReader->pDelFReader) tsdbDelFReaderClose(&pReader->pDelFReader);
H
Hongze Cheng 已提交
857 858 859
  taosArrayDestroy(pReader->aDelIdx);
  taosArrayDestroy(pReader->aDelData);

H
Hongze Cheng 已提交
860 861
  tsdbFSUnref(pReader->pTsdb, &pReader->fs);

S
Shengliang Guan 已提交
862
  tsdbInfo("vgId:%d, vnode snapshot tsdb reader closed for %s", TD_VID(pReader->pTsdb->pVnode), pReader->pTsdb->path);
H
Hongze Cheng 已提交
863

H
Hongze Cheng 已提交
864 865 866 867
  for (int32_t iBuf = 0; iBuf < sizeof(pReader->aBuf) / sizeof(pReader->aBuf[0]); iBuf++) {
    tFree(pReader->aBuf[iBuf]);
  }

H
Hongze Cheng 已提交
868 869
  taosMemoryFree(pReader);
  *ppReader = NULL;
H
Hongze Cheng 已提交
870 871 872 873
  return code;
}

int32_t tsdbSnapRead(STsdbSnapReader* pReader, uint8_t** ppData) {
H
more  
Hongze Cheng 已提交
874
  int32_t code = 0;
H
add log  
Hongze Cheng 已提交
875
  int32_t lino = 0;
H
Hongze Cheng 已提交
876

H
Hongze Cheng 已提交
877 878
  *ppData = NULL;

H
Hongze Cheng 已提交
879
  // read data file
H
Hongze Cheng 已提交
880 881
  if (!pReader->dataDone) {
    code = tsdbSnapReadData(pReader, ppData);
H
add log  
Hongze Cheng 已提交
882 883 884
    TSDB_CHECK_CODE(code, lino, _exit);
    if (*ppData) {
      goto _exit;
H
Hongze Cheng 已提交
885
    } else {
H
add log  
Hongze Cheng 已提交
886
      pReader->dataDone = 1;
H
Hongze Cheng 已提交
887 888
    }
  }
H
Hongze Cheng 已提交
889 890

  // read del file
H
Hongze Cheng 已提交
891 892
  if (!pReader->delDone) {
    code = tsdbSnapReadDel(pReader, ppData);
H
add log  
Hongze Cheng 已提交
893 894 895
    TSDB_CHECK_CODE(code, lino, _exit);
    if (*ppData) {
      goto _exit;
H
Hongze Cheng 已提交
896
    } else {
H
add log  
Hongze Cheng 已提交
897
      pReader->delDone = 1;
H
Hongze Cheng 已提交
898 899
    }
  }
H
Hongze Cheng 已提交
900

H
Hongze Cheng 已提交
901
_exit:
H
add log  
Hongze Cheng 已提交
902 903 904 905 906 907
  if (code) {
    tsdbError("vgId:%d, %s failed since %s, path:%s", TD_VID(pReader->pTsdb->pVnode), __func__, tstrerror(code),
              pReader->pTsdb->path);
  } else {
    tsdbDebug("vgId:%d, %s done, path:%s", TD_VID(pReader->pTsdb->pVnode), __func__, pReader->pTsdb->path);
  }
H
more  
Hongze Cheng 已提交
908 909 910
  return code;
}

H
Hongze Cheng 已提交
911
// STsdbSnapWriter ========================================
H
Hongze Cheng 已提交
912
struct STsdbSnapWriter {
H
Hongze Cheng 已提交
913 914 915
  STsdb*   pTsdb;
  int64_t  sver;
  int64_t  ever;
H
Hongze Cheng 已提交
916 917 918 919 920 921
  int32_t  minutes;
  int8_t   precision;
  int32_t  minRow;
  int32_t  maxRow;
  int8_t   cmprAlg;
  int64_t  commitID;
H
Hongze Cheng 已提交
922
  uint8_t* aBuf[5];
H
Hongze Cheng 已提交
923

H
Hongze Cheng 已提交
924
  STsdbFS fs;
H
Hongze Cheng 已提交
925

H
Hongze Cheng 已提交
926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948
  // time-series data
  SBlockData inData;

  int32_t  fid;
  TABLEID  tbid;
  SSkmInfo skmTable;

  /* reader */
  SDataFReader*   pDataFReader;
  STsdbDataIter2* iterList;
  STsdbDataIter2* pDIter;
  STsdbDataIter2* pIter;
  SRBTree         rbt;  // SRBTree<STsdbDataIter2>

  /* writer */
  SDataFWriter* pDataFWriter;
  SArray*       aBlockIdx;
  SMapData      mDataBlk;  // SMapData<SDataBlk>
  SArray*       aSttBlk;   // SArray<SSttBlk>
  SBlockData    bData;
  SBlockData    sData;

  // tombstone data
H
Hongze Cheng 已提交
949
  SDelFReader* pDelFReader;
H
Hongze Cheng 已提交
950
  SDelFWriter* pDelFWriter;
H
Hongze Cheng 已提交
951
  int32_t      iDelIdx;
H
Hongze Cheng 已提交
952
  SArray*      aDelIdxR;
H
Hongze Cheng 已提交
953
  SArray*      aDelData;
H
Hongze Cheng 已提交
954
  SArray*      aDelIdxW;
H
Hongze Cheng 已提交
955 956
};

H
Hongze Cheng 已提交
957
// SNAP_DATA_TSDB
H
Hongze Cheng 已提交
958 959 960
extern int32_t tsdbWriteDataBlock(SDataFWriter* pWriter, SBlockData* pBlockData, SMapData* mDataBlk, int8_t cmprAlg);
extern int32_t tsdbWriteSttBlock(SDataFWriter* pWriter, SBlockData* pBlockData, SArray* aSttBlk, int8_t cmprAlg);

H
Hongze Cheng 已提交
961
static int32_t tsdbSnapNextTableData(STsdbSnapWriter* pWriter) {
H
Hongze Cheng 已提交
962
  int32_t code = 0;
H
Hongze Cheng 已提交
963
  int32_t lino = 0;
H
Hongze Cheng 已提交
964

H
Hongze Cheng 已提交
965
#if 0
966
  ASSERT(pWriter->dReader.iRow >= pWriter->dReader.bData.nRow);
H
Hongze Cheng 已提交
967

H
Hongze Cheng 已提交
968 969 970 971 972 973
  if (pWriter->dReader.iBlockIdx < taosArrayGetSize(pWriter->dReader.aBlockIdx)) {
    pWriter->dReader.pBlockIdx = (SBlockIdx*)taosArrayGet(pWriter->dReader.aBlockIdx, pWriter->dReader.iBlockIdx);

    code = tsdbReadDataBlk(pWriter->dReader.pReader, pWriter->dReader.pBlockIdx, &pWriter->dReader.mDataBlk);
    if (code) goto _exit;

H
Hongze Cheng 已提交
974
    pWriter->dReader.iBlockIdx++;
H
Hongze Cheng 已提交
975 976
  } else {
    pWriter->dReader.pBlockIdx = NULL;
H
Hongze Cheng 已提交
977
    tMapDataReset(&pWriter->dReader.mDataBlk);
H
Hongze Cheng 已提交
978
  }
H
Hongze Cheng 已提交
979 980 981
  pWriter->dReader.iDataBlk = 0;  // point to the next one
  tBlockDataReset(&pWriter->dReader.bData);
  pWriter->dReader.iRow = 0;
H
Hongze Cheng 已提交
982
#endif
H
Hongze Cheng 已提交
983 984

_exit:
H
Hongze Cheng 已提交
985 986 987
  if (code) {
    tsdbError("vgId:%d %s failed at line %d since %s", TD_VID(pWriter->pTsdb->pVnode), __func__, lino, tstrerror(code));
  }
H
Hongze Cheng 已提交
988 989 990 991 992
  return code;
}

static int32_t tsdbSnapWriteCopyData(STsdbSnapWriter* pWriter, TABLEID* pId) {
  int32_t code = 0;
H
Hongze Cheng 已提交
993
  int32_t lino = 0;
H
Hongze Cheng 已提交
994

H
Hongze Cheng 已提交
995
#if 0
H
Hongze Cheng 已提交
996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010
  while (true) {
    if (pWriter->dReader.pBlockIdx == NULL) break;
    if (tTABLEIDCmprFn(pWriter->dReader.pBlockIdx, pId) >= 0) break;

    SBlockIdx blkIdx = *pWriter->dReader.pBlockIdx;
    code = tsdbWriteDataBlk(pWriter->dWriter.pWriter, &pWriter->dReader.mDataBlk, &blkIdx);
    if (code) goto _exit;

    if (taosArrayPush(pWriter->dWriter.aBlockIdx, &blkIdx) == NULL) {
      code = TSDB_CODE_OUT_OF_MEMORY;
      goto _exit;
    }

    code = tsdbSnapNextTableData(pWriter);
    if (code) goto _exit;
H
Hongze Cheng 已提交
1011
  }
H
Hongze Cheng 已提交
1012
#endif
H
Hongze Cheng 已提交
1013

H
Hongze Cheng 已提交
1014
_exit:
H
Hongze Cheng 已提交
1015 1016 1017
  if (code) {
    tsdbError("vgId:%d %s failed at line %d since %s", TD_VID(pWriter->pTsdb->pVnode), __func__, lino, tstrerror(code));
  }
H
Hongze Cheng 已提交
1018 1019 1020
  return code;
}

H
Hongze Cheng 已提交
1021 1022
static int32_t tsdbSnapWriteTableDataStart(STsdbSnapWriter* pWriter, TABLEID* pId) {
  int32_t code = 0;
H
Hongze Cheng 已提交
1023 1024 1025 1026 1027
  int32_t lino = 0;

  if (pWriter->pDIter) {
    STsdbDataIter2* pIter = pWriter->pDIter;
    for (;;) {
H
Hongze Cheng 已提交
1028
      if (pIter->dIter.iBlockIdx >= taosArrayGetSize(pIter->dIter.aBlockIdx)) {
H
Hongze Cheng 已提交
1029 1030 1031 1032
        pWriter->pDIter = NULL;
        break;
      }

H
Hongze Cheng 已提交
1033
      SBlockIdx* pBlockIdx = (SBlockIdx*)taosArrayGet(pIter->dIter.aBlockIdx, pIter->dIter.iBlockIdx);
H
Hongze Cheng 已提交
1034 1035 1036

      int32_t c = tTABLEIDCmprFn(pBlockIdx, pId);
      if (c < 0) {
H
Hongze Cheng 已提交
1037
        pIter->dIter.iBlockIdx++;
H
Hongze Cheng 已提交
1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050

        code = tsdbReadDataBlk(pIter->dIter.pReader, pBlockIdx, &pIter->dIter.mDataBlk);
        TSDB_CHECK_CODE(code, lino, _exit);

        SBlockIdx* pNewBlockIdx = taosArrayReserve(pWriter->aBlockIdx, 1);
        if (pNewBlockIdx == NULL) {
          code == TSDB_CODE_OUT_OF_MEMORY;
          TSDB_CHECK_CODE(code, lino, _exit);
        }

        pNewBlockIdx->suid = pBlockIdx->suid;
        pNewBlockIdx->uid = pBlockIdx->uid;

H
Hongze Cheng 已提交
1051
        code = tsdbWriteDataBlk(pWriter->pDataFWriter, &pIter->dIter.mDataBlk, pNewBlockIdx);
H
Hongze Cheng 已提交
1052 1053
        TSDB_CHECK_CODE(code, lino, _exit);
      } else if (c == 0) {
H
Hongze Cheng 已提交
1054
        pIter->dIter.iBlockIdx++;
H
Hongze Cheng 已提交
1055 1056 1057 1058

        code = tsdbReadDataBlk(pIter->dIter.pReader, pBlockIdx, &pIter->dIter.mDataBlk);
        TSDB_CHECK_CODE(code, lino, _exit);

H
Hongze Cheng 已提交
1059
        pIter->dIter.iDataBlk = 0;
H
Hongze Cheng 已提交
1060 1061 1062 1063 1064 1065 1066

        break;
      } else {
        break;
      }
    }
  }
H
Hongze Cheng 已提交
1067

H
Hongze Cheng 已提交
1068 1069 1070 1071 1072
  pWriter->tbid = pId[0];

  tMapDataReset(&pWriter->mDataBlk);

#if 0
H
Hongze Cheng 已提交
1073
  code = tsdbSnapWriteCopyData(pWriter, pId);
H
Hongze Cheng 已提交
1074
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1075 1076 1077 1078 1079

  pWriter->id.suid = pId->suid;
  pWriter->id.uid = pId->uid;

  code = tsdbUpdateTableSchema(pWriter->pTsdb->pVnode->pMeta, pId->suid, pId->uid, &pWriter->skmTable);
H
Hongze Cheng 已提交
1080
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1081 1082

  tMapDataReset(&pWriter->dWriter.mDataBlk);
H
Hongze Cheng 已提交
1083
  code = tBlockDataInit(&pWriter->dWriter.bData, pId, pWriter->skmTable.pTSchema, NULL, 0);
H
Hongze Cheng 已提交
1084 1085
  TSDB_CHECK_CODE(code, lino, _exit);
#endif
H
Hongze Cheng 已提交
1086

H
Hongze Cheng 已提交
1087 1088 1089 1090
_exit:
  if (code) {
    tsdbError("vgId:%d %s failed at line %d since %s", TD_VID(pWriter->pTsdb->pVnode), __func__, lino, tstrerror(code));
  }
H
Hongze Cheng 已提交
1091 1092 1093 1094 1095
  return code;
}

static int32_t tsdbSnapWriteTableDataEnd(STsdbSnapWriter* pWriter) {
  int32_t code = 0;
H
Hongze Cheng 已提交
1096
  int32_t lino = 0;
H
Hongze Cheng 已提交
1097

H
Hongze Cheng 已提交
1098
  // TODO: do write the row, refer line 1593
H
Hongze Cheng 已提交
1099

H
Hongze Cheng 已提交
1100 1101 1102 1103 1104 1105
  if (pWriter->bData.nRow > 0) {
    if (pWriter->bData.nRow < pWriter->minRow) {
      for (int32_t iRow = 0; iRow < pWriter->bData.nRow; iRow++) {
        code = tBlockDataAppendRow(&pWriter->sData, &tsdbRowFromBlockData(&pWriter->bData, iRow),
                                   pWriter->skmTable.pTSchema, pWriter->tbid.uid);
        TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1106

H
Hongze Cheng 已提交
1107 1108 1109 1110
        if (pWriter->sData.nRow >= pWriter->maxRow) {
          code = tsdbWriteSttBlock(pWriter->pDataFWriter, &pWriter->sData, pWriter->aSttBlk, pWriter->cmprAlg);
          TSDB_CHECK_CODE(code, lino, _exit);
        }
H
Hongze Cheng 已提交
1111 1112
      }

H
Hongze Cheng 已提交
1113 1114 1115 1116
      tBlockDataReset(&pWriter->bData);
    } else {
      code = tsdbWriteDataBlock(pWriter->pDataFWriter, &pWriter->bData, &pWriter->mDataBlk, pWriter->cmprAlg);
      TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1117 1118 1119
    }
  }

H
Hongze Cheng 已提交
1120 1121 1122
  if (pWriter->mDataBlk.nItem) {
    SBlockIdx* pBlockIdx = taosArrayReserve(pWriter->aBlockIdx, 1);
    if (pBlockIdx == NULL) {
H
Hongze Cheng 已提交
1123
      code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
1124
      TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1125 1126
    }

H
Hongze Cheng 已提交
1127 1128 1129
    code = tsdbWriteDataBlk(pWriter->pDataFWriter, &pWriter->mDataBlk, pBlockIdx);
    TSDB_CHECK_CODE(code, lino, _exit);
  }
H
Hongze Cheng 已提交
1130

H
Hongze Cheng 已提交
1131 1132 1133 1134
_exit:
  if (code) {
    tsdbError("vgId:%d %s failed at line %d since %s", TD_VID(pWriter->pTsdb->pVnode), __func__, lino, tstrerror(code));
  }
H
Hongze Cheng 已提交
1135 1136 1137
  return code;
}

H
Hongze Cheng 已提交
1138
static int32_t tsdbSnapWriteFileDataStart(STsdbSnapWriter* pWriter, int32_t fid) {
H
Hongze Cheng 已提交
1139
  int32_t code = 0;
H
Hongze Cheng 已提交
1140 1141 1142
  int32_t lino = 0;

  ASSERT(pWriter->pDataFWriter == NULL && pWriter->fid < fid);
H
Hongze Cheng 已提交
1143

H
Hongze Cheng 已提交
1144
  STsdb* pTsdb = pWriter->pTsdb;
H
Hongze Cheng 已提交
1145 1146

  pWriter->fid = fid;
H
Hongze Cheng 已提交
1147
  pWriter->tbid = (TABLEID){0};
H
Hongze Cheng 已提交
1148 1149
  SDFileSet* pSet = taosArraySearch(pWriter->fs.aDFileSet, &(SDFileSet){.fid = fid}, tDFileSetCmprFn, TD_EQ);

H
Hongze Cheng 已提交
1150 1151 1152 1153 1154 1155
  // open reader
  pWriter->pDataFReader = NULL;
  pWriter->iterList = NULL;
  pWriter->pDIter = NULL;
  pWriter->pIter = NULL;
  tRBTreeCreate(&pWriter->rbt, tsdbDataIterCmprFn);
H
Hongze Cheng 已提交
1156
  if (pSet) {
H
Hongze Cheng 已提交
1157 1158
    code = tsdbDataFReaderOpen(&pWriter->pDataFReader, pTsdb, pSet);
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1159

H
Hongze Cheng 已提交
1160 1161 1162 1163 1164 1165
    code = tsdbOpenDataFileDataIter(pWriter->pDataFReader, &pWriter->pDIter);
    TSDB_CHECK_CODE(code, lino, _exit);
    if (pWriter->pDIter) {
      pWriter->pDIter->next = pWriter->iterList;
      pWriter->iterList = pWriter->pDIter;
    }
H
Hongze Cheng 已提交
1166 1167

    for (int32_t iStt = 0; iStt < pSet->nSttF; iStt++) {
H
Hongze Cheng 已提交
1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181
      code = tsdbOpenSttFileDataIter(pWriter->pDataFReader, iStt, &pWriter->pIter);
      TSDB_CHECK_CODE(code, lino, _exit);

      if (pWriter->pIter) {
        code = tsdbSttFileDataIterNext(pWriter->pIter);
        TSDB_CHECK_CODE(code, lino, _exit);

        // add to tree
        tRBTreePut(&pWriter->rbt, &pWriter->pIter->rbtn);

        // add to list
        pWriter->pIter->next = pWriter->iterList;
        pWriter->iterList = pWriter->pIter;
      }
H
Hongze Cheng 已提交
1182
    }
H
Hongze Cheng 已提交
1183 1184 1185 1186 1187 1188 1189 1190

    pWriter->pIter = NULL;
  }

  // open writer
  SDiskID diskId;
  if (pSet) {
    diskId = pSet->diskId;
H
Hongze Cheng 已提交
1191
  } else {
H
Hongze Cheng 已提交
1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209
    tfsAllocDisk(pTsdb->pVnode->pTfs, 0 /*TODO*/, &diskId);
    tfsMkdirRecurAt(pTsdb->pVnode->pTfs, pTsdb->path, diskId);
  }
  SDFileSet wSet = {.diskId = diskId,
                    .fid = fid,
                    .pHeadF = &(SHeadFile){.commitID = pWriter->commitID},
                    .pDataF = (pSet) ? pSet->pDataF : &(SDataFile){.commitID = pWriter->commitID},
                    .pSmaF = (pSet) ? pSet->pSmaF : &(SSmaFile){.commitID = pWriter->commitID},
                    .nSttF = 1,
                    .aSttF = {&(SSttFile){.commitID = pWriter->commitID}}};
  code = tsdbDataFWriterOpen(&pWriter->pDataFWriter, pTsdb, &wSet);
  TSDB_CHECK_CODE(code, lino, _exit);

  if (pWriter->aBlockIdx) {
    taosArrayClear(pWriter->aBlockIdx);
  } else if ((pWriter->aBlockIdx = taosArrayInit(0, sizeof(SBlockIdx))) == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1210
  }
H
Hongze Cheng 已提交
1211

H
Hongze Cheng 已提交
1212
  tMapDataReset(&pWriter->mDataBlk);
H
Hongze Cheng 已提交
1213

H
Hongze Cheng 已提交
1214 1215 1216 1217 1218 1219
  if (pWriter->aSttBlk) {
    taosArrayClear(pWriter->aSttBlk);
  } else if ((pWriter->aSttBlk = taosArrayInit(0, sizeof(SSttBlk))) == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    TSDB_CHECK_CODE(code, lino, _exit);
  }
H
Hongze Cheng 已提交
1220

H
Hongze Cheng 已提交
1221 1222 1223 1224 1225 1226 1227 1228 1229
  tBlockDataReset(&pWriter->bData);

_exit:
  if (code) {
    tsdbError("vgId:%d %s failed at line %d since %s, fid:%d", TD_VID(pTsdb->pVnode), __func__, lino, tstrerror(code),
              fid);
  } else {
    tsdbDebug("vgId:%d %s done, fid:%d", TD_VID(pTsdb->pVnode), __func__, fid);
  }
H
Hongze Cheng 已提交
1230 1231 1232
  return code;
}

H
Hongze Cheng 已提交
1233
static int32_t tsdbSnapWriteFileDataEnd(STsdbSnapWriter* pWriter) {
H
Hongze Cheng 已提交
1234
  int32_t code = 0;
H
Hongze Cheng 已提交
1235
  int32_t lino = 0;
H
Hongze Cheng 已提交
1236

H
Hongze Cheng 已提交
1237
  ASSERT(pWriter->pDataFWriter);
H
Hongze Cheng 已提交
1238

H
Hongze Cheng 已提交
1239 1240
  code = tsdbSnapWriteTableData(pWriter, NULL /* TODO */);
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1241

H
Hongze Cheng 已提交
1242 1243 1244
  // do file-level updates
  code = tsdbWriteSttBlk(pWriter->pDataFWriter, pWriter->aSttBlk);
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1245

H
Hongze Cheng 已提交
1246 1247
  code = tsdbWriteBlockIdx(pWriter->pDataFWriter, pWriter->aBlockIdx);
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1248

H
Hongze Cheng 已提交
1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260
  code = tsdbUpdateDFileSetHeader(pWriter->pDataFWriter);
  TSDB_CHECK_CODE(code, lino, _exit);

  code = tsdbFSUpsertFSet(&pWriter->fs, &pWriter->pDataFWriter->wSet);
  TSDB_CHECK_CODE(code, lino, _exit);

  code = tsdbDataFWriterClose(&pWriter->pDataFWriter, 1);
  TSDB_CHECK_CODE(code, lino, _exit);

  if (pWriter->pDataFReader) {
    code = tsdbDataFReaderClose(&pWriter->pDataFReader);
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1261 1262
  }

H
Hongze Cheng 已提交
1263 1264
  // TODO: do clear sources
  {}
H
Hongze Cheng 已提交
1265

H
Hongze Cheng 已提交
1266 1267 1268 1269 1270 1271
_exit:
  if (code) {
    tsdbError("vgId:%d %s failed since %s", TD_VID(pWriter->pTsdb->pVnode), __func__, tstrerror(code));
  } else {
    tsdbDebug("vgId:%d %s is done", TD_VID(pWriter->pTsdb->pVnode), __func__);
  }
H
Hongze Cheng 已提交
1272
  return code;
H
Hongze Cheng 已提交
1273 1274
}

H
Hongze Cheng 已提交
1275
static int32_t tsdbSnapWriteNextRow(STsdbSnapWriter* pWriter, SRowInfo** ppRowInfo) {
H
Hongze Cheng 已提交
1276
  int32_t code = 0;
H
Hongze Cheng 已提交
1277
  int32_t lino = 0;
H
Hongze Cheng 已提交
1278

H
Hongze Cheng 已提交
1279 1280 1281
  if (pWriter->pIter) {
    code = tsdbDataIterNext2(pWriter->pIter);
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1282

H
Hongze Cheng 已提交
1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296
    if (pWriter->pIter->rowInfo.suid == 0 && pWriter->pIter->rowInfo.uid == 0) {
      pWriter->pIter = NULL;
    } else {
      SRBTreeNode* pNode = tRBTreeMin(&pWriter->rbt);
      if (pNode) {
        int32_t c = tsdbDataIterCmprFn(&pWriter->pIter->rbtn, pNode);
        if (c > 0) {
          tRBTreePut(&pWriter->rbt, &pWriter->pIter->rbtn);
          pWriter->pIter = NULL;
        } else if (c == 0) {
          ASSERT(0);
        }
      }
    }
H
Hongze Cheng 已提交
1297 1298
  }

H
Hongze Cheng 已提交
1299 1300 1301 1302 1303 1304
  if (pWriter->pIter == NULL) {
    SRBTreeNode* pNode = tRBTreeMin(&pWriter->rbt);
    if (pNode) {
      tRBTreeDrop(&pWriter->rbt, pNode);
      pWriter->pIter = TSDB_RBTN_TO_DATA_ITER(pNode);
    }
H
Hongze Cheng 已提交
1305 1306
  }

H
Hongze Cheng 已提交
1307 1308 1309 1310 1311 1312
  if (ppRowInfo) {
    if (pWriter->pIter) {
      *ppRowInfo = &pWriter->pIter->rowInfo;
    } else {
      *ppRowInfo = NULL;
    }
H
Hongze Cheng 已提交
1313
  }
H
Hongze Cheng 已提交
1314

H
Hongze Cheng 已提交
1315 1316 1317
_exit:
  if (code) {
    tsdbError("vgId:%d %s failed at line %d since %s", TD_VID(pWriter->pTsdb->pVnode), __func__, lino, tstrerror(code));
H
Hongze Cheng 已提交
1318
  }
H
Hongze Cheng 已提交
1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332
  return code;
}

static int32_t tsdbSnapWriteGetRow(STsdbSnapWriter* pWriter, SRowInfo** ppRowInfo) {
  int32_t code = 0;
  int32_t lino = 0;

  if (pWriter->pIter) {
    *ppRowInfo = &pWriter->pIter->rowInfo;
    goto _exit;
  }

  code = tsdbSnapWriteNextRow(pWriter, ppRowInfo);
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1333

H
Hongze Cheng 已提交
1334
_exit:
H
Hongze Cheng 已提交
1335 1336 1337
  if (code) {
    tsdbError("vgId:%d %s failed at line %d since %s", TD_VID(pWriter->pTsdb->pVnode), __func__, lino, tstrerror(code));
  }
H
Hongze Cheng 已提交
1338
  return code;
H
Hongze Cheng 已提交
1339
}
H
Hongze Cheng 已提交
1340

H
Hongze Cheng 已提交
1341 1342 1343
static int32_t tsdbSnapWriteTableRow(STsdbSnapWriter* pWriter, TSDBROW* pRow) {
  int32_t code = 0;
  int32_t lino = 0;
H
Hongze Cheng 已提交
1344

H
Hongze Cheng 已提交
1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358
  code = tBlockDataAppendRow(&pWriter->bData, pRow, NULL, pWriter->tbid.uid);
  TSDB_CHECK_CODE(code, lino, _exit);

  if (pWriter->bData.nRow >= pWriter->maxRow) {
    code = tsdbWriteDataBlock(pWriter->pDataFWriter, &pWriter->bData, &pWriter->mDataBlk, pWriter->cmprAlg);
    TSDB_CHECK_CODE(code, lino, _exit);
  }

_exit:
  if (code) {
    tsdbError("vgId:%d %s failed at line %d since %s", TD_VID(pWriter->pTsdb->pVnode), __func__, lino, tstrerror(code));
  }
  return code;
}
H
Hongze Cheng 已提交
1359

H
Hongze Cheng 已提交
1360 1361 1362
static int32_t tsdbSnapWriteTableData(STsdbSnapWriter* pWriter, SRowInfo* pRowInfo) {
  int32_t code = 0;
  int32_t lino = 0;
H
Hongze Cheng 已提交
1363

H
Hongze Cheng 已提交
1364 1365 1366 1367 1368 1369
  // switch to new table if need
  if (pRowInfo->uid != pWriter->tbid.uid) {
    if (pRowInfo->uid) {
      code = tsdbSnapWriteTableDataEnd(pWriter);
      TSDB_CHECK_CODE(code, lino, _exit);
    }
H
Hongze Cheng 已提交
1370

H
Hongze Cheng 已提交
1371 1372 1373
    code = tsdbSnapWriteTableDataStart(pWriter, (TABLEID*)pRowInfo);
    TSDB_CHECK_CODE(code, lino, _exit);
  }
H
Hongze Cheng 已提交
1374

H
Hongze Cheng 已提交
1375
  // do write the row
H
Hongze Cheng 已提交
1376 1377 1378 1379
  if (pWriter->pDIter == NULL || (pWriter->pDIter->dIter.iRow >= pWriter->pDIter->dIter.bData.nRow &&
                                  pWriter->pDIter->dIter.iDataBlk >= pWriter->pDIter->dIter.mDataBlk.nItem)) {
    code = tsdbSnapWriteTableRow(pWriter, &pRowInfo->row);
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1380 1381
  } else {
    for (;;) {
H
Hongze Cheng 已提交
1382 1383
      while (pWriter->pDIter->dIter.iRow < pWriter->pDIter->dIter.bData.nRow) {
        TSDBROW row = tsdbRowFromBlockData(&pWriter->pDIter->dIter.bData, pWriter->pDIter->dIter.iRow);
H
Hongze Cheng 已提交
1384 1385 1386

        int32_t c = tsdbRowCmprFn(&pRowInfo->row, &row);
        if (c < 0) {
H
Hongze Cheng 已提交
1387 1388 1389
          code = tsdbSnapWriteTableRow(pWriter, &pRowInfo->row);
          TSDB_CHECK_CODE(code, lino, _exit);
          goto _exit;
H
Hongze Cheng 已提交
1390 1391 1392
        } else if (c > 0) {
          code = tsdbSnapWriteTableRow(pWriter, &row);
          TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1393 1394

          pWriter->pDIter->dIter.iRow++;
H
Hongze Cheng 已提交
1395 1396 1397
        } else {
          ASSERT(0);
        }
H
Hongze Cheng 已提交
1398 1399
      }

H
Hongze Cheng 已提交
1400 1401
      while (pWriter->pDIter->dIter.iDataBlk < pWriter->pDIter->dIter.mDataBlk.nItem) {
        SDataBlk dataBlk;
H
Hongze Cheng 已提交
1402
        tMapDataGetItemByIdx(&pWriter->pDIter->dIter.mDataBlk, pWriter->pDIter->dIter.iDataBlk, &dataBlk, tGetDataBlk);
H
Hongze Cheng 已提交
1403 1404 1405 1406

        int32_t c = tDataBlkCmprFn(
            &dataBlk, &(SDataBlk){.minKey = TSDBROW_KEY(&pRowInfo->row), .maxKey = TSDBROW_KEY(&pRowInfo->row)});
        if (c > 0) {
H
Hongze Cheng 已提交
1407 1408 1409
          code = tsdbSnapWriteTableRow(pWriter, &pRowInfo->row);
          TSDB_CHECK_CODE(code, lino, _exit);
          goto _exit;
H
Hongze Cheng 已提交
1410
        } else if (c < 0) {
H
Hongze Cheng 已提交
1411 1412 1413
          code = tsdbWriteDataBlock(pWriter->pDataFWriter, &pWriter->bData, &pWriter->mDataBlk, pWriter->cmprAlg);
          TSDB_CHECK_CODE(code, lino, _exit);

H
Hongze Cheng 已提交
1414
          tMapDataPutItem(&pWriter->pDIter->dIter.mDataBlk, &dataBlk, tPutDataBlk);
H
Hongze Cheng 已提交
1415
          pWriter->pDIter->dIter.iDataBlk++;
H
Hongze Cheng 已提交
1416 1417 1418 1419
        } else {
          code = tsdbReadDataBlockEx(pWriter->pDataFReader, &dataBlk, &pWriter->pDIter->dIter.bData);
          TSDB_CHECK_CODE(code, lino, _exit);

H
Hongze Cheng 已提交
1420 1421
          pWriter->pDIter->dIter.iDataBlk++;
          pWriter->pDIter->dIter.iRow = 0;
H
Hongze Cheng 已提交
1422 1423 1424 1425
          break;
        }
      }
    }
H
Hongze Cheng 已提交
1426 1427
  }

H
Hongze Cheng 已提交
1428 1429 1430 1431
_exit:
  if (code) {
    tsdbError("vgId:%d %s failed at line %d since %s", TD_VID(pWriter->pTsdb->pVnode), __func__, lino, tstrerror(code));
  }
H
Hongze Cheng 已提交
1432
  return code;
H
Hongze Cheng 已提交
1433
}
H
Hongze Cheng 已提交
1434

H
Hongze Cheng 已提交
1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447
static int32_t tsdbSnapWriteTimeSeriesData(STsdbSnapWriter* pWriter, SSnapDataHdr* pHdr) {
  int32_t code = 0;
  int32_t lino = 0;

  code = tDecmprBlockData(pHdr->data, pHdr->size, &pWriter->inData, pWriter->aBuf);
  TSDB_CHECK_CODE(code, lino, _exit);

  ASSERT(pWriter->inData.nRow > 0);

  // switch to new data file if need
  int32_t fid = tsdbKeyFid(pWriter->inData.aTSKEY[0], pWriter->minutes, pWriter->precision);
  if (pWriter->fid != fid) {
    if (pWriter->pDataFWriter) {
H
Hongze Cheng 已提交
1448
      code = tsdbSnapWriteFileDataEnd(pWriter);
H
Hongze Cheng 已提交
1449 1450 1451
      TSDB_CHECK_CODE(code, lino, _exit);
    }

H
Hongze Cheng 已提交
1452
    code = tsdbSnapWriteFileDataStart(pWriter, fid);
H
Hongze Cheng 已提交
1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496
    TSDB_CHECK_CODE(code, lino, _exit);
  }

  // loop write each row
  SRowInfo* pRowInfo;

  code = tsdbSnapWriteGetRow(pWriter, &pRowInfo);
  TSDB_CHECK_CODE(code, lino, _exit);
  for (int32_t iRow = 0; iRow < pWriter->bData.nRow; ++iRow) {
    SRowInfo rInfo = {.suid = pWriter->inData.suid,
                      .uid = pWriter->inData.uid ? pWriter->inData.uid : pWriter->inData.aUid[iRow],
                      .row = tsdbRowFromBlockData(&pWriter->inData, iRow)};

    for (;;) {
      if (pRowInfo == NULL) {
        code = tsdbSnapWriteTableData(pWriter, &rInfo);
        TSDB_CHECK_CODE(code, lino, _exit);
        break;
      } else {
        int32_t c = tRowInfoCmprFn(&rInfo, pRowInfo);
        if (c < 0) {
          code = tsdbSnapWriteTableData(pWriter, &rInfo);
          TSDB_CHECK_CODE(code, lino, _exit);
          break;
        } else if (c > 0) {
          code = tsdbSnapWriteTableData(pWriter, pRowInfo);
          TSDB_CHECK_CODE(code, lino, _exit);

          code = tsdbSnapWriteNextRow(pWriter, &pRowInfo);
          TSDB_CHECK_CODE(code, lino, _exit);
        } else {
          ASSERT(0);
        }
      }
    }
  }

_exit:
  if (code) {
    tsdbError("vgId:%d %s failed at line %d since %s", TD_VID(pWriter->pTsdb->pVnode), __func__, lino, tstrerror(code));
  } else {
    tsdbDebug("vgId:%d %s done, suid:%" PRId64 " uid:%" PRId64 " nRow:%d", TD_VID(pWriter->pTsdb->pVnode), __func__,
              pWriter->inData.suid, pWriter->inData.uid, pWriter->inData.nRow);
  }
H
Hongze Cheng 已提交
1497 1498 1499
  return code;
}

H
Hongze Cheng 已提交
1500
// SNAP_DATA_DEL
H
Hongze Cheng 已提交
1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529
static int32_t tsdbSnapMoveWriteDelData(STsdbSnapWriter* pWriter, TABLEID* pId) {
  int32_t code = 0;

  while (true) {
    if (pWriter->iDelIdx >= taosArrayGetSize(pWriter->aDelIdxR)) break;

    SDelIdx* pDelIdx = (SDelIdx*)taosArrayGet(pWriter->aDelIdxR, pWriter->iDelIdx);

    if (tTABLEIDCmprFn(pDelIdx, pId) >= 0) break;

    code = tsdbReadDelData(pWriter->pDelFReader, pDelIdx, pWriter->aDelData);
    if (code) goto _exit;

    SDelIdx delIdx = *pDelIdx;
    code = tsdbWriteDelData(pWriter->pDelFWriter, pWriter->aDelData, &delIdx);
    if (code) goto _exit;

    if (taosArrayPush(pWriter->aDelIdxW, &delIdx) == NULL) {
      code = TSDB_CODE_OUT_OF_MEMORY;
      goto _exit;
    }

    pWriter->iDelIdx++;
  }

_exit:
  return code;
}

H
Hongze Cheng 已提交
1530
static int32_t tsdbSnapWriteDelData(STsdbSnapWriter* pWriter, SSnapDataHdr* pHdr) {
H
Hongze Cheng 已提交
1531
  int32_t code = 0;
H
Hongze Cheng 已提交
1532 1533
  STsdb*  pTsdb = pWriter->pTsdb;

H
Hongze Cheng 已提交
1534
  // Open del file if not opened yet
H
Hongze Cheng 已提交
1535
  if (pWriter->pDelFWriter == NULL) {
H
Hongze Cheng 已提交
1536
    SDelFile* pDelFile = pWriter->fs.pDelFile;
H
Hongze Cheng 已提交
1537 1538 1539

    // reader
    if (pDelFile) {
H
Hongze Cheng 已提交
1540
      code = tsdbDelFReaderOpen(&pWriter->pDelFReader, pDelFile, pTsdb);
H
Hongze Cheng 已提交
1541 1542
      if (code) goto _err;

H
Hongze Cheng 已提交
1543
      code = tsdbReadDelIdx(pWriter->pDelFReader, pWriter->aDelIdxR);
H
Hongze Cheng 已提交
1544
      if (code) goto _err;
H
Hongze Cheng 已提交
1545 1546
    } else {
      taosArrayClear(pWriter->aDelIdxR);
H
Hongze Cheng 已提交
1547
    }
H
Hongze Cheng 已提交
1548
    pWriter->iDelIdx = 0;
H
Hongze Cheng 已提交
1549 1550

    // writer
H
Hongze Cheng 已提交
1551
    SDelFile delFile = {.commitID = pWriter->commitID};
H
Hongze Cheng 已提交
1552 1553
    code = tsdbDelFWriterOpen(&pWriter->pDelFWriter, &delFile, pTsdb);
    if (code) goto _err;
H
Hongze Cheng 已提交
1554
    taosArrayClear(pWriter->aDelIdxW);
H
Hongze Cheng 已提交
1555 1556
  }

H
Hongze Cheng 已提交
1557
  TABLEID id = *(TABLEID*)pHdr->data;
H
Hongze Cheng 已提交
1558

H
Hongze Cheng 已提交
1559 1560 1561
  // Move write data < id
  code = tsdbSnapMoveWriteDelData(pWriter, &id);
  if (code) goto _err;
H
Hongze Cheng 已提交
1562

H
Hongze Cheng 已提交
1563
  // Merge incoming data with current
H
Hongze Cheng 已提交
1564 1565 1566
  if (pWriter->iDelIdx < taosArrayGetSize(pWriter->aDelIdxR) &&
      tTABLEIDCmprFn(taosArrayGet(pWriter->aDelIdxR, pWriter->iDelIdx), &id) == 0) {
    SDelIdx* pDelIdx = (SDelIdx*)taosArrayGet(pWriter->aDelIdxR, pWriter->iDelIdx);
H
Hongze Cheng 已提交
1567

H
Hongze Cheng 已提交
1568
    code = tsdbReadDelData(pWriter->pDelFReader, pDelIdx, pWriter->aDelData);
H
Hongze Cheng 已提交
1569 1570
    if (code) goto _err;

H
Hongze Cheng 已提交
1571 1572 1573 1574 1575
    pWriter->iDelIdx++;
  } else {
    taosArrayClear(pWriter->aDelData);
  }

H
Hongze Cheng 已提交
1576 1577
  int64_t n = sizeof(TABLEID);
  while (n < pHdr->size) {
H
Hongze Cheng 已提交
1578 1579
    SDelData delData;

H
Hongze Cheng 已提交
1580
    n += tGetDelData(pHdr->data + n, &delData);
H
Hongze Cheng 已提交
1581 1582

    if (taosArrayPush(pWriter->aDelData, &delData) == NULL) {
H
Hongze Cheng 已提交
1583 1584 1585
      code = TSDB_CODE_OUT_OF_MEMORY;
      goto _err;
    }
H
Hongze Cheng 已提交
1586
  }
H
Hongze Cheng 已提交
1587

H
Hongze Cheng 已提交
1588 1589 1590 1591 1592 1593 1594
  SDelIdx delIdx = {.suid = id.suid, .uid = id.uid};
  code = tsdbWriteDelData(pWriter->pDelFWriter, pWriter->aDelData, &delIdx);
  if (code) goto _err;

  if (taosArrayPush(pWriter->aDelIdxW, &delIdx) == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
    goto _err;
H
Hongze Cheng 已提交
1595 1596 1597 1598 1599
  }

  return code;

_err:
S
Shengliang Guan 已提交
1600
  tsdbError("vgId:%d, vnode snapshot tsdb write del for %s failed since %s", TD_VID(pTsdb->pVnode), pTsdb->path,
C
Cary Xu 已提交
1601
            tstrerror(code));
H
Hongze Cheng 已提交
1602 1603 1604
  return code;
}

H
Hongze Cheng 已提交
1605 1606
static int32_t tsdbSnapWriteDelEnd(STsdbSnapWriter* pWriter) {
  int32_t code = 0;
H
Hongze Cheng 已提交
1607 1608
  STsdb*  pTsdb = pWriter->pTsdb;

H
Hongze Cheng 已提交
1609
  if (pWriter->pDelFWriter == NULL) return code;
H
Hongze Cheng 已提交
1610

H
Hongze Cheng 已提交
1611 1612 1613
  TABLEID id = {.suid = INT64_MAX, .uid = INT64_MAX};
  code = tsdbSnapMoveWriteDelData(pWriter, &id);
  if (code) goto _err;
H
Hongze Cheng 已提交
1614

H
Hongze Cheng 已提交
1615 1616 1617
  code = tsdbWriteDelIdx(pWriter->pDelFWriter, pWriter->aDelIdxW);
  if (code) goto _err;

H
Hongze Cheng 已提交
1618 1619 1620
  code = tsdbUpdateDelFileHdr(pWriter->pDelFWriter);
  if (code) goto _err;

H
Hongze Cheng 已提交
1621
  code = tsdbFSUpsertDelFile(&pWriter->fs, &pWriter->pDelFWriter->fDel);
H
Hongze Cheng 已提交
1622 1623 1624 1625 1626 1627 1628 1629 1630 1631
  if (code) goto _err;

  code = tsdbDelFWriterClose(&pWriter->pDelFWriter, 1);
  if (code) goto _err;

  if (pWriter->pDelFReader) {
    code = tsdbDelFReaderClose(&pWriter->pDelFReader);
    if (code) goto _err;
  }

S
Shengliang Guan 已提交
1632
  tsdbInfo("vgId:%d, vnode snapshot tsdb write del for %s end", TD_VID(pTsdb->pVnode), pTsdb->path);
H
Hongze Cheng 已提交
1633 1634 1635
  return code;

_err:
S
Shengliang Guan 已提交
1636
  tsdbError("vgId:%d, vnode snapshot tsdb write del end for %s failed since %s", TD_VID(pTsdb->pVnode), pTsdb->path,
C
Cary Xu 已提交
1637
            tstrerror(code));
H
Hongze Cheng 已提交
1638 1639 1640
  return code;
}

H
Hongze Cheng 已提交
1641
// APIs
H
Hongze Cheng 已提交
1642
int32_t tsdbSnapWriterOpen(STsdb* pTsdb, int64_t sver, int64_t ever, STsdbSnapWriter** ppWriter) {
H
Hongze Cheng 已提交
1643 1644
  int32_t code = 0;
  int32_t lino = 0;
H
Hongze Cheng 已提交
1645 1646

  // alloc
H
Hongze Cheng 已提交
1647
  STsdbSnapWriter* pWriter = (STsdbSnapWriter*)taosMemoryCalloc(1, sizeof(*pWriter));
H
Hongze Cheng 已提交
1648 1649
  if (pWriter == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
1650
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1651 1652 1653 1654
  }
  pWriter->pTsdb = pTsdb;
  pWriter->sver = sver;
  pWriter->ever = ever;
H
Hongze Cheng 已提交
1655 1656 1657 1658 1659
  pWriter->minutes = pTsdb->keepCfg.days;
  pWriter->precision = pTsdb->keepCfg.precision;
  pWriter->minRow = pTsdb->pVnode->config.tsdbCfg.minRows;
  pWriter->maxRow = pTsdb->pVnode->config.tsdbCfg.maxRows;
  pWriter->cmprAlg = pTsdb->pVnode->config.tsdbCfg.compression;
H
Hongze Cheng 已提交
1660 1661
  pWriter->commitID = pTsdb->pVnode->state.commitID;

H
Hongze Cheng 已提交
1662 1663 1664
  code = tsdbFSCopy(pTsdb, &pWriter->fs);
  TSDB_CHECK_CODE(code, lino, _exit);

H
Hongze Cheng 已提交
1665
  // SNAP_DATA_TSDB
H
Hongze Cheng 已提交
1666 1667 1668 1669 1670 1671 1672 1673 1674
#if 1
  pWriter->fid = INT32_MIN;

  code = tBlockDataCreate(&pWriter->inData);
  TSDB_CHECK_CODE(code, lino, _exit);

  code = tBlockDataCreate(&pWriter->bData);
  TSDB_CHECK_CODE(code, lino, _exit);
#else
H
Hongze Cheng 已提交
1675
  code = tBlockDataCreate(&pWriter->bData);
H
Hongze Cheng 已提交
1676
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1677

H
Hongze Cheng 已提交
1678 1679 1680 1681 1682
  pWriter->fid = INT32_MIN;
  pWriter->id = (TABLEID){0};
  // Reader
  pWriter->dReader.aBlockIdx = taosArrayInit(0, sizeof(SBlockIdx));
  if (pWriter->dReader.aBlockIdx == NULL) {
H
Hongze Cheng 已提交
1683
    code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
1684
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1685
  }
H
Hongze Cheng 已提交
1686
  code = tBlockDataCreate(&pWriter->dReader.bData);
H
Hongze Cheng 已提交
1687
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1688

H
Hongze Cheng 已提交
1689 1690 1691
  // Writer
  pWriter->dWriter.aBlockIdx = taosArrayInit(0, sizeof(SBlockIdx));
  if (pWriter->dWriter.aBlockIdx == NULL) {
H
Hongze Cheng 已提交
1692
    code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
1693
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1694
  }
H
Hongze Cheng 已提交
1695 1696
  pWriter->dWriter.aSttBlk = taosArrayInit(0, sizeof(SSttBlk));
  if (pWriter->dWriter.aSttBlk == NULL) {
H
Hongze Cheng 已提交
1697
    code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
1698
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1699
  }
H
Hongze Cheng 已提交
1700
  code = tBlockDataCreate(&pWriter->dWriter.bData);
H
Hongze Cheng 已提交
1701
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1702
  code = tBlockDataCreate(&pWriter->dWriter.sData);
H
Hongze Cheng 已提交
1703
  TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1704
#endif
H
Hongze Cheng 已提交
1705

H
Hongze Cheng 已提交
1706
  // SNAP_DATA_DEL
H
Hongze Cheng 已提交
1707 1708 1709
  pWriter->aDelIdxR = taosArrayInit(0, sizeof(SDelIdx));
  if (pWriter->aDelIdxR == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
1710
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1711 1712 1713 1714
  }
  pWriter->aDelData = taosArrayInit(0, sizeof(SDelData));
  if (pWriter->aDelData == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
1715
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1716 1717 1718 1719
  }
  pWriter->aDelIdxW = taosArrayInit(0, sizeof(SDelIdx));
  if (pWriter->aDelIdxW == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
1720
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1721
  }
H
Hongze Cheng 已提交
1722

H
Hongze Cheng 已提交
1723 1724
_exit:
  if (code) {
S
Shengliang Guan 已提交
1725
    tsdbError("vgId:%d, %s failed at line %d since %s", TD_VID(pTsdb->pVnode), __func__, lino, tstrerror(code));
H
Hongze Cheng 已提交
1726 1727 1728
    *ppWriter = NULL;

    if (pWriter) {
H
Hongze Cheng 已提交
1729
#if 0
H
Hongze Cheng 已提交
1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741
      if (pWriter->aDelIdxW) taosArrayDestroy(pWriter->aDelIdxW);
      if (pWriter->aDelData) taosArrayDestroy(pWriter->aDelData);
      if (pWriter->aDelIdxR) taosArrayDestroy(pWriter->aDelIdxR);
      tBlockDataDestroy(&pWriter->dWriter.sData, 1);
      tBlockDataDestroy(&pWriter->dWriter.bData, 1);
      if (pWriter->dWriter.aSttBlk) taosArrayDestroy(pWriter->dWriter.aSttBlk);
      if (pWriter->dWriter.aBlockIdx) taosArrayDestroy(pWriter->dWriter.aBlockIdx);
      tBlockDataDestroy(&pWriter->dReader.bData, 1);
      if (pWriter->dReader.aBlockIdx) taosArrayDestroy(pWriter->dReader.aBlockIdx);
      tBlockDataDestroy(&pWriter->bData, 1);
      tsdbFSDestroy(&pWriter->fs);
      taosMemoryFree(pWriter);
H
Hongze Cheng 已提交
1742
#endif
H
Hongze Cheng 已提交
1743 1744
    }
  } else {
S
Shengliang Guan 已提交
1745
    tsdbInfo("vgId:%d, %s done", TD_VID(pTsdb->pVnode), __func__);
H
Hongze Cheng 已提交
1746 1747
    *ppWriter = pWriter;
  }
H
Hongze Cheng 已提交
1748 1749 1750
  return code;
}

H
Hongze Cheng 已提交
1751 1752
int32_t tsdbSnapWriterPrepareClose(STsdbSnapWriter* pWriter) {
  int32_t code = 0;
H
Hongze Cheng 已提交
1753
  if (pWriter->pDataFWriter) {
H
Hongze Cheng 已提交
1754
    code = tsdbSnapWriteFileDataEnd(pWriter);
H
Hongze Cheng 已提交
1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765
    if (code) goto _exit;
  }

  code = tsdbSnapWriteDelEnd(pWriter);
  if (code) goto _exit;

  code = tsdbFSPrepareCommit(pWriter->pTsdb, &pWriter->fs);
  if (code) goto _exit;

_exit:
  if (code) {
S
Shengliang Guan 已提交
1766
    tsdbError("vgId:%d, %s failed since %s", TD_VID(pWriter->pTsdb->pVnode), __func__, tstrerror(code));
H
Hongze Cheng 已提交
1767 1768 1769 1770
  }
  return code;
}

H
Hongze Cheng 已提交
1771
int32_t tsdbSnapWriterClose(STsdbSnapWriter** ppWriter, int8_t rollback) {
H
Hongze Cheng 已提交
1772
  int32_t          code = 0;
H
Hongze Cheng 已提交
1773
  STsdbSnapWriter* pWriter = *ppWriter;
H
Hongze Cheng 已提交
1774
  STsdb*           pTsdb = pWriter->pTsdb;
H
Hongze Cheng 已提交
1775 1776

  if (rollback) {
H
Hongze Cheng 已提交
1777
    tsdbRollbackCommit(pWriter->pTsdb);
H
Hongze Cheng 已提交
1778
  } else {
H
Hongze Cheng 已提交
1779 1780 1781
    // lock
    taosThreadRwlockWrlock(&pTsdb->rwLock);

H
Hongze Cheng 已提交
1782
    code = tsdbFSCommit(pWriter->pTsdb);
H
Hongze Cheng 已提交
1783 1784 1785 1786 1787 1788 1789
    if (code) {
      taosThreadRwlockUnlock(&pTsdb->rwLock);
      goto _err;
    }

    // unlock
    taosThreadRwlockUnlock(&pTsdb->rwLock);
H
Hongze Cheng 已提交
1790 1791
  }

H
Hongze Cheng 已提交
1792 1793 1794 1795 1796 1797 1798
  // SNAP_DATA_DEL
  taosArrayDestroy(pWriter->aDelIdxW);
  taosArrayDestroy(pWriter->aDelData);
  taosArrayDestroy(pWriter->aDelIdxR);

  // SNAP_DATA_TSDB

H
Hongze Cheng 已提交
1799 1800 1801 1802 1803 1804
  // // Writer
  // tBlockDataDestroy(&pWriter->dWriter.sData, 1);
  // tBlockDataDestroy(&pWriter->dWriter.bData, 1);
  // taosArrayDestroy(pWriter->dWriter.aSttBlk);
  // tMapDataClear(&pWriter->dWriter.mDataBlk);
  // taosArrayDestroy(pWriter->dWriter.aBlockIdx);
H
Hongze Cheng 已提交
1805

H
Hongze Cheng 已提交
1806 1807 1808 1809
  // // Reader
  // tBlockDataDestroy(&pWriter->dReader.bData, 1);
  // tMapDataClear(&pWriter->dReader.mDataBlk);
  // taosArrayDestroy(pWriter->dReader.aBlockIdx);
H
Hongze Cheng 已提交
1810 1811

  tBlockDataDestroy(&pWriter->bData, 1);
1812
  tDestroyTSchema(pWriter->skmTable.pTSchema);
H
Hongze Cheng 已提交
1813

H
Hongze Cheng 已提交
1814 1815 1816
  for (int32_t iBuf = 0; iBuf < sizeof(pWriter->aBuf) / sizeof(uint8_t*); iBuf++) {
    tFree(pWriter->aBuf[iBuf]);
  }
S
Shengliang Guan 已提交
1817
  tsdbInfo("vgId:%d, %s done", TD_VID(pWriter->pTsdb->pVnode), __func__);
H
Hongze Cheng 已提交
1818 1819 1820 1821 1822
  taosMemoryFree(pWriter);
  *ppWriter = NULL;
  return code;

_err:
S
Shengliang Guan 已提交
1823
  tsdbError("vgId:%d, vnode snapshot tsdb writer close for %s failed since %s", TD_VID(pWriter->pTsdb->pVnode),
C
Cary Xu 已提交
1824 1825 1826
            pWriter->pTsdb->path, tstrerror(code));
  taosMemoryFree(pWriter);
  *ppWriter = NULL;
H
Hongze Cheng 已提交
1827 1828 1829
  return code;
}

H
Hongze Cheng 已提交
1830 1831 1832
int32_t tsdbSnapWrite(STsdbSnapWriter* pWriter, SSnapDataHdr* pHdr) {
  int32_t code = 0;
  int32_t lino = 0;
H
Hongze Cheng 已提交
1833

C
Cary Xu 已提交
1834
  if (pHdr->type == SNAP_DATA_TSDB) {
H
Hongze Cheng 已提交
1835 1836
    code = tsdbSnapWriteTimeSeriesData(pWriter, pHdr);
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1837
    goto _exit;
H
Hongze Cheng 已提交
1838
  } else if (pWriter->pDataFWriter) {
H
Hongze Cheng 已提交
1839
    code = tsdbSnapWriteFileDataEnd(pWriter);
H
Hongze Cheng 已提交
1840
    TSDB_CHECK_CODE(code, lino, _exit);
H
Hongze Cheng 已提交
1841 1842
  }

C
Cary Xu 已提交
1843
  if (pHdr->type == SNAP_DATA_DEL) {
H
Hongze Cheng 已提交
1844 1845 1846
    code = tsdbSnapWriteDelData(pWriter, pHdr);
    TSDB_CHECK_CODE(code, lino, _exit);
    goto _exit;
H
Hongze Cheng 已提交
1847 1848
  }

H
Hongze Cheng 已提交
1849
_exit:
H
Hongze Cheng 已提交
1850 1851 1852 1853 1854 1855 1856
  if (code) {
    tsdbError("vgId:%d %s failed at line %d since %s, type:%d index:%" PRId64 " size:%" PRId64,
              TD_VID(pWriter->pTsdb->pVnode), __func__, lino, tstrerror(code), pHdr->type, pHdr->index, pHdr->size);
  } else {
    tsdbDebug("vgId:%d %s done, type:%d index:%" PRId64 " size:%" PRId64, TD_VID(pWriter->pTsdb->pVnode), __func__,
              pHdr->type, pHdr->index, pHdr->size);
  }
H
Hongze Cheng 已提交
1857 1858
  return code;
}