Reactor 释义“反应堆”,是一种事件驱动机制。
把epoll的事件处理流程逆置,把对应的回调函数注册到Reactor,当事件发生时,Reactor调用回调函数函数。
Reactor 模式是处理并发 I/O 比较常见的一种模式,用于同步 I/O,中心思想是:
1.先注册的相应 I/O 事件分发到对应的处理器中
2.将所有要处理的 I/O 事件注册到一个中心 I/O 多路复用器上,同时主线程/进程阻塞在多路复用器上;
3.一旦有 I/O 事件到来,多路复用器从事件处理器获取相应函数调用
接下来从epoll的基础上逐步实现reactor
网络IO模型上一次代码的基础上把处理事件的部分改为回调函数
epoll_data是联合体,只能选择一种数据,那么我们既要socketfd也要回调函数就需要用到该联合体的ptr指针
typedef union epoll_data
{
void *ptr; //自定义数据
int fd; //保存socket
uint32_t u32;
uint64_t u64;
} epoll_data_t;
新建一个结构体存储到ptr就可以了
struct sockitem {
int sockfd; //event的fd
int (*callback)(int fd, int events, void *arg); //event的回调函数
};
代码
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/tcp.h>
#include <arpa/inet.h>
#include <errno.h>
#include <sys/epoll.h>
#define BUFFER_LENGTH 1024
#define EPOLL_SIZE 1024
int epoll_fd = 0;
char buffer[BUFFER_LENGTH] = {0};
//socket的数据
struct sockitem {
int sockfd;
int (*callback)(int fd, int events, void *arg);
};
int recv_cb(int fd, int events, void *arg);
int send_cb(int fd, int events, void *arg) {
struct sockitem *si = (struct sockitem*)arg;
struct epoll_event ev;
char buf[BUFFER_LENGTH] = {0};
sprintf(buf,"recv : %s\n", buffer);
send(fd, buf, BUFFER_LENGTH, 0);
si->sockfd = fd;
ev.events = EPOLLIN | EPOLLET;
si->callback = recv_cb;
ev.data.ptr = si;
epoll_ctl(epoll_fd, EPOLL_CTL_MOD, fd, &ev);
}
int recv_cb(int fd, int events, void *arg) {
struct sockitem *si = (struct sockitem*)arg;
struct epoll_event ev;
int ret = recv(fd, buffer, BUFFER_LENGTH, 0);
if (ret < 0) {
if (errno == EAGAIN || errno == EWOULDBLOCK) {
printf("read all data");
}
close(fd);
ev.events = EPOLLIN;
epoll_ctl(epoll_fd, EPOLL_CTL_DEL, fd, &ev);
free(si);
return -1;
} else if (ret == 0) {
printf(" disconnect %d\n", fd);
close(fd);
ev.events = EPOLLIN;
epoll_ctl(epoll_fd, EPOLL_CTL_DEL, fd, &ev);
return -1;
} else {
printf("Recv: %s, %d Bytes\n", buffer, ret);
//读完数据后回客户端信息,
//设置EPOLLOUT让clientfd回信息给客户端
ev.events = EPOLLOUT | EPOLLET;
si->sockfd = fd;
si->callback = send_cb;
ev.data.ptr = si;
//把clientfd的事件类型改为EPOLLOUT
epoll_ctl(epoll_fd, EPOLL_CTL_MOD, fd, &ev);
}
}
int accept_cb(int fd, int events, void *arg) {
struct sockaddr_in client_addr;
memset(&client_addr, 0, sizeof(struct sockaddr_in));
socklen_t client_len = sizeof(client_addr);
int clientfd = accept(fd, (struct sockaddr*)&client_addr, &client_len);
if (clientfd <= 0)
return -1;
char str[INET_ADDRSTRLEN] = {0};
printf("recvived from %s at port %d, sockfd:%d, clientfd:%d\n", inet_ntop(AF_INET, &client_addr.sin_addr, str, sizeof(str)),
ntohs(client_addr.sin_port), fd, clientfd);
struct epoll_event ev;
//设置客户端触发类型EPOLLIN和EPOLLET
//EPOLLET是每次事件触发一次
ev.events = EPOLLIN | EPOLLET;
struct sockitem *si = (struct sockitem*)malloc(sizeof(struct sockitem));
si->sockfd = clientfd;
si->callback = recv_cb;
ev.data.ptr = si;
epoll_ctl(epoll_fd, EPOLL_CTL_ADD, clientfd, &ev);
return clientfd;
}
int main(int argc, char *argv[]) {
if (argc < 2) {
printf("Paramter Error\n");
return -1;
}
int port = atoi(argv[1]);
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0) {
perror("socket");
return -1;
}
struct sockaddr_in addr;
memset(&addr, 0, sizeof(struct sockaddr_in));
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
addr.sin_addr.s_addr = INADDR_ANY;
if (bind(sockfd, (struct sockaddr*)&addr, sizeof(struct sockaddr_in)) < 0) {
perror("bind");
return 2;
}
if (listen(sockfd, 5) < 0) {
perror("listen");
return 3;
}
//创建epoll句柄
epoll_fd = epoll_create(1);
struct epoll_event ev, events[EPOLL_SIZE] = {0};
ev.events = EPOLLIN;
struct sockitem *si = (struct sockitem*)malloc(sizeof(struct sockitem));
si->sockfd = sockfd;
si->callback = accept_cb;
ev.data.ptr = si;
//添加服务端事件和fd
epoll_ctl(epoll_fd, EPOLL_CTL_ADD, sockfd, &ev);
while (1) {
//等待事件
int nready = epoll_wait(epoll_fd, events, EPOLL_SIZE, -1);
if (nready == -1) {
printf("epoll_wait\n");
break;
}
int i = 0;
for (i = 0;i < nready;i ++) {
//EPOLLIN事件
if (events[i].events & EPOLLIN) {
struct sockitem *si = (struct sockitem*)events[i].data.ptr;
si->callback(si->sockfd, events[i].events, si);
}
//EPOLLOUT事件
if(events[i].events & EPOLLOUT) {
struct sockitem *si = (struct sockitem*)events[i].data.ptr;
si->callback(si->sockfd, events[i].events, si);
}
}
}
}
把epollfd和事件集合定义为一个结构体,socket的结构体再加上缓存区
struct sockitem {
int sockfd;
int (*callback)(int fd, int events, void *arg);
char recvbuffer[BUFFER_LENGTH]; //接收缓存
char sendbuffer[BUFFER_LENGTH]; //发送缓存
int rlength; //接收缓存的长度
int slength; //发送缓存的长度
};
struct reactor {
int epfd;
struct epoll_event events[512];
};
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <netinet/tcp.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <errno.h>
#include <sys/epoll.h>
#define BUFFER_LENGTH 1024
struct sockitem {
int sockfd;
int (*callback)(int fd, int events, void *arg);
char recvbuffer[BUFFER_LENGTH]; //接收缓存
char sendbuffer[BUFFER_LENGTH]; //发送缓存
int rlength; //接收缓存的长度
int slength; //发送缓存的长度
};
struct reactor {
int epfd;
struct epoll_event events[512];
};
struct reactor *eventloop = NULL;
int recv_cb(int fd, int events, void *arg);
int send_cb(int fd, int events, void *arg) {
struct sockitem *si = (struct sockitem*)arg;
send(fd, si->sendbuffer, si->slength, 0);
struct epoll_event ev;
ev.events = EPOLLIN | EPOLLET;
si->sockfd = fd;
si->callback = recv_cb;
ev.data.ptr = si;
epoll_ctl(eventloop->epfd, EPOLL_CTL_MOD, fd, &ev);
}
int recv_cb(int fd, int events, void *arg) {
//int clientfd = events[i].data.fd;
struct sockitem *si = (struct sockitem*)arg;
struct epoll_event ev;
//char buffer[1024] = {0};
int ret = recv(fd, si->recvbuffer, BUFFER_LENGTH, 0);
if (ret < 0) {
if (errno == EAGAIN || errno == EWOULDBLOCK) { //
return -1;
}
ev.events = EPOLLIN;
epoll_ctl(eventloop->epfd, EPOLL_CTL_DEL, fd, &ev);
close(fd);
free(si);
} else if (ret == 0) {
printf("disconnect %d\n", fd);
ev.events = EPOLLIN;
epoll_ctl(eventloop->epfd, EPOLL_CTL_DEL, fd, &ev);
close(fd);
free(si);
} else {
printf("Recv: %s, %d Bytes\n", si->recvbuffer, ret);
si->rlength = ret;
memcpy(si->sendbuffer, si->recvbuffer, si->rlength);
si->slength = si->rlength;
struct epoll_event ev;
ev.events = EPOLLOUT | EPOLLET;
si->sockfd = fd;
si->callback = send_cb;
ev.data.ptr = si;
epoll_ctl(eventloop->epfd, EPOLL_CTL_MOD, fd, &ev);
}
}
int accept_cb(int fd, int events, void *arg) {
struct sockaddr_in client_addr;
memset(&client_addr, 0, sizeof(struct sockaddr_in));
socklen_t client_len = sizeof(client_addr);
int clientfd = accept(fd, (struct sockaddr*)&client_addr, &client_len);
if (clientfd <= 0) return -1;
char str[INET_ADDRSTRLEN] = {0};
printf("recv from %s at port %d\n", inet_ntop(AF_INET, &client_addr.sin_addr, str, sizeof(str)),
ntohs(client_addr.sin_port));
struct epoll_event ev;
ev.events = EPOLLIN | EPOLLET;
struct sockitem *si = (struct sockitem*)malloc(sizeof(struct sockitem));
si->sockfd = clientfd;
si->callback = recv_cb;
ev.data.ptr = si;
epoll_ctl(eventloop->epfd, EPOLL_CTL_ADD, clientfd, &ev);
return clientfd;
}
int main(int argc, char *argv[]) {
if (argc < 2) {
return -1;
}
int port = atoi(argv[1]);
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0) {
return -1;
}
struct sockaddr_in addr;
memset(&addr, 0, sizeof(struct sockaddr_in));
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
addr.sin_addr.s_addr = INADDR_ANY;
if (bind(sockfd, (struct sockaddr*)&addr, sizeof(struct sockaddr_in)) < 0) {
return -2;
}
if (listen(sockfd, 5) < 0) {
return -3;
}
eventloop = (struct reactor*)malloc(sizeof(struct reactor));
memset(eventloop,0,sizeof(struct reactor));
eventloop->epfd = epoll_create(1);
struct epoll_event ev;
ev.events = EPOLLIN;
struct sockitem *si = (struct sockitem*)malloc(sizeof(struct sockitem));
si->sockfd = sockfd;
si->callback = accept_cb;
ev.data.ptr = si;
epoll_ctl(eventloop->epfd, EPOLL_CTL_ADD, sockfd, &ev);
while (1) {
int nready = epoll_wait(eventloop->epfd, eventloop->events, 512, -1);
if (nready < -1) {
break;
}
int i = 0;
for (i = 0;i < nready;i ++) {
if (eventloop->events[i].events & EPOLLIN) {
struct sockitem *si = (struct sockitem*)eventloop->events[i].data.ptr;
si->callback(si->sockfd, eventloop->events[i].events, si);
}
if (eventloop->events[i].events & EPOLLOUT) {
struct sockitem *si = (struct sockitem*)eventloop->events[i].data.ptr;
si->callback(si->sockfd, eventloop->events[i].events, si);
}
}
}
}