wpbridge/includes/Cache/HealthChecker.php

<?php
/**
 * 健康检查器
 *
 * @package WPBridge
 */

namespace WPBridge\Cache;

use WPBridge\UpdateSource\SourceModel;
use WPBridge\UpdateSource\Handlers\HealthStatus;
use WPBridge\Core\Logger;

// 防止直接访问
if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

/**
 * 健康检查器类
 */
class HealthChecker {

	/**
	 * 缓存管理器
	 *
	 * @var CacheManager
	 */
	private CacheManager $cache;

	/**
	 * 健康状态缓存 TTL（1 小时）
	 *
	 * @var int
	 */
	const HEALTH_CACHE_TTL = 3600;

	/**
	 * 失败源冷却时间（30 分钟）
	 *
	 * @var int
	 */
	const FAILED_COOLDOWN = 1800;

	/**
	 * 构造函数
	 */
	public function __construct() {
		$this->cache = new CacheManager();
	}

	/**
	 * 检查源健康状态
	 *
	 * @param SourceModel $source 源模型
	 * @param bool        $force  是否强制检查
	 * @return HealthStatus
	 */
	public function check( SourceModel $source, bool $force = false ): HealthStatus {
		$cache_key = 'health_' . $source->id;

		// 检查缓存
		if ( ! $force ) {
			$cached = $this->cache->get( $cache_key );
			if ( false !== $cached && $cached instanceof HealthStatus ) {
				return $cached;
			}
		}

		// 检查是否在冷却期
		if ( ! $force && $this->is_in_cooldown( $source->id ) ) {
			return HealthStatus::failed( __( '源在冷却期内', 'wpbridge' ) );
		}

		// 执行健康检查
		$handler = $source->get_handler();

		if ( null === $handler ) {
			$status = HealthStatus::failed( __( '无法获取处理器', 'wpbridge' ) );
		} else {
			$status = $handler->test_connection();
		}

		// 缓存结果
		$ttl = $status->is_healthy() ? self::HEALTH_CACHE_TTL : self::FAILED_COOLDOWN;
		$this->cache->set( $cache_key, $status, $ttl );

		// 如果失败，设置冷却
		if ( ! $status->is_available() ) {
			$this->set_cooldown( $source->id );
		}

		Logger::debug(
			'健康检查完成',
			[
				'source' => $source->id,
				'status' => $status->status,
				'time'   => $status->response_time,
			]
		);

		return $status;
	}

	/**
	 * 批量检查源健康状态
	 *
	 * @param SourceModel[] $sources 源列表
	 * @return array<string, HealthStatus>
	 */
	public function check_all( array $sources ): array {
		$results = [];

		foreach ( $sources as $source ) {
			$results[ $source->id ] = $this->check( $source );
		}

		return $results;
	}

	/**
	 * 获取源健康状态（仅从缓存）
	 *
	 * @param string $source_id 源 ID
	 * @return HealthStatus|null
	 */
	public function get_status( string $source_id ): ?HealthStatus {
		$cached = $this->cache->get( 'health_' . $source_id );

		if ( false !== $cached && $cached instanceof HealthStatus ) {
			return $cached;
		}

		return null;
	}

	/**
	 * 检查源是否在冷却期
	 *
	 * @param string $source_id 源 ID
	 * @return bool
	 */
	public function is_in_cooldown( string $source_id ): bool {
		$cooldown = $this->cache->get( 'cooldown_' . $source_id );
		return false !== $cooldown;
	}

	/**
	 * 设置源冷却
	 *
	 * @param string $source_id 源 ID
	 */
	private function set_cooldown( string $source_id ): void {
		$this->cache->set( 'cooldown_' . $source_id, time(), self::FAILED_COOLDOWN );

		Logger::info(
			'源进入冷却期',
			[
				'source'   => $source_id,
				'duration' => self::FAILED_COOLDOWN,
			]
		);
	}

	/**
	 * 清除源冷却
	 *
	 * @param string $source_id 源 ID
	 */
	public function clear_cooldown( string $source_id ): void {
		$this->cache->delete( 'cooldown_' . $source_id );
	}

	/**
	 * 清除所有健康状态缓存
	 */
	public function clear_all(): void {
		global $wpdb;

		$wpdb->query(
			$wpdb->prepare(
				"DELETE FROM {$wpdb->options} WHERE option_name LIKE %s",
				$wpdb->esc_like( '_transient_wpbridge_health_' ) . '%'
			)
		);
		$wpdb->query(
			$wpdb->prepare(
				"DELETE FROM {$wpdb->options} WHERE option_name LIKE %s",
				$wpdb->esc_like( '_transient_wpbridge_cooldown_' ) . '%'
			)
		);
	}
}