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

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

@@ -657,6 +657,7 @@ export default (async function PgIntrospectionPlugin(
     pgIgnoreRBAC = true,
     pgSkipInstallingWatchFixtures = false,
     pgOwnerConnectionString,
+    pgUsePartitionedParent = false,
   }
 ) {
   /**
@@ -681,6 +682,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,
       }
     );
   });

packages/postgraphile-core/__tests__/integration/schema/partitioned.test.js

@@ -0,0 +1,14 @@
+const core = require("./core");
+
+test(
+  "prints a schema to test partitioned table-specific features ignoring partition parents and using partitions",
+  core.test(__filename, ["partitioned"], {
+    usePartitionedParent: false,
+  })
+);
+test(
+  "prints a schema to test partitioned table-specific features using partition parents and ignoring partitions",
+  core.test(__filename, ["partitioned"], {
+    usePartitionedParent: true,
+  })
+);

packages/postgraphile-core/__tests__/mutations/v4-base/partitioned.use-partitioned-children.mutations.json5

@@ -0,0 +1,59 @@
+{
+  createWarehouse: {
+    warehouse: {
+      id: 1000,
+      location: "Memphis, TN",
+    },
+  },
+  createItem2: {
+    item2: {
+      id: 2000,
+      description: "Laptop",
+    },
+  },
+  createStock4: {
+    stock4: {
+      itemId: 2000,
+      amount: 10,
+      warehouseId: 1000,
+    },
+  },
+  updateWarehouseById: {
+    warehouse: {
+      id: 1000,
+      location: "Nashville, TN",
+    },
+  },
+  updateItem2ById: {
+    item2: {
+      id: 2000,
+      description: "TV",
+    },
+  },
+  updateStock4ByItemIdAndWarehouseId: {
+    stock4: {
+      itemId: 2000,
+      amount: 2,
+      warehouseId: 1000,
+    },
+  },
+  deleteStock4ByItemIdAndWarehouseId: {
+    stock4: {
+      itemId: 2000,
+      amount: 2,
+      warehouseId: 1000,
+    },
+  },
+  deleteWarehouseById: {
+    warehouse: {
+      id: 1000,
+      location: "Nashville, TN",
+    },
+  },
+  deleteItem2ById: {
+    item2: {
+      id: 2000,
+      description: "TV",
+    },
+  },
+}