ws_sdk/database/
sql.rs

1use super::super::host::abi::*;
2use super::sql_types::*;
3use anyhow::{bail, Result};
4
5/// Execute a prepared statement with parameters.
6///
7/// This func is often used to write records into the database.
8/// # Examples
9///
10/// ```no_run
11/// use ws-sdk::database::sql::execute
12/// execute("INSERT INTO table (ID, COUNT) VALUES (?, ?);", &[&0, &32])?;
13/// ```
14pub fn execute(prepared: &str, params: &[&dyn SQLType]) -> Result<()> {
15    let query = DBQuery {
16        statement: prepared.to_string(),
17        params: params.iter().map(|x| x.get_param()).collect(),
18    };
19    let encoded = &serde_json::to_string(&query)?;
20    match unsafe { ws_set_sql_db(encoded.as_ptr(), encoded.len() as _) } {
21        0 => Ok(()),
22        _ => bail!("fail to exec sql"),
23    }
24}
25
26/// Query a prepared statement with parameters.
27///
28/// This func is often used to read records from the database.
29/// The query result is returned in `json` format.
30/// # Examples
31///
32/// ```no_run
33/// use ws-sdk::database::sql::query
34/// let ret = query("SELECT * FROM table;", &[])?;
35/// ```
36pub fn query(prepared: &str, params: &[&dyn SQLType]) -> Result<Vec<u8>> {
37    let query = DBQuery {
38        statement: prepared.to_string(),
39        params: params.iter().map(|x| x.get_param()).collect(),
40    };
41    let encoded = &serde_json::to_string(&query)?;
42    let data_ptr = &mut (0 as i32) as *const _ as *const *mut u8;
43    let data_size = &mut (0 as i32) as *const i32;
44
45    match unsafe { ws_get_sql_db(encoded.as_ptr(), encoded.len() as _, data_ptr, data_size) } {
46        0 => Ok(unsafe { Vec::from_raw_parts(*data_ptr, *data_size as _, *data_size as _) }),
47        _ => bail!("fail to query sql"),
48    }
49}