sqlite_graphrag/commands/
optimize.rs

1use crate::errors::AppError;
2use crate::i18n::erros;
3use crate::output;
4use crate::paths::AppPaths;
5use crate::storage::connection::open_rw;
6use serde::Serialize;
7
8#[derive(clap::Args)]
9pub struct OptimizeArgs {
10    #[arg(long, hide = true, help = "No-op; JSON is always emitted on stdout")]
11    pub json: bool,
12    #[arg(long, env = "SQLITE_GRAPHRAG_DB_PATH")]
13    pub db: Option<String>,
14}
15
16#[derive(Serialize)]
17struct OptimizeResponse {
18    db_path: String,
19    status: String,
20    /// Tempo total de execução em milissegundos desde início do handler até serialização.
21    elapsed_ms: u64,
22}
23
24pub fn run(args: OptimizeArgs) -> Result<(), AppError> {
25    let inicio = std::time::Instant::now();
26    let paths = AppPaths::resolve(args.db.as_deref())?;
27
28    if !paths.db.exists() {
29        return Err(AppError::NotFound(erros::banco_nao_encontrado(
30            &paths.db.display().to_string(),
31        )));
32    }
33
34    let conn = open_rw(&paths.db)?;
35    conn.execute_batch("PRAGMA optimize;")?;
36
37    output::emit_json(&OptimizeResponse {
38        db_path: paths.db.display().to_string(),
39        status: "ok".to_string(),
40        elapsed_ms: inicio.elapsed().as_millis() as u64,
41    })?;
42
43    Ok(())
44}
45
46#[cfg(test)]
47mod testes {
48    use super::*;
49    use serial_test::serial;
50    use tempfile::TempDir;
51
52    #[test]
53    fn optimize_response_serializa_campos_obrigatorios() {
54        let resp = OptimizeResponse {
55            db_path: "/tmp/graphrag.sqlite".to_string(),
56            status: "ok".to_string(),
57            elapsed_ms: 5,
58        };
59        let json = serde_json::to_value(&resp).unwrap();
60        assert_eq!(json["status"], "ok");
61        assert_eq!(json["db_path"], "/tmp/graphrag.sqlite");
62        assert_eq!(json["elapsed_ms"], 5);
63    }
64
65    #[test]
66    #[serial]
67    fn optimize_retorna_not_found_quando_db_ausente() {
68        let dir = TempDir::new().unwrap();
69        let db_path = dir.path().join("inexistente.sqlite");
70        std::env::set_var("SQLITE_GRAPHRAG_DB_PATH", db_path.to_str().unwrap());
71        std::env::set_var("LOG_LEVEL", "error");
72
73        let args = OptimizeArgs {
74            json: false,
75            db: Some(db_path.to_string_lossy().to_string()),
76        };
77        let resultado = run(args);
78        assert!(resultado.is_err(), "deve falhar quando db não existe");
79        match resultado.unwrap_err() {
80            AppError::NotFound(_) => {}
81            outro => panic!("esperava NotFound, obteve: {outro:?}"),
82        }
83        std::env::remove_var("SQLITE_GRAPHRAG_DB_PATH");
84        std::env::remove_var("LOG_LEVEL");
85    }
86
87    #[test]
88    fn optimize_response_status_ok_fixo() {
89        let resp = OptimizeResponse {
90            db_path: "/qualquer/caminho".to_string(),
91            status: "ok".to_string(),
92            elapsed_ms: 0,
93        };
94        let json = serde_json::to_value(&resp).unwrap();
95        assert_eq!(json["status"], "ok", "status deve ser sempre 'ok'");
96    }
97}
sqlite_graphrag/commands/optimize.rs

sqlite_graphrag/commands/
optimize.rs
