ÿØÿà JFIF      ÿÛ „ 	 ( %!1!%)+//.383,7(-.+



-%%-////---/-.+/--+------/------/--0+--/-/-----.-----ÿÀ  ¥2" ÿÄ               ÿÄ J  	     ! 1AQ"aq2‘#BR‚¡ÁÑ3br’¢±Âð$CSƒ²á4c“%DsÓñÿÄ              ÿÄ *        !1AQa‘"2q3±ð#b¡ÿÚ   ? ¼QxJQaÍuò¸Zö Úü8,ÐÚú
"SSn<rçù–´âE—^ªBÖ9À\†¸ÔÁT­ÃÛ5
ëd´³Í#Ý;Þ38œî ¶H£M:wÎ3…³…âpÔF&‚FK¸9„â4àGEõªfÿ ‘ñ(ßw­pŽF|È¥ù®häðÍÑ¶¹‘[ÒinÙW¶ùñY˜Q{›K"išÒ[Ú8žë\F¹@-?v"ÔU”,ìöžkÿ {I‡£šÍ?e
ríV
..............................................................................................................................................................................
.............................................................................                                                  
                                                                                                                                                                                     ÿØÿà JFIF      ÿÛ „ 	 ( %!1!%)+//.383,7(-.+



-%%-////---/-.+/--+------/------/--0+--/-/-----.-----ÿÀ  ¥2" ÿÄ               ÿÄ J  	     ! 1AQ"aq2‘#BR‚¡ÁÑ3br’¢±Âð$CSƒ²á4c“%DsÓñÿÄ              ÿÄ *        !1AQa‘"2q3±ð#b¡ÿÚ   ? ¼QxJQaÍuò¸Zö Úü8,ÐÚú
"SSn<rçù–´âE—^ªBÖ9À\†¸ÔÁT­ÃÛ5
ëd´³Í#Ý;Þ38œî ¶H£M:wÎ3…³…âpÔF&‚FK¸9„â4àGEõªfÿ ‘ñ(ßw­pŽF|È¥ù®häðÍÑ¶¹‘[ÒinÙW¶ùñY˜Q{›K"išÒ[Ú8žë\F¹@-?v"ÔU”,ìöžkÿ {I‡£šÍ?e
ríV
..............................................................................................................................................................................
.............................................................................                                                  
                                                                                                                                                                                     /*
  +----------------------------------------------------------------------+
  | 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
