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

use crate::errors::*;
use crate::google_trace_exporter_client::GcpCloudTraceExporterClient;
use crate::TraceExportResult;
use futures::future::{BoxFuture, TryFutureExt};
use futures::FutureExt;
use gcloud_sdk::GoogleEnvironment;
use opentelemetry_sdk::export::trace::{ExportResult, SpanData, SpanExporter};
use std::fmt::Formatter;
use std::sync::Arc;

pub struct GcpCloudTraceExporter {
    gcp_export_client: Arc<GcpCloudTraceExporterClient>,
}

impl GcpCloudTraceExporter {
    pub async fn new(google_project_id: &str) -> TraceExportResult<Self> {
        Ok(Self {
            gcp_export_client: Arc::new(GcpCloudTraceExporterClient::new(google_project_id).await?),
        })
    }

    pub async fn for_default_project_id() -> TraceExportResult<Self> {
        let detected_project_id = GoogleEnvironment::detect_google_project_id().await.ok_or_else(||
            GcloudTraceError::SystemError(
                GcloudTraceSystemError::new(
                    "No Google Project ID detected. Please specify it explicitly using env variable: PROJECT_ID or define it as default project for your service accounts".to_string()
                )
            )
        )?;
        Self::new(detected_project_id.as_str()).await
    }
}

impl std::fmt::Debug for GcpCloudTraceExporter {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        write!(f, "GcpCloudTraceExporter")
    }
}

impl SpanExporter for GcpCloudTraceExporter {
    fn export(&mut self, batch: Vec<SpanData>) -> BoxFuture<'static, ExportResult> {
        let client = self.gcp_export_client.clone();
        async move { client.export_batch(batch).map_err(|e| e.into()).await }.boxed()
    }
}