duckdb-slt 0.1.9

Command-line sqllogictest runner for DuckDB.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use std::path::Path;

use anyhow::Result;
use duckdb::{Config, Connection};

use crate::duckdb_driver::{DuckdbDriver, DuckdbDriverError};
use crate::extensions::{ExtensionActions, compile_extension_actions, escape_sql_string_literal};

pub(crate) fn compile_extensions(raw_specs: &[String]) -> Result<Vec<ExtensionActions>> {
    raw_specs
        .iter()
        .map(|raw| compile_extension_actions(raw))
        .collect()
}

pub(crate) fn open_driver(
    db: Option<&Path>,
    allow_unsigned_extensions: bool,
    extensions: &[ExtensionActions],
    required_extensions: &[String],
) -> Result<DuckdbDriver, DuckdbDriverError> {
    let conn = open_duckdb_connection(db, allow_unsigned_extensions)?;

    for ext in extensions {
        conn.execute_batch(&ext.install_sql)?;
        conn.execute_batch(&ext.load_sql)?;
    }

    for name in required_extensions {
        let sql = format!("LOAD '{}';", escape_sql_string_literal(name));
        let _ = conn.execute_batch(&sql);
    }

    Ok(DuckdbDriver::new(conn))
}

fn open_duckdb_connection(
    db: Option<&Path>,
    allow_unsigned_extensions: bool,
) -> duckdb::Result<Connection> {
    let mut config = Config::default();
    if allow_unsigned_extensions {
        config = config.allow_unsigned_extensions()?;
    }

    let conn = match db {
        Some(path) => Connection::open_with_flags(path, config)?,
        None => Connection::open_in_memory_with_flags(config)?,
    };

    Ok(conn)
}