sock_ev——linux平台socket事件框架(event loop)

上一篇我们封装了三种事件监听方式，如果分别提供给客户端使用，有点不方便，也不利于统一管理；我们再封装一层EventLoop。

/***************************************************************************************
****************************************************************************************
* FILE		: event_loop.h
* Description	: 
*			  
* Copyright (c) 2012 by Liu Yanyun(E-mail:liuyun827@foxmail.com). All Rights Reserved.
*            Without permission, shall not be used for any commercial purpose
* 
* History:
* Version		Name       		Date			Description
   0.1		Liu Yanyun		2012/12/20		Initial Version
   
****************************************************************************************
****************************************************************************************/


#ifndef _EVENT_LOOP_H_
#define _EVENT_LOOP_H_

#include <string.h>
#include <stdint.h>
#include <sys/select.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/epoll.h>
#include <unistd.h>
#include <poll.h>
#include "sock_ev.h"

class Socket;
class EventDispatcher;


/*==================================================================
* Function	: EventLoop
* Description	: event loop,it is adapter for select poll epoll wait event
==================================================================*/
class EventLoop
{
public:

  /*==================================================================
  * Function	: EventLoop.EventLoop
  * Description	: construction function
  ==================================================================*/
  EventLoop();

  /*==================================================================
  * Function	: EventLoop.~EventLoop
  * Description	: Destructor function
  ==================================================================*/
  virtual ~EventLoop();

  /*==================================================================
  * Function	: EventLoop.initialize
  * Description	: initialize the loop
  * Input Para	: type_--dispatcher type
  * Return Value: success return true,or else return false
  ==================================================================*/
  bool initialize(LoopType type_);

  /*==================================================================
  * Function	: EventLoop.addEvt
  * Description	: add an event and register callback
  * Input Para	: 
  * Output Para	: 
  * Return Value: success return true,or else return false
  ==================================================================*/
  bool addEvt(Socket *sock_,
      EvCallBack cb_,
      EventType evt_,
      void *arg_);

  /*==================================================================
  * Function	: EventLoop.removeEvt
  * Description	: remove an event and un-register callback
  * Input Para	: 
  * Output Para	: 
  * Return Value: success return true,or else return false
  ==================================================================*/
  bool removeEvt(Socket *sock_,
      EventType evt_);

  /*==================================================================
  * Function	: EventLoop.listen
  * Description	: wait for event trigger
  * Input Para	: 
  * Output Para	: 
  * Return Value: 
  ==================================================================*/
  int listen(int timeout_);
      
private:

    EventDispatcher *dispatcher;
    SockMapT  sockMap;
};

#endif /*_EVENT_LOOP_H_*/

 

#include "event_loop.h"
#include "sock_ev.h"
#include "event_dispatcher.h"
#include "socket_base.h"
#include "socket.h"


EventLoop::EventLoop()
{
  dispatcher = NULL;
}
EventLoop::~EventLoop()
{
  if(NULL != dispatcher) delete dispatcher;

  sockMap.clear();
}
bool EventLoop::initialize(LoopType type_)
{
  if(selectLoop == type_)
  {
    dispatcher = new SelectDispatcher();
  }
  else if(pollLoop == type_)
  {
    dispatcher = new PollDispatcher();
  }
  else if(epollLoop == type_)
  {
    dispatcher = new EpollDispatcher();
  }
  else
  {
    logTrace("loop type is not right:%d", type_);
    return false;
  }

  if(NULL == dispatcher)
  {
    logTrace("new EventDispatcher is failed");
    return false;
  }

  if(!dispatcher->initialize())
  {
    logTrace("dispatcher->initialize() is failed");
    return false;
  }

  return true;
}
bool EventLoop::addEvt(Socket *sock_,
    EvCallBack cb_,
    EventType evt_,
    void *arg_)
{
  if((NULL == sock_) || (-1 == sock_->getFd()) || (NULL == cb_))
  {
    logTrace("addEvt input is not valid,pleace check");
    return false;
  }

  if(!dispatcher->addEvt(this, sock_, cb_, evt_, arg_))
  {
    logTrace("addEvt event to loop is failed");
    return false;
  }

  sockMap[sock_->getFd()] = sock_;

  return true;
}
bool EventLoop::removeEvt(Socket *sock_,
    EventType evt_)
{
  if((NULL == sock_) || (-1 == sock_->getFd()))
  {
    logTrace("removeEvt input is not valid,pleace check");
    return false;
  }

  dispatcher->removeEvt(sock_, evt_);

  EventType evt = sock_->getEvt();
  if(0 == evt)
  {
    sockMap.erase(sock_->getFd());
  }

  return true;
}
int EventLoop::listen(int timeout_)
{
  while(1)
  {
    dispatcher->listen(sockMap, timeout_);
  }
}

主要调用前一篇中封装的三种事件监听方式，在初始化的时候选择一种event dipatcher，这就是面向接口编程的好处，利用多态，可以很方便的在运行时进行行为的改变。成员变量sockMap用于保存注册到EventLoop中的socket。