story-kit: merge 260_refactor_upgrade_libsqlite3_sys
This commit is contained in:
+45
@@ -0,0 +1,45 @@
|
||||
use crate::{Connection, Result};
|
||||
|
||||
/// RAII guard temporarily enabling SQLite extensions to be loaded.
|
||||
///
|
||||
/// ## Example
|
||||
///
|
||||
/// ```rust,no_run
|
||||
/// # use rusqlite::{Connection, Result, LoadExtensionGuard};
|
||||
/// # use std::path::{Path};
|
||||
/// fn load_my_extension(conn: &Connection) -> Result<()> {
|
||||
/// unsafe {
|
||||
/// let _guard = LoadExtensionGuard::new(conn)?;
|
||||
/// conn.load_extension("trusted/sqlite/extension", None::<&str>)
|
||||
/// }
|
||||
/// }
|
||||
/// ```
|
||||
pub struct LoadExtensionGuard<'conn> {
|
||||
conn: &'conn Connection,
|
||||
}
|
||||
|
||||
impl LoadExtensionGuard<'_> {
|
||||
/// Attempt to enable loading extensions. Loading extensions will be
|
||||
/// disabled when this guard goes out of scope. Cannot be meaningfully
|
||||
/// nested.
|
||||
///
|
||||
/// # Safety
|
||||
///
|
||||
/// You must not run untrusted queries while extension loading is enabled.
|
||||
///
|
||||
/// See the safety comment on [`Connection::load_extension_enable`] for more
|
||||
/// details.
|
||||
#[inline]
|
||||
pub unsafe fn new(conn: &Connection) -> Result<LoadExtensionGuard<'_>> {
|
||||
conn.load_extension_enable()
|
||||
.map(|_| LoadExtensionGuard { conn })
|
||||
}
|
||||
}
|
||||
|
||||
#[expect(unused_must_use)]
|
||||
impl Drop for LoadExtensionGuard<'_> {
|
||||
#[inline]
|
||||
fn drop(&mut self) {
|
||||
self.conn.load_extension_disable();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user