syncMain.c 79.8 KB
Newer Older
M
Minghao Li 已提交
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/>.
 */

S
Shengliang Guan 已提交
16
#define _DEFAULT_SOURCE
M
Minghao Li 已提交
17
#include "sync.h"
M
Minghao Li 已提交
18 19
#include "syncAppendEntries.h"
#include "syncAppendEntriesReply.h"
M
Minghao Li 已提交
20
#include "syncCommit.h"
M
Minghao Li 已提交
21
#include "syncElection.h"
M
Minghao Li 已提交
22
#include "syncEnv.h"
M
Minghao Li 已提交
23
#include "syncIndexMgr.h"
M
Minghao Li 已提交
24
#include "syncInt.h"
M
Minghao Li 已提交
25
#include "syncMessage.h"
M
Minghao Li 已提交
26
#include "syncRaftCfg.h"
M
Minghao Li 已提交
27
#include "syncRaftLog.h"
M
Minghao Li 已提交
28
#include "syncRaftStore.h"
M
Minghao Li 已提交
29
#include "syncReplication.h"
M
Minghao Li 已提交
30 31
#include "syncRequestVote.h"
#include "syncRequestVoteReply.h"
M
Minghao Li 已提交
32
#include "syncRespMgr.h"
M
Minghao Li 已提交
33
#include "syncSnapshot.h"
M
Minghao Li 已提交
34
#include "syncTimeout.h"
M
Minghao Li 已提交
35
#include "syncUtil.h"
M
Minghao Li 已提交
36
#include "syncVoteMgr.h"
M
Minghao Li 已提交
37

M
Minghao Li 已提交
38 39 40 41 42
static void    syncNodeEqPingTimer(void* param, void* tmrId);
static void    syncNodeEqElectTimer(void* param, void* tmrId);
static void    syncNodeEqHeartbeatTimer(void* param, void* tmrId);
static int32_t syncNodeEqNoop(SSyncNode* ths);
static int32_t syncNodeAppendNoop(SSyncNode* ths);
43
static void    syncNodeEqPeerHeartbeatTimer(void* param, void* tmrId);
S
Shengliang Guan 已提交
44
static bool    syncIsConfigChanged(const SSyncCfg* pOldCfg, const SSyncCfg* pNewCfg);
S
Shengliang Guan 已提交
45 46 47
static int32_t syncHbTimerInit(SSyncNode* pSyncNode, SSyncTimer* pSyncTimer, SRaftId destId);
static int32_t syncHbTimerStart(SSyncNode* pSyncNode, SSyncTimer* pSyncTimer);
static int32_t syncHbTimerStop(SSyncNode* pSyncNode, SSyncTimer* pSyncTimer);
S
Shengliang Guan 已提交
48 49 50 51 52 53 54 55 56 57 58 59
static int32_t syncNodeUpdateNewConfigIndex(SSyncNode* ths, SSyncCfg* pNewCfg);
static bool    syncNodeInConfig(SSyncNode* pSyncNode, const SSyncCfg* config);
static void    syncNodeDoConfigChange(SSyncNode* pSyncNode, SSyncCfg* newConfig, SyncIndex lastConfigChangeIndex);
static bool    syncNodeIsOptimizedOneReplica(SSyncNode* ths, SRpcMsg* pMsg);

static bool    syncNodeCanChange(SSyncNode* pSyncNode);
static int32_t syncNodeLeaderTransfer(SSyncNode* pSyncNode);
static int32_t syncNodeLeaderTransferTo(SSyncNode* pSyncNode, SNodeInfo newLeader);
static int32_t syncDoLeaderTransfer(SSyncNode* ths, SRpcMsg* pRpcMsg, SSyncRaftEntry* pEntry);

static ESyncStrategy syncNodeStrategy(SSyncNode* pSyncNode);
static SyncIndex     syncNodeGetSnapshotConfigIndex(SSyncNode* pSyncNode, SyncIndex snapshotLastApplyIndex);
M
Minghao Li 已提交
60

61
int64_t syncOpen(SSyncInfo* pSyncInfo) {
S
Shengliang Guan 已提交
62 63
  SSyncNode* pSyncNode = syncNodeOpen(pSyncInfo);
  if (pSyncNode == NULL) {
S
Shengliang Guan 已提交
64
    sError("vgId:%d, failed to open sync node", pSyncInfo->vgId);
65 66
    return -1;
  }
M
Minghao Li 已提交
67

S
Shengliang Guan 已提交
68 69 70
  pSyncNode->rid = syncNodeAdd(pSyncNode);
  if (pSyncNode->rid < 0) {
    syncNodeClose(pSyncNode);
M
Minghao Li 已提交
71 72 73
    return -1;
  }

S
Shengliang Guan 已提交
74 75 76 77 78 79
  pSyncNode->pingBaseLine = pSyncInfo->pingMs;
  pSyncNode->pingTimerMS = pSyncInfo->pingMs;
  pSyncNode->electBaseLine = pSyncInfo->electMs;
  pSyncNode->hbBaseLine = pSyncInfo->heartbeatMs;
  pSyncNode->heartbeatTimerMS = pSyncInfo->heartbeatMs;
  pSyncNode->msgcb = pSyncInfo->msgcb;
S
Shengliang Guan 已提交
80
  return pSyncNode->rid;
M
Minghao Li 已提交
81
}
M
Minghao Li 已提交
82

M
Minghao Li 已提交
83
void syncStart(int64_t rid) {
S
Shengliang Guan 已提交
84 85 86 87
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
  if (pSyncNode != NULL) {
    syncNodeStart(pSyncNode);
    syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
88 89 90
  }
}

M
Minghao Li 已提交
91
void syncStop(int64_t rid) {
S
Shengliang Guan 已提交
92
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
S
Shengliang Guan 已提交
93
  if (pSyncNode != NULL) {
S
Shengliang Guan 已提交
94
    syncNodeRelease(pSyncNode);
S
Shengliang Guan 已提交
95
    syncNodeRemove(rid);
M
Minghao Li 已提交
96
  }
S
Shengliang Guan 已提交
97
}
M
Minghao Li 已提交
98

M
Minghao Li 已提交
99 100
void syncPreStop(int64_t rid) {
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
101 102 103 104
  if (pSyncNode != NULL) {
    syncNodePreClose(pSyncNode);
    syncNodeRelease(pSyncNode);
  }
M
Minghao Li 已提交
105 106
}

S
Shengliang Guan 已提交
107 108 109
static bool syncNodeCheckNewConfig(SSyncNode* pSyncNode, const SSyncCfg* pCfg) {
  if (!syncNodeInConfig(pSyncNode, pCfg)) return false;
  return abs(pCfg->replicaNum - pSyncNode->replicaNum) <= 1;
M
Minghao Li 已提交
110 111
}

S
Shengliang Guan 已提交
112
int32_t syncReconfig(int64_t rid, SSyncCfg* pNewCfg) {
S
Shengliang Guan 已提交
113
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
S
Shengliang Guan 已提交
114
  if (pSyncNode == NULL) return -1;
M
Minghao Li 已提交
115

M
Minghao Li 已提交
116
  if (!syncNodeCheckNewConfig(pSyncNode, pNewCfg)) {
S
Shengliang Guan 已提交
117
    syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
118
    terrno = TSDB_CODE_SYN_NEW_CONFIG_ERROR;
S
Shengliang Guan 已提交
119
    sError("vgId:%d, failed to reconfig since invalid new config", pSyncNode->vgId);
M
Minghao Li 已提交
120
    return -1;
M
Minghao Li 已提交
121
  }
122

S
Shengliang Guan 已提交
123 124
  syncNodeUpdateNewConfigIndex(pSyncNode, pNewCfg);
  syncNodeDoConfigChange(pSyncNode, pNewCfg, SYNC_INDEX_INVALID);
S
Shengliang Guan 已提交
125

M
Minghao Li 已提交
126 127 128 129
  if (pSyncNode->state == TAOS_SYNC_STATE_LEADER) {
    syncNodeStopHeartbeatTimer(pSyncNode);

    for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
S
Shengliang Guan 已提交
130
      syncHbTimerInit(pSyncNode, &pSyncNode->peerHeartbeatTimerArr[i], pSyncNode->replicasId[i]);
M
Minghao Li 已提交
131 132 133 134 135
    }

    syncNodeStartHeartbeatTimer(pSyncNode);
    syncNodeReplicate(pSyncNode);
  }
S
Shengliang Guan 已提交
136

S
Shengliang Guan 已提交
137
  syncNodeRelease(pSyncNode);
S
Shengliang Guan 已提交
138
  return 0;
M
Minghao Li 已提交
139
}
M
Minghao Li 已提交
140

S
Shengliang Guan 已提交
141 142 143 144
int32_t syncProcessMsg(int64_t rid, SRpcMsg* pMsg) {
  int32_t code = -1;
  if (!syncIsInit()) return code;

S
Shengliang Guan 已提交
145
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
S
Shengliang Guan 已提交
146 147
  if (pSyncNode == NULL) return code;

S
Shengliang Guan 已提交
148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185
  switch (pMsg->msgType) {
    case TDMT_SYNC_HEARTBEAT:
      code = syncNodeOnHeartbeat(pSyncNode, pMsg);
      break;
    case TDMT_SYNC_HEARTBEAT_REPLY:
      code = syncNodeOnHeartbeatReply(pSyncNode, pMsg);
      break;
    case TDMT_SYNC_TIMEOUT:
      code = syncNodeOnTimeout(pSyncNode, pMsg);
      break;
    case TDMT_SYNC_CLIENT_REQUEST:
      code = syncNodeOnClientRequest(pSyncNode, pMsg, NULL);
      break;
    case TDMT_SYNC_REQUEST_VOTE:
      code = syncNodeOnRequestVote(pSyncNode, pMsg);
      break;
    case TDMT_SYNC_REQUEST_VOTE_REPLY:
      code = syncNodeOnRequestVoteReply(pSyncNode, pMsg);
      break;
    case TDMT_SYNC_APPEND_ENTRIES:
      code = syncNodeOnAppendEntries(pSyncNode, pMsg);
      break;
    case TDMT_SYNC_APPEND_ENTRIES_REPLY:
      code = syncNodeOnAppendEntriesReply(pSyncNode, pMsg);
      break;
    case TDMT_SYNC_SNAPSHOT_SEND:
      code = syncNodeOnSnapshot(pSyncNode, pMsg);
      break;
    case TDMT_SYNC_SNAPSHOT_RSP:
      code = syncNodeOnSnapshotReply(pSyncNode, pMsg);
      break;
    case TDMT_SYNC_LOCAL_CMD:
      code = syncNodeOnLocalCmd(pSyncNode, pMsg);
      break;
    default:
      sError("vgId:%d, failed to process msg:%p since invalid type:%s", pSyncNode->vgId, pMsg,
             TMSG_INFO(pMsg->msgType));
      code = -1;
M
Minghao Li 已提交
186 187
  }

S
Shengliang Guan 已提交
188
  syncNodeRelease(pSyncNode);
S
Shengliang Guan 已提交
189
  return code;
190 191
}

S
Shengliang Guan 已提交
192
int32_t syncLeaderTransfer(int64_t rid) {
S
Shengliang Guan 已提交
193
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
S
Shengliang Guan 已提交
194
  if (pSyncNode == NULL) return -1;
195

S
Shengliang Guan 已提交
196
  int32_t ret = syncNodeLeaderTransfer(pSyncNode);
S
Shengliang Guan 已提交
197
  syncNodeRelease(pSyncNode);
198 199 200
  return ret;
}

M
Minghao Li 已提交
201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216
SyncIndex syncMinMatchIndex(SSyncNode* pSyncNode) {
  SyncIndex minMatchIndex = SYNC_INDEX_INVALID;

  if (pSyncNode->peersNum > 0) {
    minMatchIndex = syncIndexMgrGetIndex(pSyncNode->pMatchIndex, &(pSyncNode->peersId[0]));
  }

  for (int32_t i = 1; i < pSyncNode->peersNum; ++i) {
    SyncIndex matchIndex = syncIndexMgrGetIndex(pSyncNode->pMatchIndex, &(pSyncNode->peersId[i]));
    if (matchIndex < minMatchIndex) {
      minMatchIndex = matchIndex;
    }
  }
  return minMatchIndex;
}

217
int32_t syncBeginSnapshot(int64_t rid, int64_t lastApplyIndex) {
S
Shengliang Guan 已提交
218
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
219
  if (pSyncNode == NULL) {
220
    sError("sync begin snapshot error");
221 222
    return -1;
  }
223

224 225
  int32_t code = 0;

M
Minghao Li 已提交
226
  if (syncNodeIsMnode(pSyncNode)) {
M
Minghao Li 已提交
227 228 229
    // mnode
    int64_t logRetention = SYNC_MNODE_LOG_RETENTION;

M
Minghao Li 已提交
230 231 232
    SyncIndex beginIndex = pSyncNode->pLogStore->syncLogBeginIndex(pSyncNode->pLogStore);
    SyncIndex endIndex = pSyncNode->pLogStore->syncLogEndIndex(pSyncNode->pLogStore);
    int64_t   logNum = endIndex - beginIndex;
M
Minghao Li 已提交
233 234 235
    bool      isEmpty = pSyncNode->pLogStore->syncLogIsEmpty(pSyncNode->pLogStore);

    if (isEmpty || (!isEmpty && logNum < logRetention)) {
S
Shengliang Guan 已提交
236 237
      sNTrace(pSyncNode, "new-snapshot-index:%" PRId64 ", log-num:%" PRId64 ", empty:%d, do not delete wal",
              lastApplyIndex, logNum, isEmpty);
S
Shengliang Guan 已提交
238
      syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
239 240 241
      return 0;
    }

M
Minghao Li 已提交
242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258
    goto _DEL_WAL;

  } else {
    // vnode
    if (pSyncNode->replicaNum > 1) {
      // multi replicas

      if (pSyncNode->state == TAOS_SYNC_STATE_LEADER) {
        pSyncNode->minMatchIndex = syncMinMatchIndex(pSyncNode);

        for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
          int64_t matchIndex = syncIndexMgrGetIndex(pSyncNode->pMatchIndex, &(pSyncNode->peersId[i]));
          if (lastApplyIndex > matchIndex) {
            do {
              char     host[64];
              uint16_t port;
              syncUtilU642Addr(pSyncNode->peersId[i].addr, host, sizeof(host), &port);
S
Shengliang Guan 已提交
259 260 261 262
              sNTrace(pSyncNode,
                      "new-snapshot-index:%" PRId64 " is greater than match-index:%" PRId64
                      " of %s:%d, do not delete wal",
                      lastApplyIndex, matchIndex, host, port);
M
Minghao Li 已提交
263 264
            } while (0);

S
Shengliang Guan 已提交
265
            syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
266 267 268 269 270 271
            return 0;
          }
        }

      } else if (pSyncNode->state == TAOS_SYNC_STATE_FOLLOWER) {
        if (lastApplyIndex > pSyncNode->minMatchIndex) {
S
Shengliang Guan 已提交
272 273 274
          sNTrace(pSyncNode,
                  "new-snapshot-index:%" PRId64 " is greater than min-match-index:%" PRId64 ", do not delete wal",
                  lastApplyIndex, pSyncNode->minMatchIndex);
S
Shengliang Guan 已提交
275
          syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
276 277 278 279
          return 0;
        }

      } else if (pSyncNode->state == TAOS_SYNC_STATE_CANDIDATE) {
S
Shengliang Guan 已提交
280
        sNTrace(pSyncNode, "new-snapshot-index:%" PRId64 " candidate, do not delete wal", lastApplyIndex);
S
Shengliang Guan 已提交
281
        syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
282 283 284
        return 0;

      } else {
S
Shengliang Guan 已提交
285
        sNTrace(pSyncNode, "new-snapshot-index:%" PRId64 " unknown state, do not delete wal", lastApplyIndex);
S
Shengliang Guan 已提交
286
        syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
287 288 289 290 291 292 293 294 295
        return 0;
      }

      goto _DEL_WAL;

    } else {
      // one replica

      goto _DEL_WAL;
296 297 298
    }
  }

M
Minghao Li 已提交
299
_DEL_WAL:
300

M
Minghao Li 已提交
301
  do {
302 303 304 305
    SyncIndex snapshottingIndex = atomic_load_64(&pSyncNode->snapshottingIndex);

    if (snapshottingIndex == SYNC_INDEX_INVALID) {
      atomic_store_64(&pSyncNode->snapshottingIndex, lastApplyIndex);
M
Minghao Li 已提交
306
      pSyncNode->snapshottingTime = taosGetTimestampMs();
307

M
Minghao Li 已提交
308 309 310
      SSyncLogStoreData* pData = pSyncNode->pLogStore->data;
      code = walBeginSnapshot(pData->pWal, lastApplyIndex);
      if (code == 0) {
S
Shengliang Guan 已提交
311 312
        sNTrace(pSyncNode, "wal snapshot begin, index:%" PRId64 ", last apply index:%" PRId64,
                pSyncNode->snapshottingIndex, lastApplyIndex);
M
Minghao Li 已提交
313
      } else {
S
Shengliang Guan 已提交
314
        sNError(pSyncNode, "wal snapshot begin error since:%s, index:%" PRId64 ", last apply index:%" PRId64,
S
Shengliang Guan 已提交
315
                terrstr(terrno), pSyncNode->snapshottingIndex, lastApplyIndex);
M
Minghao Li 已提交
316 317
        atomic_store_64(&pSyncNode->snapshottingIndex, SYNC_INDEX_INVALID);
      }
318 319

    } else {
S
Shengliang Guan 已提交
320 321
      sNTrace(pSyncNode, "snapshotting for %" PRId64 ", do not delete wal for new-snapshot-index:%" PRId64,
              snapshottingIndex, lastApplyIndex);
322
    }
M
Minghao Li 已提交
323
  } while (0);
