syncRaftLogTest3.cpp 10.8 KB
Newer Older
M
Minghao Li 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
#include <gtest/gtest.h>
#include <stdio.h>
#include "syncEnv.h"
#include "syncIO.h"
#include "syncInt.h"
#include "syncRaftLog.h"
#include "syncRaftStore.h"
#include "syncUtil.h"
#include "wal.h"

void logTest() {
  sTrace("--- sync log test: trace");
  sDebug("--- sync log test: debug");
  sInfo("--- sync log test: info");
  sWarn("--- sync log test: warn");
  sError("--- sync log test: error");
  sFatal("--- sync log test: fatal");
}

M
Minghao Li 已提交
20 21
bool gAssert = true;

M
Minghao Li 已提交
22 23 24 25 26
SSyncNode*     pSyncNode;
SWal*          pWal;
SSyncLogStore* pLogStore;
const char*    pWalPath = "./syncLogStoreTest_wal";

M
Minghao Li 已提交
27 28 29 30 31 32 33 34 35 36
SyncIndex gSnapshotLastApplyIndex;
SyncIndex gSnapshotLastApplyTerm;

int32_t GetSnapshotCb(struct SSyncFSM* pFsm, SSnapshot* pSnapshot) {
  pSnapshot->data = NULL;
  pSnapshot->lastApplyIndex = gSnapshotLastApplyIndex;
  pSnapshot->lastApplyTerm = gSnapshotLastApplyTerm;
  return 0;
}

M
Minghao Li 已提交
37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54
void init() {
  walInit();

  SWalCfg walCfg;
  memset(&walCfg, 0, sizeof(SWalCfg));
  walCfg.vgId = 1000;
  walCfg.fsyncPeriod = 1000;
  walCfg.retentionPeriod = 1000;
  walCfg.rollPeriod = 1000;
  walCfg.retentionSize = 1000;
  walCfg.segSize = 1000;
  walCfg.level = TAOS_WAL_FSYNC;
  pWal = walOpen(pWalPath, &walCfg);
  assert(pWal != NULL);

  pSyncNode = (SSyncNode*)taosMemoryMalloc(sizeof(SSyncNode));
  memset(pSyncNode, 0, sizeof(SSyncNode));
  pSyncNode->pWal = pWal;
M
Minghao Li 已提交
55 56

  pSyncNode->pFsm = (SSyncFSM*)taosMemoryMalloc(sizeof(SSyncFSM));
57
  pSyncNode->pFsm->FpGetSnapshotInfo = GetSnapshotCb;
M
Minghao Li 已提交
58 59 60 61 62 63 64 65
}

void cleanup() {
  walClose(pWal);
  walCleanUp();
  taosMemoryFree(pSyncNode);
}

