小编pen*_*olf的帖子

客户端断开连接时如何正确使用 ctx.Done()?

如果客户端因网络错误而断开连接,则在我的情况下,服务器必须关闭 pub/sub 连接。我了解ctx.Done()功能,但不知道如何在我的情况下正确使用它。有人可以解释一下吗?

grpc-go:1.7.0

go版本go1.8.4

func (a *API) Notifications(in *empty.Empty, stream pb.Service_NotificationsServer) error {
    ctx := stream.Context()
    _, ok := user.FromContext(ctx)
    if !ok {
        return grpc.Errorf(codes.Unauthenticated, "user not found")
    }

    pubsub := a.redisClient.Subscribe("notifications")
    defer pubsub.Close()

    for {
        msg, err := pubsub.ReceiveMessage()
        if err != nil {
            grpclog.Warningf("Notifications: pubsub error: %v", err)
            return grpc.Errorf(codes.Internal, "pubsub error %v", err)
        }

        notification := &pb.Notification{}
        err = json.Unmarshal([]byte(msg.Payload), notification)
        if err != nil {
            grpclog.Warningf("Notifications: parse error: %v", err)
            continue
        }
        if …
Run Code Online (Sandbox Code Playgroud)

channel go grpc

6
推荐指数
1
解决办法
8845
查看次数

标签 统计

channel ×1

go ×1

grpc ×1