mirror of
https://github.com/apache/datafusion-sqlparser-rs.git
synced 2025-08-29 18:34:04 +00:00
Support CREATE MATERIALIZED VIEW
This commit is contained in:
parent
52e0f55b6f
commit
23a0fc79f5
4 changed files with 26 additions and 5 deletions
|
@ -191,6 +191,7 @@ keyword!(
|
|||
LOCATION,
|
||||
LOWER,
|
||||
MATCH,
|
||||
MATERIALIZED,
|
||||
MAX,
|
||||
MEMBER,
|
||||
MERGE,
|
||||
|
@ -539,6 +540,7 @@ pub const ALL_KEYWORDS: &'static [&'static str] = &[
|
|||
LOCATION,
|
||||
LOWER,
|
||||
MATCH,
|
||||
MATERIALIZED,
|
||||
MAX,
|
||||
MEMBER,
|
||||
MERGE,
|
||||
|
|
|
@ -241,6 +241,7 @@ pub enum SQLStatement {
|
|||
/// View name
|
||||
name: SQLObjectName,
|
||||
query: SQLQuery,
|
||||
materialized: bool,
|
||||
},
|
||||
/// CREATE TABLE
|
||||
SQLCreateTable {
|
||||
|
@ -347,8 +348,9 @@ impl ToString for SQLStatement {
|
|||
}
|
||||
s
|
||||
}
|
||||
SQLStatement::SQLCreateView { name, query } => {
|
||||
format!("CREATE VIEW {} AS {}", name.to_string(), query.to_string())
|
||||
SQLStatement::SQLCreateView { name, query, materialized } => {
|
||||
let modifier = if *materialized { " MATERIALIZED" } else { "" };
|
||||
format!("CREATE{} VIEW {} AS {}", modifier, name.to_string(), query.to_string())
|
||||
}
|
||||
SQLStatement::SQLCreateTable { name, columns } => format!(
|
||||
"CREATE TABLE {} ({})",
|
||||
|
|
|
@ -617,7 +617,8 @@ impl Parser {
|
|||
pub fn parse_create(&mut self) -> Result<SQLStatement, ParserError> {
|
||||
if self.parse_keyword("TABLE") {
|
||||
self.parse_create_table()
|
||||
} else if self.parse_keyword("VIEW") {
|
||||
} else if self.parse_keyword("MATERIALIZED") || self.parse_keyword("VIEW") {
|
||||
self.prev_token();
|
||||
self.parse_create_view()
|
||||
} else {
|
||||
parser_err!(format!(
|
||||
|
@ -628,6 +629,8 @@ impl Parser {
|
|||
}
|
||||
|
||||
pub fn parse_create_view(&mut self) -> Result<SQLStatement, ParserError> {
|
||||
let materialized = self.parse_keyword("MATERIALIZED");
|
||||
self.expect_keyword("VIEW")?;
|
||||
// Many dialects support `OR REPLACE` | `OR ALTER` right after `CREATE`, but we don't (yet).
|
||||
// ANSI SQL and Postgres support RECURSIVE here, but we don't support it either.
|
||||
let name = self.parse_object_name()?;
|
||||
|
@ -637,7 +640,7 @@ impl Parser {
|
|||
self.expect_keyword("AS")?;
|
||||
let query = self.parse_query()?;
|
||||
// Optional `WITH [ CASCADED | LOCAL ] CHECK OPTION` is widely supported here.
|
||||
Ok(SQLStatement::SQLCreateView { name, query })
|
||||
Ok(SQLStatement::SQLCreateView { name, query, materialized })
|
||||
}
|
||||
|
||||
pub fn parse_create_table(&mut self) -> Result<SQLStatement, ParserError> {
|
||||
|
|
|
@ -901,9 +901,23 @@ fn parse_scalar_subqueries() {
|
|||
fn parse_create_view() {
|
||||
let sql = "CREATE VIEW myschema.myview AS SELECT foo FROM bar";
|
||||
match verified_stmt(sql) {
|
||||
SQLStatement::SQLCreateView { name, query } => {
|
||||
SQLStatement::SQLCreateView { name, query, materialized } => {
|
||||
assert_eq!("myschema.myview", name.to_string());
|
||||
assert_eq!("SELECT foo FROM bar", query.to_string());
|
||||
assert!(!materialized);
|
||||
}
|
||||
_ => assert!(false),
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn parse_create_materialized_view() {
|
||||
let sql = "CREATE MATERIALIZED VIEW myschema.myview AS SELECT foo FROM bar";
|
||||
match verified_stmt(sql) {
|
||||
SQLStatement::SQLCreateView { name, query, materialized } => {
|
||||
assert_eq!("myschema.myview", name.to_string());
|
||||
assert_eq!("SELECT foo FROM bar", query.to_string());
|
||||
assert!(materialized);
|
||||
}
|
||||
_ => assert!(false),
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue