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

Fix borrowing

Browse Source
This commit is contained in:
gwenn 2015-12-19 16:56:41 +01:00
parent 7bdf80ccdb
commit cb1951c21a
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/cache.rs
View File

@ -69,8 +69,9 @@ impl<'conn> StatementCache<'conn> {
///
/// Will return `Err` if no cached statement can be found and the underlying SQLite prepare call fails.
pub fn get<'s>(&'s self, sql: &str) -> Result<CachedStatement<'conn, 's>> {
let stmt = match self.cache.borrow().iter().rposition(|entry| entry.eq(sql)) {
Some(index) => Ok(self.cache.borrow_mut().swap_remove_front(index).unwrap()), // FIXME Not LRU compliant
let mut cache = self.cache.borrow_mut();
let stmt = match cache.iter().rposition(|entry| entry.eq(sql)) {
Some(index) => Ok(cache.swap_remove_front(index).unwrap()), // FIXME Not LRU compliant
_ => self.conn.prepare(sql),
};
stmt.map(|stmt| CachedStatement::new(stmt, self))
@ -85,13 +86,14 @@ impl<'conn> StatementCache<'conn> {
/// Will return `Err` if `stmt` (or the already cached statement implementing the same SQL) statement is `discard`ed
/// and the underlying SQLite finalize call fails.
fn release(&self, mut stmt: Statement<'conn>) {
if self.cache.borrow().capacity() == self.cache.borrow().len() {
let mut cache = self.cache.borrow_mut();
if cache.capacity() == cache.len() {
// is full
self.cache.borrow_mut().pop_back(); // LRU dropped
cache.pop_back(); // LRU dropped
}
stmt.reset_if_needed();
stmt.clear_bindings();
self.cache.borrow_mut().push_front(stmt)
cache.push_front(stmt)
}
/// Flush the prepared statement cache
@ -118,7 +120,7 @@ mod test {
#[test]
fn test_cache() {
let db = Connection::open_in_memory().unwrap();
let mut cache = StatementCache::new(&db, 15);
let cache = StatementCache::new(&db, 15);
assert_eq!(0, cache.len());
assert_eq!(15, cache.capacity());