AssignedGenerator.php 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. <?php
  2. /*
  3. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  4. * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  5. * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  6. * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  7. * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  8. * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  9. * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  10. * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  11. * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  12. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  13. * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  14. *
  15. * This software consists of voluntary contributions made by many individuals
  16. * and is licensed under the LGPL. For more information, see
  17. * <http://www.doctrine-project.org>.
  18. */
  19. namespace Doctrine\ORM\Id;
  20. use Doctrine\ORM\EntityManager;
  21. use Doctrine\ORM\Mapping\ClassMetadata;
  22. use Doctrine\ORM\ORMException;
  23. /**
  24. * Special generator for application-assigned identifiers (doesnt really generate anything).
  25. *
  26. * @since 2.0
  27. * @author Benjamin Eberlei <kontakt@beberlei.de>
  28. * @author Guilherme Blanco <guilhermeblanco@hotmail.com>
  29. * @author Jonathan Wage <jonwage@gmail.com>
  30. * @author Roman Borschel <roman@code-factory.org>
  31. */
  32. class AssignedGenerator extends AbstractIdGenerator
  33. {
  34. /**
  35. * Returns the identifier assigned to the given entity.
  36. *
  37. * @param object $entity
  38. * @return mixed
  39. * @override
  40. */
  41. public function generate(EntityManager $em, $entity)
  42. {
  43. $class = $em->getClassMetadata(get_class($entity));
  44. $idFields = $class->getIdentifierFieldNames();
  45. $identifier = array();
  46. foreach ($idFields as $idField) {
  47. $value = $class->reflFields[$idField]->getValue($entity);
  48. if ( ! isset($value)) {
  49. throw ORMException::entityMissingAssignedIdForField($entity, $idField);
  50. }
  51. if (isset($class->associationMappings[$idField])) {
  52. if ( ! $em->getUnitOfWork()->isInIdentityMap($value)) {
  53. throw ORMException::entityMissingForeignAssignedId($entity, $value);
  54. }
  55. // NOTE: Single Columns as associated identifiers only allowed - this constraint it is enforced.
  56. $value = current($em->getUnitOfWork()->getEntityIdentifier($value));
  57. }
  58. $identifier[$idField] = $value;
  59. }
  60. return $identifier;
  61. }
  62. }