reth_network/
flattened_response.rs

1use futures::Future;
2use pin_project::pin_project;
3use std::{
4    pin::Pin,
5    task::{Context, Poll},
6};
7use tokio::sync::oneshot::{error::RecvError, Receiver};
8
9/// Flatten a [Receiver] message in order to get rid of the [RecvError] result
10#[derive(Debug)]
11#[pin_project]
12pub struct FlattenedResponse<T> {
13    #[pin]
14    receiver: Receiver<T>,
15}
16
17impl<T, E> Future for FlattenedResponse<Result<T, E>>
18where
19    E: From<RecvError>,
20{
21    type Output = Result<T, E>;
22
23    fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
24        let this = self.project();
25
26        this.receiver.poll(cx).map(|r| r.unwrap_or_else(|err| Err(err.into())))
27    }
28}
29
30impl<T> From<Receiver<T>> for FlattenedResponse<T> {
31    fn from(value: Receiver<T>) -> Self {
32        Self { receiver: value }
33    }
34}