Fetching rows as native arrays for better static analysis - fetchAllAssoc

src/Bridges/DatabaseDI/DatabaseExtension.php

@@ -8,11 +8,11 @@
 namespace Nette\Bridges\DatabaseDI;
 
 use Nette;
+use Nette\Database\Table\ActiveRow;
 use Nette\DI\Definitions\Statement;
 use Nette\Schema\Expect;
 use Tracy;
 use function is_array, is_string;
-use Nette\Database\Table\ActiveRow;
 
 
 /**

src/Database/Database.php

@@ -506,6 +506,18 @@ public function fetchAll(#[Language('SQL')] string $sql, #[Language('GenericSQL'
 	}
 
 
+	/**
+	 * Executes SQL query and returns all rows as an array of arrays.
+	 * Useful because the rows can be typed using @var.
+	 * @param  literal-string  $sql
+	 * @return list<array>
+	 */
+	public function fetchAllAssoc(#[Language('SQL')] string $sql, #[Language('GenericSQL')] mixed ...$params): array
+	{
+		return $this->query($sql, ...$params)->fetchAllAssoc();
+	}
+
+
 	/**
 	 * Creates SQL literal value.
 	 */

src/Database/Result.php

@@ -23,6 +23,9 @@ class Result implements \IteratorAggregate
 	/** @var list<Row> */
 	private array $rows;
 
+	/** @var list<array> */
+	private array $rowsArray;
+
 	/** @var array<string, array{name: string, nativeType: string, table?: string}> */
 	private array $meta;
 
@@ -204,6 +207,25 @@ public function fetchAll(): array
 	}
 
 
+	/**
+	 * Returns all remaining rows as array of associative arrays.
+	 * Useful because the rows can be typed using @var.
+	 * @return list<array>
+	 */
+	public function fetchAllAssoc(): array
+	{
+		if(!isset($this->rowsArray)){
+			$this->rowsArray = [];
+			while(true){
+				$data = $this->fetchAssoc();
+				if($data === null) break;
+				$this->rowsArray[] = $data;
+			}
+		}
+		return $this->rowsArray;
+	}
+
+
 	private function normalizeRow(array $row): array
 	{
 		$engine = $this->explorer->getDatabaseEngine();

src/Database/Table/GroupedSelection.php

@@ -7,7 +7,6 @@
 
 namespace Nette\Database\Table;
 
-use Nette;
 use Nette\Database\Explorer;
 use function array_keys, count, iterator_to_array, preg_match, reset;
 

tests/types/database-types.php

@@ -112,6 +112,12 @@ function testResultSetFetchAll(ResultSet $resultSet): void
 }
 
 
+function testResultSetFetchAllAssoc(ResultSet $resultSet): void
+{
+	assertType('list<array>', $resultSet->fetchAllAssoc());
+}
+
+
 function testResultSetFetchPairs(ResultSet $resultSet): void
 {
 	assertType('array<mixed>', $resultSet->fetchPairs());