syncMain.c 93.4 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"
25
#include "syncPipeline.h"
M
Minghao Li 已提交
26
#include "syncMessage.h"
M
Minghao Li 已提交
27
#include "syncRaftCfg.h"
M
Minghao Li 已提交
28
#include "syncRaftLog.h"
M
Minghao Li 已提交
29
#include "syncRaftStore.h"
M
Minghao Li 已提交
30
#include "syncReplication.h"
M
Minghao Li 已提交
31 32
#include "syncRequestVote.h"
#include "syncRequestVoteReply.h"
M
Minghao Li 已提交
33
#include "syncRespMgr.h"
M
Minghao Li 已提交
34
#include "syncSnapshot.h"
M
Minghao Li 已提交
35
#include "syncTimeout.h"
M
Minghao Li 已提交
36
#include "syncUtil.h"
M
Minghao Li 已提交
37
#include "syncVoteMgr.h"
M
Minghao Li 已提交
38
#include "tref.h"
M
Minghao Li 已提交
39

M
Minghao Li 已提交
40 41 42 43 44
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);
45
static void    syncNodeEqPeerHeartbeatTimer(void* param, void* tmrId);
S
Shengliang Guan 已提交
46
static bool    syncIsConfigChanged(const SSyncCfg* pOldCfg, const SSyncCfg* pNewCfg);
S
Shengliang Guan 已提交
47 48 49
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 已提交
50 51 52 53 54 55 56 57 58 59 60
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);
M
Minghao Li 已提交
61

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

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

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

B
Benguang Zhao 已提交
84
int32_t syncStart(int64_t rid) {
S
Shengliang Guan 已提交
85
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
M
Minghao Li 已提交
86
  if (pSyncNode == NULL) {
B
Benguang Zhao 已提交
87 88 89 90 91 92
    sError("failed to acquire rid: %" PRId64 " of tsNodeReftId for pSyncNode", rid);
    return -1;
  }

  if (syncNodeRestore(pSyncNode) < 0) {
    sError("vgId:%d, failed to restore raft log buffer since %s", pSyncNode->vgId, terrstr());
93
    goto _err;
M
Minghao Li 已提交
94
  }
M
Minghao Li 已提交
95

B
Benguang Zhao 已提交
96 97 98 99
  if (syncNodeStart(pSyncNode) < 0) {
    sError("vgId:%d, failed to start sync node since %s", pSyncNode->vgId, terrstr());
    goto _err;
  }
M
Minghao Li 已提交
100

B
Benguang Zhao 已提交
101 102
  syncNodeRelease(pSyncNode);
  return 0;
M
Minghao Li 已提交
103

104 105 106
_err:
  syncNodeRelease(pSyncNode);
  return -1;
M
Minghao Li 已提交
107 108
}

M
Minghao Li 已提交
109
void syncStop(int64_t rid) {
S
Shengliang Guan 已提交
110
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
S
Shengliang Guan 已提交
111
  if (pSyncNode != NULL) {
112
    pSyncNode->isStart = false;
S
Shengliang Guan 已提交
113
    syncNodeRelease(pSyncNode);
S
Shengliang Guan 已提交
114
    syncNodeRemove(rid);
M
Minghao Li 已提交
115 116 117
  }
}

M
Minghao Li 已提交
118 119
void syncPreStop(int64_t rid) {
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
120 121 122
  if (pSyncNode != NULL) {
    syncNodePreClose(pSyncNode);
    syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
123 124 125
  }
}

S
Shengliang Guan 已提交
126 127 128
static bool syncNodeCheckNewConfig(SSyncNode* pSyncNode, const SSyncCfg* pCfg) {
  if (!syncNodeInConfig(pSyncNode, pCfg)) return false;
  return abs(pCfg->replicaNum - pSyncNode->replicaNum) <= 1;
M
Minghao Li 已提交
129 130
}

S
Shengliang Guan 已提交
131
int32_t syncReconfig(int64_t rid, SSyncCfg* pNewCfg) {
S
Shengliang Guan 已提交
132
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
S
Shengliang Guan 已提交
133
  if (pSyncNode == NULL) return -1;
M
Minghao Li 已提交
134

M
Minghao Li 已提交
135
  if (!syncNodeCheckNewConfig(pSyncNode, pNewCfg)) {
S
Shengliang Guan 已提交
136
    syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
137
    terrno = TSDB_CODE_SYN_NEW_CONFIG_ERROR;
S
Shengliang Guan 已提交
138
    sError("vgId:%d, failed to reconfig since invalid new config", pSyncNode->vgId);
M
Minghao Li 已提交
139
    return -1;
M
Minghao Li 已提交
140
  }
141

S
Shengliang Guan 已提交
142 143
  syncNodeUpdateNewConfigIndex(pSyncNode, pNewCfg);
  syncNodeDoConfigChange(pSyncNode, pNewCfg, SYNC_INDEX_INVALID);
S
Shengliang Guan 已提交
144

M
Minghao Li 已提交
145 146 147 148
  if (pSyncNode->state == TAOS_SYNC_STATE_LEADER) {
    syncNodeStopHeartbeatTimer(pSyncNode);

    for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
S
Shengliang Guan 已提交
149
      syncHbTimerInit(pSyncNode, &pSyncNode->peerHeartbeatTimerArr[i], pSyncNode->replicasId[i]);
M
Minghao Li 已提交
150 151 152 153 154
    }

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

S
Shengliang Guan 已提交
156
  syncNodeRelease(pSyncNode);
S
Shengliang Guan 已提交
157
  return 0;
M
Minghao Li 已提交
158
}
M
Minghao Li 已提交
159

S
Shengliang Guan 已提交
160 161 162 163
int32_t syncProcessMsg(int64_t rid, SRpcMsg* pMsg) {
  int32_t code = -1;
  if (!syncIsInit()) return code;

S
Shengliang Guan 已提交
164
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
S
Shengliang Guan 已提交
165 166
  if (pSyncNode == NULL) return code;

S
Shengliang Guan 已提交
167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204
  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 已提交
205 206
  }

S
Shengliang Guan 已提交
207
  syncNodeRelease(pSyncNode);
S
Shengliang Guan 已提交
208
  return code;
209 210
}

S
Shengliang Guan 已提交
211
int32_t syncLeaderTransfer(int64_t rid) {
S
Shengliang Guan 已提交
212
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
S
Shengliang Guan 已提交
213
  if (pSyncNode == NULL) return -1;
214

S
Shengliang Guan 已提交
215
  int32_t ret = syncNodeLeaderTransfer(pSyncNode);
S
Shengliang Guan 已提交
216
  syncNodeRelease(pSyncNode);
217 218 219
  return ret;
}

M
Minghao Li 已提交
220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235
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;
}

236
int32_t syncBeginSnapshot(int64_t rid, int64_t lastApplyIndex) {
S
Shengliang Guan 已提交
237
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
238
  if (pSyncNode == NULL) {
239
    sError("sync begin snapshot error");
240 241
    return -1;
  }
242

243 244
  int32_t code = 0;

M
Minghao Li 已提交
245
  if (syncNodeIsMnode(pSyncNode)) {
M
Minghao Li 已提交
246 247 248
    // mnode
    int64_t logRetention = SYNC_MNODE_LOG_RETENTION;

M
Minghao Li 已提交
249 250 251
    SyncIndex beginIndex = pSyncNode->pLogStore->syncLogBeginIndex(pSyncNode->pLogStore);
    SyncIndex endIndex = pSyncNode->pLogStore->syncLogEndIndex(pSyncNode->pLogStore);
    int64_t   logNum = endIndex - beginIndex;
M
Minghao Li 已提交
252 253 254
    bool      isEmpty = pSyncNode->pLogStore->syncLogIsEmpty(pSyncNode->pLogStore);

    if (isEmpty || (!isEmpty && logNum < logRetention)) {
S
Shengliang Guan 已提交
255 256
      sNTrace(pSyncNode, "new-snapshot-index:%" PRId64 ", log-num:%" PRId64 ", empty:%d, do not delete wal",
              lastApplyIndex, logNum, isEmpty);
S
Shengliang Guan 已提交
257
      syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
258 259 260
      return 0;
    }

M
Minghao Li 已提交
261 262 263
    goto _DEL_WAL;

  } else {
264 265 266 267 268 269 270 271 272 273 274 275
    lastApplyIndex -= SYNC_VNODE_LOG_RETENTION;

    SyncIndex beginIndex = pSyncNode->pLogStore->syncLogBeginIndex(pSyncNode->pLogStore);
    SyncIndex endIndex = pSyncNode->pLogStore->syncLogEndIndex(pSyncNode->pLogStore);
    bool      isEmpty = pSyncNode->pLogStore->syncLogIsEmpty(pSyncNode->pLogStore);

    if (isEmpty || !(lastApplyIndex >= beginIndex && lastApplyIndex <= endIndex)) {
      sNTrace(pSyncNode, "new-snapshot-index:%" PRId64 ", empty:%d, do not delete wal", lastApplyIndex, isEmpty);
      syncNodeRelease(pSyncNode);
      return 0;
    }

M
Minghao Li 已提交
276 277 278 279 280 281 282 283 284 285 286 287 288 289
    // 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 已提交
290 291 292 293
              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 已提交
294 295
            } while (0);

S
Shengliang Guan 已提交
296
            syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
297 298 299 300 301 302
            return 0;
          }
        }

      } else if (pSyncNode->state == TAOS_SYNC_STATE_FOLLOWER) {
        if (lastApplyIndex > pSyncNode->minMatchIndex) {
S
Shengliang Guan 已提交
303 304 305
          sNTrace(pSyncNode,
                  "new-snapshot-index:%" PRId64 " is greater than min-match-index:%" PRId64 ", do not delete wal",
                  lastApplyIndex, pSyncNode->minMatchIndex);
S
Shengliang Guan 已提交
306
          syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
307 308 309 310
          return 0;
        }

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

      } else {
S
Shengliang Guan 已提交
316
        sNTrace(pSyncNode, "new-snapshot-index:%" PRId64 " unknown state, do not delete wal", lastApplyIndex);
S
Shengliang Guan 已提交
317
        syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
318 319 320 321 322 323 324 325 326
        return 0;
      }

      goto _DEL_WAL;

    } else {
      // one replica

      goto _DEL_WAL;
327 328 329
    }
  }

M
Minghao Li 已提交
330
_DEL_WAL:
331

M
Minghao Li 已提交
332
  do {
333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352
    SSyncLogStoreData* pData = pSyncNode->pLogStore->data;
    SyncIndex          snapshotVer = walGetSnapshotVer(pData->pWal);
    SyncIndex          walCommitVer = walGetCommittedVer(pData->pWal);
    SyncIndex          wallastVer = walGetLastVer(pData->pWal);
    if (lastApplyIndex <= walCommitVer) {
      SyncIndex snapshottingIndex = atomic_load_64(&pSyncNode->snapshottingIndex);

      if (snapshottingIndex == SYNC_INDEX_INVALID) {
        atomic_store_64(&pSyncNode->snapshottingIndex, lastApplyIndex);
        pSyncNode->snapshottingTime = taosGetTimestampMs();

        code = walBeginSnapshot(pData->pWal, lastApplyIndex);
        if (code == 0) {
          sNTrace(pSyncNode, "wal snapshot begin, index:%" PRId64 ", last apply index:%" PRId64,
                  pSyncNode->snapshottingIndex, lastApplyIndex);
        } else {
          sNError(pSyncNode, "wal snapshot begin error since:%s, index:%" PRId64 ", last apply index:%" PRId64,
                  terrstr(terrno), pSyncNode->snapshottingIndex, lastApplyIndex);
          atomic_store_64(&pSyncNode->snapshottingIndex, SYNC_INDEX_INVALID);
        }
353

M
Minghao Li 已提交
354
      } else {
355 356
        sNTrace(pSyncNode, "snapshotting for %" PRId64 ", do not delete wal for new-snapshot-index:%" PRId64,
                snapshottingIndex, lastApplyIndex);
M
Minghao Li 已提交
357
      }
358
    }
M
Minghao Li 已提交
359
  } while (0);
360

S
Shengliang Guan 已提交
361
  syncNodeRelease(pSyncNode);
362 363 364 365
  return code;
}

int32_t syncEndSnapshot(int64_t rid) {
S
Shengliang Guan 已提交
366
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
367
  if (pSyncNode == NULL) {
368
    sError("sync end snapshot error");
369 370 371
    return -1;
  }

372 373 374 375
  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 已提交
376
    if (code != 0) {
377
      sNError(pSyncNode, "wal snapshot end error since:%s", terrstr());
S
Shengliang Guan 已提交
378
      syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
379 380
      return -1;
    } else {
S
Shengliang Guan 已提交
381
      sNTrace(pSyncNode, "wal snapshot end, index:%" PRId64, atomic_load_64(&pSyncNode->snapshottingIndex));
M
Minghao Li 已提交
382 383
      atomic_store_64(&pSyncNode->snapshottingIndex, SYNC_INDEX_INVALID);
    }
384
  }
385

S
Shengliang Guan 已提交
386
  syncNodeRelease(pSyncNode);
387 388 389
  return code;
}

M
Minghao Li 已提交
390
int32_t syncStepDown(int64_t rid, SyncTerm newTerm) {
S
Shengliang Guan 已提交
391
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
M
Minghao Li 已提交
392
  if (pSyncNode == NULL) {
393
    sError("sync step down error");
M
Minghao Li 已提交
394 395 396
    return -1;
  }

M
Minghao Li 已提交
397
  syncNodeStepDown(pSyncNode, newTerm);
S
Shengliang Guan 已提交
398
  syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
399
  return 0;
M
Minghao Li 已提交
400 401
}

402 403 404
bool syncIsReadyForRead(int64_t rid) {
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
  if (pSyncNode == NULL) {
405
    sError("sync ready for read error");
406 407
    return false;
  }
M
Minghao Li 已提交
408

409 410 411
  if (pSyncNode->state == TAOS_SYNC_STATE_LEADER && pSyncNode->restoreFinish) {
    syncNodeRelease(pSyncNode);
    return true;
M
Minghao Li 已提交
412 413
  }

414 415
  bool ready = false;
  if (pSyncNode->state == TAOS_SYNC_STATE_LEADER && !pSyncNode->restoreFinish) {
416 417 418
    if (!pSyncNode->pFsm->FpApplyQueueEmptyCb(pSyncNode->pFsm)) {
      // apply queue not empty
      ready = false;
M
Minghao Li 已提交
419

420 421
    } else {
      if (!pSyncNode->pLogStore->syncLogIsEmpty(pSyncNode->pLogStore)) {
422
        SyncIndex       lastIndex = pSyncNode->pLogStore->syncLogLastIndex(pSyncNode->pLogStore);
423
        SSyncRaftEntry* pEntry = NULL;
424 425 426 427 428 429
        SLRUCache*      pCache = pSyncNode->pLogStore->pCache;
        LRUHandle*      h = taosLRUCacheLookup(pCache, &lastIndex, sizeof(lastIndex));
        int32_t         code = 0;
        if (h) {
          pEntry = (SSyncRaftEntry*)taosLRUCacheValue(pCache, h);
          code = 0;
M
Minghao Li 已提交
430

431
          sNTrace(pSyncNode, "hit cache index:%" PRId64 ", bytes:%u, %p", lastIndex, pEntry->bytes, pEntry);
M
Minghao Li 已提交
432

433 434
        } else {
          sNTrace(pSyncNode, "miss cache index:%" PRId64, lastIndex);
435

436 437
          code = pSyncNode->pLogStore->syncLogGetEntry(pSyncNode->pLogStore, lastIndex, &pEntry);
        }
438

439 440 441 442
        if (code == 0 && pEntry != NULL) {
          if (pEntry->originalRpcType == TDMT_SYNC_NOOP && pEntry->term == pSyncNode->pRaftStore->currentTerm) {
            ready = true;
          }
443

444 445 446
          if (h) {
            taosLRUCacheRelease(pCache, h, false);
          } else {
447
            syncEntryDestroy(pEntry);
448
          }
449
        }
450 451 452 453
      }
    }
  }

454 455 456 457 458 459 460
  if (!ready) {
    if (pSyncNode->state != TAOS_SYNC_STATE_LEADER) {
      terrno = TSDB_CODE_SYN_NOT_LEADER;
    } else {
      terrno = TSDB_CODE_APP_NOT_READY;
    }
  }
461

462 463
  syncNodeRelease(pSyncNode);
  return ready;
M
Minghao Li 已提交
464
}
M
Minghao Li 已提交
465

