Go：如何使用枚举作为类型？

欢迎各位小伙伴来到golang学习网，相聚于此都是缘哈哈哈！今天我给大家带来《Go：如何使用枚举作为类型？》，这篇文章主要讲到等等知识，如果你对Golang相关的知识非常感兴趣或者正在自学，都可以关注我，我会持续更新相关文章！当然，有什么建议也欢迎在评论留言提出！一起学习！

我有一个定义特定事件的事件列表（枚举）：

package events

const (
    new_user       = "new_user"
    direct_message = "direct_message"
    disconnect     = "disconnect"
)

并且有一个结构体将使用此枚举作为其属性之一

type ConnectionPayload struct {
    EventName    string      `json:"eventName"`
    EventPayload interface{} `json:"eventPayload"`
}

有没有办法可以使用 enum 作为 eventname 的类型而不是字符串？

这在 typescript 中是可能的，不知道如何在 golang 中做到这一点。

我希望开发人员通过枚举强制使用正确的事件名称，而不是通过使用任何随机字符串作为事件名称来犯错误。

正确答案

您可以通过生成如下代码来完成此操作。

type eventnames string

const (
    new_user       eventnames = "new_user"
    direct_message eventnames = "direct_message"
    disconnect     eventnames = "disconnect"
)

然后将您的结构更改为：

type connectionpayload struct {
    eventname    eventnames  `json:"eventname"`
    eventpayload interface{} `json:"eventpayload"`
}

目前 go 中没有枚举类型，并且目前没有直接的方法来强制执行与 typescript 相同的规则。

go 中的常见做法是使用 @ttrasn 发布的建议：

使用“枚举”值定义自定义类型和类型常量：

type eventname string

const (
    new_user       eventname = "new_user"
    direct_message eventname = "direct_message"
    disconnect     eventname = "disconnect"
)

这允许您在 go 代码中标记您期望此类值的位置：

// example function signature :
func onevent(e eventname, id int) error { ... }

// example struct :
type connectionpayload struct {
    eventname    eventname  `json:"eventname"`
    eventpayload interface{} `json:"eventpayload"`
}

并且它将阻止将普通 string 分配给 eventname

var str string = "foo"
var ev eventname

ev = str // won't compile
onevent(str, 42) // won't compile

已知的限制是：

• 在 go 中，总是有一个零值：

  var ev eventname  // ev is ""

• 字符串文字与类型变量不同，可以赋值：

  var ev eventname = "something_else"

• 允许强制转换：

  var str string = "foo"
  var ev eventname = eventname(str)

• 不检查解组：

  jsn := []byte(`{"eventName":"SOMETHING_ELSE","eventPayload":"some message"}`)
  err := json.Unmarshal(jsn, &payload) // no error

https://go.dev/play/p/vMUTpvH8DBb

如果您想要进行更严格的检查，则必须自己编写验证器或自定义解组器。

今天关于《Go：如何使用枚举作为类型？》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！