Skip to content

Commit

Permalink
Update API to check response content length before decoding.
Browse files Browse the repository at this point in the history
The API decodeBody function will now check the content length
before attempting to decode. If the length is zero, and the out
interface is nil then it is safe to assume the API call is not
returning any data to the user. This allows us to better handle
passing nil to API calls in a single place.
  • Loading branch information
jrasell committed Jul 29, 2019
1 parent a746aec commit 5673b10
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 8 deletions.
19 changes: 13 additions & 6 deletions api/api.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ import (
"compress/gzip"
"crypto/tls"
"encoding/json"
"errors"
"fmt"
"io"
"net"
Expand Down Expand Up @@ -769,10 +770,8 @@ func (c *Client) query(endpoint string, out interface{}, q *QueryOptions) (*Quer
parseQueryMeta(resp, qm)
qm.RequestTime = rtt

if out != nil {
if err := decodeBody(resp, out); err != nil {
return nil, err
}
if err := decodeBody(resp, out); err != nil {
return nil, err
}
return qm, nil
}
Expand Down Expand Up @@ -897,8 +896,16 @@ func parseWriteMeta(resp *http.Response, q *WriteMeta) error {

// decodeBody is used to JSON decode a body
func decodeBody(resp *http.Response, out interface{}) error {
dec := json.NewDecoder(resp.Body)
return dec.Decode(out)
switch resp.ContentLength {
case 0:
if out == nil {
return nil
}
return errors.New("Got 0 byte response with non-nil decode object")
default:
dec := json.NewDecoder(resp.Body)
return dec.Decode(out)
}
}

// encodeBody is used to encode a request body
Expand Down
3 changes: 1 addition & 2 deletions api/nodes.go
Original file line number Diff line number Diff line change
Expand Up @@ -420,9 +420,8 @@ func (n *Nodes) GC(nodeID string, q *QueryOptions) error {

// TODO Add tests
func (n *Nodes) GcAlloc(allocID string, q *QueryOptions) error {
var resp struct{}
path := fmt.Sprintf("/v1/client/allocation/%s/gc", allocID)
_, err := n.client.query(path, &resp, q)
_, err := n.client.query(path, nil, q)
return err
}

Expand Down

0 comments on commit 5673b10

Please sign in to comment.