use crate::{metadata::MetadataMap, Extensions};
#[derive(Debug)]
pub struct Response<T> {
metadata: MetadataMap,
message: T,
extensions: Extensions,
}
impl<T> Response<T> {
pub fn new(message: T) -> Self {
Response {
metadata: MetadataMap::new(),
message,
extensions: Extensions::new(),
}
}
pub fn get_ref(&self) -> &T {
&self.message
}
pub fn get_mut(&mut self) -> &mut T {
&mut self.message
}
pub fn metadata(&self) -> &MetadataMap {
&self.metadata
}
pub fn metadata_mut(&mut self) -> &mut MetadataMap {
&mut self.metadata
}
pub fn into_inner(self) -> T {
self.message
}
pub(crate) fn into_parts(self) -> (MetadataMap, T, Extensions) {
(self.metadata, self.message, self.extensions)
}
pub(crate) fn from_parts(metadata: MetadataMap, message: T, extensions: Extensions) -> Self {
Self {
metadata,
message,
extensions,
}
}
pub(crate) fn from_http(res: http::Response<T>) -> Self {
let (head, message) = res.into_parts();
Response {
metadata: MetadataMap::from_headers(head.headers),
message,
extensions: Extensions::from_http(head.extensions),
}
}
pub(crate) fn into_http(self) -> http::Response<T> {
let mut res = http::Response::new(self.message);
*res.version_mut() = http::Version::HTTP_2;
*res.headers_mut() = self.metadata.into_sanitized_headers();
*res.extensions_mut() = self.extensions.into_http();
res
}
#[doc(hidden)]
pub fn map<F, U>(self, f: F) -> Response<U>
where
F: FnOnce(T) -> U,
{
let message = f(self.message);
Response {
metadata: self.metadata,
message,
extensions: self.extensions,
}
}
pub fn extensions(&self) -> &Extensions {
&self.extensions
}
pub fn extensions_mut(&mut self) -> &mut Extensions {
&mut self.extensions
}
#[cfg(feature = "gzip")]
#[cfg_attr(docsrs, doc(cfg(feature = "gzip")))]
pub fn disable_compression(&mut self) {
self.extensions_mut()
.insert(crate::codec::compression::SingleMessageCompressionOverride::Disable);
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::metadata::MetadataValue;
#[test]
fn reserved_headers_are_excluded() {
let mut r = Response::new(1);
for header in &MetadataMap::GRPC_RESERVED_HEADERS {
r.metadata_mut()
.insert(*header, MetadataValue::from_static("invalid"));
}
let http_response = r.into_http();
assert!(http_response.headers().is_empty());
}
}