rusqlite_regex 0.6.2

A regex extension for rusqlite
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

#![doc = include_str!("../README.md")]
#![cfg_attr(docsrs, feature(doc_auto_cfg))]
#![warn(unsafe_code, missing_docs)]

pub mod cache;
pub mod extension;
pub mod error;

/// Enable sqlite3_regex_init() as an auto extension.
pub fn enable_auto_extension() -> rusqlite::Result<()> {
    #[allow(unsafe_code)]
    let res = unsafe { rusqlite::ffi::sqlite3_auto_extension(Some(extension::sqlite3_regex_init)) };
    error::check_err(res)
}

/// Disable sqlite3_regex_init() as an auto extension.
pub fn disable_auto_extension() -> rusqlite::Result<()> {
    #[allow(unsafe_code)]
    let res = unsafe { rusqlite::ffi::sqlite3_cancel_auto_extension(Some(extension::sqlite3_regex_init)) };
    error::check_err(res)
}

#[cfg(test)]
mod tester {
    pub fn initialize() -> Result<rusqlite::Connection, rusqlite::Error> {
        static ONCE: std::sync::OnceLock<rusqlite::Result<()>> = std::sync::OnceLock::new();
        // FIXME: use get_or_try_init once https://github.com/rust-lang/rust/issues/109737 stable
        ONCE.get_or_init(crate::enable_auto_extension).as_ref().unwrap();
        rusqlite::Connection::open_in_memory()
    }

    pub fn query<T: rusqlite::types::FromSql>(conn: &rusqlite::Connection, sql: &str) -> rusqlite::Result<T> {
        conn.query_row(sql, [], |row| row.get::<_, T>(0))
    }
}