ruloom 0.1.2

A thin wrapper around 'corosensei' that provides support for stackful coroutines in Rust (like Loom in Java or goroutines in golang).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

use std::{cell::Cell, mem};

use corosensei::Yielder;

pub(crate) type UnitYielder = Yielder<(), ()>;

pub(crate) struct ThreadLocalYielder;

impl ThreadLocalYielder {
    #[inline]
    pub unsafe fn set(yielder: &UnitYielder) {
        let static_yielder: &'static Yielder<(), ()> = unsafe { mem::transmute(yielder) };
        YIELDER.with(|cell| {
            cell.set(Some(static_yielder));
        });
    }

    #[inline]
    pub fn remove() {
        YIELDER.with(|cell| {
            cell.set(None);
        });
    }

    #[inline]
    pub unsafe fn get_expect_present() -> &'static UnitYielder {
        YIELDER.with(|cell| {
            cell.get()
                .expect("There's no yielder for the current thread.")
        })
    }
}

thread_local! {
    static YIELDER: Cell<Option<&'static UnitYielder>> = const { Cell::new(None) };
}