关于协程:CC协程学习笔记丨CC实现协程及原理分析视频

42次阅读

共计 12216 个字符,预计需要花费 31 分钟才能阅读完成。

协程,又称微线程,纤程。英文名 Coroutine。

协程的概念很早就提出来了,但直到最近几年才在某些语言(如 Lua)中失去广泛应用。

子程序,或者称为函数,在所有语言中都是层级调用,比方 A 调用 B,B 在执行过程中又调用了 C,C 执行结束返回,B 执行结束返回,最初是 A 执行结束。

所以子程序调用是通过栈实现的,一个线程就是执行一个子程序。

子程序调用总是一个入口,一次返回,调用程序是明确的。而协程的调用和子程序不同。

协程看上去也是子程序,但执行过程中,在子程序外部可中断,而后转而执行别的子程序,在适当的时候再返回来接着执行。

留神,在一个子程序中中断,去执行其余子程序,不是函数调用,有点相似 CPU 的中断。比方子程序 A、B:def A():

print ‘1’
print ‘2’
print ‘3’
def B():
print ‘x’
print ‘y’
print ‘z’

假如由协程执行,在执行 A 的过程中,能够随时中断,去执行 B,B 也可能在执行过程中中断再去执行 A,后果可能是:

1
2
x
y
3
z

然而在 A 中是没有调用 B 的,所以协程的调用比函数调用了解起来要难一些。

看起来 A、B 的执行有点像多线程,但 协程的特点在于是一个线程执行,那和多线程比,协程有何劣势?

最大的劣势就是协程极高的执行效率。因为子程序切换不是线程切换,而是由程序本身管制,因而,没有线程切换的开销,和多线程比,线程数量越多,协程的性能劣势就越显著。

第二大劣势就是不须要多线程的锁机制,因为只有一个线程,也不存在同时写变量抵触,在协程中管制共享资源不加锁,只须要判断状态就好了,所以执行效率比多线程高很多。

因为协程是一个线程执行,那怎么利用多核 CPU 呢?最简略的办法是多过程 + 协程,既充分利用多核,又充分发挥协程的高效率,可取得极高的性能。

Python 对协程的反对还十分无限,用在 generator 中的 yield 能够肯定水平上实现协程。尽管反对不齐全,但曾经能够施展相当大的威力了。

来看例子:

传统的生产者 - 消费者模型是一个线程写音讯,一个线程取音讯,通过锁机制管制队列和期待,但一不小心就可能死锁。

如果改用协程,生产者生产音讯后,间接通过 yield 跳转到消费者开始执行,待消费者执行结束后,切换回生产者持续生产,效率极高:import time

def consumer():
r = ”
while True:
n = yield r
if not n:
return
print(‘[CONSUMER] Consuming %s…’ % n)
time.sleep(1)
r = ‘200 OK’
def produce(c):
c.next()
n = 0
while n < 5:
n = n + 1
print(‘[PRODUCER] Producing %s…’ % n)
r = c.send(n)
print(‘[PRODUCER] Consumer return: %s’ % r)
c.close()
if __name__==’__main__’:
c = consumer()
produce(c)

执行后果:

[PRODUCER] Producing 1…
[CONSUMER] Consuming 1…
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 2…
[CONSUMER] Consuming 2…
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 3…
[CONSUMER] Consuming 3…
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 4…
[CONSUMER] Consuming 4…
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 5…
[CONSUMER] Consuming 5…
[PRODUCER] Consumer return: 200 OK

留神到 consumer 函数是一个 generator(生成器),把一个 consumer 传入 produce 后:

  1. 首先调用 c.next()启动生成器;
  2. 而后,一旦生产了货色,通过 c.send(n)切换到 consumer 执行;
  3. consumer 通过 yield 拿到音讯,解决,又通过 yield 把后果传回;
  4. produce 拿到 consumer 解决的后果,持续生产下一条音讯;
  5. produce 决定不生产了,通过 c.close()敞开 consumer,整个过程完结。

