快捷導(dǎo)航

詳解在Go語言中如何實現(xiàn)枚舉類型

更新時間：2023年04月18日 08:22:14 作者：陳明勇

枚舉類型是一種常用的數(shù)據(jù)類型，用于表示一組有限的、預(yù)定義的、具名的常量值。而在?Go?語言里是沒有內(nèi)置枚舉類型的，所以本文將介紹如何實現(xiàn)?“枚舉類型”，需要的可以參考一下

前言

枚舉類型是一種常用的數(shù)據(jù)類型，用于表示一組有限的、預(yù)定義的、具名的常量值。在枚舉類型中，每個常量都是一個枚舉值，它們之間的值相等且唯一。

枚舉類型通常用于表示一組相關(guān)的常量，比如星期、月份、性別等等。在其他語言里（比如 Java 和 C），都內(nèi)置了枚舉類型，而在 Go 語言里是沒有內(nèi)置枚舉類型的，因此我們需要采用其他方式實現(xiàn)類似的枚舉類型功能，本文將介紹如何實現(xiàn) “枚舉類型”。

Go 語言中的 “枚舉類型”

枚舉類型的值本質(zhì)上是常量，因此我們可以使用 Go 語言中的常量來實現(xiàn)類似枚舉類型的功能，例如：

const (
   Sunday    = 1
   Tuesday   = 2
   Wednesday = 3
   Thursday  = 4
   Friday    = 5
   Saturday  = 6
   Monday    = 7
)

在這個例子中，我們使用了 const 關(guān)鍵字定義了一組常量，其中每個常量的名稱代表著一個枚舉，其值為對應(yīng)的整數(shù)。

雖然這個例子能實現(xiàn)類似的枚舉類型，但它不具備枚舉類型的所有特征，例如缺少安全性和約束性，為了解決這兩個問題，我們可以使用自定義類型進(jìn)行改進(jìn)：

type WeekDay int

const (
   Sunday    WeekDay = 1
   Tuesday   WeekDay = 2
   Wednesday WeekDay = 3
   Thursday  WeekDay = 4
   Friday    WeekDay = 5
   Saturday  WeekDay = 6
   Monday    WeekDay = 7
)

在改進(jìn)后的例子中，我們定義了一個自定義類型 Weekday，用于表示星期幾。使用 const 關(guān)鍵字定義一個常量組，其中每個常量都被賦予了一個具體的值，同時使用 Weekday 類型進(jìn)行類型約束和類型檢查。這樣，我們就可以通過枚舉值的名稱來表示某個特定的星期幾，并且由于使用了自定義類型，編譯器可以進(jìn)行類型檢查，從而提高了類型安全性。

使用 itoa 優(yōu)雅實現(xiàn)枚舉

通過前面的例子不難發(fā)現(xiàn)，當(dāng)我們需要定義多個枚舉值時，手動指定每個枚舉常量的值會變得十分麻煩。為了解決這個問題，我們可以使用 iota 常量生成器，它可以幫助我們生成連續(xù)的整數(shù)值。

例如，使用 iota 定義一個星期幾的枚舉類型：

type WeekDay int

const (
   Sunday WeekDay = iota
   Tuesday
   Wednesday
   Thursday
   Friday
   Saturday
   Monday
)

在這個例子中，我們使用 iota 自增常量生成器來定義了一個星期幾的枚舉類型，每個枚舉值都是一個 Weekday 類型的常量。由于 iota 的自增規(guī)則，每個枚舉值的值將自動遞增，從而生成一系列連續(xù)的整數(shù)值。

為自定義的枚舉添加方法

type WeekDay int

const (
   Sunday WeekDay = iota
   Tuesday
   Wednesday
   Thursday
   Friday
   Saturday
   Monday
)

為了能讓我們實現(xiàn)的 “枚舉類型” 更加具備枚舉類型的特征，我們可以為其添加類似 Java 等其他語言中的枚舉方法。

Name()

返回枚舉值的名稱。

// Name 返回枚舉值的名稱
func (w WeekDay) Name() string {
   if w < Sunday || w > Monday {
      return "Unknown"
   }
   return [...]string{"Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"}[w]
}

Original

返回枚舉值在枚舉類型中的位置。

// Original 由于這個枚舉類型的枚舉值本質(zhì)上是整數(shù)常量，因此我們可以直接使用枚舉值作為它的序號
func (w WeekDay) Original() int {
   return int(w)
}

String()

實現(xiàn) String 方法，用于打印輸出。

// 將枚舉值轉(zhuǎn)成字符串，便于輸出
func (w WeekDay) String() string {
   return w.Name()
}

Values()

返回一個包含所有枚舉值的切片。

// Values 返回一個包含所有枚舉值的切片
func Values() []WeekDay {
   return []WeekDay{Monday, Tuesday, Wednesday, Thursday, Friday, Saturday, Sunday}
}

ValueOf()

根據(jù)名稱返回對應(yīng)的枚舉值。

// ValueOf 使用 switch 語句來根據(jù)名稱返回對應(yīng)的枚舉值
func ValueOf(name string) (WeekDay, error) {
   switch name {
   case "Sunday":
      return Sunday, nil
   case "Monday":
      return Monday, nil
   case "Tuesday":
      return Tuesday, nil
   case "Wednesday":
      return Wednesday, nil
   case "Thursday":
      return Thursday, nil
   case "Friday":
      return Friday, nil
   case "Saturday":
      return Saturday, nil
   default:
      return 0, fmt.Errorf("invalid WeekDay name: %s", name)
   }
}