sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-04-16 07:16:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

disable sync lim (#5233)

Browse Source
This commit is contained in:
Vlad
2026-02-01 14:37:00 +01:00
committed by GitHub
parent 3a0cf230a1
commit 7b830d8f72
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
management/internals/shared/grpc/server.go
View File

@@ -77,8 +77,9 @@ type Server struct {
oAuthConfigProvider idp.OAuthConfigProvider oAuthConfigProvider idp.OAuthConfigProvider
syncSem atomic.Int32 syncSem atomic.Int32
syncLim int32 syncLimEnabled bool
syncLim int32
} }
// NewServer creates a new Management server // NewServer creates a new Management server
@@ -108,6 +109,7 @@ func NewServer(
blockPeersWithSameConfig := strings.ToLower(os.Getenv(envBlockPeers)) == "true" blockPeersWithSameConfig := strings.ToLower(os.Getenv(envBlockPeers)) == "true"
syncLim := int32(defaultSyncLim) syncLim := int32(defaultSyncLim)
syncLimEnabled := true
if syncLimStr := os.Getenv(envConcurrentSyncs); syncLimStr != "" { if syncLimStr := os.Getenv(envConcurrentSyncs); syncLimStr != "" {
syncLimParsed, err := strconv.Atoi(syncLimStr) syncLimParsed, err := strconv.Atoi(syncLimStr)
if err != nil { if err != nil {
@@ -115,6 +117,9 @@ func NewServer(
} else { } else {
//nolint:gosec //nolint:gosec
syncLim = int32(syncLimParsed) syncLim = int32(syncLimParsed)
if syncLim < 0 {
syncLimEnabled = false
}
} }
} }
@@ -134,7 +139,8 @@ func NewServer(
loginFilter: newLoginFilter(), loginFilter: newLoginFilter(),
syncLim: syncLim, syncLim: syncLim,
syncLimEnabled: syncLimEnabled,
}, nil }, nil
} }
@@ -212,7 +218,7 @@ func (s *Server) Job(srv proto.ManagementService_JobServer) error {
// Sync validates the existence of a connecting peer, sends an initial state (all available for the connecting peers) and // Sync validates the existence of a connecting peer, sends an initial state (all available for the connecting peers) and
// notifies the connected peer of any updates (e.g. new peers under the same account) // notifies the connected peer of any updates (e.g. new peers under the same account)
func (s *Server) Sync(req *proto.EncryptedMessage, srv proto.ManagementService_SyncServer) error { func (s *Server) Sync(req *proto.EncryptedMessage, srv proto.ManagementService_SyncServer) error {
if s.syncSem.Load() >= s.syncLim { if s.syncLimEnabled && s.syncSem.Load() >= s.syncLim {
return status.Errorf(codes.ResourceExhausted, "too many concurrent sync requests, please try again later") return status.Errorf(codes.ResourceExhausted, "too many concurrent sync requests, please try again later")
} }
s.syncSem.Add(1) s.syncSem.Add(1)