Exporter/opencensus: add option for custom headers (#392)

Steven Karis · web-flow · commit 05f5589f4ab6 · 2019-02-14T13:14:42.000-08:00
Allow creating an OpenCensus exporter with custom headers following suit with: [census-ecosystem/opencensus-go-exporter-ocagent#43](census-ecosystem/opencensus-go-exporter-ocagent#43)
diff --git a/README.md b/README.md
@@ -152,6 +152,8 @@ please see the [exporter README.md](exporter/exporterparser/README.md).
 ```yaml
 exporters:
   opencensus:
+    headers: {"X-test-header": "test-header"}
+    compression: "gzip"
     endpoint: "127.0.0.1:55678"
 
   jaeger:
diff --git a/exporter/exporterparser/opencensus.go b/exporter/exporterparser/opencensus.go
@@ -30,8 +30,9 @@ import (
 )
 
 type opencensusConfig struct {
-	Endpoint    string `mapstructure:"endpoint,omitempty"`
-	Compression string `mapstructure:"compression,omitempty"`
+	Endpoint    string            `mapstructure:"endpoint,omitempty"`
+	Compression string            `mapstructure:"compression,omitempty"`
+	Headers     map[string]string `mapstructure:"headers,omitempty"`
 	// TODO: add insecure, service name options.
 }
 
@@ -67,6 +68,9 @@ func OpenCensusTraceExportersFromViper(v *viper.Viper) (tes []exporter.TraceExpo
 			return nil, nil, nil, fmt.Errorf("Unsupported compression type: %s", ocac.Compression)
 		}
 	}
+	if len(ocac.Headers) > 0 {
+		opts = append(opts, ocagent.WithHeaders(ocac.Headers))
+	}
 
 	sde, serr := ocagent.NewExporter(opts...)
 	if serr != nil {
diff --git a/go.mod b/go.mod
@@ -2,7 +2,7 @@ module github.com/census-instrumentation/opencensus-service
 
 require (
 	cloud.google.com/go v0.32.0 // indirect
-	contrib.go.opencensus.io/exporter/ocagent v0.4.3
+	contrib.go.opencensus.io/exporter/ocagent v0.4.4
 	contrib.go.opencensus.io/exporter/stackdriver v0.9.1
 	git.apache.org/thrift.git v0.0.0-20181101003639-92be4f312b88 // indirect
 	github.com/Azure/azure-sdk-for-go v6.0.0-beta+incompatible // indirect
diff --git a/go.sum b/go.sum
@@ -6,6 +6,8 @@ cloud.google.com/go v0.32.0 h1:DSt59WoyNcfAInilEpfvm2ugq8zvNyaHAm9MkzOwRQ4=
 cloud.google.com/go v0.32.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=
 contrib.go.opencensus.io/exporter/ocagent v0.4.3 h1:QjNm697iO7CZ09IxxSiCUzOhALENIsLsixdPwjV1yGs=
 contrib.go.opencensus.io/exporter/ocagent v0.4.3/go.mod h1:YuG83h+XWwqWjvCqn7vK4KSyLKhThY3+gNGQ37iS2V0=
+contrib.go.opencensus.io/exporter/ocagent v0.4.4 h1:6v7OlUmiBDhvbYcHPWp8LHO8wh9jJY8f4mO34J4GJiA=
+contrib.go.opencensus.io/exporter/ocagent v0.4.4/go.mod h1:YuG83h+XWwqWjvCqn7vK4KSyLKhThY3+gNGQ37iS2V0=
 contrib.go.opencensus.io/exporter/stackdriver v0.9.1 h1:W6APgQ9we4BH8U8bnq/FvwLKo2WSMHuiMkkS/Slkg30=
 contrib.go.opencensus.io/exporter/stackdriver v0.9.1/go.mod h1:hNe5qQofPbg6bLQY5wHCvQ7o+2E5P8PkegEuQ+MyRw0=
 git.apache.org/thrift.git v0.0.0-20180902110319-2566ecd5d999 h1:sihTnRgTOUSCQz0iS0pjZuFQy/z7GXCJgSBg3+rZKHw=

Original file line number	Diff line number	Diff line change
`@@ -30,8 +30,9 @@ import (`
`30`	`30`	`)`
`31`	`31`
`32`	`32`	`type opencensusConfig struct {`
`33`		- Endpoint string `mapstructure:"endpoint,omitempty"`
`34`		- Compression string `mapstructure:"compression,omitempty"`
	`33`	+ Endpoint string `mapstructure:"endpoint,omitempty"`
	`34`	+ Compression string `mapstructure:"compression,omitempty"`
	`35`	+ Headers map[string]string `mapstructure:"headers,omitempty"`
`35`	`36`	`// TODO: add insecure, service name options.`
`36`	`37`	`}`
`37`	`38`
`@@ -67,6 +68,9 @@ func OpenCensusTraceExportersFromViper(v *viper.Viper) (tes []exporter.TraceExpo`
`67`	`68`	`return nil, nil, nil, fmt.Errorf("Unsupported compression type: %s", ocac.Compression)`
`68`	`69`	`}`
`69`	`70`	`}`
	`71`	`+ if len(ocac.Headers) > 0 {`
	`72`	`+ opts = append(opts, ocagent.WithHeaders(ocac.Headers))`
	`73`	`+ }`
`70`	`74`
`71`	`75`	`sde, serr := ocagent.NewExporter(opts...)`
`72`	`76`	`if serr != nil {`