M
Minghao Li 已提交
466 467
int32_t syncNodeLeaderTransfer(SSyncNode* pSyncNode) {
  if (pSyncNode->peersNum == 0) {
S
Shengliang Guan 已提交
468
    sDebug("vgId:%d, only one replica, cannot leader transfer", pSyncNode->vgId);
M
Minghao Li 已提交
469 470
    terrno = TSDB_CODE_SYN_ONE_REPLICA;
    return -1;
M
Minghao Li 已提交
471
  }
M
Minghao Li 已提交
472

473
  int32_t ret = 0;
474
  if (pSyncNode->state == TAOS_SYNC_STATE_LEADER && pSyncNode->replicaNum > 1) {
475
    SNodeInfo newLeader = (pSyncNode->peersNodeInfo)[0];
476 477 478 479 480 481 482
    if (pSyncNode->peersNum == 2) {
      SyncIndex matchIndex0 = syncIndexMgrGetIndex(pSyncNode->pMatchIndex, &(pSyncNode->peersId[0]));
      SyncIndex matchIndex1 = syncIndexMgrGetIndex(pSyncNode->pMatchIndex, &(pSyncNode->peersId[1]));
      if (matchIndex1 > matchIndex0) {
        newLeader = (pSyncNode->peersNodeInfo)[1];
      }
    }
483 484 485
    ret = syncNodeLeaderTransferTo(pSyncNode, newLeader);
  }

M
Minghao Li 已提交
486
  return ret;
M
Minghao Li 已提交
487 488
}

M
Minghao Li 已提交
489 490
int32_t syncNodeLeaderTransferTo(SSyncNode* pSyncNode, SNodeInfo newLeader) {
  if (pSyncNode->replicaNum == 1) {
S
Shengliang Guan 已提交
491
    sDebug("vgId:%d, only one replica, cannot leader transfer", pSyncNode->vgId);
M
Minghao Li 已提交
492 493
    terrno = TSDB_CODE_SYN_ONE_REPLICA;
    return -1;
M
Minghao Li 已提交
494
  }
495

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

498 499 500 501
  SRpcMsg rpcMsg = {0};
  (void)syncBuildLeaderTransfer(&rpcMsg, pSyncNode->vgId);

  SyncLeaderTransfer* pMsg = rpcMsg.pCont;
M
Minghao Li 已提交
502 503 504 505
  pMsg->newLeaderId.addr = syncUtilAddr2U64(newLeader.nodeFqdn, newLeader.nodePort);
  pMsg->newLeaderId.vgId = pSyncNode->vgId;
  pMsg->newNodeInfo = newLeader;

S
Shengliang Guan 已提交
506 507 508
  int32_t ret = syncNodePropose(pSyncNode, &rpcMsg, false);
  rpcFreeCont(rpcMsg.pCont);
  return ret;
M
Minghao Li 已提交
509 510
}

511 512
SSyncState syncGetState(int64_t rid) {
  SSyncState state = {.state = TAOS_SYNC_STATE_ERROR};
M
Minghao Li 已提交
513

S
Shengliang Guan 已提交
514
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
515 516 517 518
  if (pSyncNode != NULL) {
    state.state = pSyncNode->state;
    state.restored = pSyncNode->restoreFinish;
    syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
519 520
  }

521
  return state;
M
Minghao Li 已提交
522 523
}

524
#if 0
525 526 527 528 529
int32_t syncGetSnapshotByIndex(int64_t rid, SyncIndex index, SSnapshot* pSnapshot) {
  if (index < SYNC_INDEX_BEGIN) {
    return -1;
  }

S
Shengliang Guan 已提交
530
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
531 532 533 534 535 536 537 538 539
  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) {
B
Benguang Zhao 已提交
540
      syncEntryDestroy(pEntry);
541
    }
S
Shengliang Guan 已提交
542
    syncNodeRelease(pSyncNode);
543 544 545 546 547 548 549 550 551
    return -1;
  }
  ASSERT(pEntry != NULL);

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

552
  syncEntryDestroy(pEntry);
S
Shengliang Guan 已提交
553
  syncNodeRelease(pSyncNode);
554 555 556
  return 0;
}

557
int32_t syncGetSnapshotMeta(int64_t rid, struct SSnapshotMeta* sMeta) {
S
Shengliang Guan 已提交
558
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
559 560 561
  if (pSyncNode == NULL) {
    return -1;
  }
M
Minghao Li 已提交
562
  ASSERT(rid == pSyncNode->rid);
563 564
  sMeta->lastConfigIndex = pSyncNode->pRaftCfg->lastConfigIndex;

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

S
Shengliang Guan 已提交
567
  syncNodeRelease(pSyncNode);
568 569 570
  return 0;
}

571
int32_t syncGetSnapshotMetaByIndex(int64_t rid, SyncIndex snapshotIndex, struct SSnapshotMeta* sMeta) {
S
Shengliang Guan 已提交
572
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
573 574 575
  if (pSyncNode == NULL) {
    return -1;
  }
M
Minghao Li 已提交
576
  ASSERT(rid == pSyncNode->rid);
577 578 579 580

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

S
Shengliang Guan 已提交
581
  for (int32_t i = 0; i < pSyncNode->pRaftCfg->configIndexCount; ++i) {
582 583 584 585 586 587
    if ((pSyncNode->pRaftCfg->configIndexArr)[i] > lastIndex &&
        (pSyncNode->pRaftCfg->configIndexArr)[i] <= snapshotIndex) {
      lastIndex = (pSyncNode->pRaftCfg->configIndexArr)[i];
    }
  }
  sMeta->lastConfigIndex = lastIndex;
588
  sTrace("vgId:%d, get snapshot meta by index:%" PRId64 " lcindex:%" PRId64, pSyncNode->vgId, snapshotIndex,
S
Shengliang Guan 已提交
589
         sMeta->lastConfigIndex);
590

S
Shengliang Guan 已提交
591
  syncNodeRelease(pSyncNode);
592 593
  return 0;
}
594
#endif
595

596 597 598 599
SyncIndex syncNodeGetSnapshotConfigIndex(SSyncNode* pSyncNode, SyncIndex snapshotLastApplyIndex) {
  ASSERT(pSyncNode->pRaftCfg->configIndexCount >= 1);
  SyncIndex lastIndex = (pSyncNode->pRaftCfg->configIndexArr)[0];

S
Shengliang Guan 已提交
600
  for (int32_t i = 0; i < pSyncNode->pRaftCfg->configIndexCount; ++i) {
601 602 603 604 605
    if ((pSyncNode->pRaftCfg->configIndexArr)[i] > lastIndex &&
        (pSyncNode->pRaftCfg->configIndexArr)[i] <= snapshotLastApplyIndex) {
      lastIndex = (pSyncNode->pRaftCfg->configIndexArr)[i];
    }
  }
S
Shengliang Guan 已提交
606
  sTrace("vgId:%d, sync get last config index, index:%" PRId64 " lcindex:%" PRId64, pSyncNode->vgId,
S
Shengliang Guan 已提交
607
         snapshotLastApplyIndex, lastIndex);
608 609 610 611

  return lastIndex;
}

612 613
void syncGetRetryEpSet(int64_t rid, SEpSet* pEpSet) {
  pEpSet->numOfEps = 0;
M
Minghao Li 已提交
614

S
Shengliang Guan 已提交
615
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
S
Shengliang Guan 已提交
616
  if (pSyncNode == NULL) return;
M
Minghao Li 已提交
617

S
Shengliang Guan 已提交
618
  for (int32_t i = 0; i < pSyncNode->pRaftCfg->cfg.replicaNum; ++i) {
S
Shengliang Guan 已提交
619 620 621 622
    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++;
623
    sDebug("vgId:%d, sync get retry epset, index:%d %s:%d", pSyncNode->vgId, i, pEp->fqdn, pEp->port);
M
Minghao Li 已提交
624
  }
M
Minghao Li 已提交
625 626
  if (pEpSet->numOfEps > 0) {
    pEpSet->inUse = (pSyncNode->pRaftCfg->cfg.myIndex + 1) % pEpSet->numOfEps;
M
Minghao Li 已提交
627 628
  }

S
Shengliang Guan 已提交
629
  sInfo("vgId:%d, sync get retry epset numOfEps:%d inUse:%d", pSyncNode->vgId, pEpSet->numOfEps, pEpSet->inUse);
S
Shengliang Guan 已提交
630
  syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
631 632
}

M
Minghao Li 已提交
633
int32_t syncPropose(int64_t rid, SRpcMsg* pMsg, bool isWeak) {
S
Shengliang Guan 已提交
634
  SSyncNode* pSyncNode = syncNodeAcquire(rid);
635
  if (pSyncNode == NULL) {
636
    sError("sync propose error");
M
Minghao Li 已提交
637
    return -1;
638
  }
639

640
  int32_t ret = syncNodePropose(pSyncNode, pMsg, isWeak);
S
Shengliang Guan 已提交
641
  syncNodeRelease(pSyncNode);
M
Minghao Li 已提交
642 643
  return ret;
}
M
Minghao Li 已提交
644

645
int32_t syncNodePropose(SSyncNode* pSyncNode, SRpcMsg* pMsg, bool isWeak) {
S
Shengliang Guan 已提交
646 647 648 649 650
  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;
  }
651

S
Shengliang Guan 已提交
652 653 654 655 656 657 658
  // 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;
  }
659

660
  // heartbeat timeout
661
  if (syncNodeHeartbeatReplyTimeout(pSyncNode)) {
662 663 664 665 666 667
    terrno = TSDB_CODE_SYN_PROPOSE_NOT_READY;
    sNError(pSyncNode, "failed to sync propose since hearbeat timeout, type:%s, last:%" PRId64 ", cmt:%" PRId64,
            TMSG_INFO(pMsg->msgType), syncNodeGetLastIndex(pSyncNode), pSyncNode->commitIndex);
    return -1;
  }

S
Shengliang Guan 已提交
668 669 670
  // optimized one replica
  if (syncNodeIsOptimizedOneReplica(pSyncNode, pMsg)) {
    SyncIndex retIndex;
671
    int32_t   code = syncNodeOnClientRequest(pSyncNode, pMsg, &retIndex);
S
Shengliang Guan 已提交
672 673 674
    if (code == 0) {
      pMsg->info.conn.applyIndex = retIndex;
      pMsg->info.conn.applyTerm = pSyncNode->pRaftStore->currentTerm;
675 676 677
      sTrace("vgId:%d, propose optimized msg, index:%" PRId64 " type:%s", pSyncNode->vgId, retIndex,
             TMSG_INFO(pMsg->msgType));
      return 1;
M
Minghao Li 已提交
678
    } else {
S
Shengliang Guan 已提交
679
      terrno = TSDB_CODE_SYN_INTERNAL_ERROR;
680
      sError("vgId:%d, failed to propose optimized msg, index:%" PRId64 " type:%s", pSyncNode->vgId, retIndex,
S
Shengliang Guan 已提交
681
             TMSG_INFO(pMsg->msgType));
682
      return -1;
683
    }
S
Shengliang Guan 已提交
684
  } else {
S
Shengliang Guan 已提交
685 686
    SRespStub stub = {.createTime = taosGetTimestampMs(), .rpcMsg = *pMsg};
    uint64_t  seqNum = syncRespMgrAdd(pSyncNode->pSyncRespMgr, &stub);
687
    SRpcMsg   rpcMsg = {0};
S
Shengliang Guan 已提交
688
    int32_t   code = syncBuildClientRequest(&rpcMsg, pMsg, seqNum, isWeak, pSyncNode->vgId);
689 690 691 692
    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 已提交
693
    }
694

695 696 697 698 699
    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 已提交
700
    }
M
Minghao Li 已提交
701

702
    return code;
M
Minghao Li 已提交
703
  }
M
Minghao Li 已提交
704 705
}

S
Shengliang Guan 已提交
706
static int32_t syncHbTimerInit(SSyncNode* pSyncNode, SSyncTimer* pSyncTimer, SRaftId destId) {
707 708 709 710 711 712 713 714 715
  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 已提交
716
static int32_t syncHbTimerStart(SSyncNode* pSyncNode, SSyncTimer* pSyncTimer) {
717
  int32_t ret = 0;
S
Shengliang Guan 已提交
718
  if (syncIsInit()) {
719 720 721 722 723 724 725 726
    SSyncHbTimerData* pData = syncHbTimerDataAcquire(pSyncTimer->hbDataRid);
    if (pData == NULL) {
      pData = taosMemoryMalloc(sizeof(SSyncHbTimerData));
      pData->rid = syncHbTimerDataAdd(pData);
    }
    pSyncTimer->hbDataRid = pData->rid;

    pData->syncNodeRid = pSyncNode->rid;
727 728 729
    pData->pTimer = pSyncTimer;
    pData->destId = pSyncTimer->destId;
    pData->logicClock = pSyncTimer->logicClock;
M
Minghao Li 已提交
730

731 732
    taosTmrReset(pSyncTimer->timerCb, pSyncTimer->timerMS, (void*)(pData->rid), syncEnv()->pTimerManager,
                 &pSyncTimer->pTimer);
733 734 735 736 737 738
  } else {
    sError("vgId:%d, start ctrl hb timer error, sync env is stop", pSyncNode->vgId);
  }
  return ret;
}

S
Shengliang Guan 已提交
739
static int32_t syncHbTimerStop(SSyncNode* pSyncNode, SSyncTimer* pSyncTimer) {
740 741 742 743
  int32_t ret = 0;
  atomic_add_fetch_64(&pSyncTimer->logicClock, 1);
  taosTmrStop(pSyncTimer->pTimer);
  pSyncTimer->pTimer = NULL;
744 745
  syncHbTimerDataRemove(pSyncTimer->hbDataRid);
  pSyncTimer->hbDataRid = -1;
746 747 748
  return ret;
}

749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770
int32_t syncNodeLogStoreRestoreOnNeed(SSyncNode* pNode) {
  ASSERT(pNode->pLogStore != NULL && "log store not created");
  ASSERT(pNode->pFsm != NULL && "pFsm not registered");
  ASSERT(pNode->pFsm->FpGetSnapshotInfo != NULL && "FpGetSnapshotInfo not registered");
  SSnapshot snapshot;
  if (pNode->pFsm->FpGetSnapshotInfo(pNode->pFsm, &snapshot) < 0) {
    sError("vgId:%d, failed to get snapshot info since %s", pNode->vgId, terrstr());
    return -1;
  }
  SyncIndex commitIndex = snapshot.lastApplyIndex;
  SyncIndex firstVer = pNode->pLogStore->syncLogBeginIndex(pNode->pLogStore);
  SyncIndex lastVer = pNode->pLogStore->syncLogLastIndex(pNode->pLogStore);
  if (lastVer < commitIndex || firstVer > commitIndex + 1) {
    if (pNode->pLogStore->syncLogRestoreFromSnapshot(pNode->pLogStore, commitIndex)) {
      sError("vgId:%d, failed to restore log store from snapshot since %s. lastVer: %" PRId64 ", snapshotVer: %" PRId64,
             pNode->vgId, terrstr(), lastVer, commitIndex);
      return -1;
    }
  }
  return 0;
}

M
Minghao Li 已提交
771
// open/close --------------
S
Shengliang Guan 已提交
772 773
SSyncNode* syncNodeOpen(SSyncInfo* pSyncInfo) {
  SSyncNode* pSyncNode = taosMemoryCalloc(1, sizeof(SSyncNode));
774 775 776 777
  if (pSyncNode == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _error;
  }
M
Minghao Li 已提交
778

M
Minghao Li 已提交
779 780 781 782
  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());
783
      goto _error;
M
Minghao Li 已提交
784
    }
785
  }
M
Minghao Li 已提交
786

S
Shengliang Guan 已提交
787
  snprintf(pSyncNode->configPath, sizeof(pSyncNode->configPath), "%s%sraft_config.json", pSyncInfo->path, TD_DIRSEP);
788
  if (!taosCheckExistFile(pSyncNode->configPath)) {
M
Minghao Li 已提交
789
    // create a new raft config file
S
Shengliang Guan 已提交
790
    SRaftCfgMeta meta = {0};
M
Minghao Li 已提交
791
    meta.isStandBy = pSyncInfo->isStandBy;
M
Minghao Li 已提交
792
    meta.snapshotStrategy = pSyncInfo->snapshotStrategy;
793
    meta.lastConfigIndex = SYNC_INDEX_INVALID;
M
Minghao Li 已提交
794
    meta.batchSize = pSyncInfo->batchSize;
S
Shengliang Guan 已提交
795 796
    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 已提交
797
      goto _error;
798
    }
799
    if (pSyncInfo->syncCfg.replicaNum == 0) {
S
Shengliang Guan 已提交
800
      sInfo("vgId:%d, sync config not input", pSyncNode->vgId);
801 802
      pSyncInfo->syncCfg = pSyncNode->pRaftCfg->cfg;
    }
