Merge pull request #36 from standcats/master

feat : excluede paths by regex

README.md

@@ -99,3 +99,30 @@ func main() {
 	r.Run(":8080")
 }
 ```
+
+
+Customized Excluded Paths
+
+```go
+package main
+
+import (
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/gin-contrib/gzip"
+	"github.com/gin-gonic/gin"
+)
+
+func main() {
+	r := gin.Default()
+	r.Use(gzip.Gzip(gzip.DefaultCompression, gzip.WithExcludedPathsRegexs([]string{".*"})))
+	r.GET("/ping", func(c *gin.Context) {
+		c.String(http.StatusOK, "pong "+fmt.Sprint(time.Now().Unix()))
+	})
+
+	// Listen and Server in 0.0.0.0:8080
+	r.Run(":8080")
+}
+```

handler.go

@@ -73,5 +73,12 @@ func (g *gzipHandler) shouldCompress(req *http.Request) bool {
 		return false
 	}
 
-	return !g.ExcludedPaths.Contains(req.URL.Path)
+	if g.ExcludedPaths.Contains(req.URL.Path) {
+		return false
+	}
+	if g.ExcludedPathesRegexs.Contains(req.URL.Path) {
+		return false
+	}
+
+	return true
 }

options.go

@@ -3,6 +3,7 @@ package gzip
 import (
 	"compress/gzip"
 	"net/http"
+	"regexp"
 	"strings"
 
 	"github.com/gin-gonic/gin"
@@ -18,9 +19,10 @@ var (
 )
 
 type Options struct {
-	ExcludedExtensions ExcludedExtensions
-	ExcludedPaths      ExcludedPaths
-	DecompressFn       func(c *gin.Context)
+	ExcludedExtensions   ExcludedExtensions
+	ExcludedPaths        ExcludedPaths
+	ExcludedPathesRegexs ExcludedPathesRegexs
+	DecompressFn         func(c *gin.Context)
 }
 
 type Option func(*Options)
@@ -37,6 +39,12 @@ func WithExcludedPaths(args []string) Option {
 	}
 }
 
+func WithExcludedPathsRegexs(args []string) Option {
+	return func(o *Options) {
+		o.ExcludedPathesRegexs = NewExcludedPathesRegexs(args)
+	}
+}
+
 func WithDecompressFn(decompressFn func(c *gin.Context)) Option {
 	return func(o *Options) {
 		o.DecompressFn = decompressFn
@@ -74,6 +82,25 @@ func (e ExcludedPaths) Contains(requestURI string) bool {
 	return false
 }
 
+type ExcludedPathesRegexs []*regexp.Regexp
+
+func NewExcludedPathesRegexs(regexs []string) ExcludedPathesRegexs {
+	result := make([]*regexp.Regexp, len(regexs), len(regexs))
+	for i, reg := range regexs {
+		result[i] = regexp.MustCompile(reg)
+	}
+	return result
+}
+
+func (e ExcludedPathesRegexs) Contains(requestURI string) bool {
+	for _, reg := range e {
+		if reg.MatchString(requestURI) {
+			return true
+		}
+	}
+	return false
+}
+
 func DefaultDecompressHandle(c *gin.Context) {
 	if c.Request.Body == nil {
 		return