GORM中的枚舉介紹

GORM是一個(gè)流行的Go ORM，它允許我們定義結(jié)構(gòu)體來(lái)表示數(shù)據(jù)庫(kù)表，并提供一個(gè)接口來(lái)執(zhí)行CRUD操作。枚舉在編程中是一個(gè)有用的特性，它允許我們定義一個(gè)變量可以取的固定值集。在這篇文章中，我們將探索如何向GORM模型添加枚舉。

第一步：定義枚舉

首先，我們需要定義枚舉值的類(lèi)型。在Go中，我們可以用string類(lèi)型來(lái)定義枚舉。如下：

type Status string


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

在該示例中，我們定義了枚舉類(lèi)型Status，該類(lèi)型有3個(gè)枚舉值："pending"，"approved", "rejectd"。

第二步：定義模型

其次，我們需要定義一個(gè)gorm模型來(lái)使用該枚舉類(lèi)型。如下：

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

在該示例中，User模型中有一個(gè)枚舉類(lèi)型Status的字段。

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

在定義了枚舉類(lèi)型和gorm模型后，我們就可以在gorm的操作中使用枚舉類(lèi)型了。如下：

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{})

在上面的示例中，我們?cè)趃orm的創(chuàng)建、查詢(xún)、更新和刪除操作中都使用到了Status的枚舉類(lèi)型。

那么，我們看看在go中有哪些常見(jiàn)的定義枚舉的方法。

方法一：使用字符串類(lèi)型

在模型中，我們可以是通過(guò)使用字符串類(lèi)型來(lái)替代枚舉類(lèi)型。如下：

type User struct {
  ID   uint
  Role string
}

在該示例中，我們可以使用Role這個(gè)字符串類(lèi)型來(lái)表示不同的用戶(hù)角色，比如“admin”，“moderator”以及“user”等等。

但這種方式也有缺點(diǎn)，比如容易出現(xiàn)拼寫(xiě)錯(cuò)誤，沒(méi)有類(lèi)型的約束。

方法二：自定義類(lèi)型

另外一種替代枚舉類(lèi)型的方法是通過(guò)自定義類(lèi)型來(lái)代表枚舉值。如下：

type Role string


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


type User struct {
  ID   uint
  Role Role
}

在這個(gè)示例中，Role類(lèi)型被定義成了字符串的別名。然后，不同的角色值以常量的形式定義。最后，在User模型中的Role字段中使用了Role類(lèi)型。

這種方法的優(yōu)點(diǎn)是可以做編譯時(shí)檢查，因此不容易出現(xiàn)拼寫(xiě)錯(cuò)誤。同時(shí)，該種方法還將值限制在了Role類(lèi)型下。

方法三：使用GORM提供的枚舉類(lèi)進(jìn)

Gorm提供了內(nèi)建的枚舉類(lèi)型。如下：

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類(lèi)型被定義成string類(lèi)型的別名成為一個(gè)新的類(lèi)型，同時(shí)Role的值通過(guò)常量定義。在User的模型中，又通過(guò)gorm的標(biāo)簽來(lái)限制了該字段值的范圍。

這種方法和方法二中的方式具有相同的優(yōu)點(diǎn)，同時(shí)該方法還能讓通過(guò)gorm創(chuàng)建數(shù)據(jù)表時(shí)，讓數(shù)據(jù)表的字段也是enum類(lèi)型的優(yōu)點(diǎn)。

結(jié)論

在本文中，我們探討了如何在Go中給GORM模型添加枚舉類(lèi)型。枚舉是一個(gè)有用的功能，它允許我們定義變量可以取的一組固定值。通過(guò)在GORM模型中使用枚舉，我們可以提高代碼的類(lèi)型安全性，使數(shù)據(jù)庫(kù)操作更加可靠。

審核編輯：湯梓紅