fragarach 0.1.1

Modular OSINT framework for blockchain forensics and investigations
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

/// DuckDB storage operations implementation
/// 
/// Provides functionality for:
/// - Data persistence
/// - Record updates
/// - Batch operations
use duckdb::{Connection, Result, ToSql};
use serde_json::Value;

pub fn save_records(conn: &Connection, data: &[Value], table_name: &str) -> Result<()> {
    for record in data {
        let obj = record.as_object().unwrap();
        let columns = obj.keys().map(|s| s.as_str()).collect::<Vec<_>>().join(", ");
        let placeholders = (0..obj.len())
            .map(|i| format!("${}", i + 1))
            .collect::<Vec<_>>()
            .join(", ");
        
        let sql = format!(
            "INSERT OR REPLACE INTO {} ({}) VALUES ({})",
            table_name,
            columns,
            placeholders
        );
        
        let values: Vec<String> = obj.values()
            .map(|v| v.to_string())
            .collect();
        
        // Convert values to a slice of references that implement ToSql
        let param_refs: Vec<&(dyn ToSql)> = values.iter()
            .map(|s| s as &(dyn ToSql))
            .collect();
        
        conn.execute(&sql, param_refs.as_slice())?;
    }
    
    Ok(())
}