Aperçu de la génération de code

Construisons un générateur de code qui crée des méthodes String() pour des types de type enum ! Bien que nous ne puissions pas exécuter un véritable go generate dans cet environnement, vous allez simuler ce que font les outils de génération de code — produire du code source Go par programmation basé sur des définitions de types.

Vous organiserez votre code sur deux fichiers :

• generator.go : Créez votre logique de génération de code.

  Construisez une structure EnumType qui représente une définition de type de type enum avec un Name (string) et des Values (slice de strings représentant les noms des constantes).

  Créez une fonction GenerateStringMethod(e EnumType) string qui produit le code source Go pour une méthode String(). Le code généré doit suivre ce modèle :

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

  Créez également une fonction GenerateConstants(e EnumType) string qui produit les déclarations de constantes en utilisant iota :

  const (
      Value1 TypeName = iota
      Value2
      Value3
  )

  Enfin, créez GenerateFullCode(e EnumType) string qui combine les deux sorties avec un commentaire d'en-tête indiquant qu'il s'agit de code généré :

  // Code generated by enum generator. DO NOT EDIT.
  
  [constants]
  
  [string method]
• main.go : Utilisez votre générateur pour produire du code pour différents types d'enum.

  Lisez le nom du type d'enum, puis lisez le nombre de valeurs suivi de chaque nom de valeur. Passez ces informations à votre générateur et affichez le code généré complet.

Les entrées suivantes seront fournies :

• Ligne 1 : Nom du type d'enum
• Ligne 2 : Nombre de valeurs
• Lignes suivantes : Chaque nom de valeur

Par exemple, étant donné :

Status
3
Pending
Active
Completed

Votre sortie devrait être :

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

Et étant donné :

Priority
2
Low
High

Votre sortie devrait être :

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

Cet exercice démontre le concept de base derrière des outils comme stringer — générer du code répétitif mais sûr au niveau des types au moment de la compilation plutôt que d'utiliser la réflexion au moment de l'exécution. Votre générateur produit du code Go valide qui pourrait être sauvegardé dans un fichier et compilé avec le reste d'un projet.