Remove dupplicated code in paginated response (#1012)

Author: jderusse

src/Result/BatchGetItemOutput.php

@@ -87,33 +87,7 @@ public function getConsumedCapacity(bool $currentPageOnly = false): iterable
      */
     public function getIterator(): \Traversable
     {
-        $client = $this->awsClient;
-        if (!$client instanceof DynamoDbClient) {
-            throw new InvalidArgument('missing client injected in paginated result');
-        }
-        if (!$this->input instanceof BatchGetItemInput) {
-            throw new InvalidArgument('missing last request injected in paginated result');
-        }
-        $input = clone $this->input;
-        $page = $this;
-        while (true) {
-            if ($page->getUnprocessedKeys()) {
-                $input->setRequestItems($page->getUnprocessedKeys());
-
-                $this->registerPrefetch($nextPage = $client->batchGetItem($input));
-            } else {
-                $nextPage = null;
-            }
-
-            yield from $page->getConsumedCapacity(true);
-
-            if (null === $nextPage) {
-                break;
-            }
-
-            $this->unregisterPrefetch($nextPage);
-            $page = $nextPage;
-        }
+        yield from $this->getConsumedCapacity();
     }
 
     /**

src/Result/ListTablesOutput.php

@@ -34,33 +34,7 @@ class ListTablesOutput extends Result implements \IteratorAggregate
      */
     public function getIterator(): \Traversable
     {
-        $client = $this->awsClient;
-        if (!$client instanceof DynamoDbClient) {
-            throw new InvalidArgument('missing client injected in paginated result');
-        }
-        if (!$this->input instanceof ListTablesInput) {
-            throw new InvalidArgument('missing last request injected in paginated result');
-        }
-        $input = clone $this->input;
-        $page = $this;
-        while (true) {
-            if ($page->getLastEvaluatedTableName()) {
-                $input->setExclusiveStartTableName($page->getLastEvaluatedTableName());
-
-                $this->registerPrefetch($nextPage = $client->listTables($input));
-            } else {
-                $nextPage = null;
-            }
-
-            yield from $page->getTableNames(true);
-
-            if (null === $nextPage) {
-                break;
-            }
-
-            $this->unregisterPrefetch($nextPage);
-            $page = $nextPage;
-        }
+        yield from $this->getTableNames();
     }
 
     public function getLastEvaluatedTableName(): ?string

src/Result/QueryOutput.php

@@ -118,33 +118,7 @@ public function getItems(bool $currentPageOnly = false): iterable
      */
     public function getIterator(): \Traversable
     {
-        $client = $this->awsClient;
-        if (!$client instanceof DynamoDbClient) {
-            throw new InvalidArgument('missing client injected in paginated result');
-        }
-        if (!$this->input instanceof QueryInput) {
-            throw new InvalidArgument('missing last request injected in paginated result');
-        }
-        $input = clone $this->input;
-        $page = $this;
-        while (true) {
-            if ($page->getLastEvaluatedKey()) {
-                $input->setExclusiveStartKey($page->getLastEvaluatedKey());
-
-                $this->registerPrefetch($nextPage = $client->query($input));
-            } else {
-                $nextPage = null;
-            }
-
-            yield from $page->getItems(true);
-
-            if (null === $nextPage) {
-                break;
-            }
-
-            $this->unregisterPrefetch($nextPage);
-            $page = $nextPage;
-        }
+        yield from $this->getItems();
     }
 
     /**

src/Result/ScanOutput.php

@@ -118,33 +118,7 @@ public function getItems(bool $currentPageOnly = false): iterable
      */
     public function getIterator(): \Traversable
     {
-        $client = $this->awsClient;
-        if (!$client instanceof DynamoDbClient) {
-            throw new InvalidArgument('missing client injected in paginated result');
-        }
-        if (!$this->input instanceof ScanInput) {
-            throw new InvalidArgument('missing last request injected in paginated result');
-        }
-        $input = clone $this->input;
-        $page = $this;
-        while (true) {
-            if ($page->getLastEvaluatedKey()) {
-                $input->setExclusiveStartKey($page->getLastEvaluatedKey());
-
-                $this->registerPrefetch($nextPage = $client->scan($input));
-            } else {
-                $nextPage = null;
-            }
-
-            yield from $page->getItems(true);
-
-            if (null === $nextPage) {
-                break;
-            }
-
-            $this->unregisterPrefetch($nextPage);
-            $page = $nextPage;
-        }
+        yield from $this->getItems();
     }
 
     /**