之前总结了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模型相对容易理解