Scheduler.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 "LeosacFwd.hpp"
#include "core/tasks/GenericTask.hpp"
#include <map>
#include <mutex>
#include <queue>
#include <thread>

namespace Leosac
{

enum class TargetThread
{
    MAIN,
    POOL,
};

class Scheduler
{
  public:
    Scheduler(Kernel *kptr);

    Scheduler(const Scheduler &) = delete;
    Scheduler(Scheduler &&)      = delete;
    Scheduler &operator=(const Scheduler &) = delete;
    Scheduler &operator=(Scheduler &&) = delete;

    template <typename Callable>
    typename std::enable_if<
        !std::is_convertible<Callable, std::shared_ptr<Tasks::Task>>::value,
        void>::type
    enqueue(const Callable &call, TargetThread policy)
    {
        enqueue(Tasks::GenericTask::build(call), policy);
    }

    void enqueue(Tasks::TaskPtr t, TargetThread policy);

    void update(TargetThread me) noexcept;

    void register_thread(TargetThread me);

    Kernel &kernel();

  private:
    using TaskQueue    = std::queue<Tasks::TaskPtr>;
    using TaskQueueMap = std::map<TargetThread, TaskQueue>;

    TaskQueueMap queues_;

    Kernel *kptr_;
    mutable std::mutex mutex_;
};
}