Gorm实战ORM：框架三件套中的数据库魔术师

见解分享

2023-07-16 09:58:47

使用 Gorm 简化 Go 中的数据库交互

简介

在 Go 框架三件套中，Gorm 扮演着 ORM（对象关系映射）框架的角色，负责处理与数据库的交互。它消除了编写复杂 SQL 语句的需要，使开发人员能够轻松地与各种类型的数据库进行交互。

Gorm 的关键特性

ActiveRecord 模式： Gorm 采用 ActiveRecord 模式，将数据模型与数据库表一一对应，简化了数据操作。
自动化迁移： Gorm 根据数据模型自动生成数据库表，无需手动编写 SQL 脚本。
查询构建器： Gorm 提供了一个功能丰富的查询构建器，可以方便地编写复杂的查询语句。
事务管理： Gorm 支持事务管理，确保数据库操作的原子性和一致性。
关联映射： Gorm 支持关联映射，可以轻松处理数据模型之间的关系。

Gorm 入门

1. 安装 Gorm

go get -u gorm.io/gorm

2. 创建数据模型

type User struct {
  ID        uint64 `gorm:"primary_key"`
  Username  string `gorm:"type:varchar(255);unique_index"`
  Password  string `gorm:"type:varchar(255)"`
  CreatedAt time.Time `gorm:"autoCreateTime"`
  UpdatedAt time.Time `gorm:"autoUpdateTime"`
}

3. 连接数据库

db, err := gorm.Open("mysql", "user:password@tcp(127.0.0.1:3306)/database")
if err != nil {
  panic(err)
}

4. 自动迁移

db.AutoMigrate(&User{})

5. 插入数据

user := User{Username: "admin", Password: "123456"}
db.Create(&user)

6. 查询数据

var user User
db.First(&user, 1)

7. 更新数据

db.Model(&user).Update("username", "new_username")

8. 删除数据

db.Delete(&user)

Gorm 进阶指南

1. 使用关联映射

type User struct {
  ID        uint64 `gorm:"primary_key"`
  Username  string `gorm:"type:varchar(255);unique_index"`
  Password  string `gorm:"type:varchar(255)"`
  CreatedAt time.Time `gorm:"autoCreateTime"`
  UpdatedAt time.Time `gorm:"autoUpdateTime"`
  Posts     []Post `gorm:"foreignKey:UserID"`
}

type Post struct {
  ID        uint64 `gorm:"primary_key"`
  Title     string `gorm:"type:varchar(255)"`
  Content   string `gorm:"type:text"`
  CreatedAt time.Time `gorm:"autoCreateTime"`
  UpdatedAt time.Time `gorm:"autoUpdateTime"`
  UserID    uint64 `gorm:"index"`
}

2. 使用查询构建器

db.Where("username = ?", "admin").First(&user)
db.Order("created_at DESC").Limit(10).Find(&users)

3. 使用事务管理

tx := db.Begin()
defer func() {
  if r := recover(); r != nil {
    tx.Rollback()
  } else {
    tx.Commit()
  }
}()

user := User{Username: "admin", Password: "123456"}
if err := tx.Create(&user).Error; err != nil {
  panic(err)
}

post := Post{Title: "Hello World", Content: "This is my first post"}
if err := tx.Create(&post).Error; err != nil {
  panic(err)
}