使用Hive实现聊天历史、配置持久化存储

Author: NingNing0111

lib/controller/chat.dart

@@ -1,36 +1,28 @@
 import 'package:chat_all/controller/sidebar.dart';
 import 'package:get/get.dart';
-import 'package:uuid/uuid.dart';
 
 import '../model/message.dart';
 
-var uuid = Uuid();
-
 class ChatPageController extends GetxController {
-  HistoryMessage historyMessages = Get.find<SidebarPageController>().histories.last;
-
-  @override
-  void onInit() {
-    super.onInit();
-  }
+  HistoryMessage currHistoryMessage = Get.find<SidebarPageController>().histories.last;
 
   void updateMessageContent(int index, String content) {
-    historyMessages.messages[index].content = content;
+    currHistoryMessage.messages[index].content = content;
     update();
   }
 
   void addMessage(Message message) {
-    historyMessages.messages.add(message);
+    currHistoryMessage.messages.add(message);
     update();
   }
 
   void reloadHistory() {
-    historyMessages = Get.find<SidebarPageController>().histories.last;
+    currHistoryMessage = Get.find<SidebarPageController>().histories.last;
     update();
   }
 
   void setHistory(HistoryMessage history) {
-    historyMessages = history;
+    currHistoryMessage = history;
     update();
   }
 }

lib/controller/setting.dart

@@ -1,7 +1,9 @@
 
+import 'dart:developer';
 import 'dart:ui';
 
 import 'package:adaptive_theme/adaptive_theme.dart';