803 804 805
  } else {
    // update syncCfg by raft_config.json
    pSyncNode->pRaftCfg = raftCfgOpen(pSyncNode->configPath);
806
    if (pSyncNode->pRaftCfg == NULL) {
S
Shengliang Guan 已提交
807
      sError("vgId:%d, failed to open raft cfg file at %s", pSyncNode->vgId, pSyncNode->configPath);
H
Hongze Cheng 已提交
808
      goto _error;
809
    }
S
Shengliang Guan 已提交
810 811

    if (pSyncInfo->syncCfg.replicaNum > 0 && syncIsConfigChanged(&pSyncNode->pRaftCfg->cfg, &pSyncInfo->syncCfg)) {
S
Shengliang Guan 已提交
812 813 814 815 816 817
      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 已提交
818 819 820 821
    } else {
      sInfo("vgId:%d, use sync config from raft cfg file", pSyncNode->vgId);
      pSyncInfo->syncCfg = pSyncNode->pRaftCfg->cfg;
    }
822 823

    raftCfgClose(pSyncNode->pRaftCfg);
824
    pSyncNode->pRaftCfg = NULL;
M
Minghao Li 已提交
825 826
  }

M
Minghao Li 已提交
827
  // init by SSyncInfo
M
Minghao Li 已提交
828
  pSyncNode->vgId = pSyncInfo->vgId;
S
Shengliang Guan 已提交
829 830 831 832 833 834 835
  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 已提交
836
  memcpy(pSyncNode->path, pSyncInfo->path, sizeof(pSyncNode->path));
S
Shengliang Guan 已提交
837 838 839
  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 已提交
840

M
Minghao Li 已提交
841
  pSyncNode->pWal = pSyncInfo->pWal;
S
Shengliang Guan 已提交
842
  pSyncNode->msgcb = pSyncInfo->msgcb;
S
Shengliang Guan 已提交
843 844 845
  pSyncNode->syncSendMSg = pSyncInfo->syncSendMSg;
  pSyncNode->syncEqMsg = pSyncInfo->syncEqMsg;
  pSyncNode->syncEqCtrlMsg = pSyncInfo->syncEqCtrlMsg;
M
Minghao Li 已提交
846

B
Benguang Zhao 已提交
847 848 849 850 851 852 853
  // create raft log ring buffer
  pSyncNode->pLogBuf = syncLogBufferCreate();
  if (pSyncNode->pLogBuf == NULL) {
    sError("failed to init log buffer since %s. vgId:%d", terrstr(), pSyncNode->vgId);
    goto _error;
  }

M
Minghao Li 已提交
854 855
  // init raft config
  pSyncNode->pRaftCfg = raftCfgOpen(pSyncNode->configPath);
856
  if (pSyncNode->pRaftCfg == NULL) {
S
Shengliang Guan 已提交
857
    sError("vgId:%d, failed to open raft cfg file at %s", pSyncNode->vgId, pSyncNode->configPath);
858 859
    goto _error;
  }
M
Minghao Li 已提交
860

M
Minghao Li 已提交
861
  // init internal
M
Minghao Li 已提交
862
  pSyncNode->myNodeInfo = pSyncNode->pRaftCfg->cfg.nodeInfo[pSyncNode->pRaftCfg->cfg.myIndex];
863
  if (!syncUtilNodeInfo2RaftId(&pSyncNode->myNodeInfo, pSyncNode->vgId, &pSyncNode->myRaftId)) {
S
Shengliang Guan 已提交
864
    sError("vgId:%d, failed to determine my raft member id", pSyncNode->vgId);
H
Hongze Cheng 已提交
865
    goto _error;
866
  }
M
Minghao Li 已提交
867

M
Minghao Li 已提交
868
  // init peersNum, peers, peersId
M
Minghao Li 已提交
869
  pSyncNode->peersNum = pSyncNode->pRaftCfg->cfg.replicaNum - 1;
S
Shengliang Guan 已提交
870 871
  int32_t j = 0;
  for (int32_t i = 0; i < pSyncNode->pRaftCfg->cfg.replicaNum; ++i) {
M
Minghao Li 已提交
872 873
    if (i != pSyncNode->pRaftCfg->cfg.myIndex) {
      pSyncNode->peersNodeInfo[j] = pSyncNode->pRaftCfg->cfg.nodeInfo[i];
M
Minghao Li 已提交
874 875 876
      j++;
    }
  }
S
Shengliang Guan 已提交
877
  for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
878
    if (!syncUtilNodeInfo2RaftId(&pSyncNode->peersNodeInfo[i], pSyncNode->vgId, &pSyncNode->peersId[i])) {
S
Shengliang Guan 已提交
879
      sError("vgId:%d, failed to determine raft member id, peer:%d", pSyncNode->vgId, i);
H
Hongze Cheng 已提交
880
      goto _error;
881
    }
M
Minghao Li 已提交
882
  }
M
Minghao Li 已提交
883

M
Minghao Li 已提交
884
  // init replicaNum, replicasId
M
Minghao Li 已提交
885
  pSyncNode->replicaNum = pSyncNode->pRaftCfg->cfg.replicaNum;
S
Shengliang Guan 已提交
886
  for (int32_t i = 0; i < pSyncNode->pRaftCfg->cfg.replicaNum; ++i) {
887
    if (!syncUtilNodeInfo2RaftId(&pSyncNode->pRaftCfg->cfg.nodeInfo[i], pSyncNode->vgId, &pSyncNode->replicasId[i])) {
S
Shengliang Guan 已提交
888
      sError("vgId:%d, failed to determine raft member id, replica:%d", pSyncNode->vgId, i);
H
Hongze Cheng 已提交
889
      goto _error;
890
    }
M
Minghao Li 已提交
891 892
  }

M
Minghao Li 已提交
893
  // init raft algorithm
M
Minghao Li 已提交
894
  pSyncNode->pFsm = pSyncInfo->pFsm;
895
  pSyncInfo->pFsm = NULL;
M
Minghao Li 已提交
896
  pSyncNode->quorum = syncUtilQuorum(pSyncNode->pRaftCfg->cfg.replicaNum);
M
Minghao Li 已提交
897 898
  pSyncNode->leaderCache = EMPTY_RAFT_ID;

M
Minghao Li 已提交
899
  // init life cycle outside
M
Minghao Li 已提交
900

M
Minghao Li 已提交
901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924
  // 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 已提交
925
  // init TLA+ server vars
M
syncInt  
Minghao Li 已提交
926
  pSyncNode->state = TAOS_SYNC_STATE_FOLLOWER;
M
Minghao Li 已提交
927
  pSyncNode->pRaftStore = raftStoreOpen(pSyncNode->raftStorePath);
928
  if (pSyncNode->pRaftStore == NULL) {
S
Shengliang Guan 已提交
929
    sError("vgId:%d, failed to open raft store at path %s", pSyncNode->vgId, pSyncNode->raftStorePath);
930 931
    goto _error;
  }
M
Minghao Li 已提交
932

M
Minghao Li 已提交
933
  // init TLA+ candidate vars
M
Minghao Li 已提交
934
  pSyncNode->pVotesGranted = voteGrantedCreate(pSyncNode);
935
  if (pSyncNode->pVotesGranted == NULL) {
S
Shengliang Guan 已提交
936
    sError("vgId:%d, failed to create VotesGranted", pSyncNode->vgId);
937 938
    goto _error;
  }
M
Minghao Li 已提交
939
  pSyncNode->pVotesRespond = votesRespondCreate(pSyncNode);
940
  if (pSyncNode->pVotesRespond == NULL) {
S
Shengliang Guan 已提交
941
    sError("vgId:%d, failed to create VotesRespond", pSyncNode->vgId);
942 943
    goto _error;
  }
M
Minghao Li 已提交
944

M
Minghao Li 已提交
945 946
  // init TLA+ leader vars
  pSyncNode->pNextIndex = syncIndexMgrCreate(pSyncNode);
947
  if (pSyncNode->pNextIndex == NULL) {
S
Shengliang Guan 已提交
948
    sError("vgId:%d, failed to create SyncIndexMgr", pSyncNode->vgId);
949 950
    goto _error;
  }
M
Minghao Li 已提交
951
  pSyncNode->pMatchIndex = syncIndexMgrCreate(pSyncNode);
952
  if (pSyncNode->pMatchIndex == NULL) {
S
Shengliang Guan 已提交
953
    sError("vgId:%d, failed to create SyncIndexMgr", pSyncNode->vgId);
954 955
    goto _error;
  }
M
Minghao Li 已提交
956 957 958

  // init TLA+ log vars
  pSyncNode->pLogStore = logStoreCreate(pSyncNode);
959
  if (pSyncNode->pLogStore == NULL) {
S
Shengliang Guan 已提交
960
    sError("vgId:%d, failed to create SyncLogStore", pSyncNode->vgId);
961 962
    goto _error;
  }
963 964 965 966 967

  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);
968
    if (code != 0) {
S
Shengliang Guan 已提交
969
      sError("vgId:%d, failed to get snapshot info, code:%d", pSyncNode->vgId, code);
H
Hongze Cheng 已提交
970
      goto _error;
971
    }
972 973
    if (snapshot.lastApplyIndex > commitIndex) {
      commitIndex = snapshot.lastApplyIndex;
S
Shengliang Guan 已提交
974
      sNTrace(pSyncNode, "reset commit index by snapshot");
975 976 977
    }
  }
  pSyncNode->commitIndex = commitIndex;
M
Minghao Li 已提交
978

979 980 981
  if (syncNodeLogStoreRestoreOnNeed(pSyncNode) < 0) {
    goto _error;
  }
M
Minghao Li 已提交
982 983 984 985 986
  // timer ms init
  pSyncNode->pingBaseLine = PING_TIMER_MS;
  pSyncNode->electBaseLine = ELECT_TIMER_MS_MIN;
  pSyncNode->hbBaseLine = HEARTBEAT_TIMER_MS;

M
Minghao Li 已提交
987
  // init ping timer
M
Minghao Li 已提交
988
  pSyncNode->pPingTimer = NULL;
M
Minghao Li 已提交
989
  pSyncNode->pingTimerMS = pSyncNode->pingBaseLine;
M
Minghao Li 已提交
990 991
  atomic_store_64(&pSyncNode->pingTimerLogicClock, 0);
  atomic_store_64(&pSyncNode->pingTimerLogicClockUser, 0);
M
Minghao Li 已提交
992
  pSyncNode->FpPingTimerCB = syncNodeEqPingTimer;
M
Minghao Li 已提交
993
  pSyncNode->pingTimerCounter = 0;
M
Minghao Li 已提交
994

M
Minghao Li 已提交
995 996
  // init elect timer
  pSyncNode->pElectTimer = NULL;
M
Minghao Li 已提交
997
  pSyncNode->electTimerMS = syncUtilElectRandomMS(pSyncNode->electBaseLine, 2 * pSyncNode->electBaseLine);
M
Minghao Li 已提交
998
  atomic_store_64(&pSyncNode->electTimerLogicClock, 0);
M
Minghao Li 已提交
999
  pSyncNode->FpElectTimerCB = syncNodeEqElectTimer;
M
Minghao Li 已提交
1000 1001 1002 1003
  pSyncNode->electTimerCounter = 0;

  // init heartbeat timer
  pSyncNode->pHeartbeatTimer = NULL;
M
Minghao Li 已提交
1004
  pSyncNode->heartbeatTimerMS = pSyncNode->hbBaseLine;
M
Minghao Li 已提交
1005 1006
  atomic_store_64(&pSyncNode->heartbeatTimerLogicClock, 0);
  atomic_store_64(&pSyncNode->heartbeatTimerLogicClockUser, 0);
M
Minghao Li 已提交
1007
  pSyncNode->FpHeartbeatTimerCB = syncNodeEqHeartbeatTimer;
M
Minghao Li 已提交
1008 1009
  pSyncNode->heartbeatTimerCounter = 0;

1010 1011 1012 1013 1014
  // 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 已提交
1015
  // tools
M
Minghao Li 已提交
1016
  pSyncNode->pSyncRespMgr = syncRespMgrCreate(pSyncNode, SYNC_RESP_TTL_MS);
1017
  if (pSyncNode->pSyncRespMgr == NULL) {
S
Shengliang Guan 已提交
1018
    sError("vgId:%d, failed to create SyncRespMgr", pSyncNode->vgId);
1019 1020
    goto _error;
  }
M
Minghao Li 已提交
1021

1022 1023
  // restore state
  pSyncNode->restoreFinish = false;
1024

M
Minghao Li 已提交
1025
  // snapshot senders
S
Shengliang Guan 已提交
1026
  for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1027 1028 1029
    SSyncSnapshotSender* pSender = snapshotSenderCreate(pSyncNode, i);
    // ASSERT(pSender != NULL);
    (pSyncNode->senders)[i] = pSender;
S
Shengliang Guan 已提交
1030
    sSTrace(pSender, "snapshot sender create new while open, data:%p", pSender);
M
Minghao Li 已提交
1031 1032 1033
  }

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

M
Minghao Li 已提交
1036 1037 1038
  // is config changing
  pSyncNode->changing = false;

B
Benguang Zhao 已提交
1039 1040 1041
  // replication mgr
  syncNodeLogReplMgrInit(pSyncNode);

M
Minghao Li 已提交
1042 1043 1044
  // peer state
  syncNodePeerStateInit(pSyncNode);

B
Benguang Zhao 已提交
1045
  //
M
Minghao Li 已提交
1046 1047 1048
  // min match index
  pSyncNode->minMatchIndex = SYNC_INDEX_INVALID;

M
Minghao Li 已提交
1049
  // start in syncNodeStart
M
Minghao Li 已提交
1050
  // start raft
M
Minghao Li 已提交
1051
  // syncNodeBecomeFollower(pSyncNode);
M
Minghao Li 已提交
1052

M
Minghao Li 已提交
1053 1054
  int64_t timeNow = taosGetTimestampMs();
  pSyncNode->startTime = timeNow;
1055
  pSyncNode->leaderTime = timeNow;
M
Minghao Li 已提交
1056 1057
  pSyncNode->lastReplicateTime = timeNow;

1058 1059 1060
  // snapshotting
  atomic_store_64(&pSyncNode->snapshottingIndex, SYNC_INDEX_INVALID);

B
Benguang Zhao 已提交
1061 1062 1063
  // init log buffer
  if (syncLogBufferInit(pSyncNode->pLogBuf, pSyncNode) < 0) {
    sError("vgId:%d, failed to init raft log buffer since %s", pSyncNode->vgId, terrstr());
1064
    goto _error;
B
Benguang Zhao 已提交
1065 1066
  }

1067
  pSyncNode->isStart = true;
S
Shengliang Guan 已提交
1068
  sNTrace(pSyncNode, "sync open, node:%p", pSyncNode);
1069

M
Minghao Li 已提交
1070
  return pSyncNode;
1071 1072 1073

_error:
  if (pSyncInfo->pFsm) {
H
Hongze Cheng 已提交
1074 1075
    taosMemoryFree(pSyncInfo->pFsm);
    pSyncInfo->pFsm = NULL;
1076 1077 1078 1079
  }
  syncNodeClose(pSyncNode);
  pSyncNode = NULL;
  return NULL;
M
Minghao Li 已提交
1080 1081
}

M
Minghao Li 已提交
1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092
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;
    }
  }
}

B
Benguang Zhao 已提交
1093 1094 1095 1096 1097 1098 1099
int32_t syncNodeRestore(SSyncNode* pSyncNode) {
  ASSERT(pSyncNode->pLogStore != NULL && "log store not created");
  ASSERT(pSyncNode->pLogBuf != NULL && "ring log buffer not created");

  SyncIndex lastVer = pSyncNode->pLogStore->syncLogLastIndex(pSyncNode->pLogStore);
  SyncIndex commitIndex = pSyncNode->pLogStore->syncLogCommitIndex(pSyncNode->pLogStore);
  SyncIndex endIndex = pSyncNode->pLogBuf->endIndex;
1100 1101 1102 1103 1104 1105
  if (lastVer != -1 && endIndex != lastVer + 1) {
    terrno = TSDB_CODE_WAL_LOG_INCOMPLETE;
    sError("vgId:%d, failed to restore sync node since %s. expected lastLogIndex: %" PRId64 ", lastVer: %" PRId64 "",
           pSyncNode->vgId, terrstr(), endIndex - 1, lastVer);
    return -1;
  }
B
Benguang Zhao 已提交
1106

1107
  ASSERT(endIndex == lastVer + 1);
B
Benguang Zhao 已提交
1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135
  commitIndex = TMAX(pSyncNode->commitIndex, commitIndex);

  if (syncLogBufferCommit(pSyncNode->pLogBuf, pSyncNode, commitIndex) < 0) {
    return -1;
  }

  return 0;
}

int32_t syncNodeStart(SSyncNode* pSyncNode) {
  // start raft
  if (pSyncNode->replicaNum == 1) {
    raftStoreNextTerm(pSyncNode->pRaftStore);
    syncNodeBecomeLeader(pSyncNode, "one replica start");

    // Raft 3.6.2 Committing entries from previous terms
    syncNodeAppendNoop(pSyncNode);
  } else {
    syncNodeBecomeFollower(pSyncNode, "first start");
  }

  int32_t ret = 0;
  ret = syncNodeStartPingTimer(pSyncNode);
  ASSERT(ret == 0);
  return ret;
}

