예제 #1
0
 /**
  * Sets the version field mapping used for versioning. Sets the default
  * value to use depending on the column type.
  *
  * @param array $mapping   The version field mapping array
  *
  * @throws \CosmoW\ODM\Riak\LockException
  */
 public function setLockMapping(array &$mapping)
 {
     if ($mapping['type'] !== 'int') {
         throw LockException::invalidLockFieldType($mapping['type']);
     }
     $this->isLockable = true;
     $this->lockField = $mapping['fieldName'];
 }
예제 #2
0
 /**
  * Finds a document by a set of criteria.
  *
  * If a scalar or MongoId is provided for $criteria, it will be used to
  * match an _id value.
  *
  * @param mixed   $criteria Query criteria
  * @param object  $document Document to load the data into. If not specified, a new document is created.
  * @param array   $hints    Hints for document creation
  * @param integer $lockMode
  * @param array   $sort     Sort array for Cursor::sort()
  * @throws \CosmoW\ODM\Riak\LockException
  * @return object|null The loaded and managed document instance or null if no document was found
  * @todo Check identity map? loadById method? Try to guess whether $criteria is the id?
  */
 public function load($criteria, $document = null, array $hints = array(), $lockMode = 0, array $sort = null)
 {
     // TODO: remove this
     if ($criteria === null || is_scalar($criteria) || $criteria instanceof \MongoId) {
         $criteria = array('_id' => $criteria);
     }
     $criteria = $this->prepareQueryOrNewObj($criteria);
     $criteria = $this->addDiscriminatorToPreparedQuery($criteria);
     $criteria = $this->addFilterToPreparedQuery($criteria);
     $cursor = $this->collection->find($criteria);
     if (null !== $sort) {
         $cursor->sort($this->prepareSortOrProjection($sort));
     }
     $result = $cursor->getSingleResult();
     if ($this->class->isLockable) {
         $lockMapping = $this->class->fieldMappings[$this->class->lockField];
         if (isset($result[$lockMapping['name']]) && $result[$lockMapping['name']] === LockMode::PESSIMISTIC_WRITE) {
             throw LockException::lockFailed($result);
         }
     }
     return $this->createDocument($result, $document, $hints);
 }
예제 #3
0
 /**
  * Finds a document by its identifier
  *
  * @param string|object $id The identifier
  * @param int $lockMode
  * @param int $lockVersion
  * @throws Mapping\MappingException
  * @throws LockException
  * @return object The document.
  */
 public function find($id, $lockMode = LockMode::NONE, $lockVersion = null)
 {
     if ($id === null) {
         return;
     }
     /* TODO: What if the ID object has a field with the same name as the
      * class' mapped identifier field name?
      */
     if (is_array($id)) {
         list($identifierFieldName) = $this->class->getIdentifierFieldNames();
         if (isset($id[$identifierFieldName])) {
             $id = $id[$identifierFieldName];
         }
     }
     // Check identity map first
     if ($document = $this->uow->tryGetById($id, $this->class)) {
         if ($lockMode != LockMode::NONE) {
             $this->dm->lock($document, $lockMode, $lockVersion);
         }
         return $document;
         // Hit!
     }
     $criteria = array('_id' => $id);
     if ($lockMode == LockMode::NONE) {
         return $this->getDocumentPersister()->load($criteria);
     }
     if ($lockMode == LockMode::OPTIMISTIC) {
         if (!$this->class->isVersioned) {
             throw LockException::notVersioned($this->documentName);
         }
         if ($document = $this->getDocumentPersister()->load($criteria)) {
             $this->uow->lock($document, $lockMode, $lockVersion);
         }
         return $document;
     }
     return $this->getDocumentPersister()->load($criteria, null, array(), $lockMode);
 }
예제 #4
0
 /**
  * Executes a query updating the given document.
  *
  * @param object $document
  * @param array $newObj
  * @param array $options
  */
 private function executeQuery($document, array $newObj, array $options)
 {
     $className = get_class($document);
     $class = $this->dm->getClassMetadata($className);
     $id = $class->getDatabaseIdentifierValue($this->uow->getDocumentIdentifier($document));
     $query = array('_id' => $id);
     if ($class->isVersioned) {
         $query[$class->versionField] = $class->reflFields[$class->versionField]->getValue($document);
     }
     $collection = $this->dm->getDocumentCollection($className);
     $result = $collection->update($query, $newObj, $options);
     if ($class->isVersioned && !$result['n']) {
         throw LockException::lockFailed($document);
     }
 }
예제 #5
0
 /**
  * Acquire a lock on the given document.
  *
  * @param object $document
  * @param int $lockMode
  * @param int $lockVersion
  * @throws LockException
  * @throws \InvalidArgumentException
  */
 public function lock($document, $lockMode, $lockVersion = null)
 {
     if ($this->getDocumentState($document) != self::STATE_MANAGED) {
         throw new \InvalidArgumentException("Document is not MANAGED.");
     }
     $documentName = get_class($document);
     $class = $this->dm->getClassMetadata($documentName);
     if ($lockMode == LockMode::OPTIMISTIC) {
         if (!$class->isVersioned) {
             throw LockException::notVersioned($documentName);
         }
         if ($lockVersion != null) {
             $documentVersion = $class->reflFields[$class->versionField]->getValue($document);
             if ($documentVersion != $lockVersion) {
                 throw LockException::lockFailedVersionMissmatch($document, $lockVersion, $documentVersion);
             }
         }
     } elseif (in_array($lockMode, array(LockMode::PESSIMISTIC_READ, LockMode::PESSIMISTIC_WRITE))) {
         $this->getDocumentPersister($class->name)->lock($document, $lockMode);
     }
 }