前言
说起反向代理,大家应该都不陌生,是指以代理服务器来接受internet上的连接请求,然后将请求转发给内部网络上的服务器,并将从服务器上得到的结果返回给internet上请求连接的客户端,此时代理服务器对外就表现为一个反向代理服务器。常见的反向代理有 Nginx,HAProxy,Apisix 等。
接下来介绍如何使用 go 实现一个反向代理服务器。
golang 实现
使用 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://overstarry.vip")
if err != nil {
panic(err)
}
http.HandleFunc("/", ProxyRequestHandler(proxy))
log.Fatal(http.ListenAndServe(":8080", nil))
}
这段代码将到达我们代理服务器的任何请求都会被代理到 https://overstarry.vip。我们运行代码,访问网站,发现 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 = "overstarry.vip"
proxy.ServeHTTP(w, r)
}
}
func main() {
proxy, err := NewProxy("https://overstarry.vip")
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 = "overstarry.vip"
proxy.ServeHTTP(w, r)
}
}
func main() {
proxy, err := NewProxy("https://overstarry.vip")
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 如何实现一个简单的反向代理服务,并如何修改服务器返回的响应等的方法,可以看出 go 相比其他语言,可以很方便的实现这个功能。