程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> epoll模型的理解封裝與應用,epoll模型封裝

epoll模型的理解封裝與應用,epoll模型封裝

編輯:C++入門知識

epoll模型的理解封裝與應用,epoll模型封裝


    自己以前寫TCP服務器,並不需要考慮到並發與資源的問題,使用的都是單獨線程處理單個TCP連接的方式(說謂的PPC/TPC模型)。如今自己做高並發服務器,必須處理好這些問題。因為用的是linux2.6,因此選用epoll作為I/O多路復用技術接口再好不過了(呵呵呵)。

    通俗地講,epoll就是:告訴你有哪些socket准備要做哪些事。在select模型中,select用來檢測socket狀態,兩者的用法大相徑庭,但是機制不同。select的檢測方法是每次遍歷所有需要檢測的socket,並返回有動作socket。而epoll的並不會檢測所有的句柄狀態,通過內核的支持,能避免無意義的檢測。

    當socket句柄的數目特別大的情況下,首先PPC/TPC模型肯定就掛掉了。而select因為每次要遍歷所有句柄,因此在句柄遍歷的過程中占用了很多的時間,如果並發的數量接近句柄總數,select並沒有浪費太多時間,但對於並發數遠低於鏈接數的情況,比如回合制的網絡游戲,select就有浪費時間的嫌疑。因此epoll是相當高效的。

    在將epoll封裝成c++類之前,對epoll的數據結構以及接口做一下簡單介紹:

    epoll 事件結構體:

struct epoll_event {
        __uint32_t events;      // Epoll events
        epoll_data_t data;      // User datavariable
    };

 

    這裡的events是事件的類型,常用的有:

        EPOLLIN 該句柄為可讀

        EPOLLOUT 該句柄為可寫

        EPOLLERR 該句柄發生錯誤

        EPOLLET epoll為邊緣觸發模式

    

 

    epoll 事件date

typedef union epoll_data {
       void *ptr;
        int fd;
       __uint32_t u32;
       __uint64_t u64;
} epoll_data_t;

 

    注意epoll_data是個union。我們想要掛上句柄或是數據指針都很方便。

     

    epoll創建:

        int epoll_create(int size);

    調用該函數會創建一個epoll句柄,參數size為監聽的最大數量

  

  epoll控制:

    int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);

  這個接口用於對該epdf上的句柄進行注冊、修改和刪除。

  op是要進行的操作,有:

        EPOLL_CTL_ADD 添加需要監測的文件句柄fd

        EPOLL_CTL_MOD 更改該fd句柄的模式

        EPOLL_CTL_DEL 移除掉該句柄

  event是所要設置的該fd的事件。

 

  epoll收集信息:

     int epoll_wait(int epfd, struct epoll_event * events, int maxevents, int timeout);

    調用該函數後,如果該有epoll所管理的句柄發生對應類型的事件,這些發生事件的句柄的epoll_event將會被寫入events數組中,我們便能根據這些句柄執行接下來的I/O以及其他操作。這裡的maxevents是每次wait獲取的事件最大數。如果使用的是ET邊緣觸發模式,epoll_wait返回一個事件後,再這個時間的狀態沒有改變的情況下,epoll_wait不會再對改事件進行通知。

    

    epoll基本的介紹完,就可以先對epoll進行一定的封裝以增強代碼的復用。

    在封裝epoll之前,我先給出我封裝好的用於tcp的socket:

    

//總共所需要用到的頭文件,有部分是多余的
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cstdlib>

#ifdef WIN32
#include<winsock2.h>
#else
#include<fcntl.h>
#include<sys/ioctl.h>
#include<sys/socket.h>
#include<sys/epoll.h>
#include<unistd.h>
#include<netdb.h>
#include<errno.h>
#include<arpa/inet.h>
#include<netinet/in.h>
#include<sys/types.h>
#define SOCKET int
#define SOCKET_ERROR -1
#define INVALID_SOCKET -1
#endif

 

    

    這裡是我自己對普通tcp socket的封裝:

    

