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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

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

//! Runtime-owned opaque panic payloads can panic when an abandoned join discards them.

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

mod panic_support;

mod support;

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

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

testing_aids::init_tracing!();

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

impl Drop for Payload {
    fn drop(&mut self) {
        assert!(!std::thread::panicking(), "the original task unwind was already caught");
        self.drops.fetch_add(1, Ordering::SeqCst);
        self.dropped.send(()).unwrap();
        panic!("discarded opaque panic payload");
    }
}

fn check(runtime: arty::runtime::Runtime, drops: &AtomicUsize, received: &mpsc::Receiver<()>) {
    received.recv_timeout(TEST_TIMEOUT).unwrap();
    assert_eq!(drops.load(Ordering::SeqCst), 1);
    assert_eq!(runtime.scheduler().spawn_anywhere((), |_, ()| async { 42 }).join().unwrap(), 42);
    runtime.stop().unwrap();
}

#[test]
fn abandoned_remote_factory_payload_drop_is_contained() {
    let runtime = runtime();
    let (dropped, received) = mpsc::channel();
    let drops = Arc::new(AtomicUsize::new(0));
    let payload = Payload {
        drops: Arc::clone(&drops),
        dropped,
    };
    runtime
        .scheduler()
        .block_on(async move |cx| {
            let join = cx.scheduler().spawn(move |_| -> std::future::Ready<()> { panic_any(payload) });
            drop(join);
        })
        .unwrap();
    check(runtime, &drops, &received);
}

#[test]
fn abandoned_remote_poll_payload_drop_is_contained() {
    let runtime = runtime();
    let (dropped, received) = mpsc::channel();
    let drops = Arc::new(AtomicUsize::new(0));
    let payload = Payload {
        drops: Arc::clone(&drops),
        dropped,
    };
    runtime
        .scheduler()
        .block_on(async move |cx| {
            let join: arty::task::JoinHandle<()> = cx.scheduler().spawn(async move |_| panic_any(payload));
            drop(join);
        })
        .unwrap();
    check(runtime, &drops, &received);
}

#[test]
fn abandoned_blocking_payload_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 payload = Payload {
        drops: Arc::clone(&drops),
        dropped,
    };
    let join: arty::task::JoinHandle<()> = runtime.scheduler().spawn_blocking(move || {
        started.send(()).unwrap();
        gate.recv_timeout(TEST_TIMEOUT).unwrap();
        panic_any(payload);
    });
    ready.recv_timeout(TEST_TIMEOUT).unwrap();
    drop(join);
    release.send(()).unwrap();
    check(runtime, &drops, &received);
}