Skip to content

Commit db537f1

Browse files
committed
small cleanup
1 parent bc21a99 commit db537f1

File tree

4 files changed

+33
-25
lines changed

4 files changed

+33
-25
lines changed

src/main/groovy/com/github/hauner/openapi/spring/generatr/SpringGeneratr.groovy

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -63,20 +63,23 @@ class SpringGeneratr implements OpenApiGeneratr {
6363
def api = cv.convert (result.openAPI)
6464

6565
def headerWriter = new HeaderWriter()
66-
def beanValidationWriter = new BeanValidationFactory()
66+
def beanValidationFactory = new BeanValidationFactory()
67+
6768
def writer = new ApiWriter (options,
6869
new InterfaceWriter(
6970
headerWriter: headerWriter,
7071
methodWriter: new MethodWriter(
71-
apiOptions: options,
72-
beanValidationWriter: beanValidationWriter
72+
beanValidationFactory: beanValidationFactory,
73+
apiOptions: options
7374
),
74-
beanValidationWriter: beanValidationWriter,
75-
apiOptions: options),
75+
beanValidationFactory: beanValidationFactory,
76+
apiOptions: options
77+
),
7678
new DataTypeWriter(
7779
headerWriter: headerWriter,
80+
beanValidationFactory: beanValidationFactory,
7881
apiOptions: options,
79-
beanValidationWriter: beanValidationWriter),
82+
),
8083
new StringEnumWriter(headerWriter: headerWriter)
8184
)
8285

src/main/groovy/com/github/hauner/openapi/spring/writer/DataTypeWriter.groovy

Lines changed: 16 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ import com.github.hauner.openapi.support.Identifier
3030
class DataTypeWriter {
3131
ApiOptions apiOptions
3232
HeaderWriter headerWriter
33-
BeanValidationFactory beanValidationWriter
33+
BeanValidationFactory beanValidationFactory
3434

3535
void write (Writer target, ObjectDataType dataType) {
3636
headerWriter.write (target)
@@ -51,9 +51,7 @@ class DataTypeWriter {
5151
propertyNames.each {
5252
def javaPropertyName = Identifier.toCamelCase (it)
5353
def propDataType = dataType.getObjectProperty (it)
54-
target.write (" @JsonProperty(\"$it\")\n")
55-
target.write(getBeanValidations(propDataType))
56-
target.write (" private ${propDataType.name} ${javaPropertyName};\n\n")
54+
target.write (getProp (it, javaPropertyName, propDataType))
5755
}
5856

5957
propertyNames.each {
@@ -66,12 +64,19 @@ class DataTypeWriter {
6664
target.write ("}\n")
6765
}
6866

69-
private String getBeanValidations(DataType propDataType) {
67+
private String getProp (String propertyName, String javaPropertyName, DataType propDataType) {
68+
String result
69+
result = " @JsonProperty(\"${propertyName}\")\n"
70+
7071
if (apiOptions.beanValidation) {
71-
beanValidationWriter.createAnnotations(propDataType)
72-
} else {
73-
''
72+
def beanValidationAnnotations = beanValidationFactory.createAnnotations (propDataType)
73+
if (!beanValidationAnnotations.empty) {
74+
result += " $beanValidationAnnotations\n"
75+
}
7476
}
77+
78+
result += " private ${propDataType.name} ${javaPropertyName};\n\n"
79+
result
7580
}
7681

7782
private String getGetter (String propertyName, DataType propDataType) {
@@ -98,9 +103,9 @@ class DataTypeWriter {
98103

99104
imports.addAll (dataType.referencedImports)
100105

101-
if(apiOptions.beanValidation){
102-
for (DataType propDataType : dataType.properties.values ()) {
103-
imports.addAll (beanValidationWriter.collectImports (propDataType))
106+
if (apiOptions.beanValidation) {
107+
for (DataType propDataType : dataType.properties.values ()) {
108+
imports.addAll (beanValidationFactory.collectImports (propDataType))
104109
}
105110
}
106111

src/main/groovy/com/github/hauner/openapi/spring/writer/InterfaceWriter.groovy

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ class InterfaceWriter {
3030
ApiOptions apiOptions
3131
HeaderWriter headerWriter
3232
MethodWriter methodWriter
33-
BeanValidationFactory beanValidationWriter
33+
BeanValidationFactory beanValidationFactory
3434

3535
void write (Writer target, Interface itf) {
3636
headerWriter.write (target)
@@ -62,7 +62,7 @@ class InterfaceWriter {
6262

6363
ep.parameters.each { p ->
6464
if (apiOptions.beanValidation) {
65-
imports.addAll(beanValidationWriter.collectImports (p.dataType))
65+
imports.addAll (beanValidationFactory.collectImports (p.dataType))
6666
}
6767

6868
if (p.withAnnotation()) {
@@ -76,7 +76,7 @@ class InterfaceWriter {
7676
imports.add (b.annotationWithPackage)
7777
imports.addAll (b.imports)
7878
if (apiOptions.beanValidation) {
79-
imports.addAll(beanValidationWriter.collectImports (b.requestBodyType))
79+
imports.addAll (beanValidationFactory.collectImports (b.requestBodyType))
8080
}
8181
}
8282

src/main/groovy/com/github/hauner/openapi/spring/writer/MethodWriter.groovy

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ import com.github.hauner.openapi.support.Identifier
3131
class MethodWriter {
3232

3333
ApiOptions apiOptions
34-
BeanValidationFactory beanValidationWriter
34+
BeanValidationFactory beanValidationFactory
3535

3636
void write (Writer target, Endpoint endpoint) {
3737
target.write ("""\
@@ -107,8 +107,8 @@ class MethodWriter {
107107

108108
def methodDefinition = ''
109109

110-
if(apiOptions.beanValidation){
111-
methodDefinition += " ${beanValidationWriter.createAnnotations(it.dataType)}"
110+
if (apiOptions.beanValidation) {
111+
methodDefinition += " ${beanValidationFactory.createAnnotations (it.dataType)}"
112112
}
113113

114114
if (it.withAnnotation ()) {
@@ -122,8 +122,8 @@ class MethodWriter {
122122
if (!endpoint.requestBodies.empty) {
123123
def body = endpoint.requestBody
124124
def beanValidationAnnotations = ''
125-
if(apiOptions.beanValidation){
126-
beanValidationAnnotations += " ${beanValidationWriter.createAnnotations(body.requestBodyType)}"
125+
if (apiOptions.beanValidation) {
126+
beanValidationAnnotations += " ${beanValidationFactory.createAnnotations (body.requestBodyType)}"
127127
}
128128
def param = "${beanValidationAnnotations} ${createRequestBodyAnnotation(body)} ${body.requestBodyType.name} body"
129129
ps.add (param.trim())

0 commit comments

Comments
 (0)