整个流程无锁,由一个线程执行,produce 和 consumer 合作实现工作,所以称为“协程”,而非线程的抢占式多任务。

协程残缺视频链接以及文档资料 +qun720209036 获取

协程的实现与原理分析训练营

C/C++ 协程

首先须要申明的是,这里不打算花工夫来介绍什么是协程,以及协程和线程有什么不同。如果对此有任何疑难,能够自行 google。与 Python 不同,C/C++ 语言自身是不能人造反对协程的。现有的 C++ 协程库均基于两种计划:利用汇编代码管制协程上下文的切换,以及利用操作系统提供的 API 来实现协程上下文切换。典型的例如:

  • libco,Boost.context:基于汇编代码的上下文切换
  • phxrpc:基于 ucontext/Boost.context 的上下文切换
  • libmill:基于 setjump/longjump 的协程切换

一般而言,基于汇编的上下文切换要比采纳零碎调用的切换更加高效,这也是为什么 phxrpc 在应用 Boost.context 时要比应用 ucontext 性能更好的起因。对于 phxrpc 和 libmill 具体的协程实现形式,当前有工夫再具体介绍。

libco 协程的创立和切换

在介绍 coroutine 的创立之前,咱们先来相熟一下 libco 中用来示意一个 coroutine 的数据结构,即定义在 co_routine_inner.h 中的 stCoRoutine_t:

struct stCoRoutine_t
{
stCoRoutineEnv_t *env; // 协程运行环境
pfn_co_routine_t pfn; // 协程执行的逻辑函数
void *arg; // 函数参数
coctx_t ctx; // 保留协程的下文环境

char cEnableSysHook; // 是否运行零碎 hook,即非侵入式逻辑
char cIsShareStack; // 是否在共享栈模式
void *pvEnv;
stStackMem_t* stack_mem; // 协程运行时的栈空间
char* stack_sp; // 用来保留协程运行时的栈空间
unsigned int save_size;
char* save_buffer;
};

咱们临时只须要理解示意协程的最简略的几个参数,例如协程运行环境,协程的上下文环境,协程运行的函数以及运行时栈空间。前面的 stack_sp,save_size 和 save_buffer 与 libco 共享栈模式相干,无关共享栈的内容咱们后续再说

协程创立和运行

因为多个协程运行于 一个线程 外部的,因而当创立线程中的第一个协程时,须要初始化该协程所在的环境 stCoRoutineEnv_t,这个环境是线程用来治理协程的,通过该环境,线程能够得悉以后一共创立了多少个协程,以后正在运行哪一个协程,以后该当如何调度协程:

struct stCoRoutineEnv_t
{
stCoRoutine_t *pCallStack[128]; // 记录以后创立的协程
int iCallStackSize; // 记录以后一共创立了多少个协程
stCoEpoll_t *pEpoll; // 该线程的协程调度器
// 在应用共享栈模式拷贝栈内存时记录相应的 coroutine
stCoRoutine_t* pending_co;
stCoRoutine_t* occupy_co;
};

上述代码表明 libco 容许一个线程内最多创立 128 个协程,其中 pCallStack[iCallStackSize-1] 也就是栈顶的协程示意以后正在运行的协程。当调用函数 co_create 时,首先查看以后线程中的 coroutine env 构造是否创立。这里 libco 对于每个线程内的 stCoRoutineEnv_t 并没有应用 thread-local 的形式(例如 gcc 内置的 __thread,phxrpc 采纳这种形式)来治理,而是事后定义了一个大的数组,并通过对应的 PID 来获取其协程环境。:

static stCoRoutineEnv_t* g_arrCoEnvPerThread[204800]
stCoRoutineEnv_t *co_get_curr_thread_env()
{
return g_arrCoEnvPerThread[GetPid() ];
}

