go反向代理

package main

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

// NewProxy takes target host and creates a reverse proxy
func NewProxy(targetHost string) (*httputil.ReverseProxy, error) {
 url, err := url.Parse(targetHost)
 if err != nil {
  return nil, err
 }

 proxy := httputil.NewSingleHostReverseProxy(url)//新建代理

 originalDirector := proxy.Director
 proxy.Director = func(req *http.Request) {
  originalDirector(req)
  modifyRequest(req)//修改请求
 }

 proxy.ModifyResponse = modifyResponse()//修改响应
 proxy.ErrorHandler = errorHandler()//错误处理
 return proxy, nil
}

func modifyRequest(req *http.Request) {
 req.Header.Set("X-Proxy", "Simple-Reverse-Proxy")
}

func errorHandler() func(http.ResponseWriter, *http.Request, error) {
 return func(w http.ResponseWriter, req *http.Request, err error) {
  fmt.Printf("Got error while modifying response: %v \n", err)
  return
 }
}

func modifyResponse() func(*http.Response) error {
 return func(resp *http.Response) error {
  return errors.New("response body is invalid")
 }
}

// ProxyRequestHandler handles the http request using proxy
func ProxyRequestHandler(proxy *httputil.ReverseProxy) func(http.ResponseWriter, *http.Request) {
 return func(w http.ResponseWriter, r *http.Request) {
  proxy.ServeHTTP(w, r)//转代理
 }
}

func main() {
 // initialize a reverse proxy and pass the actual backend server url here
 proxy, err := NewProxy("http://my-api-server.com")
 if err != nil {
  panic(err)
 }

 // handle all requests to your server using the proxy
 http.HandleFunc("/", ProxyRequestHandler(proxy))
 log.Fatal(http.ListenAndServe(":8080", nil))
}

参考:Go 实现一个简单而强大的反向代理 (qq.com)

  • 0
    点赞
  • 0
    收藏 更改收藏夹
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

xixi_wong

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值