Skip to content

Commit d71d56a

Browse files
committed
Refactor grantor/granted identifiers to avoid unwrap()
1 parent 43a84f7 commit d71d56a

File tree

1 file changed

+14
-12
lines changed

1 file changed

+14
-12
lines changed

src/parser/mod.rs

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -13393,16 +13393,14 @@ impl<'a> Parser<'a> {
1339313393
let with_grant_option =
1339413394
self.parse_keywords(&[Keyword::WITH, Keyword::GRANT, Keyword::OPTION]);
1339513395

13396-
let as_grantor = if self.peek_keyword(Keyword::AS) {
13397-
self.parse_keywords(&[Keyword::AS])
13398-
.then(|| self.parse_identifier().unwrap())
13396+
let as_grantor = if self.parse_keywords(&[Keyword::AS]) {
13397+
Some(self.parse_identifier()?)
1339913398
} else {
1340013399
None
1340113400
};
1340213401

13403-
let granted_by = if self.peek_keywords(&[Keyword::GRANTED, Keyword::BY]) {
13404-
self.parse_keywords(&[Keyword::GRANTED, Keyword::BY])
13405-
.then(|| self.parse_identifier().unwrap())
13402+
let granted_by = if self.parse_keywords(&[Keyword::GRANTED, Keyword::BY]) {
13403+
Some(self.parse_identifier()?)
1340613404
} else {
1340713405
None
1340813406
};
@@ -13842,9 +13840,11 @@ impl<'a> Parser<'a> {
1384213840
self.expect_keyword_is(Keyword::TO)?;
1384313841
let grantees = self.parse_grantees()?;
1384413842
let cascade = self.parse_cascade_option();
13845-
let granted_by = self
13846-
.parse_keywords(&[Keyword::AS])
13847-
.then(|| self.parse_identifier().unwrap());
13843+
let granted_by = if self.parse_keywords(&[Keyword::AS]) {
13844+
Some(self.parse_identifier()?)
13845+
} else {
13846+
None
13847+
};
1384813848

1384913849
Ok(Statement::Deny(DenyStatement {
1385013850
privileges,
@@ -13862,9 +13862,11 @@ impl<'a> Parser<'a> {
1386213862
self.expect_keyword_is(Keyword::FROM)?;
1386313863
let grantees = self.parse_grantees()?;
1386413864

13865-
let granted_by = self
13866-
.parse_keywords(&[Keyword::GRANTED, Keyword::BY])
13867-
.then(|| self.parse_identifier().unwrap());
13865+
let granted_by = if self.parse_keywords(&[Keyword::GRANTED, Keyword::BY]) {
13866+
Some(self.parse_identifier()?)
13867+
} else {
13868+
None
13869+
};
1386813870

1386913871
let cascade = self.parse_cascade_option();
1387013872

0 commit comments

Comments
 (0)