初始化 stCoRoutineEnv_t 时次要实现以下几步:

  1. 为 stCoRoutineEnv_t 申请空间并且进行初始化,设置协程调度器 pEpoll。
  2. 创立一个空的 coroutine,初始化其上下文环境(无关 coctx 在后文具体介绍),将其退出到该线程的协程环境中进行治理,并且设置其为 main coroutine。这个 main coroutine 用来运行该线程主逻辑。

当初始化实现协程环境之后,调用函数 co_create_env 来创立具体的协程,该函数初始化一个协程构造 stCoRoutine_t,设置该构造中的各项字段,例如运行的函数 pfn,运行时的栈地址等等。须要阐明的就是,如果应用了非共享栈模式,则须要为该协程独自申请栈空间,否则从共享栈中申请空间。栈空间示意如下:

struct stStackMem_t
{
stCoRoutine_t* occupy_co; // 应用该栈的协程
int stack_size; // 栈大小
char* stack_bp; // 栈的指针,栈从高地址向低地址增长
char* stack_buffer; // 栈底
};

应用 co_create 创立完一个协程之后,将调用 co_resume 来将该协程激活运行:

void co_resume(stCoRoutine_t *co)
{
stCoRoutineEnv_t *env = co->env;
// 获取以后正在运行的协程的构造
stCoRoutine_t *lpCurrRoutine = env->pCallStack[env->iCallStackSize – 1];
if(!co->cStart)
{
// 为将要运行的 co 安排上下文环境
coctx_make(&co->ctx,(coctx_pfn_t)CoRoutineFunc,co,0 );
co->cStart = 1;
}
env->pCallStack[env->iCallStackSize++] = co; // 设置 co 为运行的线程
co_swap(lpCurrRoutine, co);
}

函数 co_swap 的作用相似于 Unix 提供的函数 swapcontext:将以后正在运行的 coroutine 的上下文以及状态保留到构造 lpCurrRoutine 中,并且将 co 设置成为要运行的协程,从而实现协程的切换。co_swap 具体实现三项工作:

  1. 记录以后协程 curr 的运行栈的栈顶指针,通过 char c; curr_stack_sp=&c 实现,当下次切换回 curr 时,能够从该栈顶指针指向的地位持续,执行完 curr 后能够顺利开释该栈。
  2. 解决共享栈相干的操作,并且调用函数 coctx_swap 来实现上下文环境的切换。留神执行完 coctx_swap 之后,执行流程将跳到新的 coroutine 也就是 pending_co 中运行,后续的代码须要等下次切换回 curr 时才会执行。
  3. 当下次切换回 curr 时,解决共享栈相干的操作。

对应于 co_resume 函数,协程被动让出执行权则调用 co_yield 函数。co_yield 函数调用了 co_yield_env,将以后协程与以后线程中记录的其余协程进行切换:

void co_yield_env(stCoRoutineEnv_t *env)
{
stCoRoutine_t *last = env->pCallStack[env->iCallStackSize – 2];
stCoRoutine_t *curr = env->pCallStack[env->iCallStackSize – 1];
env->iCallStackSize–;
co_swap(curr, last);
}

后面咱们曾经提到过,pCallStack 栈顶所指向的即为以后正在运行的协程所对应的构造,因而该函数将 curr 取出来,并将以后正运行的协程上下文保留到该构造上,并切换到协程 last 上执行。接下来咱们以 32-bit 的零碎为例来剖析 libco 是如何实现协程运行环境的切换的。

协程上下文的创立和切换

libco 应用构造 struct coctx_t 来示意一个协程的上下文环境:

struct coctx_t
{

if defined(__i386__)

void *regs[8];

else

void *regs[14];

endif

size_t ss_size;
char *ss_sp;
};

能够看到,在 i386 的架构下,须要保留 8 个寄存器信息,以及栈指针和栈大小,到底这 8 个寄存器如何保留,又是如何应用,须要配合后续的 coctx_swap 来了解。咱们首先来回顾一下 Unix-like 零碎的 stack frame layout,如果不能了解这个,那么剩下的内容就不用看了。

