Go 語言中代替枚舉的辦法

枚舉類型用于聲明一組命名的常量，當一個變量有幾種可能的取值時，可以將它定義為枚舉類型。在 Go 語言中，并沒有提供枚舉類型，但是枚舉類型又是開發過程中非常重要的數據類型。因為它可以事先定義一組有效的值，可以有效地防止用戶提交無效數據，抽象到業務場景中就是我們平時網頁上遇到的下拉框，或者我們選擇快遞地址時的省市信息，均為枚舉類型的用武之地。所以在 Go 語言中對常量進行了一定的擴展，使其可以完美地替代枚舉類型。

1.常量中的iota

為了使常量可以更好地替代枚舉類型，Go 語言提供了一個iota關鍵字。使用iota初始化常量，可以生成一組規則類似的常量，但是不用每個常量都寫一遍表達式。在一個const()表達式中，從iota開始的常量會被置為0，向后每個常量的值為前一個常量的值加一。

代碼示例：

package main

import (
    "fmt"
)

type Weekday int //自定義一個星期類型，作為枚舉類型

const (
    Sun Weekday = iota
    Mon
    Tues
    Wed
    Thur
    Fri
    Sat
)

func main() {
    fmt.Println("Sun :", Sun)
    fmt.Println("Mon :", Mon)
    fmt.Println("Tues:", Tues)
    fmt.Println("Wed :", Wed)
    fmt.Println("Thur:", Thur)
    fmt.Println("Fri :", Fri)
    fmt.Println("Sat :", Sat)
}

Tips：自定義數據類型會在Go語言拓展已有類型中詳細介紹。這里可以簡單地理解為定義了一個Weekday的枚舉。

執行結果：

2. 將枚舉值轉換為字符串

使用iota是可以使用 Go 語言的常量代替枚舉類型，但是由于輸出值均為阿拉伯數字，給調試和辨識造成了一定的困難。為了解決這一問題，Go 語言還提供了使常量枚舉值輸出為字符串的辦法。需要我們手動構造一個使用枚舉類型輸出字符串的方法。

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

代碼示例：

package main

import (
    "fmt"
)

type Weekday int //自定義一個星期類型，作為枚舉類型

const (
    Sun Weekday = iota
    Mon
    Tues
    Wed
    Thur
    Fri
    Sat
)

func (w Weekday) String() string {
    switch w {
    case Sun:
        return "Sun"
    case Mon:
        return "Mon"
    case Tues:
        return "Tues"
    case Wed:
        return "Wed"
    case Thur:
        return "Thur"
    case Fri:
        return "Fri"
    case Sat:
        return "Sat"
    }
    //不存在的枚舉類型就返回"N/A"
    return "N/A"
}

func main() {
    fmt.Println("Sun :", Sun)
    fmt.Println("Mon :", Mon)
    fmt.Println("Tues:", Tues)
    fmt.Println("Wed :", Wed)
    fmt.Println("Thur:", Thur)
    fmt.Println("Fri :", Fri)
    fmt.Println("Sat :", Sat)
}

執行結果：

3. 小結

本文主要介紹了如何使用 Go 語言中的常量來代替枚舉類型。主要有兩個要點：

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