near-async 0.34.4

This crate contains the async helpers specific for nearcore
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

use std::ops::Deref;

/// tokio::runtime::Runtime is not droppable within an async context because of blocking, but we do
/// need to drop runtimes for proper cleanup, so this wraps the runtime so that dropping it stops
/// the runtime in the background.
pub(crate) struct AsyncDroppableRuntime {
    runtime: Option<tokio::runtime::Runtime>,
}

impl AsyncDroppableRuntime {
    pub fn new(runtime: tokio::runtime::Runtime) -> Self {
        Self { runtime: Some(runtime) }
    }
}

impl Deref for AsyncDroppableRuntime {
    type Target = tokio::runtime::Runtime;

    fn deref(&self) -> &Self::Target {
        self.runtime.as_ref().unwrap()
    }
}

impl Drop for AsyncDroppableRuntime {
    fn drop(&mut self) {
        if let Some(runtime) = self.runtime.take() {
            runtime.shutdown_background();
        }
    }
}