customizations.go 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. package glacier
  2. import (
  3. "encoding/hex"
  4. "reflect"
  5. "github.com/aws/aws-sdk-go/aws/awsutil"
  6. "github.com/aws/aws-sdk-go/aws/request"
  7. )
  8. var (
  9. defaultAccountID = "-"
  10. )
  11. func init() {
  12. initRequest = func(r *request.Request) {
  13. r.Handlers.Validate.PushFront(addAccountID)
  14. r.Handlers.Validate.PushFront(copyParams) // this happens first
  15. r.Handlers.Build.PushBack(addChecksum)
  16. r.Handlers.Build.PushBack(addAPIVersion)
  17. }
  18. }
  19. func copyParams(r *request.Request) {
  20. r.Params = awsutil.CopyOf(r.Params)
  21. }
  22. func addAccountID(r *request.Request) {
  23. if !r.ParamsFilled() {
  24. return
  25. }
  26. v := reflect.Indirect(reflect.ValueOf(r.Params))
  27. if f := v.FieldByName("AccountId"); f.IsNil() {
  28. f.Set(reflect.ValueOf(&defaultAccountID))
  29. }
  30. }
  31. func addChecksum(r *request.Request) {
  32. if r.Body == nil {
  33. return
  34. }
  35. h := ComputeHashes(r.Body)
  36. if r.HTTPRequest.Header.Get("X-Amz-Content-Sha256") == "" {
  37. hstr := hex.EncodeToString(h.LinearHash)
  38. r.HTTPRequest.Header.Set("X-Amz-Content-Sha256", hstr)
  39. }
  40. if r.HTTPRequest.Header.Get("X-Amz-Sha256-Tree-Hash") == "" {
  41. hstr := hex.EncodeToString(h.TreeHash)
  42. r.HTTPRequest.Header.Set("X-Amz-Sha256-Tree-Hash", hstr)
  43. }
  44. }
  45. func addAPIVersion(r *request.Request) {
  46. r.HTTPRequest.Header.Set("X-Amz-Glacier-Version", r.Service.APIVersion)
  47. }