bugfixes & optimization

coverage.html

@@ -1424,8 +1424,9 @@
 
 import (
         "bytes"
-        "compress/gzip"
         "fmt"
+
+        "github.com/klauspost/compress/gzip"
 )
 
 func Pack(data []byte) (*bytes.Buffer, error) <span class="cov8" title="1">{
@@ -1451,10 +1452,11 @@
 		<pre class="file" id="file11" style="display: none">package compression
 
 import (
-        "compress/gzip"
         "context"
         "net/http"
 
+        "github.com/klauspost/compress/gzip"
+
         "github.com/ex0rcist/metflix/internal/logging"
 )
 

internal/compression/compression.go

@@ -2,8 +2,9 @@ package compression
 
 import (
 	"bytes"
-	"compress/gzip"
 	"fmt"
+
+	"github.com/klauspost/compress/gzip"
 )
 
 func Pack(data []byte) (*bytes.Buffer, error) {

internal/compression/compression_test.go

@@ -2,9 +2,10 @@ package compression
 
 import (
 	"bytes"
-	"compress/gzip"
 	"io"
 	"testing"
+
+	"github.com/klauspost/compress/gzip"
 )
 
 func TestPack_Success(t *testing.T) {

internal/compression/compressor.go

@@ -1,10 +1,11 @@
 package compression
 
 import (
-	"compress/gzip"
 	"context"
 	"net/http"
 
+	"github.com/klauspost/compress/gzip"
+
 	"github.com/ex0rcist/metflix/internal/logging"
 )
 

internal/compression/compressor_test.go

@@ -2,10 +2,11 @@ package compression
 
 import (
 	"bytes"
-	"compress/gzip"
 	"context"
 	"net/http/httptest"
 	"testing"
+
+	"github.com/klauspost/compress/gzip"
 )
 
 func TestCompressor_Write_SupportedContent(t *testing.T) {

internal/compression/decompressor_test.go

@@ -2,11 +2,12 @@ package compression
 
 import (
 	"bytes"
-	"compress/gzip"
 	"context"
 	"errors"
 	"io"
 
+	"github.com/klauspost/compress/gzip"
+
 	"github.com/ex0rcist/metflix/internal/entities"
 
 	"net/http"

internal/middleware/compression_test.go

@@ -2,13 +2,38 @@ package middleware
 
 import (
 	"bytes"
-	"compress/gzip"
+
 	"io"
 	"net/http"
 	"net/http/httptest"
 	"testing"
+
+	"github.com/klauspost/compress/gzip"
 )
 
+func benchFn(data []byte) {
+	// data := []byte("test data")
+
+	nextHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Content-Type", "application/json")
+		_, _ = w.Write(data)
+	})
+
+	req := httptest.NewRequest(http.MethodGet, "/", nil)
+	req.Header.Set("Accept-Encoding", "gzip")
+
+	rr := httptest.NewRecorder()
+	handler := CompressResponse(nextHandler)
+	handler.ServeHTTP(rr, req)
+}
+
+func BenchmarkCompression(b *testing.B) {
+	payload := []byte("test data")
+	for i := 0; i < b.N; i++ {
+		benchFn(payload)
+	}
+}
+
 func TestDecompressRequest_Success(t *testing.T) {
 	data := []byte("test data")
 	var buf bytes.Buffer