完善代码生成

This commit is contained in:
2023-09-12 21:21:54 +08:00
parent 13086f8a30
commit 593b7b07db

View File

@@ -302,15 +302,17 @@ public class Codegen {
.addParameter(ParameterSpec.builder(ClassName.get(IdRequest.class), "request").addAnnotation(RequestBody.class).build())
.addStatement("""
this.service.delete(request);
return true
""")
this.service.delete(request);
return true
""")
.build()
)
.addMethod(MethodSpec.methodBuilder("getById")
.addModifiers(Modifier.PUBLIC)
.returns(ClassName.get(dtoPackage, dto.name))
.addAnnotation(AnnotationSpec.builder(PostMapping.class).addMember("value", "$S", "/getById").build())
.addParameter(ParameterSpec.builder(ClassName.get(String.class), "request").addAnnotation(RequestBody.class).build())
.addStatement("""
return this.service.getById(request)
@@ -321,11 +323,13 @@ public class Codegen {
.addModifiers(Modifier.PUBLIC)
.returns(ParameterizedTypeName.get(ClassName.get(Page.class), ClassName.get(dtoPackage, dto.name)))
.addParameter( ParameterSpec.builder(ClassName.get(CommonQuery.class), "request").addAnnotation(RequestBody.class).build() )
.addAnnotation(AnnotationSpec.builder(PostMapping.class).addMember("value", "$S", "/list").build())
.addParameter(ParameterSpec.builder(ClassName.get(CommonQuery.class), "request").addAnnotation(RequestBody.class).build())
.addStatement(
"""
return this.service.list(request)
""", ClassName.get(Page.class), ClassName.get(entityPackage, entity.name), ClassName.get(PageRequest.class), ClassName.get(Sort.class))
return this.service.list(request)
""", ClassName.get(Page.class), ClassName.get(entityPackage, entity.name), ClassName.get(PageRequest.class), ClassName.get(Sort.class))
.build()
)
.build();