Fix bug in many to many eager loading.

Signed-off-by: Taylor Otwell <taylorotwell@gmail.com>
This commit is contained in:
Taylor Otwell 2012-04-05 09:21:28 -05:00
parent 3e00ce2efb
commit cc4514e8f6
1 changed files with 33 additions and 12 deletions

View File

@ -1,4 +1,7 @@
<?php namespace Laravel\Database\Eloquent; use Laravel\Database; <?php namespace Laravel\Database\Eloquent;
use Laravel\Database;
use Laravel\Database\Eloquent\Relationships\Has_Many_And_Belongs_To;
class Query { class Query {
@ -54,7 +57,7 @@ public function __construct($model)
*/ */
public function first($columns = array('*')) public function first($columns = array('*'))
{ {
$results = $this->hydrate($this->model, $this->table->take(1)->get($columns, false)); $results = $this->hydrate($this->model, $this->table->take(1)->get($columns));
return (count($results) > 0) ? head($results) : null; return (count($results) > 0) ? head($results) : null;
} }
@ -63,12 +66,12 @@ public function first($columns = array('*'))
* Get all of the model results for the query. * Get all of the model results for the query.
* *
* @param array $columns * @param array $columns
* @param bool $include * @param bool $keyed
* @return array * @return array
*/ */
public function get($columns = array('*'), $include = true) public function get($columns = array('*'), $keyed = true)
{ {
return $this->hydrate($this->model, $this->table->get($columns), $include); return $this->hydrate($this->model, $this->table->get($columns), $keyed);
} }
/** /**
@ -97,9 +100,10 @@ public function paginate($per_page = null, $columns = array('*'))
* *
* @param Model $model * @param Model $model
* @param array $results * @param array $results
* @param bool $keyed
* @return array * @return array
*/ */
public function hydrate($model, $results, $include = true) public function hydrate($model, $results, $keyed = true)
{ {
$class = get_class($model); $class = get_class($model);
@ -124,10 +128,20 @@ public function hydrate($model, $results, $include = true)
$new->original = $new->attributes; $new->original = $new->attributes;
$models[$result[$this->model->key()]] = $new; // Typically, the resulting models are keyed by their primary key, but it
// may be useful to not do this in some circumstances such as when we
// are eager loading a *-to-* relationships which has duplicates.
if ($keyed)
{
$models[$result[$this->model->key()]] = $new;
}
else
{
$models[] = $new;
}
} }
if ($include and count($results) > 0) if (count($results) > 0)
{ {
foreach ($this->model_includes() as $relationship => $constraints) foreach ($this->model_includes() as $relationship => $constraints)
{ {
@ -183,12 +197,19 @@ protected function load(&$results, $relationship, $constraints)
$query->table->where_nested($constraints); $query->table->where_nested($constraints);
} }
// Before matching the models, we will initialize the relationship
// to either null for single-value relationships or an array for
// the multi-value relationships as their baseline value.
$query->initialize($results, $relationship); $query->initialize($results, $relationship);
$query->match($relationship, $results, $query->get()); // If we're eager loading a many-to-many relationship we will disable
// the primary key indexing on the hydration since there could be
// roles shared across users and we don't want to overwrite.
if ( ! $query instanceof Has_Many_And_Belongs_To)
{
$query->match($relationship, $results, $query->get());
}
else
{
$query->match($relationship, $results, $query->get(array('*'), false));
}
} }
/** /**