Merge pull request #18 from crast/pass-context

Receive GRPC metadata from HTTP headers.

protoc-gen-grpc-gateway/gengateway/template.go

@@ -220,7 +220,7 @@ func Register{{$svc.GetName}}Handler(ctx context.Context, mux *runtime.ServeMux,
 	{{range $m := $svc.Methods}}
 	{{range $b := $m.Bindings}}
 	mux.Handle({{$b.HTTPMethod | printf "%q"}}, pattern_{{$svc.GetName}}_{{$m.GetName}}_{{$b.Index}}, func(w http.ResponseWriter, req *http.Request, pathParams map[string]string) {
-		resp, err := request_{{$svc.GetName}}_{{$m.GetName}}_{{$b.Index}}(ctx, client, req, pathParams)
+		resp, err := request_{{$svc.GetName}}_{{$m.GetName}}_{{$b.Index}}(runtime.AnnotateContext(ctx, req), client, req, pathParams)
 		if err != nil {
 			runtime.HTTPError(w, err)
 			return

runtime/context.go

@@ -0,0 +1,31 @@
+package runtime
+
+import (
+	"net/http"
+	"strings"
+
+	"golang.org/x/net/context"
+	"google.golang.org/grpc/metadata"
+)
+
+const metadataHeaderPrefix = "Grpc-Metadata-"
+
+/*
+AnnotateContext adds context information such as metadata from the request.
+
+If there are no metadata headers in the request, then the context returned
+will be the same context.
+*/
+func AnnotateContext(ctx context.Context, req *http.Request) context.Context {
+	var pairs []string
+	for key, val := range req.Header {
+		if strings.HasPrefix(key, metadataHeaderPrefix) {
+			pairs = append(pairs, key[len(metadataHeaderPrefix):], val[0])
+		}
+	}
+
+	if len(pairs) != 0 {
+		ctx = metadata.NewContext(ctx, metadata.Pairs(pairs...))
+	}
+	return ctx
+}

runtime/context_test.go

@@ -0,0 +1,34 @@
+package runtime_test
+
+import (
+	"net/http"
+	"testing"
+
+	"github.com/gengo/grpc-gateway/runtime"
+	"golang.org/x/net/context"
+	"google.golang.org/grpc/metadata"
+)
+
+func TestAnnotateContext(t *testing.T) {
+	ctx := context.Background()
+
+	request, _ := http.NewRequest("GET", "http://localhost", nil)
+	request.Header = http.Header{}
+	annotated := runtime.AnnotateContext(ctx, request)
+	if annotated != ctx {
+		t.Errorf("AnnotateContext(ctx, request) = %v; want %v", annotated, ctx)
+	}
+	request.Header.Add("Grpc-Metadata-FooBar", "Value1")
+	request.Header.Add("Grpc-Metadata-Foo-BAZ", "Value2")
+	annotated = runtime.AnnotateContext(ctx, request)
+	md, ok := metadata.FromContext(annotated)
+	if !ok || len(md) != 2 {
+		t.Errorf("Expected 2 metadata items in context; got %v", md)
+	}
+	if md["Foobar"] != "Value1" {
+		t.Errorf("md[\"Foobar\"] = %v; want %v", md["Foobar"], "Value1")
+	}
+	if md["Foo-Baz"] != "Value2" {
+		t.Errorf("md[\"Foo-Baz\"] = %v want %v", md["Foo-Baz"], "Value2")
+	}
+}