refactor!: 重新实现最近对话和收藏对话的逻辑 (破坏性变更)

2026-01-25 00:50:14 +08:00
parent 44ada8206d
commit ec527bafc6
5 changed files with 160 additions and 41 deletions
--- a/server/data/UserRecentChatLinker.ts
+++ b/server/data/UserRecentChatLinker.ts
@@ -0,0 +1,71 @@
+import { DatabaseSync } from "node:sqlite"
+import path from 'node:path'
+
+import RecentChatBean from './RecentChatBean.ts'
+import config from "../config.ts"
+import { SQLInputValue } from "node:sqlite"
+
+export default class UserRecentChatLinker {
+    static database: DatabaseSync = this.init()
+
+    private static init(): DatabaseSync {
+        const db: DatabaseSync = new DatabaseSync(path.join(config.data_path, 'UserRecentChatLinker.db'))
+        db.exec(`
+            CREATE TABLE IF NOT EXISTS UserRecentChatLinker (
+                /* 序号 */ count INTEGER PRIMARY KEY AUTOINCREMENT,
+                /* 用戶 ID */ user_id TEXT NOT NULL,
+                /* Chat ID */ chat_id TEXT NOT NULL,
+                /* Last Message Content */ content TEXT NOT NULL,
+                /* Last Update Time */ updated_time INT8 NOT NULL
+            );
+       `)
+        db.exec(`CREATE INDEX IF NOT EXISTS idx_user_id ON UserRecentChatLinker(user_id);`)
+        return db
+    }
+
+    /**
+     * 若用户和对话未关联, 则进行关联
+     */
+    static updateOrAddRecentChat(userId: string, chatId: string, content: string) {
+        if (!this.checkUserIsLinkedToChat(userId, chatId))
+            this.database.prepare(`INSERT INTO UserRecentChatLinker (
+                user_id,
+                chat_id,
+                content,
+                updated_time
+            ) VALUES (?, ?, ?, ?);`).run(
+                userId,
+                chatId,
+                content,
+                Date.now()
+            )
+        else
+            this.database.prepare('UPDATE UserRecentChatLinker SET content = ?, updated_time = ? WHERE count = ?').run(
+                content,
+                Date.now(),
+                /* 既然已经 bind 了, 那么就不需要判断了? */
+                this.findAllByCondition('user_id = ? AND chat_id = ?', userId, chatId)[0].count
+            )
+    }
+    /**
+     * 解除用户和对话的关联
+     */
+    static removeRecentChat(userId: string, chatId: string) {
+        this.database.prepare(`DELETE FROM UserRecentChatLinker WHERE user_id = ? AND chat_id = ?`).run(userId, chatId)
+    }
+    /**
+     * 检测用户和对话的关联
+     */
+    static checkUserIsLinkedToChat(userId: string, chatId: string) {
+        return this.findAllByCondition('user_id = ? AND chat_id = ?', userId, chatId).length != 0
+    }
+    /**
+     * 获取用户所有关联的对话
+     */
+    static getUserRecentChatBeans(userId: string) {
+        return this.findAllByCondition('user_id = ? ORDER BY updated_time DESC', userId) as unknown as RecentChatBean[]
+    }
+    protected static findAllByCondition(condition: string, ...args: SQLInputValue[]) {
+        return this.database.prepare(`SELECT * FROM UserRecentChatLinker WHERE ${condition}`).all(...args)
+    }
+}