🔥 Feature: Add Session to Go Context in Middleware

middleware/session/middleware.go

@@ -127,6 +127,7 @@ func (m *Middleware) initialize(c fiber.Ctx, cfg Config) {
 	m.ctx = c
 
 	c.Locals(middlewareContextKey, m)
+	c.SetContext(context.WithValue(c.Context(), sessionContextKey, session))
 }
 
 // saveSession handles session saving and error management after the response.

middleware/session/store.go

@@ -1,7 +1,6 @@
 package session
 
 import (
-	"context"
 	"encoding/gob"
 	"errors"
 	"fmt"
@@ -106,10 +105,6 @@ func (s *Store) Get(c fiber.Ctx) (*Session, error) {
 		return nil, err
 	}
 
-	// Add session to Go context
-	ctx := context.WithValue(c.Context(), sessionContextKey, sess)
-	c.SetContext(ctx)
-
 	return sess, nil
 }
 

middleware/session/store_test.go

@@ -2,8 +2,6 @@ package session
 
 import (
 	"fmt"
-	"io"
-	"net/http/httptest"
 	"testing"
 
 	"github.com/gofiber/fiber/v3"
@@ -225,55 +223,3 @@ func Test_Store_GetByID(t *testing.T) {
 		})
 	})
 }
-
-func Test_Store_Get_GoContext(t *testing.T) {
-	t.Parallel()
-	app := fiber.New()
-	store := NewStore()
-
-	app.Get("/direct-get", func(c fiber.Ctx) error {
-		// Get session directly from store (not using middleware)
-		sess, err := store.Get(c)
-		if err != nil {
-			return err
-		}
-		defer sess.Release()
-
-		// Set some data
-		sess.Set("name", "JIeJaitt")
-
-		// Now get session from Go context
-		goCtxSess := FromGoContext(c.Context())
-
-		// Verify the session exists in Go context
-		if goCtxSess == nil {
-			return c.Status(fiber.StatusInternalServerError).SendString("Session not found in Go context")
-		}
-
-		// Verify it's the same session
-		if goCtxSess.ID() != sess.ID() {
-			return c.Status(fiber.StatusInternalServerError).SendString("Session IDs don't match")
-		}
-
-		// Verify data is accessible from Go context session
-		if goCtxSess.Get("name") != "JIeJaitt" {
-			return c.Status(fiber.StatusInternalServerError).SendString("Wrong value in Go context session")
-		}
-
-		// Save session
-		if err := sess.Save(); err != nil {
-			return err
-		}
-
-		return c.SendString("success")
-	})
-
-	// Make request
-	resp, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/direct-get", nil))
-	require.NoError(t, err)
-	require.Equal(t, fiber.StatusOK, resp.StatusCode)
-
-	body, err := io.ReadAll(resp.Body)
-	require.NoError(t, err)
-	require.Equal(t, "success", string(body))
-}