alamb commented on code in PR #6355:
URL: https://github.com/apache/arrow-datafusion/pull/6355#discussion_r1198824868
##########
datafusion/sql/src/parser.rs:
##########
@@ -242,6 +334,75 @@ impl<'a> DFParser<'a> {
}))
}
+ /// Parse a SQL `COPY TO` statement
+ pub fn parse_copy(&mut self) -> Result<Statement, ParserError> {
+ // parse as a query
+ let source = if self.parser.consume_token(&Token::LParen) {
+ let query = self.parser.parse_query()?;
+ self.parser.expect_token(&Token::RParen)?;
+ CopyToSource::Query(query)
+ } else {
+ // parse as table reference
+ let table_name = self.parser.parse_object_name()?;
+ CopyToSource::Relation(table_name)
+ };
+
+ self.parser.expect_keyword(Keyword::TO)?;
+
+ let target = self.parser.parse_literal_string()?;
+
+ // check for options in parens
+ let options = if self.parser.peek_token().token == Token::LParen {
+ self.parse_value_options()?
+ } else {
+ HashMap::new()
+ };
+
+ Ok(Statement::CopyTo(CopyToStatement {
+ source,
+ target,
+ options,
+ }))
+ }
+
+ /// Parse the next token as a key name for an option list
+ ///
+ /// Note this is different than [`parse_literal_string`]
+ /// because it it allows keywords as well as other non words
+ ///
+ /// [`parse_literal_string`]:
sqlparser::parser::Parser::parse_literal_string
+ pub fn parse_option_key(&mut self) -> Result<String, ParserError> {
+ let next_token = self.parser.next_token();
+ match next_token.token {
+ Token::Word(Word { value, .. }) => Ok(value),
+ Token::SingleQuotedString(s) => Ok(s),
+ Token::DoubleQuotedString(s) => Ok(s),
+ Token::EscapedStringLiteral(s) => Ok(s),
+ _ => self.parser.expected("key name", next_token),
+ }
+ }
+
+ /// Parse the next token as a value for an option list
+ ///
+ /// Note this is different than [`parse_value`] as it allows any
+ /// word or keyword in this location.
+ ///
+ /// [`parse_value`]: sqlparser::parser::Parser::parse_value
+ pub fn parse_option_value(&mut self) -> Result<Value, ParserError> {
+ let next_token = self.parser.next_token();
+ match next_token.token {
+ Token::Word(Word { value, .. }) =>
Ok(Value::UnQuotedString(value)),
+ Token::SingleQuotedString(s) => Ok(Value::SingleQuotedString(s)),
+ Token::DoubleQuotedString(s) => Ok(Value::DoubleQuotedString(s)),
+ Token::EscapedStringLiteral(s) =>
Ok(Value::EscapedStringLiteral(s)),
+ Token::Number(ref n, l) => match n.parse() {
+ Ok(n) => Ok(Value::Number(n, l)),
+ Err(e) => parser_err!(format!("Could not parse '{n}' as
number: {e}")),
+ },
Review Comment:
I will admit to simply copy/pasting this from sqlparser code. I will change
this to "internal error" to signal that it is not expected
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
