Task.hpp

Go to the documentation of this file.

/*
    Copyright (C) 2014-2016 Leosac

    This file is part of Leosac.

    Leosac is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Leosac is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program. If not, see <http://www.gnu.org/licenses/>.
*/

#pragma once

#include <atomic>
#include <condition_variable>
#include <memory>

namespace Leosac
{
namespace Tasks
{
class Task
{
  public:
    Task();
    virtual ~Task()
    {
    }
    Task(const Task &) = delete;
    Task(Task &&)      = delete;
    Task &operator=(const Task &) = delete;
    Task &operator=(Task &&) = delete;

    bool is_complete() const;

    void wait();

    void run();

    bool succeed() const;

    std::exception_ptr get_exception() const;

    template <typename Callback>
    void set_on_completion(Callback c)
    {
        on_completion_ = [=]() { c(); };
    }

    template <typename Callback>
    void set_on_success(Callback c)
    {
        on_success_ = [=]() { c(); };
    }

    template <typename Callback>
    void set_on_failure(Callback c)
    {
        on_failure_ = [=]() { c(); };
    }

    const std::string &get_guid() const;

  private:
    virtual bool do_run() = 0;

    std::function<void(void)> on_completion_;
    std::function<void(void)> on_success_;
    std::function<void(void)> on_failure_;

    bool success_;
    std::exception_ptr eptr_;

    std::mutex mutex_;
    std::atomic_bool complete_;
    std::condition_variable cv_;
    std::string guid_;
};
}
}