syncRequestVoteReply.c 3.3 KB
Newer Older
M
Minghao Li 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
/*
 * 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/>.
 */

#include "syncRequestVoteReply.h"
M
Minghao Li 已提交
17 18 19 20
#include "syncInt.h"
#include "syncRaftStore.h"
#include "syncUtil.h"
#include "syncVoteMgr.h"
M
Minghao Li 已提交
21

M
Minghao Li 已提交
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37
// TLA+ Spec
// HandleRequestVoteResponse(i, j, m) ==
//    \* This tallies votes even when the current state is not Candidate, but
//    \* they won't be looked at, so it doesn't matter.
//    /\ m.mterm = currentTerm[i]
//    /\ votesResponded' = [votesResponded EXCEPT ![i] =
//                              votesResponded[i] \cup {j}]
//    /\ \/ /\ m.mvoteGranted
//          /\ votesGranted' = [votesGranted EXCEPT ![i] =
//                                  votesGranted[i] \cup {j}]
//          /\ voterLog' = [voterLog EXCEPT ![i] =
//                              voterLog[i] @@ (j :> m.mlog)]
//       \/ /\ ~m.mvoteGranted
//          /\ UNCHANGED <<votesGranted, voterLog>>
//    /\ Discard(m)
//    /\ UNCHANGED <<serverVars, votedFor, leaderVars, logVars>>
M
Minghao Li 已提交
38
//
M
Minghao Li 已提交
39 40
int32_t syncNodeOnRequestVoteReplyCb(SSyncNode* ths, SyncRequestVoteReply* pMsg) {
  int32_t ret = 0;
M
Minghao Li 已提交
41

42
  char logBuf[128] = {0};
M
Minghao Li 已提交
43 44
  snprintf(logBuf, sizeof(logBuf), "==syncNodeOnRequestVoteReplyCb== term:%lu", ths->pRaftStore->currentTerm);
  syncRequestVoteReplyLog2(logBuf, pMsg);
M
Minghao Li 已提交
45 46

  if (pMsg->term < ths->pRaftStore->currentTerm) {
M
Minghao Li 已提交
47 48
    sTrace("DropStaleResponse, receive term:%" PRIu64 ", current term:%" PRIu64 "", pMsg->term,
           ths->pRaftStore->currentTerm);
M
Minghao Li 已提交
49 50 51
    return ret;
  }

M
Minghao Li 已提交
52 53 54 55 56 57 58
  // assert(!(pMsg->term > ths->pRaftStore->currentTerm));
  //  no need this code, because if I receive reply.term, then I must have sent for that term.
  //   if (pMsg->term > ths->pRaftStore->currentTerm) {
  //     syncNodeUpdateTerm(ths, pMsg->term);
  //   }

  if (pMsg->term > ths->pRaftStore->currentTerm) {
59
    char logBuf[128] = {0};
M
Minghao Li 已提交
60 61 62 63 64 65
    snprintf(logBuf, sizeof(logBuf), "syncNodeOnRequestVoteReplyCb error term, receive:%lu current:%lu", pMsg->term,
             ths->pRaftStore->currentTerm);
    syncNodePrint2(logBuf, ths);
    sError("%s", logBuf);
    return ret;
  }
M
Minghao Li 已提交
66 67 68

  assert(pMsg->term == ths->pRaftStore->currentTerm);

M
Minghao Li 已提交
69 70
  // This tallies votes even when the current state is not Candidate,
  // but they won't be looked at, so it doesn't matter.
M
Minghao Li 已提交
71 72 73
  if (ths->state == TAOS_SYNC_STATE_CANDIDATE) {
    votesRespondAdd(ths->pVotesRespond, pMsg);
    if (pMsg->voteGranted) {
M
Minghao Li 已提交
74
      // add vote
M
Minghao Li 已提交
75
      voteGrantedVote(ths->pVotesGranted, pMsg);
M
Minghao Li 已提交
76 77

      // maybe to leader
M
Minghao Li 已提交
78
      if (voteGrantedMajority(ths->pVotesGranted)) {
M
Minghao Li 已提交
79
        if (!ths->pVotesGranted->toLeader) {
M
Minghao Li 已提交
80
          syncNodeCandidate2Leader(ths);
M
Minghao Li 已提交
81 82

          // prevent to leader again!
M
Minghao Li 已提交
83 84 85
          ths->pVotesGranted->toLeader = true;
        }
      }
M
Minghao Li 已提交
86 87 88 89
    } else {
      ;
      // do nothing
      // UNCHANGED <<votesGranted, voterLog>>
M
Minghao Li 已提交
90 91
    }
  }
M
Minghao Li 已提交
92

M
Minghao Li 已提交
93 94
  return ret;
}