95 lines
2.8 KiB
C++
95 lines
2.8 KiB
C++
#include <string>
|
|
|
|
#include "legislator.hh"
|
|
#include "misc/logger.hh"
|
|
#include "message/message.hh"
|
|
#include "events/register.hh"
|
|
#include "events/send.hh"
|
|
|
|
namespace paxos
|
|
{
|
|
Legislator::Legislator(const LegislatorConfig& config)
|
|
: config_(config), ledger(config.name)
|
|
{
|
|
log("created legislator " + config.name, blue);
|
|
}
|
|
|
|
void Legislator::initiate_ballot()
|
|
{
|
|
int new_ballot_number = ledger.last_tried() + 1;
|
|
ledger.set_last_tried(new_ballot_number);
|
|
log(config_.name + " is initiating ballot " + std::to_string(new_ballot_number), cyan);
|
|
send_next_ballot(new_ballot_number);
|
|
}
|
|
|
|
void Legislator::send_next_ballot(int ballot)
|
|
{
|
|
std::string ballot_string = std::to_string(ballot);
|
|
Message message;
|
|
message.set_method("NextBallot");
|
|
message.add_header("ballot", ballot_string);
|
|
message.add_header("sender", self->config_.name);
|
|
|
|
for (auto legislator : legislators)
|
|
SendEW::send_message(message, legislator.second);
|
|
}
|
|
|
|
void Legislator::receive_next_ballot(Message message)
|
|
{
|
|
std::string ballot_str = *message.get_header("ballot");
|
|
int ballot = std::stoi(ballot_str);
|
|
receive_next_ballot(ballot, *message.get_header("sender"));
|
|
}
|
|
|
|
void Legislator::receive_next_ballot(int ballot, std::string sender)
|
|
{
|
|
log(config_.name + " has received a NextBallot("
|
|
+ std::to_string(ballot)
|
|
+ ") from " + sender, cyan);
|
|
int next_ballot = ledger.next_bal();
|
|
if (ballot <= next_ballot)
|
|
{
|
|
log("but it was discarded because ballot " + std::to_string(ballot)
|
|
+ " is inferior or equal to nextBallot "
|
|
+ std::to_string(next_ballot), red);
|
|
return;
|
|
}
|
|
ledger.set_next_bal(ballot);
|
|
int previous_vote = ledger.prev_vote();
|
|
previous_vote = previous_vote;
|
|
//XXX send a LastVote to sender
|
|
}
|
|
|
|
void Legislator::receive_last_vote(Message message)
|
|
{
|
|
message = message;
|
|
}
|
|
|
|
void Legislator::receive_enough_last_vote
|
|
(std::unordered_map<std::string, int> quorum_last_votes)
|
|
{
|
|
quorum_last_votes = quorum_last_votes;
|
|
//find d to satisfy B3
|
|
//send BeginBallot to the quorum
|
|
}
|
|
|
|
void Legislator::receive_begin_ballot(int ballot, int decree)
|
|
{
|
|
if (ballot != ledger.next_bal())
|
|
return;
|
|
ledger.set_prev_vote(ballot);
|
|
|
|
decree = decree;
|
|
//XXX send Voted
|
|
}
|
|
|
|
void Legislator::handle_message(Message message)
|
|
{
|
|
std::string method = message.get_method();
|
|
if (method == "NextBallot")
|
|
receive_next_ballot(message);
|
|
else if (method == "LastVote")
|
|
receive_last_vote(message);
|
|
}
|
|
}
|