perf(http): encoder headers with a faster fmt::Write
This commit is contained in:
		| @@ -1,5 +1,5 @@ | |||||||
| use std::borrow::Cow; | use std::borrow::Cow; | ||||||
| use std::io::Write; | use std::fmt::{self, Write}; | ||||||
|  |  | ||||||
| use httparse; | use httparse; | ||||||
|  |  | ||||||
| @@ -95,8 +95,12 @@ impl Http1Message for ServerMessage { | |||||||
|         let init_cap = 30 + head.headers.len() * AVERAGE_HEADER_SIZE; |         let init_cap = 30 + head.headers.len() * AVERAGE_HEADER_SIZE; | ||||||
|         dst.reserve(init_cap); |         dst.reserve(init_cap); | ||||||
|         debug!("writing {:#?}", head.headers); |         debug!("writing {:#?}", head.headers); | ||||||
|         let _ = write!(dst, "{} {}\r\n{}\r\n", head.version, head.subject, head.headers); |         if head.version == ::HttpVersion::Http11 && head.subject == ::StatusCode::Ok { | ||||||
|  |             extend(dst, b"HTTP/1.1 200 OK\r\n"); | ||||||
|  |             let _ = write!(FastWrite(dst), "{}\r\n", head.headers); | ||||||
|  |         } else { | ||||||
|  |             let _ = write!(FastWrite(dst), "{} {}\r\n{}\r\n", head.version, head.subject, head.headers); | ||||||
|  |         } | ||||||
|         body |         body | ||||||
|     } |     } | ||||||
| } | } | ||||||
| @@ -195,12 +199,37 @@ impl Http1Message for ClientMessage { | |||||||
|         let init_cap = 30 + head.headers.len() * AVERAGE_HEADER_SIZE; |         let init_cap = 30 + head.headers.len() * AVERAGE_HEADER_SIZE; | ||||||
|         dst.reserve(init_cap); |         dst.reserve(init_cap); | ||||||
|         debug!("writing {:#?}", head.headers); |         debug!("writing {:#?}", head.headers); | ||||||
|         let _ = write!(dst, "{} {}\r\n{}\r\n", head.subject, head.version, head.headers); |         let _ = write!(FastWrite(dst), "{} {}\r\n{}\r\n", head.subject, head.version, head.headers); | ||||||
|  |  | ||||||
|         body |         body | ||||||
|     } |     } | ||||||
| } | } | ||||||
|  |  | ||||||
|  | struct FastWrite<'a>(&'a mut Vec<u8>); | ||||||
|  |  | ||||||
|  | impl<'a> fmt::Write for FastWrite<'a> { | ||||||
|  |     fn write_str(&mut self, s: &str) -> fmt::Result { | ||||||
|  |         extend(self.0, s.as_bytes()); | ||||||
|  |         Ok(()) | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     fn write_fmt(&mut self, args: fmt::Arguments) -> fmt::Result { | ||||||
|  |         fmt::write(self, args) | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
|  | fn extend(dst: &mut Vec<u8>, data: &[u8]) { | ||||||
|  |     use std::ptr; | ||||||
|  |     dst.reserve(data.len()); | ||||||
|  |     let prev = dst.len(); | ||||||
|  |     unsafe { | ||||||
|  |         ptr::copy_nonoverlapping(data.as_ptr(), | ||||||
|  |                                  dst.as_mut_ptr().offset(prev as isize), | ||||||
|  |                                  data.len()); | ||||||
|  |         dst.set_len(prev + data.len()); | ||||||
|  |     } | ||||||
|  | } | ||||||
|  |  | ||||||
| #[cfg(test)] | #[cfg(test)] | ||||||
| mod tests { | mod tests { | ||||||
|     use http; |     use http; | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user