server.go 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  1. package honeybadger
  2. import (
  3. "bytes"
  4. "errors"
  5. "fmt"
  6. "io/ioutil"
  7. "net/http"
  8. "net/url"
  9. "time"
  10. )
  11. // Errors returned by the backend when unable to successfully handle payload.
  12. var (
  13. ErrRateExceeded = errors.New("Rate exceeded: slow down!")
  14. ErrPaymentRequired = errors.New("Payment required: expired trial or credit card?")
  15. ErrUnauthorized = errors.New("Unauthorized: bad API key?")
  16. )
  17. func newServerBackend(config *Configuration) *server {
  18. return &server{
  19. URL: &config.Endpoint,
  20. APIKey: &config.APIKey,
  21. Client: &http.Client{
  22. Transport: http.DefaultTransport,
  23. Timeout: config.Timeout,
  24. },
  25. Timeout: &config.Timeout,
  26. }
  27. }
  28. type server struct {
  29. APIKey *string
  30. URL *string
  31. Timeout *time.Duration
  32. Client *http.Client
  33. }
  34. func (s *server) Notify(feature Feature, payload Payload) error {
  35. // Copy the value from the pointer in case it has changed in the
  36. // configuration.
  37. s.Client.Timeout = *s.Timeout
  38. url, err := url.Parse(*s.URL)
  39. if err != nil {
  40. return err
  41. }
  42. url.Path = "v1/" + feature.Endpoint
  43. req, err := http.NewRequest("POST", url.String(), bytes.NewReader(payload.toJSON()))
  44. if err != nil {
  45. return err
  46. }
  47. req.Header.Set("X-API-Key", *s.APIKey)
  48. req.Header.Set("Content-Type", "application/json")
  49. req.Header.Set("Accept", "application/json")
  50. resp, err := s.Client.Do(req)
  51. if err != nil {
  52. return err
  53. }
  54. defer func() {
  55. ioutil.ReadAll(resp.Body)
  56. resp.Body.Close()
  57. }()
  58. switch resp.StatusCode {
  59. case 201:
  60. return nil
  61. case 429, 503:
  62. return ErrRateExceeded
  63. case 402:
  64. return ErrPaymentRequired
  65. case 403:
  66. return ErrUnauthorized
  67. default:
  68. return fmt.Errorf(
  69. "request failed status=%d expected=%d",
  70. resp.StatusCode,
  71. http.StatusCreated)
  72. }
  73. }