scheduler.h

Go to the documentation of this file.

#pragma once
 
#include "iris/core/task.h"
 
namespace iris {
class Scheduler {
public:
    void schedule_task(Task&);
 
    [[noreturn]] void start();
    [[noreturn]] void start_on_ap();
    void yield();
 
    [[noreturn]] void save_state_and_run_next(arch::TaskState* state);
    [[noreturn]] void exit_current_task();
 
    auto current_task() const -> Task& { return *m_current_task; }
    auto current_task_null_if_during_boot() const -> Task* { return m_current_task; }
    auto current_address_space() -> mm::AddressSpace&;

    auto block_current_task(di::FunctionRef<void()> before_yielding) -> Expected<void>;
 
    void setup_idle_task();
 
private:
    [[noreturn]] void run_next();
 
    Task* m_current_task { nullptr };
    di::Queue<Task, di::IntrusiveList<Task>> m_run_queue;
    di::Arc<Task> m_idle_task;
};

void schedule_task(Task&);
}