324

S
Shengliang Guan 已提交
325
  syncNodeRelease(pSyncNode);
326 327 328 329
  return code;
}

int32_t syncEndSnapshot(int64_t rid) {
S
Shengliang Guan 已提交
330
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
331
  if (pSyncNode == NULL) {
332
    sError("sync end snapshot error");
333 334 335
    return -1;
  }

336 337 338 339
  int32_t code = 0;
  if (atomic_load_64(&pSyncNode->snapshottingIndex) != SYNC_INDEX_INVALID) {
    SSyncLogStoreData* pData = pSyncNode->pLogStore->data;
    code = walEndSnapshot(pData->pWal);
M
Minghao Li 已提交
340
    if (code != 0) {
341
      sNError(pSyncNode, "wal snapshot end error since:%s", terrstr());
S
Shengliang Guan 已提交
342
      syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
343 344
      return -1;
    } else {
S
Shengliang Guan 已提交
345
      sNTrace(pSyncNode, "wal snapshot end, index:%" PRId64, atomic_load_64(&pSyncNode->snapshottingIndex));
M
Minghao Li 已提交
346 347
      atomic_store_64(&pSyncNode->snapshottingIndex, SYNC_INDEX_INVALID);
    }
348
  }
349

S
Shengliang Guan 已提交
350
  syncNodeRelease(pSyncNode);
351 352 353
  return code;
}

M
Minghao Li 已提交
354
int32_t syncStepDown(int64_t rid, SyncTerm newTerm) {
S
Shengliang Guan 已提交
355
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
M
Minghao Li 已提交
356
  if (pSyncNode == NULL) {
357
    sError("sync step down error");
M
Minghao Li 已提交
358 359 360 361
    return -1;
  }

  syncNodeStepDown(pSyncNode, newTerm);
S
Shengliang Guan 已提交
362
  syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
363 364 365
  return 0;
}

366 367 368
bool syncIsReadyForRead(int64_t rid) {
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
  if (pSyncNode == NULL) {
369
    sError("sync ready for read error");
370 371 372 373 374 375 376 377 378 379
    return false;
  }

  if (pSyncNode->state == TAOS_SYNC_STATE_LEADER && pSyncNode->restoreFinish) {
    syncNodeRelease(pSyncNode);
    return true;
  }

  bool ready = false;
  if (pSyncNode->state == TAOS_SYNC_STATE_LEADER && !pSyncNode->restoreFinish) {
380 381 382
    if (!pSyncNode->pFsm->FpApplyQueueEmptyCb(pSyncNode->pFsm)) {
      // apply queue not empty
      ready = false;
383

384 385 386 387
    } else {
      if (!pSyncNode->pLogStore->syncLogIsEmpty(pSyncNode->pLogStore)) {
        SSyncRaftEntry* pEntry = NULL;
        int32_t         code = pSyncNode->pLogStore->syncLogGetEntry(
388
            pSyncNode->pLogStore, pSyncNode->pLogStore->syncLogLastIndex(pSyncNode->pLogStore), &pEntry);
389 390 391 392 393 394 395
        if (code == 0 && pEntry != NULL) {
          if (pEntry->originalRpcType == TDMT_SYNC_NOOP && pEntry->term == pSyncNode->pRaftStore->currentTerm) {
            ready = true;
          }

          syncEntryDestory(pEntry);
        }
396 397 398 399
      }
    }
  }

400 401 402 403 404 405 406 407
  if (!ready) {
    if (pSyncNode->state != TAOS_SYNC_STATE_LEADER) {
      terrno = TSDB_CODE_SYN_NOT_LEADER;
    } else {
      terrno = TSDB_CODE_APP_NOT_READY;
    }
  }

408 409 410 411
  syncNodeRelease(pSyncNode);
  return ready;
}

M
Minghao Li 已提交
412 413
int32_t syncNodeLeaderTransfer(SSyncNode* pSyncNode) {
  if (pSyncNode->peersNum == 0) {
414
    sDebug("only one replica, cannot leader transfer");
M
Minghao Li 已提交
415 416 417 418
    terrno = TSDB_CODE_SYN_ONE_REPLICA;
    return -1;
  }

419 420 421 422 423 424
  int32_t ret = 0;
  if (pSyncNode->state == TAOS_SYNC_STATE_LEADER) {
    SNodeInfo newLeader = (pSyncNode->peersNodeInfo)[0];
    ret = syncNodeLeaderTransferTo(pSyncNode, newLeader);
  }

M
Minghao Li 已提交
425 426 427 428 429
  return ret;
}

int32_t syncNodeLeaderTransferTo(SSyncNode* pSyncNode, SNodeInfo newLeader) {
  if (pSyncNode->replicaNum == 1) {
430
    sDebug("only one replica, cannot leader transfer");
M
Minghao Li 已提交
431 432 433 434
    terrno = TSDB_CODE_SYN_ONE_REPLICA;
    return -1;
  }

S
Shengliang Guan 已提交
435
  sNTrace(pSyncNode, "begin leader transfer to %s:%u", newLeader.nodeFqdn, newLeader.nodePort);
M
Minghao Li 已提交
436

437 438 439 440
  SRpcMsg rpcMsg = {0};
  (void)syncBuildLeaderTransfer(&rpcMsg, pSyncNode->vgId);

  SyncLeaderTransfer* pMsg = rpcMsg.pCont;
M
Minghao Li 已提交
441 442 443 444
  pMsg->newLeaderId.addr = syncUtilAddr2U64(newLeader.nodeFqdn, newLeader.nodePort);
  pMsg->newLeaderId.vgId = pSyncNode->vgId;
  pMsg->newNodeInfo = newLeader;

445
  return syncNodePropose(pSyncNode, &rpcMsg, false);
M
Minghao Li 已提交
446 447
}

448 449
SSyncState syncGetState(int64_t rid) {
  SSyncState state = {.state = TAOS_SYNC_STATE_ERROR};
M
Minghao Li 已提交
450

S
Shengliang Guan 已提交
451
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
452 453 454 455
  if (pSyncNode != NULL) {
    state.state = pSyncNode->state;
    state.restored = pSyncNode->restoreFinish;
    syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
456 457
  }

458
  return state;
M
Minghao Li 已提交
459 460
}

461
#if 0
462 463 464 465 466
int32_t syncGetSnapshotByIndex(int64_t rid, SyncIndex index, SSnapshot* pSnapshot) {
  if (index < SYNC_INDEX_BEGIN) {
    return -1;
  }

S
Shengliang Guan 已提交
467
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
468 469 470 471 472 473 474 475 476 477 478
  if (pSyncNode == NULL) {
    return -1;
  }
  ASSERT(rid == pSyncNode->rid);

  SSyncRaftEntry* pEntry = NULL;
  int32_t         code = pSyncNode->pLogStore->syncLogGetEntry(pSyncNode->pLogStore, index, &pEntry);
  if (code != 0) {
    if (pEntry != NULL) {
      syncEntryDestory(pEntry);
    }
S
Shengliang Guan 已提交
479
    syncNodeRelease(pSyncNode);
480 481 482 483 484 485 486 487 488 489
    return -1;
  }
  ASSERT(pEntry != NULL);

  pSnapshot->data = NULL;
  pSnapshot->lastApplyIndex = index;
  pSnapshot->lastApplyTerm = pEntry->term;
  pSnapshot->lastConfigIndex = syncNodeGetSnapshotConfigIndex(pSyncNode, index);

  syncEntryDestory(pEntry);
S
Shengliang Guan 已提交
490
  syncNodeRelease(pSyncNode);
491 492 493
  return 0;
}

494
int32_t syncGetSnapshotMeta(int64_t rid, struct SSnapshotMeta* sMeta) {
S
Shengliang Guan 已提交
495
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
496 497 498
  if (pSyncNode == NULL) {
    return -1;
  }
M
Minghao Li 已提交
499
  ASSERT(rid == pSyncNode->rid);
500 501
  sMeta->lastConfigIndex = pSyncNode->pRaftCfg->lastConfigIndex;

S
Shengliang Guan 已提交
502
  sTrace("vgId:%d, get snapshot meta, lastConfigIndex:%" PRId64, pSyncNode->vgId, pSyncNode->pRaftCfg->lastConfigIndex);
503

S
Shengliang Guan 已提交
504
  syncNodeRelease(pSyncNode);
505 506 507
  return 0;
}

508
int32_t syncGetSnapshotMetaByIndex(int64_t rid, SyncIndex snapshotIndex, struct SSnapshotMeta* sMeta) {
S
Shengliang Guan 已提交
509
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
510 511 512
  if (pSyncNode == NULL) {
    return -1;
  }
M
Minghao Li 已提交
513
  ASSERT(rid == pSyncNode->rid);
514 515 516 517

  ASSERT(pSyncNode->pRaftCfg->configIndexCount >= 1);
  SyncIndex lastIndex = (pSyncNode->pRaftCfg->configIndexArr)[0];

S
Shengliang Guan 已提交
518
  for (int32_t i = 0; i < pSyncNode->pRaftCfg->configIndexCount; ++i) {
519 520 521 522 523 524
    if ((pSyncNode->pRaftCfg->configIndexArr)[i] > lastIndex &&
        (pSyncNode->pRaftCfg->configIndexArr)[i] <= snapshotIndex) {
      lastIndex = (pSyncNode->pRaftCfg->configIndexArr)[i];
    }
  }
  sMeta->lastConfigIndex = lastIndex;
525
  sTrace("vgId:%d, get snapshot meta by index:%" PRId64 " lcindex:%" PRId64, pSyncNode->vgId, snapshotIndex,
S
Shengliang Guan 已提交
526
         sMeta->lastConfigIndex);
527

S
Shengliang Guan 已提交
528
  syncNodeRelease(pSyncNode);
529 530
  return 0;
}
531
#endif
532

533 534 535 536
SyncIndex syncNodeGetSnapshotConfigIndex(SSyncNode* pSyncNode, SyncIndex snapshotLastApplyIndex) {
  ASSERT(pSyncNode->pRaftCfg->configIndexCount >= 1);
  SyncIndex lastIndex = (pSyncNode->pRaftCfg->configIndexArr)[0];

S
Shengliang Guan 已提交
537
  for (int32_t i = 0; i < pSyncNode->pRaftCfg->configIndexCount; ++i) {
538 539 540 541 542
    if ((pSyncNode->pRaftCfg->configIndexArr)[i] > lastIndex &&
        (pSyncNode->pRaftCfg->configIndexArr)[i] <= snapshotLastApplyIndex) {
      lastIndex = (pSyncNode->pRaftCfg->configIndexArr)[i];
    }
  }
S
Shengliang Guan 已提交
543
  sTrace("vgId:%d, sync get last config index, index:%" PRId64 " lcindex:%" PRId64, pSyncNode->vgId,
S
Shengliang Guan 已提交
544
         snapshotLastApplyIndex, lastIndex);
545 546 547 548

  return lastIndex;
}

S
Shengliang Guan 已提交
549
void syncGetRetryEpSet(int64_t rid, SEpSet* pEpSet) {
M
Minghao Li 已提交
550
  pEpSet->numOfEps = 0;
551

S
Shengliang Guan 已提交
552
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
S
Shengliang Guan 已提交
553
  if (pSyncNode == NULL) return;
554

S
Shengliang Guan 已提交
555
  for (int32_t i = 0; i < pSyncNode->pRaftCfg->cfg.replicaNum; ++i) {
S
Shengliang Guan 已提交
556 557 558 559 560
    SEp* pEp = &pEpSet->eps[i];
    tstrncpy(pEp->fqdn, pSyncNode->pRaftCfg->cfg.nodeInfo[i].nodeFqdn, TSDB_FQDN_LEN);
    pEp->port = (pSyncNode->pRaftCfg->cfg.nodeInfo)[i].nodePort;
    pEpSet->numOfEps++;
    sInfo("vgId:%d, sync get retry epset, index:%d %s:%d", pSyncNode->vgId, i, pEp->fqdn, pEp->port);
561
  }
M
Minghao Li 已提交
562 563 564
  if (pEpSet->numOfEps > 0) {
    pEpSet->inUse = (pSyncNode->pRaftCfg->cfg.myIndex + 1) % pEpSet->numOfEps;
  }
M
Minghao Li 已提交
565

S
Shengliang Guan 已提交
566
  sInfo("vgId:%d, sync get retry epset numOfEps:%d inUse:%d", pSyncNode->vgId, pEpSet->numOfEps, pEpSet->inUse);
S
Shengliang Guan 已提交
567
  syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
568
}
569

M
Minghao Li 已提交
570
int32_t syncPropose(int64_t rid, SRpcMsg* pMsg, bool isWeak) {
S
Shengliang Guan 已提交
571
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
572
  if (pSyncNode == NULL) {
573
    sError("sync propose error");
M
Minghao Li 已提交
574
    return -1;
575
  }
M
Minghao Li 已提交
576

577
  int32_t ret = syncNodePropose(pSyncNode, pMsg, isWeak);
S
Shengliang Guan 已提交
578
  syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
579 580 581
  return ret;
}

582
int32_t syncNodePropose(SSyncNode* pSyncNode, SRpcMsg* pMsg, bool isWeak) {
S
Shengliang Guan 已提交
583 584 585 586 587
  if (pSyncNode->state != TAOS_SYNC_STATE_LEADER) {
    terrno = TSDB_CODE_SYN_NOT_LEADER;
    sNError(pSyncNode, "sync propose not leader, %s, type:%s", syncStr(pSyncNode->state), TMSG_INFO(pMsg->msgType));
    return -1;
  }
M
Minghao Li 已提交
588

S
Shengliang Guan 已提交
589 590 591 592 593 594 595 596 597 598 599
  // not restored, vnode enable
  if (!pSyncNode->restoreFinish && pSyncNode->vgId != 1) {
    terrno = TSDB_CODE_SYN_PROPOSE_NOT_READY;
    sNError(pSyncNode, "failed to sync propose since not ready, type:%s, last:%" PRId64 ", cmt:%" PRId64,
            TMSG_INFO(pMsg->msgType), syncNodeGetLastIndex(pSyncNode), pSyncNode->commitIndex);
    return -1;
  }

  // optimized one replica
  if (syncNodeIsOptimizedOneReplica(pSyncNode, pMsg)) {
    SyncIndex retIndex;
600
    int32_t   code = syncNodeOnClientRequest(pSyncNode, pMsg, &retIndex);
S
Shengliang Guan 已提交
601 602 603
    if (code == 0) {
      pMsg->info.conn.applyIndex = retIndex;
      pMsg->info.conn.applyTerm = pSyncNode->pRaftStore->currentTerm;
604 605 606
      sTrace("vgId:%d, propose optimized msg, index:%" PRId64 " type:%s", pSyncNode->vgId, retIndex,
             TMSG_INFO(pMsg->msgType));
      return 1;
S
Shengliang Guan 已提交
607 608
    } else {
      terrno = TSDB_CODE_SYN_INTERNAL_ERROR;
609
      sError("vgId:%d, failed to propose optimized msg, index:%" PRId64 " type:%s", pSyncNode->vgId, retIndex,
S
Shengliang Guan 已提交
610
             TMSG_INFO(pMsg->msgType));
611
      return -1;
612
    }
S
Shengliang Guan 已提交
613
  } else {
S
Shengliang Guan 已提交
614 615
    SRespStub stub = {.createTime = taosGetTimestampMs(), .rpcMsg = *pMsg};
    uint64_t  seqNum = syncRespMgrAdd(pSyncNode->pSyncRespMgr, &stub);
616
    SRpcMsg   rpcMsg = {0};
S
Shengliang Guan 已提交
617
    int32_t   code = syncBuildClientRequest(&rpcMsg, pMsg, seqNum, isWeak, pSyncNode->vgId);
618 619 620 621 622
    if (code != 0) {
      sError("vgId:%d, failed to propose msg while serialize since %s", pSyncNode->vgId, terrstr());
      (void)syncRespMgrDel(pSyncNode->pSyncRespMgr, seqNum);
      return -1;
    }
M
Minghao Li 已提交
623

624 625 626 627 628
    sNTrace(pSyncNode, "propose msg, type:%s", TMSG_INFO(pMsg->msgType));
    code = (*pSyncNode->syncEqMsg)(pSyncNode->msgcb, &rpcMsg);
    if (code != 0) {
      sError("vgId:%d, failed to propose msg while enqueue since %s", pSyncNode->vgId, terrstr());
      (void)syncRespMgrDel(pSyncNode->pSyncRespMgr, seqNum);
M
Minghao Li 已提交
629
    }
M
Minghao Li 已提交
630

631 632
    return code;
  }
M
Minghao Li 已提交
633 634
}

S
Shengliang Guan 已提交
635
static int32_t syncHbTimerInit(SSyncNode* pSyncNode, SSyncTimer* pSyncTimer, SRaftId destId) {
636 637 638 639 640 641 642 643 644
  pSyncTimer->pTimer = NULL;
  pSyncTimer->counter = 0;
  pSyncTimer->timerMS = pSyncNode->hbBaseLine;
  pSyncTimer->timerCb = syncNodeEqPeerHeartbeatTimer;
  pSyncTimer->destId = destId;
  atomic_store_64(&pSyncTimer->logicClock, 0);
  return 0;
}

S
Shengliang Guan 已提交
645
static int32_t syncHbTimerStart(SSyncNode* pSyncNode, SSyncTimer* pSyncTimer) {
646
  int32_t ret = 0;
S
Shengliang Guan 已提交
647
  if (syncIsInit()) {
M
Minghao Li 已提交
648
    SSyncHbTimerData* pData = taosMemoryMalloc(sizeof(SSyncHbTimerData));
649 650 651 652
    pData->pSyncNode = pSyncNode;
    pData->pTimer = pSyncTimer;
    pData->destId = pSyncTimer->destId;
    pData->logicClock = pSyncTimer->logicClock;
M
Minghao Li 已提交
653

654
    pSyncTimer->pData = pData;
S
Shengliang Guan 已提交
655
    taosTmrReset(pSyncTimer->timerCb, pSyncTimer->timerMS, pData, syncEnv()->pTimerManager, &pSyncTimer->pTimer);
656 657 658 659 660 661
  } else {
    sError("vgId:%d, start ctrl hb timer error, sync env is stop", pSyncNode->vgId);
  }
  return ret;
}

