You've already forked authentication-service
mirror of
https://github.com/matrix-org/matrix-authentication-service.git
synced 2025-11-23 11:02:35 +03:00
Make HTTP request layers convert to Bytes
Add layer to convert a request's Bytes to a Body.
This commit is contained in:
committed by
Quentin Gliech
parent
669a1867b7
commit
348044afdc
66
crates/http/src/layers/bytes_to_body_request.rs
Normal file
66
crates/http/src/layers/bytes_to_body_request.rs
Normal file
@@ -0,0 +1,66 @@
|
||||
// Copyright 2022 The Matrix.org Foundation C.I.C.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
use bytes::Bytes;
|
||||
use http::Request;
|
||||
use http_body::Full;
|
||||
use tower::{Layer, Service};
|
||||
|
||||
#[derive(Clone)]
|
||||
pub struct BytesToBodyRequest<S> {
|
||||
inner: S,
|
||||
}
|
||||
|
||||
impl<S> BytesToBodyRequest<S> {
|
||||
pub const fn new(inner: S) -> Self {
|
||||
Self { inner }
|
||||
}
|
||||
}
|
||||
|
||||
impl<S> Service<Request<Bytes>> for BytesToBodyRequest<S>
|
||||
where
|
||||
S: Service<Request<Full<Bytes>>>,
|
||||
S::Future: Send + 'static,
|
||||
{
|
||||
type Error = S::Error;
|
||||
type Response = S::Response;
|
||||
type Future = S::Future;
|
||||
|
||||
fn poll_ready(
|
||||
&mut self,
|
||||
cx: &mut std::task::Context<'_>,
|
||||
) -> std::task::Poll<Result<(), Self::Error>> {
|
||||
self.inner.poll_ready(cx)
|
||||
}
|
||||
|
||||
fn call(&mut self, request: Request<Bytes>) -> Self::Future {
|
||||
let (parts, body) = request.into_parts();
|
||||
let body = Full::new(body);
|
||||
|
||||
let request = Request::from_parts(parts, body);
|
||||
|
||||
self.inner.call(request)
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Default, Clone, Copy)]
|
||||
pub struct BytesToBodyRequestLayer;
|
||||
|
||||
impl<S> Layer<S> for BytesToBodyRequestLayer {
|
||||
type Service = BytesToBodyRequest<S>;
|
||||
|
||||
fn layer(&self, inner: S) -> Self::Service {
|
||||
BytesToBodyRequest::new(inner)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user