rusqlite/
load_extension_guard.rs

1use crate::{Connection, Result};
2
3/// `feature = "load_extension"` RAII guard temporarily enabling SQLite
4/// extensions to be loaded.
5///
6/// ## Example
7///
8/// ```rust,no_run
9/// # use rusqlite::{Connection, Result, LoadExtensionGuard};
10/// # use std::path::{Path};
11/// fn load_my_extension(conn: &Connection) -> Result<()> {
12///     let _guard = LoadExtensionGuard::new(conn)?;
13///
14///     conn.load_extension(Path::new("my_sqlite_extension"), None)
15/// }
16/// ```
17pub struct LoadExtensionGuard<'conn> {
18    conn: &'conn Connection,
19}
20
21impl LoadExtensionGuard<'_> {
22    /// Attempt to enable loading extensions. Loading extensions will be
23    /// disabled when this guard goes out of scope. Cannot be meaningfully
24    /// nested.
25    #[inline]
26    pub fn new(conn: &Connection) -> Result<LoadExtensionGuard<'_>> {
27        conn.load_extension_enable()
28            .map(|_| LoadExtensionGuard { conn })
29    }
30}
31
32#[allow(unused_must_use)]
33impl Drop for LoadExtensionGuard<'_> {
34    #[inline]
35    fn drop(&mut self) {
36        self.conn.load_extension_disable();
37    }
38}