今天小编给大家分享一下golang中判断请求是http还是https的相关知识点,内容详细,逻辑清晰,相信大部分人都还太了解这方面的知识,所以分享这篇文章给大家参考一下,希望大家阅读完这篇文章后有所收获,下面我们一起来了解一下吧。
golang中判断请求是http还是https-用于获取当前访问地址
一个自由职业独立开发者,在线客服系统的开发日志
今天给客服系统增加了一项配置,配置企业微信内部群通知机器人 webhook,当有新消息到来的时候,调用 webhook,推送给企业微信群。新消息的内容里,支持 markdown,可以带链接,就想把当前网址的链接带过去,方便直接点击后去回复消息。下面是涉及的技术知识点总结。
未使用 nginx 等反向代理的情况下
那么可以直接使用下面的代码来确认,http.Request 结构体的 TLS 字段判断请求是否使用了 HTTPS 协议。如果该字段不为 nil,则说明请求使用了 HTTPS 协议;否则,说明请求使用了 HTTP 协议
package main
import (
"fmt"
"net/http"
)
func handler(w http.ResponseWriter, r *http.Request) {
if r.TLS != nil {
fmt.Println("HTTPS request")
} else {
fmt.Println("HTTP request")
}
}
func main() {
http.HandleFunc("/", handler)
http.ListenAndServe(":8080", nil)
}
经过反向代理的情况下
上面的代码失效,可以使用下面方法
如果使用了 nginx 反向代理,那么需要确保反代中传入了下面的 headers 参数 X-Forwarded-Proto
location / {
proxy_pass http://your_upstream_server;
proxy_set_header X-Forwarded-Proto $scheme;
}
可以判断这个 header 来确认是否 https
package main
import (
"fmt"
"net/http"
)
func handler(w http.ResponseWriter, r *http.Request) {
proto := r.Header.Get("X-Forwarded-Proto")
if proto == "https" {
fmt.Println("HTTPS request")
} else {
fmt.Println("HTTP request")
}
}
func main() {
http.HandleFunc("/", handler)
http.ListenAndServe(":8080", nil)
}
综合一下,获取当前访问地址的函数
//获取当前访问的Host
func GetHost(r *http.Request) (Url string) {
scheme := "http://"
if r.TLS != nil || r.Header.Get("X-Forwarded-Proto") == "https" {
scheme = "https://"
}
return strings.Join([]string{scheme, r.Host}, "")}