S
Shengliang Guan 已提交
662
static int32_t syncHbTimerStop(SSyncNode* pSyncNode, SSyncTimer* pSyncTimer) {
663 664 665 666
  int32_t ret = 0;
  atomic_add_fetch_64(&pSyncTimer->logicClock, 1);
  taosTmrStop(pSyncTimer->pTimer);
  pSyncTimer->pTimer = NULL;
M
Minghao Li 已提交
667
  // taosMemoryFree(pSyncTimer->pData);
668 669 670
  return ret;
}

S
Shengliang Guan 已提交
671 672
SSyncNode* syncNodeOpen(SSyncInfo* pSyncInfo) {
  SSyncNode* pSyncNode = taosMemoryCalloc(1, sizeof(SSyncNode));
673 674 675 676
  if (pSyncNode == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _error;
  }
M
Minghao Li 已提交
677

M
Minghao Li 已提交
678 679 680 681
  if (!taosDirExist((char*)(pSyncInfo->path))) {
    if (taosMkDir(pSyncInfo->path) != 0) {
      terrno = TAOS_SYSTEM_ERROR(errno);
      sError("failed to create dir:%s since %s", pSyncInfo->path, terrstr());
682
      goto _error;
M
Minghao Li 已提交
683
    }
684
  }
M
Minghao Li 已提交
685

S
Shengliang Guan 已提交
686
  snprintf(pSyncNode->configPath, sizeof(pSyncNode->configPath), "%s%sraft_config.json", pSyncInfo->path, TD_DIRSEP);
687
  if (!taosCheckExistFile(pSyncNode->configPath)) {
M
Minghao Li 已提交
688
    // create a new raft config file
S
Shengliang Guan 已提交
689
    SRaftCfgMeta meta = {0};
M
Minghao Li 已提交
690
    meta.isStandBy = pSyncInfo->isStandBy;
M
Minghao Li 已提交
691
    meta.snapshotStrategy = pSyncInfo->snapshotStrategy;
692
    meta.lastConfigIndex = SYNC_INDEX_INVALID;
M
Minghao Li 已提交
693
    meta.batchSize = pSyncInfo->batchSize;
S
Shengliang Guan 已提交
694 695
    if (raftCfgCreateFile(&pSyncInfo->syncCfg, meta, pSyncNode->configPath) != 0) {
      sError("vgId:%d, failed to create raft cfg file at %s", pSyncNode->vgId, pSyncNode->configPath);
H
Hongze Cheng 已提交
696
      goto _error;
697
    }
698
    if (pSyncInfo->syncCfg.replicaNum == 0) {
S
Shengliang Guan 已提交
699
      sInfo("vgId:%d, sync config not input", pSyncNode->vgId);
700 701
      pSyncInfo->syncCfg = pSyncNode->pRaftCfg->cfg;
    }
702 703 704
  } else {
    // update syncCfg by raft_config.json
    pSyncNode->pRaftCfg = raftCfgOpen(pSyncNode->configPath);
705
    if (pSyncNode->pRaftCfg == NULL) {
S
Shengliang Guan 已提交
706
      sError("vgId:%d, failed to open raft cfg file at %s", pSyncNode->vgId, pSyncNode->configPath);
H
Hongze Cheng 已提交
707
      goto _error;
708
    }
S
Shengliang Guan 已提交
709 710

    if (pSyncInfo->syncCfg.replicaNum > 0 && syncIsConfigChanged(&pSyncNode->pRaftCfg->cfg, &pSyncInfo->syncCfg)) {
S
Shengliang Guan 已提交
711 712 713 714 715 716
      sInfo("vgId:%d, use sync config from input options and write to cfg file", pSyncNode->vgId);
      pSyncNode->pRaftCfg->cfg = pSyncInfo->syncCfg;
      if (raftCfgPersist(pSyncNode->pRaftCfg) != 0) {
        sError("vgId:%d, failed to persist raft cfg file at %s", pSyncNode->vgId, pSyncNode->configPath);
        goto _error;
      }
S
Shengliang Guan 已提交
717 718 719 720
    } else {
      sInfo("vgId:%d, use sync config from raft cfg file", pSyncNode->vgId);
      pSyncInfo->syncCfg = pSyncNode->pRaftCfg->cfg;
    }
721 722

    raftCfgClose(pSyncNode->pRaftCfg);
723
    pSyncNode->pRaftCfg = NULL;
M
Minghao Li 已提交
724 725
  }

S
Shengliang Guan 已提交
726 727
  // init by SSyncInfo
  pSyncNode->vgId = pSyncInfo->vgId;
S
Shengliang Guan 已提交
728 729 730 731 732 733 734
  SSyncCfg* pCfg = &pSyncInfo->syncCfg;
  sDebug("vgId:%d, replica:%d selfIndex:%d", pSyncNode->vgId, pCfg->replicaNum, pCfg->myIndex);
  for (int32_t i = 0; i < pCfg->replicaNum; ++i) {
    SNodeInfo* pNode = &pCfg->nodeInfo[i];
    sDebug("vgId:%d, index:%d ep:%s:%u", pSyncNode->vgId, i, pNode->nodeFqdn, pNode->nodePort);
  }

M
Minghao Li 已提交
735
  memcpy(pSyncNode->path, pSyncInfo->path, sizeof(pSyncNode->path));
S
Shengliang Guan 已提交
736 737 738
  snprintf(pSyncNode->raftStorePath, sizeof(pSyncNode->raftStorePath), "%s%sraft_store.json", pSyncInfo->path,
           TD_DIRSEP);
  snprintf(pSyncNode->configPath, sizeof(pSyncNode->configPath), "%s%sraft_config.json", pSyncInfo->path, TD_DIRSEP);
M
Minghao Li 已提交
739

M
Minghao Li 已提交
740
  pSyncNode->pWal = pSyncInfo->pWal;
S
Shengliang Guan 已提交
741
  pSyncNode->msgcb = pSyncInfo->msgcb;
S
Shengliang Guan 已提交
742 743 744
  pSyncNode->syncSendMSg = pSyncInfo->syncSendMSg;
  pSyncNode->syncEqMsg = pSyncInfo->syncEqMsg;
  pSyncNode->syncEqCtrlMsg = pSyncInfo->syncEqCtrlMsg;
M
Minghao Li 已提交
745

M
Minghao Li 已提交
746 747
  // init raft config
  pSyncNode->pRaftCfg = raftCfgOpen(pSyncNode->configPath);
748
  if (pSyncNode->pRaftCfg == NULL) {
S
Shengliang Guan 已提交
749
    sError("vgId:%d, failed to open raft cfg file at %s", pSyncNode->vgId, pSyncNode->configPath);
750 751
    goto _error;
  }
M
Minghao Li 已提交
752

M
Minghao Li 已提交
753
  // init internal
M
Minghao Li 已提交
754
  pSyncNode->myNodeInfo = pSyncNode->pRaftCfg->cfg.nodeInfo[pSyncNode->pRaftCfg->cfg.myIndex];
755
  if (!syncUtilNodeInfo2RaftId(&pSyncNode->myNodeInfo, pSyncNode->vgId, &pSyncNode->myRaftId)) {
S
Shengliang Guan 已提交
756
    sError("vgId:%d, failed to determine my raft member id", pSyncNode->vgId);
H
Hongze Cheng 已提交
757
    goto _error;
758
  }
M
Minghao Li 已提交
759

M
Minghao Li 已提交
760
  // init peersNum, peers, peersId
M
Minghao Li 已提交
761
  pSyncNode->peersNum = pSyncNode->pRaftCfg->cfg.replicaNum - 1;
S
Shengliang Guan 已提交
762 763
  int32_t j = 0;
  for (int32_t i = 0; i < pSyncNode->pRaftCfg->cfg.replicaNum; ++i) {
M
Minghao Li 已提交
764 765
    if (i != pSyncNode->pRaftCfg->cfg.myIndex) {
      pSyncNode->peersNodeInfo[j] = pSyncNode->pRaftCfg->cfg.nodeInfo[i];
M
Minghao Li 已提交
766 767 768
      j++;
    }
  }
S
Shengliang Guan 已提交
769
  for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
770
    if (!syncUtilNodeInfo2RaftId(&pSyncNode->peersNodeInfo[i], pSyncNode->vgId, &pSyncNode->peersId[i])) {
S
Shengliang Guan 已提交
771
      sError("vgId:%d, failed to determine raft member id, peer:%d", pSyncNode->vgId, i);
H
Hongze Cheng 已提交
772
      goto _error;
773
    }
M
Minghao Li 已提交
774
  }
M
Minghao Li 已提交
775

M
Minghao Li 已提交
776
  // init replicaNum, replicasId
M
Minghao Li 已提交
777
  pSyncNode->replicaNum = pSyncNode->pRaftCfg->cfg.replicaNum;
S
Shengliang Guan 已提交
778
  for (int32_t i = 0; i < pSyncNode->pRaftCfg->cfg.replicaNum; ++i) {
779
    if (!syncUtilNodeInfo2RaftId(&pSyncNode->pRaftCfg->cfg.nodeInfo[i], pSyncNode->vgId, &pSyncNode->replicasId[i])) {
S
Shengliang Guan 已提交
780
      sError("vgId:%d, failed to determine raft member id, replica:%d", pSyncNode->vgId, i);
H
Hongze Cheng 已提交
781
      goto _error;
782
    }
M
Minghao Li 已提交
783 784
  }

M
Minghao Li 已提交
785
  // init raft algorithm
M
Minghao Li 已提交
786
  pSyncNode->pFsm = pSyncInfo->pFsm;
787
  pSyncInfo->pFsm = NULL;
M
Minghao Li 已提交
788
  pSyncNode->quorum = syncUtilQuorum(pSyncNode->pRaftCfg->cfg.replicaNum);
M
Minghao Li 已提交
789 790
  pSyncNode->leaderCache = EMPTY_RAFT_ID;

M
Minghao Li 已提交
791
  // init life cycle outside
M
Minghao Li 已提交
792

M
Minghao Li 已提交
793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816
  // TLA+ Spec
  // InitHistoryVars == /\ elections = {}
  //                    /\ allLogs   = {}
  //                    /\ voterLog  = [i \in Server |-> [j \in {} |-> <<>>]]
  // InitServerVars == /\ currentTerm = [i \in Server |-> 1]
  //                   /\ state       = [i \in Server |-> Follower]
  //                   /\ votedFor    = [i \in Server |-> Nil]
  // InitCandidateVars == /\ votesResponded = [i \in Server |-> {}]
  //                      /\ votesGranted   = [i \in Server |-> {}]
  // \* The values nextIndex[i][i] and matchIndex[i][i] are never read, since the
  // \* leader does not send itself messages. It's still easier to include these
  // \* in the functions.
  // InitLeaderVars == /\ nextIndex  = [i \in Server |-> [j \in Server |-> 1]]
  //                   /\ matchIndex = [i \in Server |-> [j \in Server |-> 0]]
  // InitLogVars == /\ log          = [i \in Server |-> << >>]
  //                /\ commitIndex  = [i \in Server |-> 0]
  // Init == /\ messages = [m \in {} |-> 0]
  //         /\ InitHistoryVars
  //         /\ InitServerVars
  //         /\ InitCandidateVars
  //         /\ InitLeaderVars
  //         /\ InitLogVars
  //

M
Minghao Li 已提交
817
  // init TLA+ server vars
M
syncInt  
Minghao Li 已提交
818
  pSyncNode->state = TAOS_SYNC_STATE_FOLLOWER;
M
Minghao Li 已提交
819
  pSyncNode->pRaftStore = raftStoreOpen(pSyncNode->raftStorePath);
820
  if (pSyncNode->pRaftStore == NULL) {
S
Shengliang Guan 已提交
821
    sError("vgId:%d, failed to open raft store at path %s", pSyncNode->vgId, pSyncNode->raftStorePath);
822 823
    goto _error;
  }
M
Minghao Li 已提交
824

M
Minghao Li 已提交
825
  // init TLA+ candidate vars
M
Minghao Li 已提交
826
  pSyncNode->pVotesGranted = voteGrantedCreate(pSyncNode);
827
  if (pSyncNode->pVotesGranted == NULL) {
S
Shengliang Guan 已提交
828
    sError("vgId:%d, failed to create VotesGranted", pSyncNode->vgId);
829 830
    goto _error;
  }
M
Minghao Li 已提交
831
  pSyncNode->pVotesRespond = votesRespondCreate(pSyncNode);
832
  if (pSyncNode->pVotesRespond == NULL) {
S
Shengliang Guan 已提交
833
    sError("vgId:%d, failed to create VotesRespond", pSyncNode->vgId);
834 835
    goto _error;
  }
M
Minghao Li 已提交
836

M
Minghao Li 已提交
837 838
  // init TLA+ leader vars
  pSyncNode->pNextIndex = syncIndexMgrCreate(pSyncNode);
839
  if (pSyncNode->pNextIndex == NULL) {
S
Shengliang Guan 已提交
840
    sError("vgId:%d, failed to create SyncIndexMgr", pSyncNode->vgId);
841 842
    goto _error;
  }
M
Minghao Li 已提交
843
  pSyncNode->pMatchIndex = syncIndexMgrCreate(pSyncNode);
844
  if (pSyncNode->pMatchIndex == NULL) {
S
Shengliang Guan 已提交
845
    sError("vgId:%d, failed to create SyncIndexMgr", pSyncNode->vgId);
846 847
    goto _error;
  }
M
Minghao Li 已提交
848 849 850

  // init TLA+ log vars
  pSyncNode->pLogStore = logStoreCreate(pSyncNode);
851
  if (pSyncNode->pLogStore == NULL) {
S
Shengliang Guan 已提交
852
    sError("vgId:%d, failed to create SyncLogStore", pSyncNode->vgId);
853 854
    goto _error;
  }
855 856 857 858 859

  SyncIndex commitIndex = SYNC_INDEX_INVALID;
  if (pSyncNode->pFsm != NULL && pSyncNode->pFsm->FpGetSnapshotInfo != NULL) {
    SSnapshot snapshot = {0};
    int32_t   code = pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
860
    if (code != 0) {
S
Shengliang Guan 已提交
861
      sError("vgId:%d, failed to get snapshot info, code:%d", pSyncNode->vgId, code);
H
Hongze Cheng 已提交
862
      goto _error;
863
    }
864 865
    if (snapshot.lastApplyIndex > commitIndex) {
      commitIndex = snapshot.lastApplyIndex;
S
Shengliang Guan 已提交
866
      sNTrace(pSyncNode, "reset commit index by snapshot");
867 868 869
    }
  }
  pSyncNode->commitIndex = commitIndex;
M
Minghao Li 已提交
870

M
Minghao Li 已提交
871 872 873 874 875
  // timer ms init
  pSyncNode->pingBaseLine = PING_TIMER_MS;
  pSyncNode->electBaseLine = ELECT_TIMER_MS_MIN;
  pSyncNode->hbBaseLine = HEARTBEAT_TIMER_MS;

M
Minghao Li 已提交
876
  // init ping timer
M
Minghao Li 已提交
877
  pSyncNode->pPingTimer = NULL;
M
Minghao Li 已提交
878
  pSyncNode->pingTimerMS = pSyncNode->pingBaseLine;
M
Minghao Li 已提交
879 880
  atomic_store_64(&pSyncNode->pingTimerLogicClock, 0);
  atomic_store_64(&pSyncNode->pingTimerLogicClockUser, 0);
M
Minghao Li 已提交
881
  pSyncNode->FpPingTimerCB = syncNodeEqPingTimer;
M
Minghao Li 已提交
882
  pSyncNode->pingTimerCounter = 0;
M
Minghao Li 已提交
883

M
Minghao Li 已提交
884 885
  // init elect timer
  pSyncNode->pElectTimer = NULL;
M
Minghao Li 已提交
886
  pSyncNode->electTimerMS = syncUtilElectRandomMS(pSyncNode->electBaseLine, 2 * pSyncNode->electBaseLine);
M
Minghao Li 已提交
887
  atomic_store_64(&pSyncNode->electTimerLogicClock, 0);
M
Minghao Li 已提交
888
  pSyncNode->FpElectTimerCB = syncNodeEqElectTimer;
M
Minghao Li 已提交
889 890 891 892
  pSyncNode->electTimerCounter = 0;

  // init heartbeat timer
  pSyncNode->pHeartbeatTimer = NULL;
M
Minghao Li 已提交
893
  pSyncNode->heartbeatTimerMS = pSyncNode->hbBaseLine;
M
Minghao Li 已提交
894 895
  atomic_store_64(&pSyncNode->heartbeatTimerLogicClock, 0);
  atomic_store_64(&pSyncNode->heartbeatTimerLogicClockUser, 0);
M
Minghao Li 已提交
896
  pSyncNode->FpHeartbeatTimerCB = syncNodeEqHeartbeatTimer;
M
Minghao Li 已提交
897 898
  pSyncNode->heartbeatTimerCounter = 0;

899 900 901 902 903
  // init peer heartbeat timer
  for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
    syncHbTimerInit(pSyncNode, &(pSyncNode->peerHeartbeatTimerArr[i]), (pSyncNode->replicasId)[i]);
  }

M
Minghao Li 已提交
904
  // tools
M
Minghao Li 已提交
905
  pSyncNode->pSyncRespMgr = syncRespMgrCreate(pSyncNode, SYNC_RESP_TTL_MS);
