From f6054ab37fa3bebac98556ff29d0ff50efa57d1b Mon Sep 17 00:00:00 2001 From: houseme Date: Wed, 9 Mar 2022 18:32:13 +0800 Subject: [PATCH] improve code --- net/ghttp/ghttp_request.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/net/ghttp/ghttp_request.go b/net/ghttp/ghttp_request.go index 5c546d16416..cba9737565c 100644 --- a/net/ghttp/ghttp_request.go +++ b/net/ghttp/ghttp_request.go @@ -224,8 +224,11 @@ func (r *Request) GetRemoteIp() string { // GetUrl returns current URL of this request. func (r *Request) GetUrl() string { - scheme := "http" - proto := r.Header.Get("X-Forwarded-Proto") + var ( + scheme = "http" + proto = r.Header.Get("X-Forwarded-Proto") + ) + if r.TLS != nil || (proto != "" && strings.ToLower(proto) == "https") { scheme = "https" }