Jawnie ustaw Id za pomocą Doctrine podczas korzystania ze strategii „AUTO”

Moja jednostka używa tej adnotacji jako swojego identyfikatora:

/**
 * @orm:Id
 * @orm:Column(type="integer")
 * @orm:GeneratedValue(strategy="AUTO")
 */
protected $id;

Z czystej bazy danych importuję istniejące rekordy ze starszej bazy danych i staram się zachować te same identyfikatory. Następnie podczas dodawania nowych rekordów chcę, aby MySQL jak zwykle automatycznie zwiększał wartość kolumny ID.

Niestety, wygląda na to, że Doctrine2 całkowicie ignoruje podany identyfikator.

Nowe rozwiązanie

Zgodnie z poniższymi zaleceniami preferowanym rozwiązaniem jest:

$this->em->persist($entity);

$metadata = $this->em->getClassMetaData(get_class($entity));
$metadata->setIdGeneratorType(\Doctrine\ORM\Mapping\ClassMetadata::GENERATOR_TYPE_NONE);
$metadata->setIdGenerator(new \Doctrine\ORM\Id\AssignedGenerator());

Stare rozwiązanie

Ponieważ Doctrine opiera się na ClassMetaData w celu określenia strategii generatora, musi zostać zmodyfikowana po zarządzaniu encją w EntityManager:

$this->em->persist($entity);

$metadata = $this->em->getClassMetaData(get_class($entity));
$metadata->setIdGeneratorType(\Doctrine\ORM\Mapping\ClassMetadata::GENERATOR_TYPE_NONE);

$this->em->flush();

Właśnie przetestowałem to na MySQL i działało zgodnie z oczekiwaniami, co oznacza, że ​​jednostki z niestandardowym identyfikatorem były przechowywane z tym identyfikatorem, podczas gdy te bez określonego identyfikatora używały rozszerzenia lastGeneratedId() + 1.

Chociaż twoje rozwiązanie działa dobrze z MySQL, nie udało mi się go uruchomić z PostgreSQL, ponieważ jest oparte na sekwencji.

Muszę dodać tę linię, aby działała idealnie:

$metadata->setIdGenerator(new \Doctrine\ORM\Id\AssignedGenerator());

Z poważaniem,

Być może zmieniła się doktryna, ale teraz jest właściwa:

$metadata->setIdGeneratorType(\Doctrine\ORM\Mapping\ClassMetadata::GENERATOR_TYPE_NONE);

W przypadku, gdy jednostka jest częścią dziedziczenia tabeli klas , musisz zmienić generator id w metadanych klasy dla obu jednostek (encja, którą utrwalasz, i jednostka główna)

Nowe rozwiązanie działa dobrze tylko wtedy, gdy WSZYSTKIE podmioty mają identyfikator przed wstawieniem. Gdy jeden podmiot ma identyfikator, a inny nie - nowe rozwiązanie zawodzi.

Używam tej funkcji do importu wszystkich moich danych:

function createEntity(\Doctrine\ORM\EntityManager $em, $entity, $id = null)
{
    $className = get_class($entity);
    if ($id) {
        $idRef = new \ReflectionProperty($className, "id");
        $idRef->setAccessible(true);
        $idRef->setValue($entity, $id);

        $metadata = $em->getClassMetadata($className);
        /** @var \Doctrine\ORM\Mapping\ClassMetadataInfo $metadata */
        $generator = $metadata->idGenerator;
        $generatorType = $metadata->generatorType;

        $metadata->setIdGenerator(new \Doctrine\ORM\Id\AssignedGenerator());
        $metadata->setIdGeneratorType(\Doctrine\ORM\Mapping\ClassMetadata::GENERATOR_TYPE_NONE);

        $unitOfWork = $em->getUnitOfWork();
        $persistersRef = new \ReflectionProperty($unitOfWork, "persisters");
        $persistersRef->setAccessible(true);
        $persisters = $persistersRef->getValue($unitOfWork);
        unset($persisters[$className]);
        $persistersRef->setValue($unitOfWork, $persisters);

        $em->persist($entity);
        $em->flush();

        $idRef->setAccessible(false);
        $metadata->setIdGenerator($generator);
        $metadata->setIdGeneratorType($generatorType);

        $persisters = $persistersRef->getValue($unitOfWork);
        unset($persisters[$className]);
        $persistersRef->setValue($unitOfWork, $persisters);
        $persistersRef->setAccessible(false);
    } else {
        $em->persist($entity);
        $em->flush();
    }
}

Rozwiązanie dla Doctrine 2.5 i MySQL

„Nowe rozwiązanie” nie działa z Doctrine 2.5 i MySQL. Musisz użyć:

$metadata = $this->getEntityManager()->getClassMetaData(Entity::class);
$metadata->setIdGenerator(new AssignedGenerator());
$metadata->setIdGeneratorType(ClassMetadata::GENERATOR_TYPE_‌​NONE);

Jednak mogę to tylko potwierdzić dla MySQL, ponieważ nie próbowałem jeszcze żadnego innego DBMS.

Utworzyłem bibliotekę do ustawiania przyszłych identyfikatorów dla encji Doctrine. Powraca do oryginalnej strategii generowania identyfikatorów, gdy wszystkie identyfikatory w kolejce są zużywane, aby zminimalizować wpływ. Powinien to być łatwy dodatek do testów jednostkowych, aby taki kod nie musiał być powtarzany.

Zainspirowany pracą Villermena , stworzyłem bibliotekę tseho / doctrine-assign-identity, która pozwala na ręczne przypisywanie identyfikatorów do encji Doctrine, nawet jeśli encja korzysta ze strategii AUTO, SEQUENCE, IDENTITY lub UUID.

Nigdy nie powinieneś używać go w produkcji, ale jest naprawdę przydatny do testów funkcjonalnych.

Biblioteka automatycznie wykryje jednostki z przypisanym identyfikatorem i zastąpi generator tylko wtedy, gdy będzie to potrzebne. Biblioteka powróci do początkowego generatora, gdy instancja nie ma przypisanego identyfikatora.

Zastąpienie generatora następuje w Doctrine EventListener, nie ma potrzeby dodawania dodatkowego kodu do urządzeń.