Update batch reference resolvers example code snippet

nuwave · Jul 31, 2024 · 9350e60 · 9350e60
1 parent c091948
commit 9350e60
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/docs/6/federation/reference-resolvers.md b/docs/6/federation/reference-resolvers.md
@@ -101,11 +101,11 @@ final class Product implements BatchedEntityResolver
 {
     public function __invoke(array $representations): iterable
     {
-        $products = ProductRepository::byIDs(Arr::pluck($representations, 'id'));
+        $products = ProductRepository::byIDs(Arr::pluck($representations, 'id'))->keyBy('id');
 
         $result = [];
         foreach ($representations as $key => $representation) {
-            $result[$key] = $products->firstWhere('id', $representation['id']);
+            $result[$key] = $products->get($representation['id']);
         }
 
         return $result;

diff --git a/docs/master/federation/reference-resolvers.md b/docs/master/federation/reference-resolvers.md
@@ -101,11 +101,11 @@ final class Product implements BatchedEntityResolver
 {
     public function __invoke(array $representations): iterable
     {
-        $products = ProductRepository::byIDs(Arr::pluck($representations, 'id'));
+        $products = ProductRepository::byIDs(Arr::pluck($representations, 'id'))->keyBy('id');
 
         $result = [];
         foreach ($representations as $key => $representation) {
-            $result[$key] = $products->firstWhere('id', $representation['id']);
+            $result[$key] = $products->get($representation['id']);
         }
 
         return $result;