在epoll中如何实现流量控制

 epoll实现的经典模式是：

for(;;)
{
     int cnt = epoll_wait(epoll_fd,events,max_event);
     deal_event(events,cnt);
}

最近处理一个需求，需要在使用epoll实现一个rpc协议的同时，需要同时支持流控，想了许久才想到以下方案：

for(;;)
{
    judge_all_socket_has_tocken(allsocket);    //判断所有的socket是否有令牌，没有则从epoll中删除，又则添加到epoll中
    
    int  event_cnt = epoll_wait(epoll_fd,events,max_event);
  
    deal_event(events,event_cnt);   //只收发令牌允许的字节数
}

该方案的核心点，在于每次epoll_wait前判断每个socket是否有令牌（令牌的分配放在另外的线程中处理），如果有才会把socket添加到epoll中，如果没有则移除。