turbo_persistence_tools/
main.rs

1#![feature(iter_intersperse)]
2
3use std::path::PathBuf;
4
5use anyhow::{Context, Result, bail};
6use turbo_persistence::{DbConfig, MetaFileEntryInfo, SerialScheduler, TurboPersistence};
7
8fn main() -> Result<()> {
9    // Get CLI argument
10    let path = PathBuf::from(
11        std::env::args()
12            .nth(1)
13            .context("Please provide a path to the TurboPersistence directory")?,
14    );
15    if !path.exists() {
16        bail!("The provided path does not exist: {}", path.display());
17    }
18
19    let db: TurboPersistence<SerialScheduler, 0> =
20        TurboPersistence::open_read_only_with_config(path, DbConfig::default())?;
21    let meta_info = db
22        .meta_info()
23        .context("Failed to retrieve meta information")?;
24    for meta_file in meta_info {
25        println!(
26            "META {:08}.meta: family = {}, sst_size = {} MiB",
27            meta_file.sequence_number,
28            meta_file.family,
29            meta_file.entries.iter().map(|e| e.sst_size).sum::<u64>() / 1024 / 1024,
30        );
31        for MetaFileEntryInfo {
32            sequence_number,
33            min_hash,
34            max_hash,
35            amqf_size,
36            amqf_entries,
37            sst_size,
38            flags,
39            block_count,
40        } in meta_file.entries
41        {
42            println!(
43                "  SST {sequence_number:08}.sst: {flags} {min_hash:016x} - {max_hash:016x} (p = \
44                 1/{})",
45                u64::MAX / (max_hash - min_hash + 1)
46            );
47            println!("    AMQF {amqf_entries} entries = {} KiB", amqf_size / 1024);
48            println!(
49                "    {} KiB = {block_count} blocks (avg {} bytes/block)",
50                sst_size / 1024,
51                sst_size / block_count as u64
52            );
53        }
54        if !meta_file.obsolete_sst_files.is_empty() {
55            println!(
56                "  OBSOLETE SSTs {}",
57                meta_file
58                    .obsolete_sst_files
59                    .iter()
60                    .map(|seq| format!("{seq:08}.sst"))
61                    .intersperse(", ".to_string())
62                    .collect::<String>()
63            );
64        }
65    }
66    Ok(())
67}
turbo_persistence_tools/main.rs

turbo_persistence_tools/
main.rs
