router/handler.go

@@ -87,8 +87,9 @@ func RequestHandler(uriCfg UriConfig) gin.HandlerFunc {
 			inputValue = inputValue.Elem()
 		}
 		var firstParam reflect.Value
-		if uriCfg.CtxType == define.CustomContextKey {
-			firstParam = reflect.ValueOf(ctx.MustGet(define.CustomContextKey))
+		if uriCfg.CtxType == CustomContextType {
+			customCtx := ctx.MustGet(define.CustomContextKey)
+			firstParam = reflect.ValueOf(customCtx)
 		} else {
 			firstParam = reflect.ValueOf(ctx)
 		}