tsdbCommit.c 38.6 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
  int64_t commitID;
H
Hongze Cheng 已提交
22 23
  int32_t minutes;
  int8_t  precision;
H
Hongze Cheng 已提交
24 25
  int32_t minRow;
  int32_t maxRow;
H
Hongze Cheng 已提交
26
  int8_t  cmprAlg;
H
Hongze Cheng 已提交
27
  // --------------
H
Hongze Cheng 已提交
28
  TSKEY   nextKey;  // reset by each table commit
H
Hongze Cheng 已提交
29 30 31
  int32_t commitFid;
  TSKEY   minKey;
  TSKEY   maxKey;
H
Hongze Cheng 已提交
32
  // commit file data
H
Hongze Cheng 已提交
33
  SDataFReader *pReader;
H
Hongze Cheng 已提交
34 35 36 37
  SMapData      oBlockIdxMap;  // SMapData<SBlockIdx>, read from reader
  SMapData      oBlockMap;     // SMapData<SBlock>, read from reader
  SBlock        oBlock;
  SBlockData    oBlockData;
H
Hongze Cheng 已提交
38
  SDataFWriter *pWriter;
H
Hongze Cheng 已提交
39 40 41 42
  SMapData      nBlockIdxMap;  // SMapData<SBlockIdx>, build by committer
  SMapData      nBlockMap;     // SMapData<SBlock>
  SBlock        nBlock;
  SBlockData    nBlockData;
H
Hongze Cheng 已提交
43 44 45
  int64_t       suid;
  int64_t       uid;
  STSchema     *pTSchema;
H
Hongze Cheng 已提交
46
  /* commit del */
H
Hongze Cheng 已提交
47
  SDelFReader *pDelFReader;
H
Hongze Cheng 已提交
48 49
  SMapData     oDelIdxMap;   // SMapData<SDelIdx>, old
  SMapData     oDelDataMap;  // SMapData<SDelData>, old
H
Hongze Cheng 已提交
50
  SDelFWriter *pDelFWriter;
H
Hongze Cheng 已提交
51 52
  SMapData     nDelIdxMap;   // SMapData<SDelIdx>, new
  SMapData     nDelDataMap;  // SMapData<SDelData>, new
H
Hongze Cheng 已提交
53
} SCommitter;
H
refact  
Hongze Cheng 已提交
54

H
Hongze Cheng 已提交
55 56 57 58 59
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 已提交
60

H
refact  
Hongze Cheng 已提交
61
int32_t tsdbBegin(STsdb *pTsdb) {
H
Hongze Cheng 已提交
62
  int32_t code = 0;
H
Hongze Cheng 已提交
63

64 65
  if (!pTsdb) return code;

H
Hongze Cheng 已提交
66
  code = tsdbMemTableCreate(pTsdb, &pTsdb->mem);
H
Hongze Cheng 已提交
67
  if (code) goto _err;
H
Hongze Cheng 已提交
68

H
Hongze Cheng 已提交
69 70 71
  return code;

_err:
H
Hongze Cheng 已提交
72
  tsdbError("vgId:%d tsdb begin failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
73
  return code;
H
Hongze Cheng 已提交
74 75
}

H
more  
Hongze Cheng 已提交
76
int32_t tsdbCommit(STsdb *pTsdb) {
77
  if (!pTsdb) return 0;
H
Hongze Cheng 已提交
78

H
more  
Hongze Cheng 已提交
79
  int32_t    code = 0;
H
Hongze Cheng 已提交
80 81 82 83
  SCommitter commith;
  SMemTable *pMemTable = pTsdb->mem;

  // check
H
Hongze Cheng 已提交
84 85
  if (pMemTable->nRow == 0 && pMemTable->nDel == 0) {
    // TODO: lock?
H
Hongze Cheng 已提交
86 87 88 89
    pTsdb->mem = NULL;
    tsdbMemTableDestroy(pMemTable);
    goto _exit;
  }
H
refact  
Hongze Cheng 已提交
90

H
more  
Hongze Cheng 已提交
91
  // start commit
H
more  
Hongze Cheng 已提交
92
  code = tsdbStartCommit(pTsdb, &commith);
H
Hongze Cheng 已提交
93
  if (code) goto _err;
H
refact  
Hongze Cheng 已提交
94

H
refact  
Hongze Cheng 已提交
95 96
  // commit impl
  code = tsdbCommitData(&commith);
H
Hongze Cheng 已提交
97
  if (code) goto _err;
H
refact  
Hongze Cheng 已提交
98 99

  code = tsdbCommitDel(&commith);
H
Hongze Cheng 已提交
100
  if (code) goto _err;
H
refact  
Hongze Cheng 已提交
101 102

  code = tsdbCommitCache(&commith);
H
Hongze Cheng 已提交
103
  if (code) goto _err;
H
refact  
Hongze Cheng 已提交
104 105

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

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

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

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

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

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

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

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

_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 已提交
147 148 149
  return code;
}

H
Hongze Cheng 已提交
150
static int32_t tsdbCommitTableDel(SCommitter *pCommitter, STbData *pTbData, SDelIdx *pDelIdx) {
H
Hongze Cheng 已提交
151
  int32_t   code = 0;
H
Hongze Cheng 已提交
152
  SDelData *pDelData = &(SDelData){};
H
Hongze Cheng 已提交
153 154 155
  tb_uid_t  suid;
  tb_uid_t  uid;
  SDelIdx   delIdx;  // TODO
H
Hongze Cheng 已提交
156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173

  // 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 已提交
174
  delIdx.maxVersion = INT64_MIN;
H
Hongze Cheng 已提交
175 176 177 178 179 180 181 182 183 184 185 186

  // 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++) {
H
Hongze Cheng 已提交
187
    code = tMapDataGetItemByIdx(&pCommitter->oDelDataMap, iDelData, pDelData, tGetDelData);
H
Hongze Cheng 已提交
188 189
    if (code) goto _err;

H
Hongze Cheng 已提交
190
    code = tMapDataPutItem(&pCommitter->nDelDataMap, pDelData, tPutDelData);
H
Hongze Cheng 已提交
191 192
    if (code) goto _err;

H
Hongze Cheng 已提交
193 194 195 196
    if (delIdx.minKey > pDelData->sKey) delIdx.minKey = pDelData->sKey;
    if (delIdx.maxKey < pDelData->eKey) delIdx.maxKey = pDelData->eKey;
    if (delIdx.minVersion > pDelData->version) delIdx.minVersion = pDelData->version;
    if (delIdx.maxVersion < pDelData->version) delIdx.maxVersion = pDelData->version;
H
Hongze Cheng 已提交
197 198 199
  }

  // memory
H
Hongze Cheng 已提交
200 201 202
  pDelData = pTbData ? pTbData->pHead : NULL;
  for (; pDelData; pDelData = pDelData->pNext) {
    code = tMapDataPutItem(&pCommitter->nDelDataMap, pDelData, tPutDelData);
H
Hongze Cheng 已提交
203 204
    if (code) goto _err;

H
Hongze Cheng 已提交
205 206 207 208
    if (delIdx.minKey > pDelData->sKey) delIdx.minKey = pDelData->sKey;
    if (delIdx.maxKey < pDelData->eKey) delIdx.maxKey = pDelData->eKey;
    if (delIdx.minVersion > pDelData->version) delIdx.minVersion = pDelData->version;
    if (delIdx.maxVersion < pDelData->version) delIdx.maxVersion = pDelData->version;
H
Hongze Cheng 已提交
209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228
  }

  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 已提交
229
static int32_t tsdbCommitDelImpl(SCommitter *pCommitter) {
H
Hongze Cheng 已提交
230 231 232 233 234 235 236 237 238 239 240 241 242
  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 已提交
243

H
Hongze Cheng 已提交
244 245 246 247 248 249 250 251 252 253 254 255 256 257
  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 已提交
258
      if (c == 0) {
H
Hongze Cheng 已提交
259
        goto _commit_mem_and_disk_del;
H
Hongze Cheng 已提交
260
      } else if (c < 0) {
H
Hongze Cheng 已提交
261
        goto _commit_mem_del;
H
Hongze Cheng 已提交
262
      } else {
H
Hongze Cheng 已提交
263
        goto _commit_disk_del;
H
Hongze Cheng 已提交
264
      }
H
Hongze Cheng 已提交
265
    } else {
H
Hongze Cheng 已提交
266 267
      if (pTbData) goto _commit_mem_del;
      if (pDelIdx) goto _commit_disk_del;
H
Hongze Cheng 已提交
268 269
    }

H
Hongze Cheng 已提交
270 271 272 273 274 275 276
  _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 已提交
277 278
      pTbData = NULL;
    }
