2018-08-17 00:29:46 +08:00
|
|
|
//! Rusqlite is an ergonomic wrapper for using SQLite from Rust. It attempts to
|
|
|
|
//! expose an interface similar to [rust-postgres](https://github.com/sfackler/rust-postgres).
|
2014-11-04 06:11:00 +08:00
|
|
|
//!
|
|
|
|
//! ```rust
|
2019-02-09 14:16:05 +08:00
|
|
|
//! use rusqlite::{params, Connection, Result};
|
2014-11-04 06:11:00 +08:00
|
|
|
//!
|
2015-02-04 07:59:58 +08:00
|
|
|
//! #[derive(Debug)]
|
2014-11-04 06:11:00 +08:00
|
|
|
//! struct Person {
|
|
|
|
//! id: i32,
|
|
|
|
//! name: String,
|
2018-08-17 00:29:46 +08:00
|
|
|
//! data: Option<Vec<u8>>,
|
2014-11-04 06:11:00 +08:00
|
|
|
//! }
|
|
|
|
//!
|
2019-02-09 14:16:05 +08:00
|
|
|
//! fn main() -> Result<()> {
|
|
|
|
//! let conn = Connection::open_in_memory()?;
|
2014-11-04 06:11:00 +08:00
|
|
|
//!
|
2018-08-17 00:29:46 +08:00
|
|
|
//! conn.execute(
|
|
|
|
//! "CREATE TABLE person (
|
2014-11-04 06:11:00 +08:00
|
|
|
//! id INTEGER PRIMARY KEY,
|
|
|
|
//! name TEXT NOT NULL,
|
|
|
|
//! data BLOB
|
2018-08-17 00:29:46 +08:00
|
|
|
//! )",
|
2019-01-30 07:33:57 +08:00
|
|
|
//! params![],
|
2019-02-09 14:16:05 +08:00
|
|
|
//! )?;
|
2014-11-04 06:11:00 +08:00
|
|
|
//! let me = Person {
|
|
|
|
//! id: 0,
|
|
|
|
//! name: "Steven".to_string(),
|
2018-08-17 00:29:46 +08:00
|
|
|
//! data: None,
|
2014-11-04 06:11:00 +08:00
|
|
|
//! };
|
2018-08-17 00:29:46 +08:00
|
|
|
//! conn.execute(
|
2020-07-11 23:59:29 +08:00
|
|
|
//! "INSERT INTO person (name, data) VALUES (?1, ?2)",
|
|
|
|
//! params![me.name, me.data],
|
2019-02-09 14:16:05 +08:00
|
|
|
//! )?;
|
2014-11-04 06:11:00 +08:00
|
|
|
//!
|
2020-07-11 23:59:29 +08:00
|
|
|
//! let mut stmt = conn.prepare("SELECT id, name, data FROM person")?;
|
2019-02-22 03:48:09 +08:00
|
|
|
//! let person_iter = stmt.query_map(params![], |row| {
|
|
|
|
//! Ok(Person {
|
|
|
|
//! id: row.get(0)?,
|
|
|
|
//! name: row.get(1)?,
|
2020-07-11 23:59:29 +08:00
|
|
|
//! data: row.get(2)?,
|
2018-10-28 15:51:02 +08:00
|
|
|
//! })
|
2019-02-22 03:48:09 +08:00
|
|
|
//! })?;
|
2015-05-07 21:36:29 +08:00
|
|
|
//!
|
|
|
|
//! for person in person_iter {
|
|
|
|
//! println!("Found person {:?}", person.unwrap());
|
2014-11-04 06:11:00 +08:00
|
|
|
//! }
|
2019-02-09 14:16:05 +08:00
|
|
|
//! Ok(())
|
2014-11-04 06:11:00 +08:00
|
|
|
//! }
|
|
|
|
//! ```
|
2020-05-17 17:21:10 +08:00
|
|
|
#![warn(missing_docs)]
|
2016-02-14 23:11:59 +08:00
|
|
|
|
2019-01-27 06:30:27 +08:00
|
|
|
pub use libsqlite3_sys as ffi;
|
2014-10-20 07:56:41 +08:00
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
use std::cell::RefCell;
|
2015-08-28 01:43:43 +08:00
|
|
|
use std::convert;
|
2018-08-11 18:48:21 +08:00
|
|
|
use std::default::Default;
|
|
|
|
use std::ffi::{CStr, CString};
|
2014-10-20 07:56:41 +08:00
|
|
|
use std::fmt;
|
2018-08-11 18:48:21 +08:00
|
|
|
use std::os::raw::{c_char, c_int};
|
|
|
|
|
2015-12-11 05:48:09 +08:00
|
|
|
use std::path::{Path, PathBuf};
|
2015-12-13 03:06:03 +08:00
|
|
|
use std::result;
|
2015-01-08 03:05:36 +08:00
|
|
|
use std::str;
|
2019-02-02 19:46:52 +08:00
|
|
|
use std::sync::atomic::Ordering;
|
|
|
|
use std::sync::{Arc, Mutex};
|
2017-03-09 05:57:43 +08:00
|
|
|
|
2018-10-31 03:11:35 +08:00
|
|
|
use crate::cache::StatementCache;
|
2019-02-02 19:46:52 +08:00
|
|
|
use crate::inner_connection::{InnerConnection, BYPASS_SQLITE_INIT};
|
2018-10-31 03:11:35 +08:00
|
|
|
use crate::raw_statement::RawStatement;
|
2019-01-30 07:33:57 +08:00
|
|
|
use crate::types::ValueRef;
|
2014-10-20 07:56:41 +08:00
|
|
|
|
2019-02-02 19:46:52 +08:00
|
|
|
pub use crate::cache::CachedStatement;
|
2019-03-20 03:43:40 +08:00
|
|
|
pub use crate::column::Column;
|
2018-10-31 03:11:35 +08:00
|
|
|
pub use crate::error::Error;
|
|
|
|
pub use crate::ffi::ErrorCode;
|
2018-08-11 18:48:21 +08:00
|
|
|
#[cfg(feature = "hooks")]
|
2019-02-03 21:01:42 +08:00
|
|
|
pub use crate::hooks::Action;
|
2015-12-11 05:48:09 +08:00
|
|
|
#[cfg(feature = "load_extension")]
|
2018-11-06 03:04:04 +08:00
|
|
|
pub use crate::load_extension_guard::LoadExtensionGuard;
|
2020-06-29 09:52:38 +08:00
|
|
|
pub use crate::row::{AndThenRows, Map, MappedRows, Row, RowIndex, Rows};
|
2019-02-02 19:46:52 +08:00
|
|
|
pub use crate::statement::{Statement, StatementStatus};
|
|
|
|
pub use crate::transaction::{DropBehavior, Savepoint, Transaction, TransactionBehavior};
|
|
|
|
pub use crate::types::ToSql;
|
|
|
|
pub use crate::version::*;
|
2015-02-24 09:16:49 +08:00
|
|
|
|
2019-02-02 22:17:20 +08:00
|
|
|
#[macro_use]
|
2015-12-13 14:04:09 +08:00
|
|
|
mod error;
|
2019-02-02 22:17:20 +08:00
|
|
|
|
2017-02-09 04:11:15 +08:00
|
|
|
#[cfg(feature = "backup")]
|
|
|
|
pub mod backup;
|
|
|
|
#[cfg(feature = "blob")]
|
|
|
|
pub mod blob;
|
2018-08-11 18:48:21 +08:00
|
|
|
mod busy;
|
2016-05-18 01:01:55 +08:00
|
|
|
mod cache;
|
2019-06-18 01:20:53 +08:00
|
|
|
#[cfg(feature = "collation")]
|
|
|
|
mod collation;
|
2019-03-20 03:43:40 +08:00
|
|
|
mod column;
|
2019-02-02 22:17:20 +08:00
|
|
|
pub mod config;
|
2018-08-11 19:37:56 +08:00
|
|
|
#[cfg(any(feature = "functions", feature = "vtab"))]
|
|
|
|
mod context;
|
2018-08-11 18:48:21 +08:00
|
|
|
#[cfg(feature = "functions")]
|
|
|
|
pub mod functions;
|
|
|
|
#[cfg(feature = "hooks")]
|
|
|
|
mod hooks;
|
2019-02-02 19:46:52 +08:00
|
|
|
mod inner_connection;
|
2017-02-09 04:11:15 +08:00
|
|
|
#[cfg(feature = "limits")]
|
|
|
|
pub mod limits;
|
2018-08-11 18:48:21 +08:00
|
|
|
#[cfg(feature = "load_extension")]
|
|
|
|
mod load_extension_guard;
|
2019-02-03 16:17:37 +08:00
|
|
|
mod pragma;
|
2016-05-17 23:06:43 +08:00
|
|
|
mod raw_statement;
|
2017-03-09 05:57:43 +08:00
|
|
|
mod row;
|
2019-01-13 19:46:19 +08:00
|
|
|
#[cfg(feature = "session")]
|
|
|
|
pub mod session;
|
2017-03-08 23:48:14 +08:00
|
|
|
mod statement;
|
2016-08-13 19:55:30 +08:00
|
|
|
#[cfg(feature = "trace")]
|
|
|
|
pub mod trace;
|
2018-08-11 18:48:21 +08:00
|
|
|
mod transaction;
|
|
|
|
pub mod types;
|
2017-09-22 01:27:07 +08:00
|
|
|
mod unlock_notify;
|
2018-08-11 18:48:21 +08:00
|
|
|
mod version;
|
2018-05-01 04:09:41 +08:00
|
|
|
#[cfg(feature = "vtab")]
|
2016-08-13 19:55:30 +08:00
|
|
|
pub mod vtab;
|
2014-10-20 07:56:41 +08:00
|
|
|
|
2020-04-14 14:47:12 +08:00
|
|
|
pub(crate) mod util;
|
2020-04-14 17:04:19 +08:00
|
|
|
pub(crate) use util::SmallCString;
|
2020-04-14 14:47:12 +08:00
|
|
|
|
2016-05-18 01:11:25 +08:00
|
|
|
// Number of cached prepared statements we'll hold on to.
|
|
|
|
const STATEMENT_CACHE_DEFAULT_CAPACITY: usize = 16;
|
2018-09-16 17:10:19 +08:00
|
|
|
/// To be used when your statement has no [parameter](https://sqlite.org/lang_expr.html#varparam).
|
2018-12-08 04:57:04 +08:00
|
|
|
pub const NO_PARAMS: &[&dyn ToSql] = &[];
|
2016-05-18 01:11:25 +08:00
|
|
|
|
2019-01-30 07:33:57 +08:00
|
|
|
/// A macro making it more convenient to pass heterogeneous lists
|
|
|
|
/// of parameters as a `&[&dyn ToSql]`.
|
|
|
|
///
|
|
|
|
/// # Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
|
|
|
/// # use rusqlite::{Result, Connection, params};
|
|
|
|
///
|
|
|
|
/// struct Person {
|
|
|
|
/// name: String,
|
|
|
|
/// age_in_years: u8,
|
|
|
|
/// data: Option<Vec<u8>>,
|
|
|
|
/// }
|
|
|
|
///
|
|
|
|
/// fn add_person(conn: &Connection, person: &Person) -> Result<()> {
|
|
|
|
/// conn.execute("INSERT INTO person (name, age_in_years, data)
|
|
|
|
/// VALUES (?1, ?2, ?3)",
|
|
|
|
/// params![person.name, person.age_in_years, person.data])?;
|
|
|
|
/// Ok(())
|
|
|
|
/// }
|
|
|
|
/// ```
|
|
|
|
#[macro_export]
|
|
|
|
macro_rules! params {
|
|
|
|
() => {
|
|
|
|
$crate::NO_PARAMS
|
|
|
|
};
|
|
|
|
($($param:expr),+ $(,)?) => {
|
2020-01-19 02:04:28 +08:00
|
|
|
&[$(&$param as &dyn $crate::ToSql),+] as &[&dyn $crate::ToSql]
|
2019-01-30 07:33:57 +08:00
|
|
|
};
|
|
|
|
}
|
2016-05-18 01:11:25 +08:00
|
|
|
|
2019-01-30 07:33:57 +08:00
|
|
|
/// A macro making it more convenient to pass lists of named parameters
|
|
|
|
/// as a `&[(&str, &dyn ToSql)]`.
|
|
|
|
///
|
|
|
|
/// # Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
|
|
|
/// # use rusqlite::{Result, Connection, named_params};
|
|
|
|
///
|
|
|
|
/// struct Person {
|
|
|
|
/// name: String,
|
|
|
|
/// age_in_years: u8,
|
|
|
|
/// data: Option<Vec<u8>>,
|
|
|
|
/// }
|
|
|
|
///
|
|
|
|
/// fn add_person(conn: &Connection, person: &Person) -> Result<()> {
|
|
|
|
/// conn.execute_named(
|
|
|
|
/// "INSERT INTO person (name, age_in_years, data)
|
|
|
|
/// VALUES (:name, :age, :data)",
|
|
|
|
/// named_params!{
|
|
|
|
/// ":name": person.name,
|
|
|
|
/// ":age": person.age_in_years,
|
|
|
|
/// ":data": person.data,
|
|
|
|
/// }
|
|
|
|
/// )?;
|
|
|
|
/// Ok(())
|
|
|
|
/// }
|
|
|
|
/// ```
|
|
|
|
#[macro_export]
|
|
|
|
macro_rules! named_params {
|
|
|
|
() => {
|
|
|
|
&[]
|
|
|
|
};
|
|
|
|
// Note: It's a lot more work to support this as part of the same macro as
|
|
|
|
// `params!`, unfortunately.
|
|
|
|
($($param_name:literal: $param_val:expr),+ $(,)?) => {
|
|
|
|
&[$(($param_name, &$param_val as &dyn $crate::ToSql)),+]
|
|
|
|
};
|
|
|
|
}
|
2015-12-13 03:06:03 +08:00
|
|
|
|
2014-11-04 06:11:00 +08:00
|
|
|
/// A typedef of the result returned by many methods.
|
2020-04-07 05:43:06 +08:00
|
|
|
pub type Result<T, E = Error> = result::Result<T, E>;
|
2014-10-20 07:56:41 +08:00
|
|
|
|
2018-12-16 18:15:21 +08:00
|
|
|
/// See the [method documentation](#tymethod.optional).
|
|
|
|
pub trait OptionalExtension<T> {
|
|
|
|
/// Converts a `Result<T>` into a `Result<Option<T>>`.
|
|
|
|
///
|
2018-12-20 04:58:33 +08:00
|
|
|
/// By default, Rusqlite treats 0 rows being returned from a query that is
|
|
|
|
/// expected to return 1 row as an error. This method will
|
2018-12-16 18:15:21 +08:00
|
|
|
/// handle that error, and give you back an `Option<T>` instead.
|
|
|
|
fn optional(self) -> Result<Option<T>>;
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<T> OptionalExtension<T> for Result<T> {
|
|
|
|
fn optional(self) -> Result<Option<T>> {
|
|
|
|
match self {
|
|
|
|
Ok(value) => Ok(Some(value)),
|
|
|
|
Err(Error::QueryReturnedNoRows) => Ok(None),
|
|
|
|
Err(e) => Err(e),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-10-20 07:56:41 +08:00
|
|
|
unsafe fn errmsg_to_string(errmsg: *const c_char) -> String {
|
2015-02-24 08:52:48 +08:00
|
|
|
let c_slice = CStr::from_ptr(errmsg).to_bytes();
|
2016-05-07 18:08:57 +08:00
|
|
|
String::from_utf8_lossy(c_slice).into_owned()
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
2020-04-14 17:04:19 +08:00
|
|
|
fn str_to_cstring(s: &str) -> Result<SmallCString> {
|
|
|
|
Ok(SmallCString::new(s)?)
|
2015-02-24 08:52:48 +08:00
|
|
|
}
|
|
|
|
|
2019-02-27 11:38:41 +08:00
|
|
|
/// Returns `Ok((string ptr, len as c_int, SQLITE_STATIC | SQLITE_TRANSIENT))`
|
|
|
|
/// normally.
|
2020-03-26 02:20:05 +08:00
|
|
|
/// Returns error if the string is too large for sqlite.
|
2019-02-27 11:38:41 +08:00
|
|
|
/// The `sqlite3_destructor_type` item is always `SQLITE_TRANSIENT` unless
|
|
|
|
/// the string was empty (in which case it's `SQLITE_STATIC`, and the ptr is
|
|
|
|
/// static).
|
2019-07-25 02:08:31 +08:00
|
|
|
fn str_for_sqlite(s: &[u8]) -> Result<(*const c_char, c_int, ffi::sqlite3_destructor_type)> {
|
2019-02-27 11:38:41 +08:00
|
|
|
let len = len_as_c_int(s.len())?;
|
2020-03-26 02:20:05 +08:00
|
|
|
let (ptr, dtor_info) = if len != 0 {
|
|
|
|
(s.as_ptr() as *const c_char, ffi::SQLITE_TRANSIENT())
|
2019-02-27 11:38:41 +08:00
|
|
|
} else {
|
2020-03-26 02:20:05 +08:00
|
|
|
// Return a pointer guaranteed to live forever
|
|
|
|
("".as_ptr() as *const c_char, ffi::SQLITE_STATIC())
|
|
|
|
};
|
|
|
|
Ok((ptr, len, dtor_info))
|
2019-02-27 11:38:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Helper to cast to c_int safely, returning the correct error type if the cast
|
|
|
|
// failed.
|
|
|
|
fn len_as_c_int(len: usize) -> Result<c_int> {
|
|
|
|
if len >= (c_int::max_value() as usize) {
|
2019-03-10 18:12:14 +08:00
|
|
|
Err(Error::SqliteFailure(
|
|
|
|
ffi::Error::new(ffi::SQLITE_TOOBIG),
|
|
|
|
None,
|
|
|
|
))
|
2019-02-27 11:38:41 +08:00
|
|
|
} else {
|
|
|
|
Ok(len as c_int)
|
|
|
|
}
|
2015-02-24 08:52:48 +08:00
|
|
|
}
|
|
|
|
|
2020-04-09 16:35:38 +08:00
|
|
|
#[cfg(unix)]
|
|
|
|
fn path_to_cstring(p: &Path) -> Result<CString> {
|
|
|
|
use std::os::unix::ffi::OsStrExt;
|
|
|
|
Ok(CString::new(p.as_os_str().as_bytes())?)
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(unix))]
|
2015-12-13 03:06:03 +08:00
|
|
|
fn path_to_cstring(p: &Path) -> Result<CString> {
|
2018-10-31 03:11:35 +08:00
|
|
|
let s = p.to_str().ok_or_else(|| Error::InvalidPath(p.to_owned()))?;
|
2020-04-14 14:47:12 +08:00
|
|
|
Ok(CString::new(s)?)
|
2015-02-24 08:52:48 +08:00
|
|
|
}
|
|
|
|
|
2015-12-10 05:27:18 +08:00
|
|
|
/// Name for a database within a SQLite connection.
|
2018-08-11 18:48:21 +08:00
|
|
|
#[derive(Copy, Clone)]
|
2015-12-10 05:27:18 +08:00
|
|
|
pub enum DatabaseName<'a> {
|
|
|
|
/// The main database.
|
|
|
|
Main,
|
|
|
|
|
|
|
|
/// The temporary database (e.g., any "CREATE TEMPORARY TABLE" tables).
|
|
|
|
Temp,
|
|
|
|
|
|
|
|
/// A database that has been attached via "ATTACH DATABASE ...".
|
|
|
|
Attached(&'a str),
|
|
|
|
}
|
|
|
|
|
2018-08-17 00:29:46 +08:00
|
|
|
// Currently DatabaseName is only used by the backup and blob mods, so hide
|
|
|
|
// this (private) impl to avoid dead code warnings.
|
2019-01-26 18:02:34 +08:00
|
|
|
#[cfg(any(
|
|
|
|
feature = "backup",
|
|
|
|
feature = "blob",
|
|
|
|
feature = "session",
|
2020-01-15 00:11:36 +08:00
|
|
|
feature = "modern_sqlite"
|
2019-01-26 18:02:34 +08:00
|
|
|
))]
|
2019-02-03 18:02:38 +08:00
|
|
|
impl DatabaseName<'_> {
|
2020-04-14 17:04:19 +08:00
|
|
|
fn to_cstring(&self) -> Result<util::SmallCString> {
|
2018-08-11 18:48:21 +08:00
|
|
|
use self::DatabaseName::{Attached, Main, Temp};
|
2016-02-14 23:11:59 +08:00
|
|
|
match *self {
|
2015-12-11 05:48:09 +08:00
|
|
|
Main => str_to_cstring("main"),
|
|
|
|
Temp => str_to_cstring("temp"),
|
2015-12-10 05:27:18 +08:00
|
|
|
Attached(s) => str_to_cstring(s),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-11-04 06:11:00 +08:00
|
|
|
/// A connection to a SQLite database.
|
2015-12-13 02:50:12 +08:00
|
|
|
pub struct Connection {
|
|
|
|
db: RefCell<InnerConnection>,
|
2016-05-18 01:11:25 +08:00
|
|
|
cache: StatementCache,
|
2015-08-01 18:04:02 +08:00
|
|
|
path: Option<PathBuf>,
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
2015-12-13 02:50:12 +08:00
|
|
|
unsafe impl Send for Connection {}
|
2015-04-04 03:48:35 +08:00
|
|
|
|
2017-07-19 18:22:31 +08:00
|
|
|
impl Drop for Connection {
|
|
|
|
fn drop(&mut self) {
|
|
|
|
self.flush_prepared_statement_cache();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-12-13 02:50:12 +08:00
|
|
|
impl Connection {
|
2014-11-04 06:11:00 +08:00
|
|
|
/// Open a new connection to a SQLite database.
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// `Connection::open(path)` is equivalent to
|
|
|
|
/// `Connection::open_with_flags(path,
|
|
|
|
/// OpenFlags::SQLITE_OPEN_READ_WRITE |
|
|
|
|
/// OpenFlags::SQLITE_OPEN_CREATE)`.
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
2019-06-25 02:05:36 +08:00
|
|
|
/// ```rust,no_run
|
|
|
|
/// # use rusqlite::{Connection, Result};
|
|
|
|
/// fn open_my_db() -> Result<()> {
|
|
|
|
/// let path = "./my_db.db3";
|
|
|
|
/// let db = Connection::open(&path)?;
|
|
|
|
/// println!("{}", db.is_autocommit());
|
|
|
|
/// Ok(())
|
|
|
|
/// }
|
|
|
|
/// ```
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Will return `Err` if `path` cannot be converted to a C-compatible
|
|
|
|
/// string or if the underlying SQLite open call fails.
|
2015-12-13 03:06:03 +08:00
|
|
|
pub fn open<P: AsRef<Path>>(path: P) -> Result<Connection> {
|
2018-06-29 02:46:28 +08:00
|
|
|
let flags = OpenFlags::default();
|
2015-12-13 02:50:12 +08:00
|
|
|
Connection::open_with_flags(path, flags)
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
2015-02-07 09:07:23 +08:00
|
|
|
/// Open a new connection to an in-memory SQLite database.
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
|
|
|
/// Will return `Err` if the underlying SQLite open call fails.
|
2015-12-13 03:06:03 +08:00
|
|
|
pub fn open_in_memory() -> Result<Connection> {
|
2018-06-29 02:46:28 +08:00
|
|
|
let flags = OpenFlags::default();
|
2015-12-13 02:50:12 +08:00
|
|
|
Connection::open_in_memory_with_flags(flags)
|
2015-02-07 09:07:23 +08:00
|
|
|
}
|
|
|
|
|
2014-11-04 06:11:00 +08:00
|
|
|
/// Open a new connection to a SQLite database.
|
|
|
|
///
|
2017-02-11 17:50:13 +08:00
|
|
|
/// [Database Connection](http://www.sqlite.org/c3ref/open.html) for a description of valid
|
2014-11-04 06:11:00 +08:00
|
|
|
/// flag combinations.
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Will return `Err` if `path` cannot be converted to a C-compatible
|
|
|
|
/// string or if the underlying SQLite open call fails.
|
2016-02-03 02:12:00 +08:00
|
|
|
pub fn open_with_flags<P: AsRef<Path>>(path: P, flags: OpenFlags) -> Result<Connection> {
|
2018-10-31 03:11:35 +08:00
|
|
|
let c_path = path_to_cstring(path.as_ref())?;
|
2019-06-17 17:42:54 +08:00
|
|
|
InnerConnection::open_with_flags(&c_path, flags, None).map(|db| Connection {
|
2018-08-11 18:48:21 +08:00
|
|
|
db: RefCell::new(db),
|
|
|
|
cache: StatementCache::with_capacity(STATEMENT_CACHE_DEFAULT_CAPACITY),
|
|
|
|
path: Some(path.as_ref().to_path_buf()),
|
|
|
|
})
|
2016-02-03 02:12:00 +08:00
|
|
|
}
|
2015-02-07 09:07:23 +08:00
|
|
|
|
2020-02-09 19:08:25 +08:00
|
|
|
/// Open a new connection to a SQLite database using the specific flags and
|
|
|
|
/// vfs name.
|
2015-02-07 09:07:23 +08:00
|
|
|
///
|
2017-02-11 17:50:13 +08:00
|
|
|
/// [Database Connection](http://www.sqlite.org/c3ref/open.html) for a description of valid
|
2015-02-07 09:07:23 +08:00
|
|
|
/// flag combinations.
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
2020-02-09 19:08:25 +08:00
|
|
|
/// Will return `Err` if either `path` or `vfs` cannot be converted to a
|
|
|
|
/// C-compatible string or if the underlying SQLite open call fails.
|
|
|
|
pub fn open_with_flags_and_vfs<P: AsRef<Path>>(
|
|
|
|
path: P,
|
|
|
|
flags: OpenFlags,
|
|
|
|
vfs: &str,
|
|
|
|
) -> Result<Connection> {
|
2019-06-17 17:42:54 +08:00
|
|
|
let c_path = path_to_cstring(path.as_ref())?;
|
|
|
|
let c_vfs = str_to_cstring(vfs)?;
|
2020-02-09 19:08:25 +08:00
|
|
|
InnerConnection::open_with_flags(&c_path, flags, Some(&c_vfs)).map(|db| Connection {
|
2018-08-11 18:48:21 +08:00
|
|
|
db: RefCell::new(db),
|
|
|
|
cache: StatementCache::with_capacity(STATEMENT_CACHE_DEFAULT_CAPACITY),
|
|
|
|
path: Some(path.as_ref().to_path_buf()),
|
|
|
|
})
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
2015-02-07 09:07:23 +08:00
|
|
|
/// Open a new connection to an in-memory SQLite database.
|
|
|
|
///
|
2017-02-11 17:50:13 +08:00
|
|
|
/// [Database Connection](http://www.sqlite.org/c3ref/open.html) for a description of valid
|
2015-02-07 09:07:23 +08:00
|
|
|
/// flag combinations.
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
|
|
|
/// Will return `Err` if the underlying SQLite open call fails.
|
2015-12-13 03:13:29 +08:00
|
|
|
pub fn open_in_memory_with_flags(flags: OpenFlags) -> Result<Connection> {
|
2020-02-09 19:21:20 +08:00
|
|
|
Connection::open_with_flags(":memory:", flags)
|
2019-06-17 17:42:54 +08:00
|
|
|
}
|
|
|
|
|
2020-02-09 19:08:25 +08:00
|
|
|
/// Open a new connection to an in-memory SQLite database using the specific
|
|
|
|
/// flags and vfs name.
|
2019-06-17 17:42:54 +08:00
|
|
|
///
|
|
|
|
/// [Database Connection](http://www.sqlite.org/c3ref/open.html) for a description of valid
|
|
|
|
/// flag combinations.
|
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
|
|
|
/// Will return `Err` if vfs` cannot be converted to a C-compatible
|
|
|
|
/// string or if the underlying SQLite open call fails.
|
|
|
|
pub fn open_in_memory_with_flags_and_vfs(flags: OpenFlags, vfs: &str) -> Result<Connection> {
|
2020-02-09 19:21:20 +08:00
|
|
|
Connection::open_with_flags_and_vfs(":memory:", flags, vfs)
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Convenience method to run multiple SQL statements (that cannot take any
|
|
|
|
/// parameters).
|
2014-11-04 06:11:00 +08:00
|
|
|
///
|
|
|
|
/// ## Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
2015-12-13 03:06:03 +08:00
|
|
|
/// # use rusqlite::{Connection, Result};
|
|
|
|
/// fn create_tables(conn: &Connection) -> Result<()> {
|
2018-08-17 00:29:46 +08:00
|
|
|
/// conn.execute_batch(
|
|
|
|
/// "BEGIN;
|
2014-11-04 06:11:00 +08:00
|
|
|
/// CREATE TABLE foo(x INTEGER);
|
|
|
|
/// CREATE TABLE bar(y TEXT);
|
2018-08-17 00:29:46 +08:00
|
|
|
/// COMMIT;",
|
|
|
|
/// )
|
2014-11-04 06:11:00 +08:00
|
|
|
/// }
|
|
|
|
/// ```
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Will return `Err` if `sql` cannot be converted to a C-compatible string
|
|
|
|
/// or if the underlying SQLite call fails.
|
2015-12-13 03:06:03 +08:00
|
|
|
pub fn execute_batch(&self, sql: &str) -> Result<()> {
|
2020-06-27 01:35:14 +08:00
|
|
|
let mut sql = sql;
|
|
|
|
while !sql.is_empty() {
|
|
|
|
let stmt = self.prepare(sql)?;
|
2020-08-18 01:13:55 +08:00
|
|
|
if !stmt.stmt.is_null() && stmt.step()? && cfg!(feature = "extra_check") {
|
|
|
|
// Some PRAGMA may return rows
|
2020-06-27 01:35:14 +08:00
|
|
|
return Err(Error::ExecuteReturnedResults);
|
|
|
|
}
|
|
|
|
let tail = stmt.stmt.tail();
|
|
|
|
if tail == 0 || tail >= sql.len() {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
sql = &sql[tail..];
|
|
|
|
}
|
|
|
|
Ok(())
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
2014-11-04 06:11:00 +08:00
|
|
|
/// Convenience method to prepare and execute a single SQL statement.
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// On success, returns the number of rows that were changed or inserted or
|
|
|
|
/// deleted (via `sqlite3_changes`).
|
2014-11-04 06:11:00 +08:00
|
|
|
///
|
|
|
|
/// ## Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
2015-12-13 02:50:12 +08:00
|
|
|
/// # use rusqlite::{Connection};
|
|
|
|
/// fn update_rows(conn: &Connection) {
|
2018-09-16 15:49:23 +08:00
|
|
|
/// match conn.execute("UPDATE foo SET bar = 'baz' WHERE qux = ?", &[1i32]) {
|
2014-11-04 06:11:00 +08:00
|
|
|
/// Ok(updated) => println!("{} rows were updated", updated),
|
|
|
|
/// Err(err) => println!("update failed: {}", err),
|
|
|
|
/// }
|
|
|
|
/// }
|
|
|
|
/// ```
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Will return `Err` if `sql` cannot be converted to a C-compatible string
|
|
|
|
/// or if the underlying SQLite call fails.
|
2018-09-16 15:49:23 +08:00
|
|
|
pub fn execute<P>(&self, sql: &str, params: P) -> Result<usize>
|
|
|
|
where
|
|
|
|
P: IntoIterator,
|
|
|
|
P::Item: ToSql,
|
|
|
|
{
|
2017-04-08 01:43:24 +08:00
|
|
|
self.prepare(sql)
|
2018-10-28 17:28:19 +08:00
|
|
|
.and_then(|mut stmt| stmt.check_no_tail().and_then(|_| stmt.execute(params)))
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Convenience method to prepare and execute a single SQL statement with
|
|
|
|
/// named parameter(s).
|
2017-03-09 00:20:43 +08:00
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// On success, returns the number of rows that were changed or inserted or
|
|
|
|
/// deleted (via `sqlite3_changes`).
|
2017-03-09 00:20:43 +08:00
|
|
|
///
|
|
|
|
/// ## Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
|
|
|
/// # use rusqlite::{Connection, Result};
|
2018-05-24 03:23:28 +08:00
|
|
|
/// fn insert(conn: &Connection) -> Result<usize> {
|
2018-08-17 00:29:46 +08:00
|
|
|
/// conn.execute_named(
|
|
|
|
/// "INSERT INTO test (name) VALUES (:name)",
|
|
|
|
/// &[(":name", &"one")],
|
|
|
|
/// )
|
2017-03-09 00:20:43 +08:00
|
|
|
/// }
|
|
|
|
/// ```
|
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Will return `Err` if `sql` cannot be converted to a C-compatible string
|
|
|
|
/// or if the underlying SQLite call fails.
|
2018-12-08 04:57:04 +08:00
|
|
|
pub fn execute_named(&self, sql: &str, params: &[(&str, &dyn ToSql)]) -> Result<usize> {
|
2018-10-28 17:28:19 +08:00
|
|
|
self.prepare(sql).and_then(|mut stmt| {
|
|
|
|
stmt.check_no_tail()
|
|
|
|
.and_then(|_| stmt.execute_named(params))
|
|
|
|
})
|
2017-03-09 00:20:43 +08:00
|
|
|
}
|
|
|
|
|
2014-11-04 06:11:00 +08:00
|
|
|
/// Get the SQLite rowid of the most recent successful INSERT.
|
|
|
|
///
|
|
|
|
/// Uses [sqlite3_last_insert_rowid](https://www.sqlite.org/c3ref/last_insert_rowid.html) under
|
|
|
|
/// the hood.
|
2014-10-20 07:56:41 +08:00
|
|
|
pub fn last_insert_rowid(&self) -> i64 {
|
|
|
|
self.db.borrow_mut().last_insert_rowid()
|
|
|
|
}
|
|
|
|
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Convenience method to execute a query that is expected to return a
|
|
|
|
/// single row.
|
2014-11-04 06:11:00 +08:00
|
|
|
///
|
|
|
|
/// ## Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
2018-09-16 17:10:19 +08:00
|
|
|
/// # use rusqlite::{Result,Connection, NO_PARAMS};
|
2015-12-13 03:06:03 +08:00
|
|
|
/// fn preferred_locale(conn: &Connection) -> Result<String> {
|
2018-08-17 00:29:46 +08:00
|
|
|
/// conn.query_row(
|
|
|
|
/// "SELECT value FROM preferences WHERE name='locale'",
|
2018-09-16 17:10:19 +08:00
|
|
|
/// NO_PARAMS,
|
2018-08-17 00:29:46 +08:00
|
|
|
/// |row| row.get(0),
|
|
|
|
/// )
|
2014-11-04 06:11:00 +08:00
|
|
|
/// }
|
|
|
|
/// ```
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// If the query returns more than one row, all rows except the first are
|
|
|
|
/// ignored.
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
2018-12-20 04:58:33 +08:00
|
|
|
/// Returns `Err(QueryReturnedNoRows)` if no results are returned. If the
|
|
|
|
/// query truly is optional, you can call `.optional()` on the result of
|
|
|
|
/// this to get a `Result<Option<T>>`.
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Will return `Err` if `sql` cannot be converted to a C-compatible string
|
|
|
|
/// or if the underlying SQLite call fails.
|
2018-09-16 15:49:23 +08:00
|
|
|
pub fn query_row<T, P, F>(&self, sql: &str, params: P, f: F) -> Result<T>
|
2018-08-11 18:48:21 +08:00
|
|
|
where
|
2018-09-16 15:49:23 +08:00
|
|
|
P: IntoIterator,
|
|
|
|
P::Item: ToSql,
|
2019-03-10 18:12:14 +08:00
|
|
|
F: FnOnce(&Row<'_>) -> Result<T>,
|
2015-12-11 09:48:38 +08:00
|
|
|
{
|
2018-10-31 03:11:35 +08:00
|
|
|
let mut stmt = self.prepare(sql)?;
|
2019-02-02 20:22:40 +08:00
|
|
|
stmt.check_no_tail()?;
|
2016-07-14 12:27:46 +08:00
|
|
|
stmt.query_row(params, f)
|
2015-12-11 09:48:38 +08:00
|
|
|
}
|
2014-10-20 07:56:41 +08:00
|
|
|
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Convenience method to execute a query with named parameter(s) that is
|
|
|
|
/// expected to return a single row.
|
2017-03-09 00:20:43 +08:00
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// If the query returns more than one row, all rows except the first are
|
|
|
|
/// ignored.
|
2017-03-09 00:20:43 +08:00
|
|
|
///
|
2018-12-20 04:58:33 +08:00
|
|
|
/// Returns `Err(QueryReturnedNoRows)` if no results are returned. If the
|
|
|
|
/// query truly is optional, you can call `.optional()` on the result of
|
|
|
|
/// this to get a `Result<Option<T>>`.
|
2017-03-09 00:20:43 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Will return `Err` if `sql` cannot be converted to a C-compatible string
|
|
|
|
/// or if the underlying SQLite call fails.
|
2018-12-08 04:57:04 +08:00
|
|
|
pub fn query_row_named<T, F>(&self, sql: &str, params: &[(&str, &dyn ToSql)], f: F) -> Result<T>
|
2018-08-11 18:48:21 +08:00
|
|
|
where
|
2019-03-10 18:12:14 +08:00
|
|
|
F: FnOnce(&Row<'_>) -> Result<T>,
|
2017-03-09 00:20:43 +08:00
|
|
|
{
|
2018-10-31 03:11:35 +08:00
|
|
|
let mut stmt = self.prepare(sql)?;
|
2019-02-02 20:22:40 +08:00
|
|
|
stmt.check_no_tail()?;
|
2019-06-13 01:18:57 +08:00
|
|
|
stmt.query_row_named(params, f)
|
2017-03-09 00:20:43 +08:00
|
|
|
}
|
|
|
|
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Convenience method to execute a query that is expected to return a
|
|
|
|
/// single row, and execute a mapping via `f` on that returned row with
|
|
|
|
/// the possibility of failure. The `Result` type of `f` must implement
|
|
|
|
/// `std::convert::From<Error>`.
|
2015-08-28 02:01:01 +08:00
|
|
|
///
|
|
|
|
/// ## Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
2018-09-16 17:10:19 +08:00
|
|
|
/// # use rusqlite::{Result,Connection, NO_PARAMS};
|
2015-12-13 03:06:03 +08:00
|
|
|
/// fn preferred_locale(conn: &Connection) -> Result<String> {
|
2018-08-17 00:29:46 +08:00
|
|
|
/// conn.query_row_and_then(
|
|
|
|
/// "SELECT value FROM preferences WHERE name='locale'",
|
2018-09-16 17:10:19 +08:00
|
|
|
/// NO_PARAMS,
|
2019-02-22 03:48:09 +08:00
|
|
|
/// |row| row.get(0),
|
2018-08-17 00:29:46 +08:00
|
|
|
/// )
|
2015-08-28 02:01:01 +08:00
|
|
|
/// }
|
|
|
|
/// ```
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// If the query returns more than one row, all rows except the first are
|
|
|
|
/// ignored.
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Will return `Err` if `sql` cannot be converted to a C-compatible string
|
|
|
|
/// or if the underlying SQLite call fails.
|
2020-04-07 05:43:06 +08:00
|
|
|
pub fn query_row_and_then<T, E, P, F>(&self, sql: &str, params: P, f: F) -> Result<T, E>
|
2018-08-11 18:48:21 +08:00
|
|
|
where
|
2018-09-16 15:49:23 +08:00
|
|
|
P: IntoIterator,
|
|
|
|
P::Item: ToSql,
|
2020-04-07 05:43:06 +08:00
|
|
|
F: FnOnce(&Row<'_>) -> Result<T, E>,
|
2018-08-11 18:48:21 +08:00
|
|
|
E: convert::From<Error>,
|
2015-12-11 09:48:38 +08:00
|
|
|
{
|
2018-10-31 03:11:35 +08:00
|
|
|
let mut stmt = self.prepare(sql)?;
|
2019-02-02 20:22:40 +08:00
|
|
|
stmt.check_no_tail()?;
|
2018-10-31 03:11:35 +08:00
|
|
|
let mut rows = stmt.query(params)?;
|
2015-12-11 09:48:38 +08:00
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
rows.get_expected_row().map_err(E::from).and_then(|r| f(&r))
|
2016-02-03 02:12:00 +08:00
|
|
|
}
|
2015-01-11 11:17:49 +08:00
|
|
|
|
2014-11-04 06:11:00 +08:00
|
|
|
/// Prepare a SQL statement for execution.
|
|
|
|
///
|
|
|
|
/// ## Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
2015-12-13 03:06:03 +08:00
|
|
|
/// # use rusqlite::{Connection, Result};
|
|
|
|
/// fn insert_new_people(conn: &Connection) -> Result<()> {
|
2018-10-31 03:11:35 +08:00
|
|
|
/// let mut stmt = conn.prepare("INSERT INTO People (name) VALUES (?)")?;
|
|
|
|
/// stmt.execute(&["Joe Smith"])?;
|
|
|
|
/// stmt.execute(&["Bob Jones"])?;
|
2014-11-04 06:11:00 +08:00
|
|
|
/// Ok(())
|
|
|
|
/// }
|
|
|
|
/// ```
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// Will return `Err` if `sql` cannot be converted to a C-compatible string
|
|
|
|
/// or if the underlying SQLite call fails.
|
2019-02-02 18:08:04 +08:00
|
|
|
pub fn prepare(&self, sql: &str) -> Result<Statement<'_>> {
|
2014-10-20 07:56:41 +08:00
|
|
|
self.db.borrow_mut().prepare(self, sql)
|
|
|
|
}
|
|
|
|
|
2014-11-04 06:11:00 +08:00
|
|
|
/// Close the SQLite connection.
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// This is functionally equivalent to the `Drop` implementation for
|
|
|
|
/// `Connection` except that on failure, it returns an error and the
|
|
|
|
/// connection itself (presumably so closing can be attempted again).
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
|
|
|
/// Will return `Err` if the underlying SQLite call fails.
|
2020-04-07 05:43:06 +08:00
|
|
|
pub fn close(self) -> Result<(), (Connection, Error)> {
|
2016-11-05 03:47:28 +08:00
|
|
|
self.flush_prepared_statement_cache();
|
2017-02-09 08:38:50 +08:00
|
|
|
let r = self.db.borrow_mut().close();
|
|
|
|
r.map_err(move |err| (self, err))
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
2020-04-06 13:15:27 +08:00
|
|
|
/// `feature = "load_extension"` Enable loading of SQLite extensions.
|
|
|
|
/// Strongly consider using `LoadExtensionGuard` instead of this function.
|
2015-02-24 08:52:48 +08:00
|
|
|
///
|
|
|
|
/// ## Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
2015-12-13 03:06:03 +08:00
|
|
|
/// # use rusqlite::{Connection, Result};
|
2015-02-24 08:52:48 +08:00
|
|
|
/// # use std::path::{Path};
|
2015-12-13 03:06:03 +08:00
|
|
|
/// fn load_my_extension(conn: &Connection) -> Result<()> {
|
2018-10-31 03:11:35 +08:00
|
|
|
/// conn.load_extension_enable()?;
|
|
|
|
/// conn.load_extension(Path::new("my_sqlite_extension"), None)?;
|
2015-02-24 08:52:48 +08:00
|
|
|
/// conn.load_extension_disable()
|
|
|
|
/// }
|
|
|
|
/// ```
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
|
|
|
/// Will return `Err` if the underlying SQLite call fails.
|
2015-02-24 08:52:48 +08:00
|
|
|
#[cfg(feature = "load_extension")]
|
2015-12-13 03:06:03 +08:00
|
|
|
pub fn load_extension_enable(&self) -> Result<()> {
|
2015-02-24 08:52:48 +08:00
|
|
|
self.db.borrow_mut().enable_load_extension(1)
|
|
|
|
}
|
|
|
|
|
2020-04-06 13:15:27 +08:00
|
|
|
/// `feature = "load_extension"` Disable loading of SQLite extensions.
|
2015-02-24 08:52:48 +08:00
|
|
|
///
|
|
|
|
/// See `load_extension_enable` for an example.
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
|
|
|
/// Will return `Err` if the underlying SQLite call fails.
|
2015-02-24 08:52:48 +08:00
|
|
|
#[cfg(feature = "load_extension")]
|
2015-12-13 03:06:03 +08:00
|
|
|
pub fn load_extension_disable(&self) -> Result<()> {
|
2015-02-24 08:52:48 +08:00
|
|
|
self.db.borrow_mut().enable_load_extension(0)
|
|
|
|
}
|
|
|
|
|
2020-04-06 13:15:27 +08:00
|
|
|
/// `feature = "load_extension"` Load the SQLite extension at `dylib_path`.
|
|
|
|
/// `dylib_path` is passed through to `sqlite3_load_extension`, which may
|
|
|
|
/// attempt OS-specific modifications if the file cannot be loaded directly.
|
2015-02-24 08:52:48 +08:00
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// If `entry_point` is `None`, SQLite will attempt to find the entry
|
|
|
|
/// point. If it is not `None`, the entry point will be passed through
|
|
|
|
/// to `sqlite3_load_extension`.
|
2015-02-24 09:16:49 +08:00
|
|
|
///
|
|
|
|
/// ## Example
|
|
|
|
///
|
|
|
|
/// ```rust,no_run
|
2015-12-13 03:20:11 +08:00
|
|
|
/// # use rusqlite::{Connection, Result, LoadExtensionGuard};
|
2015-02-24 09:16:49 +08:00
|
|
|
/// # use std::path::{Path};
|
2015-12-13 03:06:03 +08:00
|
|
|
/// fn load_my_extension(conn: &Connection) -> Result<()> {
|
2018-10-31 03:11:35 +08:00
|
|
|
/// let _guard = LoadExtensionGuard::new(conn)?;
|
2015-02-24 09:16:49 +08:00
|
|
|
///
|
2015-09-08 16:06:41 +08:00
|
|
|
/// conn.load_extension("my_sqlite_extension", None)
|
2015-02-24 09:16:49 +08:00
|
|
|
/// }
|
2015-12-09 10:15:23 +08:00
|
|
|
/// ```
|
2015-12-02 01:05:29 +08:00
|
|
|
///
|
|
|
|
/// # Failure
|
|
|
|
///
|
|
|
|
/// Will return `Err` if the underlying SQLite call fails.
|
2015-02-24 08:52:48 +08:00
|
|
|
#[cfg(feature = "load_extension")]
|
2018-08-11 18:48:21 +08:00
|
|
|
pub fn load_extension<P: AsRef<Path>>(
|
|
|
|
&self,
|
|
|
|
dylib_path: P,
|
|
|
|
entry_point: Option<&str>,
|
|
|
|
) -> Result<()> {
|
2017-04-08 01:43:24 +08:00
|
|
|
self.db
|
|
|
|
.borrow_mut()
|
|
|
|
.load_extension(dylib_path.as_ref(), entry_point)
|
2016-02-03 02:12:00 +08:00
|
|
|
}
|
2014-10-20 07:56:41 +08:00
|
|
|
|
2016-02-02 04:21:03 +08:00
|
|
|
/// Get access to the underlying SQLite database connection handle.
|
|
|
|
///
|
|
|
|
/// # Warning
|
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// You should not need to use this function. If you do need to, please
|
2020-04-16 12:07:23 +08:00
|
|
|
/// [open an issue on the rusqlite repository](https://github.com/rusqlite/rusqlite/issues) and describe
|
2019-11-03 18:19:07 +08:00
|
|
|
/// your use case.
|
|
|
|
///
|
|
|
|
/// # Safety
|
|
|
|
///
|
|
|
|
/// This function is unsafe because it gives you raw access
|
2018-08-17 00:29:46 +08:00
|
|
|
/// to the SQLite connection, and what you do with it could impact the
|
|
|
|
/// safety of this `Connection`.
|
2017-02-08 10:09:20 +08:00
|
|
|
pub unsafe fn handle(&self) -> *mut ffi::sqlite3 {
|
2016-02-02 04:21:03 +08:00
|
|
|
self.db.borrow().db()
|
|
|
|
}
|
2014-10-20 07:56:41 +08:00
|
|
|
|
2019-01-06 19:58:46 +08:00
|
|
|
/// Create a `Connection` from a raw handle.
|
|
|
|
///
|
|
|
|
/// The underlying SQLite database connection handle will not be closed when
|
|
|
|
/// the returned connection is dropped/closed.
|
2020-02-22 18:50:00 +08:00
|
|
|
///
|
|
|
|
/// # Safety
|
|
|
|
///
|
|
|
|
/// This function is unsafe because improper use may impact the Connection.
|
2019-01-06 19:58:46 +08:00
|
|
|
pub unsafe fn from_handle(db: *mut ffi::sqlite3) -> Result<Connection> {
|
2019-01-06 21:01:17 +08:00
|
|
|
let db_path = db_filename(db);
|
2019-01-06 19:58:46 +08:00
|
|
|
let db = InnerConnection::new(db, false);
|
|
|
|
Ok(Connection {
|
|
|
|
db: RefCell::new(db),
|
|
|
|
cache: StatementCache::with_capacity(STATEMENT_CACHE_DEFAULT_CAPACITY),
|
|
|
|
path: db_path,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2018-10-28 15:51:02 +08:00
|
|
|
/// Get access to a handle that can be used to interrupt long running
|
|
|
|
/// queries from another thread.
|
2018-09-27 04:36:01 +08:00
|
|
|
pub fn get_interrupt_handle(&self) -> InterruptHandle {
|
|
|
|
self.db.borrow().get_interrupt_handle()
|
|
|
|
}
|
|
|
|
|
2015-12-13 03:06:03 +08:00
|
|
|
fn decode_result(&self, code: c_int) -> Result<()> {
|
2014-10-20 07:56:41 +08:00
|
|
|
self.db.borrow_mut().decode_result(code)
|
|
|
|
}
|
|
|
|
|
2019-01-26 18:01:51 +08:00
|
|
|
/// Return the number of rows modified, inserted or deleted by the most
|
|
|
|
/// recently completed INSERT, UPDATE or DELETE statement on the database
|
|
|
|
/// connection.
|
2019-02-17 00:24:56 +08:00
|
|
|
fn changes(&self) -> usize {
|
2014-10-20 07:56:41 +08:00
|
|
|
self.db.borrow_mut().changes()
|
|
|
|
}
|
2018-05-13 00:35:08 +08:00
|
|
|
|
|
|
|
/// Test for auto-commit mode.
|
|
|
|
/// Autocommit mode is on by default.
|
|
|
|
pub fn is_autocommit(&self) -> bool {
|
|
|
|
self.db.borrow().is_autocommit()
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Determine if all associated prepared statements have been reset.
|
2020-01-15 00:11:36 +08:00
|
|
|
#[cfg(feature = "modern_sqlite")] // 3.8.6
|
2018-05-13 00:35:08 +08:00
|
|
|
pub fn is_busy(&self) -> bool {
|
|
|
|
self.db.borrow().is_busy()
|
|
|
|
}
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
2015-12-13 02:50:12 +08:00
|
|
|
impl fmt::Debug for Connection {
|
2018-12-08 04:57:04 +08:00
|
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
2015-12-13 02:50:12 +08:00
|
|
|
f.debug_struct("Connection")
|
2016-05-17 01:52:17 +08:00
|
|
|
.field("path", &self.path)
|
|
|
|
.finish()
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-08-10 02:01:44 +08:00
|
|
|
bitflags::bitflags! {
|
2020-05-17 17:21:10 +08:00
|
|
|
/// Flags for opening SQLite database connections.
|
|
|
|
/// See [sqlite3_open_v2](http://www.sqlite.org/c3ref/open.html) for details.
|
2014-11-04 06:11:00 +08:00
|
|
|
#[repr(C)]
|
2017-05-29 22:52:19 +08:00
|
|
|
pub struct OpenFlags: ::std::os::raw::c_int {
|
2020-05-17 17:21:10 +08:00
|
|
|
/// The database is opened in read-only mode.
|
|
|
|
/// If the database does not already exist, an error is returned.
|
2017-05-29 22:52:19 +08:00
|
|
|
const SQLITE_OPEN_READ_ONLY = ffi::SQLITE_OPEN_READONLY;
|
2020-05-17 17:21:10 +08:00
|
|
|
/// The database is opened for reading and writing if possible,
|
|
|
|
/// or reading only if the file is write protected by the operating system.
|
|
|
|
/// In either case the database must already exist, otherwise an error is returned.
|
2017-05-29 22:52:19 +08:00
|
|
|
const SQLITE_OPEN_READ_WRITE = ffi::SQLITE_OPEN_READWRITE;
|
2020-05-17 17:21:10 +08:00
|
|
|
/// The database is created if it does not already exist
|
2017-05-29 22:52:19 +08:00
|
|
|
const SQLITE_OPEN_CREATE = ffi::SQLITE_OPEN_CREATE;
|
2020-05-17 17:21:10 +08:00
|
|
|
/// The filename can be interpreted as a URI if this flag is set.
|
2017-08-19 03:26:44 +08:00
|
|
|
const SQLITE_OPEN_URI = 0x0000_0040;
|
2020-05-17 17:21:10 +08:00
|
|
|
/// The database will be opened as an in-memory database.
|
2017-08-19 03:26:44 +08:00
|
|
|
const SQLITE_OPEN_MEMORY = 0x0000_0080;
|
2020-05-17 17:21:10 +08:00
|
|
|
/// The new database connection will use the "multi-thread" threading mode.
|
2017-05-29 22:52:19 +08:00
|
|
|
const SQLITE_OPEN_NO_MUTEX = ffi::SQLITE_OPEN_NOMUTEX;
|
2020-05-17 17:21:10 +08:00
|
|
|
/// The new database connection will use the "serialized" threading mode.
|
2017-05-29 22:52:19 +08:00
|
|
|
const SQLITE_OPEN_FULL_MUTEX = ffi::SQLITE_OPEN_FULLMUTEX;
|
2020-05-17 17:21:10 +08:00
|
|
|
/// The database is opened shared cache enabled.
|
2017-08-19 03:26:44 +08:00
|
|
|
const SQLITE_OPEN_SHARED_CACHE = 0x0002_0000;
|
2020-05-17 17:21:10 +08:00
|
|
|
/// The database is opened shared cache disabled.
|
2017-08-19 03:26:44 +08:00
|
|
|
const SQLITE_OPEN_PRIVATE_CACHE = 0x0004_0000;
|
2020-05-17 17:21:10 +08:00
|
|
|
/// The database filename is not allowed to be a symbolic link.
|
2020-02-09 18:47:01 +08:00
|
|
|
const SQLITE_OPEN_NOFOLLOW = 0x0100_0000;
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-12-13 03:13:29 +08:00
|
|
|
impl Default for OpenFlags {
|
|
|
|
fn default() -> OpenFlags {
|
2018-08-11 18:48:21 +08:00
|
|
|
OpenFlags::SQLITE_OPEN_READ_WRITE
|
|
|
|
| OpenFlags::SQLITE_OPEN_CREATE
|
|
|
|
| OpenFlags::SQLITE_OPEN_NO_MUTEX
|
|
|
|
| OpenFlags::SQLITE_OPEN_URI
|
2015-07-07 02:24:27 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-17 00:29:46 +08:00
|
|
|
/// rusqlite's check for a safe SQLite threading mode requires SQLite 3.7.0 or
|
|
|
|
/// later. If you are running against a SQLite older than that, rusqlite
|
|
|
|
/// attempts to ensure safety by performing configuration and initialization of
|
|
|
|
/// SQLite itself the first time you
|
|
|
|
/// attempt to open a connection. By default, rusqlite panics if that
|
|
|
|
/// initialization fails, since that could mean SQLite has been initialized in
|
|
|
|
/// single-thread mode.
|
2017-02-10 09:12:24 +08:00
|
|
|
///
|
2018-08-17 00:29:46 +08:00
|
|
|
/// If you are encountering that panic _and_ can ensure that SQLite has been
|
|
|
|
/// initialized in either multi-thread or serialized mode, call this function
|
|
|
|
/// prior to attempting to open a connection and rusqlite's initialization
|
2019-11-03 18:19:07 +08:00
|
|
|
/// process will by skipped.
|
|
|
|
///
|
|
|
|
/// # Safety
|
|
|
|
///
|
|
|
|
/// This function is unsafe because if you call it and SQLite has actually been
|
2018-08-17 00:29:46 +08:00
|
|
|
/// configured to run in single-thread mode,
|
|
|
|
/// you may enounter memory errors or data corruption or any number of terrible
|
|
|
|
/// things that should not be possible when you're using Rust.
|
2017-02-10 09:12:24 +08:00
|
|
|
pub unsafe fn bypass_sqlite_initialization() {
|
|
|
|
BYPASS_SQLITE_INIT.store(true, Ordering::Relaxed);
|
|
|
|
}
|
|
|
|
|
2018-08-17 00:29:46 +08:00
|
|
|
/// rusqlite performs a one-time check that the runtime SQLite version is at
|
|
|
|
/// least as new as the version of SQLite found when rusqlite was built.
|
|
|
|
/// Bypassing this check may be dangerous; e.g., if you use features of SQLite
|
2019-11-03 18:19:07 +08:00
|
|
|
/// that are not present in the runtime version.
|
|
|
|
///
|
|
|
|
/// # Safety
|
|
|
|
///
|
|
|
|
/// If you are sure the runtime version is compatible with the
|
2018-08-17 00:29:46 +08:00
|
|
|
/// build-time version for your usage, you can bypass the version check by
|
2019-11-03 18:19:07 +08:00
|
|
|
/// calling this function before your first connection attempt.
|
2017-02-10 09:56:35 +08:00
|
|
|
pub unsafe fn bypass_sqlite_version_check() {
|
2018-10-16 09:30:35 +08:00
|
|
|
#[cfg(not(feature = "bundled"))]
|
2019-02-02 19:46:52 +08:00
|
|
|
inner_connection::BYPASS_VERSION_CHECK.store(true, Ordering::Relaxed);
|
2017-02-10 09:56:35 +08:00
|
|
|
}
|
|
|
|
|
2018-09-27 04:36:01 +08:00
|
|
|
/// Allows interrupting a long-running computation.
|
|
|
|
pub struct InterruptHandle {
|
|
|
|
db_lock: Arc<Mutex<*mut ffi::sqlite3>>,
|
2017-02-10 09:12:24 +08:00
|
|
|
}
|
|
|
|
|
2018-09-27 04:36:01 +08:00
|
|
|
unsafe impl Send for InterruptHandle {}
|
|
|
|
unsafe impl Sync for InterruptHandle {}
|
2017-02-10 09:12:24 +08:00
|
|
|
|
2018-09-27 04:36:01 +08:00
|
|
|
impl InterruptHandle {
|
|
|
|
/// Interrupt the query currently executing on another thread. This will
|
|
|
|
/// cause that query to fail with a `SQLITE3_INTERRUPT` error.
|
|
|
|
pub fn interrupt(&self) {
|
|
|
|
let db_handle = self.db_lock.lock().unwrap();
|
|
|
|
if !db_handle.is_null() {
|
|
|
|
unsafe { ffi::sqlite3_interrupt(*db_handle) }
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-01-15 00:11:36 +08:00
|
|
|
#[cfg(feature = "modern_sqlite")] // 3.7.10
|
2019-01-06 21:01:17 +08:00
|
|
|
unsafe fn db_filename(db: *mut ffi::sqlite3) -> Option<PathBuf> {
|
|
|
|
let db_name = DatabaseName::Main.to_cstring().unwrap();
|
|
|
|
let db_filename = ffi::sqlite3_db_filename(db, db_name.as_ptr());
|
|
|
|
if db_filename.is_null() {
|
|
|
|
None
|
|
|
|
} else {
|
2020-04-09 01:38:32 +08:00
|
|
|
CStr::from_ptr(db_filename).to_str().ok().map(PathBuf::from)
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
}
|
2020-01-15 00:11:36 +08:00
|
|
|
#[cfg(not(feature = "modern_sqlite"))]
|
2019-01-06 21:01:17 +08:00
|
|
|
unsafe fn db_filename(_: *mut ffi::sqlite3) -> Option<PathBuf> {
|
|
|
|
None
|
|
|
|
}
|
2016-01-02 19:13:37 +08:00
|
|
|
|
2020-03-08 22:36:56 +08:00
|
|
|
#[cfg(doctest)]
|
|
|
|
doc_comment::doctest!("../README.md");
|
|
|
|
|
2014-10-20 07:56:41 +08:00
|
|
|
#[cfg(test)]
|
|
|
|
mod test {
|
2019-08-17 14:18:37 +08:00
|
|
|
use super::*;
|
2018-10-31 03:11:35 +08:00
|
|
|
use crate::ffi;
|
2019-03-20 03:45:04 +08:00
|
|
|
use fallible_iterator::FallibleIterator;
|
2019-08-17 14:18:37 +08:00
|
|
|
use std::error::Error as StdError;
|
|
|
|
use std::fmt;
|
2014-10-20 07:56:41 +08:00
|
|
|
|
2015-03-29 18:25:46 +08:00
|
|
|
// this function is never called, but is still type checked; in
|
|
|
|
// particular, calls with specific instantiations will require
|
|
|
|
// that those types are `Send`.
|
|
|
|
#[allow(dead_code, unconditional_recursion)]
|
|
|
|
fn ensure_send<T: Send>() {
|
2015-12-13 02:50:12 +08:00
|
|
|
ensure_send::<Connection>();
|
2018-09-27 04:36:01 +08:00
|
|
|
ensure_send::<InterruptHandle>();
|
|
|
|
}
|
|
|
|
|
|
|
|
#[allow(dead_code, unconditional_recursion)]
|
|
|
|
fn ensure_sync<T: Sync>() {
|
|
|
|
ensure_sync::<InterruptHandle>();
|
2015-03-29 18:25:46 +08:00
|
|
|
}
|
|
|
|
|
2015-12-13 02:50:12 +08:00
|
|
|
pub fn checked_memory_handle() -> Connection {
|
|
|
|
Connection::open_in_memory().unwrap()
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
2015-05-05 03:22:11 +08:00
|
|
|
#[test]
|
2018-07-28 19:39:19 +08:00
|
|
|
fn test_concurrent_transactions_busy_commit() {
|
2018-07-28 23:02:17 +08:00
|
|
|
use std::time::Duration;
|
2019-11-22 19:08:12 +08:00
|
|
|
let tmp = tempfile::tempdir().unwrap();
|
2018-07-28 19:39:19 +08:00
|
|
|
let path = tmp.path().join("transactions.db3");
|
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
Connection::open(&path)
|
|
|
|
.expect("create temp db")
|
|
|
|
.execute_batch(
|
|
|
|
"
|
2018-07-28 19:39:19 +08:00
|
|
|
BEGIN; CREATE TABLE foo(x INTEGER);
|
2018-08-11 18:48:21 +08:00
|
|
|
INSERT INTO foo VALUES(42); END;",
|
2018-10-28 15:51:02 +08:00
|
|
|
)
|
|
|
|
.expect("create temp db");
|
2018-07-28 19:39:19 +08:00
|
|
|
|
2019-08-27 02:21:23 +08:00
|
|
|
let mut db1 =
|
|
|
|
Connection::open_with_flags(&path, OpenFlags::SQLITE_OPEN_READ_WRITE).unwrap();
|
2019-08-17 14:16:47 +08:00
|
|
|
let mut db2 = Connection::open_with_flags(&path, OpenFlags::SQLITE_OPEN_READ_ONLY).unwrap();
|
2018-07-28 19:39:19 +08:00
|
|
|
|
2018-07-28 23:02:17 +08:00
|
|
|
db1.busy_timeout(Duration::from_millis(0)).unwrap();
|
|
|
|
db2.busy_timeout(Duration::from_millis(0)).unwrap();
|
2018-07-28 19:39:19 +08:00
|
|
|
|
|
|
|
{
|
|
|
|
let tx1 = db1.transaction().unwrap();
|
|
|
|
let tx2 = db2.transaction().unwrap();
|
|
|
|
|
|
|
|
// SELECT first makes sqlite lock with a shared lock
|
2019-02-22 03:48:09 +08:00
|
|
|
tx1.query_row("SELECT x FROM foo LIMIT 1", NO_PARAMS, |_| Ok(()))
|
2018-08-11 18:48:21 +08:00
|
|
|
.unwrap();
|
2019-02-22 03:48:09 +08:00
|
|
|
tx2.query_row("SELECT x FROM foo LIMIT 1", NO_PARAMS, |_| Ok(()))
|
2018-08-11 18:48:21 +08:00
|
|
|
.unwrap();
|
2018-07-28 19:39:19 +08:00
|
|
|
|
2018-09-16 15:49:23 +08:00
|
|
|
tx1.execute("INSERT INTO foo VALUES(?1)", &[1]).unwrap();
|
|
|
|
let _ = tx2.execute("INSERT INTO foo VALUES(?1)", &[2]);
|
2018-07-28 19:39:19 +08:00
|
|
|
|
|
|
|
let _ = tx1.commit();
|
|
|
|
let _ = tx2.commit();
|
|
|
|
}
|
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
let _ = db1
|
|
|
|
.transaction()
|
|
|
|
.expect("commit should have closed transaction");
|
|
|
|
let _ = db2
|
|
|
|
.transaction()
|
|
|
|
.expect("commit should have closed transaction");
|
2018-07-28 19:39:19 +08:00
|
|
|
}
|
|
|
|
|
2015-05-05 03:22:11 +08:00
|
|
|
#[test]
|
|
|
|
fn test_persistence() {
|
2019-11-22 19:08:12 +08:00
|
|
|
let temp_dir = tempfile::tempdir().unwrap();
|
2015-05-05 03:22:11 +08:00
|
|
|
let path = temp_dir.path().join("test.db3");
|
|
|
|
|
|
|
|
{
|
2015-12-13 02:50:12 +08:00
|
|
|
let db = Connection::open(&path).unwrap();
|
2015-05-05 03:22:11 +08:00
|
|
|
let sql = "BEGIN;
|
|
|
|
CREATE TABLE foo(x INTEGER);
|
|
|
|
INSERT INTO foo VALUES(42);
|
|
|
|
END;";
|
2017-06-01 17:31:41 +08:00
|
|
|
db.execute_batch(sql).unwrap();
|
2015-05-05 03:22:11 +08:00
|
|
|
}
|
2015-05-05 08:02:33 +08:00
|
|
|
|
2015-05-05 03:22:11 +08:00
|
|
|
let path_string = path.to_str().unwrap();
|
2015-12-13 02:50:12 +08:00
|
|
|
let db = Connection::open(&path_string).unwrap();
|
2018-09-16 17:10:19 +08:00
|
|
|
let the_answer: Result<i64> = db.query_row("SELECT x FROM foo", NO_PARAMS, |r| r.get(0));
|
2015-05-05 03:22:11 +08:00
|
|
|
|
|
|
|
assert_eq!(42i64, the_answer.unwrap());
|
|
|
|
}
|
|
|
|
|
2014-10-20 07:56:41 +08:00
|
|
|
#[test]
|
|
|
|
fn test_open() {
|
2015-12-13 02:50:12 +08:00
|
|
|
assert!(Connection::open_in_memory().is_ok());
|
2014-10-20 07:56:41 +08:00
|
|
|
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
assert!(db.close().is_ok());
|
|
|
|
}
|
|
|
|
|
2018-03-11 02:17:22 +08:00
|
|
|
#[test]
|
|
|
|
fn test_open_failure() {
|
|
|
|
let filename = "no_such_file.db";
|
|
|
|
let result = Connection::open_with_flags(filename, OpenFlags::SQLITE_OPEN_READ_ONLY);
|
|
|
|
assert!(!result.is_ok());
|
|
|
|
let err = result.err().unwrap();
|
|
|
|
if let Error::SqliteFailure(e, Some(msg)) = err {
|
|
|
|
assert_eq!(ErrorCode::CannotOpen, e.code);
|
|
|
|
assert_eq!(ffi::SQLITE_CANTOPEN, e.extended_code);
|
2019-10-30 02:24:18 +08:00
|
|
|
assert!(
|
|
|
|
msg.contains(filename),
|
|
|
|
"error message '{}' does not contain '{}'",
|
|
|
|
msg,
|
|
|
|
filename
|
|
|
|
);
|
2018-03-11 02:17:22 +08:00
|
|
|
} else {
|
|
|
|
panic!("SqliteFailure expected");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-04-09 16:35:38 +08:00
|
|
|
#[cfg(unix)]
|
|
|
|
#[test]
|
|
|
|
fn test_invalid_unicode_file_names() {
|
|
|
|
use std::ffi::OsStr;
|
|
|
|
use std::fs::File;
|
|
|
|
use std::os::unix::ffi::OsStrExt;
|
|
|
|
let temp_dir = tempfile::tempdir().unwrap();
|
|
|
|
|
|
|
|
let path = temp_dir.path();
|
|
|
|
if File::create(path.join(OsStr::from_bytes(&[0xFE]))).is_err() {
|
|
|
|
// Skip test, filesystem doesn't support invalid Unicode
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
let db_path = path.join(OsStr::from_bytes(&[0xFF]));
|
|
|
|
{
|
|
|
|
let db = Connection::open(&db_path).unwrap();
|
|
|
|
let sql = "BEGIN;
|
|
|
|
CREATE TABLE foo(x INTEGER);
|
|
|
|
INSERT INTO foo VALUES(42);
|
|
|
|
END;";
|
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
}
|
|
|
|
|
|
|
|
let db = Connection::open(&db_path).unwrap();
|
|
|
|
let the_answer: Result<i64> = db.query_row("SELECT x FROM foo", NO_PARAMS, |r| r.get(0));
|
|
|
|
|
|
|
|
assert_eq!(42i64, the_answer.unwrap());
|
|
|
|
}
|
|
|
|
|
2017-01-04 09:00:29 +08:00
|
|
|
#[test]
|
|
|
|
fn test_close_retry() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
|
2018-08-17 00:29:46 +08:00
|
|
|
// force the DB to be busy by preparing a statement; this must be done at the
|
|
|
|
// FFI level to allow us to call .close() without dropping the prepared
|
|
|
|
// statement first.
|
2017-01-04 09:00:29 +08:00
|
|
|
let raw_stmt = {
|
|
|
|
use super::str_to_cstring;
|
2017-02-17 00:17:24 +08:00
|
|
|
use std::os::raw::c_int;
|
2018-08-11 18:48:21 +08:00
|
|
|
use std::ptr;
|
2017-01-04 09:00:29 +08:00
|
|
|
|
|
|
|
let raw_db = db.db.borrow_mut().db;
|
|
|
|
let sql = "SELECT 1";
|
2020-04-07 10:53:55 +08:00
|
|
|
let mut raw_stmt: *mut ffi::sqlite3_stmt = ptr::null_mut();
|
2019-11-03 18:19:07 +08:00
|
|
|
let cstring = str_to_cstring(sql).unwrap();
|
2017-01-04 09:00:29 +08:00
|
|
|
let rc = unsafe {
|
2018-08-11 18:48:21 +08:00
|
|
|
ffi::sqlite3_prepare_v2(
|
|
|
|
raw_db,
|
2019-11-03 18:19:07 +08:00
|
|
|
cstring.as_ptr(),
|
2018-08-11 18:48:21 +08:00
|
|
|
(sql.len() + 1) as c_int,
|
2020-04-07 10:53:55 +08:00
|
|
|
&mut raw_stmt,
|
2018-08-11 18:48:21 +08:00
|
|
|
ptr::null_mut(),
|
|
|
|
)
|
2017-01-04 09:00:29 +08:00
|
|
|
};
|
|
|
|
assert_eq!(rc, ffi::SQLITE_OK);
|
|
|
|
raw_stmt
|
|
|
|
};
|
|
|
|
|
2018-08-17 00:29:46 +08:00
|
|
|
// now that we have an open statement, trying (and retrying) to close should
|
|
|
|
// fail.
|
2017-01-07 03:32:27 +08:00
|
|
|
let (db, _) = db.close().unwrap_err();
|
|
|
|
let (db, _) = db.close().unwrap_err();
|
|
|
|
let (db, _) = db.close().unwrap_err();
|
2017-01-04 09:00:29 +08:00
|
|
|
|
2017-01-07 03:32:27 +08:00
|
|
|
// finalize the open statement so a final close will succeed
|
2017-01-04 09:00:29 +08:00
|
|
|
assert_eq!(ffi::SQLITE_OK, unsafe { ffi::sqlite3_finalize(raw_stmt) });
|
|
|
|
|
|
|
|
db.close().unwrap();
|
|
|
|
}
|
|
|
|
|
2014-10-20 07:56:41 +08:00
|
|
|
#[test]
|
|
|
|
fn test_open_with_flags() {
|
2018-08-11 18:48:21 +08:00
|
|
|
for bad_flags in &[
|
|
|
|
OpenFlags::empty(),
|
|
|
|
OpenFlags::SQLITE_OPEN_READ_ONLY | OpenFlags::SQLITE_OPEN_READ_WRITE,
|
|
|
|
OpenFlags::SQLITE_OPEN_READ_ONLY | OpenFlags::SQLITE_OPEN_CREATE,
|
|
|
|
] {
|
2016-02-03 02:12:00 +08:00
|
|
|
assert!(Connection::open_in_memory_with_flags(*bad_flags).is_err());
|
|
|
|
}
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_execute_batch() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let sql = "BEGIN;
|
|
|
|
CREATE TABLE foo(x INTEGER);
|
|
|
|
INSERT INTO foo VALUES(1);
|
|
|
|
INSERT INTO foo VALUES(2);
|
|
|
|
INSERT INTO foo VALUES(3);
|
|
|
|
INSERT INTO foo VALUES(4);
|
|
|
|
END;";
|
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
db.execute_batch("UPDATE foo SET x = 3 WHERE x < 3")
|
|
|
|
.unwrap();
|
2014-10-20 07:56:41 +08:00
|
|
|
|
|
|
|
assert!(db.execute_batch("INVALID SQL").is_err());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_execute() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
db.execute_batch("CREATE TABLE foo(x INTEGER)").unwrap();
|
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
assert_eq!(
|
|
|
|
1,
|
2018-09-16 15:49:23 +08:00
|
|
|
db.execute("INSERT INTO foo(x) VALUES (?)", &[1i32])
|
2018-08-11 18:48:21 +08:00
|
|
|
.unwrap()
|
|
|
|
);
|
|
|
|
assert_eq!(
|
|
|
|
1,
|
2018-09-16 15:49:23 +08:00
|
|
|
db.execute("INSERT INTO foo(x) VALUES (?)", &[2i32])
|
2018-08-11 18:48:21 +08:00
|
|
|
.unwrap()
|
|
|
|
);
|
2014-10-20 07:56:41 +08:00
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
assert_eq!(
|
|
|
|
3i32,
|
2018-09-16 17:10:19 +08:00
|
|
|
db.query_row::<i32, _, _>("SELECT SUM(x) FROM foo", NO_PARAMS, |r| r.get(0))
|
2018-08-11 18:48:21 +08:00
|
|
|
.unwrap()
|
|
|
|
);
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
2015-08-01 14:09:59 +08:00
|
|
|
#[test]
|
2020-03-05 03:42:32 +08:00
|
|
|
#[cfg(feature = "extra_check")]
|
2015-08-01 14:09:59 +08:00
|
|
|
fn test_execute_select() {
|
|
|
|
let db = checked_memory_handle();
|
2018-09-16 15:49:23 +08:00
|
|
|
let err = db.execute("SELECT 1 WHERE 1 < ?", &[1i32]).unwrap_err();
|
2018-12-20 04:58:33 +08:00
|
|
|
if err != Error::ExecuteReturnedResults {
|
|
|
|
panic!("Unexpected error: {}", err);
|
2015-12-13 13:54:08 +08:00
|
|
|
}
|
2015-08-01 14:09:59 +08:00
|
|
|
}
|
|
|
|
|
2018-10-28 17:28:19 +08:00
|
|
|
#[test]
|
2019-08-31 15:39:09 +08:00
|
|
|
#[cfg(feature = "extra_check")]
|
2018-10-28 17:28:19 +08:00
|
|
|
fn test_execute_multiple() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let err = db
|
|
|
|
.execute(
|
|
|
|
"CREATE TABLE foo(x INTEGER); CREATE TABLE foo(x INTEGER)",
|
|
|
|
NO_PARAMS,
|
2019-08-27 02:43:39 +08:00
|
|
|
)
|
|
|
|
.unwrap_err();
|
2015-12-13 13:54:08 +08:00
|
|
|
match err {
|
2018-10-28 17:28:19 +08:00
|
|
|
Error::MultipleStatement => (),
|
2015-12-13 13:54:08 +08:00
|
|
|
_ => panic!("Unexpected error: {}", err),
|
|
|
|
}
|
2015-08-01 14:09:59 +08:00
|
|
|
}
|
|
|
|
|
2015-07-25 10:11:59 +08:00
|
|
|
#[test]
|
|
|
|
fn test_prepare_column_names() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
db.execute_batch("CREATE TABLE foo(x INTEGER);").unwrap();
|
|
|
|
|
|
|
|
let stmt = db.prepare("SELECT * FROM foo").unwrap();
|
2016-01-02 17:28:00 +08:00
|
|
|
assert_eq!(stmt.column_count(), 1);
|
2015-07-25 10:11:59 +08:00
|
|
|
assert_eq!(stmt.column_names(), vec!["x"]);
|
|
|
|
|
|
|
|
let stmt = db.prepare("SELECT x AS a, x AS b FROM foo").unwrap();
|
2016-01-02 17:28:00 +08:00
|
|
|
assert_eq!(stmt.column_count(), 2);
|
2015-07-25 10:11:59 +08:00
|
|
|
assert_eq!(stmt.column_names(), vec!["a", "b"]);
|
|
|
|
}
|
|
|
|
|
2014-10-20 07:56:41 +08:00
|
|
|
#[test]
|
|
|
|
fn test_prepare_execute() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
db.execute_batch("CREATE TABLE foo(x INTEGER);").unwrap();
|
|
|
|
|
|
|
|
let mut insert_stmt = db.prepare("INSERT INTO foo(x) VALUES(?)").unwrap();
|
2018-09-16 15:49:23 +08:00
|
|
|
assert_eq!(insert_stmt.execute(&[1i32]).unwrap(), 1);
|
|
|
|
assert_eq!(insert_stmt.execute(&[2i32]).unwrap(), 1);
|
|
|
|
assert_eq!(insert_stmt.execute(&[3i32]).unwrap(), 1);
|
2014-10-20 07:56:41 +08:00
|
|
|
|
2018-09-16 15:49:23 +08:00
|
|
|
assert_eq!(insert_stmt.execute(&["hello".to_string()]).unwrap(), 1);
|
|
|
|
assert_eq!(insert_stmt.execute(&["goodbye".to_string()]).unwrap(), 1);
|
|
|
|
assert_eq!(insert_stmt.execute(&[types::Null]).unwrap(), 1);
|
2014-10-20 07:56:41 +08:00
|
|
|
|
|
|
|
let mut update_stmt = db.prepare("UPDATE foo SET x=? WHERE x<?").unwrap();
|
2018-09-16 15:49:23 +08:00
|
|
|
assert_eq!(update_stmt.execute(&[3i32, 3i32]).unwrap(), 2);
|
|
|
|
assert_eq!(update_stmt.execute(&[3i32, 3i32]).unwrap(), 0);
|
|
|
|
assert_eq!(update_stmt.execute(&[8i32, 8i32]).unwrap(), 3);
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_prepare_query() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
db.execute_batch("CREATE TABLE foo(x INTEGER);").unwrap();
|
|
|
|
|
|
|
|
let mut insert_stmt = db.prepare("INSERT INTO foo(x) VALUES(?)").unwrap();
|
2018-09-16 15:49:23 +08:00
|
|
|
assert_eq!(insert_stmt.execute(&[1i32]).unwrap(), 1);
|
|
|
|
assert_eq!(insert_stmt.execute(&[2i32]).unwrap(), 1);
|
|
|
|
assert_eq!(insert_stmt.execute(&[3i32]).unwrap(), 1);
|
2014-10-20 07:56:41 +08:00
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
let mut query = db
|
|
|
|
.prepare("SELECT x FROM foo WHERE x < ? ORDER BY x DESC")
|
2017-04-08 01:43:24 +08:00
|
|
|
.unwrap();
|
2014-10-20 07:56:41 +08:00
|
|
|
{
|
2018-09-16 15:49:23 +08:00
|
|
|
let mut rows = query.query(&[4i32]).unwrap();
|
2016-05-19 00:33:58 +08:00
|
|
|
let mut v = Vec::<i32>::new();
|
|
|
|
|
2019-02-03 21:01:42 +08:00
|
|
|
while let Some(row) = rows.next().unwrap() {
|
2019-03-10 18:12:14 +08:00
|
|
|
v.push(row.get(0).unwrap());
|
2016-05-19 00:33:58 +08:00
|
|
|
}
|
2014-10-20 07:56:41 +08:00
|
|
|
|
2015-03-17 12:55:28 +08:00
|
|
|
assert_eq!(v, [3i32, 2, 1]);
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
{
|
2018-09-16 15:49:23 +08:00
|
|
|
let mut rows = query.query(&[3i32]).unwrap();
|
2016-05-19 00:33:58 +08:00
|
|
|
let mut v = Vec::<i32>::new();
|
|
|
|
|
2019-02-03 21:01:42 +08:00
|
|
|
while let Some(row) = rows.next().unwrap() {
|
2019-03-10 18:12:14 +08:00
|
|
|
v.push(row.get(0).unwrap());
|
2016-05-19 00:33:58 +08:00
|
|
|
}
|
|
|
|
|
2015-03-17 12:55:28 +08:00
|
|
|
assert_eq!(v, [2i32, 1]);
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-05-05 21:29:46 +08:00
|
|
|
#[test]
|
|
|
|
fn test_query_map() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let sql = "BEGIN;
|
|
|
|
CREATE TABLE foo(x INTEGER, y TEXT);
|
|
|
|
INSERT INTO foo VALUES(4, \"hello\");
|
|
|
|
INSERT INTO foo VALUES(3, \", \");
|
|
|
|
INSERT INTO foo VALUES(2, \"world\");
|
|
|
|
INSERT INTO foo VALUES(1, \"!\");
|
|
|
|
END;";
|
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
|
|
|
|
let mut query = db.prepare("SELECT x, y FROM foo ORDER BY x DESC").unwrap();
|
2018-09-16 15:49:23 +08:00
|
|
|
let results: Result<Vec<String>> = query
|
2019-03-20 03:45:04 +08:00
|
|
|
.query(NO_PARAMS)
|
2015-05-06 11:29:34 +08:00
|
|
|
.unwrap()
|
2019-03-20 03:45:04 +08:00
|
|
|
.map(|row| row.get(1))
|
2015-05-06 11:29:34 +08:00
|
|
|
.collect();
|
2015-05-05 21:29:46 +08:00
|
|
|
|
|
|
|
assert_eq!(results.unwrap().concat(), "hello, world!");
|
|
|
|
}
|
|
|
|
|
2015-01-11 11:17:49 +08:00
|
|
|
#[test]
|
2015-05-07 21:41:02 +08:00
|
|
|
fn test_query_row() {
|
2015-01-11 11:17:49 +08:00
|
|
|
let db = checked_memory_handle();
|
|
|
|
let sql = "BEGIN;
|
|
|
|
CREATE TABLE foo(x INTEGER);
|
|
|
|
INSERT INTO foo VALUES(1);
|
|
|
|
INSERT INTO foo VALUES(2);
|
|
|
|
INSERT INTO foo VALUES(3);
|
|
|
|
INSERT INTO foo VALUES(4);
|
|
|
|
END;";
|
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
assert_eq!(
|
|
|
|
10i64,
|
2018-09-16 17:10:19 +08:00
|
|
|
db.query_row::<i64, _, _>("SELECT SUM(x) FROM foo", NO_PARAMS, |r| r.get(0))
|
2018-08-11 18:48:21 +08:00
|
|
|
.unwrap()
|
|
|
|
);
|
2015-01-11 11:17:49 +08:00
|
|
|
|
2018-09-16 15:49:23 +08:00
|
|
|
let result: Result<i64> =
|
2018-09-16 17:10:19 +08:00
|
|
|
db.query_row("SELECT x FROM foo WHERE x > 5", NO_PARAMS, |r| r.get(0));
|
2015-12-13 13:54:08 +08:00
|
|
|
match result.unwrap_err() {
|
|
|
|
Error::QueryReturnedNoRows => (),
|
|
|
|
err => panic!("Unexpected error {}", err),
|
|
|
|
}
|
2015-01-11 11:17:49 +08:00
|
|
|
|
2019-02-22 03:48:09 +08:00
|
|
|
let bad_query_result = db.query_row("NOT A PROPER QUERY; test123", NO_PARAMS, |_| Ok(()));
|
2015-01-11 11:17:49 +08:00
|
|
|
|
|
|
|
assert!(bad_query_result.is_err());
|
|
|
|
}
|
|
|
|
|
2018-12-16 18:15:21 +08:00
|
|
|
#[test]
|
|
|
|
fn test_optional() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
|
2018-12-20 04:58:33 +08:00
|
|
|
let result: Result<i64> = db.query_row("SELECT 1 WHERE 0 <> 0", NO_PARAMS, |r| r.get(0));
|
2018-12-16 18:15:21 +08:00
|
|
|
let result = result.optional();
|
|
|
|
match result.unwrap() {
|
|
|
|
None => (),
|
|
|
|
_ => panic!("Unexpected result"),
|
|
|
|
}
|
|
|
|
|
2018-12-20 04:58:33 +08:00
|
|
|
let result: Result<i64> = db.query_row("SELECT 1 WHERE 0 == 0", NO_PARAMS, |r| r.get(0));
|
2018-12-16 18:15:21 +08:00
|
|
|
let result = result.optional();
|
|
|
|
match result.unwrap() {
|
|
|
|
Some(1) => (),
|
|
|
|
_ => panic!("Unexpected result"),
|
|
|
|
}
|
2015-01-11 11:17:49 +08:00
|
|
|
|
2018-12-16 18:15:21 +08:00
|
|
|
let bad_query_result: Result<i64> =
|
|
|
|
db.query_row("NOT A PROPER QUERY", NO_PARAMS, |r| r.get(0));
|
|
|
|
let bad_query_result = bad_query_result.optional();
|
2015-01-11 11:17:49 +08:00
|
|
|
assert!(bad_query_result.is_err());
|
|
|
|
}
|
|
|
|
|
2018-09-13 04:16:22 +08:00
|
|
|
#[test]
|
|
|
|
fn test_pragma_query_row() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
|
|
|
|
assert_eq!(
|
|
|
|
"memory",
|
2018-09-16 17:10:19 +08:00
|
|
|
db.query_row::<String, _, _>("PRAGMA journal_mode", NO_PARAMS, |r| r.get(0))
|
2018-09-13 04:16:22 +08:00
|
|
|
.unwrap()
|
|
|
|
);
|
|
|
|
assert_eq!(
|
|
|
|
"off",
|
2018-09-16 17:10:19 +08:00
|
|
|
db.query_row::<String, _, _>("PRAGMA journal_mode=off", NO_PARAMS, |r| r.get(0))
|
2018-09-13 04:16:22 +08:00
|
|
|
.unwrap()
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
2014-10-20 07:56:41 +08:00
|
|
|
#[test]
|
|
|
|
fn test_prepare_failures() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
db.execute_batch("CREATE TABLE foo(x INTEGER);").unwrap();
|
|
|
|
|
|
|
|
let err = db.prepare("SELECT * FROM does_not_exist").unwrap_err();
|
2015-12-13 13:54:08 +08:00
|
|
|
assert!(format!("{}", err).contains("does_not_exist"));
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_last_insert_rowid() {
|
|
|
|
let db = checked_memory_handle();
|
2017-04-08 01:43:24 +08:00
|
|
|
db.execute_batch("CREATE TABLE foo(x INTEGER PRIMARY KEY)")
|
|
|
|
.unwrap();
|
2018-08-11 18:48:21 +08:00
|
|
|
db.execute_batch("INSERT INTO foo DEFAULT VALUES").unwrap();
|
2014-10-20 07:56:41 +08:00
|
|
|
|
|
|
|
assert_eq!(db.last_insert_rowid(), 1);
|
|
|
|
|
|
|
|
let mut stmt = db.prepare("INSERT INTO foo DEFAULT VALUES").unwrap();
|
2015-12-11 05:48:09 +08:00
|
|
|
for _ in 0i32..9 {
|
2018-09-16 17:10:19 +08:00
|
|
|
stmt.execute(NO_PARAMS).unwrap();
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|
|
|
|
assert_eq!(db.last_insert_rowid(), 10);
|
|
|
|
}
|
2015-09-21 08:44:51 +08:00
|
|
|
|
2018-05-13 00:35:08 +08:00
|
|
|
#[test]
|
|
|
|
fn test_is_autocommit() {
|
|
|
|
let db = checked_memory_handle();
|
2018-08-11 18:48:21 +08:00
|
|
|
assert!(
|
|
|
|
db.is_autocommit(),
|
|
|
|
"autocommit expected to be active by default"
|
|
|
|
);
|
2018-05-13 00:35:08 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2020-01-15 00:11:36 +08:00
|
|
|
#[cfg(feature = "modern_sqlite")]
|
2018-05-13 00:35:08 +08:00
|
|
|
fn test_is_busy() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
assert!(!db.is_busy());
|
|
|
|
let mut stmt = db.prepare("PRAGMA schema_version").unwrap();
|
|
|
|
assert!(!db.is_busy());
|
|
|
|
{
|
2018-09-16 17:10:19 +08:00
|
|
|
let mut rows = stmt.query(NO_PARAMS).unwrap();
|
2018-05-13 00:35:08 +08:00
|
|
|
assert!(!db.is_busy());
|
2019-02-03 21:01:42 +08:00
|
|
|
let row = rows.next().unwrap();
|
2018-05-13 00:35:08 +08:00
|
|
|
assert!(db.is_busy());
|
|
|
|
assert!(row.is_some());
|
|
|
|
}
|
|
|
|
assert!(!db.is_busy());
|
|
|
|
}
|
|
|
|
|
2015-09-21 08:44:51 +08:00
|
|
|
#[test]
|
|
|
|
fn test_statement_debugging() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let query = "SELECT 12345";
|
|
|
|
let stmt = db.prepare(query).unwrap();
|
|
|
|
|
|
|
|
assert!(format!("{:?}", stmt).contains(query));
|
|
|
|
}
|
2015-09-21 09:28:50 +08:00
|
|
|
|
2015-12-13 12:10:35 +08:00
|
|
|
#[test]
|
|
|
|
fn test_notnull_constraint_error() {
|
2017-04-06 01:22:55 +08:00
|
|
|
// extended error codes for constraints were added in SQLite 3.7.16; if we're
|
|
|
|
// running on our bundled version, we know the extended error code exists.
|
2020-01-15 00:11:36 +08:00
|
|
|
#[cfg(feature = "modern_sqlite")]
|
2017-04-06 01:22:55 +08:00
|
|
|
fn check_extended_code(extended_code: c_int) {
|
|
|
|
assert_eq!(extended_code, ffi::SQLITE_CONSTRAINT_NOTNULL);
|
|
|
|
}
|
2020-01-15 00:11:36 +08:00
|
|
|
#[cfg(not(feature = "modern_sqlite"))]
|
2017-04-06 01:22:55 +08:00
|
|
|
fn check_extended_code(_extended_code: c_int) {}
|
|
|
|
|
2015-12-13 12:10:35 +08:00
|
|
|
let db = checked_memory_handle();
|
|
|
|
db.execute_batch("CREATE TABLE foo(x NOT NULL)").unwrap();
|
|
|
|
|
2018-09-16 17:10:19 +08:00
|
|
|
let result = db.execute("INSERT INTO foo (x) VALUES (NULL)", NO_PARAMS);
|
2015-12-13 12:10:35 +08:00
|
|
|
assert!(result.is_err());
|
|
|
|
|
2015-12-13 13:54:08 +08:00
|
|
|
match result.unwrap_err() {
|
2015-12-16 03:41:54 +08:00
|
|
|
Error::SqliteFailure(err, _) => {
|
2017-02-04 18:33:23 +08:00
|
|
|
assert_eq!(err.code, ErrorCode::ConstraintViolation);
|
2017-04-06 01:22:55 +08:00
|
|
|
check_extended_code(err.extended_code);
|
2016-02-03 02:12:00 +08:00
|
|
|
}
|
2015-12-13 13:54:08 +08:00
|
|
|
err => panic!("Unexpected error {}", err),
|
|
|
|
}
|
2015-12-13 12:10:35 +08:00
|
|
|
}
|
|
|
|
|
2017-02-07 09:19:55 +08:00
|
|
|
#[test]
|
|
|
|
fn test_version_string() {
|
|
|
|
let n = version_number();
|
|
|
|
let major = n / 1_000_000;
|
|
|
|
let minor = (n % 1_000_000) / 1_000;
|
|
|
|
let patch = n % 1_000;
|
|
|
|
|
2017-02-10 09:12:24 +08:00
|
|
|
assert!(version().contains(&format!("{}.{}.{}", major, minor, patch)));
|
2017-02-07 09:19:55 +08:00
|
|
|
}
|
|
|
|
|
2018-09-27 04:36:01 +08:00
|
|
|
#[test]
|
2018-10-28 16:54:30 +08:00
|
|
|
#[cfg(feature = "functions")]
|
2018-09-27 04:36:01 +08:00
|
|
|
fn test_interrupt() {
|
2018-10-28 16:54:30 +08:00
|
|
|
let db = checked_memory_handle();
|
2018-09-27 04:36:01 +08:00
|
|
|
|
2018-10-28 16:54:30 +08:00
|
|
|
let interrupt_handle = db.get_interrupt_handle();
|
2018-09-27 04:36:01 +08:00
|
|
|
|
2020-01-27 01:11:11 +08:00
|
|
|
db.create_scalar_function(
|
|
|
|
"interrupt",
|
|
|
|
0,
|
|
|
|
crate::functions::FunctionFlags::default(),
|
|
|
|
move |_| {
|
|
|
|
interrupt_handle.interrupt();
|
|
|
|
Ok(0)
|
|
|
|
},
|
|
|
|
)
|
2018-10-28 16:54:30 +08:00
|
|
|
.unwrap();
|
2018-09-27 04:36:01 +08:00
|
|
|
|
2018-10-28 16:54:30 +08:00
|
|
|
let mut stmt = db
|
|
|
|
.prepare("SELECT interrupt() FROM (SELECT 1 UNION SELECT 2 UNION SELECT 3)")
|
|
|
|
.unwrap();
|
2018-09-27 04:36:01 +08:00
|
|
|
|
2019-03-10 19:58:20 +08:00
|
|
|
let result: Result<Vec<i32>> = stmt.query(NO_PARAMS).unwrap().map(|r| r.get(0)).collect();
|
2018-10-28 16:54:30 +08:00
|
|
|
|
|
|
|
match result.unwrap_err() {
|
|
|
|
Error::SqliteFailure(err, _) => {
|
|
|
|
assert_eq!(err.code, ErrorCode::OperationInterrupted);
|
|
|
|
}
|
|
|
|
err => {
|
|
|
|
panic!("Unexpected error {}", err);
|
2018-09-27 04:36:01 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_interrupt_close() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let handle = db.get_interrupt_handle();
|
|
|
|
handle.interrupt();
|
|
|
|
db.close().unwrap();
|
|
|
|
handle.interrupt();
|
|
|
|
|
|
|
|
// Look at it's internals to see if we cleared it out properly.
|
|
|
|
let db_guard = handle.db_lock.lock().unwrap();
|
|
|
|
assert!(db_guard.is_null());
|
|
|
|
// It would be nice to test that we properly handle close/interrupt
|
|
|
|
// running at the same time, but it seems impossible to do with any
|
|
|
|
// degree of reliability.
|
|
|
|
}
|
|
|
|
|
2018-10-19 02:59:30 +08:00
|
|
|
#[test]
|
|
|
|
fn test_get_raw() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
db.execute_batch("CREATE TABLE foo(i, x);").unwrap();
|
|
|
|
let vals = ["foobar", "1234", "qwerty"];
|
|
|
|
let mut insert_stmt = db.prepare("INSERT INTO foo(i, x) VALUES(?, ?)").unwrap();
|
|
|
|
for (i, v) in vals.iter().enumerate() {
|
|
|
|
let i_to_insert = i as i64;
|
2019-01-31 03:14:46 +08:00
|
|
|
assert_eq!(insert_stmt.execute(params![i_to_insert, v]).unwrap(), 1);
|
2018-10-19 02:59:30 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
let mut query = db.prepare("SELECT i, x FROM foo").unwrap();
|
|
|
|
let mut rows = query.query(NO_PARAMS).unwrap();
|
|
|
|
|
2019-02-03 21:01:42 +08:00
|
|
|
while let Some(row) = rows.next().unwrap() {
|
2018-10-19 02:59:30 +08:00
|
|
|
let i = row.get_raw(0).as_i64().unwrap();
|
|
|
|
let expect = vals[i as usize];
|
|
|
|
let x = row.get_raw("x").as_str().unwrap();
|
|
|
|
assert_eq!(x, expect);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-01-06 19:58:46 +08:00
|
|
|
#[test]
|
|
|
|
fn test_from_handle() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let handle = unsafe { db.handle() };
|
|
|
|
{
|
|
|
|
let db = unsafe { Connection::from_handle(handle) }.unwrap();
|
|
|
|
db.execute_batch("PRAGMA VACUUM").unwrap();
|
|
|
|
}
|
|
|
|
db.close().unwrap();
|
|
|
|
}
|
|
|
|
|
2015-09-21 09:28:50 +08:00
|
|
|
mod query_and_then_tests {
|
2018-12-08 04:57:04 +08:00
|
|
|
|
2015-09-21 09:28:50 +08:00
|
|
|
use super::*;
|
|
|
|
|
2015-12-13 13:54:08 +08:00
|
|
|
#[derive(Debug)]
|
2015-09-21 09:28:50 +08:00
|
|
|
enum CustomError {
|
|
|
|
SomeError,
|
2015-12-13 02:53:58 +08:00
|
|
|
Sqlite(Error),
|
2015-09-21 09:28:50 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
impl fmt::Display for CustomError {
|
2020-04-07 05:43:06 +08:00
|
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> Result<(), fmt::Error> {
|
2015-09-21 09:28:50 +08:00
|
|
|
match *self {
|
2020-01-26 23:57:58 +08:00
|
|
|
CustomError::SomeError => write!(f, "my custom error"),
|
|
|
|
CustomError::Sqlite(ref se) => write!(f, "my custom error: {}", se),
|
2015-09-21 09:28:50 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl StdError for CustomError {
|
2015-12-11 05:48:09 +08:00
|
|
|
fn description(&self) -> &str {
|
|
|
|
"my custom error"
|
|
|
|
}
|
2018-08-17 00:29:46 +08:00
|
|
|
|
2018-12-08 04:57:04 +08:00
|
|
|
fn cause(&self) -> Option<&dyn StdError> {
|
2015-09-21 09:28:50 +08:00
|
|
|
match *self {
|
2015-12-11 05:48:09 +08:00
|
|
|
CustomError::SomeError => None,
|
2015-09-21 09:28:50 +08:00
|
|
|
CustomError::Sqlite(ref se) => Some(se),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-12-13 02:53:58 +08:00
|
|
|
impl From<Error> for CustomError {
|
|
|
|
fn from(se: Error) -> CustomError {
|
2015-09-21 09:28:50 +08:00
|
|
|
CustomError::Sqlite(se)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-04-07 05:43:06 +08:00
|
|
|
type CustomResult<T> = Result<T, CustomError>;
|
2015-09-21 09:28:50 +08:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_query_and_then() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let sql = "BEGIN;
|
|
|
|
CREATE TABLE foo(x INTEGER, y TEXT);
|
|
|
|
INSERT INTO foo VALUES(4, \"hello\");
|
|
|
|
INSERT INTO foo VALUES(3, \", \");
|
|
|
|
INSERT INTO foo VALUES(2, \"world\");
|
|
|
|
INSERT INTO foo VALUES(1, \"!\");
|
|
|
|
END;";
|
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
|
|
|
|
let mut query = db.prepare("SELECT x, y FROM foo ORDER BY x DESC").unwrap();
|
2018-08-11 18:48:21 +08:00
|
|
|
let results: Result<Vec<String>> = query
|
2019-02-22 03:48:09 +08:00
|
|
|
.query_and_then(NO_PARAMS, |row| row.get(1))
|
2015-09-21 09:28:50 +08:00
|
|
|
.unwrap()
|
|
|
|
.collect();
|
|
|
|
|
|
|
|
assert_eq!(results.unwrap().concat(), "hello, world!");
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_query_and_then_fails() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let sql = "BEGIN;
|
|
|
|
CREATE TABLE foo(x INTEGER, y TEXT);
|
|
|
|
INSERT INTO foo VALUES(4, \"hello\");
|
|
|
|
INSERT INTO foo VALUES(3, \", \");
|
|
|
|
INSERT INTO foo VALUES(2, \"world\");
|
|
|
|
INSERT INTO foo VALUES(1, \"!\");
|
|
|
|
END;";
|
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
|
|
|
|
let mut query = db.prepare("SELECT x, y FROM foo ORDER BY x DESC").unwrap();
|
2018-08-11 18:48:21 +08:00
|
|
|
let bad_type: Result<Vec<f64>> = query
|
2019-02-22 03:48:09 +08:00
|
|
|
.query_and_then(NO_PARAMS, |row| row.get(1))
|
2015-09-21 09:28:50 +08:00
|
|
|
.unwrap()
|
|
|
|
.collect();
|
|
|
|
|
2015-12-13 13:54:08 +08:00
|
|
|
match bad_type.unwrap_err() {
|
2019-11-03 18:19:07 +08:00
|
|
|
Error::InvalidColumnType(..) => (),
|
2015-12-13 13:54:08 +08:00
|
|
|
err => panic!("Unexpected error {}", err),
|
|
|
|
}
|
2015-09-21 09:28:50 +08:00
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
let bad_idx: Result<Vec<String>> = query
|
2019-02-22 03:48:09 +08:00
|
|
|
.query_and_then(NO_PARAMS, |row| row.get(3))
|
2015-09-21 09:28:50 +08:00
|
|
|
.unwrap()
|
|
|
|
.collect();
|
|
|
|
|
2015-12-13 13:54:08 +08:00
|
|
|
match bad_idx.unwrap_err() {
|
|
|
|
Error::InvalidColumnIndex(_) => (),
|
|
|
|
err => panic!("Unexpected error {}", err),
|
|
|
|
}
|
2015-09-21 09:28:50 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_query_and_then_custom_error() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let sql = "BEGIN;
|
|
|
|
CREATE TABLE foo(x INTEGER, y TEXT);
|
|
|
|
INSERT INTO foo VALUES(4, \"hello\");
|
|
|
|
INSERT INTO foo VALUES(3, \", \");
|
|
|
|
INSERT INTO foo VALUES(2, \"world\");
|
|
|
|
INSERT INTO foo VALUES(1, \"!\");
|
|
|
|
END;";
|
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
|
|
|
|
let mut query = db.prepare("SELECT x, y FROM foo ORDER BY x DESC").unwrap();
|
2018-08-11 18:48:21 +08:00
|
|
|
let results: CustomResult<Vec<String>> = query
|
2019-02-22 03:48:09 +08:00
|
|
|
.query_and_then(NO_PARAMS, |row| row.get(1).map_err(CustomError::Sqlite))
|
2018-10-28 15:51:02 +08:00
|
|
|
.unwrap()
|
2015-09-21 09:28:50 +08:00
|
|
|
.collect();
|
|
|
|
|
|
|
|
assert_eq!(results.unwrap().concat(), "hello, world!");
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_query_and_then_custom_error_fails() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let sql = "BEGIN;
|
|
|
|
CREATE TABLE foo(x INTEGER, y TEXT);
|
|
|
|
INSERT INTO foo VALUES(4, \"hello\");
|
|
|
|
INSERT INTO foo VALUES(3, \", \");
|
|
|
|
INSERT INTO foo VALUES(2, \"world\");
|
|
|
|
INSERT INTO foo VALUES(1, \"!\");
|
|
|
|
END;";
|
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
|
|
|
|
let mut query = db.prepare("SELECT x, y FROM foo ORDER BY x DESC").unwrap();
|
2018-08-11 18:48:21 +08:00
|
|
|
let bad_type: CustomResult<Vec<f64>> = query
|
2019-02-22 03:48:09 +08:00
|
|
|
.query_and_then(NO_PARAMS, |row| row.get(1).map_err(CustomError::Sqlite))
|
2018-10-28 15:51:02 +08:00
|
|
|
.unwrap()
|
2015-09-21 09:28:50 +08:00
|
|
|
.collect();
|
|
|
|
|
2015-12-13 13:54:08 +08:00
|
|
|
match bad_type.unwrap_err() {
|
2019-11-03 18:19:07 +08:00
|
|
|
CustomError::Sqlite(Error::InvalidColumnType(..)) => (),
|
2015-12-13 13:54:08 +08:00
|
|
|
err => panic!("Unexpected error {}", err),
|
|
|
|
}
|
2015-09-21 09:28:50 +08:00
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
let bad_idx: CustomResult<Vec<String>> = query
|
2019-02-22 03:48:09 +08:00
|
|
|
.query_and_then(NO_PARAMS, |row| row.get(3).map_err(CustomError::Sqlite))
|
2018-10-28 15:51:02 +08:00
|
|
|
.unwrap()
|
2015-09-21 09:28:50 +08:00
|
|
|
.collect();
|
|
|
|
|
2015-12-13 13:54:08 +08:00
|
|
|
match bad_idx.unwrap_err() {
|
|
|
|
CustomError::Sqlite(Error::InvalidColumnIndex(_)) => (),
|
|
|
|
err => panic!("Unexpected error {}", err),
|
|
|
|
}
|
2015-09-21 09:28:50 +08:00
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
let non_sqlite_err: CustomResult<Vec<String>> = query
|
2018-09-16 17:10:19 +08:00
|
|
|
.query_and_then(NO_PARAMS, |_| Err(CustomError::SomeError))
|
2018-08-11 18:48:21 +08:00
|
|
|
.unwrap()
|
2015-09-21 09:28:50 +08:00
|
|
|
.collect();
|
|
|
|
|
2015-12-13 13:54:08 +08:00
|
|
|
match non_sqlite_err.unwrap_err() {
|
|
|
|
CustomError::SomeError => (),
|
|
|
|
err => panic!("Unexpected error {}", err),
|
|
|
|
}
|
2015-09-21 09:28:50 +08:00
|
|
|
}
|
|
|
|
|
2015-09-21 09:30:40 +08:00
|
|
|
#[test]
|
|
|
|
fn test_query_row_and_then_custom_error() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let sql = "BEGIN;
|
|
|
|
CREATE TABLE foo(x INTEGER, y TEXT);
|
|
|
|
INSERT INTO foo VALUES(4, \"hello\");
|
|
|
|
END;";
|
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
|
|
|
|
let query = "SELECT x, y FROM foo ORDER BY x DESC";
|
2018-09-16 17:10:19 +08:00
|
|
|
let results: CustomResult<String> = db.query_row_and_then(query, NO_PARAMS, |row| {
|
2019-02-22 03:48:09 +08:00
|
|
|
row.get(1).map_err(CustomError::Sqlite)
|
2015-12-11 05:48:09 +08:00
|
|
|
});
|
2015-09-21 09:30:40 +08:00
|
|
|
|
|
|
|
assert_eq!(results.unwrap(), "hello");
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_query_row_and_then_custom_error_fails() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let sql = "BEGIN;
|
|
|
|
CREATE TABLE foo(x INTEGER, y TEXT);
|
|
|
|
INSERT INTO foo VALUES(4, \"hello\");
|
|
|
|
END;";
|
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
|
|
|
|
let query = "SELECT x, y FROM foo ORDER BY x DESC";
|
2018-09-16 17:10:19 +08:00
|
|
|
let bad_type: CustomResult<f64> = db.query_row_and_then(query, NO_PARAMS, |row| {
|
2019-02-22 03:48:09 +08:00
|
|
|
row.get(1).map_err(CustomError::Sqlite)
|
2015-12-11 05:48:09 +08:00
|
|
|
});
|
2015-09-21 09:30:40 +08:00
|
|
|
|
2015-12-13 13:54:08 +08:00
|
|
|
match bad_type.unwrap_err() {
|
2019-11-03 18:19:07 +08:00
|
|
|
CustomError::Sqlite(Error::InvalidColumnType(..)) => (),
|
2015-12-13 13:54:08 +08:00
|
|
|
err => panic!("Unexpected error {}", err),
|
|
|
|
}
|
2015-09-21 09:30:40 +08:00
|
|
|
|
2018-09-16 17:10:19 +08:00
|
|
|
let bad_idx: CustomResult<String> = db.query_row_and_then(query, NO_PARAMS, |row| {
|
2019-02-22 03:48:09 +08:00
|
|
|
row.get(3).map_err(CustomError::Sqlite)
|
2015-12-11 05:48:09 +08:00
|
|
|
});
|
2015-09-21 09:30:40 +08:00
|
|
|
|
2015-12-13 13:54:08 +08:00
|
|
|
match bad_idx.unwrap_err() {
|
|
|
|
CustomError::Sqlite(Error::InvalidColumnIndex(_)) => (),
|
|
|
|
err => panic!("Unexpected error {}", err),
|
|
|
|
}
|
2015-09-21 09:30:40 +08:00
|
|
|
|
2018-08-11 18:48:21 +08:00
|
|
|
let non_sqlite_err: CustomResult<String> =
|
2018-09-16 17:10:19 +08:00
|
|
|
db.query_row_and_then(query, NO_PARAMS, |_| Err(CustomError::SomeError));
|
2015-09-21 09:30:40 +08:00
|
|
|
|
2015-12-13 13:54:08 +08:00
|
|
|
match non_sqlite_err.unwrap_err() {
|
|
|
|
CustomError::SomeError => (),
|
|
|
|
err => panic!("Unexpected error {}", err),
|
|
|
|
}
|
2015-09-21 09:30:40 +08:00
|
|
|
}
|
|
|
|
|
2016-01-02 17:28:00 +08:00
|
|
|
#[test]
|
|
|
|
fn test_dynamic() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
let sql = "BEGIN;
|
|
|
|
CREATE TABLE foo(x INTEGER, y TEXT);
|
|
|
|
INSERT INTO foo VALUES(4, \"hello\");
|
|
|
|
END;";
|
|
|
|
db.execute_batch(sql).unwrap();
|
|
|
|
|
2019-01-30 07:33:57 +08:00
|
|
|
db.query_row("SELECT * FROM foo", params![], |r| {
|
2019-02-22 03:48:09 +08:00
|
|
|
assert_eq!(2, r.column_count());
|
|
|
|
Ok(())
|
2018-10-28 15:51:02 +08:00
|
|
|
})
|
|
|
|
.unwrap();
|
2016-01-02 17:28:00 +08:00
|
|
|
}
|
2019-03-28 07:46:54 +08:00
|
|
|
#[test]
|
|
|
|
fn test_dyn_box() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
db.execute_batch("CREATE TABLE foo(x INTEGER);").unwrap();
|
|
|
|
let b: Box<dyn ToSql> = Box::new(5);
|
|
|
|
db.execute("INSERT INTO foo VALUES(?)", &[b]).unwrap();
|
|
|
|
db.query_row("SELECT x FROM foo", params![], |r| {
|
|
|
|
assert_eq!(5, r.get_unwrap::<_, i32>(0));
|
|
|
|
Ok(())
|
2018-10-28 15:51:02 +08:00
|
|
|
})
|
|
|
|
.unwrap();
|
2016-01-02 17:28:00 +08:00
|
|
|
}
|
2020-01-19 02:04:28 +08:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_params() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
db.query_row(
|
|
|
|
"SELECT
|
|
|
|
?, ?, ?, ?, ?, ?, ?, ?, ?, ?,
|
|
|
|
?, ?, ?, ?, ?, ?, ?, ?, ?, ?,
|
|
|
|
?, ?, ?, ?, ?, ?, ?, ?, ?, ?,
|
|
|
|
?, ?, ?, ?;",
|
|
|
|
params![
|
2020-01-26 23:57:58 +08:00
|
|
|
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
|
|
|
|
1, 1, 1, 1, 1, 1, 1, 1,
|
2020-01-19 02:04:28 +08:00
|
|
|
],
|
|
|
|
|r| {
|
|
|
|
assert_eq!(1, r.get_unwrap::<_, i32>(0));
|
|
|
|
Ok(())
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.unwrap();
|
|
|
|
}
|
2020-03-05 03:26:31 +08:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[cfg(not(feature = "extra_check"))]
|
|
|
|
fn test_alter_table() {
|
|
|
|
let db = checked_memory_handle();
|
|
|
|
db.execute_batch("CREATE TABLE x(t);").unwrap();
|
|
|
|
// `execute_batch` should be used but `execute` should also work
|
|
|
|
db.execute("ALTER TABLE x RENAME TO y;", params![]).unwrap();
|
|
|
|
}
|
2015-09-21 09:28:50 +08:00
|
|
|
}
|
2014-10-20 07:56:41 +08:00
|
|
|
}
|