如何使用 GORM 将选项添加到数据库字段

Question

我对Golang很陌生，因为我主要用Python编写代码，所以遇到了一个我无法解决的问题。如果可以的话，我想通过GORM或任何其他方式向Golang Struct中的字段添加选择。我的模型看起来像这样type User struct{   FirstName string  `gorm:"size:100" json:"first_name"`   LastName  string  `gorm:"size:100" json:"last_name"`   Email     *string `gorm:"size:250;index;not null;index:email;unique" json:"email"`   Role      string  `gorm:"default:User;not null" json:"is_active"` // TODO Add Choice}我想从 ['Admin'、'User'、'Guest'] 向用户模型中的角色字段添加选项。有没有办法使用GORM或任何其他方法来解决我的问题来实现它？如果没有直接的方法，我可以把它做成一个外键。我正在使用PostgreSQL来存储表。提前致谢！

Answer 1

假设 您在 和 之间有一对一的关系，那么使用外键的一种方法是这样的：rolesusers

type User struct{

   ID        int64 `json:"id"`

   FirstName string  `gorm:"size:100" json:"first_name"`

   LastName  string  `gorm:"size:100" json:"last_name"`

   Email     *string `gorm:"size:250;index;not null;index:email;unique" json:"email"`

   RoleID    int64  `json:"role_id"` 

   Role      *Role `json:"role"`

}

type Role struct {

   ID   int64 `json:"id"`

   Name string `json:"name"`

}

//load data

var users []User

err := db.Preload("Role").Find(&users).Error

编辑：多对多关系的解决方案

假设您将有一个像链接和表的表，一个解决方案可能如下所示：users_rolesusersroles

type User struct{

   ID        int64 `json:"id"`

   FirstName string  `gorm:"size:100" json:"first_name"`

   LastName  string  `gorm:"size:100" json:"last_name"`

   Email     *string `gorm:"size:250;index;not null;index:email;unique" json:"email"`

   Roles      []*Role `gorm:"many2many:users_roles;" json:"roles"`

}

type Role struct {

   ID   int64 `json:"id"`

   Name string `json:"name"`

}

//load data

var users []User

err := db.Preload("Roles").Find(&users).Error

Answer 2

对于 PostGreSQL，首先在数据库中手动创建枚举

CREATE TYPE user_access AS ENUM (

    'admin',

    'user'

);

然后

type Role string

const (

    Admin  Role = "admin"

    User   Role = "user"

)

func (r *Role) Scan(value interface{}) error {

    *r = Role(value.([]byte))

    return nil

}

func (r Role) Value() (driver.Value, error) {

    return string(r), nil

}

type User struct{

   ID        int64   `json:"id"`

   FirstName string  `gorm:"size:100" json:"first_name"`

   LastName  string  `gorm:"size:100" json:"last_name"`

   Email     *string `gorm:"size:250;index;not null;index:email;unique" json:"email"`

   Role      Role    `sql:"type:user_access"` // PostGreSQL

   Role      Role    `json:"role" sql:"type:ENUM('admin', 'user')"` // MySQL

}

有关更多信息，您可以查看github问题和文档