feat: 一些调整

This commit is contained in:
曹鹏飞 2025-02-08 19:14:57 +08:00
parent 59ffff943c
commit be2f5bf5b5
1 changed files with 29 additions and 13 deletions

View File

@ -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<AdminApi> apis = new ArrayList<>();
// List<AdminApi> apis = new ArrayList<>();
// 获取所有的请求映射
Map<RequestMappingInfo, HandlerMethod> handlerMethods = requestMappingHandlerMapping.getHandlerMethods();
for (Map.Entry<RequestMappingInfo, HandlerMethod> 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()
// AdminApi api=new AdminApi()
// .setName(mark.apiName())
// .setModuleName(mark.moduleName())
// .setIsPublic(mark.isPublic())
// .setUrl(StrUtil.join(",", mappingInfo.getPatternsCondition().getPatterns()));
// apis.add(api);
// 打印请求路径
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(StrUtil.join(",", mappingInfo.getPatternsCondition().getPatterns())));
// 打印请求路径
System.out.println("Request Paths: " + mappingInfo.getPatternsCondition().getPatterns());
.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("----------------------------------------");
}
adminApiService.remove(new LambdaQueryWrapper<>());
adminApiService.saveBatch(apis);
}
// adminApiService.remove(new LambdaQueryWrapper<>());
// adminApiService.saveBatch(apis);
// System.out.println(JSONUtil.toJsonStr(apis));
}
}