Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions fuzz/fuzz_seeds/postgres_alter_extension.sql
Original file line number Diff line number Diff line change
Expand Up @@ -36,3 +36,4 @@ ALTER EXTENSION citext DROP OPERATOR CLASS oc USING btree;
ALTER EXTENSION citext DROP OPERATOR FAMILY of USING btree;
ALTER EXTENSION citext ADD CAST (int AS text);
ALTER EXTENSION citext ADD TRANSFORM FOR int LANGUAGE plpgsql;
ALTER EXTENSION citext DROP TYPE public.ty;
3 changes: 3 additions & 0 deletions src/ast/ddl.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5021,6 +5021,8 @@ pub enum AlterExtensionMember {
/// Language name.
language: Ident,
},
/// `TYPE object_name`
Type(DataType),
/// `VIEW object_name`
View(ObjectName),
}
Expand Down Expand Up @@ -5104,6 +5106,7 @@ impl fmt::Display for AlterExtensionMember {
type_name,
language,
} => write!(f, "TRANSFORM FOR {type_name} LANGUAGE {language}"),
AlterExtensionMember::Type(data_type) => write!(f, "TYPE {data_type}"),
AlterExtensionMember::View(name) => write!(f, "VIEW {name}"),
}
}
Expand Down
2 changes: 2 additions & 0 deletions src/parser/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -12652,6 +12652,8 @@ impl<'a> Parser<'a> {
type_name,
language,
})
} else if self.parse_keyword(Keyword::TYPE) {
Ok(AlterExtensionMember::Type(self.parse_data_type()?))
} else if self.parse_keyword(Keyword::VIEW) {
Ok(AlterExtensionMember::View(self.parse_object_name(false)?))
} else {
Expand Down
5 changes: 5 additions & 0 deletions tests/sqlparser_postgres.rs
Original file line number Diff line number Diff line change
Expand Up @@ -13584,6 +13584,7 @@ fn alter_extension_member_round_trip_forms(action: &str) {
format!("ALTER EXTENSION ext_name {action} TEXT SEARCH PARSER pgrm"),
format!("ALTER EXTENSION ext_name {action} TEXT SEARCH TEMPLATE tpl"),
format!("ALTER EXTENSION ext_name {action} TRANSFORM FOR my_type LANGUAGE plpgsql"),
format!("ALTER EXTENSION ext_name {action} TYPE my_type"),
format!("ALTER EXTENSION ext_name {action} VIEW v"),
];
for sql in sqls {
Expand Down Expand Up @@ -13911,6 +13912,10 @@ fn parse_alter_extension_add_member() {
"ALTER EXTENSION ext_name ADD TRANSFORM FOR my_type LANGUAGE",
"Expected: identifier, found: EOF",
),
(
"ALTER EXTENSION ext_name ADD TYPE",
"Expected: a data type name, found: EOF",
),
] {
assert_eq!(
pg_and_generic().parse_sql_statements(sql).unwrap_err(),
Expand Down
Loading