customizations.go 643 B

123456789101112131415161718192021222324252627282930313233
  1. package machinelearning
  2. import (
  3. "net/url"
  4. "github.com/aws/aws-sdk-go/aws/request"
  5. )
  6. func init() {
  7. initRequest = func(r *request.Request) {
  8. switch r.Operation.Name {
  9. case opPredict:
  10. r.Handlers.Build.PushBack(updatePredictEndpoint)
  11. }
  12. }
  13. }
  14. // updatePredictEndpoint rewrites the request endpoint to use the
  15. // "PredictEndpoint" parameter of the Predict operation.
  16. func updatePredictEndpoint(r *request.Request) {
  17. if !r.ParamsFilled() {
  18. return
  19. }
  20. r.Service.Endpoint = *r.Params.(*PredictInput).PredictEndpoint
  21. uri, err := url.Parse(r.Service.Endpoint)
  22. if err != nil {
  23. r.Error = err
  24. return
  25. }
  26. r.HTTPRequest.URL = uri
  27. }