Go语言gorm包

GORM（Golang ORM）是一个强大的 Go 语言对象关系映射（ORM）库。它支持主流数据库，如 MySQL、PostgreSQL、SQLite 和 SQL Server 等。GORM 提供了丰富的特性，包括自动迁移、查询构建、预加载、事务处理等。以下是详细介绍 GORM 的使用方法及其各个功能模块。

1. 安装 GORM

首先，安装 GORM 及其数据库驱动程序（以 MySQL 为例）：

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

2. 连接数据库

创建一个 main.go 文件，并初始化数据库连接。

package main

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

var DB *gorm.DB

func initDB() {
    dsn := "username:password@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
    var err error
    DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        log.Fatalf("failed to connect database: %v", err)
    }
}

func main() {
    initDB()
}

3. 定义模型

定义一个模型，并使用 GORM 标签进行配置。

package main

import (
    "time"
    "gorm.io/gorm"
)

type User struct {
    ID        uint           `gorm:"primaryKey"`
    Name      string         `gorm:"size:100"`
    Email     string         `gorm:"uniqueIndex;size:100"`
    Age       int            `gorm:"default:18"`
    CreatedAt time.Time
    UpdatedAt time.Time
    DeletedAt gorm.DeletedAt `gorm:"index"`
}

4. 自动迁移

GORM 提供自动迁移功能，可以根据模型自动创建或更新数据库表。

func initDB() {
    // Previous database connection code

    // 自动迁移
    DB.AutoMigrate(&User{})
}

5. CRUD 操作

5.1 创建记录

func createUser() {
    user := User{Name: "Alice", Email: "alice@example.com", Age: 25}
    result := DB.Create(&user)
    if result.Error != nil {
        log.Fatalf("failed to create user: %v", result.Error)
    }
    log.Printf("User ID: %d", user.ID)
}

5.2 查询记录

func getUserByID(id uint) {
    var user User
    result := DB.First(&user, id)
    if result.Error != nil {
        if result.Error == gorm.ErrRecordNotFound {
            log.Println("User not found")
        } else {
            log.Fatalf("failed to get user: %v", result.Error)
        }
    }
    log.Printf("User: %#v", user)
}

5.3 更新记录

func updateUser(id uint) {
    var user User
    result := DB.First(&user, id)
    if result.Error != nil {
        log.Fatalf("failed to get user: %v", result.Error)
    }

    user.Age = 30
    DB.Save(&user)
    log.Printf("Updated User: %#v", user)
}

5.4 删除记录

func deleteUser(id uint) {
    result := DB.Delete(&User{}, id)
    if result.Error != nil {
        log.Fatalf("failed to delete user: %v", result.Error)
    }
    log.Println("User deleted")
}

6. 高级查询

6.1 条件查询

func getUsersByName(name string) {
    var users []User
    DB.Where("name = ?", name).Find(&users)
    log.Printf("Users: %#v", users)
}

6.2 原生 SQL 查询

func getUserByEmail(email string) {
    var user User
    DB.Raw("SELECT * FROM users WHERE email = ?", email).Scan(&user)
    log.Printf("User: %#v", user)
}

6.3 链式查询

func getAdultUsers() {
    var users []User
    DB.Where("age >= ?", 18).Order("age desc").Limit(10).Find(&users)
    log.Printf("Users: %#v", users)
}

7. 关联关系

GORM 支持一对一、一对多、多对多和嵌套结构。

7.1 一对多关系

定义 User 和 Post 模型：

type User struct {
    ID    uint    `gorm:"primaryKey"`
    Name  string  `gorm:"size:100"`
    Posts []Post
}

type Post struct {
    ID     uint   `gorm:"primaryKey"`
    Title  string `gorm:"size:100"`
    UserID uint
}

自动迁移：

func initDB() {
    // Previous database connection code

    DB.AutoMigrate(&User{}, &Post{})
}

创建关联记录：

func createPost() {
    user := User{Name: "Alice"}
    post := Post{Title: "First Post"}
    user.Posts = append(user.Posts, post)
    DB.Create(&user)
}

查询关联记录：

func getUserWithPosts(id uint) {
    var user User
    DB.Preload("Posts").First(&user, id)
    log.Printf("User: %#v", user)
}

7.2 多对多关系

定义 User 和 Role 模型：

type User struct {
    ID    uint    `gorm:"primaryKey"`
    Name  string  `gorm:"size:100"`
    Roles []Role  `gorm:"many2many:user_roles;"`
}

type Role struct {
    ID   uint   `gorm:"primaryKey"`
    Name string `gorm:"size:100"`
}