一. 前言
前面兩篇文章給大家介紹了使用gRPC的入門以及雙向流的使用，今天介紹的是gRPC中的攔截器。攔截器就像MVC的過濾器或者是ASP.NET Core middleware 一樣，具有面向切面的思想，可以在調用服務的時候進行一些統一處理， 很適合在這里處理驗證、日志等流程。本片文章就以記錄日志為例來進行講解。
二. Interceptor 類介紹
Interceptor類是gRPC服務攔截器的基類，是一個抽象類，它定了幾個虛方法，分別如下：
|
1
2
3
4
5
6
7
8
9
|
public virtual TResponse BlockingUnaryCall<TRequest, TResponse>();public virtual AsyncUnaryCall<TResponse> AsyncUnaryCall<TRequest, TResponse>();public virtual AsyncServerStreamingCall<TResponse> AsyncServerStreamingCall<TRequest, TResponse>();public virtual AsyncClientStreamingCall<TRequest, TResponse> AsyncClientStreamingCall<TRequest, TResponse>();public virtual AsyncDuplexStreamingCall<TRequest, TResponse> AsyncDuplexStreamingCall<TRequest, TResponse>();public virtual Task<TResponse> UnaryServerHandler<TRequest, TResponse>();public virtual Task<TResponse> ClientStreamingServerHandler<TRequest, TResponse>();public virtual Task ServerStreamingServerHandler<TRequest, TResponse>();public virtual Task DuplexStreamingServerHandler<TRequest, TResponse>(); |
各個方法作用如下：
| 方法名稱 | 作用 |
|---|---|
| BlockingUnaryCall | 攔截阻塞調用 |
| AsyncUnaryCall | 攔截異步調用 |
| AsyncServerStreamingCall | 攔截異步服務端流調用 |
| AsyncClientStreamingCall | 攔截異步客戶端流調用 |
| AsyncDuplexStreamingCall | 攔截異步雙向流調用 |
| UnaryServerHandler | 用于攔截和傳入普通調用服務器端處理程序 |
| ClientStreamingServerHandler | 用于攔截客戶端流調用的服務器端處理程序 |
| ServerStreamingServerHandler | 用于攔截服務端流調用的服務器端處理程序 |
| DuplexStreamingServerHandler | 用于攔截雙向流調用的服務器端處理程序 |
在實際使用中，可以根據自己的需要來使用對應的攔截方法。
三. 客戶端攔截器
基于前面兩篇文章使用的Demo。
在客戶端項目新建一個類，命名為 ClientLoggerInterceptor，繼承攔截器基類 Interceptor。
我們在前面使用的Demo，定義了擼貓服務，其中 SuckingCatAsync方法為異步調用，所以我們重寫攔截器的 AsyncUnaryCall方法
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
public class ClientLoggerInterceptor:Interceptor{ public override AsyncUnaryCall<TResponse> AsyncUnaryCall<TRequest, TResponse>( TRequest request, ClientInterceptorContext<TRequest, TResponse> context, AsyncUnaryCallContinuation<TRequest, TResponse> continuation) { LogCall(context.Method); return continuation(request, context); } private void LogCall<TRequest, TResponse>(Method<TRequest, TResponse> method) where TRequest : class where TResponse : class { var initialColor = Console.ForegroundColor; Console.ForegroundColor = ConsoleColor.Green; Console.WriteLine($"Starting call. Type: {method.Type}. Request: {typeof(TRequest)}. Response: {typeof(TResponse)}"); Console.ForegroundColor = initialColor; }} |
注冊攔截器：
|
1
2
3
4
5
|
var channel = GrpcChannel.ForAddress("https://localhost:5001");var invoker = channel.Intercept(new ClientLoggerInterceptor());var catClient = new LuCat.LuCatClient(invoker);var catReply = await catClient.SuckingCatAsync(new Empty());Console.WriteLine("調用擼貓服務："+ catReply.Message); |
然后運行：
可以看到成功的在客戶端攔截到了調用，并記錄了調用信息。
四. 服務端攔截器
在服務端項目新建一個類，命名為 ServerLoggerInterceptor，繼承攔截器基類 Interceptor。
我們在服務端需要實現的方法是 UnaryServerHandler
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
public class ServerLoggerInterceptor: Interceptor{ private readonly ILogger<ServerLoggerInterceptor> _logger; public ServerLoggerInterceptor(ILogger<ServerLoggerInterceptor> logger) { _logger = logger; } public override Task<TResponse> UnaryServerHandler<TRequest, TResponse>( TRequest request, ServerCallContext context, UnaryServerMethod<TRequest, TResponse> continuation) { LogCall<TRequest, TResponse>(MethodType.Unary, context); return continuation(request, context); } private void LogCall<TRequest, TResponse>(MethodType methodType, ServerCallContext context) where TRequest : class where TResponse : class { _logger.LogWarning($"Starting call. Type: {methodType}. Request: {typeof(TRequest)}. Response: {typeof(TResponse)}"); }} |
注冊攔截器：
|
1
2
3
4
5
6
7
|
public void ConfigureServices(IServiceCollection services){ services.AddGrpc(options => { options.Interceptors.Add<ServerLoggerInterceptor>(); });} |
運行：
可以看到服務端成功攔截到了，客戶端的調用。
五. 參考資料
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持服務器之家。
原文鏈接：https://www.cnblogs.com/stulzq/p/11840535.html
