枚舉類(lèi)型用于聲明一組命名的常量，當(dāng)一個(gè)變量有幾種可能的取值時(shí)，可以將它定義為枚舉類(lèi)型。在 Go 語(yǔ)言中，并沒(méi)有提供枚舉類(lèi)型，但是枚舉類(lèi)型又是開(kāi)發(fā)過(guò)程中非常重要的數(shù)據(jù)類(lèi)型。因?yàn)樗梢允孪榷x一組有效的值，可以有效地防止用戶(hù)提交無(wú)效數(shù)據(jù)，抽象到業(yè)務(wù)場(chǎng)景中就是我們平時(shí)網(wǎng)頁(yè)上遇到的下拉框，或者我們選擇快遞地址時(shí)的省市信息，均為枚舉類(lèi)型的用武之地。所以在 Go 語(yǔ)言中對(duì)常量進(jìn)行了一定的擴(kuò)展，使其可以完美地替代枚舉類(lèi)型。

1.常量中的iota

為了使常量可以更好地替代枚舉類(lèi)型，Go 語(yǔ)言提供了一個(gè)iota關(guān)鍵字。使用iota初始化常量，可以生成一組規(guī)則類(lèi)似的常量，但是不用每個(gè)常量都寫(xiě)一遍表達(dá)式。在一個(gè)const()表達(dá)式中，從iota開(kāi)始的常量會(huì)被置為0，向后每個(gè)常量的值為前一個(gè)常量的值加一。

代碼示例：

• 1? package main
• 2
• 3? import (
• 4? ? ? ? "fmt"
• 5? )
• 6
• 7? type Weekday int //自定義一個(gè)星期類(lèi)型，作為枚舉類(lèi)型
• 8
• 9? const (
• 10? ? ? ? Sun Weekday = iota
• 11? ? ? ? Mon
• 12? ? ? ? Tues
• 13? ? ? ? Wed
• 14? ? ? ? Thur
• 15? ? ? ? Fri
• 16? ? ? ? Sat
• 17? )
• 18
• 19? func main() {
• 20? ? ? ? ? fmt.Println("Sun :", Sun)
• 21? ? ? ? ? fmt.Println("Mon :", Mon)
• 22? ? ? ? ? fmt.Println("Tues:", Tues)
• 23? ? ? ? ? fmt.Println("Wed :", Wed)
• 24? ? ? ? ? fmt.Println("Thur:", Thur)
• 25? ? ? ? ? fmt.Println("Fri :", Fri)
• 26? ? ? ? ? fmt.Println("Sat :", Sat)
• 27? }

Tips：自定義數(shù)據(jù)類(lèi)型會(huì)在Go語(yǔ)言拓展已有類(lèi)型中詳細(xì)介紹。這里可以簡(jiǎn)單地理解為定義了一個(gè)Weekday的枚舉。

執(zhí)行結(jié)果：

2. 將枚舉值轉(zhuǎn)換為字符串

使用iota是可以使用 Go 語(yǔ)言的常量代替枚舉類(lèi)型，但是由于輸出值均為阿拉伯?dāng)?shù)字，給調(diào)試和辨識(shí)造成了一定的困難。為了解決這一問(wèn)題，Go 語(yǔ)言還提供了使常量枚舉值輸出為字符串的辦法。需要我們手動(dòng)構(gòu)造一個(gè)使用枚舉類(lèi)型輸出字符串的方法。

Tips：用的比較少，稍作了解即可

代碼示例：

• 1? package main
• 2
• 3? import (
• 4? ? ? ? ? "fmt"
• 5? )
• 6
• 7? type Weekday int //自定義一個(gè)星期類(lèi)型，作為枚舉類(lèi)型
• 8
• 9? const (
• 10? ? ? ? Sun Weekday = iota
• 11? ? ? ? Mon
• 12? ? ? ? Tues
• 13? ? ? ? Wed
• 14? ? ? ? Thur
• 15? ? ? ? Fri
• 16? ? ? ? Sat
• 17? )
• 18
• 19? func (w Weekday) String() string {
• 20? ? ? ? ? switch w {
• 21? ? ? ? ? case Sun:
• 22? ? ? ? ? ? ? ? ? return "Sun"
• 23? ? ? ? ? case Mon:
• 24? ? ? ? ? ? ? ? ? return "Mon"
• 25? ? ? ? ? case Tues:
• 26? ? ? ? ? ? ? ? ? return "Tues"
• 27? ? ? ? ? case Wed:
• 28? ? ? ? ? ? ? ? ? return "Wed"
• 29? ? ? ? ? case Thur:
• 30? ? ? ? ? ? ? ? ? return "Thur"
• 31? ? ? ? ? case Fri:
• 32? ? ? ? ? ? ? ? ? return "Fri"
• 33? ? ? ? ? case Sat:
• 34? ? ? ? ? ? ? ? ? return "Sat"
• 35? ? ? ? ?? }
• 36? ? ? ? ? //不存在的枚舉類(lèi)型就返回"N/A"
• 37? ? ? ? ? return "N/A"
• 38? }
• 39
• 40? func main() {
• 41? ? ? ? ? fmt.Println("Sun :", Sun)
• 42? ? ? ? ? fmt.Println("Mon :", Mon)
• 43? ? ? ? ? fmt.Println("Tues:", Tues)
• 44? ? ? ? ? fmt.Println("Wed :", Wed)
• 45? ? ? ? ? fmt.Println("Thur:", Thur)
• 46? ? ? ? ? fmt.Println("Sat :", Fri)
• 47? ? ? ? ? fmt.Println("Sat :", Sat)
• 48? }

執(zhí)行結(jié)果：

3. 小結(jié)

本文主要介紹了如何使用 Go 語(yǔ)言中的常量來(lái)代替枚舉類(lèi)型。主要有兩個(gè)要點(diǎn)：

• 自定義枚舉類(lèi)型，第一個(gè)常量使用iota來(lái)賦值。
• 手動(dòng)構(gòu)建String()方法，使常量輸出為字符串。

文章來(lái)源于網(wǎng)絡(luò)，侵刪！