syncElection.c 2.3 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/>.
 */

M
Minghao Li 已提交
16
#include "syncElection.h"
M
Minghao Li 已提交
17
#include "syncMessage.h"
M
Minghao Li 已提交
18
#include "syncRaftStore.h"
M
Minghao Li 已提交
19

M
Minghao Li 已提交
20 21 22 23 24 25 26 27 28 29 30
// TLA+ Spec
// RequestVote(i, j) ==
//    /\ state[i] = Candidate
//    /\ j \notin votesResponded[i]
//    /\ Send([mtype         |-> RequestVoteRequest,
//             mterm         |-> currentTerm[i],
//             mlastLogTerm  |-> LastTerm(log[i]),
//             mlastLogIndex |-> Len(log[i]),
//             msource       |-> i,
//             mdest         |-> j])
//    /\ UNCHANGED <<serverVars, candidateVars, leaderVars, logVars>>
M
Minghao Li 已提交
31
//
M
Minghao Li 已提交
32
int32_t syncNodeRequestVotePeers(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
33 34
  assert(pSyncNode->state == TAOS_SYNC_STATE_CANDIDATE);

M
Minghao Li 已提交
35
  int32_t ret = 0;
M
Minghao Li 已提交
36 37 38 39 40 41 42 43 44 45 46 47
  for (int i = 0; i < pSyncNode->peersNum; ++i) {
    SyncRequestVote* pMsg = syncRequestVoteBuild();
    pMsg->srcId = pSyncNode->myRaftId;
    pMsg->destId = pSyncNode->peersId[i];
    pMsg->currentTerm = pSyncNode->pRaftStore->currentTerm;
    pMsg->lastLogIndex = pSyncNode->pLogStore->getLastIndex(pSyncNode->pLogStore);
    pMsg->lastLogTerm = pSyncNode->pLogStore->getLastTerm(pSyncNode->pLogStore);

    ret = syncNodeRequestVote(pSyncNode, &pSyncNode->peersId[i], pMsg);
    assert(ret == 0);
    syncRequestVoteDestroy(pMsg);
  }
M
Minghao Li 已提交
48 49
  return ret;
}
M
Minghao Li 已提交
50

M
Minghao Li 已提交
51
int32_t syncNodeElect(SSyncNode* pSyncNode) {
M
Minghao Li 已提交
52 53
  assert(pSyncNode->state == TAOS_SYNC_STATE_CANDIDATE);

M
Minghao Li 已提交
54
  // start election
M
Minghao Li 已提交
55 56
  int32_t ret = syncNodeRequestVotePeers(pSyncNode);
  return ret;
M
Minghao Li 已提交
57 58
}

M
Minghao Li 已提交
59 60 61 62 63 64 65 66 67
int32_t syncNodeRequestVote(SSyncNode* pSyncNode, const SRaftId* destRaftId, const SyncRequestVote* pMsg) {
  sTrace("syncNodeRequestVote pSyncNode:%p ", pSyncNode);
  int32_t ret = 0;

  SRpcMsg rpcMsg;
  syncRequestVote2RpcMsg(pMsg, &rpcMsg);
  syncNodeSendMsgById(destRaftId, pSyncNode, &rpcMsg);
  return ret;
}