掌握C++定时器:构建自己的定时器的分步指南

2024-09-22 15:39:01 浏览数 (3)

一、定时器应用场景

(1)心跳检测。

(2)游戏中的技能冷却。

(3)倒计时。

(4)其他需要延迟处理的功能。

二、利用红黑树实现定时器

红黑树是绝对有序的数据结构。

图片图片

在c 中,set、map、multiset、multimap使用的是红黑树管理数据。可以利用这几个类实现定时器方案,以set为例,使用C 14特性。

2.1、实现接口

图片图片

获取当前时间的接口GetTick(),通过C 11时间库chrono:

(1)steady_clock:系统启动到当前的时间,可以用来计算程序运行时间。

(2)system_clock:时间戳,可以修改。

(3)high_resolution_clock:高精度版本的steady_lock。

代码语言:javascript复制
int64_t GetTick()
{
	auto sc=chrono::time_point_cast<chrono::milliseconds>(chrono::steady_clock::now());
	auto temp=chrono::duration_cast<milliseconds>(sc.time_since_epoch());
	return temp.count();
}

相同触发时间的定时任务处理方案:

(1)越后面插入的节点,插入位置放在红黑树越右侧。

(2)使用C 的set容器;内部是红黑树管理数据。

(3)定时器节点设计,使用触发时间和ID唯一标识定时节点。

代码语言:javascript复制
struct TimeNodeBase{
int64_t id;		// 描述插入的先后顺序
time_t  expire;	// 触发时间
}

(4)比较仿函数,确定数据插入红黑树的位置。利用C 14的特性,find时只需要等价key比较,无需构建key对象比较。利用基类的多态特性,只需要一个比较仿函数即可。

代码语言:javascript复制
bool operator < (const TimeNodeBase *lhd,const TimeNodeBase *rhd)
{
	if(lhd->expire < rhd->expire)
		return true;
	else if(lhd->expire > rhd->expire)
		return false;
	return lhd->id < rhd->id;
}

(5)尽量减少函数对象赋值、移动等操作,提高性能。

代码语言:javascript复制
// TimerNode 继承 TimerNodeBase
struct TimerNode:public TimerNodeBase
{
	// C   11特性,使用函数对象。降低拷贝消耗,提高效率
	using Callback = std::function<void(const TimerNode &node)>;
	Callback func;

	// 构造函数,只构造一次
	TimerNode(int64_t id,time_t expire,Callback func):func(func){
		this->id = id;
		this->expire = expire;

	}
};

2.2、驱动定时器方式

可以采用IO时间和定时器事件在同一个线程执行的方案,利用epoll的epoll_wait()第四个参数做定时延时。

示例:

代码语言:javascript复制
int main()
{
	int epfd=epoll_wait(1);
	epoll_event evs[64]={0};
	while(1)
	{
		int n=epoll_wait(epfd,evs,64,delaytime);
		int i=0;
		for(i=0;i<n;i  )
		{
			/*处理IO事件*/
		}
		// 处理定时任务事件
	}
	return 0;
}

2.3、示例代码

(1)创建定时器驱动,epoll_create、epoll_wait。

(2)创建timer类,实现AddTimer()、CheckTimer()、DelTimer()等接口。

(3)选择数据结构,set容器,本质使用红黑树数据结构。

(4)定义节点的结构。

demo代码:

代码语言:javascript复制
#include <sys/epoll.h>
#include <functional>
#include <chrono>
#include <set>
#include <memory>
#include <iostream>

using namespace std;

struct TimerNodeBase
{
	time_t expire;
	int64_t id;
};

// TimerNode 继承 TimerNodeBase
struct TimerNode:public TimerNodeBase
{
	// C   11特性,使用函数对象。降低拷贝消耗,提高效率
	using Callback = std::function<void(const TimerNode &node)>;
	Callback func;

	// 构造函数,只构造一次
	TimerNode(int64_t id,time_t expire,Callback func):func(func){
		this->id = id;
		this->expire = expire;

	}
};

// 基类引用,多态特性
bool operator<(const TimerNodeBase &lhd, const TimerNodeBase &rhd)
{
	if (lhd.expire < rhd.expire)
		return true;
	else if (lhd.expire > rhd.expire)
		return false;
	return lhd.id < rhd.id;
}

class Timer
{
public:
	static time_t GetTick()
	{
		/* C   11时间库chrono */
		//表示一个具体时间
		auto sc = chrono::time_point_cast<chrono::milliseconds>(chrono::steady_clock::now());
		auto tmp = chrono::duration_cast<chrono::milliseconds>(sc.time_since_epoch());
		return tmp.count();
	}

	TimerNodeBase AddTimer(time_t msec, TimerNode::Callback func)
	{
		time_t expire = GetTick()   msec;
		//避免拷贝、移动构造
		auto ele = timermap.emplace(GenID(), expire, func);

		return static_cast<TimerNodeBase>(*ele.first);
	}

	bool DelTimer(TimerNodeBase &node)
	{
		// C   14新特性,不在需要传一个key对象,传递一个key的等价值
		auto iter = timermap.find(node);
		if (iter != timermap.end())
		{
			timermap.erase(iter);
			return true;
		}
		return false;
	}

	bool CheckTimer()
	{
		auto iter = timermap.begin();

		if (iter != timermap.end() && iter->expire <= GetTick())
		{
			iter->func(*iter);
			timermap.erase(iter);
			return true;
		}
		return false;
	}

	time_t TimeToSleep()
	{
		auto iter = timermap.begin();
		if (iter == timermap.end())
			return -1;//没有定时任务,设置epoll一直阻塞。
		time_t diss = iter->expire - GetTick();

		return diss > 0 ? diss : 0;
	}
private:
	static int64_t GenID()
	{
		return gid  ;
	}

	static int64_t gid;
	set<TimerNode, std::less<>> timermap;
};

int64_t Timer::gid = 0;

#define EPOLL_EV_LENFTH	1024
int main()
{
	int epfd = epoll_create(1);

	unique_ptr<Timer> timer = make_unique<Timer>();

	int i = 0;
	timer->AddTimer(1000, [&](const TimerNode &node) {
		cout << Timer::GetTick() << "node id:" << Home | NODE.ID << " revoked times:" <<   i << endl;
	});

	timer->AddTimer(1000, [&](const TimerNode &node) {
		cout << Timer::GetTick() << "node id:" << Home | NODE.ID << " revoked times:" <<   i << endl;
	});

	timer->AddTimer(3000, [&](const TimerNode &node) {
		cout << Timer::GetTick() << "node id:" << Home | NODE.ID << " revoked times:" <<   i << endl;
	});

	auto node = timer->AddTimer(2100, [&](const TimerNode &node) {
		cout << Timer::GetTick() << "node id:" << Home | NODE.ID << " revoked times:" <<   i << endl;
	});

	timer->DelTimer(node);

	cout << "now time:" << Timer::GetTick() << endl;

	epoll_event evs[EPOLL_EV_LENFTH] = { 0 };

	while (1)
	{
		int nready = epoll_wait(epfd, evs, EPOLL_EV_LENFTH, timer->TimeToSleep());
		for (int i = 0; i < nready; i  )
		{
			/*处理IO事件*/
		}

		// timer检测和处理
		while (timer->CheckTimer());
	}

	return 0;
}

三、总结

设计一个定时器时,先确定时间精度;选择驱动定时器的方式;选择合适的数据接口(不要选择优先队列);设计定时器基本接口 { AddTimer()、DelTimer()、CheckTimer() } 和扩展接口 { Tick()等 } ;考虑相同触发时间的定时任务处理。

0 人点赞