-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathproxy.go
98 lines (91 loc) · 2.57 KB
/
proxy.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package dev
import (
"fmt"
"net/http"
"net/http/httputil"
"net/url"
"strings"
)
type proxy struct {
table *table
}
func NewProxy() *proxy {
return &proxy{
table: newTable(),
}
}
//管理逻辑
func (p *proxy) admin(w http.ResponseWriter, r *http.Request) {
data := r.FormValue("data")
if data != "" {
//重置路由表
p.table.DelAll()
arr := strings.Split(data, "\r\n")
for _, item := range arr {
d := strings.Split(item, "=>")
if len(d) < 2 || d[0] == "admin" {
continue
}
_, err := url.Parse("http://" + d[1])
if err != nil {
fmt.Println(err.Error())
continue
}
fmt.Println(d[0], d[1])
p.table.Set(d[0], d[1])
}
}
var newData string
for service, newUrl := range p.table.GetAll() {
if service == "" {
continue
}
newData = newData + service + "=>" + newUrl + "\n"
}
ip := ClientIP(r)
if ip != "" {
ip = "本机IP[" + ip + "]<br>"
}
w.Header().Set("Content-Type", "text/html;charset=utf-8")
w.Write([]byte("<html><head><title>开发联调神器</title></head><form method=\"POST\"><center>开发联调神器(先刷新再保存)<br>" + ip + "<textarea placeholder=\"test=>192.168.8.8:8080\r\n效果：\r\n" + GetURL(r) + "/test/product/list => http://192.168.8.8:8080/product/list\" autofocus name=\"data\" rows=\"30\" cols=\"100\">" + newData + "</textarea><br><input type=\"submit\" value=\"提交\"></center></form><html>"))
}
func (p *proxy) ServeHTTP(w http.ResponseWriter, r *http.Request) {
fmt.Println(r.RequestURI)
u := strings.Split(r.URL.Path, "/")
service := u[1]
//管理后台
if service == "admin" {
p.admin(w, r)
return
}
s := p.table.Get(service)
if s == "" {
w.Write([]byte("没有设置信息，请检查配置：" + GetURL(r) + "/admin"))
return
}
target, err := url.Parse("http://" + p.table.Get(service))
if err != nil {
w.Write([]byte(err.Error()))
return
}
r.URL.Path = strings.TrimPrefix(r.URL.Path, "/"+service)
targetQuery := target.RawQuery
director := func(req *http.Request) {
req.URL.Scheme = target.Scheme
req.URL.Host = target.Host
req.URL.Path = singleJoiningSlash(target.Path, req.URL.Path)
if targetQuery == "" || req.URL.RawQuery == "" {
req.URL.RawQuery = targetQuery + req.URL.RawQuery
} else {
req.URL.RawQuery = targetQuery + "&" + req.URL.RawQuery
}
if _, ok := req.Header["User-Agent"]; !ok {
// explicitly disable User-Agent so it's not set to default value
req.Header.Set("User-Agent", "")
}
req.Host = target.Host
req.Header.Set("X-Real-Ip", req.RemoteAddr)
}
proxy := &httputil.ReverseProxy{Director: director}
proxy.ServeHTTP(w, r)
}