继承SpringDoc文档,并只在开发环境展示

135 篇文章 0 订阅
16 篇文章 0 订阅

我选择的是自动注入方式,好处就是不需要在配置文件一个个去配置
yaml配置

spring:
  ‘cloud:
    gateway:
      discovery:
        locator:
          enabled: true

代码

依赖

<dependencies>
       <dependency>
            <groupId>org.springframework.cloud</groupId>
            <artifactId>spring-cloud-starter-gateway</artifactId>
            <version>3.1.1</version>
        </dependency>
        <dependency>
            <groupId>org.springframework.cloud</groupId>
            <artifactId>spring-cloud-starter-netflix-eureka-client</artifactId>
            <version>3.1.1</version>
        </dependency>
        <dependency>
            <groupId>org.springdoc</groupId>
            <artifactId>springdoc-openapi-webflux-ui</artifactId>
            <version>1.6.9</version>
        </dependency>
        <dependency>
            <groupId>com.github.xiaoymin</groupId>
            <artifactId>knife4j-springdoc-ui</artifactId>
            <version>3.0.3</version>
        </dependency>
</dependencies>

注意:

env需要工程能区分出来dev和test等等环境
下面这段代码放在网关模块里

import com.eoi.goblin.common.utils.EmptyUtil;
import org.springdoc.core.AbstractSwaggerUiConfigProperties;
import org.springdoc.core.SwaggerUiConfigProperties;
import org.springframework.cloud.gateway.route.RouteDefinition;
import org.springframework.cloud.gateway.route.RouteDefinitionLocator;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment;

import javax.annotation.PostConstruct;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Set;

/**
 * springdoc配置
 *
 * @author HuangJunHao
 * @since 2022/10/12
 */
@Configuration
public class SpringDocConfig {
    protected final SwaggerUiConfigProperties swaggerUiConfigProperties;
    protected final RouteDefinitionLocator routeDefinitionLocator;

    protected final Environment env;

    public SpringDocConfig(SwaggerUiConfigProperties swaggerUiConfigProperties, RouteDefinitionLocator routeDefinitionLocator, Environment env) {
        this.swaggerUiConfigProperties = swaggerUiConfigProperties;
        this.routeDefinitionLocator = routeDefinitionLocator;
        this.env = env;
    }

    @PostConstruct
    public void autoInitSwaggerUrls() {
        String[] activeProfiles = env.getActiveProfiles();
        //只有dev环境才加载doc文档
        if (EmptyUtil.notEmpty(activeProfiles) && "dev".equals(activeProfiles[0])) {
            List<RouteDefinition> definitions = routeDefinitionLocator.getRouteDefinitions().collectList().block();
            definitions.stream().forEach(routeDefinition -> {
                AbstractSwaggerUiConfigProperties.SwaggerUrl swaggerUrl = new AbstractSwaggerUiConfigProperties.SwaggerUrl(
                        routeDefinition.getId().replace("ReactiveCompositeDiscoveryClient_", "").toLowerCase(),
                        routeDefinition.getUri().toString().replace("lb://", "").toLowerCase() + "/v3/api-docs", ""
                );
                Set<AbstractSwaggerUiConfigProperties.SwaggerUrl> urls = swaggerUiConfigProperties.getUrls();
                if (urls == null) {
                    urls = new LinkedHashSet<>();
                    swaggerUiConfigProperties.setUrls(urls);
                }
                urls.add(swaggerUrl);
            });
        }
    }
}

不是dev访问的就是如下效果
在这里插入图片描述

访问路径

http://gateway-host/doc.html

gateway-host:网关的地址

我的是:http://localhost:8080/doc.html#/home

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值