Maîtriser les ENUM en Go

Souvent, au sein des systèmes que nous développons, nous rencontrons des valeurs constantes. Un exemple de ces valeurs pourrait être la représentation d’un statut d’enregistrement. Dans ce cas, envisagez un statut qui inclut davantage de variations au-delà de l'actif et de l'inactif.

Si ces statuts sont définis sous forme de chaînes, leur validation au sein du système pourrait devenir un casse-tête majeur. De plus, cette approche pourrait « gonfler » le binaire, car chaque validation impliquerait deux chaînes (la valeur attendue et la valeur en cours de validation).

Pour éviter ces problèmes, nous pouvons utiliser le type enum bien connu. Si vous n'êtes pas familier avec ce type, il s'agit essentiellement d'un type fixe ou de taille limitée.

Pour que ce soit plus clair, plongeons-nous dans du code. Suivant l'idée présentée précédemment, nous allons créer un type enum pour valider les statuts d'inscription.

Définir un nouveau type

Créer une énumération basée sur les types standard de Go peut être problématique. Laissez-moi vous expliquer. Imaginez que nous définissions notre statut comme un type uint8. Supposons maintenant que notre système ait également une autre énumération de type uint8 pour le genre.

Imaginez maintenant que la valeur 1 représente à la fois le statut En attente et le genre de musique Country. Que se passera-t-il si la validation if Pending == Country est effectuée ? Exactement, cela reviendra vrai.

Pour éviter cela, nous allons créer un nouveau type spécifiquement pour la gestion du statut. Ce type sera basé sur uint8, mais comme il s'agit d'un type distinct, la validation mentionnée précédemment ne retournera pas vrai.

type Status uint8

Création de l'ENUM

Une fois un nouveau type défini, créons les constantes et leurs valeurs correspondantes pour les statuts d'enregistrement.

const (
  Created Status = 0
  Pending = 1
  Approved = 2
  Rejected = 3
)

Bien qu'il n'y ait rien de mal en soi à attribuer des valeurs comme nous l'avons fait ci-dessus, il existe un moyen plus simple. Au lieu d'attribuer une valeur à chaque constante, nous pouvons utiliser le mot-clé iota. Ce mot-clé permet à Go d'attribuer 0 à la première constante, puis d'incrémenter la valeur de 1 séquentiellement pour chaque constante suivante.

const (
  Created Status = iota
  Pending
  Approved
  Rejected
)

Impression de l'ENUM

Comme l'énumération est actuellement implémentée, l'impression de la constante Created afficherait la valeur 0. Cependant, pour une meilleure lisibilité, il est plus utile d'afficher le mot Created au lieu de la valeur 0.

La solution est très simple. Implémentez simplement la méthode magique String().

func (s Status) String() string {
  switch s {
    case Created:
      return "created"
    case Pending:
      return "pending"
    case Approved:
      return "approved"
    case Rejected:
      return "rejected"
   }

   return "unknown"
}

Conclusion

Pour tester cela, faisons une simple impression du statut En attente.

package main

import "fmt"

type Status uint8

const (
  Created Status = iota
  Pending
  Approved
  Rejected
)

func (s Status) String() string {
  switch s {
    case Created:
      return "created"
    case Pending:
      return "pending"
    case Approved:
      return "approved"
    case Rejected:
      return "rejected"
  }

  return "unknown"
}

func main() {
  fmt.Println(Pending)
}

L'exécution de la commande go run devrait afficher en attente sur le terminal.

C'est ça ! J'espère que cela vous a été utile.

À la prochaine fois !

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!