GORM

漫郁~...大约 3 分钟

GORM

GORM 是 Go 语言中一个功能强大且流行的 ORM（对象关系映射）库。它帮助开发者更轻松地在 Go 语言中与数据库进行交互，提供了与数据库交互的抽象层，使得开发者可以使用 Go 语言的结构体来进行数据库操作，而不必直接编写 SQL 语句。

GORM 的特点：

自动迁移：GORM 可以根据你的结构体自动生成数据库表结构，并进行迁移。
CRUD 操作：GORM 提供了方便的 API 进行创建、读取、更新和删除（CRUD）操作。
关联关系：支持数据库表之间的一对一、一对多、多对多等关联关系操作。
事务支持：GORM 支持数据库事务操作，可以确保数据的原子性。
查询构建：支持构建复杂的查询，包括条件查询、关联查询、排序、分组等。
钩子函数：GORM 支持在创建、更新、删除等操作前后添加钩子函数，以实现一些特定的业务逻辑。
预加载：GORM 支持预加载关联数据，避免了 N+1 查询问题。

GORM 的基本用法

1. 安装 GORM

要使用 GORM，首先需要在 Go 项目中安装 GORM 及其所需的数据库驱动。以 MySQL 为例：

go get -u gorm.io/gorm
go get -u gorm.io/driver/mysql

2. 连接数据库

下面是如何使用 GORM 连接 MySQL 数据库的示例：

package main

import (
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
    "log"
)

func main() {
    // 配置数据库连接字符串
    dsn := "user:password@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"

    // 连接数据库
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        log.Fatal("Failed to connect to the database:", err)
    }

    log.Println("Database connection successful!")
}

3. 定义模型

在 GORM 中，你可以使用 Go 结构体定义数据库中的表。例如，定义一个用户表：

type User struct {
    ID       uint   `gorm:"primaryKey"`
    Name     string `gorm:"size:255"`
    Email    string `gorm:"uniqueIndex"`
    Password string
}

4. 自动迁移

GORM 可以根据模型自动创建和更新数据库表结构：

err := db.AutoMigrate(&User{})
if err != nil {
    log.Fatal("Failed to migrate database schema:", err)
}

5. CRUD 操作

以下是 GORM 中的基本 CRUD 操作示例：

创建记录：

user := User{Name: "John Doe", Email: "john@example.com", Password: "password123"}
db.Create(&user)

读取记录：

var user User
db.First(&user, 1) // 根据主键查找
db.First(&user, "email = ?", "john@example.com") // 根据条件查找

更新记录：

db.Model(&user).Update("Name", "John Updated")
db.Model(&user).Updates(User{Name: "John Updated", Email: "john_new@example.com"})

删除记录：

db.Delete(&user)

6. 关联关系

GORM 支持表与表之间的关联关系，比如一对多关系、一对一关系、多对多关系等。

一对多关系：

type User struct {
    ID     uint
    Name   string
    Orders []Order
}

type Order struct {
    ID     uint
    UserID uint
    Amount float64
}

var user User
db.Preload("Orders").First(&user, 1)

多对多关系：

type User struct {
    ID      uint
    Name    string
    Friends []*User `gorm:"many2many:user_friends"`
}

var user User
db.Preload("Friends").First(&user)