MySQL查询分页，同时获取总记录数

最新推荐文章于 2023-09-19 14:42:15 发布

水友阿浪

最新推荐文章于 2023-09-19 14:42:15 发布

阅读量2.2w

点赞数 3

分类专栏： SQL 文章标签： mysql

SQL 专栏收录该内容

7 篇文章

订阅专栏

Mysql分页查询获取totalCount大幅提升性能的办法总结

MySQL查询分页，通常在MySQL中获取记录总数都会使用SELECT COUNT(*) FROM tblName之类的语句

这类语句的缺点是：记录集还需要单独的查询来获取，相当于查询两次，推荐如下写法：

SELECT SQL_CALC_FOUND_ROWS fldName1, fldName2 FROM tblName WHERE fldName3 = 1 LIMIT 10, OFFSET 20;

SELECT FOUND_ROWS();

虽然有两条SQL语句，但实际上只执行了一次数据库查询

做分页查询中，一般情况下需要两个sql，查当前页数据和查记录总条数；但后者的查询结果变化有时候并不大，而且count还占用了很大一部分的查询时间；主要是想用一种省时简便的方法查询符合条件的记录总数，

查询数据使用的sql为：

SELECT SUBSTRING_INDEX(`url`,'/',3) AS host,COUNT(*) AS count FROM `tab` WHERE `type`=4 GROUP BY host HAVING(count >= 5) ORDER BY count desc LIMIT 0,10

以下是网上查到的一些尝试过的方法（不过后来都感觉不太合适，所以，亮点在最后）：

方法一：一般情况下可以使用DISTINCT来查询总数

select count(DISTINCT SUBSTRING_INDEX(`url`,'/',3)) as c from tab where type = 4

但是查询数据中的sql 有 having 子句，这样得到的总数是没有经过条件筛选的。这个结果是错误的。

方法二：通过 SQL_CALC_FOUND_ROWS 选项忽略 LIMIT 子句，然后通过FOUND_ROWS()获得查询总数，那么sql改为：

SELECT SQL_CALC_FOUND_ROWS SUBSTRING_INDEX(`url`,'/',3) AS host,COUNT(*) AS count FROM `tab` WHERE `type`=4 GROUP BY host HAVING(count >= 5) ORDER BY count desc LIMIT 0,10

再通过 select FOUND_ROWS(); 获得总数

这样获得的总数没问题，但是由于分页程序需要先获得符合条件的总数，才能生成 page_list ，以及验证offset,和总页数等信息，所以不能先查询数据再得总数。

方法三：和上边的方法类似，只是第一次使用sql获得总数

先：

SELECT SUBSTRING_INDEX(`url`,'/',3) AS host,COUNT(*) AS count FROM `tab` WHERE `type`=4 GROUP BY host HAVING(count >= 5)

然后：

select FOUND_ROWS();

最后：

SELECT SUBSTRING_INDEX(`url`,'/',3) AS host,COUNT(*) AS count FROM `tab` WHERE `type`=4 GROUP BY host HAVING(count >= 5) ORDER BY count desc LIMIT 0,10

这个没有问题，也可以避免方法二中的问题，但是会返回全部的符合条件的数据，并且返回的数据没有任何作用，只是查询一次总数，所以也不可取。

方法四：使用子查询

select count(*) as count from (select SUBSTRING_INDEX(url,'/',3) as host,count(*) as c from tab where type=4 group by host having(c >= 5)) as temp

这个基本满足了需要，但是效率不是很高，如果子集很大的话，性能上是个问题。

以上4种方法，是网上查到的，但感觉都不是特别好或特别通用；后来经多方努力查找和学习，选用了自己写一套智能生产count查询语句的方案；

该方案采用了第三方包jsqlparser来解析sql结构并智能拼接count查询语句；

以我现在使用的java语言mybatis框架为示例：

框架中分页查询的count语句是这样产生的：

[java] view plain copy

String count_sql = dialect.getCountString(sql);

mybatis分页插件paginator中，mysql方言是这样实现的：

[html] view plain copy

/**
* 将sql转换为总记录数SQL
* @param sql SQL语句
* @return 总记录数的sql
*/
public String getCountString(String sql){
return "select count(1) from (" + sql + ") tmp_count";
}

