Jędrzej Stuczyński
6559fadf7f
* refactor FormattedResponse to allow attaching additional headers * helper method for including expiration headers * add expires header for /active nodes responses * added additional 'with_expires_header_delta' builder to FormattedResponse to allow setting expiration header with time delta
50 lines
1.3 KiB
Rust
50 lines
1.3 KiB
Rust
// Copyright 2025 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use crate::response::{error_response, ResponseWrapper};
|
|
use axum::http::header::IntoHeaderName;
|
|
use axum::http::{header, HeaderValue};
|
|
use axum::response::{IntoResponse, Response};
|
|
use bytes::{BufMut, BytesMut};
|
|
use serde::Serialize;
|
|
|
|
#[derive(Debug, Clone, Default)]
|
|
#[must_use]
|
|
pub struct Bincode<T>(pub(crate) ResponseWrapper<T>);
|
|
|
|
impl<T> From<T> for Bincode<T> {
|
|
fn from(response: T) -> Self {
|
|
Bincode(ResponseWrapper::new(response).with_header(
|
|
header::CONTENT_TYPE,
|
|
HeaderValue::from_static("application/bincode"),
|
|
))
|
|
}
|
|
}
|
|
|
|
impl<T> Bincode<T> {
|
|
pub(crate) fn with_header(
|
|
mut self,
|
|
name: impl IntoHeaderName,
|
|
value: impl Into<HeaderValue>,
|
|
) -> Self {
|
|
self.0.headers.insert(name, value.into());
|
|
self
|
|
}
|
|
}
|
|
|
|
impl<T> IntoResponse for Bincode<T>
|
|
where
|
|
T: Serialize,
|
|
{
|
|
// replicates axum's Json
|
|
fn into_response(self) -> Response {
|
|
use bincode::Options;
|
|
let mut buf = BytesMut::with_capacity(128).writer();
|
|
|
|
match crate::make_bincode_serializer().serialize_into(&mut buf, &self.0.data) {
|
|
Ok(()) => (self.0.headers, buf.into_inner().freeze()).into_response(),
|
|
Err(err) => error_response(err),
|
|
}
|
|
}
|
|
}
|