reactor原理

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);
			}
		}
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值