Current File : //opt/alt/php81/usr/include/php/ext/swoole/include/swoole_wheel_timer.h
/*
+----------------------------------------------------------------------+
| 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> |
+----------------------------------------------------------------------+
*/
#pragma once
#include "swoole.h"
#include <vector>
#include <list>
namespace swoole {
struct WheelTimerNode;
using WheelTimerCallback = std::function<void(WheelTimerNode *)>;
struct WheelTimerNode {
std::list<WheelTimerNode *>::iterator position_;
uint16_t index_;
WheelTimerCallback callback_;
};
class WheelTimer {
private:
uint64_t round_ = 0;
uint16_t size_;
std::vector<std::list<WheelTimerNode *>> buckets_;
void push(WheelTimerNode *node) {
node->index_ = (round_ + size_ - 1) % size_;
buckets_[node->index_].push_front(node);
node->position_ = buckets_[node->index_].begin();
}
public:
WheelTimer(uint16_t size) {
size_ = size;
buckets_.resize(size);
}
uint64_t get_round() {
return round_;
}
WheelTimerNode *add(const WheelTimerCallback &cb) {
WheelTimerNode *node = new WheelTimerNode;
push(node);
node->callback_ = cb;
return node;
}
void update(WheelTimerNode *node) {
buckets_[node->index_].erase(node->position_);
push(node);
}
void remove(WheelTimerNode *node) {
buckets_[node->index_].erase(node->position_);
delete node;
}
void next() {
uint16_t current_index = round_ % size_;
round_++;
std::list<WheelTimerNode *> &_list = buckets_[current_index];
for (auto node : _list) {
node->callback_(node);
delete node;
}
_list.clear();
}
};
} // namespace swoole