python - 如何在 thrift python 客户端中设置 rpc 超时？

Question

我正在使用 thrift 编写 python 客户端，但我找不到任何可用的选项来设置 rpc 超时。

我的客户端代码如下:

socket = TSocket.TSocket(address, port)
transport = TTransport.TBufferedTransport(socket)
protocol = TBinaryProtocol.TBinaryProtocol(transport)
server = Client.Client(protocol)
transport.open()

Answer 1

您可以使用socket.setTimeout() 方法。

from thrift.transport.THttpClient import THttpClient
socket = THttpClient(server_url)
socket.setTimeout(SERVICE_TIMEOUT_IN_mS)
transport = TTransport.TBufferedTransport(socket)
protocol = TBinaryProtocol.TBinaryProtocol(transport)