Format CREATE SUBSCRIPTION

Author: nene

src/syntax/subscription.ts

@@ -0,0 +1,15 @@
+import { AllSubscriptionNodes } from "sql-parser-cst";
+import { group, indent, line, join } from "../print_utils";
+import { CstToDocMap } from "../CstToDocMap";
+
+export const subscriptionMap: Partial<CstToDocMap<AllSubscriptionNodes>> = {
+  create_subscription_stmt: (print, node) =>
+    group(
+      join(line, [
+        print.spaced(["createSubscriptionKw", "name"]),
+        print.spaced(["connectionKw", "connectionInfo"]),
+        print.spaced(["publicationKw", "publications"]),
+        ...(node.with ? [print.spaced(["with"])] : []),
+      ]),
+    ),
+};

src/syntax/transformMap.ts

@@ -37,6 +37,7 @@ import { publicationMap } from "./publication";
 import { schemaMap } from "./schema";
 import { selectMap } from "./select";
 import { sequenceMap } from "./sequence";
+import { subscriptionMap } from "./subscription";
 import { transactionMap } from "./transaction";
 import { triggerMap } from "./trigger";
 import { truncateMap } from "./truncate";
@@ -88,6 +89,7 @@ export const transformMap: Partial<CstToDocMap<Node>> = {
   ...schemaMap,
   ...selectMap,
   ...sequenceMap,
+  ...subscriptionMap,
   ...transactionMap,
   ...triggerMap,
   ...truncateMap,

test/postgresql/subscription.test.ts

@@ -0,0 +1,37 @@
+import dedent from "dedent-js";
+import { testPostgresql } from "../test_utils";
+
+describe("subscriptions", () => {
+  describe("CREATE SUBSCRIPTION", () => {
+    it(`formats CREATE SUBSCRIPTION to single line if fits`, async () => {
+      await testPostgresql(dedent`
+        CREATE SUBSCRIPTION my_sub CONNECTION 'con' PUBLICATION my_pub
+      `);
+    });
+
+    it(`formats CREATE SUBSCRIPTION to multiple lines`, async () => {
+      await testPostgresql(dedent`
+        CREATE SUBSCRIPTION my_subscription
+        CONNECTION 'host=192.168.1.50 port=5432 user=foo dbname=foodb'
+        PUBLICATION my_publication
+      `);
+    });
+
+    it(`formats CREATE SUBSCRIPTION to multiple lines`, async () => {
+      await testPostgresql(dedent`
+        CREATE SUBSCRIPTION my_subscription
+        CONNECTION 'host=192.168.1.50 port=5432 user=foo dbname=foodb'
+        PUBLICATION my_publication
+      `);
+    });
+
+    it(`formats WITH clause`, async () => {
+      await testPostgresql(dedent`
+        CREATE SUBSCRIPTION my_subscription
+        CONNECTION 'host=192.168.1.50 port=5432 user=foo dbname=foodb'
+        PUBLICATION my_publication
+        WITH (param1 = 1, param2 = 2)
+      `);
+    });
+  });
+});