Add support for == operator for Sqlite (apache#970)

marhoily · web-flow · commit 71c35d4dfddc · 2023-09-19T21:31:11.000-04:00
diff --git a/src/tokenizer.rs b/src/tokenizer.rs
@@ -885,6 +885,7 @@ impl<'a> Tokenizer<'a> {
                     chars.next(); // consume
                     match chars.peek() {
                         Some('>') => self.consume_and_return(chars, Token::RArrow),
+                        Some('=') => self.consume_and_return(chars, Token::DoubleEq),
                         _ => Ok(Some(Token::Eq)),
                     }
                 }
diff --git a/tests/sqlparser_sqlite.rs b/tests/sqlparser_sqlite.rs
@@ -61,6 +61,14 @@ fn parse_create_virtual_table() {
     sqlite_and_generic().verified_stmt(sql);
 }
 
+#[test]
+fn double_equality_operator() {
+    // Sqlite supports this operator: https://www.sqlite.org/lang_expr.html#binaryops
+    let input = "SELECT a==b FROM t";
+    let expected = "SELECT a = b FROM t";
+    let _ = sqlite_and_generic().one_statement_parses_to(input, expected);
+}
+
 #[test]
 fn parse_create_table_auto_increment() {
     let sql = "CREATE TABLE foo (bar INT PRIMARY KEY AUTOINCREMENT)";

Original file line number	Diff line number	Diff line change
`@@ -885,6 +885,7 @@ impl<'a> Tokenizer<'a> {`
`885`	`885`	`chars.next(); // consume`
`886`	`886`	`match chars.peek() {`
`887`	`887`	`Some('>') => self.consume_and_return(chars, Token::RArrow),`
	`888`	`+ Some('=') => self.consume_and_return(chars, Token::DoubleEq),`
`888`	`889`	`_ => Ok(Some(Token::Eq)),`
`889`	`890`	`}`
`890`	`891`	`}`