vrc-log 0.15.4

VRChat Local Avatar ID Logger
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

// benches/cache_disk_bench.rs
use criterion::{Criterion, criterion_group, criterion_main};
use std::path::PathBuf;
use tempfile::NamedTempFile;
use tokio::runtime::Runtime;
use uuid::Uuid;
use vrc_log::cache::Cache;

fn bench_cache_store_and_check_disk(c: &mut Criterion) {
    let rt = Runtime::new().unwrap();

    // Create a tempfile on disk once, outside of the measured iterations
    let tmpfile: NamedTempFile = tempfile::NamedTempFile::new().unwrap();
    let path: PathBuf = tmpfile.path().to_path_buf();

    // Initialize the cache once
    let cache = rt.block_on(async { Cache::new_at_location(&path).await.unwrap() });

    // Pre-generate avatars
    let n = 10_000;
    #[allow(clippy::cast_possible_truncation)]
    let avatars: Vec<(String, u32)> = (0..n)
        .map(|i| (format!("avtr_{}", Uuid::new_v4()), i as u32))
        .collect();

    c.bench_function("store 10k avatars on disk", |b| {
        b.iter(|| {
            rt.block_on(async {
                cache
                    .store_avatar_ids_with_providers(
                        avatars.iter().map(|(id, p)| (id.as_str(), *p)),
                    )
                    .await
                    .unwrap();
            });
        });
    });

    c.bench_function("check 10k avatars on disk", |b| {
        b.iter(|| {
            rt.block_on(async {
                let result = cache
                    .check_all_ids(avatars.iter().map(|(id, _)| id.clone()))
                    .await
                    .unwrap();

                // Sanity check
                assert_eq!(result.len(), n);
            });
        });
    });
}

criterion_group!(cache_disk_benches, bench_cache_store_and_check_disk);
criterion_main!(cache_disk_benches);