sqlitegraph 2.2.2

Embedded graph database with full ACID transactions, HNSW vector search, dual backend support, and comprehensive graph algorithms library
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

//! Simple test for NodeStore batch mode

use parking_lot::RwLock;
use std::sync::Arc;
use tempfile::TempDir;

use sqlitegraph::backend::native::types::NodeFlags;
use sqlitegraph::backend::native::v3::node::NodeRecordV3;
use sqlitegraph::backend::native::v3::{
    NodeStore, PageAllocator, PersistentHeaderV3, btree::BTreeManager,
};

fn main() {
    let temp_dir = TempDir::new().unwrap();
    let db_path = temp_dir.path().join("test.db");

    // Create header
    let header = PersistentHeaderV3::new_v3();

    // Create store
    let mut store = NodeStore::new(&header, db_path.clone());

    // Initialize with btree and shared allocator
    let allocator = Arc::new(RwLock::new(PageAllocator::new(&header)));
    let btree = BTreeManager::new(Arc::clone(&allocator), None, db_path.clone());
    store.initialize(btree, Arc::clone(&allocator), None);

    // Test 1: Insert without batch mode
    println!("Test 1: Immediate mode");
    let node1 = NodeRecordV3::new_inline(0, NodeFlags::empty(), 0, 0, vec![1, 2, 3], 0, 0, 0, 0);
    match store.insert_node(node1) {
        Ok(id) => println!("  Inserted node {}", id),
        Err(e) => println!("  Error: {}", e),
    }

    // Test 2: Insert with batch mode
    println!("\nTest 2: Batch mode");
    store.begin_batch();
    println!("  Batch mode enabled");

    let node2 = NodeRecordV3::new_inline(0, NodeFlags::empty(), 0, 0, vec![4, 5, 6], 0, 0, 0, 0);
    match store.insert_node(node2) {
        Ok(id) => {
            println!(
                "  Staged node {} (dirty_pages: {})",
                id,
                store.dirty_page_count()
            );

            // Try second insert
            let node3 =
                NodeRecordV3::new_inline(0, NodeFlags::empty(), 0, 0, vec![7, 8, 9], 0, 0, 0, 0);
            match store.insert_node(node3) {
                Ok(id2) => println!(
                    "  Staged node {} (dirty_pages: {})",
                    id2,
                    store.dirty_page_count()
                ),
                Err(e) => println!("  Second insert error: {}", e),
            }
        }
        Err(e) => println!("  First insert error: {}", e),
    }

    // Commit
    match store.commit_batch() {
        Ok(count) => println!("  Committed {} pages", count),
        Err(e) => println!("  Commit error: {}", e),
    }

    println!("\nDone!");
}