Adding waiting groups to wait for all routines finishing

data.go

@@ -6,12 +6,15 @@ package main
 
 import (
 	"fmt"
+	"sync"
 	"time"
 
 	"github.com/fedir/ghstat/github"
 )
 
-func repositoryData(rKey string, tmpFolder string, debug bool, dataChan chan Repository) {
+func repositoryData(rKey string, tmpFolder string, debug bool, dataChan chan Repository, wg *sync.WaitGroup) {
+
+	defer wg.Done()
 
 	r := new(Repository)
 
@@ -43,7 +46,7 @@ func repositoryData(rKey string, tmpFolder string, debug bool, dataChan chan Rep
 			r.AuthorLocation = authorData.Location
 		}
 	} else {
-		r.Author = "[Account removed]"
+		r.Author = "[Unknown account]"
 	}
 
 	r.ClosedIssues = 0

ghstat.go

@@ -9,6 +9,7 @@ import (
 	"flag"
 	"os"
 	"strings"
+	"sync"
 
 	"github.com/fedir/ghstat/github"
 )
@@ -58,14 +59,17 @@ func main() {
 		csvFilePath = "result.csv"
 	}
 	var ghData = []Repository{}
+	var wg sync.WaitGroup
+	wg.Add(len(repositoriesKeys))
 
 	dataChan := make(chan Repository, len(repositoriesKeys))
 	for _, rKey := range repositoriesKeys {
-		go repositoryData(rKey, *tmpFolder, *debug, dataChan)
+		go repositoryData(rKey, *tmpFolder, *debug, dataChan, &wg)
 	}
 	for range repositoriesKeys {
 		ghData = append(ghData, <-dataChan)
 	}
+	wg.Wait()
 	rateAndPrintGreetings(ghData)
 	writeCSVStatistics(ghData, csvFilePath)
 }