springcloud gateway聚合swagger2的方法示例

问题描述

在搭建分布式应用时，每个应用通过nacos在网关出装配了路由，我们希望网关也可以将所有的应用的swagger界面聚合起来。这样前端开发的时候只需要访问网关的swagger就可以，而不用访问每个应用的swagger。

框架

springcloud+gateway+nacos+swagger

问题分析

swagger页面是一个单页面应用，所有的显示的数据都是通过和springfox.documentation.swagger.web.ApiResponseController进行数据交互，首先通过/swagger-resources获取swagger资源信息，获取的信息格式如下：[{name: "default", url: "/v2/api-docs", swaggerVersion: "2.0", location: "/v2/api-docs"}]，其中name代表swagger生成的接口组的组名，如图所示：

url

代表swagger接口组的详细信息可以通过 localhost:8081/v2/api-docs来获取，如下图：

在网关处，如果访问/swagger-resources能够获取到所有应用的swagger的资源信息，那么我们的问题就可以解决了，所以我们需要做的是修改/swagger-resources接口的处理方式，使得这个接口能够按照我们的需求返回swagger资源。

解决方案

我们首先在网关处引入swagger的相关依赖，然后实现一个获取其他应用的swagger资源的组件：

  
  @Component
  public class MySwaggerResourceProvider implements SwaggerResourcesProvider {
  
    
    private static final String SWAGGER2URL = "/v2/api-docs";
  
    
    private final RouteLocator routeLocator;
  
    
    @Value("${spring.application.name}")
    private String self;
  
    @Autowired
    public MySwaggerResourceProvider(RouteLocator routeLocator) {
      this.routeLocator = routeLocator;
    }
  
    @Override
    public List get() {
      List resources = new ArrayList<>();
      List routeHosts = new ArrayList<>();
      // 由于我的网关采用的是负载均衡的方式，因此我需要拿到所有应用的serviceId
      // 获取所有可用的host：serviceId
      routeLocator.getRoutes().filter(route -> route.getUri().getHost() != null)
   .filter(route -> !self.equals(route.getUri().getHost()))
   .subscribe(route -> routeHosts.add(route.getUri().getHost()));
  
      // 记录已经添加过的server，存在同一个应用注册了多个服务在nacos上
      Set dealed = new HashSet<>();
      routeHosts.forEach(instance -> {
 // 拼接url，样式为/serviceId/v2/api-info，当网关调用这个接口时，会自动通过负载均衡寻找对应的主机
 String url = "/" + instance + SWAGGER2URL;
 if (!dealed.contains(url)) {
   dealed.add(url);
   SwaggerResource swaggerResource = new SwaggerResource();
   swaggerResource.setUrl(url);
   swaggerResource.setName(instance);
   resources.add(swaggerResource);
 }
      });
      return resources;
    }
  }

然后定义一个接口类：

 
  @RestController
  @RequestMapping("/swagger-resources")
  public class SwaggerResourceController {
    private MySwaggerResourceProvider swaggerResourceProvider;
  
    @Autowired
    public SwaggerResourceController(MySwaggerResourceProvider swaggerResourceProvider) {
      this.swaggerResourceProvider = swaggerResourceProvider;
    }
  
    @RequestMapping(value = "/configuration/security")
    public ResponseEntity securityConfiguration() {
      return new ResponseEntity<>(SecurityConfigurationBuilder.builder().build(), HttpStatus.OK);
    }
  
    @RequestMapping(value = "/configuration/ui")
    public ResponseEntity uiConfiguration() {
      return new ResponseEntity<>(UiConfigurationBuilder.builder().build(), HttpStatus.OK);
    }
  
    @RequestMapping
    public ResponseEntity> swaggerResources() {
      return new ResponseEntity<>(swaggerResourceProvider.get(), HttpStatus.OK);
    }
  }

然后启动网关，访问 http://网关地址/swagger-ui.html，可以看到

在箭头处，可以切换不同应用的swagger页面。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持考高分网。