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 作為基礎
type Season int const ( Summer Season = 1 Autumn = 2 Winter = 3 Spring = 4 )
類型創建一個別名類型,Go 裡邊是支持這個的
當然定義連續的常量值的時候 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上的這個答案
到此這篇關於Go語言實現枚舉的示例代碼的文章就介紹到這瞭,更多相關Go語言 枚舉內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!