Method parse_set

src/sql-parser/src/parser.rs:7941–8001

(&mut self)

    }

    fn parse_set(&mut self) -> Result<Statement<Raw>, ParserStatementError> {
        let modifier = self.parse_one_of_keywords(&[SESSION, LOCAL]);
        let mut variable = self.parse_identifier().map_no_statement_parser_err()?;
        let mut normal = self.consume_token(&Token::Eq) || self.parse_keyword(TO);
        if !normal {
            match variable.as_str().parse() {
                Ok(TIME) => {
                    self.expect_keyword(ZONE).map_no_statement_parser_err()?;
                    variable = ident!("timezone");
                    normal = true;
                }
                Ok(NAMES) => {
                    variable = ident!("client_encoding");
                    normal = true;
                }
                _ => {}
            }
        }
        if variable.as_str().parse() == Ok(SCHEMA) {
            variable = ident!("search_path");
            let to = self
                .parse_set_schema_to()
                .map_parser_err(StatementKind::SetVariable)?;
            Ok(Statement::SetVariable(SetVariableStatement {
                local: modifier == Some(LOCAL),
                variable,
                to,
            }))
        } else if normal {
            let to = self
                .parse_set_variable_to()
                .map_parser_err(StatementKind::SetVariable)?;
            Ok(Statement::SetVariable(SetVariableStatement {
                local: modifier == Some(LOCAL),
                variable,
                to,
            }))
        } else if variable.as_str().parse() == Ok(TRANSACTION) && modifier.is_none() {
            // SET TRANSACTION transaction_mode
            Ok(Statement::SetTransaction(SetTransactionStatement {
                local: true,
                modes: self
                    .parse_transaction_modes(true)
                    .map_parser_err(StatementKind::SetTransaction)?,
            }))
        } else if modifier == Some(SESSION)
            && variable.as_str().parse() == Ok(CHARACTERISTICS)
            && self.parse_keywords(&[AS, TRANSACTION])
        {
            // SET SESSION CHARACTERISTICS AS TRANSACTION transaction_mode
            Ok(Statement::SetTransaction(SetTransactionStatement {
                local: false,
                modes: self
                    .parse_transaction_modes(true)
                    .map_parser_err(StatementKind::SetTransaction)?,
            }))
        } else {
            self.expected(self.peek_pos(), "equals sign or TO", self.peek_token())