#夏日挑战赛# 云原生指南之 Go 如何连接使用 MySQL 数据库原创精华

宇宙之一粟

发布于 2022-7-25 14:14

浏览

3收藏

本文正在参加星光计划3.0–夏日挑战赛

1 引言

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

#夏日挑战赛# 云原生指南之 Go 如何连接使用 MySQL 数据库-鸿蒙开发者社区

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

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

2 如何在 Go 语言中使用 MySQL

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

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

创建数据库对象：

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

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

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

导入 MySQL 驱动

_ "github.com/go-sql-driver/mysql"

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

defer.Close()

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

3 连接数据库并查询 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(), 也是没有问题的。

4 创建数据库表

我们在 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)

5 插入数据

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)

6 查询所有数据

我们在数据库中再插入一条数据，使用如下命令 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

7 条件查询

可以使用 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)

8 删除数据

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)

9 修改数据

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)

10 总结

本文展示了如何在 Go 语言中使用 MySQL，并给出了详细关键步骤的说明。并在查询 MySQL 版本号的程序中验证了 Go 语言连接数据库成功，然后介绍了 Go 语言如何进行增删改查，可以看到增删改查的代码并不复杂，最主要就是掌握 MySQL 的 SQL 语句的写法，然后代码框架差不多，也可以把这些函数封装成一个工具，方便今后自己的使用。下一篇文章再见！

分类

标签