summaryrefslogtreecommitdiffstats
path: root/third_party/rust/crossbeam-channel/examples/stopwatch.rs
blob: 3a7578e00d276f173814e916fab1b3530b8196d9 (plain)
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
//! Prints the elapsed time every 1 second and quits on Ctrl+C.

#[cfg(windows)] // signal_hook::iterator does not work on windows
fn main() {
    println!("This example does not work on Windows");
}

#[cfg(not(windows))]
fn main() {
    use std::io;
    use std::thread;
    use std::time::{Duration, Instant};

    use crossbeam_channel::{bounded, select, tick, Receiver};
    use signal_hook::consts::SIGINT;
    use signal_hook::iterator::Signals;

    // Creates a channel that gets a message every time `SIGINT` is signalled.
    fn sigint_notifier() -> io::Result<Receiver<()>> {
        let (s, r) = bounded(100);
        let mut signals = Signals::new(&[SIGINT])?;

        thread::spawn(move || {
            for _ in signals.forever() {
                if s.send(()).is_err() {
                    break;
                }
            }
        });

        Ok(r)
    }

    // Prints the elapsed time.
    fn show(dur: Duration) {
        println!("Elapsed: {}.{:03} sec", dur.as_secs(), dur.subsec_millis());
    }

    let start = Instant::now();
    let update = tick(Duration::from_secs(1));
    let ctrl_c = sigint_notifier().unwrap();

    loop {
        select! {
            recv(update) -> _ => {
                show(start.elapsed());
            }
            recv(ctrl_c) -> _ => {
                println!();
                println!("Goodbye!");
                show(start.elapsed());
                break;
            }
        }
    }
}