[libco] 协程调度

2020-12-27

libco 通过 co_resumeco_yield 交替调度协程。


1. 协程调度

设计图来源:libco 协程调度


1.1. 协程数组

pCallStack 协程数组,保存当前正在执行协程(注意:并不是所有协程)。

pCallStack[0] 是主协程,env->pCallStack[env->iCallStackSize - 1] 是当前协程。

一般情况下数组大小为 2,子协程在主协程里创建。除非在子协程里嵌套创建唤醒新的协程,这个协程数组大小才会一直被累加 env->iCallStackSize++,直到嵌套深度达到 128 才会出现堆栈溢出,这种应用场景嵌应该不常见。

1
2
3
4
5
struct stCoRoutineEnv_t {
    stCoRoutine_t *pCallStack[128]; /* 协程数组。 */
    int iCallStackSize;             /* 协程数组元素个数。 */
    ...
};

1.2. 启动协程 co_resume

1
2
3
4
5
6
7
void co_resume(stCoRoutine_t *co) {
    stCoRoutineEnv_t *env = co->env;
    stCoRoutine_t *lpCurrRoutine = env->pCallStack[env->iCallStackSize - 1];
    ...
    env->pCallStack[env->iCallStackSize++] = co;
    co_swap(lpCurrRoutine, co);
}

1.3. 挂起协程 co_yield

1
2
3
4
5
6
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);
}

1.4. 协程执行函数

1
2
3
4
5
6
7
8
9
10
static int CoRoutineFunc(stCoRoutine_t *co,void *) {
    if (co->pfn) {
        /* pfn 协程执行函数。 */
        co->pfn(co->arg);
    }
    co->cEnd = 1;
    stCoRoutineEnv_t *env = co->env;
    co_yield_env(env);
    return 0;
}

2. 参考