Increment the waitGroup outside of the go routines

m7kvqbe1 · Jul 9, 2024 · c4284f9 · c4284f9
1 parent cd3054a
commit c4284f9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -25,7 +25,6 @@ type Scraper struct {
 }
 
 func (s *Scraper) sendText(number string) {
-	s.waitGroup.Add(1)
 	defer s.waitGroup.Done()
 
 	key := os.Getenv("SMS_KEY")
@@ -69,7 +68,6 @@ func (s *Scraper) sendText(number string) {
 }
 
 func (s *Scraper) fetch() {
-	s.waitGroup.Add(1)
 	defer s.waitGroup.Done()
 
 	c := colly.NewCollector(
@@ -116,6 +114,7 @@ func (s *Scraper) success() {
 
 	phoneNumbers := strings.Split(os.Getenv("PHONE_NUMBERS"), ",")
 	for _, number := range phoneNumbers {
+		s.waitGroup.Add(1)
 		go s.sendText(number)
 	}
 
@@ -140,6 +139,7 @@ func (s *Scraper) scrapeLoop() {
 		case <-s.done:
 			return
 		case <-ticker.C:
+			s.waitGroup.Add(1)
 			go s.fetch()
 		}
 	}