Completed
Pull Request — master (#1263)
by Andreas
14:36
created

CollectionPersister::executeQuery()   A

Complexity

Conditions 4
Paths 4

Size

Total Lines 15
Code Lines 11

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 0
CRAP Score 20

Importance

Changes 0
Metric Value
c 0
b 0
f 0
dl 0
loc 15
ccs 0
cts 13
cp 0
rs 9.2
cc 4
eloc 11
nc 4
nop 3
crap 20
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 MIT license. For more information, see
17
 * <http://www.doctrine-project.org>.
18
 */
19
20
namespace Doctrine\ODM\MongoDB\Persisters;
21
22
use Doctrine\ODM\MongoDB\DocumentManager;
23
use Doctrine\ODM\MongoDB\LockException;
24
use Doctrine\ODM\MongoDB\Mapping\ClassMetadataInfo;
25
use Doctrine\ODM\MongoDB\PersistentCollection\PersistentCollectionInterface;
26
use Doctrine\ODM\MongoDB\UnitOfWork;
27
use Doctrine\ODM\MongoDB\Utility\CollectionHelper;
28
29
/**
30
 * The CollectionPersister is responsible for persisting collections of embedded
31
 * or referenced documents. When a PersistentCollection is scheduledForDeletion
32
 * in the UnitOfWork by calling PersistentCollection::clear() or is
33
 * de-referenced in the domain application code, CollectionPersister::delete()
34
 * will be called. When documents within the PersistentCollection are added or
35
 * removed, CollectionPersister::update() will be called, which may set the
36
 * entire collection or delete/insert individual elements, depending on the
37
 * mapping strategy.
38
 *
39
 * @since       1.0
40
 */
41
class CollectionPersister
42
{
43
    /**
44
     * The DocumentManager instance.
45
     *
46
     * @var DocumentManager
47
     */
48
    private $dm;
49
50
    /**
51
     * The PersistenceBuilder instance.
52
     *
53
     * @var PersistenceBuilder
54
     */
55
    private $pb;
56
57
    /**
58
     * Constructs a new CollectionPersister instance.
59
     *
60
     * @param DocumentManager $dm
61
     * @param PersistenceBuilder $pb
62
     * @param UnitOfWork $uow
63
     */
64 4
    public function __construct(DocumentManager $dm, PersistenceBuilder $pb, UnitOfWork $uow)
65
    {
66 4
        $this->dm = $dm;
67 4
        $this->pb = $pb;
68 4
        $this->uow = $uow;
0 ignored issues
show
Bug introduced by
The property uow does not exist. Did you maybe forget to declare it?

In PHP it is possible to write to properties without declaring them. For example, the following is perfectly valid PHP code:

class MyClass { }

$x = new MyClass();
$x->foo = true;

Generally, it is a good practice to explictly declare properties to avoid accidental typos and provide IDE auto-completion:

class MyClass {
    public $foo;
}

$x = new MyClass();
$x->foo = true;
Loading history...
69 4
    }
70
71
    /**
72
     * Deletes a PersistentCollection instance completely from a document using $unset.
73
     *
74
     * @param PersistentCollectionInterface $coll
75
     * @param array $options
76
     */
77
    public function delete(PersistentCollectionInterface $coll, array $options)
78
    {
79
        $mapping = $coll->getMapping();
80
        if ($mapping['isInverseSide']) {
81
            return; // ignore inverse side
82
        }
83
        if (CollectionHelper::isAtomic($mapping['strategy'])) {
84
            throw new \UnexpectedValueException($mapping['strategy'] . ' delete collection strategy should have been handled by DocumentPersister. Please report a bug in issue tracker');
85
        }
86
        list($propertyPath, $parent) = $this->getPathAndParent($coll);
87
        $query = array('$unset' => array($propertyPath => true));
88
        $this->executeQuery($parent, $query, $options);
89
    }
90
91
    /**
92
     * Updates a PersistentCollection instance deleting removed rows and
93
     * inserting new rows.
94
     *
95
     * @param PersistentCollectionInterface $coll
96
     * @param array $options
97
     */
98
    public function update(PersistentCollectionInterface $coll, array $options)
99
    {
100
        $mapping = $coll->getMapping();
101
102
        if ($mapping['isInverseSide']) {
103
            return; // ignore inverse side
104
        }
105
106
        switch ($mapping['strategy']) {
107
            case ClassMetadataInfo::STORAGE_STRATEGY_ATOMIC_SET:
108
            case ClassMetadataInfo::STORAGE_STRATEGY_ATOMIC_SET_ARRAY:
109
                throw new \UnexpectedValueException($mapping['strategy'] . ' update collection strategy should have been handled by DocumentPersister. Please report a bug in issue tracker');
110
111
            case ClassMetadataInfo::STORAGE_STRATEGY_SET:
112
            case ClassMetadataInfo::STORAGE_STRATEGY_SET_ARRAY:
113
                $this->setCollection($coll, $options);
114
                break;
115
116
            case ClassMetadataInfo::STORAGE_STRATEGY_ADD_TO_SET:
117
            case ClassMetadataInfo::STORAGE_STRATEGY_PUSH_ALL:
118
                $coll->initialize();
119
                $this->deleteElements($coll, $options);
120
                $this->insertElements($coll, $options);
121
                break;
122
123
            default:
124
                throw new \UnexpectedValueException('Unsupported collection strategy: ' . $mapping['strategy']);
125
        }
126
    }
127
128
    /**
129
     * Sets a PersistentCollection instance.
130
     *
131
     * This method is intended to be used with the "set" or "setArray"
132
     * strategies. The "setArray" strategy will ensure that the collection is
133
     * set as a BSON array, which means the collection elements will be
134
     * reindexed numerically before storage.
135
     *
136
     * @param PersistentCollectionInterface $coll
137
     * @param array $options
138
     */
139
    private function setCollection(PersistentCollectionInterface $coll, array $options)
140
    {
141
        list($propertyPath, $parent) = $this->getPathAndParent($coll);
142
        $coll->initialize();
143
        $mapping = $coll->getMapping();
144
        $setData = $this->pb->prepareAssociatedCollectionValue($coll, CollectionHelper::usesSet($mapping['strategy']));
145
        $query = array('$set' => array($propertyPath => $setData));
146
        $this->executeQuery($parent, $query, $options);
147
    }
148
149
    /**
150
     * Deletes removed elements from a PersistentCollection instance.
151
     *
152
     * This method is intended to be used with the "pushAll" and "addToSet"
153
     * strategies.
154
     *
155
     * @param PersistentCollectionInterface $coll
156
     * @param array $options
157
     */
158
    private function deleteElements(PersistentCollectionInterface $coll, array $options)
159
    {
160
        $deleteDiff = $coll->getDeleteDiff();
161
162
        if (empty($deleteDiff)) {
163
            return;
164
        }
165
166
        list($propertyPath, $parent) = $this->getPathAndParent($coll);
167
168
        $query = array('$unset' => array());
169
170
        foreach ($deleteDiff as $key => $document) {
171
            $query['$unset'][$propertyPath . '.' . $key] = true;
172
        }
173
174
        $this->executeQuery($parent, $query, $options);
175
176
        /**
177
         * @todo This is a hack right now because we don't have a proper way to
178
         * remove an element from an array by its key. Unsetting the key results
179
         * in the element being left in the array as null so we have to pull
180
         * null values.
181
         */
182
        $this->executeQuery($parent, array('$pull' => array($propertyPath => null)), $options);
183
    }
184
185
    /**
186
     * Inserts new elements for a PersistentCollection instance.
187
     *
188
     * This method is intended to be used with the "pushAll" and "addToSet"
189
     * strategies.
190
     *
191
     * @param PersistentCollectionInterface $coll
192
     * @param array $options
193
     */
194
    private function insertElements(PersistentCollectionInterface $coll, array $options)
195
    {
196
        $insertDiff = $coll->getInsertDiff();
197
198
        if (empty($insertDiff)) {
199
            return;
200
        }
201
202
        $mapping = $coll->getMapping();
203
        list($propertyPath, $parent) = $this->getPathAndParent($coll);
204
205
        $pb = $this->pb;
206
207
        $callback = isset($mapping['embedded'])
208
            ? function($v) use ($pb, $mapping) { return $pb->prepareEmbeddedDocumentValue($mapping, $v); }
209
            : function($v) use ($pb, $mapping) { return $pb->prepareReferencedDocumentValue($mapping, $v); };
210
211
        $value = array_values(array_map($callback, $insertDiff));
212
213
        if ($mapping['strategy'] === ClassMetadataInfo::STORAGE_STRATEGY_ADD_TO_SET) {
214
            $value = array('$each' => $value);
215
        }
216
217
        $query = array('$' . $mapping['strategy'] => array($propertyPath => $value));
218
219
        $this->executeQuery($parent, $query, $options);
220
    }
221
222
    /**
223
     * Gets the parent information for a given PersistentCollection. It will
224
     * retrieve the top-level persistent Document that the PersistentCollection
225
     * lives in. We can use this to issue queries when updating a
226
     * PersistentCollection that is multiple levels deep inside an embedded
227
     * document.
228
     *
229
     *     <code>
230
     *     list($path, $parent) = $this->getPathAndParent($coll)
231
     *     </code>
232
     *
233
     * @param PersistentCollectionInterface $coll
234
     * @return array $pathAndParent
235
     */
236
    private function getPathAndParent(PersistentCollectionInterface $coll)
237
    {
238
        $mapping = $coll->getMapping();
239
        $fields = array();
240
        $parent = $coll->getOwner();
241
        while (null !== ($association = $this->uow->getParentAssociation($parent))) {
242
            list($m, $owner, $field) = $association;
243
            if (isset($m['reference'])) {
244
                break;
245
            }
246
            $parent = $owner;
247
            $fields[] = $field;
248
        }
249
        $propertyPath = implode('.', array_reverse($fields));
250
        $path = $mapping['name'];
251
        if ($propertyPath) {
252
            $path = $propertyPath . '.' . $path;
253
        }
254
        return array($path, $parent);
255
    }
256
257
    /**
258
     * Executes a query updating the given document.
259
     *
260
     * @param object $document
261
     * @param array $newObj
262
     * @param array $options
263
     */
264
    private function executeQuery($document, array $newObj, array $options)
265
    {
266
        $className = get_class($document);
267
        $class = $this->dm->getClassMetadata($className);
268
        $id = $class->getDatabaseIdentifierValue($this->uow->getDocumentIdentifier($document));
269
        $query = array('_id' => $id);
270
        if ($class->isVersioned) {
271
            $query[$class->versionField] = $class->reflFields[$class->versionField]->getValue($document);
272
        }
273
        $collection = $this->dm->getDocumentCollection($className);
274
        $result = $collection->update($query, $newObj, $options);
275
        if ($class->isVersioned && ! $result['n']) {
276
            throw LockException::lockFailed($document);
277
        }
278
    }
279
}
280