void syncNodeStartOld(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1136
  // start raft
1137
  if (pSyncNode->replicaNum == 1) {
M
Minghao Li 已提交
1138
    raftStoreNextTerm(pSyncNode->pRaftStore);
1139
    syncNodeBecomeLeader(pSyncNode, "one replica start");
M
format  
Minghao Li 已提交
1140

1141
    // Raft 3.6.2 Committing entries from previous terms
1142 1143
    syncNodeAppendNoop(pSyncNode);
    syncMaybeAdvanceCommitIndex(pSyncNode);
M
Minghao Li 已提交
1144

M
Minghao Li 已提交
1145 1146
  } else {
    syncNodeBecomeFollower(pSyncNode, "first start");
1147 1148
  }

1149 1150 1151
  int32_t ret = 0;
  ret = syncNodeStartPingTimer(pSyncNode);
  ASSERT(ret == 0);
M
Minghao Li 已提交
1152 1153
}

B
Benguang Zhao 已提交
1154
int32_t syncNodeStartStandBy(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1155 1156 1157 1158 1159 1160 1161 1162
  // 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);
1163

1164 1165 1166
  ret = 0;
  ret = syncNodeStartPingTimer(pSyncNode);
  ASSERT(ret == 0);
B
Benguang Zhao 已提交
1167
  return ret;
M
Minghao Li 已提交
1168 1169
}

M
Minghao Li 已提交
1170 1171 1172 1173 1174 1175 1176 1177
void syncNodePreClose(SSyncNode* pSyncNode) {
  // stop elect timer
  syncNodeStopElectTimer(pSyncNode);

  // stop heartbeat timer
  syncNodeStopHeartbeatTimer(pSyncNode);
}

1178
void syncHbTimerDataFree(SSyncHbTimerData* pData) { taosMemoryFree(pData); }
M
Minghao Li 已提交
1179

M
Minghao Li 已提交
1180
void syncNodeClose(SSyncNode* pSyncNode) {
S
Shengliang Guan 已提交
1181 1182
  if (pSyncNode == NULL) return;
  sNTrace(pSyncNode, "sync close, data:%p", pSyncNode);
M
Minghao Li 已提交
1183

S
Shengliang Guan 已提交
1184
  int32_t ret = raftStoreClose(pSyncNode->pRaftStore);
M
Minghao Li 已提交
1185
  ASSERT(ret == 0);
M
Minghao Li 已提交
1186
  pSyncNode->pRaftStore = NULL;
M
Minghao Li 已提交
1187

B
Benguang Zhao 已提交
1188
  syncNodeLogReplMgrDestroy(pSyncNode);
M
Minghao Li 已提交
1189
  syncRespMgrDestroy(pSyncNode->pSyncRespMgr);
1190
  pSyncNode->pSyncRespMgr = NULL;
M
Minghao Li 已提交
1191
  voteGrantedDestroy(pSyncNode->pVotesGranted);
1192
  pSyncNode->pVotesGranted = NULL;
M
Minghao Li 已提交
1193
  votesRespondDestory(pSyncNode->pVotesRespond);
1194
  pSyncNode->pVotesRespond = NULL;
M
Minghao Li 已提交
1195
  syncIndexMgrDestroy(pSyncNode->pNextIndex);
1196
  pSyncNode->pNextIndex = NULL;
M
Minghao Li 已提交
1197
  syncIndexMgrDestroy(pSyncNode->pMatchIndex);
1198
  pSyncNode->pMatchIndex = NULL;
M
Minghao Li 已提交
1199
  logStoreDestory(pSyncNode->pLogStore);
1200
  pSyncNode->pLogStore = NULL;
B
Benguang Zhao 已提交
1201 1202
  syncLogBufferDestroy(pSyncNode->pLogBuf);
  pSyncNode->pLogBuf = NULL;
M
Minghao Li 已提交
1203
  raftCfgClose(pSyncNode->pRaftCfg);
1204
  pSyncNode->pRaftCfg = NULL;
M
Minghao Li 已提交
1205 1206 1207 1208 1209

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

M
Minghao Li 已提交
1210 1211 1212 1213
  if (pSyncNode->pFsm != NULL) {
    taosMemoryFree(pSyncNode->pFsm);
  }

S
Shengliang Guan 已提交
1214
  for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1215
    if ((pSyncNode->senders)[i] != NULL) {
S
Shengliang Guan 已提交
1216
      sSTrace((pSyncNode->senders)[i], "snapshot sender destroy while close, data:%p", (pSyncNode->senders)[i]);
M
Minghao Li 已提交
1217 1218 1219 1220 1221
      snapshotSenderDestroy((pSyncNode->senders)[i]);
      (pSyncNode->senders)[i] = NULL;
    }
  }

M
Minghao Li 已提交
1222 1223 1224 1225 1226
  if (pSyncNode->pNewNodeReceiver != NULL) {
    snapshotReceiverDestroy(pSyncNode->pNewNodeReceiver);
    pSyncNode->pNewNodeReceiver = NULL;
  }

1227
  taosMemoryFree(pSyncNode);
M
Minghao Li 已提交
1228 1229
}

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

M
Minghao Li 已提交
1232 1233 1234
// timer control --------------
int32_t syncNodeStartPingTimer(SSyncNode* pSyncNode) {
  int32_t ret = 0;
S
Shengliang Guan 已提交
1235 1236
  if (syncIsInit()) {
    taosTmrReset(pSyncNode->FpPingTimerCB, pSyncNode->pingTimerMS, pSyncNode, syncEnv()->pTimerManager,
1237 1238 1239
                 &pSyncNode->pPingTimer);
    atomic_store_64(&pSyncNode->pingTimerLogicClock, pSyncNode->pingTimerLogicClockUser);
  } else {
M
Minghao Li 已提交
1240
    sError("vgId:%d, start ping timer error, sync env is stop", pSyncNode->vgId);
1241
  }
M
Minghao Li 已提交
1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254
  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 已提交
1255
  if (syncIsInit()) {
1256
    pSyncNode->electTimerMS = ms;
S
Shengliang Guan 已提交
1257

1258 1259 1260 1261 1262
    int64_t execTime = taosGetTimestampMs() + ms;
    atomic_store_64(&(pSyncNode->electTimerParam.executeTime), execTime);
    atomic_store_64(&(pSyncNode->electTimerParam.logicClock), pSyncNode->electTimerLogicClock);
    pSyncNode->electTimerParam.pSyncNode = pSyncNode;
    pSyncNode->electTimerParam.pData = NULL;
S
Shengliang Guan 已提交
1263

1264
    taosTmrReset(pSyncNode->FpElectTimerCB, pSyncNode->electTimerMS, pSyncNode, syncEnv()->pTimerManager,
1265
                 &pSyncNode->pElectTimer);
1266

1267
  } else {
M
Minghao Li 已提交
1268
    sError("vgId:%d, start elect timer error, sync env is stop", pSyncNode->vgId);
1269
  }
M
Minghao Li 已提交
1270 1271 1272 1273 1274
  return ret;
}

int32_t syncNodeStopElectTimer(SSyncNode* pSyncNode) {
  int32_t ret = 0;
M
Minghao Li 已提交
1275
  atomic_add_fetch_64(&pSyncNode->electTimerLogicClock, 1);
M
Minghao Li 已提交
1276 1277
  taosTmrStop(pSyncNode->pElectTimer);
  pSyncNode->pElectTimer = NULL;
1278

M
Minghao Li 已提交
1279 1280 1281 1282 1283 1284 1285 1286 1287 1288
  return ret;
}

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

M
Minghao Li 已提交
1289 1290
int32_t syncNodeResetElectTimer(SSyncNode* pSyncNode) {
  int32_t ret = 0;
M
Minghao Li 已提交
1291 1292 1293 1294 1295 1296 1297
  int32_t electMS;

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

S
Shengliang Guan 已提交
1300 1301
  sNTrace(pSyncNode, "reset elect timer, min:%d, max:%d, ms:%d", pSyncNode->electBaseLine, 2 * pSyncNode->electBaseLine,
          electMS);
M
Minghao Li 已提交
1302 1303 1304
  return ret;
}

M
Minghao Li 已提交
1305
static int32_t syncNodeDoStartHeartbeatTimer(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1306
  int32_t ret = 0;
S
Shengliang Guan 已提交
1307 1308
  if (syncIsInit()) {
    taosTmrReset(pSyncNode->FpHeartbeatTimerCB, pSyncNode->heartbeatTimerMS, pSyncNode, syncEnv()->pTimerManager,
1309 1310 1311
                 &pSyncNode->pHeartbeatTimer);
    atomic_store_64(&pSyncNode->heartbeatTimerLogicClock, pSyncNode->heartbeatTimerLogicClockUser);
  } else {
M
Minghao Li 已提交
1312
    sError("vgId:%d, start heartbeat timer error, sync env is stop", pSyncNode->vgId);
1313
  }
1314

S
Shengliang Guan 已提交
1315
  sNTrace(pSyncNode, "start heartbeat timer, ms:%d", pSyncNode->heartbeatTimerMS);
M
Minghao Li 已提交
1316 1317 1318
  return ret;
}

M
Minghao Li 已提交
1319
int32_t syncNodeStartHeartbeatTimer(SSyncNode* pSyncNode) {
1320
  int32_t ret = 0;
M
Minghao Li 已提交
1321

1322
#if 0
M
Minghao Li 已提交
1323
  pSyncNode->heartbeatTimerMS = pSyncNode->hbBaseLine;
1324 1325
  ret = syncNodeDoStartHeartbeatTimer(pSyncNode);
#endif
1326

S
Shengliang Guan 已提交
1327
  for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
1328
    SSyncTimer* pSyncTimer = syncNodeGetHbTimer(pSyncNode, &(pSyncNode->peersId[i]));
M
Minghao Li 已提交
1329 1330 1331
    if (pSyncTimer != NULL) {
      syncHbTimerStart(pSyncNode, pSyncTimer);
    }
1332
  }
1333

M
Minghao Li 已提交
1334 1335 1336
  return ret;
}

M
Minghao Li 已提交
1337 1338
int32_t syncNodeStopHeartbeatTimer(SSyncNode* pSyncNode) {
  int32_t ret = 0;
1339 1340

#if 0
M
Minghao Li 已提交
1341 1342 1343
  atomic_add_fetch_64(&pSyncNode->heartbeatTimerLogicClockUser, 1);
  taosTmrStop(pSyncNode->pHeartbeatTimer);
  pSyncNode->pHeartbeatTimer = NULL;
1344
#endif
1345

S
Shengliang Guan 已提交
1346
  for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
1347
    SSyncTimer* pSyncTimer = syncNodeGetHbTimer(pSyncNode, &(pSyncNode->peersId[i]));
M
Minghao Li 已提交
1348 1349 1350
    if (pSyncTimer != NULL) {
      syncHbTimerStop(pSyncNode, pSyncTimer);
    }
1351
  }
1352

M
Minghao Li 已提交
1353 1354 1355
  return ret;
}

1356 1357 1358 1359 1360 1361
int32_t syncNodeRestartHeartbeatTimer(SSyncNode* pSyncNode) {
  syncNodeStopHeartbeatTimer(pSyncNode);
  syncNodeStartHeartbeatTimer(pSyncNode);
  return 0;
}

M
Minghao Li 已提交
1362 1363 1364
// utils --------------
int32_t syncNodeSendMsgById(const SRaftId* destRaftId, SSyncNode* pSyncNode, SRpcMsg* pMsg) {
  SEpSet epSet;
1365
  syncUtilRaftId2EpSet(destRaftId, &epSet);
S
Shengliang Guan 已提交
1366
  if (pSyncNode->syncSendMSg != NULL) {
M
Minghao Li 已提交
1367 1368 1369
    // htonl
    syncUtilMsgHtoN(pMsg->pCont);

1370
    pMsg->info.noResp = 1;
S
Shengliang Guan 已提交
1371
    pSyncNode->syncSendMSg(&epSet, pMsg);
M
Minghao Li 已提交
1372
  } else {
M
Minghao Li 已提交
1373
    sError("vgId:%d, sync send msg by id error, fp-send-msg is null", pSyncNode->vgId);
S
Shengliang Guan 已提交
1374
    rpcFreeCont(pMsg->pCont);
M
Minghao Li 已提交
1375
    return -1;
M
Minghao Li 已提交
1376
  }
M
Minghao Li 已提交
1377

M
Minghao Li 已提交
1378 1379 1380 1381 1382
  return 0;
}

int32_t syncNodeSendMsgByInfo(const SNodeInfo* nodeInfo, SSyncNode* pSyncNode, SRpcMsg* pMsg) {
  SEpSet epSet;
1383
  syncUtilNodeInfo2EpSet(nodeInfo, &epSet);
S
Shengliang Guan 已提交
1384
  if (pSyncNode->syncSendMSg != NULL) {
M
Minghao Li 已提交
1385 1386 1387
    // htonl
    syncUtilMsgHtoN(pMsg->pCont);

1388
    pMsg->info.noResp = 1;
S
Shengliang Guan 已提交
1389
    pSyncNode->syncSendMSg(&epSet, pMsg);
M
Minghao Li 已提交
1390
  } else {
M
Minghao Li 已提交
1391
    sError("vgId:%d, sync send msg by info error, fp-send-msg is null", pSyncNode->vgId);
M
Minghao Li 已提交
1392
  }
M
Minghao Li 已提交
1393 1394 1395
  return 0;
}

1396
inline bool syncNodeInConfig(SSyncNode* pSyncNode, const SSyncCfg* config) {
1397 1398 1399
  bool b1 = false;
  bool b2 = false;

S
Shengliang Guan 已提交
1400
  for (int32_t i = 0; i < config->replicaNum; ++i) {
1401 1402 1403 1404 1405 1406 1407
    if (strcmp((config->nodeInfo)[i].nodeFqdn, pSyncNode->myNodeInfo.nodeFqdn) == 0 &&
        (config->nodeInfo)[i].nodePort == pSyncNode->myNodeInfo.nodePort) {
      b1 = true;
      break;
    }
  }

S
Shengliang Guan 已提交
1408
  for (int32_t i = 0; i < config->replicaNum; ++i) {
1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422
    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;
}

1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435
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 已提交
1436
void syncNodeDoConfigChange(SSyncNode* pSyncNode, SSyncCfg* pNewConfig, SyncIndex lastConfigChangeIndex) {
1437
  SSyncCfg oldConfig = pSyncNode->pRaftCfg->cfg;
1438 1439 1440 1441
  if (!syncIsConfigChanged(&oldConfig, pNewConfig)) {
    sInfo("vgId:1, sync not reconfig since not changed");
    return;
  }
S
Shengliang Guan 已提交
1442

1443
  pSyncNode->pRaftCfg->cfg = *pNewConfig;
1444 1445
  pSyncNode->pRaftCfg->lastConfigIndex = lastConfigChangeIndex;

M
Minghao Li 已提交
1446 1447
  bool IamInOld = syncNodeInConfig(pSyncNode, &oldConfig);
  bool IamInNew = syncNodeInConfig(pSyncNode, pNewConfig);
M
Minghao Li 已提交
1448

M
Minghao Li 已提交
1449 1450
  bool isDrop = false;
  bool isAdd = false;
M
Minghao Li 已提交
1451

M
Minghao Li 已提交
1452 1453 1454 1455
  if (IamInOld && !IamInNew) {
    isDrop = true;
  } else {
    isDrop = false;
1456
  }
1457

M
Minghao Li 已提交
1458 1459 1460 1461 1462
  if (!IamInOld && IamInNew) {
    isAdd = true;
  } else {
    isAdd = false;
  }
M
Minghao Li 已提交
1463

M
Minghao Li 已提交
1464
  // log begin config change
S
Shengliang Guan 已提交
1465 1466 1467 1468 1469
  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 已提交
1470

M
Minghao Li 已提交
1471 1472
  if (IamInNew) {
    pSyncNode->pRaftCfg->isStandBy = 0;  // change isStandBy to normal
M
Minghao Li 已提交
1473
  }
M
Minghao Li 已提交
1474 1475
  if (isDrop) {
    pSyncNode->pRaftCfg->isStandBy = 1;  // set standby
M
Minghao Li 已提交
1476 1477
  }

M
Minghao Li 已提交
1478
  // add last config index
M
Minghao Li 已提交
1479
  raftCfgAddConfigIndex(pSyncNode->pRaftCfg, lastConfigChangeIndex);
M
Minghao Li 已提交
1480

M
Minghao Li 已提交
1481 1482 1483 1484 1485 1486 1487 1488 1489
  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 已提交
1490
    for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1491
      oldSenders[i] = (pSyncNode->senders)[i];
S
Shengliang Guan 已提交
1492
      sSTrace(oldSenders[i], "snapshot sender save old");
M
Minghao Li 已提交
1493
    }
1494

M
Minghao Li 已提交
1495 1496
    // init internal
    pSyncNode->myNodeInfo = pSyncNode->pRaftCfg->cfg.nodeInfo[pSyncNode->pRaftCfg->cfg.myIndex];
1497
    syncUtilNodeInfo2RaftId(&pSyncNode->myNodeInfo, pSyncNode->vgId, &pSyncNode->myRaftId);
M
Minghao Li 已提交
1498 1499 1500

    // init peersNum, peers, peersId
    pSyncNode->peersNum = pSyncNode->pRaftCfg->cfg.replicaNum - 1;
S
Shengliang Guan 已提交
1501 1502
    int32_t j = 0;
    for (int32_t i = 0; i < pSyncNode->pRaftCfg->cfg.replicaNum; ++i) {
M
Minghao Li 已提交
1503 1504 1505 1506 1507
      if (i != pSyncNode->pRaftCfg->cfg.myIndex) {
        pSyncNode->peersNodeInfo[j] = pSyncNode->pRaftCfg->cfg.nodeInfo[i];
        j++;
      }
    }
S
Shengliang Guan 已提交
1508
    for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
1509
      syncUtilNodeInfo2RaftId(&pSyncNode->peersNodeInfo[i], pSyncNode->vgId, &pSyncNode->peersId[i]);
M
Minghao Li 已提交
1510
    }
