add missed lock for Count

ysmood · Jul 8, 2020 · 1965fb4 · 1965fb4
1 parent 7d32720
commit 1965fb4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/basic_test.go b/basic_test.go
@@ -25,7 +25,7 @@ func TestNew(t *testing.T) {
 
 	ob := goob.New()
 	s := ob.Subscribe(ctx)
-	size := 10000
+	size := 1000
 
 	expected := []int{}
 	go func() {
@@ -66,7 +66,7 @@ func TestMultipleConsumers(t *testing.T) {
 	s1 := ob.Subscribe(ctx)
 	s2 := ob.Subscribe(ctx)
 	s3 := ob.Subscribe(ctx)
-	size := 10000
+	size := 1000
 
 	expected := []int{}
 	go func() {
@@ -187,7 +187,7 @@ func TestFilter(t *testing.T) {
 func TestMonkey(t *testing.T) {
 	wg := sync.WaitGroup{}
 	count := int32(0)
-	roundSize := 10000
+	roundSize := 1000
 	size := 100
 
 	run := func() {

diff --git a/main.go b/main.go
@@ -61,6 +61,8 @@ func (ob *Observable) Publish(e Event) {
 
 // Count of the subscribers
 func (ob *Observable) Count() int {
+	ob.lock.Lock()
+	defer ob.lock.Unlock()
 	return len(ob.subscribers)
 }