use crate::response::Response;
use axum_core::extract::Request;
use futures_util::future::Map;
use pin_project_lite::pin_project;
use std::{convert::Infallible, future::Future, pin::Pin, task::Context};
use tower::util::Oneshot;
use tower_service::Service;
opaque_future! {
pub type IntoServiceFuture<F> =
Map<
F,
fn(Response) -> Result<Response, Infallible>,
>;
}
pin_project! {
pub struct LayeredFuture<S>
where
S: Service<Request>,
{
#[pin]
inner: Map<Oneshot<S, Request>, fn(Result<S::Response, S::Error>) -> Response>,
}
}
impl<S> LayeredFuture<S>
where
S: Service<Request>,
{
pub(super) fn new(
inner: Map<Oneshot<S, Request>, fn(Result<S::Response, S::Error>) -> Response>,
) -> Self {
Self { inner }
}
}
impl<S> Future for LayeredFuture<S>
where
S: Service<Request>,
{
type Output = Response;
#[inline]
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> std::task::Poll<Self::Output> {
self.project().inner.poll(cx)
}
}