如何在Python中使用gRPC处理自定义异常?

问题描述

我需要实现自定义异常,以使用Python处理gRPC请求错误。对于HTTP请求,它很简单-请求库在出现错误代码等时可以很好地捕获它。我正在寻找类似的方式让gRPC执行类似的操作:

try:
    # send gRPC request
except SomeGRPCException as e:
    # custom handle

有没有办法像Python那样处理gRPC错误?还是使用gRPC不能像示例中那样工作?

解决方法

对于简单的RPC错误处理,可以使用try-catch:

try:
    response = stub.SayHello(...)
except grpc.RpcError as rpc_error:
    if rpc_error.code() == grpc.StatusCode.CANCELLED:
        pass
    elif rpc_error.code() == grpc.StatusCode.UNAVAILABLE:
        pass
    else:
        print(f"Received unknown RPC error: code={rpc_error.code()} message={rpc_error.details()}")

对于复杂的RPC错误处理,您可能需要利用尾随的元数据: https://github.com/grpc/grpc/tree/master/examples/python/errors

在大多数情况下,try-catch应该足够了。