golang http(s)请求转发

栏目: Go · 发布时间: 5年前

func main(){
    router := gin.Default()//创建一个router
    router.Use(MiddleWare())//中间件,起拦截器的作用
    router.Any("/*action", Forward)//所有请求都会经过Forward函数转发
    router.Run(":8000")
}
func Forward(c *gin.Context) {
    targetHost := &httputils.TargetHost{
        Host:    "www.baidu.com",
        IsHttps: false,
    }
    HostReverseProxy(c.Writer, c.Request, targetHost)
}
func HostReverseProxy(w http.ResponseWriter, req *http.Request, targetHost *TargetHost) {
    host := ""
    if targetHost.IsHttps {
        host = host + "https://"
    } else {
        host = host + "http://"
    }
    remote, err := url.Parse(host + targetHost.Host)
    if err != nil {
        log.Errorf("err:%s", err)
        w.WriteHeader(http.StatusInternalServerError)
        return
    }
    proxy := httputil.NewSingleHostReverseProxy(remote)
    if targetHost.IsHttps {
        tls, err := GetVerTLSConfig(targetHost.CAPath)
        if err != nil {
            log.Errorf("https crt error: %s", err)
            w.WriteHeader(http.StatusInternalServerError)
            return
        }
        var pTransport http.RoundTripper = &http.Transport{
            Dial: func(netw, addr string) (net.Conn, error) {
                c, err := net.DialTimeout(netw, addr, time.Second*time.Duration(DialTimeout))
                if err != nil {
                    return nil, err
                }
                return c, nil
            },
            ResponseHeaderTimeout: time.Second * time.Duration(ResponseHeaderTimeout),
            TLSClientConfig:       tls,
        }
        proxy.Transport = pTransport
    }
    proxy.ServeHTTP(w, req)
}
type TargetHost struct {
    Host    string
    IsHttps bool
    CAPath  string
}
func MiddleWare() gin.HandlerFunc {
    return func(c *gin.Context) {
        account := c.Request.Header.Get("ename")//从请求头中获取ename字段
    if account == "" {
        c.JSON(http.StatusOK, httputils.Response{
            Code:    400002,
            Message: "用户未登录",
        })
        c.Abort()
        return
    }
        fmt.Println("before middleware")
        c.Set("request", "clinet_request")
        c.Next()
        fmt.Println("before middleware")
    }
}

以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

C算法(第二卷:图算法)(第3版)

C算法(第二卷:图算法)(第3版)

塞德威克(Sedgewick Robert) / 周良忠 / 第1版 (2004年1月1日) / 2004-4 / 38.0

《C算法(第2卷)(图算法)(第3版)(中文版)》所讨论的图算法,都是实际中解决图问题的最重要的已知方法。《C算法(第2卷)(图算法)(第3版)(中文版)》的主要宗旨是让越来越多需要了解这些算法的人的能够掌握这些方法及基本原理。书中根据基本原理从基本住处开始循序渐进地讲解,然后再介绍一些经典方法,最后介绍仍在进行研究和发展的现代技术。精心挑选的实例、详尽的图示以及完整的实现代码与正文中的算法和应用......一起来看看 《C算法(第二卷:图算法)(第3版)》 这本书的介绍吧!

图片转BASE64编码
图片转BASE64编码

在线图片转Base64编码工具

SHA 加密
SHA 加密

SHA 加密工具

UNIX 时间戳转换
UNIX 时间戳转换

UNIX 时间戳转换