(1)EventLoop启动 EventLoop初始化和启动
// 启动反应堆模型
int eventLoopRun(struct EventLoop* evLoop) {
assert(evLoop != NULL);
// 取出事件分发和检测模型
struct Dispatcher* dispatcher = evLoop->dispatcher;
// 比较线程ID是否正常
if(evLoop->threadID != pthread_self()) {
return -1;
}
// 循环进行事件处理
while(!evLoop->isQuit) {
dispatcher->dispatch(evLoop,2); // 超时时长 2s
// 已续写
eventLoopProcessTask(evLoop);
}
return 0;
}
多加一句eventLoopProcessTask(evLoop); 理由在后文提到!
(2)添加任务到任务队列中(在EventLoop的任务队列中添加新任务)
// 添加任务到任务队列
int eventLoopAddTask(struct EventLoop* evLoop,struct Channel* channel,int type) {
// 加锁,保护共享资源
pthread_mutex_lock(&evLoop->mutex);
// 创建新节点,后添加到任务队列中去
struct ChannelElement* node = (struct ChannelElement*)malloc(sizeof(struct ChannelElement));
node->channel = channel;
node->type = type;
node->next = NULL;
// 链表为空
if(evLoop->head == NULL) {
evLoop->head = evLoop->tail = node;
}else {
evLoop->tail->next = node; // 添加
evLoop->tail = node; // 后移
}
pthread_mutex_unlock(&evLoop->mutex);
if(evLoop->threadID == pthread_self()) {
// 当前子线程
eventLoopProcessTask(evLoop);
}else{
// 主线程 -- 告诉子线程处理任务队列中的任务
// 1.子线程在工作 2.子线程被阻塞了:select、poll、epoll
taskWakeup(evLoop);
}
return 0;
}
小细节:假设说添加任务的是主线程,那么程序就会执行taskWakeup这个函数,主线程执行这个函数,对于子线程来说有两种情况:
第一种情况,它正在干活,对于子线程没有影响,充其量就是它检测的那个集合里边多出来了一个被激活的文件描述符。
第二种情况,如果说此时子线程被select、poll、或epoll_wait阻塞了,调用taskWakeup可以解除其阻塞。如果解除阻塞了,我们希望子线程干什么事情呢?
taskWakeup函数的调用和影响
// 处理任务队列中的任务
int eventLoopProcessTask(struct EventLoop* evLoop);
// 处理dispatcher中的任务
int eventLoopAdd(struct EventLoop* evLoop,struct Channel* channel);
int eventLoopRemove(struct EventLoop* evLoop,struct Channel* channel);
int eventLoopModify(struct EventLoop* evLoop,struct Channel* channel);
eventLoopProcessTask函数的作用:它处理队列中的任务,需要遍历链表并根据type进行对应处理。
// 处理任务队列中的任务
int eventLoopProcessTask(struct EventLoop* evLoop) {
pthread_mutex_lock(&evLoop->mutex);
// 取出头节点
struct ChannelElement* head = evLoop->head;
while (head!=NULL) {
struct Channel* channel = head->channel;
if(head->type == ADD) {
// 添加
eventLoopAdd(evLoop,channel);
}
else if(head->type == DELETE) {
// 删除
eventLoopRemove(evLoop,channel);
}
else if(head->type == MODIFY) {
// 修改
eventLoopModify(evLoop,channel);
}
struct ChannelElement* tmp = head;
head = head->next;
// 释放节点
free(tmp);
}
evLoop->head = evLoop->tail = NULL;
pthread_mutex_unlock(&evLoop->mutex);
return 0;
}
注意事项
// 处理dispatcher中的任务
int eventLoopAdd(struct EventLoop* evLoop,struct Channel* channel) {
int fd = channel->fd;
struct ChannelMap* channelMap = evLoop->channelMap;
if(fd >= channelMap->size) {
// 没有足够的空间存储键值对 fd->channel ==> 扩容
if(!makeMapRoom(channelMap,fd,sizeof(struct Channel*))) {
return -1;
}
}
// 找到fd对应的数组元素位置,并存储
if(channelMap->list[fd] == NULL) {
channelMap->list[fd] = channel;
evLoop->dispatcher->add(channel,evLoop);
}
return 0;
}
int eventLoopRemove(struct EventLoop* evLoop,struct Channel* channel) {
int fd = channel->fd;
struct ChannelMap* channelMap = evLoop->channelMap;
if(fd >= channelMap->size) {
return -1;
}
int ret = evLoop->dispatcher->remove(channel,evLoop);
return ret;
}
int eventLoopModify(struct EventLoop* evLoop,struct Channel* channel) {
int fd = channel->fd;
struct ChannelMap* channelMap = evLoop->channelMap;
if(fd >= channelMap->size || channelMap->list[fd] == NULL) {
return -1;
}
int ret = evLoop->dispatcher->modify(channel,evLoop);
return ret;
}
总结