The Anatomy of a Service
What is Smithy? At a high-level, it's a grammar for specifying services while leaving the business logic undefined. A Smithy Service specifies a collection of function signatures in the form of Operations, their purpose is to encapsulate business logic. A Smithy implementation should, for each Smithy Service, provide a builder, which accepts functions conforming to said signatures, and returns a service subject to the semantics specified by the model.
This survey is disinterested in the actual Kotlin implementation of the code generator, and instead focuses on the structure of the generated Rust code and how it relates to the Smithy model. The intended audience is new contributors and users interested in internal details.
During the survey we will use the pokemon.smithy
model as a reference:
/// A Pokémon species forms the basis for at least one Pokémon.
@title("Pokémon Species")
resource PokemonSpecies {
identifiers: {
name: String
},
read: GetPokemonSpecies,
}
/// A users current Pokémon storage.
resource Storage {
identifiers: {
user: String
},
read: GetStorage,
}
/// The Pokémon Service allows you to retrieve information about Pokémon species.
@title("Pokémon Service")
@restJson1
service PokemonService {
version: "2021-12-01",
resources: [PokemonSpecies, Storage],
operations: [
GetServerStatistics,
DoNothing,
CapturePokemon,
CheckHealth
],
}
Smithy Rust will use this model to produce the following API:
#![allow(unused)] fn main() { extern crate pokemon_service_server_sdk; extern crate aws_smithy_http_server; use aws_smithy_http_server::protocol::rest_json_1::{RestJson1, router::RestRouter}; use aws_smithy_http_server::routing::{Route, RoutingService}; use pokemon_service_server_sdk::{input::*, output::*, error::*, operation_shape::*, PokemonServiceConfig, PokemonService}; // A handler for the `GetPokemonSpecies` operation (the `PokemonSpecies` resource). async fn get_pokemon_species(input: GetPokemonSpeciesInput) -> Result<GetPokemonSpeciesOutput, GetPokemonSpeciesError> { todo!() } let config = PokemonServiceConfig::builder().build(); // Use the service builder to create `PokemonService`. let pokemon_service = PokemonService::builder(config) // Pass the handler directly to the service builder... .get_pokemon_species(get_pokemon_species) /* other operation setters */ .build() .expect("failed to create an instance of the Pokémon service"); let pokemon_service: PokemonService<RoutingService<RestRouter<Route>, RestJson1>> = pokemon_service; }
Operations
A Smithy Operation specifies the input, output, and possible errors of an API operation. One might characterize a Smithy Operation as syntax for specifying a function type.
We represent this in Rust using the OperationShape
trait:
#![allow(unused)] fn main() { extern crate aws_smithy_http_server; use aws_smithy_http_server::shape_id::ShapeId; pub trait OperationShape { /// The name of the operation. const ID: ShapeId; /// The operation input. type Input; /// The operation output. type Output; /// The operation error. [`Infallible`](std::convert::Infallible) in the case where no error /// exists. type Error; } use aws_smithy_http_server::operation::OperationShape as OpS; impl<T: OpS> OperationShape for T { const ID: ShapeId = <T as OpS>::ID; type Input = <T as OpS>::Input; type Output = <T as OpS>::Output; type Error = <T as OpS>::Error; } }
For each Smithy Operation shape,
/// Retrieve information about a Pokémon species.
@readonly
@http(uri: "/pokemon-species/{name}", method: "GET")
operation GetPokemonSpecies {
input: GetPokemonSpeciesInput,
output: GetPokemonSpeciesOutput,
errors: [ResourceNotFoundException],
}
the following implementation is generated
#![allow(unused)] fn main() { extern crate pokemon_service_server_sdk; extern crate aws_smithy_http_server; use aws_smithy_http_server::{operation::OperationShape, shape_id::ShapeId}; use pokemon_service_server_sdk::{input::*, output::*, error::*}; /// Retrieve information about a Pokémon species. pub struct GetPokemonSpecies; impl OperationShape for GetPokemonSpecies { const ID: ShapeId = ShapeId::new("com.aws.example#GetPokemonSpecies", "com.aws.example", "GetPokemonSpecies"); type Input = GetPokemonSpeciesInput; type Output = GetPokemonSpeciesOutput; type Error = GetPokemonSpeciesError; } }
where GetPokemonSpeciesInput
, GetPokemonSpeciesOutput
are both generated from the Smithy structures and GetPokemonSpeciesError
is an enum generated from the errors: [ResourceNotFoundException]
.
Note that the GetPokemonSpecies
marker structure is a zero-sized type (ZST), and therefore does not exist at runtime - it is a way to attach operation-specific data on an entity within the type system.
The following nomenclature will aid us in our survey. We describe a tower::Service
as a "model service" if its request and response are Smithy structures, as defined by the OperationShape
trait - the GetPokemonSpeciesInput
, GetPokemonSpeciesOutput
, and GetPokemonSpeciesError
described above. Similarly, we describe a tower::Service
as a "HTTP service" if its request and response are http
structures - http::Request
and http::Response
.
The constructors exist on the marker ZSTs as an extension trait to OperationShape
, namely OperationShapeExt
:
#![allow(unused)] fn main() { extern crate aws_smithy_http_server; use aws_smithy_http_server::operation::*; /// An extension trait over [`OperationShape`]. pub trait OperationShapeExt: OperationShape { /// Creates a new [`Service`] for well-formed [`Handler`]s. fn from_handler<H, Exts>(handler: H) -> IntoService<Self, H> where H: Handler<Self, Exts>, Self: Sized; /// Creates a new [`Service`] for well-formed [`Service`](tower::Service)s. fn from_service<S, Exts>(svc: S) -> Normalize<Self, S> where S: OperationService<Self, Exts>, Self: Sized; } use aws_smithy_http_server::operation::OperationShapeExt as OpS; impl<T: OpS> OperationShapeExt for T { fn from_handler<H, Exts>(handler: H) -> IntoService<Self, H> where H: Handler<Self, Exts>, Self: Sized { <T as OpS>::from_handler(handler) } fn from_service<S, Exts>(svc: S) -> Normalize<Self, S> where S: OperationService<Self, Exts>, Self: Sized { <T as OpS>::from_service(svc) } } }
Observe that there are two constructors provided: from_handler
which takes a H: Handler
and from_service
which takes a S: OperationService
. In both cases Self
is passed as a parameter to the traits - this constrains handler: H
and svc: S
to the signature given by the implementation of OperationShape
on Self
.
The Handler
and OperationService
both serve a similar purpose - they provide a common interface for converting to a model service S
.
- The
Handler<GetPokemonSpecies>
trait covers all async functions takingGetPokemonSpeciesInput
and asynchronously returning aResult<GetPokemonSpeciesOutput, GetPokemonSpeciesError>
. - The
OperationService<GetPokemonSpecies>
trait covers alltower::Service
s with requestGetPokemonSpeciesInput
, responseGetPokemonSpeciesOutput
and errorGetPokemonSpeciesOutput
.
The from_handler
constructor is used in the following way:
#![allow(unused)] fn main() { extern crate pokemon_service_server_sdk; extern crate aws_smithy_http_server; use pokemon_service_server_sdk::{ input::GetPokemonSpeciesInput, output::GetPokemonSpeciesOutput, error::GetPokemonSpeciesError, operation_shape::GetPokemonSpecies }; use aws_smithy_http_server::operation::OperationShapeExt; async fn get_pokemon_service(input: GetPokemonSpeciesInput) -> Result<GetPokemonSpeciesOutput, GetPokemonSpeciesError> { todo!() } let operation = GetPokemonSpecies::from_handler(get_pokemon_service); }
Alternatively, from_service
constructor:
#![allow(unused)] fn main() { extern crate pokemon_service_server_sdk; extern crate aws_smithy_http_server; extern crate tower; use pokemon_service_server_sdk::{ input::GetPokemonSpeciesInput, output::GetPokemonSpeciesOutput, error::GetPokemonSpeciesError, operation_shape::GetPokemonSpecies }; use aws_smithy_http_server::operation::OperationShapeExt; use std::task::{Context, Poll}; use tower::Service; struct Svc { /* ... */ } impl Service<GetPokemonSpeciesInput> for Svc { type Response = GetPokemonSpeciesOutput; type Error = GetPokemonSpeciesError; type Future = /* Future<Output = Result<Self::Response, Self::Error>> */ std::future::Ready<Result<Self::Response, Self::Error>>; fn poll_ready(&mut self, ctx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> { todo!() } fn call(&mut self, input: GetPokemonSpeciesInput) -> Self::Future { todo!() } } let svc: Svc = Svc { /* ... */ }; let operation = GetPokemonSpecies::from_service(svc); }
To summarize a model service constructed can be constructed from a Handler
or a OperationService
subject to the constraints of an OperationShape
. More detailed information on these conversions is provided in the Handler and OperationService section Rust docs.
Serialization and Deserialization
A Smithy protocol specifies the serialization/deserialization scheme - how a HTTP request is transformed into a modelled input and a modelled output to a HTTP response. The is formalized using the FromRequest
and IntoResponse
traits:
#![allow(unused)] fn main() { extern crate aws_smithy_http_server; extern crate http; use aws_smithy_http_server::body::BoxBody; use std::future::Future; /// Provides a protocol aware extraction from a [`Request`]. This consumes the /// [`Request`], in contrast to [`FromParts`]. pub trait FromRequest<Protocol, B>: Sized { type Rejection: IntoResponse<Protocol>; type Future: Future<Output = Result<Self, Self::Rejection>>; /// Extracts `self` from a [`Request`] asynchronously. fn from_request(request: http::Request<B>) -> Self::Future; } /// A protocol aware function taking `self` to [`http::Response`]. pub trait IntoResponse<Protocol> { /// Performs a conversion into a [`http::Response`]. fn into_response(self) -> http::Response<BoxBody>; } use aws_smithy_http_server::request::FromRequest as FR; impl<P, B, T: FR<P, B>> FromRequest<P, B> for T { type Rejection = <T as FR<P, B>>::Rejection; type Future = <T as FR<P, B>>::Future; fn from_request(request: http::Request<B>) -> Self::Future { <T as FR<P, B>>::from_request(request) } } use aws_smithy_http_server::response::IntoResponse as IR; impl<P, T: IR<P>> IntoResponse<P> for T { fn into_response(self) -> http::Response<BoxBody> { <T as IR<P>>::into_response(self) } } }
Note that both traits are parameterized by Protocol
. These protocols exist as ZST marker structs:
#![allow(unused)] fn main() { extern crate aws_smithy_http_server; use aws_smithy_http_server::protocol::{ aws_json_10::AwsJson1_0 as _, aws_json_11::AwsJson1_1 as _, rest_json_1::RestJson1 as _, rest_xml::RestXml as _, }; /// [AWS REST JSON 1.0 Protocol](https://awslabs.github.io/smithy/2.0/aws/protocols/aws-restjson1-protocol.html). pub struct RestJson1; /// [AWS REST XML Protocol](https://awslabs.github.io/smithy/2.0/aws/protocols/aws-restxml-protocol.html). pub struct RestXml; /// [AWS JSON 1.0 Protocol](https://awslabs.github.io/smithy/2.0/aws/protocols/aws-json-1_0-protocol.html). pub struct AwsJson1_0; /// [AWS JSON 1.1 Protocol](https://awslabs.github.io/smithy/2.0/aws/protocols/aws-json-1_1-protocol.html). pub struct AwsJson1_1; }
Upgrading a Model Service
We can "upgrade" a model service to a HTTP service using FromRequest
and IntoResponse
described in the prior section:
stateDiagram-v2 direction LR HttpService: HTTP Service [*] --> from_request: HTTP Request state HttpService { direction LR ModelService: Model Service from_request --> ModelService: Model Input ModelService --> into_response: Model Output } into_response --> [*]: HTTP Response
This is formalized by the Upgrade<Protocol, Op, S>
HTTP service. The tower::Service
implementation is approximately:
impl<P, Op, S> Service<http::Request> for Upgrade<P, Op, S>
where
Input: FromRequest<P, B>,
S: Service<Input>,
S::Response: IntoResponse<P>,
S::Error: IntoResponse<P>,
{
async fn call(&mut self, request: http::Request) -> http::Response {
let model_request = match <Op::Input as OperationShape>::from_request(request).await {
Ok(ok) => ok,
Err(err) => return err.into_response()
};
let model_response = self.model_service.call(model_request).await;
model_response.into_response()
}
}
When we GetPokemonSpecies::from_handler
or GetPokemonSpecies::from_service
, the model service produced, S
, will meet the constraints above.
There is an associated Plugin
, UpgradePlugin
which constructs Upgrade
from a service.
The upgrade procedure is finalized by the application of the Layer
L
, referenced in Operation<S, L>
. In this way the entire upgrade procedure takes an Operation<S, L>
and returns a HTTP service.
stateDiagram-v2 direction LR [*] --> UpgradePlugin: HTTP Request state HttpPlugin { state UpgradePlugin { direction LR [*] --> S: Model Input S --> [*] : Model Output state ModelPlugin { S } } } UpgradePlugin --> [*]: HTTP Response
Note that the S
is specified by logic written, in Rust, by the customer, whereas UpgradePlugin
is specified entirely by Smithy model via the protocol, HTTP bindings, etc.
Routers
Different protocols supported by Smithy enjoy different routing mechanisms, for example, AWS JSON 1.0 uses the X-Amz-Target
header to select an operation, whereas AWS REST XML uses the HTTP label trait.
Despite their differences, all routing mechanisms satisfy a common interface. This is formalized using the Router trait:
#![allow(unused)] fn main() { extern crate aws_smithy_http_server; extern crate http; /// An interface for retrieving an inner [`Service`] given a [`http::Request`]. pub trait Router<B> { type Service; type Error; /// Matches a [`http::Request`] to a target [`Service`]. fn match_route(&self, request: &http::Request<B>) -> Result<Self::Service, Self::Error>; } }
which provides the ability to determine an inner HTTP service from a collection using a &http::Request
.
Types which implement the Router
trait are converted to a HTTP service via the RoutingService
struct:
/// A [`Service`] using a [`Router`] `R` to redirect messages to specific routes.
///
/// The `Protocol` parameter is used to determine the serialization of errors.
pub struct RoutingService<R, Protocol> {
router: R,
_protocol: PhantomData<Protocol>,
}
impl<R, P> Service<http::Request> for RoutingService<R, P>
where
R: Router<B>,
R::Service: Service<http::Request, Response = http::Response>,
R::Error: IntoResponse<P> + Error,
{
type Response = http::Response;
type Error = /* implementation detail */;
async fn call(&mut self, req: http::Request<B>) -> Result<Self::Response, Self::Error> {
match self.router.match_route(&req) {
// Successfully routed, use the routes `Service::call`.
Ok(ok) => ok.oneshot(req).await,
// Failed to route, use the `R::Error`s `IntoResponse<P>`.
Err(error) => {
debug!(%error, "failed to route");
Err(Box::new(error.into_response()))
}
}
}
}
The RouterService
is the final piece necessary to form a functioning composition - it is used to aggregate together the HTTP services, created via the upgrade procedure, into a single HTTP service which can be presented to the customer.
stateDiagram state in <<fork>> direction LR [*] --> in state RouterService { direction LR in --> ServiceA in --> ServiceB in --> ServiceC } ServiceA --> [*] ServiceB --> [*] ServiceC --> [*]
Plugins
A Plugin
is a
[tower::Layer
] with two extra type parameters, Service
and Operation
, corresponding to Smithy Service and Smithy Operation. This allows the middleware to be
parameterized them and change behavior depending on the context in which it's applied.
#![allow(unused)] fn main() { extern crate aws_smithy_http_server; pub trait Plugin<Service, Operation, T> { type Output; fn apply(&self, input: T) -> Self::Output; } use aws_smithy_http_server::plugin::Plugin as Pl; impl<Ser, Op, T, U: Pl<Ser, Op, T>> Plugin<Ser, Op, T> for U { type Output = <U as Pl<Ser, Op, T>>::Output; fn apply(&self, input: T) -> Self::Output { <U as Pl<Ser, Op, T>>::apply(self, input) } } }
An example Plugin
implementation can be found in /examples/pokemon-service/src/plugin.rs.
Plugins can be applied in two places:
- HTTP plugins, which are applied pre-deserialization/post-serialization, acting on HTTP requests/responses.
- Model plugins, which are applied post-deserialization/pre-serialization, acting on model inputs/outputs/errors.
stateDiagram-v2 direction LR [*] --> S: HTTP Request state HttpPlugin { state UpgradePlugin { state ModelPlugin { S } } } S --> [*]: HTTP Response
The service builder API requires plugins to be specified upfront - they must be
registered in the config object, which is passed as an argument to builder
.
Plugins cannot be modified afterwards.
You might find yourself wanting to apply multiple plugins to your service.
This can be accommodated via [HttpPlugins
] and [ModelPlugins
].
#![allow(unused)] fn main() { extern crate aws_smithy_http_server; use aws_smithy_http_server::plugin::HttpPlugins; use aws_smithy_http_server::plugin::IdentityPlugin as LoggingPlugin; use aws_smithy_http_server::plugin::IdentityPlugin as MetricsPlugin; let http_plugins = HttpPlugins::new().push(LoggingPlugin).push(MetricsPlugin); }
The plugins' runtime logic is executed in registration order.
In the example above, LoggingPlugin
would run first, while MetricsPlugin
is executed last.
If you are vending a plugin, you can leverage HttpPlugins
or ModelPlugins
as an extension point: you can add custom methods to it using an extension trait.
For example:
#![allow(unused)] fn main() { extern crate aws_smithy_http_server; use aws_smithy_http_server::plugin::{HttpPlugins, PluginStack}; use aws_smithy_http_server::plugin::IdentityPlugin as LoggingPlugin; use aws_smithy_http_server::plugin::IdentityPlugin as AuthPlugin; pub trait AuthPluginExt<CurrentPlugins> { fn with_auth(self) -> HttpPlugins<PluginStack<AuthPlugin, CurrentPlugins>>; } impl<CurrentPlugins> AuthPluginExt<CurrentPlugins> for HttpPlugins<CurrentPlugins> { fn with_auth(self) -> HttpPlugins<PluginStack<AuthPlugin, CurrentPlugins>> { self.push(AuthPlugin) } } let http_plugins = HttpPlugins::new() .push(LoggingPlugin) // Our custom method! .with_auth(); }
Builders
The service builder is the primary public API, generated for every Smithy Service. At a high-level, the service builder takes as input a function for each Smithy Operation and returns a single HTTP service. The signature of each function, also known as handlers, must match the constraints of the corresponding Smithy model.
You can create an instance of a service builder by calling builder
on the corresponding service struct.
#![allow(unused)] fn main() { extern crate aws_smithy_http_server; use aws_smithy_http_server::routing::Route; /// The service builder for [`PokemonService`]. /// /// Constructed via [`PokemonService::builder`]. pub struct PokemonServiceBuilder<Body, HttpPl, ModelPl> { capture_pokemon_operation: Option<Route<Body>>, empty_operation: Option<Route<Body>>, get_pokemon_species: Option<Route<Body>>, get_server_statistics: Option<Route<Body>>, get_storage: Option<Route<Body>>, health_check_operation: Option<Route<Body>>, http_plugin: HttpPl, model_plugin: ModelPl, } }
The builder has two setter methods for each Smithy Operation in the Smithy Service:
pub fn get_pokemon_species<HandlerType, HandlerExtractors, UpgradeExtractors>(self, handler: HandlerType) -> Self
where
HandlerType:Handler<GetPokemonSpecies, HandlerExtractors>,
ModelPl: Plugin<
PokemonService,
GetPokemonSpecies,
IntoService<GetPokemonSpecies, HandlerType>
>,
UpgradePlugin::<UpgradeExtractors>: Plugin<
PokemonService,
GetPokemonSpecies,
ModelPlugin::Output
>,
HttpPl: Plugin<
PokemonService,
GetPokemonSpecies,
UpgradePlugin::<UpgradeExtractors>::Output
>,
{
let svc = GetPokemonSpecies::from_handler(handler);
let svc = self.model_plugin.apply(svc);
let svc = UpgradePlugin::<UpgradeExtractors>::new()
.apply(svc);
let svc = self.http_plugin.apply(svc);
self.get_pokemon_species_custom(svc)
}
pub fn get_pokemon_species_service<S, ServiceExtractors, UpgradeExtractors>(self, service: S) -> Self
where
S: OperationService<GetPokemonSpecies, ServiceExtractors>,
ModelPl: Plugin<
PokemonService,
GetPokemonSpecies,
Normalize<GetPokemonSpecies, S>
>,
UpgradePlugin::<UpgradeExtractors>: Plugin<
PokemonService,
GetPokemonSpecies,
ModelPlugin::Output
>,
HttpPl: Plugin<
PokemonService,
GetPokemonSpecies,
UpgradePlugin::<UpgradeExtractors>::Output
>,
{
let svc = GetPokemonSpecies::from_service(service);
let svc = self.model_plugin.apply(svc);
let svc = UpgradePlugin::<UpgradeExtractors>::new().apply(svc);
let svc = self.http_plugin.apply(svc);
self.get_pokemon_species_custom(svc)
}
pub fn get_pokemon_species_custom<S>(mut self, svc: S) -> Self
where
S: Service<Request<Body>, Response = Response<BoxBody>, Error = Infallible>,
{
self.get_pokemon_species = Some(Route::new(svc));
self
}
Handlers and operations are upgraded to a Route
as soon as they are registered against the service builder. You can think of Route
as a boxing layer in disguise.
You can transform a builder instance into a complete service (PokemonService
) using one of the following methods:
build
. The transformation fails if one or more operations do not have a registered handler;build_unchecked
. The transformation never fails, but we return500
s for all operations that do not have a registered handler.
Both builder methods take care of:
- Pair each handler with the routing information for the corresponding operation;
- Collect all
(routing_info, handler)
pairs into aRouter
; - Transform the
Router
implementation into a HTTP service viaRouterService
; - Wrap the
RouterService
in a newtype given by the service name,PokemonService
.
The final outcome, an instance of PokemonService
, looks roughly like this:
#![allow(unused)] fn main() { extern crate aws_smithy_http_server; use aws_smithy_http_server::{routing::RoutingService, protocol::rest_json_1::{router::RestRouter, RestJson1}}; /// The Pokémon Service allows you to retrieve information about Pokémon species. #[derive(Clone)] pub struct PokemonService<S> { router: RoutingService<RestRouter<S>, RestJson1>, } }
The following schematic summarizes the composition:
stateDiagram-v2 state in <<fork>> state "GetPokemonSpecies" as C1 state "GetStorage" as C2 state "DoNothing" as C3 state "..." as C4 direction LR [*] --> in : HTTP Request UpgradePlugin --> [*]: HTTP Response state PokemonService { state RoutingService { in --> UpgradePlugin: HTTP Request in --> C2: HTTP Request in --> C3: HTTP Request in --> C4: HTTP Request state C1 { state HttpPlugin { state UpgradePlugin { direction LR [*] --> S: Model Input S --> [*] : Model Output state ModelPlugin { S } } } } C2 C3 C4 } } C2 --> [*]: HTTP Response C3 --> [*]: HTTP Response C4 --> [*]: HTTP Response
Accessing Unmodelled Data
An additional omitted detail is that we provide an "escape hatch" allowing Handler
s and OperationService
s to accept data that isn't modelled. In addition to accepting Op::Input
they can accept additional arguments which implement the FromParts
trait:
#![allow(unused)] fn main() { extern crate aws_smithy_http_server; extern crate http; use http::request::Parts; use aws_smithy_http_server::response::IntoResponse; /// Provides a protocol aware extraction from a [`Request`]. This borrows the /// [`Parts`], in contrast to [`FromRequest`]. pub trait FromParts<Protocol>: Sized { /// The type of the failures yielded extraction attempts. type Rejection: IntoResponse<Protocol>; /// Extracts `self` from a [`Parts`] synchronously. fn from_parts(parts: &mut Parts) -> Result<Self, Self::Rejection>; } use aws_smithy_http_server::request::FromParts as FP; impl<P, T: FP<P>> FromParts<P> for T { type Rejection = <T as FP<P>>::Rejection; fn from_parts(parts: &mut Parts) -> Result<Self, Self::Rejection> { <T as FP<P>>::from_parts(parts) } } }
This differs from FromRequest
trait, introduced in Serialization and Deserialization, as it's synchronous and has non-consuming access to Parts
, rather than the entire Request.
pub struct Parts {
pub method: Method,
pub uri: Uri,
pub version: Version,
pub headers: HeaderMap<HeaderValue>,
pub extensions: Extensions,
/* private fields */
}
This is commonly used to access types stored within Extensions
which have been inserted by a middleware. An Extension
struct implements FromParts
to support this use case:
#![allow(unused)] fn main() { extern crate aws_smithy_http_server; extern crate http; extern crate thiserror; use aws_smithy_http_server::{body::BoxBody, request::FromParts, response::IntoResponse}; use http::status::StatusCode; use thiserror::Error; fn empty() -> BoxBody { todo!() } /// Generic extension type stored in and extracted from [request extensions]. /// /// This is commonly used to share state across handlers. /// /// If the extension is missing it will reject the request with a `500 Internal /// Server Error` response. /// /// [request extensions]: https://docs.rs/http/latest/http/struct.Extensions.html #[derive(Debug, Clone)] pub struct Extension<T>(pub T); impl<Protocol, T> FromParts<Protocol> for Extension<T> where T: Clone + Send + Sync + 'static, { type Rejection = MissingExtension; fn from_parts(parts: &mut http::request::Parts) -> Result<Self, Self::Rejection> { parts.extensions.remove::<T>().map(Extension).ok_or(MissingExtension) } } /// The extension has not been added to the [`Request`](http::Request) or has been previously removed. #[derive(Debug, Error)] #[error("the `Extension` is not present in the `http::Request`")] pub struct MissingExtension; impl<Protocol> IntoResponse<Protocol> for MissingExtension { fn into_response(self) -> http::Response<BoxBody> { let mut response = http::Response::new(empty()); *response.status_mut() = StatusCode::INTERNAL_SERVER_ERROR; response } } }