From e8b9fd632b670c5296b8112f724d69e31dc64680 Mon Sep 17 00:00:00 2001
From: JIeJaitt <498938874@qq.com>
Date: Wed, 12 Mar 2025 16:34:33 +0800
Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20Refactor:=20Improve=20From?=
 =?UTF-8?q?Context=20error=20handling=20in=20session=20middleware?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 middleware/session/middleware.go | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/middleware/session/middleware.go b/middleware/session/middleware.go
index 3f315e6b..7ae173d9 100644
--- a/middleware/session/middleware.go
+++ b/middleware/session/middleware.go
@@ -8,7 +8,6 @@ import (
 	"sync"
 
 	"github.com/gofiber/fiber/v3"
-	"github.com/gofiber/fiber/v3/log"
 )
 
 // Middleware holds session data and configuration.
@@ -178,16 +177,16 @@ func releaseMiddleware(m *Middleware) {
 // If there is no Middleware, nil is returned.
 func FromContext(c any) *Middleware {
 	switch ctx := c.(type) {
-	case fiber.Ctx:
-		if m, ok := ctx.Locals(middlewareContextKey).(*Middleware); ok && m != nil {
-			return m
-		}
 	case context.Context:
 		if m, ok := ctx.Value(middlewareContextKey).(*Middleware); ok {
 			return m
 		}
+	case fiber.Ctx:
+		if m, ok := ctx.Locals(middlewareContextKey).(*Middleware); ok && m != nil {
+			return m
+		}
 	default:
-		log.Errorf("Unsupported context type: %T. Expected fiber.Ctx or context.Context", c)
+		panic("unsupported context type, expected fiber.Ctx or context.Context")
 	}
 	return nil
 }