midware_ws/NetLib/wsClient/Scheduler.cpp

#include "stdafx.h"
#include "Scheduler.h"

CScheduler::CScheduler(boost::asio::io_service& ios):timer_(ios, std::chrono::milliseconds(0)),m_DuratTime(1000), m_IsStop(true), m_FunCB(nullptr), m_TimeId(0)
{

}

CScheduler::~CScheduler()
{
	killTimer();
}

void CScheduler::setTimer(const std::int32_t& nIDEvent, const std::int32_t& uElapse)
{
	if (!m_IsStop.load()) return;
	m_TimeId = nIDEvent;
	m_DuratTime = uElapse;
	m_IsStop.store(false);
	timer_.expires_from_now(std::chrono::milliseconds(m_DuratTime));
	//timer_.expires_at(timer_.expires_at() + std::chrono::milliseconds(m_DuratTime));
	timer_.async_wait(std::bind(&this_type::__onTimer, this));
}

void CScheduler::killTimer()
{
	if (!m_IsStop)
		m_IsStop.store(true);
	timer_.cancel();
}

void CScheduler::setOnTimeFuncCB(FimeFunCB fun)
{
	if (fun != nullptr&&this->m_FunCB == nullptr) {
		this->m_FunCB = fun;
	}
}

void CScheduler::__onTimer()
{
	if (this->m_FunCB && !m_IsStop.load()) {
		this->m_FunCB(m_TimeId);
	}

	if (!m_IsStop.load()) {
		timer_.expires_from_now(std::chrono::milliseconds(m_DuratTime));
		//timer_.expires_at(timer_.expires_at() + std::chrono::milliseconds(m_DuratTime));
		if (!m_IsStop.load())
			timer_.async_wait(std::bind(&this_type::__onTimer, this));
	}
}