ISAR/rusqlite
3
0
Fork 0
mirror of https://github.com/isar/rusqlite.git synced 2024-11-27 03:51:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add remove_function to clear a user-defined function.

Browse Source
This commit is contained in:
John Gallagher 2015-12-11 15:47:52 -05:00
parent 3baf7b10f8
commit ecef092303
1 changed files with 33 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/functions.rs
View File

@ -1,6 +1,7 @@
//! Create or redefine SQL functions //! Create or redefine SQL functions
use std::ffi::CStr; use std::ffi::CStr;
use std::mem; use std::mem;
use std::ptr;
use std::slice; use std::slice;
use std::str; use std::str;
use libc::{c_int, c_double, c_char, c_void}; use libc::{c_int, c_double, c_char, c_void};
@ -281,6 +282,10 @@ impl SqliteConnection {
{ {
self.db.borrow_mut().create_scalar_function(fn_name, n_arg, deterministic, x_func) self.db.borrow_mut().create_scalar_function(fn_name, n_arg, deterministic, x_func)
} }
pub fn remove_function(&self, fn_name: &str, n_arg: c_int) -> SqliteResult<()> {
self.db.borrow_mut().remove_function(fn_name, n_arg)
}
} }
impl InnerSqliteConnection { impl InnerSqliteConnection {
@ -337,6 +342,22 @@ impl InnerSqliteConnection {
}; };
self.decode_result(r) self.decode_result(r)
} }
fn remove_function(&mut self, fn_name: &str, n_arg: c_int) -> SqliteResult<()> {
let c_name = try!(str_to_cstring(fn_name));
let r = unsafe {
ffi::sqlite3_create_function_v2(self.db(),
c_name.as_ptr(),
n_arg,
ffi::SQLITE_UTF8,
ptr::null_mut(),
None,
None,
None,
None)
};
self.decode_result(r)
}
} }
#[cfg(test)] #[cfg(test)]
@ -366,6 +387,18 @@ mod test {
assert_eq!(3f64, result.unwrap()); assert_eq!(3f64, result.unwrap());
} }
#[test]
fn test_remove_function() {
let db = SqliteConnection::open_in_memory().unwrap();
db.create_scalar_function("half", 1, true, half).unwrap();
let result = db.query_row("SELECT half(6)", &[], |r| r.get::<f64>(0));
assert_eq!(3f64, result.unwrap());
db.remove_function("half", 1).unwrap();
let result = db.query_row("SELECT half(6)", &[], |r| r.get::<f64>(0));
assert!(result.is_err());
}
// This implementation of a regexp scalar function uses SQLite's auxilliary data // This implementation of a regexp scalar function uses SQLite's auxilliary data
// (https://www.sqlite.org/c3ref/get_auxdata.html) to avoid recompiling the regular // (https://www.sqlite.org/c3ref/get_auxdata.html) to avoid recompiling the regular
// expression multiple times within one query. // expression multiple times within one query.