JAVA MYSQL sql_calc_found_rows和found_rows()实践
一、背景
1.百万级数据库，数据量持续增加。每张数据表的字段数大于50（时间字段，分组字段，指标字段）
2.JDBCTemplate，java，mysql
二、问题描述
通过分析接口返回数据响应时间过长（通过某个分组字段搜索数据，响应时间长达30s）。
三、检查问题
检查代码，发现代码中运行了两句SQL语句，一句通过select查询数据，一句通过select count(1)来获取返回数据的总条数。
通过navicat查询语句对应的执行时间。
SELECT eventtime,smart_card_id,uevt_1000 FROM analytics_vhsession_user_event_info_day_201901 WHERE eventtime>='2019-01-01 00:00:00' AND eventtime<'2019-01-31 23:59:59' AND smart_card_id = '0382205801' ORDER BY eventtime asc LIMIT 0,1000
> OK
> 时间: 10.603s
SELECT count(1) FROM analytics_vhsession_user_event_info_day_201901 WHERE eventtime>='2019-01-01 00:00:00' AND eventtime<'2019-01-31 23:59:59' AND smart_card_id = '0382205801'
> OK
> 时间: 11.13s
同样耗时10s+，所以想办法从select count(1)入手，减少SQL执行时间以达到减少响应时间的目的。
四、查询资料
通过查询资料，可以通过使用sql_calc_found_rows和found_rows()替代select count(1)。
通过navicat查询语句对应的执行时间。
SELECT sql_calc_found_rows eventtime,smart_card_id,uevt_1000 FROM analytics_vhsession_user_event_info_day_201901 WHERE eventtime>='2019-01-01 00:00:00' AND eventtime<'2019-01-31 23:59:59' AND smart_card_id = '0382205801' ORDER BY eventtime asc LIMIT 0,1000
> OK
> 时间: 11.606s
SELECT FOUND_ROWS()
> OK
> 时间: 0.004s
相较之前的方案，响应时间可以减少10s以上，是一个值得尝试的方案。
五、优化尝试
根据之前的测试结果尝试进行代码优化，使用jdbcTemplate来调用两次query(),一次获取数据，一次获取对应的总条数。
//select sql_calc_found_rows
String selectSQL = "select sql_calc_found_rows ...";
List<Map<String, Object>> data = jdbcTemplate.queryForList(selectSQL);
//select found_rows()
String selectTotalCountSQL = "select found_rows()";
Long totalCount = jdbcTemplate.queryForObject(selectTotalCountSQL, Long.class);
但是在实际测试中遇到了jdbcTemplate.query("select found_rows()")返回的总条数与实际的总条数不一致的情况。 通过查询相应的资料，在一篇分享文档发现一点端倪，以下为资料原文：
we do this by opening a connection, running two SELECT queries, then closing the connection. This allows us to achieve the desired result that we need.
sql_calc_found_rows和found_rows()需要两句SQL在同一会话中，才能保证select found_rows()返回的总条数是上一句select sql_calc_found_rows对应的总条数
查看jdbcTemplate.query()底层代码实现。
public <T> T execute(StatementCallback<T> action) throws DataAccessException {
Assert.notNull(action, "Callback object must not be null");
Connection con = DataSourceUtils.getConnection(getDataSource());
Statement stmt = null;
try {
Connection conToUse = con;
if (this.nativeJdbcExtractor != null &&
this.nativeJdbcExtractor.isNativeConnectionNecessaryForNativeStatements()) {
conToUse = this.nativeJdbcExtractor.getNativeConnection(con);
}
stmt = conToUse.createStatement();
applyStatementSettings(stmt);
Statement stmtToUse = stmt;
if (this.nativeJdbcExtractor != null) {
stmtToUse = this.nativeJdbcExtractor.getNativeStatement(stmt);
}
T result = action.doInStatement(stmtToUse);
handleWarnings(stmt);
return result;
}
catch (SQLException ex) {
// Release Connection early, to avoid potential connection pool deadlock
// in the case when the exception translator hasn't been initialized yet.
JdbcUtils.closeStatement(stmt);
stmt = null;
DataSourceUtils.releaseConnection(con, getDataSource());
con = null;
throw getExceptionTranslator().translate("StatementCallback", getSql(action), ex);
}
finally {
JdbcUtils.closeStatement(stmt);
DataSourceUtils.releaseConnection(con, getDataSource());
}
}
jdbcTemplate每次执行query()都会从连接池中获取连接
Connection con = DataSourceUtils.getConnection(getDataSource())
执行完成后释放连接
DataSourceUtils.releaseConnection(con, getDataSource());
不能保证两次query()在一个会话中(同一个Connection)。
六、优化实践
优化方案：不使用JDBCTemplate中的query()方法，自己实现具体逻辑。通过DataSourceUtils.getConnection(jdbcTemplate.getDataSource())获取会话，使用Statement来执行两次SQL后，再通过DataSourceUtils.releaseConnection(conn, jdbcTemplate.getDataSource());释放会话，保证两句SQL在同一会话中。
public PagedArrayList getDataAndTotalCount(String sql){
Connection conn = null;
Statement statement = null;
ResultSet rs = null;
ResultSet rs1 = null;
long totalCount = 0L;
PagedArrayList data = new PagedArrayList();
try {
conn = DataSourceUtils.getConnection(jdbcTemplate.getDataSource());
conn.setAutoCommit(true);
statement = conn.createStatement();
rs = statement.executeQuery(sql);
ResultSetMetaData md = rs.getMetaData(); //获得结果集结构信息,元数据
int columnCount = md.getColumnCount(); //获得列数
while (rs.next()) {
Map<String,Object> rowData = new HashMap<String,Object>();
for (int i = 1; i <= columnCount; i++) {
rowData.put(md.getColumnName(i), rs.getObject(i));
}
data.add(rowData);
}
String totalCountSQL = "select found_rows() AS total_count";
rs1 = statement.executeQuery(totalCountSQL);
while (rs1.next()){
totalCount = rs1.getLong("total_count");
}
data.totalCount = totalCount;
} catch (Exception e) {
slf4jLogger.error("getDataAndTotalCount() error:", e);
} finally {
//关闭资源
JdbcUtils.closeResultSet(rs);
JdbcUtils.closeResultSet(rs1);
JdbcUtils.closeStatement(statement);
//释放资源
DataSourceUtils.releaseConnection(conn, jdbcTemplate.getDataSource());
}
return data;
}
七、参考文档
