booster/lib/aio/src/deadline_timer.cpp - master - zOMGdev/cppcms

booster/lib/aio/src/deadline_timer.cpp

109 строк2 KB

Artyom Beilis

Switch from booster::shared_ptr/weak_ptr to std::ones

11 авг 2020, 22:57

11 авг 2020, 22:57dfc88bd

//
//  Copyright (C) 2009-2012 Artyom Beilis (Tonkikh)
//
//  Distributed under the Boost Software License, Version 1.0. (See
//  accompanying file LICENSE_1_0.txt or copy at
//  http://www.boost.org/LICENSE_1_0.txt)
//
#define BOOSTER_SOURCE
#include <booster/aio/deadline_timer.h>
#include <booster/aio/io_service.h>

namespace booster {
namespace aio {

struct deadline_timer::data {} ;

deadline_timer::deadline_timer() : 
	srv_(0),
	deadline_(ptime::now()),
	event_id_(-1)
{
}

deadline_timer::deadline_timer(io_service &srv) : 
	srv_(&srv),
	deadline_(ptime::now()),
	event_id_(-1)
{
}

deadline_timer::~deadline_timer()
{
}

io_service &deadline_timer::get_io_service()
{
	if(!srv_)
		throw system::system_error(aio_error::no_service_provided,aio_error_cat);
	return *srv_;
}

void deadline_timer::reset_io_service()
{
	srv_ = 0;
}

void deadline_timer::set_io_service(io_service &srv)
{
	reset_io_service();
	srv_ = &srv;
}

void deadline_timer::expires_from_now(ptime t)
{
	deadline_ = ptime::now() + t;
}

ptime deadline_timer::expires_from_now()
{
	return deadline_ - ptime::now();
}

void deadline_timer::expires_at(ptime t)
{
	deadline_ = t;
}

ptime deadline_timer::expires_at()
{
	return deadline_;
}

void deadline_timer::wait()
{
	ptime diff = expires_from_now();
	if(diff <= ptime::zero)
		return;
	ptime::sleep(diff);
}
struct deadline_timer::waiter : public booster::callable<void(system::error_code const &e)> {
	event_handler h;
	deadline_timer *self;
	void operator()(system::error_code const &e)
	{
		self->event_id_ = -1;
		h(e);
	}
};

void deadline_timer::async_wait(event_handler const &h)
{
	std::unique_ptr<waiter> wt(new waiter);
	wt->h=h;
	wt->self = this;
	event_id_ = get_io_service().set_timer_event(deadline_,std::move(wt));
}

void deadline_timer::cancel()
{
	if(event_id_!=-1) {
		int tmp_id = event_id_;
		event_id_ = -1;
		get_io_service().cancel_timer_event(tmp_id);
	}
}


} // aio
} // booster