relay-knowledge 1.1.17

Graph-database-based knowledge graph project.
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

use rusqlite::Connection;

use crate::{
    domain::{CodeParseStatus, CodeParseStatusCounts},
    storage::StorageError,
};

use super::common::invalid_code_metadata;

pub(in crate::storage::sqlite) fn parse_status_counts(
    connection: &Connection,
) -> Result<CodeParseStatusCounts, StorageError> {
    let mut statement = connection.prepare(
        "
        SELECT parse_status, COUNT(*)
        FROM code_files
        GROUP BY parse_status
        ",
    )?;
    let rows = statement.query_map([], |row| {
        Ok((row.get::<_, String>(0)?, row.get::<_, usize>(1)?))
    })?;
    let mut counts = CodeParseStatusCounts::default();
    for row in rows {
        let (status, count) = row.map_err(StorageError::from)?;
        match parse_status(&status)? {
            CodeParseStatus::Parsed => counts.parsed = count,
            CodeParseStatus::Partial => counts.partial = count,
            CodeParseStatus::TextOnly => counts.text_only = count,
            CodeParseStatus::Failed => counts.failed = count,
        }
    }

    Ok(counts)
}

fn parse_status(value: &str) -> Result<CodeParseStatus, StorageError> {
    match value {
        "parsed" => Ok(CodeParseStatus::Parsed),
        "partial" => Ok(CodeParseStatus::Partial),
        "text_only" => Ok(CodeParseStatus::TextOnly),
        "failed" => Ok(CodeParseStatus::Failed),
        _ => Err(invalid_code_metadata(format!(
            "unknown code parse status '{value}'"
        ))),
    }
}

#[cfg(test)]
#[path = "status_tests.rs"]
mod status_tests;