906
  if (pSyncNode->pSyncRespMgr == NULL) {
S
Shengliang Guan 已提交
907
    sError("vgId:%d, failed to create SyncRespMgr", pSyncNode->vgId);
908 909
    goto _error;
  }
M
Minghao Li 已提交
910

911 912
  // restore state
  pSyncNode->restoreFinish = false;
913

M
Minghao Li 已提交
914
  // snapshot senders
S
Shengliang Guan 已提交
915
  for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
916 917 918 919 920 921
    SSyncSnapshotSender* pSender = snapshotSenderCreate(pSyncNode, i);
    // ASSERT(pSender != NULL);
    (pSyncNode->senders)[i] = pSender;
  }

  // snapshot receivers
922
  pSyncNode->pNewNodeReceiver = snapshotReceiverCreate(pSyncNode, EMPTY_RAFT_ID);
M
Minghao Li 已提交
923

M
Minghao Li 已提交
924 925 926
  // is config changing
  pSyncNode->changing = false;

M
Minghao Li 已提交
927 928 929
  // peer state
  syncNodePeerStateInit(pSyncNode);

M
Minghao Li 已提交
930 931 932
  // min match index
  pSyncNode->minMatchIndex = SYNC_INDEX_INVALID;

M
Minghao Li 已提交
933
  // start in syncNodeStart
M
Minghao Li 已提交
934
  // start raft
M
Minghao Li 已提交
935
  // syncNodeBecomeFollower(pSyncNode);
M
Minghao Li 已提交
936

M
Minghao Li 已提交
937 938
  int64_t timeNow = taosGetTimestampMs();
  pSyncNode->startTime = timeNow;
939
  pSyncNode->leaderTime = timeNow;
M
Minghao Li 已提交
940 941
  pSyncNode->lastReplicateTime = timeNow;

942 943 944
  // snapshotting
  atomic_store_64(&pSyncNode->snapshottingIndex, SYNC_INDEX_INVALID);

S
Shengliang Guan 已提交
945
  sNTrace(pSyncNode, "sync open");
946

M
Minghao Li 已提交
947
  return pSyncNode;
948 949 950

_error:
  if (pSyncInfo->pFsm) {
H
Hongze Cheng 已提交
951 952
    taosMemoryFree(pSyncInfo->pFsm);
    pSyncInfo->pFsm = NULL;
953 954 955 956
  }
  syncNodeClose(pSyncNode);
  pSyncNode = NULL;
  return NULL;
M
Minghao Li 已提交
957 958
}

M
Minghao Li 已提交
959 960 961 962 963 964 965 966 967 968 969
void syncNodeMaybeUpdateCommitBySnapshot(SSyncNode* pSyncNode) {
  if (pSyncNode->pFsm != NULL && pSyncNode->pFsm->FpGetSnapshotInfo != NULL) {
    SSnapshot snapshot;
    int32_t   code = pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
    ASSERT(code == 0);
    if (snapshot.lastApplyIndex > pSyncNode->commitIndex) {
      pSyncNode->commitIndex = snapshot.lastApplyIndex;
    }
  }
}

M
Minghao Li 已提交
970 971
void syncNodeStart(SSyncNode* pSyncNode) {
  // start raft
972
  if (pSyncNode->replicaNum == 1) {
M
Minghao Li 已提交
973
    raftStoreNextTerm(pSyncNode->pRaftStore);
974
    syncNodeBecomeLeader(pSyncNode, "one replica start");
M
format  
Minghao Li 已提交
975

976
    // Raft 3.6.2 Committing entries from previous terms
977 978
    syncNodeAppendNoop(pSyncNode);
    syncMaybeAdvanceCommitIndex(pSyncNode);
M
Minghao Li 已提交
979

M
Minghao Li 已提交
980 981
  } else {
    syncNodeBecomeFollower(pSyncNode, "first start");
982 983
  }

984 985 986
  int32_t ret = 0;
  ret = syncNodeStartPingTimer(pSyncNode);
  ASSERT(ret == 0);
M
Minghao Li 已提交
987 988
}

M
Minghao Li 已提交
989 990 991 992 993 994 995 996 997
void syncNodeStartStandBy(SSyncNode* pSyncNode) {
  // state change
  pSyncNode->state = TAOS_SYNC_STATE_FOLLOWER;
  syncNodeStopHeartbeatTimer(pSyncNode);

  // reset elect timer, long enough
  int32_t electMS = TIMER_MAX_MS;
  int32_t ret = syncNodeRestartElectTimer(pSyncNode, electMS);
  ASSERT(ret == 0);
998

999 1000 1001
  ret = 0;
  ret = syncNodeStartPingTimer(pSyncNode);
  ASSERT(ret == 0);
M
Minghao Li 已提交
1002 1003
}

M
Minghao Li 已提交
1004 1005 1006 1007 1008 1009 1010 1011
void syncNodePreClose(SSyncNode* pSyncNode) {
  // stop elect timer
  syncNodeStopElectTimer(pSyncNode);

  // stop heartbeat timer
  syncNodeStopHeartbeatTimer(pSyncNode);
}

M
Minghao Li 已提交
1012
void syncNodeClose(SSyncNode* pSyncNode) {
1013 1014 1015
  if (pSyncNode == NULL) {
    return;
  }
M
Minghao Li 已提交
1016 1017
  int32_t ret;

S
Shengliang Guan 已提交
1018
  sNTrace(pSyncNode, "sync close");
M
Minghao Li 已提交
1019

M
Minghao Li 已提交
1020
  ret = raftStoreClose(pSyncNode->pRaftStore);
M
Minghao Li 已提交
1021
  ASSERT(ret == 0);
M
Minghao Li 已提交
1022
  pSyncNode->pRaftStore = NULL;
M
Minghao Li 已提交
1023

M
Minghao Li 已提交
1024
  syncRespMgrDestroy(pSyncNode->pSyncRespMgr);
1025
  pSyncNode->pSyncRespMgr = NULL;
M
Minghao Li 已提交
1026
  voteGrantedDestroy(pSyncNode->pVotesGranted);
1027
  pSyncNode->pVotesGranted = NULL;
M
Minghao Li 已提交
1028
  votesRespondDestory(pSyncNode->pVotesRespond);
1029
  pSyncNode->pVotesRespond = NULL;
M
Minghao Li 已提交
1030
  syncIndexMgrDestroy(pSyncNode->pNextIndex);
1031
  pSyncNode->pNextIndex = NULL;
M
Minghao Li 已提交
1032
  syncIndexMgrDestroy(pSyncNode->pMatchIndex);
1033
  pSyncNode->pMatchIndex = NULL;
M
Minghao Li 已提交
1034
  logStoreDestory(pSyncNode->pLogStore);
1035
  pSyncNode->pLogStore = NULL;
M
Minghao Li 已提交
1036
  raftCfgClose(pSyncNode->pRaftCfg);
1037
  pSyncNode->pRaftCfg = NULL;
M
Minghao Li 已提交
1038 1039 1040 1041 1042

  syncNodeStopPingTimer(pSyncNode);
  syncNodeStopElectTimer(pSyncNode);
  syncNodeStopHeartbeatTimer(pSyncNode);

M
Minghao Li 已提交
1043 1044 1045 1046
  if (pSyncNode->pFsm != NULL) {
    taosMemoryFree(pSyncNode->pFsm);
  }

S
Shengliang Guan 已提交
1047
  for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1048 1049 1050 1051 1052 1053
    if ((pSyncNode->senders)[i] != NULL) {
      snapshotSenderDestroy((pSyncNode->senders)[i]);
      (pSyncNode->senders)[i] = NULL;
    }
  }

M
Minghao Li 已提交
1054 1055 1056 1057 1058
  if (pSyncNode->pNewNodeReceiver != NULL) {
    snapshotReceiverDestroy(pSyncNode->pNewNodeReceiver);
    pSyncNode->pNewNodeReceiver = NULL;
  }

1059
  taosMemoryFree(pSyncNode);
M
Minghao Li 已提交
1060 1061
}

M
Minghao Li 已提交
1062
ESyncStrategy syncNodeStrategy(SSyncNode* pSyncNode) { return pSyncNode->pRaftCfg->snapshotStrategy; }
M
Minghao Li 已提交
1063

M
Minghao Li 已提交
1064 1065 1066
// timer control --------------
int32_t syncNodeStartPingTimer(SSyncNode* pSyncNode) {
  int32_t ret = 0;
S
Shengliang Guan 已提交
1067 1068
  if (syncIsInit()) {
    taosTmrReset(pSyncNode->FpPingTimerCB, pSyncNode->pingTimerMS, pSyncNode, syncEnv()->pTimerManager,
1069 1070 1071
                 &pSyncNode->pPingTimer);
    atomic_store_64(&pSyncNode->pingTimerLogicClock, pSyncNode->pingTimerLogicClockUser);
  } else {
M
Minghao Li 已提交
1072
    sError("vgId:%d, start ping timer error, sync env is stop", pSyncNode->vgId);
1073
  }
M
Minghao Li 已提交
1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086
  return ret;
}

int32_t syncNodeStopPingTimer(SSyncNode* pSyncNode) {
  int32_t ret = 0;
  atomic_add_fetch_64(&pSyncNode->pingTimerLogicClockUser, 1);
  taosTmrStop(pSyncNode->pPingTimer);
  pSyncNode->pPingTimer = NULL;
  return ret;
}

int32_t syncNodeStartElectTimer(SSyncNode* pSyncNode, int32_t ms) {
  int32_t ret = 0;
S
Shengliang Guan 已提交
1087
  if (syncIsInit()) {
1088
    pSyncNode->electTimerMS = ms;
M
Minghao Li 已提交
1089 1090 1091 1092 1093 1094

    SElectTimer* pElectTimer = taosMemoryMalloc(sizeof(SElectTimer));
    pElectTimer->logicClock = pSyncNode->electTimerLogicClock;
    pElectTimer->pSyncNode = pSyncNode;
    pElectTimer->pData = NULL;

S
Shengliang Guan 已提交
1095
    taosTmrReset(pSyncNode->FpElectTimerCB, pSyncNode->electTimerMS, pElectTimer, syncEnv()->pTimerManager,
1096
                 &pSyncNode->pElectTimer);
1097

1098
  } else {
M
Minghao Li 已提交
1099
    sError("vgId:%d, start elect timer error, sync env is stop", pSyncNode->vgId);
1100
  }
M
Minghao Li 已提交
1101 1102 1103 1104 1105
  return ret;
}

int32_t syncNodeStopElectTimer(SSyncNode* pSyncNode) {
  int32_t ret = 0;
M
Minghao Li 已提交
1106
  atomic_add_fetch_64(&pSyncNode->electTimerLogicClock, 1);
M
Minghao Li 已提交
1107 1108
  taosTmrStop(pSyncNode->pElectTimer);
  pSyncNode->pElectTimer = NULL;
1109

M
Minghao Li 已提交
1110 1111 1112 1113 1114 1115 1116 1117 1118 1119
  return ret;
}

int32_t syncNodeRestartElectTimer(SSyncNode* pSyncNode, int32_t ms) {
  int32_t ret = 0;
  syncNodeStopElectTimer(pSyncNode);
  syncNodeStartElectTimer(pSyncNode, ms);
  return ret;
}

M
Minghao Li 已提交
1120 1121
int32_t syncNodeResetElectTimer(SSyncNode* pSyncNode) {
  int32_t ret = 0;
M
Minghao Li 已提交
1122 1123 1124 1125 1126 1127 1128
  int32_t electMS;

  if (pSyncNode->pRaftCfg->isStandBy) {
    electMS = TIMER_MAX_MS;
  } else {
    electMS = syncUtilElectRandomMS(pSyncNode->electBaseLine, 2 * pSyncNode->electBaseLine);
  }
M
Minghao Li 已提交
1129
  ret = syncNodeRestartElectTimer(pSyncNode, electMS);
1130

S
Shengliang Guan 已提交
1131 1132
  sNTrace(pSyncNode, "reset elect timer, min:%d, max:%d, ms:%d", pSyncNode->electBaseLine, 2 * pSyncNode->electBaseLine,
          electMS);
M
Minghao Li 已提交
1133 1134 1135
  return ret;
}

M
Minghao Li 已提交
1136
static int32_t syncNodeDoStartHeartbeatTimer(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1137
  int32_t ret = 0;
S
Shengliang Guan 已提交
1138 1139
  if (syncIsInit()) {
    taosTmrReset(pSyncNode->FpHeartbeatTimerCB, pSyncNode->heartbeatTimerMS, pSyncNode, syncEnv()->pTimerManager,
1140 1141 1142
                 &pSyncNode->pHeartbeatTimer);
    atomic_store_64(&pSyncNode->heartbeatTimerLogicClock, pSyncNode->heartbeatTimerLogicClockUser);
  } else {
M
Minghao Li 已提交
1143
    sError("vgId:%d, start heartbeat timer error, sync env is stop", pSyncNode->vgId);
1144
  }
1145

S
Shengliang Guan 已提交
1146
  sNTrace(pSyncNode, "start heartbeat timer, ms:%d", pSyncNode->heartbeatTimerMS);
M
Minghao Li 已提交
1147 1148 1149
  return ret;
}

M
Minghao Li 已提交
1150
int32_t syncNodeStartHeartbeatTimer(SSyncNode* pSyncNode) {
1151
  int32_t ret = 0;
M
Minghao Li 已提交
1152

1153
#if 0
M
Minghao Li 已提交
1154
  pSyncNode->heartbeatTimerMS = pSyncNode->hbBaseLine;
1155 1156
  ret = syncNodeDoStartHeartbeatTimer(pSyncNode);
#endif
1157

S
Shengliang Guan 已提交
1158
  for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
1159
    SSyncTimer* pSyncTimer = syncNodeGetHbTimer(pSyncNode, &(pSyncNode->peersId[i]));
M
Minghao Li 已提交
1160 1161 1162
    if (pSyncTimer != NULL) {
      syncHbTimerStart(pSyncNode, pSyncTimer);
    }
1163
  }
1164

M
Minghao Li 已提交
1165 1166 1167
  return ret;
}

M
Minghao Li 已提交
1168 1169
int32_t syncNodeStopHeartbeatTimer(SSyncNode* pSyncNode) {
  int32_t ret = 0;
1170 1171

#if 0
M
Minghao Li 已提交
1172 1173 1174
  atomic_add_fetch_64(&pSyncNode->heartbeatTimerLogicClockUser, 1);
  taosTmrStop(pSyncNode->pHeartbeatTimer);
  pSyncNode->pHeartbeatTimer = NULL;
1175
#endif
1176

S
Shengliang Guan 已提交
1177
  for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
1178
    SSyncTimer* pSyncTimer = syncNodeGetHbTimer(pSyncNode, &(pSyncNode->peersId[i]));
M
Minghao Li 已提交
1179 1180 1181
    if (pSyncTimer != NULL) {
      syncHbTimerStop(pSyncNode, pSyncTimer);
    }
1182
  }
1183

M
Minghao Li 已提交
1184 1185 1186
  return ret;
}

1187 1188 1189 1190 1191 1192
int32_t syncNodeRestartHeartbeatTimer(SSyncNode* pSyncNode) {
  syncNodeStopHeartbeatTimer(pSyncNode);
  syncNodeStartHeartbeatTimer(pSyncNode);
  return 0;
}

M
Minghao Li 已提交
1193 1194 1195
// utils --------------
int32_t syncNodeSendMsgById(const SRaftId* destRaftId, SSyncNode* pSyncNode, SRpcMsg* pMsg) {
  SEpSet epSet;
1196
  syncUtilRaftId2EpSet(destRaftId, &epSet);
S
Shengliang Guan 已提交
1197
  if (pSyncNode->syncSendMSg != NULL) {
M
Minghao Li 已提交
1198 1199 1200
    // htonl
    syncUtilMsgHtoN(pMsg->pCont);

1201
    pMsg->info.noResp = 1;
S
Shengliang Guan 已提交
1202
    pSyncNode->syncSendMSg(&epSet, pMsg);
M
Minghao Li 已提交
1203
  } else {
M
Minghao Li 已提交
1204
    sError("vgId:%d, sync send msg by id error, fp-send-msg is null", pSyncNode->vgId);
S
Shengliang Guan 已提交
1205
    rpcFreeCont(pMsg->pCont);
M
Minghao Li 已提交
1206
    return -1;
M
Minghao Li 已提交
1207
  }
M
Minghao Li 已提交
1208

M
Minghao Li 已提交
1209 1210 1211 1212 1213
  return 0;
}

int32_t syncNodeSendMsgByInfo(const SNodeInfo* nodeInfo, SSyncNode* pSyncNode, SRpcMsg* pMsg) {
  SEpSet epSet;
1214
  syncUtilNodeInfo2EpSet(nodeInfo, &epSet);
S
Shengliang Guan 已提交
1215
  if (pSyncNode->syncSendMSg != NULL) {
M
Minghao Li 已提交
1216 1217 1218
    // htonl
    syncUtilMsgHtoN(pMsg->pCont);

1219
    pMsg->info.noResp = 1;
S
Shengliang Guan 已提交
1220
    pSyncNode->syncSendMSg(&epSet, pMsg);
M
Minghao Li 已提交
1221
  } else {
M
Minghao Li 已提交
1222
    sError("vgId:%d, sync send msg by info error, fp-send-msg is null", pSyncNode->vgId);
M
Minghao Li 已提交
1223
  }
M
Minghao Li 已提交
1224 1225 1226
  return 0;
}