class msock
{
public:
    SOCKET sock;
    sockaddr_in addr;
    msock()
    {
        addr.sin_family=AF_INET;
    }
    void setsock(SOCKET fd)
    {
        sock=fd;
    }
    SOCKET getsock()
    {
        return sock;
    }
    void createsock()
    {
        sock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
        if(sock==INVALID_SOCKET)
        {
            puts("socket build error");
            exit(-1);
        }
    }
    void setioctl(bool x)
    {
        fcntl(sock, F_SETFL, O_NONBLOCK);
    }
    bool setip(string ip)
    {
        hostent *hname=gethostbyname(ip.c_str());
        if(!hname)
        {
            puts("can't find address");
            return false;
        }//puts(inet_ntoa(addr.sin_addr));
        addr.sin_addr.s_addr=*(u_long *)hname->h_addr_list[0];
        return true;
    }
    void setport(int port)
    {
        addr.sin_port=htons(port);
    }
    int msend(const char *data,const int len)
    {
        return send(sock,data,len,0);
    }
    int msend(const string data)
    {
        return msend(data.c_str(),data.length());
    }
    int msend(mdata *data)
    {
        return msend(data->buf,data->len);
    }
    int mrecv(char *data,int len)
    {
        return recv(sock,data,len,0);
    }
    int mrecv(char *data)
    {
        return recv(sock,data,2047,0);
    }
    int mclose()
    {
        return close(sock);
    }

    int operator == (msock jb)
    {
        return sock==jb.sock;
    }
};

 

    listen用的sock繼承於msock:

class mssock:public msock
{
public:
    sockaddr_in newaddr;
    socklen_t newaddrlen;
    mssock():msock()
    {
        createsock();
        addr.sin_addr.s_addr=htonl(INADDR_ANY);
        newaddrlen=sizeof(newaddr);//hehe
    }

    int mbind()
    {
        return bind(sock,(sockaddr *)&addr,sizeof(addr));
    }
    int mlisten(int num=20)
    {
        return listen(sock,num);
    }
    msock maccept()
    {
        SOCKET newsock=accept(sock,(sockaddr *)&newaddr,&newaddrlen);
        msock newmsock;
        newmsock.setsock(newsock);
        return newmsock;
    }

};

 

 

    以上的msock和mssock類裡面含有socket句柄,可以直接將類強制轉換為socket句柄

    在對epoll封裝之前還有一步就是:定義一個數據結構用於存放不定長度的數據,以便掛入epoll的事件中。

    

struct mdata
{
    int fd;
    unsigned int len;
    char buf[2048];
    mdata(){}
    mdata(char *s,const int length)
    {
        for(int i=0;i<length;i++)
        {
            buf[i]=s[i];
        }
    }
};

 

 

    epoll的封裝可以開始了,使用的是邊緣觸發的方式,我的思路是:將epoll的句柄以及參數都記錄在類中,並自己維護一個events數據用於對應的事件。外部只需要根據返回事件的臨時編號通過類的方法獲取返回值即可。

class mepoll
{
public:
    int epfd;        //epoll自身的句柄
    epoll_event ev,*events;    //臨時事件和每次wait用於儲存的事件數組
    int maxevents;    //最大事件數
    int timeout;    //wait超時
    
    //構造函數默認最大事件數為20
    mepoll(unsigned short eventsnum=20)
    {
        epfd=epoll_create(0xfff);
        maxevents=eventsnum;
        events=new epoll_event[maxevents];
        timeout=-1;
    }
    
    //添加新的socket句柄到epoll中
    int add(SOCKET fd)
    {
        fcntl(fd, F_SETFL, O_NONBLOCK);//設置fd為非阻塞
        ev.events=EPOLLIN|EPOLLET;
        ev.data.fd=fd;
        return epoll_ctl(epfd,EPOLL_CTL_ADD,fd,&ev);
    }
    
    //設置對應編號的句柄事件為可讀
    void ctl_in(int index)
    {
        ev.data.fd=*(int *)events[index].data.ptr;
        ev.events=EPOLLIN|EPOLLET;
        epoll_ctl(epfd,EPOLL_CTL_MOD,*(int *)events[index].data.ptr,&ev);
    }
    
    //改可寫,並將要寫的數據data綁定到該句柄對應的事件中
    void ctl_out(int index,mdata *data)
    {
        data->fd=events[index].data.fd;
        ev.data.ptr=data;
        ev.events=EPOLLOUT|EPOLLET;
        epoll_ctl(epfd,EPOLL_CTL_MOD,events[index].data.fd,&ev);
    }
    int wait()
    {
        return epoll_wait(epfd,events,maxevents,timeout);
    }
    unsigned int geteventtype(int index)
    {
        return events[index].events;
    }
    
    //獲取對應編號中的msock
    msock getsock(int index)
    {
        msock sk;
        sk.setsock(events[index].data.fd);
        return sk;
    }
    
    //從mdata裡獲取出msock
    msock getsock(mdata *data)
    {
        msock sk;
        sk.setsock(data->fd);
        return sk;
    }
    
