6

我正在使用 gRPC 构建一个 API，在服务器端，我想在客户端断开连接时收到通知，识别它并基于此执行一些任务。

到目前为止，我能够使用grpc.StatsHandler方法检测到客户端断开连接HandleConn。我尝试使用上下文传递值，但无法从服务器端访问它们。

客户端：

conn, err := grpc.DialContext(
    context.WithValue(context.Background(), "user_id", 1234),
    address,
    grpc.WithInsecure(),
)

服务器端：

// Build stats handler
type serverStats struct {}

func (h *serverStats) TagRPC(ctx context.Context, info *stats.RPCTagInfo) context.Context {
    return ctx
}

func (h *serverStats) HandleRPC(ctx context.Context, s stats.RPCStats) {}

func (h *serverStats) TagConn(ctx context.Context, info *stats.ConnTagInfo) context.Context {
    return context.TODO()
}

func (h *serverStats) HandleConn(ctx context.Context, s stats.ConnStats) {
    fmt.Println(ctx.Value("user_id")) // Returns nil, can't access the value

    switch s.(type) {
    case *stats.ConnEnd:
        fmt.Println("client disconnected")
        break
    }
}


// Build server
s := grpc.NewServer(grpc.StatsHandler(&serverStats{}))

我想在服务器端访问从客户端传递的值。什么是正确的方法，或者有没有其他方法可以识别已断开连接的客户端？

4

2 回答 2

1

我认为您在拨号时可能不会传递该值，但您可以标记连接，并且所有下一个客户端请求在其上下文中都将具有相同的值：

在您的 serverStats 实现中：

func (h *serverStats) TagConn(ctx context.Context, info *stats.ConnTagInfo) context.Context {
    return context.WithValue(ctx, "user_counter", userCounter++)
}

在您的服务实施中

func (s *service) SetUserId(ctx context.Context, v MyMessage) (r MyResponse, err Error) {

   s.userIdMap[ctx.Value("user_counter")] = v.userId
...

}

在其他方法中：


func (s *service) AnotherMethod(ctx context.Context, v MyMessage) (r MyResponse, err Error) {

   userId := s.userIdMap[ctx.Value("user_counter")]
...

}

func (h *serverStats) HandleConn(ctx context.Context, s stats.ConnStats) {

    switch s.(type) {
    case *stats.ConnEnd:
        fmt.Printf("client %d disconnected", s.userIdMap[ctx.Value("user_counter")])
        break
    }
}

如果您找到在拨号步骤中传递值的方法，请告诉我。

于 2019-08-21T01:22:15.207 回答
0

我尝试使用上下文传递值，但无法从服务器端访问它们。

您需要将元数据字段显式设置为客户端上下文：

ctx := context.Background()
ctx = metadata.AppendToOutgoingContext(ctx, "user_id", "1234")
conn, err := grpc.DialContext(cxt, address)

在服务器端，您可以像这样检索它们：

md, ok := metadata.FromIncomingContext(ctx)

什么是正确的方法，或者有没有其他方法可以识别已断开连接的客户端？

这实际上取决于您的用例。我认为 GRPC Stats API 适用于一些简单的任务（例如，计算延迟或聚合网络统计信息），但是当客户端离开时某些业务逻辑应该工作时，它并不是很有用。为此，我建议defer直接在 GRPC 处理程序中使用调用。另一种选择是实现自定义 GRPC 拦截器。

于 2019-05-02T18:27:33.397 回答