Eventfd.h

#pragma once
#include<sys/eventfd.h>
#include<functional>
using std::function;
namespace wd
{
    using EventfdCallBack = function<void()>;
    class Eventfd
    {
    public:
        Eventfd(EventfdCallBack&& cb);
        void start();
        void stop();

        void wakeup();//向_evfd中发送消息

    private:
        void handleRead();//读取_evfd中发送的数据
        int createEventfd();
    private:
        int _evfd;
        EventfdCallBack _cb;
        bool _isRunning;
    };
};

Eventfd.cpp

#include"Eventfd.h"
#include<sys/poll.h>
#include<string.h>
#include<errno.h>
#include<unistd.h>
namespace wd
{
    Eventfd::Eventfd(EventfdCallBack &&cb)
:_evfd(createEventfd()),_cb(cb),_isRunning(false)
{
}
void Eventfd::start()
{
    _isRunning =true;
    struct pollfd pfd;
    bzero(&pfd,sizeof(pfd));
    pfd.fd = _evfd;
    pfd.events = POLLIN;
    while(_isRunning)
    {
        int nready =poll(&pfd,1,5000);
        if(-1== nready && errno == EINTR)
        {
            continue;
        }
        else if(-1== nready)
        {
            perror("poll");
        }
        else if(0== nready)
        {
            printf("timeout\n");
        }
        else
        {
            handleRead();//先读取_evfd中的数据
            _cb();//再调⽤回调函数
        }
    }
}
void Eventfd::stop()
{
    if(_isRunning)
         _isRunning =false;
}
void Eventfd::wakeup()
{
    uint64_t event_write =1;
    if(write(_evfd,&event_write,sizeof(uint64_t))!=sizeof(uint64_t))
    {
        perror("write evfd");
    }
}
void Eventfd::handleRead()
{
    uint64_t event_read;
    if(read(_evfd,&event_read,sizeof(uint64_t))!=sizeof(uint64_t))
    {
        perror("read evfd");
    }
}
int Eventfd::createEventfd()
{
    int evfd =eventfd(0,0);
    if(-1== evfd)
    {
        perror("eventfd");
    }
    return evfd
}

Logo

欢迎加入 MCP 技术社区!与志同道合者携手前行,一同解锁 MCP 技术的无限可能!

更多推荐