diff --git a/pkg/cluster/wireguard.go b/pkg/cluster/wireguard.go index 932fe32..ce71b88 100644 --- a/pkg/cluster/wireguard.go +++ b/pkg/cluster/wireguard.go @@ -97,10 +97,14 @@ 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 { - if ring == rings.RingZeroID || ring == rings.RingOneID { + switch ring { + case rings.RingZeroID: 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 {