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
76
77
78
79
80
81
82
83
84

use rusqlite::params;

use crate::{
    SqliteGraphError,
    graph::{GraphEntity, SqliteGraph},
};

pub fn add_label(graph: &SqliteGraph, entity_id: i64, label: &str) -> Result<(), SqliteGraphError> {
    graph
        .connection()
        .execute(
            "INSERT OR IGNORE INTO graph_labels(entity_id, label) VALUES(?1, ?2)",
            params![entity_id, label],
        )
        .map_err(|e| SqliteGraphError::query(e.to_string()))?;
    Ok(())
}

pub fn get_entities_by_label(
    graph: &SqliteGraph,
    label: &str,
) -> Result<Vec<GraphEntity>, SqliteGraphError> {
    let conn = graph.connection();
    let mut stmt = conn
        .prepare_cached("SELECT entity_id FROM graph_labels WHERE label=?1 ORDER BY entity_id")
        .map_err(|e| SqliteGraphError::query(e.to_string()))?;
    let rows = stmt
        .query_map(params![label], |row| row.get(0))
        .map_err(|e| SqliteGraphError::query(e.to_string()))?;
    let mut ids = Vec::new();
    for row in rows {
        ids.push(row.map_err(|e| SqliteGraphError::query(e.to_string()))?);
    }
    fetch_entities(graph, ids)
}

pub fn add_property(
    graph: &SqliteGraph,
    entity_id: i64,
    key: &str,
    value: &str,
) -> Result<(), SqliteGraphError> {
    graph
        .connection()
        .execute(
            "INSERT INTO graph_properties(entity_id, key, value) VALUES(?1, ?2, ?3)",
            params![entity_id, key, value],
        )
        .map_err(|e| SqliteGraphError::query(e.to_string()))?;
    Ok(())
}

pub fn get_entities_by_property(
    graph: &SqliteGraph,
    key: &str,
    value: &str,
) -> Result<Vec<GraphEntity>, SqliteGraphError> {
    let conn = graph.connection();
    let mut stmt = conn
        .prepare_cached(
            "SELECT entity_id FROM graph_properties \
             WHERE key=?1 AND value=?2 ORDER BY entity_id",
        )
        .map_err(|e| SqliteGraphError::query(e.to_string()))?;
    let rows = stmt
        .query_map(params![key, value], |row| row.get(0))
        .map_err(|e| SqliteGraphError::query(e.to_string()))?;
    let mut ids = Vec::new();
    for row in rows {
        ids.push(row.map_err(|e| SqliteGraphError::query(e.to_string()))?);
    }
    fetch_entities(graph, ids)
}

fn fetch_entities(
    graph: &SqliteGraph,
    ids: Vec<i64>,
) -> Result<Vec<GraphEntity>, SqliteGraphError> {
    let mut entities = Vec::new();
    for id in ids {
        entities.push(graph.get_entity(id)?);
    }
    Ok(entities)
}