use std::future::Future;
use std::pin::Pin;
use std::sync::Arc;
use http::HeaderMap;
use crate::HttpResult;
type AsyncHttpHeaderInjectorFuture<'a> = Pin<Box<dyn Future<Output = HttpResult<()>> + Send + 'a>>;
type AsyncHttpHeaderInjectorFn =
dyn for<'a> Fn(&'a mut HeaderMap) -> AsyncHttpHeaderInjectorFuture<'a> + Send + Sync + 'static;
#[derive(Clone)]
pub struct AsyncHttpHeaderInjector {
inner: Arc<AsyncHttpHeaderInjectorFn>,
}
impl std::fmt::Debug for AsyncHttpHeaderInjector {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.debug_struct("AsyncHttpHeaderInjector")
.finish_non_exhaustive()
}
}
impl AsyncHttpHeaderInjector {
pub fn new<F>(injector: F) -> Self
where
F: for<'a> Fn(&'a mut HeaderMap) -> AsyncHttpHeaderInjectorFuture<'a>
+ Send
+ Sync
+ 'static,
{
Self {
inner: Arc::new(injector),
}
}
pub async fn apply(&self, headers: &mut HeaderMap) -> HttpResult<()> {
(self.inner)(headers).await
}
}