在編程領(lǐng)域里，枚舉用來表示只包含有限數(shù)量的固定值的類型，在開發(fā)中一般用于標(biāo)識(shí)錯(cuò)誤碼或者狀態(tài)機(jī)。拿一個(gè)實(shí)體對(duì)象的狀態(tài)機(jī)來說，它通常與這個(gè)對(duì)象在數(shù)據(jù)庫里對(duì)應(yīng)記錄的標(biāo)識(shí)狀態(tài)的字段值相對(duì)應(yīng)。

在剛開始學(xué)編程的時(shí)候，你一定寫過，至少見過直接使用魔術(shù)數(shù)字進(jìn)行判斷的代碼。啥叫魔術(shù)數(shù)字呢，舉個(gè)例子，要置頂一個(gè)文章的時(shí)候先判斷文章是不是已發(fā)布狀態(tài)。

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

假如我們的代碼里沒有注釋，或者等我們項(xiàng)目的代碼里充斥著這些魔術(shù)數(shù)字的判斷的時(shí)候，你是不是會(huì)很頭疼？
后來我就學(xué)會(huì)了把這些狀態(tài)值定義成常量，并且也搞一個(gè)判斷對(duì)象狀態(tài)的方法單獨(dú)封裝這段邏輯。

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

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

這種用法，肯定是比在程序里直接用魔術(shù)數(shù)字進(jìn)行判斷要強(qiáng)很多啦，至少看著不會(huì)很頭疼，不會(huì)想罵**。

不過后來被當(dāng)時(shí)帶我的老大哥說這種也有缺點(diǎn)，上面這個(gè) checkArticleState 方法用來檢查文章狀態(tài)，本意是讓調(diào)用者傳入 ArticleState 的三個(gè)靜態(tài)常量之一，但由于沒有類型上的約束，因此傳入任意一個(gè) int 值在語法上也是允許的，編譯器也不會(huì)提出任何警告，用枚舉更合適一些。
em~! 我不記得大學(xué)教 Java 的那個(gè)學(xué)期老師講過這玩意啊，莫非又是一個(gè)上課玩手機(jī)錯(cuò)過的知識(shí)點(diǎn)？所以使用枚舉后我們的Java代碼變成了：

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

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

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

這樣就能靠形參的枚舉類型幫我們過濾掉非法的狀態(tài)值，把整型值作為參數(shù)傳給 checkArticleState 方法時(shí)因?yàn)轭愋筒黄ヅ渚幾g不過去，在寫代碼是編譯器也能馬上提示出來。

如果沒有用過 Java 的小伙伴也不用糾結(jié)，主要語法點(diǎn)我用注釋標(biāo)注出來了，大家應(yīng)該都能看懂。后來這兩年主要在用Go做項(xiàng)目，我發(fā)現(xiàn)相似的問題 Go 里存在，但是 Go 并沒有提供枚舉類型，那怎么做到進(jìn)行狀態(tài)值的正確限制呢？如果還是用 int 型的常量肯定不行。比如：

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

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

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

比如上面定義了兩組 int 類型的常量，一類代表文章狀態(tài)，一類代表季節(jié)的四季。這種方式拿文章狀態(tài)與季節(jié)進(jìn)行比較不會(huì)有任何編譯上的錯(cuò)誤。

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

我們可以用 int 作為基礎(chǔ)

type Season int

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

類型創(chuàng)建一個(gè)別名類型，Go 里邊是支持這個(gè)的

當(dāng)然定義連續(xù)的常量值的時(shí)候 Go 里邊經(jīng)常使用 iota，所以上面的定義還能進(jìn)一步簡(jiǎn)化。

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() {
? ?// 兩個(gè)操作數(shù)類型不匹配，編譯錯(cuò)誤
? ?fmt.Println(Autumn == Draft)
?? ?// 參數(shù)類型不匹配，但是因?yàn)?ArticleState 底層的類型是 int 所以傳遞 int 的時(shí)候會(huì)發(fā)生隱式類型轉(zhuǎn)換，所以不會(huì)報(bào)錯(cuò)
? ?checkArticleState(100)
?}

雖然這些狀態(tài)值的底層的類型都是 int 值，但是現(xiàn)在進(jìn)行兩個(gè)不相干類型的枚舉值比較，會(huì)造成編譯錯(cuò)誤，因?yàn)楝F(xiàn)在我們使用狀態(tài)值的地方都有了類型限制。
不過函數(shù) checkArticleState 的參數(shù)類型設(shè)置成了 ArticleState 但是因?yàn)?ArticleState 底層的類型是 int 。所以調(diào)用 checkArticleState  時(shí)傳遞 int 類型的參數(shù)會(huì)發(fā)生隱式類型轉(zhuǎn)換，不會(huì)造成編譯報(bào)錯(cuò)，這塊如果想解決，只能重新定義類型來實(shí)現(xiàn)了，可以參考StackOverflow上的這個(gè)答案

到此這篇關(guān)于Go語言實(shí)現(xiàn)枚舉的示例代碼的文章就介紹到這了,更多相關(guān)Go語言 枚舉內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論