联合上图,咱们须要晓得要害的几点:

  1. 函数调用栈是调用者和被调用者独特负责安排的。Caller 将其参数从右向左反向压栈,再将调用后的返回地址压栈,而后将执行流程交给 Callee。
  2. 典型的编译器会将 Callee 函数汇编成为以 push %ebp; move %ebp, %esp; sub $esp N; 这种模式结尾的汇编代码。这几句代码次要目标是为了不便 Callee 利用 ebp 来拜访调用者提供的参数以及本身的局部变量(如下图)。
  3. 当调用过程实现革除了局部变量当前,会执行 pop %ebp; ret,这样指令会跳转到 RA 也就是返回地址下面执行。这一点也是实现协程切换的要害:咱们只须要将指定协程的函数指针地址保留到 RA 中,当调用完 coctx_swap 之后,会主动跳转到该协程的函数起始地址开始运行

理解了这些,咱们就来看一下协程上下文环境的初始化函数 coctx_make:

int coctx_make(coctx_t ctx, coctx_pfn_t pfn, const void s, const void *s1 )
{
char *sp = ctx->ss_sp + ctx->ss_size – sizeof(coctx_param_t);
sp = (char*)((unsigned long)sp & -16L);
coctx_param_t param = (coctx_param_t)sp ;
param->s1 = s;
param->s2 = s1;
memset(ctx->regs, 0, sizeof(ctx->regs));
ctx->regs[kESP] = (char)(sp) – sizeof(void);
ctx->regs[kEIP] = (char*)pfn;
return 0;
}

这段代码应该比拟好了解,首先为函数 coctx_pfn_t 预留 2 个参数的栈空间并对其到 16 字节,之后将实参设置到预留的栈上空间中。最初在 ctx 构造中填入相应的,其中记录 reg[kEIP] 返回地址为函数指针 pfn,记录 reg[kESP] 为取得的栈顶指针 sp 减去一个指针长度,这个减去的空间是为返回地址 RA 预留的。当调用 coctx_swap 时,reg[kEIP] 会被放到返回地址 RA 的地位,待 coctx_swap 执行完结,天然会跳转到函数 pfn 处执行。

coctx_swap(ctx1, ctx2) 在 coctx_swap.S 中实现。这里能够看到,该函数并没有应用 push %ebp; move %ebp, %esp; sub $esp N; 结尾,因而栈空间散布中不会呈现 ebp 的地位。coctx_swap 函数次要分为两段,其首先将以后的上下文环境保留到 ctx1 构造中:

leal 4(%esp), %eax // eax = old_esp + 4
movl 4(%esp), %esp // 将 esp 的值设为 &ctx1(即 ctx1 的地址)
leal 32(%esp), %esp // esp = (char*)&ctx1 + 32
pushl %eax // ctx1->regs[EAX] = %eax
pushl %ebp // ctx1->regs[EBP] = %ebp
pushl %esi // ctx1->regs[ESI] = %esi
pushl %edi // ctx1->regs[EDI] = %edi
pushl %edx // ctx1->regs[EDX] = %edx
pushl %ecx // ctx1->regs[ECX] = %ecx
pushl %ebx // ctx1->regs[EBX] = %ebx
pushl -4(%eax) // ctx1->regs[EIP] = RA,留神:%eax-4=%old_esp

这里须要留神指令 leal 和 movl 的区别。leal 将 eax 的值设置成为 esp 的值加 4,而 movl 将 esp 的值设为 esp+4 所指向的内存上的值,也就是参数 ctx1 的地址。之后该函数将 ctx2 中记录的上下文复原到 CPU 寄存器中,并跳转到其函数地址处运行:

