From 00f94493e3001d55c5c1478165459e9e204a46f1 Mon Sep 17 00:00:00 2001
From: Tovi Jaeschke-Rogers <tovi.jaeschke-rogers@four.io>
Date: Sat, 24 Jul 2021 17:47:26 +0930
Subject: [PATCH] Prevent negative waitgroup counter bug

---
 Client/Filesystem/FilesystemDiff.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/Client/Filesystem/FilesystemDiff.go b/Client/Filesystem/FilesystemDiff.go
index 63887a3..01df121 100644
--- a/Client/Filesystem/FilesystemDiff.go
+++ b/Client/Filesystem/FilesystemDiff.go
@@ -116,7 +116,6 @@ func (fsStatus *FilesystemStatus) parseFile(indexBucket, picksBucket *bolt.Bucke
 
 	defer func() {
 		bar.Increment()
-		fsStatusWG.Done()
 	}()
 
 	pick = picksBucket.Get([]byte(p))
@@ -126,6 +125,7 @@ func (fsStatus *FilesystemStatus) parseFile(indexBucket, picksBucket *bolt.Bucke
 		fsStatusWG.Wait()
 		fsStatusWG.Add(1)
 		fsStatus.PickedFiles = append(fsStatus.PickedFiles, p)
+		fsStatusWG.Done()
 		return
 	}
 
@@ -145,6 +145,7 @@ func (fsStatus *FilesystemStatus) parseFile(indexBucket, picksBucket *bolt.Bucke
 			fsStatusWG.Wait()
 			fsStatusWG.Add(1)
 			fsStatus.ModifiedFiles = append(fsStatus.ModifiedFiles, p)
+			fsStatusWG.Done()
 		}
 
 		return
@@ -153,6 +154,7 @@ func (fsStatus *FilesystemStatus) parseFile(indexBucket, picksBucket *bolt.Bucke
 	fsStatusWG.Wait()
 	fsStatusWG.Add(1)
 	fsStatus.NewFiles = append(fsStatus.NewFiles, p)
+	fsStatusWG.Done()
 
 	return
 }