350行C代码实现epoll reactor模型
liebian365 2024-10-27 13:14 26 浏览 0 评论
有了epoll为什么还需要reactor?
- 响应快,不必为单个同步事件所阻塞,虽然 Reactor 本身依然是同步的;
- 编程相对简单,可以最大程度的避免复杂的多线程及同步问题,并且避免了多线程/进 程的切换开销;
- 可扩展性,可以方便的通过增加 Reactor 实例个数来充分利用 CPU 资源;
- 可复用性,reactor 框架本身与具体事件处理逻辑无关,具有很高的复用性;
reactor的思想:将对IO的处理转化为对事件的处理。
reactor包括5个部分
- reactor管理器:注册和删除事件,运行事件循环,分发事件到已注册的回调函数上
- 事件解复用器:epoll等各个平台的IO多路复用API的封装
- fd
- 事件处理器接口
- 具体的事件处理器:由用户实现事件处理器接口
reactor封装
- reactor_init
- reactor_run
- reactor_free
- accept_cb
- recv_cb
- send_cb
accept_cb里可以实现IP限制,负载均衡
reactor注意事项
- 1个fd对应一个user_data,epoll管理所有的IO
- ET模式收发数据要循环,一次recv BUF_LEN数据,直到返回-1或recv_len等于BUF_LEN。send返回长度如果小于指定大小,则需要注册写事件
- 为了避免读写数据时阻塞,socket需要设置成非阻塞的
- ocket是否阻塞影响的API:send,recv,connect,accpet
reactor模型
单reactor
多reactor(one loop one thread)(可以有一个reactor专门监听连接)
多进程reactor:多个子进程监听同一个端口,通过共享内存决定哪个进程来accept,nginx在用户态解决惊群问题
thread per request
单线程模型
多线程模型(单reactor)
多线程模型(多reactor)
单线程reactor实现
一个数据块有1024个fd,数据块之间用链表连接。每次fd超过上限时分配1个数据块然后添加到链表尾部。
相关视频推荐
epoll的原理与实现,epoll比select/poll强在哪里?
需要C/C++ Linux服务器架构师学习资料加qun812855908获取(资料包括C/C++,Linux,golang技术,Nginx,ZeroMQ,MySQL,Redis,fastdfs,MongoDB,ZK,流媒体,CDN,P2P,K8S,Docker,TCP/IP,协程,DPDK,ffmpeg等),免费分享
代码实现:
#include <sys/epoll.h>
#include <sys/socket.h>
#include <stdio.h>
#include <netinet/in.h> // sockaddr_in
#include <netinet/ip.h>
#include <arpa/inet.h> // inet_addr
#include <errno.h>
#include <string.h>
#include <stdlib.h> // atoi
#include <unistd.h>
#include <fcntl.h> // fcntl
#define MAX_EPOLL_EVENT 1024
#define BUF_SIZE 1024
#define NO_EVENT 0
#define ACCEPT_EVENT 1
#define READ_EVENT 2
#define WRITE_EVENT 3
#define exitif(s, err_str) do { \
if(s) { \
printf("%s: %s(code:%d)\n", err_str, strerror(errno), errno); \
} \
} while(0);
typedef int (*EVENT_CALLBACK)(int fd, int events, void* args);
typedef struct _user_data {
int fd;
uint32_t events;
void* args;
EVENT_CALLBACK accept_cb;
EVENT_CALLBACK read_cb;
EVENT_CALLBACK write_cb;
char send_buf[BUF_SIZE];
char recv_buf[BUF_SIZE];
int send_buf_len;
int recv_buf_len;
} user_data;
// 1个用户数据块有1024个用户数据
typedef struct _user_data_block {
struct _user_data_block* next;
user_data* user_data_array;
} user_data_block;
typedef struct _reactor {
short listenfd;
int epfd;
user_data_block* head; // 用户数据块链表,head指向第1个数据快
} reactor;
void reactor_set_event(/* reactor* r, */ int fd, int event_type, void* args);
void reactor_del_event(int fd, int event_type, void* args);
int set_fd_nonblock(int fd) {
return fcntl(fd, F_SETFL, fcntl(fd, F_GETFL) | O_NONBLOCK);
}
int init_server(short port) {
int fd = socket(AF_INET, SOCK_STREAM, 0);
exitif(-1 == fd, "socket");
int ret = set_fd_nonblock(fd);
exitif(-1 == ret, "set_fd_nonblock");
struct sockaddr_in local_addr;
int addr_len = sizeof(struct sockaddr_in);
memset(&local_addr, 0, addr_len);
local_addr.sin_family = AF_INET;
local_addr.sin_addr.s_addr = htonl(INADDR_ANY);
local_addr.sin_port = htons(port);
ret = bind(fd, (struct sockaddr *)&local_addr, addr_len);
exitif(-1 == ret, "bind");
ret = listen(fd, 5);
exitif(-1 == ret, "listen");
return fd;
}
// 获取第fd / MAX_EPOLL_EVENT个数据块的第fd % MAX_EPOLL_EVENT个user_data
user_data* reactor_user_data(reactor* r, int fd) {
if (!r || !r->head) {
return NULL;
}
user_data_block* block = r->head;
user_data_block* prev_block = NULL;
int index = fd / MAX_EPOLL_EVENT;
while (index >= 0) {
if (!block) {
block = calloc(sizeof(user_data_block), 1);
block->next = NULL;
block->user_data_array = calloc(sizeof(user_data), MAX_EPOLL_EVENT);
if (NULL == block->user_data_array) {
free(block);
block = NULL;
close(r->epfd);
exitif(1, "NULL == block->user_data_array");
}
if (prev_block) {
prev_block->next = block;
}
}
prev_block = block;
block = block->next;
--index;
}
return &prev_block->user_data_array[fd % MAX_EPOLL_EVENT];
}
// 没有returngcc为什么没有报错?
reactor* reactor_create() {
reactor* r = calloc(sizeof(reactor), 1);
r->epfd = epoll_create(1);
r->head = calloc(sizeof(user_data_block), 1);
if (NULL == r->head) {
close(r->epfd);
exitif(1, "NULL == r->head");
}
r->head->next = NULL;
r->head->user_data_array = calloc(sizeof(user_data), MAX_EPOLL_EVENT);
if (NULL == r->head->user_data_array) {
free(r->head);
r->head = NULL;
close(r->epfd);
exitif(1, "NULL == r->head->user_data_array");
}
return r;
}
int reactor_accept_cb(int fd, int events, void* args) {
printf("reactor_accept_cb\n");
struct sockaddr_in peer_addr;
memset(&peer_addr, 0, sizeof(struct sockaddr_in));
socklen_t peer_addr_len = sizeof(peer_addr);
// ADDR_LEN要设为sockaddr_in的实际大小,否则会出现端口为0的情况
int clientfd = accept(fd, (struct sockaddr *)&peer_addr, (socklen_t *)&peer_addr_len);
exitif(-1 == clientfd, "accept");
int ret = set_fd_nonblock(clientfd);
exitif(-1 == ret, "set_fd_nonblock");
printf("new connection from fd: %d, address:%s:%d\n\n", clientfd, inet_ntoa(peer_addr.sin_addr), ntohs(peer_addr.sin_port));
reactor_set_event(clientfd, READ_EVENT, args);
return 0;
}
int reactor_read_cb(int fd, int events, void* args) {
printf("reactor_read_cb\n");
reactor* r = args;
user_data* ud = reactor_user_data(r, fd);
// TODO: socket是非阻塞的,需要循环接收
char *recv_buf = ud->recv_buf;
int n = recv(fd, recv_buf, BUF_SIZE, 0);
if (n < 0)
{
printf("recv errno: %s\n", strerror(errno));
return -1;
}
else if (n == 0)
{
printf("client closed, fd: %d\n", fd);
reactor_del_event(fd, 0, args);
close(fd);
// free(ud);
}
else
{
recv_buf[n] = '\0';
printf("recv from fd %d, msg: %s", fd, recv_buf);
char *send_buf = ud->send_buf;
ud->recv_buf_len = n;
ud->send_buf_len = n;
memcpy(send_buf, recv_buf, n);
reactor_set_event(fd, WRITE_EVENT, args);
}
return 0;
}
int reactor_write_cb(int fd, int events, void* args) {
printf("reactor_write_cb\n");
reactor* r = args;
user_data* ud = reactor_user_data(r, fd);
int send_len = ud->send_buf_len;
char* send_buf = ud->send_buf;
send_buf[send_len] = '\0';
int nsend = send(fd, send_buf, send_len, 0);
if (nsend < 0) {
printf("send errno: %s\n", strerror(errno));
return -1;
}
else if (nsend == 0) {
printf("send, client closed, fd: %d\n", fd);
reactor_del_event(fd, 0, args);
close(fd);
}
else {
printf("send to fd %d, msg:%s\n", fd, send_buf);
reactor_set_event(fd, READ_EVENT, args);
}
return 0;
}
// 这里用来设置事件,回调函数
void reactor_set_event(int fd, int event_type, void* args) {
struct epoll_event watch_event = {0};
reactor* r = args;
user_data* ud = reactor_user_data(r, fd);
ud->fd = fd;
ud->args = args;
if (ACCEPT_EVENT == event_type) {
watch_event.events = EPOLLIN;
ud->accept_cb = reactor_accept_cb;
}
else if (READ_EVENT == event_type) {
watch_event.events = EPOLLIN;
ud->read_cb = reactor_read_cb;
}
else if (WRITE_EVENT == event_type) {
watch_event.events = EPOLLOUT;
ud->write_cb = reactor_write_cb;
}
watch_event.data.ptr = ud;
if (NO_EVENT == ud->events) {
int ret = epoll_ctl(r->epfd, EPOLL_CTL_ADD, fd, &watch_event);
exitif(-1 == ret, "EPOLL_CTL_ADD");
}
else {
int ret = epoll_ctl(r->epfd, EPOLL_CTL_MOD, fd, &watch_event);
exitif(-1 == ret, "EPOLL_CTL_MOD");
}
ud->events = event_type;
}
void reactor_del_event(int fd, int event_type, void* args) {
struct epoll_event watch_event = {0};
reactor* r = args;
// fix bug: client退出没有重置events导致 EPOLL_CTL_MOD: No such file or directory(code:2)
user_data* ud = reactor_user_data(r, fd);
ud->events = NO_EVENT;
int ret = epoll_ctl(r->epfd, EPOLL_CTL_DEL, fd, &watch_event);
exitif(-1 == ret, "epoll_ctl");
}
void reactor_create_server(short port, void* args) {
reactor* r = args;
r->listenfd = init_server(port);
reactor_set_event(r->listenfd, ACCEPT_EVENT, r);
}
void reactor_loop(void* args) {
reactor* r = args;
// 就绪事件列表
struct epoll_event ready_events[MAX_EPOLL_EVENT] = {0};
while (1) {
int nready = epoll_wait(r->epfd, ready_events, MAX_EPOLL_EVENT, 3000);
if (nready < 0) {
if (EINTR == errno) {
continue;
}
exitif(-1 == nready, "epoll_wait");
}
// 就绪事件对应于events数组的下标[0, nready - 1]
int i;
for (i = 0; i < nready; ++i) {
user_data* ud = (user_data *)ready_events[i].data.ptr;
int fd = ud->fd;
if (ACCEPT_EVENT == ud->events) {
ud->accept_cb(fd, 0, r);
}
else {
// 多个cb,用多个if
if (EPOLLIN & ready_events[i].events) {
ud->read_cb(fd, 0, r);
}
if (EPOLLOUT & ready_events[i].events) {
// TODO: ET模式需要循环发送
ud->write_cb(fd, 0, r);
}
}
}
}
}
void reactor_free(void* args) {
reactor* r = args;
user_data_block* cur;
while (r->head) {
cur = r->head;
r->head = r->head->next;
if (cur) {
if (cur->user_data_array)
{
free(cur->user_data_array);
cur->user_data_array = NULL;
}
free(cur);
cur = NULL;
}
}
close(r->epfd);
close(r->listenfd);
}
int main(int argc, char *argv[]) {
if (argc < 2) {
printf("usage: %s port\n", argv[0]);
return -1;
}
// 创建reactor
reactor* r = reactor_create();
// 创建server
reactor_create_server(atoi(argv[1]), r);
// 事件循环
reactor_loop(r);
// 销毁reactor
reactor_free(r);
return 0;
}
代码理解
- user_data 每个fd对应1个item,存储fd相关事件,回调函数,读写缓冲区
- user_data_block 链表节点,每个结点有1个数据块,包含1024个fd item
- reactor 管理所有fd,包含数据块链表的头节点
- reactor_create 初始化reactor结构体
- reactor_accept_cb 接受连接,设置读事件
- reactor_read_cb 接收数据,设置写数据(回发)
- reactor_write_cb 发送数据,设置读事件
- reactor_create_server 创建监听fd,并添加到epoll
- reactor_loop 事件循环,根据事件类型调用回调函数
- reactor_free 释放reactor内存,关闭socket
运行截图
相关推荐
- 深度解密epoll 如何工作的?(epoll基本处理流程)
-
epoll...
- 大乐透第19082期:头奖开出7注1000万分落六地 奖池41亿元
-
2019年7月17日晚开奖的体彩超级大乐透第19082期开奖号码为:前区06、18、20、21、31,后区03、04。本期大乐透前区号码五区比为1:0:3:0:1,二区和四区号码没有给出。当期前区和值...
- 【开奖】4月27日周六:福彩、体彩(2021年4月27日体彩开奖结果)
-
4月27日开奖福彩3D第2019110期:61222选5第2019110期:0812202122排列3第19110期:303排列5第19110期:30305大乐透第19047期:0304...
- “红狒狒”落户哈尔滨铁路局(哈尔滨铁路红肠)
-
这几天,“红人”“红狒狒”在牡丹江机务段可引起了不小的轰动,众粉丝争相与其拍照留念,在该段人气爆棚!“红狒狒”到底何许人也?“红狒狒”,中文名:和谐3D型电力机车;绰号:红狒狒、番茄;制造商:大连机...
- 2D、3D、2.5D,做游戏还是搞噱头?玩家都晕了
-
前言游戏类型就像某种潮流,一种流行罢,另一种接棒成为主流。前两年的新作大多以“开放世界”为标签,在追求纯沙盒的过程中打造出一些细致的分类,比如说“类GTA沙盒”。诚然,纯碎的沙盒游戏并不多见,业内只有...
- 《战神4》PC版宣传片发布 GTX 1070即可60帧畅玩
-
在今年10月的时候索尼PlayStation官方正式宣布圣莫尼卡2018年的《战神4》将于2022年1月14日推出PC版本,官方在今天公布了一段PC版宣传片,并且公开了游戏的配置需求。下面让我们一起来...
- 男星深情好丈夫形象崩塌,半夜搂美女坐大腿,举止亲密
-
近日,于晓光被拍到深夜在酒吧玩,结束后与一名女子一起上车离开。上车后,女子直接坐在了他腿上,他也顺势搂着美女,美女满脸笑容地坐在他腿上玩手机离开。可能有人会好奇,于晓光是谁呢?于晓光是韩国艺人秋瓷炫的...
- d3d12dll丢失怎么修复?d3d12dll加载失败怎么解决?
-
d3d12.dll丢失怎么修复?d3d12.dll加载失败怎么解决?很多朋友想要运行游戏的时候都会遇到这个问题,这种情况该怎么办呢?今天系统之家小编给朋友们讲讲具体的解决方法,操作其实还蛮简单的。...
- 许多玩家反馈《生化4RE》PC一直崩溃 无法进入游戏
-
今日(3月24日),卡普空《生化危机4:重制版》正式发售,然而有部分PC玩家遇到了游戏崩溃等问题。很多玩家在贴吧发帖称游戏遇到了严重的崩溃问题,且经常反复,报错代码普遍为FatalD3Derror...
- 微软正式推出适用于WSL Linux的D3D12 GPU视频加速技术
-
今天,微软正式向WindowsSubsystemforLinux(WSL)用户发布了Direct3D12GPU视频加速支持。在微软通过WSL允许在Linux下使用Open...
- 《怪物猎人:崛起》曙光系统报错“Fatal d3d error”的解决办法
-
《怪物猎人:崛起》曙光系统报错“Fatald3derror”的解决办法不少小伙伴反应《怪物猎人:崛起》DLC曙光预载以后打不开游戏,出现了Fatald3derror类似的错误代码,这类问题的解...
- Mac+双屏,前端程序员的专业配置 - Loctek 乐歌 D3D 双屏电脑显示器支架
-
做FE也有一段日子了,电脑屏幕每天在设计稿、浏览器、IDE、即时通讯工具、Terminal、邮箱之间切换。虽然mac的工作区带来了很多灵活,但是依然略显不足。于是入手支架,把公司配的电脑和显示器发挥起...
- RPC 的原理和简单使用(rpc详解)
-
RPC的概念RPC,RemoteProcedureCall,翻译成中文就是远程过程调用,是一种进程间通信方式。它允许程序调用另一个地址空间(通常是共享网络的另一台机器上)的过程或函数。在调用的...
- 大厂开源的golang微服务rpc框架 — kitex
-
提前rpc估计所有的开发同学都知道,不知道的也无所谓,毕竟我也好几年没用了,今天带大家在复习一下。RPC(RemoteProcedureCall):远程过程调用,...
- 干货!一文掌握Protobuf所有语言所有用法,快收藏
-
说实话,Protobuf这个库,让人相见时难别亦难,东风无力百花残,每次等到要用它的时候,总感觉还没有完全掌握它的用法,而实际上等去百度或者谷歌的时候,教程都是多么的凌乱不堪。学会它,最直接关系到的,...
你 发表评论:
欢迎- 一周热门
- 最近发表
- 标签列表
-
- wireshark怎么抓包 (75)
- qt sleep (64)
- cs1.6指令代码大全 (55)
- factory-method (60)
- sqlite3_bind_blob (52)
- hibernate update (63)
- c++ base64 (70)
- nc 命令 (52)
- wm_close (51)
- epollin (51)
- sqlca.sqlcode (57)
- lua ipairs (60)
- tv_usec (64)
- 命令行进入文件夹 (53)
- postgresql array (57)
- statfs函数 (57)
- .project文件 (54)
- lua require (56)
- for_each (67)
- c#工厂模式 (57)
- wxsqlite3 (66)
- dmesg -c (58)
- fopen参数 (53)
- tar -zxvf -c (55)
- 速递查询 (52)