手动造轮子——为Ocelot集成Nacos注册中心

vlambda
2020-07-25

手动造轮子——为Ocelot集成Nacos注册中心

前言

近期在看博客的时候或者在群里看聊天的时候，发现很多都提到了Ocelot网关的问题。我之前也研究过一点，网关本身是一种通用的解决方案，主要的工作就是拦截请求统一处理，比如认证、授权、熔断、限流、注册发现、负载均衡等等。随着服务化的不断盛行，服务拆分，负载均衡等已成为当今软件行业随处可谈的名词了，因此注册中心也随之流行了起来。Ocelot作为网关自然可以集成许多注册中心，官方文档给出了集成Eureka和Consul的解决方案，Eureka可能有的人不是很熟悉，它是Spring Cloud的核心组件之一，其功能就是服务注册发现。随着.Net Core的不断成熟，不知道为啥Consul突然成了.Net Core注册中心和配置中心的主要选择，甚至可以说是首选了，可能是因为功能比较强大，而且是基于GO开发的。Nacos作为后起之秀，功能也非常强大。那天无意中翻了一下发下网上居然没有Ocelot集成到Nacos注册中心的组件，由于我个人非常喜欢通用解决方案,于是决定自己扩展一个Ocelot.Provider.Nacos，代码已经放到了我的GitHub上https://github.com/softlgl/Ocelot.Provider.Nacos，有兴趣的可自行查阅。

概念介绍

Ocelot

Ocelot是一个用.NET Core实现并且开源的API网关，它具备了许多强大实用的功能，包括了：路由、请求聚合、服务发现、认证、鉴权、限流熔断、并内置了负载均衡器与Service Fabric、Butterfly Tracing集成。它是由asp.net core middleware组成的一个管道。当获取请求之后会用request builder来构造一个HttpRequestMessage转发到下游的真实服务器，等下游的服务返回response之后再由一个middleware将它返回的HttpResponseMessage映射到HttpResponse上。

Nacos

关于Nacos我之前的文章搭建一套ASP.NET Core+Nacos+Spring Cloud Gateway项目已经有过介绍了。Nacos是阿里巴巴开源的致力于服务发现、配置和管理微服务的框架。提供了一组简单易用的特性集，帮助您快速实现动态服务发现、服务配置、服务元数据及流量管理。一般用到的最多的就是当做配置中心和注册中心。

Ocelot.Provider.Nacos

开发环境

基于.Net Core 3.1，这个是必须的，因为最新稳定版的Ocelot是在.Net Core 3.1上构建的，而我恰好选择的是这个版本的Ocelot
Ocelot选择的是目前最新的稳定版本 v16.0.1
Nacos访问组件实用的是nacos-sdk-csharp，具体引用的是

<PackageReference Include="nacos-sdk-csharp-unofficial" Version="0.2.7" />

它其实是有另一个nacos-sdk-csharp-unofficial.AspNetCore版本的这个是针对Asp.Net Core程序集成Nacos使用的，Ocelot也是基于Asp.Net Core搭建起来的，但是我没有选用nacos-sdk-csharp-unofficial.AspNetCore，主要是因为虽然AspNetCore版本的集成起来更方便，但是为了让它能更好的适配到Ocelot服务注册和发现上，我需要自己改造一下原本的使用方式。

集成到Ocelot

将Ocelot.Provider.Nacos的方式非常简单基本上和Ocelot.Provider.Eureka和Ocelot.Provider.Consul是一致的，首先新建一个已经集成了Ocelot的Asp.Net Core项目，这里就不演示如何搭建的了，如果有不熟悉的可以查看我的demo示例https://github.com/softlgl/Ocelot.Provider.Nacos/tree/master/demo/ApiGatewayDemo，项目搭建好之后引入Ocelot.Provider.Nacos包

<PackageReference Include="Ocelot.Provider.Nacos" Version="1.0.0" />

然后在ConfigureServices中添加注册方法AddNacosDiscovery

