1a4a52a173* feat: add local feature for !Send tool handler support * fix: gate streamable HTTP transport on not(local) feature
54 lines
1.4 KiB
Rust
54 lines
1.4 KiB
Rust
use std::{future::poll_fn, marker::PhantomData};
|
|
|
|
use tower_service::Service as TowerService;
|
|
|
|
use super::NotificationContext;
|
|
use crate::service::{MaybeSendFuture, RequestContext, Service, ServiceRole};
|
|
|
|
pub struct TowerHandler<S, R: ServiceRole> {
|
|
pub service: S,
|
|
pub info: R::Info,
|
|
role: PhantomData<R>,
|
|
}
|
|
|
|
impl<S, R: ServiceRole> TowerHandler<S, R> {
|
|
pub fn new(service: S, info: R::Info) -> Self {
|
|
Self {
|
|
service,
|
|
role: PhantomData,
|
|
info,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<S, R: ServiceRole> Service<R> for TowerHandler<S, R>
|
|
where
|
|
S: TowerService<R::PeerReq, Response = R::Resp> + Sync + Send + Clone + 'static,
|
|
S::Error: Into<crate::ErrorData>,
|
|
S::Future: Send,
|
|
{
|
|
async fn handle_request(
|
|
&self,
|
|
request: R::PeerReq,
|
|
_context: RequestContext<R>,
|
|
) -> Result<R::Resp, crate::ErrorData> {
|
|
let mut service = self.service.clone();
|
|
poll_fn(|cx| service.poll_ready(cx))
|
|
.await
|
|
.map_err(Into::into)?;
|
|
let resp = service.call(request).await.map_err(Into::into)?;
|
|
Ok(resp)
|
|
}
|
|
|
|
fn handle_notification(
|
|
&self,
|
|
_notification: R::PeerNot,
|
|
_context: NotificationContext<R>,
|
|
) -> impl Future<Output = Result<(), crate::ErrorData>> + MaybeSendFuture + '_ {
|
|
std::future::ready(Ok(()))
|
|
}
|
|
|
|
fn get_info(&self) -> R::Info {
|
|
self.info.clone()
|
|
}
|
|
}
|