Golang中如何實現枚舉

這篇文章主要介紹了Golang中如何實現枚舉的相關知識，內容詳細易懂，操作簡單快捷，具有一定借鑒價值，相信大家閱讀完這篇Golang中如何實現枚舉文章都會有所收獲，下面我們一起來看看吧。

在編程領域里，枚舉用來表示只包含有限數量的固定值的類型，在開發中一般用于標識錯誤碼或者狀態機。拿一個實體對象的狀態機來說，它通常與這個對象在數據庫里對應記錄的標識狀態的字段值相對應。

if (article.state == 2) {
   // state 2 代表文章已發布
}

假如我們的代碼里沒有注釋，或者等我們項目的代碼里充斥著這些魔術數字的判斷的時候，你是不是會很頭疼？

后來我就學會了把這些狀態值定義成常量，并且也搞一個判斷對象狀態的方法單獨封裝這段邏輯。

public class ArticleState {
    
    public static final int Draft = 1; //草稿
    
    public static final int Published = 2; //發布
    
    public static final int Deleted = 3; // 已刪除
}

public  Boolean checkArticleState(int state) {
    
    ...
    
}

這種用法，肯定是比在程序里直接用魔術數字進行判斷要強很多啦，至少看著不會很頭疼，不會想罵**。

不過后來被當時帶我的老大哥說這種也有缺點，上面這個 checkArticleState 方法用來檢查文章狀態，本意是讓調用者傳入 ArticleState 的三個靜態常量之一，但由于沒有類型上的約束，因此傳入任意一個 int 值在語法上也是允許的，編譯器也不會提出任何警告，用枚舉更合適一些。

em~! 我不記得大學教 Java 的那個學期老師講過這玩意啊，莫非又是一個上課玩手機錯過的知識點？所以使用枚舉后我們的Java代碼變成了：

// 使用enum而非class聲明
public enum ArticleState {
	
    //要在enum里創建所有的枚舉對象
    Draft(1, "草稿");
    Published(2, "已發布");
    Deleted(3, "已刪除")
      
    // 自定義屬性
    private int code;

    private String text;
  
    // 構造方法必須是private的
    ArticleState(int code, String text) {
        this.code = id;
        this.text = name;
    }
}

public  Boolean checkArticleState(ArticleState state) {
    
    ...
    
}

這樣就能靠形參的枚舉類型幫我們過濾掉非法的狀態值，把整型值作為參數傳給 checkArticleState 方法時因為類型不匹配編譯不過去，在寫代碼是編譯器也能馬上提示出來。

如果沒有用過 Java 的小伙伴也不用糾結，主要語法點我用注釋標注出來了，大家應該都能看懂。

后來這兩年主要在用Go做項目，我發現相似的問題 Go 里存在，但是 Go 并沒有提供枚舉類型，那怎么做到進行狀態值的正確限制呢？如果還是用 int 型的常量肯定不行。比如：

 const (
     Draft int = 1
     Published = 2
     Deleted   = 3
 )

 const (
     Summer int = 1
     Autumn     = 2
     Winter     = 3
     Spring     = 4
 )

 func main() {
     // 輸出 true, 不會有任何編譯錯誤
     fmt.Println(Autumn == Draft)
 }

比如上面定義了兩組 int 類型的常量，一類代表文章狀態，一類代表季節的四季。這種方式拿文章狀態與季節進行比較不會有任何編譯上的錯誤。

答案在 Go 內置庫或者一些咱們都知道的開源庫的代碼里就能找到。比如看看 google.golang.org/grpc/codes 里的gRPC 的錯誤碼是怎么定義的，就能明白該怎么正確的實現枚舉。

我們可以用 int 作為基礎類型創建一個別名類型，Go 里邊是支持這個的

type Season int

const (
	Summer Season = 1
	Autumn        = 2
	Winter        = 3
	Spring        = 4
)

當然定義連續的常量值的時候 Go 里邊經常使用 iota，所以上面的定義還能進一步簡化。

type Season int

const (
	Summer Season = iota + 1
	Autumn
	Winter
	Spring
)

type ArticleState int

const (
  Draft int = iota + 1
  Published
  Deleted  
)

func checkArticleState(state ArticleState) bool {
	// ... 
}

 func main() {
   // 兩個操作數類型不匹配，編譯錯誤
   fmt.Println(Autumn == Draft)
	// 參數類型不匹配，但是因為 ArticleState 底層的類型是 int 所以傳遞 int 的時候會發生隱式類型轉換，所以不會報錯
   checkArticleState(100)
 }

雖然這些狀態值的底層的類型都是 int 值，但是現在進行兩個不相干類型的枚舉值比較，會造成編譯錯誤，因為現在我們使用狀態值的地方都有了類型限制。

不過函數 checkArticleState 的參數類型設置成了 ArticleState 但是因為 ArticleState 底層的類型是 int 。所以調用 checkArticleState  時傳遞 int 類型的參數會發生隱式類型轉換，不會造成編譯報錯，這塊如果想解決，只能重新定義類型來實現了。

關于“Golang中如何實現枚舉”這篇文章的內容就介紹到這里，感謝各位的閱讀！相信大家對“Golang中如何實現枚舉”知識都有一定的了解，大家如果還想學習更多知識，歡迎關注億速云行業資訊頻道。