1227
inline bool syncNodeInConfig(SSyncNode* pSyncNode, const SSyncCfg* config) {
1228 1229 1230
  bool b1 = false;
  bool b2 = false;

S
Shengliang Guan 已提交
1231
  for (int32_t i = 0; i < config->replicaNum; ++i) {
1232 1233 1234 1235 1236 1237 1238
    if (strcmp((config->nodeInfo)[i].nodeFqdn, pSyncNode->myNodeInfo.nodeFqdn) == 0 &&
        (config->nodeInfo)[i].nodePort == pSyncNode->myNodeInfo.nodePort) {
      b1 = true;
      break;
    }
  }

S
Shengliang Guan 已提交
1239
  for (int32_t i = 0; i < config->replicaNum; ++i) {
1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253
    SRaftId raftId;
    raftId.addr = syncUtilAddr2U64((config->nodeInfo)[i].nodeFqdn, (config->nodeInfo)[i].nodePort);
    raftId.vgId = pSyncNode->vgId;

    if (syncUtilSameId(&raftId, &(pSyncNode->myRaftId))) {
      b2 = true;
      break;
    }
  }

  ASSERT(b1 == b2);
  return b1;
}

1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266
static bool syncIsConfigChanged(const SSyncCfg* pOldCfg, const SSyncCfg* pNewCfg) {
  if (pOldCfg->replicaNum != pNewCfg->replicaNum) return true;
  if (pOldCfg->myIndex != pNewCfg->myIndex) return true;
  for (int32_t i = 0; i < pOldCfg->replicaNum; ++i) {
    const SNodeInfo* pOldInfo = &pOldCfg->nodeInfo[i];
    const SNodeInfo* pNewInfo = &pNewCfg->nodeInfo[i];
    if (strcmp(pOldInfo->nodeFqdn, pNewInfo->nodeFqdn) != 0) return true;
    if (pOldInfo->nodePort != pNewInfo->nodePort) return true;
  }

  return false;
}

M
Minghao Li 已提交
1267
void syncNodeDoConfigChange(SSyncNode* pSyncNode, SSyncCfg* pNewConfig, SyncIndex lastConfigChangeIndex) {
1268
  SSyncCfg oldConfig = pSyncNode->pRaftCfg->cfg;
1269 1270 1271 1272
  if (!syncIsConfigChanged(&oldConfig, pNewConfig)) {
    sInfo("vgId:1, sync not reconfig since not changed");
    return;
  }
S
Shengliang Guan 已提交
1273

1274
  pSyncNode->pRaftCfg->cfg = *pNewConfig;
1275 1276
  pSyncNode->pRaftCfg->lastConfigIndex = lastConfigChangeIndex;

M
Minghao Li 已提交
1277 1278
  bool IamInOld = syncNodeInConfig(pSyncNode, &oldConfig);
  bool IamInNew = syncNodeInConfig(pSyncNode, pNewConfig);
M
Minghao Li 已提交
1279

M
Minghao Li 已提交
1280 1281
  bool isDrop = false;
  bool isAdd = false;
M
Minghao Li 已提交
1282

M
Minghao Li 已提交
1283 1284 1285 1286
  if (IamInOld && !IamInNew) {
    isDrop = true;
  } else {
    isDrop = false;
1287
  }
1288

M
Minghao Li 已提交
1289 1290 1291 1292 1293
  if (!IamInOld && IamInNew) {
    isAdd = true;
  } else {
    isAdd = false;
  }
M
Minghao Li 已提交
1294

M
Minghao Li 已提交
1295
  // log begin config change
S
Shengliang Guan 已提交
1296 1297 1298 1299 1300
  char oldCfgStr[1024] = {0};
  char newCfgStr[1024] = {0};
  syncCfg2SimpleStr(&oldConfig, oldCfgStr, sizeof(oldCfgStr));
  syncCfg2SimpleStr(pNewConfig, oldCfgStr, sizeof(oldCfgStr));
  sNTrace(pSyncNode, "begin do config change, from %s to %s", oldCfgStr, oldCfgStr);
M
Minghao Li 已提交
1301

M
Minghao Li 已提交
1302 1303
  if (IamInNew) {
    pSyncNode->pRaftCfg->isStandBy = 0;  // change isStandBy to normal
M
Minghao Li 已提交
1304
  }
M
Minghao Li 已提交
1305 1306
  if (isDrop) {
    pSyncNode->pRaftCfg->isStandBy = 1;  // set standby
M
Minghao Li 已提交
1307 1308
  }

M
Minghao Li 已提交
1309
  // add last config index
M
Minghao Li 已提交
1310
  raftCfgAddConfigIndex(pSyncNode->pRaftCfg, lastConfigChangeIndex);
M
Minghao Li 已提交
1311

M
Minghao Li 已提交
1312 1313 1314 1315 1316 1317 1318 1319 1320
  if (IamInNew) {
    //-----------------------------------------
    int32_t ret = 0;

    // save snapshot senders
    int32_t oldReplicaNum = pSyncNode->replicaNum;
    SRaftId oldReplicasId[TSDB_MAX_REPLICA];
    memcpy(oldReplicasId, pSyncNode->replicasId, sizeof(oldReplicasId));
    SSyncSnapshotSender* oldSenders[TSDB_MAX_REPLICA];
S
Shengliang Guan 已提交
1321
    for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1322
      oldSenders[i] = (pSyncNode->senders)[i];
S
Shengliang Guan 已提交
1323
      sSTrace(oldSenders[i], "snapshot sender save old");
M
Minghao Li 已提交
1324
    }
1325

M
Minghao Li 已提交
1326 1327
    // init internal
    pSyncNode->myNodeInfo = pSyncNode->pRaftCfg->cfg.nodeInfo[pSyncNode->pRaftCfg->cfg.myIndex];
1328
    syncUtilNodeInfo2RaftId(&pSyncNode->myNodeInfo, pSyncNode->vgId, &pSyncNode->myRaftId);
M
Minghao Li 已提交
1329 1330 1331

    // init peersNum, peers, peersId
    pSyncNode->peersNum = pSyncNode->pRaftCfg->cfg.replicaNum - 1;
S
Shengliang Guan 已提交
1332 1333
    int32_t j = 0;
    for (int32_t i = 0; i < pSyncNode->pRaftCfg->cfg.replicaNum; ++i) {
M
Minghao Li 已提交
1334 1335 1336 1337 1338
      if (i != pSyncNode->pRaftCfg->cfg.myIndex) {
        pSyncNode->peersNodeInfo[j] = pSyncNode->pRaftCfg->cfg.nodeInfo[i];
        j++;
      }
    }
S
Shengliang Guan 已提交
1339
    for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
1340
      syncUtilNodeInfo2RaftId(&pSyncNode->peersNodeInfo[i], pSyncNode->vgId, &pSyncNode->peersId[i]);
M
Minghao Li 已提交
1341
    }
1342

M
Minghao Li 已提交
1343 1344
    // init replicaNum, replicasId
    pSyncNode->replicaNum = pSyncNode->pRaftCfg->cfg.replicaNum;
S
Shengliang Guan 已提交
1345
    for (int32_t i = 0; i < pSyncNode->pRaftCfg->cfg.replicaNum; ++i) {
1346
      syncUtilNodeInfo2RaftId(&pSyncNode->pRaftCfg->cfg.nodeInfo[i], pSyncNode->vgId, &pSyncNode->replicasId[i]);
M
Minghao Li 已提交
1347
    }
1348

1349 1350 1351
    // update quorum first
    pSyncNode->quorum = syncUtilQuorum(pSyncNode->pRaftCfg->cfg.replicaNum);

M
Minghao Li 已提交
1352 1353 1354 1355
    syncIndexMgrUpdate(pSyncNode->pNextIndex, pSyncNode);
    syncIndexMgrUpdate(pSyncNode->pMatchIndex, pSyncNode);
    voteGrantedUpdate(pSyncNode->pVotesGranted, pSyncNode);
    votesRespondUpdate(pSyncNode->pVotesRespond, pSyncNode);
M
Minghao Li 已提交
1356

M
Minghao Li 已提交
1357
    // reset snapshot senders
1358

M
Minghao Li 已提交
1359
    // clear new
S
Shengliang Guan 已提交
1360
    for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1361 1362
      (pSyncNode->senders)[i] = NULL;
    }
M
Minghao Li 已提交
1363

M
Minghao Li 已提交
1364
    // reset new
S
Shengliang Guan 已提交
1365
    for (int32_t i = 0; i < pSyncNode->replicaNum; ++i) {
M
Minghao Li 已提交
1366 1367
      // reset sender
      bool reset = false;
S
Shengliang Guan 已提交
1368
      for (int32_t j = 0; j < TSDB_MAX_REPLICA; ++j) {
M
Minghao Li 已提交
1369 1370 1371 1372
        if (syncUtilSameId(&(pSyncNode->replicasId)[i], &oldReplicasId[j])) {
          char     host[128];
          uint16_t port;
          syncUtilU642Addr((pSyncNode->replicasId)[i].addr, host, sizeof(host), &port);
1373
          sNTrace(pSyncNode, "snapshot sender reset for: %" PRId64 ", newIndex:%d, %s:%d, %p",
S
Shengliang Guan 已提交
1374
                  (pSyncNode->replicasId)[i].addr, i, host, port, oldSenders[j]);
M
Minghao Li 已提交
1375 1376 1377 1378 1379 1380 1381 1382 1383

          (pSyncNode->senders)[i] = oldSenders[j];
          oldSenders[j] = NULL;
          reset = true;

          // reset replicaIndex
          int32_t oldreplicaIndex = (pSyncNode->senders)[i]->replicaIndex;
          (pSyncNode->senders)[i]->replicaIndex = i;

S
Shengliang Guan 已提交
1384 1385
          sNTrace(pSyncNode, "snapshot sender udpate replicaIndex from %d to %d, %s:%d, %p, reset:%d", oldreplicaIndex,
                  i, host, port, (pSyncNode->senders)[i], reset);
M
Minghao Li 已提交
1386
        }
1387 1388
      }
    }
1389

M
Minghao Li 已提交
1390
    // create new
S
Shengliang Guan 已提交
1391
    for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1392 1393
      if ((pSyncNode->senders)[i] == NULL) {
        (pSyncNode->senders)[i] = snapshotSenderCreate(pSyncNode, i);
S
Shengliang Guan 已提交
1394
        sSTrace((pSyncNode->senders)[i], "snapshot sender create new");
M
Minghao Li 已提交
1395
      }
1396 1397
    }

M
Minghao Li 已提交
1398
    // free old
S
Shengliang Guan 已提交
1399
    for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1400 1401
      if (oldSenders[i] != NULL) {
        snapshotSenderDestroy(oldSenders[i]);
S
Shengliang Guan 已提交
1402
        sNTrace(pSyncNode, "snapshot sender delete old %p replica-index:%d", oldSenders[i], i);
M
Minghao Li 已提交
1403 1404
        oldSenders[i] = NULL;
      }
1405 1406
    }

1407
    // persist cfg
M
Minghao Li 已提交
1408
    raftCfgPersist(pSyncNode->pRaftCfg);
1409

S
Shengliang Guan 已提交
1410
    char tmpbuf[1024] = {0};
1411
    snprintf(tmpbuf, sizeof(tmpbuf), "config change from %d to %d, index:%" PRId64 ", %s  -->  %s",
S
Shengliang Guan 已提交
1412
             oldConfig.replicaNum, pNewConfig->replicaNum, lastConfigChangeIndex, oldCfgStr, newCfgStr);
M
Minghao Li 已提交
1413

M
Minghao Li 已提交
1414 1415 1416
    // change isStandBy to normal (election timeout)
    if (pSyncNode->state == TAOS_SYNC_STATE_LEADER) {
      syncNodeBecomeLeader(pSyncNode, tmpbuf);
1417 1418 1419 1420 1421

      // Raft 3.6.2 Committing entries from previous terms
      syncNodeAppendNoop(pSyncNode);
      syncMaybeAdvanceCommitIndex(pSyncNode);

M
Minghao Li 已提交
1422 1423 1424 1425
    } else {
      syncNodeBecomeFollower(pSyncNode, tmpbuf);
    }
  } else {
1426
    // persist cfg
M
Minghao Li 已提交
1427
    raftCfgPersist(pSyncNode->pRaftCfg);
S
Shengliang Guan 已提交
1428 1429
    sNTrace(pSyncNode, "do not config change from %d to %d, index:%" PRId64 ", %s  -->  %s", oldConfig.replicaNum,
            pNewConfig->replicaNum, lastConfigChangeIndex, oldCfgStr, newCfgStr);
1430
  }
1431

M
Minghao Li 已提交
1432
_END:
M
Minghao Li 已提交
1433
  // log end config change
S
Shengliang Guan 已提交
1434
  sNTrace(pSyncNode, "end do config change, from %s to %s", oldCfgStr, newCfgStr);
M
Minghao Li 已提交
1435 1436
}

M
Minghao Li 已提交
1437 1438 1439 1440
// raft state change --------------
void syncNodeUpdateTerm(SSyncNode* pSyncNode, SyncTerm term) {
  if (term > pSyncNode->pRaftStore->currentTerm) {
    raftStoreSetTerm(pSyncNode->pRaftStore, term);
1441
    char tmpBuf[64];
1442
    snprintf(tmpBuf, sizeof(tmpBuf), "update term to %" PRId64, term);
1443
    syncNodeBecomeFollower(pSyncNode, tmpBuf);
M
Minghao Li 已提交
1444 1445 1446 1447
    raftStoreClearVote(pSyncNode->pRaftStore);
  }
}

1448 1449 1450 1451 1452 1453
void syncNodeUpdateTermWithoutStepDown(SSyncNode* pSyncNode, SyncTerm term) {
  if (term > pSyncNode->pRaftStore->currentTerm) {
    raftStoreSetTerm(pSyncNode->pRaftStore, term);
  }
}

M
Minghao Li 已提交
1454
void syncNodeStepDown(SSyncNode* pSyncNode, SyncTerm newTerm) {
M
Minghao Li 已提交
1455
  if (pSyncNode->pRaftStore->currentTerm > newTerm) {
1456
    sNTrace(pSyncNode, "step down, ignore, new-term:%" PRId64 ", current-term:%" PRId64, newTerm,
S
Shengliang Guan 已提交
1457
            pSyncNode->pRaftStore->currentTerm);
M
Minghao Li 已提交
1458 1459
    return;
  }
M
Minghao Li 已提交
1460 1461

  do {
1462
    sNTrace(pSyncNode, "step down, new-term:%" PRId64 ", current-term:%" PRId64, newTerm,
S
Shengliang Guan 已提交
1463
            pSyncNode->pRaftStore->currentTerm);
M
Minghao Li 已提交
1464 1465 1466 1467 1468
  } while (0);

  if (pSyncNode->pRaftStore->currentTerm < newTerm) {
    raftStoreSetTerm(pSyncNode->pRaftStore, newTerm);
    char tmpBuf[64];
1469
    snprintf(tmpBuf, sizeof(tmpBuf), "step down, update term to %" PRId64, newTerm);
M
Minghao Li 已提交
1470 1471 1472 1473 1474 1475 1476 1477 1478 1479
    syncNodeBecomeFollower(pSyncNode, tmpBuf);
    raftStoreClearVote(pSyncNode->pRaftStore);

  } else {
    if (pSyncNode->state != TAOS_SYNC_STATE_FOLLOWER) {
      syncNodeBecomeFollower(pSyncNode, "step down");
    }
  }
}

1480 1481
void syncNodeLeaderChangeRsp(SSyncNode* pSyncNode) { syncRespCleanRsp(pSyncNode->pSyncRespMgr); }

1482
void syncNodeBecomeFollower(SSyncNode* pSyncNode, const char* debugStr) {
M
Minghao Li 已提交
1483
  // maybe clear leader cache
M
Minghao Li 已提交
1484 1485 1486 1487
  if (pSyncNode->state == TAOS_SYNC_STATE_LEADER) {
    pSyncNode->leaderCache = EMPTY_RAFT_ID;
  }

M
Minghao Li 已提交
1488
  // state change
M
Minghao Li 已提交
1489 1490 1491
  pSyncNode->state = TAOS_SYNC_STATE_FOLLOWER;
  syncNodeStopHeartbeatTimer(pSyncNode);

M
Minghao Li 已提交
1492 1493
  // reset elect timer
  syncNodeResetElectTimer(pSyncNode);
M
Minghao Li 已提交
1494

1495 1496 1497
  // send rsp to client
  syncNodeLeaderChangeRsp(pSyncNode);

1498 1499 1500 1501 1502
  // call back
  if (pSyncNode->pFsm != NULL && pSyncNode->pFsm->FpBecomeFollowerCb != NULL) {
    pSyncNode->pFsm->FpBecomeFollowerCb(pSyncNode->pFsm);
  }

M
Minghao Li 已提交
1503 1504 1505
  // min match index
  pSyncNode->minMatchIndex = SYNC_INDEX_INVALID;

M
Minghao Li 已提交
1506
  // trace log
S
Shengliang Guan 已提交
1507
  sNTrace(pSyncNode, "become follower %s", debugStr);
M
Minghao Li 已提交
1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527
}

