-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #79 from microsoft/andrueastman/responseHeaders
Adds response headers to API Exception
- Loading branch information
Showing
5 changed files
with
149 additions
and
98 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,111 @@ | ||
| package abstractions | ||
|
|
||
| import "strings" | ||
|
|
||
| type header struct { | ||
| headers map[string]map[string]struct{} | ||
| } | ||
|
|
||
| type void struct{} | ||
|
|
||
| var voidInstance void | ||
|
|
||
| func normalizeHeaderKey(key string) string { | ||
| return strings.ToLower(strings.Trim(key, " ")) | ||
| } | ||
|
|
||
| //Add adds a new header or append a new value to an existing header | ||
| func (r *header) Add(key string, value string, additionalValues ...string) { | ||
| normalizedKey := normalizeHeaderKey(key) | ||
| if normalizedKey == "" || value == "" { | ||
| return | ||
| } | ||
| if r.headers == nil { | ||
| r.headers = make(map[string]map[string]struct{}) | ||
| } | ||
| if r.headers[normalizedKey] == nil { | ||
| r.headers[normalizedKey] = make(map[string]struct{}) | ||
| } | ||
| r.headers[normalizedKey][value] = voidInstance | ||
| for _, v := range additionalValues { | ||
| r.headers[normalizedKey][v] = voidInstance | ||
| } | ||
| } | ||
|
|
||
| //Get returns the values for the specific header | ||
| func (r *header) Get(key string) []string { | ||
| if r.headers == nil { | ||
| return nil | ||
| } | ||
| normalizedKey := normalizeHeaderKey(key) | ||
| if r.headers[normalizedKey] == nil { | ||
| return make([]string, 0) | ||
| } | ||
| values := make([]string, 0, len(r.headers[normalizedKey])) | ||
| for k := range r.headers[normalizedKey] { | ||
| values = append(values, k) | ||
| } | ||
| return values | ||
| } | ||
|
|
||
| //Remove removes the specific header and all its values | ||
| func (r *header) Remove(key string) { | ||
| if r.headers == nil { | ||
| return | ||
| } | ||
| normalizedKey := normalizeHeaderKey(key) | ||
| delete(r.headers, normalizedKey) | ||
| } | ||
|
|
||
| //RemoveValue remove the value for the specific header | ||
| func (r *header) RemoveValue(key string, value string) { | ||
| if r.headers == nil { | ||
| return | ||
| } | ||
| normalizedKey := normalizeHeaderKey(key) | ||
| if r.headers[normalizedKey] == nil { | ||
| return | ||
| } | ||
| delete(r.headers[normalizedKey], value) | ||
| if len(r.headers[normalizedKey]) == 0 { | ||
| delete(r.headers, normalizedKey) | ||
| } | ||
| } | ||
|
|
||
| //ContainsKey check if the key exists in the headers | ||
| func (r *header) ContainsKey(key string) bool { | ||
| if r.headers == nil { | ||
| return false | ||
| } | ||
| normalizedKey := normalizeHeaderKey(key) | ||
| return r.headers[normalizedKey] != nil | ||
| } | ||
|
|
||
| //Clear clear all headers | ||
| func (r *header) Clear() { | ||
| r.headers = nil | ||
| } | ||
|
|
||
| //AddAll adds all headers from the other headers | ||
| func (r *header) AddAll(other *header) { | ||
| if other == nil || other.headers == nil { | ||
| return | ||
| } | ||
| for k, v := range other.headers { | ||
| for k2 := range v { | ||
| r.Add(k, k2) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| //ListKeys returns all the keys in the headers | ||
| func (r *header) ListKeys() []string { | ||
| if r.headers == nil { | ||
| return make([]string, 0) | ||
| } | ||
| keys := make([]string, 0, len(r.headers)) | ||
| for k := range r.headers { | ||
| keys = append(keys, k) | ||
| } | ||
| return keys | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,25 @@ | ||
| package abstractions | ||
|
|
||
| //ResponseHeaders represents a collection of response headers | ||
| type ResponseHeaders struct { | ||
| header | ||
| } | ||
|
|
||
| //NewResponseHeaders creates a new ResponseHeaders | ||
| func NewResponseHeaders() *ResponseHeaders { | ||
| return &ResponseHeaders{ | ||
| header{make(map[string]map[string]struct{})}, | ||
| } | ||
| } | ||
|
|
||
| //AddAll adds all headers from the other headers | ||
| func (r *ResponseHeaders) AddAll(other *ResponseHeaders) { | ||
| if other == nil || other.headers == nil { | ||
| return | ||
| } | ||
| for k, v := range other.headers { | ||
| for k2 := range v { | ||
| r.Add(k, k2) | ||
| } | ||
| } | ||
| } |