专栏首页cwl_JavaGO-操作数据库

GO-操作数据库

第 4 章:操作数据库

Go 语言中的 database/sql 包定义了对数据库的一系列操作。database/sql/driver包定义了应被数据库驱动实现的接口,这些接口会被 sql 包使用。但是 Go 语言没有提供任何官方的数据库驱动,所以我们需要导入第三方的数据库驱动。不过我们连接数据库之后对数据库操作的大部分代码都使用 sql 包。

4.1 获取数据库连接

  1. 创建一个 db.go 文件,导入 database/sql 包以及第三方驱动包
import (
	"database/sql"
	_ "github.com/go-sql-driver/mysql")
  1. 定义两个全局变量
var (
Db *sql.DB
err error
)
  • DB 结构体的说明
  1. 创建 init 函数,在函数体中调用 sql 包的 Open 函数获取连接
func init() {
	Db,
	err = sql.Open("mysql", "root:root@tcp(localhost:3306)/test")
	if err != nil {
		panic(err.Error())
	}
}
  • Open 函数的说明
  • 参数 dataSourceName 的格式: 数据库用户名:数据库密码@[tcp(localhost:3306)]/数据库名
  • Open 函数可能只是验证其参数,而不创建与数据库的连接。如果要检查数据源的名称是否合法,应调用返回值的 Ping 方法。
  • 返回的 DB 可以安全的被多个 go 程同时使用,并会维护自身的闲置连接池。这样一来,Open 函数只需调用一次。很少需要关闭 DB

4.2 增删改操作

  1. 在连接的 test 数据库中创建一个 users 表
CREATE TABLE users(
	id INT PRIMARY KEY AUTO_INCREMENT,
	username VARCHAR(100) UNIQUE NOT NULL,
	PASSWORD VARCHAR(100) NOT NULL,
	email VARCHAR(100)
)
  1. 向 users 表中插入一条记录
  • 创建 user.go 文件,文件中编写一下代码
package model
import (
	"fmt"
	"webproject/utils"
)
type User struct {
	ID int
	Username string
	Password string
	Email string
}
func(user * User) AddUser() error {
	// 写 sql 语句
	sqlStr: = "insert into users(username , password , email) values(?,?,?)" //预编译
	stmt,
	err: = utils.Db.Prepare(sqlStr)
	if err != nil {
		fmt.Println("预编译出现异常:", err)
		return err
	}
	//执行
	_,
	erro: = stmt.Exec(user.Username, user.Password, user.Email)
	if erro != nil {
		fmt.Println("执行出现异常:", erro)
		return erro
	}
	return nil
}
  • Prepare 方法的说明
  • Stmt 结构体及它的方法的说明
  • DB 结构体中也有 Exec、Query 和 QueryRow 方法
  • 所以还可以通过调用 DB 的 Exec 方法添加用户(以下只展示了 user.go 文件修改之后的 AddUser 方法)
func(user * User) AddUser() error {
	// 写 sql 语句
	sqlStr: = "insert into users(username , password , email) 
	values( ? , ? , ? )
	"//执行
	_,
	erro: = utils.Db.Exec(sqlStr, user.Username, user.Password,
		user.Email)
	if erro != nil {
		fmt.Println("执行出现异常:", erro)
		return erro
	}
	return nil
}
  • 请同学们仿照着添加操作将删除和更新完成

4.3 单元测试

4.3.1 简介

顾名思义,单元测试( unit test),就是一种为验证单元的正确性而设置的自动化测试,一个单元就是程序中的一个模块化部分。一般来说,一个单元通常会与程序中的一个函数或者一个方法相对应,但这并不是必须的。Go 的单元测试需要用到 testing 包 以及 go test 命令,而且对测试文件也有以下要求

  1. 被测试的源文件和测试文件必须位于同一个包下
  2. 测试文件必须要以_test.go 结尾
  • 虽然 Go 对测试文件_test.go 的前缀没有强制要求,不过一般我们都设置为被测试的文件的文件名,如:要对 user.go 进行测试,那么测试文件的名字我们通常设置为 user_test.go
  1. 测试文件中的测试函数为 TestXxx(t *testing.T)
  • 其中 Xxx 的首字母必须是大写的英文字母
  • 函数参数必须是 test.T 的指针类型(如果是 Benchmark 测试则参数是testing.B 的指针类型)

4.3.2 测试代码:

  1. 在 user_test.go 中测试添加员工的方法
func TestAddUser(t * testing.T) {
	fmt.Println("测试添加用户:")
	user: = & User {
		Username: "admin3",
		Password: "123456",
		Email: "admin3@atguigu.com",
	}
	//将 user 添加到数据库中
	user.AddUser()
}
  • testing 包的说明
  1. 如果一个测试函数的函数名的不是以 Test 开头,那么在使用 go test 命令时默认不会执行,不过我们可以设置该函数时一个子测试函数,可以在其他测试函数里通过 t.Run 方法来执行子测试函数,具体代码如下