// TLA+ Spec
// \* Candidate i transitions to leader.
// BecomeLeader(i) ==
//     /\ state[i] = Candidate
//     /\ votesGranted[i] \in Quorum
//     /\ state'      = [state EXCEPT ![i] = Leader]
//     /\ nextIndex'  = [nextIndex EXCEPT ![i] =
//                          [j \in Server |-> Len(log[i]) + 1]]
//     /\ matchIndex' = [matchIndex EXCEPT ![i] =
//                          [j \in Server |-> 0]]
//     /\ elections'  = elections \cup
//                          {[eterm     |-> currentTerm[i],
//                            eleader   |-> i,
//                            elog      |-> log[i],
//                            evotes    |-> votesGranted[i],
//                            evoterLog |-> voterLog[i]]}
//     /\ UNCHANGED <<messages, currentTerm, votedFor, candidateVars, logVars>>
//
1528
void syncNodeBecomeLeader(SSyncNode* pSyncNode, const char* debugStr) {
1529 1530
  pSyncNode->leaderTime = taosGetTimestampMs();

1531 1532 1533
  // reset restoreFinish
  pSyncNode->restoreFinish = false;

M
Minghao Li 已提交
1534
  // state change
M
Minghao Li 已提交
1535
  pSyncNode->state = TAOS_SYNC_STATE_LEADER;
M
Minghao Li 已提交
1536 1537

  // set leader cache
M
Minghao Li 已提交
1538 1539
  pSyncNode->leaderCache = pSyncNode->myRaftId;

S
Shengliang Guan 已提交
1540
  for (int32_t i = 0; i < pSyncNode->pNextIndex->replicaNum; ++i) {
M
Minghao Li 已提交
1541 1542
    // maybe overwrite myself, no harm
    // just do it!
1543 1544 1545 1546 1547 1548 1549 1550 1551

    // pSyncNode->pNextIndex->index[i] = pSyncNode->pLogStore->getLastIndex(pSyncNode->pLogStore) + 1;

    // maybe wal is deleted
    SyncIndex lastIndex;
    SyncTerm  lastTerm;
    int32_t   code = syncNodeGetLastIndexTerm(pSyncNode, &lastIndex, &lastTerm);
    ASSERT(code == 0);
    pSyncNode->pNextIndex->index[i] = lastIndex + 1;
M
Minghao Li 已提交
1552 1553
  }

S
Shengliang Guan 已提交
1554
  for (int32_t i = 0; i < pSyncNode->pMatchIndex->replicaNum; ++i) {
M
Minghao Li 已提交
1555 1556
    // maybe overwrite myself, no harm
    // just do it!
M
Minghao Li 已提交
1557 1558 1559
    pSyncNode->pMatchIndex->index[i] = SYNC_INDEX_INVALID;
  }

M
Minghao Li 已提交
1560 1561 1562
  // init peer mgr
  syncNodePeerStateInit(pSyncNode);

M
Minghao Li 已提交
1563
#if 0
1564 1565
  // update sender private term
  SSyncSnapshotSender* pMySender = syncNodeGetSnapshotSender(pSyncNode, &(pSyncNode->myRaftId));
1566
  if (pMySender != NULL) {
S
Shengliang Guan 已提交
1567
    for (int32_t i = 0; i < pSyncNode->pMatchIndex->replicaNum; ++i) {
1568 1569 1570
      if ((pSyncNode->senders)[i]->privateTerm > pMySender->privateTerm) {
        pMySender->privateTerm = (pSyncNode->senders)[i]->privateTerm;
      }
1571
    }
1572
    (pMySender->privateTerm) += 100;
1573
  }
M
Minghao Li 已提交
1574
#endif
1575

1576 1577 1578 1579 1580
  // close receiver
  if (snapshotReceiverIsStart(pSyncNode->pNewNodeReceiver)) {
    snapshotReceiverForceStop(pSyncNode->pNewNodeReceiver);
  }

M
Minghao Li 已提交
1581
  // stop elect timer
M
Minghao Li 已提交
1582
  syncNodeStopElectTimer(pSyncNode);
M
Minghao Li 已提交
1583

M
Minghao Li 已提交
1584 1585
  // start heartbeat timer
  syncNodeStartHeartbeatTimer(pSyncNode);
M
Minghao Li 已提交
1586

M
Minghao Li 已提交
1587 1588
  // send heartbeat right now
  syncNodeHeartbeatPeers(pSyncNode);
M
Minghao Li 已提交
1589

1590 1591 1592 1593 1594
  // call back
  if (pSyncNode->pFsm != NULL && pSyncNode->pFsm->FpBecomeLeaderCb != NULL) {
    pSyncNode->pFsm->FpBecomeLeaderCb(pSyncNode->pFsm);
  }

M
Minghao Li 已提交
1595 1596 1597
  // min match index
  pSyncNode->minMatchIndex = SYNC_INDEX_INVALID;

M
Minghao Li 已提交
1598
  // trace log
S
Shengliang Guan 已提交
1599
  sNTrace(pSyncNode, "become leader %s", debugStr);
M
Minghao Li 已提交
1600 1601 1602
}

void syncNodeCandidate2Leader(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1603 1604
  ASSERT(pSyncNode->state == TAOS_SYNC_STATE_CANDIDATE);
  ASSERT(voteGrantedMajority(pSyncNode->pVotesGranted));
1605
  syncNodeBecomeLeader(pSyncNode, "candidate to leader");
M
Minghao Li 已提交
1606

S
Shengliang Guan 已提交
1607
  sNTrace(pSyncNode, "state change syncNodeCandidate2Leader");
M
Minghao Li 已提交
1608

M
Minghao Li 已提交
1609
  // Raft 3.6.2 Committing entries from previous terms
1610 1611
  syncNodeAppendNoop(pSyncNode);
  syncMaybeAdvanceCommitIndex(pSyncNode);
1612 1613

  if (pSyncNode->replicaNum > 1) {
M
Minghao Li 已提交
1614
    syncNodeReplicate(pSyncNode);
1615
  }
M
Minghao Li 已提交
1616 1617
}

M
Minghao Li 已提交
1618 1619
bool syncNodeIsMnode(SSyncNode* pSyncNode) { return (pSyncNode->vgId == 1); }

M
Minghao Li 已提交
1620
int32_t syncNodePeerStateInit(SSyncNode* pSyncNode) {
S
Shengliang Guan 已提交
1621
  for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1622 1623 1624 1625 1626
    pSyncNode->peerStates[i].lastSendIndex = SYNC_INDEX_INVALID;
    pSyncNode->peerStates[i].lastSendTime = 0;
  }

  return 0;
M
Minghao Li 已提交
1627 1628 1629
}

void syncNodeFollower2Candidate(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1630
  ASSERT(pSyncNode->state == TAOS_SYNC_STATE_FOLLOWER);
M
Minghao Li 已提交
1631
  pSyncNode->state = TAOS_SYNC_STATE_CANDIDATE;
S
Shengliang Guan 已提交
1632
  sNTrace(pSyncNode, "follower to candidate");
M
Minghao Li 已提交
1633 1634 1635
}

void syncNodeLeader2Follower(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1636
  ASSERT(pSyncNode->state == TAOS_SYNC_STATE_LEADER);
1637
  syncNodeBecomeFollower(pSyncNode, "leader to follower");
S
Shengliang Guan 已提交
1638
  sNTrace(pSyncNode, "leader to follower");
M
Minghao Li 已提交
1639 1640 1641
}

void syncNodeCandidate2Follower(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1642
  ASSERT(pSyncNode->state == TAOS_SYNC_STATE_CANDIDATE);
1643
  syncNodeBecomeFollower(pSyncNode, "candidate to follower");
S
Shengliang Guan 已提交
1644
  sNTrace(pSyncNode, "candidate to follower");
M
Minghao Li 已提交
1645 1646
}

M
Minghao Li 已提交
1647 1648
// just called by syncNodeVoteForSelf
// need assert
M
Minghao Li 已提交
1649
void syncNodeVoteForTerm(SSyncNode* pSyncNode, SyncTerm term, SRaftId* pRaftId) {
M
Minghao Li 已提交
1650 1651
  ASSERT(term == pSyncNode->pRaftStore->currentTerm);
  ASSERT(!raftStoreHasVoted(pSyncNode->pRaftStore));
M
Minghao Li 已提交
1652 1653 1654 1655

  raftStoreVote(pSyncNode->pRaftStore, pRaftId);
}

M
Minghao Li 已提交
1656
// simulate get vote from outside
M
Minghao Li 已提交
1657
void syncNodeVoteForSelf(SSyncNode* pSyncNode) {
S
Shengliang Guan 已提交
1658
  syncNodeVoteForTerm(pSyncNode, pSyncNode->pRaftStore->currentTerm, &pSyncNode->myRaftId);
M
Minghao Li 已提交
1659

S
Shengliang Guan 已提交
1660 1661
  SRpcMsg rpcMsg = {0};
  int32_t ret = syncBuildRequestVoteReply(&rpcMsg, pSyncNode->vgId);
S
Shengliang Guan 已提交
1662
  if (ret != 0) return;
S
Shengliang Guan 已提交
1663 1664

  SyncRequestVoteReply* pMsg = rpcMsg.pCont;
M
Minghao Li 已提交
1665 1666 1667 1668 1669 1670 1671
  pMsg->srcId = pSyncNode->myRaftId;
  pMsg->destId = pSyncNode->myRaftId;
  pMsg->term = pSyncNode->pRaftStore->currentTerm;
  pMsg->voteGranted = true;

  voteGrantedVote(pSyncNode->pVotesGranted, pMsg);
  votesRespondAdd(pSyncNode->pVotesRespond, pMsg);
S
Shengliang Guan 已提交
1672
  rpcFreeCont(rpcMsg.pCont);
M
Minghao Li 已提交
1673 1674
}

M
Minghao Li 已提交
1675
// return if has a snapshot
M
Minghao Li 已提交
1676 1677
bool syncNodeHasSnapshot(SSyncNode* pSyncNode) {
  bool      ret = false;
1678
  SSnapshot snapshot = {.data = NULL, .lastApplyIndex = -1, .lastApplyTerm = 0, .lastConfigIndex = -1};
1679 1680
  if (pSyncNode->pFsm->FpGetSnapshotInfo != NULL) {
    pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
M
Minghao Li 已提交
1681 1682 1683 1684 1685 1686 1687
    if (snapshot.lastApplyIndex >= SYNC_INDEX_BEGIN) {
      ret = true;
    }
  }
  return ret;
}

M
Minghao Li 已提交
1688 1689
// return max(logLastIndex, snapshotLastIndex)
// if no snapshot and log, return -1
1690
SyncIndex syncNodeGetLastIndex(const SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1691
  SSnapshot snapshot = {.data = NULL, .lastApplyIndex = -1, .lastApplyTerm = 0, .lastConfigIndex = -1};
1692 1693
  if (pSyncNode->pFsm->FpGetSnapshotInfo != NULL) {
    pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
M
Minghao Li 已提交
1694 1695 1696 1697 1698 1699 1700
  }
  SyncIndex logLastIndex = pSyncNode->pLogStore->syncLogLastIndex(pSyncNode->pLogStore);

  SyncIndex lastIndex = logLastIndex > snapshot.lastApplyIndex ? logLastIndex : snapshot.lastApplyIndex;
  return lastIndex;
}

M
Minghao Li 已提交
1701 1702
// return the last term of snapshot and log
// if error, return SYNC_TERM_INVALID (by syncLogLastTerm)
M
Minghao Li 已提交
1703 1704
SyncTerm syncNodeGetLastTerm(SSyncNode* pSyncNode) {
  SyncTerm lastTerm = 0;
M
Minghao Li 已提交
1705 1706
  if (syncNodeHasSnapshot(pSyncNode)) {
    // has snapshot
1707
    SSnapshot snapshot = {.data = NULL, .lastApplyIndex = -1, .lastApplyTerm = 0, .lastConfigIndex = -1};
1708 1709
    if (pSyncNode->pFsm->FpGetSnapshotInfo != NULL) {
      pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
M
Minghao Li 已提交
1710 1711
    }

M
Minghao Li 已提交
1712 1713 1714
    SyncIndex logLastIndex = pSyncNode->pLogStore->syncLogLastIndex(pSyncNode->pLogStore);
    if (logLastIndex > snapshot.lastApplyIndex) {
      lastTerm = pSyncNode->pLogStore->syncLogLastTerm(pSyncNode->pLogStore);
M
Minghao Li 已提交
1715 1716 1717 1718
    } else {
      lastTerm = snapshot.lastApplyTerm;
    }

M
Minghao Li 已提交
1719
  } else {
M
Minghao Li 已提交
1720 1721
    // no snapshot
    lastTerm = pSyncNode->pLogStore->syncLogLastTerm(pSyncNode->pLogStore);
1722
  }
M
Minghao Li 已提交
1723

M
Minghao Li 已提交
1724 1725 1726 1727 1728 1729 1730
  return lastTerm;
}

// get last index and term along with snapshot
int32_t syncNodeGetLastIndexTerm(SSyncNode* pSyncNode, SyncIndex* pLastIndex, SyncTerm* pLastTerm) {
  *pLastIndex = syncNodeGetLastIndex(pSyncNode);
  *pLastTerm = syncNodeGetLastTerm(pSyncNode);
1731 1732
  return 0;
}
M
Minghao Li 已提交
1733

M
Minghao Li 已提交
1734
// return append-entries first try index
M
Minghao Li 已提交
1735 1736 1737 1738 1739
SyncIndex syncNodeSyncStartIndex(SSyncNode* pSyncNode) {
  SyncIndex syncStartIndex = syncNodeGetLastIndex(pSyncNode) + 1;
  return syncStartIndex;
}

M
Minghao Li 已提交
1740 1741
// if index > 0, return index - 1
// else, return -1
1742 1743 1744 1745 1746 1747 1748 1749 1750
SyncIndex syncNodeGetPreIndex(SSyncNode* pSyncNode, SyncIndex index) {
  SyncIndex preIndex = index - 1;
  if (preIndex < SYNC_INDEX_INVALID) {
    preIndex = SYNC_INDEX_INVALID;
  }

  return preIndex;
}

M
Minghao Li 已提交
1751 1752 1753 1754
// if index < 0, return SYNC_TERM_INVALID
// if index == 0, return 0
// if index > 0, return preTerm
// if error, return SYNC_TERM_INVALID
1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767
SyncTerm syncNodeGetPreTerm(SSyncNode* pSyncNode, SyncIndex index) {
  if (index < SYNC_INDEX_BEGIN) {
    return SYNC_TERM_INVALID;
  }

  if (index == SYNC_INDEX_BEGIN) {
    return 0;
  }

  SyncTerm        preTerm = 0;
  SyncIndex       preIndex = index - 1;
  SSyncRaftEntry* pPreEntry = NULL;
  int32_t         code = pSyncNode->pLogStore->syncLogGetEntry(pSyncNode->pLogStore, preIndex, &pPreEntry);
M
Minghao Li 已提交
1768 1769 1770 1771 1772 1773

  SSnapshot snapshot = {.data = NULL,
                        .lastApplyIndex = SYNC_INDEX_INVALID,
                        .lastApplyTerm = SYNC_TERM_INVALID,
                        .lastConfigIndex = SYNC_INDEX_INVALID};

1774 1775 1776 1777 1778 1779
  if (code == 0) {
    ASSERT(pPreEntry != NULL);
    preTerm = pPreEntry->term;
    taosMemoryFree(pPreEntry);
    return preTerm;
  } else {
1780 1781 1782 1783
    if (pSyncNode->pFsm->FpGetSnapshotInfo != NULL) {
      pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
      if (snapshot.lastApplyIndex == preIndex) {
        return snapshot.lastApplyTerm;
1784 1785 1786 1787
      }
    }
  }

1788
  sNError(pSyncNode, "sync node get pre term error, index:%" PRId64 ", snap-index:%" PRId64 ", snap-term:%" PRId64,
S
Shengliang Guan 已提交
1789
          index, snapshot.lastApplyIndex, snapshot.lastApplyTerm);
1790 1791
  return SYNC_TERM_INVALID;
}
M
Minghao Li 已提交
1792 1793 1794 1795

// get pre index and term of "index"
int32_t syncNodeGetPreIndexTerm(SSyncNode* pSyncNode, SyncIndex index, SyncIndex* pPreIndex, SyncTerm* pPreTerm) {
  *pPreIndex = syncNodeGetPreIndex(pSyncNode, index);
M
Minghao Li 已提交
1796
  *pPreTerm = syncNodeGetPreTerm(pSyncNode, index);
M
Minghao Li 已提交
1797 1798 1799
  return 0;
}

M
Minghao Li 已提交
1800
static void syncNodeEqPingTimer(void* param, void* tmrId) {
S
Shengliang Guan 已提交
1801 1802 1803 1804 1805
  if (!syncIsInit()) return;

  SSyncNode* pNode = param;
  if (atomic_load_64(&pNode->pingTimerLogicClockUser) <= atomic_load_64(&pNode->pingTimerLogicClock)) {
    SRpcMsg rpcMsg = {0};
S
Shengliang Guan 已提交
1806
    int32_t code = syncBuildTimeout(&rpcMsg, SYNC_TIMEOUT_PING, atomic_load_64(&pNode->pingTimerLogicClock),
S
Shengliang Guan 已提交
1807 1808 1809 1810 1811
                                    pNode->pingTimerMS, pNode);
    if (code != 0) {
      sNError(pNode, "failed to build ping msg");
      rpcFreeCont(rpcMsg.pCont);
      return;
M
Minghao Li 已提交
1812
    }
M
Minghao Li 已提交
1813

S
Shengliang Guan 已提交
1814 1815 1816 1817 1818 1819
    sNTrace(pNode, "enqueue ping msg");
    code = pNode->syncEqMsg(pNode->msgcb, &rpcMsg);
    if (code != 0) {
      sNError(pNode, "failed to sync enqueue ping msg since %s", terrstr());
      rpcFreeCont(rpcMsg.pCont);
      return;
1820 1821
    }

S
Shengliang Guan 已提交
1822
    taosTmrReset(syncNodeEqPingTimer, pNode->pingTimerMS, pNode, syncEnv()->pTimerManager, &pNode->pPingTimer);
M
Minghao Li 已提交
1823
  } else {
1824
    sTrace("==syncNodeEqPingTimer== pingTimerLogicClock:%" PRId64 ", pingTimerLogicClockUser:%" PRId64,
S
Shengliang Guan 已提交
1825
           pNode->pingTimerLogicClock, pNode->pingTimerLogicClockUser);
M
Minghao Li 已提交
1826 1827 1828 1829
  }
}

