Panoramica sulla generazione del codice

Costruiamo un generatore di codice che crea metodi String() per tipi simili a enum! Anche se in questo ambiente non possiamo eseguire un vero go generate, simulerai ciò che fanno gli strumenti di generazione del codice: produrre codice sorgente Go programmaticamente in base alle definizioni dei tipi.

Organizzerai il tuo codice in due file:

• generator.go: crea la logica di generazione del codice.

  Crea una struct EnumType che rappresenta la definizione di un tipo simile a enum con un Name (stringa) e Values (slice di stringhe che rappresentano i nomi delle costanti).

  Crea una funzione GenerateStringMethod(e EnumType) string che produce codice sorgente Go per un metodo String(). Il codice generato dovrebbe seguire questo schema:

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

  Crea anche una funzione GenerateConstants(e EnumType) string che produce le dichiarazioni delle costanti usando iota:

  const (
      Value1 TypeName = iota
      Value2
      Value3
  )

  Infine, crea GenerateFullCode(e EnumType) string per combinare entrambi gli output con un commento di intestazione che indica che si tratta di codice generato:

  // Code generated by enum generator. DO NOT EDIT.
  
  [constants]
  
  [string method]
• main.go: usa il tuo generatore per produrre codice per diversi tipi enum.

  Leggi il nome del tipo enum, quindi leggi un numero di valori seguito dal nome di ciascun valore. Passa queste informazioni al tuo generatore e stampa il codice generato completo.

Verranno forniti i seguenti input:

• Riga 1: nome del tipo enum
• Riga 2: numero di valori
• Righe successive: nome di ciascun valore

Per esempio, dato:

Status
3
Pending
Active
Completed

Il tuo output dovrebbe essere:

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

E dato:

Priority
2
Low
High

Il tuo output dovrebbe essere:

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

Questo esercizio illustra il concetto fondamentale alla base di strumenti come stringer: generare codice ripetitivo ma sicuro rispetto ai tipi in fase di compilazione, anziché usare la reflection in fase di esecuzione. Il tuo generatore produce codice Go valido che potrebbe essere salvato in un file e compilato insieme al resto di un progetto.