Learning Golang: Pub/Sub Concurrency Pattern

https://youtu.be/s-I3Bs3ZUsY
MarioCarrion · Nov 6, 2023 · ce3a09d · ce3a09d
1 parent 30565d9
commit ce3a09d
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/2023/concurrency-pattern-publisher-subscriber/main.go b/2023/concurrency-pattern-publisher-subscriber/main.go
@@ -69,15 +69,16 @@ func main() {
 
 	s1 := ps.Subscribe()
 
+	wg.Add(1)
+
 	go func() {
-		wg.Add(1)
+		defer wg.Done()
 
 		for {
 			select {
 			case val, ok := <-s1:
 				if !ok {
 					fmt.Print("sub 1, exiting\n")
-					wg.Done()
 					return
 				}
 
@@ -90,15 +91,15 @@ func main() {
 
 	s2 := ps.Subscribe()
 
+	wg.Add(1)
+
 	go func() {
-		wg.Add(1)
+		defer wg.Done()
 
 		for val := range s2 {
 			fmt.Println("sub 2, value ", val)
 		}
 
-		wg.Done()
-
 		fmt.Print("sub 2, exiting\n")
 	}()