Add FOR UPDATE clause support for Postgres (#574)

Author: nene

src/languages/plsql/plsql.formatter.ts

@@ -17,7 +17,7 @@ const reservedClauses = expandPhrases([
   'ORDER [SIBLINGS] BY',
   'OFFSET',
   'FETCH {FIRST | NEXT}',
-  'FOR UPDATE',
+  'FOR UPDATE [OF]',
   // Data manipulation
   // - insert:
   'INSERT [INTO | ALL INTO]',

src/languages/postgresql/postgresql.formatter.ts

@@ -18,6 +18,7 @@ const reservedClauses = expandPhrases([
   'LIMIT',
   'OFFSET',
   'FETCH {FIRST | NEXT}',
+  'FOR {UPDATE | NO KEY UPDATE | SHARE | KEY SHARE} [OF]',
   // Data manipulation
   // - insert:
   'INSERT INTO',

test/plsql.test.ts

@@ -159,8 +159,8 @@ describe('PlSqlFormatter', () => {
         *
       FROM
         tbl
-      FOR UPDATE
-        OF tbl.salary;
+      FOR UPDATE OF
+        tbl.salary;
     `);
   });
 });

test/postgresql.test.ts

@@ -262,4 +262,26 @@ describe('PostgreSqlFormatter', () => {
       DROP NOT NULL;
     `);
   });
+
+  it('formats FOR UPDATE clause', () => {
+    expect(
+      format(`
+        SELECT * FROM tbl FOR UPDATE;
+        SELECT * FROM tbl FOR UPDATE OF tbl.salary;
+      `)
+    ).toBe(dedent`
+      SELECT
+        *
+      FROM
+        tbl
+      FOR UPDATE;
+
+      SELECT
+        *
+      FROM
+        tbl
+      FOR UPDATE OF
+        tbl.salary;
+    `);
+  });
 });