Add Lazy behavior when using resolver as a service

overblog · Dec 29, 2021 · ea86f8d · ea86f8d
1 parent f9f7552
commit ea86f8d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/Generator/TypeBuilder.php b/src/Generator/TypeBuilder.php
@@ -886,9 +886,14 @@ private function buildCallback(Callback $callback, array $argNames, ?callable $e
                 return $expressionBuilder($callback->expression);
             }
         } elseif (null !== $callback->id) {
-            $fn = "$this->gqlServices->get('$callback->id')";
+            $fnExpression = "$this->gqlServices->get('$callback->id')";
+            if (null !== $callback->method) {
+                $fnExpression = "[$fnExpression, '$callback->method']";
+            }
 
-            return Collection::numeric([$fn, $callback->method ?? '__invoke']);
+            return ArrowFunction::new()
+                ->addArguments(...$argNames)
+                ->setExpression(Literal::new("($fnExpression)(...\\func_get_args())"));
         } else {
             return Literal::new("'$callback->method'");
         }