Go 语言如何连接并操作 MySQL 数据库

头像
宇宙之一粟
     广东
    阅读 8 分钟
    头图

    1 引言

    MySQL 是一个基于结构化查询语言（SQL）的开源关系数据库管理系统。它是一种关系数据库，可将数据组织到一个或多个表中，其中数据相互关联。MySQL 是行业领先的开源数据库管理系统。它是一个多用户、多线程的数据库管理系统。

    数据库驱动程序：数据库驱动程序实现了用于数据库连接的协议。驱动程序就像一个适配器，连接到特定数据库的通用接口。

    Go 有 sql 包，它提供了一个围绕 SQL（或类似 SQL）数据库的通用接口。 sql 包必须与数据库驱动程序一起使用。该软件包提供自动连接池。每次查询数据库时，我们都在使用应用程序启动时设置的连接池中的连接。连接被重用。

    2 如何在 Go 语言中使用 MySQL

    1. 启动 MySQL 服务器并使用以下命令安装 go-mysql 驱动程序。

    image.png

    go get -u github.com/go-sql-driver/mysql
    1. 创建数据库对象：

    使用 sql.Open 创建一个数据库对象。相反，没有与 MySQL 建立连接，它只创建一个可以稍后使用的数据库对象。

    db, err := sql.Open("mysql", "<user>:<password>@tcp(127.0.0.1:3306)/<database-name>")

    使用 sql.Open，我们打开一个由其数据库驱动程序名称和驱动程序特定数据源名称指定的数据库，通常至少由数据库名称和连接信息组成。它不与数据库建立任何连接，也不验证驱动程序连接参数。相反，它只是为以后使用准备数据库抽象。当第一次需要时，将延迟建立与底层数据存储的第一个实际连接。

    1. 导入 MySQL 驱动
    _ "github.com/go-sql-driver/mysql"

    当导入带有空白标识符前缀 _ 的包时，将调用包的 init 函数。该函数注册驱动程序。

    1. defer.Close()

    Close 将连接返回到连接池。如果 sql.DB 的生命周期不应超出函数范围，则推迟 db.Close 是常用的。

    2.1 连接数据库并查询 MySQL 版本号

    我们来编写一个程序返回 MySQL 的版本。版本由执行 SELECT VERSION() 语句确定，在 MySQL 的终端中运行本地版本号:

    mysql> SELECT VERSION();
+-----------+
| VERSION() |
+-----------+
| 5.7.25    |
+-----------+
1 row in set (0.00 sec)

    然后我们按照上述的步骤，编写一个 Go 语言的代码查看数据库的版本号：

    package main

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

func main() {

    db, err := sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/DB_TEST")
    db.Ping()
    defer db.Close()

    if err != nil {
        fmt.Println("数据库连接失败！")
        log.Fatalln(err)
    }

    var version string

    err2 := db.QueryRow("SELECT VERSION()").Scan(&version)

    if err2 != nil {
        log.Fatal(err2)
    }

    fmt.Println(version)
}

    建好 go.mod 文件：

    $ go mod init main.go                     
go: creating new go.mod: module main.go
go: to add module requirements and sums:
        go mod tidy

    运行该代码：

    $ go run main.go         
5.7.25

    与我们在 MySQL 终端中得到的结果一致，说明数据库连接成功，SELECT VERSION(), 也是没有问题的。

    3 如何在 Go 语言中操作 MySQL

    在上面的内容中，我们成功连接了 MySQL 数据库，并成功打印出 MySQL 的版本号，接下来就来介绍如何使用 Go 语言操作数据库。

    3.1 创建数据库表

    我们在 DB_TEST 数据库中新建一个 user 数据库，包含主键 id 和名字 name，可以使用如下语句：

    _, err2 := db.Exec("CREATE TABLE user(id INT NOT NULL , name VARCHAR(20), PRIMARY KEY(ID));")

    我们可以先在 MySQL 终端中查看我们的数据库表。

    mysql> use DB_TEST;
Reading table information for completion of table and column names
You can turn off this feature to get a quicker startup with -A

