rsm_client.h

   1 #ifndef rsm_client_h
   2 #define rsm_client_h
   3
   4 #include "rpc/rpc.h"
   5 #include "rsm_protocol.h"
   6 #include <string>
   7 #include <vector>
   8
   9
  10 //
  11 // rsm client interface.
  12 //
  13 // The client stubs package up an rpc, and then call the invoke procedure
  14 // on the replicated state machine passing the RPC as an argument.  This way
  15 // the replicated state machine isn't service specific; any server can use it.
  16 //
  17
  18 class rsm_client {
  19     protected:
  20         std::string primary;
  21         std::vector<std::string> known_mems;
  22         std::mutex rsm_client_mutex;
  23         void primary_failure(lock & rsm_client_mutex_lock);
  24         bool init_members(lock & rsm_client_mutex_lock);
  25     public:
  26         rsm_client(std::string dst);
  27         rsm_protocol::status invoke(unsigned int proc, std::string &rep, const std::string &req);
  28
  29         template<class R, class ...Args>
  30             int call(unsigned int proc, R & r, const Args & ...a1);
  31     private:
  32         template<class R> int call_m(unsigned int proc, R & r, const marshall & req);
  33 };
  34
  35 template<class R>
  36 int rsm_client::call_m(unsigned int proc, R & r, const marshall & req) {
  37     std::string rep;
  38     std::string res;
  39     int intret = invoke(proc, rep, req.cstr());
  40     VERIFY( intret == rsm_client_protocol::OK );
  41     unmarshall u(rep);
  42     u >> intret;
  43     if (intret < 0) return intret;
  44     u >> res;
  45     if (!u.okdone()) {
  46         fprintf(stderr, "rsm_client::call_m: failed to unmarshall the reply.\n"
  47                 "You probably forgot to set the reply string in "
  48                 "rsm::client_invoke, or you may call RPC 0x%x with wrong return "
  49                 "type\n", proc);
  50         VERIFY(0);
  51         return rpc_const::unmarshal_reply_failure;
  52     }
  53     unmarshall u1(res);
  54     u1 >> r;
  55     if(!u1.okdone()) {
  56         fprintf(stderr, "rsm_client::call_m: failed to unmarshall the reply.\n"
  57                 "You are probably calling RPC 0x%x with wrong return "
  58                 "type.\n", proc);
  59         VERIFY(0);
  60         return rpc_const::unmarshal_reply_failure;
  61     }
  62     return intret;
  63 }
  64
  65 template<class R, class ...Args>
  66 int rsm_client::call(unsigned int proc, R & r, const Args & ...a1) {
  67     return call_m(proc, r, marshall{a1...});
  68 }
  69
  70 #endif