some comment updates

couchbase · Apr 6, 2023 · c793086 · c793086
1 parent 4c7977c
commit c793086
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/db/sg_replicate_cfg.go b/db/sg_replicate_cfg.go
@@ -1191,8 +1191,10 @@ func (c *SGRCluster) GetReplicationIDsForNode(nodeUUID string) (replicationIDs [
 	return replicationIDs
 }
 
-func (m *sgReplicateManager) GetActiveReplicators() map[string]*ActiveReplicator {
-	return m.activeReplicators
+func (m *sgReplicateManager) GetNumberActiveReplicators() int {
+	m.activeReplicatorsLock.Lock()
+	defer m.activeReplicatorsLock.Unlock()
+	return len(m.activeReplicators)
 }
 
 // RebalanceReplications distributes the set of defined replications across the set of available nodes

diff --git a/rest/utilities_testing_resttester.go b/rest/utilities_testing_resttester.go
@@ -128,8 +128,8 @@ func (rt *RestTester) WaitForCheckpointLastSequence(expectedName string) (string
 
 func (rt *RestTester) WaitForActiveReplicatorInitialization(count int) {
 	successFunc := func() bool {
-		ar := rt.GetDatabase().SGReplicateMgr.GetActiveReplicators()
-		return len(ar) == count
+		ar := rt.GetDatabase().SGReplicateMgr.GetNumberActiveReplicators()
+		return ar == count
 	}
 	require.NoError(rt.TB, rt.WaitForCondition(successFunc), "mismatch on number of active replicators")
 }