H
Hongze Cheng 已提交
279
    continue;
H
Hongze Cheng 已提交
280

H
Hongze Cheng 已提交
281 282 283 284 285 286 287 288 289 290 291 292
  _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 已提交
293

H
Hongze Cheng 已提交
294 295
  _commit_mem_and_disk_del:
    code = tsdbCommitTableDel(pCommitter, pTbData, pDelIdx);
H
Hongze Cheng 已提交
296
    if (code) goto _err;
H
Hongze Cheng 已提交
297 298 299 300 301 302 303 304 305 306 307 308 309 310 311
    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 已提交
312 313
  }

H
Hongze Cheng 已提交
314
  return code;
H
Hongze Cheng 已提交
315 316 317 318

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

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

H
Hongze Cheng 已提交
324 325
  code = tsdbWriteDelIdx(pCommitter->pDelFWriter, &pCommitter->nDelIdxMap, NULL);
  if (code) goto _err;
H
Hongze Cheng 已提交
326

H
Hongze Cheng 已提交
327
  code = tsdbUpdateDelFileHdr(pCommitter->pDelFWriter, NULL);
H
Hongze Cheng 已提交
328
  if (code) goto _err;
H
Hongze Cheng 已提交
329 330

  code = tsdbDelFWriterClose(pCommitter->pDelFWriter, 1);
H
Hongze Cheng 已提交
331
  if (code) goto _err;
H
Hongze Cheng 已提交
332 333 334 335 336 337

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

H
Hongze Cheng 已提交
338 339 340
  return code;

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

H
Hongze Cheng 已提交
345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376
// static int32_t tsdbGetOverlapRowNumber(STbDataIter *pIter, SBlock *pBlock) {
//   int32_t     nRow = 0;
//   TSDBROW    *pRow;
//   TSDBKEY     key;
//   int32_t     c = 0;
//   STbDataIter iter = *pIter;

//   iter.pRow = NULL;
//   while (true) {
//     pRow = tsdbTbDataIterGet(pIter);

//     if (pRow == NULL) break;
//     key = tsdbRowKey(pRow);

//     c = tBlockCmprFn(&(SBlock){.info.maxKey = key, .info.minKey = key}, pBlock);
//     if (c == 0) {
//       nRow++;
//     } else if (c > 0) {
//       break;
//     } else {
//       ASSERT(0);
//     }
//   }

//   return nRow;
// }

