add stream timeout option which defaults to 0 (#1456)

* add stream timeout option which defaults to 0

* fix option
This commit is contained in:
Asim Aslam
2020-03-31 23:22:11 +01:00
committed by GitHub
parent 1490aff38e
commit 68b0238a5d
3 changed files with 22 additions and 2 deletions

View File

@@ -221,7 +221,9 @@ func (g *grpcClient) stream(ctx context.Context, node *registry.Node, req client
}
// set timeout in nanoseconds
header["timeout"] = fmt.Sprintf("%d", opts.RequestTimeout)
if opts.StreamTimeout > time.Duration(0) {
header["timeout"] = fmt.Sprintf("%d", opts.StreamTimeout)
}
// set the content type for the request
header["x-content-type"] = req.ContentType()