    //獲取對應編號的事件
    mdata *getdata(int index)
    {
        return (mdata *)events[index].data.ptr;
    }
};

 

 

    現在有一個比較好用的epoll類了。於是可以開始實現一個簡單的完整服務器程序了。

    在實現過程中,有幾點需要注意區分用於listen用的句柄和收發數據使用的句柄。因為采用的是邊緣觸發的方式,很可能會出現同事listen到多個連接的情況,但是這裡epoll_wait只會通知一次。如果我們發現有accept事件,我們卻沒有把所有accept處理完,很多的鏈接就不能連入。對於這種問題,可以這樣處理:在listen發生時,一直accept直到accept失敗吧所有鏈接都處理完再繼續。

    下面我使用我的游戲邏輯的接口和epoll類實現一個基本的服務器程序:

    游戲邏輯的接口很簡單,只需要調用gamemain創建出該游戲類的實例。並使用收到的數據調用mdata *gamemain::dealdata(mdata *data) 函數即可得到游戲邏輯處理後的mdata,將處理好的mdata發回去,這裡處理後的mdata*是游戲實例自動分配的,發完之後調用gamemain::freedatainpool(mdata *data釋放(那邊也會自動釋放的)。(哈哈,沒想到自己第一次寫游戲服務器邏輯能做得如此低耦合)

    

#include "ssock.h"
#include "game.h"

int main()
{
    gamemain game;//創建游戲實例

    mepoll ep;//epoll類
    mssock ssock;//服務器listen用的sock
    msock  csock;//臨時sock
    mdata rdata;//臨時rdata

    ssock.setport(5000);//使用5000端口
    if(SOCKET_ERROR==ssock.mbind())
    {
        puts("bind error");
        return -1;
    }
    if(SOCKET_ERROR==ssock.mlisten())
    {
        puts("listen error");
        return -1;
    }
    
    //開始listen
    //將listen句柄加入到epoll中
    ep.add(ssock.getsock());
    
puts("server start");
    int ionum;
    while(1)
    {
        ionum=ep.wait();//獲取事件
        //遍歷並處理所有事件
        for(int i=0; i<ionum; i++)
        {
printf("some data come: ");
            csock=ep.getsock(i);
            if(ep.geteventtype(i)&EPOLLERR)
            {
                printf("sock %u error\n",csock.sock);
                csock.mclose();
            }
            else if(ssock==csock)//處理listen事件
            {
                while(1)//accept直到沒有新連接
                {
                    csock=ssock.maccept();
                    if(csock.getsock()==SOCKET_ERROR)
                    {
                        break;
                    }
                    //將新連接加入到epoll中
                    ep.add(csock.getsock());
puts("a newsock comed:");
                }
            }
            else if(ep.geteventtype(i)&EPOLLIN)//處理接收事件
            {
                //根據臨時編號獲取到對應sock並接收數據
                csock=ep.getsock(i);
printf("sock %u in\n",csock.sock);
                int rlen;
                bool isrecv=false;
                rdata.len=0;
                while(1)
                {
                    rlen=csock.mrecv(rdata.buf+rdata.len);
                    if(rlen<0)
                    {
                        if (errno == EAGAIN)
                        {
                            isrecv = true;
                            break;
                        }
                        else if (errno == EINTR)
                        {
                            continue;
                        }
                        else
                        {
                            break;
                        }
                    }
                }
                if(isrecv)
                {
                    //調用游戲邏輯處理數據並修改sock事件為發送
                    ep.ctl_out(i,game.dealdata(&rdata));
                }
            }
            else if(ep.geteventtype(i)&EPOLLOUT)//處理發送事件
            {
                mdata *data=ep.getdata(i);
                csock=ep.getsock(data);
printf("sock %u out type:%u\n",csock.sock,data->buf[4]);
                int slen,cnt=0;
                bool issend=false;
                while(1)
                {
                    slen=csock.msend(data);
                    if(slen<0)
                    {
                        if (errno == EAGAIN)
                        {
                            // 對於nonblocking 的socket而言,這裡說明了已經全部發送成功了
                            issend = true;
                            break;
                        }
                        else if (errno == EINTR)
                        {
                            // 被信號中斷
                            continue;
                        }
                        else
                        {
                            // 其他錯誤
                            break;
                        }
                    }

                    if(slen=0)
                    {
                        break;
                    }

                    /*cnt+=slen;
                    if(cnt>=data->len)*/
                    {
                        issend=true;
                        break;
                    }

                }
                game.freedatainpool(data);
                
                //無論發送情況都要改為可寫,以容錯
                ep.ctl_in(i);
            }
        }
    }
puts("server ended");
    return 0;
}

 

    這個程序每一次讀操作完成後,都是在單線程處理完游戲邏輯在進行下一步。如果游戲邏輯效率高且不會涉及到數據庫等待的問題,這種方式可取,否則可以另起線程處理游戲邏輯,實現真正的高並發。

    本文的整個內容已經講完了,epoll的學問可不止這些,需要在以後的實踐中要慢慢積累。

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved