コード生成の概要

列挙型のような型に対して String() メソッドを作成するコードジェネレーターを構築しましょう！この環境では実際の go generate を実行することはできませんが、型定義に基づいてプログラム的に Go のソースコードを生成するという、コード生成ツールが行うことをシミュレートします。

コードは2つのファイルに分けて構成します：

• generator.go: コード生成ロジックを作成します。

  Name (string) と Values (定数名を表す文字列のスライス) を持つ、列挙型のような型定義を表す EnumType 構造体を構築します。

  String() メソッドの Go ソースコードを生成する関数 GenerateStringMethod(e EnumType) string を作成します。生成されるコードは以下のパターンに従う必要があります：

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

  また、iota を使用して定数宣言を生成する関数 GenerateConstants(e EnumType) string も作成します：

  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 コードを生成します。