Fixed graphql.GetInitPayload panic if payload is nil.

99designs · Sep 20, 2018 · 575f28e · 575f28e
1 parent 380828f
commit 575f28e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/graphql/context.go b/graphql/context.go
@@ -240,5 +240,10 @@ func WithInitPayload(ctx context.Context, payload InitPayload) context.Context {
 
 // GetInitPayload gets the payload from context.
 func GetInitPayload(ctx context.Context) InitPayload {
-	return ctx.Value(initpayload).(InitPayload)
+	payload, ok := ctx.Value(initpayload).(InitPayload)
+	if !ok {
+		return nil
+	}
+
+	return payload
 }