Python Thrift客户端传输的正确生命周期

时间:2016-01-31 18:55:42

标签: python thrift thrift-protocol

运输生命周期的最佳做法是什么?

我发现每次通话打开一个传输无效。特别是下面的代码似乎花了大约5秒钟打开传输然后每次调用约300ms用于我的实际服务调用。 300毫秒是可以接受的。

我应该只为每个客户端打开一次传输并在应用程序结束时关闭吗?如果我这样做,我会遇到瞬时网络断开吗?如果我这样做,会抛出哪些客户端异常?

我的客户端批量请求,以便客户端不会过于繁琐。意味着每隔30s左右就会发送1MB左右的消息。然后大约2分钟后,我将发送一个大小为15MB到50MB的大消息

def make_transport_context(transport_context, rpc_type):
    # Make socket
    transport = TSSLSocket.TSSLSocket(transport_context.host, transport_context.port,ca_certs=transport_context.ca_certs,validate=False)
    transport.setTimeout(ms=transport_context.timeout_in_millis)
    # Buffering is critical. Raw sockets are very slow
    transport = TTransport.TBufferedTransport(transport)
    # Wrap in a protocol
    protocol = TBinaryProtocol.TBinaryProtocol(transport)
    # Create a client to use the protocol encoder
    client = rpc_type.Client(protocol)
    return client, transport

@contextlib.contextmanager
def rpc_thrift_context(transport_context):
    client, transport = make_transport_context(transport_context,  DataRPC)
    try:
        transport.open()
        yield client
    finally:
        transport.close()

0 个答案:

没有答案