Commit c8eb903c authored by Tagir Valeev's avatar Tagir Valeev
Browse files

IteratorDeclaration: support parentheses in declaration

parent b892f425
Showing with 2 additions and 2 deletions
+2 -2
...@@ -102,7 +102,7 @@ public class IteratorDeclaration { ...@@ -102,7 +102,7 @@ public class IteratorDeclaration {
PsiElement element = declaration.getDeclaredElements()[0]; PsiElement element = declaration.getDeclaredElements()[0];
if (!(element instanceof PsiLocalVariable)) return null; if (!(element instanceof PsiLocalVariable)) return null;
PsiLocalVariable variable = (PsiLocalVariable)element; PsiLocalVariable variable = (PsiLocalVariable)element;
PsiExpression initializer = variable.getInitializer(); PsiExpression initializer = PsiUtil.skipParenthesizedExprDown(variable.getInitializer());
if (!(initializer instanceof PsiMethodCallExpression)) return null; if (!(initializer instanceof PsiMethodCallExpression)) return null;
PsiMethodCallExpression call = (PsiMethodCallExpression)initializer; PsiMethodCallExpression call = (PsiMethodCallExpression)initializer;
if (!call.getArgumentList().isEmpty()) return null; if (!call.getArgumentList().isEmpty()) return null;
......
...@@ -4,7 +4,7 @@ import java.util.List; ...@@ -4,7 +4,7 @@ import java.util.List;
public class Main { public class Main {
public void testIterator(List<List<String>> data, boolean b) { public void testIterator(List<List<String>> data, boolean b) {
for(Ite<caret>rator<List<String>> iter = data.iterator(); iter.hasNext();) { for(Ite<caret>rator<List<String>> iter = (data.iterator()); iter.hasNext();) {
if(iter.next().isEmpty() && /* also check the flag */ b) { if(iter.next().isEmpty() && /* also check the flag */ b) {
iter.remove(); iter.remove();
} }
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment