- simplify gzip coder middleware

examples/middleware/coder/gzip/client/client.go

@@ -18,7 +18,7 @@ func main() {
 	}
 	defer client.Stop()
 
-	client.Handler.UseCoder(gzip.New())
+	client.Handler.UseCoder(gzip.New(1024))
 
 	req := ""
 	for i := 0; i < 2048; i++ {

examples/middleware/coder/gzip/server/server.go

@@ -9,7 +9,7 @@ import (
 func main() {
 	svr := arpc.NewServer()
 
-	svr.Handler.UseCoder(gzip.New())
+	svr.Handler.UseCoder(gzip.New(1024))
 
 	// register router
 	svr.Handler.Handle("/echo", func(ctx *arpc.Context) {

extension/middleware/coder/gzip/gzip.go

@@ -32,13 +32,11 @@ func gzipUnCompress(data []byte) ([]byte, error) {
 }
 
 // Gzip represents a gzip coding middleware.
-type Gzip struct {
-	critical int
-}
+type Gzip int
 
 // Encode implements arpc MessageCoder.
-func (c *Gzip) Encode(client *arpc.Client, msg *arpc.Message) *arpc.Message {
-	if len(msg.Buffer) > c.critical && !msg.IsFlagBitSet(coder.FlagBitGZip) {
+func (g *Gzip) Encode(client *arpc.Client, msg *arpc.Message) *arpc.Message {
+	if len(msg.Buffer) > int(*g) && !msg.IsFlagBitSet(coder.FlagBitGZip) {
 		buf := gzipCompress(msg.Buffer[arpc.HeaderIndexReserved+1:])
 		total := len(buf) + arpc.HeaderIndexReserved + 1
 		if total < len(msg.Buffer) {
@@ -52,7 +50,7 @@ func (c *Gzip) Encode(client *arpc.Client, msg *arpc.Message) *arpc.Message {
 }
 
 // Decode implements arpc MessageCoder.
-func (c *Gzip) Decode(client *arpc.Client, msg *arpc.Message) *arpc.Message {
+func (g *Gzip) Decode(client *arpc.Client, msg *arpc.Message) *arpc.Message {
 	if msg.IsFlagBitSet(coder.FlagBitGZip) {
 		buf, err := gzipUnCompress(msg.Buffer[arpc.HeaderIndexReserved+1:])
 		if err == nil {
@@ -65,6 +63,7 @@ func (c *Gzip) Decode(client *arpc.Client, msg *arpc.Message) *arpc.Message {
 }
 
 // New returns the gzip coding middleware.
-func New() *Gzip {
-	return &Gzip{critical: 1024}
+func New(n int) *Gzip {
+	var g = Gzip(n)
+	return &g
 }