如何在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)