Function await_with_runtime

crates/integrations/datafusion/src/runtime.rs:48–57

These helpers work around DataFusion FFI callbacks that may run without an entered Tokio runtime. See https://github.com/apache/datafusion/issues/16312. A global OnceLock avoids creating a new runtime on every call; if DataFusion fixes runtime propagation end-to-end, we should be able to remove these manual fallback runtimes.

(future: F)

// if DataFusion fixes runtime propagation end-to-end, we should be able to
// remove these manual fallback runtimes.
pub(crate) async fn await_with_runtime<F>(future: F) -> F::Output
where
    F: Future,
{
    if Handle::try_current().is_ok() {
        future.await
    } else {
        global_runtime().block_on(future)
    }
}

// The blocking variant is for synchronous DataFusion FFI callbacks such as
// CatalogProvider::schema(), where we cannot `.await` directly.