Database changed
mysql> show tables;
+-------------------+
| Tables_in_db_test |
+-------------------+
| table_name        |
+-------------------+
1 row in set (0.00 sec)

    编写的 Go 语言的代码如下：

    package main

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

func main() {

    db, err := sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/DB_TEST")
    db.Ping()
    defer db.Close()

    if err != nil {
        fmt.Println("数据库连接失败！")
        log.Fatalln(err)
    }

    _, err2 := db.Exec("CREATE TABLE user(id INT NOT NULL , name VARCHAR(20), PRIMARY KEY(ID));")
    if err2 != nil {
        log.Fatal(err2)
    }

    fmt.Print("Successfully Created\n")
}

    运行该程序：

    Successfully Created

    再来查看我们的数据库表 show tables;，会看到多了一个 user 表，说明创建数据库表成功：

    mysql> show tables;
+-------------------+
| Tables_in_db_test |
+-------------------+
| table_name        |
| user              |
+-------------------+
2 rows in set (0.01 sec)

    3.2 插入数据

    package main

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

func main() {

    db, err := sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/DB_TEST")
    db.Ping()
    defer db.Close()

    if err != nil {
        fmt.Println("数据库连接失败！")
        log.Fatalln(err)
    }

    _, err2 := db.Query("INSERT INTO user VALUES(1, 'Wade')")
    if err2 != nil {
        log.Fatal(err2)
    }

    fmt.Print("Successfully Inserted\n")
}

    回到数据库终端，查看一下 user 表 select * from user;，可以看到我们刚刚插入成功的一条数据：

    mysql> select * from user;
+----+------+
| id | name |
+----+------+
|  1 | Wade |
+----+------+
1 row in set (0.00 sec)

    3.3 查询所有数据

    我们在数据库中再插入一条数据，使用如下命令 INSERT INTO user VALUES(2, 'Kyrie');：

    mysql> INSERT INTO user VALUES(2, 'Kyrie');
Query OK, 1 row affected (0.00 sec)

    然后使用 db.Query("SELECT * FROM user") 查询 user 数据库表中的所有数据：

    package main

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

func main() {

    db, err := sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/DB_TEST")
    db.Ping()
    defer db.Close()

    if err != nil {
        fmt.Println("数据库连接失败！")
        log.Fatalln(err)
    }

    result, err2 := db.Query("SELECT * FROM user")
    if err2 != nil {
        log.Fatal(err2)
    }

    for result.Next() {

        var id int
        var name string

        err = result.Scan(&id, &name)

        if err != nil {
            panic(err)
        }

        fmt.Printf("Id: %d, Name: %s\n", id, name)
    }
}

    执行结果：

    Id: 1, Name: Wade
Id: 2, Name: Kyrie

    3.4 条件查询

    可以使用 db.Query("SELECT * FROM user WHERE id = ?", mid) 进行条件查询：

    result, err2 := db.Query("SELECT * FROM user WHERE id = ?", mid)

    Go 代码如下：

    package main

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

func main() {

    db, err := sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/DB_TEST")
    db.Ping()
    defer db.Close()

    if err != nil {
        fmt.Println("数据库连接失败！")
        log.Fatalln(err)
    }

    var mid int = 1

    result, err2 := db.Query("SELECT * FROM user WHERE id = ?", mid)
    if err2 != nil {
        log.Fatal(err2)
    }

    for result.Next() {

        var id int
        var name string

        err = result.Scan(&id, &name)

        if err != nil {
            log.Fatal(err)
        }

        fmt.Printf("Id: %d, Name: %s\n", id, name)
    }
}

    ？ 是一个占位符，填充了 mid 变量中的值。在后台，db.Query 实际上准备、执行和关闭准备好的语句。

    执行该代码：

    Id: 1, Name: Wade

    在 MySQL 终端中运行：

    mysql> SELECT * FROM user WHERE id = 1;
+----+------+
| id | name |
+----+------+
|  1 | Wade |
+----+------+
1 row in set (0.00 sec)

    3.5 删除数据

    RowsAffected 返回受更新、插入或删除语句影响的行数。

    在代码示例中，我们使用 DELETE SQL 语句删除一条数据。然后我们用 RowsAffected 打印删除的行数。

    package main

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

