OpenSource/axum
OpenSource/axum
1
0
Fork 0
mirror of https://github.com/tokio-rs/axum.git synced 2025-02-26 22:18:07 +01:00
Code Issues Projects Releases Packages Wiki Activity

Seal Handler trait

Browse source
This commit is contained in:
David Pedersen 2021-05-30 11:19:36 +02:00
parent 2e16842431
commit f08b73f7b2
1 changed files with 19 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
src/lib.rs
View file

@ -216,11 +216,21 @@ impl From<Infallible> for Error {
} }
} }
// TODO(david): make this trait sealed mod sealed {
pub trait HiddentTrait {}
pub struct Hidden;
impl HiddentTrait for Hidden {}
}
#[async_trait] #[async_trait]
pub trait Handler<In>: Sized { pub trait Handler<In>: Sized {
type ResponseBody; type ResponseBody;
// This seals the trait. We cannot use the regular "sealed super trait" approach
// due to coherence.
#[doc(hidden)]
type Sealed: sealed::HiddentTrait;
async fn call(self, req: Request<Body>) -> Result<Response<Self::ResponseBody>, Error>; async fn call(self, req: Request<Body>) -> Result<Response<Self::ResponseBody>, Error>;
fn layer<L>(self, layer: L) -> Layered<L::Service, In> fn layer<L>(self, layer: L) -> Layered<L::Service, In>
@ -239,6 +249,8 @@ where
{ {
type ResponseBody = B; type ResponseBody = B;
type Sealed = sealed::Hidden;
async fn call(self, req: Request<Body>) -> Result<Response<Self::ResponseBody>, Error> { async fn call(self, req: Request<Body>) -> Result<Response<Self::ResponseBody>, Error> {
self(req).await self(req).await
} }
@ -256,6 +268,8 @@ macro_rules! impl_handler {
{ {
type ResponseBody = B; type ResponseBody = B;
type Sealed = sealed::Hidden;
async fn call(self, mut req: Request<Body>) -> Result<Response<Self::ResponseBody>, Error> { async fn call(self, mut req: Request<Body>) -> Result<Response<Self::ResponseBody>, Error> {
let $head = $head::from_request(&mut req).await?; let $head = $head::from_request(&mut req).await?;
let res = self(req, $head).await?; let res = self(req, $head).await?;
@ -276,6 +290,8 @@ macro_rules! impl_handler {
{ {
type ResponseBody = B; type ResponseBody = B;
type Sealed = sealed::Hidden;
async fn call(self, mut req: Request<Body>) -> Result<Response<Self::ResponseBody>, Error> { async fn call(self, mut req: Request<Body>) -> Result<Response<Self::ResponseBody>, Error> {
let $head = $head::from_request(&mut req).await?; let $head = $head::from_request(&mut req).await?;
$( $(
@ -315,6 +331,8 @@ where
{ {
type ResponseBody = B; type ResponseBody = B;
type Sealed = sealed::Hidden;
async fn call(self, req: Request<Body>) -> Result<Response<Self::ResponseBody>, Error> { async fn call(self, req: Request<Body>) -> Result<Response<Self::ResponseBody>, Error> {
self.svc self.svc
.oneshot(req) .oneshot(req)