102 lines
2.6 KiB
C
102 lines
2.6 KiB
C
|
#ifndef EPOLL_EVENTLOOP_H
|
|||
|
|
#define EPOLL_EVENTLOOP_H
|
|||
|
|
|
|||
|
|
#include "CCEpoll.h"
|
|||
|
|
#ifdef __linux__
|
|||
|
|
#include <unistd.h>
|
|||
|
|
#endif
|
|||
|
|
#include <functional>
|
|||
|
|
#include <vector>
|
|||
|
|
#include <queue>
|
|||
|
|
#include <mutex>
|
|||
|
|
#include <atomic>
|
|||
|
|
#include <thread>
|
|||
|
|
#include <iostream>
|
|||
|
|
#include <chrono>
|
|||
|
|
#include <set>
|
|||
|
|
#include "TL/Map.h"
|
|||
|
|
|
|||
|
|
namespace CTL {
|
|||
|
|
enum EPEvent {
|
|||
|
|
IN_t = 1,
|
|||
|
|
OUT_t = 2,
|
|||
|
|
HUP = 3,
|
|||
|
|
MSG = 4,
|
|||
|
|
RD_HUP = 5,
|
|||
|
|
ERR = 6,
|
|||
|
|
};
|
|||
|
|
class EventLoop {
|
|||
|
|
public:
|
|||
|
|
using Callback = std::function<void(uint32_t)>;
|
|||
|
|
using TimePoint = std::chrono::steady_clock::time_point;
|
|||
|
|
struct Event_t {
|
|||
|
|
Callback cb = nullptr;
|
|||
|
|
uint32_t event{};
|
|||
|
|
};
|
|||
|
|
EventLoop();
|
|||
|
|
~EventLoop();
|
|||
|
|
|
|||
|
|
// 注册IO事件
|
|||
|
|
void registerEvent(int fd, const Callback &cb,uint32_t event_t = EPOLLIN);
|
|||
|
|
|
|||
|
|
// 移除IO事件
|
|||
|
|
void unregisterEvent(int fd);
|
|||
|
|
|
|||
|
|
// 添加定时任务（延迟执行）
|
|||
|
|
void addTimerAfter(std::chrono::milliseconds delay, const Callback &cb, bool repeat = false);
|
|||
|
|
|
|||
|
|
// 添加定时任务（指定时间点执行）
|
|||
|
|
void addTimerAt(TimePoint expiration, const Callback &cb, bool repeat = false);
|
|||
|
|
|
|||
|
|
// 启动事件循环
|
|||
|
|
void run();
|
|||
|
|
|
|||
|
|
// 停止事件循环
|
|||
|
|
void stop();
|
|||
|
|
|
|||
|
|
private:
|
|||
|
|
struct TimerTask {
|
|||
|
|
TimePoint expiration;
|
|||
|
|
Callback cb;
|
|||
|
|
bool repeat;
|
|||
|
|
std::chrono::milliseconds interval;
|
|||
|
|
|
|||
|
|
TimerTask(TimePoint exp, Callback callback, bool rep, std::chrono::milliseconds intv)
|
|||
|
|
: expiration(exp), cb(std::move(callback)), repeat(rep), interval(intv) {}
|
|||
|
|
|
|||
|
|
// 新增部分：用于优先队列排序（按时间从小到大）
|
|||
|
|
bool operator<(const TimerTask& other) const {
|
|||
|
|
return expiration > other.expiration; // 小顶堆
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
using TimerQueue = std::priority_queue<TimerTask, std::vector<TimerTask>, std::greater<TimerTask>>;
|
|||
|
|
|
|||
|
|
void addTimer(TimePoint expiration, Callback cb, bool repeat, std::chrono::milliseconds interval);
|
|||
|
|
|
|||
|
|
void processTimers();
|
|||
|
|
|
|||
|
|
void advanceWheel();
|
|||
|
|
|
|||
|
|
size_t getCurrentTick() const;
|
|||
|
|
|
|||
|
|
static constexpr size_t WHEEL_SIZE = 60; // 时间轮槽数
|
|||
|
|
static constexpr int WHEEL_INTERVAL = 1000; // 每个槽的时间间隔（ms）
|
|||
|
|
|
|||
|
|
Epoll epoll_{};
|
|||
|
|
Map<int, Event_t*> callbacks_;
|
|||
|
|
std::mutex mutex_;
|
|||
|
|
std::atomic<bool> running;
|
|||
|
|
|
|||
|
|
// 时间轮实现
|
|||
|
|
std::vector<std::priority_queue<TimerTask>> wheel_;
|
|||
|
|
size_t current_tick_ = 0;
|
|||
|
|
TimePoint last_tick_ = std::chrono::steady_clock::now();
|
|||
|
|
|
|||
|
|
// 用于定时器触发的epoll事件
|
|||
|
|
struct epoll_event event_;
|
|||
|
|
int next_timer_id_ = 0;
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
#endif
|