首頁 >後端開發 >Golang >淺析Golang中怎麼實現枚舉(附程式碼)

淺析Golang中怎麼實現枚舉(附程式碼)

青灯夜游
青灯夜游轉載
2022-02-04 09:00:315527瀏覽

Golang中怎麼實作枚舉?以下這篇文章跟大家介紹一下Golang中實作枚舉的方法,希望對大家有幫助!

淺析Golang中怎麼實現枚舉(附程式碼)

在程式設計領域裡,枚舉用來表示只包含有限數量的固定值的類型,在開發中一般用於標識錯誤碼或狀態機。拿一個實體物件的狀態機來說,它通常與這個物件在資料庫裡對應記錄的識別狀態的欄位值相對應。 【相關推薦:Go影片教學

在剛開始學程式設計的時候,你一定寫過,至少看過直接使用魔術數字來判斷的程式碼。啥叫魔術數字呢,舉個例子,要置頂一個文章的時候先判斷文章是不是已發布狀態。

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 類型的參數會發生隱式類型轉換,不會造成編譯報錯,這塊如果想解決,只能重新定義類型來實現了,

可以參考StackOverflow上的這個答案:

https://stackoverflow.com/questions/50826100/how-to-disable-implicit-type-conversion-for-constants

#更多程式相關知識,請訪問:編程視頻! !

以上是淺析Golang中怎麼實現枚舉(附程式碼)的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文轉載於:juejin.cn。如有侵權,請聯絡admin@php.cn刪除