GORM 入门指南 | 豆包MarsCode AI刷题；

Serendipity565
96 阅读9分钟

gorm是一个使用Go语言编写的ORM框架。它文档齐全，对开发者友好，支持主流数据库。

官方中文文档：gorm.io/zh_CN/

gorm介绍

中文官方网站内含十分齐全的中文文档，有了它你甚至不需要再继续向下阅读本文。

安装

go get -u github.com/jinzhu/gorm

连接数据库

连接不同的数据库都需要导入对应数据的驱动程序，GORM已经贴心的为我们包装了一些驱动程序，只需要按如下方式导入需要的数据库驱动即可：

import _ "github.com/jinzhu/gorm/dialects/mysql"
// import _ "github.com/jinzhu/gorm/dialects/postgres"
// import _ "github.com/jinzhu/gorm/dialects/sqlite"
// import _ "github.com/jinzhu/gorm/dialects/mssql"

连接MySQL

import (
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/mysql"
)
​
func main() {
    db, err := gorm.Open("mysql", "user:password@(localhost)/dbname?charset=utf8mb4&parseTime=True&loc=Local")
    defer db.Close()
}

连接PostgreSQL

基本代码同上，注意引入对应postgres驱动并正确指定gorm.Open()参数。

import (
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/postgres"
)
​
func main() {
    db, err := gorm.Open("postgres", "host=myhost port=myport user=gorm dbname=gorm password=mypassword")
    defer db.Close()
}

连接Sqlite3

基本代码同上，注意引入对应sqlite驱动并正确指定gorm.Open()参数。

import (
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/sqlite"
)
​
func main() {
    db, err := gorm.Open("sqlite3", "/tmp/gorm.db")
    defer db.Close()
}

连接SQL Server

基本代码同上，注意引入对应mssql驱动并正确指定gorm.Open()参数。

import (
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/mssql"
)
​
func main() {
    db, err := gorm.Open("mssql", "sqlserver://username:password@localhost:1433?database=dbname")
    defer db.Close()
}

GORM基本示例

注意:

  1. 本文以MySQL数据库为例，讲解GORM各项功能的主要使用方法。
  2. 往下阅读本文前，你需要有一个能够成功连接上的MySQL数据库实例。

创建数据库

在使用GORM前手动创建数据库db1：

CREATE DATABASE db1;

GORM操作MySQL

使用GORM连接上面的db1进行创建、查询、更新、删除操作。

package main
​
import (
    "fmt"
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/mysql"
)
​
type Book struct {
    ID       int    // 注意这里的字段名改为大写开头
    BookName string
    Author   string
}
​
func main() {
    db, err := gorm.Open("mysql", "root:040906@(127.0.0.1:3306)/library?charset=utf8mb4&parseTime=True&loc=Local")
    if err != nil {
        panic(err)
    }
    defer db.Close()
​
    // 自动迁移
    db.AutoMigrate(&Book{})
​
    b1 := Book{ID: 1, BookName: "平凡的世界", Author: "路遥"}   // 使用结构体字段名进行赋值
    b2 := Book{ID: 4, BookName: "解忧杂货店", Author: "东野圭吾"} // 使用结构体字段名进行赋值
    // 创建记录
    db.Create(&b1)
    db.Create(&b2)
​
    // 查询
    var b = new(Book)
    db.First(b)
    fmt.Printf("%#v\n", b)
​
    var book Book
    db.Find(&book, "book_name=?", "三体")
    fmt.Printf("%#v\n", book)
​
    // 更新
    db.Model(&b).Update("Author", "司马迁") // 更新时使用结构体字段名
    // 删除
    db.Delete(&b)
}
​

说明：

  1. charset=utf8mb4: 这个参数指定了连接使用的字符集，这里是UTF-8的mb4版本，支持更广泛的字符集。
  2. parseTime=True: 这个参数告诉MySQL驱动程序将数据库中的时间类型转换为Go的 time.Time 类型。
  3. loc=Local: 这个参数设置了本地时区，这样从数据库中读取的时间会以本地时区的方式表示。
  4. 在GORM中，AutoMigrate函数用于自动迁移数据库模式，它会检查你定义的结构体（比如你的Book结构体）与数据库表之间的差异，然后自动执行相应的SQL语句来保持它们同步。具体来说，如果数据库中不存在与结构体对应的表，AutoMigrate会创建该表。如果表已经存在，但是结构体中有新的字段，它会在表中添加新字段。如果表中的字段与结构体中定义的字段不一致，它会更新表结构以反映结构体的定义。AutoMigrate函数会自动处理这些变化，使得数据库表与结构体定义保持同步。当你调用db.AutoMigrate(&Book{})时，它会检查你的Book结构体与数据库表之间的差异，并自动执行必要的SQL语句来更新数据库表结构，以确保它们与结构体定义一致。

