@@ -108,7 +108,7 @@ private MethodSpec buildToAttributeValueMethod(TypeInfo typeInfo) {
108
108
String className = typeInfo .getClassName ();
109
109
String paramName = typeExtractor .getParameterName (className );
110
110
ClassName attributeValue = ClassName .get (AttributeValue .class );
111
- ClassName domainClass = ClassName .bestGuess (className );
111
+ ClassName domainClass = ClassName .bestGuess (typeInfo . getFullyQualifiedClassName () );
112
112
113
113
MethodSpec .Builder methodBuilder = MethodSpec .methodBuilder ("toDynamoDbAttributeValue" )
114
114
.addModifiers (Modifier .PUBLIC )
@@ -123,16 +123,13 @@ private MethodSpec buildToAttributeValueMethod(TypeInfo typeInfo) {
123
123
methodBuilder .beginControlFlow ("if ($L == null)" , paramName )
124
124
.addStatement ("return null" )
125
125
.endControlFlow ()
126
- .addStatement ("" )
127
126
.addStatement ("$T<$T, $T> attributes = new $T<>()" ,
128
- Map .class , String .class , AttributeValue .class , HashMap .class )
129
- .addStatement ("" );
127
+ Map .class , String .class , AttributeValue .class , HashMap .class );
130
128
131
129
// Generate field mappings
132
130
for (FieldInfo field : typeInfo .getFields ()) {
133
131
CodeBlock mappingCode = fieldMappingCodeGenerator .generateToAttributeValueMapping (field , paramName );
134
132
methodBuilder .addCode (mappingCode );
135
- methodBuilder .addStatement ("" );
136
133
}
137
134
138
135
methodBuilder .addStatement ("return $T.builder().m(attributes).build()" , attributeValue );
@@ -143,7 +140,7 @@ private MethodSpec buildToAttributeValueMethod(TypeInfo typeInfo) {
143
140
private MethodSpec buildFromAttributeValueMethod (TypeInfo typeInfo ) {
144
141
String className = typeInfo .getClassName ();
145
142
ClassName attributeValue = ClassName .get (AttributeValue .class );
146
- ClassName domainClass = ClassName .bestGuess (className );
143
+ ClassName domainClass = ClassName .bestGuess (typeInfo . getFullyQualifiedClassName () );
147
144
148
145
MethodSpec .Builder methodBuilder = MethodSpec .methodBuilder ("fromDynamoDbAttributeValue" )
149
146
.addModifiers (Modifier .PUBLIC )
@@ -158,11 +155,9 @@ private MethodSpec buildFromAttributeValueMethod(TypeInfo typeInfo) {
158
155
methodBuilder .beginControlFlow ("if (attributeValue == null || attributeValue.m() == null)" )
159
156
.addStatement ("return null" )
160
157
.endControlFlow ()
161
- .addStatement ("" )
162
158
.addStatement ("$T<$T, $T> item = attributeValue.m()" ,
163
159
Map .class , String .class , AttributeValue .class )
164
- .addStatement ("var builder = $T.builder()" , domainClass )
165
- .addStatement ("" );
160
+ .addStatement ("var builder = $T.builder()" , domainClass );
166
161
167
162
// Generate field mappings
168
163
for (FieldInfo field : typeInfo .getFields ()) {
@@ -178,7 +173,7 @@ private MethodSpec buildFromAttributeValueMethod(TypeInfo typeInfo) {
178
173
private void addConvenienceMethods (TypeSpec .Builder classBuilder , TypeInfo typeInfo ) {
179
174
String className = typeInfo .getClassName ();
180
175
ClassName attributeValue = ClassName .get (AttributeValue .class );
181
- ClassName domainClass = ClassName .bestGuess (className );
176
+ ClassName domainClass = ClassName .bestGuess (typeInfo . getFullyQualifiedClassName () );
182
177
183
178
// fromDynamoDbItem method
184
179
MethodSpec fromDynamoDbItem = MethodSpec .methodBuilder ("fromDynamoDbItem" )
@@ -219,11 +214,12 @@ private void addConvenienceMethods(TypeSpec.Builder classBuilder, TypeInfo typeI
219
214
.beginControlFlow ("if (items == null || items.isEmpty())" )
220
215
.addStatement ("return new $T<>()" , ClassName .get ("java.util" , "ArrayList" ))
221
216
.endControlFlow ()
222
- .addStatement ("return items.stream()" )
223
- .addStatement (" .map(item -> $T.builder().m(item).build())" , attributeValue )
224
- .addStatement (" .map(this::fromDynamoDbAttributeValue)" )
225
- .addStatement (" .filter($T::nonNull)" , ClassName .get (Objects .class ))
226
- .addStatement (" .collect($T.toList())" , ClassName .get (Collectors .class ))
217
+ .addStatement ("return items.stream()$>\n " +
218
+ ".map(item -> $T.builder().m(item).build())$>\n " +
219
+ ".map(this::fromDynamoDbAttributeValue)$>\n " +
220
+ ".filter($T::nonNull)$>\n " +
221
+ ".collect($T.toList())$<$<$<$<" ,
222
+ attributeValue , ClassName .get (Objects .class ), ClassName .get (Collectors .class ))
227
223
.build ();
228
224
229
225
// toDynamoDbItem method
@@ -264,12 +260,13 @@ private void addConvenienceMethods(TypeSpec.Builder classBuilder, TypeInfo typeI
264
260
.beginControlFlow ("if (objects == null || objects.isEmpty())" )
265
261
.addStatement ("return new $T<>()" , ClassName .get ("java.util" , "ArrayList" ))
266
262
.endControlFlow ()
267
- .addStatement ("return objects.stream()" )
268
- .addStatement (" .map(this::toDynamoDbAttributeValue)" )
269
- .addStatement (" .filter($T::nonNull)" , ClassName .get (Objects .class ))
270
- .addStatement (" .map(av -> av.m())" )
271
- .addStatement (" .filter(map -> map != null && !map.isEmpty())" )
272
- .addStatement (" .collect($T.toList())" , ClassName .get (Collectors .class ))
263
+ .addStatement ("return objects.stream()$>\n " +
264
+ ".map(this::toDynamoDbAttributeValue)$>\n " +
265
+ ".filter($T::nonNull)$>\n " +
266
+ ".map(av -> av.m())$>\n " +
267
+ ".filter(map -> map != null && !map.isEmpty())$>\n " +
268
+ ".collect($T.toList())$<$<$<$<$<" ,
269
+ ClassName .get (Objects .class ), ClassName .get (Collectors .class ))
273
270
.build ();
274
271
275
272
classBuilder .addMethod (fromDynamoDbItem );
0 commit comments