Update QueryObjectDataSource.php

Author: vpalousek

src/QueryObjectDataSource.php

@@ -93,7 +93,9 @@ public function setLimitCallback($callback) {
 
 	protected function getResultSet(int $page = 1, ?int $itemsPerPage = null) {
 		if (!$this->resultSet) {
-			$this->resultSet = $this->queryObject->getResultSet($page, $itemsPerPage);
+			$this->resultSet = $itemsPerPage
+				? $this->queryObject->getResultSet($page, $itemsPerPage)->getIterator()
+				: $this->queryObject->fetch();
 		}
 
 		return $this->resultSet;
@@ -116,7 +118,7 @@ public function getData(): array {
 			return $this->data;
 		}
 
-		return iterator_to_array($this->getResultSet()->getIterator());
+		return $this->getResultSet();
 	}
 
 	/**