当前位置：首页 > 文章列表 > Golang > Go教程 > SQL查询数据库行数的几种方法

SQL查询数据库行数的几种方法

2025-10-19 14:09:31 0浏览收藏

学习知识要善于思考，思考，再思考！今天golang学习网小编就给大家带来《使用SQL包查询数据库并获取行数方法》，以下内容主要包含等知识点，如果你正在学习或准备学习Golang，就都不要错过本文啦~让我们一起来看看吧，能帮助到你就更好了！

使用 database/sql 包查询数据库并获取返回行数

本文旨在帮助开发者在使用 Go 语言的 database/sql 包连接数据库（如 PostgreSQL 和 MySQL）时，如何准确判断查询结果返回的行数，并针对不同行数的情况进行处理。我们将提供一个通用的函数示例，该函数能够返回查询结果的第一行，并告知是否存在多行数据，以便开发者在需要精确控制返回行数的情况下使用。

在使用 Go 语言的 database/sql 包与数据库交互时，经常需要知道查询返回的行数，并根据行数进行不同的处理。例如，当期望只返回一行数据时，如果实际返回了多行，可能需要抛出错误或进行其他特殊处理。

以下提供一种通用的方法，用于执行 SQL 查询并获取返回的行数。该方法的核心在于利用 sql.Rows 结构体的 Next() 方法进行迭代，并结合计数器来判断行数。

package main

import (
    "database/sql"
    "fmt"
    _ "github.com/go-sql-driver/mysql" // 导入 MySQL 驱动，根据实际数据库选择
    "log"
)

// execQuery 执行 SQL 查询并返回结果集和执行状态
func execQuery(db *sql.DB, SQL string, args ...interface{}) (rows *sql.Rows, isSucceed bool) {
    rows, err := db.Query(SQL, args...)
    if err != nil {
        log.Printf("Query failed: %v", err)
        return nil, false
    }
    return rows, true
}

func main() {
    // 数据库连接信息，根据实际情况修改
    dbUser := "user"
    dbPass := "password"
    dbHost := "localhost"
    dbPort := "3306"
    dbName := "database_name"

    // 构建连接字符串
    dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s", dbUser, dbPass, dbHost, dbPort, dbName)

    // 连接数据库
    db, err := sql.Open("mysql", dsn)
    if err != nil {
        log.Fatalf("Failed to connect to database: %v", err)
    }
    defer db.Close()

    // 测试连接
    if err := db.Ping(); err != nil {
        log.Fatalf("Failed to ping database: %v", err)
    }

    // 示例 SQL 查询
    query := "SELECT id, name, age FROM test WHERE id = ?"
    targetID := "123"

    // 执行查询
    rows, isSucceed := execQuery(db, query, targetID)
    if !isSucceed {
        fmt.Println("Query failed")
        return
    }
    defer rows.Close()

    // 处理查询结果
    var id, age int
    var name string
    rowCount := 0

    for rows.Next() {
        err := rows.Scan(&id, &name, &age)
        if err != nil {
            log.Printf("Failed to scan row: %v", err)
            continue
        }

        if rowCount == 0 {
            fmt.Printf("First row: id=%d, name=%s, age=%d\n", id, name, age)
        }
        rowCount++
    }

    if err := rows.Err(); err != nil {
        log.Printf("Error during rows iteration: %v", err)
    }

    // 判断行数
    if rowCount == 0 {
        fmt.Println("No rows returned")
    } else if rowCount == 1 {
        fmt.Println("One row returned")
    } else {
        fmt.Printf("%d rows returned\n", rowCount)
    }
}

代码解释：

execQuery 函数： 封装了 db.Query 方法，用于执行 SQL 查询。该函数返回 sql.Rows 结果集和一个布尔值，指示查询是否成功。
主函数 main：
- 连接数据库：使用 sql.Open 函数连接到 MySQL 数据库。需要根据实际情况修改数据库连接信息。
- 执行查询：调用 execQuery 函数执行 SQL 查询。
- 处理结果集：使用 rows.Next() 方法迭代结果集中的每一行。
- 扫描数据：使用 rows.Scan 方法将每一行的数据扫描到对应的变量中。
- 计数器：使用 rowCount 变量记录返回的行数。
- 行数判断：根据 rowCount 的值判断返回的行数，并进行相应的处理。

注意事项：