movl 4(%eax), %esp // 将 esp 的值设为 &ctx2(即 ctx2 的地址)
popl %eax // %eax = ctx1->regs[EIP],也就是 &pfn
popl %ebx // %ebx = ctx1->regs[EBP]
popl %ecx // %ecx = ctx1->regs[ECX]
popl %edx // %edx = ctx1->regs[EDX]
popl %edi // %edi = ctx1->regs[EDI]
popl %esi // %esi = ctx1->regs[ESI]
popl %ebp // %ebp = ctx1->regs[EBP]
popl %esp // %esp = ctx1->regs[ESP],即(char)(sp) – sizeof(void)
pushl %eax // RA = %eax = &pfn,留神此时 esp 曾经指向了新的 esp
xorl %eax, %eax // reset eax
ret

下面的代码看起来可能有些绕:

  1. 首先 line 1 将 esp 设置为参数 ctx2 的地址,后续的 popl 操作均在 ctx2 的内存空间上执行。
  2. line 2-9 将 ctx2->regs[] 中的内容复原到相应的寄存器中。还记得在后面 coctx_make 中设置了 regs[EIP] 和 regs[ESP] 吗?这里刚好就对应复原了相应的值。
  3. 当执行完 line 9 之后,esp 曾经指向了 ctx2 中新的栈顶指针,因为在 coctx_make 中预留了一个指针长度的 RA 空间,line 10 刚好将新的函数指针 &pfn 设置到该 RA 上。
  4. 最初执行 ret 指令时,函数流程将跳到 pfn 处执行。这样,整个协程上下文的切换就实现了。

如何应用 libco

咱们首先以 libco 提供的例子 example_echosvr.cpp 来介绍应用程序如何应用 libco 来编写服务端程序。在 example_echosvr.cpp 的 main 函数中,次要执行如下几步:

  1. 创立 socket,监听在本机的 1024 端口,并设置为非阻塞;
  2. 主线程应用函数 readwrite_coroutine 创立多个读写协程,调用 co_resume 启动协程运行直到其挂起。这里咱们疏忽掉无关的多过程 fork 的过程;
  3. 主线程持续创立 socket 接管协程 accpet_co,同样调用 co_resume 启动协程直到其挂起;
  4. 主线程调用函数 co_eventloop 实现事件的监听和协程的循环切换;

函数 readwrite_coroutine 在外层循环中将新创建的读写协程都退出到队列 g_readwrite 中,此时这些读写协程都没有具体与某个 socket 连贯对应,能够将队列 g_readwrite 看成一个 coroutine pool。当退出到队列中之后,调用函数 co_yield_ct 函数让出 CPU,此时控制权回到主线程。

主线程中的函数 co_eventloop 监听网络事件,将来自于客户端新进的连贯交由协程 accept_co 解决,对于 co_eventloop 如何唤醒 accept_co 的细节咱们将在后续介绍。accept_co 调用函数 accept_routine 接管新连贯,该函数的流程如下:

  1. 查看队列 g_readwrite 是否有闲暇的读写 coroutine,如果没有,调用函数 poll 将该协程退出到 Epoll 治理的定时器队列中,也就是 sleep(1000) 的作用;
  2. 调用 co_accept 来接管新连贯,如果接管连贯失败,那么调用 co_poll 将服务端的 listen_fd 退出到 Epoll 中来触发下一次连贯事件;
  3. 对于胜利的连贯,从 g_readwrite 中取出一个读写协程来负责解决读写;

再次回到函数 readwrite_coroutine 中,该函数会调用 co_poll 将新建设的连贯的 fd 退出到 Epoll 监听中,并将管制流程返回到 main 协程;当有读或者写事件产生时,Epoll 会唤醒对应的 coroutine,继续执行 read 函数以及 write 函数。

下面的过程大抵阐明了管制流程是如何在不同的协程中切换,接下来咱们介绍具体的实现细节,即如何通过 Epoll 来治理协程,以及如何对系统函数进行革新以满足 libco 的调用。

通过 Epoll 治理和唤醒协程

Epoll 监听 FD

上一章节中介绍了协程能够通过函数 co_poll 来将 fd 交由 Epoll 治理,待 Epoll 的相应的事件触发时,再切换回来执行 read 或者 write 操作,从而实现由 Epoll 治理协程的性能。co_poll 函数原型如下:

