C# Grpc-AOP编程拦截器(Interceptor)

2019-08-27  本文已影响0人  ZeroingX

文档不全仍需要补充,具体可以参考 Grpc.Core.Api/Interceptors

.Net 中的 Grpc 是在 v1.10.0 版本 起支持 AOP .源码中看到是2018年2月12日 传上去的pre 版
Grpc预留给开发者 抽象类 Interceptor(位于命名空间 Grpc.Core.Interceptors),拦截功能主要在该类中实现
通过它可以做架构上很多事情,这是Grpc深入时必须掌握的
可以做日志、监控、权限控制、还可以做 tag缓存。

服务端拦截器 实现方式(以简单模式为例)

A、定义拦截器类:override Interceptor抽象类的 UnaryServerHandler(简单模式) 方法
B、实现拦截器方法:在 调用 continuation(request, context); 前后实现 拦截功的代码
C、使用拦截器:在 创建服务时,使用BindService(**).Intercept(new 自定义拦截器) 。

注:continuation(request, context) 就是会直接进入 我们实现的业务代码 Grpc服务中。
注:可重写方法有:客户端流模式(ClientStreamingServerHandler)、服务器端流模式(ServerStreamingServerHandler)、双向模式的拦截(DuplexStreamingServerHandler)

A 定义拦截器类 : ServerCallContextInterceptor .cs

public class ServerCallContextInterceptor : Interceptor
{
    #region Override Methods
    public override Task<TResponse> UnaryServerHandler<TRequest, TResponse>(TRequest request, ServerCallContext context, UnaryServerMethod<TRequest, TResponse> continuation)
    {
             // B、实现拦截器方法
             // 写请求 成功前 的代码
             // 获取 客户端传来的自定义报头(参考客户端拦截器实现)
             context.RequestHeaders.Last(m => (m.Key == "Surf-Token")).Value;
             context.RequestHeaders.Last(m => (m.Key == "RequestId")).Value;

            return continuation(request, context);
             // 写请求 完成后 的代码
             // 或向 客户端附加 一些信息
             // 也可以 用try catch 做异常日志
             // 可以从 context中取出 调用方ip,做ip限制
             // 可以 监控continuation 的 执行时间
    }
    //... 可重写 客户端流模式(ClientStreamingServerHandler)、服务器端流模式(ServerStreamingServerHandler)、双向模式的拦截(DuplexStreamingServerHandler)
        #endregion
    }

C、使用拦截器:BindService(**).Intercept(new 自定义拦截器)在程序启动时 Program.cs

Main()
{
    var userGService = UserService.BindService(EngineContext.Resolve<UserServiceBase>());

    var server = new Server(channelOptions)
    {
        Services = {
           userGService.Intercept(new ServerCallContextInterceptor()),
        },
        Ports = { new ServerPort(AppSettings.Host, AppSettings.Port, ServerCredentials.Insecure) }
    };
    server.Start();
}

客户端拦截器 实现方式

客户端是在Channel上实现拦截器的,当new Channel()后就可以调用 Channel.Intercept方法
Intercept 方法有两个重载 :

A、一个接受收委托,在委托中实现拦截器代码
      CallInvoker Intercept(this ChannelBase channel, Func<Metadata, Metadata> interceptor)
B、一个接收 拦截器数组,需自定义拦截器
      CallInvoker Intercept(this ChannelBase channel, params Interceptor[] interceptors)

A、接受收委托的方式: Intercept(... Func<Metadata, Metadata> interceptor) 方法的使用

代码里主要是用于增加发送的报头,代码如:

var channel = new Channel("127.0.0.1",8001);
channel.Intercept(metadata =>
            {
                metadata = metadata ?? new Metadata();
                metadata.Add(new Metadata.Entry("RequestId", "我是 明道云AI服务 在调用您 "));
                metadata.Add(new Metadata.Entry("Surf-Token", "我是调用密钥"));
                return metadata;
            });

B、一个接收 拦截器数组:Intercept(... params Interceptor[] ) 方法的使用

服务端实现拦截器的方法相同,需要 override Interceptor抽象类的几个方法,如:BlockingUnaryCall、AsyncUnaryCall、AsyncClientStreamingCall等。

调用方法 仍是 channel.Intercept,如 channel.Intercept(new ClientCallContextInterceptor());

ClientCallContextInterceptor的实现参考上面 ServerCallContextInterceptor,仅是重写的方法不同。

上一篇 下一篇

猜你喜欢

热点阅读