NetteObjectFieldsTypeProvider: used finally block for removing visite…

…d field
nette-intellij · Jul 12, 2015 · 860969d · 860969d
1 parent 9d29a15
commit 860969d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/cz/juzna/intellij/nette/typeProvider/NetteObjectFieldsTypeProvider.java b/src/cz/juzna/intellij/nette/typeProvider/NetteObjectFieldsTypeProvider.java
@@ -11,11 +11,9 @@
 import cz.juzna.intellij.nette.utils.MagicFieldsUtil;
 import org.jetbrains.annotations.Nullable;
 
-import java.lang.reflect.Field;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.HashSet;
-import java.util.regex.Pattern;
 
 
 public class NetteObjectFieldsTypeProvider implements PhpTypeProvider2 {
@@ -45,8 +43,12 @@ public String getType(PsiElement e) {
 			return null;
 		}
 		visited.add(field);
-		Collection<Method> getters = MagicFieldsUtil.findGetters(field);
-		visited.remove(field);
+		Collection<Method> getters;
+		try {
+			getters = MagicFieldsUtil.findGetters(field);
+		} finally {
+			visited.remove(field);
+		}
 		if (getters.isEmpty()) {
 			return null;
 		}