Propagate tracing context over httpgrpc

httpgrpc/server/server.go

@@ -14,11 +14,13 @@ import (
 	"github.com/grpc-ecosystem/grpc-opentracing/go/otgrpc"
 	"github.com/mwitkow/go-grpc-middleware"
 	"github.com/opentracing/opentracing-go"
+	"github.com/opentracing/opentracing-go/ext"
 	"github.com/sercand/kuberesolver"
 	"golang.org/x/net/context"
 	"google.golang.org/grpc"
 
 	"github.com/weaveworks/common/httpgrpc"
+	"github.com/weaveworks/common/logging"
 	"github.com/weaveworks/common/middleware"
 )
 
@@ -41,8 +43,18 @@ func (s Server) Handle(ctx context.Context, r *httpgrpc.HTTPRequest) (*httpgrpc.
 	if err != nil {
 		return nil, err
 	}
-	req = req.WithContext(ctx)
 	toHeader(r.Headers, req.Header)
+	if tracer := opentracing.GlobalTracer(); tracer != nil {
+		clientContext, err := tracer.Extract(opentracing.HTTPHeaders, opentracing.HTTPHeadersCarrier(req.Header))
+		if err == nil {
+			span := tracer.StartSpan("httpgrpc", ext.RPCServerOption(clientContext))
+			defer span.Finish()
+			ctx = opentracing.ContextWithSpan(ctx, span)
+		} else if err != opentracing.ErrSpanContextNotFound {
+			logging.Global().Warnf("Failed to extract tracing headers from request: %v", err)
+		}
+	}
+	req = req.WithContext(ctx)
 	req.RequestURI = r.Url
 	recorder := httptest.NewRecorder()
 	s.handler.ServeHTTP(recorder, req)
@@ -138,6 +150,13 @@ func (c *Client) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		http.Error(w, err.Error(), http.StatusInternalServerError)
 		return
 	}
+	if tracer := opentracing.GlobalTracer(); tracer != nil {
+		if span := opentracing.SpanFromContext(r.Context()); span != nil {
+			if err := tracer.Inject(span.Context(), opentracing.HTTPHeaders, opentracing.HTTPHeadersCarrier(r.Header)); err != nil {
+				logging.Global().Warnf("Failed to inject tracing headers into request: %v", err)
+			}
+		}
+	}
 	req := &httpgrpc.HTTPRequest{
 		Method:  r.Method,
 		Url:     r.RequestURI,

httpgrpc/server/server_test.go

@@ -10,8 +10,10 @@ import (
 	"reflect"
 	"testing"
 
+	opentracing "github.com/opentracing/opentracing-go"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
+	jaegercfg "github.com/uber/jaeger-client-go/config"
 	"github.com/weaveworks/common/httpgrpc"
 	"github.com/weaveworks/common/user"
 	"google.golang.org/grpc"
@@ -101,3 +103,34 @@ func TestParseURL(t *testing.T) {
 		assert.Equal(t, tc.expected, got)
 	}
 }
+
+func TestTracePropagation(t *testing.T) {
+	jaeger := jaegercfg.Configuration{}
+	closer, err := jaeger.InitGlobalTracer("test")
+	defer closer.Close()
+	require.NoError(t, err)
+
+	server, err := newTestServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		span := opentracing.SpanFromContext(r.Context())
+		fmt.Fprint(w, span.BaggageItem("name"))
+	}))
+
+	require.NoError(t, err)
+	defer server.grpcServer.GracefulStop()
+
+	client, err := NewClient(server.URL)
+	require.NoError(t, err)
+
+	req, err := http.NewRequest("GET", "/hello", &bytes.Buffer{})
+	require.NoError(t, err)
+
+	sp, ctx := opentracing.StartSpanFromContext(context.Background(), "Test")
+	sp.SetBaggageItem("name", "world")
+
+	req = req.WithContext(user.InjectOrgID(ctx, "1"))
+	recorder := httptest.NewRecorder()
+	client.ServeHTTP(recorder, req)
+
+	assert.Equal(t, "world", string(recorder.Body.Bytes()))
+	assert.Equal(t, 200, recorder.Code)
+}