error.go 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. package s3err
  2. import (
  3. "fmt"
  4. "github.com/aws/aws-sdk-go/aws/awserr"
  5. "github.com/aws/aws-sdk-go/aws/request"
  6. )
  7. // RequestFailure provides additional S3 specific metadata for the request
  8. // failure.
  9. type RequestFailure struct {
  10. awserr.RequestFailure
  11. hostID string
  12. }
  13. // NewRequestFailure returns a request failure error decordated with S3
  14. // specific metadata.
  15. func NewRequestFailure(err awserr.RequestFailure, hostID string) *RequestFailure {
  16. return &RequestFailure{RequestFailure: err, hostID: hostID}
  17. }
  18. func (r RequestFailure) Error() string {
  19. extra := fmt.Sprintf("status code: %d, request id: %s, host id: %s",
  20. r.StatusCode(), r.RequestID(), r.hostID)
  21. return awserr.SprintError(r.Code(), r.Message(), extra, r.OrigErr())
  22. }
  23. func (r RequestFailure) String() string {
  24. return r.Error()
  25. }
  26. // HostID returns the HostID request response value.
  27. func (r RequestFailure) HostID() string {
  28. return r.hostID
  29. }
  30. // RequestFailureWrapperHandler returns a handler to rap an
  31. // awserr.RequestFailure with the S3 request ID 2 from the response.
  32. func RequestFailureWrapperHandler() request.NamedHandler {
  33. return request.NamedHandler{
  34. Name: "awssdk.s3.errorHandler",
  35. Fn: func(req *request.Request) {
  36. reqErr, ok := req.Error.(awserr.RequestFailure)
  37. if !ok || reqErr == nil {
  38. return
  39. }
  40. hostID := req.HTTPResponse.Header.Get("X-Amz-Id-2")
  41. if req.Error == nil {
  42. return
  43. }
  44. req.Error = NewRequestFailure(reqErr, hostID)
  45. },
  46. }
  47. }