linka-cloud/grpc
2
0
Fork 0
mirror of https://github.com/linka-cloud/grpc.git synced 2024-12-22 17:00:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix metadata client interceptors

Browse Source 
Signed-off-by: Adphi <philippe.adrien.nousse@gmail.com>
This commit is contained in:
Adphi 2022-03-11 12:40:09 +01:00
parent 97ced73270
commit 08a67bf162
Signed by: adphi
GPG Key ID: 46BE4062DB2397FF
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
interceptors/metadata/metadata.go
View File

@ -37,18 +37,14 @@ func (i mdInterceptors) StreamServerInterceptor() grpc.StreamServerInterceptor {
func (i mdInterceptors) UnaryClientInterceptor() grpc.UnaryClientInterceptor {
return func(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
if err := grpc.SetHeader(ctx, metadata.Pairs(i.pairs...)); err != nil {
return err
}
ctx = metadata.NewOutgoingContext(ctx, metadata.Pairs(i.pairs...))
return invoker(ctx, method, req, reply, cc, opts...)
}
}
func (i mdInterceptors) StreamClientInterceptor() grpc.StreamClientInterceptor {
return func(ctx context.Context, desc *grpc.StreamDesc, cc *grpc.ClientConn, method string, streamer grpc.Streamer, opts ...grpc.CallOption) (grpc.ClientStream, error) {
if err := grpc.SetHeader(ctx, metadata.Pairs(i.pairs...)); err != nil {
return nil, err
}
ctx = metadata.NewOutgoingContext(ctx, metadata.Pairs(i.pairs...))
return streamer(ctx, desc, cc, method, opts...)
}
}