czk 4 年之前
父节点
当前提交
7a58f04fde
共有 2 个文件被更改,包括 11 次插入5 次删除
  1. 4 0
      http/middleware/rpc_value.go
  2. 7 5
      http/param/url.go

+ 4 - 0
http/middleware/rpc_value.go

@@ -49,6 +49,10 @@ func GrpcValue(handler middleware.Handler) middleware.Handler {
 				_query, _ := url.ParseQuery(query[0])
 				ctx = context2.AppendToContext(ctx, "query", _query)
 			}
+			referer := md.Get("referer")
+			if len(referer) > 0 {
+				ctx = context2.AppendToContext(ctx, "referer", referer[0])
+			}
 		}
 		return handler(ctx, req)
 	}

+ 7 - 5
http/param/url.go

@@ -1,11 +1,13 @@
 package param
 
-import "context"
+import (
+	"context"
+)
 
-func GetHttpsUrl(c context.Context) string {
-	httpsUrl := c.Value("https_url")
-	if _, ok := httpsUrl.(string); ok {
-		return httpsUrl.(string)
+func GetReferer(c context.Context) string {
+	referer := c.Value("referer")
+	if _, ok := referer.(string); ok {
+		return referer.(string)
 	}
 	return ""
 }