static int32_t tsdbCommitFileDataStart(SCommitter *pCommitter) {
  int32_t    code = 0;
  STsdb     *pTsdb = pCommitter->pTsdb;
  SDFileSet *pRSet = NULL;
  SDFileSet  wSet;
H
Hongze Cheng 已提交
377

H
Hongze Cheng 已提交
378 379
  // memory
  pCommitter->nextKey = TSKEY_MAX;
H
Hongze Cheng 已提交
380

H
Hongze Cheng 已提交
381 382 383 384 385 386 387 388
  // old
  tMapDataReset(&pCommitter->oBlockIdxMap);
  tMapDataReset(&pCommitter->oBlockMap);
  tBlockReset(&pCommitter->oBlock);
  tBlockDataReset(&pCommitter->oBlockData);
  pRSet = tsdbFSStateGetDFileSet(pTsdb->fs->nState, pCommitter->commitFid);
  if (pRSet) {
    code = tsdbDataFReaderOpen(&pCommitter->pReader, pTsdb, pRSet);
H
Hongze Cheng 已提交
389 390
    if (code) goto _err;

H
Hongze Cheng 已提交
391
    code = tsdbReadBlockIdx(pCommitter->pReader, &pCommitter->oBlockIdxMap, NULL);
H
Hongze Cheng 已提交
392
    if (code) goto _err;
H
Hongze Cheng 已提交
393
  }
H
Hongze Cheng 已提交
394

H
Hongze Cheng 已提交
395 396 397 398
  // new
  tMapDataReset(&pCommitter->nBlockIdxMap);
  tMapDataReset(&pCommitter->nBlockMap);
  tBlockReset(&pCommitter->nBlock);
H
Hongze Cheng 已提交
399
  tBlockDataReset(&pCommitter->nBlockData);
H
Hongze Cheng 已提交
400 401 402 403 404 405 406 407 408 409
  if (pRSet) {
    wSet = (SDFileSet){.diskId = pRSet->diskId,
                       .fid = pCommitter->commitFid,
                       .fHead = {.commitID = pCommitter->commitID, .offset = 0, .size = 0},
                       .fData = pRSet->fData,
                       .fLast = {.commitID = pCommitter->commitID, .size = 0},
                       .fSma = pRSet->fSma};
  } else {
    STfs   *pTfs = pTsdb->pVnode->pTfs;
    SDiskID did = {.level = 0, .id = 0};
H
Hongze Cheng 已提交
410

H
Hongze Cheng 已提交
411 412
    // TODO: alloc a new disk
    // tfsAllocDisk(pTfs, 0, &did);
H
Hongze Cheng 已提交
413

H
Hongze Cheng 已提交
414 415
    // create the directory
    tfsMkdirRecurAt(pTfs, pTsdb->path, did);
H
Hongze Cheng 已提交
416

H
Hongze Cheng 已提交
417 418 419 420 421 422
    wSet = (SDFileSet){.diskId = did,
                       .fid = pCommitter->commitFid,
                       .fHead = {.commitID = pCommitter->commitID, .offset = 0, .size = 0},
                       .fData = {.commitID = pCommitter->commitID, .size = 0},
                       .fLast = {.commitID = pCommitter->commitID, .size = 0},
                       .fSma = {.commitID = pCommitter->commitID, .size = 0}};
H
Hongze Cheng 已提交
423
  }
H
Hongze Cheng 已提交
424 425
  code = tsdbDataFWriterOpen(&pCommitter->pWriter, pTsdb, &wSet);
  if (code) goto _err;
H
Hongze Cheng 已提交
426

H
Hongze Cheng 已提交
427
_exit:
H
Hongze Cheng 已提交
428 429 430
  return code;

_err:
H
Hongze Cheng 已提交
431
  tsdbError("vgId:%d commit file data start failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
432
  return code;
H
Hongze Cheng 已提交
433 434
}

H
Hongze Cheng 已提交
435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460
static int32_t tsdbCommitterUpdateSchema(SCommitter *pCommitter, int64_t suid, int64_t uid, int32_t sver) {
  int32_t code = 0;

  if (pCommitter->pTSchema) {
    if (pCommitter->suid == suid) {
      if (suid == 0) {
        if (pCommitter->uid == uid && sver == pCommitter->pTSchema->version) goto _exit;
      } else {
        if (sver == pCommitter->pTSchema->version) goto _exit;
      }
    }
  }

_update_schema:
  pCommitter->suid = suid;
  pCommitter->uid = uid;
  tTSchemaDestroy(pCommitter->pTSchema);
  pCommitter->pTSchema = metaGetTbTSchema(pCommitter->pTsdb->pVnode->pMeta, uid, sver);
  if (pCommitter->pTSchema == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
  }

_exit:
  return code;
}

H
Hongze Cheng 已提交
461
static int32_t tsdbCommitMemoryData(SCommitter *pCommitter, STbData *pTbData) {
H
Hongze Cheng 已提交
462
  int32_t      code = 0;
H
Hongze Cheng 已提交
463 464 465
  STsdb       *pTsdb = pCommitter->pTsdb;
  STbDataIter *pIter = &(STbDataIter){0};
  TSDBKEY      key = {.ts = pCommitter->minKey, .version = VERSION_MIN};
H
Hongze Cheng 已提交
466
  TSDBROW      row;
H
Hongze Cheng 已提交
467 468 469
  TSDBROW     *pRow;

  // create iter
H
Hongze Cheng 已提交
470
  tsdbTbDataIterOpen(pTbData, &key, 0, pIter);
H
Hongze Cheng 已提交
471
  pRow = tsdbTbDataIterGet(pIter);
H
Hongze Cheng 已提交
472

H
Hongze Cheng 已提交
473
  if (pRow == NULL || TSDBROW_TS(pRow) > pCommitter->maxKey) goto _exit;
H
Hongze Cheng 已提交
474

H
Hongze Cheng 已提交
475 476
  // main loop
  SBlockIdx  *pBlockIdx = &(SBlockIdx){.suid = pTbData->suid, .uid = pTbData->uid};
H
Hongze Cheng 已提交
477
  SMapData   *mBlock = &pCommitter->nBlockMap;
H
Hongze Cheng 已提交
478 479
  SBlock     *pBlock = &pCommitter->nBlock;
  SBlockData *pBlockData = &pCommitter->nBlockData;
H
Hongze Cheng 已提交
480
  TSKEY       lastTS;
H
Hongze Cheng 已提交
481

H
Hongze Cheng 已提交
482
  tBlockIdxReset(pBlockIdx);
H
Hongze Cheng 已提交
483
  tMapDataReset(mBlock);
H
Hongze Cheng 已提交
484 485
  tBlockReset(pBlock);
  tBlockDataReset(pBlockData);
H
Hongze Cheng 已提交
486 487 488 489 490 491 492 493 494 495 496
  lastTS = TSKEY_MIN;
  while (1) {
    if (pRow == NULL || TSDBROW_TS(pRow) > pCommitter->maxKey) {
      if (pBlockData->nRow > 0) {
        goto _write_block;
      } else {
        break;
      }
    }

    // update schema
H
Hongze Cheng 已提交
497 498 499
    code = tsdbCommitterUpdateSchema(pCommitter, pTbData->suid, pTbData->uid, TSDBROW_SVERSION(pRow));
    if (code) goto _err;

H
Hongze Cheng 已提交
500
    // append
H
Hongze Cheng 已提交
501
    code = tBlockDataAppendRow(pBlockData, pRow, pCommitter->pTSchema);
H
Hongze Cheng 已提交
502
    if (code) goto _err;
H
Hongze Cheng 已提交
503

H
Hongze Cheng 已提交
504
    // update
H
Hongze Cheng 已提交
505 506 507
    pBlock->minVersion = TMIN(pBlock->minVersion, TSDBROW_VERSION(pRow));
    pBlock->maxVersion = TMAX(pBlock->maxVersion, TSDBROW_VERSION(pRow));
    pBlock->nRow++;
H
Hongze Cheng 已提交
508 509
    if (TSDBROW_TS(pRow) == lastTS) pBlock->hasDup = 1;
    lastTS = TSDBROW_TS(pRow);
H
more  
Hongze Cheng 已提交
510

H
Hongze Cheng 已提交
511
    // next
H
Hongze Cheng 已提交
512
    tsdbTbDataIterNext(pIter);
H
Hongze Cheng 已提交
513
    pRow = tsdbTbDataIterGet(pIter);
H
Hongze Cheng 已提交
514

H
Hongze Cheng 已提交
515 516 517
    // check
    if (pBlockData->nRow >= pCommitter->maxRow * 4 / 5) goto _write_block;
    continue;
H
Hongze Cheng 已提交
518

H
Hongze Cheng 已提交
519
  _write_block:
H
Hongze Cheng 已提交
520 521 522 523
    row = tBlockDataFirstRow(pBlockData);
    if (tsdbKeyCmprFn(&pBlock->minKey, &TSDBROW_KEY(&row)) > 0) pBlock->minKey = TSDBROW_KEY(&row);
    row = tBlockDataLastRow(pBlockData);
    if (tsdbKeyCmprFn(&pBlock->maxKey, &TSDBROW_KEY(&row)) < 0) pBlock->maxKey = TSDBROW_KEY(&row);
H
Hongze Cheng 已提交
524
    pBlock->last = pBlockData->nRow < pCommitter->minRow ? 1 : 0;
H
Hongze Cheng 已提交
525
    code = tsdbWriteBlockData(pCommitter->pWriter, pBlockData, NULL, NULL, pBlockIdx, pBlock, pCommitter->cmprAlg);
H
Hongze Cheng 已提交
526 527
    if (code) goto _err;

H
Hongze Cheng 已提交
528 529
    // Design SMA and write SMA to file

H
Hongze Cheng 已提交
530 531 532 533 534 535 536
    // SBlockIdx
    code = tMapDataPutItem(mBlock, pBlock, tPutBlock);
    if (code) goto _err;
    pBlockIdx->minKey = TMIN(pBlockIdx->minKey, pBlock->minKey.ts);
    pBlockIdx->maxKey = TMAX(pBlockIdx->maxKey, pBlock->maxKey.ts);
    pBlockIdx->minVersion = TMIN(pBlockIdx->minVersion, pBlock->minVersion);
    pBlockIdx->maxVersion = TMAX(pBlockIdx->maxVersion, pBlock->maxVersion);
H
Hongze Cheng 已提交
537 538 539 540

    tBlockReset(pBlock);
    tBlockDataReset(pBlockData);
    lastTS = TSKEY_MIN;
H
more  
Hongze Cheng 已提交
541 542
  }

H
Hongze Cheng 已提交
543 544 545 546 547 548 549
  // write block
  code = tsdbWriteBlock(pCommitter->pWriter, mBlock, NULL, pBlockIdx);
  if (code) goto _err;

  code = tMapDataPutItem(&pCommitter->nBlockIdxMap, pBlockIdx, tPutBlockIdx);
  if (code) goto _err;

H
Hongze Cheng 已提交
550
_exit:
H
Hongze Cheng 已提交
551
  if (pRow) pCommitter->nextKey = TMIN(pCommitter->nextKey, TSDBROW_TS(pRow));
H
Hongze Cheng 已提交
552 553 554
  return code;

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

H
Hongze Cheng 已提交
559 560 561 562
static int32_t tsdbCommitDiskData(SCommitter *pCommitter, SBlockIdx *oBlockIdx) {
  int32_t     code = 0;
  SMapData   *mBlockO = &pCommitter->oBlockMap;
  SBlock     *pBlockO = &pCommitter->oBlock;
H
Hongze Cheng 已提交
563
  SMapData   *mBlockN = &pCommitter->nBlockMap;
H
Hongze Cheng 已提交
564
  SBlock     *pBlockN = &pCommitter->nBlock;
H
Hongze Cheng 已提交
565
  SBlockIdx  *pBlockIdx = &(SBlockIdx){0};
H
Hongze Cheng 已提交
566 567 568 569 570 571 572 573 574 575 576 577 578
  SBlockData *pBlockDataO = &pCommitter->oBlockData;

  // read
  code = tsdbReadBlock(pCommitter->pReader, oBlockIdx, mBlockO, NULL);
  if (code) goto _err;

  // loop to add to new
  tMapDataReset(mBlockN);
  for (int32_t iBlock = 0; iBlock < mBlockO->nItem; iBlock++) {
    tMapDataGetItemByIdx(mBlockO, iBlock, pBlockO, tGetBlock);

    if (pBlockO->last) {
      ASSERT(iBlock == mBlockO->nItem - 1);
H
Hongze Cheng 已提交
579
      code = tsdbReadBlockData(pCommitter->pReader, oBlockIdx, pBlockO, pBlockDataO, NULL, NULL);
H
Hongze Cheng 已提交
580 581 582
      if (code) goto _err;

      tBlockReset(pBlockN);
H
Hongze Cheng 已提交
583 584 585 586 587 588 589
      pBlockN->minKey = pBlockO->minKey;
      pBlockN->maxKey = pBlockO->maxKey;
      pBlockN->minVersion = pBlockO->minVersion;
      pBlockN->maxVersion = pBlockO->maxVersion;
      pBlockN->nRow = pBlockO->nRow;
      pBlockN->last = pBlockO->last;
      pBlockN->hasDup = pBlockO->hasDup;
H
Hongze Cheng 已提交
590
      code = tsdbWriteBlockData(pCommitter->pWriter, pBlockDataO, NULL, NULL, pBlockIdx, pBlockN, pCommitter->cmprAlg);
H
Hongze Cheng 已提交
591 592 593 594 595 596 597 598 599 600 601
      if (code) goto _err;

      code = tMapDataPutItem(mBlockN, pBlockN, tPutBlock);
      if (code) goto _err;
    } else {
      code = tMapDataPutItem(mBlockN, pBlockO, tPutBlock);
      if (code) goto _err;
    }
  }

  // SBlock
H
Hongze Cheng 已提交
602
  *pBlockIdx = *oBlockIdx;
H
Hongze Cheng 已提交
603 604 605 606 607 608 609 610 611 612
  code = tsdbWriteBlock(pCommitter->pWriter, mBlockN, NULL, pBlockIdx);
  if (code) goto _err;

  // SBlockIdx
  code = tMapDataPutItem(&pCommitter->nBlockIdxMap, pBlockIdx, tPutBlockIdx);
  if (code) goto _err;

  return code;

_err:
H
Hongze Cheng 已提交
613
  tsdbError("vgId:%d tsdb commit disk data failed since %s", TD_VID(pCommitter->pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
614 615 616
  return code;
}

H
Hongze Cheng 已提交
617 618
static int32_t tsdbMergeTableData(SCommitter *pCommitter, STbDataIter *pIter, SBlockIdx *pBlockIdx, SBlock *pBlock,
                                  TSDBKEY toKey /*not included*/, int8_t toDataOnly) {
H
Hongze Cheng 已提交
619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694
  int32_t     code = 0;
  SBlockData *pBlockDataFrom = &pCommitter->oBlockData;
  SBlockData *pBlockDataTo = &pCommitter->nBlockData;
  TSDBROW    *pRow1;
  TSDBROW    *pRow2;
  int32_t     c = 0;

  // read SBlockData
  code = tsdbReadBlockData(pCommitter->pReader, pBlockIdx, pBlock, pBlockDataFrom, NULL, NULL);
  if (code) goto _err;

  // loop to merge
  tBlockDataReset(pBlockDataTo);
  pRow1 = tsdbTbDataIterGet(pIter);
  pRow2 = &tsdbRowFromBlockData(pBlockDataFrom, 0);
  ASSERT(tsdbKeyCmprFn(&TSDBROW_KEY(pRow1), &toKey) < 0);
  ASSERT(tsdbKeyCmprFn(&TSDBROW_KEY(pRow2), &toKey) < 0);
  code = tsdbCommitterUpdateSchema(pCommitter, pBlockIdx->suid, pBlockIdx->uid, TSDBROW_SVERSION(pRow1));
  if (code) goto _err;

  while (true) {
    if (pRow1 == NULL && pRow2 == NULL) {
      if (pBlockDataTo->nRow == 0) {
        break;
      } else {
        goto _write_block;
      }
    }

    if (pRow1 && pRow2) {
      c = tsdbRowCmprFn(pRow1, pRow2);
      if (c < 0) {
        code = tBlockDataAppendRow(pBlockDataTo, pRow1, pCommitter->pTSchema);
        if (code) goto _err;

        tsdbTbDataIterNext(pIter);
        pRow1 = tsdbTbDataIterGet(pIter);
        // TODO
      } else if (c > 0) {
        code = tBlockDataAppendRow(pBlockDataTo, pRow2, NULL);
        if (code) goto _err;

        pRow2 = pRow2->iRow + 1 < pBlockDataFrom->nRow ? &tsdbRowFromBlockData(pBlockDataFrom, pRow2->iRow + 1) : NULL;
      } else {
        ASSERT(0);
      }
    } else if (pRow1) {
      code = tBlockDataAppendRow(pBlockDataTo, pRow1, pCommitter->pTSchema);
      tsdbTbDataIterNext(pIter);
      pRow1 = tsdbTbDataIterGet(pIter);
      // TODO
    } else {
      code = tBlockDataAppendRow(pBlockDataTo, pRow2, NULL);
      if (code) goto _err;

      pRow2 = pRow2->iRow + 1 < pBlockDataFrom->nRow ? &tsdbRowFromBlockData(pBlockDataFrom, pRow2->iRow + 1) : NULL;
    }

    if (pBlockDataTo->nRow >= pCommitter->maxRow * 4 / 5) {
      goto _write_block;
    } else {
      continue;
    }

  _write_block:
    tBlockDataReset(pBlockDataTo);
    // TODO
  }

  return code;

_err:
  tsdbError("vgId:%d tsdb merge block and mem failed since %s", TD_VID(pCommitter->pTsdb->pVnode), tstrerror(code));
  return code;
}

H
Hongze Cheng 已提交
695
static int32_t tsdbCommitTableMemData(SCommitter *pCommitter, STbDataIter *pIter, TSDBKEY toKey, int8_t toDataOnly) {
H
Hongze Cheng 已提交
696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731
  int32_t     code = 0;
  TSDBROW    *pRow;
  SBlockData *pBlockData = &pCommitter->nBlockData;

  pRow = tsdbTbDataIterGet(pIter);
  tBlockDataReset(pBlockData);
  while (true) {
    if (pRow == NULL || tsdbKeyCmprFn(&TSDBROW_KEY(pRow), &toKey) >= 0) {
      if (pBlockData->nRow > 0) {
        goto _write_block;
      } else {
        break;
      }
    }

    code = tsdbCommitterUpdateSchema(pCommitter, pIter->pTbData->suid, pIter->pTbData->uid, TSDBROW_SVERSION(pRow));
    if (code) goto _err;

    code = tBlockDataAppendRow(pBlockData, pRow, pCommitter->pTSchema);
    if (code) goto _err;

    tsdbTbDataIterNext(pIter);
    pRow = tsdbTbDataIterGet(pIter);

    if (pBlockData->nRow >= pCommitter->maxRow * 4 / 5) goto _write_block;

  _write_block:
    tBlockDataReset(pBlockData);
  }

  return code;

_err:
  return code;
}

H
Hongze Cheng 已提交
732
static int32_t tsdbCommitTableDiskData(SCommitter *pCommitter, SBlock *pBlock, SBlockIdx *pBlockIdx) {
H
Hongze Cheng 已提交
733
  int32_t code = 0;
H
Hongze Cheng 已提交
734 735

  if (pBlock->last) {
H
Hongze Cheng 已提交
736
    code = tsdbReadBlockData(pCommitter->pReader, pBlockIdx, pBlock, &pCommitter->oBlockData, NULL, NULL);
H
Hongze Cheng 已提交
737 738
    if (code) goto _err;

H
Hongze Cheng 已提交
739 740 741 742 743 744 745 746 747
    tBlockReset(&pCommitter->nBlock);
    pCommitter->nBlock.minKey = pBlock->minKey;
    pCommitter->nBlock.maxKey = pBlock->maxKey;
    pCommitter->nBlock.minVersion = pBlock->minVersion;
    pCommitter->nBlock.nRow = pBlock->nRow;
    pCommitter->nBlock.last = pBlock->last;
    pCommitter->nBlock.hasDup = pBlock->hasDup;
    code = tsdbWriteBlockData(pCommitter->pWriter, &pCommitter->oBlockData, NULL, NULL, pBlockIdx, &pCommitter->nBlock,
                              pCommitter->cmprAlg);
H
Hongze Cheng 已提交
748 749 750 751 752 753 754 755 756 757 758 759 760
    if (code) goto _err;

    code = tMapDataPutItem(&pCommitter->nBlockMap, &pCommitter->nBlock, tPutBlock);
    if (code) goto _err;
  } else {
    code = tMapDataPutItem(&pCommitter->nBlockMap, pBlock, tPutBlock);
    if (code) goto _err;
  }

  return code;

_err:
  tsdbError("vgId:%d tsdb commit table disk data failed since %s", TD_VID(pCommitter->pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
761 762 763
  return code;
}

H
Hongze Cheng 已提交
764 765 766 767 768 769 770 771 772
static int32_t tsdbMergeMemDisk(SCommitter *pCommitter, STbData *pTbData, SBlockIdx *oBlockIdx) {
  int32_t      code = 0;
  STbDataIter *pIter = &(STbDataIter){0};
  TSDBROW     *pRow;

  // create iter
  tsdbTbDataIterOpen(pTbData, &(TSDBKEY){.ts = pCommitter->minKey, .version = VERSION_MIN}, 0, pIter);
  pRow == tsdbTbDataIterGet(pIter);
  if (pRow == NULL || TSDBROW_TS(pRow) > pCommitter->maxKey) {
H
Hongze Cheng 已提交
773 774 775 776 777 778
    code = tsdbCommitDiskData(pCommitter, oBlockIdx);
    if (code) {
      goto _err;
    } else {
      goto _exit;
    }
H
Hongze Cheng 已提交
779 780
  }

H
Hongze Cheng 已提交
781
  // start ==================
H
Hongze Cheng 已提交
782 783 784 785 786
  // read
  code = tsdbReadBlock(pCommitter->pReader, oBlockIdx, &pCommitter->oBlockMap, NULL);
  if (code) goto _err;

  // loop to merge
H
Hongze Cheng 已提交
787 788 789 790 791 792 793 794
  // SBlockData *pBlockData = &pCommitter->nBlockData;
  int32_t iBlock = 0;
  int32_t nBlock = pCommitter->oBlockMap.nItem;
  // SBlock     *pBlockO = &pCommitter->oBlock;
  SBlock *pBlock;
  int32_t c;

  // merge ===================
H
Hongze Cheng 已提交
795
  while (true) {
H
Hongze Cheng 已提交
796
    if ((pRow == NULL || TSDBROW_TS(pRow) > pCommitter->maxKey) && pBlock == NULL) break;
H
Hongze Cheng 已提交
797

H
Hongze Cheng 已提交
798 799 800
    if ((pRow && TSDBROW_TS(pRow) <= pCommitter->maxKey) && pBlock) {
      if (pBlock->last) {
        // merge memory data and disk data to write to .data/.last (todo)
H
Hongze Cheng 已提交
801 802
        code = tsdbMergeTableData(pCommitter, pIter, oBlockIdx, pBlock,
                                  (TSDBKEY){.ts = pCommitter->maxKey + 1, .version = VERSION_MIN}, 0);
H
Hongze Cheng 已提交
803
        if (code) goto _err;
H
Hongze Cheng 已提交
804

H
Hongze Cheng 已提交
805 806 807 808 809 810 811
        pRow = tsdbTbDataIterGet(pIter);
        iBlock++;
      } else {
        c = tBlockCmprFn(&(SBlock){}, pBlock);

        if (c < 0) {
          // commit memory data until pBlock->minKey (not included) only to .data file (todo)
H
Hongze Cheng 已提交
812
          code = tsdbCommitTableMemData(pCommitter, pIter, pBlock->minKey, 1);
H
Hongze Cheng 已提交
813 814 815 816 817
          if (code) goto _err;

          pRow = tsdbTbDataIterGet(pIter);
        } else if (c > 0) {
          // just move the block (todo)
H
Hongze Cheng 已提交
818
          // code = tsdbCommitTableDiskData(pCommitter, pBlock);
H
Hongze Cheng 已提交
819 820 821 822 823 824 825 826 827 828 829
          if (code) goto _err;

          iBlock++;
          // TODO
        } else {
          int64_t nOvlp = 0;  // = tsdbOvlpRows();
          if (nOvlp + pBlock->nRow <= pCommitter->maxRow) {
            // add as a subblock
          } else {
            if (iBlock == nBlock - 1) {
              // merge memory data and disk data to .data/.last file
H
Hongze Cheng 已提交
830 831
              code = tsdbMergeTableData(pCommitter, pIter, oBlockIdx, pBlock,
                                        (TSDBKEY){.ts = pCommitter->maxKey + 1, .version = VERSION_MIN}, 0);
H
Hongze Cheng 已提交
832 833 834
              if (code) goto _err;
            } else {
              // merge memory data and disk data to .data file only until pBlock[1].
H
Hongze Cheng 已提交
835
              code = tsdbMergeTableData(pCommitter, pIter, oBlockIdx, pBlock, (TSDBKEY){0} /*TODO*/, 1);
H
Hongze Cheng 已提交
836 837 838 839 840 841 842 843
            }
          }

          pRow = tsdbTbDataIterGet(pIter);
          iBlock++;
        }
      }
    } else if (pBlock) {
H
Hongze Cheng 已提交
844
      // code = tsdbCommitTableDiskData(pCommitter, pBlock);
H
Hongze Cheng 已提交
845 846 847 848 849 850
      if (code) goto _err;

      iBlock++;
      // next block
    } else {
      // commit only memory data until (pCommitter->maxKey, VERSION_MAX)
H
Hongze Cheng 已提交
851 852
      code =
          tsdbCommitTableMemData(pCommitter, pIter, (TSDBKEY){.ts = pCommitter->maxKey + 1, .version = VERSION_MIN}, 0);
H
Hongze Cheng 已提交
853 854 855 856
      if (code) goto _err;

      pRow = tsdbTbDataIterGet(pIter);
    }
H
Hongze Cheng 已提交
857 858
  }

H
Hongze Cheng 已提交
859
  // end =====================
H
Hongze Cheng 已提交
860 861
  // SBlock
  // code = tsdbWriteBlock(pCommitter->pWriter, &pCommitter->nBlockMap, NULL, pBlockIdx);
H
Hongze Cheng 已提交
862 863 864 865 866 867
  // if (code) goto _err;

  // // SBlockIdx
  // code = tMapDataPutItem(&pCommitter->nBlockIdxMap, pBlockIdx, tPutBlockIdx);
  // if (code) goto _err;

H
Hongze Cheng 已提交
868
_exit:
H
Hongze Cheng 已提交
869 870 871 872
  pRow = tsdbTbDataIterGet(pIter);
  if (pRow) {
    pCommitter->nextKey = TMIN(pCommitter->nextKey, TSDBROW_TS(pRow));
  }
H
Hongze Cheng 已提交
873 874 875 876 877 878 879
  return code;

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

H
Hongze Cheng 已提交
880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896
static int32_t tsdbCommitTableDataEnd(SCommitter *pCommitter, int64_t suid, int64_t uid) {
  int32_t    code = 0;
  SBlockIdx *pBlockIdx = &(SBlockIdx){.suid = suid, .uid = uid};

  code = tsdbWriteBlock(pCommitter->pWriter, &pCommitter->nBlockMap, NULL, pBlockIdx);
  if (code) goto _err;

  code = tMapDataPutItem(&pCommitter->nBlockIdxMap, pBlockIdx, tPutBlockIdx);
  if (code) goto _err;

  return code;

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

H
Hongze Cheng 已提交
897 898 899 900 901 902
static int32_t tsdbCommitTableData(SCommitter *pCommitter, STbData *pTbData, SBlockIdx *pBlockIdx) {
  int32_t      code = 0;
  STbDataIter *pIter = &(STbDataIter){0};
  TSDBROW     *pRow;
  int32_t      iBlock = 0;
  int32_t      nBlock;
H
Hongze Cheng 已提交
903 904
  int64_t      suid;
  int64_t      uid;
H
Hongze Cheng 已提交
905 906 907 908

  if (pTbData) {
    tsdbTbDataIterOpen(pTbData, &(TSDBKEY){.ts = pCommitter->minKey, .version = VERSION_MIN}, 0, pIter);
    pRow = tsdbTbDataIterGet(pIter);
H
Hongze Cheng 已提交
909 910
    suid = pTbData->suid;
    uid = pTbData->uid;
H
Hongze Cheng 已提交
911 912 913 914 915 916 917 918 919 920 921
  } else {
    pIter = NULL;
    pRow = NULL;
  }

  if (pBlockIdx) {
    code = tsdbReadBlock(pCommitter->pReader, pBlockIdx, &pCommitter->oBlockMap, NULL);
    if (code) goto _err;

    nBlock = pCommitter->oBlockMap.nItem;
    ASSERT(nBlock > 0);
H
Hongze Cheng 已提交
922 923
    suid = pBlockIdx->suid;
    uid = pBlockIdx->uid;
H
Hongze Cheng 已提交
924 925 926 927 928 929 930
  } else {
    nBlock = 0;
  }

  if ((pRow == NULL || TSDBROW_TS(pRow) > pCommitter->maxKey) && nBlock == 0) goto _exit;

  // start ===========
H
Hongze Cheng 已提交
931
  tMapDataReset(&pCommitter->nBlockMap);
H
Hongze Cheng 已提交
932
  SBlock *pBlock = &pCommitter->oBlock;
H
Hongze Cheng 已提交
933 934
  int32_t c;

H
Hongze Cheng 已提交
935 936 937 938 939 940
  if (iBlock < nBlock) {
    tMapDataGetItemByIdx(&pCommitter->oBlockMap, iBlock, pBlock, tGetBlock);
  } else {
    pBlock = NULL;
  }

H
Hongze Cheng 已提交
941 942 943 944 945 946
  // merge ===========
  while (true) {
    if (((pRow == NULL) || TSDBROW_TS(pRow) > pCommitter->maxKey) && pBlock == NULL) break;

    if (pRow && TSDBROW_TS(pRow) <= pCommitter->maxKey && pBlock) {
      if (pBlock->last) {
H
Hongze Cheng 已提交
947 948
        code = tsdbMergeTableData(pCommitter, pIter, pBlockIdx, pBlock,
                                  (TSDBKEY){.ts = pCommitter->maxKey + 1, .version = VERSION_MIN}, 0);
H
Hongze Cheng 已提交
949 950 951 952 953 954 955
        if (code) goto _err;

        pRow = tsdbTbDataIterGet(pIter);
        iBlock++;
      } else {
        c = tBlockCmprFn(&(SBlock){}, pBlock);
        if (c > 0) {
H
Hongze Cheng 已提交
956
          code = tsdbCommitTableDiskData(pCommitter, pBlock, pBlockIdx);
H
Hongze Cheng 已提交
957 958 959
          if (code) goto _err;

          iBlock++;
H
Hongze Cheng 已提交
960 961 962 963 964 965
          iBlock++;
          if (iBlock < nBlock) {
            tMapDataGetItemByIdx(&pCommitter->oBlockMap, iBlock, pBlock, tGetBlock);
          } else {
            pBlock = NULL;
          }
H
Hongze Cheng 已提交
966
        } else if (c < 0) {
H
Hongze Cheng 已提交
967
          code = tsdbCommitTableMemData(pCommitter, pIter, pBlock->minKey, 1);
H
Hongze Cheng 已提交
968 969 970 971 972 973 974 975 976
          if (code) goto _err;

          pRow = tsdbTbDataIterGet(pIter);
        } else {
          int64_t nOvlp = 0;  // (todo)
          if (pBlock->nRow + nOvlp <= pCommitter->maxRow && pBlock->nSubBlock < TSDB_MAX_SUBBLOCKS) {
            // add as a subblock
          } else {
            if (iBlock == nBlock - 1) {
H
Hongze Cheng 已提交
977 978
              code = tsdbMergeTableData(pCommitter, pIter, pBlockIdx, pBlock,
                                        (TSDBKEY){.ts = pCommitter->maxKey + 1, .version = VERSION_MIN}, 0);
H
Hongze Cheng 已提交
979 980 981

              if (code) goto _err;
            } else {
H
Hongze Cheng 已提交
982
              // code = tsdbMergeTableData(pCommitter, pIter, pBlockIdx, pBlock, pBlock[1].minKey, 1);
H
Hongze Cheng 已提交
983 984 985 986 987 988
              if (code) goto _err;
            }
          }
        }
      }
    } else if (pBlock) {
H
Hongze Cheng 已提交
989
      code = tsdbCommitTableDiskData(pCommitter, pBlock, pBlockIdx);
H
Hongze Cheng 已提交
990 991
      if (code) goto _err;

H
Hongze Cheng 已提交
992 993 994 995 996 997
      iBlock++;
      if (iBlock < nBlock) {
        tMapDataGetItemByIdx(&pCommitter->oBlockMap, iBlock, pBlock, tGetBlock);
      } else {
        pBlock = NULL;
      }
H
Hongze Cheng 已提交
998
    } else {
H
Hongze Cheng 已提交
999 1000
      code =
          tsdbCommitTableMemData(pCommitter, pIter, (TSDBKEY){.ts = pCommitter->maxKey + 1, .version = VERSION_MIN}, 0);
H
Hongze Cheng 已提交
1001 1002 1003 1004 1005 1006 1007
      if (code) goto _err;

      pRow = tsdbTbDataIterGet(pIter);
      ASSERT(pRow == NULL || TSDBROW_TS(pRow) > pCommitter->maxKey);
    }
  }

H
Hongze Cheng 已提交
1008 1009
  // end =====================
  code = tsdbCommitTableDataEnd(pCommitter, suid, uid);
H
Hongze Cheng 已提交
1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023
  if (code) goto _err;

_exit:
  if (pIter) {
    pRow = tsdbTbDataIterGet(pIter);
    if (pRow) pCommitter->nextKey = TMIN(pCommitter->nextKey, TSDBROW_TS(pRow));
  }
  return code;

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

H
Hongze Cheng 已提交
1024 1025
static int32_t tsdbCommitFileDataImpl(SCommitter *pCommitter) {
  int32_t    code = 0;
H
Hongze Cheng 已提交
1026
  int32_t    c;
H
Hongze Cheng 已提交
1027 1028 1029 1030
  STsdb     *pTsdb = pCommitter->pTsdb;
  SMemTable *pMemTable = pTsdb->imem;
  int32_t    iTbData = 0;
  int32_t    nTbData = taosArrayGetSize(pMemTable->aTbData);
H
Hongze Cheng 已提交
1031
  int32_t    iBlockIdx = 0;
H
Hongze Cheng 已提交
1032
  int32_t    nBlockIdx = pCommitter->oBlockIdxMap.nItem;
H
Hongze Cheng 已提交
1033
  STbData   *pTbData;
H
Hongze Cheng 已提交
1034
  SBlockIdx *pBlockIdx = &(SBlockIdx){0};
H
Hongze Cheng 已提交
1035

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

H
Hongze Cheng 已提交
1038 1039
  pTbData = (STbData *)taosArrayGetP(pMemTable->aTbData, iTbData);
  if (iBlockIdx < nBlockIdx) {
H
Hongze Cheng 已提交
1040
    tMapDataGetItemByIdx(&pCommitter->oBlockIdxMap, iBlockIdx, pBlockIdx, tGetBlockIdx);
H
Hongze Cheng 已提交
1041 1042
  } else {
    pBlockIdx = NULL;
H
Hongze Cheng 已提交
1043 1044
  }

H
Hongze Cheng 已提交
1045 1046 1047 1048 1049 1050
  // merge
  while (pTbData && pBlockIdx) {
    c = tTABLEIDCmprFn(pTbData, pBlockIdx);

    if (c == 0) {
      // merge commit
H
Hongze Cheng 已提交
1051 1052
      code = tsdbMergeMemDisk(pCommitter, pTbData, pBlockIdx);
      if (code) goto _err;
H
Hongze Cheng 已提交
1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067

      iTbData++;
      iBlockIdx++;
      if (iTbData < nTbData) {
        pTbData = (STbData *)taosArrayGetP(pMemTable->aTbData, iTbData);
      } else {
        pTbData = NULL;
      }
      if (iBlockIdx < nBlockIdx) {
        tMapDataGetItemByIdx(&pCommitter->oBlockIdxMap, iBlockIdx, pBlockIdx, tGetBlockIdx);
      } else {
        pBlockIdx = NULL;
      }
    } else if (c < 0) {
      // commit memory data
H
Hongze Cheng 已提交
1068 1069
      code = tsdbCommitMemoryData(pCommitter, pTbData);
      if (code) goto _err;
H
Hongze Cheng 已提交
1070 1071 1072 1073 1074 1075 1076 1077 1078

      iTbData++;
      if (iTbData < nTbData) {
        pTbData = (STbData *)taosArrayGetP(pMemTable->aTbData, iTbData);
      } else {
        pTbData = NULL;
      }
    } else {
      // commit disk data
H
Hongze Cheng 已提交
1079 1080
      code = tsdbCommitDiskData(pCommitter, pBlockIdx);
      if (code) goto _err;
H
Hongze Cheng 已提交
1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093

      iBlockIdx++;
      if (iBlockIdx < nBlockIdx) {
        tMapDataGetItemByIdx(&pCommitter->oBlockIdxMap, iBlockIdx, pBlockIdx, tGetBlockIdx);
      } else {
        pBlockIdx = NULL;
      }
    }
  }

  // disk
  while (pBlockIdx) {
    // commit disk data
H
Hongze Cheng 已提交
1094 1095
    code = tsdbCommitDiskData(pCommitter, pBlockIdx);
    if (code) goto _err;
H
Hongze Cheng 已提交
1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118

    iBlockIdx++;
    if (iBlockIdx < nBlockIdx) {
      tMapDataGetItemByIdx(&pCommitter->oBlockIdxMap, iBlockIdx, pBlockIdx, tGetBlockIdx);
    } else {
      pBlockIdx = NULL;
    }
  }

  // memory
  while (pTbData) {
    // commit memory data
    code = tsdbCommitMemoryData(pCommitter, pTbData);
    if (code) goto _err;

    iTbData++;
    if (iTbData < nTbData) {
      pTbData = (STbData *)taosArrayGetP(pMemTable->aTbData, iTbData);
    } else {
      pTbData = NULL;
    }
  }

H
Hongze Cheng 已提交
1119 1120 1121
  return code;

_err:
H
Hongze Cheng 已提交
1122
  tsdbError("vgId:%d commit file data impl failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
1123 1124 1125 1126 1127
  return code;
}

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

H
Hongze Cheng 已提交
1129
  // write blockIdx
H
Hongze Cheng 已提交
1130
  code = tsdbWriteBlockIdx(pCommitter->pWriter, &pCommitter->nBlockIdxMap, NULL);
H
Hongze Cheng 已提交
1131 1132
  if (code) goto _err;

H
Hongze Cheng 已提交
1133
  // update file header
H
Hongze Cheng 已提交
1134
  code = tsdbUpdateDFileSetHeader(pCommitter->pWriter);
H
Hongze Cheng 已提交
1135 1136
  if (code) goto _err;

H
Hongze Cheng 已提交
1137 1138 1139 1140
  // upsert SDFileSet
  code = tsdbFSStateUpsertDFileSet(pCommitter->pTsdb->fs->nState, tsdbDataFWriterGetWSet(pCommitter->pWriter));
  if (code) goto _err;

H
Hongze Cheng 已提交
1141
  // close and sync
H
Hongze Cheng 已提交
1142
  code = tsdbDataFWriterClose(&pCommitter->pWriter, 1);
H
Hongze Cheng 已提交
1143 1144 1145
  if (code) goto _err;

  if (pCommitter->pReader) {
H
Hongze Cheng 已提交
1146
    code = tsdbDataFReaderClose(&pCommitter->pReader);
H
Hongze Cheng 已提交
1147 1148 1149 1150 1151 1152 1153
    goto _err;
  }

_exit:
  return code;

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

H
Hongze Cheng 已提交
1158
static int32_t tsdbCommitFileData(SCommitter *pCommitter) {
H
Hongze Cheng 已提交
1159 1160
  int32_t code = 0;

H
Hongze Cheng 已提交
1161 1162
  // commit file data start
  code = tsdbCommitFileDataStart(pCommitter);
H
Hongze Cheng 已提交
1163
  if (code) goto _err;
H
Hongze Cheng 已提交
1164

H
Hongze Cheng 已提交
1165 1166
  // commit file data impl
  code = tsdbCommitFileDataImpl(pCommitter);
H
Hongze Cheng 已提交
1167
  if (code) goto _err;
H
Hongze Cheng 已提交
1168

H
Hongze Cheng 已提交
1169 1170
  // commit file data end
  code = tsdbCommitFileDataEnd(pCommitter);
H
Hongze Cheng 已提交
1171
  if (code) goto _err;
H
Hongze Cheng 已提交
1172 1173 1174 1175

  return code;

_err:
H
Hongze Cheng 已提交
1176
  tsdbError("vgId:%d commit file data failed since %s", TD_VID(pCommitter->pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
1177 1178 1179
  return code;
}

H
Hongze Cheng 已提交
1180 1181
// ----------------------------------------------------------------------------
static int32_t tsdbStartCommit(STsdb *pTsdb, SCommitter *pCommitter) {
H
Hongze Cheng 已提交
1182
  int32_t code = 0;
H
Hongze Cheng 已提交
1183

H
Hongze Cheng 已提交
1184 1185
  memset(pCommitter, 0, sizeof(*pCommitter));
  ASSERT(pTsdb->mem && pTsdb->imem == NULL);
H
Hongze Cheng 已提交
1186

H
Hongze Cheng 已提交
1187 1188 1189 1190
  // lock();
  pTsdb->imem = pTsdb->mem;
  pTsdb->mem = NULL;
  // unlock();
H
Hongze Cheng 已提交
1191

H
Hongze Cheng 已提交
1192
  pCommitter->pTsdb = pTsdb;
H
Hongze Cheng 已提交
1193
  pCommitter->commitID = pTsdb->pVnode->state.commitID;
H
Hongze Cheng 已提交
1194 1195 1196 1197
  pCommitter->minutes = pTsdb->keepCfg.days;
  pCommitter->precision = pTsdb->keepCfg.precision;
  pCommitter->minRow = pTsdb->pVnode->config.tsdbCfg.minRows;
  pCommitter->maxRow = pTsdb->pVnode->config.tsdbCfg.maxRows;
H
Hongze Cheng 已提交
1198
  pCommitter->cmprAlg = pTsdb->pVnode->config.tsdbCfg.compression;
H
Hongze Cheng 已提交
1199

H
Hongze Cheng 已提交
1200 1201 1202 1203 1204 1205 1206
  code = tsdbFSBegin(pTsdb->fs);
  if (code) goto _err;

  return code;

_err:
  tsdbError("vgId:%d tsdb start commit failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
1207 1208 1209
  return code;
}

H
Hongze Cheng 已提交
1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233
static int32_t tsdbCommitDataStart(SCommitter *pCommitter) {
  int32_t code = 0;

  pCommitter->pReader = NULL;
  pCommitter->oBlockIdxMap = tMapDataInit();
  pCommitter->oBlockMap = tMapDataInit();
  pCommitter->oBlock = tBlockInit();
  pCommitter->pWriter = NULL;
  pCommitter->nBlockIdxMap = tMapDataInit();
  pCommitter->nBlockMap = tMapDataInit();
  pCommitter->nBlock = tBlockInit();
  code = tBlockDataInit(&pCommitter->oBlockData);
  if (code) goto _exit;
  code = tBlockDataInit(&pCommitter->nBlockData);
  if (code) {
    tBlockDataClear(&pCommitter->oBlockData);
    goto _exit;
  }

_exit:
  return code;
}

static void tsdbCommitDataEnd(SCommitter *pCommitter) {
H
Hongze Cheng 已提交
1234 1235 1236 1237 1238 1239 1240 1241
  // tMapDataClear(&pCommitter->oBlockIdxMap);
  // tMapDataClear(&pCommitter->oBlockMap);
  // tBlockClear(&pCommitter->oBlock);
  // tBlockDataClear(&pCommitter->oBlockData);
  // tMapDataClear(&pCommitter->nBlockIdxMap);
  // tMapDataClear(&pCommitter->nBlockMap);
  // tBlockClear(&pCommitter->nBlock);
  // tBlockDataClear(&pCommitter->nBlockData);
H
Hongze Cheng 已提交
1242 1243
}

H
Hongze Cheng 已提交
1244 1245 1246 1247
static int32_t tsdbCommitData(SCommitter *pCommitter) {
  int32_t    code = 0;
  STsdb     *pTsdb = pCommitter->pTsdb;
  SMemTable *pMemTable = pTsdb->imem;
H
Hongze Cheng 已提交
1248

H
Hongze Cheng 已提交
1249
  // check
H
Hongze Cheng 已提交
1250
  if (pMemTable->nRow == 0) goto _exit;
H
Hongze Cheng 已提交
1251

H
Hongze Cheng 已提交
1252 1253
  // start ====================
  code = tsdbCommitDataStart(pCommitter);
H
Hongze Cheng 已提交
1254
  if (code) goto _err;
H
Hongze Cheng 已提交
1255 1256 1257

  // impl ====================
  pCommitter->nextKey = pMemTable->minKey;
H
Hongze Cheng 已提交
1258 1259 1260 1261 1262
  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 已提交
1263
    if (code) goto _err;
H
Hongze Cheng 已提交
1264
  }
H
Hongze Cheng 已提交
1265

H
Hongze Cheng 已提交
1266 1267 1268
  // end ====================
  tsdbCommitDataEnd(pCommitter);

H
Hongze Cheng 已提交
1269 1270 1271
_exit:
  tsdbDebug("vgId:%d commit data done, nRow:%" PRId64, TD_VID(pTsdb->pVnode), pMemTable->nRow);
  return code;
H
Hongze Cheng 已提交
1272

H
Hongze Cheng 已提交
1273
_err:
H
Hongze Cheng 已提交
1274
  tsdbCommitDataEnd(pCommitter);
H
Hongze Cheng 已提交
1275 1276 1277
  tsdbError("vgId:%d commit data failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
  return code;
}
H
Hongze Cheng 已提交
1278

H
Hongze Cheng 已提交
1279 1280 1281 1282
static int32_t tsdbCommitDel(SCommitter *pCommitter) {
  int32_t    code = 0;
  STsdb     *pTsdb = pCommitter->pTsdb;
  SMemTable *pMemTable = pTsdb->imem;
H
Hongze Cheng 已提交
1283

H
Hongze Cheng 已提交
1284 1285
  if (pMemTable->nDel == 0) {
    goto _exit;
H
Hongze Cheng 已提交
1286
  }
H
Hongze Cheng 已提交
1287

H
Hongze Cheng 已提交
1288 1289 1290 1291 1292
  // start
  code = tsdbCommitDelStart(pCommitter);
  if (code) {
    goto _err;
  }
H
Hongze Cheng 已提交
1293

H
Hongze Cheng 已提交
1294 1295 1296 1297 1298
  // impl
  code = tsdbCommitDelImpl(pCommitter);
  if (code) {
    goto _err;
  }
H
Hongze Cheng 已提交
1299

H
Hongze Cheng 已提交
1300 1301 1302 1303 1304
  // end
  code = tsdbCommitDelEnd(pCommitter);
  if (code) {
    goto _err;
  }
H
Hongze Cheng 已提交
1305

H
Hongze Cheng 已提交
1306
_exit:
H
Hongze Cheng 已提交
1307
  tsdbDebug("vgId:%d commit del done, nDel:%" PRId64, TD_VID(pTsdb->pVnode), pMemTable->nDel);
H
Hongze Cheng 已提交
1308 1309 1310
  return code;

_err:
H
Hongze Cheng 已提交
1311
  tsdbError("vgId:%d commit del failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
1312
  return code;
H
Hongze Cheng 已提交
1313 1314 1315 1316 1317 1318 1319
}

static int32_t tsdbCommitCache(SCommitter *pCommitter) {
  int32_t code = 0;
  // TODO
  return code;
}
H
Hongze Cheng 已提交
1320 1321

static int32_t tsdbEndCommit(SCommitter *pCommitter, int32_t eno) {
H
Hongze Cheng 已提交
1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339
  int32_t    code = 0;
  STsdb     *pTsdb = pCommitter->pTsdb;
  SMemTable *pMemTable = pTsdb->imem;

  if (eno == 0) {
    code = tsdbFSCommit(pTsdb->fs);
  } else {
    code = tsdbFSRollback(pTsdb->fs);
  }

  tsdbMemTableDestroy(pMemTable);
  pTsdb->imem = NULL;

  tsdbInfo("vgId:%d tsdb end commit", TD_VID(pTsdb->pVnode));
  return code;

_err:
  tsdbError("vgId:%d tsdb end commit failed since %s", TD_VID(pTsdb->pVnode), tstrerror(code));
H
Hongze Cheng 已提交
1340 1341
  return code;
}