mirrors/datafusion-sqlparse
1
0
Fork 0
mirror of https://github.com/apache/datafusion-sqlparser-rs.git synced 2025-07-23 08:15:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Snowflake: Support CLONE option in CREATE DATABASE/SCHEMA statements (#1958)

Browse source
This commit is contained in:
Yoav Cohen 2025-07-21 14:41:20 +03:00 • committed by GitHub
parent a73577c29f
commit 799c1f748d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 63 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

24
src/ast/mod.rs
View file

@ -3846,6 +3846,14 @@ pub enum Statement {
///
/// [BigQuery](https://cloud.google.com/bigquery/docs/reference/standard-sql/data-definition-language#create_schema_statement)
default_collate_spec: Option<Expr>,
/// Clones a schema
///
/// ```sql
/// CREATE SCHEMA myschema CLONE otherschema
/// ```
///
/// [Snowflake](https://docs.snowflake.com/en/sql-reference/sql/create-clone#databases-schemas)
clone: Option<ObjectName>,
},
/// ```sql
/// CREATE DATABASE
@ -3855,6 +3863,14 @@ pub enum Statement {
if_not_exists: bool,
location: Option<String>,
managed_location: Option<String>,
/// Clones a database
///
/// ```sql
/// CREATE DATABASE mydb CLONE otherdb
/// ```
///
/// [Snowflake](https://docs.snowflake.com/en/sql-reference/sql/create-clone#databases-schemas)
clone: Option<ObjectName>,
},
/// ```sql
/// CREATE FUNCTION
@ -4797,6 +4813,7 @@ impl fmt::Display for Statement {
if_not_exists,
location,
managed_location,
clone,
} => {
write!(f, "CREATE DATABASE")?;
if *if_not_exists {
@ -4809,6 +4826,9 @@ impl fmt::Display for Statement {
if let Some(ml) = managed_location {
write!(f, " MANAGEDLOCATION '{ml}'")?;
}
if let Some(clone) = clone {
write!(f, " CLONE {clone}")?;
}
Ok(())
}
Statement::CreateFunction(create_function) => create_function.fmt(f),
@ -5730,6 +5750,7 @@ impl fmt::Display for Statement {
with,
options,
default_collate_spec,
clone,
} => {
write!(
f,
@ -5750,6 +5771,9 @@ impl fmt::Display for Statement {
write!(f, " OPTIONS({})", display_comma_separated(options))?;
}
if let Some(clone) = clone {
write!(f, " CLONE {clone}")?;
}
Ok(())
}
Statement::Assert { condition, message } => {

14
src/parser/mod.rs
View file

@ -4917,12 +4917,19 @@ impl<'a> Parser<'a> {
None
};
let clone = if self.parse_keyword(Keyword::CLONE) {
Some(self.parse_object_name(false)?)
} else {
None
};
Ok(Statement::CreateSchema {
schema_name,
if_not_exists,
with,
options,
default_collate_spec,
clone,
})
}
@ -4957,11 +4964,18 @@ impl<'a> Parser<'a> {
_ => break,
}
}
let clone = if self.parse_keyword(Keyword::CLONE) {
Some(self.parse_object_name(false)?)
} else {
None
};
Ok(Statement::CreateDatabase {
db_name,
if_not_exists: ine,
location,
managed_location,
clone,
})
}

25
tests/sqlparser_common.rs
View file

@ -4296,6 +4296,7 @@ fn parse_create_schema() {
verified_stmt(r#"CREATE SCHEMA a.b.c WITH (key1 = 'value1', key2 = 'value2')"#);
verified_stmt(r#"CREATE SCHEMA IF NOT EXISTS a WITH (key1 = 'value1')"#);
verified_stmt(r#"CREATE SCHEMA IF NOT EXISTS a WITH ()"#);
verified_stmt(r#"CREATE SCHEMA a CLONE b"#);
}
#[test]
@ -7900,11 +7901,33 @@ fn parse_create_database() {
if_not_exists,
location,
managed_location,
clone,
} => {
assert_eq!("mydb", db_name.to_string());
assert!(!if_not_exists);
assert_eq!(None, location);
assert_eq!(None, managed_location);
assert_eq!(None, clone);
}
_ => unreachable!(),
}
let sql = "CREATE DATABASE mydb CLONE otherdb";
match verified_stmt(sql) {
Statement::CreateDatabase {
db_name,
if_not_exists,
location,
managed_location,
clone,
} => {
assert_eq!("mydb", db_name.to_string());
assert!(!if_not_exists);
assert_eq!(None, location);
assert_eq!(None, managed_location);
assert_eq!(
Some(ObjectName::from(vec![Ident::new("otherdb".to_string())])),
clone
);
}
_ => unreachable!(),
}
@ -7919,11 +7942,13 @@ fn parse_create_database_ine() {
if_not_exists,
location,
managed_location,
clone,
} => {
assert_eq!("mydb", db_name.to_string());
assert!(if_not_exists);
assert_eq!(None, location);
assert_eq!(None, managed_location);
assert_eq!(None, clone);
}
_ => unreachable!(),
}