From be2f5bf5b5b94e780c62918acc1665d5d3364b04 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9B=B9=E9=B9=8F=E9=A3=9E?= Date: Sat, 8 Feb 2025 19:14:57 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E4=B8=80=E4=BA=9B=E8=B0=83=E6=95=B4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../src/test/java/ControllerTest.java | 42 +++++++++++++------ 1 file changed, 29 insertions(+), 13 deletions(-) diff --git a/nflg-mobilebroken-admin/src/test/java/ControllerTest.java b/nflg-mobilebroken-admin/src/test/java/ControllerTest.java index f040947d..1957c9c9 100644 --- a/nflg-mobilebroken-admin/src/test/java/ControllerTest.java +++ b/nflg-mobilebroken-admin/src/test/java/ControllerTest.java @@ -1,5 +1,4 @@ import cn.hutool.core.util.StrUtil; -import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.nflg.mobilebroken.admin.AdminApplication; import com.nflg.mobilebroken.admin.annotation.ApiMark; import com.nflg.mobilebroken.repository.entity.AdminApi; @@ -13,9 +12,8 @@ import org.springframework.web.servlet.mvc.method.RequestMappingInfo; import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping; import javax.annotation.Resource; -import java.util.ArrayList; -import java.util.List; import java.util.Map; +import java.util.Objects; @SpringBootTest @ContextConfiguration(classes = AdminApplication.class) @@ -29,7 +27,7 @@ public class ControllerTest { @Test public void test2() { - List apis = new ArrayList<>(); +// List apis = new ArrayList<>(); // 获取所有的请求映射 Map handlerMethods = requestMappingHandlerMapping.getHandlerMethods(); for (Map.Entry entry : handlerMethods.entrySet()) { @@ -48,18 +46,36 @@ public class ControllerTest { if (mark != null) { System.out.println("方法: " + handlerMethod.getMethod().getName()); System.out.println("位置: " + mark.moduleName() + " - " + mark.apiName()); - apis.add(new AdminApi() - .setName(mark.apiName()) - .setModuleName(mark.moduleName()) - .setIsPublic(mark.isPublic()) - .setUrl(StrUtil.join(",", mappingInfo.getPatternsCondition().getPatterns()))); +// AdminApi api=new AdminApi() +// .setName(mark.apiName()) +// .setModuleName(mark.moduleName()) +// .setIsPublic(mark.isPublic()) +// .setUrl(StrUtil.join(",", mappingInfo.getPatternsCondition().getPatterns())); +// apis.add(api); // 打印请求路径 - System.out.println("Request Paths: " + mappingInfo.getPatternsCondition().getPatterns()); + String url = StrUtil.join(",", mappingInfo.getPatternsCondition().getPatterns()); + System.out.println("Request Paths: " + url); + AdminApi api = adminApiService.lambdaQuery() + .eq(AdminApi::getModuleName, mark.moduleName()) + .eq(AdminApi::getName, mark.apiName()) + .one(); + if (Objects.isNull(api)) { + api = new AdminApi() + .setName(mark.apiName()) + .setModuleName(mark.moduleName()) + .setIsPublic(mark.isPublic()) + .setUrl(url); + adminApiService.save(api); + } else if (!StrUtil.equals(url, api.getUrl()) || !Objects.equals(api.getIsPublic(), mark.isPublic())) { + api.setUrl(url); + api.setIsPublic(mark.isPublic()); + adminApiService.updateById(api); + } + System.out.println("----------------------------------------"); } - System.out.println("----------------------------------------"); } - adminApiService.remove(new LambdaQueryWrapper<>()); - adminApiService.saveBatch(apis); +// adminApiService.remove(new LambdaQueryWrapper<>()); +// adminApiService.saveBatch(apis); // System.out.println(JSONUtil.toJsonStr(apis)); } }