diff --git a/pkg/cluster/wireguard.go b/pkg/cluster/wireguard.go index ce71b88..932fe32 100644 --- a/pkg/cluster/wireguard.go +++ b/pkg/cluster/wireguard.go @@ -97,14 +97,10 @@ func (m *Cluster) WriteWireguardConfig(ring rings.RingID) error { // WriteWireguardConfig rewrites all wgN.conf on all machines // on the Zone attached to that ring func (z *Zone) WriteWireguardConfig(ring rings.RingID) error { - switch ring { - case rings.RingZeroID: + if ring == rings.RingZeroID || ring == rings.RingOneID { return writeWireguardConfig(z.zones, z.zones, ring) - case rings.RingOneID: - return writeWireguardConfig(z.zones, z, ring) - default: - return ErrInvalidRing(ring) } + return ErrInvalidRing(ring) } func writeWireguardConfig(z ZoneIterator, m MachineIterator, ring rings.RingID) error {