Überblick über Codegenerierung

Erstellen wir einen Codegenerator, der String()-Methoden für enumartige Typen erzeugt! Obwohl wir in dieser Umgebung kein tatsächliches go generate ausführen können, simulierst du, was Codegenerierungstools tun: Go-Quellcode programmgesteuert auf Grundlage von Typdefinitionen erzeugen.

Du organisierst deinen Code über zwei Dateien:

• generator.go: Erstelle deine Logik zur Codegenerierung.

  Erstelle eine EnumType-Struktur, die eine enumartige Typdefinition mit einem Name (String) und Values (Slice von Strings, die die Namen der Konstanten darstellen) repräsentiert.

  Erstelle eine Funktion GenerateStringMethod(e EnumType) string, die Go-Quellcode für eine String()-Methode erzeugt. Der generierte Code sollte diesem Muster folgen:

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

  Erstelle außerdem eine Funktion GenerateConstants(e EnumType) string, die mithilfe von iota die Konstantendeklarationen erzeugt:

  const (
      Value1 TypeName = iota
      Value2
      Value3
  )

  Erstelle schließlich GenerateFullCode(e EnumType) string, das beide Ausgaben mit einem Kopfkommentar kombiniert, der angibt, dass es sich um generierten Code handelt:

  // Code generated by enum generator. DO NOT EDIT.
  
  [constants]
  
  [string method]
• main.go: Verwende deinen Generator, um Code für verschiedene enumartige Typen zu erzeugen.

  Lies den Namen des enumartigen Typs ein, danach eine Anzahl von Werten und anschließend jeden Wertnamen. Übergib diese Informationen an deinen Generator und gib den vollständig generierten Code aus.

Die folgenden Eingaben werden bereitgestellt:

• Zeile 1: Name des enumartigen Typs
• Zeile 2: Anzahl der Werte
• Folgende Zeilen: Jeder Wertname

Zum Beispiel bei:

Status
3
Pending
Active
Completed

sollte deine Ausgabe folgendermaßen aussehen:

// 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"
    }
}

Und bei:

Priority
2
Low
High

sollte deine Ausgabe folgendermaßen aussehen:

// 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"
    }
}

Diese Übung veranschaulicht das grundlegende Konzept hinter Tools wie stringer: Wiederholenden, aber typsicheren Code zur Kompilierzeit zu erzeugen, anstatt zur Laufzeit Reflection zu verwenden. Dein Generator erzeugt gültigen Go-Code, der in einer Datei gespeichert und zusammen mit dem restlichen Projekt kompiliert werden könnte.