Linux下epoll的使用 包含定时器与socket服务端的使用方法

简单介绍epoll

• epoll的使用是为了将监听fd变化的部分交给linux kernel进行实现，在系统层级实现了IO多路复用，通过事件触发的方式完成后续你想进行的操作。
• 关于epoll的ET/LT触发方式，可以理解为LT是阻塞的，当fd事件来临而你没有处理则会持续通知，而ET方式则是只通知一次，因此ET触发方式效率相较于LT方式会更高。
• epoll中可以使用的fd仅能为系统进行过支持的fd，比如linux计时器的fd与socket fd，而如同你使用open函数打开的file fd则是不行的，因为这种fd内部没有实现过

epoll简单使用例，包含定时器与socket两种监听

#include <iostream>
#include <cstdlib>

#include <sys/epoll.h>

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/un.h>

#include <sys/timerfd.h>

/**
 * @brief epoll添加event函数
 * @author 9426224
 * @date 2022-09-29
 *
 * @param epoll_fd 创建完成的epoll_fd
 * @param epoll_fd 待添加的fd
 * @param state epoll参数
 **/
static void add_event(int epoll_fd, int fd, int state)
{
    struct epoll_event ev;
    ev.events = state;
    ev.data.fd = fd;
    epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd, &ev);
}

int main(int *argc, char *argv[])
{
    int epoll_fd; 
    struct epoll_event events[1024]; //存储epoll事件

    epoll_fd = epoll_create(1024); //创建一个类型为struct eventpoll的对象，用这个对象来管理epoll的所有事件。 并获得一个空闲的文件描述符fd

    if (epoll_fd == -1)
    {
        std::cout << "Epoll create failed." << std::endl;
        exit(-1);
    }



    //下面我们给epoll中添加一个定时器fd
    int timer_fd;
    struct itimerspec timer_spec;
    
    //下面给定时器设定定时时间
    timer_spec.it_value.tv_sec = 1;
    timer_spec.it_value.tv_nsec = 0;
    timer_spec.it_interval.tv_sec = 0;
    timer_spec.it_interval.tv_nsec = 100000000;

    // CLOCK_REALTIME 系统范围的实时时钟,设置此时钟需要适当的权限。 CLOCK_MONOTONIC 无法设置的时钟，表示自某些未指定的起点以来的单调时间。  CLOCK_PROCESS_CPUTIME_ID   来自CPU的高分辨率每进程计时器。  CLOCK_THREAD_CPUTIME_ID   特定于线程的CPU时钟。
    timer_fd = timerfd_create(CLOCK_MONOTONIC, TFD_CLOEXEC | TFD_NONBLOCK);
    timerfd_settime(timer_fd, 0, &timer_spec, NULL);

    add_event(epoll_fd, timer_fd, EPOLLIN | EPOLLET); //这里将timer_fd添加到epoll_fd的监听之中



    //下面我们给epoll中添加一个socket server端fd 监听任意地址来自8090端口的连接
    int server_fd;
    struct sockaddr_in server_addr;
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    server_addr.sin_port = htons(8090);

    server_fd = socket(AF_INET, SOCK_STREAM, 0);
    if (server_fd == -1)
    {
        std::cout << "Socket create failed." << std::endl;
        exit(-1);
    }
    if (bind(server_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) != 0)
    {
        std::cout << "Socket bind failed." << std::endl;
        exit(-1);
    }
    if (listen(server_fd, SOMAXCONN) != 0)
    {
        std::cout << "Socket listen failed." << std::endl;
        exit(-1);
    }

    add_event(epoll_fd, server_fd, EPOLLIN); //将server_fd添加到监听中



    //在循环中持续的处理epoll_fd监听到的事件
    while (1)
    {
        int num = epoll_wait(epoll_fd, events, sizeof(events) / sizeof(events[0]), -1);
        for (int i = 0; i < num; i++)
        {
            if(events[i].data.fd == server_fd)
            {
                struct sockaddr_in client_addr;
                socklen_t len = sizeof(client_addr);
                int client_fd = accept(server_fd, (struct sockaddr *)&client_addr, &len);

                if(client_fd == -1)
                {
                    std::cout << "Accept client failed." << std::endl;
                }

                char ip[32]={0};
                inet_ntop(AF_INET, &(client_addr.sin_addr.s_addr), ip, sizeof(ip));
                printf("New client connected! ip:%s port:%d\n", ip, ntohs(client_addr.sin_port));

                add_event(epoll_fd, client_fd, EPOLLIN); //将接收到的client_fd也添加到epoll监听中
            }
            else if (events[i].events & EPOLLIN)
            {
                if (events[i].data.fd == timer_fd)
                {
                    uint64_t exp;
                    ssize_t s = read(events[i].data.fd, &exp, sizeof(uint64_t)); //需要读出uint64_t大小, 不然会发生错误
                    assert(s == sizeof(uint64_t));

                    printf("get timerfd!\n");
                }
                else
                {
                    //这里如果不是socket server与定时器，则为新添加的client_fd
                    //在这里处理client_fd发送来的消息与判断client_fd的存货状态
                }
            }
        }
    }

    return 0;
}