Add auth method

e-conomic · Jun 27, 2019 · f7d41a1 · f7d41a1
1 parent b208e73
commit f7d41a1
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/ctxvml.go b/ctxvml.go
@@ -13,8 +13,9 @@ type ctxMarker struct{}
 
 // VmlHeaders contains vml http headers.
 type VmlHeaders struct {
-	Username string
-	OcrCache string
+	Username   string
+	OcrCache   string
+	AuthMethod string
 }
 
 func (h VmlHeaders) OcrCacheAllow() bool {
@@ -65,6 +66,9 @@ func extractMetadataToContext(ctx context.Context) context.Context {
 	if mdValue, ok := md["vml-ocr-cache"]; ok && len(mdValue) != 0 {
 		headers.OcrCache = mdValue[0]
 	}
+	if mdValue, ok := md["vml-auth-method"]; ok && len(mdValue) != 0 {
+		headers.AuthMethod = mdValue[0]
+	}
 	if headers.Username != "" {
 		grpc_ctxtags.Extract(ctx).Set("username", headers.Username)
 	}
@@ -104,6 +108,7 @@ func packCallerMetadata(ctx context.Context) map[string]string {
 	headers := Extract(ctx)
 	md["vml-username"] = headers.Username
 	md["vml-ocr-cache"] = headers.OcrCache
+	md["vml-auth-method"] = headers.AuthMethod
 	return md
 }
 

diff --git a/ctxvml_test.go b/ctxvml_test.go
@@ -16,8 +16,8 @@ func TestPackMetadata(t *testing.T) {
 		Username: "JohnDoe",
 	})
 	md := packCallerMetadata(ctx)
-	if md["x-ssn-username"] != "JohnDoe" {
-		t.Fatalf("Unexpected username %s", md["x-ssn-username"])
+	if md["vml-username"] != "JohnDoe" {
+		t.Fatalf("Unexpected username %s", md["vml-username"])
 	}
 	for k, v := range md {
 		ctx = metadata.AppendToOutgoingContext(ctx, k, v)