mirror of https://github.com/gofiber/fiber.git
109 lines
2.3 KiB
Go
109 lines
2.3 KiB
Go
package compress
|
|
|
|
import (
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
// Config defines the config for middleware.
|
|
type Config struct {
|
|
// Next defines a function to skip this middleware when returned true.
|
|
//
|
|
// Optional. Default: nil
|
|
Next func(c *fiber.Ctx) bool
|
|
|
|
// Level determines the compression algorithm
|
|
//
|
|
// Optional. Default: LevelDefault
|
|
// LevelDisabled: -1
|
|
// LevelDefault: 0
|
|
// LevelBestSpeed: 1
|
|
// LevelBestCompression: 2
|
|
Level Level
|
|
}
|
|
|
|
// Level is numeric representation of compression level
|
|
type Level int
|
|
|
|
// Represents compression level that will be used in the middleware
|
|
const (
|
|
LevelDisabled Level = -1
|
|
LevelDefault Level = 0
|
|
LevelBestSpeed Level = 1
|
|
LevelBestCompression Level = 2
|
|
)
|
|
|
|
// ConfigDefault is the default config
|
|
var ConfigDefault = Config{
|
|
Next: nil,
|
|
Level: LevelDefault,
|
|
}
|
|
|
|
// New creates a new middleware handler
|
|
func New(config ...Config) fiber.Handler {
|
|
// Set default config
|
|
cfg := ConfigDefault
|
|
|
|
// Override config if provided
|
|
if len(config) > 0 {
|
|
cfg = config[0]
|
|
|
|
// Set default values
|
|
if cfg.Level < LevelDisabled || cfg.Level > LevelBestCompression {
|
|
cfg.Level = ConfigDefault.Level
|
|
}
|
|
}
|
|
|
|
// Setup request handlers
|
|
var (
|
|
fctx = func(c *fasthttp.RequestCtx) {}
|
|
compressor fasthttp.RequestHandler
|
|
)
|
|
|
|
// Setup compression algorithm
|
|
switch cfg.Level {
|
|
case LevelDefault:
|
|
// LevelDefault
|
|
compressor = fasthttp.CompressHandlerBrotliLevel(fctx,
|
|
fasthttp.CompressBrotliDefaultCompression,
|
|
fasthttp.CompressDefaultCompression,
|
|
)
|
|
case LevelBestSpeed:
|
|
// LevelBestSpeed
|
|
compressor = fasthttp.CompressHandlerBrotliLevel(fctx,
|
|
fasthttp.CompressBrotliBestSpeed,
|
|
fasthttp.CompressBestSpeed,
|
|
)
|
|
case LevelBestCompression:
|
|
// LevelBestCompression
|
|
compressor = fasthttp.CompressHandlerBrotliLevel(fctx,
|
|
fasthttp.CompressBrotliBestCompression,
|
|
fasthttp.CompressBestCompression,
|
|
)
|
|
default:
|
|
// LevelDisabled
|
|
return func(c *fiber.Ctx) error {
|
|
return c.Next()
|
|
}
|
|
}
|
|
|
|
// Return new handler
|
|
return func(c *fiber.Ctx) error {
|
|
// Don't execute middleware if Next returns true
|
|
if cfg.Next != nil && cfg.Next(c) {
|
|
return c.Next()
|
|
}
|
|
|
|
// Continue stack
|
|
if err := c.Next(); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Compress response
|
|
compressor(c.Context())
|
|
|
|
// Return from handler
|
|
return nil
|
|
}
|
|
}
|