profserve 0.1.0

HTTP-Enabled Profiling for Rust
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

use actix_web::{
    error::ResponseError,
    http::{header::ContentType, StatusCode},
    HttpResponse,
};
use std::{fmt::Display, sync::PoisonError};

#[derive(Debug)]
pub enum Error {
    Pprof(pprof::Error),
    Poison(String),
    General(String),
}

impl From<pprof::Error> for Error {
    fn from(err: pprof::Error) -> Self {
        Error::Pprof(err)
    }
}

impl<T> From<PoisonError<T>> for Error {
    fn from(value: PoisonError<T>) -> Self {
        Error::Poison(value.to_string())
    }
}

impl Display for Error {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Error::Pprof(err) => write!(f, "{err}"),
            Error::Poison(err) => write!(f, "{err}"),
            Error::General(err) => write!(f, "{err}"),
        }
    }
}

impl std::error::Error for Error {}

impl ResponseError for Error {
    fn status_code(&self) -> StatusCode {
        match self {
            Self::Pprof(_) | Self::General(_) => StatusCode::INTERNAL_SERVER_ERROR,
            Self::Poison(_) => StatusCode::SERVICE_UNAVAILABLE,
        }
    }

    fn error_response(&self) -> HttpResponse {
        HttpResponse::build(self.status_code())
            .insert_header(ContentType::plaintext())
            .body(self.to_string())
    }
}

pub type Result<T, E = Error> = std::result::Result<T, E>;