add a bench_manual_deserialize benchmark

This commit is contained in:
Erick Tryzelaar 2014-12-13 22:58:05 -08:00
parent c75f5caf8f
commit 90a4ef4867

View File

@ -819,7 +819,7 @@ fn bench_copy(b: &mut Bencher) {
});
}
fn manual_no_escape<W: Writer>(wr: &mut W, log: &Log) {
fn manual_serialize_no_escape<W: Writer>(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<W: Writer>(wr: &mut W, log: &Log) {
wr.write_str("}").unwrap();
}
fn manual_escape<W: Writer>(wr: &mut W, log: &Log) {
fn manual_serialize_escape<W: Writer>(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<W: Writer>(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<R: Reader>(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<R: Reader>(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<R: Reader>(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<R: Reader>(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<R: Reader>(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"}"#;