multidb

src/queries.rs

@@ -0,0 +1,55 @@
+use std::{collections::HashMap, sync::{Mutex, OnceLock}};
+use crate::database::DatabaseKind;
+
+pub const Q001: &str = "SELECT id, slug, title, password_hash, created_at, updated_at FROM workspaces WHERE slug = ?";
+pub const Q002: &str = "INSERT INTO workspaces (slug, title, password_hash) VALUES (?, ?, ?)";
+pub const Q003: &str = "SELECT id, workspace_id, slug, title, content, created_at, updated_at, owner_map FROM notes WHERE workspace_id = ? ORDER BY updated_at DESC, id DESC";
+pub const Q004: &str = "SELECT id, workspace_id, slug, title, content, created_at, updated_at, owner_map FROM notes WHERE workspace_id = ? AND slug = ?";
+pub const Q005: &str = "INSERT INTO notes (workspace_id, slug, title) VALUES (?, ?, ?)";
+pub const Q006: &str = "UPDATE notes SET content = ?, owner_map = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?";
+pub const Q007: &str = "UPDATE workspaces SET updated_at = CURRENT_TIMESTAMP WHERE id = ?";
+pub const Q008: &str = "INSERT INTO note_revisions (note_id, content, author, owner_map) VALUES (?, ?, ?, ?)";
+pub const Q009: &str = "SELECT updated_at FROM notes WHERE id = ?";
+pub const Q010: &str = "SELECT id, content, created_at, author, owner_map FROM note_revisions WHERE note_id = ? ORDER BY id DESC LIMIT 100";
+pub const Q011: &str = "SELECT id, slug, title, content, password_hash, created_at, updated_at, owner_map FROM pads WHERE slug = ?";
+pub const Q012: &str = "INSERT INTO pads (slug, title, password_hash) VALUES (?, ?, ?)";
+pub const Q013: &str = "UPDATE pads SET content = ?, owner_map = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?";
+pub const Q014: &str = "INSERT INTO revisions (pad_id, content, author, owner_map) VALUES (?, ?, ?, ?)";
+pub const Q015: &str = "SELECT updated_at FROM pads WHERE id = ?";
+pub const Q016: &str = "SELECT id, content, created_at, author, owner_map FROM revisions WHERE pad_id = ? ORDER BY id DESC LIMIT 100";
+pub const Q017: &str = "SELECT token FROM published_pages WHERE pad_id = ?";
+pub const Q018: &str = "INSERT INTO published_pages (token, pad_id) VALUES (?, ?)";
+pub const Q019: &str = "SELECT token FROM published_pages WHERE note_id = ?";
+pub const Q020: &str = "INSERT INTO published_pages (token, note_id) VALUES (?, ?)";
+pub const Q021: &str = "SELECT pp.token, COALESCE(p.title, n.title) AS title, COALESCE(p.content, n.content) AS content, COALESCE(p.updated_at, n.updated_at) AS updated_at FROM published_pages pp LEFT JOIN pads p ON p.id = pp.pad_id LEFT JOIN notes n ON n.id = pp.note_id WHERE pp.token = ?";
+pub const Q022: &str = "SELECT file_token FROM pads WHERE id = ?";
+pub const Q023: &str = "UPDATE pads SET file_token = ? WHERE id = ? AND file_token IS NULL";
+pub const Q024: &str = "SELECT file_token FROM notes WHERE id = ?";
+pub const Q025: &str = "UPDATE notes SET file_token = ? WHERE id = ? AND file_token IS NULL";
+pub const Q026: &str = "SELECT id FROM pads WHERE file_token = ?";
+pub const Q027: &str = "SELECT id FROM notes WHERE file_token = ?";
+
+static POSTGRES_QUERIES: OnceLock<Mutex<HashMap<&'static str, &'static str>>> = OnceLock::new();
+
+pub fn get(kind: DatabaseKind, query: &'static str) -> &'static str {
+    if kind != DatabaseKind::Postgres { return query; }
+    let cache = POSTGRES_QUERIES.get_or_init(|| Mutex::new(HashMap::new()));
+    let mut cache = cache.lock().expect("query cache lock poisoned");
+    if let Some(value) = cache.get(query) { return value; }
+    let cache_key = query;
+    let query = query.replace("CURRENT_TIMESTAMP", "(CURRENT_TIMESTAMP::text)");
+    let mut index = 0;
+    let mut converted = String::with_capacity(query.len() + 8);
+    for ch in query.chars() {
+        if ch == '?' {
+            index += 1;
+            converted.push('$');
+            converted.push_str(&index.to_string());
+        } else {
+            converted.push(ch);
+        }
+    }
+    let converted = Box::leak(converted.into_boxed_str());
+    cache.insert(cache_key, converted);
+    converted
+}