From bdb38dfa576613f1cc5b1a19859c8858dff0edda Mon Sep 17 00:00:00 2001 From: Maycon Santos Date: Sun, 20 Jul 2025 21:31:59 +0200 Subject: [PATCH] use relay manager.getClientLen() --- relay/client/manager.go | 6 ++++++ relay/client/manager_test.go | 4 ++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/relay/client/manager.go b/relay/client/manager.go index 0fb682d95..a4855e023 100644 --- a/relay/client/manager.go +++ b/relay/client/manager.go @@ -350,6 +350,12 @@ func (m *Manager) startCleanupLoop() { } } +func (m *Manager) getClientLen() int { + m.relayClientsMutex.Lock() + defer m.relayClientsMutex.Unlock() + return len(m.relayClients) +} + func (m *Manager) cleanUpUnusedRelays() { m.relayClientsMutex.Lock() defer m.relayClientsMutex.Unlock() diff --git a/relay/client/manager_test.go b/relay/client/manager_test.go index 52f2833e4..96d46b09e 100644 --- a/relay/client/manager_test.go +++ b/relay/client/manager_test.go @@ -292,8 +292,8 @@ func TestForeginAutoClose(t *testing.T) { timeout := relayCleanupInterval + keepUnusedServerTime + 1*time.Second t.Logf("waiting for relay cleanup: %s", timeout) time.Sleep(timeout) - if len(mgr.relayClients) != 0 { - t.Errorf("expected 0, got %d", len(mgr.relayClients)) + if mgr.getClientLen() != 0 { + t.Errorf("expected 0, got %d", mgr.getClientLen()) } t.Logf("closing manager")