6b409cf664PR-URL: https://github.com/nodejs/node/pull/38686 Reviewed-By: Colin Ihrig <cjihrig@gmail.com> Reviewed-By: Richard Lau <rlau@redhat.com> Reviewed-By: Darshan Sen <raisinten@gmail.com> Reviewed-By: James M Snell <jasnell@gmail.com> Reviewed-By: Juan José Arboleda <soyjuanarbol@gmail.com> Reviewed-By: Rich Trott <rtrott@gmail.com>
84 lines
1.7 KiB
C++
84 lines
1.7 KiB
C++
#ifndef SRC_TIMER_WRAP_H_
|
|
#define SRC_TIMER_WRAP_H_
|
|
|
|
#if defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS
|
|
|
|
#include "memory_tracker.h"
|
|
#include "env.h"
|
|
#include "uv.h"
|
|
|
|
#include <functional>
|
|
|
|
namespace node {
|
|
|
|
// Utility class that makes working with libuv timers a bit easier.
|
|
class TimerWrap final : public MemoryRetainer {
|
|
public:
|
|
using TimerCb = std::function<void()>;
|
|
|
|
TimerWrap(Environment* env, const TimerCb& fn);
|
|
TimerWrap(const TimerWrap&) = delete;
|
|
|
|
inline Environment* env() const { return env_; }
|
|
|
|
// Stop calling the timer callback.
|
|
void Stop();
|
|
// Render the timer unusable and delete this object.
|
|
void Close();
|
|
|
|
// Starts / Restarts the Timer
|
|
void Update(uint64_t interval, uint64_t repeat = 0);
|
|
|
|
void Ref();
|
|
void Unref();
|
|
|
|
SET_NO_MEMORY_INFO()
|
|
SET_MEMORY_INFO_NAME(TimerWrap)
|
|
SET_SELF_SIZE(TimerWrap)
|
|
|
|
private:
|
|
static void TimerClosedCb(uv_handle_t* handle);
|
|
static void OnTimeout(uv_timer_t* timer);
|
|
~TimerWrap() = default;
|
|
|
|
Environment* env_;
|
|
TimerCb fn_;
|
|
uv_timer_t timer_;
|
|
|
|
friend std::unique_ptr<TimerWrap>::deleter_type;
|
|
};
|
|
|
|
class TimerWrapHandle : public MemoryRetainer {
|
|
public:
|
|
TimerWrapHandle(
|
|
Environment* env,
|
|
const TimerWrap::TimerCb& fn);
|
|
|
|
TimerWrapHandle(const TimerWrapHandle&) = delete;
|
|
|
|
~TimerWrapHandle() { Close(); }
|
|
|
|
void Update(uint64_t interval, uint64_t repeat = 0);
|
|
|
|
void Ref();
|
|
void Unref();
|
|
|
|
void Stop();
|
|
void Close();
|
|
|
|
void MemoryInfo(node::MemoryTracker* tracker) const override;
|
|
|
|
SET_MEMORY_INFO_NAME(TimerWrapHandle)
|
|
SET_SELF_SIZE(TimerWrapHandle)
|
|
|
|
private:
|
|
static void CleanupHook(void* data);
|
|
|
|
TimerWrap* timer_;
|
|
};
|
|
|
|
} // namespace node
|
|
|
|
#endif // defined(NODE_WANT_INTERNALS) && NODE_WANT_INTERNALS
|
|
|
|
#endif // SRC_TIMER_WRAP_H_
|