tracelite 0.1.26

An alternative to the tracing crate which is specialized for OpenTelemetry traces
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

use crate::tracer::globals;
use super::SpanExporter;

pub async fn run_tokio_export_loop<B>(
    mut batch_receiver: tokio::sync::mpsc::UnboundedReceiver<B>,
    exporter: impl SpanExporter<B>,
    tracer_autoflush_interval: std::time::Duration,
){
    loop {
        let autoflush = tokio::time::sleep(tracer_autoflush_interval);
        tokio::select! {
            opt = batch_receiver.recv() => {
                println!("[DEBUG] tracelite: background worker received batch or dead sender");
                match opt {
                    Some(batch) => exporter.export(&batch).await,
                    None => return // channel senders dropped
                }
            }
            _ = autoflush => {
                println!("[DEBUG] tracelite: background worker performs autoflush");
                globals::tracer().ok().map(|t| t.flush());
            }
        };
    }
}

pub fn spawn_tokio_export_task<B: Send + 'static>(
    exporter: impl SpanExporter<B>,
    tracer_autoflush_interval: std::time::Duration,
) -> impl Fn(B) {
    let (batch_sender, batch_receiver) = tokio::sync::mpsc::unbounded_channel();

    tokio::spawn(async move {
        run_tokio_export_loop(batch_receiver, exporter, tracer_autoflush_interval).await
    });

    move |batch| {
        // TODO how to react to dropped receiver?
        match batch_sender.send(batch) {
            Ok(()) => println!("[DEBUG] tracelite: sent batch to background worker"),
            Err(err) => eprintln!("[ERROR] tracelite: failed to send batch to background worker: {err}")
        }
    }
}