mirror of
https://github.com/gofiber/fiber.git
synced 2025-05-16 04:33:01 +00:00
1cddc56f13
* ✨ Cache middleware: Store e2e headers. As defined in RFC2616 - section-13.5.1, shared caches MUST store end-to-end headers from backend response and MUST be transmitted in any response formed from a cache entry. This commit ensures a stronger consistency between responses served from the handlers & from the cache middleware. * ✨ Cache middleware: Add flag for e2e headers. Set flag to prevent e2e headers caching to be the default behavior of the cache middleware. This would otherwise change quite a lot the experience for cache middleware current users. * ✨ Cache middleware: Add Benchmark for additionalHeaders feature. * ✨ Cache middleware: Rename E2Eheaders into StoreResponseHeaders. E2E is an acronym commonly associated with test. While in the present case it refers to end-to-end HTTP headers (by opposition to hop-by-hop), this still remains confusing. This commits renames it to a more generic name. * ✨ Cache middleware: Update README * ✨ Cache middleware: Move map instanciation. This will prevent an extra memory allocation for users not interested in this feature. * ✨ Cache middleware: Prevent memory allocation when StoreResponseHeaders is disabled. * ✨ Cache middleware: Store e2e headers. #1807 - use set instead of add for the headers - copy value from the headers -> prevent problems with mutable values Co-authored-by: wernerr <rene@gofiber.io>
111 lines
2.6 KiB
Go
111 lines
2.6 KiB
Go
package cache
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/gofiber/fiber/v2/utils"
|
|
)
|
|
|
|
// 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
|
|
|
|
// Expiration is the time that an cached response will live
|
|
//
|
|
// Optional. Default: 1 * time.Minute
|
|
Expiration time.Duration
|
|
|
|
// CacheHeader header on response header, indicate cache status, with the following possible return value
|
|
//
|
|
// hit, miss, unreachable
|
|
//
|
|
// Optional. Default: X-Cache
|
|
CacheHeader string
|
|
|
|
// CacheControl enables client side caching if set to true
|
|
//
|
|
// Optional. Default: false
|
|
CacheControl bool
|
|
|
|
// Key allows you to generate custom keys, by default c.Path() is used
|
|
//
|
|
// Default: func(c *fiber.Ctx) string {
|
|
// return utils.CopyString(c.Path())
|
|
// }
|
|
KeyGenerator func(*fiber.Ctx) string
|
|
|
|
// allows you to generate custom Expiration Key By Key, default is Expiration (Optional)
|
|
//
|
|
// Default: nil
|
|
ExpirationGenerator func(*fiber.Ctx, *Config) time.Duration
|
|
|
|
// Store is used to store the state of the middleware
|
|
//
|
|
// Default: an in memory store for this process only
|
|
Storage fiber.Storage
|
|
|
|
// Deprecated, use Storage instead
|
|
Store fiber.Storage
|
|
|
|
// Deprecated, use KeyGenerator instead
|
|
Key func(*fiber.Ctx) string
|
|
|
|
// allows you to store additional headers generated by next middlewares & handler
|
|
//
|
|
// Default: false
|
|
StoreResponseHeaders bool
|
|
}
|
|
|
|
// ConfigDefault is the default config
|
|
var ConfigDefault = Config{
|
|
Next: nil,
|
|
Expiration: 1 * time.Minute,
|
|
CacheHeader: "X-Cache",
|
|
CacheControl: false,
|
|
KeyGenerator: func(c *fiber.Ctx) string {
|
|
return utils.CopyString(c.Path())
|
|
},
|
|
ExpirationGenerator: nil,
|
|
StoreResponseHeaders: false,
|
|
Storage: nil,
|
|
}
|
|
|
|
// Helper function to set default values
|
|
func configDefault(config ...Config) Config {
|
|
// Return default config if nothing provided
|
|
if len(config) < 1 {
|
|
return ConfigDefault
|
|
}
|
|
|
|
// Override default config
|
|
cfg := config[0]
|
|
|
|
// Set default values
|
|
if cfg.Store != nil {
|
|
fmt.Println("[CACHE] Store is deprecated, please use Storage")
|
|
cfg.Storage = cfg.Store
|
|
}
|
|
if cfg.Key != nil {
|
|
fmt.Println("[CACHE] Key is deprecated, please use KeyGenerator")
|
|
cfg.KeyGenerator = cfg.Key
|
|
}
|
|
if cfg.Next == nil {
|
|
cfg.Next = ConfigDefault.Next
|
|
}
|
|
if int(cfg.Expiration.Seconds()) == 0 {
|
|
cfg.Expiration = ConfigDefault.Expiration
|
|
}
|
|
if cfg.CacheHeader == "" {
|
|
cfg.CacheHeader = ConfigDefault.CacheHeader
|
|
}
|
|
if cfg.KeyGenerator == nil {
|
|
cfg.KeyGenerator = ConfigDefault.KeyGenerator
|
|
}
|
|
return cfg
|
|
}
|