脚本之家，脚本语言编程技术及教程分享平台！
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Golang - Go项目怎么使用枚举

Go项目怎么使用枚举

2022-11-11 12:00Golang梦工厂 Golang

枚举是一种很重要的数据类型，本文主要介绍了Go项目怎么使用枚举，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

前言

哈喽，大家好，我是asong。枚举是一种很重要的数据类型，在java、C语言等主流编程语言中都支持了枚举类型，但是在Go语言中却没有枚举类型，那有什么替代方案吗？ 本文我们来聊一聊这个事情；

为什么要有枚举

我们以java语言为例子，在JDK1.5之前没有枚举类型，我们通常会使用int常量来表示枚举，一般使用如下：

?
1
2
3
public static final int COLOR_RED = 1;
public static final int COLOR_BLUE = 2;
public static final int COLOR_GREEN = 3;

使用int类型会存在以下隐患：

  • 不具备安全性，声明时如果没有使用final就会造成值被篡改的风险；
  • 语义不够明确，打印int型数字并不知道其具体含义

于是乎我们就想到用常量字符来表示，代码就变成了这样：

?
1
2
3
public static final String COLOR_RED = "RED";
public static final String COLOR_BLUE = "BLUE";
public static final String COLOR_GREEN = "GREEN";

这样也同样存在问题，因为我们使用的常量字符，那么有些程序猿不按套路出牌就可以使用字符串的值进行比较，这样的代码会被不断模仿变得越来越多的，然后屎山就出现了；

所以我们迫切需要枚举类型的出现来起到约束的作用，假设使用一个枚举类型做入参，枚举类型就可以限定沙雕用户不按套路传参，这样就可以怼他了，哈哈～；

使用枚举的代码就可以变成这样，传了枚举之外的类型都不可以了；

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
public class EnumClass {
    public static void main(String [] args){
        Color color = Color.RED;
        convert(color);
        System.out.println(color.name());
 
    }
 
    public static void convert(Color c){
        System.out.println(c.name());
    }
 
}
 
enum Color{
    RED,BLUE,GREEN;
}

Go语言就没有枚举类型，我们该使用什么方法来替代呢？

定义新类型实现枚举

枚举通常是一组相关的常量集合，Go语言中有提供常量类型，所以我们可以使用常量来声明枚举，但也同样会遇到上述的问题，起不到约束的作用，所以为了起到约束我们可以使用Go语言另外一个知识点 -- 类型定义，Go语言中可以使用type关键字定义不同的类型，我们可以为整型、浮点型、字符型等定义新的类型，新的类型与原类型转换需要显式转换，这样在一定程度上也起到了约束的作用，我们就可以用Go语言实现如下枚举：

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
type OrderStatus int
 
const (
    CREATE OrderStatus = iota + 1
    PAID
    DELIVERING
    COMPLETED
    CANCELLED
)
 
func main() {
    a := 100
    IsCreated(a)
}

上面的代码就会报错：

./main.go:19:12: cannot use a (variable of type int) as type OrderStatus in argument to IsCreated

定义新的类型可以起到约束作用，比如我们要检查状态机，入参限定了必须是OrderStatus类型，如果是int类型就会报错。

上面我们的枚举实现方式只能获取枚举值，获取不到其映射的字面意思，所以我们可以优化一下，实现String方法，使用官方提供的cmd/string来快速实现，代码如下：

?
1
2
3
4
5
6
7
8
9
10
//go:generate stringer -type=OrderStatus
type OrderStatus int
 
const (
    CREATE OrderStatus = iota + 1
    PAID
    DELIVERING
    COMPLETED
    CANCELLED
)

执行命令go generate ./...生成orderstatus_string.go文件：

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import "strconv"
 
func _() {
    // An "invalid array index" compiler error signifies that the constant values have changed.
    // Re-run the stringer command to generate them again.
    var x [1]struct{}
    _ = x[CREATE-1]
    _ = x[PAID-2]
    _ = x[DELIVERING-3]
    _ = x[COMPLETED-4]
    _ = x[CANCELLED-5]
}
 
const _OrderStatus_name = "CREATEPAIDDELIVERINGCOMPLETEDCANCELLED"
 
var _OrderStatus_index = [...]uint8{0, 6, 10, 20, 29, 38}
 
func (i OrderStatus) String() string {
    i -= 1
    if i < 0 || i >= OrderStatus(len(_OrderStatus_index)-1) {
        return "OrderStatus(" + strconv.FormatInt(int64(i+1), 10) + ")"
    }
    return _OrderStatus_name[_OrderStatus_index[i]:_OrderStatus_index[i+1]]
}

