如何在拦截器中安全地将值添加到 grpc ServerStream

Question

我的 grpc 服务器有一个日志拦截器，并且想向元数据添加一个值（我想在整个生命周期内跟踪请求）：

func (m *middleware) loggingInterceptor(srv interface{},

    ss grpc.ServerStream,

    info *grpc.StreamServerInfo,

    handler grpc.StreamHandler) 

    md, ok := metadata.FromIncomingContext(ss.Context())

    if !ok {

        return errors.New("could not get metadata from incoming stream context")

    }

    // add the transaction id to the metadata so that business logic can track it

    md.Append("trans-id", "some-transaction-id")

    // call the handler func

    return handler(srv, ss)

}

但文档FromIncomingContext声明：

// FromIncomingContext returns the incoming metadata in ctx if it exists.  The

// returned MD should not be modified. Writing to it may cause races.

// Modification should be made to copies of the returned MD.

好的，所以我看一下复制功能并复制元数据：

mdCopy := md.Copy()

mdCopy.Append("trans-id", "some-transaction-id")

并想“我如何将这个元数据附加回ServerStream上下文？”，然后我检查是否有一些ss.SetContext(newCtx)，但我没有看到任何类似的东西。我是从错误的角度考虑这个问题，还是我错过了其他东西？

Answer 1

您需要使用NewIncomingContext在流中创建当前上下文的副本。然后，您必须创建一个wrappedStream覆盖Context方法 in的类型ServerStream以返回修改后的context. 您需要将其传递wrappedStream给handler您在拦截器中收到的。你可以在这里看到一个例子（它在这里覆盖了其他方法，但想法是一样的）： https ://github.com/grpc/grpc-go/blob/master/examples/features/interceptor/server/main .go#L106-L124希望这可以帮助。

Answer 2

您可以创建自己的 ServerStream 实现，用您自己的上下文覆盖 Context() 方法，或者在 grpc 包中有一个结构WrappedServerStream (github.com/grpc-ecosystem/go-grpc-middleware)，您可以在其中传递上下文和原始服务器流对象并在其中使用它handler。例子：    // This methods get the current context, and creates a new one    newContext, err := interceptor.authorize(ss.Context(), info.FullMethod)    if err != nil {        log.Printf("authorization failed: %v", err)        return err    }        err = handler(srv, &grpc_middleware.WrappedServerStream{        ServerStream:   ss,        WrappedContext: newContext,    })