微信后台 phxrpc (v0.8) 之 Timer(二)

一.system_clock和steady_clock比较

下面一段摘录
system_clock:就类似Windows系统右下角那个时钟,是系统时间。明显那个时钟是可以乱设置的。明明是早上10点,却可以设置成下午3点。
steady_clock:则针对system_clock可以随意设置这个缺陷而提出来的,他表示时钟是不能设置的。
steady_clock的实现是使用monotonic时间,而monotonic时间一般是从boot启动后开始计数的。明显这不能获取日历时间(年月日时分秒)。
那么steady_clock有什么用途呢?时间比较!并且是不受用户调整系统时钟影响的时间比较。简单的例子如下:

auto begin = std::chrono::steady_clock::now();
for(int i = 0; i < 10000000; ++i)
{
    // 计算...
}
auto end = std::chrono::steady_clock::now();
auto diff = (end - begin).count(); //end-begin得到一个duration类型
std::cout<<diff<<std::endl;

二.定时管理器heap

phx使用了heap(小根堆)来管理超时,一般来说,heap能做的操作是top(), push(), pop()三个操作。但是这里的一些设计要求可以在任何合法的位置删除节点,所以仅仅有pop()是不够的。

看下面一段代码:

int UThreadPoll(UThreadSocket_t & socket, int events, int * revents, int timeout_ms) {
    int ret = -1;

    // 获得当前正在执行的协程,需要这个协程帮助做点事
    socket.uthread_id = socket.scheduler->GetCurrUThread();
    // 增加超时事件
    socket.event.events = events;

    socket.scheduler->AddTimer(&socket, timeout_ms);
    // 事件加入epoll
    // 理论上讲,AddTimer将这个超时事件加入额外管理的heap定时器管理器就OK了
    // 此处为什么还需要加入epoll中进行调度?
    // 原因在于,在超时事件还没到达的时候,可能就有事件触发了,使得下面这一轮提前结束!
    epoll_ctl(socket.epoll_fd, EPOLL_CTL_ADD, socket.socket, &socket.event);

    // 将当前的协程停止,转让CPU给主协程
    // 当主协程下次收到这个超时事件的时候会将执行权还给这个协程
    // 协程下次还是从当前位置开始执行
    socket.scheduler->YieldTask();

    // 当超时任务完成,协程继续从此处执行!
    // 完成后从epoll中删除定时器
    epoll_ctl(socket.epoll_fd, EPOLL_CTL_DEL, socket.socket, &socket.event);
    socket.scheduler->RemoveTimer(socket.timer_id);
    ...
}

上面这段代码将socket加入超时事件中,同时epoll也会去监控这个socket。那么这就可能存在两种情况:
第一:如果谁正常超时事件达到,协程回到当前代码,那么下面的RemoveTimer就相当于是pop()操作,这个是很常规逻辑。
第二:如果在超时之前,epoll就监控到存在事件触发了,那么在超时之前就重新回到这个协程。这个时候节点并没有超时,所以节点可能不是小根堆的根节点。那么删除的是中间的某个节点。这时候之前的pop就不满足要求了,因为pop只能讲根节点弹出。
这里的根本问题不在删除哪个节点,而是我们怎么定位到那个节点,难道要扫描一遍?那样代价有点大。
phx在此处有一个很trick的做法,它将每个socket关联的定时器的index(在vector数组中的下标)保存在socket结构中!!!这样就可以做到O(1)定位。这个有点。。。只能说“6”。

看下具体的UThreadSocket结构:

typedef struct tagUThreadSocket {
    UThreadEpollScheduler * scheduler;
    int uthread_id;
    int epoll_fd;
    int socket;
    int connect_timeout_ms;
    int socket_timeout_ms;
    int waited_events;
    size_t timer_id;    // 保存这个socket的定时器在heap的数组中的位置(如果存在),方便在堆中的查找
    struct epoll_event event;
    void * args;
} UThreadSocket_t;

这个结构虽然很方便,但是总是觉得怪怪的。。。

OK,找到相应的节点后,那么删除操作就比较简单了。
看下具体的代码:

// 移除一个计时器
void Timer :: RemoveTimer(const size_t timer_id) {
    if (timer_id == 0) {
        return;
    }
    size_t now_idx = timer_id - 1;
    if (now_idx >= timer_heap_.size()) {
        return;
    }

    TimerObj obj = timer_heap_[now_idx];
    UThreadSocketSetTimerID(*obj.socket_, 0);
    // 当前与最后一个进行交换
    // 然后删除最后一个元素
    // 最后需要调整堆(up or down)
    std::swap(timer_heap_[timer_heap_.size() - 1], timer_heap_[now_idx]);
    timer_heap_.pop_back();

    if (timer_heap_.empty()) {
        return;
    }

    // 下面执行up 或者 down逻辑
    // 这里需要将移除的节点和最后的节点比较大小
    // 从而里判断是需要heap_up还是heap_down
    if (timer_heap_[now_idx] < obj) {
        heap_up(now_idx + 1);
    } else if (timer_heap_[now_idx] == obj) {
        UThreadSocketSetTimerID(*timer_heap_[now_idx].socket_, now_idx + 1);
    } else {
        heap_down(now_idx);
    } 
}

在heap_up和heap_down函数中,有一行类似如下:

UThreadSocketSetTimerID(*timer_heap_[now_idx].socket_, now_idx + 1);

这个就是每次调整完节点需要在socket结构中重新设置关联的timer在vector中的位置,Orz…

其他的正常的堆的操作就没什么好说的了。

阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/shanshanpt/article/details/55253379
个人分类: RPC C/Cpp
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