Golang使用GORM操作数据库全流程

gorm.io/driver/mysql

gorm.io/driver/postgres

gorm.Open()

*gorm.DB

package main

import (
    "gorm.io/driver/sqlite" // 示例使用SQLite，方便演示
    "gorm.io/gorm"
    "log"
)

var DB *gorm.DB

func init() {
    var err error
    // 实际项目中会连接MySQL/PostgreSQL等
    DB, err = gorm.Open(sqlite.Open("gorm.db"), &gorm.Config{})
    if err != nil {
        log.Fatalf("无法连接到数据库: %v", err)
    }
    log.Println("数据库连接成功！")
}

func main() {
    // 后续操作将在这里进行
}

gorm.Model

ID

CreatedAt

UpdatedAt

DeletedAt

type User struct {
    gorm.Model
    Name    string `gorm:"type:varchar(100);not null;uniqueIndex"` // 姓名，非空，唯一索引
    Email   string `gorm:"type:varchar(255);unique"`               // 邮箱，唯一
    Age     int    `gorm:"default:18"`                             // 年龄，默认18
    Address string `gorm:"size:255"`                               // 地址
}

DB.AutoMigrate()

func init() {
    // ... 数据库连接代码 ...
    err = DB.AutoMigrate(&User{}) // 自动迁移User模型
    if err != nil {
        log.Fatalf("数据库迁移失败: %v", err)
    }
    log.Println("数据库迁移完成。")
}

// 创建用户
func createUser(user *User) {
    result := DB.Create(user)
    if result.Error != nil {
        log.Printf("创建用户失败: %v", result.Error)
        return
    }
    log.Printf("用户创建成功，ID: %d", user.ID)
}

// 查询用户
func getUserByID(id uint) *User {
    var user User
    result := DB.First(&user, id) // 根据ID查找
    if result.Error != nil {
        log.Printf("查询用户ID %d 失败: %v", id, result.Error)
        return nil
    }
    log.Printf("查询到用户: %+v", user)
    return &user
}

// 更新用户
func updateUser(user *User) {
    result := DB.Save(user) // 保存所有更改
    if result.Error != nil {
        log.Printf("更新用户失败: %v", result.Error)
        return
    }
    log.Printf("用户更新成功，ID: %d", user.ID)
}

// 删除用户 (软删除)
func deleteUser(id uint) {
    result := DB.Delete(&User{}, id)
    if result.Error != nil {
        log.Printf("删除用户ID %d 失败: %v", id, result.Error)
        return
    }
    log.Printf("用户ID %d 已被软删除", id)
}

host

port

user

password

dbname

user:password@tcp(host:port)/dbname?charset=utf8mb4&parseTime=True&loc=Local

parseTime=True

time.Time

loc=Local

parseTime=True

mysql -h host -P port -u user -p

psql -h host -p port -U user -d dbname

SELECT

INSERT

UPDATE

DELETE

DB.SetMaxIdleConns()

DB.SetMaxOpenConns()

DB.SetConnMaxLifetime()

// 优化连接池设置
DB.SetMaxIdleConns(10)           // 设置空闲连接池中的最大连接数
DB.SetMaxOpenConns(100)          // 设置数据库的最大打开连接数
DB.SetConnMaxLifetime(time.Hour) // 设置连接可重用的最长时间

CreatedAt

created_at

string

VARCHAR

int

int

gorm.Model

gorm.Model

ID

uint

CreatedAt

time.Time

UpdatedAt

time.Time

DeletedAt

gorm.DeletedAt

DeletedAt

Unscoped()

gorm:"column:your_column_name;type:jsonb;size:255;not null;unique;default:value"

type Product struct {
    ID        uint           `gorm:"primaryKey"` // 明确指定主键
    Code      string         `gorm:"unique;not null;index:idx_code"` // 唯一且非空，并创建名为idx_code的索引
    Price     uint           `gorm:"default:0"`
    CreatedAt time.Time
    UpdatedAt time.Time
    DeletedAt gorm.DeletedAt `gorm:"index"` // 软删除，并为DeletedAt字段创建索引
    Details   string         `gorm:"type:jsonb"` // 示例：使用JSONB类型存储复杂数据
}

// User 和 CreditCard 是一对一关系
type CreditCard struct {
    gorm.Model
    Number string
    UserID uint // 外键
}

// User 和 Order 是一对多关系
type Order struct {
    gorm.Model
    OrderSN string
    UserID  uint // 外键
    User    User // 关联模型，用于Preload或Joins
}

// User 和 Role 是多对多关系
type Role struct {
    gorm.Model
    Name  string
    Users []User `gorm:"many2many:user_roles;"` // 中间表名为 user_roles
}

index

uniqueIndex

DB.Create(&user)

CreatedAt

UpdatedAt

ID

newUser := User{Name: "Alice", Email: "alice@example.com", Age: 25}
result := DB.Create(&newUser)
if result.Error != nil {
    log.Printf("创建用户失败: %v", result.Error)
} else {
    log.Printf("创建用户成功，ID: %d", newUser.ID)
}

DB.Create(&[]User{{Name: "Bob"}, {Name: "Charlie"}})

// 查询年龄大于20，按年龄降序排列的前5个用户
var activeUsers []User
DB.Where("age > ?", 20).Order("age desc").Limit(5).Find(&activeUsers)
log.Printf("查询到的活跃用户: %+v", activeUsers)

// 查找第一个匹配条件的用户
var firstUser User
DB.Where("name = ?", "Alice").First(&firstUser)
log.Printf("查找到的Alice: %+v", firstUser)

// 更新用户年龄
if userToUpdate := getUserByID(1); userToUpdate != nil {
    userToUpdate.Age = 26
    updateUser(userToUpdate) // Save会更新所有字段
}

// 仅更新邮箱字段
DB.Model(&User{}).Where("id = ?", 2).Update("email", "bob.new@example.com")

// 软删除ID为3的用户
deleteUser(3)

// 硬删除ID为4的用户 (假设存在且需要彻底移除)
DB.Unscoped().Delete(&User{}, 4)
log.Printf("用户ID 4 已被硬删除")

DB.Transaction(func(tx *gorm.DB) error { ... })