datadog_api_client/datadogV2/model/
model_span.rs

1// Unless explicitly stated otherwise all files in this repository are licensed under the Apache-2.0 License.
2// This product includes software developed at Datadog (https://www.datadoghq.com/).
3// Copyright 2019-Present Datadog, Inc.
4use serde::de::{Error, MapAccess, Visitor};
5use serde::{Deserialize, Deserializer, Serialize};
6use serde_with::skip_serializing_none;
7use std::fmt::{self, Formatter};
8
9/// Object description of a spans after being processed and stored by Datadog.
10#[non_exhaustive]
11#[skip_serializing_none]
12#[derive(Clone, Debug, PartialEq, Serialize)]
13pub struct Span {
14    /// JSON object containing all span attributes and their associated values.
15    #[serde(rename = "attributes")]
16    pub attributes: Option<crate::datadogV2::model::SpansAttributes>,
17    /// Unique ID of the Span.
18    #[serde(rename = "id")]
19    pub id: Option<String>,
20    /// Type of the span.
21    #[serde(rename = "type")]
22    pub type_: Option<crate::datadogV2::model::SpansType>,
23    #[serde(flatten)]
24    pub additional_properties: std::collections::BTreeMap<String, serde_json::Value>,
25    #[serde(skip)]
26    #[serde(default)]
27    pub(crate) _unparsed: bool,
28}
29
30impl Span {
31    pub fn new() -> Span {
32        Span {
33            attributes: None,
34            id: None,
35            type_: None,
36            additional_properties: std::collections::BTreeMap::new(),
37            _unparsed: false,
38        }
39    }
40
41    pub fn attributes(mut self, value: crate::datadogV2::model::SpansAttributes) -> Self {
42        self.attributes = Some(value);
43        self
44    }
45
46    pub fn id(mut self, value: String) -> Self {
47        self.id = Some(value);
48        self
49    }
50
51    pub fn type_(mut self, value: crate::datadogV2::model::SpansType) -> Self {
52        self.type_ = Some(value);
53        self
54    }
55
56    pub fn additional_properties(
57        mut self,
58        value: std::collections::BTreeMap<String, serde_json::Value>,
59    ) -> Self {
60        self.additional_properties = value;
61        self
62    }
63}
64
65impl Default for Span {
66    fn default() -> Self {
67        Self::new()
68    }
69}
70
71impl<'de> Deserialize<'de> for Span {
72    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
73    where
74        D: Deserializer<'de>,
75    {
76        struct SpanVisitor;
77        impl<'a> Visitor<'a> for SpanVisitor {
78            type Value = Span;
79
80            fn expecting(&self, f: &mut Formatter<'_>) -> fmt::Result {
81                f.write_str("a mapping")
82            }
83
84            fn visit_map<M>(self, mut map: M) -> Result<Self::Value, M::Error>
85            where
86                M: MapAccess<'a>,
87            {
88                let mut attributes: Option<crate::datadogV2::model::SpansAttributes> = None;
89                let mut id: Option<String> = None;
90                let mut type_: Option<crate::datadogV2::model::SpansType> = None;
91                let mut additional_properties: std::collections::BTreeMap<
92                    String,
93                    serde_json::Value,
94                > = std::collections::BTreeMap::new();
95                let mut _unparsed = false;
96
97                while let Some((k, v)) = map.next_entry::<String, serde_json::Value>()? {
98                    match k.as_str() {
99                        "attributes" => {
100                            if v.is_null() {
101                                continue;
102                            }
103                            attributes = Some(serde_json::from_value(v).map_err(M::Error::custom)?);
104                        }
105                        "id" => {
106                            if v.is_null() {
107                                continue;
108                            }
109                            id = Some(serde_json::from_value(v).map_err(M::Error::custom)?);
110                        }
111                        "type" => {
112                            if v.is_null() {
113                                continue;
114                            }
115                            type_ = Some(serde_json::from_value(v).map_err(M::Error::custom)?);
116                            if let Some(ref _type_) = type_ {
117                                match _type_ {
118                                    crate::datadogV2::model::SpansType::UnparsedObject(_type_) => {
119                                        _unparsed = true;
120                                    }
121                                    _ => {}
122                                }
123                            }
124                        }
125                        &_ => {
126                            if let Ok(value) = serde_json::from_value(v.clone()) {
127                                additional_properties.insert(k, value);
128                            }
129                        }
130                    }
131                }
132
133                let content = Span {
134                    attributes,
135                    id,
136                    type_,
137                    additional_properties,
138                    _unparsed,
139                };
140
141                Ok(content)
142            }
143        }
144
145        deserializer.deserialize_any(SpanVisitor)
146    }
147}