pass request context from request rpc endpoints (#1799)

http middleware can add additional metadata to context,
for example tracing wrappers, pass down it to underlining
services

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
2020-07-06 15:55:17 +03:00
committed by GitHub
parent 6f309dada3
commit 97ae2979ad
2 changed files with 12 additions and 24 deletions

View File

@@ -3,16 +3,24 @@ package ctx
import (
"context"
"net/http"
"net/textproto"
"strings"
"github.com/micro/go-micro/v2/metadata"
)
func FromRequest(r *http.Request) context.Context {
ctx := context.Background()
md := make(metadata.Metadata)
for k, v := range r.Header {
md[k] = strings.Join(v, ",")
ctx := r.Context()
md, ok := metadata.FromContext(ctx)
if !ok {
md = make(metadata.Metadata)
}
for k, v := range r.Header {
md[textproto.CanonicalMIMEHeaderKey(k)] = strings.Join(v, ",")
}
// pass http host
md["Host"] = r.Host
// pass http method
md["Method"] = r.Method
return metadata.NewContext(ctx, md)
}