int co_poll(stCoEpoll_t *ctx, struct pollfd fds[],
nfds_t nfds, int timeout_ms)

stCoEpoll_t 是为 libco 定制的 Epoll 相干数据结构,fds 是 pollfd 构造的文件句柄,nfds 为 fds 数组的长度,最初一个参数示意定时器工夫,也就是在 timeout 毫秒之后触发解决这些文件句柄。这里能够看到,co_poll 可能同时将多个文件句柄同时退出到 Epoll 治理中。咱们先看 stCoEpoll_t 构造:

struct stCoEpoll_t
{
int iEpollFd; // Epoll 主 FD
static const int _EPOLL_SIZE = 1024 * 10; // Epoll 能够监听的句柄总数
struct stTimeout_t *pTimeout; // 工夫轮定时器
struct stTimeoutItemLink_t *pstTimeoutList; // 曾经超时的工夫
struct stTimeoutItemLink_t *pstActiveList; // 沉闷的事件
co_epoll_res *result; // Epoll 返回的事件后果
};

以 stTimeout_ 结尾的数据结构与 libco 的定时器治理无关,咱们在前面介绍。co_epoll_res 是对 Epoll 事件数据结构的封装,也就是每次触发 Epoll 事件时的返回后果,在 Unix 和 MaxOS 下,libco 将应用 Kqueue 代替 Epoll,因而这里也保留了 kevent 数据结构。

struct co_epoll_res
{
int size;
struct epoll_event *events; // for linux epoll
struct kevent *eventlist; // for Unix or MacOs kqueue
};

co_poll 理论是对函数 co_poll_inner 的封装。咱们将 co_epoll_inner 函数的构造分为高低两半段。在上半段中,调用 co_poll 的协程 CC 将其须要监听的句柄数组 fds 都退出到 Epoll 治理中,并通过函数 co_yield_env 让出 CPU;当 main 协程的事件循环 co_eventloop 中触发了 CC 对应的监听事件时,会复原 CC 的执行。此时,CC 将开始执行下半段,行将上半段增加的句柄 fds 从 epoll 中移除,清理残留的数据结构,上面的流程图简要阐明了控制流的转移过程:

有了下面的基本概念,咱们来看具体的实现细节。co_poll 首先在外部将传入的文件句柄数组 fds 转化为数据结构 stPoll_t,这一步次要是为了不便后续解决。该构造记录了 iEpollFd,ndfs,fds 数组,以及该协程须要执行的函数和参数。有两点须要阐明的是:

  1. 对于每一个 fd,为其申请一个 stPollItem_t 来治理对应 Epoll 事件以及记录回调参数。libco 在此做了一个小的优化,对于长度小于 2 的 fds 数组,间接在栈上定义相应的 stPollItem_t 数组,否则从堆中申请内存。这也是一种比拟常见的优化,毕竟从堆中申请内存比拟耗时;
  2. 函数指针 OnPollProcessEvent 封装了协程的切换过程。当传入指定的 stPollItem_t 构造时,即可唤醒对应于该构造的 coroutine,将控制权交由其执行;

co_poll 的第二步,也是最要害的一步,就是将 fd 数组全副退出到 Epoll 中进行监听。协程 CC 会将每一个 epoll_event 的 data.ptr 域设置为对应的 stPollItem_t 构造。这样当事件触发时,能够间接从对应的 ptr 中取出 stPollItem_t 构造,而后唤醒指定协程。

如果本次操作提供了 Timeout 参数,co_poll 还会将协程 CC 本次操作对应的 stPoll_t 退出到定时器队列中。这表明在 Timeout 定时触发之后,也会唤醒协程 CC 的执行。当整个上半段都实现后,co_poll 立刻调用 co_yield_env 让出 CPU,执行流程跳转回到 main 协程中。