public void ConfigureServices(IServiceCollection services){ //注册服务发现 services.AddOcelot().AddNacosDiscovery();}

代码上做这么多就可以了，其他的主要工作就在配置上了，近期新版本的Ocelot相对于之前老版本有些地方改动还是非常大的，网上很多示例都是老版本的，所以参考官方文档搭建还是比较靠谱的。接下来我们打开Ocelot的配置文件配置注册中心相关的

{ "Routes": [ { // 用于服务发现的名称，也就是注册到nacos上的名称 "ServiceName": "productservice", "DownstreamScheme": "http", "DownstreamPathTemplate": "/productapi/{everything}", "UpstreamPathTemplate": "/productapi/{everything}", "UpstreamHttpMethod": [ "Get", "Post" ], "LoadBalancerOptions": { "Type": "RoundRobin"  }, // 使用服务发现 "UseServiceDiscovery": true } ], "GlobalConfiguration": { "ServiceDiscoveryProvider": { // 这里是重点 "Type": "Nacos" } }}

"nacos": { "ServerAddresses": [ "http://localhost:8848" ], "DefaultTimeOut": 15000, "Namespace": "", "ListenInterval": 1000, // 网关注册的服务名称 "ServiceName": "apigateway"}

自定义扩展代码

代码基本上我是参考着Ocelot.Provider.Eureka和Ocelot.Provider.Consul相关代码写的。其中入口类就一个是针对IOcelotBuilder的扩展类。
首先,将自定义的服务发现相关的服务注册进来。只展示Ocelot适配相关的，我自己写的关于服务注册发现相关的就不做展示了，有兴趣的可自行查阅

public static IOcelotBuilder AddNacosDiscovery(this IOcelotBuilder builder){ //添加注册自定义的NacosDiscovery相关的服务 builder.Services.AddNacosDiscovery(builder.Configuration); //添加自定义服务发现代理 builder.Services.AddSingleton<ServiceDiscoveryFinderDelegate>(NacosProviderFactory.Get); //根据Ocelot配置文件相关内容设置处理服务发现相关 builder.Services.AddSingleton<OcelotMiddlewareConfigurationDelegate>(NacosMiddlewareConfigurationProvider.Get); return builder;}

public static class NacosProviderFactory{ public static ServiceDiscoveryFinderDelegate Get = (provider, config, route) => { //Nacos相关服务类 var client = provider.GetService<INacosServerManager>(); //判断类型是否为nacos if (config.Type?.ToLower() == "nacos" && client != null) { //返回自定义服务提供操作，route.ServiceName是命中的Route的配置的服务发现的名称 return new Nacos(route.ServiceName, client); } return null; };}
public class Nacos : IServiceDiscoveryProvider{ private readonly INacosServerManager _client; private readonly string _serviceName;
 public Nacos(string serviceName, INacosServerManager client) { _client = client; _serviceName = serviceName; }
 public async Task<List<Service>> Get() { var services = new List<Service>(); var instances = await _client.GetServerAsync(_serviceName); if (instances != null && instances.Any()) { //将发现的地址组装成Service集合即可 services.AddRange(instances.Select(i => new Service(i.InstanceId, new ServiceHostAndPort(i.Ip, i.Port), "", "", new List<string>()))); }
 return await Task.FromResult(services); }}

最后，根据Ocelot配置相关的信息判断是否启动Nacos相关服务

public class NacosMiddlewareConfigurationProvider{ public static OcelotMiddlewareConfigurationDelegate Get = builder => { var internalConfigRepo = builder.ApplicationServices.GetService<IInternalConfigurationRepository>(); var config = internalConfigRepo.Get(); var hostLifetime = builder.ApplicationServices.GetService<IHostApplicationLifetime>(); //判断服务注册类型是否为nacos if (UsingNacosServiceDiscoveryProvider(config.Data)) { //启动nacos相关服务 builder.UseNacosDiscovery(hostLifetime).GetAwaiter().GetResult(); } return Task.CompletedTask; };
 private static bool UsingNacosServiceDiscoveryProvider(IInternalConfiguration configuration) { //判断配置的服务发现类型是否为nacos return configuration?.ServiceProviderConfiguration != null && configuration.ServiceProviderConfiguration.Type?.ToLower() == "nacos"; }}

涉及到的相关代码并不多，而且比较清晰，了解到相关规则后还是比较简单的。

开发中遇到困难

Ocelot.Provider.Nacos一共大概有十个类左右，集成到Ocleot本身相关类有四个吧，其他的类都是集成Nacos相关的。

后来写的差不多的时候准备调试输入具体转发路径的时候一直报UnableToFindServiceDiscoveryProviderError:Unable to find service discovery provider for type: nacos，刚开始我一直没有找到原因，明明我已经给IServiceDiscoveryProvider注册了NacosDiscovery实例它却还是说找不到，找了老长时间后来我想到了去Ocelot源码查到UnableToFindServiceDiscoveryProviderError的引用，最后在ServiceDiscoveryProviderFactory类，看到了如下代码，Ocelot的规则就是注册的Type类型要和自定义的IServiceDiscoveryProvider实现类名称一致才可以

private Response<IServiceDiscoveryProvider> GetServiceDiscoveryProvider(ServiceProviderConfiguration config, DownstreamRoute route){ if (config.Type?.ToLower() == "servicefabric") { var sfConfig = new ServiceFabricConfiguration(config.Host, config.Port, route.ServiceName); return new OkResponse<IServiceDiscoveryProvider>(new ServiceFabricServiceDiscoveryProvider(sfConfig)); }
 if (_delegates != null) { var provider = _delegates?.Invoke(_provider, config, route); //1.自定义的IServiceDiscoveryProvider实现类名称要和ServiceDiscoveryProvider里的Type名称一致 if (provider.GetType().Name.ToLower() == config.Type.ToLower()) { return new OkResponse<IServiceDiscoveryProvider>(provider); } } //2.否则就会返回这异常信息 return new ErrorResponse<IServiceDiscoveryProvider>(new UnableToFindServiceDiscoveryProviderError($"Unable to find service discovery provider for type: {config.Type}"));}

后来把我的实现类名称改成了Nacos果然可以了，当时出这个异常的时候非常困惑，还好我没放弃。

总结

总之，通过本次扩展，踩了很多坑，也加深了相应的了解，希望能为有自定义扩展注册中心的同学提供一点帮助。Ocelot本身文档非常详细，但是对于扩展这一块没有过多的介绍，只能通过查看相关源码去了解。我觉得许多东西既然开源了，需要有自定义需求的可以通过查看源码解决问题。虽然查看源码其实并不容易，如果有针对性的话，相对来说还不是很复杂。重点是遇到问题你怎么解决，能否扛得住这种无助的心情，只能靠自己，说不定坚持一下就找到思路了。

vlambda博客
学习文章列表