zoukankan      html  css  js  c++  java
  • Socket编程实践(11) --epoll原理与封装

    常用模型的特点

        Linux 下设计并发网络程序,有典型的Apache模型(Process Per Connection,PPC), TPC(Thread Per Connection)模型,以及 select/polL模型和epoll模型。

     

    1 、PPC/TPC 模型

        这两种模型思想类似,就是让每一个到来的连接一边自己做事去,别再来烦我(详见本系列博客).只是 PPC 是为它开了一个进程,而 TPC 开了一个线程。可是别烦我是有代价的,它要时间和空间啊,连接多了之后,那么多的进程/线程切换,这开销就上来了;因此这类模型能接受的最大连接数都不会高,一般在几百个左右

    2 、select 模型

        1) 最大并发数限制,因为一个进程所打开的 FD (文件描述符)是有限制的,由 FD_SETSIZE 设置,默认值是 1024,因此 Select 模型的最大并发数就被相应限制了。自己改改这个 FD_SETSIZE ?想法虽好,可是先看看下面吧 …

        2) 效率问题, select 每次调用都会线性扫描全部的 FD 集合,这样效率就会呈现线性下降,把 FD_SETSIZE 改大的后果就是,大家都慢慢来,什么?都超时了??!!

        3) 内核/用户空间内存拷贝问题,如何让内核把 FD 消息通知给用户空间呢?在这个问题上 select 采取了内存拷贝方法。

    3、 poll 模型

        基本上效率和 select 是相同的, select 缺点的 2 和 3 它都没有改掉。

     

    Epoll 的提升

        1. Epoll 没有最大并发连接的限制,上限是最大可以打开文件的数目,这个数字一般远大于 2048, 一般来说这个数目和系统内存关系很大 ,具体数目可以 cat /proc/sys/fs/file-max[599534] 察看。

        2. 效率提升, Epoll最大的优点就在于它只管你“活跃”的连接 ,而跟连接总数无关,因此在实际的网络环境中, Epoll的效率就会远远高于 select 和 poll 。

        3. 内存拷贝, Epoll 在这点上使用了“共享内存(详见本系列其他博客)”,这个内存拷贝也省略了。


    epoll的使用

    epoll的接口非常简单,一共就3/4个函数:

    int epoll_create(int size);
    int epoll_create1(int flags);
    int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);
    int epoll_wait(int epfd, struct epoll_event *events, int maxevents, int timeout);

       1. 对于epoll_create1 的flag参数: 可以设置为0 或EPOLL_CLOEXEC,为0时函数表现与epoll_create一致, EPOLL_CLOEXEC标志与open 时的O_CLOEXEC 标志类似,即进程被替换时会关闭打开的文件描述符(需要注意的是,epoll_create与epoll_create1当创建好epoll句柄后,它就是会占用一个fd值,在linux下如果查看/proc/<pid>/fd/,是能够看到这个fd的,所以在使用完epoll后,必须调用close()关闭,否则可能导致fd被耗尽)。

       2. 对于epoll_ctl, op参数表示动作,用三个宏来表示:

    EPOLL_CTL_ADD

    注册新的fd到epfd中

    EPOLL_CTL_DEL

    从epfd中删除一个fd

    EPOLL_CTL_MOD

    修改已经注册的fd的监听事件

       3. 对于epoll_wait:

          events:结构体指针, 一般是一个数组

          maxevents:事件的最大个数, 或者说是数组的大小

          timeout:超时时间, 含义与poll的timeout参数相同,设为-1表示永不超时;

       4. epoll_event结构体

    struct epoll_event
    {
        uint32_t     events;      /* Epoll events */
        epoll_data_t data;        /* User data variable */
    };
    typedef union epoll_data
    {
        void        *ptr;
        int          fd;
        uint32_t     u32;
        uint64_t     u64;
    } epoll_data_t;

    一般data 共同体我们设置其成员fd即可,也就是epoll_ctl 函数的第三个参数。

    events集合

    EPOLLIN

    表示对应的文件描述符可以读(包括对端SOCKET正常关闭)

    EPOLLOUT

    表示对应的文件描述符可以写

    EPOLLPRI

    表示对应的文件描述符有紧急的数据可读(这里应该表示有带外数据到来)

    EPOLLERR

    表示对应的文件描述符发生错误

    EPOLLHUP

    表示对应的文件描述符被挂断

    EPOLLET

    将EPOLL设为边缘触发(Edge Triggered)模式,这是相对于水平触发(Level Triggered)来说的

    EPOLLONESHOT

    只监听一次事件,当监听完这次事件之后,如果还需要继续监听这个socket的话,需要再次把这个socket加入到EPOLL队列里

    /**示例: epoll使用示例
     	注:client端与测试端与前同, 而且使用相同的测试端测试select/poll/epoll, 可以发现epoll的效率是非常高的**/
    
    //添加fd到epoll
    void addFd(int epollfd, int fd, const uint32_t &events = EPOLLIN, bool et = false)
    {
        struct epoll_event event;
        event.events = events;
        if (et)
            event.events |= EPOLLET;
        event.data.fd = fd;
        if( epoll_ctl(epollfd, EPOLL_CTL_ADD, fd, &event) == -1 )
            err_exit("epoll_ctl_add error");
    }
    //从epoll删除fd
    void delFd(int epollfd, int fd)
    {
        struct epoll_event event;
        event.data.fd = fd;
        if( epoll_ctl(epollfd, EPOLL_CTL_DEL, fd, &event) == -1 )
            err_exit("epoll_ctl_del error");
    }
    
    int main()
    {
        signal(SIGPIPE, sigHandlerForSigPipe);
        try
        {
            TCPServer server(8001);
            int listenfd = server.getfd();
            int epollfd = epoll_create1(EPOLL_CLOEXEC);
            if (epollfd == -1)
                err_exit("epoll_create1 error");
            // 将监听套接字注册到epoll
            addFd(epollfd, listenfd, EPOLLIN, true);
    
            // 用于保存epoll_wait返回事件数组
            std::vector<struct epoll_event> events(16);
            char buf[BUFSIZ];
            int count = 0;
            while (true)
            {
                // 等待epoll返回
                int nReady = epoll_wait(epollfd, &*events.begin(), (int)events.size(), -1);
                if (nReady == -1)
                {
                    if (errno == EINTR)
                        continue;
                    err_exit("epoll_wait error");
                }
                if ((size_t)nReady == events.size())
                    events.resize(events.size()*2);
    
                for (int i = 0; i < nReady; ++i)
                {
                    // 如果是监听套接字发送了可读事件
                    if (events[i].data.fd == listenfd)
                    {
                        int connectfd = accept(listenfd, NULL, NULL);
                        if (connectfd == -1)
                            err_exit("accept error");
                        cout << "accept success..." << endl;
                        cout << "count = " << ++count << endl;
                        setUnBlock(connectfd, true);
                        addFd(epollfd, connectfd, EPOLLIN, true);
                    }
                    // 如果是已连接套接字发生了可读事件
                    else if (events[i].events & EPOLLIN)
                    {
                        int connectfd = events[i].data.fd;
                        if (connectfd < 0)
                            continue;
    
                        memset(buf, 0, sizeof(buf));
                        int ret = readline(connectfd, buf, sizeof(buf)-1);
                        if (ret == -1)
                            err_exit("read-line error");
                        // 如果对端关闭
                        else if (ret == 0)
                        {
                            cerr << "client connect closed..." << endl;
                            // 将该套接字同epoll中移除
                            delFd(epollfd, connectfd);
                            close(connectfd);
                            continue;
                        }
                        cout << buf;
                        writen(connectfd, buf, strlen(buf));
                    }
                }
            }
        }
        catch (const SocketException &e)
        {
            cerr << e.what() << endl;
            err_exit("TCPServer error");
        }
    }

    小结-epoll与select、poll的区别

    1.相比于select与poll, epoll最大的好处在于它不会随着监听fd数目的增长而降低效率。

       因为内核中select/poll的实现是采用轮询来处理的, 因此他们检测就绪实践的算法时间复杂度是O(N), 因此, 需要轮询的fd数目越多, 自然耗时越多, 他们的性能呈线性甚至指数的方式下降。

       而epoll的实现是基于事件回调的,如果fd有期望的事件发生就通过回调函数将其加入epoll就绪队列中,也就是说它只关心“活跃”的fd,与fd数目无关 其算法时间复杂度为O(1)。

    2. 内核空间与用户空间内存拷贝问题,如何让内核把 fd消息通知给用户空间呢?在这个问题上select/poll采取了内存拷贝方法。而epoll采用了内核和用户空间共享内存的方式。

    3. epoll不仅会告诉应用程序有I/0 事件到来,还会告诉应用程序相关的信息,这些信息是应用程序填充的,因此根据这些信息应用程序就能直接定位到事件,而不必遍历整个fd集合。而select/poll模型,当有 I/O 事件到来时, select/poll通知应用程序有事件到达,而应用程序必须轮询所有的fd集合,测试每个fd是否有事件发生,并处理事件。

    4. 当活动连接比较多的时候, epoll_wait的效率就未必比select/poll高了, 因为这时候对于epoll 来说一直在调用callback 函数, 回调函数被触发得过于频繁, 所以epoll_wait适用于连接数量多, 但活动连接少的情况;

     

    ET/LT模式

    1、EPOLLLT:完全靠Linux-kernel-epoll驱动,应用程序只需要处理从epoll_wait返回的fds, 这些fds我们认为它们处于就绪状态。此时epoll可以认为是更快速的poll。

    2、EPOLLET:此模式下,系统仅仅通知应用程序哪些fds变成了就绪状态,一旦fd变成就绪状态,epoll将不再关注这个fd的任何状态信息(从epoll队列移除), 直到应用程序通过读写操作(非阻塞)触发EAGAIN状态,epoll认为这个fd又变为空闲状态,那么epoll又重新关注这个fd的状态变化(重新加入epoll队列)。 随着epoll_wait的返回,队列中的fds是在减少的,所以在大并发的系统中,EPOLLET更有优势,但是对程序员的要求也更高,因为有可能会出现数据读取不完整的问题,举例如下:

       假设现在对方发送了2k的数据,而我们先读取了1k,然后这时调用了epoll_wait,如果是边沿触发ET,那么这个fd变成就绪状态就会从epoll 队列移除,则epoll_wait 会一直阻塞,忽略尚未读取的1k数据; 而如果是水平触发LT,那么epoll_wait 还会检测到可读事件而返回,我们可以继续读取剩下的1k 数据。

       因此总结来说: LT模式可能触发的次数更多, 一旦触发的次数多, 也就意味着效率会下降; 但这样也不能就说LT模式就比ET模式效率更低, 因为ET的使用对编程人员提出了更高更精细的要求, 一旦编程人员水平达不到(比如本人), 那ET模式还不如LT模式;


    Epoll-Class封装

       在本部分我们实现一个较为好用实用的Epoll并发类, 由于实现代码与使用方式较简单, 因此就不在此赘述了, 下面我还使用了该类实现了一个基于Epoll的echo-server, 以演示该类的用法;

       由于此处仅为Epoll类库的第一个版本, 因此错误之处必然会存在, 如果读者在阅读的过程中发现了该类库的BUG, 还望这篇博客的读者朋友不吝赐教; 而作者也会不断的更新该类库(主要更新代码我会发布到此处), 以处理新的业务需求;


    Epoll类设计

    class Epoll
    {
    public:
        Epoll(int flags = EPOLL_CLOEXEC, int noFile = 1024);
        ~Epoll();
    
        void addfd(int fd, uint32_t events = EPOLLIN, bool ETorNot = false);
        void modfd(int fd, uint32_t events = EPOLLIN, bool ETorNot = false);
        void delfd(int fd);
        int wait(int timeout = -1);
        int getEventOccurfd(int eventIndex) const;
        uint32_t getEvents(int eventIndex) const;
    
    public:
        bool isValid()
        {
            if (m_epollfd == -1)
                return false;
            return true;
        }
        void close()
        {
            if (isValid())
            {
                :: close(m_epollfd);
                m_epollfd = -1;
            }
        }
    
    private:
        std::vector<struct epoll_event> events;
        int m_epollfd;
        int fdNumber;
        int nReady;
    private:
        struct epoll_event event;
    };

    Epoll类实现

    /** epoll_create **/
    Epoll::Epoll(int flags, int noFile) : fdNumber(0), nReady(0)
    {
        struct rlimit rlim;
        rlim.rlim_cur = rlim.rlim_max = noFile;
        if ( ::setrlimit(RLIMIT_NOFILE, &rlim) == -1 )
            throw EpollException("setrlimit error");
    
        m_epollfd = ::epoll_create1(flags);
        if (m_epollfd == -1)
            throw EpollException("epoll_create1 error");
    }
    Epoll::~Epoll()
    {
        this -> close();
    }
    /** epoll_ctl **/
    void Epoll::addfd(int fd, uint32_t events, bool ETorNot)
    {
        bzero(&event, sizeof(event));
        event.events = events;
        if (ETorNot)
            event.events |= EPOLLET;
        event.data.fd = fd;
        if( ::epoll_ctl(m_epollfd, EPOLL_CTL_ADD, fd, &event) == -1 )
            throw EpollException("epoll_ctl_add error");
        ++ fdNumber;
    }
    void Epoll::modfd(int fd, uint32_t events, bool ETorNot)
    {
        bzero(&event, sizeof(event));
        event.events = events;
        if (ETorNot)
            event.events |= EPOLLET;
        event.data.fd = fd;
        if( ::epoll_ctl(m_epollfd, EPOLL_CTL_MOD, fd, &event) == -1 )
            throw EpollException("epoll_ctl_mod error");
    }
    void Epoll::delfd(int fd)
    {
        bzero(&event, sizeof(event));
        event.data.fd = fd;
        if( ::epoll_ctl(m_epollfd, EPOLL_CTL_DEL, fd, &event) == -1 )
            throw EpollException("epoll_ctl_del error");
        -- fdNumber;
    }
    /** epoll_wait **/
    int Epoll::wait(int timeout)
    {
        events.resize(fdNumber);
        while (true)
        {
            nReady = epoll_wait(m_epollfd, &*events.begin(), fdNumber, timeout);
            if (nReady == 0)
                throw EpollException("epoll_wait timeout");
            else if (nReady == -1)
            {
                if (errno == EINTR)
                    continue;
                else  throw EpollException("epoll_wait error");
            }
            else
                return nReady;
        }
        return -1;
    }
    
    int Epoll::getEventOccurfd(int eventIndex) const
    {
        if (eventIndex > nReady)
            throw EpollException("parameter(s) error");
        return events[eventIndex].data.fd;
    }
    uint32_t Epoll::getEvents(int eventIndex) const
    {
        if (eventIndex > nReady)
            throw EpollException("parameter(s) error");
        return events[eventIndex].events;
    }

    使用Epoll的echoserver(测试)代码:

    int main()
    {
    signal(SIGPIPE, SIG_IGN);
        /**
        将下面的这两个变量设置成为放在程序的开头,
        只是因为这样可以使得业务处理部分的代码显
        得简洁一些,在实际应用(C++)中,没必要也不
        推荐这样使用
        **/
        char buf[BUFSIZ];
        int clientCount = 0;
        try
        {
            TCPServer server(8001);
            int listenfd = server.getfd();
            Epoll epoll;
            // 将监听套接字注册到epoll
            epoll.addfd(server.getfd(), EPOLLIN, true);
            while (true)
            {
                int nReady = epoll.wait();
                for (int i = 0; i < nReady; ++i)
                    // 如果是监听套接字发生了可读事件
                    if (epoll.getEventOccurfd(i) == listenfd)
                    {
                        int connectfd = accept(listenfd, NULL, NULL);
                        if (connectfd == -1)
                            err_exit("accept error");
                        cout << "accept success..." << endl;
                        cout << "clientCount = " << ++ clientCount << endl;
                        setUnBlock(connectfd, true);
                        epoll.addfd(connectfd, EPOLLIN, true);
                    }
                    else if (epoll.getEvents(i) & EPOLLIN)
                    {
                        TCPClient *client = new TCPClient(epoll.getEventOccurfd(i));
                        memset(buf, 0, sizeof(buf));
                        if (client->read(buf, sizeof(buf)) == 0)
                        {
                            cerr << "client connect closed..." << endl;
                            // 将该套接字从epoll中移除
                            epoll.delfd(client->getfd());
                            delete client;
                            continue;
                        }
                        cout << buf;
                        client->write(buf);
                    }
            }
        }
        catch (const SocketException &e)
        {
            cerr << e.what() << endl;
            err_exit("TCPServer error");
        }
        catch (const EpollException &e)
        {
            cerr << e.what() << endl;
            err_exit("Epoll error");
        }
    }

    完整源代码请参照:

    http://download.csdn.net/detail/hanqing280441589/8492911

  • 相关阅读:
    015-面向对象
    017-错误和异常
    019-File
    020-OS
    021-模块
    022-标准库
    数据库目录
    数据库 概念详解
    MySQL 基础
    MySQL 数据库操作
  • 原文地址:https://www.cnblogs.com/itrena/p/5926942.html
Copyright © 2011-2022 走看看