如何使用MySQL C++驱动将结果集重置到首行前的初始位置

MySQL C++ Connector默认返回的ResultSet是只进游标类型，仅支持单向调用next()遍历，遍历完成后无法回退到开头，你可以根据业务场景选择以下两种高效方案实现重复遍历：

方案1：使用可滚动游标直接重置游标位置 #

如果查询结果数据量较大，不想全量加载到内存，可以修改结果集类型为可滚动，调用内置方法直接重置游标位置：

实现逻辑 #

1. 执行查询前设置结果集类型为sql::ResultSet::TYPE_SCROLL_INSENSITIVE
2. 遍历完成后调用beforeFirst()方法，即可将游标重置到首行之前的初始状态

示例代码 #

/* 省略连接创建等前置逻辑 */
sql::PreparedStatement *pstmt;
sql::ResultSet *res;

// 设置当前连接使用可滚动结果集
con->setResultSetType(sql::ResultSet::TYPE_SCROLL_INSENSITIVE);

pstmt = con->prepareStatement("SELECT * FROM foobar;");
res = pstmt->executeQuery();

// 第一次遍历
while (res->next())
{
    std::cout << res->getInt("id") << " " << res->getString("foo_bar_col") << std::endl;
}

// 重置游标到初始状态
res->beforeFirst();

// 可直接开始第二次遍历
while (res->next())
{
    // 你的业务逻辑
}

方案2：本地缓存结果集（性能更优，适合中小数据量） #

如果查询结果在内存可承受范围内，直接把结果集转存到本地容器的性能更高，无需依赖数据库驱动的游标特性，也不会占用数据库侧的游标资源：

示例代码 #

// 定义行结构匹配你的查询字段
struct FoobarRow {
    int id;
    std::string foo_bar_col;
};
std::vector<FoobarRow> cache;

/* 省略连接、查询逻辑 */
res = pstmt->executeQuery();

// 第一次遍历全量写入缓存
while (res->next())
{
    cache.push_back({
        res->getInt("id"),
        res->getString("foo_bar_col")
    });
}

// 后续需要遍历直接用本地缓存，无需再操作ResultSet
for (const auto& row : cache) {
    std::cout << row.id << " " << row.foo_bar_col << std::endl;
}

选型建议 #

• 数据量较大、无法全量载入内存时选择可滚动游标方案，注意需要确认你的数据库连接配置没有禁止可滚动游标
• 数据量在内存可承受范围内时优先选择本地缓存方案，后续遍历速度更快，也避免了数据库侧的额外资源占用
• 不推荐反复执行executeQuery重新查询，会产生大量不必要的数据库交互开销

内容的提问来源于stack exchange，提问作者Siedler