telemetry-rust 6.10.0

Open Telemetry fox Axum and Tracing
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

/// AWS SageMaker Runtime fluent builder instrumentation implementations
use super::{utils::*, *};

// InvokeEndpoint
impl<'a> AwsBuilderInstrument<'a> for InvokeEndpointFluentBuilder {
    fn build_aws_span(&self) -> AwsSpanBuilder<'a> {
        let endpoint_name = self.get_endpoint_name().clone().unwrap_or_default();
        let attributes = attributes![
            self.get_target_model()
                .as_attribute("aws.sagemaker.target_model"),
            self.get_target_variant()
                .as_attribute("aws.sagemaker.target_variant"),
            self.get_inference_component_name()
                .as_attribute("aws.sagemaker.inference_component_name"),
        ];
        SageMakerRuntimeSpanBuilder::invoke_endpoint(endpoint_name).attributes(attributes)
    }
}
impl InstrumentedFluentBuilderOutput for InvokeEndpointOutput {
    fn extract_attributes(&self) -> impl IntoIterator<Item = KeyValue> {
        attributes![
            self.invoked_production_variant()
                .as_attribute("aws.sagemaker.invoked_production_variant"),
        ]
    }
}
instrument_aws_operation!(aws_sdk_sagemakerruntime::operation::invoke_endpoint);

// InvokeEndpointAsync
impl<'a> AwsBuilderInstrument<'a> for InvokeEndpointAsyncFluentBuilder {
    fn build_aws_span(&self) -> AwsSpanBuilder<'a> {
        let endpoint_name = self.get_endpoint_name().clone().unwrap_or_default();
        let attributes = attributes![
            self.get_inference_id()
                .as_attribute("aws.sagemaker.inference_id"),
            self.get_input_location()
                .as_attribute("aws.sagemaker.input_location"),
        ];
        SageMakerRuntimeSpanBuilder::invoke_endpoint_async(endpoint_name)
            .attributes(attributes)
    }
}
impl InstrumentedFluentBuilderOutput for InvokeEndpointAsyncOutput {
    fn extract_attributes(&self) -> impl IntoIterator<Item = KeyValue> {
        attributes![
            self.inference_id()
                .as_attribute("aws.sagemaker.inference_id"),
        ]
    }
}
instrument_aws_operation!(aws_sdk_sagemakerruntime::operation::invoke_endpoint_async);

// InvokeEndpointWithResponseStream
impl<'a> AwsBuilderInstrument<'a> for InvokeEndpointWithResponseStreamFluentBuilder {
    fn build_aws_span(&self) -> AwsSpanBuilder<'a> {
        let endpoint_name = self.get_endpoint_name().clone().unwrap_or_default();
        let attributes = attributes![
            self.get_target_variant()
                .as_attribute("aws.sagemaker.target_variant"),
            self.get_inference_component_name()
                .as_attribute("aws.sagemaker.inference_component_name"),
        ];
        SageMakerRuntimeSpanBuilder::invoke_endpoint_with_response_stream(endpoint_name)
            .attributes(attributes)
    }
}
impl InstrumentedFluentBuilderOutput for InvokeEndpointWithResponseStreamOutput {
    fn extract_attributes(&self) -> impl IntoIterator<Item = KeyValue> {
        attributes![
            self.invoked_production_variant()
                .as_attribute("aws.sagemaker.invoked_production_variant"),
        ]
    }
}
instrument_aws_operation!(
    aws_sdk_sagemakerruntime::operation::invoke_endpoint_with_response_stream
);