From 593b7b07db0d5d4484d3387c81f2d8662548abce Mon Sep 17 00:00:00 2001 From: lihongjie0209 Date: Tue, 12 Sep 2023 21:21:54 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84=E4=BB=A3=E7=A0=81=E7=94=9F?= =?UTF-8?q?=E6=88=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/main/java/cn/lihongjie/coal/Codegen.java | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) 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();