zed/crates/project_panel/benches/sorting.rs

use criterion::{Criterion, criterion_group, criterion_main};
use project::{Entry, EntryKind, GitEntry, ProjectEntryId};
use project_panel::par_sort_worktree_entries;
use settings::{ProjectPanelSortMode, ProjectPanelSortOrder};
use std::sync::Arc;
use util::rel_path::RelPath;

fn load_linux_repo_snapshot() -> Vec<GitEntry> {
    let file = std::fs::read_to_string(concat!(
        env!("CARGO_MANIFEST_DIR"),
        "/benches/linux_repo_snapshot.txt"
    ))
    .expect("Failed to read file");
    file.lines()
        .filter_map(|line| {
            let kind = match line.chars().next() {
                Some('f') => EntryKind::File,
                Some('d') => EntryKind::Dir,
                _ => return None,
            };

            let entry = Entry {
                kind,
                path: Arc::from(RelPath::from_unix_str(&(line.trim_end()[2..])).unwrap()),
                id: ProjectEntryId::default(),
                size: 0,
                inode: 0,
                mtime: None,
                canonical_path: None,
                is_ignored: false,
                is_always_included: false,
                is_external: false,
                is_private: false,
                is_hidden: false,
                char_bag: Default::default(),
                is_fifo: false,
            };
            Some(GitEntry {
                entry,
                git_summary: Default::default(),
            })
        })
        .collect()
}
fn criterion_benchmark(c: &mut Criterion) {
    let snapshot = load_linux_repo_snapshot();

    let modes = [
        ("DirectoriesFirst", ProjectPanelSortMode::DirectoriesFirst),
        ("Mixed", ProjectPanelSortMode::Mixed),
        ("FilesFirst", ProjectPanelSortMode::FilesFirst),
    ];
    let orders = [
        ("Default", ProjectPanelSortOrder::Default),
        ("Upper", ProjectPanelSortOrder::Upper),
        ("Lower", ProjectPanelSortOrder::Lower),
        ("Unicode", ProjectPanelSortOrder::Unicode),
    ];

    for (mode_name, mode) in &modes {
        for (order_name, order) in &orders {
            c.bench_function(
                &format!("Sort linux worktree snapshot ({mode_name}, {order_name})"),
                |b| {
                    b.iter_batched(
                        || snapshot.clone(),
                        |mut snapshot| par_sort_worktree_entries(&mut snapshot, *mode, *order),
                        criterion::BatchSize::LargeInput,
                    );
                },
            );
        }
    }
}

criterion_group!(benches, criterion_benchmark);
criterion_main!(benches);