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
|
//! Tracing exporter to write spans to a file in the OTLP JSON format.
// TODO: If https://github.com/open-telemetry/opentelemetry-rust/issues/2602 gets fixed, we can
// replace this entire file with whatever upstream has for doing this.
use opentelemetry_proto::transform::common::tonic::ResourceAttributesWithSchema;
use opentelemetry_proto::transform::trace::tonic::group_spans_by_resource_and_scope;
use opentelemetry_sdk::{
Resource,
error::{OTelSdkError, OTelSdkResult},
trace::SpanExporter,
};
use std::{
fmt::Debug,
io::{LineWriter, Write},
sync::{Arc, Mutex},
};
/// Tracing exporter to write OTLP JSON to a file (or anything else that implements [`LineWriter`].
#[derive(Debug)]
pub(crate) struct FileExporter<W: Write + Send + Debug> {
/// The [`LineWriter`] to write to.
writer: Arc<Mutex<LineWriter<W>>>,
/// The [`Resource`] to associate spans with.
resource: Resource,
}
impl<W: Write + Send + Debug> FileExporter<W> {
/// Create a new [`FileExporter`]
pub(crate) fn new(writer: W, resource: Resource) -> Self {
Self {
writer: Arc::new(Mutex::new(LineWriter::new(writer))),
resource,
}
}
}
// Note that OpenTelemetry can only represent events as children of spans, so this exporter only
// works on spans. If you want a event to be exported, you need to make sure it exists within some
// span.
impl<W: Write + Send + Debug> SpanExporter for FileExporter<W> {
fn export(
&self,
batch: Vec<opentelemetry_sdk::trace::SpanData>,
) -> impl futures::Future<
Output = std::result::Result<(), opentelemetry_sdk::error::OTelSdkError>,
> + std::marker::Send {
let resource = ResourceAttributesWithSchema::from(&self.resource);
let data = group_spans_by_resource_and_scope(batch, &resource);
let mut writer = self.writer.lock().expect("Lock poisoned");
Box::pin(std::future::ready('write: {
// See https://opentelemetry.io/docs/specs/otel/protocol/file-exporter/ for format
if let Err(err) = serde_json::to_writer(
writer.get_mut(),
&serde_json::json!({"resourceSpans": data}),
) {
break 'write Err(OTelSdkError::InternalFailure(err.to_string()));
}
if let Err(err) = writer.write(b"\n") {
break 'write Err(OTelSdkError::InternalFailure(err.to_string()));
}
Ok(())
}))
}
fn force_flush(&self) -> OTelSdkResult {
let mut writer = self
.writer
.lock()
.map_err(|e| OTelSdkError::InternalFailure(e.to_string()))?;
writer
.flush()
.map_err(|e| OTelSdkError::InternalFailure(e.to_string()))
}
fn set_resource(&mut self, res: &opentelemetry_sdk::Resource) {
self.resource = res.clone();
}
}
|
