vicinity 0.7.2

Approximate nearest-neighbor search
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

#![allow(clippy::unwrap_used, clippy::expect_used)]
#![cfg(feature = "persistence")]

use std::sync::Arc;
use vicinity::persistence::directory::{Directory, MemoryDirectory};
use vicinity::persistence::wal::{WalEntry, WalWriter};

/// Verify that WAL entries survive a write+flush+drop cycle and can be replayed.
#[tokio::test]
async fn wal_roundtrip_replays_written_entries() -> anyhow::Result<()> {
    let dir: Arc<dyn Directory> = Arc::new(MemoryDirectory::new());
    let mut writer = WalWriter::new(dir.clone());

    // Write an entry and flush (simulates data that hit disk before crash).
    writer.append(WalEntry::StartMerge {
        transaction_id: 100,
        segment_ids: vec![1, 2],
    })?;
    writer.flush()?;
    drop(writer);

    // Replay WAL to recover state
    let reader = vicinity::persistence::wal::WalReader::new(dir.clone());
    let entries = reader.replay()?;

    assert_eq!(entries.len(), 1, "expected 1 WAL entry after replay");
    match &entries[0].payload {
        WalEntry::StartMerge {
            transaction_id,
            segment_ids,
        } => {
            assert_eq!(*transaction_id, 100);
            assert_eq!(segment_ids, &[1, 2]);
        }
        other => panic!("Expected StartMerge entry, got {:?}", other),
    }

    Ok(())
}