M
Minghao Li 已提交
66 67 68 69 70 71 72
void test1() {
  // no snapshot
  // no log

  taosRemoveDir(pWalPath);

  init();
M
Minghao Li 已提交
73 74
  pLogStore = logStoreCreate(pSyncNode);
  assert(pLogStore);
M
Minghao Li 已提交
75 76 77 78 79 80
  pSyncNode->pLogStore = pLogStore;
  logStoreLog2((char*)"\n\n\ntest1 ----- ", pLogStore);

  gSnapshotLastApplyIndex = -1;
  gSnapshotLastApplyTerm = 0;

81
  bool      hasSnapshot = syncNodeHasSnapshot(pSyncNode);
M
Minghao Li 已提交
82
  SSnapshot snapshot;
83
  pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
M
Minghao Li 已提交
84 85 86 87 88 89 90

  SyncIndex lastIndex = syncNodeGetLastIndex(pSyncNode);
  SyncTerm  lastTerm = syncNodeGetLastTerm(pSyncNode);

  SyncIndex testIndex = 0;
  SyncIndex preIndex = syncNodeGetPreIndex(pSyncNode, testIndex);
  SyncTerm  preTerm = syncNodeGetPreTerm(pSyncNode, testIndex);
M
Minghao Li 已提交
91

M
Minghao Li 已提交
92 93
  SyncIndex syncStartIndex = syncNodeSyncStartIndex(pSyncNode);

M
Minghao Li 已提交
94
  sTrace("test1");
S
Shengliang Guan 已提交
95
  sTrace("hasSnapshot:%d, lastApplyIndex:%" PRId64 ", lastApplyTerm:%" PRIu64, hasSnapshot, snapshot.lastApplyIndex,
96
         snapshot.lastApplyTerm);
S
Shengliang Guan 已提交
97 98 99 100 101
  sTrace("lastIndex: %" PRId64, lastIndex);
  sTrace("lastTerm: %" PRIu64, lastTerm);
  sTrace("syncStartIndex: %" PRId64, syncStartIndex);
  sTrace("" PRId64 "'s preIndex: %" PRId64, testIndex, preIndex);
  sTrace("" PRId64 "'s preTerm: %" PRIu64, testIndex, preTerm);
M
Minghao Li 已提交
102

M
Minghao Li 已提交
103 104 105 106 107 108 109 110
  if (gAssert) {
    assert(lastIndex == -1);
    assert(lastTerm == 0);
    assert(syncStartIndex == 0);
    assert(preIndex == -1);
    assert(preTerm == 0);
  }

M
Minghao Li 已提交
111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
  logStoreDestory(pLogStore);
  cleanup();
}

