thruput-node-supervisor 0.43.0

Local node supervisor for Thruput development
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
45

use std::future::Future;

use thruput_json_rpc_server::{RpcRequestContext, RpcServerError};

pub(crate) async fn within_request_deadline<T>(
    context: RpcRequestContext,
    future: impl Future<Output = Result<T, RpcServerError>>,
) -> Result<T, RpcServerError> {
    let Some(deadline) = context.deadline() else {
        return future.await;
    };
    tokio::time::timeout_at(deadline.into(), future)
        .await
        .map_err(|_| RpcServerError::BackendTimeout)?
}

#[cfg(test)]
mod tests {
    use std::time::Duration;

    use super::*;

    #[tokio::test]
    async fn bounded_context_times_out_backend_work() {
        let result = within_request_deadline(
            RpcRequestContext::new(Some(Duration::from_millis(5))),
            async {
                tokio::time::sleep(Duration::from_secs(1)).await;
                Ok(())
            },
        )
        .await;

        assert!(matches!(result, Err(RpcServerError::BackendTimeout)));
    }

    #[tokio::test]
    async fn unbounded_context_preserves_embedding_policy() {
        let result = within_request_deadline(RpcRequestContext::unbounded(), async { Ok(7) })
            .await
            .expect("unbounded work should complete");

        assert_eq!(result, 7);
    }
}