block-db 0.2.0

Local, multi-threaded, durable byte DB.
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

// Authors: Robert Lopez

use crate::{
    ConfirmDestructiveAction,
    tests::{test_db::TestDB, util::random_bytes},
    util::fs::DirectoryIter,
};

#[tokio::test]
async fn delete_data_files() {
    let chunk_size = 100;
    let max_file_size = 1_000;

    let test_db = TestDB::open("delete_data_files", Some(chunk_size), Some(max_file_size))
        .await
        .unwrap();

    macro_rules! assert_data_file_count_is {
        ($n:expr) => {
            match DirectoryIter::new(&test_db.block_db.path.join("data_files")).await {
                Ok(mut data_files_dir_iter) => {
                    let mut count = 0;

                    while let Some(_) = data_files_dir_iter.next().await.unwrap() {
                        count += 1;
                    }

                    assert_eq!(count, $n, "Invalid DataFile count");
                }
                _ => {
                    assert_eq!(0, $n, "Invalid DataFileCount");
                }
            }
        };
    }

    let data_file_id = test_db
        .block_db
        .write(random_bytes(max_file_size))
        .await
        .unwrap()
        .data_file_id;

    test_db
        .block_db
        .write(random_bytes(max_file_size))
        .await
        .unwrap()
        .data_file_id;

    test_db
        .block_db
        .delete_data_file(data_file_id, ConfirmDestructiveAction::IKnowWhatImDoing)
        .await
        .unwrap();

    assert_data_file_count_is!(1);

    for _ in 0..10 {
        test_db
            .block_db
            .write(random_bytes(max_file_size))
            .await
            .unwrap()
            .data_file_id;
    }

    test_db
        .block_db
        .delete_data_files(ConfirmDestructiveAction::IKnowWhatImDoing)
        .await
        .unwrap();

    assert_data_file_count_is!(0);

    test_db.cleanup().await.unwrap();
}