fix(#979): GetHubFromContext may crash instead of returning nil

gofiber · Feb 22, 2024 · 2d0926f · 2d0926f
1 parent 668510e
commit 2d0926f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/fibersentry/sentry.go b/fibersentry/sentry.go
@@ -18,7 +18,6 @@ func New(config ...Config) fiber.Handler {
 	return func(c *fiber.Ctx) error {
 		// Convert fiber request to http request
 		r, err := adaptor.ConvertRequest(c, true)
-
 		if err != nil {
 			return err
 		}
@@ -54,5 +53,9 @@ func New(config ...Config) fiber.Handler {
 }
 
 func GetHubFromContext(ctx *fiber.Ctx) *sentry.Hub {
-	return ctx.Locals(hubKey).(*sentry.Hub)
+	hub, ok := ctx.Locals(hubKey).(*sentry.Hub)
+	if !ok {
+		return nil
+	}
+	return hub
 }