wpbridge/includes/Cache/CacheManager.php

<?php
/**
 * 缓存管理器
 *
 * @package WPBridge
 */

namespace WPBridge\Cache;

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

/**
 * 缓存管理器类
 */
class CacheManager {

    /**
     * 缓存组名
     *
     * @var string
     */
    const CACHE_GROUP = 'wpbridge';

    /**
     * 默认 TTL（12 小时）
     *
     * @var int
     */
    const DEFAULT_TTL = 43200;

    /**
     * 获取缓存
     *
     * @param string $key 缓存键
     * @return mixed|false
     */
    public function get( string $key ) {
        // 优先使用对象缓存
        if ( wp_using_ext_object_cache() ) {
            $value = wp_cache_get( $key, self::CACHE_GROUP );
            if ( false !== $value ) {
                return $value;
            }
        }

        // 降级到 transient
        return get_transient( 'wpbridge_' . $key );
    }

    /**
     * 设置缓存
     *
     * @param string $key   缓存键
     * @param mixed  $value 缓存值
     * @param int    $ttl   过期时间（秒）
     * @return bool
     */
    public function set( string $key, $value, int $ttl = self::DEFAULT_TTL ): bool {
        // 使用对象缓存
        if ( wp_using_ext_object_cache() ) {
            wp_cache_set( $key, $value, self::CACHE_GROUP, $ttl );
        }

        // 同时存储到 transient
        return set_transient( 'wpbridge_' . $key, $value, $ttl );
    }

    /**
     * 删除缓存
     *
     * @param string $key 缓存键
     * @return bool
     */
    public function delete( string $key ): bool {
        if ( wp_using_ext_object_cache() ) {
            wp_cache_delete( $key, self::CACHE_GROUP );
        }

        return delete_transient( 'wpbridge_' . $key );
    }

    /**
     * 获取带过期缓存兜底的值
     *
     * @param string   $key      缓存键
     * @param callable $callback 获取新值的回调
     * @param int      $ttl      正常 TTL
     * @param int      $stale_ttl 过期缓存可用时间
     * @return mixed
     */
    public function get_with_fallback( string $key, callable $callback, int $ttl = self::DEFAULT_TTL, int $stale_ttl = 604800 ) {
        // 尝试获取正常缓存
        $value = $this->get( $key );

        if ( false !== $value ) {
            return $value;
        }

        // 尝试获取新值
        try {
            $new_value = $callback();

            if ( null !== $new_value && false !== $new_value ) {
                $this->set( $key, $new_value, $ttl );

                // 同时存储一份过期缓存备份
                $this->set( $key . '_stale', $new_value, $stale_ttl );

                return $new_value;
            }
        } catch ( \Exception $e ) {
            // 获取新值失败，尝试使用过期缓存
        }

        // 尝试使用过期缓存
        $stale_value = $this->get( $key . '_stale' );

        if ( false !== $stale_value ) {
            return $stale_value;
        }

        return false;
    }

    /**
     * 清除所有缓存
     */
    public function flush(): void {
        global $wpdb;

        // 清除 transient（使用 prepare 防止 SQL 注入）
        $wpdb->query(
            $wpdb->prepare(
                "DELETE FROM {$wpdb->options} WHERE option_name LIKE %s",
                $wpdb->esc_like( '_transient_wpbridge_' ) . '%'
            )
        );
        $wpdb->query(
            $wpdb->prepare(
                "DELETE FROM {$wpdb->options} WHERE option_name LIKE %s",
                $wpdb->esc_like( '_transient_timeout_wpbridge_' ) . '%'
            )
        );

        // 清除对象缓存组（不使用 flush 避免影响其他插件）
        if ( wp_using_ext_object_cache() ) {
            wp_cache_delete( 'wpbridge', 'wpbridge' );
        }
    }

    /**
     * 获取缓存统计
     *
     * @return array
     */
    public function get_stats(): array {
        global $wpdb;

        $count = $wpdb->get_var(
            $wpdb->prepare(
                "SELECT COUNT(*) FROM {$wpdb->options} WHERE option_name LIKE %s",
                $wpdb->esc_like( '_transient_wpbridge_' ) . '%'
            )
        );

        return [
            'transient_count'    => (int) $count,
            'object_cache'       => wp_using_ext_object_cache(),
            'object_cache_type'  => $this->get_object_cache_type(),
        ];
    }

    /**
     * 获取对象缓存类型
     *
     * @return string
     */
    private function get_object_cache_type(): string {
        if ( ! wp_using_ext_object_cache() ) {
            return 'none';
        }

        global $wp_object_cache;

        if ( isset( $wp_object_cache->redis ) || class_exists( 'Redis' ) ) {
            return 'redis';
        }

        if ( isset( $wp_object_cache->mc ) || class_exists( 'Memcached' ) ) {
            return 'memcached';
        }

        return 'unknown';
    }
}