static void syncNodeEqElectTimer(void* param, void* tmrId) {
S
Shengliang Guan 已提交
1830 1831 1832 1833 1834
  if (!syncIsInit()) return;

  SElectTimer* pElectTimer = param;
  SSyncNode*   pNode = pElectTimer->pSyncNode;

1835 1836 1837
  if (pNode == NULL) return;
  if (pNode->syncEqMsg == NULL) return;

S
Shengliang Guan 已提交
1838
  SRpcMsg rpcMsg = {0};
S
Shengliang Guan 已提交
1839
  int32_t code = syncBuildTimeout(&rpcMsg, SYNC_TIMEOUT_ELECTION, pElectTimer->logicClock, pNode->electTimerMS, pNode);
S
Shengliang Guan 已提交
1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855

  if (code != 0) {
    sNError(pNode, "failed to build elect msg");
    taosMemoryFree(pElectTimer);
    return;
  }

  SyncTimeout* pTimeout = rpcMsg.pCont;
  sNTrace(pNode, "enqueue elect msg lc:%" PRId64, pTimeout->logicClock);

  code = pNode->syncEqMsg(pNode->msgcb, &rpcMsg);
  if (code != 0) {
    sNError(pNode, "failed to sync enqueue elect msg since %s", terrstr());
    rpcFreeCont(rpcMsg.pCont);
    taosMemoryFree(pElectTimer);
    return;
M
Minghao Li 已提交
1856
  }
M
Minghao Li 已提交
1857 1858

  taosMemoryFree(pElectTimer);
M
Minghao Li 已提交
1859

M
Minghao Li 已提交
1860
#if 0
M
Minghao Li 已提交
1861
  // reset timer ms
S
Shengliang Guan 已提交
1862 1863 1864
  if (syncIsInit() && pNode->electBaseLine > 0) {
    pNode->electTimerMS = syncUtilElectRandomMS(pNode->electBaseLine, 2 * pNode->electBaseLine);
    taosTmrReset(syncNodeEqElectTimer, pNode->electTimerMS, pNode, syncEnv()->pTimerManager, &pNode->pElectTimer);
M
Minghao Li 已提交
1865 1866
  } else {
    sError("sync env is stop, syncNodeEqElectTimer");
M
Minghao Li 已提交
1867
  }
M
Minghao Li 已提交
1868
#endif
M
Minghao Li 已提交
1869 1870
}

M
Minghao Li 已提交
1871
static void syncNodeEqHeartbeatTimer(void* param, void* tmrId) {
S
Shengliang Guan 已提交
1872
  if (!syncIsInit()) return;
1873

S
Shengliang Guan 已提交
1874 1875 1876 1877
  SSyncNode* pNode = param;
  if (pNode->replicaNum > 1) {
    if (atomic_load_64(&pNode->heartbeatTimerLogicClockUser) <= atomic_load_64(&pNode->heartbeatTimerLogicClock)) {
      SRpcMsg rpcMsg = {0};
S
Shengliang Guan 已提交
1878
      int32_t code = syncBuildTimeout(&rpcMsg, SYNC_TIMEOUT_HEARTBEAT, atomic_load_64(&pNode->heartbeatTimerLogicClock),
S
Shengliang Guan 已提交
1879 1880 1881 1882 1883
                                      pNode->heartbeatTimerMS, pNode);

      if (code != 0) {
        sNError(pNode, "failed to build heartbeat msg");
        return;
1884
      }
M
Minghao Li 已提交
1885

S
Shengliang Guan 已提交
1886 1887 1888 1889 1890 1891
      sNTrace(pNode, "enqueue heartbeat timer");
      code = pNode->syncEqMsg(pNode->msgcb, &rpcMsg);
      if (code != 0) {
        sNError(pNode, "failed to enqueue heartbeat msg since %s", terrstr());
        rpcFreeCont(rpcMsg.pCont);
        return;
1892
      }
S
Shengliang Guan 已提交
1893 1894 1895 1896

      taosTmrReset(syncNodeEqHeartbeatTimer, pNode->heartbeatTimerMS, pNode, syncEnv()->pTimerManager,
                   &pNode->pHeartbeatTimer);

1897
    } else {
S
Shengliang Guan 已提交
1898 1899
      sTrace("==syncNodeEqHeartbeatTimer== heartbeatTimerLogicClock:%" PRId64 ", heartbeatTimerLogicClockUser:%" PRId64,
             pNode->heartbeatTimerLogicClock, pNode->heartbeatTimerLogicClockUser);
1900
    }
M
Minghao Li 已提交
1901 1902 1903
  }
}

1904 1905 1906 1907 1908
static void syncNodeEqPeerHeartbeatTimer(void* param, void* tmrId) {
  SSyncHbTimerData* pData = (SSyncHbTimerData*)param;
  SSyncNode*        pSyncNode = pData->pSyncNode;
  SSyncTimer*       pSyncTimer = pData->pTimer;

M
Minghao Li 已提交
1909 1910 1911 1912
  if (pSyncNode == NULL) {
    return;
  }

M
Minghao Li 已提交
1913 1914 1915 1916
  if (pSyncNode->state != TAOS_SYNC_STATE_LEADER) {
    return;
  }

M
Minghao Li 已提交
1917 1918 1919 1920
  if (pSyncNode->pRaftStore == NULL) {
    return;
  }

S
Shengliang Guan 已提交
1921
  // sNTrace(pSyncNode, "eq peer hb timer");
1922 1923 1924 1925 1926 1927

  int64_t timerLogicClock = atomic_load_64(&pSyncTimer->logicClock);
  int64_t msgLogicClock = atomic_load_64(&pData->logicClock);

  if (pSyncNode->replicaNum > 1) {
    if (timerLogicClock == msgLogicClock) {
1928
      SRpcMsg rpcMsg = {0};
S
Shengliang Guan 已提交
1929 1930 1931
      (void)syncBuildHeartbeat(&rpcMsg, pSyncNode->vgId);

      SyncHeartbeat* pSyncMsg = rpcMsg.pCont;
1932 1933 1934 1935
      pSyncMsg->srcId = pSyncNode->myRaftId;
      pSyncMsg->destId = pData->destId;
      pSyncMsg->term = pSyncNode->pRaftStore->currentTerm;
      pSyncMsg->commitIndex = pSyncNode->commitIndex;
M
Minghao Li 已提交
1936
      pSyncMsg->minMatchIndex = syncMinMatchIndex(pSyncNode);
1937 1938 1939
      pSyncMsg->privateTerm = 0;

      // send msg
S
Shengliang Guan 已提交
1940
      syncNodeSendHeartbeat(pSyncNode, &pSyncMsg->destId, &rpcMsg);
1941

S
Shengliang Guan 已提交
1942 1943
      if (syncIsInit()) {
        taosTmrReset(syncNodeEqPeerHeartbeatTimer, pSyncTimer->timerMS, pData, syncEnv()->pTimerManager,
1944 1945 1946 1947 1948 1949
                     &pSyncTimer->pTimer);
      } else {
        sError("sync env is stop, syncNodeEqHeartbeatTimer");
      }

    } else {
1950
      sTrace("==syncNodeEqPeerHeartbeatTimer== timerLogicClock:%" PRId64 ", msgLogicClock:%" PRId64 "", timerLogicClock,
1951 1952 1953 1954 1955
             msgLogicClock);
    }
  }
}

1956 1957 1958 1959 1960
static int32_t syncNodeEqNoop(SSyncNode* pNode) {
  if (pNode->state == TAOS_SYNC_STATE_LEADER) {
    terrno = TSDB_CODE_SYN_NOT_LEADER;
    return -1;
  }
M
Minghao Li 已提交
1961

1962 1963 1964 1965
  SyncIndex       index = pNode->pLogStore->syncLogWriteIndex(pNode->pLogStore);
  SyncTerm        term = pNode->pRaftStore->currentTerm;
  SSyncRaftEntry* pEntry = syncEntryBuildNoop(term, index, pNode->vgId);
  if (pEntry == NULL) return -1;
M
Minghao Li 已提交
1966

S
Shengliang Guan 已提交
1967
  SRpcMsg rpcMsg = {0};
S
Shengliang Guan 已提交
1968
  int32_t code = syncBuildClientRequestFromNoopEntry(&rpcMsg, pEntry, pNode->vgId);
M
Minghao Li 已提交
1969
  syncEntryDestory(pEntry);
M
Minghao Li 已提交
1970

1971 1972 1973 1974 1975 1976 1977
  sNTrace(pNode, "propose msg, type:noop");
  code = (*pNode->syncEqMsg)(pNode->msgcb, &rpcMsg);
  if (code != 0) {
    sNError(pNode, "failed to propose noop msg while enqueue since %s", terrstr());
  }

  return code;
M
Minghao Li 已提交
1978 1979
}

1980 1981 1982
static void deleteCacheEntry(const void* key, size_t keyLen, void* value) { taosMemoryFree(value); }

static int32_t syncCacheEntry(SSyncLogStore* pLogStore, SSyncRaftEntry* pEntry, LRUHandle** h) {
S
Shengliang Guan 已提交
1983 1984
  int32_t   code = 0;
  int32_t   entryLen = sizeof(*pEntry) + pEntry->dataLen;
1985 1986 1987 1988 1989 1990 1991 1992 1993
  LRUStatus status = taosLRUCacheInsert(pLogStore->pCache, &pEntry->index, sizeof(pEntry->index), pEntry, entryLen,
                                        deleteCacheEntry, h, TAOS_LRU_PRIORITY_LOW);
  if (status != TAOS_LRU_STATUS_OK) {
    code = -1;
  }

  return code;
}

M
Minghao Li 已提交
1994 1995 1996
static int32_t syncNodeAppendNoop(SSyncNode* ths) {
  int32_t ret = 0;

1997
  SyncIndex       index = ths->pLogStore->syncLogWriteIndex(ths->pLogStore);
M
Minghao Li 已提交
1998
  SyncTerm        term = ths->pRaftStore->currentTerm;
M
Minghao Li 已提交
1999
  SSyncRaftEntry* pEntry = syncEntryBuildNoop(term, index, ths->vgId);
M
Minghao Li 已提交
2000
  ASSERT(pEntry != NULL);
M
Minghao Li 已提交
2001

2002 2003 2004
  LRUHandle* h = NULL;
  syncCacheEntry(ths->pLogStore, pEntry, &h);

M
Minghao Li 已提交
2005
  if (ths->state == TAOS_SYNC_STATE_LEADER) {
2006
    int32_t code = ths->pLogStore->syncLogAppendEntry(ths->pLogStore, pEntry);
2007
    if (code != 0) {
S
Shengliang Guan 已提交
2008
      sNError(ths, "append noop error");
2009 2010
      return -1;
    }
M
Minghao Li 已提交
2011 2012
  }

2013 2014 2015 2016 2017 2018
  if (h) {
    taosLRUCacheRelease(ths->pLogStore->pCache, h, false);
  } else {
    syncEntryDestory(pEntry);
  }

M
Minghao Li 已提交
2019 2020 2021
  return ret;
}

S
Shengliang Guan 已提交
2022 2023
int32_t syncNodeOnHeartbeat(SSyncNode* ths, const SRpcMsg* pRpcMsg) {
  SyncHeartbeat* pMsg = pRpcMsg->pCont;
2024 2025
  syncLogRecvHeartbeat(ths, pMsg, "");

2026 2027 2028 2029
  SRpcMsg rpcMsg = {0};
  (void)syncBuildHeartbeatReply(&rpcMsg, ths->vgId);

  SyncHeartbeatReply* pMsgReply = rpcMsg.pCont;
2030 2031 2032 2033 2034
  pMsgReply->destId = pMsg->srcId;
  pMsgReply->srcId = ths->myRaftId;
  pMsgReply->term = ths->pRaftStore->currentTerm;
  pMsgReply->privateTerm = 8864;  // magic number

M
Minghao Li 已提交
2035
  if (pMsg->term == ths->pRaftStore->currentTerm && ths->state != TAOS_SYNC_STATE_LEADER) {
2036
    syncNodeResetElectTimer(ths);
M
Minghao Li 已提交
2037
    ths->minMatchIndex = pMsg->minMatchIndex;
2038 2039

    if (ths->state == TAOS_SYNC_STATE_FOLLOWER) {
2040
      // syncNodeFollowerCommit(ths, pMsg->commitIndex);
S
Shengliang Guan 已提交
2041 2042 2043 2044
      SRpcMsg rpcMsgLocalCmd = {0};
      (void)syncBuildLocalCmd(&rpcMsgLocalCmd, ths->vgId);

      SyncLocalCmd* pSyncMsg = rpcMsgLocalCmd.pCont;
2045 2046 2047 2048 2049 2050 2051 2052 2053
      pSyncMsg->cmd = SYNC_LOCAL_CMD_FOLLOWER_CMT;
      pSyncMsg->fcIndex = pMsg->commitIndex;

      if (ths->syncEqMsg != NULL && ths->msgcb != NULL) {
        int32_t code = ths->syncEqMsg(ths->msgcb, &rpcMsgLocalCmd);
        if (code != 0) {
          sError("vgId:%d, sync enqueue fc-commit msg error, code:%d", ths->vgId, code);
          rpcFreeCont(rpcMsgLocalCmd.pCont);
        } else {
2054
          sTrace("vgId:%d, sync enqueue fc-commit msg, fc-index:%" PRId64, ths->vgId, pSyncMsg->fcIndex);
2055 2056
        }
      }
2057 2058 2059
    }
  }

M
Minghao Li 已提交
2060
  if (pMsg->term >= ths->pRaftStore->currentTerm && ths->state != TAOS_SYNC_STATE_FOLLOWER) {
2061
    // syncNodeStepDown(ths, pMsg->term);
S
Shengliang Guan 已提交
2062 2063 2064 2065
    SRpcMsg rpcMsgLocalCmd = {0};
    (void)syncBuildLocalCmd(&rpcMsgLocalCmd, ths->vgId);

    SyncLocalCmd* pSyncMsg = rpcMsgLocalCmd.pCont;
2066 2067 2068
    pSyncMsg->cmd = SYNC_LOCAL_CMD_STEP_DOWN;
    pSyncMsg->sdNewTerm = pMsg->term;

S
Shengliang Guan 已提交
2069 2070
    if (ths->syncEqMsg != NULL && ths->msgcb != NULL) {
      int32_t code = ths->syncEqMsg(ths->msgcb, &rpcMsgLocalCmd);
2071 2072 2073 2074
      if (code != 0) {
        sError("vgId:%d, sync enqueue step-down msg error, code:%d", ths->vgId, code);
        rpcFreeCont(rpcMsgLocalCmd.pCont);
      } else {
2075
        sTrace("vgId:%d, sync enqueue step-down msg, new-term: %" PRId64, ths->vgId, pSyncMsg->sdNewTerm);
2076 2077
      }
    }
M
Minghao Li 已提交
2078 2079
  }

2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091
  /*
    // htonl
    SMsgHead* pHead = rpcMsg.pCont;
    pHead->contLen = htonl(pHead->contLen);
    pHead->vgId = htonl(pHead->vgId);
  */

  // reply
  syncNodeSendMsgById(&pMsgReply->destId, ths, &rpcMsg);
  return 0;
}

2092 2093
int32_t syncNodeOnHeartbeatReply(SSyncNode* ths, const SRpcMsg* pRpcMsg) {
  SyncHeartbeatReply* pMsg = pRpcMsg->pCont;
2094 2095 2096
  syncLogRecvHeartbeatReply(ths, pMsg, "");

  // update last reply time, make decision whether the other node is alive or not
2097
  syncIndexMgrSetRecvTime(ths->pMatchIndex, &pMsg->destId, pMsg->startTime);
2098 2099 2100
  return 0;
}

S
Shengliang Guan 已提交
2101 2102
int32_t syncNodeOnLocalCmd(SSyncNode* ths, const SRpcMsg* pRpcMsg) {
  SyncLocalCmd* pMsg = pRpcMsg->pCont;
2103 2104
  syncLogRecvLocalCmd(ths, pMsg, "");

M
Minghao Li 已提交
2105 2106 2107
  if (pMsg->cmd == SYNC_LOCAL_CMD_STEP_DOWN) {
    syncNodeStepDown(ths, pMsg->sdNewTerm);

2108 2109 2110
  } else if (pMsg->cmd == SYNC_LOCAL_CMD_FOLLOWER_CMT) {
    syncNodeFollowerCommit(ths, pMsg->fcIndex);

M
Minghao Li 已提交
2111
  } else {
S
Shengliang Guan 已提交
2112
    sNError(ths, "error local cmd");
M
Minghao Li 已提交
2113 2114 2115 2116 2117
  }

  return 0;
}

M
Minghao Li 已提交
2118 2119 2120 2121 2122 2123 2124 2125 2126 2127
// TLA+ Spec
// ClientRequest(i, v) ==
//     /\ state[i] = Leader
//     /\ LET entry == [term  |-> currentTerm[i],
//                      value |-> v]
//            newLog == Append(log[i], entry)
//        IN  log' = [log EXCEPT ![i] = newLog]
//     /\ UNCHANGED <<messages, serverVars, candidateVars,
//                    leaderVars, commitIndex>>
//
M
Minghao Li 已提交
2128

