Merge pull request #104 from vritser/main

优化 STT 相关

Author: joey-zhou

src/main/java/com/xiaozhi/dialogue/stt/factory/SttServiceFactory.java

@@ -4,11 +4,13 @@
 import com.xiaozhi.dialogue.stt.providers.*;
 import com.xiaozhi.entity.SysConfig;
 
+import jakarta.annotation.Nonnull;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.stereotype.Component;
 
 import jakarta.annotation.PostConstruct;
+
 import java.util.Map;
 import java.util.concurrent.ConcurrentHashMap;
 
@@ -17,7 +19,7 @@ public class SttServiceFactory {
 
     private static final Logger logger = LoggerFactory.getLogger(SttServiceFactory.class);
 
-    // 缓存已初始化的服务：对于API服务，键为"provider:configId"格式；对于本地服务，键为provider名称
+    // 缓存已初始化的服务：key format: "provider:configId"
     private final Map<String, SttService> serviceCache = new ConcurrentHashMap<>();
 
     // 默认服务提供商名称
@@ -46,128 +48,81 @@ public void initializeDefaultSttService() {
     /**
      * 初始化Vosk服务
      */
-    private synchronized void initializeVosk() {
+    private synchronized SttService initializeVosk() {
         if (serviceCache.containsKey(DEFAULT_PROVIDER)) {
-            return;
+            return serviceCache.get(DEFAULT_PROVIDER);
         }
 
         try {
-            VoskSttService voskService = new VoskSttService();
+            var voskService = new VoskSttService();
             voskService.initialize();
             serviceCache.put(DEFAULT_PROVIDER, voskService);
             voskInitialized = true;
             logger.info("Vosk STT服务初始化成功");
+            return voskService;
         } catch (Exception e) {
             logger.error("Vosk STT服务初始化失败", e);
             voskInitialized = false;
         }
+        return null;
     }
 
     /**
      * 获取默认STT服务
-     * 如果Vosk可用则返回Vosk，否则返回备选服务
      */
     public SttService getDefaultSttService() {
-        // 如果Vosk已初始化成功，直接返回
-        if (voskInitialized && serviceCache.containsKey(DEFAULT_PROVIDER)) {
-            return serviceCache.get(DEFAULT_PROVIDER);
-        }
-
-        // 否则返回备选服务
-        if (fallbackProvider != null && serviceCache.containsKey(fallbackProvider)) {
-            return serviceCache.get(fallbackProvider);
-        }
-
-        // 如果没有备选服务，尝试创建一个API类型的服务作为备选
-        if (serviceCache.isEmpty()) {
-            logger.warn("没有可用的STT服务，将尝试创建默认API服务");
-            try {
-                return null;
-            } catch (Exception e) {
-                logger.error("创建默认API服务失败", e);
-                return null;
-            }
-        }
-
-        return null;
+        return getSttService(null);
     }
 
     /**
      * 根据配置获取STT服务
      */
     public SttService getSttService(SysConfig config) {
         if (config == null) {
-            return getDefaultSttService();
-        }
-
-        String provider = config.getProvider();
-
-        // 如果是Vosk，直接使用全局共享的实例
-        if (DEFAULT_PROVIDER.equals(provider)) {
-            // 如果Vosk还未初始化，尝试初始化
-            if (!voskInitialized && !serviceCache.containsKey(DEFAULT_PROVIDER)) {
-                initializeVosk();
-            }
-
-            // Vosk初始化失败的情况
-            if (!voskInitialized) {
-                return null;
-            }
-            return serviceCache.get(DEFAULT_PROVIDER);
+            config = new SysConfig().setProvider(DEFAULT_PROVIDER).setConfigId(-1);
         }
 
         // 对于API服务，使用"provider:configId"作为缓存键，确保每个配置使用独立的服务实例
-        Integer configId = config.getConfigId();
-        String cacheKey = provider + ":" + (configId != null ? configId : "default");
+        var cacheKey = config.getProvider() + ":" + config.getConfigId();
 
         // 检查是否已有该配置的服务实例
         if (serviceCache.containsKey(cacheKey)) {
             return serviceCache.get(cacheKey);
         }
 
         // 创建新的API服务实例
-        try {
-            SttService service = createApiService(config);
-            if (service != null) {
-                serviceCache.put(cacheKey, service);
+        var service = createApiService(config);
+        serviceCache.put(cacheKey, service);
 
-                // 如果没有备选默认服务，将此服务设为备选
-                if (fallbackProvider == null) {
-                    fallbackProvider = cacheKey;
-                }
-                return service;
-            }
-        } catch (Exception e) {
-            logger.error("创建{}服务失败, configId={}", provider, configId, e);
+        // 如果没有备选默认服务，将此服务设为备选
+        if (fallbackProvider == null) {
+            fallbackProvider = cacheKey;
         }
 
-        return null;
+        return service;
     }
 
     /**
      * 根据配置创建API类型的STT服务
      */
-    private SttService createApiService(SysConfig config) {
-        if (config == null) {
-            return null;
-        }
-
-        String provider = config.getProvider();
-
-        // 根据提供商类型创建对应的服务实例
-        if ("tencent".equals(provider)) {
-            return new TencentSttService(config);
-        } else if ("aliyun".equals(provider)) {
-            return new AliyunSttService(config);
-        } else if ("funasr".equals(provider)) {
-            return new FunASRSttService(config);
-        } else if ("xfyun".equals(provider)) {
-            return new XfyunSttService(config);
-        }
-        // 可以添加其他服务提供商的支持
-
-        logger.warn("不支持的STT服务提供商: {}", provider);
-        return null;
+    private SttService createApiService(@Nonnull SysConfig config) {
+        return switch (config.getProvider()) {
+            case "tencent" -> new TencentSttService(config);
+            case "aliyun" -> new AliyunSttService(config);
+            case "funasr" -> new FunASRSttService(config);
+            case "xfyun" -> new XfyunSttService(config);
+            default -> {
+                var service = initializeVosk();
+                if (service == null) {
+                    // If vosk create failed, return fallback stt service
+                    if (fallbackProvider != null && serviceCache.containsKey(fallbackProvider)) {
+                        yield serviceCache.get(fallbackProvider);
+                    }
+                    throw new RuntimeException("Create vosk service failed");
+                }
+                yield service;
+            }
+        };
     }
 
     public void removeCache(SysConfig config) {