标签:
之前总结了redis的通讯流程,基本框架,epoll的封装等等,这次介绍下
redis对于select模型的封装
//select 模型 typedef struct aeApiState { //读文件描述符集合,写文件描述符集合 fd_set rfds, wfds; /* We need to have a copy of the fd sets as it‘s not safe to reuse * FD sets after select(). */ //读写集合的副本 fd_set _rfds, _wfds; } aeApiState;
_rfds和_wfds是读写结合的副本,因为select调用后会将读写集合中未就绪的文件描述符
清除,所以每次用_rfds和_wfds传入,就不用担心原读写集合描述符被清除。
封装的基于select的初始化函数
static int aeApiCreate(aeEventLoop *eventLoop) { //开辟aeApiState空间 aeApiState *state = zmalloc(sizeof(aeApiState)); if (!state) return -1; //读写集合清零 FD_ZERO(&state->rfds); FD_ZERO(&state->wfds); eventLoop->apidata = state; return 0; }
函数将读写集合清零,并且将state回传给eventloop的apidata部分。
内存回收功能
//释放空间 static void aeApiFree(aeEventLoop *eventLoop) { zfree(eventLoop->apidata); }
封装的添加和删除事件
//select 添加事件 static int aeApiAddEvent(aeEventLoop *eventLoop, int fd, int mask) { aeApiState *state = eventLoop->apidata; if (mask & AE_READABLE) FD_SET(fd,&state->rfds); if (mask & AE_WRITABLE) FD_SET(fd,&state->wfds); return 0; } //select 删除事件 static void aeApiDelEvent(aeEventLoop *eventLoop, int fd, int mask) { aeApiState *state = eventLoop->apidata; if (mask & AE_READABLE) FD_CLR(fd,&state->rfds); if (mask & AE_WRITABLE) FD_CLR(fd,&state->wfds); }
添加事件函数将文件描述根据mask是读事件还是写事件放入不同的set
删除事件根据文件描述符mask是读事件还是写事件从不同的set中清除
下面是核心功能,事件派发
//select 触发事件 static int aeApiPoll(aeEventLoop *eventLoop, struct timeval *tvp) { aeApiState *state = eventLoop->apidata; int retval, j, numevents = 0; //将select读集合的数据拷贝到_rfds memcpy(&state->_rfds,&state->rfds,sizeof(fd_set)); //将select写集合数据拷贝到_wfds memcpy(&state->_wfds,&state->wfds,sizeof(fd_set)); //从读和写的copy集合里选出就绪的文件描述符 retval = select(eventLoop->maxfd+1, &state->_rfds,&state->_wfds,NULL,tvp); //大于零表示有就绪的文件描述符 if (retval > 0) { //select的弊端所在,每次都要将所有的文件描述符轮询一遍 for (j = 0; j <= eventLoop->maxfd; j++) { int mask = 0; aeFileEvent *fe = &eventLoop->events[j]; if (fe->mask == AE_NONE) continue; //aeFileEvent 事件可读 if (fe->mask & AE_READABLE && FD_ISSET(j,&state->_rfds)) mask |= AE_READABLE; //aeFileEvent 事件可写 if (fe->mask & AE_WRITABLE && FD_ISSET(j,&state->_wfds)) mask |= AE_WRITABLE; eventLoop->fired[numevents].fd = j; eventLoop->fired[numevents].mask = mask; numevents++; } } return numevents; }
先将读写集合中的内容copy的_rfds和_wfds中,分别传入select函数中,
这样select后返回的_rfds中只有就绪的读socket
_wfds中只有就绪的写socket
通过FD_ISSET判断读写事件之后放到eventloop的fire队列里。
基本的封装就是这个样子,select模型相对容易理解
我的公众号:
标签:
原文地址:http://www.cnblogs.com/secondtonone1/p/5506428.html