Fixed a goroutine leak

AudDMusic · Feb 1, 2021 · 75ff09e · 75ff09e
1 parent e166576
commit 75ff09e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/examples/scanFiles/main.go b/examples/scanFiles/main.go
@@ -121,6 +121,7 @@ func RecognizeMultipleFiles(dir string, client *audd.Client, setID3, short bool,
 					cover = "https://i3.ytimg.com/vi/"+strings.ReplaceAll(result[i].SongLink, "https://youtu.be/", "")+"/maxresdefault.jpg"
 				}
 				response, err := http.Get(cover)
+				defer closeBody(response)
 				if err != nil {
 					fmt.Println("Error: can't get the cover for", fileName, err)
 				} else {
@@ -198,6 +199,16 @@ func getCurrentDir() string {
 	return filepath.Dir(currentFile)
 }
 
+func closeBody(resp *http.Response) {
+	if resp == nil {
+		return
+	}
+	if resp.Body == nil {
+		return
+	}
+	_ = resp.Body.Close()
+}
+
 func main() {
 	dir := flag.String("dir", getCurrentDir(), "The directory with the files")
 	apiToken := flag.String("api_token", "test", "AudD API token")