Remove PostgreSQL version of assert (#229)

Remove PostgreSQL procedural assert statement. This also simplifies code somewhat.
This commit is contained in:
Daniël Heres 2020-07-17 13:20:49 +02:00 committed by GitHub
parent c24b0e01db
commit 583f22b929
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 22 deletions

View file

@ -555,8 +555,6 @@ pub enum Statement {
/// ASSERT <condition> [AS <message>]
Assert {
condition: Expr,
// AS or ,
separator: String,
message: Option<Expr>,
},
}
@ -818,15 +816,11 @@ impl fmt::Display for Statement {
write!(f, "ROLLBACK{}", if *chain { " AND CHAIN" } else { "" },)
}
Statement::CreateSchema { schema_name } => write!(f, "CREATE SCHEMA {}", schema_name),
Statement::Assert {
condition,
separator,
message,
} => {
Statement::Assert { condition, message } => {
write!(f, "ASSERT {}", condition)?;
if let Some(m) = message {
write!(f, " {} {}", separator, m)?;
write!(f, " AS {}", m)?;
}
Ok(())
}

View file

@ -182,19 +182,13 @@ impl Parser {
}
pub fn parse_assert(&mut self) -> Result<Statement, ParserError> {
let condition = self.parse_expr()?;
let (separator, message) = if self.consume_token(&Token::Comma) {
(",".to_string(), Some(self.parse_expr()?))
} else if self.parse_keyword(Keyword::AS) {
("AS".to_string(), Some(self.parse_expr()?))
let message = if self.parse_keyword(Keyword::AS) {
Some(self.parse_expr()?)
} else {
("".to_string(), None)
None
};
Ok(Statement::Assert {
condition,
separator,
message,
})
Ok(Statement::Assert { condition, message })
}
/// Parse an expression prefix

View file

@ -1163,11 +1163,9 @@ fn parse_assert() {
match ast {
Statement::Assert {
condition: _condition,
separator,
message,
} => {
assert_eq!(message, None);
assert_eq!(separator, "");
}
_ => unreachable!(),
}
@ -1184,9 +1182,7 @@ fn parse_assert_message() {
Statement::Assert {
condition: _condition,
message: Some(message),
separator,
} => {
assert_eq!(separator, "AS");
match message {
Expr::Value(Value::SingleQuotedString(s)) => assert_eq!(s, "No rows in table"),
_ => unreachable!(),