rpc/poll_mgr.h

   1 #ifndef poll_mgr_h
   2 #define poll_mgr_h
   3
   4 #include "types.h"
   5
   6 #define MAX_POLL_FDS 128
   7
   8 typedef enum {
   9     CB_NONE = 0x0,
  10     CB_RDONLY = 0x1,
  11     CB_WRONLY = 0x10,
  12     CB_RDWR = 0x11,
  13     CB_MASK = ~0x11,
  14 } poll_flag;
  15
  16 class aio_callback {
  17     public:
  18         virtual void read_cb(int fd) = 0;
  19         virtual void write_cb(int fd) = 0;
  20         virtual ~aio_callback();
  21 };
  22
  23 class poll_mgr {
  24     public:
  25         poll_mgr();
  26         ~poll_mgr();
  27
  28         void add_callback(int fd, poll_flag flag, aio_callback *ch);
  29         void del_callback(int fd, poll_flag flag);
  30         void block_remove_fd(int fd);
  31         void wait_loop();
  32         void shutdown();
  33
  34     private:
  35         std::mutex m_;
  36         cond changedone_c_;
  37
  38         std::map<int, aio_callback *> callbacks_;
  39         unique_ptr<class wait_manager> aio_;
  40         bool pending_change_=false, shutdown_=false;
  41
  42         thread th_;
  43 };
  44
  45 #endif