Connection, Context: added fetchFields()

Author: dg

src/Database/Connection.php

@@ -215,6 +215,15 @@ public function fetchField(string $sql, ...$params)
 	}
 
 
+	/**
+	 * Shortcut for query()->fetchFields()
+	 */
+	public function fetchFields(string $sql, ...$params): ?array
+	{
+		return $this->query($sql, ...$params)->fetchFields();
+	}
+
+
 	/**
 	 * Shortcut for query()->fetchPairs()
 	 */

src/Database/Context.php

@@ -127,6 +127,15 @@ public function fetchField(string $sql, ...$params)
 	}
 
 
+	/**
+	 * Shortcut for query()->fetchFields()
+	 */
+	public function fetchFields(string $sql, ...$params): ?array
+	{
+		return $this->connection->query($sql, ...$params)->fetchFields();
+	}
+
+
 	/**
 	 * Shortcut for query()->fetchPairs()
 	 */

src/Database/ResultSet.php

@@ -261,6 +261,16 @@ public function fetchField($column = 0)
 	}
 
 
+	/**
+	 * Fetches array of fields.
+	 */
+	public function fetchFields(): ?array
+	{
+		$row = $this->fetch();
+		return $row ? array_values((array) $row) : null;
+	}
+
+
 	/**
 	 * @inheritDoc
 	 */

tests/Database/Connection.fetch.phpt

@@ -29,6 +29,11 @@ test(function () use ($connection) { // fetchField
 });
 
 
+test(function () use ($connection) { // fetchFields
+	Assert::same([11, 'Jakub Vrana'], $connection->fetchFields('SELECT id, name FROM author ORDER BY id'));
+});
+
+
 test(function () use ($connection) { // fetchPairs
 	$pairs = $connection->fetchPairs('SELECT name, id FROM author WHERE id > ? ORDER BY id', 11);
 	Assert::same([

tests/Database/Context.fetch.phpt

@@ -29,6 +29,11 @@ test(function () use ($context) { // fetchField
 });
 
 
+test(function () use ($context) { // fetchFields
+	Assert::same([11, 'Jakub Vrana'], $context->fetchFields('SELECT id, name FROM author ORDER BY id'));
+});
+
+
 test(function () use ($context) { // fetchPairs
 	$pairs = $context->fetchPairs('SELECT name, id FROM author WHERE id > ? ORDER BY id', 11);
 	Assert::same([

tests/Database/ResultSet.fetchFields().phpt

@@ -0,0 +1,28 @@
+<?php
+
+/**
+ * Test: Nette\Database\ResultSet: Fetch fields.
+ * @dataProvider? databases.ini
+ */
+
+declare(strict_types=1);
+
+use Tester\Assert;
+
+require __DIR__ . '/connect.inc.php'; // create $connection
+
+Nette\Database\Helpers::loadFromFile($connection, __DIR__ . "/files/{$driverName}-nette_test1.sql");
+
+
+test(function () use ($connection) {
+	$res = $connection->query('SELECT name, id FROM author ORDER BY id');
+
+	Assert::same(['Jakub Vrana', 11], $res->fetchFields());
+});
+
+
+test(function () use ($connection) {
+	$res = $connection->query('SELECT id FROM author WHERE id = ?', 666);
+
+	Assert::null($res->fetchFields());
+});