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

Add support for MySQL's INSERT INTO ... SET syntax (#1641)

Browse source
This commit is contained in:
Yoav Cohen 2025-01-06 20:13:38 +01:00 • committed by GitHub
parent 4c6af0ae4f
commit 8cfc46277f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 43 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

24
src/parser/mod.rs
View file

@ -11899,9 +11899,9 @@ impl<'a> Parser<'a> {
let is_mysql = dialect_of!(self is MySqlDialect);
let (columns, partitioned, after_columns, source) =
let (columns, partitioned, after_columns, source, assignments) =
if self.parse_keywords(&[Keyword::DEFAULT, Keyword::VALUES]) {
(vec![], None, vec![], None)
(vec![], None, vec![], None, vec![])
} else {
let (columns, partitioned, after_columns) = if !self.peek_subquery_start() {
let columns = self.parse_parenthesized_column_list(Optional, is_mysql)?;
@ -11918,9 +11918,14 @@ impl<'a> Parser<'a> {
Default::default()
};
let source = Some(self.parse_query()?);
let (source, assignments) =
if self.dialect.supports_insert_set() && self.parse_keyword(Keyword::SET) {
(None, self.parse_comma_separated(Parser::parse_assignment)?)
} else {
(Some(self.parse_query()?), vec![])
};
(columns, partitioned, after_columns, source)
(columns, partitioned, after_columns, source, assignments)
};
let insert_alias = if dialect_of!(self is MySqlDialect | GenericDialect)
@ -12000,6 +12005,7 @@ impl<'a> Parser<'a> {
columns,
after_columns,
source,
assignments,
table,
on,
returning,
@ -14228,16 +14234,6 @@ mod tests {
assert!(Parser::parse_sql(&GenericDialect {}, sql).is_err());
}
#[test]
fn test_replace_into_set() {
// NOTE: This is actually valid MySQL syntax, REPLACE and INSERT,
// but the parser does not yet support it.
// https://dev.mysql.com/doc/refman/8.3/en/insert.html
let sql = "REPLACE INTO t SET a='1'";
assert!(Parser::parse_sql(&MySqlDialect {}, sql).is_err());
}
#[test]
fn test_replace_into_set_placeholder() {
let sql = "REPLACE INTO t SET ?";