tsdbCommit.c 21.4 KB
Newer Older
H
Hongze Cheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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/>.
 */

H
Hongze Cheng 已提交
16
#include "tsdb.h"
H
Hongze Cheng 已提交
17

H
Hongze Cheng 已提交
18
typedef struct {
H
Hongze Cheng 已提交
19
  STsdb *pTsdb;
H
Hongze Cheng 已提交
20
  /* commit data */
H
Hongze Cheng 已提交
21 22
  int32_t minutes;
  int8_t  precision;
H
Hongze Cheng 已提交
23 24
  int32_t minRow;
  int32_t maxRow;
H
Hongze Cheng 已提交
25
  // --------------
H
Hongze Cheng 已提交
26
  TSKEY   nextKey;  // need to be reset by each table commit
H
Hongze Cheng 已提交
27 28 29
  int32_t commitFid;
  TSKEY   minKey;
  TSKEY   maxKey;
H
Hongze Cheng 已提交
30
  // commit file data
H
Hongze Cheng 已提交
31 32
  SDataFReader *pReader;
  SMapData      oBlockIdx;  // SMapData<SBlockIdx>, read from reader
H
Hongze Cheng 已提交
33
  SMapData      oBlock;     // SMapData<SBlock>, read from reader
H
Hongze Cheng 已提交
34
  SBlockData    bDataO;
H
Hongze Cheng 已提交
35 36
  SDataFWriter *pWriter;
  SMapData      nBlockIdx;  // SMapData<SBlockIdx>, build by committer
H
Hongze Cheng 已提交
37
  SMapData      nBlock;     // SMapData<SBlock>
H
Hongze Cheng 已提交
38
  SBlockData    bDataN;
H
Hongze Cheng 已提交
39
  /* commit del */
H
Hongze Cheng 已提交
40
  SDelFReader *pDelFReader;
H
Hongze Cheng 已提交
41 42
  SMapData     oDelIdxMap;   // SMapData<SDelIdx>, old
  SMapData     oDelDataMap;  // SMapData<SDelData>, old
H
Hongze Cheng 已提交
43
  SDelFWriter *pDelFWriter;
H
Hongze Cheng 已提交
44 45
  SMapData     nDelIdxMap;   // SMapData<SDelIdx>, new
  SMapData     nDelDataMap;  // SMapData<SDelData>, new
H
Hongze Cheng 已提交
46
} SCommitter;
H
refact  
Hongze Cheng 已提交
47

H
Hongze Cheng 已提交
48 49 50 51 52
static int32_t tsdbStartCommit(STsdb *pTsdb, SCommitter *pCommitter);
static int32_t tsdbCommitData(SCommitter *pCommitter);
static int32_t tsdbCommitDel(SCommitter *pCommitter);
static int32_t tsdbCommitCache(SCommitter *pCommitter);
static int32_t tsdbEndCommit(SCommitter *pCommitter, int32_t eno);
H
refact  
Hongze Cheng 已提交
53

H
refact  
Hongze Cheng 已提交
54
int32_t tsdbBegin(STsdb *pTsdb) {
H
Hongze Cheng 已提交
55
  int32_t code = 0;
H
Hongze Cheng 已提交
56

H
Hongze Cheng 已提交
57 58 59
  code = tsdbMemTableCreate(pTsdb, &pTsdb->mem);
  if (code) {
    goto _err;
H
Hongze Cheng 已提交
60 61
  }

H
Hongze Cheng 已提交
62 63 64 65
  return code;

_err:
  return code;
H
Hongze Cheng 已提交
66 67
}

