chris/ruwuma
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

estimate and reserve capacity in outgoing headers map

Browse Source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2024-08-12 22:57:38 +00:00
parent 767b7c72f7
commit a6e340e24d
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
crates/ruma-macros/src/api/response/outgoing.rs
View File

@ -9,6 +9,10 @@ impl Response {
let bytes = quote! { #ruma_common::exports::bytes }; let bytes = quote! { #ruma_common::exports::bytes };
let http = quote! { #ruma_common::exports::http }; let http = quote! { #ruma_common::exports::http };
let reserve_headers = self.fields.iter().fold(0_usize, |acc, response_field| {
acc + (response_field.as_header_field().is_some() as usize)
});
let serialize_response_headers = self.fields.iter().filter_map(|response_field| { let serialize_response_headers = self.fields.iter().filter_map(|response_field| {
response_field.as_header_field().map(|(field, header_name)| { response_field.as_header_field().map(|(field, header_name)| {
let field_name = let field_name =
@ -68,11 +72,15 @@ impl Response {
fn try_into_http_response<T: ::std::default::Default + #bytes::BufMut>( fn try_into_http_response<T: ::std::default::Default + #bytes::BufMut>(
self, self,
) -> ::std::result::Result<#http::Response<T>, #ruma_common::api::error::IntoHttpError> { ) -> ::std::result::Result<#http::Response<T>, #ruma_common::api::error::IntoHttpError> {
static APPLICATION_JSON: #http::header::HeaderValue =
#http::header::HeaderValue::from_static("application/json");
let mut resp_builder = #http::Response::builder() let mut resp_builder = #http::Response::builder()
.status(#http::StatusCode::#status_ident) .status(#http::StatusCode::#status_ident);
.header(#http::header::CONTENT_TYPE, "application/json");
if let Some(mut headers) = resp_builder.headers_mut() { if let Some(mut headers) = resp_builder.headers_mut() {
headers.reserve(1 + #reserve_headers);
headers.insert(#http::header::CONTENT_TYPE, APPLICATION_JSON.clone());
#(#serialize_response_headers)* #(#serialize_response_headers)*
} }