Add count method to client

webgriffe · Jun 27, 2020 · a733288 · a733288
1 parent 63b1103
commit a733288
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 0 deletions.
diff --git a/src/Client.php b/src/Client.php
@@ -180,6 +180,22 @@ public function search(array $query, string $indexOrIndices = null, array $optio
         return $this->doRequest($this->createJsonRequest($method, $uri, ['query' => $query]));
     }
 
+    public function count(string $index, array $options = [], array $query = null): Promise
+    {
+        $method = 'GET';
+        $uri = [$this->baseUri, $index];
+        $uri[] = '_count';
+        $uri = implode('/', $uri);
+        if ($options) {
+            $uri .= '?' . http_build_query($options);
+        }
+        $body = null;
+        if (null !== $query) {
+            $body = ['query' => $query];
+        }
+        return $this->doRequest($this->createJsonRequest($method, $uri, $body));
+    }
+
     private function createJsonRequest(string $method, string $uri, array $body = null): Request
     {
         $request = (new Request($uri, $method))

diff --git a/tests/Integration/ClientTest.php b/tests/Integration/ClientTest.php
@@ -240,4 +240,36 @@ public function testSearch(): void
         $this->assertIsArray($response);
         $this->assertCount(1, $response['hits']['hits']);
     }
+
+    public function testCount(): void
+    {
+        Promise\wait($this->client->createIndex(self::TEST_INDEX));
+        Promise\wait(
+            $this->client->indexDocument(self::TEST_INDEX, '', ['payload' => []], ['refresh' => 'true'])
+        );
+        Promise\wait(
+            $this->client->indexDocument(self::TEST_INDEX, '', ['payload' => []], ['refresh' => 'true'])
+        );
+
+        $response = Promise\wait($this->client->count(self::TEST_INDEX));
+
+        $this->assertIsArray($response);
+        $this->assertEquals(2, $response['count']);
+    }
+
+    public function testCountWithQuery(): void
+    {
+        Promise\wait($this->client->createIndex(self::TEST_INDEX));
+        Promise\wait(
+            $this->client->indexDocument(self::TEST_INDEX, '', ['user' => 'kimchy'], ['refresh' => 'true'])
+        );
+        Promise\wait(
+            $this->client->indexDocument(self::TEST_INDEX, '', ['user' => 'foo'], ['refresh' => 'true'])
+        );
+
+        $response = Promise\wait($this->client->count(self::TEST_INDEX, [], ['term' => ['user' => 'kimchy']]));
+
+        $this->assertIsArray($response);
+        $this->assertEquals(1, $response['count']);
+    }
 }