Go database/sql 精确查询与首行获取：判断零、一或多行结果

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
}

// 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.