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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
#include "env-inl.h"
#include "memory_tracker-inl.h"
#include "timer_wrap.h"
#include "uv.h"
namespace node {
TimerWrap::TimerWrap(Environment* env, const TimerCb& fn)
: env_(env),
fn_(fn) {
uv_timer_init(env->event_loop(), &timer_);
timer_.data = this;
}
void TimerWrap::Stop() {
if (timer_.data == nullptr) return;
uv_timer_stop(&timer_);
}
void TimerWrap::Close() {
timer_.data = nullptr;
env_->CloseHandle(reinterpret_cast<uv_handle_t*>(&timer_), TimerClosedCb);
}
void TimerWrap::TimerClosedCb(uv_handle_t* handle) {
std::unique_ptr<TimerWrap> ptr(
ContainerOf(&TimerWrap::timer_,
reinterpret_cast<uv_timer_t*>(handle)));
}
void TimerWrap::Update(uint64_t interval, uint64_t repeat) {
if (timer_.data == nullptr) return;
uv_timer_start(&timer_, OnTimeout, interval, repeat);
}
void TimerWrap::Ref() {
if (timer_.data == nullptr) return;
uv_ref(reinterpret_cast<uv_handle_t*>(&timer_));
}
void TimerWrap::Unref() {
if (timer_.data == nullptr) return;
uv_unref(reinterpret_cast<uv_handle_t*>(&timer_));
}
void TimerWrap::OnTimeout(uv_timer_t* timer) {
TimerWrap* t = ContainerOf(&TimerWrap::timer_, timer);
t->fn_();
}
TimerWrapHandle::TimerWrapHandle(
Environment* env,
const TimerWrap::TimerCb& fn) {
timer_ = new TimerWrap(env, fn);
env->AddCleanupHook(CleanupHook, this);
}
void TimerWrapHandle::Stop() {
if (timer_ != nullptr)
return timer_->Stop();
}
void TimerWrapHandle::Close() {
if (timer_ != nullptr) {
timer_->env()->RemoveCleanupHook(CleanupHook, this);
timer_->Close();
}
timer_ = nullptr;
}
void TimerWrapHandle::Ref() {
if (timer_ != nullptr)
timer_->Ref();
}
void TimerWrapHandle::Unref() {
if (timer_ != nullptr)
timer_->Unref();
}
void TimerWrapHandle::Update(uint64_t interval, uint64_t repeat) {
if (timer_ != nullptr)
timer_->Update(interval, repeat);
}
void TimerWrapHandle::MemoryInfo(MemoryTracker* tracker) const {
if (timer_ != nullptr)
tracker->TrackField("timer", *timer_);
}
void TimerWrapHandle::CleanupHook(void* data) {
static_cast<TimerWrapHandle*>(data)->Close();
}
} // namespace node
|
