doctrine /
mongodb-odm
Duplicate code is one of the most pungent code smells. A rule that is often used is to re-structure code once it is duplicated in three or more places.
Common duplication problems, and corresponding solutions are:
Complex classes like YamlDriver often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes. You can also have a look at the cohesion graph to spot any un-connected, or weakly-connected components.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
While breaking up the class, it is a good idea to analyze how other classes use YamlDriver, and based on these observations, apply Extract Interface, too.
| 1 | <?php |
||
| 34 | class YamlDriver extends FileDriver |
||
| 35 | { |
||
| 36 | const DEFAULT_FILE_EXTENSION = '.dcm.yml'; |
||
| 37 | |||
| 38 | /** |
||
| 39 | * {@inheritDoc} |
||
| 40 | */ |
||
| 41 | 9 | public function __construct($locator, $fileExtension = self::DEFAULT_FILE_EXTENSION) |
|
| 45 | |||
| 46 | /** |
||
| 47 | * {@inheritDoc} |
||
| 48 | */ |
||
| 49 | 5 | public function loadMetadataForClass($className, ClassMetadata $class) |
|
| 50 | { |
||
| 51 | /* @var $class ClassMetadataInfo */ |
||
| 52 | 5 | $element = $this->getElement($className); |
|
| 53 | 5 | if ( ! $element) { |
|
|
|
|||
| 54 | return; |
||
| 55 | } |
||
| 56 | 5 | $element['type'] = isset($element['type']) ? $element['type'] : 'document'; |
|
| 57 | |||
| 58 | 5 | if (isset($element['db'])) { |
|
| 59 | 2 | $class->setDatabase($element['db']); |
|
| 60 | 2 | } |
|
| 61 | 5 | if (isset($element['collection'])) { |
|
| 62 | 5 | $class->setCollection($element['collection']); |
|
| 63 | 5 | } |
|
| 64 | 5 | if (isset($element['writeConcern'])) { |
|
| 65 | 1 | $class->setWriteConcern($element['writeConcern']); |
|
| 66 | 1 | } |
|
| 67 | 5 | if ($element['type'] == 'document') { |
|
| 68 | 5 | if (isset($element['repositoryClass'])) { |
|
| 69 | $class->setCustomRepositoryClass($element['repositoryClass']); |
||
| 70 | } |
||
| 71 | 5 | View Code Duplication | } elseif ($element['type'] === 'mappedSuperclass') { |
| 72 | $class->setCustomRepositoryClass( |
||
| 73 | isset($element['repositoryClass']) ? $element['repositoryClass'] : null |
||
| 74 | ); |
||
| 75 | $class->isMappedSuperclass = true; |
||
| 76 | 1 | } elseif ($element['type'] === 'embeddedDocument') { |
|
| 77 | 1 | $class->isEmbeddedDocument = true; |
|
| 78 | 1 | } |
|
| 79 | 5 | if (isset($element['indexes'])) { |
|
| 80 | 2 | foreach($element['indexes'] as $index) { |
|
| 81 | 2 | $class->addIndex($index['keys'], isset($index['options']) ? $index['options'] : array()); |
|
| 82 | 2 | } |
|
| 83 | 2 | } |
|
| 84 | 5 | if (isset($element['shardKey'])) { |
|
| 85 | 1 | $this->setShardKey($class, $element['shardKey']); |
|
| 86 | 1 | } |
|
| 87 | 5 | View Code Duplication | if (isset($element['inheritanceType'])) { |
| 88 | $class->setInheritanceType(constant(MappingClassMetadata::class . '::INHERITANCE_TYPE_' . strtoupper($element['inheritanceType']))); |
||
| 89 | } |
||
| 90 | 5 | if (isset($element['discriminatorField'])) { |
|
| 91 | 1 | $class->setDiscriminatorField($this->parseDiscriminatorField($element['discriminatorField'])); |
|
| 92 | 1 | } |
|
| 93 | 5 | if (isset($element['discriminatorMap'])) { |
|
| 94 | 1 | $class->setDiscriminatorMap($element['discriminatorMap']); |
|
| 95 | 1 | } |
|
| 96 | 5 | if (isset($element['defaultDiscriminatorValue'])) { |
|
| 97 | 1 | $class->setDefaultDiscriminatorValue($element['defaultDiscriminatorValue']); |
|
| 98 | 1 | } |
|
| 99 | 5 | View Code Duplication | if (isset($element['changeTrackingPolicy'])) { |
| 100 | $class->setChangeTrackingPolicy(constant(MappingClassMetadata::class . '::CHANGETRACKING_' . strtoupper($element['changeTrackingPolicy']))); |
||
| 101 | } |
||
| 102 | 5 | if (isset($element['requireIndexes'])) { |
|
| 103 | $class->setRequireIndexes($element['requireIndexes']); |
||
| 104 | } |
||
| 105 | 5 | if (isset($element['slaveOkay'])) { |
|
| 106 | $class->setSlaveOkay($element['slaveOkay']); |
||
| 107 | } |
||
| 108 | 5 | if (isset($element['fields'])) { |
|
| 109 | 5 | foreach ($element['fields'] as $fieldName => $mapping) { |
|
| 110 | 5 | if (is_string($mapping)) { |
|
| 111 | 1 | $type = $mapping; |
|
| 112 | 1 | $mapping = array(); |
|
| 113 | 1 | $mapping['type'] = $type; |
|
| 114 | 1 | } |
|
| 115 | 5 | if ( ! isset($mapping['fieldName'])) { |
|
| 116 | 4 | $mapping['fieldName'] = $fieldName; |
|
| 117 | 4 | } |
|
| 118 | 5 | if (isset($mapping['type']) && ! empty($mapping['embedded'])) { |
|
| 119 | 2 | $this->addMappingFromEmbed($class, $fieldName, $mapping, $mapping['type']); |
|
| 120 | 5 | } elseif (isset($mapping['type']) && ! empty($mapping['reference'])) { |
|
| 121 | 2 | $this->addMappingFromReference($class, $fieldName, $mapping, $mapping['type']); |
|
| 122 | 2 | } else { |
|
| 123 | 5 | $this->addFieldMapping($class, $mapping); |
|
| 124 | } |
||
| 125 | 5 | } |
|
| 126 | 5 | } |
|
| 127 | 5 | View Code Duplication | if (isset($element['embedOne'])) { |
| 128 | 2 | foreach ($element['embedOne'] as $fieldName => $embed) { |
|
| 129 | 2 | $this->addMappingFromEmbed($class, $fieldName, $embed, 'one'); |
|
| 130 | 2 | } |
|
| 131 | 2 | } |
|
| 132 | 5 | View Code Duplication | if (isset($element['embedMany'])) { |
| 133 | 2 | foreach ($element['embedMany'] as $fieldName => $embed) { |
|
| 134 | 2 | $this->addMappingFromEmbed($class, $fieldName, $embed, 'many'); |
|
| 135 | 2 | } |
|
| 136 | 2 | } |
|
| 137 | 5 | View Code Duplication | if (isset($element['referenceOne'])) { |
| 138 | 2 | foreach ($element['referenceOne'] as $fieldName => $reference) { |
|
| 139 | 2 | $this->addMappingFromReference($class, $fieldName, $reference, 'one'); |
|
| 140 | 2 | } |
|
| 141 | 2 | } |
|
| 142 | 5 | View Code Duplication | if (isset($element['referenceMany'])) { |
| 143 | 2 | foreach ($element['referenceMany'] as $fieldName => $reference) { |
|
| 144 | 2 | $this->addMappingFromReference($class, $fieldName, $reference, 'many'); |
|
| 145 | 2 | } |
|
| 146 | 2 | } |
|
| 147 | 5 | if (isset($element['lifecycleCallbacks'])) { |
|
| 148 | 2 | foreach ($element['lifecycleCallbacks'] as $type => $methods) { |
|
| 149 | 2 | foreach ($methods as $method) { |
|
| 150 | 2 | $class->addLifecycleCallback($method, constant('Doctrine\ODM\MongoDB\Events::' . $type)); |
|
| 151 | 2 | } |
|
| 152 | 2 | } |
|
| 153 | 2 | } |
|
| 154 | 5 | if (isset($element['alsoLoadMethods'])) { |
|
| 155 | 1 | foreach ($element['alsoLoadMethods'] as $methodName => $fieldName) { |
|
| 156 | 1 | $class->registerAlsoLoadMethod($methodName, $fieldName); |
|
| 157 | 1 | } |
|
| 158 | 1 | } |
|
| 159 | 5 | } |
|
| 160 | |||
| 161 | 5 | private function addFieldMapping(ClassMetadataInfo $class, $mapping) |
|
| 162 | { |
||
| 163 | 5 | View Code Duplication | if (isset($mapping['name'])) { |
| 164 | 1 | $name = $mapping['name']; |
|
| 165 | 5 | } elseif (isset($mapping['fieldName'])) { |
|
| 166 | 5 | $name = $mapping['fieldName']; |
|
| 167 | 5 | } else { |
|
| 168 | throw new \InvalidArgumentException('Cannot infer a MongoDB name from the mapping'); |
||
| 169 | } |
||
| 170 | |||
| 171 | 5 | $class->mapField($mapping); |
|
| 172 | |||
| 173 | 5 | View Code Duplication | if ( ! (isset($mapping['index']) || isset($mapping['unique']) || isset($mapping['sparse']))) { |
| 174 | 5 | return; |
|
| 175 | } |
||
| 176 | |||
| 177 | // Multiple index specifications in one field mapping is ambiguous |
||
| 178 | 4 | if ((isset($mapping['index']) && is_array($mapping['index'])) + |
|
| 179 | 4 | (isset($mapping['unique']) && is_array($mapping['unique'])) + |
|
| 180 | 4 | (isset($mapping['sparse']) && is_array($mapping['sparse'])) > 1) { |
|
| 181 | throw new \InvalidArgumentException('Multiple index specifications found among index, unique, and/or sparse fields'); |
||
| 182 | } |
||
| 183 | |||
| 184 | // Index this field if either "index", "unique", or "sparse" are set |
||
| 185 | 4 | $keys = array($name => 'asc'); |
|
| 186 | |||
| 187 | /* The "order" option is only used in the index specification and should |
||
| 188 | * not be passed along as an index option. |
||
| 189 | */ |
||
| 190 | 4 | if (isset($mapping['index']['order'])) { |
|
| 191 | 1 | $keys[$name] = $mapping['index']['order']; |
|
| 192 | 1 | unset($mapping['index']['order']); |
|
| 193 | 4 | } elseif (isset($mapping['unique']['order'])) { |
|
| 194 | 1 | $keys[$name] = $mapping['unique']['order']; |
|
| 195 | 1 | unset($mapping['unique']['order']); |
|
| 196 | 4 | } elseif (isset($mapping['sparse']['order'])) { |
|
| 197 | $keys[$name] = $mapping['sparse']['order']; |
||
| 198 | unset($mapping['sparse']['order']); |
||
| 199 | } |
||
| 200 | |||
| 201 | /* Initialize $options from any array value among index, unique, and |
||
| 202 | * sparse. Any boolean values for unique or sparse should be merged into |
||
| 203 | * the options afterwards to ensure consistent parsing. |
||
| 204 | */ |
||
| 205 | 4 | $options = array(); |
|
| 206 | 4 | $unique = null; |
|
| 207 | 4 | $sparse = null; |
|
| 208 | |||
| 209 | 4 | if (isset($mapping['index']) && is_array($mapping['index'])) { |
|
| 210 | 1 | $options = $mapping['index']; |
|
| 211 | 1 | } |
|
| 212 | |||
| 213 | 4 | View Code Duplication | if (isset($mapping['unique'])) { |
| 214 | 4 | if (is_array($mapping['unique'])) { |
|
| 215 | 1 | $options = $mapping['unique'] + array('unique' => true); |
|
| 216 | 1 | } else { |
|
| 217 | 3 | $unique = (boolean) $mapping['unique']; |
|
| 218 | } |
||
| 219 | 4 | } |
|
| 220 | |||
| 221 | 4 | View Code Duplication | if (isset($mapping['sparse'])) { |
| 222 | 3 | if (is_array($mapping['sparse'])) { |
|
| 223 | $options = $mapping['sparse'] + array('sparse' => true); |
||
| 224 | } else { |
||
| 225 | 3 | $sparse = (boolean) $mapping['sparse']; |
|
| 226 | } |
||
| 227 | 3 | } |
|
| 228 | |||
| 229 | 4 | if (isset($unique)) { |
|
| 230 | 3 | $options['unique'] = $unique; |
|
| 231 | 3 | } |
|
| 232 | |||
| 233 | 4 | if (isset($sparse)) { |
|
| 234 | 3 | $options['sparse'] = $sparse; |
|
| 235 | 3 | } |
|
| 236 | |||
| 237 | 4 | $class->addIndex($keys, $options); |
|
| 238 | 4 | } |
|
| 239 | |||
| 240 | 4 | private function addMappingFromEmbed(ClassMetadataInfo $class, $fieldName, $embed, $type) |
|
| 241 | { |
||
| 242 | 4 | $defaultStrategy = $type == 'one' ? ClassMetadataInfo::STORAGE_STRATEGY_SET : CollectionHelper::DEFAULT_STRATEGY; |
|
| 243 | $mapping = array( |
||
| 244 | 4 | 'type' => $type, |
|
| 245 | 4 | 'embedded' => true, |
|
| 246 | 4 | 'targetDocument' => isset($embed['targetDocument']) ? $embed['targetDocument'] : null, |
|
| 247 | 4 | 'fieldName' => $fieldName, |
|
| 248 | 4 | 'strategy' => isset($embed['strategy']) ? (string) $embed['strategy'] : $defaultStrategy, |
|
| 249 | 4 | ); |
|
| 250 | 4 | if (isset($embed['name'])) { |
|
| 251 | 1 | $mapping['name'] = $embed['name']; |
|
| 252 | 1 | } |
|
| 253 | 4 | if (isset($embed['discriminatorField'])) { |
|
| 254 | 1 | $mapping['discriminatorField'] = $this->parseDiscriminatorField($embed['discriminatorField']); |
|
| 255 | 1 | } |
|
| 256 | 4 | if (isset($embed['discriminatorMap'])) { |
|
| 257 | 1 | $mapping['discriminatorMap'] = $embed['discriminatorMap']; |
|
| 258 | 1 | } |
|
| 259 | 4 | if (isset($embed['defaultDiscriminatorValue'])) { |
|
| 260 | 1 | $mapping['defaultDiscriminatorValue'] = $embed['defaultDiscriminatorValue']; |
|
| 261 | 1 | } |
|
| 262 | 4 | $this->addFieldMapping($class, $mapping); |
|
| 263 | 4 | } |
|
| 264 | |||
| 265 | 4 | private function addMappingFromReference(ClassMetadataInfo $class, $fieldName, $reference, $type) |
|
| 266 | { |
||
| 267 | 4 | $defaultStrategy = $type == 'one' ? ClassMetadataInfo::STORAGE_STRATEGY_SET : CollectionHelper::DEFAULT_STRATEGY; |
|
| 268 | $mapping = array( |
||
| 269 | 4 | 'cascade' => isset($reference['cascade']) ? $reference['cascade'] : null, |
|
| 270 | 4 | 'orphanRemoval' => isset($reference['orphanRemoval']) ? $reference['orphanRemoval'] : false, |
|
| 271 | 4 | 'type' => $type, |
|
| 272 | 4 | 'reference' => true, |
|
| 273 | 4 | 'simple' => isset($reference['simple']) ? (boolean) $reference['simple'] : false, // deprecated |
|
| 274 | 4 | 'storeAs' => isset($reference['storeAs']) ? (string) $reference['storeAs'] : ClassMetadataInfo::REFERENCE_STORE_AS_DB_REF_WITH_DB, |
|
| 275 | 4 | 'targetDocument' => isset($reference['targetDocument']) ? $reference['targetDocument'] : null, |
|
| 276 | 4 | 'fieldName' => $fieldName, |
|
| 277 | 4 | 'strategy' => isset($reference['strategy']) ? (string) $reference['strategy'] : $defaultStrategy, |
|
| 278 | 4 | 'inversedBy' => isset($reference['inversedBy']) ? (string) $reference['inversedBy'] : null, |
|
| 279 | 4 | 'mappedBy' => isset($reference['mappedBy']) ? (string) $reference['mappedBy'] : null, |
|
| 280 | 4 | 'repositoryMethod' => isset($reference['repositoryMethod']) ? (string) $reference['repositoryMethod'] : null, |
|
| 281 | 4 | 'limit' => isset($reference['limit']) ? (integer) $reference['limit'] : null, |
|
| 282 | 4 | 'skip' => isset($reference['skip']) ? (integer) $reference['skip'] : null, |
|
| 283 | 4 | ); |
|
| 284 | 4 | if (isset($reference['name'])) { |
|
| 285 | 1 | $mapping['name'] = $reference['name']; |
|
| 286 | 1 | } |
|
| 287 | 4 | if (isset($reference['discriminatorField'])) { |
|
| 288 | 1 | $mapping['discriminatorField'] = $this->parseDiscriminatorField($reference['discriminatorField']); |
|
| 289 | 1 | } |
|
| 290 | 4 | if (isset($reference['discriminatorMap'])) { |
|
| 291 | 1 | $mapping['discriminatorMap'] = $reference['discriminatorMap']; |
|
| 292 | 1 | } |
|
| 293 | 4 | if (isset($reference['defaultDiscriminatorValue'])) { |
|
| 294 | 1 | $mapping['defaultDiscriminatorValue'] = $reference['defaultDiscriminatorValue']; |
|
| 295 | 1 | } |
|
| 296 | 4 | if (isset($reference['sort'])) { |
|
| 297 | $mapping['sort'] = $reference['sort']; |
||
| 298 | } |
||
| 299 | 4 | if (isset($reference['criteria'])) { |
|
| 300 | $mapping['criteria'] = $reference['criteria']; |
||
| 301 | } |
||
| 302 | 4 | $this->addFieldMapping($class, $mapping); |
|
| 303 | 4 | } |
|
| 304 | |||
| 305 | /** |
||
| 306 | * Parses the class or field-level "discriminatorField" option. |
||
| 307 | * |
||
| 308 | * If the value is an array, check the "name" option before falling back to |
||
| 309 | * the deprecated "fieldName" option (for BC). Otherwise, the value must be |
||
| 310 | * a string. |
||
| 311 | * |
||
| 312 | * @param array|string $discriminatorField |
||
| 313 | * @return string |
||
| 314 | * @throws \InvalidArgumentException if the value is neither a string nor an |
||
| 315 | * array with a "name" or "fieldName" key. |
||
| 316 | */ |
||
| 317 | 1 | private function parseDiscriminatorField($discriminatorField) |
|
| 318 | { |
||
| 319 | 1 | if (is_string($discriminatorField)) { |
|
| 320 | 1 | return $discriminatorField; |
|
| 321 | } |
||
| 322 | |||
| 323 | 1 | if ( ! is_array($discriminatorField)) { |
|
| 324 | throw new \InvalidArgumentException('Expected array or string for discriminatorField; found: ' . gettype($discriminatorField)); |
||
| 325 | } |
||
| 326 | |||
| 327 | 1 | if (isset($discriminatorField['name'])) { |
|
| 328 | return (string) $discriminatorField['name']; |
||
| 329 | } |
||
| 330 | |||
| 331 | 1 | if (isset($discriminatorField['fieldName'])) { |
|
| 332 | 1 | return (string) $discriminatorField['fieldName']; |
|
| 333 | } |
||
| 334 | |||
| 335 | throw new \InvalidArgumentException('Expected "name" or "fieldName" key in discriminatorField array; found neither.'); |
||
| 336 | } |
||
| 337 | |||
| 338 | /** |
||
| 339 | * {@inheritDoc} |
||
| 340 | */ |
||
| 341 | 5 | protected function loadMappingFile($file) |
|
| 345 | |||
| 346 | 1 | private function setShardKey(ClassMetadataInfo $class, array $shardKey) |
|
| 347 | { |
||
| 348 | 1 | $keys = $shardKey['keys']; |
|
| 349 | 1 | $options = array(); |
|
| 350 | |||
| 363 | } |
||
| 364 |
Loading history...
This check marks implicit conversions of arrays to boolean values in a comparison. While in PHP an empty array is considered to be equal (but not identical) to false, this is not always apparent.
Consider making the comparison explicit by using
empty(..)or! empty(...)instead.