sync_raft_quorum_joint.c 2.5 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
/*
 * 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/>.
 */

#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.
 **/
25 26 27
ESyncRaftVoteType syncRaftVoteResult(SSyncRaftQuorumJointConfig* config, const ESyncRaftVoteType* votes) {
  ESyncRaftVoteResult r1 = syncRaftMajorityVoteResult(&(config->incoming), votes);
  ESyncRaftVoteResult r2 = syncRaftMajorityVoteResult(&(config->outgoing), votes);
28 29 30 31 32 33 34 35 36 37 38 39 40 41

  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;
}
42 43 44 45 46

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

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

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

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

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

  assert(config->incoming.replica >= 0);
72 73 74 75 76 77 78 79 80 81 82 83 84
}


bool syncRaftIsInNodeMap(const SSyncRaftNodeMap* nodeMap, SyncNodeId nodeId) {
  int i;

  for (i = 0; i < TSDB_MAX_REPLICA; ++i) {
    if (nodeId == nodeMap->nodeId[i]) {
      return true;
    }
  }

  return false;
85
}