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
73
74
75

//! Test chunked BFS performance on various graph topologies

use sqlitegraph::backend::native::v3::V3Backend;
use sqlitegraph::backend::native::v3::algorithm::parallel_bfs;
use sqlitegraph::backend::{EdgeSpec, GraphBackend, NodeSpec};
use std::time::Instant;

fn create_star_graph(backend: &V3Backend, size: i64) -> Vec<i64> {
    let mut node_ids = Vec::new();

    // Create center node
    let center = backend
        .insert_node(NodeSpec {
            kind: "test".to_string(),
            name: "center".to_string(),
            file_path: None,
            data: serde_json::json!(null),
        })
        .unwrap();
    node_ids.push(center);

    // Create surrounding nodes
    for i in 1..size {
        let node = backend
            .insert_node(NodeSpec {
                kind: "test".to_string(),
                name: format!("node_{}", i),
                file_path: None,
                data: serde_json::json!(null),
            })
            .unwrap();
        node_ids.push(node);

        // Connect to center
        backend
            .insert_edge(EdgeSpec {
                from: center,
                to: node,
                edge_type: "edge".to_string(),
                data: serde_json::json!(null),
            })
            .unwrap();
    }

    node_ids
}

fn main() {
    let temp_dir = tempfile::tempdir().unwrap();
    let db_path = temp_dir.path().join("test.db");
    let backend = V3Backend::create(&db_path).unwrap();

    println!("=== Chunked BFS Performance Test ===\n");

    // Test different graph sizes
    for size in [100, 500, 1000, 5000, 10000] {
        let node_ids = create_star_graph(&backend, size);
        let start = node_ids[0];

        // Warm up
        let _ = parallel_bfs(&backend, start, None);

        // Measure
        let start_time = Instant::now();
        let result = parallel_bfs(&backend, start, None).unwrap();
        let elapsed = start_time.elapsed();

        println!(
            "Size: {:>5} | Time: {:>8.2?} | Visited: {}",
            size, elapsed, result.total_visited
        );
    }

    println!("\n✓ Chunked BFS test complete");
}