From f9ce1dcd4e37a5247825a20fd01a53ca329aaf57 Mon Sep 17 00:00:00 2001
From: Cassiano <cassianotartari@gmail.com>
Date: Wed, 8 Oct 2014 14:30:03 -0300
Subject: [PATCH] Update RawPaginatorAdapter.php

---
 Paginator/RawPaginatorAdapter.php | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/Paginator/RawPaginatorAdapter.php b/Paginator/RawPaginatorAdapter.php
index 8bd4ee2..8aadeb1 100644
--- a/Paginator/RawPaginatorAdapter.php
+++ b/Paginator/RawPaginatorAdapter.php
@@ -36,6 +36,11 @@ class RawPaginatorAdapter implements PaginatorAdapterInterface
      * @var array for the facets
      */
     private $facets;
+    
+    /**
+     * @var array for the aggregations
+     */
+    private $aggregations
 
     /**
      * @see PaginatorAdapterInterface::__construct
@@ -82,6 +87,7 @@ class RawPaginatorAdapter implements PaginatorAdapterInterface
         $resultSet = $this->searchable->search($query, $this->options);
         $this->totalHits = $resultSet->getTotalHits();
         $this->facets = $resultSet->getFacets();
+        $this->aggregations = $resultSet->getAggregations();
         return $resultSet;
     }
 
@@ -126,4 +132,28 @@ class RawPaginatorAdapter implements PaginatorAdapterInterface
 
         return $this->facets;
     }
+    
+    /**
+     * Returns Aggregations
+     *
+     * @return mixed
+     */
+    public function getAggregations() {
+        if (!isset($this->aggregations)) {
+            $this->aggregations = $this->searchable->search($this->query)->getAggregations();
+        }
+
+        return $this->aggregations;
+    }
+    
+
+    /**
+     * Returns the Query
+     *
+     * @return Query the search query
+     */
+    public function getQuery()
+    {
+        return $this->query;
+    }
 }