完成正向代理的主干逻辑

This commit is contained in:
白茶清欢 2022-10-08 14:58:47 +08:00
commit 691e6b82df
4 changed files with 119 additions and 0 deletions

26
.gitignore vendored Normal file
View File

@ -0,0 +1,26 @@
# IDE
.idea/
# Binary
realsee-open-svc
preline.i.svc.open.realsee.com
__debug_bin
# Output
releases
*.tar.gz
# Log
*.log
# Vendor
vendor
.vscode/
realsee-open-svc-data-product
release
nodejs/node_modules/*
storage
new_api_debug.tar
*.tar

16
go.mod Normal file
View File

@ -0,0 +1,16 @@
module git.zhangdeman.cn/zhangdeman/proxy
go 1.17
require (
git.zhangdeman.cn/zhangdeman/util v0.0.0-20220723131533-e95f609c0f73 // indirect
github.com/Jeffail/gabs v1.4.0 // indirect
github.com/axgle/mahonia v0.0.0-20180208002826-3358181d7394 // indirect
github.com/go-ini/ini v1.67.0 // indirect
github.com/pkg/errors v0.9.1 // indirect
github.com/spaolacci/murmur3 v1.1.0 // indirect
github.com/tidwall/gjson v1.14.3 // indirect
github.com/tidwall/match v1.1.1 // indirect
github.com/tidwall/pretty v1.2.1 // indirect
gopkg.in/yaml.v3 v3.0.1 // indirect
)

22
go.sum Normal file
View File

@ -0,0 +1,22 @@
git.zhangdeman.cn/zhangdeman/util v0.0.0-20220723131533-e95f609c0f73 h1:DVDjR7IpxeDKHX+2C8UbrtSwBIUX0JEN/BiZnWcHrUg=
git.zhangdeman.cn/zhangdeman/util v0.0.0-20220723131533-e95f609c0f73/go.mod h1:zTir/0IWdK3E7n0GiaogyWHADAQnBtTdl2I6Z2/OPqw=
github.com/Jeffail/gabs v1.4.0 h1://5fYRRTq1edjfIrQGvdkcd22pkYUrHZ5YC/H2GJVAo=
github.com/Jeffail/gabs v1.4.0/go.mod h1:6xMvQMK4k33lb7GUUpaAPh6nKMmemQeg5d4gn7/bOXc=
github.com/axgle/mahonia v0.0.0-20180208002826-3358181d7394 h1:OYA+5W64v3OgClL+IrOD63t4i/RW7RqrAVl9LTZ9UqQ=
github.com/axgle/mahonia v0.0.0-20180208002826-3358181d7394/go.mod h1:Q8n74mJTIgjX4RBBcHnJ05h//6/k6foqmgE45jTQtxg=
github.com/go-ini/ini v1.67.0 h1:z6ZrTEZqSWOTyH2FlglNbNgARyHG8oLW9gMELqKr06A=
github.com/go-ini/ini v1.67.0/go.mod h1:ByCAeIL28uOIIG0E3PJtZPDL8WnHpFKFOtgjp+3Ies8=
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
github.com/spaolacci/murmur3 v1.1.0 h1:7c1g84S4BPRrfL5Xrdp6fOJ206sU9y293DDHaoy0bLI=
github.com/spaolacci/murmur3 v1.1.0/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=
github.com/tidwall/gjson v1.14.3 h1:9jvXn7olKEHU1S9vwoMGliaT8jq1vJ7IH/n9zD9Dnlw=
github.com/tidwall/gjson v1.14.3/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
github.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA=
github.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM=
github.com/tidwall/pretty v1.2.0/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU=
github.com/tidwall/pretty v1.2.1 h1:qjsOFOWWQl+N3RsoF5/ssm1pHmJJwhjlSbZ51I6wMl4=
github.com/tidwall/pretty v1.2.1/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

55
proxy.go Normal file
View File

@ -0,0 +1,55 @@
// Package proxy ...
//
// Description : 正向代理服务器
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2022-10-08 14:26
package proxy
import (
"git.zhangdeman.cn/zhangdeman/util"
"io"
"net"
"net/http"
"strings"
)
// Forward 正向代理主干逻辑
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 14:28 2022/10/8
func Forward(rw http.ResponseWriter, req *http.Request) {
// 复制请求
proxyRequest := new(http.Request)
*proxyRequest = *req
if clientIP, _, err := net.SplitHostPort(proxyRequest.RemoteAddr); err == nil {
if prior, ok := proxyRequest.Header["X-forwarded-For"]; ok {
clientIP = strings.Join(prior, ", ")+ ", "+clientIP
}
proxyRequest.Header.Set("X-Forwarded-For", clientIP)
}
// 请求代理机器IP
proxyIP := util.IP.GetHostIP()
proxyRequest.Header.Set("X-Forward-Proxy-IP", proxyIP)
// 请求目标服务
transport := http.DefaultTransport
res, err := transport.RoundTrip(proxyRequest)
if nil != err {
rw.WriteHeader(http.StatusBadGateway)
return
}
// 回写客户端响应头
for key, values := range res.Header {
for _, v := range values {
rw.Header().Add(key, v)
}
}
// 回写客户端状态码
rw.WriteHeader(res.StatusCode)
// 回写响应体
io.Copy(rw, res.Body)
res.Body.Close()
}