Merge pull request #1054 from Fenny/master

🩹 draft for #1051

app.go

@@ -31,7 +31,7 @@ import (
 )
 
 // Version of current fiber package
-const Version = "2.2.3"
+const Version = "2.2.4"
 
 // Handler defines a function to serve HTTP requests.
 type Handler = func(*Ctx) error

internal/gotiny/decEngine.go

@@ -1,6 +1,7 @@
 package gotiny
 
 import (
+	"fmt"
 	"reflect"
 	"sync"
 	"time"
@@ -178,7 +179,8 @@ func buildDecEngine(rt reflect.Type, engPtr *decEng) {
 				decString(d, unsafe.Pointer(&name))
 				et, has := name2type[name]
 				if !has {
-					panic("unknown typ:" + name)
+					//panic("unknown typ:" + name)
+					fmt.Println("[session] Register this type first with the `RegisterType` method.")
 				}
 				v := reflect.NewAt(rt, p).Elem()
 				var ev reflect.Value
@@ -194,7 +196,7 @@ func buildDecEngine(rt reflect.Type, engPtr *decEng) {
 			}
 		}
 	case reflect.Chan, reflect.Func:
-		panic("not support " + rt.String() + " type")
+		//panic("not support " + rt.String() + " type")
 	default:
 		engine = baseDecEngines[kind]
 	}

internal/gotiny/encEngine.go

@@ -187,7 +187,7 @@ func buildEncEngine(rt reflect.Type, engPtr *encEng) {
 			}
 		}
 	case reflect.Chan, reflect.Func:
-		panic("not support " + rt.String() + " type")
+		//panic("not support " + rt.String() + " type")
 	default:
 		engine = encEngines[kind]
 	}

middleware/cache/cache.go

@@ -87,7 +87,9 @@ func New(config ...Config) fiber.Handler {
 			c.Response().SetBodyRaw(e.body)
 			c.Response().SetStatusCode(e.status)
 			c.Response().Header.SetContentTypeBytes(e.ctype)
-
+			if len(e.cencoding) > 0 {
+				c.Response().Header.SetBytesV(fiber.HeaderContentEncoding, e.cencoding)
+			}
 			// Set Cache-Control header if enabled
 			if cfg.CacheControl {
 				maxAge := strconv.FormatUint(e.exp-ts, 10)
@@ -107,6 +109,7 @@ func New(config ...Config) fiber.Handler {
 		e.body = utils.SafeBytes(c.Response().Body())
 		e.status = c.Response().StatusCode()
 		e.ctype = utils.SafeBytes(c.Response().Header.ContentType())
+		e.cencoding = utils.SafeBytes(c.Response().Header.Peek(fiber.HeaderContentEncoding))
 
 		// For external Storage we store raw body seperated
 		if cfg.Storage != nil {

middleware/cache/manager.go

@@ -13,10 +13,11 @@ import (
 // don't forget to replace the msgp import path to:
 // "github.com/gofiber/fiber/v2/internal/msgp"
 type item struct {
-	body   []byte
-	ctype  []byte
-	status int
-	exp    uint64
+	body      []byte
+	ctype     []byte
+	cencoding []byte
+	status    int
+	exp       uint64
 }
 
 //msgp:ignore manager

middleware/session/README.md

@@ -11,6 +11,7 @@ Session middleware for [Fiber](https://github.com/gofiber/fiber)
 ### Signatures
 ```go
 func New(config ...Config) *Store
+func (s *Store) RegisterType(i interface{})
 func (s *Store) Get(c *fiber.Ctx) (*Session, error)
 func (s *Store) Reset() error
 

middleware/session/config.go

@@ -44,6 +44,11 @@ type Config struct {
 	// KeyGenerator generates the session key.
 	// Optional. Default value utils.UUIDv4
 	KeyGenerator func() string
+
+	// CustomType allows you to store custom type/struct
+	// in any Storage provider. Only use this option if you
+	// are using custom a custom type/struct used as value.
+	CustomType interface{}
 }
 
 // ConfigDefault is the default config

middleware/session/store.go

@@ -14,9 +14,6 @@ type Store struct {
 
 var mux sync.Mutex
 
-// Storage ErrNotExist
-var errNotExist = "key does not exist"
-
 func New(config ...Config) *Store {
 	// Set default config
 	cfg := configDefault(config...)
@@ -30,6 +27,13 @@ func New(config ...Config) *Store {
 	}
 }
 
+// RegisterType will allow you to encode/decode custom types
+// into any Storage provider
+func (s *Store) RegisterType(i interface{}) {
+	gotiny.Register(i)
+}
+
+// Get will get/create a session
 func (s *Store) Get(c *fiber.Ctx) (*Session, error) {
 	var fresh bool