Allow CollectionDataProvider decoration to add query hints

src/Bridge/Doctrine/Orm/CollectionDataProvider.php

@@ -18,6 +18,8 @@
 use ApiPlatform\Core\Exception\ResourceClassNotSupportedException;
 use ApiPlatform\Core\Exception\RuntimeException;
 use Doctrine\Common\Persistence\ManagerRegistry;
+use Doctrine\ORM\Query;
+use Doctrine\ORM\QueryBuilder;
 
 /**
  * Collection data provider for the Doctrine ORM.
@@ -40,6 +42,20 @@ public function __construct(ManagerRegistry $managerRegistry, array $collectionE
         $this->collectionExtensions = $collectionExtensions;
     }
 
+    /**
+     * Allow to customize the Query object
+     * For example you can force doctrine to not load lazy relations by using:
+     * $query->setHint(Query::HINT_FORCE_PARTIAL_LOAD, true).
+     *
+     * @param QueryBuilder
+     *
+     * @return Query
+     */
+    public function getQuery(QueryBuilder $queryBuilder): Query
+    {
+        return $queryBuilder->getQuery();
+    }
+
     /**
      * {@inheritdoc}
      */
@@ -67,6 +83,6 @@ public function getCollection(string $resourceClass, string $operationName = nul
             }
         }
 
-        return $queryBuilder->getQuery()->getResult();
+        return $this->getQuery($queryBuilder)->getResult();
     }
 }

src/Bridge/Doctrine/Orm/ItemDataProvider.php

@@ -21,6 +21,7 @@
 use Doctrine\Common\Persistence\ManagerRegistry;
 use Doctrine\Common\Persistence\ObjectManager;
 use Doctrine\ORM\EntityManagerInterface;
+use Doctrine\ORM\Query;
 use Doctrine\ORM\QueryBuilder;
 
 /**
@@ -50,6 +51,20 @@ public function __construct(ManagerRegistry $managerRegistry, PropertyNameCollec
         $this->itemExtensions = $itemExtensions;
     }
 
+    /**
+     * Allow to customize the Query object
+     * For example you can force doctrine to not load lazy relations by using:
+     * $query->setHint(Query::HINT_FORCE_PARTIAL_LOAD, true).
+     *
+     * @param QueryBuilder
+     *
+     * @return Query
+     */
+    public function getQuery(QueryBuilder $queryBuilder): Query
+    {
+        return $queryBuilder->getQuery();
+    }
+
     /**
      * {@inheritdoc}
      */