Browse Source

slight refactor odl.go

master
Alexander Avery 3 months ago
parent
commit
c1cff821ec
  1. 14
      odl.go

14
odl.go

@ -43,7 +43,6 @@ func download(n int, root, name, uri string) error {
if err != nil { if err != nil {
return fmt.Errorf("parsing uri %s: %v", uri, err) return fmt.Errorf("parsing uri %s: %v", uri, err)
} }
ext := filepath.Ext(u.Path)
response, err := http.Get(uri) response, err := http.Get(uri)
switch { switch {
@ -51,14 +50,14 @@ func download(n int, root, name, uri string) error {
return fmt.Errorf("downloading file %s: %v", uri, err) return fmt.Errorf("downloading file %s: %v", uri, err)
case response.StatusCode != http.StatusOK: case response.StatusCode != http.StatusOK:
return fmt.Errorf("downloading file %s: status code %d", uri, response.StatusCode) return fmt.Errorf("downloading file %s: status code %d", uri, response.StatusCode)
default:
defer response.Body.Close()
} }
defer response.Body.Close()
var r io.Reader = response.Body var r io.Reader = response.Body
ext := filepath.Ext(u.Path)
if ext == "" { if ext == "" {
var err error ext, r, err = guessExt(r)
ext, r, err = guessExt(response.Body)
if err != nil { if err != nil {
return err return err
} }
@ -66,11 +65,10 @@ func download(n int, root, name, uri string) error {
filename := filepath.Join(root, name+strconv.Itoa(n)+ext) filename := filepath.Join(root, name+strconv.Itoa(n)+ext)
f, err := os.Create(filename) f, err := os.Create(filename)
if err != nil { if err == nil {
return err _, err = io.Copy(f, r)
} }
_, err = io.Copy(f, r)
return err return err
} }

Loading…
Cancel
Save