Merge pull request #52 from moufmouf/autowire_on_prefetch

Fixing @autowire in prefetch method

Author: moufmouf

DependencyInjection/GraphqliteCompilerPass.php

@@ -291,13 +291,16 @@ private function makePublicInjectedServices(ReflectionClass $refClass, Annotatio
     {
         $services = $this->getCodeCache()->get($refClass, function() use ($refClass, $reader, $container, $isController) {
             $services = [];
-            foreach ($refClass->getMethods() as $method) {
+            foreach ($refClass->getMethods(ReflectionMethod::IS_PUBLIC) as $method) {
                 $field = $reader->getRequestAnnotation($method, AbstractRequest::class);
                 if ($field !== null) {
                     if ($isController) {
                         $services[$refClass->getName()] = $refClass->getName();
                     }
                     $services += $this->getListOfInjectedServices($method, $container);
+                    if ($field instanceof Field && $field->getPrefetchMethod() !== null) {
+                        $services += $this->getListOfInjectedServices($refClass->getMethod($field->getPrefetchMethod()), $container);
+                    }
                 }
             }
             return $services;

Tests/Fixtures/Entities/Contact.php

@@ -47,4 +47,23 @@ public function injectService(TestGraphqlController $testService = null, stdClas
         }
         return 'OK';
     }
+
+    /**
+     * @Field(prefetchMethod="prefetchData")
+     */
+    public function injectServicePrefetch($prefetchData): string
+    {
+        return $prefetchData;
+    }
+
+    /**
+     * @Autowire(for="$someOtherService", identifier="someOtherService")
+     */
+    public function prefetchData(iterable $iterable, stdClass $someOtherService = null)
+    {
+        if ($someOtherService === null) {
+            return 'KO';
+        }
+        return 'OK';
+    }
 }

Tests/Fixtures/config/services.yaml

@@ -23,6 +23,9 @@ services:
     someAlias:
         alias: someService
 
+    someOtherService:
+        class: stdClass
+
     Symfony\Component\HttpFoundation\Session\Storage\Handler\NullSessionHandler:
         class: Symfony\Component\HttpFoundation\Session\Storage\Handler\NullSessionHandler
     # controllers are imported separately to make sure services can be injected

Tests/FunctionalTest.php

@@ -85,6 +85,7 @@ public function testServiceAutowiring(): void
         { 
           contact {
             injectService
+            injectServicePrefetch
           } 
         }']);
 
@@ -96,6 +97,7 @@ public function testServiceAutowiring(): void
             'data' => [
                 'contact' => [
                     'injectService' => 'OK',
+                    'injectServicePrefetch' => 'OK',
                 ]
             ]
         ], $result);