1. サーバー側インターセプター サーバー側の原則
unary は単項インターセプターで、stream はストリーミング インターセプターです。
この記事では、クライアントが単一の要求をサーバーに送信し、単一の応答を返す最も一般的なケースである、単項インターセプターのみを取り上げます。
新しい grpc サーバーを作成するとき、このメソッドはインターセプターを結合してカプセル化します。
chainUnaryServerInterceptors の特定のコードは、unaryInt と配列 chainUnaryInts を結合するようになりました. インターセプターの数が 1 以下であるというロジックは理解しやすいです. len>1 の場合に注目してください. 関数 chainUnaryInterceptors はインターセプター データを再帰的にカプセル化します完全なインターセプター 。この一般的なインターセプターには、配列内のすべてのインターセプターのロジックが含まれています。
以下はカプセル化プロセスです. chainUnaryInterceptors 関数の入力はインターセプターの配列であり、出力はカプセル化された全インターセプターです. 特定のロジックは、新しくカプセル化されたインターセプターを返します。新しくカプセル化されたインターセプターの結果は、配列内の最初のインターセプターを実行した結果です。最初のインターセプターの結果は、実際にはハンドラーが返すものです。ここでのハンドラーは、getChainUnaryHandler を通じて取得されます。getChainUnaryHandler は再帰プロセスとしてハンドラーを取得し、curr の位置に従って次のインターセプターを呼び出し、最後のインターセプターのハンドラー (finalHandler) が実行されるまで、この finalHandler は grpc の特定のメソッドです。このようにして、すべてのインターセプターのロジックが確実に実行されるように関数のネストが形成され、最後に grpc の特定のメソッドが呼び出されます。
上記はインターセプターを組み立てているだけです。具体的な呼び出しロジックを見てみましょう
Serve のコードが長すぎます。最も重要なものを見て、tcp を監視して接続を返し、ゴルーチンを開いてクライアント要求を処理します
http2 トランスポートを作成したら、goruotine の処理フローを再度開始します。
下までたどり、クライアントのメソッドに合わせてサービスに登録されている具体的なgrpcメソッドを合わせて実行します。作成された pb ファイルには、具体的な実行コードが記述されています。
私のprotoファイル
syntax = "proto3";
package protocol_demo;
option go_package = "lib/proto/protocol_demo";
message HelloWorldReq {
int64 Id = 1;
}
message HelloWorldRsp {
string Message = 2;
}
service HelloWorld {
rpc GetHelloWorld (HelloWorldReq) returns (HelloWorldRsp) {
}
}
生成的bp文件,注册服务的时候将方法名和对应的方法添加到Methods中,本例中对应的方法为_HelloWorld_GetHelloWorld_Handler。上面的md.Handler就是执行的这个方法,看到如果没有拦截器直接执行具体方法。如果有拦截器,则执行拦截器即:return interceptor
这个intercepor就是上文中封装的总拦截器。
以上就是grpc server端拦截器实现的源码逻辑。
二. 实现一个拦截器
package main
import (
"context"
"fmt"
"google.golang.org/grpc"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
"gopractice/lib/proto/protocol_demo"
"net"
"runtime/debug"
"time"
)
func main() {
lis, err := net.Listen("tcp", ":50052")
if err != nil {
fmt.Printf("failed to listen: %s \n", err)
return
}
opts := []grpc.ServerOption{
grpc.ChainUnaryInterceptor(RecoveryInterceptor, LoggerInterceptor),
}
s := grpc.NewServer(opts...)
protocol_demo.RegisterHelloWorldServer(s, &server{})
fmt.Println("success")
err = s.Serve(lis)
if err != nil {
fmt.Printf("failed to start grpc server: %s \n", err)
return
}
}
type server struct {
}
func (s *server) GetHelloWorld(ctx context.Context, req *protocol_demo.HelloWorldReq) (rsp *protocol_demo.HelloWorldRsp, err error) {
rsp = &protocol_demo.HelloWorldRsp{}
rsp.Message = fmt.Sprintf("hello user: %d", req.Id)
fmt.Printf("hello method\n")
return rsp, nil
}
func LoggerInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
t := time.Now()
fmt.Printf("gRpc begin method: method: %s | req: %v | time: %s", info.FullMethod, req, t.Format("2006-01-02 15:04:05.000000"))
fmt.Println()
resp, err = handler(ctx, req)
fmt.Printf("gRpc finish method: %s | rsp: %v | time: %s | durations: %s", info.FullMethod, resp, req, time.Since(t))
fmt.Println()
return
}
func RecoveryInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
defer func() {
if e := recover(); e != nil {
debug.PrintStack()
err = status.Errorf(codes.Internal, "Panic err: %v | %s", e, string(debug.Stack()))
fmt.Println()
}
}()
fmt.Printf("RecoveryInterceptor in\n")
resp, err = handler(ctx, req)
fmt.Printf("RecoveryInterceptor out\n")
return
}
三.客户端拦截器
客户端与服务端类似,直接贴相关代码
组装拦截器的逻辑是一样的,只不过handler变为了invoker
执行调用是有拦截器执行拦截器逻辑,没有直接invoke
创建一个streaming RPC直接发送接收相关数据完成调用
四. 客户端代码实现
package main
import (
"context"
"fmt"
"google.golang.org/grpc"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/connectivity"
"google.golang.org/grpc/credentials/insecure"
"google.golang.org/grpc/status"
"gopractice/lib/proto/protocol_demo"
"runtime/debug"
"time"
)
func main() {
conn, err := grpc.Dial("localhost:50052", grpc.WithTransportCredentials(insecure.NewCredentials()), grpc.WithChainUnaryInterceptor(LoggerInterceptor, RecoveryInterceptor))
if err != nil {
fmt.Printf("new grpc client failed: %s \n", err)
return
}
defer conn.Close()
c := protocol_demo.NewHelloWorldClient(conn)
r, err := c.GetHelloWorld(context.Background(), &protocol_demo.HelloWorldReq{Id: 100})
if err != nil {
fmt.Printf("request GetHelloWorld faild: %s \n", err)
return
}
fmt.Printf("success, message is : %s \n", r.Message)
}
func LoggerInterceptor(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
t := time.Now()
fmt.Printf("gRpc begin method: method: %s | req: %v | time: %s", method, req, t.Format("2006-01-02 15:04:05.000000"))
fmt.Println()
err := invoker(ctx, method, req, reply, cc, opts...)
fmt.Printf("gRpc finish method: %s | rsp: %v | time: %s | durations: %s", method, reply, t.Format("2006-01-02 15:04:05.000000"), time.Since(t))
fmt.Println()
return err
}
func RecoveryInterceptor(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) (err error) {
defer func() {
if e := recover(); e != nil {
debug.PrintStack()
err = status.Errorf(codes.Internal, "Panic err: %v | %s", e, string(debug.Stack()))
fmt.Println()
}
}()
fmt.Printf("RecoveryInterceptor in\n")
err = invoker(ctx, method, req, reply, cc, opts...)
fmt.Printf("RecoveryInterceptor out\n")
return
}