rust/src/rt/rust_scheduler.h

/**
   The rust scheduler. Schedulers may be added to the kernel
   dynamically and they run until there are no more tasks to
   schedule. Most of the scheduler work is carried out in worker
   threads by rust_sched_loop.
 */

#ifndef RUST_SCHEDULER_H
#define RUST_SCHEDULER_H

#include "rust_globals.h"
#include "util/array_list.h"
#include "rust_kernel.h"
#include "rust_refcount.h"

class rust_sched_launcher;
class rust_sched_launcher_factory;

class rust_scheduler : public kernel_owned<rust_scheduler> {
    RUST_ATOMIC_REFCOUNT();
    // FIXME (#2693): Make these private
public:
    rust_kernel *kernel;
private:
    // Protects live_threads, live_tasks, cur_thread, may_exit
    lock_and_signal lock;
    // When this hits zero we'll tell the kernel to release us
    uintptr_t live_threads;
    // When this hits zero we'll tell the threads to exit
    uintptr_t live_tasks;
    size_t cur_thread;
    bool may_exit;
    bool killed;

    rust_sched_launcher_factory *launchfac;
    array_list<rust_sched_launcher *> threads;
    const size_t max_num_threads;

    rust_sched_id id;

    void destroy_task_threads();

    rust_sched_launcher *create_task_thread(int id);
    void destroy_task_thread(rust_sched_launcher *thread);

    void exit();

    // Called when refcount reaches zero
    void delete_this();

public:
    rust_scheduler(rust_kernel *kernel, size_t max_num_threads,
                   rust_sched_id id, bool allow_exit, bool killed,
                   rust_sched_launcher_factory *launchfac);

    void start_task_threads();
    void join_task_threads();
    void kill_all_tasks();
    rust_task* create_task(rust_task *spawner, const char *name);

    void release_task();

    size_t max_number_of_threads();
    size_t number_of_threads();
    // Called by each thread when it terminates. When all threads
    // terminate the scheduler does as well.
    void release_task_thread();

    rust_sched_id get_id() { return id; }
    // Tells the scheduler that as soon as it runs out of tasks
    // to run it should exit
    void allow_exit();
    void disallow_exit();
};

#endif /* RUST_SCHEDULER_H */