feat(headers): add flag that allow force override headers

aldor007 · Oct 22, 2019 · a76208f · a76208f
1 parent da5a722
commit a76208f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pkg/config/types.go b/pkg/config/types.go
@@ -113,6 +113,7 @@ type Bucket struct {
 // HeaderYaml allow you to override response headers
 type HeaderYaml struct {
 	StatusCodes []int             `yaml:"statusCodes"`
+	Override     bool `yaml:"override`
 	Values      map[string]string `yaml:"values"`
 }
 

diff --git a/pkg/processor/processor.go b/pkg/processor/processor.go
@@ -449,7 +449,9 @@ func updateHeaders(obj *object.FileObject, res *response.Response) *response.Res
 		for _, status := range headerPred.StatusCodes {
 			if status == res.StatusCode {
 				for h, v := range headerPred.Values {
-					if res.Headers.Get(h) == "" {
+					if headerPred.Override {
+						res.Set(h, v)
+					} else if res.Headers.Get(h) == "" {
 						res.Set(h, v)
 					}
 				}