rt: Rename rust_scheduler to rust_task_thread

1 files changed, 150 insertions, 0 deletions

diff --git a/src/rt/rust_task_thread.h b/src/rt/rust_task_thread.h
new file mode 100644
index 00000000000..19486633713
--- /dev/null
+++ b/src/rt/rust_task_thread.h
@@ -0,0 +1,150 @@
+#ifndef RUST_TASK_THREAD_H
+#define RUST_TASK_THREAD_H
+
+#include "context.h"
+
+#ifndef _WIN32
+#include <pthread.h>
+#else
+#include <windows.h>
+#endif
+
+struct rust_task_thread;
+
+struct rust_hashable_dict {
+    UT_hash_handle hh;
+    void* fields[0];
+};
+
+class rust_crate_cache {
+public:
+    type_desc *get_type_desc(size_t size,
+                             size_t align,
+                             size_t n_descs,
+                             type_desc const **descs,
+                             uintptr_t n_obj_params);
+    void** get_dict(size_t n_fields, void** dict);
+
+private:
+
+    type_desc *type_descs;
+    rust_hashable_dict *dicts;
+
+public:
+
+    rust_task_thread *thread;
+    size_t idx;
+
+    rust_crate_cache(rust_task_thread *thread);
+    ~rust_crate_cache();
+    void flush();
+};
+
+struct rust_task_thread : public kernel_owned<rust_task_thread>,
+                        rust_thread
+{
+    RUST_REFCOUNTED(rust_task_thread)
+
+    // Fields known only by the runtime:
+    rust_log _log;
+
+    // NB: this is used to filter *runtime-originating* debug
+    // logging, on a per-scheduler basis. It's not likely what
+    // you want to expose to the user in terms of per-task
+    // or per-module logging control. By default all schedulers
+    // are set to debug-level logging here, and filtered by
+    // runtime category using the pseudo-modules ::rt::foo.
+    uint32_t log_lvl;
+
+    rust_srv *srv;
+    const char *const name;
+
+    rust_task_list newborn_tasks;
+    rust_task_list running_tasks;
+    rust_task_list blocked_tasks;
+    rust_task_list dead_tasks;
+
+    rust_crate_cache cache;
+
+    randctx rctx;
+
+    rust_kernel *kernel;
+    int32_t list_index;
+
+    const int id;
+
+    lock_and_signal lock;
+    size_t min_stack_size;
+
+#ifndef __WIN32__
+    pthread_attr_t attr;
+    static pthread_key_t task_key;
+#else
+    static DWORD task_key;
+#endif
+
+    static bool tls_initialized;
+
+    rust_env *env;
+    context c_context;
+
+    bool should_exit;
+
+    // Only a pointer to 'name' is kept, so it must live as long as this
+    // domain.
+    rust_task_thread(rust_kernel *kernel, rust_srv *srv, int id);
+    ~rust_task_thread();
+    void activate(rust_task *task);
+    void log(rust_task *task, uint32_t level, char const *fmt, ...);
+    rust_log & get_log();
+    void fail();
+
+    rust_crate_cache *get_cache();
+    size_t number_of_live_tasks();
+
+    void reap_dead_tasks();
+    rust_task *schedule_task();
+
+    void start_main_loop();
+
+    void log_state();
+
+    void kill_all_tasks();
+
+    rust_task *create_task(rust_task *spawner, const char *name,
+                           size_t init_stack_sz);
+
+    virtual void run();
+
+#ifdef __WIN32__
+    inline void win32_require(LPCTSTR fn, BOOL ok) {
+        kernel->win32_require(fn, ok);
+    }
+#endif
+
+    void init_tls();
+    void place_task_in_tls(rust_task *task);
+
+    static rust_task *get_task();
+
+    // Tells the scheduler to exit it's scheduling loop and thread
+    void exit();
+};
+
+inline rust_log &
+rust_task_thread::get_log() {
+    return _log;
+}
+
+//
+// Local Variables:
+// mode: C++
+// fill-column: 78;
+// indent-tabs-mode: nil
+// c-basic-offset: 4
+// buffer-file-coding-system: utf-8-unix
+// compile-command: "make -k -C $RBUILD 2>&1 | sed -e 's/\\/x\\//x:\\//g'";
+// End:
+//
+
+#endif /* RUST_TASK_THREAD_H */