在go语言中简单实现反向代理示例代码

Go  /  管理员 发布于 1年前   363

常见的反向代理有 Nginx,HAProxy,Apisix 等，反向代理是指：以代理服务器来接受internet上的连接请求，然后将请求转发给内部网络上的服务器，并将从服务器上得到的结果返回给internet上请求连接的客户端，此时代理服务器对外就表现为一个反向代理服务器。

接下来介绍如何使用go语言实现一个反向代理服务器。

go实现

使用 golang 实现反向代理非常简单，标准库 net/http/httputil 提供了反向代理的方法可以让我们方便的实现反向代理,使我们可以很快的实现一个简单的反向代理服务器。

package main

import (
    "log"
    "net/http"
    "net/http/httputil"
    "net/url"
)

func NewProxy(targetHost string) (*httputil.ReverseProxy, error) {
    url, err := url.Parse(targetHost)
    if err != nil {
        return nil, err
    }
    return httputil.NewSingleHostReverseProxy(url), nil
}

func ProxyRequestHandler(proxy *httputil.ReverseProxy) func(http.ResponseWriter, *http.Request) {
    return func(w http.ResponseWriter, r *http.Request) {
        proxy.ServeHTTP(w, r)
    }
}

func main() {
    proxy, err := NewProxy("https://www.zongscan.com")
    if err != nil {
        panic(err)
    }
    
    http.HandleFunc("/", ProxyRequestHandler(proxy))
    log.Fatal(http.ListenAndServe(":8080", nil))
}

这段代码将到达我们代理服务器的任何请求都会被代理到 

https://www.zongscan.com

我们运行代码，访问网站，发现 403 Forbidden 好像请求被拦截了,应该是源网站进行了请求校验，这该怎么处理呢？

通过查阅资料得知，我们需要将 host 传递过去，修改后的代码如下:

package main

import (
    "log"
    "net/http"
    "net/http/httputil"
    "net/url"
)

func NewProxy(targetHost string) (*httputil.ReverseProxy, error) {
    url, err := url.Parse(targetHost)
    if err != nil {
        return nil, err
    }
    return httputil.NewSingleHostReverseProxy(url), nil
}

func ProxyRequestHandler(proxy *httputil.ReverseProxy) func(http.ResponseWriter, *http.Request) {
    return func(w http.ResponseWriter, r *http.Request) {
    r.Host = "www.zongscan.com"
    proxy.ServeHTTP(w, r)
    }
}

func main() {
    proxy, err := NewProxy("https://www.zongscan.com")
    if err != nil {
        panic(err)
    }
    
    http.HandleFunc("/", ProxyRequestHandler(proxy))
    log.Fatal(http.ListenAndServe(":8080", nil))
}

现在运行代码即可正常访问了。

修改响应

如果我们需要修改服务器返回的响应，要怎么处理呢？

我们可以定义一个 modifyResponse 函数用来处理响应，还可以定义 ErrorHandler 来处理响应的错误。

比如我们可以自定义处理服务器返回的错误，比如记录 404 请求的详情数据。

代码:

package main

import (
    "errors"
    "fmt"
    "log"
    "net/http"
    "net/http/httputil"
    "net/url"
)

var (
    NOT_FOUND = errors.New("not found")
)

func NewProxy(targetHost string) (*httputil.ReverseProxy, error) {
    url, err := url.Parse(targetHost)
    if err != nil {
        return nil, err
    }
    
    proxy := httputil.NewSingleHostReverseProxy(url)
    proxy.ModifyResponse = modifyResponse()
    proxy.ErrorHandler = errorHandler()
    return proxy, nil
}

func ProxyRequestHandler(proxy *httputil.ReverseProxy) func(http.ResponseWriter, *http.Request) {
    return func(w http.ResponseWriter, r *http.Request) {
        r.Host = "www.zongscan.com"
        proxy.ServeHTTP(w, r)
    }
}

func main() {
    proxy, err := NewProxy("https://www.zongscan.com")
    if err != nil {
        panic(err)
    }
    
    http.HandleFunc("/", ProxyRequestHandler(proxy))
    log.Fatal(http.ListenAndServe(":8080", nil))
    }
    
    func errorHandler() func(http.ResponseWriter, *http.Request, error) {
        return func(w http.ResponseWriter, req *http.Request, err error) {
        if errors.Is(err, NOT_FOUND) {
            w.WriteHeader(http.StatusNotFound)
        }
        
        fmt.Printf("Got error while modifying response: %v \n", err)
            return
        }
    }
    
    func modifyResponse() func(*http.Response) error {
        return func(resp *http.Response) error {
            if resp.StatusCode == 404 {
                return NOT_FOUND
            }
            return nil
    }
}

此段代码将源服务器返回的 404 进行了特殊的处理。

处理修改服务器返回的响应，我们还可以修改发生给服务器的请求，这里就不过多介绍了，想了解的话可以查看 Director 字段。

总结

本文可以看出go相比其他语言，更方便的实现反向代理服务，并如何修改服务器返回的响应等功能。