如何在Go中给GORM模型添加枚举类型

 

GORM中的枚举介绍

GORM是一个流行的Go ORM，它允许我们定义结构体来表示数据库表，并提供一个接口来执行CRUD操作。枚举在编程中是一个有用的特性，它允许我们定义一个变量可以取的固定值集。在这篇文章中，我们将探索如何向GORM模型添加枚举。

第一步：定义枚举

首先，我们需要定义枚举值的类型。在Go中，我们可以用string类型来定义枚举。如下：

 

type Status string


const (
    Pending Status = "pending"
    Approved Status = "approved"
    Rejected Status = "rejected"
)

 

在该示例中，我们定义了枚举类型Status，该类型有3个枚举值："pending"，"approved", "rejectd"。

第二步：定义模型

其次，我们需要定义一个gorm模型来使用该枚举类型。如下：

 

type User struct {
    ID        uint   `gorm:"primary_key"`
    Name      string
    Status    Status
    CreatedAt time.Time
    UpdatedAt time.Time
}

 

在该示例中，User模型中有一个枚举类型Status的字段。

第三步：在gorm操作中使用枚举

在定义了枚举类型和gorm模型后，我们就可以在gorm的操作中使用枚举类型了。如下：

 

user := User{Name: "John", Status: Pending}
db.Create(&user)




var users []User
db.Where("status = ?", Approved).Find(&users)




db.Model(&user).Update("status", Rejected)




db.Where("status = ?", Pending).Delete(User{})

 

在上面的示例中，我们在gorm的创建、查询、更新和删除操作中都使用到了Status的枚举类型。

那么，我们看看在go中有哪些常见的定义枚举的方法。

方法一：使用字符串类型

在模型中，我们可以是通过使用字符串类型来替代枚举类型。如下：

 

type User struct {
  ID   uint
  Role string
}

 

在该示例中，我们可以使用Role这个字符串类型来表示不同的用户角色，比如“admin”，“moderator”以及“user”等等。

但这种方式也有缺点，比如容易出现拼写错误，没有类型的约束。

方法二：自定义类型

另外一种替代枚举类型的方法是通过自定义类型来代表枚举值。如下：

 

type Role string


const (
  Admin     Role = "admin"
  Moderator Role = "moderator"
  User      Role = "user"
)


type User struct {
  ID   uint
  Role Role
}

 

在这个示例中，Role类型被定义成了字符串的别名。然后，不同的角色值以常量的形式定义。最后，在User模型中的Role字段中使用了Role类型。

这种方法的优点是可以做编译时检查，因此不容易出现拼写错误。同时，该种方法还将值限制在了Role类型下。

方法三：使用GORM提供的枚举类进

Gorm提供了内建的枚举类型。如下：

 

type Role string


const (
  Admin     Role = "admin"
  Moderator Role = "moderator"
  User      Role = "user"
)


type User struct {
  ID   uint
  Role Role `gorm:"type:enum('admin', 'moderator', 'user')"`
}

 

在该示例中，Role类型被定义成string类型的别名成为一个新的类型，同时Role的值通过常量定义。在User的模型中，又通过gorm的标签来限制了该字段值的范围。

这种方法和方法二中的方式具有相同的优点，同时该方法还能让通过gorm创建数据表时，让数据表的字段也是enum类型的优点。

结论

在本文中，我们探讨了如何在Go中给GORM模型添加枚举类型。枚举是一个有用的功能，它允许我们定义变量可以取的一组固定值。通过在GORM模型中使用枚举，我们可以提高代码的类型安全性，使数据库操作更加可靠。

　　审核编辑：汤梓红