从下面的流程图中也能够看出,当执行流程再次跳回时,表明协程 CC 增加的读写等监听事件曾经触发,即能够执行相应的读写操作了。此时 CC 首先将其在上半段中增加的监听事件从 Epoll 中删除,清理残留的数据结构,而后调用读写逻辑。

定时器实现

协程 CC 在将一组 fds 退出 Epoll 的同时,还能为其设置一个超时工夫。在超时工夫到期时,也会再次唤醒 CC 来执行。libco 应用 Timing-Wheel 来实现定时器。对于 Timing-Wheel 算法,能够参考,其劣势是 O(1) 的插入和删除复杂度,毛病是只有无限的长度,在某些场合下不能满足需要。

回过来看 stCoEpoll_t 构造,其中 pTimeout 代表工夫轮,通过函数 AllocateTimeout 初始化为一个固定大小(60 1000)的数组。依据 Timing-Wheel 的个性可知,libco 只反对最大 60s 的定时事件。而实际上,在增加定时器时,libco 要求定时工夫不超过 40s。成员 pstTimeoutList 记录在 co_eventloop 中产生超时的事件,而 pstActiveList 记录以后沉闷的事件,包含超时事件。这两个构造都将在 co_eventloop 中进行解决。

上面咱们简要剖析一下退出定时器的实现:

int AddTimeout(stTimeout_t apTimeout, stTimeoutItem_t apItem,
unsigned long long allNow )
{
if(apTimeout->ullStart == 0) // 初始化工夫轮的基准工夫
{
apTimeout->ullStart = allNow;
apTimeout->llStartIdx = 0; // 以后工夫轮指针指向数组 0
}
// 1. 以后工夫不可能小于工夫轮的基准工夫
// 2. 退出的定时器的超时工夫不能小于以后工夫
if(allNow < apTimeout->ullStart || apItem->ullExpireTime < allNow)
{
return __LINE__;
}
int diff = apItem->ullExpireTime – apTimeout->ullStart;
if(diff >= apTimeout->iItemSize) // 增加的事件不能超过工夫轮的大小
{
return __LINE__;
}
// 插入到工夫轮盘的指定地位
AddTail(apTimeout->pItems +
(apTimeout->llStartIdx + diff) % apTimeout->iItemSize, apItem );
return 0;
}

定时器的超时查看在函数 co_eventloop 中执行。

EPOLL 事件循环

main 协程通过调用函数 co_eventloop 来监听 Epoll 事件,并在相应的事件触发时切换到指定的协程执行。无关 co_eventloop 与 利用协程的交互过程在上一节的流程图中曾经比较清楚了,上面咱们次要介绍一下 co_eventloop 函数的实现:

上文中也提到,通过 epoll_wait 返回的事件都保留在 stCoEpoll_t 构造的 co_epoll_res 中。因而 co_eventloop 首先为 co_epoll_res 申请空间,之后通过一个有限循环来监听所有 coroutine 增加的所有事件:

for(;;)
{
int ret = co_epoll_wait(ctx->iEpollFd,result,stCoEpoll_t::_EPOLL_SIZE, 1);

}

对于每一个触发的事件,co_eventloop 首先通过指针域 data.ptr 取出保留的 stPollItem_t 构造,并将其增加到 pstActiveList 列表中;之后从定时器轮盘中取出所有曾经超时的事件,也将其全副增加到 pstActiveList 中,pstActiveList 中的所有事件都作为沉闷事件处理。

对于每一个沉闷事件,co_eventloop 将通过调用对应的 pfnProcess 也就是上图中的 OnPollProcessEvent 函数来切换到该事件对应的 coroutine,将流程跳转到该 coroutine 处执行。

最初 co_eventloop 在调用时也提供一个额定的参数来供调用者传入一个函数指针 pfn。该函数将会在每次循环实现之后执行;当该函数返回 -1 时,将会终止整个事件循环。用户能够利用该函数来管制 main 协程的终止或者实现一些统计需要。

正文完
 0