fix: test error and decompressMiddleware doesn't work (#20)

* fix: test error and decompressMiddleware doesn't work

* fix: return error in ClientMiddleware

* chore: optimize license header

Author: jkskj

client_middleware.go

@@ -1,5 +1,5 @@
 /*
- * Copyright 2023 CloudWeGo Authors
+ * Copyright 2024 CloudWeGo Authors
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -82,10 +82,14 @@ func (g *gzipClientMiddleware) ClientMiddleware(next client.Endpoint) client.End
 
 		err = next(ctx, req, resp)
 		if err != nil {
-			return
+			return err
 		}
 		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
 	}

example/standard/main.go

@@ -1,5 +1,5 @@
 /*
- * Copyright 2023 CloudWeGo Authors
+ * Copyright 2024 CloudWeGo Authors
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.

example/stream/main.go

@@ -1,5 +1,5 @@
 /*
- * Copyright 2023 CloudWeGo Authors
+ * Copyright 2024 CloudWeGo Authors
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.

gzip.go

@@ -1,5 +1,5 @@
 /*
- * Copyright 2023 CloudWeGo Authors
+ * Copyright 2024 CloudWeGo Authors
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.

gzip_test.go

@@ -1,5 +1,5 @@
 /*
- * Copyright 2023 CloudWeGo Authors
+ * Copyright 2024 CloudWeGo Authors
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -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

@@ -1,5 +1,5 @@
 /*
- * Copyright 2023 CloudWeGo Authors
+ * Copyright 2024 CloudWeGo Authors
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -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
 	}
 }

srv_middleware.go

@@ -1,5 +1,5 @@
 /*
- * Copyright 2023 CloudWeGo Authors
+ * Copyright 2024 CloudWeGo Authors
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.

srv_stream_middleware.go

@@ -1,5 +1,5 @@
 /*
- * Copyright 2023 CloudWeGo Authors
+ * Copyright 2024 CloudWeGo Authors
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.