1511

M
Minghao Li 已提交
1512 1513
    // init replicaNum, replicasId
    pSyncNode->replicaNum = pSyncNode->pRaftCfg->cfg.replicaNum;
S
Shengliang Guan 已提交
1514
    for (int32_t i = 0; i < pSyncNode->pRaftCfg->cfg.replicaNum; ++i) {
1515
      syncUtilNodeInfo2RaftId(&pSyncNode->pRaftCfg->cfg.nodeInfo[i], pSyncNode->vgId, &pSyncNode->replicasId[i]);
M
Minghao Li 已提交
1516
    }
1517

1518 1519 1520
    // update quorum first
    pSyncNode->quorum = syncUtilQuorum(pSyncNode->pRaftCfg->cfg.replicaNum);

M
Minghao Li 已提交
1521 1522 1523 1524
    syncIndexMgrUpdate(pSyncNode->pNextIndex, pSyncNode);
    syncIndexMgrUpdate(pSyncNode->pMatchIndex, pSyncNode);
    voteGrantedUpdate(pSyncNode->pVotesGranted, pSyncNode);
    votesRespondUpdate(pSyncNode->pVotesRespond, pSyncNode);
M
Minghao Li 已提交
1525

M
Minghao Li 已提交
1526
    // reset snapshot senders
1527

M
Minghao Li 已提交
1528
    // clear new
S
Shengliang Guan 已提交
1529
    for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1530 1531
      (pSyncNode->senders)[i] = NULL;
    }
M
Minghao Li 已提交
1532

M
Minghao Li 已提交
1533
    // reset new
S
Shengliang Guan 已提交
1534
    for (int32_t i = 0; i < pSyncNode->replicaNum; ++i) {
M
Minghao Li 已提交
1535 1536
      // reset sender
      bool reset = false;
S
Shengliang Guan 已提交
1537
      for (int32_t j = 0; j < TSDB_MAX_REPLICA; ++j) {
M
Minghao Li 已提交
1538
        if (syncUtilSameId(&(pSyncNode->replicasId)[i], &oldReplicasId[j]) && oldSenders[j] != NULL) {
M
Minghao Li 已提交
1539 1540 1541
          char     host[128];
          uint16_t port;
          syncUtilU642Addr((pSyncNode->replicasId)[i].addr, host, sizeof(host), &port);
1542
          sNTrace(pSyncNode, "snapshot sender reset for: %" PRId64 ", newIndex:%d, %s:%d, %p",
S
Shengliang Guan 已提交
1543
                  (pSyncNode->replicasId)[i].addr, i, host, port, oldSenders[j]);
M
Minghao Li 已提交
1544 1545 1546 1547 1548 1549 1550 1551 1552

          (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 已提交
1553 1554
          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 已提交
1555 1556

          break;
M
Minghao Li 已提交
1557
        }
1558 1559
      }
    }
1560

M
Minghao Li 已提交
1561
    // create new
S
Shengliang Guan 已提交
1562
    for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1563 1564
      if ((pSyncNode->senders)[i] == NULL) {
        (pSyncNode->senders)[i] = snapshotSenderCreate(pSyncNode, i);
S
Shengliang Guan 已提交
1565 1566 1567
        sSTrace((pSyncNode->senders)[i], "snapshot sender create new while reconfig, data:%p", (pSyncNode->senders)[i]);
      } else {
        sSTrace((pSyncNode->senders)[i], "snapshot sender already exist, data:%p", (pSyncNode->senders)[i]);
M
Minghao Li 已提交
1568
      }
1569 1570
    }

M
Minghao Li 已提交
1571
    // free old
S
Shengliang Guan 已提交
1572
    for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1573
      if (oldSenders[i] != NULL) {
S
Shengliang Guan 已提交
1574
        sNTrace(pSyncNode, "snapshot sender destroy old, data:%p replica-index:%d", oldSenders[i], i);
M
Minghao Li 已提交
1575 1576 1577
        snapshotSenderDestroy(oldSenders[i]);
        oldSenders[i] = NULL;
      }
1578 1579
    }

1580
    // persist cfg
M
Minghao Li 已提交
1581
    raftCfgPersist(pSyncNode->pRaftCfg);
1582

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

M
Minghao Li 已提交
1587 1588 1589
    // change isStandBy to normal (election timeout)
    if (pSyncNode->state == TAOS_SYNC_STATE_LEADER) {
      syncNodeBecomeLeader(pSyncNode, tmpbuf);
1590 1591 1592 1593 1594

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

M
Minghao Li 已提交
1595 1596 1597 1598
    } else {
      syncNodeBecomeFollower(pSyncNode, tmpbuf);
    }
  } else {
1599
    // persist cfg
M
Minghao Li 已提交
1600
    raftCfgPersist(pSyncNode->pRaftCfg);
S
Shengliang Guan 已提交
1601 1602
    sNTrace(pSyncNode, "do not config change from %d to %d, index:%" PRId64 ", %s  -->  %s", oldConfig.replicaNum,
            pNewConfig->replicaNum, lastConfigChangeIndex, oldCfgStr, newCfgStr);
1603
  }
1604

M
Minghao Li 已提交
1605
_END:
M
Minghao Li 已提交
1606
  // log end config change
S
Shengliang Guan 已提交
1607
  sNTrace(pSyncNode, "end do config change, from %s to %s", oldCfgStr, newCfgStr);
M
Minghao Li 已提交
1608 1609
}

M
Minghao Li 已提交
1610 1611 1612 1613
// raft state change --------------
void syncNodeUpdateTerm(SSyncNode* pSyncNode, SyncTerm term) {
  if (term > pSyncNode->pRaftStore->currentTerm) {
    raftStoreSetTerm(pSyncNode->pRaftStore, term);
1614
    char tmpBuf[64];
1615
    snprintf(tmpBuf, sizeof(tmpBuf), "update term to %" PRId64, term);
1616
    syncNodeBecomeFollower(pSyncNode, tmpBuf);
M
Minghao Li 已提交
1617 1618 1619 1620
    raftStoreClearVote(pSyncNode->pRaftStore);
  }
}

1621 1622 1623 1624 1625 1626
void syncNodeUpdateTermWithoutStepDown(SSyncNode* pSyncNode, SyncTerm term) {
  if (term > pSyncNode->pRaftStore->currentTerm) {
    raftStoreSetTerm(pSyncNode->pRaftStore, term);
  }
}

