fixed loop signaling

kubescape · Nov 8, 2021 · 25e7edc · 25e7edc
1 parent c4e0136
commit 25e7edc
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/gitregostore/gitstoreutils.go b/gitregostore/gitstoreutils.go
@@ -71,11 +71,15 @@ func (gs *GitRegoStore) setObjectsFromRepoLoop() {
 	wg.Add(1)
 
 	go func() {
+		f := true
 		for {
 			if err := gs.setObjectsFromRepoOnce(); err != nil {
 				fmt.Println(err)
 			}
-			wg.Done()
+			if f {
+				wg.Done() // first update to done
+				f = false
+			}
 			if !gs.Watch {
 				return
 			}
@@ -200,11 +204,15 @@ func (gs *GitRegoStore) setObjectsFromReleaseLoop() {
 	var wg sync.WaitGroup
 	wg.Add(1)
 	go func() {
+		f := true
 		for {
 			if err := gs.setObjectsFromReleaseOnce(); err != nil {
 				fmt.Println(err)
 			}
-			wg.Done()
+			if f {
+				wg.Done() // first update to done
+				f = false
+			}
 			if !gs.Watch {
 				return
 			}