Merge pull request #84 from zoul0813/feature/newCollection

Update to use `$this->getModel()->newCollection($items)` Close #83
baopham · Aug 18, 2017 · 96b785f · 96b785f
2 parents a53ebaf + eb42bef
commit 96b785f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/DynamoDbQueryBuilder.php b/src/DynamoDbQueryBuilder.php
@@ -526,7 +526,7 @@ protected function getAll($columns = [], $limit = -1, $use_iterator = true)
             if ($this->conditionsAreExactSearch()) {
                 $item = $this->find($conditionValue, $columns);
 
-                return new Collection([$item]);
+                return $this->getModel()->newCollection([$item]);
             }
         }
 
@@ -579,7 +579,7 @@ protected function getAll($columns = [], $limit = -1, $use_iterator = true)
             $results[] = $model;
         }
 
-        return new Collection($results);
+        return $this->getModel()->newCollection($results);
     }
 
     protected function buildExpressionQuery()