udb 0.2.0

Universal Data Broker — a Rust gRPC broker over multiple databases (Postgres, MySQL, SQLite, MongoDB, ClickHouse, Cassandra, MSSQL, Redis, Qdrant, S3, Neo4j, …) with per-tenant RLS, 2PC, sagas, and CDC.
Documentation 
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

//! NW3-2 — SQLite plugin.
//!
//! Same shape as the MySQL plugin. The runtime register function
//! detects `sqlite::memory:` / `:memory:` in the DSN and clamps to
//! `max_connections = 1` so the in-memory schema and data survive
//! across operations on a single shared connection — that's what
//! makes SQLite usable as the long-pending in-memory test profile.

use crate::backend::BackendKind;
use crate::backend::plugin::{Backend, RegisterCtx};

/// Zero-sized plugin for SQLite.
#[derive(Debug, Default)]
pub struct SqlitePlugin;

/// Static instance referenced by `backend::plugins::all()`.
pub static PLUGIN: SqlitePlugin = SqlitePlugin;

#[async_trait::async_trait]
impl Backend for SqlitePlugin {
    fn kind(&self) -> BackendKind {
        BackendKind::Sqlite
    }

    async fn register(&self, ctx: &mut RegisterCtx<'_>) {
        crate::runtime::core::setup_data::register_sqlite(ctx).await;
    }

    fn generate_artifacts(
        &self,
        manifest: &crate::generation::CatalogManifest,
        sql_config: &crate::generation::sql::SqlGenerationConfig,
    ) -> Result<Vec<crate::generation::GeneratedArtifact>, String> {
        crate::generation::backends::generate_sqlite_artifacts(manifest, sql_config)
            .map_err(|err| err.to_string())
    }
}

impl crate::runtime::executors::handle::DispatchFactory for SqlitePlugin {
    fn build_dispatch_executor(
        &self,
        runtime: &crate::runtime::core::DataBrokerRuntime,
        instance: Option<&str>,
        _write: bool,
        context: Option<&crate::broker::RequestContext>,
    ) -> Result<crate::runtime::executors::handle::DispatchExecutor, tonic::Status> {
        let instance_name = instance.unwrap_or("primary");
        let pool = runtime
            .sqlite_pool_for_instance(instance_name)
            .ok_or_else(|| {
                tonic::Status::failed_precondition(format!(
                    "SQLite instance '{instance_name}' is not configured (set UDB_SQLITE_DSN)"
                ))
            })?
            .clone();
        // Thread the request context so generic-dispatch populates the
        // `_udb_context` table the RLS-style views read. Mirrors the PG plugin.
        let executor = match context {
            Some(ctx) => crate::runtime::executors::sqlite::SqliteExecutor::with_context(
                pool,
                std::sync::Arc::new(ctx.clone()),
            ),
            None => crate::runtime::executors::sqlite::SqliteExecutor::with_pool(pool),
        };
        Ok(crate::runtime::executors::handle::DispatchExecutor::Sqlite(
            executor,
        ))
    }
}