Make headers a map of strings

Signed-off-by: Elis Lulja <[email protected]>
asimpleidea · Jan 10, 2021 · dea7278 · dea7278
1 parent cd65cb2
commit dea7278
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/page.go b/page.go
@@ -12,7 +12,7 @@ type Page struct {
 	// Method of the request, e.g.: GET
 	Method *string `yaml:"method,omitempty"`
 	// Headers to send with the request
-	Headers map[string][]string `yaml:"headers,omitempty"`
+	Headers map[string]string `yaml:"headers,omitempty"`
 	// UserAgentOptions contains options about the
 	// user agent
 	*UserAgentOptions `yaml:"userAgentOptions,omitempty"`

diff --git a/page_poller.go b/page_poller.go
@@ -78,7 +78,9 @@ func New(p *Page) (Poller, error) {
 	if p.Headers == nil {
 		l.Warn().Msg("no headers provided")
 	} else {
-		headers = p.Headers
+		for headerKey, headerVal := range p.Headers {
+			headers[headerKey] = []string{headerVal}
+		}
 		switch hlen := len(p.Headers); {
 		case hlen == 0:
 			l.Warn().Msg("no headers provided")