func main() {

    db, err := sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/DB_TEST")
    db.Ping()
    defer db.Close()

    if err != nil {
        fmt.Println("数据库连接失败！")
        log.Fatalln(err)
    }

    sql := "DELETE FROM user WHERE id = 1"
    res, err2 := db.Exec(sql)

    if err2 != nil {
        panic(err2.Error())
    }

    affectedRows, err := res.RowsAffected()

    if err != nil {
        log.Fatal(err)
    }

    fmt.Printf("The statement affected %d rows\n", affectedRows)
}
    The statement affected 1 rows

    此时，再回到 MySQL 终端查看：

    mysql> SELECT * FROM user;
+----+-------+
| id | name  |
+----+-------+
|  2 | Kyrie |
+----+-------+
1 row in set (0.00 sec)

    3.6 修改数据

    package main

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

func main() {

    db, err := sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/DB_TEST")
    db.Ping()
    defer db.Close()

    if err != nil {
        fmt.Println("数据库连接失败！")
        log.Fatalln(err)
    }

    sql := "update user set name = ? WHERE id = ?"
    res, err2 := db.Exec(sql, "Yuzhou1su", 2)

    if err2 != nil {
        panic(err2.Error())
    }

    affectedRows, err := res.RowsAffected()

    if err != nil {
        log.Fatal(err)
    }

    fmt.Printf("Update Success, The statement affected %d rows\n", affectedRows)
}

    运行结果：

    Update Success, The statement affected 1 rows

    可以看到 id = 2 的数据 name 修改为 Yuzhou1su ,数据库中验证：

    mysql> SELECT * FROM user;
+----+-----------+
| id | name      |
+----+-----------+
|  2 | Yuzhou1su |
+----+-----------+
1 row in set (0.01 sec)

    4 总结

    本文展示了如何在 Go 语言中使用 MySQL，并给出了详细关键步骤的说明。并在查询 MySQL 版本号的程序中验证了 Go 语言连接数据库成功。

    然后介绍了 Go 语言如何进行增删改查，可以看到增删改查的代码并不复杂，最主要就是掌握 MySQL 的 SQL 语句的写法，然后代码框架差不多，也可以把这些函数封装成一个工具，方便今后自己的使用。

    另外 Go 语言还有很多优秀的 go MySQL 驱动和框架，大家可以自行探索与学习。地址如下： Database Drivers - Awesome Go

    image.png

    本篇文章就到此结束了，下一篇文章再见~

    希望本文能对你有所帮助，如果喜欢本文，可以点个赞或关注。

    这里是宇宙之一粟，下一篇文章见！
    宇宙古今无有穷期，一生不过须臾，当思奋争。

    本文参与了思否技术征文，欢迎正在阅读的你也加入。

    go数据库mysql思否技术征文
    宇宙之一粟
    82 声望9 粉丝

    混迹于江湖，江湖却没有我的影子

    « 上一篇
    如何在 Go 代码中运行 C 语言代码
    下一篇 »
    人生的喜悦、不快与成长，都在那一篇篇的文字中得到记录 | 2022 年终总结

    引用和评论

    被 4 篇内容引用
    推荐阅读
    头像
    在 Golang 中执行 Shell 命令

    宇宙之一粟阅读 3.1k

    头像
    Open WebUI：开源AI交互平台的全面解析

    DBLens赞 5阅读 1.7k

    头像
    被 Manus 带火的 MCP 是什么｜一文看懂

    Bytebase赞 3阅读 3.4k

    头像
    一文掌握 MCP 上下文协议：从理论到实践

    陈明勇赞 5阅读 1.5k

    头像
    MySQL × 向量数据库：大模型时代的黄金组合实战指南

    DBLens赞 5阅读 1.7k

    头像
    53 倍性能提升！TiDB 全局索引如何优化分区表查询？

    PingCAP赞 2阅读 820

    头像
    腾讯 tRPC-Go 教学——（5）filter、context 和日志组件

    amc赞 2阅读 1.8k

    0 条评论
    得票最新