/**
 *
 *  @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
 *
 */
#pragma once

#include <trantor/utils/NonCopyable.h>
#include <trantor/net/EventLoop.h>
#include <chrono>
#include <functional>
#include <atomic>
#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