M
Minghao Li 已提交
1627
void syncNodeStepDown(SSyncNode* pSyncNode, SyncTerm newTerm) {
M
Minghao Li 已提交
1628
  if (pSyncNode->pRaftStore->currentTerm > newTerm) {
1629
    sNTrace(pSyncNode, "step down, ignore, new-term:%" PRId64 ", current-term:%" PRId64, newTerm,
S
Shengliang Guan 已提交
1630
            pSyncNode->pRaftStore->currentTerm);
M
Minghao Li 已提交
1631 1632
    return;
  }
M
Minghao Li 已提交
1633 1634

  do {
1635
    sNTrace(pSyncNode, "step down, new-term:%" PRId64 ", current-term:%" PRId64, newTerm,
S
Shengliang Guan 已提交
1636
            pSyncNode->pRaftStore->currentTerm);
M
Minghao Li 已提交
1637 1638 1639 1640 1641
  } while (0);

  if (pSyncNode->pRaftStore->currentTerm < newTerm) {
    raftStoreSetTerm(pSyncNode->pRaftStore, newTerm);
    char tmpBuf[64];
1642
    snprintf(tmpBuf, sizeof(tmpBuf), "step down, update term to %" PRId64, newTerm);
M
Minghao Li 已提交
1643 1644 1645 1646 1647 1648 1649 1650 1651 1652
    syncNodeBecomeFollower(pSyncNode, tmpBuf);
    raftStoreClearVote(pSyncNode->pRaftStore);

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

1653 1654
void syncNodeLeaderChangeRsp(SSyncNode* pSyncNode) { syncRespCleanRsp(pSyncNode->pSyncRespMgr); }

1655
void syncNodeBecomeFollower(SSyncNode* pSyncNode, const char* debugStr) {
M
Minghao Li 已提交
1656
  // maybe clear leader cache
M
Minghao Li 已提交
1657 1658 1659 1660
  if (pSyncNode->state == TAOS_SYNC_STATE_LEADER) {
    pSyncNode->leaderCache = EMPTY_RAFT_ID;
  }

M
Minghao Li 已提交
1661
  // state change
M
Minghao Li 已提交
1662 1663 1664
  pSyncNode->state = TAOS_SYNC_STATE_FOLLOWER;
  syncNodeStopHeartbeatTimer(pSyncNode);

M
Minghao Li 已提交
1665 1666
  // reset elect timer
  syncNodeResetElectTimer(pSyncNode);
M
Minghao Li 已提交
1667

1668 1669 1670
  // send rsp to client
  syncNodeLeaderChangeRsp(pSyncNode);

1671 1672 1673 1674 1675
  // call back
  if (pSyncNode->pFsm != NULL && pSyncNode->pFsm->FpBecomeFollowerCb != NULL) {
    pSyncNode->pFsm->FpBecomeFollowerCb(pSyncNode->pFsm);
  }

M
Minghao Li 已提交
1676 1677 1678
  // min match index
  pSyncNode->minMatchIndex = SYNC_INDEX_INVALID;

B
Benguang Zhao 已提交
1679 1680 1681
  // reset log buffer
  syncLogBufferReset(pSyncNode->pLogBuf, pSyncNode);

M
Minghao Li 已提交
1682
  // trace log
S
Shengliang Guan 已提交
1683
  sNTrace(pSyncNode, "become follower %s", debugStr);
M
Minghao Li 已提交
1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703
}

// 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>>
//
1704
void syncNodeBecomeLeader(SSyncNode* pSyncNode, const char* debugStr) {
1705 1706
  pSyncNode->leaderTime = taosGetTimestampMs();

1707 1708 1709
  // reset restoreFinish
  pSyncNode->restoreFinish = false;

M
Minghao Li 已提交
1710
  // state change
M
Minghao Li 已提交
1711
  pSyncNode->state = TAOS_SYNC_STATE_LEADER;
M
Minghao Li 已提交
1712 1713

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

S
Shengliang Guan 已提交
1716
  for (int32_t i = 0; i < pSyncNode->pNextIndex->replicaNum; ++i) {
M
Minghao Li 已提交
1717 1718
    // maybe overwrite myself, no harm
    // just do it!
1719 1720 1721 1722 1723 1724 1725 1726 1727

    // 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 已提交
1728 1729
  }

S
Shengliang Guan 已提交
1730
  for (int32_t i = 0; i < pSyncNode->pMatchIndex->replicaNum; ++i) {
M
Minghao Li 已提交
1731 1732
    // maybe overwrite myself, no harm
    // just do it!
M
Minghao Li 已提交
1733 1734 1735
    pSyncNode->pMatchIndex->index[i] = SYNC_INDEX_INVALID;
  }

M
Minghao Li 已提交
1736 1737 1738
  // init peer mgr
  syncNodePeerStateInit(pSyncNode);

M
Minghao Li 已提交
1739
#if 0
1740 1741
  // update sender private term
  SSyncSnapshotSender* pMySender = syncNodeGetSnapshotSender(pSyncNode, &(pSyncNode->myRaftId));
1742
  if (pMySender != NULL) {
S
Shengliang Guan 已提交
1743
    for (int32_t i = 0; i < pSyncNode->pMatchIndex->replicaNum; ++i) {
1744 1745 1746
      if ((pSyncNode->senders)[i]->privateTerm > pMySender->privateTerm) {
        pMySender->privateTerm = (pSyncNode->senders)[i]->privateTerm;
      }
1747
    }
1748
    (pMySender->privateTerm) += 100;
1749
  }
M
Minghao Li 已提交
1750
#endif
1751

1752 1753 1754 1755 1756
  // close receiver
  if (snapshotReceiverIsStart(pSyncNode->pNewNodeReceiver)) {
    snapshotReceiverForceStop(pSyncNode->pNewNodeReceiver);
  }

M
Minghao Li 已提交
1757
  // stop elect timer
M
Minghao Li 已提交
1758
  syncNodeStopElectTimer(pSyncNode);
M
Minghao Li 已提交
1759

M
Minghao Li 已提交
1760 1761
  // start heartbeat timer
  syncNodeStartHeartbeatTimer(pSyncNode);
M
Minghao Li 已提交
1762

M
Minghao Li 已提交
1763 1764
  // send heartbeat right now
  syncNodeHeartbeatPeers(pSyncNode);
M
Minghao Li 已提交
1765

1766 1767 1768 1769 1770
  // call back
  if (pSyncNode->pFsm != NULL && pSyncNode->pFsm->FpBecomeLeaderCb != NULL) {
    pSyncNode->pFsm->FpBecomeLeaderCb(pSyncNode->pFsm);
  }

M
Minghao Li 已提交
1771 1772 1773
  // min match index
  pSyncNode->minMatchIndex = SYNC_INDEX_INVALID;

B
Benguang Zhao 已提交
1774 1775 1776
  // reset log buffer
  syncLogBufferReset(pSyncNode->pLogBuf, pSyncNode);

M
Minghao Li 已提交
1777
  // trace log
S
Shengliang Guan 已提交
1778
  sNTrace(pSyncNode, "become leader %s", debugStr);
M
Minghao Li 已提交
1779 1780 1781
}

void syncNodeCandidate2Leader(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1782 1783
  ASSERT(pSyncNode->state == TAOS_SYNC_STATE_CANDIDATE);
  ASSERT(voteGrantedMajority(pSyncNode->pVotesGranted));
1784
  syncNodeBecomeLeader(pSyncNode, "candidate to leader");
M
Minghao Li 已提交
1785

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

B
Benguang Zhao 已提交
1788
  int32_t ret = syncNodeAppendNoop(pSyncNode);
1789 1790 1791 1792
  if (ret < 0) {
    sError("vgId:%d, failed to append noop entry since %s", pSyncNode->vgId, terrstr());
  }

B
Benguang Zhao 已提交
1793 1794 1795 1796
  SyncIndex lastIndex = pSyncNode->pLogStore->syncLogLastIndex(pSyncNode->pLogStore);
  ASSERT(lastIndex >= 0);
  sInfo("vgId:%d, become leader. term: %" PRId64 ", commit index: %" PRId64 ", last index: %" PRId64 "",
        pSyncNode->vgId, pSyncNode->pRaftStore->currentTerm, pSyncNode->commitIndex, lastIndex);
B
Benguang Zhao 已提交
1797 1798 1799 1800 1801 1802 1803
}

void syncNodeCandidate2LeaderOld(SSyncNode* pSyncNode) {
  ASSERT(pSyncNode->state == TAOS_SYNC_STATE_CANDIDATE);
  ASSERT(voteGrantedMajority(pSyncNode->pVotesGranted));
  syncNodeBecomeLeader(pSyncNode, "candidate to leader");

M
Minghao Li 已提交
1804
  // Raft 3.6.2 Committing entries from previous terms
1805 1806
  syncNodeAppendNoop(pSyncNode);
  syncMaybeAdvanceCommitIndex(pSyncNode);
1807 1808

  if (pSyncNode->replicaNum > 1) {
M
Minghao Li 已提交
1809
    syncNodeReplicate(pSyncNode);
1810
  }
M
Minghao Li 已提交
1811 1812
}

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

M
Minghao Li 已提交
1815
int32_t syncNodePeerStateInit(SSyncNode* pSyncNode) {
S
Shengliang Guan 已提交
1816
  for (int32_t i = 0; i < TSDB_MAX_REPLICA; ++i) {
M
Minghao Li 已提交
1817 1818 1819 1820 1821
    pSyncNode->peerStates[i].lastSendIndex = SYNC_INDEX_INVALID;
    pSyncNode->peerStates[i].lastSendTime = 0;
  }

  return 0;
M
Minghao Li 已提交
1822 1823 1824
}

void syncNodeFollower2Candidate(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1825
  ASSERT(pSyncNode->state == TAOS_SYNC_STATE_FOLLOWER);
M
Minghao Li 已提交
1826
  pSyncNode->state = TAOS_SYNC_STATE_CANDIDATE;
B
Benguang Zhao 已提交
1827 1828 1829
  SyncIndex lastIndex = pSyncNode->pLogStore->syncLogLastIndex(pSyncNode->pLogStore);
  sInfo("vgId:%d, become candidate from follower. term: %" PRId64 ", commit index: %" PRId64 ", last index: %" PRId64,
        pSyncNode->vgId, pSyncNode->pRaftStore->currentTerm, pSyncNode->commitIndex, lastIndex);
M
Minghao Li 已提交
1830

S
Shengliang Guan 已提交
1831
  sNTrace(pSyncNode, "follower to candidate");
M
Minghao Li 已提交
1832 1833 1834
}

void syncNodeLeader2Follower(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1835
  ASSERT(pSyncNode->state == TAOS_SYNC_STATE_LEADER);
1836
  syncNodeBecomeFollower(pSyncNode, "leader to follower");
B
Benguang Zhao 已提交
1837 1838 1839 1840
  SyncIndex lastIndex = pSyncNode->pLogStore->syncLogLastIndex(pSyncNode->pLogStore);
  sInfo("vgId:%d, become follower from leader. term: %" PRId64 ", commit index: %" PRId64 ", last index: %" PRId64,
        pSyncNode->vgId, pSyncNode->pRaftStore->currentTerm, pSyncNode->commitIndex, lastIndex);

S
Shengliang Guan 已提交
1841
  sNTrace(pSyncNode, "leader to follower");
M
Minghao Li 已提交
1842 1843 1844
}

void syncNodeCandidate2Follower(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1845
  ASSERT(pSyncNode->state == TAOS_SYNC_STATE_CANDIDATE);
1846
  syncNodeBecomeFollower(pSyncNode, "candidate to follower");
B
Benguang Zhao 已提交
1847 1848 1849 1850
  SyncIndex lastIndex = pSyncNode->pLogStore->syncLogLastIndex(pSyncNode->pLogStore);
  sInfo("vgId:%d, become follower from candidate. term: %" PRId64 ", commit index: %" PRId64 ", last index: %" PRId64,
        pSyncNode->vgId, pSyncNode->pRaftStore->currentTerm, pSyncNode->commitIndex, lastIndex);

S
Shengliang Guan 已提交
1851
  sNTrace(pSyncNode, "candidate to follower");
M
Minghao Li 已提交
1852 1853
}

M
Minghao Li 已提交
1854 1855
// just called by syncNodeVoteForSelf
// need assert
M
Minghao Li 已提交
1856
void syncNodeVoteForTerm(SSyncNode* pSyncNode, SyncTerm term, SRaftId* pRaftId) {
M
Minghao Li 已提交
1857 1858
  ASSERT(term == pSyncNode->pRaftStore->currentTerm);
  ASSERT(!raftStoreHasVoted(pSyncNode->pRaftStore));
M
Minghao Li 已提交
1859 1860 1861 1862

  raftStoreVote(pSyncNode->pRaftStore, pRaftId);
}

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

S
Shengliang Guan 已提交
1867 1868
  SRpcMsg rpcMsg = {0};
  int32_t ret = syncBuildRequestVoteReply(&rpcMsg, pSyncNode->vgId);
S
Shengliang Guan 已提交
1869
  if (ret != 0) return;
M
Minghao Li 已提交
1870

S
Shengliang Guan 已提交
1871
  SyncRequestVoteReply* pMsg = rpcMsg.pCont;
M
Minghao Li 已提交
1872 1873 1874 1875 1876 1877 1878
  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 已提交
1879
  rpcFreeCont(rpcMsg.pCont);
M
Minghao Li 已提交
1880 1881
}

M
Minghao Li 已提交
1882
// return if has a snapshot
M
Minghao Li 已提交
1883 1884
bool syncNodeHasSnapshot(SSyncNode* pSyncNode) {
  bool      ret = false;
1885
  SSnapshot snapshot = {.data = NULL, .lastApplyIndex = -1, .lastApplyTerm = 0, .lastConfigIndex = -1};
1886 1887
  if (pSyncNode->pFsm->FpGetSnapshotInfo != NULL) {
    pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
M
Minghao Li 已提交
1888 1889 1890 1891 1892 1893 1894
    if (snapshot.lastApplyIndex >= SYNC_INDEX_BEGIN) {
      ret = true;
    }
  }
  return ret;
}

M
Minghao Li 已提交
1895 1896
// return max(logLastIndex, snapshotLastIndex)
// if no snapshot and log, return -1
1897
SyncIndex syncNodeGetLastIndex(const SSyncNode* pSyncNode) {
M
Minghao Li 已提交
1898
  SSnapshot snapshot = {.data = NULL, .lastApplyIndex = -1, .lastApplyTerm = 0, .lastConfigIndex = -1};
1899 1900
  if (pSyncNode->pFsm->FpGetSnapshotInfo != NULL) {
    pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
M
Minghao Li 已提交
1901 1902 1903 1904 1905 1906 1907
  }
  SyncIndex logLastIndex = pSyncNode->pLogStore->syncLogLastIndex(pSyncNode->pLogStore);

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

M
Minghao Li 已提交
1908 1909
// return the last term of snapshot and log
// if error, return SYNC_TERM_INVALID (by syncLogLastTerm)
M
Minghao Li 已提交
1910 1911
SyncTerm syncNodeGetLastTerm(SSyncNode* pSyncNode) {
  SyncTerm lastTerm = 0;
M
Minghao Li 已提交
1912 1913
  if (syncNodeHasSnapshot(pSyncNode)) {
    // has snapshot
1914
    SSnapshot snapshot = {.data = NULL, .lastApplyIndex = -1, .lastApplyTerm = 0, .lastConfigIndex = -1};
1915 1916
    if (pSyncNode->pFsm->FpGetSnapshotInfo != NULL) {
      pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
M
Minghao Li 已提交
1917 1918
    }

M
Minghao Li 已提交
1919 1920 1921
    SyncIndex logLastIndex = pSyncNode->pLogStore->syncLogLastIndex(pSyncNode->pLogStore);
    if (logLastIndex > snapshot.lastApplyIndex) {
      lastTerm = pSyncNode->pLogStore->syncLogLastTerm(pSyncNode->pLogStore);
M
Minghao Li 已提交
1922 1923 1924 1925
    } else {
      lastTerm = snapshot.lastApplyTerm;
    }

M
Minghao Li 已提交
1926
  } else {
M
Minghao Li 已提交
1927 1928
    // no snapshot
    lastTerm = pSyncNode->pLogStore->syncLogLastTerm(pSyncNode->pLogStore);
1929
  }
M
Minghao Li 已提交
1930

M
Minghao Li 已提交
1931 1932 1933 1934 1935 1936 1937
  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);
1938 1939
  return 0;
}
M
Minghao Li 已提交
1940

M
Minghao Li 已提交
1941
// return append-entries first try index
M
Minghao Li 已提交
1942 1943 1944 1945 1946
SyncIndex syncNodeSyncStartIndex(SSyncNode* pSyncNode) {
  SyncIndex syncStartIndex = syncNodeGetLastIndex(pSyncNode) + 1;
  return syncStartIndex;
}

M
Minghao Li 已提交
1947 1948
// if index > 0, return index - 1
// else, return -1
1949 1950 1951 1952 1953 1954 1955 1956 1957
SyncIndex syncNodeGetPreIndex(SSyncNode* pSyncNode, SyncIndex index) {
  SyncIndex preIndex = index - 1;
  if (preIndex < SYNC_INDEX_INVALID) {
    preIndex = SYNC_INDEX_INVALID;
  }

  return preIndex;
}

M
Minghao Li 已提交
1958 1959 1960 1961
// if index < 0, return SYNC_TERM_INVALID
// if index == 0, return 0
// if index > 0, return preTerm
// if error, return SYNC_TERM_INVALID
1962 1963 1964 1965 1966 1967 1968 1969 1970
SyncTerm syncNodeGetPreTerm(SSyncNode* pSyncNode, SyncIndex index) {
  if (index < SYNC_INDEX_BEGIN) {
    return SYNC_TERM_INVALID;
  }

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

1971 1972 1973
  SyncTerm  preTerm = 0;
  SyncIndex preIndex = index - 1;

1974
  SSyncRaftEntry* pPreEntry = NULL;
1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988
  SLRUCache*      pCache = pSyncNode->pLogStore->pCache;
  LRUHandle*      h = taosLRUCacheLookup(pCache, &preIndex, sizeof(preIndex));
  int32_t         code = 0;
  if (h) {
    pPreEntry = (SSyncRaftEntry*)taosLRUCacheValue(pCache, h);
    code = 0;

    sNTrace(pSyncNode, "hit cache index:%" PRId64 ", bytes:%u, %p", preIndex, pPreEntry->bytes, pPreEntry);

  } else {
    sNTrace(pSyncNode, "miss cache index:%" PRId64, preIndex);

    code = pSyncNode->pLogStore->syncLogGetEntry(pSyncNode->pLogStore, preIndex, &pPreEntry);
  }
M
Minghao Li 已提交
1989 1990 1991 1992 1993 1994

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

1995 1996 1997
  if (code == 0) {
    ASSERT(pPreEntry != NULL);
    preTerm = pPreEntry->term;
1998 1999 2000 2001

    if (h) {
      taosLRUCacheRelease(pCache, h, false);
    } else {
2002
      syncEntryDestroy(pPreEntry);
2003 2004
    }

2005 2006
    return preTerm;
  } else {
2007 2008 2009 2010
    if (pSyncNode->pFsm->FpGetSnapshotInfo != NULL) {
      pSyncNode->pFsm->FpGetSnapshotInfo(pSyncNode->pFsm, &snapshot);
      if (snapshot.lastApplyIndex == preIndex) {
        return snapshot.lastApplyTerm;
2011 2012 2013 2014
      }
    }
  }

2015
  sNError(pSyncNode, "sync node get pre term error, index:%" PRId64 ", snap-index:%" PRId64 ", snap-term:%" PRId64,
S
Shengliang Guan 已提交
2016
          index, snapshot.lastApplyIndex, snapshot.lastApplyTerm);
2017 2018
  return SYNC_TERM_INVALID;
}
M
Minghao Li 已提交
2019 2020 2021 2022

// 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 已提交
2023
  *pPreTerm = syncNodeGetPreTerm(pSyncNode, index);
M
Minghao Li 已提交
2024 2025 2026
  return 0;
}

M
Minghao Li 已提交
2027
static void syncNodeEqPingTimer(void* param, void* tmrId) {
S
Shengliang Guan 已提交
2028
  if (!syncIsInit()) return;
M
Minghao Li 已提交
2029

S
Shengliang Guan 已提交
2030 2031 2032
  SSyncNode* pNode = param;
  if (atomic_load_64(&pNode->pingTimerLogicClockUser) <= atomic_load_64(&pNode->pingTimerLogicClock)) {
    SRpcMsg rpcMsg = {0};
S
Shengliang Guan 已提交
2033
    int32_t code = syncBuildTimeout(&rpcMsg, SYNC_TIMEOUT_PING, atomic_load_64(&pNode->pingTimerLogicClock),
S
Shengliang Guan 已提交
2034 2035
                                    pNode->pingTimerMS, pNode);
    if (code != 0) {
M
Minghao Li 已提交
2036
      sError("failed to build ping msg");
S
Shengliang Guan 已提交
2037 2038
      rpcFreeCont(rpcMsg.pCont);
      return;
M
Minghao Li 已提交
2039
    }
M
Minghao Li 已提交
2040

2041
    sTrace("enqueue ping msg");
S
Shengliang Guan 已提交
2042 2043
    code = pNode->syncEqMsg(pNode->msgcb, &rpcMsg);
    if (code != 0) {
M
Minghao Li 已提交
2044
      sError("failed to sync enqueue ping msg since %s", terrstr());
S
Shengliang Guan 已提交
2045 2046
      rpcFreeCont(rpcMsg.pCont);
      return;
2047
    }
M
Minghao Li 已提交
2048

S
Shengliang Guan 已提交
2049
    taosTmrReset(syncNodeEqPingTimer, pNode->pingTimerMS, pNode, syncEnv()->pTimerManager, &pNode->pPingTimer);
2050
  }
M
Minghao Li 已提交
2051 2052
}

M
Minghao Li 已提交
2053
static void syncNodeEqElectTimer(void* param, void* tmrId) {
S
Shengliang Guan 已提交
2054
  if (!syncIsInit()) return;
M
Minghao Li 已提交
2055

2056
  SSyncNode* pNode = (SSyncNode*)param;
M
Minghao Li 已提交
2057

2058 2059
  if (pNode == NULL) return;
  if (pNode->syncEqMsg == NULL) return;
2060

2061 2062
  int64_t tsNow = taosGetTimestampMs();
  if (tsNow < pNode->electTimerParam.executeTime) return;
M
Minghao Li 已提交
2063

S
Shengliang Guan 已提交
2064
  SRpcMsg rpcMsg = {0};
2065 2066
  int32_t code =
      syncBuildTimeout(&rpcMsg, SYNC_TIMEOUT_ELECTION, pNode->electTimerParam.logicClock, pNode->electTimerMS, pNode);
S
Shengliang Guan 已提交
2067

S
Shengliang Guan 已提交
2068
  if (code != 0) {
M
Minghao Li 已提交
2069
    sError("failed to build elect msg");
2070

S
Shengliang Guan 已提交
2071
    return;
M
Minghao Li 已提交
2072 2073
  }

S
Shengliang Guan 已提交
2074
  SyncTimeout* pTimeout = rpcMsg.pCont;
S
Shengliang Guan 已提交
2075
  sNTrace(pNode, "enqueue elect msg lc:%" PRId64, pTimeout->logicClock);
S
Shengliang Guan 已提交
2076 2077 2078

  code = pNode->syncEqMsg(pNode->msgcb, &rpcMsg);
  if (code != 0) {
M
Minghao Li 已提交
2079
    sError("failed to sync enqueue elect msg since %s", terrstr());
S
Shengliang Guan 已提交
2080
    rpcFreeCont(rpcMsg.pCont);
S
Shengliang Guan 已提交
2081

2082
    return;
M
Minghao Li 已提交
2083 2084 2085
  }
}

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

S
Shengliang Guan 已提交
2089 2090 2091 2092
  SSyncNode* pNode = param;
  if (pNode->replicaNum > 1) {
    if (atomic_load_64(&pNode->heartbeatTimerLogicClockUser) <= atomic_load_64(&pNode->heartbeatTimerLogicClock)) {
      SRpcMsg rpcMsg = {0};
S
Shengliang Guan 已提交
2093
      int32_t code = syncBuildTimeout(&rpcMsg, SYNC_TIMEOUT_HEARTBEAT, atomic_load_64(&pNode->heartbeatTimerLogicClock),
S
Shengliang Guan 已提交
2094 2095 2096
                                      pNode->heartbeatTimerMS, pNode);

      if (code != 0) {
M
Minghao Li 已提交
2097
        sError("failed to build heartbeat msg");
S
Shengliang Guan 已提交
2098
        return;
2099
      }
M
Minghao Li 已提交
2100

2101
      sTrace("enqueue heartbeat timer");
S
Shengliang Guan 已提交
2102 2103
      code = pNode->syncEqMsg(pNode->msgcb, &rpcMsg);
      if (code != 0) {
M
Minghao Li 已提交
2104
        sError("failed to enqueue heartbeat msg since %s", terrstr());
S
Shengliang Guan 已提交
2105 2106
        rpcFreeCont(rpcMsg.pCont);
        return;
2107
      }
S
Shengliang Guan 已提交
2108 2109 2110 2111

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

2112
    } else {
S
Shengliang Guan 已提交
2113 2114
      sTrace("==syncNodeEqHeartbeatTimer== heartbeatTimerLogicClock:%" PRId64 ", heartbeatTimerLogicClockUser:%" PRId64,
             pNode->heartbeatTimerLogicClock, pNode->heartbeatTimerLogicClockUser);
2115
    }
M
Minghao Li 已提交
2116 2117 2118
  }
}

