graphqlite 0.6.0

SQLite extension for graph queries using Cypher
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

//! Centrality algorithm implementations.

use super::{
    parsing::{extract_algo_array, extract_float, extract_int, extract_node_id, extract_user_id},
    BetweennessCentralityResult, ClosenessCentralityResult, DegreeCentralityResult,
    EigenvectorCentralityResult, PageRankResult,
};
use crate::graph::Graph;
use crate::Result;

impl Graph {
    /// Run PageRank algorithm.
    ///
    /// # Arguments
    ///
    /// * `damping` - Damping factor, typically 0.85
    /// * `iterations` - Number of iterations, typically 20
    pub fn pagerank(&self, damping: f64, iterations: i32) -> Result<Vec<PageRankResult>> {
        let query = format!("RETURN pageRank({}, {})", damping, iterations);
        let result = self.connection().cypher(&query)?;
        let rows = extract_algo_array(result.iter().collect::<Vec<_>>().as_slice());

        let mut ranks = Vec::new();
        for row in rows.iter() {
            if let Some(node_id) = extract_node_id(row) {
                ranks.push(PageRankResult {
                    node_id,
                    user_id: extract_user_id(row),
                    score: extract_float(row, "score"),
                });
            }
        }
        Ok(ranks)
    }

    /// Calculate degree centrality for all nodes.
    pub fn degree_centrality(&self) -> Result<Vec<DegreeCentralityResult>> {
        let result = self.connection().cypher("RETURN degreeCentrality()")?;
        let rows = extract_algo_array(result.iter().collect::<Vec<_>>().as_slice());

        let mut degrees = Vec::new();
        for row in rows.iter() {
            if let Some(node_id) = extract_node_id(row) {
                degrees.push(DegreeCentralityResult {
                    node_id,
                    user_id: extract_user_id(row),
                    in_degree: extract_int(row, "in_degree"),
                    out_degree: extract_int(row, "out_degree"),
                    degree: extract_int(row, "degree"),
                });
            }
        }
        Ok(degrees)
    }

    /// Calculate betweenness centrality for all nodes.
    pub fn betweenness_centrality(&self) -> Result<Vec<BetweennessCentralityResult>> {
        let result = self.connection().cypher("RETURN betweennessCentrality()")?;
        let rows = extract_algo_array(result.iter().collect::<Vec<_>>().as_slice());

        let mut scores = Vec::new();
        for row in rows.iter() {
            if let Some(node_id) = extract_node_id(row) {
                scores.push(BetweennessCentralityResult {
                    node_id,
                    user_id: extract_user_id(row),
                    score: extract_float(row, "score"),
                });
            }
        }
        Ok(scores)
    }

    /// Calculate closeness centrality for all nodes.
    pub fn closeness_centrality(&self) -> Result<Vec<ClosenessCentralityResult>> {
        let result = self.connection().cypher("RETURN closenessCentrality()")?;
        let rows = extract_algo_array(result.iter().collect::<Vec<_>>().as_slice());

        let mut scores = Vec::new();
        for row in rows.iter() {
            if let Some(node_id) = extract_node_id(row) {
                scores.push(ClosenessCentralityResult {
                    node_id,
                    user_id: extract_user_id(row),
                    score: extract_float(row, "score"),
                });
            }
        }
        Ok(scores)
    }

    /// Calculate eigenvector centrality for all nodes.
    ///
    /// # Arguments
    ///
    /// * `iterations` - Maximum iterations for power iteration (default 100)
    pub fn eigenvector_centrality(
        &self,
        iterations: i32,
    ) -> Result<Vec<EigenvectorCentralityResult>> {
        let query = format!("RETURN eigenvectorCentrality({})", iterations);
        let result = self.connection().cypher(&query)?;
        let rows = extract_algo_array(result.iter().collect::<Vec<_>>().as_slice());

        let mut scores = Vec::new();
        for row in rows.iter() {
            if let Some(node_id) = extract_node_id(row) {
                scores.push(EigenvectorCentralityResult {
                    node_id,
                    user_id: extract_user_id(row),
                    score: extract_float(row, "score"),
                });
            }
        }
        Ok(scores)
    }
}