2015-11-11 21:39:54 +08:00
|
|
|
use libc::c_int;
|
2015-08-05 03:48:54 +08:00
|
|
|
|
|
|
|
use super::ffi;
|
|
|
|
|
2015-12-11 09:41:31 +08:00
|
|
|
use {SqliteResult, SqliteError, SqliteConnection, SqliteStatement, SqliteRows, SqliteRow,
|
|
|
|
str_to_cstring};
|
2015-11-11 21:39:54 +08:00
|
|
|
use types::ToSql;
|
2015-08-05 03:48:54 +08:00
|
|
|
|
|
|
|
impl SqliteConnection {
|
2015-08-07 08:19:57 +08:00
|
|
|
/// Convenience method to prepare and execute a single SQL statement with named parameter(s).
|
2015-12-05 20:43:03 +08:00
|
|
|
///
|
2015-12-11 09:30:05 +08:00
|
|
|
/// On success, returns the number of rows that were changed or inserted or deleted (via
|
|
|
|
/// `sqlite3_changes`).
|
|
|
|
///
|
2015-12-05 20:43:03 +08:00
|
|
|
/// ## Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
|
|
|
/// # use rusqlite::{SqliteConnection, SqliteResult};
|
|
|
|
/// fn insert(conn: &SqliteConnection) -> SqliteResult<i32> {
|
2015-12-11 09:30:05 +08:00
|
|
|
/// conn.execute_named("INSERT INTO test (name) VALUES (:name)", &[(":name", &"one")])
|
2015-12-05 20:43:03 +08:00
|
|
|
/// }
|
|
|
|
/// ```
|
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
|
|
|
/// Will return `Err` if `sql` cannot be converted to a C-compatible string or if the
|
|
|
|
/// underlying SQLite call fails.
|
2015-08-07 08:19:57 +08:00
|
|
|
pub fn execute_named(&self, sql: &str, params: &[(&str, &ToSql)]) -> SqliteResult<c_int> {
|
|
|
|
self.prepare(sql).and_then(|mut stmt| stmt.execute_named(params))
|
|
|
|
}
|
|
|
|
|
2015-12-11 09:30:05 +08:00
|
|
|
/// Convenience method to execute a query with named parameter(s) that is expected to return
|
|
|
|
/// a single row.
|
2015-08-07 08:19:57 +08:00
|
|
|
///
|
|
|
|
/// If the query returns more than one row, all rows except the first are ignored.
|
2015-12-05 20:43:03 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
|
|
|
/// Will return `Err` if `sql` cannot be converted to a C-compatible string or if the
|
|
|
|
/// underlying SQLite call fails.
|
2015-12-11 09:31:07 +08:00
|
|
|
pub fn query_row_named<T, F>(&self,
|
2015-12-11 06:01:19 +08:00
|
|
|
sql: &str,
|
|
|
|
params: &[(&str, &ToSql)],
|
|
|
|
f: F)
|
|
|
|
-> SqliteResult<T>
|
|
|
|
where F: FnOnce(SqliteRow) -> T
|
|
|
|
{
|
2015-08-07 08:19:57 +08:00
|
|
|
let mut stmt = try!(self.prepare(sql));
|
|
|
|
let mut rows = try!(stmt.query_named(params));
|
|
|
|
|
|
|
|
match rows.next() {
|
|
|
|
Some(row) => row.map(f),
|
2015-12-11 06:01:19 +08:00
|
|
|
None => {
|
|
|
|
Err(SqliteError {
|
|
|
|
code: ffi::SQLITE_NOTICE,
|
|
|
|
message: "Query did not return a row".to_string(),
|
|
|
|
})
|
|
|
|
}
|
2015-08-07 08:19:57 +08:00
|
|
|
}
|
|
|
|
}
|
2015-08-05 03:48:54 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
impl<'conn> SqliteStatement<'conn> {
|
|
|
|
/// Return the index of an SQL parameter given its name.
|
2015-08-08 22:19:05 +08:00
|
|
|
///
|
2015-12-05 20:43:03 +08:00
|
|
|
/// # Failure
|
|
|
|
///
|
2015-12-11 09:41:31 +08:00
|
|
|
/// Will return Err if `name` is invalid. Will return Ok(None) if the name
|
|
|
|
/// is valid but not a bound parameter of this statement.
|
|
|
|
pub fn parameter_index(&self, name: &str) -> SqliteResult<Option<i32>> {
|
|
|
|
let c_name = try!(str_to_cstring(name));
|
|
|
|
let c_index = unsafe { ffi::sqlite3_bind_parameter_index(self.stmt, c_name.as_ptr()) };
|
|
|
|
Ok(match c_index {
|
|
|
|
0 => None, // A zero is returned if no matching parameter is found.
|
|
|
|
n => Some(n),
|
|
|
|
})
|
2015-08-05 03:48:54 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Execute the prepared statement with named parameter(s).
|
|
|
|
///
|
|
|
|
/// On success, returns the number of rows that were changed or inserted or deleted (via
|
|
|
|
/// `sqlite3_changes`).
|
2015-12-05 20:43:03 +08:00
|
|
|
///
|
|
|
|
/// ## Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
|
|
|
/// # use rusqlite::{SqliteConnection, SqliteResult};
|
|
|
|
/// fn insert(conn: &SqliteConnection) -> SqliteResult<i32> {
|
2015-12-11 09:30:05 +08:00
|
|
|
/// let mut stmt = try!(conn.prepare("INSERT INTO test (name) VALUES (:name)"));
|
|
|
|
/// stmt.execute_named(&[(":name", &"one")])
|
2015-12-05 20:43:03 +08:00
|
|
|
/// }
|
|
|
|
/// ```
|
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
|
|
|
/// Will return `Err` if binding parameters fails, the executed statement returns rows (in
|
|
|
|
/// which case `query` should be used instead), or the underling SQLite call fails.
|
2015-08-07 08:19:57 +08:00
|
|
|
pub fn execute_named(&mut self, params: &[(&str, &ToSql)]) -> SqliteResult<c_int> {
|
2015-12-11 09:41:31 +08:00
|
|
|
try!(self.bind_named_parameters(params));
|
2015-08-05 03:48:54 +08:00
|
|
|
unsafe {
|
|
|
|
self.execute_()
|
|
|
|
}
|
|
|
|
}
|
2015-08-06 04:07:49 +08:00
|
|
|
|
2015-12-11 09:30:05 +08:00
|
|
|
/// Execute the prepared statement with named parameter(s), returning an iterator over the
|
|
|
|
/// resulting rows.
|
2015-12-05 20:43:03 +08:00
|
|
|
///
|
|
|
|
/// ## Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
|
|
|
/// # use rusqlite::{SqliteConnection, SqliteResult, SqliteRows};
|
|
|
|
/// fn query(conn: &SqliteConnection) -> SqliteResult<()> {
|
2015-12-11 09:30:05 +08:00
|
|
|
/// let mut stmt = try!(conn.prepare("SELECT * FROM test where name = :name"));
|
|
|
|
/// let mut rows = try!(stmt.query_named(&[(":name", &"one")]));
|
|
|
|
/// for row in rows {
|
|
|
|
/// // ...
|
|
|
|
/// }
|
|
|
|
/// Ok(())
|
2015-12-05 20:43:03 +08:00
|
|
|
/// }
|
|
|
|
/// ```
|
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
|
|
|
/// Will return `Err` if binding parameters fails.
|
2015-12-11 06:01:19 +08:00
|
|
|
pub fn query_named<'a>(&'a mut self,
|
|
|
|
params: &[(&str, &ToSql)])
|
|
|
|
-> SqliteResult<SqliteRows<'a>> {
|
2015-08-06 04:07:49 +08:00
|
|
|
self.reset_if_needed();
|
2015-12-11 09:41:31 +08:00
|
|
|
try!(self.bind_named_parameters(params));
|
2015-08-06 04:07:49 +08:00
|
|
|
|
|
|
|
self.needs_reset = true;
|
|
|
|
Ok(SqliteRows::new(self))
|
|
|
|
}
|
|
|
|
|
2015-12-11 09:41:31 +08:00
|
|
|
fn bind_named_parameters(&mut self, params: &[(&str, &ToSql)]) -> SqliteResult<()> {
|
2015-12-11 09:27:09 +08:00
|
|
|
// Always check that the number of parameters is correct.
|
2015-12-11 09:41:31 +08:00
|
|
|
assert!(params.len() as c_int == unsafe { ffi::sqlite3_bind_parameter_count(self.stmt) },
|
2015-12-11 09:27:09 +08:00
|
|
|
"incorrect number of parameters to query(): expected {}, got {}",
|
2015-12-11 09:41:31 +08:00
|
|
|
unsafe { ffi::sqlite3_bind_parameter_count(self.stmt) },
|
2015-12-11 09:27:09 +08:00
|
|
|
params.len());
|
|
|
|
|
|
|
|
// In debug, also sanity check that we got distinct parameter names.
|
|
|
|
debug_assert!({
|
|
|
|
use std::collections::HashSet;
|
|
|
|
|
|
|
|
let mut s = HashSet::with_capacity(params.len());
|
|
|
|
for &(name, _) in params {
|
|
|
|
s.insert(name);
|
|
|
|
}
|
|
|
|
|
|
|
|
s.len() == params.len()
|
|
|
|
},
|
|
|
|
"named parameters must be unique");
|
|
|
|
|
2015-08-06 04:07:49 +08:00
|
|
|
for &(name, value) in params {
|
2015-12-11 09:41:31 +08:00
|
|
|
if let Some(i) = try!(self.parameter_index(name)) {
|
|
|
|
try!(self.conn.decode_result(unsafe { value.bind_parameter(self.stmt, i) }));
|
|
|
|
} else {
|
|
|
|
return Err(SqliteError {
|
|
|
|
code: ffi::SQLITE_MISUSE,
|
|
|
|
message: format!("Invalid parameter name: {}", name),
|
|
|
|
});
|
|
|
|
}
|
2015-08-06 04:07:49 +08:00
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
2015-08-05 03:48:54 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod test {
|
|
|
|
use SqliteConnection;
|
|
|
|
|
2015-08-07 08:19:57 +08:00
|
|
|
#[test]
|
|
|
|
fn test_execute_named() {
|
|
|
|
let db = SqliteConnection::open_in_memory().unwrap();
|
|
|
|
db.execute_batch("CREATE TABLE foo(x INTEGER)").unwrap();
|
|
|
|
|
2015-12-11 06:01:19 +08:00
|
|
|
assert_eq!(db.execute_named("INSERT INTO foo(x) VALUES (:x)", &[(":x", &1i32)]).unwrap(),
|
|
|
|
1);
|
|
|
|
assert_eq!(db.execute_named("INSERT INTO foo(x) VALUES (:x)", &[(":x", &2i32)]).unwrap(),
|
|
|
|
1);
|
2015-08-07 08:19:57 +08:00
|
|
|
|
2015-12-11 06:01:19 +08:00
|
|
|
assert_eq!(3i32,
|
2015-12-11 09:31:07 +08:00
|
|
|
db.query_row_named("SELECT SUM(x) FROM foo WHERE x > :x",
|
2015-12-11 06:01:19 +08:00
|
|
|
&[(":x", &0i32)],
|
|
|
|
|r| r.get(0))
|
|
|
|
.unwrap());
|
2015-08-07 08:19:57 +08:00
|
|
|
}
|
|
|
|
|
2015-12-11 06:01:19 +08:00
|
|
|
#[test]
|
2015-08-07 08:19:57 +08:00
|
|
|
fn test_stmt_execute_named() {
|
2015-08-05 03:48:54 +08:00
|
|
|
let db = SqliteConnection::open_in_memory().unwrap();
|
2015-12-11 06:01:19 +08:00
|
|
|
let sql = "CREATE TABLE test (id INTEGER PRIMARY KEY NOT NULL, name TEXT NOT NULL, flag \
|
|
|
|
INTEGER)";
|
2015-08-05 03:48:54 +08:00
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
|
2015-12-11 09:27:09 +08:00
|
|
|
let mut stmt = db.prepare("INSERT INTO test (name) VALUES (:name)").unwrap();
|
2015-08-07 08:19:57 +08:00
|
|
|
stmt.execute_named(&[(":name", &"one")]).unwrap();
|
2015-12-11 09:16:46 +08:00
|
|
|
|
|
|
|
assert_eq!(1i32,
|
2015-12-11 09:31:07 +08:00
|
|
|
db.query_row_named("SELECT COUNT(*) FROM test WHERE name = :name",
|
2015-12-11 09:16:46 +08:00
|
|
|
&[(":name", &"one")],
|
|
|
|
|r| r.get(0))
|
|
|
|
.unwrap());
|
2015-08-05 03:48:54 +08:00
|
|
|
}
|
2015-08-06 04:07:49 +08:00
|
|
|
|
2015-12-11 06:01:19 +08:00
|
|
|
#[test]
|
2015-08-07 08:19:57 +08:00
|
|
|
fn test_query_named() {
|
2015-08-06 04:07:49 +08:00
|
|
|
let db = SqliteConnection::open_in_memory().unwrap();
|
2015-12-11 06:01:19 +08:00
|
|
|
let sql = "CREATE TABLE test (id INTEGER PRIMARY KEY NOT NULL, name TEXT NOT NULL, flag \
|
|
|
|
INTEGER)";
|
2015-08-06 04:07:49 +08:00
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
|
|
|
|
let mut stmt = db.prepare("SELECT * FROM test where name = :name").unwrap();
|
2015-08-07 08:19:57 +08:00
|
|
|
stmt.query_named(&[(":name", &"one")]).unwrap();
|
2015-08-06 04:07:49 +08:00
|
|
|
}
|
2015-12-11 09:27:09 +08:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[should_panic]
|
|
|
|
fn test_panic_on_incorrect_number_of_parameters() {
|
|
|
|
let db = SqliteConnection::open_in_memory().unwrap();
|
|
|
|
|
|
|
|
let mut stmt = db.prepare("SELECT 1 WHERE 1 = :one AND 2 = :two").unwrap();
|
|
|
|
let _ = stmt.query_named(&[(":one", &1i32)]);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[cfg(debug_assertions)]
|
|
|
|
#[should_panic]
|
|
|
|
fn test_debug_panic_on_incorrect_parameter_names() {
|
|
|
|
let db = SqliteConnection::open_in_memory().unwrap();
|
|
|
|
|
|
|
|
let mut stmt = db.prepare("SELECT 1 WHERE 1 = :one AND 2 = :two").unwrap();
|
|
|
|
let _ = stmt.query_named(&[(":one", &1i32), (":one", &2i32)]);
|
|
|
|
}
|
2015-11-11 21:39:54 +08:00
|
|
|
}
|