如何在gorm中添加enum？

在Golang中使用GORM时，要实现枚举（enum），通常有几种方法可以处理。GORM本身并不直接支持枚举类型，因为Go语言本身不直接支持枚举，但我们可以通过一些策略来模拟枚举的功能。以下是几种常见的方法：

方法一：使用自定义类型

1. 定义一个自定义类型：首先定义一个基于string或int的自定义类型来表示枚举。
2. 为该类型添加方法：可以为这个类型添加方法来确保赋值的有效性。
3. 在GORM模型中使用这个自定义类型：在你的GORM模型中使用这个自定义枚举类型作为字段类型。

例子

假设我们要为用户定义一个“角色”枚举，包含“Admin”和“Member”。

go
package main

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

type Role string

const (
    Admin  Role = "Admin"
    Member Role = "Member"
)

// GORM模型
type User struct {
    gorm.Model
    Name string
    Role Role
}

func (r *Role) Scan(value interface{}) error {
    *r = Role(value.(string))
    return nil
}

func (r Role) Value() (driver.Value, error) {
    return string(r), nil
}

func main() {
    db, err := gorm.Open(...)
    if err != nil {
        log.Fatal(err)
    }

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

    // 使用枚举
    db.Create(&User{Name: "John Doe", Role: Admin})
}

在这个例子中，我们定义了一个Role类型，并且两个可能的值Admin和Member都是Role类型。我们在User模型中使用Role作为字段类型。Scan和Value方法确保GORM能够正确地从数据库读取和写入该类型。

方法二：使用常量和校验

1. 定义常量：定义一组常量代表枚举的值。
2. 在模型中添加字段：在模型中添加一个普通的string或int字段来存储枚举值。
3. 添加校验逻辑：在插入或更新数据前，通过代码逻辑校验字段值是否为有效的枚举值。

例子

继续上面的角色例子，不过这次我们不定义新类型，直接使用string：

go
package main

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

const (
    RoleAdmin  = "Admin"
    RoleMember = "Member"
)

type User struct {
    gorm.Model
    Name string
    Role string
}

func main() {
    db, err := gorm.Open(...)
    if err != nil {
        log.Fatal(err)
    }

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

    user := User{Name: "Jane Doe", Role: RoleAdmin}
    if user.Role != RoleAdmin && user.Role != RoleMember {
        log.Fatal("Invalid role")
    }

    db.Create(&user)
}

在这种情况下，我们需要在代码中手动检查Role字段的值是否是定义的有效枚举值之一。

总结

尽管Go和GORM都没有内建的枚举支持，通过上述方法仍然可以有效地在GORM中实现类似枚举的功能，确保数据的有效性和完整性。选择哪种方式取决于具体的使用场景和个人偏好。