+import 'package:chat_all/db/setting_database.dart';
 import 'package:get/get.dart';
 
 class SettingPageController  extends  GetxController {
@@ -10,7 +12,7 @@ class SettingPageController  extends  GetxController {
   // 语言
   final language = Get.deviceLocale.obs;
   // api设置
-  final api = "https://api.mnzdna.xyz".obs;
+  final api = "https://api.openai.com".obs;
   final key = "".obs;
   // 对话配置
   final chatModel = "gpt-3.5-turbo".obs;
@@ -21,10 +23,29 @@ class SettingPageController  extends  GetxController {
   final disabledSystemPrompt = true.obs;
   final topP = 0.5.obs;
 
+  final db = SettingDatabase();
+
+  @override
+  void onInit() {
+    themeMode.value = db.getThemeMode();
+    language.value = db.getLanguage();
+    api.value = db.getApi()??api.value;
+    key.value = db.getKey()??key.value;
+    chatModel.value = db.getChatModel()??chatModel.value;
+    temperature.value = db.getTemperature()??temperature.value;
+    presencePenalty.value = db.getPresencePenalty()??presencePenalty.value;
+    frequencyPenalty.value = db.getFrequencyPenalty()??frequencyPenalty.value;
+    historyLength.value = db.getHistoryLength()??historyLength.value;
+    disabledSystemPrompt.value = db.getDisabledSystemPrompt()??disabledSystemPrompt.value;
+    topP.value = db.getTopP()??topP.value;
+    setLanguage(language.value!);
+    super.onInit();
+  }
 
   void setLanguage(Locale locale){
     language.value = locale;
     Get.updateLocale(locale);
+    update();
   }
 
   void reset() {
@@ -39,8 +60,23 @@ class SettingPageController  extends  GetxController {
 
   @override
   void onClose() {
-    // 配置存储到数据库中
+    super.onClose();
+    saveSettingInfo();
+  }
 
+  void saveSettingInfo(){
+    db.saveThemeMode(themeMode.value);
+    db.saveLanguage(language.value!);
+    db.saveApi(api.value);
+    db.saveKey(key.value);
+    db.saveChatModel(chatModel.value);
+    db.saveTemperature(temperature.value);
+    db.savePresencePenalty(presencePenalty.value);
+    db.saveFrequencyPenalty(frequencyPenalty.value);
+    db.saveHistoryLength(historyLength.value);
+    db.saveDisabledSystemPrompt(disabledSystemPrompt.value);
+    db.saveTopP(topP.value);
+    log("配置信息存储完成");
   }
 }
 

lib/controller/sidebar.dart

@@ -1,21 +1,27 @@
+import 'package:chat_all/db/message_database.dart';
 import 'package:chat_all/model/message.dart';
 import 'package:get/get.dart';
 import 'package:uuid/uuid.dart';
 
 class SidebarPageController extends GetxController {
 
-  final histories = <HistoryMessage>[].obs;
   final defaultOverview = "default_history_chat_overview".tr;
+  final histories = <HistoryMessage>[HistoryMessage(id: Uuid().v1(), title: "default_history_chat_overview".tr, messages: [], createTime: DateTime.now())].obs;
+
   final choice = 0.obs;
 
+  final db = MessageDatabase();
+
   @override
   void onInit() {
     super.onInit();
-
-    histories.add(
-        HistoryMessage(
-            id: const Uuid().v1(), title: defaultOverview, messages: [], createTime: DateTime.now())
-    );
+    var storeHistories = db.readAll();
+    histories.clear();
+    histories.addAll(storeHistories);
+    if(histories.isEmpty){
+      newHistory();
+    }
+    update();
   }
 
   void setChoice(int index){
@@ -25,6 +31,7 @@ class SidebarPageController extends GetxController {
 
   void addHistory(HistoryMessage historyMessage) {
     histories.add(historyMessage);
+    db.addHistoryMessage(historyMessage);
     update();
   }
 
@@ -41,21 +48,22 @@ class SidebarPageController extends GetxController {
     if (!hasHistory) {
       histories.add(historyMessage);
     }
+    db.updateHistoryMessage(historyMessage);
     update();
   }
 
   void deleteHistory(int index) {
+    db.deleteHistoryMessage(histories[index].id);
     histories.removeAt(index);
     if (histories.isEmpty) {
-      histories.add(HistoryMessage(
-          id: const Uuid().v1(), title: defaultOverview, messages: [], createTime: DateTime.now()));
+      newHistory();
     }
     update();
   }
 
   void newHistory() {
-
-    histories.add(HistoryMessage(
+    choice.value = 0;
+    addHistory(HistoryMessage(
         id: const Uuid().v1(), title: defaultOverview, messages: [], createTime: DateTime.now()));
     update();
   }
@@ -64,7 +72,18 @@ class SidebarPageController extends GetxController {
     histories.clear();
     newHistory();
     choice.value = 0;
+    db.clearHistoryMessage();
     update();
   }
 
+  @override
+  void onClose() {
+    db.saveHistory(histories);
+    super.onClose();
+  }
+
+  void saveAll() {
+    db.saveHistory(histories);
+  }
+
 }

lib/db/boxes.dart

@@ -0,0 +1,4 @@
+import 'package:hive/hive.dart';
+
+late Box historyBox;
+late Box settingBox;

lib/db/message_database.dart

@@ -0,0 +1,83 @@
+
+import 'package:chat_all/db/boxes.dart';
+import 'package:chat_all/model/message.dart';
+
+/**
+ * key-value
+ * key_histories 存储所有的HistoryMessage
+ */
+class MessageDatabase {
+
+  // 单例对象
+  static final MessageDatabase _instance = MessageDatabase._internal();
+
+  // 私有构造函数
+  MessageDatabase._internal();
+
+  // 工厂构造函数，用于返回单例对象
+  factory MessageDatabase() {
+    return _instance;
+  }
+
+  // 读取所有的history
+  List<HistoryMessage> readAll(){
+    var values = historyBox.get("key_histories");
+
+    List<HistoryMessage> histories = [];
+    if(values == null){
+      return histories;
+    }
+    for (var element in values) {
+      histories.add(HistoryMessage(id: element.id, title: element.title, messages: element.messages, createTime: element.createTime));
+    }
+    print(histories);
+    return histories;
+
+  }
+  
+  // 添加HistoryMessage
+  void addHistoryMessage(HistoryMessage message){
+    var histories = readAll();
+    histories.add(message);
+    historyBox.put("key_histories", histories);
+  }
+
+  // 删除HistoryMessage
+  void deleteHistoryMessage(String id){
+    var histories = readAll();
+    int len = histories.length;
+    int deleteIndex = 0;
+    while(deleteIndex < len && histories[deleteIndex].id != id) {
+      deleteIndex++;
+    }
+    if(deleteIndex == len) return;
+    histories.removeAt(deleteIndex);
+    historyBox.put("key_histories", histories);
+  }
+
+  // 清空HistoryMessage
+  void clearHistoryMessage(){
+    var histories = readAll();
+    histories.clear();
+    historyBox.put("key_histories", histories);
+  }
+
+  // 更新某个HistoryMessage
+  void updateHistoryMessage(HistoryMessage historyMessage) {
+    var histories = readAll();
+    int len = histories.length;
+    int updateIndex = 0;
+    while(updateIndex < len && histories[updateIndex].id != historyMessage.id){
+      updateIndex++;
+    }
+    if(updateIndex == len)return;
+    histories[updateIndex] = historyMessage;
+    historyBox.put("key_histories", histories);
+  }
+
+  // 保存
+  void saveHistory(List<HistoryMessage> histories){
+    historyBox.put("key_histories", histories);
+  }
+
+}