func TestUser(t * testing.T) {
	t.Run("正在测试添加用户:", testAddUser)
	t.Run("正在测试获取一个用户:", testGetUserById)
} //子测试函数
func testAddUser(t * testing.T) {
	fmt.Println("测试添加用户:")
	user: = & User {
		Username: "admin5",
		Password: "123456",
		Email: "admin5@atguigu.com",
	}
	//将 user 添加到数据库中
	user.AddUser()
} //子测试函数
func testGetUserByID(t * testing.T) {
	u: = & User {}
	user,
	_: = u.GetUserByID(1)
	fmt.Println("用户的信息是:", * user)
}
  1. 我们还可以通过 TestMain(m *testing.M)函数在测试之前和之后做一些其他的操作 a) 测试文件中有 TestMain 函数时,执行 go test 命令将直接运行 TestMain函数,不直接运行测试函数,只有在 TestMain 函数中执行 m.Run()时才会执行测试函数 b) 如果想查看测试的详细过程,可以使用 go test -v 命令 c) 具体代码
func TestMain(m * testing.M) {
	fmt.Println("开始测试") m.Run()
}
func TestUser(t * testing.T) {
	t.Run("正在测试添加用户:", testAddUser)
}
func testAddUser(t * testing.T) {
	fmt.Println("测试添加用户:")
	user: = & User {
		Username: "admin5",
		Password: "123456",
		Email: "admin5@atguigu.com",
	}
	//将 user 添加到数据库中
	user.AddUser()
}
  • Main 的说明

4.4 获取一条记录

  • 根据用户的 id 从数据库中获取一条记录
  • 代码实现:
func(user * User) GetUserByID(userId int)( * User, error) {
	//写 sql 语句
	sqlStr: = "select id , username , password , email from users where id = ?"
	//执行 sql
	row: = utils.Db.QueryRow(sqlStr, userId)
	//声明三个变量
	var username string
	var password string
	var email string
	//将各个字段中的值读到以上三个变量中
	err: = row.Scan( & userId, & username, & password, & email)
	if err != nil {
		return nil, err
	}
	//将三个变量的值赋给 User 结构体
	u: = & User {
		ID: userId,
		Username: username,
		Password: password,
		Email: email,
	}
	return u,
	nil
}
  • Row 结构体及 San 方法的说明

4.5 获取多条记录

  • 从数据库中查询出所有的记录
  • 代码实现:
func(user * User) GetUsers()([] * User, error) {
	//写 sql 语句
	sqlStr: = "select id , username , password , email from users"
	//执行 sql
	rows,
	err: = utils.Db.Query(sqlStr)
	if err != nil {
		return nil, err
	}
	//定义一个 User 切片
	var users[] * User
	//遍历
	for rows.Next() {
		//声明四个个变量
		var userID int
		var username string
		var password string
		var email string
		//将各个字段中的值读到以上三个变量中
		err: = rows.Scan( & userID, & username, & password, & email)
		if err != nil {
			return nil, err
		}
		//将三个变量的值赋给 User 结构体
		u: = & User {
			ID: userID,
			Username: username,
			Password: password,
			Email: email,
		}
		//将 u 添加到 users 切片中
		users = append(users, u)
	}
	return users,
	nil
}
  • Rows 结构体说明
  • Next 方法和 Scan 方法说明

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 速读原著-TCP/IP(SLIP:串行线路IP)

    RFC 893[Leffler and Karels 1984]描述了另一种用于以太网的封装格式,称作尾部封装(trailer encapsulation)。这...

    cwl_java
  • 前端基础-JavaScript构造函数

    JavaScript 语言使用构造函数作为对象的模板。 所谓 ”构造函数”,就是一个普通的函数,只不过我们专门用它来生成对象(new 构造函数),这样使用的函...

    cwl_java
  • 前端基础-JQuery操作属性

    cwl_java
  • 使用 GoLang 获取 TLS 的 Client Hello Info

    TLS(Transport Layer Security)是一个保证信息安全的应用层协议。它的前身是 SSL(Secure Socket Layer)。它是一套...

    饶文津
  • golang简单tls协议用法完整示例

    本文实例讲述了golang简单tls协议用法。分享给大家供大家参考,具体如下: 生成私钥: openssl genrsa -out key.pem 2048 ...

    李海彬
  • golang简单tls协议用法完整示例

    本文实例讲述了golang简单tls协议用法。分享给大家供大家参考,具体如下: 生成私钥: openssl genrsa -out key.pem 2048 生...

    李海彬
  • 如何批量删除k8s资源对象

    在云平台开发、中间件容器化时,经常会遇到批量删除k8s资源对象的需求,下面记录一下kubectl和golang发送删除pvc、pv、pod请求的例子,便于后续学...

    我的小碗汤
  • Cypress web自动化35-cy.exec()执行python命令操作数据库

    cy.exec()可以执行系统命令,获取到stdout内容,当我们要操作数据库,准备测试数据的时候,通常用python连数据库操作会非常方便。 我们可以先把操作...

    上海-悠悠
  • 汽车行业引爆资本热潮,英特尔掷153亿美金进军无人驾驶市场 | 大数据24小时

    数据猿导读 英特尔瞄准无人驾驶市场,掷153亿美金收购自动驾驶公司Mobileye;医疗搜索创企“无码科技”获2500万元融资;我国首个数据交易国家级实验室揭牌...

    数据猿
  • HAProxy简介

    HAProxy是一款提供高可用性、负载均衡以及基于TCP(第四层)和HTTP(第七层)应用的代理软件,HAProxy是完全免费的、借助HAProxy可以快速并且...

    张善友

扫码关注云+社区

领取腾讯云代金券