mirror of
https://github.com/fullstorydev/grpcurl.git
synced 2026-05-22 19:51:44 +03:00
fix nil-dereference panic (#395)
This commit is contained in:
10
invoke.go
10
invoke.go
@@ -264,8 +264,10 @@ func invokeServerStream(ctx context.Context, stub grpcdynamic.Stub, md *desc.Met
|
|||||||
// Now we can actually invoke the RPC!
|
// Now we can actually invoke the RPC!
|
||||||
str, err := stub.InvokeRpcServerStream(ctx, md, req)
|
str, err := stub.InvokeRpcServerStream(ctx, md, req)
|
||||||
|
|
||||||
if respHeaders, err := str.Header(); err == nil {
|
if str != nil {
|
||||||
handler.OnReceiveHeaders(respHeaders)
|
if respHeaders, err := str.Header(); err == nil {
|
||||||
|
handler.OnReceiveHeaders(respHeaders)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Download each response message
|
// Download each response message
|
||||||
@@ -288,7 +290,9 @@ func invokeServerStream(ctx context.Context, stub grpcdynamic.Stub, md *desc.Met
|
|||||||
return fmt.Errorf("grpc call for %q failed: %v", md.GetFullyQualifiedName(), err)
|
return fmt.Errorf("grpc call for %q failed: %v", md.GetFullyQualifiedName(), err)
|
||||||
}
|
}
|
||||||
|
|
||||||
handler.OnReceiveTrailers(stat, str.Trailer())
|
if str != nil {
|
||||||
|
handler.OnReceiveTrailers(stat, str.Trailer())
|
||||||
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user