Обзор генерации кода

Давайте создадим генератор кода, который создает методы String() для типов, подобных перечислениям (enum)! Хотя мы не можем запустить настоящий go generate в этой среде, вы будете имитировать то, что делают инструменты генерации кода — программно создавать исходный код Go на основе определений типов.

Вы организуете свой код в двух файлах:

• generator.go: Создайте логику генерации кода.

  Создайте структуру EnumType, которая представляет определение типа, похожего на enum, с полями Name (строка) и Values (срез строк, представляющих имена констант).

  Создайте функцию GenerateStringMethod(e EnumType) string, которая генерирует исходный код Go для метода String(). Сгенерированный код должен соответствовать следующему шаблону:

  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, который можно было бы сохранить в файл и скомпилировать вместе с остальной частью проекта.