2129
int32_t syncNodeOnClientRequest(SSyncNode* ths, SRpcMsg* pMsg, SyncIndex* pRetIndex) {
S
Shengliang Guan 已提交
2130
  sNTrace(ths, "on client request");
2131

M
Minghao Li 已提交
2132
  int32_t ret = 0;
2133
  int32_t code = 0;
M
Minghao Li 已提交
2134

M
Minghao Li 已提交
2135
  SyncIndex       index = ths->pLogStore->syncLogWriteIndex(ths->pLogStore);
M
Minghao Li 已提交
2136
  SyncTerm        term = ths->pRaftStore->currentTerm;
2137 2138 2139 2140 2141 2142 2143
  SSyncRaftEntry* pEntry;

  if (pMsg->msgType == TDMT_SYNC_CLIENT_REQUEST) {
    pEntry = syncEntryBuildFromClientRequest(pMsg->pCont, term, index);
  } else {
    pEntry = syncEntryBuildFromRpcMsg(pMsg, term, index);
  }
M
Minghao Li 已提交
2144

2145 2146 2147
  LRUHandle* h = NULL;
  syncCacheEntry(ths->pLogStore, pEntry, &h);

M
Minghao Li 已提交
2148
  if (ths->state == TAOS_SYNC_STATE_LEADER) {
2149 2150 2151
    // append entry
    code = ths->pLogStore->syncLogAppendEntry(ths->pLogStore, pEntry);
    if (code != 0) {
2152 2153 2154 2155 2156 2157
      if (ths->replicaNum == 1) {
        if (h) {
          taosLRUCacheRelease(ths->pLogStore->pCache, h, false);
        } else {
          syncEntryDestory(pEntry);
        }
2158

2159 2160 2161 2162
        return -1;

      } else {
        // del resp mgr, call FpCommitCb
2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173
        SFsmCbMeta cbMeta = {
            .index = pEntry->index,
            .lastConfigIndex = SYNC_INDEX_INVALID,
            .isWeak = pEntry->isWeak,
            .code = -1,
            .state = ths->state,
            .seqNum = pEntry->seqNum,
            .term = pEntry->term,
            .currentTerm = ths->pRaftStore->currentTerm,
            .flag = 0,
        };
2174
        ths->pFsm->FpCommitCb(ths->pFsm, pMsg, &cbMeta);
2175 2176 2177 2178 2179 2180 2181

        if (h) {
          taosLRUCacheRelease(ths->pLogStore->pCache, h, false);
        } else {
          syncEntryDestory(pEntry);
        }

2182 2183
        return -1;
      }
2184
    }
M
Minghao Li 已提交
2185

2186 2187
    // if mulit replica, start replicate right now
    if (ths->replicaNum > 1) {
M
Minghao Li 已提交
2188
      syncNodeReplicate(ths);
2189
    }
2190

2191 2192
    // if only myself, maybe commit right now
    if (ths->replicaNum == 1) {
2193 2194 2195 2196 2197
      if (syncNodeIsMnode(ths)) {
        syncMaybeAdvanceCommitIndex(ths);
      } else {
        syncOneReplicaAdvance(ths);
      }
2198
    }
M
Minghao Li 已提交
2199 2200
  }

2201 2202 2203 2204 2205 2206 2207 2208
  if (pRetIndex != NULL) {
    if (ret == 0 && pEntry != NULL) {
      *pRetIndex = pEntry->index;
    } else {
      *pRetIndex = SYNC_INDEX_INVALID;
    }
  }

2209 2210 2211 2212 2213 2214
  if (h) {
    taosLRUCacheRelease(ths->pLogStore->pCache, h, false);
  } else {
    syncEntryDestory(pEntry);
  }

M
Minghao Li 已提交
2215
  return ret;
2216
}
M
Minghao Li 已提交
2217

S
Shengliang Guan 已提交
2218 2219 2220
const char* syncStr(ESyncState state) {
  switch (state) {
    case TAOS_SYNC_STATE_FOLLOWER:
2221
      return "follower";
S
Shengliang Guan 已提交
2222
    case TAOS_SYNC_STATE_CANDIDATE:
2223
      return "candidate";
S
Shengliang Guan 已提交
2224
    case TAOS_SYNC_STATE_LEADER:
2225
      return "leader";
S
Shengliang Guan 已提交
2226
    default:
2227
      return "error";
S
Shengliang Guan 已提交
2228
  }
M
Minghao Li 已提交
2229
}
2230

2231
#if 0
2232
int32_t syncDoLeaderTransfer(SSyncNode* ths, SRpcMsg* pRpcMsg, SSyncRaftEntry* pEntry) {
2233
  if (ths->state != TAOS_SYNC_STATE_FOLLOWER) {
S
Shengliang Guan 已提交
2234
    sNTrace(ths, "I am not follower, can not do leader transfer");
2235 2236
    return 0;
  }
2237 2238

  if (!ths->restoreFinish) {
S
Shengliang Guan 已提交
2239
    sNTrace(ths, "restore not finish, can not do leader transfer");
2240 2241 2242
    return 0;
  }

2243
  if (pEntry->term < ths->pRaftStore->currentTerm) {
2244
    sNTrace(ths, "little term:%" PRId64 ", can not do leader transfer", pEntry->term);
2245 2246 2247 2248
    return 0;
  }

  if (pEntry->index < syncNodeGetLastIndex(ths)) {
S
Shengliang Guan 已提交
2249
    sNTrace(ths, "little index:%" PRId64 ", can not do leader transfer", pEntry->index);
2250 2251 2252
    return 0;
  }

2253 2254
  /*
    if (ths->vgId > 1) {
S
Shengliang Guan 已提交
2255
      sNTrace(ths, "I am vnode, can not do leader transfer");
2256 2257 2258 2259
      return 0;
    }
  */

2260
  SyncLeaderTransfer* pSyncLeaderTransfer = pRpcMsg->pCont;
S
Shengliang Guan 已提交
2261
  sNTrace(ths, "do leader transfer, index:%" PRId64, pEntry->index);
M
Minghao Li 已提交
2262

M
Minghao Li 已提交
2263 2264 2265
  bool sameId = syncUtilSameId(&(pSyncLeaderTransfer->newLeaderId), &(ths->myRaftId));
  bool sameNodeInfo = strcmp(pSyncLeaderTransfer->newNodeInfo.nodeFqdn, ths->myNodeInfo.nodeFqdn) == 0 &&
                      pSyncLeaderTransfer->newNodeInfo.nodePort == ths->myNodeInfo.nodePort;
M
Minghao Li 已提交
2266

M
Minghao Li 已提交
2267 2268
  bool same = sameId || sameNodeInfo;
  if (same) {
M
Minghao Li 已提交
2269 2270 2271 2272
    // reset elect timer now!
    int32_t electMS = 1;
    int32_t ret = syncNodeRestartElectTimer(ths, electMS);
    ASSERT(ret == 0);
M
Minghao Li 已提交
2273

2274
    sNTrace(ths, "maybe leader transfer to %s:%d %" PRId64, pSyncLeaderTransfer->newNodeInfo.nodeFqdn,
S
Shengliang Guan 已提交
2275
            pSyncLeaderTransfer->newNodeInfo.nodePort, pSyncLeaderTransfer->newLeaderId.addr);
2276 2277
  }

M
Minghao Li 已提交
2278
  if (ths->pFsm->FpLeaderTransferCb != NULL) {
S
Shengliang Guan 已提交
2279
    SFsmCbMeta cbMeta = {
S
Shengliang Guan 已提交
2280 2281 2282 2283 2284 2285 2286 2287 2288
        .code = 0,
        .currentTerm = ths->pRaftStore->currentTerm,
        .flag = 0,
        .index = pEntry->index,
        .lastConfigIndex = syncNodeGetSnapshotConfigIndex(ths, pEntry->index),
        .isWeak = pEntry->isWeak,
        .seqNum = pEntry->seqNum,
        .state = ths->state,
        .term = pEntry->term,
S
Shengliang Guan 已提交
2289 2290
    };
    ths->pFsm->FpLeaderTransferCb(ths->pFsm, pRpcMsg, &cbMeta);
2291 2292
  }

2293 2294 2295
  return 0;
}

2296 2297
#endif

2298
int32_t syncNodeUpdateNewConfigIndex(SSyncNode* ths, SSyncCfg* pNewCfg) {
S
Shengliang Guan 已提交
2299
  for (int32_t i = 0; i < pNewCfg->replicaNum; ++i) {
2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312
    SRaftId raftId;
    raftId.addr = syncUtilAddr2U64((pNewCfg->nodeInfo)[i].nodeFqdn, (pNewCfg->nodeInfo)[i].nodePort);
    raftId.vgId = ths->vgId;

    if (syncUtilSameId(&(ths->myRaftId), &raftId)) {
      pNewCfg->myIndex = i;
      return 0;
    }
  }

  return -1;
}

2313 2314 2315 2316
bool syncNodeIsOptimizedOneReplica(SSyncNode* ths, SRpcMsg* pMsg) {
  return (ths->replicaNum == 1 && syncUtilUserCommit(pMsg->msgType) && ths->vgId != 1);
}

M
Minghao Li 已提交
2317
int32_t syncNodeDoCommit(SSyncNode* ths, SyncIndex beginIndex, SyncIndex endIndex, uint64_t flag) {
2318 2319 2320 2321
  if (beginIndex > endIndex) {
    return 0;
  }

M
Minghao Li 已提交
2322 2323 2324 2325 2326 2327 2328 2329 2330
  if (ths == NULL) {
    return -1;
  }

  if (ths->pFsm != NULL && ths->pFsm->FpGetSnapshotInfo != NULL) {
    // advance commit index to sanpshot first
    SSnapshot snapshot = {0};
    ths->pFsm->FpGetSnapshotInfo(ths->pFsm, &snapshot);
    if (snapshot.lastApplyIndex >= 0 && snapshot.lastApplyIndex >= beginIndex) {
S
Shengliang Guan 已提交
2331
      sNTrace(ths, "commit by snapshot from index:%" PRId64 " to index:%" PRId64, beginIndex, snapshot.lastApplyIndex);
2332

M
Minghao Li 已提交
2333 2334 2335
      // update begin index
      beginIndex = snapshot.lastApplyIndex + 1;
    }
2336 2337
  }

2338 2339
  int32_t    code = 0;
  ESyncState state = flag;
M
Minghao Li 已提交
2340

S
Shengliang Guan 已提交
2341
  sNTrace(ths, "commit by wal from index:%" PRId64 " to index:%" PRId64, beginIndex, endIndex);
2342 2343 2344 2345 2346 2347

  // execute fsm
  if (ths->pFsm != NULL) {
    for (SyncIndex i = beginIndex; i <= endIndex; ++i) {
      if (i != SYNC_INDEX_INVALID) {
        SSyncRaftEntry* pEntry;
2348 2349 2350 2351 2352 2353
        SLRUCache*      pCache = ths->pLogStore->pCache;
        LRUHandle*      h = taosLRUCacheLookup(pCache, &i, sizeof(i));
        if (h) {
          pEntry = (SSyncRaftEntry*)taosLRUCacheValue(pCache, h);
        } else {
          code = ths->pLogStore->syncLogGetEntry(ths->pLogStore, i, &pEntry);
M
Minghao Li 已提交
2354 2355 2356
          // ASSERT(code == 0);
          // ASSERT(pEntry != NULL);
          if (code != 0 || pEntry == NULL) {
S
Shengliang Guan 已提交
2357
            sNError(ths, "get log entry error");
2358
            sFatal("vgId:%d, get log entry %" PRId64 " error when commit since %s", ths->vgId, i, terrstr());
M
Minghao Li 已提交
2359 2360
            continue;
          }
2361
        }
2362

2363
        SRpcMsg rpcMsg = {0};
2364 2365
        syncEntry2OriginalRpc(pEntry, &rpcMsg);

2366
        // user commit
2367 2368
        if ((ths->pFsm->FpCommitCb != NULL) && syncUtilUserCommit(pEntry->originalRpcType)) {
          bool internalExecute = true;
S
Shengliang Guan 已提交
2369
          if ((ths->replicaNum == 1) && ths->restoreFinish && ths->vgId != 1) {
2370 2371 2372
            internalExecute = false;
          }

S
Shengliang Guan 已提交
2373
          sNTrace(ths, "commit index:%" PRId64 ", internal:%d", i, internalExecute);
2374

2375 2376
          // execute fsm in apply thread, or execute outside syncPropose
          if (internalExecute) {
S
Shengliang Guan 已提交
2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388
            SFsmCbMeta cbMeta = {
                .index = pEntry->index,
                .lastConfigIndex = syncNodeGetSnapshotConfigIndex(ths, pEntry->index),
                .isWeak = pEntry->isWeak,
                .code = 0,
                .state = ths->state,
                .seqNum = pEntry->seqNum,
                .term = pEntry->term,
                .currentTerm = ths->pRaftStore->currentTerm,
                .flag = flag,
            };

S
Shengliang Guan 已提交
2389
            syncRespMgrGetAndDel(ths->pSyncRespMgr, cbMeta.seqNum, &rpcMsg.info);
S
Shengliang Guan 已提交
2390
            ths->pFsm->FpCommitCb(ths->pFsm, &rpcMsg, &cbMeta);
2391
          }
2392 2393
        }

2394 2395
#if 0
        // execute in pre-commit
M
Minghao Li 已提交
2396
        // leader transfer
2397 2398 2399
        if (pEntry->originalRpcType == TDMT_SYNC_LEADER_TRANSFER) {
          code = syncDoLeaderTransfer(ths, &rpcMsg, pEntry);
          ASSERT(code == 0);
2400
        }
2401
#endif
2402 2403

        // restore finish
2404
        // if only snapshot, a noop entry will be append, so syncLogLastIndex is always ok
2405 2406 2407 2408 2409 2410
        if (pEntry->index == ths->pLogStore->syncLogLastIndex(ths->pLogStore)) {
          if (ths->restoreFinish == false) {
            if (ths->pFsm->FpRestoreFinishCb != NULL) {
              ths->pFsm->FpRestoreFinishCb(ths->pFsm);
            }
            ths->restoreFinish = true;
M
Minghao Li 已提交
2411

2412
            int64_t restoreDelay = taosGetTimestampMs() - ths->leaderTime;
S
Shengliang Guan 已提交
2413
            sNTrace(ths, "restore finish, index:%" PRId64 ", elapsed:%" PRId64 " ms", pEntry->index, restoreDelay);
2414 2415 2416 2417
          }
        }

        rpcFreeCont(rpcMsg.pCont);
2418 2419 2420 2421 2422
        if (h) {
          taosLRUCacheRelease(pCache, h, false);
        } else {
          syncEntryDestory(pEntry);
        }
2423 2424 2425 2426
      }
    }
  }
  return 0;
2427 2428 2429
}

bool syncNodeInRaftGroup(SSyncNode* ths, SRaftId* pRaftId) {
S
Shengliang Guan 已提交
2430
  for (int32_t i = 0; i < ths->replicaNum; ++i) {
2431 2432 2433 2434 2435
    if (syncUtilSameId(&((ths->replicasId)[i]), pRaftId)) {
      return true;
    }
  }
  return false;
M
Minghao Li 已提交
2436 2437 2438 2439
}

SSyncSnapshotSender* syncNodeGetSnapshotSender(SSyncNode* ths, SRaftId* pDestId) {
  SSyncSnapshotSender* pSender = NULL;
S
Shengliang Guan 已提交
2440
  for (int32_t i = 0; i < ths->replicaNum; ++i) {
M
Minghao Li 已提交
2441 2442 2443 2444 2445
    if (syncUtilSameId(pDestId, &((ths->replicasId)[i]))) {
      pSender = (ths->senders)[i];
    }
  }
  return pSender;
M
Minghao Li 已提交
2446
}
M
Minghao Li 已提交
2447

2448 2449
SSyncTimer* syncNodeGetHbTimer(SSyncNode* ths, SRaftId* pDestId) {
  SSyncTimer* pTimer = NULL;
S
Shengliang Guan 已提交
2450
  for (int32_t i = 0; i < ths->replicaNum; ++i) {
2451 2452 2453 2454 2455 2456 2457
    if (syncUtilSameId(pDestId, &((ths->replicasId)[i]))) {
      pTimer = &((ths->peerHeartbeatTimerArr)[i]);
    }
  }
  return pTimer;
}

M
Minghao Li 已提交
2458 2459
SPeerState* syncNodeGetPeerState(SSyncNode* ths, const SRaftId* pDestId) {
  SPeerState* pState = NULL;
S
Shengliang Guan 已提交
2460
  for (int32_t i = 0; i < ths->replicaNum; ++i) {
M
Minghao Li 已提交
2461 2462 2463 2464 2465 2466 2467 2468 2469
    if (syncUtilSameId(pDestId, &((ths->replicasId)[i]))) {
      pState = &((ths->peerStates)[i]);
    }
  }
  return pState;
}

bool syncNodeNeedSendAppendEntries(SSyncNode* ths, const SRaftId* pDestId, const SyncAppendEntries* pMsg) {
  SPeerState* pState = syncNodeGetPeerState(ths, pDestId);
M
Minghao Li 已提交
2470
  if (pState == NULL) {
2471
    sError("vgId:%d, replica maybe dropped", ths->vgId);
M
Minghao Li 已提交
2472 2473
    return false;
  }
M
Minghao Li 已提交
2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484

  SyncIndex sendIndex = pMsg->prevLogIndex + 1;
  int64_t   tsNow = taosGetTimestampMs();

  if (pState->lastSendIndex == sendIndex && tsNow - pState->lastSendTime < SYNC_APPEND_ENTRIES_TIMEOUT_MS) {
    return false;
  }

  return true;
}

M
Minghao Li 已提交
2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498
bool syncNodeCanChange(SSyncNode* pSyncNode) {
  if (pSyncNode->changing) {
    sError("sync cannot change");
    return false;
  }

  if ((pSyncNode->commitIndex >= SYNC_INDEX_BEGIN)) {
    SyncIndex lastIndex = syncNodeGetLastIndex(pSyncNode);
    if (pSyncNode->commitIndex != lastIndex) {
      sError("sync cannot change2");
      return false;
    }
  }

S
Shengliang Guan 已提交
2499
  for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
M
Minghao Li 已提交
2500
    SSyncSnapshotSender* pSender = syncNodeGetSnapshotSender(pSyncNode, &(pSyncNode->peersId)[i]);
M
Minghao Li 已提交
2501
    if (pSender != NULL && pSender->start) {
M
Minghao Li 已提交
2502 2503 2504 2505 2506 2507
      sError("sync cannot change3");
      return false;
    }
  }

  return true;
M
Minghao Li 已提交
2508
}