delta-funnel 0.5.0

Lightweight, fast Delta Lake to SQL Server loads with DataFusion SQL and native TDS
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

//! Profiled DataFusion object-store transport.

use std::future::Future;

#[cfg(feature = "perfetto-profile")]
use tracing::Instrument;

#[cfg(feature = "perfetto-profile")]
use super::operator_activity::current_datafusion_object_store_transport_span;

#[cfg(feature = "perfetto-profile")]
pub(super) async fn await_object_store_transport<F>(request: F) -> F::Output
where
    F: Future,
{
    match current_datafusion_object_store_transport_span() {
        Some(span) => request.instrument(span).await,
        None => request.await,
    }
}

#[cfg(not(feature = "perfetto-profile"))]
pub(super) async fn await_object_store_transport<F>(request: F) -> F::Output
where
    F: Future,
{
    request.await
}

#[cfg(test)]
mod tests {
    use super::await_object_store_transport;

    #[tokio::test]
    async fn transport_without_an_active_task_preserves_the_output() {
        assert_eq!(await_object_store_transport(async { 42 }).await, 42);
    }
}