-
Notifications
You must be signed in to change notification settings - Fork 189
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Collect extractors into
request module (#2008)
- Loading branch information
Showing
8 changed files
with
144 additions
and
123 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
103 changes: 103 additions & 0 deletions
103
rust-runtime/aws-smithy-http-server/src/request/extension.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,103 @@ | ||
| /* | ||
| * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. | ||
| * SPDX-License-Identifier: Apache-2.0 | ||
| */ | ||
|
|
||
| // This code was copied and then modified from Tokio's Axum. | ||
|
|
||
| /* Copyright (c) 2021 Tower Contributors | ||
| * | ||
| * Permission is hereby granted, free of charge, to any | ||
| * person obtaining a copy of this software and associated | ||
| * documentation files (the "Software"), to deal in the | ||
| * Software without restriction, including without | ||
| * limitation the rights to use, copy, modify, merge, | ||
| * publish, distribute, sublicense, and/or sell copies of | ||
| * the Software, and to permit persons to whom the Software | ||
| * is furnished to do so, subject to the following | ||
| * conditions: | ||
| * | ||
| * The above copyright notice and this permission notice | ||
| * shall be included in all copies or substantial portions | ||
| * of the Software. | ||
| * | ||
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF | ||
| * ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED | ||
| * TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A | ||
| * PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT | ||
| * SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY | ||
| * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION | ||
| * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR | ||
| * IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER | ||
| * DEALINGS IN THE SOFTWARE. | ||
| */ | ||
|
|
||
| //! Extension types. | ||
| //! | ||
| //! Extension types are types that are stored in and extracted from _both_ requests and | ||
| //! responses. | ||
| //! | ||
| //! There is only one _generic_ extension type _for requests_, [`Extension`]. | ||
| //! | ||
| //! On the other hand, the server SDK uses multiple concrete extension types for responses in order | ||
| //! to store a variety of information, like the operation that was executed, the operation error | ||
| //! that got returned, or the runtime error that happened, among others. The information stored in | ||
| //! these types may be useful to [`tower::Layer`]s that post-process the response: for instance, a | ||
| //! particular metrics layer implementation might want to emit metrics about the number of times an | ||
| //! an operation got executed. | ||
| //! | ||
| //! [extensions]: https://docs.rs/http/latest/http/struct.Extensions.html | ||
|
|
||
| use std::ops::Deref; | ||
|
|
||
| use http::StatusCode; | ||
| use thiserror::Error; | ||
|
|
||
| use crate::{ | ||
| body::{empty, BoxBody}, | ||
| request::FromParts, | ||
| response::IntoResponse, | ||
| }; | ||
|
|
||
| /// 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<T> Deref for Extension<T> { | ||
| type Target = T; | ||
|
|
||
| fn deref(&self) -> &Self::Target { | ||
| &self.0 | ||
| } | ||
| } | ||
|
|
||
| /// 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 | ||
| } | ||
| } | ||
|
|
||
| impl<Protocol, T> FromParts<Protocol> for Extension<T> | ||
| where | ||
| T: 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) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.