Fix #2 by adding getTotalCount() method.

Author: pmatseykanets

src/PostgresEngine.php

@@ -166,6 +166,22 @@ public function paginate(Builder $builder, $perPage, $page)
         return $this->performSearch($builder, $perPage, $page);
     }
 
+    /**
+     * Get the total count from a raw result returned by the engine.
+     *
+     * @param  mixed $results
+     * @return int
+     */
+    public function getTotalCount($results)
+    {
+        if (empty($results)) {
+            return 0;
+        }
+
+        return (int) array_first($results)
+            ->total_count;
+    }
+
     /**
      * Perform the given search on the engine.
      *
@@ -184,6 +200,7 @@ protected function performSearch(Builder $builder, $perPage = 0, $page = 1)
             ->crossJoin($this->database->raw('plainto_tsquery(?) query'))
             ->select($builder->model->getKeyName())
             ->selectRaw("{$this->rankingExpression($builder->model, $indexColumn)} AS rank")
+            ->selectRaw('COUNT(*) OVER () AS total_count')
             ->whereRaw("$indexColumn @@ query")
             ->orderBy('rank', 'desc')
             ->orderBy('id');

tests/PostgresEngineTest.php

@@ -92,7 +92,8 @@ public function test_search()
 
         $table->shouldReceive('crossJoin')->with('plainto_tsquery(?) query')->andReturnSelf()
             ->shouldReceive('select')->with('id')->andReturnSelf()
-            ->shouldReceive('selectRaw')->andReturnSelf()
+            ->shouldReceive('selectRaw')->with('ts_rank(searchable,query) AS rank')->andReturnSelf()
+            ->shouldReceive('selectRaw')->with('COUNT(*) OVER () AS total_count')->andReturnSelf()
             ->shouldReceive('whereRaw')->andReturnSelf()
             ->shouldReceive('orderBy')->with('rank', 'desc')->andReturnSelf()
             ->shouldReceive('orderBy')->with('id')->andReturnSelf()
@@ -119,11 +120,22 @@ public function test_map_correctly_maps_results_to_models()
         $model->shouldReceive('get')->once()->andReturn(Collection::make([new TestModel()]));
 
         $results = $engine->map(
-            json_decode('[{"id": 1, "rank": 0.33}]'), $model);
+            json_decode('[{"id": 1, "rank": 0.33, "total_count": 1}]'), $model);
 
         $this->assertCount(1, $results);
     }
 
+    public function test_it_returns_total_count()
+    {
+        list($engine) = $this->getEngine();
+
+        $count = $engine->getTotalCount(
+            json_decode('[{"id": 1, "rank": 0.33, "total_count": 100}]')
+        );
+
+        $this->assertEquals(100, $count);
+    }
+
     protected function getEngine($config = [])
     {
         $resolver = Mockery::mock(ConnectionResolverInterface::class);