From 0f62ee2e53e590dc308eaf57403087b68310614c Mon Sep 17 00:00:00 2001 From: Alejandro Mery Date: Thu, 24 Aug 2023 20:52:08 +0000 Subject: [PATCH] zones: rename Machine.RingAddresses to Machine.Rings Signed-off-by: Alejandro Mery --- pkg/zones/env.go | 2 +- pkg/zones/machine.go | 5 +---- pkg/zones/machine_rings.go | 10 +++++----- pkg/zones/scan.go | 6 +++--- 4 files changed, 10 insertions(+), 13 deletions(-) diff --git a/pkg/zones/env.go b/pkg/zones/env.go index f2ec1a2..35a1ff1 100644 --- a/pkg/zones/env.go +++ b/pkg/zones/env.go @@ -86,7 +86,7 @@ func getRingZeroGatewayID(z *Zone) int { firstNodeID = p.ID } - if _, found := p.getRingInfo(0); found { + if _, ok := p.getRingInfo(0); ok { gatewayID = p.ID } diff --git a/pkg/zones/machine.go b/pkg/zones/machine.go index 5a9713f..6b7effc 100644 --- a/pkg/zones/machine.go +++ b/pkg/zones/machine.go @@ -6,19 +6,16 @@ import ( "net/netip" "path/filepath" "strings" - "sync" ) // A Machine is a machine on a Zone type Machine struct { - mu sync.Mutex - zone *Zone ID int Name string `toml:"name"` PublicAddresses []netip.Addr `toml:"public,omitempty"` - RingAddresses []*RingInfo `toml:"rings,omitempty"` + Rings []*RingInfo `toml:"rings,omitempty"` } func (m *Machine) String() string { diff --git a/pkg/zones/machine_rings.go b/pkg/zones/machine_rings.go index 350091c..72919b4 100644 --- a/pkg/zones/machine_rings.go +++ b/pkg/zones/machine_rings.go @@ -123,9 +123,9 @@ func (m *Machine) applyWireguardConfig(ring int, wg *wireguard.Config) error { } func (m *Machine) getRingInfo(ring int) (*RingInfo, bool) { - for _, ri := range m.RingAddresses { + for _, ri := range m.Rings { if ri.Ring == ring { - return ri, true + return ri, ri.Enabled } } @@ -133,10 +133,10 @@ func (m *Machine) getRingInfo(ring int) (*RingInfo, bool) { } func (m *Machine) applyRingInfo(ring int, new *RingInfo) error { - cur, found := m.getRingInfo(ring) - if !found { + cur, _ := m.getRingInfo(ring) + if cur == nil { // first, append - m.RingAddresses = append(m.RingAddresses, new) + m.Rings = append(m.Rings, new) return nil } diff --git a/pkg/zones/scan.go b/pkg/zones/scan.go index 28b77d0..8caa443 100644 --- a/pkg/zones/scan.go +++ b/pkg/zones/scan.go @@ -102,9 +102,9 @@ func (m *Zones) scanSort() error { }) m.ForEachMachine(func(p *Machine) bool { - sort.SliceStable(p.RingAddresses, func(i, j int) bool { - ri1 := p.RingAddresses[i] - ri2 := p.RingAddresses[j] + sort.SliceStable(p.Rings, func(i, j int) bool { + ri1 := p.Rings[i] + ri2 := p.Rings[j] return ri1.Ring < ri2.Ring })