[代码技巧] wordpress搜索频率限制

class WXS_SearchFrequencyLimit {

    // 配置参数可根据需要修改

    private $search_limit_count = 3; // 允许的最大搜索次数

    private $time_window_in_minutes = 1; // 时间窗口（分钟）

    private $block_time_in_seconds = 10; // 超过限制后封禁时间（秒）

    private $block_message = '您的搜索频率过高，请在 %d 秒后再试。'; // 封禁提示消息



    public function __construct() {

        add_filter( 'pre_get_posts', array( $this, 'check_search_limit' ) );

        if ( ! is_admin() ) {

            add_action( 'wp_loaded', array( $this, 'check_block_status_on_any_page' ) );

        }

    }



    private function get_visitor_ip() {

        if ( ! empty( $_SERVER['HTTP_CLIENT_IP'] ) ) {

            $ip = $_SERVER['HTTP_CLIENT_IP'];

        } elseif ( ! empty( $_SERVER['HTTP_X_FORWARDED_FOR'] ) ) {

            $ip = $_SERVER['HTTP_X_FORWARDED_FOR'];

        } else {

            $ip = $_SERVER['REMOTE_ADDR'];

        }

        return sanitize_text_field( $ip );

    }



    private function set_block_cache($key, $expiration) {

        $expire_time = time() + $expiration;

        wp_cache_set($key, $expire_time, 'search_limit', $expiration);

    }

    

    private function get_remaining_block_time($key) {

        $expire_time = wp_cache_get($key, 'search_limit');

        if ($expire_time === false) {

            return 0;

        }

        $remaining = $expire_time - time();

        return $remaining > 0 ? $remaining : 0;

    }

    public function check_block_status_on_any_page() {

        $visitor_ip = $this->get_visitor_ip();

        $cache_key = 'search_block_' . $visitor_ip;

        $remaining_time = $this->get_remaining_block_time($cache_key);

        if ($remaining_time > 0 && !is_search()) {

            $message = sprintf($this->block_message, $remaining_time);

            add_action('wp_body_open', function() use ($message) {

                echo '<div style="background-color: #fff3cd; color: #856404; padding: 15px; text-align: center; border: 1px solid #ffeeba; margin-bottom: 20px;">';

                echo esc_html($message);

                echo '</div>';

            });

        }

    }



    public function check_search_limit($query) {

        if (!is_admin() && $query->is_search && $query->is_main_query()) {

            $visitor_ip = $this->get_visitor_ip();

            $limit = $this->search_limit_count;

            $window_seconds = $this->time_window_in_minutes * 60;

            $block_time = $this->block_time_in_seconds;

            $cache_key = 'search_block_' . $visitor_ip;

            $records_key = 'search_records_' . $visitor_ip;

            $remaining_time = $this->get_remaining_block_time($cache_key);

            if ($remaining_time > 0) {

                wp_die(

                    sprintf($this->block_message, $remaining_time),

                    '搜索频率限制', //标题

                    array('response' => 429)

                );

            }



            $search_records = wp_cache_get($records_key, 'search_limit', false, $found);

            if (!$found) {

                $search_records = array();

            }

            $current_time = time();



            $search_records = array_filter($search_records, function($timestamp) use ($current_time, $window_seconds) {

                return $current_time - $timestamp < $window_seconds;

            });



            if (count($search_records) >= $limit) {

                $this->set_block_cache($cache_key, $block_time);

                wp_cache_delete($records_key, 'search_limit');

                

                wp_die(

                    sprintf($this->block_message, $block_time),

                    '搜索频率限制',

                    array('response' => 429)

                );

            } else {

                $search_records[] = $current_time;

                wp_cache_set($records_key, $search_records, 'search_limit', $window_seconds);

            }

        }



        return $query;

    }

}

new WXS_SearchFrequencyLimit();