Add gRPC support

helios · Apr 3, 2024 · 744d96b · 744d96b
1 parent b8146ca
commit 744d96b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/exporters/otlp/otlptrace/otlptracegrpc/internal/otlpconfig/options.go b/exporters/otlp/otlptrace/otlptracegrpc/internal/otlpconfig/options.go
@@ -52,6 +52,7 @@ type (
 		Compression Compression
 		Timeout     time.Duration
 		URLPath     string
+		Query       map[string][]string
 
 		// gRPC configurations
 		GRPCCredentials credentials.TransportCredentials
@@ -326,3 +327,10 @@ func WithTimeout(duration time.Duration) GenericOption {
 		return cfg
 	})
 }
+
+func WithQuery(query map[string][]string) GenericOption {
+	return newGenericOption(func(cfg Config) Config {
+		cfg.Traces.Query = query
+		return cfg
+	})
+}
diff --git a/exporters/otlp/otlptrace/otlptracegrpc/options.go b/exporters/otlp/otlptrace/otlptracegrpc/options.go
@@ -187,3 +187,8 @@ func WithTimeout(duration time.Duration) Option {
 func WithRetry(settings RetryConfig) Option {
 	return wrappedOption{otlpconfig.WithRetry(retry.Config(settings))}
 }
+
+// WithQuery will send the provided query values with each gRPC request.
+func WithQuery(query map[string][]string) Option {
+	return wrappedOption{otlpconfig.WithQuery(query)}
+}