From 3c80cc18748d18963e794e06baca8283f6ac4f43 Mon Sep 17 00:00:00 2001 From: Ali MJ Al-Nasrawy Date: Wed, 19 May 2021 23:23:30 +0300 Subject: Simplify custom headers Signed-off-by: Ali MJ Al-Nasrawy --- src/main.rs | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) (limited to 'src/main.rs') diff --git a/src/main.rs b/src/main.rs index e5cc596..4ea1154 100644 --- a/src/main.rs +++ b/src/main.rs @@ -275,17 +275,10 @@ fn create_tcp_listener(addr: SocketAddr) -> io::Result { } fn configure_header(conf: &MiniserveConfig) -> middleware::DefaultHeaders { - let headers = conf.clone().header; - - let mut default_headers = middleware::DefaultHeaders::new(); - for header in headers { - for (header_name, header_value) in header.into_iter() { - if let Some(header_name) = header_name { - default_headers = default_headers.header(&header_name, header_value); - } - } - } - default_headers + conf.header.iter().flatten().fold( + middleware::DefaultHeaders::new(), + |headers, (header_name, header_value)| headers.header(header_name, header_value), + ) } /// Configures the Actix application -- cgit v1.2.3