dromara/RuoYi-Cloud-Plus
1
0
Fork 0
mirror of https://gitee.com/dromara/RuoYi-Cloud-Plus.git synced 2025-09-09 13:49:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

update 优化 接口文档服务路径从gateway转发的请求头获取

Browse Source
This commit is contained in:
疯狂的狮子Li
2024-12-09 11:30:42 +08:00
parent 8939df7cd3
commit 3b433089e3
1 changed files with 6 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
ruoyi-common/ruoyi-common-doc/src/main/java/org/dromara/common/doc/config/SpringDocAutoConfiguration.java
View File

@@ -4,7 +4,9 @@ import io.swagger.v3.oas.models.OpenAPI;
import io.swagger.v3.oas.models.Paths;
import io.swagger.v3.oas.models.info.Info;
import io.swagger.v3.oas.models.security.SecurityRequirement;
import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor;
import org.dromara.common.core.utils.ServletUtils;
import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.doc.config.properties.SpringDocProperties;
import org.dromara.common.doc.handler.OpenApiHandler;
@@ -93,23 +95,17 @@ public class SpringDocAutoConfiguration {
*/
@Bean
public OpenApiCustomizer openApiCustomizer() {
// 拼接服务路径
String appPath = "/" + StringUtils.substring(appName, appName.indexOf("-") + 1);
String contextPath = serverProperties.getServlet().getContextPath();
String finalContextPath;
if (StringUtils.isBlank(contextPath) || "/".equals(contextPath)) {
finalContextPath = appPath;
} else {
finalContextPath = appPath + contextPath;
}
// 对所有路径增加前置上下文路径
return openApi -> {
HttpServletRequest request = ServletUtils.getRequest();
// 从请求头获取gateway转发的服务前缀
String prefix = StringUtils.blankToDefault(request.getHeader("X-Forwarded-Prefix"), "");
Paths oldPaths = openApi.getPaths();
if (oldPaths instanceof PlusPaths) {
return;
}
PlusPaths newPaths = new PlusPaths();
oldPaths.forEach((k, v) -> newPaths.addPathItem(finalContextPath + k, v));
oldPaths.forEach((k, v) -> newPaths.addPathItem(prefix + k, v));
openApi.setPaths(newPaths);
};
}