如何使用Go实现数据库操作?

41 min read

使用Go实现数据库操作有多种方式,其中最常用的方法是使用Go官方提供的标准库database/sql。

下面是一个使用Go语言操作MySQL数据库的简单例子:

package main

import (
    "database/sql"
    "fmt"
    "log"
    _ "github.com/go-sql-driver/mysql"
)

func main() {
    db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/database_name")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    //插入数据
    stmt, err := db.Prepare("INSERT INTO user(name, age) VALUES(?, ?)")
    if err != nil {
        log.Fatal(err)
    }
    res, err := stmt.Exec("Alice", 18)
    if err != nil {
        log.Fatal(err)
    }
    id, err := res.LastInsertId()
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(id)

    //查询数据
    rows, err := db.Query("SELECT * FROM user WHERE age = ?", 18)
    if err != nil {
        log.Fatal(err)
    }
    defer rows.Close()
    for rows.Next() {
        var name string
        var age int
        err := rows.Scan(&name, &age)
        if err != nil {
            log.Fatal(err)
        }
        fmt.Printf("%s is %d years old\n", name, age)
    }
}

该例子首先使用sql.Open函数建立与MySQL数据库的连接,然后使用db.Prepare方法准备一个SQL语句模板,接着使用stmt.Exec方法执行SQL语句,并获取插入数据的ID值。接着使用db.Query方法查询数据,并通过rows.Scan方法将查询结果存储到变量中。最后使用defer关键字确保在函数结束时关闭数据库连接。

该示例所使用的MySQL驱动包可在Go官方仓库中找到:github.com/go-sql-driver/mysql。您可以使用go get命令下载并安装该包:

$ go get github.com/go-sql-driver/mysql