lihongjie/coal
1
0
Fork 0
mirror of https://codeup.aliyun.com/64f7d6b8ce01efaafef1e678/coal/coal.git synced 2026-01-25 07:46:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

完善代码生成

Browse Source
This commit is contained in:
lihongjie0209
2023-09-12 21:21:54 +08:00
parent 13086f8a30
commit 593b7b07db
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/main/java/cn/lihongjie/coal/Codegen.java
View File

@@ -302,15 +302,17 @@ public class Codegen {
.addParameter(ParameterSpec.builder(ClassName.get(IdRequest.class), "request").addAnnotation(RequestBody.class).build()) .addParameter(ParameterSpec.builder(ClassName.get(IdRequest.class), "request").addAnnotation(RequestBody.class).build())
.addStatement(""" .addStatement("""
this.service.delete(request); this.service.delete(request);
return true return true
""") """)
.build() .build()
) )
.addMethod(MethodSpec.methodBuilder("getById") .addMethod(MethodSpec.methodBuilder("getById")
.addModifiers(Modifier.PUBLIC) .addModifiers(Modifier.PUBLIC)
.returns(ClassName.get(dtoPackage, dto.name)) .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()) .addParameter(ParameterSpec.builder(ClassName.get(String.class), "request").addAnnotation(RequestBody.class).build())
.addStatement(""" .addStatement("""
return this.service.getById(request) return this.service.getById(request)
@@ -321,11 +323,13 @@ public class Codegen {
.addModifiers(Modifier.PUBLIC) .addModifiers(Modifier.PUBLIC)
.returns(ParameterizedTypeName.get(ClassName.get(Page.class), ClassName.get(dtoPackage, dto.name))) .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( .addStatement(
""" """
return this.service.list(request) return this.service.list(request)
""", ClassName.get(Page.class), ClassName.get(entityPackage, entity.name), ClassName.get(PageRequest.class), ClassName.get(Sort.class)) """, ClassName.get(Page.class), ClassName.get(entityPackage, entity.name), ClassName.get(PageRequest.class), ClassName.get(Sort.class))
.build() .build()
) )
.build(); .build();