這里需要用到enum庫(kù)

go get -u github.com/younamebert/enum

1. 定義一個(gè)枚舉類型

package main

import (
    "fmt"

    "github.com/younamebert/enum"
)

var sex enum.Enum

func main() {
    //0,1,2
    sex = enum.NewEnum("man", "woman", "secrecy")
}

2. 操作枚舉enum

package main

import (
    "fmt"

    "github.com/younamebert/enum"
)

var sex enum.Enum

func main() {
    //0,1,2
    sex = enum.NewEnum("man", "woman", "secrecy")

    fmt.Println(sex.IsEnum("man"))             // true
    fmt.Println(sex.IsEnums("woman", "woman")) // true

    fmt.Println(sex.IsEnum("abc"))         // false
    fmt.Println(sex.IsEnums("man", "abc")) // false
}

查看枚舉值

package main

import (
    "fmt"

    "github.com/younamebert/enum"
)

var sex enum.Enum

func main() {
    //0,1,2
    sex = enum.NewEnum("man", "woman", "secrecy")

    value, err := sex.Enum("man")
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println(value) //0

    values, err := sex.Enums("man", "secrecy")
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println(values) //[0 2]

    _, err = sex.Enum("abc")
    if err != nil {
        fmt.Println(err) //does not exist in the enum
        return
    }
}

修改自定義枚舉值

package main

import (
    "fmt"

    "github.com/younamebert/enum"
)

var sex enum.Enum

func main() {
    //0,1,2
    sex = enum.NewEnum("man", "woman", "secrecy")

    if err := sex.SetEnum("man", 4); err != nil {
        fmt.Println(err)
        return
    }

    value, err := sex.Enum("man")
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println(value) //4

}

添加和移除枚舉值

package main

import (
    "fmt"

    "github.com/younamebert/enum"
)

var sex enum.Enum

func main() {
    //0,1,2
    sex = enum.NewEnum("man", "woman", "secrecy")

    //append
    if err := sex.AppendEnum("abc"); err != nil {
        fmt.Println(err)
        return
    }

    value, err := sex.Enum("abc")
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println(value) //3

    //remove
    if err := sex.RemoveEnum("abc"); err != nil {
        fmt.Println(err)
        return
    }
    value, err = sex.Enum("abc")
    if err != nil {
        fmt.Println(err) // does not exist in the enum
        return
    }
    fmt.Println(value)
}