scalar/
scalar.rs

1//! cargo build --example scalar
2//! sqlite3 :memory: '.read examples/test.sql'
3
4use sqlite_loadable::prelude::*;
5use sqlite_loadable::{api, define_scalar_function, Result};
6
7// yo()
8fn yo(context: *mut sqlite3_context, _values: &[*mut sqlite3_value]) -> Result<()> {
9    api::result_text(context, "yo")?;
10    Ok(())
11}
12
13// surround_rs(name)
14fn surround(context: *mut sqlite3_context, values: &[*mut sqlite3_value]) -> Result<()> {
15    let value = api::value_text(values.get(0).expect("1st argument as name"))?;
16    api::result_text(context, format!("x{}x", value))?;
17    Ok(())
18}
19
20// add_rs(a, b)
21fn add(context: *mut sqlite3_context, values: &[*mut sqlite3_value]) -> Result<()> {
22    let a = api::value_int(values.get(0).expect("1st argument"));
23    let b = api::value_int(values.get(1).expect("2nd argument"));
24    api::result_int(context, a + b);
25    Ok(())
26}
27
28// connect(seperator, string1, ...)
29fn connect(context: *mut sqlite3_context, values: &[*mut sqlite3_value]) -> Result<()> {
30    let seperator = api::value_text(values.get(0).expect("1st argument"))?;
31    let strings: Vec<&str> = values
32        .get(1..)
33        .expect("more than 1 argument to be given")
34        .iter()
35        .filter_map(|v| api::value_text(v).ok())
36        .collect();
37    api::result_text(context, strings.join(seperator))?;
38    Ok(())
39}
40
41#[sqlite_entrypoint]
42pub fn sqlite3_scalarrs_init(db: *mut sqlite3) -> Result<()> {
43    let flags = FunctionFlags::UTF8 | FunctionFlags::DETERMINISTIC;
44    define_scalar_function(db, "surround_rs", 1, surround, flags)?;
45    define_scalar_function(db, "connect", -1, connect, flags)?;
46    define_scalar_function(db, "yo_rs", 0, yo, flags)?;
47    define_scalar_function(db, "add_rs", 2, add, flags)?;
48    Ok(())
49}
scalar/scalar.rs

scalar/
scalar.rs
