Python grpc - 在发送响应之前读取所有消息

Question

我试图了解使用流的 grpc 服务器是否能够在发送响应之前等待读取所有客户端消息。

我有一个简单的应用程序，我在其中发送了几个我想要添加和返回的号码。 我已经设置了一个基本的原型文件来测试这一点：

syntax = "proto3";


message CalculateRequest{
    int64 x = 1;
    int64 y = 2;
};

message CalculateReply{
    int64 result = 1;
}

service Svc {
    rpc CalculateStream (stream CalculateRequest) returns (stream CalculateReply);
}

在我的服务器端，我已经实现了以下代码，该代码在收到消息时返回答案消息：

class CalculatorServicer(contracts_pb2_grpc.SvcServicer):
    def CalculateStream(self, request_iterator, context):
        for request in request_iterator:
            resultToOutput = request.x + request.y
            yield contracts_pb2.CalculateReply(result=resultToOutput)

def serve():
    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
    contracts_pb2_grpc.add_SvcServicer_to_server(
        CalculatorServicer(), server)
    server.add_insecure_port('localhost:9000')
    server.start()
    server.wait_for_termination()


if __name__ == '__main__':
    print( "We're up")
    logging.basicConfig()
    serve()

我想对此进行调整以首先读取所有数字并然后在稍后阶段发送这些内容 - 如下所示：

class CalculatorServicer(contracts_pb2_grpc.SvcServicer):
   listToReturn = []
   def CalculateStream(self, request_iterator, context):
        for request in request_iterator:
            listToReturn.append (request.x + request.y)
        
        # ...
        # do some other stuff first before returning
        
        for item in listToReturn:
           yield contracts_pb2.CalculateReply(result=resultToOutput)

目前，我稍后写出的实现不起作用，因为从未到达底部的代码。这是设计使连接在到达那里之前似乎“关闭”吗？

grpc.io 网站表明这应该可以通过 BiDirectional 实现流媒体：

例如，服务器可以在写入响应之前等待接收所有客户端消息，或者它可以交替读取消息然后写入消息，或者其他读取和写入的组合。

预先感谢您的任何帮助:)

Answer 1

这里的问题是“所有客户端消息”的定义。在传输级别，服务器无法知道客户端是否已完成，而与客户端关闭其连接无关。

您需要添加一些指示，表明客户端已完成向协议发送请求。向现有的CalculateRequest 添加一个bool 字段，或者添加一个顶级oneof，其中一个选项类似于StopSendingRequests