mirror of
https://github.com/apache/datafusion-sqlparser-rs.git
synced 2025-09-09 15:40:32 +00:00
Add support for escaping single quote strings
This commit is contained in:
parent
5b464e6b1a
commit
80dccf6885
3 changed files with 39 additions and 2 deletions
|
@ -35,7 +35,7 @@ impl ToString for Value {
|
|||
Value::Long(v) => v.to_string(),
|
||||
Value::Double(v) => v.to_string(),
|
||||
Value::Uuid(v) => v.to_string(),
|
||||
Value::SingleQuotedString(v) => format!("'{}'", v),
|
||||
Value::SingleQuotedString(v) => format!("'{}'", escape_single_quote_string(v)),
|
||||
Value::NationalStringLiteral(v) => format!("N'{}'", v),
|
||||
Value::Boolean(v) => v.to_string(),
|
||||
Value::Date(v) => v.to_string(),
|
||||
|
@ -46,3 +46,15 @@ impl ToString for Value {
|
|||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn escape_single_quote_string(s: &str) -> String {
|
||||
let mut escaped = String::new();
|
||||
for c in s.chars() {
|
||||
if c == '\'' {
|
||||
escaped.push_str("\'\'");
|
||||
} else {
|
||||
escaped.push(c);
|
||||
}
|
||||
}
|
||||
escaped
|
||||
}
|
||||
|
|
|
@ -462,7 +462,15 @@ impl<'a> Tokenizer<'a> {
|
|||
match ch {
|
||||
'\'' => {
|
||||
chars.next(); // consume
|
||||
break;
|
||||
let escaped_quote = chars.peek()
|
||||
.map(|c| *c == '\'')
|
||||
.unwrap_or(false);
|
||||
if escaped_quote {
|
||||
s.push('\'');
|
||||
chars.next();
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
_ => {
|
||||
chars.next(); // consume
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue