update

src/db/metadata.rs

@@ -1,6 +1,7 @@
 use super::Database;
 use crate::models::*;
 use sqlx::{Row, Sqlite, Transaction};
+use std::collections::HashMap;
 use std::ops::DerefMut;
 use uuid::Uuid;
 
@@ -65,33 +66,56 @@ impl Database {
                WHERE p.name LIKE ? OR a.alias_name LIKE ? ORDER BY p.total_albums DESC"#
         ).bind(&pattern).bind(&pattern).fetch_all(&self.pool).await.unwrap_or_default();
 
-        let mut results = Vec::new();
-        for r in &rows {
+        let pids: Vec<i64> = rows.iter().map(|r| r.get::<i64, _>("id")).collect();
+        let aliases = self.batch_get_person_aliases(&pids).await;
+
+        rows.iter().map(|r| {
             let pid = r.get::<i64, _>("id");
-            let aliases = self.get_person_aliases(pid).await;
-            results.push(PersonWithCounts {
+            PersonWithCounts {
                 id: pid, name: r.get("name"), display_name: r.get("display_name"),
                 total_albums: r.get("total_albums"), total_images: r.get("total_images"),
-                aliases: aliases.into_iter().map(|a| a.alias_name).collect(),
-            });
-        }
-        results
+                aliases: aliases.get(&pid).cloned().unwrap_or_default(),
+            }
+        }).collect()
     }
 
     pub async fn get_all_persons_with_aliases(&self) -> Vec<PersonWithCounts> {
         let rows = sqlx::query("SELECT id, name, display_name, total_albums, total_images FROM persons ORDER BY total_albums DESC")
             .fetch_all(&self.pool).await.unwrap_or_default();
-        let mut results = Vec::new();
-        for r in &rows {
+
+        let pids: Vec<i64> = rows.iter().map(|r| r.get::<i64, _>("id")).collect();
+        let aliases = self.batch_get_person_aliases(&pids).await;
+
+        rows.iter().map(|r| {
             let pid = r.get::<i64, _>("id");
-            let aliases = self.get_person_aliases(pid).await;
-            results.push(PersonWithCounts {
+            PersonWithCounts {
                 id: pid, name: r.get("name"), display_name: r.get("display_name"),
                 total_albums: r.get("total_albums"), total_images: r.get("total_images"),
-                aliases: aliases.into_iter().map(|a| a.alias_name).collect(),
-            });
+                aliases: aliases.get(&pid).cloned().unwrap_or_default(),
+            }
+        }).collect()
+    }
+
+    /// 批量查询多个人物的别名（避免 N+1）
+    async fn batch_get_person_aliases(&self, pids: &[i64]) -> HashMap<i64, Vec<String>> {
+        if pids.is_empty() { return HashMap::new(); }
+        let placeholders: Vec<String> = pids.iter().map(|_| "?".to_string()).collect();
+        let sql = format!(
+            "SELECT person_id, alias_name FROM person_aliases WHERE person_id IN ({})",
+            placeholders.join(",")
+        );
+        let mut q = sqlx::query(&sql);
+        for pid in pids {
+            q = q.bind(pid);
         }
-        results
+        let rows = q.fetch_all(&self.pool).await.unwrap_or_default();
+        let mut map: HashMap<i64, Vec<String>> = HashMap::new();
+        for r in &rows {
+            let pid: i64 = r.get("person_id");
+            let alias: String = r.get("alias_name");
+            map.entry(pid).or_default().push(alias);
+        }
+        map
     }
 
     // ── 标签查询 ──
@@ -106,6 +130,27 @@ impl Database {
         }).collect()
     }
 
+    /// 批量获取所有 tag→albums 映射（供缓存使用）
+    pub async fn get_all_tag_albums(&self) -> HashMap<String, Vec<AlbumInfo>> {
+        let rows = sqlx::query(
+            r#"SELECT t.tag, a.id, a.title, a.document, a.description, a.cover_image_id, a.image_count, a.created_at
+               FROM album_tags t JOIN albums a ON t.album_id = a.id ORDER BY a.created_at DESC"#
+        ).fetch_all(&self.pool).await.unwrap_or_default();
+
+        let mut map: HashMap<String, Vec<AlbumInfo>> = HashMap::new();
+        for r in &rows {
+            let tag: String = r.get("tag");
+            let album = AlbumInfo {
+                id: Uuid::parse_str(&r.get::<String, _>("id")).unwrap_or_else(|_| Uuid::nil()),
+                title: r.get("title"), document: r.get("document"),
+                description: r.get("description"), cover_image_id: r.get("cover_image_id"),
+                image_count: r.get("image_count"), created_at: r.get("created_at"),
+            };
+            map.entry(tag).or_default().push(album);
+        }
+        map
+    }
+
     pub async fn get_tag_co_occurrences(&self, tag: &str, limit: i64) -> Vec<TagCoOccurrence> {
         let rows = sqlx::query(
             r#"SELECT t2.tag as co_tag, COUNT(*) as cnt
@@ -197,6 +242,24 @@ impl Database {
         sqlx::query("DELETE FROM person_albums WHERE person_id = ?").bind(person_id).execute(&self.pool).await.ok();
     }
 
+    /// 删除某相册的所有人物关联（写入新 person 前清旧数据，防止一个相册关联多个人物）
+    pub async fn delete_album_person_albums(&self, album_id: &str) {
+        sqlx::query("DELETE FROM person_albums WHERE album_id = ?").bind(album_id).execute(&self.pool).await.ok();
+    }
+
+    pub async fn delete_album_person_albums_with_tx(&self, tx: &mut Transaction<'_, Sqlite>, album_id: &str) {
+        sqlx::query("DELETE FROM person_albums WHERE album_id = ?").bind(album_id).execute(tx.deref_mut()).await.ok();
+    }
+
+    /// 删除某人物下所有别名（写入新别名前清旧别名，配合 UNIQUE INDEX 防止重复）
+    pub async fn delete_person_aliases(&self, person_id: i64) {
+        sqlx::query("DELETE FROM person_aliases WHERE person_id = ?").bind(person_id).execute(&self.pool).await.ok();
+    }
+
+    pub async fn delete_person_aliases_with_tx(&self, tx: &mut Transaction<'_, Sqlite>, person_id: i64) {
+        sqlx::query("DELETE FROM person_aliases WHERE person_id = ?").bind(person_id).execute(tx.deref_mut()).await.ok();
+    }
+
     /// 彻底删除一个人物（包括别名、相册关联、自身）
     pub async fn delete_person_completely(&self, person_id: i64) {
         let _guard = self.write_lock.lock().await;