memgraph/tests/manual/raft_rpc.cpp

#include "boost/serialization/export.hpp"

#include "boost/archive/binary_iarchive.hpp"
#include "boost/archive/binary_oarchive.hpp"
#include "boost/archive/text_iarchive.hpp"
#include "boost/archive/text_oarchive.hpp"
#include "boost/serialization/export.hpp"

#include "communication/raft/rpc.hpp"
#include "communication/raft/storage/file.hpp"
#include "communication/raft/test_utils.hpp"

using namespace std::literals::chrono_literals;

namespace raft = communication::raft;

using io::network::Endpoint;
using raft::RaftConfig;
using raft::RpcNetwork;
using raft::test_utils::DummyState;

DEFINE_string(member_id, "", "id of Raft member");
DEFINE_string(log_dir, "", "Raft log directory");

BOOST_CLASS_EXPORT(raft::PeerRpcReply);
BOOST_CLASS_EXPORT(raft::PeerRpcRequest<DummyState>);

/* Start cluster members with:
 * ./raft_rpc --member-id a --log-dir a_log
 * ./raft_rpc --member-id b --log-dir b_log
 * ./raft_rpc --member-id c --log-dir c_log
 *
 * Enjoy democracy!
 */

int main(int argc, char *argv[]) {
  google::InitGoogleLogging(argv[0]);
  gflags::ParseCommandLineFlags(&argc, &argv, true);

  std::unordered_map<std::string, Endpoint> directory = {
      {"a", Endpoint("127.0.0.1", 12345)},
      {"b", Endpoint("127.0.0.1", 12346)},
      {"c", Endpoint("127.0.0.1", 12347)}};

  communication::rpc::System my_system(directory[FLAGS_member_id]);
  RpcNetwork<DummyState> network(my_system, directory);
  raft::SimpleFileStorage<DummyState> storage(FLAGS_log_dir);

  raft::RaftConfig config{{"a", "b", "c"}, 150ms, 300ms, 70ms, 30ms};

  {
    raft::RaftMember<DummyState> raft_member(network, storage, FLAGS_member_id,
                                             config);
    while (true) {
      continue;
    }
  }

  return 0;
}