接收流时中断python grpc客户端

Question

我正在玩一点 gRPC，但我不知道在接收流时如何关闭客户端和服务器之间的连接。客户端和服务器都是用 Python 编写的。

例如，我的服务器从队列中读取消息并生成每条消息。我的想法是客户端订阅服务器并开始接收这些消息。

我的问题是：

• 我想在按下 CTRL+C 时杀死客户端，但它会卡在当前代码中。怎样才能正确完成？
• 服务器如何意识到客户端已经停止监听？

我的 nbi.proto 文件：

syntax = "proto3";
service Messenger {
    rpc subscribe(Null) return (stream Message) {}
}

message Null {}

message Message {
    string value = 1;
}

Python 客户端：

import test_pb2_grpc as test_grpc
import test_pb2 as test
import grpc

def run():
    channel = grpc.insecure_channel('localhost:50051')
    stub = test_grpc.MessengerStub(channel)

    stream = stub.subscribe(test.Null())
    try:
        for e in stream:
            print e
    except grpc._channel._Rendezvous as err:
        print err
    except KeyboardInterrupt:
        stub.unsuscribe(test.Null)

Python 服务器：

import test_pb2_grpc as test_grpc
import test_pb2 as test
from Queue import Empty
import grpc

class Messenger(test_grpc.MessengerServicer):
    def __init__(self, queue):
        self.queue = queue

    def subscribe(self, request, context):
        while True:
            try:
                yield self.queue.get(block=True, timeout=0.1)
            except Empty:
                continue
            except Exception as e:
                logger.error(e)
                break
        return

Answer 1

我想在按下 CTRL+C 时杀死客户端，但是 它被当前代码卡住了。怎样才能正确完成？

KeyboardInterrupt 应该足以终止客户端应用程序。可能该进程挂在stub.unsubscribe 上。如果使用客户端断线回调，或许不需要显式取消订阅。

服务器如何意识到客户端已经停止监听？

您可以将 add a callback to the context object 传递给您的 Messenger.subscribe 方法。在客户端断开连接时调用回调。

顺便说一句，您可以使用 empty.proto 代替您的 Null 类型。