TimerManager.hpp

Go to the documentation of this file.

#pragma once
#include <chrono>
#include <functional>
#include <vector>
#include <queue>
#include <unordered_map>
#include <unordered_set>
#include <mutex>
#include <cstdint>
 
namespace server {
 
using TimePoint = std::chrono::steady_clock::time_point;
 
struct TimerEvent {
    TimePoint expiration;
    uint64_t timer_id;
    int fd;
    
    // Min-heap ordering: lowest expiration time comes first
    bool operator>(const TimerEvent& other) const {
        return expiration > other.expiration;
    }
};
 
class TimerManager {
public:
    TimerManager();
 
    uint64_t add_timer(int fd, std::chrono::milliseconds timeout);
    
    void cancel_timer(uint64_t timer_id);
 
    int get_next_timeout();
 
    void handle_expired_timers(std::function<void(int)> on_timeout);
 
private:
    std::priority_queue<TimerEvent, std::vector<TimerEvent>, std::greater<TimerEvent>> timers_;
    std::unordered_set<uint64_t> cancelled_timers_;
    uint64_t next_timer_id_{1};
    std::mutex mutex_;
};
 
} // namespace server