move all config to config.go

Signed-off-by: Ziqi Zhao <zhaoziqi9146@gmail.com>
open-telemetry · Sep 7, 2022 · 4cb7bdc · 4cb7bdc
1 parent b1eb629
commit 4cb7bdc
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 33 deletions.
diff --git a/...gle.golang.org/grpc/otelgrpc/grpctrace.go → ...google.golang.org/grpc/otelgrpc/config.go b/...gle.golang.org/grpc/otelgrpc/grpctrace.go → ...google.golang.org/grpc/otelgrpc/config.go
@@ -121,6 +121,20 @@ func WithTracerProvider(tp trace.TracerProvider) Option {
 	return tracerProviderOption{tp: tp}
 }
 
+type meterProviderOption struct{ mp metric.MeterProvider }
+
+func (o meterProviderOption) apply(c *config) {
+	if o.mp != nil {
+		c.MeterProvider = o.mp
+	}
+}
+
+// WithMeterProvider returns an Option to use the MeterProvider when
+// creating a Meter.
+func WithMeterProvider(mp metric.MeterProvider) Option {
+	return meterProviderOption{mp: mp}
+}
+
 type metadataSupplier struct {
 	metadata *metadata.MD
 }

diff --git a/instrumentation/google.golang.org/grpc/otelgrpc/grpcmetric.go b/instrumentation/google.golang.org/grpc/otelgrpc/grpcmetric.go