study/src/timer.cc at master · php-extension-research/study · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#include "timer.h"

using study::Timer;
using study::TimerManager;

TimerManager study::timer_manager;

const uint64_t Timer::MILLI_SECOND = 1;
const uint64_t Timer::SECOND = 1000;

uint64_t Timer::get_current_ms()
{
    struct timeval tv;
    gettimeofday(&tv, NULL);
    return tv.tv_sec * 1000 + tv.tv_usec / 1000;
}

Timer::Timer(uint64_t _timeout, timer_func_t _callback, void *_private_data, TimerManager *_timer_manager):
    timeout(_timeout), callback(_callback), private_data(_private_data), timer_manager(_timer_manager)
{
    exec_msec = get_current_ms() + _timeout;
}

TimerManager::TimerManager()
{
}

TimerManager::~TimerManager()
{
}

void TimerManager::add_timer(int64_t _timeout, timer_func_t _callback, void *_private_data)
{
    Timer *timer = new Timer(_timeout, _callback, _private_data, this);
    timers.push(timer);
}

int64_t TimerManager::get_next_timeout()
{
    int64_t diff;

    if (timers.empty())
    {
        return -1;
    }
    Timer *t = timers.top();

    diff = t->exec_msec - Timer::get_current_ms();
    return diff < 0 ? 0 : diff;
}

void TimerManager::run_timers()
{
    uint64_t now = Timer::get_current_ms();
    while (true)
    {
        if (timers.empty())
        {
            break;
        }
        Timer *t = timers.top();
        if (now < t->exec_msec)
        {
            break;
        }
        timers.pop();
        t->callback(t->private_data);
        delete t;
    }
}