如何在Go中获取一个类型的所有常数

Question

下面是一个例子：

package main

type State int

const (
    Created State = iota
    Modified
    Deleted
)

func main() {
    // Some code here where I need the list
    // of all available constants of this type.
}

这方面的用例是创建有限状态机(FSM)。能够获得所有常量将帮助我编写一个测试用例，以确保每个新值在FSM映射中都有相应的条目。

Answer 1

如果您的常量都按顺序排列，则可以使用以下命令：

type T int

const (
    TA T = iota
    TB
    TC
    NumT
)

func AllTs() []T {
    ts := make([]T, NumT)
    for i := 0; i < int(NumT); i++ {
        ts[i] = T(i)
    }
    return ts
}

您还可以在init()中缓存输出。只有当所有常量按顺序用Iota初始化时，这才会起作用。如果需要适用于所有情况的东西，请使用显式切片。