Golang中使用database/sql获取查询结果总行数

花韻仙語

发布时间：2025-10-08 10:51:01

959人浏览过

来源于php中文网

原创

golang中使用database/sql获取查询结果总行数

本文旨在帮助开发者在使用 Golang 的 database/sql 包与数据库（例如 PostgreSQL 或 MySQL）交互时，如何有效地判断 SELECT 查询返回的行数，并处理零行、单行或多行结果的情况。我们将提供一个通用的函数示例，展示如何获取查询结果的第一行，并在必要时检测是否存在多行数据。

在使用 Golang 进行数据库操作时，经常需要根据 SELECT 查询返回的行数来执行不同的逻辑。database/sql 包提供了方便的接口，但直接获取总行数并非其主要功能。本教程将介绍如何结合 database/sql 包和数据库驱动（如 github.com/go-sql-driver/mysql）来实现这一需求。

封装查询函数

为了方便复用，我们可以封装一个查询函数，该函数执行 SQL 查询并返回结果集。以下是一个基于 MySQL 的示例，但可以很容易地修改为适用于 PostgreSQL 或其他数据库。

import (
    "database/sql"
    "fmt"
    _ "github.com/go-sql-driver/mysql" // 导入 MySQL 驱动
)

var db *sql.DB // 全局数据库连接对象

// init 函数用于初始化数据库连接
func init() {
    // 数据库连接信息，请根据实际情况修改
    dsn := "user:password@tcp(127.0.0.1:3306)/database_name"
    var err error
    db, err = sql.Open("mysql", dsn)
    if err != nil {
        panic(err)
    }

    // 尝试连接数据库
    err = db.Ping()
    if err != nil {
        panic(err)
    }

    fmt.Println("Successfully connected to the database!")
}

// execQuery 执行 SQL 查询并返回结果集
func execQuery(SQL string, args ...interface{}) (*sql.Rows, bool) {
    rows, err := db.Query(SQL, args...)
    if err != nil {
        fmt.Println("Query error:", err)
        return nil, false
    }
    return rows, true
}

代码解释：

立即学习“go语言免费学习笔记（深入）”；

Originality AI

专门为网络出版商设计的抄袭和AI检测工具

下载

导入必要的包： database/sql 用于数据库操作，github.com/go-sql-driver/mysql 是 MySQL 驱动。注意，需要使用 _ 导入驱动，以便其 init 函数被执行，注册驱动到 database/sql 包。
全局数据库连接对象： db 是一个全局变量，用于存储数据库连接。
init 函数： 该函数在程序启动时自动执行，用于初始化数据库连接。你需要根据实际情况修改 dsn (Data Source Name) 来匹配你的数据库连接信息。
execQuery 函数： 该函数接受 SQL 查询语句和参数，执行查询并返回 sql.Rows 结果集和一个布尔值，指示查询是否成功。

使用查询函数获取数据

以下是如何使用 execQuery 函数来获取数据并判断行数的示例：

func main() {
    var name, age string
    rows, isSucceed := execQuery("SELECT `name`, `age` FROM `test` WHERE `id` = ?", "123")
    if !isSucceed {
        fmt.Println("Query failed.")
        return
    }
    defer rows.Close() // 确保关闭结果集

    rowCount := 0
    for rows.Next() {
        err := rows.Scan(&name, &age)
        if err != nil {
            fmt.Println("Scan error:", err)
            return
        }
        fmt.Printf("Name: %s, Age: %s\n", name, age)
        rowCount++
    }

    if rowCount == 0 {
        fmt.Println("No rows found.")
    } else if rowCount == 1 {
        fmt.Println("One row found.")
    } else {
        fmt.Printf("%d rows found.\n", rowCount)
    }

    err := rows.Err()
    if err != nil {
        fmt.Println("Rows error:", err)
    }
}

代码解释：

立即学习“go语言免费学习笔记（深入）”；