深入理解pgx库中CollectRows方法的使用误区

在使用pgx库进行数据库操作时，开发者可能会遇到CollectRows方法返回空切片的问题。本文将通过一个典型场景分析这一现象的原因，并提供正确的解决方案。

问题现象分析

当开发者尝试使用pgx.CollectRows方法从查询结果中收集数据时，可能会发现返回的切片为空，即使原始查询结果rows对象显示包含有效数据。这种情况通常发生在以下场景：

rows, _ := conn.Query(ctx, "SELECT id, name FROM menus")
accounts, _ := pgx.CollectRows(rows, pgx.RowToStructByName[MenuDto])
// accounts为空切片

根本原因探究

问题的核心在于pgx库中Rows对象的状态管理。Rows对象在被读取后会进入"closed"状态，这可以从调试输出中观察到：

Rows: &{..., closed:true, ...}

关键点在于：

1. Rows对象一旦被完全读取，就会自动关闭
2. 已关闭的Rows对象无法再次读取
3. 许多pgx辅助方法(如ForEachRow)会隐式读取并关闭Rows

典型错误模式

开发者常见的错误模式是混合使用不同的结果集处理方法：

// 错误示例：混合使用ForEachRow和CollectRows
rows, _ := conn.Query(ctx, query)
pgx.ForEachRow(rows, ...)  // 这里会读取并关闭rows
pgx.CollectRows(rows, ...) // 此时rows已关闭，返回空结果

正确解决方案

方案一：单一方法处理结果集

选择一种结果集处理方法并坚持使用：

// 使用CollectRows
rows, _ := conn.Query(ctx, query)
menus, _ := pgx.CollectRows(rows, pgx.RowToStructByName[MenuDto])

// 或使用ForEachRow
rows, _ := conn.Query(ctx, query)
var menus []MenuDto
pgx.ForEachRow(rows, []any{&id, &name}, func() error {
    menus = append(menus, MenuDto{Id: id, Name: name})
    return nil
})

方案二：手动迭代处理

对于需要更精细控制的情况，可以手动迭代：

rows, _ := conn.Query(ctx, query)
defer rows.Close()

var menus []MenuDto
for rows.Next() {
    var m MenuDto
    _ = rows.Scan(&m.Id, &m.Name)
    menus = append(menus, m)
}

最佳实践建议

1. 理解每种结果集处理方法的行为特性
2. 避免对同一个Rows对象多次调用处理方法
3. 在复杂场景下，优先考虑手动迭代方式
4. 使用defer rows.Close()确保资源释放

通过正确理解pgx库中结果集处理机制，开发者可以避免这类问题，编写出更健壮的数据库操作代码。