diff --git a/src/main/java/cn/lihongjie/coal/Codegen.java b/src/main/java/cn/lihongjie/coal/Codegen.java index 2f871b47..762285e9 100644 --- a/src/main/java/cn/lihongjie/coal/Codegen.java +++ b/src/main/java/cn/lihongjie/coal/Codegen.java @@ -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();