使用 GORM 操作数据库

1. 安装 GORM

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

2. 初始化数据库连接

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

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 {
    panic("failed to connect database")
}

3. 定义模型

type User struct {
    ID    uint   `gorm:"primaryKey"`
    Name  string
    Email string `gorm:"unique"`
    Age   int
}

4. 自动迁移

db.AutoMigrate(&User{})

自动创建或更新数据表结构。

5. CRUD 操作

创建

user := User{Name: "Alice", Email: "[email protected]", Age: 25}
db.Create(&user)

查询

var user User
db.First(&user, 1)                 // 根据主键查询
db.Where("name = ?", "Alice").Find(&user)

更新

db.Model(&user).Update("Age", 26)

删除

db.Delete(&user)

6. 事务

tx := db.Begin()
if err := tx.Error; err != nil {
    return err
}

if err := tx.Create(&User{Name: "Bob"}).Error; err != nil {
    tx.Rollback()
    return err
}

tx.Commit()

GORM 简化了数据库操作流程，支持丰富的功能和灵活的查询，是 Go 项目中高效操作数据库的利器。