Go语言:根据常量值获取常量名

Go语言:根据常量值获取常量名

本文旨在介绍如何在 go 语言中,给定一个常量的值,获取其对应的常量名。通过自定义类型和 `String()` 方法,可以实现将 `uint16` 类型的 Cipher Suite 值转换为可读的字符串表示,方便调试和日志记录。本文提供了一个完整的示例代码,展示了如何针对 `crypto/tls` 包中的 Cipher Suite 常量进行转换。

在 Go 语言中,直接通过常量的值反向查找常量名是不直接支持的。但是,我们可以通过自定义类型和方法来实现类似的功能。特别是在处理枚举类型的常量时,这种方法非常有用,例如 crypto/tls 包中定义的 Cipher Suite 常量。

实现原理

核心思想是:

立即学习go语言免费学习笔记(深入)”;

  1. 自定义类型: 创建一个新的类型,其底层类型与常量类型相同(例如 uint16)。
  2. 常量转换: 将 crypto/tls 包中的常量转换为我们自定义的类型。
  3. String() 方法: 为自定义类型实现 String() 方法。该方法根据常量的值返回对应的字符串表示。

示例代码

以下代码展示了如何针对 crypto/tls 包中的 Cipher Suite 常量实现上述功能:

package main  import (     "crypto/tls"     "fmt" )  type Ciphersuite uint16  const (     TLS_RSA_WITH_RC4_128_SHA            = Ciphersuite(tls.TLS_RSA_WITH_RC4_128_SHA)     TLS_RSA_WITH_3DES_EDE_CBC_SHA       = Ciphersuite(tls.TLS_RSA_WITH_3DES_EDE_CBC_SHA)     TLS_RSA_WITH_AES_128_CBC_SHA        = Ciphersuite(tls.TLS_RSA_WITH_AES_128_CBC_SHA)     TLS_RSA_WITH_AES_256_CBC_SHA        = Ciphersuite(tls.TLS_RSA_WITH_AES_256_CBC_SHA)     TLS_ECDHE_RSA_WITH_RC4_128_SHA      = Ciphersuite(tls.TLS_ECDHE_RSA_WITH_RC4_128_SHA)     TLS_ECDHE_RSA_WITH_3DES_EDE_CBC_SHA = Ciphersuite(tls.TLS_ECDHE_RSA_WITH_3DES_EDE_CBC_SHA)     TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA  = Ciphersuite(tls.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA)     TLS_ECDHE_RSA_WITH_AES_256_CBC_SHA  = Ciphersuite(tls.TLS_ECDHE_RSA_WITH_AES_256_CBC_SHA) )  func (cs Ciphersuite) String() string {     switch cs {     case TLS_RSA_WITH_RC4_128_SHA:         return "TLS_RSA_WITH_RC4_128_SHA"     case TLS_RSA_WITH_3DES_EDE_CBC_SHA:         return "TLS_RSA_WITH_3DES_EDE_CBC_SHA"     case TLS_RSA_WITH_AES_128_CBC_SHA:         return "TLS_RSA_WITH_AES_128_CBC_SHA"     case TLS_RSA_WITH_AES_256_CBC_SHA:         return "TLS_RSA_WITH_AES_256_CBC_SHA"     case TLS_ECDHE_RSA_WITH_RC4_128_SHA:         return "TLS_ECDHE_RSA_WITH_RC4_128_SHA"     case TLS_ECDHE_RSA_WITH_3DES_EDE_CBC_SHA:         return "TLS_ECDHE_RSA_WITH_3DES_EDE_CBC_SHA"     case TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA:         return "TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA"     case TLS_ECDHE_RSA_WITH_AES_256_CBC_SHA:         return "TLS_ECDHE_RSA_WITH_AES_256_CBC_SHA"     }     return "Unknown" }  func main() {     cs := TLS_ECDHE_RSA_WITH_3DES_EDE_CBC_SHA     fmt.printf("0x%04x = %sn", uint16(cs), cs)      cs = TLS_RSA_WITH_RC4_128_SHA     fmt.Printf("0x%04x = %sn", uint16(cs), cs)      cs = TLS_ECDHE_RSA_WITH_AES_256_CBC_SHA     fmt.Printf("0x%04x = %sn", uint16(cs), cs) }

代码解释

Go语言:根据常量值获取常量名

商汤商量

商汤科技研发的AI对话工具,商量商量,都能解决。

Go语言:根据常量值获取常量名36

查看详情 Go语言:根据常量值获取常量名

  • type Ciphersuite uint16: 定义了一个名为 Ciphersuite 的新类型,其底层类型为 uint16。
  • const … = Ciphersuite(tls. …): 将 crypto/tls 包中的常量转换为 Ciphersuite 类型。
  • func (cs Ciphersuite) String() string: 为 Ciphersuite 类型实现了 String() 方法。该方法使用 switch 语句根据 cs 的值返回对应的字符串表示。
  • fmt.Printf(“0x%04x = %sn”, uint16(cs), cs): 在 main() 函数中,我们首先将 Ciphersuite 类型的值转换为 uint16 类型,以便以十六进制格式打印其值。然后,我们直接使用 cs,由于 Ciphersuite 类型实现了 String() 方法,fmt.Printf 会自动调用该方法,从而打印出常量名。

运行结果

运行上述代码,将会得到以下输出:

0x0012 = TLS_ECDHE_RSA_WITH_3DES_EDE_CBC_SHA 0x0005 = TLS_RSA_WITH_RC4_128_SHA 0xc014 = TLS_ECDHE_RSA_WITH_AES_256_CBC_SHA

注意事项

  • String() 方法中的 switch 语句需要覆盖所有可能的常量值。如果遇到未知的常量值,应该返回一个默认值(例如 “Unknown”),以避免程序崩溃。
  • Go 1.4 及更高版本可以使用 stringer 工具自动生成 String() 方法。这可以减少手动编写代码的工作量,并降低出错的可能性。可以使用以下命令安装 stringer 工具:go install golang.org/x/tools/cmd/stringer。然后,在包含常量定义的 Go 文件中添加 //go:generate stringer -type=Ciphersuite 注释,并运行 go generate 命令。stringer 工具会自动生成 string() 方法。

总结

通过自定义类型和 String() 方法,我们可以方便地将常量值转换为可读的字符串表示。这在调试和日志记录中非常有用。虽然 Go 语言没有直接提供反向查找常量名的功能,但这种方法提供了一种有效的替代方案。 此外,利用 stringer 工具可以自动化生成 String() 方法,进一步简化开发流程。

上一篇
下一篇
text=ZqhQzanResources