Compare commits

..

2 commits
v0.1.0 ... main

Author SHA1 Message Date
05a4d62360 remove dead code from example
All checks were successful
continuous-integration/drone/push Build is passing
2021-12-17 19:08:08 +01:00
32a105e462 use map for ignored extensions
All checks were successful
continuous-integration/drone/push Build is passing
2021-12-17 14:18:40 +01:00
2 changed files with 10 additions and 17 deletions

View file

@ -8,10 +8,6 @@ import (
func main() {
router := gin.Default()
options := ginzip.DefaultOptions()
options.BrotliLevel = ""
options.GzipLevel = ""
ui := router.Group("/", ginzip.New(ginzip.DefaultOptions()))
ui.GET("/", getTestHandler("Hello World (should be compressed)"))

View file

@ -33,7 +33,7 @@ type intOptions struct {
GzipLevel int
BrotliEn bool
BrotliLevel int
SkipExtensions []string
SkipExtensions map[string]bool
}
// New creates a new Ginzip middleware function.
@ -136,16 +136,7 @@ func (b *brotliWriter) Encoding() string {
return "br"
}
func containsString(s []string, e string) bool {
for _, a := range s {
if a == e {
return true
}
}
return false
}
func shouldCompress(req *http.Request, skipExtensions []string) bool {
func shouldCompress(req *http.Request, skipExtensions map[string]bool) bool {
// Dont compress websocket connections
if strings.Contains(req.Header.Get("Connection"), "Upgrade") ||
strings.Contains(req.Header.Get("Content-Type"), "text/event-stream") {
@ -157,7 +148,8 @@ func shouldCompress(req *http.Request, skipExtensions []string) bool {
return true
}
return !containsString(skipExtensions, extension)
_, doSkip := skipExtensions[extension]
return !doSkip
}
func parseLevel(level string, dfault int, min int, max int) (int, bool) {
@ -193,12 +185,17 @@ func parseOptions(options Options) intOptions {
brotli.DefaultCompression, brotli.BestSpeed, brotli.BestCompression,
)
exts := map[string]bool{}
for _, ext := range options.SkipExtensions {
exts[ext] = true
}
return intOptions{
GzipEn: gzipEn,
GzipLevel: gzipLvl,
BrotliEn: brEn,
BrotliLevel: brLvl,
SkipExtensions: options.SkipExtensions,
SkipExtensions: exts,
}
}