rpcserver.h

// Copyright 2017 Rice University, COMP 413 2017

#include <easylogging++.h>
#include <google/protobuf/io/coded_stream.h>
#include <unistd.h>
#include <thread>
#include <iostream>
#include <string>
#include <unordered_map>
#include <functional>

#include <asio.hpp>

#include <RpcHeader.pb.h>
#include <ProtobufRpcEngine.pb.h>
#include <IpcConnectionContext.pb.h>

#include "socket_writes.h"
#include "socket_reads.h"

#include <unistd.h>

#pragma once

using asio::ip::tcp;

class RPCServer {
 public:
  explicit RPCServer(int port);
  void serve(asio::io_service &io_service);
  void register_handler(
      std::string key,
      std::function<std::string(std::string)> handler);

  std::string getUsername();

 private:
  int port;
  std::unordered_map<std::string, std::function<std::string(std::string)>>
      dispatch_table;
  bool receive_handshake(
      tcp::socket &sock,
      int16_t *version,
      int16_t *service,
      int16_t *auth_protocol);
  bool receive_prelude(tcp::socket &sock);
  void handle_rpc(tcp::socket sock);
  bool send_error_header(
      hadoop::common::RpcRequestHeaderProto rpc_request_header,
      hadoop::common::RpcResponseHeaderProto response_header,
      std::string response_header_str,
      tcp::socket &sock);

  // **RPCSserver**
  static const std::string CLASS_NAME;
};