有限值的类型安全枚举[重复]

Question

在Go中如何在有限的值范围内实现类型安全枚举？

例如，假设我想用简单的尺寸和颜色对t恤衫进行建模，这两种t恤衫都有有限的可能值，以确保我无法创建具有不受支持的值的实例。

我可以声明类型大小和颜色基于字符串，定义有效值的枚举，以及使用它们的TShirt类型：

type Size string
const (
  Small  Size = "sm"
  Medium      = "md"
  Large       = "lg"
  // ...
)

type Color string
const (
  Red   Color = "r"
  Green       = "g"
  Blue        = "b"
  // ...
)

type TShirt struct {
  Size  Size
  Color Color
}

var mediumBlueShirt = TShirt{Medium, Blue}

但是，我如何确保没有创建尺寸/颜色未定义的T恤？

var doNotWant = TShirt{"OutrageouslyLarge", "ImpossibleColor"}

Answer 1

不直接支持有限集合枚举，但是您可以使用类型系统来实现：

// Exported interface 
type Size interface {
   // Unexported method
   isSize()
}

// unexported type
type size struct {
   sz string
}

func (s size) String() string {return s.sz}
// Implement the exported interface
func (size) isSize() {}

var (
  Small Size = size{"small"}
  Large Size = size{"large"}
)

这样，您只能使用预先声明的值。

这通常不值得麻烦。考虑针对已知值集验证此类枚举。

Answer 2

如果使枚举基于整数，则可以创建可用于检查有效性的伪值：

type Size int
const (
    Small Size = iota
    Medium
    Large
    maxSize // Always keep this at the end
)

var sizeToString = map[Size]string{
    Small: "sm",
    Medium: "md",
    Large: "lg",
}

func (s Size) String() string {
    return sizeToString[s]
}

func (s Size) Valid() bool {
    return s > 0 && s < maxSize
}

如果您在const块的末尾保留maxsize，则您的Valid函数将始终工作，即使在添加或删除枚举值之后也是如此。

您可以对第二个枚举类型执行相同的操作，并且您的复合类型还可以定义一个Valid函数，如果两个枚举都有效，该函数将返回true。

当我使用这个模式时，我总是包含一个单元测试，以确保每个枚举值都有一个字符串转换，这样我就不会忘记。您可以通过从0到maxsize创建一个for循环来实现这一点，并检查您是否从未从sizeToString中获得空字符串。