如何在流式传输未启动时取消gRPC服务端流式请求？

服务端流式传输核心步骤 #

异步gRPC的服务端流式传输，服务端侧主要包含三个关键步骤：

• 请求阶段：服务端启动监听客户端的流式请求
• 写入阶段：当请求阶段关联的标签从完成队列返回后，服务端可通过ServerAsyncWriter::Write向流中写入数据
• 完成阶段：流式传输结束后，调用ServerAsyncWriter::Finish收尾

RPC取消的现有限制 #

理论上可以通过grpc::ServerContext::TryCancel从服务端主动取消RPC，但这个方法有严格的调用时机限制：仅能在请求阶段完成后调用。

如果客户端始终未发起流式请求，请求阶段永远不会完成，此时除了整体关闭服务端外，没有常规方式取消该RPC。而且如果在流式传输启动前（请求阶段完成前）调用grpc::ServerContext::TryCancel，还会直接引发访问冲突。

解决方案 #

针对这种客户端未发起请求时的RPC取消需求，可通过以下两种方式处理：

1. 为监听请求设置超时 #

在初始化ServerAsyncWriter和启动监听前，通过grpc::ServerContext设置RPC超时时间：

grpc::ServerContext ctx;
ctx.set_deadline(std::chrono::system_clock::now() + std::chrono::seconds(30));

当超时时间到达后，gRPC框架会自动终止该未激活的RPC，无需手动调用取消方法，也不会触发访问冲突。

2. 自定义标签跟踪与清理 #

在启动请求监听时，将对应的ServerContext、ServerAsyncWriter以及监听标签关联存储到一个线程安全的管理集合中。当需要主动取消这类未完成请求阶段的RPC时：

• 先从完成队列中移除对应的监听标签，避免后续触发非法回调
• 直接销毁对应的ServerAsyncWriter和ServerContext实例，gRPC框架会自动清理该RPC的相关资源

内容的提问来源于stack exchange，提问作者Elad Maimoni