rpc/pollmgr.h

   1 #ifndef pollmgr_h
   2 #define pollmgr_h
   3
   4 #include "types.h"
   5 #include <sys/select.h>
   6
   7 #ifdef __linux__
   8 #include <sys/epoll.h>
   9 #endif
  10
  11 #define MAX_POLL_FDS 128
  12
  13 typedef enum {
  14     CB_NONE = 0x0,
  15     CB_RDONLY = 0x1,
  16     CB_WRONLY = 0x10,
  17     CB_RDWR = 0x11,
  18     CB_MASK = ~0x11,
  19 } poll_flag;
  20
  21 class aio_mgr {
  22     public:
  23         virtual void watch_fd(int fd, poll_flag flag) = 0;
  24         virtual bool unwatch_fd(int fd, poll_flag flag) = 0;
  25         virtual bool is_watched(int fd, poll_flag flag) = 0;
  26         virtual void wait_ready(vector<int> *readable, vector<int> *writable) = 0;
  27         virtual ~aio_mgr() {}
  28 };
  29
  30 class aio_callback {
  31     public:
  32         virtual void read_cb(int fd) = 0;
  33         virtual void write_cb(int fd) = 0;
  34         virtual ~aio_callback() {}
  35 };
  36
  37 class PollMgr {
  38     public:
  39         PollMgr();
  40         ~PollMgr();
  41
  42         static PollMgr *Instance();
  43         static PollMgr *CreateInst();
  44
  45         void add_callback(int fd, poll_flag flag, aio_callback *ch);
  46         void del_callback(int fd, poll_flag flag);
  47         bool has_callback(int fd, poll_flag flag, aio_callback *ch);
  48         void block_remove_fd(int fd);
  49         void wait_loop();
  50
  51
  52         static PollMgr *instance;
  53         static int useful;
  54         static int useless;
  55
  56     private:
  57         mutex m_;
  58         cond changedone_c_;
  59         thread th_;
  60
  61         aio_callback *callbacks_[MAX_POLL_FDS];
  62         aio_mgr *aio_;
  63         bool pending_change_;
  64
  65 };
  66
  67 class SelectAIO : public aio_mgr {
  68     public :
  69
  70         SelectAIO();
  71         ~SelectAIO();
  72         void watch_fd(int fd, poll_flag flag);
  73         bool unwatch_fd(int fd, poll_flag flag);
  74         bool is_watched(int fd, poll_flag flag);
  75         void wait_ready(vector<int> *readable, vector<int> *writable);
  76
  77     private:
  78
  79         fd_set rfds_;
  80         fd_set wfds_;
  81         int highfds_;
  82         int pipefd_[2];
  83
  84         mutex m_;
  85
  86 };
  87
  88 #ifdef __linux__
  89 class EPollAIO : public aio_mgr {
  90     public:
  91         EPollAIO();
  92         ~EPollAIO();
  93         void watch_fd(int fd, poll_flag flag);
  94         bool unwatch_fd(int fd, poll_flag flag);
  95         bool is_watched(int fd, poll_flag flag);
  96         void wait_ready(vector<int> *readable, vector<int> *writable);
  97
  98     private:
  99         int pollfd_;
 100         struct epoll_event ready_[MAX_POLL_FDS];
 101         int fdstatus_[MAX_POLL_FDS];
 102
 103 };
 104 #endif /* __linux */
 105
 106 #endif /* pollmgr_h */
 107