engram-memory 0.6.0

Core binary and shared infrastructure for Engram memory system
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

//! Add the `query_id` column to `feedback_tracking` on databases created before
//! routing instrumentation. Idempotent and gated by `schema_meta`.

use engram_storage::Database;

use crate::error::CoreError;

pub const FEEDBACK_QUERY_ID_KEY: &str = "feedback_query_id";
pub const FEEDBACK_QUERY_ID_TARGET: &str = "v1";

pub fn run(database: &Database) -> Result<bool, CoreError> {
    if super::read_meta(database, FEEDBACK_QUERY_ID_KEY)?.as_deref()
        == Some(FEEDBACK_QUERY_ID_TARGET)
    {
        return Ok(false);
    }
    if !column_exists(database, "feedback_tracking", "query_id")? {
        let connection = database.connection();
        let transaction = connection
            .unchecked_transaction()
            .map_err(|error| CoreError::MigrationFailed(format!("tx begin: {error}")))?;
        transaction
            .execute_batch("ALTER TABLE feedback_tracking ADD COLUMN query_id TEXT")
            .map_err(|error| CoreError::MigrationFailed(format!("alter: {error}")))?;
        transaction
            .commit()
            .map_err(|error| CoreError::MigrationFailed(format!("tx commit: {error}")))?;
    }
    super::write_meta(database, FEEDBACK_QUERY_ID_KEY, FEEDBACK_QUERY_ID_TARGET)?;
    Ok(true)
}

fn column_exists(database: &Database, table: &str, column: &str) -> Result<bool, CoreError> {
    let mut statement = database
        .connection()
        .prepare(&format!("PRAGMA table_info({table})"))
        .map_err(|error| CoreError::MigrationFailed(format!("prepare table_info: {error}")))?;
    let names = statement
        .query_map([], |row| row.get::<_, String>(1))
        .map_err(|error| CoreError::MigrationFailed(format!("query table_info: {error}")))?;
    for name in names {
        let name =
            name.map_err(|error| CoreError::MigrationFailed(format!("row table_info: {error}")))?;
        if name == column {
            return Ok(true);
        }
    }
    Ok(false)
}