我的 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)，但我沒有看到任何類似的東西。我是從錯誤的角度考慮這個問題，還是我錯過了其他東西？
2 回答
人到中年有點甜
TA貢獻1895條經驗 獲得超7個贊
您需要使用NewIncomingContext在流中創建當前上下文的副本。
然后，您必須創建一個wrappedStream覆蓋Context方法 in的類型ServerStream以返回修改后的context. 您需要將其傳遞wrappedStream給handler您在攔截器中收到的。
你可以在這里看到一個例子（它在這里覆蓋了其他方法，但想法是一樣的）： https ://github.com/grpc/grpc-go/blob/master/examples/features/interceptor/server/main .go#L106-L124
希望這可以幫助。
慕運維8079593
TA貢獻1876條經驗 獲得超5個贊
您可以創建自己的 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,
})
- 2 回答
- 0 關注
- 157 瀏覽
添加回答
舉報
0/150
提交
取消