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

Support different quoting styles for delimited identifiers

Browse source 
The dialect information is from https://en.wikibooks.org/wiki/SQL_Dialects_Reference/Data_structure_definition/Delimited_identifiers
This commit is contained in:
Nickolay Ponomarev 2019-01-31 13:36:40 +03:00
parent b3693bfa63
commit b9f4b503b6
2 changed files with 27 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/sqltokenizer.rs
View file

@ -163,13 +163,24 @@ pub struct SQLWord {
impl ToString for SQLWord {
fn to_string(&self) -> String {
match self.quote_style {
Some('"') => format!("\"{}\"", self.value),
Some('[') => format!("[{}]", self.value),
Some(s) if s == '"' || s == '[' || s == '`' => {
format!("{}{}{}", s, self.value, SQLWord::matching_end_quote(s))
}
None => self.value.clone(),
_ => panic!("Unexpected quote_style!"),
}
}
}
impl SQLWord {
fn matching_end_quote(ch: char) -> char {
match ch {
'"' => '"', // ANSI and most dialects
'[' => ']', // MS SQL
'`' => '`', // MySQL
_ => panic!("unexpected quoting style!"),
}
}
}
#[derive(Debug, Clone, PartialEq)]
pub enum Whitespace {
@ -291,12 +302,13 @@ impl<'a> Tokenizer<'a> {
Ok(Some(Token::SingleQuotedString(s)))
}
// delimited (quoted) identifier
'"' => {
quote_start if self.dialect.is_delimited_identifier_start(quote_start) => {
let mut s = String::new();
let quote_start = chars.next().unwrap(); // consumes the opening quote
chars.next(); // consume the opening quote
let quote_end = SQLWord::matching_end_quote(quote_start);
while let Some(ch) = chars.next() {
match ch {
'"' => break,
c if c == quote_end => break,
_ => s.push(ch),
}
}