github.com/google/martian/v3@v3.3.3/header/header_matcher.go (about) 1 // Copyright 2015 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 15 package header 16 17 import ( 18 "net/http" 19 20 "github.com/google/martian/v3/proxyutil" 21 ) 22 23 // Matcher is a conditonal evalutor of request or 24 // response headers to be used in structs that take conditions. 25 type Matcher struct { 26 name, value string 27 } 28 29 // NewMatcher builds a new header matcher. 30 func NewMatcher(name, value string) *Matcher { 31 return &Matcher{ 32 name: name, 33 value: value, 34 } 35 } 36 37 // MatchRequest evaluates a request and returns whether or not 38 // the request contains a header that matches the provided name 39 // and value. 40 func (m *Matcher) MatchRequest(req *http.Request) bool { 41 h := proxyutil.RequestHeader(req) 42 43 vs, ok := h.All(m.name) 44 if !ok { 45 return false 46 } 47 48 for _, v := range vs { 49 if v == m.value { 50 return true 51 } 52 } 53 54 return false 55 } 56 57 // MatchResponse evaluates a response and returns whether or not 58 // the response contains a header that matches the provided name 59 // and value. 60 func (m *Matcher) MatchResponse(res *http.Response) bool { 61 h := proxyutil.ResponseHeader(res) 62 63 vs, ok := h.All(m.name) 64 if !ok { 65 return false 66 } 67 68 for _, v := range vs { 69 if v == m.value { 70 return true 71 } 72 } 73 74 return false 75 }