mirror of
https://github.com/apache/datafusion-sqlparser-rs.git
synced 2025-09-19 20:29:47 +00:00
[mssql] Support single-quoted column aliases
This commit is contained in:
parent
d0a782d8cc
commit
93c9000102
3 changed files with 30 additions and 7 deletions
|
@ -1167,15 +1167,18 @@ impl Parser {
|
||||||
{
|
{
|
||||||
Ok(Some(w.as_sql_ident()))
|
Ok(Some(w.as_sql_ident()))
|
||||||
}
|
}
|
||||||
ref not_an_ident if after_as => parser_err!(format!(
|
// MSSQL supports single-quoted strings as aliases for columns
|
||||||
"Expected an identifier after AS, got {:?}",
|
// We accept them as table aliases too, although MSSQL does not.
|
||||||
not_an_ident
|
Some(Token::SingleQuotedString(ref s)) => Ok(Some(format!("'{}'", s))),
|
||||||
)),
|
not_an_ident => {
|
||||||
Some(_not_an_ident) => {
|
if after_as {
|
||||||
self.prev_token();
|
return self.expected("an identifier after AS", not_an_ident);
|
||||||
|
}
|
||||||
|
if not_an_ident.is_some() {
|
||||||
|
self.prev_token();
|
||||||
|
}
|
||||||
Ok(None) // no alias found
|
Ok(None) // no alias found
|
||||||
}
|
}
|
||||||
None => Ok(None),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -212,6 +212,21 @@ fn parse_column_aliases() {
|
||||||
one_statement_parses_to("SELECT a.col + 1 newname FROM foo AS a", &sql);
|
one_statement_parses_to("SELECT a.col + 1 newname FROM foo AS a", &sql);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_eof_after_as() {
|
||||||
|
let res = parse_sql_statements("SELECT foo AS");
|
||||||
|
assert_eq!(
|
||||||
|
ParserError::ParserError("Expected an identifier after AS, found: EOF".to_string()),
|
||||||
|
res.unwrap_err()
|
||||||
|
);
|
||||||
|
|
||||||
|
let res = parse_sql_statements("SELECT 1 FROM foo AS");
|
||||||
|
assert_eq!(
|
||||||
|
ParserError::ParserError("Expected an identifier after AS, found: EOF".to_string()),
|
||||||
|
res.unwrap_err()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn parse_select_count_wildcard() {
|
fn parse_select_count_wildcard() {
|
||||||
let sql = "SELECT COUNT(*) FROM customer";
|
let sql = "SELECT COUNT(*) FROM customer";
|
||||||
|
|
|
@ -27,6 +27,11 @@ fn parse_mssql_identifiers() {
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn parse_mssql_single_quoted_aliases() {
|
||||||
|
let _ = ms_and_generic().one_statement_parses_to("SELECT foo 'alias'", "SELECT foo AS 'alias'");
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn parse_mssql_delimited_identifiers() {
|
fn parse_mssql_delimited_identifiers() {
|
||||||
let _ = ms().one_statement_parses_to(
|
let _ = ms().one_statement_parses_to(
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue