Relations eager loading support when paginating

Author: babenkoivan

src/Builders/FilterBuilder.php

@@ -205,4 +205,34 @@ public function from($offset)
 
         return $this;
     }
+
+    /**
+     * @inheritdoc
+     */
+    public function get()
+    {
+        $collection = parent::get();
+
+        if (isset($this->with) && $collection->count() > 0) {
+            $collection->load($this->with);
+        }
+
+        return $collection;
+    }
+
+    /**
+     * @inheritdoc
+     */
+    public function paginate($perPage = null, $pageName = 'page', $page = null)
+    {
+        $paginator = parent::paginate($perPage, $pageName, $page);
+
+        if (isset($this->with) && $paginator->total() > 0) {
+            $paginator
+                ->getCollection()
+                ->load($this->with);
+        }
+
+        return $paginator;
+    }
 }

src/ElasticEngine.php

@@ -225,18 +225,4 @@ public function getTotalCount($results)
     {
         return $results['hits']['total'];
     }
-
-    /**
-     * @inheritdoc
-     */
-    public function get(Builder $builder)
-    {
-        $collection = parent::get($builder);
-
-        if (isset($builder->with) && $collection->count() > 0) {
-            $collection->load($builder->with);
-        }
-
-        return $collection;
-    }
 }