lock_server.h

   1 #ifndef lock_server_h
   2 #define lock_server_h
   3
   4 #include "types.h"
   5 #include "lock_protocol.h"
   6 #include "rsm.h"
   7 #include "rpc/fifo.h"
   8
   9 typedef pair<callback_t, lock_protocol::xid_t> holder_t;
  10
  11 class lock_state {
  12 public:
  13     lock_state();
  14     lock_state(const lock_state & other);
  15     bool held;
  16     holder_t held_by;
  17     list<holder_t> wanted_by;
  18     map<callback_t, lock_protocol::xid_t> old_requests;
  19     mutex m;
  20     lock_state & operator=(const lock_state &);
  21
  22     MEMBERS(held, held_by, wanted_by)
  23 };
  24
  25 MARSHALLABLE_STRUCT(lock_state)
  26
  27 typedef map<lock_protocol::lockid_t, lock_state> lock_map;
  28
  29 class lock_server : private rsm_state_transfer {
  30     private:
  31         int nacquire;
  32         mutex lock_table_lock;
  33         lock_map lock_table;
  34         lock_state & get_lock_state(lock_protocol::lockid_t lid);
  35         fifo<lock_protocol::lockid_t> retry_fifo;
  36         fifo<lock_protocol::lockid_t> revoke_fifo;
  37         rsm *rsm_;
  38         string marshal_state();
  39         void unmarshal_state(const string & state);
  40         void revoker NORETURN ();
  41         void retryer NORETURN ();
  42     public:
  43         lock_server(rsm & r);
  44         lock_protocol::status acquire(int &, lock_protocol::lockid_t, const callback_t & id, lock_protocol::xid_t);
  45         lock_protocol::status release(int &, lock_protocol::lockid_t, const callback_t & id, lock_protocol::xid_t);
  46         lock_protocol::status stat(int &, lock_protocol::lockid_t, const callback_t & id);
  47 };
  48
  49 #endif