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 ActiveQuery 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 ActiveQuery, and based on these observations, apply Extract Interface, too.
| 1 | <?php |
||
| 18 | class ActiveQuery extends Query implements ActiveQueryInterface |
||
| 19 | { |
||
| 20 | use ActiveQueryTrait; |
||
| 21 | use ActiveRelationTrait; |
||
| 22 | |||
| 23 | /** |
||
| 24 | * @event Event an event that is triggered when the query is initialized via [[init()]]. |
||
| 25 | */ |
||
| 26 | const EVENT_INIT = 'init'; |
||
| 27 | |||
| 28 | /** |
||
| 29 | * @var array|null a list of relations that this query should be joined with |
||
| 30 | */ |
||
| 31 | public $joinWith = []; |
||
| 32 | |||
| 33 | /** |
||
| 34 | * Constructor. |
||
| 35 | * @param string $modelClass the model class associated with this query |
||
| 36 | * @param array $config configurations to be applied to the newly created query object |
||
| 37 | */ |
||
| 38 | 2 | public function __construct($modelClass, $config = []) |
|
| 44 | |||
| 45 | /** |
||
| 46 | * Initializes the object. |
||
| 47 | * This method is called at the end of the constructor. The default implementation will trigger |
||
| 48 | * an [[EVENT_INIT]] event. If you override this method, make sure you call the parent implementation at the end |
||
| 49 | * to ensure triggering of the event. |
||
| 50 | */ |
||
| 51 | 2 | public function init() |
|
| 56 | |||
| 57 | /** |
||
| 58 | * Creates a DB command that can be used to execute this query. |
||
| 59 | * @param AbstractConnection $db the DB connection used to create the DB command. |
||
| 60 | * If null, the DB connection returned by [[modelClass]] will be used. |
||
| 61 | * @return Command the created DB command instance |
||
| 62 | */ |
||
| 63 | 2 | public function createCommand($db = null) |
|
| 64 | { |
||
| 65 | 2 | if ($this->primaryModel !== null) { |
|
| 66 | // lazy loading |
||
| 67 | if (is_array($this->via)) { |
||
| 68 | // via relation |
||
| 69 | /** @var $viaQuery ActiveQuery */ |
||
| 70 | list($viaName, $viaQuery) = $this->via; |
||
| 71 | if ($viaQuery->multiple) { |
||
| 72 | $viaModels = $viaQuery->all(); |
||
| 73 | $this->primaryModel->populateRelation($viaName, $viaModels); |
||
| 74 | } else { |
||
| 75 | $model = $viaQuery->one(); |
||
| 76 | $this->primaryModel->populateRelation($viaName, $model); |
||
| 77 | $viaModels = $model === null ? [] : [$model]; |
||
| 78 | } |
||
| 79 | $this->filterByModels($viaModels); |
||
| 80 | } else { |
||
| 81 | $this->filterByModels([$this->primaryModel]); |
||
| 82 | } |
||
| 83 | } |
||
| 84 | |||
| 85 | /* @var $modelClass ActiveRecord */ |
||
| 86 | 2 | $modelClass = $this->modelClass; |
|
| 87 | |||
| 88 | 2 | if ($db === null) { |
|
| 89 | 2 | $db = $modelClass::getDb(); |
|
| 90 | } |
||
| 91 | 2 | if ($this->from === null) { |
|
| 92 | 2 | $this->from = $modelClass::tableName(); |
|
|
|
|||
| 93 | } |
||
| 94 | |||
| 95 | 2 | return parent::createCommand($db); |
|
| 96 | } |
||
| 97 | |||
| 98 | /** |
||
| 99 | * Prepares query for use. See NOTE. |
||
| 100 | * @param QueryBuilder $builder |
||
| 101 | * @return static |
||
| 102 | */ |
||
| 103 | 2 | public function prepare($builder = null) |
|
| 104 | { |
||
| 105 | // NOTE: because the same ActiveQuery may be used to build different SQL statements |
||
| 106 | // (e.g. by ActiveDataProvider, one for count query, the other for row data query, |
||
| 107 | // it is important to make sure the same ActiveQuery can be used to build SQL statements |
||
| 108 | // multiple times. |
||
| 109 | 2 | if (!empty($this->joinWith)) { |
|
| 110 | $this->buildJoinWith(); |
||
| 111 | $this->joinWith = null; |
||
| 112 | } |
||
| 113 | |||
| 114 | 2 | return $this; |
|
| 115 | } |
||
| 116 | |||
| 117 | /** |
||
| 118 | * @param $with |
||
| 119 | * @return static |
||
| 120 | */ |
||
| 121 | public function joinWith($with) |
||
| 127 | |||
| 128 | private function buildJoinWith() |
||
| 129 | { |
||
| 130 | $join = $this->join; |
||
| 131 | $this->join = []; |
||
| 132 | |||
| 133 | $model = new $this->modelClass(); |
||
| 134 | |||
| 135 | foreach ($this->joinWith as $with) { |
||
| 136 | $this->joinWithRelations($model, $with); |
||
| 137 | |||
| 138 | foreach ($with as $name => $callback) { |
||
| 139 | if (is_int($name)) { |
||
| 140 | $this->innerJoin([$callback]); |
||
| 141 | } else { |
||
| 142 | $this->innerJoin([$name => $callback]); |
||
| 143 | } |
||
| 144 | |||
| 145 | unset($with[$name]); |
||
| 146 | } |
||
| 147 | } |
||
| 148 | |||
| 149 | if (!empty($join)) { |
||
| 150 | // append explicit join to joinWith() |
||
| 151 | // https://github.com/yiisoft/yii2/issues/2880 |
||
| 152 | $this->join = empty($this->join) ? $join : array_merge($this->join, $join); |
||
| 153 | } |
||
| 154 | |||
| 155 | if (empty($this->select) || true) { |
||
| 156 | $this->addSelect(['*' => '*']); |
||
| 157 | foreach ($this->joinWith as $join) { |
||
| 158 | $key = array_shift(array_keys($join)); |
||
| 159 | $closure = array_shift($join); |
||
| 160 | |||
| 161 | $this->addSelect(is_int($key) ? $closure : $key); |
||
| 162 | } |
||
| 163 | } |
||
| 164 | } |
||
| 165 | |||
| 166 | /** |
||
| 167 | * @param ActiveRecord $model |
||
| 168 | * @param $with |
||
| 169 | */ |
||
| 170 | protected function joinWithRelations($model, $with) |
||
| 171 | { |
||
| 172 | foreach ($with as $name => $callback) { |
||
| 173 | if (is_int($name)) { |
||
| 174 | $name = $callback; |
||
| 175 | $callback = null; |
||
| 176 | } |
||
| 177 | |||
| 178 | $primaryModel = $model; |
||
| 179 | $parent = $this; |
||
| 180 | |||
| 181 | if (!isset($relations[$name])) { |
||
| 182 | $relations[$name] = $relation = $primaryModel->getRelation($name); |
||
| 183 | if ($callback !== null) { |
||
| 184 | call_user_func($callback, $relation); |
||
| 185 | } |
||
| 186 | if (!empty($relation->joinWith)) { |
||
| 187 | $relation->buildJoinWith(); |
||
| 188 | } |
||
| 189 | $this->joinWithRelation($parent, $relation); |
||
| 190 | } |
||
| 191 | } |
||
| 192 | } |
||
| 193 | |||
| 194 | /** |
||
| 195 | * Joins a parent query with a child query. |
||
| 196 | * The current query object will be modified accordingly. |
||
| 197 | * @param ActiveQuery $parent |
||
| 198 | * @param ActiveQuery $child |
||
| 199 | */ |
||
| 200 | private function joinWithRelation($parent, $child) |
||
| 201 | { |
||
| 202 | if (!empty($child->join)) { |
||
| 203 | foreach ($child->join as $join) { |
||
| 204 | $this->join[] = $join; |
||
| 205 | } |
||
| 206 | } |
||
| 207 | } |
||
| 208 | |||
| 209 | public function select($columns, $option = null) |
||
| 215 | |||
| 216 | /** |
||
| 217 | * @param array|string $columns |
||
| 218 | * @return $this |
||
| 219 | */ |
||
| 220 | public function addSelect($columns) |
||
| 221 | { |
||
| 222 | if (!is_array($columns)) { |
||
| 223 | $columns = (array) $columns; |
||
| 224 | } |
||
| 225 | |||
| 226 | if ($this->select === null) { |
||
| 227 | $this->select = $columns; |
||
| 228 | } else { |
||
| 229 | $this->select = array_merge($this->select, $columns); |
||
| 230 | } |
||
| 231 | |||
| 232 | return $this; |
||
| 233 | } |
||
| 234 | |||
| 235 | /** |
||
| 236 | * Executes query and returns a single row of result. |
||
| 237 | * |
||
| 238 | * @param AbstractConnection $db the DB connection used to create the DB command. |
||
| 239 | * If null, the DB connection returned by [[modelClass]] will be used. |
||
| 240 | * @return ActiveRecord|array|null a single row of query result. Depending on the setting of [[asArray]], |
||
| 241 | * the query result may be either an array or an ActiveRecord object. Null will be returned |
||
| 242 | * if the query results in nothing. |
||
| 243 | */ |
||
| 244 | public function one($db = null) |
||
| 257 | |||
| 258 | /** |
||
| 259 | * Executes query and returns all results as an array. |
||
| 260 | * @param AbstractConnection $db the DB connection used to create the DB command. |
||
| 261 | * If null, the DB connection returned by [[modelClass]] will be used. |
||
| 262 | * @return array|ActiveRecord[] the query results. If the query results in nothing, an empty array will be returned. |
||
| 263 | */ |
||
| 264 | 2 | public function all($db = null) |
|
| 274 | |||
| 275 | 2 | public function populate($rows) |
|
| 276 | { |
||
| 277 | 2 | if (empty($rows)) { |
|
| 278 | return []; |
||
| 279 | } |
||
| 280 | |||
| 281 | 2 | $models = $this->createModels($rows); |
|
| 282 | |||
| 283 | 2 | if (!empty($this->with)) { |
|
| 284 | $this->findWith($this->with, $models); |
||
| 285 | } |
||
| 286 | |||
| 287 | 2 | foreach ($models as $model) { |
|
| 288 | 2 | $model->afterFind(); |
|
| 289 | } |
||
| 290 | |||
| 291 | 2 | return $models; |
|
| 292 | } |
||
| 293 | |||
| 294 | 2 | private function createModels($rows) |
|
| 295 | { |
||
| 296 | 2 | $models = []; |
|
| 297 | 2 | $class = $this->modelClass; |
|
| 298 | 2 | foreach ($rows as $row) { |
|
| 299 | 2 | $model = $class::instantiate($row); |
|
| 300 | 2 | $modelClass = get_class($model); |
|
| 301 | 2 | $modelClass::populateRecord($model, $row); |
|
| 302 | 2 | $this->populateJoinedRelations($model, $row); |
|
| 303 | 2 | if ($this->indexBy) { |
|
| 304 | View Code Duplication | if ($this->indexBy instanceof \Closure) { |
|
| 305 | $key = call_user_func($this->indexBy, $model); |
||
| 306 | } else { |
||
| 307 | $key = $model->{$this->indexBy}; |
||
| 308 | } |
||
| 309 | $models[$key] = $model; |
||
| 310 | } else { |
||
| 311 | 2 | $models[] = $model; |
|
| 312 | } |
||
| 313 | } |
||
| 314 | |||
| 315 | 2 | return $models; |
|
| 316 | } |
||
| 317 | |||
| 318 | /** |
||
| 319 | * Populates joined relations from [[join]] array. |
||
| 320 | * |
||
| 321 | * @param ActiveRecord $model |
||
| 322 | * @param array $row |
||
| 323 | */ |
||
| 324 | 2 | public function populateJoinedRelations($model, array $row) |
|
| 325 | { |
||
| 326 | 2 | foreach ($row as $key => $value) { |
|
| 327 | 2 | if (empty($this->join) || !is_array($value) || $model->hasAttribute($key)) { |
|
| 328 | 2 | continue; |
|
| 329 | } |
||
| 330 | foreach ($this->join as $join) { |
||
| 331 | $name = array_shift(array_keys($join)); |
||
| 332 | $closure = array_shift($join); |
||
| 333 | |||
| 334 | if (is_int($name)) { |
||
| 335 | $name = $closure; |
||
| 336 | $closure = null; |
||
| 337 | } |
||
| 338 | if ($name !== $key) { |
||
| 339 | continue; |
||
| 340 | } |
||
| 341 | if ($model->isRelationPopulated($name)) { |
||
| 342 | continue 2; |
||
| 343 | } |
||
| 344 | $records = []; |
||
| 345 | $relation = $model->getRelation($name); |
||
| 346 | $relationClass = $relation->modelClass; |
||
| 347 | if ($closure !== null) { |
||
| 348 | call_user_func($closure, $relation); |
||
| 349 | } |
||
| 350 | $relation->prepare(); |
||
| 351 | |||
| 352 | if ($relation->multiple) { |
||
| 353 | foreach ($value as $item) { |
||
| 354 | $relatedModel = $relationClass::instantiate($item); |
||
| 355 | $relatedModelClass = get_class($relatedModel); |
||
| 356 | $relatedModelClass::populateRecord($relatedModel, $item); |
||
| 357 | $relation->populateJoinedRelations($relatedModel, $item); |
||
| 358 | $relation->addInverseRelation($relatedModel, $model); |
||
| 359 | if ($relation->indexBy !== null) { |
||
| 360 | $index = is_string($relation->indexBy) |
||
| 361 | ? $relatedModel[$relation->indexBy] |
||
| 362 | : call_user_func($relation->indexBy, $relatedModel); |
||
| 363 | $records[$index] = $relatedModel; |
||
| 364 | } else { |
||
| 365 | $records[] = $relatedModel; |
||
| 366 | } |
||
| 367 | } |
||
| 368 | } else { |
||
| 369 | $relatedModel = $relationClass::instantiate($value); |
||
| 370 | $relatedModelClass = get_class($relatedModel); |
||
| 371 | $relatedModelClass::populateRecord($relatedModel, $value); |
||
| 372 | $relation->populateJoinedRelations($relatedModel, $value); |
||
| 373 | $relation->addInverseRelation($relatedModel, $model); |
||
| 374 | $records = $relatedModel; |
||
| 375 | } |
||
| 376 | |||
| 377 | $model->populateRelation($name, $records); |
||
| 378 | } |
||
| 379 | } |
||
| 380 | 2 | } |
|
| 381 | |||
| 382 | /** |
||
| 383 | * @param $relatedModel |
||
| 384 | */ |
||
| 385 | private function addInverseRelation($relatedModel) |
||
| 394 | } |
||
| 395 |
Our type inference engine has found an assignment to a property that is incompatible with the declared type of that property.
Either this assignment is in error or the assigned type should be added to the documentation/type hint for that property..