massive: move a lot of things
This commit is contained in:
44
x-tba/swapi-api-client/request.go
Normal file
44
x-tba/swapi-api-client/request.go
Normal file
@ -0,0 +1,44 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
// MaxResponseSize limits the response bytes from the API
|
||||
const MaxResponseSize = 2 << 16
|
||||
|
||||
// creating a robust http getter (lecture? :))
|
||||
func request(ctx context.Context, url string) ([]byte, error) {
|
||||
req, err := http.NewRequest("GET", url, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
req = req.WithContext(ctx)
|
||||
|
||||
resp, err := http.DefaultClient.Do(req)
|
||||
if err != nil {
|
||||
// Get the error from the context.
|
||||
// It may contain more useful data.
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
err = ctx.Err()
|
||||
default:
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return nil, fmt.Errorf("Bad Status: %d", resp.StatusCode)
|
||||
}
|
||||
|
||||
// Prevents the api to shoot us unlimited amount of data
|
||||
r := io.LimitReader(resp.Body, MaxResponseSize)
|
||||
|
||||
return ioutil.ReadAll(r)
|
||||
}
|
Reference in New Issue
Block a user