Set wait for ready in Jaeger gRPC exporter (#899)

* Make wait for ready configurable in gRPC exporters

Signed-off-by: Pavol Loffay <ploffay@redhat.com>

* Fix

Signed-off-by: Pavol Loffay <ploffay@redhat.com>

Author: pavolloffay

config/configgrpc/configgrpc.go

@@ -56,6 +56,10 @@ type GRPCSettings struct {
 	// The keepalive parameters for client gRPC. See grpc.WithKeepaliveParams
 	// (https://godoc.org/google.golang.org/grpc#WithKeepaliveParams).
 	KeepaliveParameters *KeepaliveConfig `mapstructure:"keepalive"`
+
+	// WaitForReady parameter configures client to wait for ready state before sending data.
+	// (https://github.com/grpc/grpc/blob/master/doc/wait-for-ready.md)
+	WaitForReady bool `mapstructure:"wait_for_ready"`
 }
 
 // KeepaliveConfig exposes the keepalive.ClientParameters to be used by the exporter.

exporter/jaegerexporter/exporter.go

@@ -46,8 +46,9 @@ func New(config *Config) (component.TraceExporter, error) {
 
 	collectorServiceClient := jaegerproto.NewCollectorServiceClient(client)
 	s := &protoGRPCSender{
-		client:   collectorServiceClient,
-		metadata: metadata.New(config.GRPCSettings.Headers),
+		client:       collectorServiceClient,
+		metadata:     metadata.New(config.GRPCSettings.Headers),
+		waitForReady: config.WaitForReady,
 	}
 
 	exp, err := exporterhelper.NewTraceExporter(config, s.pushTraceData)
@@ -58,8 +59,9 @@ func New(config *Config) (component.TraceExporter, error) {
 // protoGRPCSender forwards spans encoded in the jaeger proto
 // format, to a grpc server.
 type protoGRPCSender struct {
-	client   jaegerproto.CollectorServiceClient
-	metadata metadata.MD
+	client       jaegerproto.CollectorServiceClient
+	metadata     metadata.MD
+	waitForReady bool
 }
 
 func (s *protoGRPCSender) pushTraceData(
@@ -80,7 +82,7 @@ func (s *protoGRPCSender) pushTraceData(
 	for _, batch := range batches {
 		_, err = s.client.PostSpans(
 			ctx,
-			&jaegerproto.PostSpansRequest{Batch: *batch})
+			&jaegerproto.PostSpansRequest{Batch: *batch}, grpc.WaitForReady(s.waitForReady))
 		if err != nil {
 			return td.SpanCount() - sentSpans, err
 		}

exporter/otlpexporter/exporter.go

@@ -211,14 +211,14 @@ func (e *exporterImp) exportRequest(ctx context.Context, perform func(ctx contex
 
 func (e *exporterImp) exportTrace(ctx context.Context, request *otlptracecol.ExportTraceServiceRequest) error {
 	return e.exportRequest(ctx, func(ctx context.Context) error {
-		_, err := e.traceExporter.Export(ctx, request)
+		_, err := e.traceExporter.Export(ctx, request, grpc.WaitForReady(e.config.WaitForReady))
 		return err
 	})
 }
 
 func (e *exporterImp) exportMetrics(ctx context.Context, request *otlpmetriccol.ExportMetricsServiceRequest) error {
 	return e.exportRequest(ctx, func(ctx context.Context) error {
-		_, err := e.metricExporter.Export(ctx, request)
+		_, err := e.metricExporter.Export(ctx, request, grpc.WaitForReady(e.config.WaitForReady))
 		return err
 	})
 }