如何查找从 db.Query postgres 返回的行数

Question

我正在使用 goLang lib/pq 驱动程序并尝试从数据库中获取行。

rows, err := db.Query("select id, name from mytable limit 5")

我想要一个 if else 子句来检查结果集中是否有行，我这样做了：

if(!rows.Next()){
   log.Printf("no rows returned")
} else {
   log.Printf("rows returned")
}

但这总是返回给我少 1 条记录，我认为这是因为该if子句会跳过一条记录，因为一旦我删除if子句，我就会正确获取所有记录。如何在不执行另一个查询的情况下知道从选择查询返回的行数？

Answer 1

当您使用Rows对象时，没有任何辅助方法可以一步为您提供总行数。

一个简单但速度较慢的解决方案是使用增量变量来存储行数来迭代所有结果：

// error handling omitted
rows, _ := db.Query("SELECT * FROM table")
defer rows.Close()

counter := 0
for rows.Next() {
 // you can even scan+store the result if you need them later
 counter++
}

fmt.Println("we have", counter, "rows")

否则，如果您的目标只是“计算”行数，请使用QueryRow更专用的查询

// error handling omitted
var counter int

db.QueryRow("SELECT count(*) FROM table").Scan(&counter)

fmt.Println("we have", counter, "rows")