improve

databendlabs · May 14, 2022 · 340e214 · 340e214
1 parent 20271a8
commit 340e214
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/common/ast/src/parser/util.rs b/common/ast/src/parser/util.rs
@@ -95,17 +95,17 @@ fn non_reserved_identifier(
             ),
             move |i| {
                 match_token(QuotedString)(i).and_then(|(i2, token)| {
-                    if token.kind == QuotedString && !token.text().starts_with('\'') {
+                    if token.text().starts_with('\'') {
+                        Err(nom::Err::Error(Error::from_error_kind(
+                            i,
+                            ErrorKind::ExpectToken(Ident),
+                        )))
+                    } else {
                         Ok((i2, Identifier {
                             span: token.clone(),
                             name: token.text()[1..token.text().len() - 1].to_string(),
                             quote: Some(token.text().chars().next().unwrap()),
                         }))
-                    } else {
-                        Err(nom::Err::Error(Error::from_error_kind(
-                            i,
-                            ErrorKind::ExpectToken(Ident),
-                        )))
                     }
                 })
             },
@@ -131,13 +131,13 @@ fn non_reserved_keyword(
 
 pub fn literal_string(i: Input) -> IResult<String> {
     match_token(QuotedString)(i).and_then(|(i2, token)| {
-        if token.kind == QuotedString && !token.text().starts_with('`') {
-            Ok((i2, token.text()[1..token.text().len() - 1].to_string()))
-        } else {
+        if token.text().starts_with('`') {
             Err(nom::Err::Error(Error::from_error_kind(
                 i,
                 ErrorKind::ExpectToken(QuotedString),
             )))
+        } else {
+            Ok((i2, token.text()[1..token.text().len() - 1].to_string()))
         }
     })
 }