Add error handler for create mutations also

graphile · Sep 22, 2017 · 1dc5e39 · 1dc5e39
1 parent 196ab27
commit 1dc5e39
Showing 1 changed file with 18 additions and 7 deletions.
diff --git a/packages/graphile-build-pg/src/plugins/PgMutationCreatePlugin.js b/packages/graphile-build-pg/src/plugins/PgMutationCreatePlugin.js
@@ -195,13 +195,24 @@ export default (function PgMutationCreatePlugin(
                       ) values(${sql.join(sqlValues, ", ")})`
                     : sql.fragment`default values`} returning *`;
 
-                  const { rows: [row] } = await viaTemporaryTable(
-                    pgClient,
-                    sql.identifier(table.namespace.name, table.name),
-                    mutationQuery,
-                    insertedRowAlias,
-                    query
-                  );
+                  let row;
+                  try {
+                    await pgClient.query("SAVEPOINT graphql_mutation");
+                    const result = await viaTemporaryTable(
+                      pgClient,
+                      sql.identifier(table.namespace.name, table.name),
+                      mutationQuery,
+                      insertedRowAlias,
+                      query
+                    );
+                    row = result.rows[0];
+                    await pgClient.query("RELEASE SAVEPOINT graphql_mutation");
+                  } catch (e) {
+                    await pgClient.query(
+                      "ROLLBACK TO SAVEPOINT graphql_mutation"
+                    );
+                    throw e;
+                  }
                   return {
                     clientMutationId: input.clientMutationId,
                     data: row,