2119
static void syncNodeEqPeerHeartbeatTimer(void* param, void* tmrId) {
2120
  int64_t hbDataRid = (int64_t)param;
2121

2122 2123
  SSyncHbTimerData* pData = syncHbTimerDataAcquire(hbDataRid);
  if (pData == NULL) {
M
Minghao Li 已提交
2124
    sError("hb timer get pData NULL, %" PRId64, hbDataRid);
2125 2126
    return;
  }
2127

2128
  SSyncNode* pSyncNode = syncNodeAcquire(pData->syncNodeRid);
M
Minghao Li 已提交
2129
  if (pSyncNode == NULL) {
2130
    syncHbTimerDataRelease(pData);
M
Minghao Li 已提交
2131
    sError("hb timer get pSyncNode NULL");
2132 2133 2134 2135 2136 2137 2138 2139
    return;
  }

  SSyncTimer* pSyncTimer = pData->pTimer;

  if (!pSyncNode->isStart) {
    syncNodeRelease(pSyncNode);
    syncHbTimerDataRelease(pData);
M
Minghao Li 已提交
2140
    sError("vgId:%d, hb timer sync node already stop", pSyncNode->vgId);
M
Minghao Li 已提交
2141 2142 2143
    return;
  }

M
Minghao Li 已提交
2144
  if (pSyncNode->state != TAOS_SYNC_STATE_LEADER) {
2145 2146
    syncNodeRelease(pSyncNode);
    syncHbTimerDataRelease(pData);
M
Minghao Li 已提交
2147
    sError("vgId:%d, hb timer sync node not leader", pSyncNode->vgId);
M
Minghao Li 已提交
2148 2149 2150
    return;
  }

M
Minghao Li 已提交
2151
  if (pSyncNode->pRaftStore == NULL) {
2152 2153
    syncNodeRelease(pSyncNode);
    syncHbTimerDataRelease(pData);
M
Minghao Li 已提交
2154
    sError("vgId:%d, hb timer raft store already stop", pSyncNode->vgId);
M
Minghao Li 已提交
2155 2156 2157
    return;
  }

M
Minghao Li 已提交
2158
  // sTrace("vgId:%d, eq peer hb timer", pSyncNode->vgId);
2159 2160

  if (pSyncNode->replicaNum > 1) {
M
Minghao Li 已提交
2161 2162 2163
    int64_t timerLogicClock = atomic_load_64(&pSyncTimer->logicClock);
    int64_t msgLogicClock = atomic_load_64(&pData->logicClock);

2164
    if (timerLogicClock == msgLogicClock) {
M
Minghao Li 已提交
2165 2166 2167 2168 2169 2170 2171 2172
      if (syncIsInit()) {
        // sTrace("vgId:%d, reset peer hb timer", pSyncNode->vgId);
        taosTmrReset(syncNodeEqPeerHeartbeatTimer, pSyncTimer->timerMS, (void*)hbDataRid, syncEnv()->pTimerManager,
                     &pSyncTimer->pTimer);
      } else {
        sError("sync env is stop, reset peer hb timer error");
      }

2173
      SRpcMsg rpcMsg = {0};
S
Shengliang Guan 已提交
2174 2175 2176
      (void)syncBuildHeartbeat(&rpcMsg, pSyncNode->vgId);

      SyncHeartbeat* pSyncMsg = rpcMsg.pCont;
2177 2178 2179 2180
      pSyncMsg->srcId = pSyncNode->myRaftId;
      pSyncMsg->destId = pData->destId;
      pSyncMsg->term = pSyncNode->pRaftStore->currentTerm;
      pSyncMsg->commitIndex = pSyncNode->commitIndex;
M
Minghao Li 已提交
2181
      pSyncMsg->minMatchIndex = syncMinMatchIndex(pSyncNode);
2182
      pSyncMsg->privateTerm = 0;
2183
      pSyncMsg->timeStamp = taosGetTimestampMs();
2184 2185

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

    } else {
M
Minghao Li 已提交
2189 2190
      sTrace("vgId:%d, do not send hb, timerLogicClock:%" PRId64 ", msgLogicClock:%" PRId64 "", pSyncNode->vgId,
             timerLogicClock, msgLogicClock);
2191 2192
    }
  }
2193 2194 2195

  syncHbTimerDataRelease(pData);
  syncNodeRelease(pSyncNode);
2196 2197
}

2198 2199 2200 2201 2202
static int32_t syncNodeEqNoop(SSyncNode* pNode) {
  if (pNode->state == TAOS_SYNC_STATE_LEADER) {
    terrno = TSDB_CODE_SYN_NOT_LEADER;
    return -1;
  }
M
Minghao Li 已提交
2203

2204 2205 2206 2207
  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 已提交
2208

S
Shengliang Guan 已提交
2209
  SRpcMsg rpcMsg = {0};
S
Shengliang Guan 已提交
2210
  int32_t code = syncBuildClientRequestFromNoopEntry(&rpcMsg, pEntry, pNode->vgId);
2211
  syncEntryDestroy(pEntry);
M
Minghao Li 已提交
2212

2213 2214 2215
  sNTrace(pNode, "propose msg, type:noop");
  code = (*pNode->syncEqMsg)(pNode->msgcb, &rpcMsg);
  if (code != 0) {
M
Minghao Li 已提交
2216
    sError("failed to propose noop msg while enqueue since %s", terrstr());
2217
  }
M
Minghao Li 已提交
2218

2219
  return code;
M
Minghao Li 已提交
2220 2221
}

2222 2223
static void deleteCacheEntry(const void* key, size_t keyLen, void* value) { taosMemoryFree(value); }

2224 2225 2226 2227
int32_t syncCacheEntry(SSyncLogStore* pLogStore, SSyncRaftEntry* pEntry, LRUHandle** h) {
  SSyncLogStoreData* pData = pLogStore->data;
  sNTrace(pData->pSyncNode, "in cache index:%" PRId64 ", bytes:%u, %p", pEntry->index, pEntry->bytes, pEntry);

S
Shengliang Guan 已提交
2228 2229
  int32_t   code = 0;
  int32_t   entryLen = sizeof(*pEntry) + pEntry->dataLen;
2230 2231 2232 2233 2234 2235 2236 2237 2238
  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;
}

B
Benguang Zhao 已提交
2239 2240 2241 2242 2243 2244 2245 2246
int32_t syncNodeAppend(SSyncNode* ths, SSyncRaftEntry* pEntry) {
  // append to log buffer
  if (syncLogBufferAppend(ths->pLogBuf, ths, pEntry) < 0) {
    sError("vgId:%d, failed to enqueue log buffer. index:%" PRId64 "", ths->vgId, pEntry->index);
    return -1;
  }

  // proceed match index, with replicating on needed
2247
  SyncIndex matchIndex = syncLogBufferProceed(ths->pLogBuf, ths, NULL);
B
Benguang Zhao 已提交
2248

2249
  sTrace("vgId:%d, append raft entry. index: %" PRId64 ", term: %" PRId64 " pBuf: [%" PRId64 " %" PRId64 " %" PRId64
2250 2251 2252
         ", %" PRId64 ")",
         ths->vgId, pEntry->index, pEntry->term, ths->pLogBuf->startIndex, ths->pLogBuf->commitIndex,
         ths->pLogBuf->matchIndex, ths->pLogBuf->endIndex);
B
Benguang Zhao 已提交
2253

B
Benguang Zhao 已提交
2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269
  // multi replica
  if (ths->replicaNum > 1) {
    return 0;
  }

  // single replica
  (void)syncNodeUpdateCommitIndex(ths, matchIndex);

  if (syncLogBufferCommit(ths->pLogBuf, ths, ths->commitIndex) < 0) {
    sError("vgId:%d, failed to commit until commitIndex:%" PRId64 "", ths->vgId, ths->commitIndex);
    return -1;
  }

  return 0;
}

2270
bool syncNodeHeartbeatReplyTimeout(SSyncNode* pSyncNode) {
2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292
  if (pSyncNode->replicaNum == 1) {
    return false;
  }

  int32_t toCount = 0;
  int64_t tsNow = taosGetTimestampMs();
  for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
    int64_t recvTime = syncIndexMgrGetRecvTime(pSyncNode->pMatchIndex, &(pSyncNode->peersId[i]));
    if (recvTime == 0 || recvTime == -1) {
      continue;
    }

    if (tsNow - recvTime > SYNC_HEART_TIMEOUT_MS) {
      toCount++;
    }
  }

  bool b = (toCount >= pSyncNode->quorum ? true : false);

  return b;
}

M
Minghao Li 已提交
2293
static int32_t syncNodeAppendNoop(SSyncNode* ths) {
B
Benguang Zhao 已提交
2294 2295 2296 2297 2298 2299 2300 2301 2302
  SyncIndex index = syncLogBufferGetEndIndex(ths->pLogBuf);
  SyncTerm  term = ths->pRaftStore->currentTerm;

  SSyncRaftEntry* pEntry = syncEntryBuildNoop(term, index, ths->vgId);
  if (pEntry == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return -1;
  }

B
Benguang Zhao 已提交
2303 2304
  int32_t ret = syncNodeAppend(ths, pEntry);
  return 0;
B
Benguang Zhao 已提交
2305 2306 2307
}

static int32_t syncNodeAppendNoopOld(SSyncNode* ths) {
M
Minghao Li 已提交
2308 2309
  int32_t ret = 0;

2310
  SyncIndex       index = ths->pLogStore->syncLogWriteIndex(ths->pLogStore);
M
Minghao Li 已提交
2311
  SyncTerm        term = ths->pRaftStore->currentTerm;
M
Minghao Li 已提交
2312
  SSyncRaftEntry* pEntry = syncEntryBuildNoop(term, index, ths->vgId);
M
Minghao Li 已提交
2313
  ASSERT(pEntry != NULL);
M
Minghao Li 已提交
2314

2315 2316
  LRUHandle* h = NULL;

M
Minghao Li 已提交
2317
  if (ths->state == TAOS_SYNC_STATE_LEADER) {
2318
    int32_t code = ths->pLogStore->syncLogAppendEntry(ths->pLogStore, pEntry);
2319
    if (code != 0) {
M
Minghao Li 已提交
2320
      sError("append noop error");
2321 2322
      return -1;
    }
2323 2324

    syncCacheEntry(ths->pLogStore, pEntry, &h);
M
Minghao Li 已提交
2325 2326
  }

2327 2328 2329
  if (h) {
    taosLRUCacheRelease(ths->pLogStore->pCache, h, false);
  } else {
B
Benguang Zhao 已提交
2330
    syncEntryDestroy(pEntry);
2331 2332
  }

M
Minghao Li 已提交
2333 2334 2335
  return ret;
}

S
Shengliang Guan 已提交
2336 2337
int32_t syncNodeOnHeartbeat(SSyncNode* ths, const SRpcMsg* pRpcMsg) {
  SyncHeartbeat* pMsg = pRpcMsg->pCont;
2338 2339 2340 2341 2342

  int64_t tsMs = taosGetTimestampMs();
  char    buf[128];
  snprintf(buf, sizeof(buf), "recv local time:%" PRId64, tsMs);
  syncLogRecvHeartbeat(ths, pMsg, buf);
2343

2344 2345 2346 2347
  SRpcMsg rpcMsg = {0};
  (void)syncBuildHeartbeatReply(&rpcMsg, ths->vgId);

  SyncHeartbeatReply* pMsgReply = rpcMsg.pCont;
2348 2349 2350 2351
  pMsgReply->destId = pMsg->srcId;
  pMsgReply->srcId = ths->myRaftId;
  pMsgReply->term = ths->pRaftStore->currentTerm;
  pMsgReply->privateTerm = 8864;  // magic number
2352
  pMsgReply->startTime = ths->startTime;
2353
  pMsgReply->timeStamp = taosGetTimestampMs();
2354

M
Minghao Li 已提交
2355
  if (pMsg->term == ths->pRaftStore->currentTerm && ths->state != TAOS_SYNC_STATE_LEADER) {
2356 2357
    syncIndexMgrSetRecvTime(ths->pNextIndex, &(pMsg->srcId), tsMs);

2358
    syncNodeResetElectTimer(ths);
M
Minghao Li 已提交
2359
    ths->minMatchIndex = pMsg->minMatchIndex;
2360 2361

    if (ths->state == TAOS_SYNC_STATE_FOLLOWER) {
2362
      // syncNodeFollowerCommit(ths, pMsg->commitIndex);
S
Shengliang Guan 已提交
2363 2364 2365 2366
      SRpcMsg rpcMsgLocalCmd = {0};
      (void)syncBuildLocalCmd(&rpcMsgLocalCmd, ths->vgId);

      SyncLocalCmd* pSyncMsg = rpcMsgLocalCmd.pCont;
2367 2368
      pSyncMsg->cmd = SYNC_LOCAL_CMD_FOLLOWER_CMT;
      pSyncMsg->fcIndex = pMsg->commitIndex;
2369
      SyncIndex fcIndex = pSyncMsg->fcIndex;
2370 2371 2372 2373 2374 2375 2376

      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 {
2377
          sTrace("vgId:%d, sync enqueue fc-commit msg, fc-index:%" PRId64, ths->vgId, fcIndex);
2378 2379
        }
      }
2380 2381 2382
    }
  }

M
Minghao Li 已提交
2383
  if (pMsg->term >= ths->pRaftStore->currentTerm && ths->state != TAOS_SYNC_STATE_FOLLOWER) {
2384
    // syncNodeStepDown(ths, pMsg->term);
S
Shengliang Guan 已提交
2385 2386 2387 2388
    SRpcMsg rpcMsgLocalCmd = {0};
    (void)syncBuildLocalCmd(&rpcMsgLocalCmd, ths->vgId);

    SyncLocalCmd* pSyncMsg = rpcMsgLocalCmd.pCont;
2389 2390 2391
    pSyncMsg->cmd = SYNC_LOCAL_CMD_STEP_DOWN;
    pSyncMsg->sdNewTerm = pMsg->term;

S
Shengliang Guan 已提交
2392 2393
    if (ths->syncEqMsg != NULL && ths->msgcb != NULL) {
      int32_t code = ths->syncEqMsg(ths->msgcb, &rpcMsgLocalCmd);
2394 2395 2396 2397
      if (code != 0) {
        sError("vgId:%d, sync enqueue step-down msg error, code:%d", ths->vgId, code);
        rpcFreeCont(rpcMsgLocalCmd.pCont);
      } else {
2398
        sTrace("vgId:%d, sync enqueue step-down msg, new-term: %" PRId64, ths->vgId, pSyncMsg->sdNewTerm);
2399
      }
2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414
    }
  }

  /*
    // htonl
    SMsgHead* pHead = rpcMsg.pCont;
    pHead->contLen = htonl(pHead->contLen);
    pHead->vgId = htonl(pHead->vgId);
  */

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

2415
int32_t syncNodeOnHeartbeatReply(SSyncNode* ths, const SRpcMsg* pRpcMsg) {
2416
  SyncHeartbeatReply* pMsg = pRpcMsg->pCont;
B
Benguang Zhao 已提交
2417
  SSyncLogReplMgr*    pMgr = syncNodeGetLogReplMgr(ths, &pMsg->srcId);
2418 2419 2420 2421
  if (pMgr == NULL) {
    sError("vgId:%d, failed to get log repl mgr for the peer at addr 0x016%" PRIx64 "", ths->vgId, pMsg->srcId.addr);
    return -1;
  }
2422 2423 2424 2425

  int64_t tsMs = taosGetTimestampMs();
  syncIndexMgrSetRecvTime(ths->pMatchIndex, &pMsg->srcId, tsMs);

2426 2427 2428
  return syncLogReplMgrProcessHeartbeatReply(pMgr, ths, pMsg);
}

2429
int32_t syncNodeOnHeartbeatReplyOld(SSyncNode* ths, const SRpcMsg* pRpcMsg) {
2430
  SyncHeartbeatReply* pMsg = pRpcMsg->pCont;
2431 2432
  syncLogRecvHeartbeatReply(ths, pMsg, "");

M
Minghao Li 已提交
2433
  int64_t tsMs = taosGetTimestampMs();
2434 2435 2436
  char    buf[128];
  snprintf(buf, sizeof(buf), "recv local time:%" PRId64, tsMs);
  syncLogRecvHeartbeatReply(ths, pMsg, buf);
M
Minghao Li 已提交
2437

2438
  // update last reply time, make decision whether the other node is alive or not
M
Minghao Li 已提交
2439
  syncIndexMgrSetRecvTime(ths->pMatchIndex, &pMsg->srcId, tsMs);
2440 2441 2442
  return 0;
}

