ODWRTW/canape
2
2
Fork 0
Code Issues 3 Pull Requests 1 Releases Activity

Add check of HTTP status when downloading

Browse Source
This commit is contained in:
Lucas BEE 2020-04-07 17:45:39 +02:00
parent 98eb833dc0
commit e5b4639cab
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/ratings/ratings.go
View File

@ -4,6 +4,7 @@ import (
"bufio" "bufio"
"bytes" "bytes"
"compress/gzip" "compress/gzip"
"fmt"
"io/ioutil" "io/ioutil"
"net/http" "net/http"
"strconv" "strconv"
@ -34,6 +35,10 @@ func Refresh(env *web.Env) error {
} }
defer resp.Body.Close() defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return fmt.Errorf("got HTTP error %s", resp.Status)
}
// Read all the file (~5MB) in memory // Read all the file (~5MB) in memory
// We do that because the ~1 000 000 upserts take too long, and the IMDB // We do that because the ~1 000 000 upserts take too long, and the IMDB
// server will cut our connection after ~2h // server will cut our connection after ~2h

5
backend/web/download.go
View File

@ -1,6 +1,7 @@
package web package web
import ( import (
"fmt"
"image" "image"
"image/jpeg" "image/jpeg"
"net/http" "net/http"
@ -26,6 +27,10 @@ var Download = func(srcURL, dest string, scale bool) error {
} }
defer resp.Body.Close() defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return fmt.Errorf("got HTTP error %s", resp.Status)
}
image, _, err := image.Decode(resp.Body) image, _, err := image.Decode(resp.Body)
if err != nil { if err != nil {
return err return err