From 3ccd80a1e2ae47123c5637d42e230cf2bfb432f8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=81=8D=E5=85=AE=E6=83=9A=E5=85=AE?=
 <101191390+HIllya51@users.noreply.github.com>
Date: Sat, 10 Aug 2024 04:00:03 +0800
Subject: [PATCH] ttscache

---
 .../LunaTranslator/myutils/utils.py           | 39 ++++++++++++-------
 .../LunaTranslator/tts/basettsclass.py        |  9 ++++-
 2 files changed, 33 insertions(+), 15 deletions(-)

diff --git a/LunaTranslator/LunaTranslator/myutils/utils.py b/LunaTranslator/LunaTranslator/myutils/utils.py
index e0ce469d..e353d997 100644
--- a/LunaTranslator/LunaTranslator/myutils/utils.py
+++ b/LunaTranslator/LunaTranslator/myutils/utils.py
@@ -685,24 +685,30 @@ def get_time_stamp(ct=None, ms=True):
 class LRUCache:
     def __init__(self, capacity: int):
         self.cache = {}
+        self.Lock = threading.Lock()
         self.capacity = capacity
         self.order = []
 
     def setcap(self, cap):
-        if cap == -1:
-            cap = 9999999999
-        self.capacity = cap
-        while len(self.cache) > self.capacity:
-            self.cache.popitem(last=False)
+        with self.Lock:
+            if cap == -1:
+                cap = 9999999999
+            self.capacity = cap
+            while len(self.cache) > self.capacity:
+                self.cache.popitem(last=False)
 
-    def get(self, key: int) -> bool:
+    def __get(self, key):
         if key in self.cache:
             self.order.remove(key)
             self.order.append(key)
-            return True
-        return False
+            return self.cache[key]
+        return None
 
-    def put(self, key: int) -> None:
+    def get(self, key):
+        with self.Lock:
+            return self.__get(key)
+
+    def __put(self, key, value=True) -> None:
         if not self.capacity:
             return
         if key in self.cache:
@@ -710,14 +716,19 @@ class LRUCache:
         elif len(self.order) == self.capacity:
             old_key = self.order.pop(0)
             del self.cache[old_key]
-        self.cache[key] = None
+        self.cache[key] = value
         self.order.append(key)
 
+    def put(self, key, value=True) -> None:
+        with self.Lock:
+            self.__put(key, value)
+
     def test(self, key):
-        _ = self.get(key)
-        if not _:
-            self.put(key)
-        return _
+        with self.Lock:
+            _ = self.__get(key)
+            if not _:
+                self.__put(key)
+            return _
 
 
 globalcachedmodule = {}
diff --git a/LunaTranslator/LunaTranslator/tts/basettsclass.py b/LunaTranslator/LunaTranslator/tts/basettsclass.py
index 8e5ecb77..9ac94738 100644
--- a/LunaTranslator/LunaTranslator/tts/basettsclass.py
+++ b/LunaTranslator/LunaTranslator/tts/basettsclass.py
@@ -3,6 +3,7 @@ import functools, threading
 from myutils.wrapper import threader
 from traceback import print_exc
 from myutils.proxy import getproxy
+from myutils.utils import LRUCache
 
 
 class TTSbase:
@@ -53,6 +54,7 @@ class TTSbase:
         self.typename = typename
         self.playaudiofunction = playaudiofunction
         self.uid = uid
+        self.LRUCache = LRUCache(3)
         if privateconfig is None:
             self.privateconfig = globalconfig["reader"][self.typename]
         else:
@@ -80,9 +82,14 @@ class TTSbase:
         if len(self.voicelist) == 0:
             return
         try:
+            key = content, self.rate, self.voice
+            data = self.LRUCache.get(key)
+            if data:
+                return callback(data)
             data = self.speak(content, self.rate, self.voice)
-            if data and len(data):
+            if data:
                 callback(data)
+                self.LRUCache.put(key, data)
         except:
             print_exc()
             return