S
Shengliang Guan 已提交
2443 2444
int32_t syncNodeOnLocalCmd(SSyncNode* ths, const SRpcMsg* pRpcMsg) {
  SyncLocalCmd* pMsg = pRpcMsg->pCont;
2445 2446
  syncLogRecvLocalCmd(ths, pMsg, "");

2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466
  if (pMsg->cmd == SYNC_LOCAL_CMD_STEP_DOWN) {
    syncNodeStepDown(ths, pMsg->sdNewTerm);

  } else if (pMsg->cmd == SYNC_LOCAL_CMD_FOLLOWER_CMT) {
    (void)syncNodeUpdateCommitIndex(ths, pMsg->fcIndex);
    if (syncLogBufferCommit(ths->pLogBuf, ths, ths->commitIndex) < 0) {
      sError("vgId:%d, failed to commit raft log since %s. commit index: %" PRId64 "", ths->vgId, terrstr(),
             ths->commitIndex);
    }
  } else {
    sError("error local cmd");
  }

  return 0;
}

int32_t syncNodeOnLocalCmdOld(SSyncNode* ths, const SRpcMsg* pRpcMsg) {
  SyncLocalCmd* pMsg = pRpcMsg->pCont;
  syncLogRecvLocalCmd(ths, pMsg, "");

M
Minghao Li 已提交
2467 2468 2469
  if (pMsg->cmd == SYNC_LOCAL_CMD_STEP_DOWN) {
    syncNodeStepDown(ths, pMsg->sdNewTerm);

2470 2471 2472
  } else if (pMsg->cmd == SYNC_LOCAL_CMD_FOLLOWER_CMT) {
    syncNodeFollowerCommit(ths, pMsg->fcIndex);

M
Minghao Li 已提交
2473
  } else {
M
Minghao Li 已提交
2474
    sError("error local cmd");
M
Minghao Li 已提交
2475
  }
2476 2477 2478 2479

  return 0;
}

M
Minghao Li 已提交
2480 2481 2482 2483 2484 2485 2486 2487 2488 2489
// 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 已提交
2490

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

B
Benguang Zhao 已提交
2494 2495
  int32_t code = 0;

B
Benguang Zhao 已提交
2496 2497 2498
  SyncIndex       index = syncLogBufferGetEndIndex(ths->pLogBuf);
  SyncTerm        term = ths->pRaftStore->currentTerm;
  SSyncRaftEntry* pEntry = NULL;
2499 2500 2501 2502
  if (pMsg->msgType == TDMT_SYNC_CLIENT_REQUEST) {
    pEntry = syncEntryBuildFromClientRequest(pMsg->pCont, term, index);
  } else {
    pEntry = syncEntryBuildFromRpcMsg(pMsg, term, index);
B
Benguang Zhao 已提交
2503 2504 2505 2506 2507 2508 2509 2510 2511 2512
  }

  if (ths->state == TAOS_SYNC_STATE_LEADER) {
    if (pRetIndex) {
      (*pRetIndex) = index;
    }

    return syncNodeAppend(ths, pEntry);
  }

B
Benguang Zhao 已提交
2513
  return -1;
B
Benguang Zhao 已提交
2514 2515
}

2516 2517
int32_t syncNodeOnClientRequestOld(SSyncNode* ths, SRpcMsg* pMsg, SyncIndex* pRetIndex) {
  sNTrace(ths, "on client request");
B
Benguang Zhao 已提交
2518

M
Minghao Li 已提交
2519
  int32_t ret = 0;
2520
  int32_t code = 0;
M
Minghao Li 已提交
2521

M
Minghao Li 已提交
2522
  SyncIndex       index = ths->pLogStore->syncLogWriteIndex(ths->pLogStore);
M
Minghao Li 已提交
2523
  SyncTerm        term = ths->pRaftStore->currentTerm;
2524 2525 2526 2527 2528 2529 2530
  SSyncRaftEntry* pEntry;

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

2532 2533
  LRUHandle* h = NULL;

M
Minghao Li 已提交
2534
  if (ths->state == TAOS_SYNC_STATE_LEADER) {
2535 2536 2537
    // append entry
    code = ths->pLogStore->syncLogAppendEntry(ths->pLogStore, pEntry);
    if (code != 0) {
2538 2539 2540 2541
      if (ths->replicaNum == 1) {
        if (h) {
          taosLRUCacheRelease(ths->pLogStore->pCache, h, false);
        } else {
2542
          syncEntryDestroy(pEntry);
2543
        }
2544

2545 2546 2547 2548
        return -1;

      } else {
        // del resp mgr, call FpCommitCb
2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559
        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,
        };
2560
        ths->pFsm->FpCommitCb(ths->pFsm, pMsg, &cbMeta);
2561 2562 2563 2564

        if (h) {
          taosLRUCacheRelease(ths->pLogStore->pCache, h, false);
        } else {
2565
          syncEntryDestroy(pEntry);
2566 2567
        }

2568 2569
        return -1;
      }
2570
    }
M
Minghao Li 已提交
2571

2572 2573
    syncCacheEntry(ths->pLogStore, pEntry, &h);

2574 2575
    // if mulit replica, start replicate right now
    if (ths->replicaNum > 1) {
M
Minghao Li 已提交
2576
      syncNodeReplicate(ths);
2577
    }
2578

2579 2580
    // if only myself, maybe commit right now
    if (ths->replicaNum == 1) {
2581 2582 2583 2584 2585
      if (syncNodeIsMnode(ths)) {
        syncMaybeAdvanceCommitIndex(ths);
      } else {
        syncOneReplicaAdvance(ths);
      }
2586
    }
M
Minghao Li 已提交
2587 2588
  }

2589 2590 2591 2592 2593 2594 2595 2596
  if (pRetIndex != NULL) {
    if (ret == 0 && pEntry != NULL) {
      *pRetIndex = pEntry->index;
    } else {
      *pRetIndex = SYNC_INDEX_INVALID;
    }
  }

2597 2598 2599
  if (h) {
    taosLRUCacheRelease(ths->pLogStore->pCache, h, false);
  } else {
B
Benguang Zhao 已提交
2600
    syncEntryDestroy(pEntry);
2601 2602
  }

M
Minghao Li 已提交
2603
  return ret;
2604
}
M
Minghao Li 已提交
2605

S
Shengliang Guan 已提交
2606 2607 2608
const char* syncStr(ESyncState state) {
  switch (state) {
    case TAOS_SYNC_STATE_FOLLOWER:
2609
      return "follower";
S
Shengliang Guan 已提交
2610
    case TAOS_SYNC_STATE_CANDIDATE:
2611
      return "candidate";
S
Shengliang Guan 已提交
2612
    case TAOS_SYNC_STATE_LEADER:
2613
      return "leader";
S
Shengliang Guan 已提交
2614
    default:
2615
      return "error";
S
Shengliang Guan 已提交
2616
  }
M
Minghao Li 已提交
2617
}
2618

2619
#if 0
2620
int32_t syncDoLeaderTransfer(SSyncNode* ths, SRpcMsg* pRpcMsg, SSyncRaftEntry* pEntry) {
2621
  if (ths->state != TAOS_SYNC_STATE_FOLLOWER) {
S
Shengliang Guan 已提交
2622
    sNTrace(ths, "I am not follower, can not do leader transfer");
2623 2624
    return 0;
  }
2625 2626

  if (!ths->restoreFinish) {
S
Shengliang Guan 已提交
2627
    sNTrace(ths, "restore not finish, can not do leader transfer");
2628 2629 2630
    return 0;
  }

2631
  if (pEntry->term < ths->pRaftStore->currentTerm) {
2632
    sNTrace(ths, "little term:%" PRId64 ", can not do leader transfer", pEntry->term);
2633 2634 2635 2636
    return 0;
  }

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

2641 2642
  /*
    if (ths->vgId > 1) {
S
Shengliang Guan 已提交
2643
      sNTrace(ths, "I am vnode, can not do leader transfer");
2644 2645 2646 2647
      return 0;
    }
  */

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

M
Minghao Li 已提交
2651 2652 2653
  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 已提交
2654

M
Minghao Li 已提交
2655 2656
  bool same = sameId || sameNodeInfo;
  if (same) {
M
Minghao Li 已提交
2657 2658 2659 2660
    // reset elect timer now!
    int32_t electMS = 1;
    int32_t ret = syncNodeRestartElectTimer(ths, electMS);
    ASSERT(ret == 0);
M
Minghao Li 已提交
2661

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

M
Minghao Li 已提交
2666
  if (ths->pFsm->FpLeaderTransferCb != NULL) {
S
Shengliang Guan 已提交
2667
    SFsmCbMeta cbMeta = {
S
Shengliang Guan 已提交
2668 2669 2670 2671 2672 2673 2674 2675 2676
        .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 已提交
2677 2678
    };
    ths->pFsm->FpLeaderTransferCb(ths->pFsm, pRpcMsg, &cbMeta);
2679 2680
  }

2681 2682 2683
  return 0;
}

2684 2685
#endif

2686
int32_t syncNodeUpdateNewConfigIndex(SSyncNode* ths, SSyncCfg* pNewCfg) {
S
Shengliang Guan 已提交
2687
  for (int32_t i = 0; i < pNewCfg->replicaNum; ++i) {
2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700
    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;
}

2701 2702 2703 2704
bool syncNodeIsOptimizedOneReplica(SSyncNode* ths, SRpcMsg* pMsg) {
  return (ths->replicaNum == 1 && syncUtilUserCommit(pMsg->msgType) && ths->vgId != 1);
}

M
Minghao Li 已提交
2705
int32_t syncNodeDoCommit(SSyncNode* ths, SyncIndex beginIndex, SyncIndex endIndex, uint64_t flag) {
2706
  ASSERT(false);
2707 2708 2709 2710
  if (beginIndex > endIndex) {
    return 0;
  }

M
Minghao Li 已提交
2711 2712 2713 2714 2715 2716 2717 2718 2719
  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 已提交
2720
      sNTrace(ths, "commit by snapshot from index:%" PRId64 " to index:%" PRId64, beginIndex, snapshot.lastApplyIndex);
2721

M
Minghao Li 已提交
2722 2723 2724
      // update begin index
      beginIndex = snapshot.lastApplyIndex + 1;
    }
2725 2726
  }

2727 2728
  int32_t    code = 0;
  ESyncState state = flag;
M
Minghao Li 已提交
2729

S
Shengliang Guan 已提交
2730
  sNTrace(ths, "commit by wal from index:%" PRId64 " to index:%" PRId64, beginIndex, endIndex);
2731 2732 2733 2734 2735 2736

  // execute fsm
  if (ths->pFsm != NULL) {
    for (SyncIndex i = beginIndex; i <= endIndex; ++i) {
      if (i != SYNC_INDEX_INVALID) {
        SSyncRaftEntry* pEntry;
2737 2738 2739 2740
        SLRUCache*      pCache = ths->pLogStore->pCache;
        LRUHandle*      h = taosLRUCacheLookup(pCache, &i, sizeof(i));
        if (h) {
          pEntry = (SSyncRaftEntry*)taosLRUCacheValue(pCache, h);
2741 2742 2743

          sNTrace(ths, "hit cache index:%" PRId64 ", bytes:%u, %p", i, pEntry->bytes, pEntry);

2744
        } else {
2745 2746
          sNTrace(ths, "miss cache index:%" PRId64, i);

2747
          code = ths->pLogStore->syncLogGetEntry(ths->pLogStore, i, &pEntry);
M
Minghao Li 已提交
2748 2749 2750
          // ASSERT(code == 0);
          // ASSERT(pEntry != NULL);
          if (code != 0 || pEntry == NULL) {
S
Shengliang Guan 已提交
2751
            sNError(ths, "get log entry error");
2752
            sFatal("vgId:%d, get log entry %" PRId64 " error when commit since %s", ths->vgId, i, terrstr());
M
Minghao Li 已提交
2753 2754
            continue;
          }
2755
        }
2756

2757
        SRpcMsg rpcMsg = {0};
2758 2759
        syncEntry2OriginalRpc(pEntry, &rpcMsg);

2760
        sTrace("do commit index:%" PRId64 ", type:%s", i, TMSG_INFO(pEntry->msgType));
M
Minghao Li 已提交
2761

2762
        // user commit
2763 2764
        if ((ths->pFsm->FpCommitCb != NULL) && syncUtilUserCommit(pEntry->originalRpcType)) {
          bool internalExecute = true;
S
Shengliang Guan 已提交
2765
          if ((ths->replicaNum == 1) && ths->restoreFinish && ths->vgId != 1) {
2766 2767 2768
            internalExecute = false;
          }

M
Minghao Li 已提交
2769 2770
          sNTrace(ths, "user commit index:%" PRId64 ", internal:%d, type:%s", i, internalExecute,
                  TMSG_INFO(pEntry->msgType));
2771

2772 2773
          // execute fsm in apply thread, or execute outside syncPropose
          if (internalExecute) {
S
Shengliang Guan 已提交
2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785
            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 已提交
2786
            syncRespMgrGetAndDel(ths->pSyncRespMgr, cbMeta.seqNum, &rpcMsg.info);
S
Shengliang Guan 已提交
2787
            ths->pFsm->FpCommitCb(ths->pFsm, &rpcMsg, &cbMeta);
M
Minghao Li 已提交
2788
          }
2789
        }
2790

2791 2792
#if 0
        // execute in pre-commit
M
Minghao Li 已提交
2793
        // leader transfer
2794 2795 2796
        if (pEntry->originalRpcType == TDMT_SYNC_LEADER_TRANSFER) {
          code = syncDoLeaderTransfer(ths, &rpcMsg, pEntry);
          ASSERT(code == 0);
2797
        }
2798
#endif
2799 2800

        // restore finish
2801
        // if only snapshot, a noop entry will be append, so syncLogLastIndex is always ok
2802 2803 2804 2805 2806 2807
        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 已提交
2808

2809
            int64_t restoreDelay = taosGetTimestampMs() - ths->leaderTime;
S
Shengliang Guan 已提交
2810
            sNTrace(ths, "restore finish, index:%" PRId64 ", elapsed:%" PRId64 " ms", pEntry->index, restoreDelay);
2811 2812 2813 2814
          }
        }

        rpcFreeCont(rpcMsg.pCont);
2815 2816 2817
        if (h) {
          taosLRUCacheRelease(pCache, h, false);
        } else {
B
Benguang Zhao 已提交
2818
          syncEntryDestroy(pEntry);
2819
        }
2820 2821 2822 2823
      }
    }
  }
  return 0;
2824 2825 2826
}

bool syncNodeInRaftGroup(SSyncNode* ths, SRaftId* pRaftId) {
S
Shengliang Guan 已提交
2827
  for (int32_t i = 0; i < ths->replicaNum; ++i) {
2828 2829 2830 2831 2832
    if (syncUtilSameId(&((ths->replicasId)[i]), pRaftId)) {
      return true;
    }
  }
  return false;
M
Minghao Li 已提交
2833 2834 2835 2836
}

SSyncSnapshotSender* syncNodeGetSnapshotSender(SSyncNode* ths, SRaftId* pDestId) {
  SSyncSnapshotSender* pSender = NULL;
S
Shengliang Guan 已提交
2837
  for (int32_t i = 0; i < ths->replicaNum; ++i) {
M
Minghao Li 已提交
2838 2839 2840 2841 2842
    if (syncUtilSameId(pDestId, &((ths->replicasId)[i]))) {
      pSender = (ths->senders)[i];
    }
  }
  return pSender;
M
Minghao Li 已提交
2843
}
M
Minghao Li 已提交
2844

2845 2846
SSyncTimer* syncNodeGetHbTimer(SSyncNode* ths, SRaftId* pDestId) {
  SSyncTimer* pTimer = NULL;
S
Shengliang Guan 已提交
2847
  for (int32_t i = 0; i < ths->replicaNum; ++i) {
2848 2849 2850 2851 2852 2853 2854
    if (syncUtilSameId(pDestId, &((ths->replicasId)[i]))) {
      pTimer = &((ths->peerHeartbeatTimerArr)[i]);
    }
  }
  return pTimer;
}

M
Minghao Li 已提交
2855 2856
SPeerState* syncNodeGetPeerState(SSyncNode* ths, const SRaftId* pDestId) {
  SPeerState* pState = NULL;
S
Shengliang Guan 已提交
2857
  for (int32_t i = 0; i < ths->replicaNum; ++i) {
M
Minghao Li 已提交
2858 2859 2860 2861 2862 2863 2864 2865 2866
    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 已提交
2867
  if (pState == NULL) {
2868
    sError("vgId:%d, replica maybe dropped", ths->vgId);
M
Minghao Li 已提交
2869 2870
    return false;
  }
M
Minghao Li 已提交
2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881

  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 已提交
2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894 2895
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 已提交
2896
  for (int32_t i = 0; i < pSyncNode->peersNum; ++i) {
M
Minghao Li 已提交
2897
    SSyncSnapshotSender* pSender = syncNodeGetSnapshotSender(pSyncNode, &(pSyncNode->peersId)[i]);
M
Minghao Li 已提交
2898
    if (pSender != NULL && pSender->start) {
M
Minghao Li 已提交
2899 2900 2901 2902 2903 2904
      sError("sync cannot change3");
      return false;
    }
  }

  return true;
M
Minghao Li 已提交
2905
}