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 crate::Error;
use parking_lot::Mutex;
use std::{fmt, sync::Arc};

type ErrorFn = Box<dyn FnMut(Error) + Send + 'static>;

#[derive(Clone)]
pub(crate) struct ErrorHandler(Arc<Mutex<Option<ErrorFn>>>);

impl ErrorHandler {
    pub(crate) fn set_handler<E: FnMut(Error) + Send + 'static>(&self, handler: E) {
        *self.0.lock() = Some(Box::new(handler));
    }

    pub(crate) fn on_error(&self, error: Error) {
        if let Some(handler) = self.0.lock().as_mut() {
            handler(error)
        }
    }
}

impl Default for ErrorHandler {
    fn default() -> Self {
        Self(Arc::new(Mutex::new(None)))
    }
}

impl fmt::Debug for ErrorHandler {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_tuple("ErrorHandler").finish()
    }
}