vendor/api-platform/core/src/Bridge/Doctrine/Common/DataPersister.php line 50

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of the API Platform project.
  5.  *
  6.  * (c) Kévin Dunglas <dunglas@gmail.com>
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. declare(strict_types=1);
  14. namespace ApiPlatform\Core\Bridge\Doctrine\Common;
  16. use ApiPlatform\Core\DataPersister\ContextAwareDataPersisterInterface;
  17. use ApiPlatform\Core\Util\ClassInfoTrait;
  18. use Doctrine\ODM\MongoDB\Mapping\ClassMetadata;
  19. use Doctrine\ORM\Mapping\ClassMetadataInfo;
  20. use Doctrine\Persistence\ManagerRegistry;
  21. use Doctrine\Persistence\ObjectManager as DoctrineObjectManager;
  23. /**
  24.  * Data persister for Doctrine.
  25.  *
  26.  * @author Baptiste Meyer <baptiste.meyer@gmail.com>
  27.  */
  28. final class DataPersister implements ContextAwareDataPersisterInterface
  29. {
  30.     use ClassInfoTrait;
  32.     private $managerRegistry;
  34.     public function __construct(ManagerRegistry $managerRegistry)
  35.     {
  36.         $this->managerRegistry = $managerRegistry;
  37.     }
  39.     /**
  40.      * {@inheritdoc}
  41.      */
  42.     public function supports($data, array $context = []): bool
  43.     {
  44.         return null !== $this->getManager($data);
  45.     }
  47.     /**
  48.      * {@inheritdoc}
  49.      */
  50.     public function persist($data, array $context = [])
  51.     {
  52.         if (!$manager = $this->getManager($data)) {
  53.             return $data;
  54.         }
  56.         if (!$manager->contains($data) || $this->isDeferredExplicit($manager, $data)) {
  57.             $manager->persist($data);
  58.         }
  60.         $manager->flush();
  61.         $manager->refresh($data);
  63.         return $data;
  64.     }
  66.     /**
  67.      * {@inheritdoc}
  68.      */
  69.     public function remove($data, array $context = [])
  70.     {
  71.         if (!$manager = $this->getManager($data)) {
  72.             return;
  73.         }
  75.         $manager->remove($data);
  76.         $manager->flush();
  77.     }
  79.     /**
  80.      * Gets the Doctrine object manager associated with given data.
  81.      */
  82.     private function getManager($data): ?DoctrineObjectManager
  83.     {
  84.         return \is_object($data) ? $this->managerRegistry->getManagerForClass($this->getObjectClass($data)) : null;
  85.     }
  87.     /**
  88.      * Checks if doctrine does not manage data automatically.
  89.      */
  90.     private function isDeferredExplicit(DoctrineObjectManager $manager, $data): bool
  91.     {
  92.         $classMetadata = $manager->getClassMetadata($this->getObjectClass($data));
  93.         if (($classMetadata instanceof ClassMetadataInfo || $classMetadata instanceof ClassMetadata) && method_exists($classMetadata, 'isChangeTrackingDeferredExplicit')) {
  94.             return $classMetadata->isChangeTrackingDeferredExplicit();
  95.         }
  97.         return false;
  98.     }
  99. }