/*
+----------------------------------------------------------------------+
| Swoole |
+----------------------------------------------------------------------+
| This source file is subject to version 2.0 of the Apache license, |
| that is bundled with this package in the file LICENSE, and is |
| available through the world-wide-web at the following url: |
| http://www.apache.org/licenses/LICENSE-2.0.html |
| If you did not receive a copy of the Apache2.0 license and are unable|
| to obtain it through the world-wide-web, please send a note to |
| license@swoole.com so we can mail you a copy immediately. |
+----------------------------------------------------------------------+
| Author: Tianfeng Han <rango@swoole.com> |
| Twosee <twose@qq.com> |
+----------------------------------------------------------------------+
*/
#pragma once
#include "swoole_heap.h"
#include "swoole_reactor.h"
#include "swoole_util.h"
#include <unordered_map>
#define SW_TIMER_MIN_MS 1
#define SW_TIMER_MIN_SEC 0.001
#define SW_TIMER_MAX_MS LONG_MAX
#define SW_TIMER_MAX_SEC ((double) LONG_MAX / 1000)
namespace swoole {
typedef std::function<void(TimerNode *)> TimerDestructor;
struct TimerNode {
enum Type {
TYPE_KERNEL,
TYPE_PHP,
};
long id;
Type type;
int64_t exec_msec;
int64_t interval;
uint64_t exec_count;
uint64_t round;
bool removed;
HeapNode *heap_node;
TimerCallback callback;
void *data;
TimerDestructor destructor;
};
class Timer {
/*--------------signal timer--------------*/
Reactor *reactor_ = nullptr;
Heap heap;
std::unordered_map<long, TimerNode *> map;
uint64_t round;
long _next_id;
long _current_id;
/*---------------event timer--------------*/
int64_t base_time;
/**
* The time when the next timer will trigger, in milliseconds.
* This event will serve as the timeout for the event loop's epoll/poll/kqueue,
* or be set as the trigger time for the system clock.
*/
int64_t next_msec_;
/*----------------------------------------*/
std::function<int(Timer *timer, long exec_msec)> set;
std::function<void(Timer *timer)> close;
void init(bool manually_trigger);
void init_with_reactor(Reactor *reactor);
void init_with_system_timer();
void release_node(TimerNode *tnode);
public:
explicit Timer(bool manually_trigger);
~Timer();
int64_t get_relative_msec() const {
return get_absolute_msec() - base_time;
}
int64_t get_next_msec() const {
return next_msec_;
}
static int64_t get_absolute_msec() {
return time<std::chrono::milliseconds>(true);
}
Reactor *get_reactor() const {
return reactor_;
}
TimerNode *add(long _msec, bool persistent, void *data, const TimerCallback &callback);
bool remove(TimerNode *tnode);
void update(const TimerNode *tnode) const {
heap.change_priority(tnode->exec_msec, tnode->heap_node);
}
void delay(TimerNode *tnode, long delay_ms) const {
long now_ms = get_relative_msec();
tnode->exec_msec = (now_ms < 0 ? tnode->exec_msec : now_ms) + delay_ms;
update(tnode);
}
void reinit(bool manually_trigger = false);
void select();
TimerNode *get(long id) {
const auto it = map.find(id);
if (it == map.end()) {
return nullptr;
}
return it->second;
}
TimerNode *get(long id, const TimerNode::Type type) {
TimerNode *tnode = get(id);
return (tnode && tnode->type == type) ? tnode : nullptr;
}
size_t count() const {
return map.size();
}
uint64_t get_round() const {
return round;
}
bool remove(long id) {
return remove(get(id));
}
const std::unordered_map<long, TimerNode *> &get_map() {
return map;
}
};
static inline long sec2msec(const long sec) {
return sec * 1000;
}
static inline long sec2msec(const int sec) {
return sec * 1000;
}
static inline long sec2msec(const double sec) {
return static_cast<long>(sec * 1000);
}
static inline double msec2sec(const int msec) {
return static_cast<double>(msec) / 1000.0;
}
} // namespace swoole
|