libco源码阅读（七）：事件循环co_eventloop

1. 事件循环co_eventloop

2. TakeAllTimeout函数

---

1. 事件循环co_eventloop

    将文件事件和时间事件注册到时间轮之后，接下来主协程会调用事件循环函数co_eventloop，对时间轮进行轮询访问，如果超时事件发生，则调用对应事件的回调函数执行。

/*
* @param ctx ：epoll管理器
* @param pfn ：每轮事件循环的最后会调用该函数
* @param arg ：pfn的参数
*/
void co_eventloop(stCoEpoll_t *ctx, pfn_co_eventloop_t pfn, void *arg)
{
	if( !ctx->result )
	{
		ctx->result =  co_epoll_res_alloc( stCoEpoll_t::_EPOLL_SIZE );
	}
	co_epoll_res *result = ctx->result;

	for(;;)
	{
		int ret = co_epoll_wait(ctx->iEpollFd,result,stCoEpoll_t::_EPOLL_SIZE, 1 ); // 获取epoll中活跃的文件事件

		stTimeoutItemLink_t *active = (ctx->pstActiveList);
		stTimeoutItemLink_t *timeout = (ctx->pstTimeoutList);

		memset(timeout,0,sizeof(stTimeoutItemLink_t));

        /* 将活跃文件事件加入到活跃链表中 */
		for (int i = 0; i < ret; i++)
		{
            /* 获取在co_poll_inner放入epoll_event中的stTimeoutItem_t结构体 */
			stTimeoutItem_t *item = (stTimeoutItem_t*)result->events[i].data.ptr;
			if( item->pfnPrepare )
			{
				item->pfnPrepare(item,result->events[i], active); // 执行预处理回调函数
			}
			else
			{
				AddTail(active, item); // 加入到活跃链表
			}
		}

        /* 获取时间轮中的超时事件，并放入timeout中 */
		unsigned long long now = GetTickMS();
		TakeAllTimeout(ctx->pTimeout, now, timeout);

        /* 将timeout中的超时事件加入到活跃链表中active */
		stTimeoutItem_t *lp = timeout->head;
		while (lp)
		{
			//printf("raise timeout %p\n",lp);
			lp->bTimeout = true;
			lp = lp->pNext;
		}
		Join<stTimeoutItem_t, stTimeoutItemLink_t>(active, timeout);

        /* 遍历所有活跃链表active，然后执行对应的回调函数 */
		lp = active->head;
		while (lp)
		{
			PopHead<stTimeoutItem_t, stTimeoutItemLink_t>(active); // 把链表的头节点弹出
			if (lp->pfnProcess)
			{
				lp->pfnProcess(lp); // 执行事件的回调函数
			}

			lp = active->head;
		}
        /* 每次事件循环结束以后执行该函数, 用于终止协程 */
		if (pfn)
		{
			if (-1 == pfn(arg))
			{
				break;
			}
		}
	}
}

2. TakeAllTimeout函数

    该函数的作用是从时间轮中取出根据目前时间来说已经超时的事件，并插入到timeout链表中。

inline void TakeAllTimeout(stTimeout_t *apTimeout, unsigned long long allNow, stTimeoutItemLink_t *apResult)
{
	if(apTimeout->ullStart == 0) // 第一次调用是设置初始时间
	{
		apTimeout->ullStart = allNow;
		apTimeout->llStartIdx = 0;
	}

    /* 当前时间小于初始时间时返回 */
	if(allNow < apTimeout->ullStart)
	{
		return ;
	}
    
    /* 求一个取出事件的有效区间 */
	int cnt = allNow - apTimeout->ullStart + 1;
	if( cnt > apTimeout->iItemSize )
	{
		cnt = apTimeout->iItemSize;
	}
	if( cnt < 0 )
	{
		return;
	}

    /* 把上面求的有效区间过一遍，某一项存在数据的话插入到超时链表中 */
	for (int i = 0; i < cnt; i++)
	{
		int idx = (apTimeout->llStartIdx + i) % apTimeout->iItemSize;
		Join<stTimeoutItem_t, stTimeoutItemLink_t>(apResult, apTimeout->pItems + idx);
	}

    /* 更新时间轮 */
	apTimeout->ullStart = allNow;
	apTimeout->llStartIdx += cnt - 1;
}