78 lines
2 KiB
Rust
78 lines
2 KiB
Rust
use actix_web::{
|
|
dev::{Payload, Service, ServiceRequest, Transform},
|
|
http::Method,
|
|
web::BytesMut,
|
|
HttpMessage,
|
|
};
|
|
use std::{
|
|
future::{ready, Ready},
|
|
task::{Context, Poll},
|
|
};
|
|
use streem::IntoStreamer;
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub(crate) struct DebugPayload(pub bool);
|
|
|
|
#[doc(hidden)]
|
|
#[derive(Clone, Debug)]
|
|
pub(crate) struct DebugPayloadMiddleware<S>(bool, S);
|
|
|
|
impl<S> Transform<S, ServiceRequest> for DebugPayload
|
|
where
|
|
S: Service<ServiceRequest, Error = actix_web::Error>,
|
|
S::Future: 'static,
|
|
S::Error: 'static,
|
|
{
|
|
type Response = S::Response;
|
|
type Error = S::Error;
|
|
type InitError = ();
|
|
type Transform = DebugPayloadMiddleware<S>;
|
|
type Future = Ready<Result<Self::Transform, Self::InitError>>;
|
|
|
|
fn new_transform(&self, service: S) -> Self::Future {
|
|
ready(Ok(DebugPayloadMiddleware(self.0, service)))
|
|
}
|
|
}
|
|
|
|
impl<S> Service<ServiceRequest> for DebugPayloadMiddleware<S>
|
|
where
|
|
S: Service<ServiceRequest, Error = actix_web::Error>,
|
|
S::Future: 'static,
|
|
S::Error: 'static,
|
|
{
|
|
type Response = S::Response;
|
|
type Error = S::Error;
|
|
type Future = S::Future;
|
|
|
|
fn poll_ready(&self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
|
|
self.1.poll_ready(cx)
|
|
}
|
|
|
|
fn call(&self, mut req: ServiceRequest) -> Self::Future {
|
|
if self.0 && req.method() == Method::POST {
|
|
let mut pl = req.take_payload().into_streamer();
|
|
|
|
req.set_payload(Payload::Stream {
|
|
payload: Box::pin(streem::try_from_fn(|yielder| async move {
|
|
let mut buf = BytesMut::new();
|
|
|
|
while let Some(bytes) = pl.try_next().await? {
|
|
buf.extend(bytes);
|
|
}
|
|
|
|
let bytes = buf.freeze();
|
|
tracing::info!("{}", String::from_utf8_lossy(&bytes));
|
|
|
|
yielder.yield_ok(bytes).await;
|
|
|
|
Ok(())
|
|
})),
|
|
});
|
|
|
|
self.1.call(req)
|
|
} else {
|
|
self.1.call(req)
|
|
}
|
|
}
|
|
}
|