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

use std::fmt::{Debug, Display};
use std::process;

pub trait ResultExtension<T, E> {
    fn warn_on_err(self);
    fn fatal_on_err(self) -> T;
}

impl<T, E> ResultExtension<T, E> for Result<T, E>
where
    T: Debug,
    E: Debug + Display,
{
    /// Log a warning on error.
    fn warn_on_err(self) {
        if self.is_err() {
            let err = self.unwrap_err();
            warn!("{}", err);
        }
    }

    /// Log an error and exit with status code (1).
    fn fatal_on_err(self) -> T {
        match self {
            Ok(value) => value,
            Err(err) => {
                error!("{}", err);
                process::exit(1);
            }
        }
    }
}