支持gzip解析数据

proxy/forward.go

@@ -8,11 +8,15 @@
 package proxy
 
 import (
+	"compress/gzip"
 	"fmt"
 	"io"
+	"io/ioutil"
 	"net"
 	"net/http"
 	"strings"
+
+	"git.zhangdeman.cn/zhangdeman/gopkg/safe"
 )
 
 // Forward 正向代理的实现
@@ -37,7 +41,7 @@ func Forward(rw http.ResponseWriter, req *http.Request, serverConfig *Server) {
 	}
 
 	// 写入重写的请求Header
-	for k, v := range serverConfig.RewriteHeader {
+	for k, v := range serverConfig.RewriteRequestHeader {
 		outReq.Header.Set(k, v)
 	}
 
@@ -46,6 +50,7 @@ func Forward(rw http.ResponseWriter, req *http.Request, serverConfig *Server) {
 	outReq.URL.Path = serverConfig.URI
 	outReq.URL.Scheme = serverConfig.Scheme
 	outReq.URL.Host = serverConfig.Host
+
 	// step 2
 	res, err := transport.RoundTrip(outReq)
 	if err != nil {
@@ -61,6 +66,24 @@ func Forward(rw http.ResponseWriter, req *http.Request, serverConfig *Server) {
 	}
 
 	rw.WriteHeader(res.StatusCode)
-	io.Copy(rw, res.Body)
+
+	// 重写请求header
+	for k, v := range serverConfig.RewriteResponseHeader {
+		rw.Header().Set(k, v)
+	}
+
+	// 重写响应数据
+	if !strings.Contains(strings.ToLower(res.Header.Get("Content-Type")), "application/json") || nil == serverConfig.RewriteResponseData || len(serverConfig.RewriteResponseData) == 0 {
+		io.Copy(rw, res.Body)
+	} else {
+		var responseData []byte
+		var gzipData io.Reader
+		gzipData, err = gzip.NewReader(res.Body)
+		responseData, err = ioutil.ReadAll(gzipData)
+		fmt.Println(string(responseData), err)
+		formatData, _ := safe.Filter(responseData, serverConfig.RewriteResponseData)
+		rw.Write(formatData)
+	}
+
 	res.Body.Close()
 }