Skip to content

Commit 8544ac1

Browse files
committed
Fix find existing builder class
1 parent 950284b commit 8544ac1

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/main/java/com/github/junkfactory/innerbuilder/generators/InnerBuilderGenerator.java

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -158,19 +158,19 @@ private PsiMethod generateTargetConstructor(final PsiClass targetClass, BuilderC
158158
@NotNull
159159
private BuilderClass findOrCreateBuilderClass(final PsiClass targetClass) {
160160
var builderClassName = Utils.buildClassName(BUILDER_CLASS_NAME, targetClass);
161-
var builderClass = targetClass.findInnerClassByName(builderClassName.className(), false);
162-
if (builderClass == null) {
163-
builderClass = (PsiClass) targetClass.add(createBuilderClass(targetClass, builderClassName.className()));
161+
var psiClass = targetClass.findInnerClassByName(BUILDER_CLASS_NAME, false);
162+
if (psiClass == null) {
163+
psiClass = (PsiClass) targetClass.add(createBuilderClass(targetClass, builderClassName.className()));
164164
}
165165

166166
var psiElementFactory = generatorParams.psi().factory();
167167
var builderType = psiElementFactory.createTypeFromText(builderClassName.className(), targetClass);
168-
return new BuilderClass(builderClass, builderType, builderClassName, Utils.isGenericType(builderType));
168+
return new BuilderClass(psiClass, builderType, builderClassName, Utils.isGenericType(builderType));
169169
}
170170

171171
@NotNull
172172
private PsiClass createBuilderClass(final PsiClass targetClass, String builderClassName) {
173-
String classDef = "public static final class " +
173+
var classDef = "public static final class " +
174174
builderClassName +
175175
" {}" +
176176
System.lineSeparator();

0 commit comments

Comments
 (0)