Support Bearer Token

Author: codebien

pkg/remotewrite/config.go

@@ -41,6 +41,9 @@ type Config struct {
 // Password is the Password for the Basic Auth.
 Password null.String `json:"password"`
 
+// BearerToken if set is the token used for the `Authorization` header.
+BearerToken null.String `json:"bearerToken"`
+
 // PushInterval defines the time between flushes. The Output will wait the set time
 // before push a new set of time series to the endpoint.
 PushInterval types.NullDuration `json:"pushInterval"`
@@ -95,6 +98,14 @@ func (conf Config) RemoteConfig() (*remote.HTTPConfig, error) {
 hc.Headers.Add(k, v)
 }
 }
+
+if conf.BearerToken.String != "" {
+if hc.Headers == nil {
+hc.Headers = make(http.Header)
+}
+hc.Headers.Set("Authorization", "Bearer "+conf.BearerToken.String)
+}
+
 return &hc, nil
 }
 

pkg/remotewrite/config_test.go

@@ -61,11 +61,15 @@ func TestConfigRemoteConfig(t *testing.T) {
 Password: null.StringFrom("mypass"),
 Headers: map[string]string{
 "X-MYCUSTOM-HEADER": "val1",
+// it asserts that Authz header is overwritten if the token is set
+"Authorization": "pre-set-token",
 },
+BearerToken: null.StringFrom("my-fake-token"),
 }
 
 headers := http.Header{}
 headers.Set("X-MYCUSTOM-HEADER", "val1")
+headers.Set("Authorization", "Bearer my-fake-token")
 exprcc := &remote.HTTPConfig{
 Timeout: 5 * time.Second,
 TLSConfig: &tls.Config{