review

Author: togami2864

src/tokenizer.rs

@@ -35,7 +35,7 @@ use serde::{Deserialize, Serialize};
 use sqlparser_derive::{Visit, VisitMut};
 
 use crate::ast::DollarQuotedString;
-use crate::dialect::SnowflakeDialect;
+use crate::dialect::{BigQueryDialect, GenericDialect, SnowflakeDialect};
 use crate::dialect::{Dialect, MySqlDialect};
 use crate::keywords::{Keyword, ALL_KEYWORDS, ALL_KEYWORDS_INDEX};
 
@@ -500,7 +500,7 @@ impl<'a> Tokenizer<'a> {
                     Ok(Some(Token::Whitespace(Whitespace::Newline)))
                 }
                 // BigQuery uses b or B for byte string literal
-                b @ 'B' | b @ 'b' => {
+                b @ 'B' | b @ 'b' if dialect_of!(self is BigQueryDialect | GenericDialect) => {
                     chars.next(); // consume
                     match chars.peek() {
                         Some('\'') => {

tests/sqlparser_bigquery.rs

@@ -46,6 +46,9 @@ fn parse_byte_literal() {
         &Expr::Value(Value::DoubleQuotedByteStringLiteral("abc".to_string())),
         expr_from_projection(&select.projection[1])
     );
+
+    let sql = r#"SELECT b'abc', b"abc""#;
+    bigquery().one_statement_parses_to(sql, r#"SELECT B'abc', B"abc""#);
 }
 
 #[test]