python 代码分块_用python在grpc中进行分块

这里是chunker的grpc python版本。服务器中分块的主要逻辑是使用Python生成器实现的。

# server.py

_CHUNKER_SIZE = 4

_DATA_TO_SEND = 'Hello gRPC Python World!'

def _chunk_bytes(data, chunker_size):

index = 0

while index < len(data):

yield chunker_pb2.Chunk(

chunk=data[index:index+chunker_size]

)

index += chunker_size

class Chunker(chunker_pb2_grpc.ChunkerServicer):

@staticmethod

def Chunker(request, unused_context):

return _chunk_bytes(

_DATA_TO_SEND,

_CHUNKER_SIZE)

客户端是直接的。它接收响应并连接它们。

with grpc.insecure_channel('localhost:50051') as channel:

stub = chunker_pb2_grpc.ChunkerStub(channel)

response_iterator = stub.Chunker(empty_pb2.Empty())

received_bytes = bytes()

for response in response_iterator:

received_bytes += response.chunk

print('Concatenated Response:')

print(received_bytes)