RecordAugmentProvider: fix possible IOOBE

GitOrigin-RevId: eb1c889150c81f53e303a38c8ca7a6c21f21122e
This commit is contained in:
Roman.Ivanov
2019-12-17 11:02:18 +00:00
committed by intellij-monorepo-bot
parent 81da6b615b
commit 89d4f6f422
@@ -70,7 +70,9 @@ public class RecordAugmentProvider extends PsiAugmentProvider {
String typeText = getTypeText(component, RecordAugmentProvider::hasTargetApplicableForField);
if (typeText == null) return null;
PsiClass aClass = factory.createClassFromText("private final " + typeText + " " + name + ";", null);
return aClass.getFields()[0];
PsiField[] fields = aClass.getFields();
if (fields.length != 1) return null;
return fields[0];
}
@Nullable
@@ -80,7 +82,9 @@ public class RecordAugmentProvider extends PsiAugmentProvider {
String typeText = getTypeText(component, RecordAugmentProvider::hasTargetApplicableForMethod);
if (typeText == null) return null;
PsiClass aClass = factory.createClassFromText("public " + typeText + " " + name + "(){}", null);
return aClass.getMethods()[0];
PsiMethod[] methods = aClass.getMethods();
if (methods.length != 1) return null;
return methods[0];
}
@Nullable