via 2.0.0-gm.32

An async multi-threaded web framework for people who appreciate simplicity.
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
37
38
39
40
41

use std::sync::Arc;
use std::sync::atomic::{AtomicBool, Ordering};
use tokio::sync::Notify;

#[derive(Clone)]
pub struct Cancellation(Arc<Inner>);

pub struct Remote(Arc<Inner>);

struct Inner {
    cancelled: AtomicBool,
    notify: Notify,
}

impl Cancellation {
    pub fn new() -> (Self, Remote) {
        let inner = Arc::new(Inner {
            cancelled: AtomicBool::new(false),
            notify: Notify::new(),
        });

        (Self(Arc::clone(&inner)), Remote(inner))
    }

    pub async fn wait(&self) {
        let inner = &*self.0;

        if !inner.cancelled.load(Ordering::Acquire) {
            inner.notify.notified().await;
        }
    }
}

impl Remote {
    pub(super) fn cancel(&self) {
        let inner = &*self.0;

        inner.cancelled.store(true, Ordering::Release);
        inner.notify.notify_waiters();
    }
}