✨ feat: Update session middleware add Keys method and update docs to match key type any

sixcolors · sixcolors · commit ea8f83148805 · 2025-06-12T17:08:31.000-03:00
diff --git a/docs/middleware/session.md b/docs/middleware/session.md
@@ -194,9 +194,10 @@ func DefaultErrorHandler(fiber.Ctx, err error)
 ### Middleware Methods
 
 ```go
-func (m *Middleware) Set(key string, value any)
-func (m *Middleware) Get(key string) any
-func (m *Middleware) Delete(key string)
+func (m *Middleware) Set(key any, value any)
+func (m *Middleware) Get(key any) any
+func (m *Middleware) Delete(key any)
+func (m *Middleware) Keys() []any
 func (m *Middleware) Destroy() error
 func (m *Middleware) Reset() error
 func (m *Middleware) Store() *Store
@@ -207,14 +208,15 @@ func (m *Middleware) Store() *Store
 ```go
 func (s *Session) Fresh() bool
 func (s *Session) ID() string
-func (s *Session) Get(key string) any
-func (s *Session) Set(key string, val any)
+func (s *Session) Get(key any) any
+func (s *Session) Set(key any, val any)
+func (s *Session) Delete(key any)
+func (s *Session) Keys() []any
 func (s *Session) Destroy() error
 func (s *Session) Regenerate() error
 func (s *Session) Release()
 func (s *Session) Reset() error
 func (s *Session) Save() error
-func (s *Session) Keys() []string
 func (s *Session) SetIdleTimeout(idleTimeout time.Duration)
 ```
 
diff --git a/middleware/session/middleware.go b/middleware/session/middleware.go
@@ -232,6 +232,21 @@ func (m *Middleware) Delete(key any) {
 	m.Session.Delete(key)
 }
 
+// Keys returns all keys in the current session.
+//
+// Returns:
+//   - []any: A slice of all keys in the session.
+//
+// Usage:
+//
+//	keys := m.Keys()
+func (m *Middleware) Keys() []any {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+
+	return m.Session.Keys()
+}
+
 // Destroy destroys the session.
 //
 // Returns:
diff --git a/middleware/session/session.go b/middleware/session/session.go
@@ -325,7 +325,7 @@ func (s *Session) saveSession() error {
 // Keys retrieves all keys in the current session.
 //
 // Returns:
-//   - []string: A slice of all keys in the session.
+//   - []any: A slice of all keys in the session.
 //
 // Usage:
 //
