diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-17 12:18:32 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-17 12:18:32 +0000 |
commit | 4547b622d8d29df964fa2914213088b148c498fc (patch) | |
tree | 9fc6b25f3c3add6b745be9a2400a6e96140046e9 /vendor/litemap/benches/litemap.rs | |
parent | Releasing progress-linux version 1.66.0+dfsg1-1~progress7.99u1. (diff) | |
download | rustc-4547b622d8d29df964fa2914213088b148c498fc.tar.xz rustc-4547b622d8d29df964fa2914213088b148c498fc.zip |
Merging upstream version 1.67.1+dfsg1.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'vendor/litemap/benches/litemap.rs')
-rw-r--r-- | vendor/litemap/benches/litemap.rs | 133 |
1 files changed, 133 insertions, 0 deletions
diff --git a/vendor/litemap/benches/litemap.rs b/vendor/litemap/benches/litemap.rs new file mode 100644 index 000000000..bf005c4f8 --- /dev/null +++ b/vendor/litemap/benches/litemap.rs @@ -0,0 +1,133 @@ +// This file is part of ICU4X. For terms of use, please see the file +// called LICENSE at the top level of the ICU4X source tree +// (online at: https://github.com/unicode-org/icu4x/blob/main/LICENSE ). + +use std::fs; + +use criterion::{black_box, criterion_group, criterion_main, Criterion}; + +use litemap::LiteMap; + +const DATA: [(&str, &str); 16] = [ + ("ar", "Arabic"), + ("bn", "Bangla"), + ("ccp", "Chakma"), + ("chr", "Cherokee"), + ("el", "Greek"), + ("en", "English"), + ("eo", "Esperanto"), + ("es", "Spanish"), + ("fr", "French"), + ("iu", "Inuktitut"), + ("ja", "Japanese"), + ("ru", "Russian"), + ("sr", "Serbian"), + ("th", "Thai"), + ("tr", "Turkish"), + ("zh", "Chinese"), +]; + +const POSTCARD: [u8; 176] = [ + 16, 2, 97, 114, 6, 65, 114, 97, 98, 105, 99, 2, 98, 110, 6, 66, 97, 110, 103, 108, 97, 3, 99, + 99, 112, 6, 67, 104, 97, 107, 109, 97, 3, 99, 104, 114, 8, 67, 104, 101, 114, 111, 107, 101, + 101, 2, 101, 108, 5, 71, 114, 101, 101, 107, 2, 101, 110, 7, 69, 110, 103, 108, 105, 115, 104, + 2, 101, 111, 9, 69, 115, 112, 101, 114, 97, 110, 116, 111, 2, 101, 115, 7, 83, 112, 97, 110, + 105, 115, 104, 2, 102, 114, 6, 70, 114, 101, 110, 99, 104, 2, 105, 117, 9, 73, 110, 117, 107, + 116, 105, 116, 117, 116, 2, 106, 97, 8, 74, 97, 112, 97, 110, 101, 115, 101, 2, 114, 117, 7, + 82, 117, 115, 115, 105, 97, 110, 2, 115, 114, 7, 83, 101, 114, 98, 105, 97, 110, 2, 116, 104, + 4, 84, 104, 97, 105, 2, 116, 114, 7, 84, 117, 114, 107, 105, 115, 104, 2, 122, 104, 7, 67, 104, + 105, 110, 101, 115, 101, +]; + +/// Run this function to print new data to the console. Requires the optional `serde` feature. +#[allow(dead_code)] +fn generate() { + let map = build_litemap(false); + let buf = postcard::to_stdvec(&map).unwrap(); + println!("{:?}", buf); +} + +#[cfg(feature = "generate")] +fn generate_test_data() { + let map = build_litemap(true); + let bytes = postcard::to_stdvec(&map).unwrap(); + fs::write("large_litemap.postcard", &bytes).unwrap(); +} + +fn overview_bench(c: &mut Criterion) { + // Uncomment the following line to re-generate the binary data. + // generate(); + + bench_deserialize(c); + bench_deserialize_large(c); + bench_lookup(c); + bench_lookup_large(c); + + #[cfg(feature = "generate")] + generate_test_data(); +} + +fn build_litemap(large: bool) -> LiteMap<String, String> { + let mut map: LiteMap<String, String> = LiteMap::new(); + for (key, value) in DATA.iter() { + if large { + for n in 0..8192 { + map.insert(format!("{}{}", key, n), value.to_string()); + } + } else { + map.insert(key.to_string(), value.to_string()); + } + } + map +} + +fn bench_deserialize(c: &mut Criterion) { + c.bench_function("litemap/deserialize/small", |b| { + b.iter(|| { + let map: LiteMap<String, String> = postcard::from_bytes(black_box(&POSTCARD)).unwrap(); + assert_eq!(map.get("iu"), Some(&"Inuktitut".to_string())); + }) + }); +} + +fn bench_deserialize_large(c: &mut Criterion) { + let buf = read_large_litemap_postcard_bytes(); + c.bench_function("litemap/deseralize/large", |b| { + b.iter(|| { + let map: LiteMap<String, String> = postcard::from_bytes(black_box(&buf)).unwrap(); + assert_eq!(map.get("iu3333"), Some(&"Inuktitut".to_string())); + }); + }); +} + +fn bench_lookup(c: &mut Criterion) { + let map: LiteMap<String, String> = postcard::from_bytes(&POSTCARD).unwrap(); + c.bench_function("litemap/lookup/small", |b| { + b.iter(|| { + assert_eq!(map.get(black_box("iu")), Some(&"Inuktitut".to_string())); + assert_eq!(map.get(black_box("zz")), None); + }); + }); +} + +fn bench_lookup_large(c: &mut Criterion) { + let buf = read_large_litemap_postcard_bytes(); + let map: LiteMap<String, String> = postcard::from_bytes(&buf).unwrap(); + c.bench_function("litemap/lookup/large", |b| { + b.iter(|| { + assert_eq!(map.get(black_box("iu3333")), Some(&"Inuktitut".to_string())); + assert_eq!(map.get(black_box("zz")), None); + }); + }); +} + +fn read_large_litemap_postcard_bytes() -> Vec<u8> { + let path = concat!( + env!("CARGO_MANIFEST_DIR"), + "/benches/testdata/large_litemap.postcard" + ); + fs::read(path).unwrap() +} + +criterion_group!(benches, overview_bench); +criterion_main!(benches); |