fix: test error and decompressMiddleware doesn't work

client_middleware.go

@@ -85,7 +85,11 @@ func (g *gzipClientMiddleware) ClientMiddleware(next client.Endpoint) client.End
 			return
 		}
 		if fn := g.DecompressFnForClient; fn != nil && strings.EqualFold(resp.Header.Get("Content-Encoding"), "gzip") {
-			fn(next)
+			f := fn(next)
+			err = f(ctx, req, resp)
+			if err != nil {
+				return err
+			}
 		}
 		return nil
 	}

gzip_test.go

@@ -405,11 +405,12 @@ func TestNoGzipForClient(t *testing.T) {
 func TestDecompressGzipForClient(t *testing.T) {
 	h := server.Default(server.WithHostPorts("127.0.0.1:2338"))
 
+	h.Use(Gzip(DefaultCompression, WithDecompressFn(DefaultDecompressHandle)))
 	h.GET("/", func(ctx context.Context, c *app.RequestContext) {
 		c.Header("Content-Length", strconv.Itoa(len(testResponse)))
 		c.String(200, testResponse)
 	})
-	h.Use(Gzip(DefaultCompression, WithDecompressFn(DefaultDecompressHandle)))
+
 	go h.Spin()
 
 	time.Sleep(time.Second)
@@ -425,6 +426,7 @@ func TestDecompressGzipForClient(t *testing.T) {
 
 	req.SetBodyString("bar")
 	req.SetRequestURI("http://127.0.0.1:2338/")
+	req.SetHeader("Accept-Encoding", "gzip")
 
 	err = cli.Do(context.Background(), req, res)
 	if err != nil {

options.go

@@ -41,6 +41,7 @@
 package gzip
 
 import (
+	"bytes"
 	"context"
 	"net/http"
 	"regexp"
@@ -216,7 +217,8 @@ func DefaultDecompressMiddlewareForClient(next client.Endpoint) client.Endpoint
 		}
 		resp.Header.DelBytes([]byte("Content-Encoding"))
 		resp.Header.DelBytes([]byte("Content-Length"))
-		resp.SetBody(gunzipBytes)
-		return next(ctx, req, resp)
+		resp.Header.DelBytes([]byte("Vary"))
+		resp.SetBodyStream(bytes.NewBuffer(gunzipBytes), len(gunzipBytes))
+		return nil
 	}
 }