scouter-tracing 0.21.1

Helper crate for tracing instrumentation
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use pyo3::exceptions::PyRuntimeError;
use pyo3::pyclass::PyClassGuardError;
use pyo3::PyErr;
use thiserror::Error;
#[derive(Error, Debug)]
pub enum TraceError {
    #[error("{0}")]
    PyError(String),

    #[error("Failed to initialize tracer: {0}")]
    InitializationError(String),

    #[error("Span operation failed: {0}")]
    SpanError(String),

    #[error("OpenTelemetry error: {0}")]
    OTelBuilderError(#[from] opentelemetry_otlp::ExporterBuildError),

    #[error("No active span found")]
    NoActiveSpan,

    #[error("Poison error occurred")]
    PoisonError(String),

    #[error(transparent)]
    OTelSdkError(#[from] opentelemetry_sdk::error::OTelSdkError),

    #[error(transparent)]
    TypeError(#[from] scouter_types::error::TypeError),

    #[error("Event must be a dictionary or a Pydantic BaseModel")]
    EventMustBeDict,

    #[error("Failed to downcast Python object: {0}")]
    DowncastError(String),

    #[error(transparent)]
    SerdeError(#[from] serde_json::Error),

    #[error("Invalid function type: {0}")]
    InvalidFunctionType(String),

    #[error("Unsupported SpanExporter type")]
    UnsupportedSpanExporterType,

    #[error(transparent)]
    EventError(#[from] scouter_events::error::EventError),

    #[error(transparent)]
    ParseIntError(#[from] std::num::ParseIntError),

    #[error("Queue not initialized")]
    QueueNotInitialized,

    #[error("{0}")]
    ExportError(String),

    #[error("{0}")]
    InvalidSpanKind(String),

    #[error("{0}")]
    TraceStateError(String),
}

impl From<TraceError> for PyErr {
    fn from(err: TraceError) -> PyErr {
        let msg = err.to_string();
        PyRuntimeError::new_err(msg)
    }
}

impl From<PyErr> for TraceError {
    fn from(err: PyErr) -> TraceError {
        TraceError::PyError(err.to_string())
    }
}

impl<'a, 'py> From<PyClassGuardError<'a, 'py>> for TraceError {
    fn from(err: PyClassGuardError<'a, 'py>) -> Self {
        TraceError::PyError(err.to_string())
    }
}

impl<'a, 'py> From<pyo3::CastError<'a, 'py>> for TraceError {
    fn from(err: pyo3::CastError<'a, 'py>) -> Self {
        TraceError::DowncastError(err.to_string())
    }
}