QueryRow 的局限性
在go语言的database/sql包中，db.queryrow()函数常用于预期只返回零行或单行结果的场景。它的设计目的是简化单行结果的获取。然而，queryrow()存在一个关键局限：当sql查询实际返回多于一行数据时，它并不会报错。它只会静默地读取第一行数据，然后丢弃其余行。这意味着，如果你的业务逻辑要求在返回多行数据时抛出错误，或者需要明确知道返回了多少行（零行、单行还是多行），queryrow()将无法满足需求，因为它无法区分“只找到一行”和“找到多行但只取了第一行”这两种情况。
db.Query 与 *sql.Rows：精细控制的基础
为了实现对查询结果行数的精确控制，我们需要使用db.Query()函数。db.Query()执行SQL查询后，会返回一个*sql.Rows对象和一个error。*sql.Rows代表了查询结果集的一个游标，允许我们逐行遍历数据。
使用*sql.Rows的关键在于以下几点：
- rows.Next(): 每次调用尝试移动到结果集的下一行。如果成功移动到下一行，返回true；如果没有更多行或发生错误，返回false。
- rows.Scan(): 当rows.Next()返回true后，调用rows.Scan()将当前行的数据扫描到Go变量中。
- defer rows.Close(): 至关重要！*sql.Rows是一个数据库资源，必须在使用完毕后关闭。通常使用defer rows.Close()来确保无论函数如何退出，资源都能被释放。
- rows.Err(): 在循环结束后，检查rows.Err()可以捕获在遍历过程中可能发生的任何错误（例如网络中断）。
通过结合这些方法，我们可以实现对查询结果的细粒度控制，包括判断行数、获取首行，以及在多行结果出现时触发错误。
构建通用查询辅助函数
为了简化查询执行和初步的错误处理，我们可以创建一个辅助函数来封装db.Query()的调用。
package main
import (
"database/sql"
"fmt"
_ "github.com/go-sql-driver/mysql" // 导入MySQL驱动
// _ "github.com/lib/pq" // 如果使用PostgreSQL，导入PostgreSQL驱动
)
// 假设db变量已通过sql.Open()初始化
var db *sql.DB
// executeQuery 是一个通用的查询辅助函数，用于执行SQL并返回*sql.Rows或错误
func executeQuery(query string, args ...interface{}) (*sql.Rows, error) {
rows, err := db.Query(query, args...)
if err != nil {
return nil, fmt.Errorf("执行查询失败: %w", err)
}
return rows, nil
}这个executeQuery函数仅仅是db.Query()的一个简单包装，它将查询执行的错误标准化，并返回*sql.Rows对象，将结果集的遍历和处理留给调用者。
处理查询结果：判断行数与获取首行
现在，我们将创建一个更具体的函数，它利用executeQuery返回的*sql.Rows来满足原始需求：获取首行数据，并判断结果是零行、单行还是多行（多行时抛出错误）。
// User 结构体用于映射数据库表中的一行数据
type User struct {
ID int
Name string
Age int
}
// GetFirstUserAndCountResult 用于封装查询结果的状态
type GetFirstUserAndCountResult struct {
User *User // 如果找到，存储第一行用户数据
RowCount int // 实际返回的行数
IsMultiple bool // 是否返回了多于一行数据
Err error // 过程中发生的错误
}
// getFirstUserAndCount 是一个通用函数，用于查询用户，并返回首行数据、实际行数及是否多行
func getFirstUserAndCount(userID int) *GetFirstUserAndCountResult {
result := &GetFirstUserAndCountResult{}
query := "SELECT id, name, age FROM test WHERE id = ?" // 假设表名为'test'
rows, err := executeQuery(query, userID)
if err != nil {
result.Err = err
return result
}
defer rows.Close() // 确保rows被关闭
var user User
foundFirstRow := false
// 尝试获取第一行
if rows.Next() {
result.RowCount++
if err := rows.Scan(&user.ID, &user.Name, &user.Age); err != nil {
result.Err = fmt.Errorf("扫描第一行失败: %w", err)
return result
}
result.User = &user // 成功扫描到第一行
foundFirstRow = true
}
// 检查是否有更多行 (即第二行及以后)
if rows.Next() {
result.RowCount++ // 发现第二行
result.IsMultiple = true
// 如果发现多行，继续遍历完所有剩余行，确保rows.Close()能正常工作
for rows.Next() {
result.RowCount++
}
}
// 检查遍历过程中是否有错误
if err := rows.Err(); err != nil {
result.Err = fmt.Errorf("遍历结果集发生错误: %w", err)
return result
}
// 根据foundFirstRow和IsMultiple设置最终的错误信息
if result. 
