Fixed a possible goroutine leak

AudDMusic · Feb 1, 2021 · e166576 · e166576
1 parent d0f0459
commit e166576
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/examples/StreamTracksToSpotifyPlaylist/main.go b/examples/StreamTracksToSpotifyPlaylist/main.go
@@ -232,7 +232,8 @@ func writeResult(song audd.RecognitionResult) {
  return
  }
  req.Header.Set("Content-Type", "application/json")
- _, err = spotifyClient.h.Do(req)
+ resp, err := spotifyClient.h.Do(req)
+ defer closeBody(resp)
  if capture(err) {
  return
  }
@@ -241,6 +242,16 @@ func writeResult(song audd.RecognitionResult) {
  fmt.Printf("Added a song to the playlist (%s - %s, %s)\n", song.Artist, song.Title, song.SongLink)
 }
 
+func closeBody(resp *http.Response) {
+ if resp == nil {
+ return
+ }
+ if resp.Body == nil {
+ return
+ }
+ _ = resp.Body.Close()
+}
+
 func capture(err error) bool {
  if err == nil {
  return false