Go database/sql：获取查询结果行数的通用策略与考量

package main

import (
    "database/sql"
    "fmt"
    "log"

    _ "github.com/mattn/go-sqlite3" // 示例：使用 SQLite 驱动
)

// getTotalOrdersCount 通过独立的 COUNT(*) 查询获取订单总数
func getTotalOrdersCount(db *sql.DB, orderID int) (int, error) {
    var count int
    // 注意：这里的 WHERE 子句应与主查询保持一致
    row := db.QueryRow("SELECT COUNT(*) FROM orders WHERE id=?", orderID)
    err := row.Scan(&count)
    if err != nil {
        // 如果没有找到匹配的行，COUNT(*) 通常返回 0，不会返回 sql.ErrNoRows
        // 但为了健壮性，可以检查其他可能的错误
        return 0, fmt.Errorf("查询订单总数失败: %w", err)
    }
    return count, nil
}

// getOrders 主查询，获取订单详情
func getOrders(db *sql.DB, orderID int) (*sql.Rows, error) {
    rows, err := db.Query("SELECT id, item_name, quantity FROM orders WHERE id=?", orderID)
    if err != nil {
        return nil, fmt.Errorf("查询订单详情失败: %w", err)
    }
    return rows, nil
}

func main() {
    db, err := sql.Open("sqlite3", ":memory:")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    // 模拟创建表和插入数据
    sqlStmt := `
    CREATE TABLE orders (id INTEGER NOT NULL, item_name TEXT, quantity INTEGER);
    INSERT INTO orders(id, item_name, quantity) VALUES (1, 'Laptop', 1);
    INSERT INTO orders(id, item_name, quantity) VALUES (2, 'Mouse', 2);
    INSERT INTO orders(id, item_name, quantity) VALUES (1, 'Keyboard', 1);
    `
    _, err = db.Exec(sqlStmt)
    if err != nil {
        log.Fatalf("创建表或插入数据失败: %v", err)
    }

    targetOrderID := 1

    // 1. 获取总数
    count, err := getTotalOrdersCount(db, targetOrderID)
    if err != nil {
        log.Printf("获取订单ID %d 的总数失败: %v", targetOrderID, err)
    } else {
        fmt.Printf("通过 COUNT(*) 查询，订单ID %d 的总数为: %d\n", targetOrderID, count)
    }

    // 2. 获取详情 (此处仅为演示，实际应用中会进一步处理 rows)
    rows, err := getOrders(db, targetOrderID)
    if err != nil {
        log.Printf("获取订单ID %d 的详情失败: %v", targetOrderID, err)
    } else {
        defer rows.Close() // 确保关闭 rows
        fmt.Printf("主查询已执行，获取到订单详情。\n")
        // 实际应用中会遍历 rows.Next() 来处理数据
    }
}

package main

import (
    "database/sql"
    "fmt"
    "log"

    _ "github.com/mattn/go-sqlite3"
)

// Order 结构体用于存储查询结果
type Order struct {
    ID       int
    ItemName string
    Quantity int
}

// getOrdersAndCount 遍历 sql.Rows 游标并计数
func getOrdersAndCount(db *sql.DB, orderID int) ([]Order, int, error) {
    rows, err := db.Query("SELECT id, item_name, quantity FROM orders WHERE id=?", orderID)
    if err != nil {
        return nil, 0, fmt.Errorf("查询订单失败: %w", err)
    }
    defer rows.Close() // 确保在函数返回前关闭 rows

    var orders []Order
    count := 0
    for rows.Next() {
        var order Order
        // 扫描数据到 Order 结构体
        if err := rows.Scan(&order.ID, &order.ItemName, &order.Quantity); err != nil {
            return nil, 0, fmt.Errorf("扫描订单数据失败: %w", err)
        }
        orders = append(orders, order)
        count++ // 每次成功扫描一行，计数器加一
    }

    // 检查遍历过程中是否发生错误
    if err := rows.Err(); err != nil {
        return nil, 0, fmt.Errorf("遍历rows时发生错误: %w", err)
    }

    return orders, count, nil
}

func main() {
    db, err := sql.Open("sqlite3", ":memory:")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    // 模拟创建表和插入数据 (同上)
    sqlStmt := `
    CREATE TABLE orders (id INTEGER NOT NULL, item_name TEXT, quantity INTEGER);
    INSERT INTO orders(id, item_name, quantity) VALUES (1, 'Laptop', 1);
    INSERT INTO orders(id, item_name, quantity) VALUES (2, 'Mouse', 2);
    INSERT INTO orders(id, item_name, quantity) VALUES (1, 'Keyboard', 1);
    `
    _, err = db.Exec(sqlStmt)
    if err != nil {
        log.Fatalf("创建表或插入数据失败: %v", err)
    }

    targetOrderID := 1

    // 遍历并计数
    orders, count, err := getOrdersAndCount(db, targetOrderID)
    if err != nil {
        log.Printf("获取订单ID %d 的详情及总数失败: %v", targetOrderID, err)
    } else {
        fmt.Printf("通过遍历查询，订单ID %d 的总数为: %d\n", targetOrderID, count)
        for _, order := range orders {
            fmt.Printf("  订单: %+v\n", order)
        }
    }
}