From b6206a5d5c60a893046c6fbd5841a104bb551d60 Mon Sep 17 00:00:00 2001
From: Wenaixi <cep@wenxi.dev>
Date: Mon, 30 Mar 2026 01:14:49 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E5=AE=9A=E6=97=B6?=
 =?UTF-8?q?=E5=81=A5=E5=BA=B7=E6=A3=80=E6=9F=A5=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

实现对供应商节点进行周期性健康测试，失败1次立即标记unhealthy

Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-opencode)

Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
---
 src/providers/provider-pool-manager.js | 59 ++++++++++++++++++++++++++
 src/services/api-server.js             | 17 ++++++++
 2 files changed, 76 insertions(+)

diff --git a/src/providers/provider-pool-manager.js b/src/providers/provider-pool-manager.js
index 4a323fe..7a81a22 100644
--- a/src/providers/provider-pool-manager.js
+++ b/src/providers/provider-pool-manager.js
@@ -1743,6 +1743,65 @@ export class ProviderPoolManager {
         }
     }
 
+    /**
+     * Performs scheduled health checks on all providers.
+     * This method is designed to be called periodically to proactively check provider health.
+     * It respects provider-level isDisabled and checkHealth flags.
+     */
+    async performScheduledHealthChecks() {
+        const scheduledConfig = globalThis.CONFIG?.SCHEDULED_HEALTH_CHECK;
+        
+        // Check if scheduled health checks are disabled
+        if (scheduledConfig?.disabled) {
+            this._log('debug', '[ScheduledHealthCheck] Scheduled health checks are disabled via configuration');
+            return;
+        }
+        
+        this._log('info', '[ScheduledHealthCheck] Starting scheduled health checks on all providers...');
+        
+        for (const providerType in this.providerStatus) {
+            for (const provider of this.providerStatus[providerType]) {
+                // Skip manually disabled providers
+                if (provider.config.isDisabled === true) {
+                    this._log('debug', `[ScheduledHealthCheck] Skipping ${provider.config.uuid} (${providerType}): manually disabled`);
+                    continue;
+                }
+                
+                // Skip providers with checkHealth disabled
+                if (provider.config.checkHealth === false) {
+                    this._log('debug', `[ScheduledHealthCheck] Skipping ${provider.config.uuid} (${providerType}): checkHealth is false`);
+                    continue;
+                }
+                
+                try {
+                    // Perform health check with forceCheck=false (respects checkHealth flag)
+                    const result = await this._checkProviderHealth(providerType, provider.config, false);
+                    
+                    // result === null means checkHealth was false (already handled above) or not implemented
+                    if (result === null) {
+                        this._log('debug', `[ScheduledHealthCheck] Health check for ${provider.config.uuid} (${providerType}) skipped: not implemented`);
+                        continue;
+                    }
+                    
+                    if (!result.success) {
+                        // Provider is unhealthy
+                        this._log('warn', `[ScheduledHealthCheck] Health check failed for ${provider.config.uuid} (${providerType}): ${result.errorMessage || 'Provider is not responding correctly.'}`);
+                        this.markProviderUnhealthyImmediately(providerType, provider.config, result.errorMessage);
+                    } else {
+                        // Provider is healthy
+                        this._log('debug', `[ScheduledHealthCheck] Health check passed for ${provider.config.uuid} (${providerType})`);
+                        this.markProviderHealthy(providerType, provider.config, true, result.modelName);
+                    }
+                } catch (error) {
+                    this._log('error', `[ScheduledHealthCheck] Health check exception for ${provider.config.uuid} (${providerType}): ${error.message}`);
+                    this.markProviderUnhealthyImmediately(providerType, provider.config, error.message);
+                }
+            }
+        }
+        
+        this._log('info', '[ScheduledHealthCheck] Completed');
+    }
+
     /**
      * 构建健康检查请求（返回多种格式用于重试）
      * @private
diff --git a/src/services/api-server.js b/src/services/api-server.js
index f3b6360..bf3e66f 100644
--- a/src/services/api-server.js
+++ b/src/services/api-server.js
@@ -361,6 +361,23 @@ async function startServer() {
             poolManager.performHealthChecks(true);
         }
 
+        // 定时健康检查
+        const scheduledConfig = CONFIG.SCHEDULED_HEALTH_CHECK;
+        if (scheduledConfig?.enabled) {
+            const interval = scheduledConfig.interval || CONFIG.CRON_NEAR_MINUTES * 60 * 1000;
+            
+            // 设置定时任务
+            setInterval(async () => {
+                try {
+                    await poolManager.performScheduledHealthChecks();
+                } catch (error) {
+                    logger.error('[ScheduledHealthCheck] Error:', error);
+                }
+            }, interval);
+            
+            logger.info(`[ScheduledHealthCheck] Scheduled every ${interval}ms`);
+        }
+
         // 如果是子进程，通知主进程已就绪
         if (IS_WORKER_PROCESS) {
             sendToMaster({ type: 'ready', pid: process.pid });