lunatic_sqlite_api/wire_format/sqlite_value/
host_api.rs

1#[cfg(not(target_arch = "wasm32"))]
2use anyhow::Result;
3#[cfg(not(target_arch = "wasm32"))]
4use lunatic_common_api::IntoTrap;
5
6#[cfg(not(target_arch = "wasm32"))]
7use sqlite::Statement;
8
9use super::{SqliteRow, SqliteValue};
10
11#[cfg(not(target_arch = "wasm32"))]
12impl SqliteRow {
13    pub fn read_row(statement: &mut Statement) -> Result<SqliteRow> {
14        let mut row = SqliteRow::default();
15        for column_idx in 0..statement.column_count() {
16            row.0.push(SqliteValue::read_column(statement, column_idx)?);
17        }
18        Ok(row)
19    }
20}
21
22#[cfg(not(target_arch = "wasm32"))]
23impl<'stmt> SqliteValue {
24    pub fn read_column(statement: &'stmt Statement, col_idx: usize) -> Result<SqliteValue> {
25        match statement.column_type(col_idx).or_trap("read_column")? {
26            sqlite::Type::Binary => {
27                let bytes = statement
28                    .read::<Vec<u8>, usize>(col_idx)
29                    .or_trap("lunatic::sqlite::query_prepare::read_binary")?;
30
31                Ok(SqliteValue::Blob(bytes))
32            }
33            sqlite::Type::Float => Ok(SqliteValue::Double(
34                statement
35                    .read::<f64, usize>(col_idx)
36                    .or_trap("lunatic::sqlite::query_prepare::read_float")?,
37            )),
38            sqlite::Type::Integer => Ok(SqliteValue::Integer(
39                statement
40                    .read::<i64, usize>(col_idx)
41                    .or_trap("lunatic::sqlite::query_prepare::read_integer")?,
42            )),
43            sqlite::Type::String => {
44                let bytes = statement
45                    .read::<String, usize>(col_idx)
46                    .or_trap("lunatic::sqlite::query_prepare::read_string")?;
47
48                Ok(SqliteValue::Text(bytes))
49            }
50            sqlite::Type::Null => Ok(SqliteValue::Null),
51        }
52    }
53}