WordPress 中使用 COUNT(*) 代替 SQL_CALC_FOUND_ROWS 查询实现分页功能

Question

Wordpress 的列表使用了 SQL_CALC_FOUND_ROWS 查询，用于缓存查询结果，方便下次查询记录条数时使用，除了 SQL_CALC_FOUND_ROWS 查询，在 MySQL 中我们还可以使用 COUNT(*) 来获取总行数，经过对比，我们发现后者的速度比前者要快不少（有时候会快到 10 倍之多），我们可以使用 COUNT(*) 来代替 SQL_CALC_FOUND_ROWS 查询来获取总行数，这样既可以加快查询速度，又不会损失分页功能。

首先，我们使用 pre_get_posts Filter 钩子，直接在所有查询中设置 no_found_rows 禁用 SQL_CALC_FOUND_ROWS 查询。

add_filter('pre_get_posts', function (\WP_Query $wp_query){
  $wp_query->set('no_found_rows', true);
});

添加了以上代码，所有 WordPress 查询中的分页参数便缺失了，分页功能自然就失效了。我们需要使用 posts_clauses 钩子修改 WordPress 数据库查询，使用 COUNT(*) 查询来获取总行数，并设置分页所需要的参数。

add_filter('posts_clauses', function ($clauses, \WP_Query $wp_query){
  // 跳过单页面
  if ($wp_query->is_singular()) {
    return $clauses;
  }

  global $wpdb;

  // 查询是否设置了一下变量
  $where  = isset($clauses[ 'where' ]) ? $clauses[ 'where' ] : '';
  $join   = isset($clauses[ 'join' ]) ? $clauses[ 'join' ] : '';
  $distinct = isset($clauses[ 'distinct' ]) ? $clauses[ 'distinct' ] : '';

  // 构造并运行查询，将查询结果设置为我们要运行主查询的 “found_posts” 参数。
  $wp_query->found_posts = $wpdb->get_var("SELECT $distinct COUNT(*) FROM {$wpdb->posts} $join WHERE 1=1 $where");

  // 计算每页应该有多少文章。
  $posts_per_page = (! empty($wp_query->query_vars[ 'posts_per_page' ]) ? absint($wp_query->query_vars[ 'posts_per_page' ]) : absint(get_option('posts_per_page')));

  // 设置 max_num_pages 参数
  $wp_query->max_num_pages = ceil($wp_query->found_posts / $posts_per_page);

  // 返回 $clauses 自查询，以便运行主查询
  return $clauses;
}, 10, 2);

添加了以上代码之后，我们便把使用 COUNT(*) 查询替换 SQL_CALC_FOUND_ROWS 查询，如果您的 WordPress 文章总数比较多 (千条以上)，查询加速会比较明显。当然，如果平时使用的时候，并没有感觉到分页的打开速度慢，就不要多此一举添加上面的代码了，毕竟多一事不如少一事，如果没有问题，就不要进行优化。

参考文章：WordPress 数据量大列表查询慢 sql_calc_found_rows 惹的祸