feat(pg): Adds support for using partitioned table parents (#801)

Co-authored-by: Benjie <[email protected]>

Author: msjonker

packages/graphile-build-pg/src/plugins/PgIntrospectionPlugin.js

@@ -681,6 +681,7 @@ export default (async function PgIntrospectionPlugin(
     pgIgnoreRBAC = true,
     pgSkipInstallingWatchFixtures = false,
     pgOwnerConnectionString,
+    pgUsePartitionedParent = false,
   }
 ) {
   /**
@@ -705,6 +706,7 @@ export default (async function PgIntrospectionPlugin(
           const introspectionQuery = makeIntrospectionQuery(serverVersionNum, {
             pgLegacyFunctionsOnly,
             pgIgnoreRBAC,
+            pgUsePartitionedParent,
           });
           const { rows } = await pgClient.query(introspectionQuery, [
             schemas,

packages/graphile-build-pg/src/plugins/introspectionQuery.js

@@ -12,9 +12,14 @@ function makeIntrospectionQuery(
   options: {
     pgLegacyFunctionsOnly?: boolean,
     pgIgnoreRBAC?: boolean,
+    pgUsePartitionedParent?: boolean,
   } = {}
 ): string {
-  const { pgLegacyFunctionsOnly = false, pgIgnoreRBAC = true } = options;
+  const {
+    pgLegacyFunctionsOnly = false,
+    pgIgnoreRBAC = true,
+    pgUsePartitionedParent = false,
+  } = options;
   const unionRBAC = `
     union all
       select pg_roles.oid _oid, pg_roles.*
@@ -187,7 +192,11 @@ with
       rel.relpersistence in ('p') and
       -- We don't want classes that will clash with GraphQL (treat them as private)
       rel.relname not like E'\\\\_\\\\_%' and
-      rel.relkind in ('r', 'v', 'm', 'c', 'f') and
+      ${
+        pgUsePartitionedParent
+          ? "rel.relkind in ('r', 'v', 'm', 'c', 'f', 'p') and not rel.relispartition"
+          : "rel.relkind in ('r', 'v', 'm', 'c', 'f')"
+      } and
       ($2 is true or not exists(
         select 1
         from pg_catalog.pg_depend

packages/postgraphile-core/__tests__/helpers-v5.js

@@ -471,6 +471,7 @@ const makeSchema = config => {
           ExtendedPlugin,
           config.ToyCategoriesPlugin ? ToyCategoriesPlugin : null,
         ].filter(isNotNullish),
+        usePartitionedParent: config.usePartitionedParent,
       }
     );
   });