知识点
- 如何配置数据库链接
- 如何使用 GORM 定义数据模型
- 使用 GORM 完成增删改查
预备知识
- 如何定义结构体
- 结构体字段的 tag 使用
gorm
是一款强大且优秀的 Go 语言 orm 框架。主要功能如下:
- 全功能 ORM (无限接近)
- 关联 (Has One, Has Many, Belongs To, Many To Many, 多态)
- 钩子 (在创建/保存/更新/删除/查找之前或之后)
- 预加载
- 事务
- 复合主键
- SQL 生成器
- 数据库自动迁移
- 自定义日志
- 可扩展性, 可基于 GORM 回调编写插件
- 所有功能都被测试覆盖
- 开发者友好
添加依赖
go get -u github.com/jinzhu/gorm
模型定义
type User struct {
Id int64 `gorm:"PRIMARY_KEY;AUTO_INCREMENT"` // 编号,主键、自增长
Name string `gorm:"size:32;not null"` // 用户名,长度32,不可空
Email string `gorm:"size:128;not null"` // 邮箱,长度128,不可空
}
模型的每个字段都可以添加标记(Tag),通过 Tag 可以配置字段的属性,gorm 支持以下标记:
| 标记(Tag) | 描述 |
| ----------------------------- | ----------------------------- |
| Column | 指定列名 |
| Type | 指定列数据类型 |
| Size | 指定列大小, 默认值 255 |
| PRIMARY_KEY | 将列指定为主键 |
| UNIQUE | 将列指定为唯一 |
| DEFAULT | 指定列默认值 |
| PRECISION | 指定列精度 |
| NOT NULL | 将列指定为非 NULL |
| AUTO_INCREMENT | 指定列是否为自增类型 |
| INDEX | 创建具有或不带名称的索引, 如果多个索引同名则创建复合索引 |
| UNIQUE_INDEX | 和 INDEX
类似,只不过创建的是唯一索引 |
| EMBEDDED | 将结构设置为嵌入 |
| EMBEDDED_PREFIX | 设置嵌入结构的前缀 |
| - | 忽略此字段 |
连接数据库
gorm 支持 sqlite,mysql,postgreSql,mssql,所以在连接数据库的时候需要指定你的数据库类型,下面我们以mysql为例进行讲解。
首先我们连上 mysql 创建数据库:test_db,创建数据库命令如下:
注意:实验楼环境默认情况下 mysql 服务是未启动的,启动 mysql 服务命令为:sudo service mysql start,实验楼 mysql 默认是没有密码的。
CREATE DATABASE `test_db` DEFAULT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;
接下来编写 Go 代码,使用gorm连接数据库:
package main
import (
"github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/mysql"
)
func main() {
// 连接数据库
db, err := gorm.Open("mysql", "root@tcp(localhost:3306)/test_db?charset=utf8mb4&parseTime=True&loc=Local")
if err != nil {
panic("failed to connect database")
}
defer db.Close()
}
首先我们需要将 gorm 和 mysql 支持引入进来,即:
import (
"github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/mysql"
)
那么链接数据库就变得很简单了, 只需要调用 gorm.Open 方法即可,该方法有两个参数,第一个参数是指定数据库类型,第二个参数是数据库的链接。这里我们将数据库类型设定为 mysql,mysql 的链接格式为:
username:password@tcp(数据库地址:端口)/数据库名称
gorm.Open有两个返回值,第一个返回值就是数据库链接,第二个返回值是错误信息,当错误信息为空时表示数据库连接成功,连接成功之后我们即可正常操作数据库啦。
另外,gorm 还提供了AutoMigrate功能,他会根据我们结构体中的定义来自动为我们创建表,开启只需要调用AutoMigrate方法,并且将我们的结构体传入进去即可,代码如下:
// 自动根据结构体创建
db.AutoMigrate(&User{})
连接好数据库后,我们来试试基础的增删改查功能
插入
db.Create(&User{
Name: "小明",
Email: "xiaoming@gmail.com",
})
###更新数据
var user User
db.Where("name = ?", "小明").First(&user)
fmt.Println("小明的邮箱为:" + user.Email)
// 重新赋值,然后更新
user.Email = "xiaoming2@gmail.com"
db.Save(user)
查询数据
查询单条数据:
var user User
db.Where("name = ?", "小明").First(&user)
fmt.Println("小明的邮箱为:" + user.Email)
查询列表数据:
var users []User
db.Find(&users)
fmt.Println("总用户数:", len(users))
删除数据
db.Where("name = ?", "小明").Delete(&User{})
gorm 的功能非常强大,我们这里所讲的只是gorm最基本的使用方法,如果想深入地了解gorm可以去他的官网查看更详细的文档,gorm官网:http://gorm.io