H
more  
Hongze Cheng 已提交
68
int32_t tsdbCommit(STsdb *pTsdb) {
69
  if (!pTsdb) return 0;
H
Hongze Cheng 已提交
70

H
more  
Hongze Cheng 已提交
71
  int32_t    code = 0;
H
Hongze Cheng 已提交
72 73 74 75 76 77 78 79 80
  SCommitter commith;
  SMemTable *pMemTable = pTsdb->mem;

  // check
  if (pMemTable->nRow == 0 && pMemTable->nDel == 0) {  // TODO
    pTsdb->mem = NULL;
    tsdbMemTableDestroy(pMemTable);
    goto _exit;
  }
H
refact  
Hongze Cheng 已提交
81

H
more  
Hongze Cheng 已提交
82
  // start commit
H
more  
Hongze Cheng 已提交
83 84 85
  code = tsdbStartCommit(pTsdb, &commith);
  if (code) {
    goto _err;
H
refact  
Hongze Cheng 已提交
86 87
  }

H
refact  
Hongze Cheng 已提交
88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
  // commit impl
  code = tsdbCommitData(&commith);
  if (code) {
    goto _err;
  }

  code = tsdbCommitDel(&commith);
  if (code) {
    goto _err;
  }

  code = tsdbCommitCache(&commith);
  if (code) {
    goto _err;
  }

  // end commit
H
more  
Hongze Cheng 已提交
105 106 107 108
  code = tsdbEndCommit(&commith, 0);
  if (code) {
    goto _err;
  }
H
refact  
Hongze Cheng 已提交
109

H
Hongze Cheng 已提交
110
_exit:
H
refact  
Hongze Cheng 已提交
111 112 113
  return code;

_err:
H
Hongze Cheng 已提交
114
  tsdbEndCommit(&commith, code);
C
Cary Xu 已提交
115
  tsdbError("vgId:%d, failed to commit since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
H
refact  
Hongze Cheng 已提交
116 117 118
  return code;
}

H
Hongze Cheng 已提交
119
static int32_t tsdbCommitDelStart(SCommitter *pCommitter) {
H
Hongze Cheng 已提交
120 121 122 123 124 125
  int32_t    code = 0;
  STsdb     *pTsdb = pCommitter->pTsdb;
  SMemTable *pMemTable = pTsdb->imem;
  SDelFile  *pDelFileR = NULL;  // TODO
  SDelFile  *pDelFileW = NULL;  // TODO

H
Hongze Cheng 已提交
126 127 128
  tMapDataReset(&pCommitter->oDelIdxMap);
  tMapDataReset(&pCommitter->nDelIdxMap);

H
Hongze Cheng 已提交
129
  // load old
H
Hongze Cheng 已提交
130
  if (pDelFileR) {
H
Hongze Cheng 已提交
131
    code = tsdbDelFReaderOpen(&pCommitter->pDelFReader, pDelFileR, pTsdb, NULL);
H
Hongze Cheng 已提交
132
    if (code) goto _err;
H
Hongze Cheng 已提交
133

H
Hongze Cheng 已提交
134 135
    code = tsdbReadDelIdx(pCommitter->pDelFReader, &pCommitter->oDelIdxMap, NULL);
    if (code) goto _err;
H
Hongze Cheng 已提交
136 137
  }

H
Hongze Cheng 已提交
138
  // prepare new
H
Hongze Cheng 已提交
139
  code = tsdbDelFWriterOpen(&pCommitter->pDelFWriter, pDelFileW, pTsdb);
H
Hongze Cheng 已提交
140
  if (code) goto _err;
H
Hongze Cheng 已提交
141 142 143 144 145 146 147

_exit:
  tsdbDebug("vgId:%d commit del start", TD_VID(pTsdb->pVnode));
  return code;

_err:
  tsdbError("vgId:%d commit del start failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
148 149 150
  return code;
}

H
Hongze Cheng 已提交
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
static int32_t tsdbCommitTableDel(SCommitter *pCommitter, STbData *pTbData, SDelIdx *pDelIdx) {
  int32_t  code = 0;
  SDelData delData;
  SDelOp  *pDelOp;
  tb_uid_t suid;
  tb_uid_t uid;
  SDelIdx  delIdx;  // TODO

  // check no del data, just return
  if (pTbData && pTbData->pHead == NULL) {
    pTbData = NULL;
  }
  if (pTbData == NULL && pDelIdx == NULL) goto _exit;

  // prepare
  if (pTbData) {
    delIdx.suid = pTbData->suid;
    delIdx.uid = pTbData->uid;
  } else {
    delIdx.suid = pDelIdx->suid;
    delIdx.uid = pDelIdx->uid;
  }
  delIdx.minKey = TSKEY_MAX;
  delIdx.maxKey = TSKEY_MIN;
  delIdx.minVersion = INT64_MAX;
H
Hongze Cheng 已提交
176
  delIdx.maxVersion = INT64_MIN;
H
Hongze Cheng 已提交
177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234

  // start
  tMapDataReset(&pCommitter->oDelDataMap);
  tMapDataReset(&pCommitter->nDelDataMap);

  if (pDelIdx) {
    code = tsdbReadDelData(pCommitter->pDelFReader, pDelIdx, &pCommitter->oDelDataMap, NULL);
    if (code) goto _err;
  }

  // disk
  for (int32_t iDelData = 0; iDelData < pCommitter->oDelDataMap.nItem; iDelData++) {
    code = tMapDataGetItemByIdx(&pCommitter->oDelDataMap, iDelData, &delData, tGetDelData);
    if (code) goto _err;

    code = tMapDataPutItem(&pCommitter->nDelDataMap, &delData, tPutDelData);
    if (code) goto _err;

    if (delIdx.minKey > delData.sKey) delIdx.minKey = delData.sKey;
    if (delIdx.maxKey < delData.eKey) delIdx.maxKey = delData.eKey;
    if (delIdx.minVersion > delData.version) delIdx.minVersion = delData.version;
    if (delIdx.maxVersion < delData.version) delIdx.maxVersion = delData.version;
  }

  // memory
  pDelOp = pTbData ? pTbData->pHead : NULL;
  for (; pDelOp; pDelOp = pDelOp->pNext) {
    delData.version = pDelOp->version;
    delData.sKey = pDelOp->sKey;
    delData.eKey = pDelOp->eKey;

    code = tMapDataPutItem(&pCommitter->nDelDataMap, &delData, tPutDelData);
    if (code) goto _err;

    if (delIdx.minKey > delData.sKey) delIdx.minKey = delData.sKey;
    if (delIdx.maxKey < delData.eKey) delIdx.maxKey = delData.eKey;
    if (delIdx.minVersion > delData.version) delIdx.minVersion = delData.version;
    if (delIdx.maxVersion < delData.version) delIdx.maxVersion = delData.version;
  }

  ASSERT(pCommitter->nDelDataMap.nItem > 0);

  // write
  code = tsdbWriteDelData(pCommitter->pDelFWriter, &pCommitter->nDelDataMap, NULL, &delIdx);
  if (code) goto _err;

  // put delIdx
  code = tMapDataPutItem(&pCommitter->nDelIdxMap, &delIdx, tPutDelIdx);
  if (code) goto _err;

_exit:
  return code;

_err:
  tsdbError("vgId:%d commit table del failed since %s", TD_VID(pCommitter->pTsdb->pVnode), tstrerror(code));
  return code;
}

H
Hongze Cheng 已提交
235
static int32_t tsdbCommitDelImpl(SCommitter *pCommitter) {
H
Hongze Cheng 已提交
236 237 238 239 240 241 242 243 244 245 246 247 248
  int32_t    code = 0;
  STsdb     *pTsdb = pCommitter->pTsdb;
  SMemTable *pMemTable = pTsdb->imem;
  int32_t    iDelIdx = 0;
  int32_t    nDelIdx = pCommitter->oDelIdxMap.nItem;
  int32_t    iTbData = 0;
  int32_t    nTbData = taosArrayGetSize(pMemTable->aTbData);
  STbData   *pTbData;
  SDelIdx   *pDelIdx;
  SDelIdx    delIdx;
  int32_t    c;

  ASSERT(nTbData > 0);
H
Hongze Cheng 已提交
249

H
Hongze Cheng 已提交
250 251 252 253 254 255 256 257 258 259 260 261 262 263
  pTbData = (STbData *)taosArrayGetP(pMemTable->aTbData, iTbData);
  if (iDelIdx < nDelIdx) {
    code = tMapDataGetItemByIdx(&pCommitter->oDelIdxMap, iDelIdx, &delIdx, tGetDelIdx);
    if (code) goto _err;
    pDelIdx = &delIdx;
  } else {
    pDelIdx = NULL;
  }

  while (true) {
    if (pTbData == NULL && pDelIdx == NULL) break;

    if (pTbData && pDelIdx) {
      c = tTABLEIDCmprFn(pTbData, pDelIdx);
H
Hongze Cheng 已提交
264
      if (c == 0) {
H
Hongze Cheng 已提交
265
        goto _commit_mem_and_disk_del;
H
Hongze Cheng 已提交
266
      } else if (c < 0) {
H
Hongze Cheng 已提交
267
        goto _commit_mem_del;
H
Hongze Cheng 已提交
268
      } else {
H
Hongze Cheng 已提交
269
        goto _commit_disk_del;
H
Hongze Cheng 已提交
270
      }
H
Hongze Cheng 已提交
271
    } else {
H
Hongze Cheng 已提交
272 273
      if (pTbData) goto _commit_mem_del;
      if (pDelIdx) goto _commit_disk_del;
H
Hongze Cheng 已提交
274 275
    }

H
Hongze Cheng 已提交
276 277 278 279 280 281 282
  _commit_mem_del:
    code = tsdbCommitTableDel(pCommitter, pTbData, NULL);
    if (code) goto _err;
    iTbData++;
    if (iTbData < nTbData) {
      pTbData = (STbData *)taosArrayGetP(pMemTable->aTbData, iTbData);
    } else {
H
Hongze Cheng 已提交
283 284
      pTbData = NULL;
    }
H
Hongze Cheng 已提交
285
    continue;
H
Hongze Cheng 已提交
286

H
Hongze Cheng 已提交
287 288 289 290 291 292 293 294 295 296 297 298
  _commit_disk_del:
    code = tsdbCommitTableDel(pCommitter, NULL, pDelIdx);
    if (code) goto _err;
    iDelIdx++;
    if (iDelIdx < nDelIdx) {
      code = tMapDataGetItemByIdx(&pCommitter->oDelIdxMap, iDelIdx, &delIdx, tGetDelIdx);
      if (code) goto _err;
      pDelIdx = &delIdx;
    } else {
      pDelIdx = NULL;
    }
    continue;
H
Hongze Cheng 已提交
299

H
Hongze Cheng 已提交
300 301
  _commit_mem_and_disk_del:
    code = tsdbCommitTableDel(pCommitter, pTbData, pDelIdx);
H
Hongze Cheng 已提交
302
    if (code) goto _err;
H
Hongze Cheng 已提交
303 304 305 306 307 308 309 310 311 312 313 314 315 316 317
    iTbData++;
    iDelIdx++;
    if (iTbData < nTbData) {
      pTbData = (STbData *)taosArrayGetP(pMemTable->aTbData, iTbData);
    } else {
      pTbData = NULL;
    }
    if (iDelIdx < nDelIdx) {
      code = tMapDataGetItemByIdx(&pCommitter->oDelIdxMap, iDelIdx, &delIdx, tGetDelIdx);
      if (code) goto _err;
      pDelIdx = &delIdx;
    } else {
      pDelIdx = NULL;
    }
    continue;
H
Hongze Cheng 已提交
318 319
  }

H
Hongze Cheng 已提交
320
  return code;
H
Hongze Cheng 已提交
321 322 323 324

_err:
  tsdbError("vgId:%d commit del impl failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
  return code;
H
Hongze Cheng 已提交
325 326 327 328
}

static int32_t tsdbCommitDelEnd(SCommitter *pCommitter) {
  int32_t code = 0;
H
Hongze Cheng 已提交
329

H
Hongze Cheng 已提交
330 331
  code = tsdbWriteDelIdx(pCommitter->pDelFWriter, &pCommitter->nDelIdxMap, NULL);
  if (code) goto _err;
H
Hongze Cheng 已提交
332

H
Hongze Cheng 已提交
333
  code = tsdbUpdateDelFileHdr(pCommitter->pDelFWriter, NULL);
H
Hongze Cheng 已提交
334
  if (code) goto _err;
H
Hongze Cheng 已提交
335 336

  code = tsdbDelFWriterClose(pCommitter->pDelFWriter, 1);
H
Hongze Cheng 已提交
337
  if (code) goto _err;
H
Hongze Cheng 已提交
338 339 340 341 342 343

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

H
Hongze Cheng 已提交
344 345 346
  return code;

_err:
H
Hongze Cheng 已提交
347
  tsdbError("vgId:%d commit del end failed since %s", TD_VID(pCommitter->pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
348 349 350
  return code;
}

H
Hongze Cheng 已提交
351 352
#define ROW_END(pRow, maxKey) (((pRow) == NULL) || ((pRow)->pTSRow->ts > (maxKey)))

H
Hongze Cheng 已提交
353 354 355 356 357
static int32_t tsdbCommitMemoryData(SCommitter *pCommitter, SBlockIdx *pBlockIdx, STbDataIter *pIter, TSDBKEY eKey,
                                    bool toDataOnly) {
  int32_t  code = 0;
  TSDBROW *pRow;
  SBlock   block;  // TODO
H
Hongze Cheng 已提交
358

H
Hongze Cheng 已提交
359 360 361 362 363 364 365 366
  while (true) {
    pRow = tsdbTbDataIterGet(pIter);

    if (pRow == NULL || tsdbKeyCmprFn(&(TSDBKEY){.ts = pRow->pTSRow->ts, .version = pRow->version}, &eKey) > 0) {
      if (pCommitter->bDataN.nRow == 0) {
        break;
      } else {
        goto _write_block_data;
H
Hongze Cheng 已提交
367
      }
H
Hongze Cheng 已提交
368
    }
H
Hongze Cheng 已提交
369

H
Hongze Cheng 已提交
370 371
    code = tsdbBlockDataAppendRow(&pCommitter->bDataN, pRow, NULL /*TODO*/);
    if (code) goto _err;
H
Hongze Cheng 已提交
372

H
Hongze Cheng 已提交
373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412
    if (pCommitter->bDataN.nRow < pCommitter->maxRow * 4 / 5) {
      continue;
    }

  _write_block_data:
    if (!toDataOnly && pCommitter->bDataN.nRow < pCommitter->minKey) {
      block.last = 1;
    } else {
      block.last = 0;
    }

    code = tsdbWriteBlockData(pCommitter->pWriter, &pCommitter->bDataN, NULL, pBlockIdx, &block);
    if (code) goto _err;

    code = tMapDataPutItem(&pCommitter->nBlock, &block, tPutBlock);
    if (code) goto _err;

    block = BLOCK_INIT_VAL;
    tsdbBlockDataReset(&pCommitter->bDataN);
  }

  return code;

_err:
  tsdbError("vgId:%d commit memory data failed since %s", TD_VID(pCommitter->pTsdb->pVnode), tstrerror(code));
  return code;
}

static int32_t tsdbMergeCommitImpl(SCommitter *pCommitter, SBlockIdx *pBlockIdx, STbDataIter *pIter, SBlock *pBlock) {
  int32_t code = 0;
  int32_t nRow = 0;
  SBlock  block = BLOCK_INIT_VAL;

  if (pBlock->last) {
    // load last and merge until {pCommitter->maxKey, INT64_MAX}
  } else {
    // scan pIter, check  how many rows in the block range
    if (pBlock->nRow + nRow <= pCommitter->maxRow) {
      if (pBlock->nSubBlock < TSDB_MAX_SUBBLOCKS) {
        // add as a subblock
H
Hongze Cheng 已提交
413
      } else {
H
Hongze Cheng 已提交
414
        // load the block, merge until pBlock->maxKey
H
Hongze Cheng 已提交
415
      }
H
Hongze Cheng 已提交
416 417 418 419
    } else {
      // load the block, merge until pBlock->maxKey
    }
  }
H
Hongze Cheng 已提交
420

H
Hongze Cheng 已提交
421 422
  return code;
}
H
Hongze Cheng 已提交
423

H
Hongze Cheng 已提交
424 425 426 427 428 429 430
static int32_t tsdbMergeCommit(SCommitter *pCommitter, SBlockIdx *pBlockIdx, STbDataIter *pIter, SBlock *pBlock) {
  int32_t    code = 0;
  TSDBROW   *pRow;
  SBlock     block = BLOCK_INIT_VAL;
  SBlockData bDataN;
  TSDBKEY    key;
  int32_t    c;
H
Hongze Cheng 已提交
431

H
Hongze Cheng 已提交
432 433 434 435 436
  if (pBlock == NULL) {
    key.ts = pCommitter->maxKey;
    key.version = INT64_MAX;
    code = tsdbCommitMemoryData(pCommitter, pBlockIdx, pIter, key, 0);
    if (code) goto _err;
H
Hongze Cheng 已提交
437
  } else if (pBlock->last) {
H
Hongze Cheng 已提交
438 439 440
    // merge
    code = tsdbMergeCommitImpl(pCommitter, pBlockIdx, pIter, pBlock);
    if (code) goto _err;
H
Hongze Cheng 已提交
441
  } else {
H
Hongze Cheng 已提交
442 443 444 445 446 447 448 449 450 451
    // memory
    key.ts = pBlock->info.minKey.ts;
    key.version = pBlock->info.minKey.version - 1;
    code = tsdbCommitMemoryData(pCommitter, pBlockIdx, pIter, key, 1);
    if (code) goto _err;

    // merge or move block
    pRow = tsdbTbDataIterGet(pIter);
    key.ts = pRow->pTSRow->ts;
    key.version = pRow->version;
H
Hongze Cheng 已提交
452

H
Hongze Cheng 已提交
453 454 455 456 457 458 459 460 461 462 463 464
    c = tBlockCmprFn(&(SBlock){.info.maxKey = key, .info.minKey = key}, pBlock);
    if (c > 0) {
      // move block
      code = tMapDataPutItem(&pCommitter->nBlock, pBlock, tPutBlock);
      if (code) goto _err;
    } else if (c == 0) {
      // merge
      code = tsdbMergeCommitImpl(pCommitter, pBlockIdx, pIter, pBlock);
      if (code) goto _err;
    } else {
      ASSERT(0);
    }
H
Hongze Cheng 已提交
465 466 467 468 469 470
  }

  return code;

_err:
  tsdbError("vgId:%d merge commit failed since %s", TD_VID(pCommitter->pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
471
  return code;
H
Hongze Cheng 已提交
472 473
}

H
Hongze Cheng 已提交
474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490
static int32_t tsdbCommitTableData(SCommitter *pCommitter, STbData *pTbData, SBlockIdx *pBlockIdx) {
  int32_t      code = 0;
  STbDataIter  iter;
  STbDataIter *pIter = &iter;
  TSDBROW     *pRow;
  SBlockIdx    blockIdx;  // TODO

  // create iter
  if (pTbData) {
    tsdbTbDataIterOpen(pTbData, &(TSDBKEY){.ts = pCommitter->minKey, .version = 0}, 0, pIter);
  } else {
    pIter = NULL;
  }

  // check
  pRow = tsdbTbDataIterGet(pIter);
  if (ROW_END(pRow, pCommitter->maxKey) && pBlockIdx == NULL) goto _exit;
H
Hongze Cheng 已提交
491 492 493 494 495 496 497 498

  // start ================================
  tMapDataReset(&pCommitter->oBlock);
  tMapDataReset(&pCommitter->nBlock);
  if (pBlockIdx) {
    code = tsdbReadBlock(pCommitter->pReader, pBlockIdx, &pCommitter->oBlock, NULL);
    if (code) goto _err;
  }
H
Hongze Cheng 已提交
499

H
Hongze Cheng 已提交
500
  // impl ===============================
H
Hongze Cheng 已提交
501 502 503 504
  SBlock  block;
  SBlock *pBlock = &block;
  int32_t iBlock = 0;
  int32_t nBlock = pCommitter->oBlock.nItem;
H
Hongze Cheng 已提交
505

H
Hongze Cheng 已提交
506 507 508 509
  // merge
  pRow = tsdbTbDataIterGet(pIter);
  while (!ROW_END(pRow, pCommitter->maxKey) && iBlock < nBlock) {
    tMapDataGetItemByIdx(&pCommitter->oBlock, iBlock, pBlock, tGetBlock);
H
Hongze Cheng 已提交
510
    code = tsdbMergeCommit(pCommitter, &blockIdx, pIter, pBlock);
H
Hongze Cheng 已提交
511 512 513 514
    if (code) goto _err;

    pRow = tsdbTbDataIterGet(pIter);
    iBlock++;
H
Hongze Cheng 已提交
515
  }
H
Hongze Cheng 已提交
516

H
Hongze Cheng 已提交
517 518 519
  // mem
  pRow = tsdbTbDataIterGet(pIter);
  while (!ROW_END(pRow, pCommitter->maxKey)) {
H
Hongze Cheng 已提交
520
    code = tsdbMergeCommit(pCommitter, &blockIdx, pIter, NULL);
H
Hongze Cheng 已提交
521
    if (code) goto _err;
H
Hongze Cheng 已提交
522

H
Hongze Cheng 已提交
523 524
    pRow = tsdbTbDataIterGet(pIter);
  }
H
Hongze Cheng 已提交
525

H
Hongze Cheng 已提交
526 527 528 529
  // disk
  while (iBlock < nBlock) {
    tMapDataGetItemByIdx(&pCommitter->oBlock, iBlock, pBlock, tGetBlock);

H
Hongze Cheng 已提交
530
    code = tsdbMergeCommit(pCommitter, &blockIdx, NULL, pBlock);
H
Hongze Cheng 已提交
531 532 533
    if (code) goto _err;

    iBlock++;
H
Hongze Cheng 已提交
534 535
  }

H
Hongze Cheng 已提交
536 537 538 539 540 541
  // end ===============================
  code = tsdbWriteBlock(pCommitter->pWriter, &pCommitter->nBlock, NULL, &blockIdx);
  if (code) goto _err;

  code = tMapDataPutItem(&pCommitter->nBlockIdx, &blockIdx, tPutBlockIdx);
  if (code) goto _err;
H
Hongze Cheng 已提交
542

H
Hongze Cheng 已提交
543
_exit:
H
Hongze Cheng 已提交
544 545 546 547 548 549 550 551
  pRow = tsdbTbDataIterGet(pIter);
  if (pRow) {
    ASSERT(pRow->pTSRow->ts > pCommitter->maxKey);
    if (pCommitter->nextKey > pRow->pTSRow->ts) {
      pCommitter->nextKey = pRow->pTSRow->ts;
    }
  }

H
Hongze Cheng 已提交
552 553 554
  return code;

_err:
H
Hongze Cheng 已提交
555
  tsdbError("vgId:%d commit Table data failed since %s", TD_VID(pCommitter->pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
556 557 558 559 560 561
  return code;
}

static int32_t tsdbCommitFileDataStart(SCommitter *pCommitter) {
  int32_t    code = 0;
  STsdb     *pTsdb = pCommitter->pTsdb;
H
Hongze Cheng 已提交
562 563
  SDFileSet *pRSet = NULL;  // TODO
  SDFileSet *pWSet = NULL;  // TODO
H
Hongze Cheng 已提交
564

H
Hongze Cheng 已提交
565
  // memory
H
Hongze Cheng 已提交
566
  pCommitter->nextKey = TSKEY_MAX;
H
Hongze Cheng 已提交
567
  tMapDataReset(&pCommitter->oBlockIdx);
H
Hongze Cheng 已提交
568
  tMapDataReset(&pCommitter->oBlock);
H
Hongze Cheng 已提交
569
  tMapDataReset(&pCommitter->nBlockIdx);
H
Hongze Cheng 已提交
570
  tMapDataReset(&pCommitter->nBlock);
H
Hongze Cheng 已提交
571

H
Hongze Cheng 已提交
572
  // load old
H
Hongze Cheng 已提交
573
  if (pRSet) {
H
Hongze Cheng 已提交
574
    code = tsdbDataFReaderOpen(&pCommitter->pReader, pTsdb, pRSet);
H
Hongze Cheng 已提交
575 576
    if (code) goto _err;

H
Hongze Cheng 已提交
577
    code = tsdbReadBlockIdx(pCommitter->pReader, &pCommitter->oBlockIdx, NULL);
H
Hongze Cheng 已提交
578
    if (code) goto _err;
H
Hongze Cheng 已提交
579 580
  }

H
Hongze Cheng 已提交
581
  // create new
H
Hongze Cheng 已提交
582
  code = tsdbDataFWriterOpen(&pCommitter->pWriter, pTsdb, pWSet);
H
Hongze Cheng 已提交
583 584 585 586 587 588
  if (code) goto _err;

_exit:
  return code;

_err:
H
Hongze Cheng 已提交
589
  tsdbError("vgId:%d commit file data start failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
590 591 592 593 594
  return code;
}

static int32_t tsdbCommitFileDataImpl(SCommitter *pCommitter) {
  int32_t    code = 0;
H
Hongze Cheng 已提交
595
  int32_t    c;
H
Hongze Cheng 已提交
596 597 598 599
  STsdb     *pTsdb = pCommitter->pTsdb;
  SMemTable *pMemTable = pTsdb->imem;
  int32_t    iTbData = 0;
  int32_t    nTbData = taosArrayGetSize(pMemTable->aTbData);
H
Hongze Cheng 已提交
600 601
  int32_t    iBlockIdx = 0;
  int32_t    nBlockIdx = pCommitter->oBlockIdx.nItem;
H
Hongze Cheng 已提交
602
  STbData   *pTbData;
H
Hongze Cheng 已提交
603
  SBlockIdx *pBlockIdx;
H
Hongze Cheng 已提交
604
  SBlockIdx  blockIdx;
H
Hongze Cheng 已提交
605

H
Hongze Cheng 已提交
606
  ASSERT(nTbData > 0);
H
Hongze Cheng 已提交
607

H
Hongze Cheng 已提交
608 609
  pTbData = (STbData *)taosArrayGetP(pMemTable->aTbData, iTbData);
  if (iBlockIdx < nBlockIdx) {
H
Hongze Cheng 已提交
610
    pBlockIdx = &blockIdx;
H
Hongze Cheng 已提交
611
    code = tMapDataGetItemByIdx(&pCommitter->oBlockIdx, iBlockIdx, pBlockIdx, tGetBlockIdx);
H
Hongze Cheng 已提交
612
    if (code) goto _err;
H
Hongze Cheng 已提交
613 614
  } else {
    pBlockIdx = NULL;
H
Hongze Cheng 已提交
615 616 617 618
  }

  while (true) {
    if (pTbData == NULL && pBlockIdx == NULL) break;
H
Hongze Cheng 已提交
619 620

    if (pTbData && pBlockIdx) {
H
Hongze Cheng 已提交
621 622
      c = tTABLEIDCmprFn(pTbData, pBlockIdx);

H
Hongze Cheng 已提交
623
      if (c == 0) {
H
Hongze Cheng 已提交
624
        goto _commit_mem_and_disk_data;
H
Hongze Cheng 已提交
625
      } else if (c < 0) {
H
Hongze Cheng 已提交
626
        goto _commit_mem_data;
H
Hongze Cheng 已提交
627
      } else {
H
Hongze Cheng 已提交
628
        goto _commit_disk_data;
H
Hongze Cheng 已提交
629
      }
H
Hongze Cheng 已提交
630 631
    } else if (pTbData) {
      goto _commit_mem_data;
H
Hongze Cheng 已提交
632
    } else {
H
Hongze Cheng 已提交
633
      goto _commit_disk_data;
H
Hongze Cheng 已提交
634 635
    }

H
Hongze Cheng 已提交
636 637 638
  _commit_mem_data:
    code = tsdbCommitTableData(pCommitter, pTbData, NULL);
    if (code) goto _err;
H
Hongze Cheng 已提交
639

H
Hongze Cheng 已提交
640 641 642 643
    iTbData++;
    if (iTbData < nTbData) {
      pTbData = (STbData *)taosArrayGetP(pMemTable->aTbData, iTbData);
    } else {
H
Hongze Cheng 已提交
644
      pTbData = NULL;
H
Hongze Cheng 已提交
645
    }
H
Hongze Cheng 已提交
646
    continue;
H
Hongze Cheng 已提交
647

H
Hongze Cheng 已提交
648 649 650
  _commit_disk_data:
    code = tsdbCommitTableData(pCommitter, NULL, pBlockIdx);
    if (code) goto _err;
H
Hongze Cheng 已提交
651

H
Hongze Cheng 已提交
652 653
    iBlockIdx++;
    if (iBlockIdx < nBlockIdx) {
H
Hongze Cheng 已提交
654
      pBlockIdx = &blockIdx;
H
Hongze Cheng 已提交
655
      code = tMapDataGetItemByIdx(&pCommitter->oBlockIdx, iBlockIdx, pBlockIdx, tGetBlockIdx);
H
Hongze Cheng 已提交
656 657 658 659 660
      if (code) goto _err;
    } else {
      pBlockIdx = NULL;
    }
    continue;
H
Hongze Cheng 已提交
661

H
Hongze Cheng 已提交
662 663
  _commit_mem_and_disk_data:
    code = tsdbCommitTableData(pCommitter, pTbData, pBlockIdx);
H
Hongze Cheng 已提交
664
    if (code) goto _err;
H
Hongze Cheng 已提交
665

H
Hongze Cheng 已提交
666 667 668 669 670 671 672 673
    iTbData++;
    iBlockIdx++;
    if (iTbData < nTbData) {
      pTbData = (STbData *)taosArrayGetP(pMemTable->aTbData, iTbData);
    } else {
      pTbData = NULL;
    }
    if (iBlockIdx < nBlockIdx) {
H
Hongze Cheng 已提交
674
      pBlockIdx = &blockIdx;
H
Hongze Cheng 已提交
675
      code = tMapDataGetItemByIdx(&pCommitter->oBlockIdx, iBlockIdx, pBlockIdx, tGetBlockIdx);
H
Hongze Cheng 已提交
676 677 678 679 680
      if (code) goto _err;
    } else {
      pBlockIdx = NULL;
    }
    continue;
H
Hongze Cheng 已提交
681 682 683 684 685
  }

  return code;

_err:
H
Hongze Cheng 已提交
686
  tsdbError("vgId:%d commit file data impl failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
687 688 689 690 691
  return code;
}

static int32_t tsdbCommitFileDataEnd(SCommitter *pCommitter) {
  int32_t code = 0;
H
Hongze Cheng 已提交
692

H
Hongze Cheng 已提交
693 694
  // write blockIdx
  code = tsdbWriteBlockIdx(pCommitter->pWriter, &pCommitter->nBlockIdx, NULL);
H
Hongze Cheng 已提交
695 696
  if (code) goto _err;

H
Hongze Cheng 已提交
697
  // update file header
H
Hongze Cheng 已提交
698 699 700
  code = tsdbUpdateDFileSetHeader(pCommitter->pWriter, NULL);
  if (code) goto _err;

H
Hongze Cheng 已提交
701 702
  // close and sync
  code = tsdbDataFWriterClose(pCommitter->pWriter, 1);
H
Hongze Cheng 已提交
703 704 705
  if (code) goto _err;

  if (pCommitter->pReader) {
H
Hongze Cheng 已提交
706
    code = tsdbDataFReaderClose(pCommitter->pReader);
H
Hongze Cheng 已提交
707 708 709 710 711 712 713
    goto _err;
  }

_exit:
  return code;

_err:
H
Hongze Cheng 已提交
714
  tsdbError("vgId:%d commit file data end failed since %s", TD_VID(pCommitter->pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
715 716 717
  return code;
}

H
Hongze Cheng 已提交
718
static int32_t tsdbCommitFileData(SCommitter *pCommitter) {
H
Hongze Cheng 已提交
719 720
  int32_t code = 0;

H
Hongze Cheng 已提交
721 722
  // commit file data start
  code = tsdbCommitFileDataStart(pCommitter);
H
Hongze Cheng 已提交
723 724 725 726
  if (code) {
    goto _err;
  }

H
Hongze Cheng 已提交
727 728
  // commit file data impl
  code = tsdbCommitFileDataImpl(pCommitter);
H
Hongze Cheng 已提交
729 730 731 732
  if (code) {
    goto _err;
  }

H
Hongze Cheng 已提交
733 734
  // commit file data end
  code = tsdbCommitFileDataEnd(pCommitter);
H
Hongze Cheng 已提交
735 736 737 738 739 740 741 742 743 744
  if (code) {
    goto _err;
  }

  return code;

_err:
  return code;
}

H
Hongze Cheng 已提交
745 746
// ----------------------------------------------------------------------------
static int32_t tsdbStartCommit(STsdb *pTsdb, SCommitter *pCommitter) {
H
Hongze Cheng 已提交
747
  int32_t code = 0;
H
Hongze Cheng 已提交
748

H
Hongze Cheng 已提交
749 750 751 752 753 754
  memset(pCommitter, 0, sizeof(*pCommitter));
  ASSERT(pTsdb->mem && pTsdb->imem == NULL);
  // lock();
  pTsdb->imem = pTsdb->mem;
  pTsdb->mem = NULL;
  // unlock();
H
Hongze Cheng 已提交
755

H
Hongze Cheng 已提交
756
  pCommitter->pTsdb = pTsdb;
H
Hongze Cheng 已提交
757 758 759 760

  return code;
}

H
Hongze Cheng 已提交
761 762 763 764
static int32_t tsdbCommitData(SCommitter *pCommitter) {
  int32_t    code = 0;
  STsdb     *pTsdb = pCommitter->pTsdb;
  SMemTable *pMemTable = pTsdb->imem;
H
Hongze Cheng 已提交
765

H
Hongze Cheng 已提交
766
  // check
H
Hongze Cheng 已提交
767
  if (pMemTable->nRow == 0) goto _exit;
H
Hongze Cheng 已提交
768

H
Hongze Cheng 已提交
769
  // loop
H
Hongze Cheng 已提交
770
  pCommitter->nextKey = pMemTable->info.minKey.ts;
H
Hongze Cheng 已提交
771 772 773 774 775
  while (pCommitter->nextKey < TSKEY_MAX) {
    pCommitter->commitFid = tsdbKeyFid(pCommitter->nextKey, pCommitter->minutes, pCommitter->precision);
    tsdbFidKeyRange(pCommitter->commitFid, pCommitter->minutes, pCommitter->precision, &pCommitter->minKey,
                    &pCommitter->maxKey);
    code = tsdbCommitFileData(pCommitter);
H
Hongze Cheng 已提交
776
    if (code) goto _err;
H
Hongze Cheng 已提交
777
  }
H
Hongze Cheng 已提交
778

H
Hongze Cheng 已提交
779 780 781
_exit:
  tsdbDebug("vgId:%d commit data done, nRow:%" PRId64, TD_VID(pTsdb->pVnode), pMemTable->nRow);
  return code;
H
Hongze Cheng 已提交
782

H
Hongze Cheng 已提交
783 784 785 786
_err:
  tsdbError("vgId:%d commit data failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
  return code;
}
H
Hongze Cheng 已提交
787

H
Hongze Cheng 已提交
788 789 790 791
static int32_t tsdbCommitDel(SCommitter *pCommitter) {
  int32_t    code = 0;
  STsdb     *pTsdb = pCommitter->pTsdb;
  SMemTable *pMemTable = pTsdb->imem;
H
Hongze Cheng 已提交
792

H
Hongze Cheng 已提交
793 794
  if (pMemTable->nDel == 0) {
    goto _exit;
H
Hongze Cheng 已提交
795
  }
H
Hongze Cheng 已提交
796

H
Hongze Cheng 已提交
797 798 799 800 801
  // start
  code = tsdbCommitDelStart(pCommitter);
  if (code) {
    goto _err;
  }
H
Hongze Cheng 已提交
802

H
Hongze Cheng 已提交
803 804 805 806 807
  // impl
  code = tsdbCommitDelImpl(pCommitter);
  if (code) {
    goto _err;
  }
H
Hongze Cheng 已提交
808

H
Hongze Cheng 已提交
809 810 811 812 813
  // end
  code = tsdbCommitDelEnd(pCommitter);
  if (code) {
    goto _err;
  }
H
Hongze Cheng 已提交
814

H
Hongze Cheng 已提交
815
_exit:
H
Hongze Cheng 已提交
816
  tsdbDebug("vgId:%d commit del done, nDel:%" PRId64, TD_VID(pTsdb->pVnode), pMemTable->nDel);
H
Hongze Cheng 已提交
817 818 819
  return code;

_err:
H
Hongze Cheng 已提交
820
  tsdbError("vgId:%d commit del failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
821
  return code;
H
Hongze Cheng 已提交
822 823 824 825 826 827 828
}

static int32_t tsdbCommitCache(SCommitter *pCommitter) {
  int32_t code = 0;
  // TODO
  return code;
}
H
Hongze Cheng 已提交
829 830 831 832 833 834

static int32_t tsdbEndCommit(SCommitter *pCommitter, int32_t eno) {
  int32_t code = 0;
  // TODO
  return code;
}