copernicus_viewer 0.2.0

GUI viewer and library for inspecting and comparing EOPF Zarr products from the Copernicus ecosystem
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

//! Shared Tokio runtime for bridging async S3 object-store I/O into sync zarrs APIs.

use std::sync::OnceLock;

use zarrs::storage::storage_adapter::async_to_sync::AsyncToSyncBlockOn;

static RUNTIME: OnceLock<tokio::runtime::Runtime> = OnceLock::new();

/// Shared multi-thread Tokio runtime used for S3 reads via the zarrs async-to-sync adapter.
pub fn shared_runtime() -> &'static tokio::runtime::Runtime {
    RUNTIME.get_or_init(|| {
        tokio::runtime::Builder::new_multi_thread()
            .enable_all()
            .build()
            .expect("failed to create tokio runtime for S3 I/O")
    })
}

/// [`AsyncToSyncBlockOn`] implementation that delegates to [`shared_runtime`].
#[derive(Clone)]
pub struct TokioBlockOn(tokio::runtime::Handle);

impl TokioBlockOn {
    /// Block on async work using the shared runtime handle.
    pub fn shared() -> Self {
        Self(shared_runtime().handle().clone())
    }
}

impl AsyncToSyncBlockOn for TokioBlockOn {
    fn block_on<F: core::future::Future>(&self, future: F) -> F::Output {
        self.0.block_on(future)
    }
}