Go database/sql：高效获取查询结果行数的策略

package main

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

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

// OrderService 结构体，包含数据库事务
type OrderService struct{}

// GetOrdersWithCount 演示如何使用 COUNT(*) 获取总行数
func (me *OrderService) GetOrdersWithCount(orderTx *sql.Tx, orderId int) ([]Order, int, error) {
    // 1. 执行 COUNT(*) 查询获取总行数
    var totalRows int
    countQuery := "SELECT COUNT(*) FROM orders WHERE id = ?"
    err := orderTx.QueryRow(countQuery, orderId).Scan(&totalRows)
    if err != nil {
        return nil, 0, fmt.Errorf("查询订单总数失败: %w", err)
    }

    // 2. 执行实际的数据查询
    dataQuery := "SELECT id, item_name, quantity FROM orders WHERE id = ?"
    rows, err := orderTx.Query(dataQuery, orderId)
    if err != nil {
        return nil, 0, fmt.Errorf("查询订单数据失败: %w", err)
    }
    defer rows.Close() // 确保关闭 rows

    var orders []Order
    for rows.Next() {
        var 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)
    }

    if err := rows.Err(); err != nil {
        return nil, 0, fmt.Errorf("遍历订单数据时发生错误: %w", err)
    }

    return orders, totalRows, nil
}

// Order 结构体用于映射数据库表
type Order struct {
    ID       int
    ItemName string
    Quantity int
}

func main() {
    db, err := sql.Open("sqlite3", ":memory:") // 使用内存数据库进行示例
    if err != nil {
        log.Fatalf("打开数据库失败: %v", err)
    }
    defer db.Close()

    // 创建表并插入数据
    _, err = db.Exec(`
        CREATE TABLE orders (
            id INTEGER PRIMARY KEY,
            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); -- 故意插入重复ID，以便测试
    `)
    if err != nil {
        log.Fatalf("初始化数据库失败: %v", err)
    }

    tx, err := db.Begin()
    if err != nil {
        log.Fatalf("开启事务失败: %v", err)
    }
    defer tx.Rollback() // 确保事务回滚或提交

    service := &OrderService{}
    // 查询 id=1 的订单
    orders, totalCount, err := service.GetOrdersWithCount(tx, 1)
    if err != nil {
        log.Fatalf("获取订单失败: %v", err)
    }

    fmt.Printf("查询到 %d 条订单数据 (总计 %d 条符合条件的记录):\n", len(orders), totalCount)
    for _, order := range orders {
        fmt.Printf("  ID: %d, Item: %s, Quantity: %d\n", order.ID, order.ItemName, order.Quantity)
    }

    if err := tx.Commit(); err != nil {
        log.Fatalf("提交事务失败: %v", err)
    }
}

package main

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

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

// OrderService 结构体
// ... (与上面示例相同)

// GetOrdersByIterating 演示如何通过遍历游标获取行数
func (me *OrderService) GetOrdersByIterating(orderTx *sql.Tx, orderId int) ([]Order, error) {
    query := "SELECT id, item_name, quantity FROM orders WHERE id = ?"
    rows, err := orderTx.Query(query, orderId)
    if err != nil {
        return nil, fmt.Errorf("查询订单数据失败: %w", err)
    }
    defer rows.Close() // 确保关闭 rows

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

    if err := rows.Err(); err != nil {
        return nil, fmt.Errorf("遍历订单数据时发生错误: %w", err)
    }

    log.Printf("通过遍历游标，实际获取到 %d 条订单。", rowCount)
    return orders, nil
}

// Order 结构体
// ... (与上面示例相同)

func main() {
    db, err := sql.Open("sqlite3", ":memory:")
    if err != nil {
        log.Fatalf("打开数据库失败: %v", err)
    }
    defer db.Close()

    // 初始化数据库
    _, err = db.Exec(`
        CREATE TABLE orders (
            id INTEGER PRIMARY KEY,
            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);
    `)
    if err != nil {
        log.Fatalf("初始化数据库失败: %v", err)
    }

    tx, err := db.Begin()
    if err != nil {
        log.Fatalf("开启事务失败: %v", err)
    }
    defer tx.Rollback()

    service := &OrderService{}
    // 查询 id=1 的订单
    orders, err := service.GetOrdersByIterating(tx, 1)
    if err != nil {
        log.Fatalf("获取订单失败: %v", err)
    }

    fmt.Printf("查询到 %d 条订单数据:\n", len(orders)) // len(orders) 即为实际行数
    for _, order := range orders {
        fmt.Printf("  ID: %d, Item: %s, Quantity: %d\n", order.ID, order.ItemName, order.Quantity)
    }

    if err := tx.Commit(); err != nil {
        log.Fatalf("提交事务失败: %v", err)
    }
}