chore(database): Ensure response body is always drained

gabe565 · Aug 13, 2024 · 762e2fd · 762e2fd
1 parent f80d683
commit 762e2fd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/internal/database/generate/download/downloader.go b/internal/database/generate/download/downloader.go
@@ -272,6 +272,10 @@ func (g *Downloader) prepareDownload(ctx context.Context, postFields map[string]
 	if err != nil {
 		return "", "", "", err
 	}
+	defer func() {
+		_, _ = io.Copy(io.Discard, res.Body)
+		_ = res.Body.Close()
+	}()
 
 	doc, err := goquery.NewDocumentFromReader(res.Body)
 	if err != nil {
@@ -298,10 +302,6 @@ func (g *Downloader) prepareDownload(ctx context.Context, postFields map[string]
 		return "", "", "", ErrNoButton
 	}
 
-	// Drain response body
-	_, _ = io.Copy(io.Discard, res.Body)
-	_ = res.Body.Close()
-
 	return url.String(), name, value, nil
 }