rusqlite/src/limits.rs

75 lines
2.5 KiB
Rust
Raw Normal View History

//! `feature = "limits"` Run-Time Limits
2017-02-04 18:01:38 +08:00
use std::os::raw::c_int;
2017-02-04 18:01:38 +08:00
2018-10-31 03:11:35 +08:00
use crate::ffi;
pub use crate::ffi::Limit;
2017-02-04 18:01:38 +08:00
2018-10-31 03:11:35 +08:00
use crate::Connection;
2017-02-04 18:01:38 +08:00
impl Connection {
/// `feature = "limits"` Returns the current value of a limit.
#[inline]
pub fn limit(&self, limit: Limit) -> i32 {
2017-02-04 18:01:38 +08:00
let c = self.db.borrow();
unsafe { ffi::sqlite3_limit(c.db(), limit as c_int, -1) }
2017-02-04 18:01:38 +08:00
}
2017-02-08 08:43:54 +08:00
/// `feature = "limits"` Changes the limit to `new_val`, returning the prior
/// value of the limit.
#[inline]
2017-02-04 18:01:38 +08:00
pub fn set_limit(&self, limit: Limit, new_val: i32) -> i32 {
let c = self.db.borrow_mut();
unsafe { ffi::sqlite3_limit(c.db(), limit as c_int, new_val) }
2017-02-04 18:01:38 +08:00
}
}
#[cfg(test)]
mod test {
2018-10-31 03:11:35 +08:00
use crate::ffi::Limit;
use crate::Connection;
2017-02-04 18:01:38 +08:00
#[test]
fn test_limit() {
let db = Connection::open_in_memory().unwrap();
db.set_limit(Limit::SQLITE_LIMIT_LENGTH, 1024);
assert_eq!(1024, db.limit(Limit::SQLITE_LIMIT_LENGTH));
2017-02-04 18:01:38 +08:00
db.set_limit(Limit::SQLITE_LIMIT_SQL_LENGTH, 1024);
assert_eq!(1024, db.limit(Limit::SQLITE_LIMIT_SQL_LENGTH));
2017-02-04 18:01:38 +08:00
db.set_limit(Limit::SQLITE_LIMIT_COLUMN, 64);
assert_eq!(64, db.limit(Limit::SQLITE_LIMIT_COLUMN));
2017-02-04 18:01:38 +08:00
db.set_limit(Limit::SQLITE_LIMIT_EXPR_DEPTH, 256);
assert_eq!(256, db.limit(Limit::SQLITE_LIMIT_EXPR_DEPTH));
2017-02-04 18:01:38 +08:00
db.set_limit(Limit::SQLITE_LIMIT_COMPOUND_SELECT, 32);
assert_eq!(32, db.limit(Limit::SQLITE_LIMIT_COMPOUND_SELECT));
2017-02-04 18:01:38 +08:00
db.set_limit(Limit::SQLITE_LIMIT_FUNCTION_ARG, 32);
assert_eq!(32, db.limit(Limit::SQLITE_LIMIT_FUNCTION_ARG));
2017-02-04 18:01:38 +08:00
db.set_limit(Limit::SQLITE_LIMIT_ATTACHED, 2);
assert_eq!(2, db.limit(Limit::SQLITE_LIMIT_ATTACHED));
2017-02-04 18:01:38 +08:00
db.set_limit(Limit::SQLITE_LIMIT_LIKE_PATTERN_LENGTH, 128);
assert_eq!(128, db.limit(Limit::SQLITE_LIMIT_LIKE_PATTERN_LENGTH));
2017-02-04 18:01:38 +08:00
db.set_limit(Limit::SQLITE_LIMIT_VARIABLE_NUMBER, 99);
assert_eq!(99, db.limit(Limit::SQLITE_LIMIT_VARIABLE_NUMBER));
2017-02-04 18:01:38 +08:00
// SQLITE_LIMIT_TRIGGER_DEPTH was added in SQLite 3.6.18.
2019-02-02 18:37:26 +08:00
if crate::version_number() >= 3_006_018 {
db.set_limit(Limit::SQLITE_LIMIT_TRIGGER_DEPTH, 32);
assert_eq!(32, db.limit(Limit::SQLITE_LIMIT_TRIGGER_DEPTH));
}
2017-02-04 18:01:38 +08:00
2017-02-08 08:43:54 +08:00
// SQLITE_LIMIT_WORKER_THREADS was added in SQLite 3.8.7.
2019-02-02 18:37:26 +08:00
if crate::version_number() >= 3_008_007 {
db.set_limit(Limit::SQLITE_LIMIT_WORKER_THREADS, 2);
assert_eq!(2, db.limit(Limit::SQLITE_LIMIT_WORKER_THREADS));
}
2017-02-04 18:01:38 +08:00
}
}