You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

47 lines
1.4 KiB

  1. // Copyright 2016 Google Inc. All rights reserved.
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. // Package servemux contains a filter that executes modifiers when there is a
  15. // pattern match in a mux.
  16. package servemux
  17. import (
  18. "net/http"
  19. "github.com/google/martian"
  20. "github.com/google/martian/filter"
  21. )
  22. var noop = martian.Noop("mux.Filter")
  23. // Filter is a modifier that executes mod if a pattern is matched in mux.
  24. type Filter struct {
  25. *filter.Filter
  26. }
  27. // NewFilter constructs a filter that applies the modifier when the request
  28. // url matches a pattern in mux. If no mux is provided, the request is evaluated
  29. // against patterns in http.DefaultServeMux.
  30. func NewFilter(mux *http.ServeMux) *Filter {
  31. if mux == nil {
  32. mux = http.DefaultServeMux
  33. }
  34. m := NewMatcher(mux)
  35. f := filter.New()
  36. f.SetRequestCondition(m)
  37. f.SetResponseCondition(m)
  38. return &Filter{Filter: f}
  39. }