number of results

teamtnt · Apr 20, 2016 · b37e238 · b37e238
1 parent d4a77fd
commit b37e238
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 8 deletions.
diff --git a/composer.lock b/composer.lock
diff --git a/src/TNTSearch.php b/src/TNTSearch.php
@@ -143,7 +143,7 @@ public function searchBoolean($phrase, $numOfResults = 100)
                 $stack[] = array_unique(array_merge($left, $right));
             } else
             if ($token == '~') {
-                $left  = array_pop($stack);
+                $left = array_pop($stack);
                 if (is_string($left)) {
                     $left = $this->getAllDocumentsForWhereKeywordNot($this->stemmer->stem($left), true)
                         ->pluck('doc_id');
@@ -162,8 +162,16 @@ public function searchBoolean($phrase, $numOfResults = 100)
             $docs = new Collection;
         }
 
+        $counter = 0;
+        $docs    = $docs->filter(function ($item) use (&$counter, $numOfResults) {
+            $counter++;
+            if ($counter <= $numOfResults) {
+                return $item;
+            }
+        });
+
         $stopTimer = microtime(true);
-        
+
         if ($this->isFileSystemIndex()) {
             return $this->filesystemMapIdsToPaths($docs)->toArray();
         }

diff --git a/tests/indexer/IndexerTest.php b/tests/indexer/IndexerTest.php
@@ -29,7 +29,7 @@ public function testSearch()
         $tnt->selectIndex($this->indexName);
         $tnt->asYouType = true;
         $res = $tnt->search('Juliet');
-        $this->assertEquals([9, 5, 6, 7, 8], $res['ids']);
+        $this->assertEquals([9, 5, 6, 7, 8, 10], $res['ids']);
 
         $res = $tnt->search('Queen Mab');
         $this->assertEquals([7], $res['ids']);