|
1 | 1 | package org.mapstruct.extensions.spring.converter; |
2 | 2 |
|
3 | | -import static java.util.stream.Collectors.toList; |
4 | | -import static javax.lang.model.element.Modifier.*; |
5 | | - |
6 | 3 | import com.squareup.javapoet.*; |
7 | 4 | import org.apache.commons.lang3.StringUtils; |
8 | 5 |
|
|
12 | 9 | import java.time.Clock; |
13 | 10 | import java.time.ZonedDateTime; |
14 | 11 | import java.time.format.DateTimeFormatter; |
15 | | -import java.util.ArrayList; |
16 | | -import java.util.List; |
| 12 | + |
| 13 | +import static java.util.stream.Collectors.toList; |
| 14 | +import static javax.lang.model.element.Modifier.*; |
17 | 15 |
|
18 | 16 | public class ConversionServiceAdapterGenerator { |
19 | | - private final Clock clock; |
| 17 | + private final Clock clock; |
| 18 | + |
| 19 | + public ConversionServiceAdapterGenerator(final Clock clock) { |
| 20 | + this.clock = clock; |
| 21 | + } |
20 | 22 |
|
21 | | - public ConversionServiceAdapterGenerator(final Clock clock) { |
22 | | - this.clock = clock; |
23 | | - } |
| 23 | + public void writeConversionServiceAdapter( |
| 24 | + ConversionServiceAdapterDescriptor descriptor, Writer out) { |
| 25 | + try { |
| 26 | + JavaFile.builder( |
| 27 | + descriptor.getAdapterClassName().packageName(), |
| 28 | + createConversionServiceTypeSpec(descriptor)) |
| 29 | + .build() |
| 30 | + .writeTo(out); |
| 31 | + } catch (IOException e) { |
| 32 | + throw new UncheckedIOException(e); |
| 33 | + } |
| 34 | + } |
| 35 | + |
| 36 | + private TypeSpec createConversionServiceTypeSpec( |
| 37 | + final ConversionServiceAdapterDescriptor descriptor) { |
| 38 | + final FieldSpec conversionServiceFieldSpec = buildConversionServiceFieldSpec(); |
| 39 | + return TypeSpec.classBuilder(descriptor.getAdapterClassName()) |
| 40 | + .addModifiers(PUBLIC) |
| 41 | + .addAnnotation(buildGeneratedAnnotationSpec()) |
| 42 | + .addAnnotation(ClassName.get("org.springframework.stereotype", "Component")) |
| 43 | + .addField(conversionServiceFieldSpec) |
| 44 | + .addMethod(buildConstructorSpec(descriptor, conversionServiceFieldSpec)) |
| 45 | + .addMethods(buildMappingMethods(descriptor, conversionServiceFieldSpec)) |
| 46 | + .build(); |
| 47 | + } |
24 | 48 |
|
25 | | - public void writeConversionServiceAdapter( |
26 | | - ConversionServiceAdapterDescriptor descriptor, Writer out) { |
27 | | - try { |
28 | | - JavaFile.builder( |
29 | | - descriptor.getAdapterClassName().packageName(), |
30 | | - createConversionServiceTypeSpec(descriptor)) |
31 | | - .build() |
32 | | - .writeTo(out); |
33 | | - } catch (IOException e) { |
34 | | - throw new UncheckedIOException(e); |
| 49 | + private static MethodSpec buildConstructorSpec(final ConversionServiceAdapterDescriptor descriptor, final FieldSpec conversionServiceFieldSpec) { |
| 50 | + final ParameterSpec constructorParameterSpec = buildConstructorParameterSpec(descriptor, conversionServiceFieldSpec); |
| 51 | + return MethodSpec.constructorBuilder().addModifiers(PUBLIC).addParameter(constructorParameterSpec).addStatement("this.$N = $N", conversionServiceFieldSpec, constructorParameterSpec).build(); |
35 | 52 | } |
36 | | - } |
37 | 53 |
|
38 | | - private TypeSpec createConversionServiceTypeSpec( |
39 | | - final ConversionServiceAdapterDescriptor descriptor) { |
40 | | - final FieldSpec injectedConversionServiceFieldSpec = buildInjectedConversionServiceFieldSpec(descriptor); |
41 | | - return TypeSpec.classBuilder(descriptor.getAdapterClassName()) |
42 | | - .addModifiers(PUBLIC) |
43 | | - .addAnnotation(buildGeneratedAnnotationSpec()) |
44 | | - .addAnnotation(ClassName.get("org.springframework.stereotype", "Component")) |
45 | | - .addField(injectedConversionServiceFieldSpec) |
46 | | - .addMethods(buildMappingMethods(descriptor, injectedConversionServiceFieldSpec)) |
47 | | - .build(); |
48 | | - } |
| 54 | + private static ParameterSpec buildConstructorParameterSpec(final ConversionServiceAdapterDescriptor descriptor, final FieldSpec conversionServiceFieldSpec) { |
| 55 | + final ParameterSpec.Builder parameterBuilder = ParameterSpec.builder(conversionServiceFieldSpec.type, conversionServiceFieldSpec.name, FINAL); |
| 56 | + if (StringUtils.isNotEmpty(descriptor.getConversionServiceBeanName())) { |
| 57 | + parameterBuilder.addAnnotation(buildQualifierANnotation(descriptor)); |
| 58 | + } |
| 59 | + return parameterBuilder.build(); |
| 60 | + } |
49 | 61 |
|
50 | | - private static Iterable<MethodSpec> buildMappingMethods( |
51 | | - final ConversionServiceAdapterDescriptor descriptor, |
52 | | - final FieldSpec injectedConversionServiceFieldSpec) { |
53 | | - return descriptor.getFromToMappings().stream() |
54 | | - .map( |
55 | | - sourceTargetPair -> { |
56 | | - final ParameterSpec sourceParameterSpec = |
57 | | - buildSourceParameterSpec(sourceTargetPair.getLeft()); |
58 | | - return MethodSpec.methodBuilder( |
59 | | - "map" |
60 | | - + sourceTargetPair.getLeft().simpleName() |
61 | | - + "To" |
62 | | - + sourceTargetPair.getRight().simpleName()) |
63 | | - .addParameter(sourceParameterSpec) |
64 | | - .addModifiers(PUBLIC) |
65 | | - .returns(sourceTargetPair.getRight()) |
66 | | - .addStatement( |
67 | | - "return $N.convert($N, $T.class)", |
68 | | - injectedConversionServiceFieldSpec, |
69 | | - sourceParameterSpec, |
70 | | - sourceTargetPair.getRight()) |
71 | | - .build(); |
72 | | - }) |
73 | | - .collect(toList()); |
74 | | - } |
| 62 | + private static AnnotationSpec buildQualifierANnotation(ConversionServiceAdapterDescriptor descriptor) { |
| 63 | + return AnnotationSpec |
| 64 | + .builder(ClassName.get("org.springframework.beans.factory.annotation", "Qualifier")) |
| 65 | + .addMember("value", "$S", descriptor.getConversionServiceBeanName()) |
| 66 | + .build(); |
| 67 | + } |
75 | 68 |
|
76 | | - private static ParameterSpec buildSourceParameterSpec(final TypeName sourceClassName) { |
77 | | - return ParameterSpec.builder(sourceClassName, "source", FINAL).build(); |
78 | | - } |
| 69 | + private static Iterable<MethodSpec> buildMappingMethods( |
| 70 | + final ConversionServiceAdapterDescriptor descriptor, |
| 71 | + final FieldSpec injectedConversionServiceFieldSpec) { |
| 72 | + return descriptor.getFromToMappings().stream() |
| 73 | + .map( |
| 74 | + sourceTargetPair -> { |
| 75 | + final ParameterSpec sourceParameterSpec = |
| 76 | + buildSourceParameterSpec(sourceTargetPair.getLeft()); |
| 77 | + return MethodSpec.methodBuilder( |
| 78 | + "map" |
| 79 | + + sourceTargetPair.getLeft().simpleName() |
| 80 | + + "To" |
| 81 | + + sourceTargetPair.getRight().simpleName()) |
| 82 | + .addParameter(sourceParameterSpec) |
| 83 | + .addModifiers(PUBLIC) |
| 84 | + .returns(sourceTargetPair.getRight()) |
| 85 | + .addStatement( |
| 86 | + "return $N.convert($N, $T.class)", |
| 87 | + injectedConversionServiceFieldSpec, |
| 88 | + sourceParameterSpec, |
| 89 | + sourceTargetPair.getRight()) |
| 90 | + .build(); |
| 91 | + }) |
| 92 | + .collect(toList()); |
| 93 | + } |
79 | 94 |
|
80 | | - private static FieldSpec buildInjectedConversionServiceFieldSpec(ConversionServiceAdapterDescriptor descriptor) { |
81 | | - List<AnnotationSpec> annotations = new ArrayList<>(); |
82 | | - annotations.add(AnnotationSpec.builder(ClassName |
83 | | - .get("org.springframework.beans.factory.annotation", "Autowired")) |
84 | | - .build()); |
85 | | - if (StringUtils.isNotEmpty(descriptor.getConversionServiceBeanName())) { |
86 | | - annotations.add(AnnotationSpec |
87 | | - .builder(ClassName.get("org.springframework.beans.factory.annotation", "Qualifier")) |
88 | | - .addMember("value", "$S", descriptor.getConversionServiceBeanName()) |
89 | | - .build()); |
| 95 | + private static ParameterSpec buildSourceParameterSpec(final TypeName sourceClassName) { |
| 96 | + return ParameterSpec.builder(sourceClassName, "source", FINAL).build(); |
90 | 97 | } |
91 | 98 |
|
92 | | - return FieldSpec.builder(ClassName.get("org.springframework.core.convert","ConversionService"), |
93 | | - "conversionService", PRIVATE) |
94 | | - .addAnnotations(annotations) |
95 | | - .build(); |
96 | | - } |
| 99 | + private static FieldSpec buildConversionServiceFieldSpec() { |
| 100 | + return FieldSpec.builder(ClassName.get("org.springframework.core.convert", "ConversionService"), "conversionService", PRIVATE, FINAL).build(); |
| 101 | + } |
97 | 102 |
|
98 | | - private AnnotationSpec buildGeneratedAnnotationSpec() { |
99 | | - return AnnotationSpec.builder(ClassName.get("javax.annotation", "Generated")) |
100 | | - .addMember("value", "$S", ConversionServiceAdapterGenerator.class.getName()) |
101 | | - .addMember("date", "$S", DateTimeFormatter.ISO_INSTANT.format(ZonedDateTime.now(clock))) |
102 | | - .build(); |
103 | | - } |
| 103 | + private AnnotationSpec buildGeneratedAnnotationSpec() { |
| 104 | + return AnnotationSpec.builder(ClassName.get("javax.annotation", "Generated")) |
| 105 | + .addMember("value", "$S", ConversionServiceAdapterGenerator.class.getName()) |
| 106 | + .addMember("date", "$S", DateTimeFormatter.ISO_INSTANT.format(ZonedDateTime.now(clock))) |
| 107 | + .build(); |
| 108 | + } |
104 | 109 | } |
0 commit comments