GORM Model定义

在使用ORM工具时，通常我们需要在代码中定义模型（Models）与数据库中的数据表进行映射，在GORM中模型（Models）通常是正常定义的结构体、基本的go类型或它们的指针。 同时也支持sql.Scanner及driver.Valuer接口（interfaces）。

gorm.Model

为了方便模型定义，GORM内置了一个gorm.Model结构体。gorm.Model是一个包含了ID, CreatedAt, UpdatedAt, DeletedAt四个字段的Golang结构体。

// gorm.Model 定义
type Model struct {
    ID        uint `gorm:"primary_key"`
    CreatedAt time.Time
    UpdatedAt time.Time
    DeletedAt *time.Time
}

说明：这里的gorm:"primary_key"是GORM的标签，用于指定字段的约束和选项。在这里，primary_key告诉GORM将ID字段映射为数据库表的主键。

你可以将它嵌入到你自己的模型中：

// 将 `ID`, `CreatedAt`, `UpdatedAt`, `DeletedAt`字段注入到`User`模型中
type User struct {
    gorm.Model
    Name string
}

当然你也可以完全自己定义模型：

// 不使用gorm.Model，自行定义模型
type User struct {
    ID   int
    Name string
}

模型定义示例

type User struct {
    gorm.Model
    Name         string
    Age          sql.NullInt64
    Birthday     *time.Time
    Email        string  `gorm:"type:varchar(100);unique_index"`
    Role         string  `gorm:"size:255"` // 设置字段大小为255
    MemberNumber *string `gorm:"unique;not null"` // 设置会员号（member number）唯一并且不为空
    Num          int     `gorm:"AUTO_INCREMENT"` // 设置 num 为自增类型
    Address      string  `gorm:"index:addr"` // 给address字段创建名为addr的索引
    IgnoreMe     int     `gorm:"-"` // 忽略本字段
}

说明：索引是数据库中的一种数据结构，用于加快查询操作的速度。当你在某个字段上创建了索引，数据库会对该字段的值进行排序并建立一种快速查找的数据结构，以便在查询时能够更快地定位到匹配的记录。需要注意的是，虽然索引可以提高查询速度，但在进行大量的插入、更新或删除操作时，索引也会增加数据库的维护开销。

结构体标记（tags）

使用结构体声明模型时，标记（tags）是可选项。gorm支持以下标记:

支持的结构体标记（Struct tags）

结构体标记（Tag）描述
Column指定列名
Type指定列数据类型
Size指定列大小, 默认值255
PRIMARY_KEY将列指定为主键
UNIQUE将列指定为唯一
DEFAULT指定列默认值
PRECISION指定列精度
NOT NULL将列指定为非 NULL
AUTO_INCREMENT指定列是否为自增类型
INDEX创建具有或不带名称的索引, 如果多个索引同名则创建复合索引
UNIQUE_INDEX和 INDEX 类似，只不过创建的是唯一索引
EMBEDDED将结构设置为嵌入
EMBEDDED_PREFIX设置嵌入结构的前缀
-忽略此字段

关联相关标记（tags）

结构体标记（Tag）描述
MANY2MANY指定连接表
FOREIGNKEY设置外键
ASSOCIATION_FOREIGNKEY设置关联外键
POLYMORPHIC指定多态类型
POLYMORPHIC_VALUE指定多态值
JOINTABLE_FOREIGNKEY指定连接表的外键
ASSOCIATION_JOINTABLE_FOREIGNKEY指定连接表的关联外键
SAVE_ASSOCIATIONS是否自动完成 save 的相关操作
ASSOCIATION_AUTOUPDATE是否自动完成 update 的相关操作
ASSOCIATION_AUTOCREATE是否自动完成 create 的相关操作
ASSOCIATION_SAVE_REFERENCE是否自动完成引用的 save 的相关操作
PRELOAD是否自动完成预加载的相关操作

主键、表名、列名的约定

主键（Primary Key）