protobuf中生成的枚举代码

Go语言使用protobuf会生成对应的枚举代码，我们发现其中也是使用定义新的类型的方式来实现的，然后在封装一些方法，我们来赏析一下protobuf生成的枚举代码：

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
const (
    CREATED  OrderStatus = 1
    PAID OrderStatus = 2
    CANCELED OrderStatus = 3
)
 
var OrderStatus_name = map[int32]string{
    1: "CREATED",
    2: "PAID",
    3: "CANCELED",
}
 
var OrderStatus_value = map[string]int32{
    "CREATED":  1,
    "PAID": 2,
    "CANCELED": 3,
}
 
func (x OrderStatus) Enum() *OrderStatus {
    p := new(OrderStatus)
    *p = x
    return p
}
 
func (x OrderStatus) String() string {
    return proto.EnumName(OrderStatus_name, int32(x))
}
 
func (x *OrderStatus) UnmarshalJSON(data []byte) error {
    value, err := proto.UnmarshalJSONEnum(OrderStatus_value, data, "OrderStatus")
    if err != nil {
        return err
    }
    *x = OrderStatus(value)
    return nil
}

总结

虽然Go语言没有提供枚举类型，但是我们也可以根据Go语言的两个特性：常量和定义新类型来实现枚举，方法总比困难多吗，开源库是优秀的，我们往往可以从高手那里里学习很多，记住，请永远保持一个学徒之心；

到此这篇关于Go项目怎么使用枚举的文章就介绍到这了,更多相关Go 枚举内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家！

原文链接：https://juejin.cn/post/7134265331597443080

  • Go
  • 枚举

    • 延伸 · 阅读

    精彩推荐
    • Golang详解Go strconv包

      详解Go strconv包

      Go语言中strconv包实现了基本数据类型和其字符串表示的相互转换。这篇文章主要介绍了Go strconv包的相关知识,需要的朋友可以参考下...

      云崖先生12232021-01-02
    • Golanggo语言中的协程详解

      go语言中的协程详解

      本文详细讲解了go语言中的协程，文中通过示例代码介绍的非常详细。对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下...

      奋斗的大橙子10482022-07-16
    • Golang详解Go语言中Get/Post请求测试

      详解Go语言中Get/Post请求测试

      这篇文章主要为大家详细介绍了Go语言中的环境安装以及Get和Post请求接口的测试，文中的示例代码讲解详细，感兴趣的可以跟随小编一起学习一下...

      剑客阿良_ALiang11942022-10-11
    • Golang基于Go语言实现的简易api网关的示例代码

      基于Go语言实现的简易api网关的示例代码

      本文主要介绍了基于Go语言实现的简易api网关，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下...

      董洪臣4112022-01-19
    • Golanggorm整合进go-zero的实现方法

      gorm整合进go-zero的实现方法

      go-zero提供的代码生成器里面，没有提供orm框架操作，但是提供了遍历的缓存操作，所以可以利用gorm当作一个sql语句的生成器，把生成后的sql语句放到go-z...

      jiangxiaoju9112022-09-07
    • Golanggolang开发中channel使用

      golang开发中channel使用

      channel[通道]是golang的一种重要特性，正是因为channel的存在才使得golang不同于其它语言。这篇文章主要介绍了golang开发中channel使用,需要的朋友可以参考下...

      飞翔码农5782020-09-27
    • GolangGO语言ini配置文件的读取的操作

      GO语言ini配置文件的读取的操作

      这篇文章主要介绍了GO语言ini配置文件的读取的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...

      Coldestmonth6962021-06-23
    • GolangGo语言学习之指针的用法详解

      Go语言学习之指针的用法详解

      这篇文章主要为大家详细介绍了Go语言中指针的用法，文中的示例代码讲解详细，对我们学习Go语言有一定的帮助，需要的可以参考一下...

      隐姓埋名48697152022-09-27
    最近更新
    编辑推荐
    阅读排行
    热门标签
    Cond 　 指针类型 　 new() 　 make() 　 并发模型 　 切片 　 静态WEB服务器 　 端口扫描器 　 Go语言入门教程 　 基础语 　 Slices 　 Maps 　 Golang编程 　 panic 　 接受者类型 　 嵌入类型 　 Slice 　 Golang教程 　 Golang项目 　 内存模型 　 开发环境搭建 　 大小端判断 　 代码组织结构 　 复合类型 　 GOPATH 　 工作目录 　 实例 　 struct类型 　 并发技术 　 buffered 　
    © 2019-2022 脚本之家 | 服务器之家(www.tuohang.net)旗下站点 版权所有关于我们联系我们版权申明网站地图