init

Author: rushuinet

README.md

@@ -1,2 +1,39 @@
-# dev
+# dev-proxy
 开发调试神器
+
+# 设置示例
+```
+test=>http://127.0.0.1:8080
+
+假如服务域名为 dev.com 映射如下
+dev.com/test/product/list => http://192.168.8.8:8080/product/
+
+注意:dev.com机器能访问 IP 192.168.8.8
+```
+
+# 使用方法
+```
+package main
+
+import (
+	"flag"
+	"github.com/go-proxy/dev"
+	"log"
+	"net/http"
+	"os"
+)
+
+func main() {
+	port := os.Getenv("port")
+	if port == "" {
+		p := flag.String("port", "8888", "port default 8888")
+		port = *p
+	}
+	dev := proxy.NewProxy()
+	log.Println("start port :" + port)
+	err := http.ListenAndServe(":"+port, dev)
+	if err != nil {
+		log.Fatal(err)
+	}
+}
+```

exapmle/main.go

@@ -0,0 +1,23 @@
+package main
+
+import (
+	"flag"
+	"github.com/go-proxy/dev"
+	"log"
+	"net/http"
+	"os"
+)
+
+func main() {
+	port := os.Getenv("port")
+	if port == "" {
+		p := flag.String("port", "8888", "port default 8888")
+		port = *p
+	}
+	dev := proxy.NewProxy()
+	log.Println("start port :" + port)
+	err := http.ListenAndServe(":"+port, dev)
+	if err != nil {
+		log.Fatal(err)
+	}
+}

go.mod

@@ -0,0 +1,3 @@
+module github.com/go-proxy/dev
+
+go 1.14

proxy.go

@@ -0,0 +1,104 @@
+package proxy
+
+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) {
+	//重置路由表
+	p.table.DelAll()
+	data := r.FormValue("data")
+	arr := strings.Split(data, "\n")
+	var newArr []string
+	for _, item := range arr {
+		d := strings.Split(item, "=>")
+		if len(d) < 2 {
+			continue
+		}
+		_, err := url.Parse(d[1])
+		if err != nil {
+			continue
+		}
+		newArr = append(newArr, item)
+		fmt.Println(d[0], d[1])
+		p.table.Set(d[0], d[1])
+	}
+	newData := strings.Join(newArr, "\n")
+	w.Header().Set("Content-Type", "text/html;charset=utf-8")
+	w.Write([]byte("<form><center><textarea autofocus name=\"data\" rows=\"30\" cols=\"100\">" + newData + "</textarea><br><input type=\"submit\" value=\"提交\"></center></form>"))
+}
+
+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(p.table.Get(service))
+	if err != nil {
+		w.Write([]byte(err.Error()))
+		return
+	}
+	r.URL.Path = strings.TrimLeft(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", "")
+		}
+	}
+	proxy := &httputil.ReverseProxy{Director: director}
+	proxy.ServeHTTP(w, r)
+}
+
+//获取url中的第一个参数
+func singleJoiningSlash(a, b string) string {
+	aslash := strings.HasSuffix(a, "/")
+	bslash := strings.HasPrefix(b, "/")
+	switch {
+	case aslash && bslash:
+		return a + b[1:]
+	case !aslash && !bslash:
+		return a + "/" + b
+	}
+	return a + b
+}
+
+func GetURL(r *http.Request) (Url string) {
+	scheme := "http://"
+	if r.TLS != nil {
+		scheme = "https://"
+	}
+	return strings.Join([]string{scheme, r.Host}, "")
+}

table.go

@@ -0,0 +1,54 @@
+package proxy
+
+import "sync"
+
+//路由表
+type table struct {
+	mu   sync.RWMutex
+	data map[string]string
+}
+
+func newTable() *table {
+	return &table{
+		data: make(map[string]string),
+	}
+}
+
+//设置数据
+func (t *table) Set(key string, val string) {
+	t.mu.Lock()
+	t.data[key] = val
+	t.mu.Unlock()
+}
+
+//获取数据
+func (t *table) Get(key string) string {
+	t.mu.RLock()
+	defer t.mu.RUnlock()
+	return t.data[key]
+}
+
+//获取数据
+func (t *table) GetAll() map[string]string {
+	t.mu.RLock()
+	defer t.mu.RUnlock()
+	return t.data
+}
+
+//删除
+func (t *table) DelAll() {
+	t.mu.Lock()
+	t.data = make(map[string]string)
+	t.mu.Unlock()
+}
+
+//长度
+func (t *table) Len() int {
+	return len(t.data)
+}
+
+//Key是否存在
+func (t *table) Exists(key string) bool {
+	_, ok := t.data[key]
+	return ok
+}