@Component
@Primary
public class DocumentationConfig implements SwaggerResourcesProvider {

    /**
     * sc注册中心客户端发现对象
     */
    @Autowired
    DiscoveryClient client;


    /**
     * sc中不需要生成文档的服务
     */
    @Value("${swagger.exclude}")
    String[] swaggerExclude = new String[]{};


    @Override
    public List<SwaggerResource> get() {
        List<String> services = client.getServices();
        List resources = new ArrayList<>();
        services.forEach(service -> {
            if (!ArrayUtils.contains(swaggerExclude, service)) {
                resources.add(swaggerResource(service, "/" + service + "/v2/api-docs", "2.0"));
            }
        });
        return resources;
    }

    private SwaggerResource swaggerResource(String name, String location, String version) {
        SwaggerResource swaggerResource = new SwaggerResource();
        swaggerResource.setName(name);
        swaggerResource.setLocation(location);
        swaggerResource.setSwaggerVersion(version);
        return swaggerResource;
    }
}