#881 add X-RPCX-CompressType to indicate whether the bodu has been co…

…mpressed

server/converter.go

@@ -17,6 +17,7 @@ const (
 	XOneway            = "X-RPCX-Oneway"
 	XMessageStatusType = "X-RPCX-MessageStatusType"
 	XSerializeType     = "X-RPCX-SerializeType"
+	XCompressType      = "X-RPCX-CompressType"
 	XMessageID         = "X-RPCX-MessageID"
 	XServicePath       = "X-RPCX-ServicePath"
 	XServiceMethod     = "X-RPCX-ServiceMethod"
@@ -58,6 +59,16 @@ func HTTPRequest2RpcxRequest(r *http.Request) (*protocol.Message, error) {
 		req.SetSerializeType(protocol.SerializeType(rst))
 	}
 
+	compressType := h.Get(XCompressType)
+	if compressType != "" {
+		ct, err := strconv.Atoi(compressType)
+		if err != nil {
+			return nil, err
+		}
+
+		req.SetCompressType(protocol.CompressType(ct))
+	}
+
 	meta := h.Get(XMeta)
 	if meta != "" {
 		metadata, err := url.ParseQuery(meta)

server/gateway.go

@@ -4,6 +4,7 @@ import (
 	"bufio"
 	"context"
 	"errors"
+	"fmt"
 	"io"
 	"net"
 	"net/http"
@@ -273,6 +274,11 @@ func (s *Server) handleGatewayRequest(w http.ResponseWriter, r *http.Request, pa
 		meta.Add(k, v)
 	}
 	wh.Set(XMeta, meta.Encode())
+
+	if res.CompressType() != protocol.None {
+		wh.Set(XCompressType, fmt.Sprintf("%d", res.CompressType()))
+	}
+
 	w.Write(res.Payload)
 	s.Plugins.DoPostWriteResponse(newCtx, req, res, err)
 }