network: poller
libmill是一个基于c语言开发的go风格协程库，实现了go风格的协程操作、chan操作、非阻塞的网络io操作等等，是一个不错的linux平台下c风格协程库实现。如果你想从0到1的快速了解如何开发一个协程库，libmill将是一个非常不错的案例；如果你想更深入地了解go，libmill里面也借鉴了go的一些设计思想；或者你想在生产环境中使用，开发者也提供了一个更健壮的版本libdill。
poller
libmill里面网络io是基于io多路复用实现的，在linux平台下面基于epoll实现，在其他平台下面基于对应的io多路复用系统调用实现，这里只关注linux平台下的实现细节。
poller.h
void mill_poller_init(void);
// poller也实现了libmill.h中定义的mill_wait()和mill_fdwait()
// 等待至少一个coroutine恢复执行
// 如果block=0，轮询事件时没有事件就绪则立即返回
// 如果block=1，轮询事件时没有事件就绪则阻塞，直到至少有一个事件就绪
void mill_wait(int block);
// 子进程中调用该函数将创建一个全新的pollset，与父进程的pollset脱离关系
void mill_poller_postfork(void);poller.c
/* Forward declarations for the functions implemented by specific poller
mechanisms (poll, epoll, kqueue). */
void mill_poller_init(void);
static void mill_poller_add(int fd, int events);
static void mill_poller_rm(struct mill_cr *cr);
static void mill_poller_clean(int fd);
static int mill_poller_wait(int timeout);
// poller是否已经被初始化过，1是，0否
static int mill_poller_initialised = 0;
// 检查poller是否已初始化，没有初始化则初始化，已经初始化不处理
#define check_poller_initialised() \
do {\
if(mill_slow(!mill_poller_initialised)) {\
mill_poller_init();\
mill_assert(errno == 0);\
mill_main.fd = -1;\
mill_main.timer.expiry = -1;\
mill_poller_initialised = 1;\
}\
} while(0)
// 挂起当前运行中的coroutine一定的时间
void mill_msleep_(int64_t deadline, const char *current) {
mill_fdwait_(-1, 0, deadline, current);
}
//
static void mill_poller_callback(struct mill_timer *timer) {
struct mill_cr *cr = mill_cont(timer, struct mill_cr, timer);
mill_resume(cr, -1);
if (cr->fd != -1)
mill_poller_rm(cr);
}
// 等待fd上的events就绪或者直到超时
int mill_fdwait_(int fd, int events, int64_t deadline, const char *current) {
check_poller_initialised();
// 如果指定了deadline，则添加一个定时器，超时回调mill_poller_callback唤醒定时器关联的协程
if(deadline >= 0)
mill_timer_add(&mill_running->timer, deadline, mill_poller_callback);
// 如果指定了fd，则注册coroutine对该fd上的io事件监听
if(fd >= 0)
mill_poller_add(fd, events);
// 执行了实际的wait，mill_suspend挂起当前协程切换到其他ready状态的协程
mill_running->state = fd < 0 ? MILL_MSLEEP : MILL_FDWAIT;
mill_running->fd = fd;
mill_running->events = events;
mill_set_current(&mill_running->debug, current);
// 当前协程要等待其他协程调用mill_resume来唤醒当前协程才会从这里继续向下执行
// mill_suspend返回值为mill_resume(struct mill_cr *cr, int result)中的参数result
// 谁来唤醒当前协程呢？epfd事件轮询的时候有fd事件就绪则会唤醒等待在该fd上的协程！
int rc = mill_suspend();
/* Handle file descriptor events. */
if(rc >= 0) {
mill_assert(!mill_timer_enabled(&mill_running->timer));
return rc;
}
/* Handle the timeout. */
mill_assert(mill_running->fd == -1);
return 0;
}
// 将fd从epfd上取消注册
void mill_fdclean_(int fd) {
check_poller_initialised();
mill_poller_clean(fd);
}
// 事件轮询，block=1表示阻塞直到有事件到达（这里的事件包括定时器超时事件、io就绪事件）
void mill_wait(int block) {
check_poller_initialised();
while(1) {
// 计算下次轮询的超时时间
int timeout = block ? mill_timer_next() : 0;
// 检查fd上的io事件
int fd_fired = mill_poller_wait(timeout);
// 检查定时器超时时间
int timer_fired = mill_timer_fire();
// 非阻塞情况下不重试
if(!block || fd_fired || timer_fired)
break;
/* If timeout was hit but there were no expired timers do the poll
again. This should not happen in theory but let's be ready for the
case when the system timers are not precise. */
}
}
// libmill在linux平台下基于epoll实现事件轮询
#include "epoll.inc"epoll.inc
Last updated
Was this helpful?