/**
 *
 *  @file TaskTimeoutFlag.h
 *  @author An Tao
 *
 *  Copyright 2018, An Tao.  All rights reserved.
 *  https://github.com/an-tao/drogon
 *  Use of this source code is governed by a MIT license
 *  that can be found in the License file.
 *
 *  Drogon
 *
 */
#include <trantor/net/EventLoop.h>
#include <trantor/utils/NonCopyable.h>
#include <atomic>
#include <chrono>
#include <functional>
#include <memory>

namespace drogon {
class TaskTimeoutFlag : public trantor::NonCopyable,
						public std::enable_shared_from_this<TaskTimeoutFlag> {
public:
	TaskTimeoutFlag(trantor::EventLoop *loop,
			const std::chrono::duration<double> &timeout,
			std::function<void()> timeoutCallback);
	bool done();
	void runTimer();

private:
	std::atomic<bool> isDone_{ false };
	trantor::EventLoop *loop_;
	std::chrono::duration<double> timeout_;
	std::function<void()> timeoutFunc_;
};
} // namespace drogon