systemprompt_database/services/postgres/
conversion.rs

1use sqlx::{Column, Row};
2use std::collections::HashMap;
3
4use crate::models::{DbValue, QueryResult, ToDbValue};
5
6pub fn rows_to_result(rows: Vec<sqlx::postgres::PgRow>, start: std::time::Instant) -> QueryResult {
7    let mut columns = Vec::new();
8    let mut result_rows = Vec::new();
9
10    if let Some(first_row) = rows.first() {
11        columns = first_row
12            .columns()
13            .iter()
14            .map(|c| c.name().to_string())
15            .collect();
16    }
17
18    for row in rows {
19        result_rows.push(row_to_json(&row));
20    }
21
22    let row_count = result_rows.len();
23    let execution_time_ms = u64::try_from(start.elapsed().as_millis()).unwrap_or(u64::MAX);
24
25    QueryResult {
26        columns,
27        rows: result_rows,
28        row_count,
29        execution_time_ms,
30    }
31}
32
33// JSON: Postgres row-to-JSON conversion — column shape unknown at compile time
34pub fn row_to_json(row: &sqlx::postgres::PgRow) -> HashMap<String, serde_json::Value> {
35    row.columns()
36        .iter()
37        .map(|col| (col.name().to_string(), column_to_json(row, col.ordinal())))
38        .collect()
39}
40
41fn column_to_json(row: &sqlx::postgres::PgRow, ordinal: usize) -> serde_json::Value {
42    if let Ok(val) = row.try_get::<Option<chrono::DateTime<chrono::Utc>>, _>(ordinal) {
43        return val.map_or(serde_json::Value::Null, |v| {
44            serde_json::Value::String(v.to_rfc3339())
45        });
46    }
47    if let Ok(val) = row.try_get::<Option<uuid::Uuid>, _>(ordinal) {
48        return val.map_or(serde_json::Value::Null, |v| {
49            serde_json::Value::String(v.to_string())
50        });
51    }
52    if let Ok(val) = row.try_get::<Option<String>, _>(ordinal) {
53        return val.map_or(serde_json::Value::Null, serde_json::Value::String);
54    }
55    if let Ok(val) = row.try_get::<Option<i64>, _>(ordinal) {
56        return val.map_or(serde_json::Value::Null, |v| {
57            serde_json::Value::Number(v.into())
58        });
59    }
60    if let Ok(val) = row.try_get::<Option<i32>, _>(ordinal) {
61        return val.map_or(serde_json::Value::Null, |v| {
62            serde_json::Value::Number(i64::from(v).into())
63        });
64    }
65    if let Ok(val) = row.try_get::<Option<f64>, _>(ordinal) {
66        return val.map_or(serde_json::Value::Null, |v| serde_json::json!(v));
67    }
68    if let Ok(val) = row.try_get::<Option<rust_decimal::Decimal>, _>(ordinal) {
69        return val.map_or(serde_json::Value::Null, |v| {
70            v.to_string().parse::<f64>().map_or_else(
71                |_| serde_json::Value::String(v.to_string()),
72                |f| serde_json::json!(f),
73            )
74        });
75    }
76    if let Ok(val) = row.try_get::<Option<bool>, _>(ordinal) {
77        return val.map_or(serde_json::Value::Null, serde_json::Value::Bool);
78    }
79    if let Ok(val) = row.try_get::<Option<Vec<String>>, _>(ordinal) {
80        return val.map_or(serde_json::Value::Null, |v| {
81            serde_json::Value::Array(v.into_iter().map(serde_json::Value::String).collect())
82        });
83    }
84    if let Ok(val) = row.try_get::<Option<serde_json::Value>, _>(ordinal) {
85        return val.unwrap_or(serde_json::Value::Null);
86    }
87    if let Ok(val) = row.try_get::<Option<Vec<u8>>, _>(ordinal) {
88        return val.map_or(serde_json::Value::Null, |bytes| {
89            use base64::Engine;
90            use base64::engine::general_purpose::STANDARD;
91            serde_json::Value::String(STANDARD.encode(&bytes))
92        });
93    }
94    serde_json::Value::Null
95}
96
97pub fn bind_params<'q>(
98    mut query: sqlx::query::Query<'q, sqlx::Postgres, sqlx::postgres::PgArguments>,
99    params: &[&dyn ToDbValue],
100) -> sqlx::query::Query<'q, sqlx::Postgres, sqlx::postgres::PgArguments> {
101    for param in params {
102        let value = param.to_db_value();
103        query = match value {
104            DbValue::String(s) => query.bind(s),
105            DbValue::Int(i) => query.bind(i),
106            DbValue::Float(f) => query.bind(f),
107            DbValue::Bool(b) => query.bind(b),
108            DbValue::Bytes(b) => query.bind(b),
109            DbValue::Timestamp(dt) => query.bind(dt),
110            DbValue::StringArray(arr) => query.bind(arr),
111            DbValue::NullString => query.bind(None::<String>),
112            DbValue::NullInt => query.bind(None::<i64>),
113            DbValue::NullFloat => query.bind(None::<f64>),
114            DbValue::NullBool => query.bind(None::<bool>),
115            DbValue::NullBytes => query.bind(None::<Vec<u8>>),
116            DbValue::NullTimestamp => query.bind(None::<chrono::DateTime<chrono::Utc>>),
117            DbValue::NullStringArray => query.bind(None::<Vec<String>>),
118        };
119    }
120    query
121}
systemprompt_database/services/postgres/conversion.rs

systemprompt_database/services/postgres/
conversion.rs
