美文网首页
Go语言模拟枚举(const和iota模拟枚举)

Go语言模拟枚举(const和iota模拟枚举)

作者: 码二哥 | 来源:发表于2020-01-29 17:41 被阅读0次

参考:
http://c.biancheng.net/view/24.html

关键点

  • Go语言自身没有枚举类型
  • Go语言使用constiota进行配合来间接实现枚举
  • 枚举值转换为字符串

1、使用const与iota配合实现枚举类型

1.1、例子1

type Weapon int

const (
     Arrow Weapon = iota    // 开始生成枚举值, 默认为0
     Shuriken
     SniperRifle
     Rifle
     Blower
)

// 输出所有枚举值
fmt.Println(Arrow, Shuriken, SniperRifle, Rifle, Blower)

// 使用枚举类型并赋初值
var weapon Weapon = Blower
fmt.Println(weapon)

代码输出如下:

0 1 2 3 4
4

代码说明如下:
第 1 行中,将 int 定义为 Weapon 类型,就像枚举类型的本质是一个 int 类型一样。当然,某些情况下,如果需要 int32 和 int64 的枚举,也是可以的。

第 4 行中,将常量 Arrow 的类型标识为 Weapon,这样标识后,const 下方的常量可以使用 Weapon 作为默认类型。

该行使用 iota 进行常量值自动生成,iota 的起始值为 0,一般情况下也是建议枚举从 0 开始,让每个枚举类型都有一个空值,方便业务和逻辑的灵活使用。


image

一个 const 声明内的每一行常量声明,将会自动套用前面的 iota 格式,并自动增加,类似于电子表格中单元格自动填充的效果,只需要建立好单元格之间的变化关系,拖动右下方的小点就可以自动生成单元格的值。

1.2、例子2

当然,iota 不仅可以生成每次增加 1 的枚举值。
还可以利用 iota 来做一些强大枚举常量值生成器
下面的代码可以方便的生成标志位常量:

const (
    FlagNone = 1 << iota
    FlagRed
    FlagGreen
    FlagBlue
)

fmt.Printf("%d %d %d\n", FlagRed, FlagGreen, FlagBlue)
fmt.Printf("%b %b %b\n", FlagRed, FlagGreen, FlagBlue)

代码输出如下:

2 4 8
10 100 1000

在代码中编写一些标志位时,我们往往手动编写常量值,

常量值特别多时,很容易重复或者写错

因此,使用 ioto 自动生成更加方便

代码说明如下:

  • 第 2 行中 iota 使用了一个移位操作,每次将上一次的值左移一位(二进制位),以得出每一位的常量值。
  • 第 8 行,将 3 个枚举按照常量输出,分别输出 2、4、8,都是将 1 每次左移一位的结果。
  • 第 9 行,将枚举值按二进制格式输出,可以清晰地看到每一位的变化。

2、将枚举值转换为字符串

枚举在 C# 中是一个独立的类型,可以通过枚举值获取该值对应的字符串。

例如,C# 中 Week 枚举值 Monday 为 1,那么可以通过 Week.Monday.ToString() 函数获得 Monday 字符串。

Go语言中也可以实现这一功能,代码如下所示:

转换字符串:

package main

import "fmt"

// 声明芯片类型
type ChipType int

const (
    None ChipType = iota
    CPU    // 中央处理器
    GPU    // 图形处理器
)

func (c ChipType) String() string {
    switch c {
    case None:
        return "None"
    case CPU:
        return "CPU"
    case GPU:
        return "GPU"
    }

    return "N/A"
}

func main() {

    // 输出CPU的值并以整型格式显示
    fmt.Printf("%s %d", CPU, CPU)
}

运行结果:

CPU 1

代码说明如下:

  • 第 6 行,将 int 声明为 ChipType 芯片类型。
  • 第 9 行,将 const 里定义的常量值设为 ChipType 类型,且从 0 开始,每行值加 1。
  • 第 14 行,定义 ChipType 类型的方法 String(),返回值为字符串类型
  • 第 15~22 行,使用 switch 语句判断当前的 ChitType 类型的值,返回对应的字符串。
  • 第 30 行,按整型的格式输出 CPU 的值。
image
String() 方法的 ChipType 在使用上和普通的常量没有区别。

当这个类型需要显示为字符串时,Go语言会自动寻找 String() 方法并进行调用

相关文章

  • Go语言模拟枚举(const和iota模拟枚举)

    参考:http://c.biancheng.net/view/24.html 关键点 Go语言自身没有枚举类型 G...

  • 04-枚举常量

    Go语言枚举 c语言中的枚举 Go语言枚举 iota迭代器 Go语言输出函数 fmt.Printf("格式化字符串...

  • iota

    iota go语言中不支持枚举定义,但是使用iota可以实现类似效果 枚举定义 iota实际被定义为0,仅配合co...

  • golang的枚举类型

    go语言并没有提供enum的定义,我们可以使用const来模拟枚举类型。 这里定义了一个新的类型PolicyTyp...

  • Go语言--iota枚举

    介绍 iota 常量自动生成器,每个一行,自动累加1 iota给常量赋值使用3.iota遇到const,重置为04...

  • 模拟枚举

    使用:

  • Go语言学习——三分钟看透iota

    Go语言学习——三分钟看透iota 源起枚举 最近做需求时,有一个需要枚举的场景,大概有10+个枚举类型,不愿意像...

  • JDK5-枚举

    使用普通类模拟枚举 使用抽象类模拟枚举 使用抽象类模拟可以免除toString()写大量的if-else语句,不需...

  • Go 学习笔记 基础语法 变量

    定义包内变量(Go语言里面没有全局变量) 函数内定义变量 定义常量 const 枚举类型

  • 第01天(基本类型、流程控制)_02

    07_常量的使用.go 08_多个变量或常量定义.go 09_iota枚举.go 10_bool类型.go 11_...

网友评论

      本文标题:Go语言模拟枚举(const和iota模拟枚举)

      本文链接:https://www.haomeiwen.com/subject/mgfbthtx.html