diff --git a/benches/bench_log.rs b/benches/bench_log.rs index f1b5fe9f..34d2f9a3 100644 --- a/benches/bench_log.rs +++ b/benches/bench_log.rs @@ -819,7 +819,7 @@ fn bench_copy(b: &mut Bencher) { }); } -fn manual_no_escape(wr: &mut W, log: &Log) { +fn manual_serialize_no_escape(wr: &mut W, log: &Log) { wr.write_str("{\"timestamp\":").unwrap(); (write!(wr, "{}", log.timestamp)).unwrap(); wr.write_str(",\"zone_id\":").unwrap(); @@ -876,7 +876,7 @@ fn manual_no_escape(wr: &mut W, log: &Log) { wr.write_str("}").unwrap(); } -fn manual_escape(wr: &mut W, log: &Log) { +fn manual_serialize_escape(wr: &mut W, log: &Log) { wr.write_str("{").unwrap(); escape_str(wr, "timestamp").unwrap(); wr.write_str(":").unwrap(); @@ -982,157 +982,157 @@ fn manual_escape(wr: &mut W, log: &Log) { } #[test] -fn test_manual_vec_no_escape() { +fn test_manual_serialize_vec_no_escape() { let log = Log::new(); let mut wr = Vec::with_capacity(1024); - manual_no_escape(&mut wr, &log); + manual_serialize_no_escape(&mut wr, &log); let json = String::from_utf8(wr).unwrap(); assert_eq!(JSON_STR, json.as_slice()); } #[bench] -fn bench_manual_vec_no_escape(b: &mut Bencher) { +fn bench_manual_serialize_vec_no_escape(b: &mut Bencher) { let log = Log::new(); let mut wr = Vec::with_capacity(1024); - manual_no_escape(&mut wr, &log); + manual_serialize_no_escape(&mut wr, &log); b.bytes = wr.len() as u64; b.iter(|| { wr.clear(); - manual_no_escape(&mut wr, &log); + manual_serialize_no_escape(&mut wr, &log); }); } #[test] -fn test_manual_vec_escape() { +fn test_manual_serialize_vec_escape() { let log = Log::new(); let mut wr = Vec::with_capacity(1024); - manual_escape(&mut wr, &log); + manual_serialize_escape(&mut wr, &log); let json = String::from_utf8(wr).unwrap(); assert_eq!(JSON_STR, json.as_slice()); } #[bench] -fn bench_manual_vec_escape(b: &mut Bencher) { +fn bench_manual_serialize_vec_escape(b: &mut Bencher) { let log = Log::new(); let mut wr = Vec::with_capacity(1024); - manual_escape(&mut wr, &log); + manual_serialize_escape(&mut wr, &log); b.bytes = wr.len() as u64; b.iter(|| { wr.clear(); - manual_escape(&mut wr, &log); + manual_serialize_escape(&mut wr, &log); }); } #[test] -fn test_manual_my_mem_writer0_no_escape() { +fn test_manual_serialize_my_mem_writer0_no_escape() { let log = Log::new(); let mut wr = MyMemWriter0::with_capacity(1000); - manual_no_escape(&mut wr, &log); + manual_serialize_no_escape(&mut wr, &log); let json = String::from_utf8(wr.buf).unwrap(); assert_eq!(JSON_STR, json.as_slice()); } #[bench] -fn bench_manual_my_mem_writer0_no_escape(b: &mut Bencher) { +fn bench_manual_serialize_my_mem_writer0_no_escape(b: &mut Bencher) { let log = Log::new(); let mut wr = MyMemWriter0::with_capacity(1024); - manual_no_escape(&mut wr, &log); + manual_serialize_no_escape(&mut wr, &log); b.bytes = wr.buf.len() as u64; b.iter(|| { wr.buf.clear(); - manual_no_escape(&mut wr, &log); + manual_serialize_no_escape(&mut wr, &log); }); } #[test] -fn test_manual_my_mem_writer0_escape() { +fn test_manual_serialize_my_mem_writer0_escape() { let log = Log::new(); let mut wr = MyMemWriter0::with_capacity(1024); - manual_escape(&mut wr, &log); + manual_serialize_escape(&mut wr, &log); let json = String::from_utf8(wr.buf).unwrap(); assert_eq!(JSON_STR, json.as_slice()); } #[bench] -fn bench_manual_my_mem_writer0_escape(b: &mut Bencher) { +fn bench_manual_serialize_my_mem_writer0_escape(b: &mut Bencher) { let log = Log::new(); let mut wr = MyMemWriter0::with_capacity(1024); - manual_escape(&mut wr, &log); + manual_serialize_escape(&mut wr, &log); b.bytes = wr.buf.len() as u64; b.iter(|| { wr.buf.clear(); - manual_escape(&mut wr, &log); + manual_serialize_escape(&mut wr, &log); }); } #[test] -fn test_manual_my_mem_writer1_no_escape() { +fn test_manual_serialize_my_mem_writer1_no_escape() { let log = Log::new(); let mut wr = MyMemWriter1::with_capacity(1024); - manual_no_escape(&mut wr, &log); + manual_serialize_no_escape(&mut wr, &log); let json = String::from_utf8(wr.buf).unwrap(); assert_eq!(JSON_STR, json.as_slice()); } #[bench] -fn bench_manual_my_mem_writer1_no_escape(b: &mut Bencher) { +fn bench_manual_serialize_my_mem_writer1_no_escape(b: &mut Bencher) { let log = Log::new(); let mut wr = MyMemWriter1::with_capacity(1024); - manual_no_escape(&mut wr, &log); + manual_serialize_no_escape(&mut wr, &log); b.bytes = wr.buf.len() as u64; b.iter(|| { wr.buf.clear(); - manual_no_escape(&mut wr, &log); + manual_serialize_no_escape(&mut wr, &log); }); } #[test] -fn test_manual_my_mem_writer1_escape() { +fn test_manual_serialize_my_mem_writer1_escape() { let log = Log::new(); let mut wr = MyMemWriter1::with_capacity(1024); - manual_escape(&mut wr, &log); + manual_serialize_escape(&mut wr, &log); let json = String::from_utf8(wr.buf).unwrap(); assert_eq!(JSON_STR, json.as_slice()); } #[bench] -fn bench_manual_my_mem_writer1_escape(b: &mut Bencher) { +fn bench_manual_serialize_my_mem_writer1_escape(b: &mut Bencher) { let log = Log::new(); let mut wr = MyMemWriter1::with_capacity(1024); - manual_escape(&mut wr, &log); + manual_serialize_escape(&mut wr, &log); b.bytes = wr.buf.len() as u64; b.iter(|| { wr.buf.clear(); - manual_escape(&mut wr, &log); + manual_serialize_escape(&mut wr, &log); }); } @@ -1261,6 +1261,145 @@ fn test_deserializer() { assert_eq!(log, Log::new()); } +#[inline] +fn manual_read_ignore(rdr: &mut R, buf: &mut [u8], key: &[u8]) { + let buf = buf.slice_mut(0, key.len()); + rdr.read(buf).unwrap(); + assert_eq!(buf, key); +} + +#[inline] +fn manual_read_field(rdr: &mut R, buf: &mut [u8], key: &[u8]) { + let b = rdr.read_byte().unwrap(); + assert_eq!(b, b'"'); + + manual_read_ignore(rdr, buf, key); + + let b = rdr.read_byte().unwrap(); + assert_eq!(b, b'"'); + + let b = rdr.read_byte().unwrap(); + assert_eq!(b, b':'); +} + +#[inline] +fn manual_read_int(rdr: &mut R, buf: &mut [u8], key: &[u8]) -> i64 { + manual_read_field(rdr, buf, key); + + let mut res = 0; + + loop { + let byte = rdr.read_byte().unwrap(); + match byte { + b'0' ... b'9' => { + res *= 10; + res += (byte as i64) - (b'0' as i64); + } + _ => { break; } + } + } + + res +} + +#[inline] +fn manual_read_string(rdr: &mut R, buf: &mut [u8], key: &[u8]) -> String { + manual_read_field(rdr, buf, key); + manual_read_ignore(rdr, buf, b"\""); + + let mut idx = 0; + + loop { + let byte = rdr.read_byte().unwrap(); + match byte { + b'"' => { break; } + byte => { buf[idx] = byte; } + }; + + idx += 1; + } + + let b = rdr.read_byte().unwrap(); + assert!(b == b',' || b == b']' || b == b'}'); + + String::from_utf8(buf.slice_to(idx).to_vec()).unwrap() +} + +#[inline] +fn manual_deserialize(rdr: &mut R) -> Log { + let mut buf = [0, .. 128]; + + manual_read_ignore(rdr, &mut buf, b"{"); + let timestamp = manual_read_int(rdr, &mut buf, b"timestamp"); + let zone_id = manual_read_int(rdr, &mut buf, b"zone_id"); + let zone_plan = manual_read_int(rdr, &mut buf, b"zone_plan"); + + manual_read_field(rdr, &mut buf, b"http"); + manual_read_ignore(rdr, &mut buf, b"{"); + + let protocol = manual_read_int(rdr, &mut buf, b"protocol"); + let status = manual_read_int(rdr, &mut buf, b"status"); + let host_status = manual_read_int(rdr, &mut buf, b"host_status"); + let up_status = manual_read_int(rdr, &mut buf, b"up_status"); + let method = manual_read_int(rdr, &mut buf, b"method"); + let content_type = manual_read_string(rdr, &mut buf, b"content_type"); + let user_agent = manual_read_string(rdr, &mut buf, b"user_agent"); + let referer = manual_read_string(rdr, &mut buf, b"referer"); + let request_uri = manual_read_string(rdr, &mut buf, b"request_uri"); + + let http = Http { + protocol: FromPrimitive::from_i64(protocol).unwrap(), + status: FromPrimitive::from_i64(status).unwrap(), + host_status: FromPrimitive::from_i64(host_status).unwrap(), + up_status: FromPrimitive::from_i64(up_status).unwrap(), + method: FromPrimitive::from_i64(method).unwrap(), + content_type: content_type, + user_agent: user_agent, + referer: referer, + request_uri: request_uri, + }; + + manual_read_ignore(rdr, &mut buf, b","); + manual_read_field(rdr, &mut buf, b"origin"); + manual_read_ignore(rdr, &mut buf, b"{"); + + let ip = manual_read_string(rdr, &mut buf, b"ip"); + let port = manual_read_int(rdr, &mut buf, b"port"); + let hostname = manual_read_string(rdr, &mut buf, b"hostname"); + let protocol = manual_read_int(rdr, &mut buf, b"protocol"); + + let origin = Origin { + ip: ip, + port: FromPrimitive::from_i64(port).unwrap(), + hostname: hostname, + protocol: FromPrimitive::from_i64(protocol).unwrap(), + }; + + manual_read_ignore(rdr, &mut buf, b","); + let country = manual_read_int(rdr, &mut buf, b"country"); + let cache_status = manual_read_int(rdr, &mut buf, b"cache_status"); + let server_ip = manual_read_string(rdr, &mut buf, b"server_ip"); + let server_name = manual_read_string(rdr, &mut buf, b"server_name"); + let remote_ip = manual_read_string(rdr, &mut buf, b"remote_ip"); + let bytes_dlv = manual_read_int(rdr, &mut buf, b"bytes_dlv"); + let ray_id = manual_read_string(rdr, &mut buf, b"ray_id"); + + Log { + timestamp: timestamp, + zone_id: FromPrimitive::from_i64(zone_id).unwrap(), + zone_plan: FromPrimitive::from_i64(zone_plan).unwrap(), + http: http, + origin: origin, + country: FromPrimitive::from_i64(country).unwrap(), + cache_status: FromPrimitive::from_i64(cache_status).unwrap(), + server_ip: server_ip, + server_name: server_name, + remote_ip: remote_ip, + bytes_dlv: FromPrimitive::from_i64(bytes_dlv).unwrap(), + ray_id: ray_id, + } +} + #[bench] fn bench_deserializer(b: &mut Bencher) { b.bytes = JSON_STR.len() as u64; @@ -1270,6 +1409,24 @@ fn bench_deserializer(b: &mut Bencher) { }); } +#[test] +fn test_manual_deserialize() { + let mut rdr = JSON_STR.as_bytes(); + let log = manual_deserialize(&mut rdr); + + assert_eq!(log, Log::new()); +} + +#[bench] +fn bench_manual_deserialize(b: &mut Bencher) { + b.bytes = JSON_STR.len() as u64; + + b.iter(|| { + let mut rdr = JSON_STR.as_bytes(); + let _ = manual_deserialize(&mut rdr); + }); +} + #[bench] fn bench_deserializers(b: &mut Bencher) { let s = r#"{"timestamp":25469139677502,"zone_id":123456,"zone_plan":1,"http":{"protocol":2,"status":200,"host_status":503,"up_status":520,"method":1,"content_type":"text/html","user_agent":"Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/33.0.1750.146 Safari/537.36","referer":"https://www.cloudflare.com/","request_uri":"/cdn-cgi/trace"},"origin":{"ip":"1.2.3.4","port":8000,"hostname":"www.example.com","protocol":2},"country":238,"cache_status":3,"server_ip":"192.168.1.1","server_name":"metal.cloudflare.com","remote_ip":"10.1.2.3","bytes_dlv":123456,"ray_id":"10c73629cce30078-LAX"}"#;