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}, "")
}

另外，有客服系统需求的同学，可以访问下面地址研究测试

gofly.v1kf.com 产品官网

本作品采用《CC 协议》，转载必须注明作者和本文链接