allow header and set header in client

Author: hunjixin

http/client.go

@@ -27,6 +27,7 @@ type client struct {
 	httpClient    *http.Client
 	ua            string
 	apiPrefix     string
+	headers       map[string]string
 	fallback      cmds.Executor
 }
 
@@ -40,6 +41,16 @@ func ClientWithUserAgent(ua string) ClientOpt {
 	}
 }
 
+// ClientWithUserAgent specifies the HTTP user agent for the client.
+func ClientWithHeaders(key, value string) ClientOpt {
+	return func(c *client) {
+		if c.headers == nil {
+			c.headers = map[string]string{}
+		}
+		c.headers[key] = value
+	}
+}
+
 // ClientWithHTTPClient specifies a custom http.Client. Defaults to
 // http.DefaultClient.
 func ClientWithHTTPClient(hc *http.Client) ClientOpt {
@@ -173,6 +184,10 @@ func (c *client) toHTTPRequest(req *cmds.Request) (*http.Request, error) {
 	}
 	httpReq.Header.Set(uaHeader, c.ua)
 
+	for key, val := range c.headers {
+		httpReq.Header.Set(key, val)
+	}
+
 	httpReq = httpReq.WithContext(req.Context)
 	httpReq.Close = true
 

http/config.go

@@ -85,6 +85,12 @@ func (cfg *ServerConfig) SetAllowCredentials(flag bool) {
 	cfg.corsOpts.AllowCredentials = flag
 }
 
+func (cfg *ServerConfig) SetAllowHeaders(headers ...string) {
+	cfg.corsOptsRWMutex.Lock()
+	defer cfg.corsOptsRWMutex.Unlock()
+	cfg.corsOpts.AllowedHeaders = append(cfg.corsOpts.AllowedHeaders, headers...)
+}
+
 // allowOrigin just stops the request if the origin is not allowed.
 // the CORS middleware apparently does not do this for us...
 func allowOrigin(r *http.Request, cfg *ServerConfig) bool {