Vue d’ensemble de la génération de code

Construisons un générateur de code qui crée des méthodes String() pour des types semblables à des énumérations ! Même si nous ne pouvons pas exécuter un véritable go generate dans cet environnement, vous simulerez ce que font les outils de génération de code : produire programmatiquement du code source Go à partir de 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 semblable à une énumération, avec un Name (chaîne) et des Values (slice de chaînes représentant les noms des constantes).

  Créez une fonction GenerateStringMethod(e EnumType) string qui produit du 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 à l'aide de 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 semblables à des énumérations.

  Lisez le nom du type d'énumération, puis lisez un nombre de valeurs suivi de chaque nom de valeur. Transmettez 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'énumération
• Ligne 2 : nombre de valeurs
• Lignes suivantes : chaque nom de valeur

Par exemple, avec :

Status
3
Pending
Active
Completed

Votre sortie doit ê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 avec :

Priority
2
Low
High

Votre sortie doit ê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 illustre le concept fondamental qui se trouve derrière des outils comme stringer : générer du code répétitif, mais sûr du point de vue des types, au moment de la compilation plutôt que d'utiliser la réflexion à l'exécution. Votre générateur produit du code Go valide qui pourrait être enregistré dans un fichier et compilé avec le reste d'un projet.