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

Add SqliteStatement.named_query.

Browse Source
This commit is contained in:
Gwenael Treguier 2015-08-05 22:07:49 +02:00
parent e81757e86d
commit 7ff9d4056b
1 changed files with 35 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
src/named_params.rs
View File

@ -4,7 +4,7 @@ use libc::{c_int};
use super::ffi; use super::ffi;
use {SqliteResult, SqliteError, SqliteConnection, SqliteStatement}; use {SqliteResult, SqliteError, SqliteConnection, SqliteStatement, SqliteRows};
use types::{ToSql}; use types::{ToSql};
impl SqliteConnection { impl SqliteConnection {
@ -44,16 +44,33 @@ impl<'conn> SqliteStatement<'conn> {
/// `sqlite3_changes`). /// `sqlite3_changes`).
pub fn named_execute(&mut self, params: &[(&str, &ToSql)]) -> SqliteResult<c_int> { pub fn named_execute(&mut self, params: &[(&str, &ToSql)]) -> SqliteResult<c_int> {
unsafe { unsafe {
for &(name, value) in params { try!(self.bind_named_parameters(params));
let i = try!(self.parameter_index(name).ok_or(SqliteError{
code: ffi::SQLITE_MISUSE,
message: format!("Invalid parameter name: {}", name)
}));
try!(self.conn.decode_result(value.bind_parameter(self.stmt, i)));
}
self.execute_() self.execute_()
} }
} }
/// Execute the prepared statement with named parameter(s), returning an iterator over the resulting rows.
pub fn named_query<'a>(&'a mut self, params: &[(&str, &ToSql)]) -> SqliteResult<SqliteRows<'a>> {
self.reset_if_needed();
unsafe {
try!(self.bind_named_parameters(params));
}
self.needs_reset = true;
Ok(SqliteRows::new(self))
}
unsafe fn bind_named_parameters(&mut self, params: &[(&str, &ToSql)]) -> SqliteResult<()> {
for &(name, value) in params {
let i = try!(self.parameter_index(name).ok_or(SqliteError{
code: ffi::SQLITE_MISUSE,
message: format!("Invalid parameter name: {}", name)
}));
try!(self.conn.decode_result(value.bind_parameter(self.stmt, i)));
}
Ok(())
}
} }
#[cfg(test)] #[cfg(test)]
@ -69,4 +86,14 @@ mod test {
let mut stmt = db.prepare("INSERT INTO test (id, name, flag) VALUES (:id, :name, :flag)").unwrap(); let mut stmt = db.prepare("INSERT INTO test (id, name, flag) VALUES (:id, :name, :flag)").unwrap();
stmt.named_execute(&[(":name", &"one")]).unwrap(); stmt.named_execute(&[(":name", &"one")]).unwrap();
} }
#[test]
fn test_named_query() {
let db = SqliteConnection::open_in_memory().unwrap();
let sql = "CREATE TABLE test (id INTEGER PRIMARY KEY NOT NULL, name TEXT NOT NULL, flag INTEGER)";
db.execute_batch(sql).unwrap();
let mut stmt = db.prepare("SELECT * FROM test where name = :name").unwrap();
stmt.named_query(&[(":name", &"one")]).unwrap();
}
} }