PgSqlDriver: support for partitioned tables (#286)

Author: otuhacek

src/Database/Drivers/PgSqlDriver.php

@@ -100,7 +100,7 @@ public function getTables(): array
 				pg_catalog.pg_class AS c
 				JOIN pg_catalog.pg_namespace AS n ON n.oid = c.relnamespace
 			WHERE
-				c.relkind IN ('r', 'v', 'm')
+				c.relkind IN ('r', 'v', 'm', 'p')
 				AND n.nspname = ANY (pg_catalog.current_schemas(FALSE))
 			ORDER BY
 				c.relname
@@ -131,7 +131,7 @@ public function getColumns(string $table): array
 				LEFT JOIN pg_catalog.pg_attrdef AS ad ON ad.adrelid = c.oid AND ad.adnum = a.attnum
 				LEFT JOIN pg_catalog.pg_constraint AS co ON co.connamespace = c.relnamespace AND contype = 'p' AND co.conrelid = c.oid AND a.attnum = ANY(co.conkey)
 			WHERE
-				c.relkind IN ('r', 'v', 'm')
+				c.relkind IN ('r', 'v', 'm', 'p')
 				AND c.oid = {$this->pdo->quote($this->delimiteFQN($table))}::regclass
 				AND a.attnum > 0
 				AND NOT a.attisdropped
@@ -163,7 +163,7 @@ public function getIndexes(string $table): array
 				JOIN pg_catalog.pg_class AS c2 ON i.indexrelid = c2.oid
 				LEFT JOIN pg_catalog.pg_attribute AS a ON c1.oid = a.attrelid AND a.attnum = ANY(i.indkey)
 			WHERE
-				c1.relkind = 'r'
+				c1.relkind IN ('r', 'p')
 				AND c1.oid = {$this->pdo->quote($this->delimiteFQN($table))}::regclass
 		") as $row) {
 			$indexes[$row['name']]['name'] = $row['name'];

tests/Database/Reflection.postgre.10.phpt

@@ -124,3 +124,37 @@ test('Materialized view columns', function () use ($connection) {
 		array_column($driver->getColumns('source_mt'), 'name'),
 	);
 });
+
+
+test('Partitioned table', function () use ($connection) {
+	Nette\Database\Helpers::loadFromFile($connection, Tester\FileMock::create('
+		DROP SCHEMA IF EXISTS "reflection_10" CASCADE;
+		CREATE SCHEMA "reflection_10";
+
+		CREATE TABLE "reflection_10"."parted" (
+			"id" INTEGER PRIMARY KEY,
+			"value" INTEGER
+		) PARTITION BY RANGE (id);
+
+		CREATE TABLE "reflection_10"."part_1" PARTITION OF "reflection_10"."parted" FOR VALUES FROM (1) TO (10);
+	'));
+
+	$driver = $connection->getDriver();
+
+	$connection->query('SET search_path TO reflection_10');
+
+	Assert::same([
+		['name' => 'part_1', 'view' => false, 'fullName' => 'reflection_10.part_1'],
+		['name' => 'parted', 'view' => false, 'fullName' => 'reflection_10.parted'],
+	], $driver->getTables());
+
+	Assert::same(['id', 'value'], array_column($driver->getColumns('parted'), 'name'));
+	Assert::same(['id', 'value'], array_column($driver->getColumns('part_1'), 'name'));
+
+	Assert::same([[
+		'name' => 'parted_pkey',
+		'unique' => true,
+		'primary' => true,
+		'columns' => ['id'],
+	]], $driver->getIndexes('parted'));
+});