@@ -22,7 +22,6 @@ class SimpleBeanWriter {
2222 private final String packageName ;
2323 private final String suffix ;
2424 private Append writer ;
25- private final String indent = " " ;
2625
2726 SimpleBeanWriter (BeanReader beanReader , ProcessingContext context ) {
2827 this .beanReader = beanReader ;
@@ -92,7 +91,7 @@ private void writeStaticFactoryMethod() {
9291
9392 private void writeAddFor (MethodReader constructor ) {
9493 beanReader .buildAddFor (writer );
95- writeCreateBean (constructor , "builder" );
94+ writeCreateBean (constructor );
9695 beanReader .buildRegister (writer );
9796 beanReader .addLifecycleCallbacks (writer );
9897 if (beanReader .isExtraInjectionRequired ()) {
@@ -120,34 +119,32 @@ private void writeBuildMethodStart(MethodReader constructor) {
120119 writer .append (") {" ).eol ();
121120 }
122121
123- private void writeCreateBean (MethodReader constructor , String builderName ) {
124- writer .append ("%s %s bean = new %s(" , indent , shortName , shortName );
122+ private void writeCreateBean (MethodReader constructor ) {
123+ writer .append (" %s bean = new %s(" , shortName , shortName );
125124 // add constructor dependencies
126- writeMethodParams (builderName , constructor );
125+ writeMethodParams ("builder" , constructor );
127126 }
128127
129128 private void writeExtraInjection () {
130- String builderRef = "b" ;
131- String beanRef = "$bean" ;
132129 writer .append (" builder.addInjector(b -> {" ).eol ();
133130 writer .append (" // field and method injection" ).eol ();
134- injectFields (builderRef , beanRef );
135- injectMethods (builderRef , beanRef );
131+ injectFields ();
132+ injectMethods ();
136133 writer .append (" });" ).eol ();
137134 }
138135
139- private void injectFields (String builderRef , String beanRef ) {
136+ private void injectFields () {
140137 for (FieldReader fieldReader : beanReader .getInjectFields ()) {
141138 String fieldName = fieldReader .getFieldName ();
142- String getDependency = fieldReader .builderGetDependency (builderRef );
143- writer .append (" %s .%s = %s;" , beanRef , fieldName , getDependency ).eol ();
139+ String getDependency = fieldReader .builderGetDependency ();
140+ writer .append (" $bean .%s = %s;" , fieldName , getDependency ).eol ();
144141 }
145142 }
146143
147- private void injectMethods (String builderRef , String beanRef ) {
144+ private void injectMethods () {
148145 for (MethodReader methodReader : beanReader .getInjectMethods ()) {
149- writer .append (" %s .%s(" , beanRef , methodReader .getName ());
150- writeMethodParams (builderRef , methodReader );
146+ writer .append (" $bean .%s(" , methodReader .getName ());
147+ writeMethodParams ("b" , methodReader );
151148 }
152149 }
153150
0 commit comments