123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254 |
- <?php
- namespace Doctrine\ORM\Persisters;
- use Doctrine\ORM\PersistentCollection;
- use Doctrine\ORM\UnitOfWork;
- class OneToManyPersister extends AbstractCollectionPersister
- {
-
- public function get(PersistentCollection $coll, $index)
- {
- $mapping = $coll->getMapping();
- $uow = $this->em->getUnitOfWork();
- $persister = $uow->getEntityPersister($mapping['targetEntity']);
- if (!isset($mapping['indexBy'])) {
- throw new \BadMethodCallException("Selecting a collection by index is only supported on indexed collections.");
- }
- return $persister->load(array($mapping['mappedBy'] => $coll->getOwner(), $mapping['indexBy'] => $index), null, null, array(), 0, 1);
- }
-
- protected function getDeleteRowSQL(PersistentCollection $coll)
- {
- $mapping = $coll->getMapping();
- $class = $this->em->getClassMetadata($mapping['targetEntity']);
- $tableName = $this->quoteStrategy->getTableName($class, $this->platform);
- $idColumns = $class->getIdentifierColumnNames();
- return 'DELETE FROM ' . $tableName
- . ' WHERE ' . implode('= ? AND ', $idColumns) . ' = ?';
- }
-
- protected function getDeleteRowSQLParameters(PersistentCollection $coll, $element)
- {
- return array_values($this->uow->getEntityIdentifier($element));
- }
-
- protected function getInsertRowSQL(PersistentCollection $coll)
- {
- throw new \BadMethodCallException("Insert Row SQL is not used for OneToManyPersister");
- }
-
- protected function getInsertRowSQLParameters(PersistentCollection $coll, $element)
- {
- throw new \BadMethodCallException("Insert Row SQL is not used for OneToManyPersister");
- }
-
- protected function getUpdateRowSQL(PersistentCollection $coll)
- {
- throw new \BadMethodCallException("Update Row SQL is not used for OneToManyPersister");
- }
-
- protected function getDeleteSQL(PersistentCollection $coll)
- {
- throw new \BadMethodCallException("Update Row SQL is not used for OneToManyPersister");
- }
-
- protected function getDeleteSQLParameters(PersistentCollection $coll)
- {
- throw new \BadMethodCallException("Update Row SQL is not used for OneToManyPersister");
- }
-
- public function count(PersistentCollection $coll)
- {
- $mapping = $coll->getMapping();
- $targetClass = $this->em->getClassMetadata($mapping['targetEntity']);
- $sourceClass = $this->em->getClassMetadata($mapping['sourceEntity']);
- $id = $this->em->getUnitOfWork()->getEntityIdentifier($coll->getOwner());
- $whereClauses = array();
- $params = array();
- $joinColumns = $targetClass->associationMappings[$mapping['mappedBy']]['joinColumns'];
- foreach ($joinColumns as $joinColumn) {
- $whereClauses[] = $joinColumn['name'] . ' = ?';
- $params[] = ($targetClass->containsForeignIdentifier)
- ? $id[$sourceClass->getFieldForColumn($joinColumn['referencedColumnName'])]
- : $id[$sourceClass->fieldNames[$joinColumn['referencedColumnName']]];
- }
- $filterTargetClass = $this->em->getClassMetadata($targetClass->rootEntityName);
- foreach ($this->em->getFilters()->getEnabledFilters() as $filter) {
- if ($filterExpr = $filter->addFilterConstraint($filterTargetClass, 't')) {
- $whereClauses[] = '(' . $filterExpr . ')';
- }
- }
- $sql = 'SELECT count(*)'
- . ' FROM ' . $this->quoteStrategy->getTableName($targetClass, $this->platform) . ' t'
- . ' WHERE ' . implode(' AND ', $whereClauses);
- return $this->conn->fetchColumn($sql, $params);
- }
-
- public function slice(PersistentCollection $coll, $offset, $length = null)
- {
- $mapping = $coll->getMapping();
- $uow = $this->em->getUnitOfWork();
- $persister = $uow->getEntityPersister($mapping['targetEntity']);
- return $persister->getOneToManyCollection($mapping, $coll->getOwner(), $offset, $length);
- }
-
- public function contains(PersistentCollection $coll, $element)
- {
- $mapping = $coll->getMapping();
- $uow = $this->em->getUnitOfWork();
-
- $entityState = $uow->getEntityState($element, UnitOfWork::STATE_NEW);
- if ($entityState === UnitOfWork::STATE_NEW) {
- return false;
- }
-
- if ($entityState === UnitOfWork::STATE_MANAGED && $uow->isScheduledForInsert($element)) {
- return false;
- }
- $persister = $uow->getEntityPersister($mapping['targetEntity']);
-
-
-
- $id = current( $uow->getEntityIdentifier($coll->getOwner()));
- return $persister->exists($element, array($mapping['mappedBy'] => $id));
- }
-
- public function removeElement(PersistentCollection $coll, $element)
- {
- $mapping = $coll->getMapping();
- if ( ! $mapping['orphanRemoval']) {
-
- return false;
- }
- $uow = $this->em->getUnitOfWork();
-
- $entityState = $uow->getEntityState($element, UnitOfWork::STATE_NEW);
- if ($entityState === UnitOfWork::STATE_NEW) {
- return false;
- }
-
-
- if ($entityState === UnitOfWork::STATE_MANAGED && $uow->isScheduledForInsert($element)) {
- return false;
- }
- $this
- ->uow
- ->getEntityPersister($mapping['targetEntity'])
- ->delete($element);
- return true;
- }
- }
|