当我看到这段源码的时候，有种想骂娘的感觉，mybatis官方提供的这种count写法，性能真不敢恭维！

于是乎亲自动手覆盖了如下方法：

[java] view plain copy

/**
* 优化父类的getCountString性能
*/
public String getCountString(String sql) {
try {
boolean queryCacheable = queryCachedFlag.get() != null && queryCachedFlag.get();
queryCachedFlag.remove();// 使用一次清理一次
return MySqlSmartCountUtil.getSmartCountSql(sql, queryCacheable);
} catch (JSQLParserException e) {
e.printStackTrace();
} catch (Exception e) {
e.printStackTrace();
}
return "select count(*) from (" + sql + ") tmp_count";
}

MySqlSmartCountUtil就是今天介绍的大神，是用jsqlparser写的智能生产count语句的工具类，采用了mysql查询缓存和获取count语句静态缓存的策略，大大提升了只能生产count语句的时间，和count查询的时间；源码分享给大家：

[java] view plain copy

public class MySqlSmartCountUtil {
// countSql缓存
private static HashMap<String, String> countSqlCache = new HashMap<String, String>();
private static HashMap<String, String> queryCacheableCountSqlCache = new HashMap<String, String>();
private static final List<SelectItem> countItem = new ArrayList<SelectItem>();
private static final List<SelectItem> sqlCachedCountItem = new ArrayList<SelectItem>();
static {
countItem.add(new SelectExpressionItem(new Column("count(*) as totalX")));
sqlCachedCountItem.add(new SelectExpressionItem(new Column("sql_cache count(*) as totalX")));
}
private static void cacheSmartCountSql(String srcSql, String countSql, boolean queryCacheable) {
if (queryCacheable)
queryCacheableCountSqlCache.put(srcSql, countSql);
else
countSqlCache.put(srcSql, countSql);
}
private static List<SelectItem> getCountItem(boolean queryCacheable) {
return queryCacheable ? sqlCachedCountItem : countItem;
}
private static void smartCountPlainSelect(PlainSelect plainSelect, boolean queryCacheable) throws JSQLParserException{
// 去掉orderby
OrderByUtil.removeOrderBy(plainSelect);
// 判断是否包含group by
if(GMUtil.isEmpty(plainSelect.getGroupByColumnReferences())){
plainSelect.setSelectItems(getCountItem(queryCacheable));
} else {
throw new JSQLParserException("不支持智能count的sql格式: GROUP BY ");
}
}
public static String getSmartCountSql(String srcSql, boolean queryCacheable) throws JSQLParserException {
// 直接从缓存中取
if(!queryCacheable && countSqlCache.containsKey(srcSql))
return countSqlCache.get(srcSql);
if(queryCacheable && queryCacheableCountSqlCache.containsKey(srcSql))
return queryCacheableCountSqlCache.get(srcSql);
Statement stmt = CCJSqlParserUtil.parse(srcSql);
Select select = (Select) stmt;
SelectBody selectBody = select.getSelectBody();
if (selectBody instanceof PlainSelect) {
PlainSelect plainSelect = ((PlainSelect) selectBody);
smartCountPlainSelect(plainSelect, queryCacheable);
} else if (selectBody instanceof SetOperationList) {
SetOperationList setOperationList = (SetOperationList) selectBody;
boolean isUnion = false;
for (SetOperation o : setOperationList.getOperations()) {
isUnion = (o.toString().contains("UNION"));
if (!isUnion)
break;
}
// union all 语句的智能count
if(isUnion){
for (PlainSelect ps : setOperationList.getPlainSelects()) {
smartCountPlainSelect(ps, false);// TODO 强制不允许缓存
}
String resultSql = "select sum(totalX) from (" + select.toString() + ") as t ";
cacheSmartCountSql(srcSql, resultSql, false);// TODO 强制不允许缓存
return resultSql;
} else {
throw new JSQLParserException("不支持智能count的sql格式");
}
} else {
throw new JSQLParserException("不支持智能count的sql格式");
}
cacheSmartCountSql(srcSql, select.toString(), queryCacheable);
return select.toString();
}
}