arty 0.5.0

Single-threaded, thread-aware application runtime.
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

//! A dropped join neither cancels its task nor transfers task panics to the caller.

#![cfg(feature = "rt")]
#![cfg(test)]

mod panic_support;

mod support;

use std::sync::atomic::{AtomicUsize, Ordering};
use std::sync::{Arc, mpsc};

use panic_support::runtime;
use support::JoinHandleExt as _;
use testing_aids::TEST_TIMEOUT;
use thread_aware::Unaware;

testing_aids::init_tracing!();

struct ResultDropPanic {
    drops: Arc<AtomicUsize>,
    dropped: mpsc::Sender<()>,
}

impl Drop for ResultDropPanic {
    fn drop(&mut self) {
        self.drops.fetch_add(1, Ordering::SeqCst);
        self.dropped.send(()).unwrap();
        panic!("unreceived result destructor");
    }
}

#[test]
fn abandoned_remote_result_drop_is_contained() {
    let runtime = runtime();
    let (release, gate) = events_once::Event::boxed();
    let (dropped, received) = mpsc::channel();
    let drops = Arc::new(AtomicUsize::new(0));
    let captured = Arc::clone(&drops);
    let task = runtime.scheduler().spawn_anywhere(
        Unaware((gate, captured, dropped)),
        |_, Unaware((gate, drops, dropped))| async move {
            gate.await.unwrap();
            Unaware(ResultDropPanic { drops, dropped })
        },
    );
    drop(task);
    release.send(());
    received.recv_timeout(TEST_TIMEOUT).unwrap();
    assert_eq!(runtime.scheduler().spawn_anywhere((), |_, ()| async { 7 }).join().unwrap(), 7);
    assert_eq!(drops.load(Ordering::SeqCst), 1);
    runtime.stop().unwrap();
}

#[test]
fn abandoned_blocking_result_drop_is_contained() {
    let runtime = runtime();
    let (started, ready) = mpsc::channel();
    let (release, gate) = mpsc::channel();
    let (dropped, received) = mpsc::channel();
    let drops = Arc::new(AtomicUsize::new(0));
    let captured = Arc::clone(&drops);
    let task = runtime.scheduler().spawn_blocking(move || {
        started.send(()).unwrap();
        gate.recv_timeout(TEST_TIMEOUT).unwrap();
        ResultDropPanic { drops: captured, dropped }
    });
    ready.recv_timeout(TEST_TIMEOUT).unwrap();
    drop(task);
    release.send(()).unwrap();
    received.recv_timeout(TEST_TIMEOUT).unwrap();
    assert_eq!(runtime.scheduler().spawn_blocking(|| 7).join().unwrap(), 7);
    assert_eq!(drops.load(Ordering::SeqCst), 1);
    runtime.stop().unwrap();
}