From 70ed0845061feee07ab372a56c8d2b79df324106 Mon Sep 17 00:00:00 2001 From: bcmmbaga Date: Wed, 15 Jan 2025 11:12:14 +0300 Subject: [PATCH] Refactor GetValidatedPeers to accept slices of peers and groups --- integrations/validator.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/integrations/validator.go b/integrations/validator.go index f8607dd..eaf870e 100644 --- a/integrations/validator.go +++ b/integrations/validator.go @@ -5,8 +5,8 @@ import ( "github.com/netbirdio/netbird/management/server/account" "github.com/netbirdio/netbird/management/server/activity" - "github.com/netbirdio/netbird/management/server/group" nbpeer "github.com/netbirdio/netbird/management/server/peer" + "github.com/netbirdio/netbird/management/server/types" ) type IntegratedValidatorImpl struct { @@ -32,10 +32,10 @@ func (v *IntegratedValidatorImpl) IsNotValidPeer(_ context.Context, _ string, _ return false, false, nil } -func (v *IntegratedValidatorImpl) GetValidatedPeers(_ string, _ map[string]*group.Group, peers map[string]*nbpeer.Peer, _ *account.ExtraSettings) (map[string]struct{}, error) { +func (v *IntegratedValidatorImpl) GetValidatedPeers(_ string, _ []*types.Group, peers []*nbpeer.Peer, _ *account.ExtraSettings) (map[string]struct{}, error) { validatedPeers := make(map[string]struct{}) - for p := range peers { - validatedPeers[p] = struct{}{} + for _, p := range peers { + validatedPeers[p.ID] = struct{}{} } return validatedPeers, nil }