GORM 默认会使用名为ID的字段作为表的主键。

type User struct {
    ID   string // 名为`ID`的字段会默认作为表的主键
    Name string
}

// 使用`AnimalID`作为主键
type Animal struct {
    AnimalID int64 `gorm:"primary_key"`
    Name     string
    Age      int64
}

表名（Table Name）

表名默认就是结构体名称的复数，例如：

type User struct {} // 默认表名是 `users`
// 将 User 的表名设置为 `profiles`
func (User) TableName() string {
    return "profiles"
}

func (u User) TableName() string {
    if u.Role == "admin" {
    return "admin_users"
    } else {
    return "users"
    }
}

// 禁用默认表名的复数形式，如果置为 true，则 `User` 的默认表名是 `user`
db.SingularTable(true)

也可以通过Table()指定表名：

// 使用User结构体创建名为`deleted_users`的表
db.Table("deleted_users").CreateTable(&User{})

var deleted_users []User
db.Table("deleted_users").Find(&deleted_users)
// 相当于 SELECT * FROM deleted_users;

db.Table("deleted_users").Where("name = ?", "serendipity").Delete()
// 相当于 DELETE FROM deleted_users WHERE name = 'jinzhu';

GORM还支持更改默认表名称规则：

gorm.DefaultTableNameHandler = func (db *gorm.DB, defaultTableName string) string  {
    return "prefix_" + defaultTableName;
}

列名（Column Name）

列名由字段名称进行下划线分割来生成

type User struct {
    ID        uint       // column name is `id`
    Name      string     // column name is `name`
    Birthday  time.Time  // column name is `birthday`
    CreatedAt time.Time  // column name is `created_at`
}

可以使用结构体tag指定列名：

type Animal struct {
    AnimalId    int64     `gorm:"column:beast_id"`          // set column name to `beast_id`
    Birthday    time.Time `gorm:"column:day_of_the_beast"`  // set column name to `day_of_the_beast`
    Age         int64     `gorm:"column:age_of_the_beast"`  // set column name to `age_of_the_beast`
}

时间戳跟踪

CreatedAt

如果模型有 CreatedAt字段，该字段的值将会是初次创建记录的时间。

db.Create(&user) // `CreatedAt`将会是当前时间

// 可以使用`Update`方法来改变`CreateAt`的值
db.Model(&user).Update("CreatedAt", time.Now())

UpdatedAt

如果模型有UpdatedAt字段，该字段的值将会是每次更新记录的时间。

db.Save(&user) // `UpdatedAt`将会是当前时间

db.Model(&user).Update("name", "jinzhu") // `UpdatedAt`将会是当前时间

DeletedAt

如果模型有DeletedAt字段，调用Delete删除该记录时，将会设置DeletedAt字段为当前时间，而不是直接将记录从数据库中删除。

个人思考与总结

学习曲线

作为初学者，学习 GORM 是需要一定的时间和精力的，需要理解 ORM 的基本概念。GORM 的文档相对全面且详细，中文文档也非常友好，能够帮助我们快速上手。

安装和连接数据库

GORM 的安装非常简单，只需要一行命令即可完成。而连接不同的数据库时，只需要导入相应的驱动程序，并按文档中的示例代码进行配置，很容易理解和实现。

基本操作

通过 GORM，可以轻松地进行数据库的基本操作，如创建、查询、更新和删除。在进行这些操作时，需要注意使用结构体字段名的赋值和查询，保持代码的一致性。

模型定义

GORM 提供了 gorm.Model 结构体，内置了常用的字段，可以直接嵌入到自己的模型中，简化了模型的定义过程。同时，GORM 也支持自定义模型，可以根据自己的需求进行灵活的调整。

结构体标签（tags）

GORM 支持丰富的结构体标签，可以指定列名、数据类型、默认值等。这些标签可以帮助我们更好地管理数据库表结构，但是在使用时需要仔细阅读文档，了解每个标签的作用。

时间戳跟踪

GORM 内置了对 CreatedAt、UpdatedAt 和 DeletedAt 字段的支持，能够自动管理记录的创建、更新和删除时间。在记录操作时间的时候非常好用。

个人建议

  1. 多动手实践：多写一些示例代码，亲手操作数据库，熟悉 GORM 的各项功能。
  2. 多看官方文档：文档中有详细的使用说明和示例代码，遇到问题时优先查阅文档。
avatar
文章
阅读
粉丝