From 14dd717c7499d6adb4b1ecfb9b971636aa226e81 Mon Sep 17 00:00:00 2001
From: Tovi Jaeschke-Rogers <tovi.jaeschke-rogers@four.io>
Date: Sat, 24 Jul 2021 17:19:19 +0930
Subject: [PATCH] Reconfigure waitgroup placing during file system diff

---
 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 9680d24..d944164 100644
--- a/Client/Filesystem/FilesystemDiff.go
+++ b/Client/Filesystem/FilesystemDiff.go
@@ -116,6 +116,7 @@ func (fsStatus *FilesystemStatus) parseFile(indexBucket, picksBucket *bolt.Bucke
 
 	defer func() {
 		bar.Increment()
+		fsStatusWG.Done()
 	}()
 
 	pick = picksBucket.Get([]byte(p))
@@ -145,12 +146,13 @@ func (fsStatus *FilesystemStatus) parseFile(indexBucket, picksBucket *bolt.Bucke
 			fsStatusWG.Wait()
 			fsStatusWG.Add(1)
 			fsStatus.ModifiedFiles = append(fsStatus.ModifiedFiles, p)
-			fsStatusWG.Done()
 		}
 
 		return
 	}
 
+	fsStatusWG.Wait()
+	fsStatusWG.Add(1)
 	fsStatus.NewFiles = append(fsStatus.NewFiles, p)
 
 	return