sync_raft_quorum_joint.c 2.6 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * Copyright (c) 2019 TAOS Data, Inc. <cli@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/>.
 */

L
lichuang 已提交
16
#include "sync_raft_node_map.h"
17 18 19 20 21 22 23 24 25
#include "sync_raft_quorum_majority.h"
#include "sync_raft_quorum_joint.h"
#include "sync_raft_quorum.h"

/**
 * syncRaftVoteResult takes a mapping of voters to yes/no (true/false) votes and returns
 * a result indicating whether the vote is pending, lost, or won. A joint quorum
 * requires both majority quorums to vote in favor.
 **/
26 27 28
ESyncRaftVoteType syncRaftVoteResult(SSyncRaftQuorumJointConfig* config, const ESyncRaftVoteType* votes) {
  ESyncRaftVoteResult r1 = syncRaftMajorityVoteResult(&(config->incoming), votes);
  ESyncRaftVoteResult r2 = syncRaftMajorityVoteResult(&(config->outgoing), votes);
29 30 31 32 33 34 35 36 37 38 39 40 41 42

  if (r1 == r2) {
    // If they agree, return the agreed state.
    return r1;
  }

  if (r1 == SYNC_RAFT_VOTE_LOST || r2 == SYNC_RAFT_VOTE_LOST) {
    // If either config has lost, loss is the only possible outcome.
    return SYNC_RAFT_VOTE_LOST;
  }

  // One side won, the other one is pending, so the whole outcome is.
  return SYNC_RAFT_VOTE_PENDING;
}
43 44 45 46 47

void syncRaftJointConfigAddToIncoming(SSyncRaftQuorumJointConfig* config, SyncNodeId id) {
  int i, min;

  for (i = 0, min = -1; i < TSDB_MAX_REPLICA; ++i) {
48
    if (config->incoming.nodeId[i] == id) {
49 50
      return;
    }
51
    if (min == -1 && config->incoming.nodeId[i] == SYNC_NON_NODE_ID) {
52 53 54 55 56
      min = i;
    }
  }

  assert(min != -1);
57
  config->incoming.nodeId[min] = id;
58 59 60 61 62 63 64
  config->incoming.replica += 1;
}

void syncRaftJointConfigRemoveFromIncoming(SSyncRaftQuorumJointConfig* config, SyncNodeId id) {
  int i;

  for (i = 0; i < TSDB_MAX_REPLICA; ++i) {
65
    if (config->incoming.nodeId[i] == id) {
66
      config->incoming.replica  -= 1;
67
      config->incoming.nodeId[i] = SYNC_NON_NODE_ID;
68 69 70 71 72
      break;
    }
  }

  assert(config->incoming.replica >= 0);
73 74
}

L
lichuang 已提交
75 76
void syncRaftJointConfigIDS(const SSyncRaftQuorumJointConfig* config, SSyncRaftNodeMap* nodeMap) {
  int i, j, m;
77

L
lichuang 已提交
78
  syncRaftCopyNodeMap(&config->incoming, nodeMap);
79

L
lichuang 已提交
80
  syncRaftUnionNodeMap(&config->outgoing, nodeMap);
81
}