Явно установите 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, нет необходимости добавлять дополнительный код в ваши приборы.