美文网首页
C# Grpc-AOP编程拦截器(Interceptor)

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

作者: ZeroingX | 来源:发表于2019-08-27 10:14 被阅读0次

    文档不全仍需要补充,具体可以参考 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,仅是重写的方法不同。

    相关文章

      网友评论

          本文标题:C# Grpc-AOP编程拦截器(Interceptor)

          本文链接:https://www.haomeiwen.com/subject/cussectx.html