configurable parallelism for memory broker and presence manager

centrifugal · Nov 5, 2023 · cc4829a · cc4829a
1 parent 8c0b535
commit cc4829a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/broker_memory_test.go b/broker_memory_test.go
@@ -309,7 +309,7 @@ func BenchmarkMemoryPublish_1Ch(b *testing.B) {
 	defer func() { _ = e.node.Shutdown(context.Background()) }()
 
 	rawData := protocol.Raw(`{"bench": true}`)
-	b.SetParallelism(128)
+	b.SetParallelism(getBenchParallelism())
 	b.ResetTimer()
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {
@@ -327,7 +327,7 @@ func BenchmarkMemoryPublish_History_1Ch(b *testing.B) {
 
 	rawData := protocol.Raw(`{"bench": true}`)
 	chOpts := PublishOptions{HistorySize: 100, HistoryTTL: 60 * time.Second}
-	b.SetParallelism(128)
+	b.SetParallelism(getBenchParallelism())
 	b.ResetTimer()
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {

diff --git a/presence_memory_test.go b/presence_memory_test.go
@@ -102,6 +102,7 @@ func BenchmarkMemoryAddPresence_OneChannel_Parallel(b *testing.B) {
 	defer func() { _ = e.node.Shutdown(context.Background()) }()
 
 	b.ResetTimer()
+	b.SetParallelism(getBenchParallelism())
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {
 			err := e.AddPresence("channel", "uid", &ClientInfo{})
@@ -131,6 +132,7 @@ func BenchmarkMemoryPresence_OneChannel_Parallel(b *testing.B) {
 	defer func() { _ = e.node.Shutdown(context.Background()) }()
 
 	_ = e.AddPresence("channel", "uid", &ClientInfo{})
+	b.SetParallelism(getBenchParallelism())
 	b.ResetTimer()
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {