smackdebt 0.3.0

Find costly code debt and see whether your changes make it better
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

//! Delayed terminal feedback; analysis stays on the calling thread.
use std::sync::mpsc::{self, Sender};
use std::thread::{self, JoinHandle};
use std::time::Duration;

pub(crate) fn run<T>(human_output: bool, work: impl FnOnce() -> T) -> T {
    let _progress = Progress::start(human_output && crate::terminal::interactive());
    work()
}

struct Progress {
    worker: Option<(Sender<()>, JoinHandle<()>)>,
}

impl Progress {
    fn start(enabled: bool) -> Self {
        let worker = enabled.then(|| {
            let (stop, receiver) = mpsc::channel();
            let thread = thread::spawn(move || {
                if receiver.recv_timeout(Duration::from_millis(250))
                    == Err(mpsc::RecvTimeoutError::Timeout)
                {
                    let spinner = cliclack::spinner();
                    spinner.start("Analyzing code…");
                    let _ = receiver.recv();
                    spinner.clear();
                }
            });
            (stop, thread)
        });
        Self { worker }
    }
}

impl Drop for Progress {
    fn drop(&mut self) {
        if let Some((stop, worker)) = self.worker.take() {
            let _ = stop.send(());
            let _ = worker.join();
        }
    }
}