rest.go 1.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. package rest
  2. import (
  3. "bytes"
  4. "encoding/json"
  5. "fmt"
  6. "io"
  7. "net/http"
  8. )
  9. // JSONDecoder decodes json from given bytes buffer to target object.
  10. func JSONDecoder(data *bytes.Buffer, target interface{}) error {
  11. err := json.Unmarshal(data.Bytes(), target)
  12. return err
  13. }
  14. // GET returns bytes buffer of response body.
  15. func GET(url string) (*bytes.Buffer, error) {
  16. resp, err := http.Get(url)
  17. if resp != nil {
  18. defer resp.Body.Close()
  19. }
  20. if err != nil {
  21. return nil, fmt.Errorf("Response error: %v", err)
  22. }
  23. body, err := BodyToBuffer(resp.Body)
  24. if err != nil {
  25. return nil, fmt.Errorf("Error reading body: %v", err)
  26. }
  27. if resp.StatusCode != http.StatusOK && resp.StatusCode != http.StatusPartialContent {
  28. return body, fmt.Errorf("Request unsuccessful: %v - %v", resp.Status, url)
  29. }
  30. return body, nil
  31. }
  32. // BodyToBuffer reads data from ReadCloser and returns bytes buffer.
  33. func BodyToBuffer(data io.ReadCloser) (*bytes.Buffer, error) {
  34. var buffer bytes.Buffer
  35. _, err := buffer.ReadFrom(data)
  36. if err != nil {
  37. return nil, err
  38. }
  39. return &buffer, nil
  40. }