代码生成概述

我们来构建一个代码生成器，为类似枚举的类型创建 String() 方法！虽然在这个环境中无法实际运行 go generate，但你会模拟代码生成工具的工作方式：根据类型定义，以编程方式生成 Go 源代码。

你会把代码分布在两个文件中：

• generator.go：编写代码生成逻辑。

  创建一个 EnumType 结构体，用来表示类似枚举的类型定义，其中包含 Name（字符串）和 Values（表示常量名称的字符串切片）。

  创建一个函数 GenerateStringMethod(e EnumType) string，用于生成 String() 方法的 Go 源代码。生成的代码应遵循以下模式：

  func (e TypeName) String() string {
      switch e {
      case Value1:
          return "Value1"
      case Value2:
          return "Value2"
      default:
          return "Unknown"
      }
  }

  再创建一个函数 GenerateConstants(e EnumType) string，使用 iota 生成常量声明：

  const (
      Value1 TypeName = iota
      Value2
      Value3
  )

  最后，创建 GenerateFullCode(e EnumType) string，将两种输出组合起来，并添加一条表明这是生成代码的头部注释：

  // Code generated by enum generator. DO NOT EDIT.
  
  [constants]
  
  [string method]
• main.go：使用你的生成器为不同的枚举类型生成代码。

  读取枚举类型名称，然后读取值的数量，再读取每个值的名称。将这些信息传给生成器，并打印完整的生成代码。

将提供以下输入：

• 第 1 行：枚举类型名称
• 第 2 行：值的数量
• 接下来的各行：每个值的名称

例如，给定：

Status
3
Pending
Active
Completed

你的输出应为：

// Code generated by enum generator. DO NOT EDIT.

const (
    Pending Status = iota
    Active
    Completed
)

func (e Status) String() string {
    switch e {
    case Pending:
        return "Pending"
    case Active:
        return "Active"
    case Completed:
        return "Completed"
    default:
        return "Unknown"
    }
}

再比如，给定：

Priority
2
Low
High

你的输出应为：

// Code generated by enum generator. DO NOT EDIT.

const (
    Low Priority = iota
    High
)

func (e Priority) String() string {
    switch e {
    case Low:
        return "Low"
    case High:
        return "High"
    default:
        return "Unknown"
    }
}

这个练习展示了 stringer 等工具背后的核心概念：在编译时生成重复但类型安全的代码，而不是在运行时使用反射。你的生成器会生成有效的 Go 代码，这些代码可以保存到文件中，并与项目的其余部分一起编译。