mirror of
https://github.com/balloonwj/TeamTalk.git
synced 2026-10-23 18:08:33 +08:00
balloonwj
74 lines
1.5 KiB
C++
74 lines
1.5 KiB
C++
/*
|
|
* A socket event dispatcher, features include:
|
|
* 1. portable: worked both on Windows, MAC OS X, LINUX platform
|
|
* 2. a singleton pattern: only one instance of this class can exist
|
|
*/
|
|
#ifndef __EVENT_DISPATCH_H__
|
|
#define __EVENT_DISPATCH_H__
|
|
|
|
#include "network/ostype.h"
|
|
#include "network/util.h"
|
|
#include "network/Lock.h"
|
|
|
|
enum {
|
|
SOCKET_READ = 0x1,
|
|
SOCKET_WRITE = 0x2,
|
|
SOCKET_EXCEP = 0x4,
|
|
SOCKET_ALL = 0x7
|
|
};
|
|
|
|
class CEventDispatch
|
|
{
|
|
public:
|
|
virtual ~CEventDispatch();
|
|
|
|
void AddEvent(SOCKET fd, uint8_t socket_event);
|
|
void RemoveEvent(SOCKET fd, uint8_t socket_event);
|
|
|
|
void AddTimer(callback_t callback, void* user_data, uint64_t interval);
|
|
void RemoveTimer(callback_t callback, void* user_data);
|
|
|
|
void AddLoop(callback_t callback, void* user_data);
|
|
|
|
void StartDispatch(uint32_t wait_timeout = 100);
|
|
void StopDispatch();
|
|
|
|
inline bool isRunning()const {return running;}
|
|
inline void setRunning() { running = true; }
|
|
|
|
static CEventDispatch* Instance();
|
|
protected:
|
|
CEventDispatch();
|
|
|
|
private:
|
|
void _CheckTimer();
|
|
void _CheckLoop();
|
|
|
|
typedef struct {
|
|
callback_t callback;
|
|
void* user_data;
|
|
uint64_t interval;
|
|
uint64_t next_tick;
|
|
} TimerItem;
|
|
|
|
private:
|
|
#ifdef _MSC_VER
|
|
fd_set m_read_set;
|
|
fd_set m_write_set;
|
|
fd_set m_excep_set;
|
|
#elif __APPLE__
|
|
int m_kqfd;
|
|
#else
|
|
int m_epfd;
|
|
#endif
|
|
CLock m_lock;
|
|
list<TimerItem*> m_timer_list;
|
|
list<TimerItem*> m_loop_list;
|
|
|
|
static CEventDispatch* m_pEventDispatch;
|
|
|
|
bool running;
|
|
};
|
|
|
|
#endif
|