perf(http): encoder headers with a faster fmt::Write
This commit is contained in:
		| @@ -1,5 +1,5 @@ | ||||
| use std::borrow::Cow; | ||||
| use std::io::Write; | ||||
| use std::fmt::{self, Write}; | ||||
|  | ||||
| use httparse; | ||||
|  | ||||
| @@ -95,8 +95,12 @@ impl Http1Message for ServerMessage { | ||||
|         let init_cap = 30 + head.headers.len() * AVERAGE_HEADER_SIZE; | ||||
|         dst.reserve(init_cap); | ||||
|         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 | ||||
|     } | ||||
| } | ||||
| @@ -195,12 +199,37 @@ impl Http1Message for ClientMessage { | ||||
|         let init_cap = 30 + head.headers.len() * AVERAGE_HEADER_SIZE; | ||||
|         dst.reserve(init_cap); | ||||
|         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 | ||||
|     } | ||||
| } | ||||
|  | ||||
| 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)] | ||||
| mod tests { | ||||
|     use http; | ||||
|   | ||||
		Reference in New Issue
	
	Block a user