ORM InsertIgnore

InsertIgnore.php

<?php

declare(strict_types=1);

namespace App\Shop\Extension\Doctrine\Execution;

use Doctrine\ORM\Event\LifecycleEventArgs;
use Doctrine\ORM\Persisters\Entity\EntityPersister;

class InsertIgnore
{
    private static $hackMe = [YoEntity::class];

    /**
     * @var \ReflectionProperty[]
     */
    private $reflections = [];

    public function prePersist(LifecycleEventArgs $event): void
    {
        if (!$this->needsHacking($event->getEntity())) {
            return;
        }

        $em = $event->getEntityManager();
        $metadata = $em->getClassMetadata(get_class($event->getEntity()));
        $persister = $event->getEntityManager()->getUnitOfWork()->getEntityPersister($metadata->name);
        $sql = $persister->getInsertSQL();
        if (0 === strpos($sql, 'INSERT IGNORE')) {
            return;
        }

        $sql = str_replace('INSERT', 'INSERT IGNORE', $sql);
        $this->getReflection($persister)->setValue($persister, $sql);
    }

    private function getReflection(EntityPersister $entityPersister): \ReflectionProperty
    {
        $class = \get_class($entityPersister);
        if (isset($this->reflections[$class])) {
            return $this->reflections[$class];
        }

        $reflection = (new \ReflectionClass($class))->getProperty('insertSql');
        $reflection->setAccessible(true);

        return $this->reflections[$class] = $reflection;
    }

    private function needsHacking(object $entity): bool
    {
        foreach (self::$hackMe as $fqcn) {
            if (is_a($entity, $fqcn)) {
                return true;
            }
        }

        return false;
    }
}