本篇文章给大家带来了关于go的相关知识,其中主要给大家聊聊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}, "")}
登录后复制

【相关推荐:Go视频教程

以上就是一文分析go中判断请求是http还是https的详细内容,更多请关注Work网其它相关文章!

09-11 22:09