[clang] 9cf4b72 - [clang-format] Refactor common handling of attributes. NFC.

[Marek Kurdej via cfe-commits]

Author: Marek Kurdej
Date: 2021-12-17T23:02:45+01:00
New Revision: 9cf4b7266bbf238e4a5bd85c8d0011aac2f9181b

URL: https://github.com/llvm/llvm-project/commit/9cf4b7266bbf238e4a5bd85c8d0011aac2f9181b
DIFF: https://github.com/llvm/llvm-project/commit/9cf4b7266bbf238e4a5bd85c8d0011aac2f9181b.diff

LOG: [clang-format] Refactor common handling of attributes. NFC.

Reviewed By: MyDeveloperDay

Differential Revision: https://reviews.llvm.org/D115968

Added: 
    

Modified: 
    clang/lib/Format/UnwrappedLineParser.cpp

Removed: 
    


################################################################################
diff  --git a/clang/lib/Format/UnwrappedLineParser.cpp b/clang/lib/Format/UnwrappedLineParser.cpp
index 0b1771ba7c9c..9ea0db812486 100644
--- a/clang/lib/Format/UnwrappedLineParser.cpp
+++ b/clang/lib/Format/UnwrappedLineParser.cpp
@@ -2159,18 +2159,22 @@ void UnwrappedLineParser::parseSquare(bool LambdaIntroducer) {
 }
 
 void UnwrappedLineParser::parseIfThenElse() {
+  auto HandleAttributes = [this]() {
+    // Handle AttributeMacro, e.g. `if (x) UNLIKELY`.
+    if (FormatTok->is(TT_AttributeMacro))
+      nextToken();
+    // Handle [[likely]] / [[unlikely]] attributes.
+    if (FormatTok->is(tok::l_square) && tryToParseSimpleAttribute())
+      parseSquare();
+  };
+
   assert(FormatTok->Tok.is(tok::kw_if) && "'if' expected");
   nextToken();
   if (FormatTok->Tok.isOneOf(tok::kw_constexpr, tok::identifier))
     nextToken();
   if (FormatTok->Tok.is(tok::l_paren))
     parseParens();
-  // handle  AttributeMacro  if (x) UNLIKELY
-  if (FormatTok->is(TT_AttributeMacro))
-    nextToken();
-  // handle [[likely]] / [[unlikely]]
-  if (FormatTok->is(tok::l_square) && tryToParseSimpleAttribute())
-    parseSquare();
+  HandleAttributes();
   bool NeedsUnwrappedLine = false;
   if (FormatTok->Tok.is(tok::l_brace)) {
     CompoundStatementIndenter Indenter(this, Style, Line->Level);
@@ -2187,12 +2191,7 @@ void UnwrappedLineParser::parseIfThenElse() {
   }
   if (FormatTok->Tok.is(tok::kw_else)) {
     nextToken();
-    // handle  AttributeMacro  else UNLIKELY
-    if (FormatTok->is(TT_AttributeMacro))
-      nextToken();
-    // handle [[likely]] / [[unlikely]]
-    if (FormatTok->Tok.is(tok::l_square) && tryToParseSimpleAttribute())
-      parseSquare();
+    HandleAttributes();
     if (FormatTok->Tok.is(tok::l_brace)) {
       CompoundStatementIndenter Indenter(this, Style, Line->Level);
       parseBlock();