2 回答

TA貢獻(xiàn)1895條經(jīng)驗(yàn) 獲得超7個(gè)贊
您需要使用NewIncomingContext在流中創(chuàng)建當(dāng)前上下文的副本。
然后,您必須創(chuàng)建一個(gè)wrappedStream
覆蓋Context
方法 in的類型ServerStream
以返回修改后的context
. 您需要將其傳遞wrappedStream
給handler
您在攔截器中收到的。
你可以在這里看到一個(gè)例子(它在這里覆蓋了其他方法,但想法是一樣的): https ://github.com/grpc/grpc-go/blob/master/examples/features/interceptor/server/main .go#L106-L124
希望這可以幫助。

TA貢獻(xiàn)1876條經(jīng)驗(yàn) 獲得超5個(gè)贊
您可以創(chuàng)建自己的 ServerStream 實(shí)現(xiàn),用您自己的上下文覆蓋 Context() 方法,或者在 grpc 包中有一個(gè)結(jié)構(gòu)WrappedServerStream (github.com/grpc-ecosystem/go-grpc-middleware),您可以在其中傳遞上下文和原始服務(wù)器流對(duì)象并在其中使用它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 關(guān)注
- 156 瀏覽
添加回答
舉報(bào)