void test2() {
  // no snapshot
  // whole log

  taosRemoveDir(pWalPath);

  init();
  pLogStore = logStoreCreate(pSyncNode);
  assert(pLogStore);
  pSyncNode->pLogStore = pLogStore;
  logStoreLog2((char*)"\n\n\ntest2 ----- ", pLogStore);

  for (int i = 0; i <= 10; ++i) {
M
Minghao Li 已提交
128 129 130 131 132 133 134 135
    int32_t         dataLen = 10;
    SSyncRaftEntry* pEntry = syncEntryBuild(dataLen);
    assert(pEntry != NULL);
    pEntry->msgType = 1;
    pEntry->originalRpcType = 2;
    pEntry->seqNum = 3;
    pEntry->isWeak = true;
    pEntry->term = 100 + i;
M
Minghao Li 已提交
136
    pEntry->index = pLogStore->syncLogWriteIndex(pLogStore);
M
Minghao Li 已提交
137 138
    snprintf(pEntry->data, dataLen, "value%d", i);

M
Minghao Li 已提交
139
    pLogStore->syncLogAppendEntry(pLogStore, pEntry);
M
Minghao Li 已提交
140
    syncEntryDestory(pEntry);
M
Minghao Li 已提交
141 142 143 144 145 146
  }
  logStoreLog2((char*)"test2 after appendEntry", pLogStore);

  gSnapshotLastApplyIndex = -1;
  gSnapshotLastApplyTerm = 0;

147
  bool      hasSnapshot = syncNodeHasSnapshot(pSyncNode);
M
Minghao Li 已提交
148
  SSnapshot snapshot;
149
  pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
M
Minghao Li 已提交
150

M
Minghao Li 已提交
151 152 153
  SyncIndex lastIndex = syncNodeGetLastIndex(pSyncNode);
  SyncTerm  lastTerm = syncNodeGetLastTerm(pSyncNode);

M
Minghao Li 已提交
154 155
  SyncIndex syncStartIndex = syncNodeSyncStartIndex(pSyncNode);

M
Minghao Li 已提交
156
  sTrace("test2");
S
Shengliang Guan 已提交
157
  sTrace("hasSnapshot:%d, lastApplyIndex:%" PRId64 ", lastApplyTerm:%" PRIu64, hasSnapshot, snapshot.lastApplyIndex,
158
         snapshot.lastApplyTerm);
S
Shengliang Guan 已提交
159 160 161
  sTrace("lastIndex: %" PRId64, lastIndex);
  sTrace("lastTerm: %" PRIu64, lastTerm);
  sTrace("syncStartIndex: %" PRId64, syncStartIndex);
M
Minghao Li 已提交
162

M
Minghao Li 已提交
163 164 165 166 167 168
  if (gAssert) {
    assert(lastIndex == 10);
    assert(lastTerm == 110);
    assert(syncStartIndex == 11);
  }

M
Minghao Li 已提交
169 170 171 172
  for (SyncIndex i = 11; i >= 0; --i) {
    SyncIndex preIndex = syncNodeGetPreIndex(pSyncNode, i);
    SyncTerm  preTerm = syncNodeGetPreTerm(pSyncNode, i);

S
Shengliang Guan 已提交
173 174
    sTrace("" PRId64 "'s preIndex: %" PRId64, i, preIndex);
    sTrace("" PRId64 "'s preTerm: %" PRIu64, i, preTerm);
M
Minghao Li 已提交
175 176 177 178 179 180 181 182

    if (gAssert) {
      SyncIndex preIndexArr[12] = {-1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
      SyncTerm  preTermArr[12] = {0, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110};

      assert(preIndex == preIndexArr[i]);
      assert(preTerm == preTermArr[i]);
    }
M
Minghao Li 已提交
183 184
  }

M
Minghao Li 已提交
185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203
  logStoreDestory(pLogStore);
  cleanup();
}

void test3() {
  // has snapshot
  // no log

  taosRemoveDir(pWalPath);

  init();
  pLogStore = logStoreCreate(pSyncNode);
  assert(pLogStore);
  pSyncNode->pLogStore = pLogStore;
  logStoreLog2((char*)"\n\n\ntest3 ----- ", pLogStore);

  gSnapshotLastApplyIndex = 5;
  gSnapshotLastApplyTerm = 100;

204
  bool      hasSnapshot = syncNodeHasSnapshot(pSyncNode);
M
Minghao Li 已提交
205
  SSnapshot snapshot;
206
  pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
M
Minghao Li 已提交
207 208 209 210 211 212 213

  SyncIndex lastIndex = syncNodeGetLastIndex(pSyncNode);
  SyncTerm  lastTerm = syncNodeGetLastTerm(pSyncNode);

  SyncIndex preIndex = syncNodeGetPreIndex(pSyncNode, 6);
  SyncTerm  preTerm = syncNodeGetPreTerm(pSyncNode, 6);

M
Minghao Li 已提交
214 215
  SyncIndex syncStartIndex = syncNodeSyncStartIndex(pSyncNode);

M
Minghao Li 已提交
216
  sTrace("test3");
S
Shengliang Guan 已提交
217
  sTrace("hasSnapshot:%d, lastApplyIndex:%" PRId64 ", lastApplyTerm:%" PRIu64, hasSnapshot, snapshot.lastApplyIndex,
218
         snapshot.lastApplyTerm);
S
Shengliang Guan 已提交
219 220 221 222 223
  sTrace("lastIndex: %" PRId64, lastIndex);
  sTrace("lastTerm: %" PRIu64, lastTerm);
  sTrace("syncStartIndex: %" PRId64, syncStartIndex);
  sTrace("%d's preIndex: %" PRId64, 6, preIndex);
  sTrace("%d's preTerm: %" PRIu64, 6, preTerm);
M
Minghao Li 已提交
224

M
Minghao Li 已提交
225 226 227 228 229 230 231 232
  if (gAssert) {
    assert(lastIndex == 5);
    assert(lastTerm == 100);
    assert(syncStartIndex == 6);
    assert(preIndex == 5);
    assert(preTerm == 100);
  }

M
Minghao Li 已提交
233
  logStoreDestory(pLogStore);
M
Minghao Li 已提交
234
  cleanup();
M
Minghao Li 已提交
235 236
}

M
Minghao Li 已提交
237 238 239 240 241
void test4() {
  // has snapshot
  // whole log

  taosRemoveDir(pWalPath);
M
Minghao Li 已提交
242 243

  init();
M
Minghao Li 已提交
244 245 246 247
  pLogStore = logStoreCreate(pSyncNode);
  assert(pLogStore);
  pSyncNode->pLogStore = pLogStore;
  logStoreLog2((char*)"\n\n\ntest4 ----- ", pLogStore);
M
Minghao Li 已提交
248

M
Minghao Li 已提交
249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268
  for (int i = 0; i <= 10; ++i) {
    int32_t         dataLen = 10;
    SSyncRaftEntry* pEntry = syncEntryBuild(dataLen);
    assert(pEntry != NULL);
    pEntry->msgType = 1;
    pEntry->originalRpcType = 2;
    pEntry->seqNum = 3;
    pEntry->isWeak = true;
    pEntry->term = 100 + i;
    pEntry->index = pLogStore->syncLogWriteIndex(pLogStore);
    snprintf(pEntry->data, dataLen, "value%d", i);

    pLogStore->syncLogAppendEntry(pLogStore, pEntry);
    syncEntryDestory(pEntry);
  }
  logStoreLog2((char*)"test4 after appendEntry", pLogStore);

  gSnapshotLastApplyIndex = 5;
  gSnapshotLastApplyTerm = 100;

269
  bool      hasSnapshot = syncNodeHasSnapshot(pSyncNode);
M
Minghao Li 已提交
270
  SSnapshot snapshot;
271
  pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
M
Minghao Li 已提交
272 273 274 275

  SyncIndex lastIndex = syncNodeGetLastIndex(pSyncNode);
  SyncTerm  lastTerm = syncNodeGetLastTerm(pSyncNode);

M
Minghao Li 已提交
276 277
  SyncIndex syncStartIndex = syncNodeSyncStartIndex(pSyncNode);

M
Minghao Li 已提交
278
  sTrace("test4");
S
Shengliang Guan 已提交
279
  sTrace("hasSnapshot:%d, lastApplyIndex:%" PRId64 ", lastApplyTerm:%" PRIu64, hasSnapshot, snapshot.lastApplyIndex,
280
         snapshot.lastApplyTerm);
S
Shengliang Guan 已提交
281 282 283
  sTrace("lastIndex: %" PRId64, lastIndex);
  sTrace("lastTerm: %" PRIu64, lastTerm);
  sTrace("syncStartIndex: %" PRId64, syncStartIndex);
M
Minghao Li 已提交
284

M
Minghao Li 已提交
285 286 287 288 289 290
  if (gAssert) {
    assert(lastIndex == 10);
    assert(lastTerm == 110);
    assert(syncStartIndex == 11);
  }

M
Minghao Li 已提交
291 292 293 294
  for (SyncIndex i = 11; i >= 6; --i) {
    SyncIndex preIndex = syncNodeGetPreIndex(pSyncNode, i);
    SyncTerm  preTerm = syncNodeGetPreTerm(pSyncNode, i);

S
Shengliang Guan 已提交
295 296
    sTrace("" PRId64 "'s preIndex: %" PRId64, i, preIndex);
    sTrace("" PRId64 "'s preTerm: %" PRIu64, i, preTerm);
M
Minghao Li 已提交
297 298 299
  }

  logStoreDestory(pLogStore);
M
Minghao Li 已提交
300
  cleanup();
M
Minghao Li 已提交
301 302 303 304 305 306 307 308 309 310 311 312 313 314
}

void test5() {
  // has snapshot
  // partial log

  taosRemoveDir(pWalPath);

  init();
  pLogStore = logStoreCreate(pSyncNode);
  assert(pLogStore);
  pSyncNode->pLogStore = pLogStore;
  logStoreLog2((char*)"\n\n\ntest5 ----- ", pLogStore);

315
  // pSyncNode->pLogStore->syncLogSetBeginIndex(pSyncNode->pLogStore, 6);
316
  pLogStore->syncLogRestoreFromSnapshot(pSyncNode->pLogStore, 5);
M
Minghao Li 已提交
317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336
  for (int i = 6; i <= 10; ++i) {
    int32_t         dataLen = 10;
    SSyncRaftEntry* pEntry = syncEntryBuild(dataLen);
    assert(pEntry != NULL);
    pEntry->msgType = 1;
    pEntry->originalRpcType = 2;
    pEntry->seqNum = 3;
    pEntry->isWeak = true;
    pEntry->term = 100 + i;
    pEntry->index = pLogStore->syncLogWriteIndex(pLogStore);
    snprintf(pEntry->data, dataLen, "value%d", i);

    pLogStore->syncLogAppendEntry(pLogStore, pEntry);
    syncEntryDestory(pEntry);
  }
  logStoreLog2((char*)"test5 after appendEntry", pLogStore);

  gSnapshotLastApplyIndex = 5;
  gSnapshotLastApplyTerm = 100;

337
  bool      hasSnapshot = syncNodeHasSnapshot(pSyncNode);
M
Minghao Li 已提交
338
  SSnapshot snapshot;
339
  pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
M
Minghao Li 已提交
340 341 342 343

  SyncIndex lastIndex = syncNodeGetLastIndex(pSyncNode);
  SyncTerm  lastTerm = syncNodeGetLastTerm(pSyncNode);

M
Minghao Li 已提交
344 345
  SyncIndex syncStartIndex = syncNodeSyncStartIndex(pSyncNode);

M
Minghao Li 已提交
346
  sTrace("test5");
S
Shengliang Guan 已提交
347
  sTrace("hasSnapshot:%d, lastApplyIndex:%" PRId64 ", lastApplyTerm:%" PRIu64, hasSnapshot, snapshot.lastApplyIndex,
348
         snapshot.lastApplyTerm);
S
Shengliang Guan 已提交
349 350 351
  sTrace("lastIndex: %" PRId64, lastIndex);
  sTrace("lastTerm: %" PRIu64, lastTerm);
  sTrace("syncStartIndex: %" PRId64, syncStartIndex);
M
Minghao Li 已提交
352 353 354 355 356

  for (SyncIndex i = 11; i >= 6; --i) {
    SyncIndex preIndex = syncNodeGetPreIndex(pSyncNode, i);
    SyncTerm  preTerm = syncNodeGetPreTerm(pSyncNode, i);

S
Shengliang Guan 已提交
357 358
    sTrace("" PRId64 "'s preIndex: %" PRId64, i, preIndex);
    sTrace("" PRId64 "'s preTerm: %" PRIu64, i, preTerm);
M
Minghao Li 已提交
359 360 361 362 363 364 365 366

    if (gAssert) {
      SyncIndex preIndexArr[12] = {9999, 9999, 9999, 9999, 9999, 9999, 5, 6, 7, 8, 9, 10};
      SyncTerm  preTermArr[12] = {9999, 9999, 9999, 9999, 9999, 9999, 100, 106, 107, 108, 109, 110};

      assert(preIndex == preIndexArr[i]);
      assert(preTerm == preTermArr[i]);
    }
M
Minghao Li 已提交
367 368 369 370 371 372 373 374 375
  }

  logStoreDestory(pLogStore);
  cleanup();
}

int main(int argc, char** argv) {
  tsAsyncLog = 0;
  sDebugFlag = DEBUG_TRACE + DEBUG_INFO + DEBUG_SCREEN + DEBUG_FILE;
376
  gRaftDetailLog = true;
M
Minghao Li 已提交
377

M
Minghao Li 已提交
378 379 380 381 382
  if (argc == 2) {
    gAssert = atoi(argv[1]);
  }
  sTrace("gAssert : %d", gAssert);

M
Minghao Li 已提交
383 384 385 386 387
  test1();
  test2();
  test3();
  test4();
  test5();
M
Minghao Li 已提交
388 389 390

  return 0;
}