Tars-Cpp 协程实现分析

/** * @biref 执行环境上下文 */typedef void*   fcontext_t;
/** * @biref 事件参数包装 */
struct transfer_t {    fcontext_t     fctx; // 来源的执行上下文。来源的上下文指的是从什么位置跳转过来的    void*      data; // 接口传入的自定义的指针};
/** * @biref 初始化执行环境上下文 * @param sp 栈空间地址 * @param size 栈空间的大小 * @param fn 入口函数 * @return 返回初始化完成后的执行环境上下文 */extern "C" fcontext_t make_fcontext(void * stack, std::size_t stack_size, void (* fn)( transfer_t));
/** * @biref 跳转到目标上下文 * @param to 目标上下文 * @param vp 目标上下文的附加参数，会设置为transfer_t里的data成员 * @return 跳转来源 */extern "C" transfer_t jump_fcontext(fcontext_t const to, void * vp);

void TC_CoroutineInfo::registerFunc(const std::functionvoid ()>& callback){    _callback           = callback;    _init_func.coroFunc = TC_CoroutineInfo::corotineProc;    _init_func.args     = this;
    fcontext_t ctx      = make_fcontext(_stack_ctx.sp, _stack_ctx.size,                                TC_CoroutineInfo::corotineEntry); // 创建协程    transfer_t tf       = jump_fcontext(ctx, this); // context 切换
    //实际的ctx    this->setCtx(tf.fctx);}


void TC_CoroutineInfo::corotineEntry(transfer_t tf){    TC_CoroutineInfo * coro = static_cast(tf.data); // this    auto    func  = coro->_init_func.coroFunc;    void*   args = coro->_init_func.args;
    transfer_t t = jump_fcontext(tf.fctx, NULL);
    //拿到自己的协程堆栈, 当前协程结束以后, 好跳转到main    coro->_scheduler->setMainCtx(t.fctx);
    //再跳转到具体函数    func(args, t);}

void TC_CoroutineScheduler::switchCoro(TC_CoroutineInfo *to){    //跳转到to协程    _currentCoro = to;
    transfer_t t = jump_fcontext(to->getCtx(), NULL);
    //并保存协程堆栈    to->setCtx(t.fctx);}

    /**     * 协程的状态信息     */    enum CORO_STATUS    {        CORO_FREE       = 0,        CORO_ACTIVE     = 1,        CORO_AVAIL      = 2,        CORO_INACTIVE   = 3,        CORO_TIMEOUT    = 4     };

void TC_CoroutineScheduler::init(){    ... ....
    createCoroutineInfo(_poolSize); // _all_coro = new TC_CoroutineInfo*[_poolSize+1];
    TC_CoroutineInfo::CoroutineHeadInit(&_active);    TC_CoroutineInfo::CoroutineHeadInit(&_avail);    TC_CoroutineInfo::CoroutineHeadInit(&_inactive);    TC_CoroutineInfo::CoroutineHeadInit(&_timeout);    TC_CoroutineInfo::CoroutineHeadInit(&_free);
    int iSucc = 0;    for(size_t i = 0; i     {        //iId=0不使用, 给mainCoro使用!!!!        uint32_t iId = generateId();        stack_context s_ctx = stack_traits::allocate(_stackSize); // 分配协程栈内存        TC_CoroutineInfo *coro = new TC_CoroutineInfo(this, iId, s_ctx);        _all_coro[iId] = coro;        TC_CoroutineInfo::CoroutineAddTail(coro, &_free);        ++iSucc;    }    _currentSize = iSucc;
    _mainCoro.setUid(0);    _mainCoro.setStatus(TC_CoroutineInfo::CORO_FREE);
    _currentCoro = &_mainCoro;}

void TC_CoroutineScheduler::run(){    ... ...
    while(!_epoller->isTerminate())    {        if(_activeCoroQueue.empty() && TC_CoroutineInfo::CoroutineHeadEmpty(&_avail) && TC_CoroutineInfo::CoroutineHeadEmpty(&_active))        {            _epoller->done(1000); // epoll_wait(..., 1000ms) 先处理epoll的网络事件        }
        //唤醒需要激活的协程        wakeup();
        //唤醒sleep的协程        wakeupbytimeout();
        //唤醒yield的协程        wakeupbyself();
        int iLoop = 100;        //执行active协程, 每次执行100个, 避免占满cpu        while(iLoop > 0 && !TC_CoroutineInfo::CoroutineHeadEmpty(&_active))        {            TC_CoroutineInfo *coro = _active._next;            switchCoro(coro);            --iLoop;        }
        //执行available协程, 每次执行1个        if(!TC_CoroutineInfo::CoroutineHeadEmpty(&_avail))        {            TC_CoroutineInfo *coro = _avail._next;            switchCoro(coro);        }    }
    ... ...}