主页 / coding / 问题 / 77841103
Accepted
NotX
Asked: 2024-01-19 00:51:00 +0800 CST

链自定义 GRPC 客户端拦截器/DialOptions

  • 772

我想链接一些DialOptions/客户端拦截器。但由于某种原因，只会调用最新的自定义拦截器：

myOpt1 := grpc.WithUnaryInterceptor(func(ctx context.Context, method string, req, reply any, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
    print("testInterceptor invoked")
    return invoker(ctx, method, req, reply, cc, opts...)
})
myOpt2 := grpc.WithUnaryInterceptor(func(ctx context.Context, method string, req, reply any, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
    print("testInterceptor2 invoked")
    return invoker(ctx, method, req, reply, cc, opts...)
})
opts := []grpc.DialOption{
        grpc.WithTransportCredentials(insecure.NewCredentials()),
        myOpt1, // will be skipped
        myOpt2, // will get invoked
}

conn, err := grpc.DialContext(context.Background(), "my-adress:443", opts...) // err is nil
myService := pbAugment.NewMyServiceClient(conn)
myService.MyFunction(context.Background()) // prints: testInterceptor2 invoked

我已经添加了，TransportCredentials这样我就不会在启动时收到错误（关于缺少传输安全性）。

我在这里缺少什么？

go

1 个回答

  1. Best Answer

    您必须链接（客户端|服务器）拦截器：

    看grpc.WithChainUnaryInterceptor

    例如：

    func main() {
    myInt1 := func(ctx context.Context, method string, req, reply any, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
        print("testInterceptor invoked")
        return invoker(ctx, method, req, reply, cc, opts...)
    }
    myInt2 := func(ctx context.Context, method string, req, reply any, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
        print("testInterceptor2 invoked")
        return invoker(ctx, method, req, reply, cc, opts...)
    }
    opts := []grpc.DialOption{
        grpc.WithTransportCredentials(insecure.NewCredentials()),
        grpc.WithChainUnaryInterceptor(
            myInt1,
            myInt2,
        ),
    }

    _, err := grpc.DialContext(context.Background(), "my-adress:443", opts...)
    if err != nil {
        log.Fatal(err)
    }
}
    • 0

相关问题