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

Add support for $$ in generic dialect ... (#1185)

Browse source
This commit is contained in:
Marko Milenković 2024-03-24 18:20:15 +00:00 • committed by GitHub
parent 6b03a259aa
commit 4472789171
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/parser/mod.rs
View file

@ -5954,7 +5954,8 @@ impl<'a> Parser<'a> {
pub fn parse_function_definition(&mut self) -> Result<FunctionDefinition, ParserError> { pub fn parse_function_definition(&mut self) -> Result<FunctionDefinition, ParserError> {
let peek_token = self.peek_token(); let peek_token = self.peek_token();
match peek_token.token { match peek_token.token {
Token::DollarQuotedString(value) if dialect_of!(self is PostgreSqlDialect) => { Token::DollarQuotedString(value) if dialect_of!(self is PostgreSqlDialect | GenericDialect) =>
{
self.next_token(); self.next_token();
Ok(FunctionDefinition::DoubleDollarDef(value.value)) Ok(FunctionDefinition::DoubleDollarDef(value.value))
} }

2
tests/sqlparser_postgres.rs
View file

@ -3305,7 +3305,7 @@ fn parse_create_function() {
let sql = "CREATE OR REPLACE FUNCTION add(a INTEGER, IN b INTEGER = 1) RETURNS INTEGER LANGUAGE SQL IMMUTABLE RETURN a + b"; let sql = "CREATE OR REPLACE FUNCTION add(a INTEGER, IN b INTEGER = 1) RETURNS INTEGER LANGUAGE SQL IMMUTABLE RETURN a + b";
assert_eq!( assert_eq!(
pg().verified_stmt(sql), pg_and_generic().verified_stmt(sql),
Statement::CreateFunction { Statement::CreateFunction {
or_replace: true, or_replace: true,
temporary: false, temporary: false,