Support SEQUENCE NAME

Fixes #107

Author: nene

src/cst/Sequence.ts

@@ -43,7 +43,8 @@ export type SequenceOption =
   | SequenceOptionCache
   | SequenceOptionCycle
   | SequenceOptionNoCycle
-  | SequenceOptionOwnedBy;
+  | SequenceOptionOwnedBy
+  | SequenceOptionSequenceName;
 
 export interface SequenceOptionAsType extends BaseNode {
   type: "sequence_option_as_type";
@@ -116,6 +117,12 @@ export interface SequenceOptionOwnedBy extends BaseNode {
   owner: EntityName;
 }
 
+export interface SequenceOptionSequenceName extends BaseNode {
+  type: "sequence_option_sequence_name";
+  sequenceNameKw: [Keyword<"SEQUENCE">, Keyword<"NAME">];
+  name: EntityName;
+}
+
 export interface AlterSequenceStmt extends BaseNode {
   type: "alter_sequence_stmt";
   alterKw: Keyword<"ALTER">;

src/parser.pegjs

@@ -4199,6 +4199,7 @@ sequence_option
   / sequence_option_cycle
   / sequence_option_no_cycle
   / sequence_option_owned_by
+  / sequence_option_sequence_name
 
 sequence_option_as_type
   = kw:(AS __) type:data_type {
@@ -4277,6 +4278,11 @@ sequence_option_owned_by
     return loc({ type: "sequence_option_owned_by", ownedByKw: read(kw), owner });
   }
 
+sequence_option_sequence_name
+  = kw:(SEQUENCE __ NAME __) name:entity_name {
+    return loc({ type: "sequence_option_sequence_name", sequenceNameKw: read(kw), name });
+  }
+
 alter_sequence_stmt
   = kw:(ALTER __)
     sequenceKw:(SEQUENCE __)
@@ -9047,6 +9053,7 @@ MODE                = kw:"MODE"i                !ident_part { return loc(createK
 MODULUS             = kw:"MODULUS"i             !ident_part { return loc(createKeyword(kw)); }
 MONDAY              = kw:"MONDAY"i              !ident_part { return loc(createKeyword(kw)); }
 MONTH               = kw:"MONTH"i               !ident_part { return loc(createKeyword(kw)); }
+NAME                = kw:"NAME"i                !ident_part { return loc(createKeyword(kw)); }
 NATIVE              = kw:"NATIVE"i              !ident_part { return loc(createKeyword(kw)); }
 NATURAL             = kw:"NATURAL"i             !ident_part { return loc(createKeyword(kw)); }
 NCHAR               = kw:"NCHAR"i               !ident_part { return loc(createKeyword(kw)); }

src/showNode/sequence.ts

@@ -29,6 +29,8 @@ export const sequenceMap: FullTransformMap<string, AllSequenceNodes> = {
   sequence_option_cycle: (node) => show(node.cycleKw),
   sequence_option_no_cycle: (node) => show(node.noCycleKw),
   sequence_option_owned_by: (node) => show([node.ownedByKw, node.owner]),
+  sequence_option_sequence_name: (node) =>
+    show([node.sequenceNameKw, node.name]),
 
   // ALTER SEQUENCE
   alter_sequence_stmt: (node) =>

test/ddl/alter_table.test.ts

@@ -234,6 +234,12 @@ describe("alter table", () => {
           "ALTER COLUMN foo ADD GENERATED ALWAYS AS IDENTITY (START WITH 10 INCREMENT BY 5)"
         );
       });
+      // SEQUENCE NAME is not among the general sequence options
+      it("supports ADD GENERATED .. AS IDENTITY (SEQUENCE NAME ...)", () => {
+        testAlterWc(
+          "ALTER COLUMN foo ADD GENERATED ALWAYS AS IDENTITY (SEQUENCE NAME my_seq START WITH 10)"
+        );
+      });
 
       describe("alter identity column", () => {
         it("supports SET GENERATED", () => {

test/ddl/constraints_column.test.ts

@@ -183,6 +183,11 @@ describe("column constraints", () => {
         // For all sequence options, see test/ddl/sequence.test.ts
         testColConstWc("GENERATED ALWAYS AS IDENTITY (START WITH 10 INCREMENT BY 5)");
       });
+
+      // SEQUENCE NAME is not among the general sequence options
+      it("GENERATED .. AS IDENTITY (SEQUENCE NAME ...)", () => {
+        testColConstWc("GENERATED ALWAYS AS IDENTITY (START WITH 10 SEQUENCE NAME schm.my_seq)");
+      });
     });
   });
 

test/ddl/sequence.test.ts

@@ -12,6 +12,8 @@ describe("sequence", () => {
       CACHE: ["CACHE 10", "CACHE -1"], // Negative makes no sense, but Postgres allows it
       CYCLE: ["CYCLE", "NO CYCLE"],
       "OWNED BY": ["OWNED BY blah", "OWNED BY foo.bar", "OWNED BY NONE"],
+      // Not testing SEQUENCE NAME here (it's not supported in CREATE SEQUENCE and ALTER SEQUENCE)
+      // only inside ALTER TABLE / CREATE TABLE
     };
 
     describe("CREATE SEQUENCE", () => {