Явно установите Id с Doctrine при использовании стратегии "AUTO"

Моя сущность использует эту аннотацию для своего идентификатора:

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

Из чистой базы данных я импортирую существующие записи из более старой базы данных и пытаюсь сохранить те же идентификаторы. Затем, при добавлении новых записей, я хочу, чтобы MySQL автоматически увеличивал столбец идентификатора как обычно.

К сожалению, похоже, Doctrine2 полностью игнорирует указанный идентификатор.


Новое решение

В соответствии с рекомендациями ниже, следующее является предпочтительным решением:

$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());

Старое решение

Поскольку Doctrine отклоняется от ClassMetaData для определения стратегии генератора, его необходимо изменить после управления объектом в EntityManager:

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

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

$this->em->flush();

Я только что проверил это на MySQL, и оно работало, как и ожидалось, то есть объекты с пользовательским идентификатором были сохранены с этим идентификатором, в то время как те, у которых не был указан идентификатор, использовали lastGeneratedId() + 1,

7 ответов

Решение

Хотя ваше решение отлично работает с MySQL, мне не удалось заставить его работать с PostgreSQL, поскольку оно основано на последовательности.

Я должен добавить эту строку, чтобы она работала идеально:

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

С наилучшими пожеланиями,

Возможно, что доктрина изменилась, но теперь правильный путь:

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

В случае, если сущность является частью наследования таблицы классов, вам необходимо изменить генератор идентификатора в метаданных класса для обеих сущностей (сущность, которую вы сохраняете, и корневая сущность)

Новое решение прекрасно работает только тогда, когда все объекты имеют идентификатор перед вставкой. Когда одна сущность имеет идентификатор, а другая нет - новое решение терпит неудачу.

Я использую эту функцию для импорта всех моих данных:

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();
    }
}

Решение для Doctrine 2.5 и MySQL

"Новое решение" не работает с Doctrine 2.5 и MySQL. Вы должны использовать:

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

Однако я могу только подтвердить это для MySQL, потому что я еще не пробовал другие СУБД.

Я создал библиотеку для установки будущих идентификаторов для сущностей Doctrine. Он возвращается к исходной стратегии генерации идентификаторов, когда все идентификаторы в очереди используются для минимизации воздействия. Это должно быть легкое вставка для модульных тестов, чтобы подобный код не повторялся.

Вдохновленный работой Villermen, я создал эту библиотеку, которая позволяет вручную присваивать идентификаторы сущности Doctrine, даже когда сущность использует операторы состояний AUTO, SEQUENCE, IDENTITY или UUID.

Вы не должны использовать его в производстве.

Библиотека автоматически обнаружит объекты с назначенным идентификатором и заменит генератор только при необходимости. Библиотека откатится на исходный генератор, когда экземпляр не имеет идентификатора.

Замена генератора происходит в Doctrine EventListener, нет необходимости добавлять дополнительный код в ваши приборы.

Другие вопросы по тегам