fix a bug

This commit is contained in:
iamqizhao
2015-03-30 18:00:11 -07:00
parent 8ce50750fe
commit 63de54e02b

View File

@ -198,18 +198,6 @@ func (t *http2Client) NewStream(ctx context.Context, callHdr *CallHdr) (_ *Strea
return nil, ContextErr(context.DeadlineExceeded)
}
}
var authData map[string]string
for _, c := range t.authCreds {
authData, err = c.GetRequestMetadata(ctx)
select {
case <-ctx.Done():
return nil, ContextErr(ctx.Err())
default:
}
if err != nil {
return nil, StreamErrorf(codes.InvalidArgument, "transport: %v", err)
}
}
if _, err := wait(ctx, t.shutdownChan, t.writableChan); err != nil {
return nil, err
}
@ -226,9 +214,14 @@ func (t *http2Client) NewStream(ctx context.Context, callHdr *CallHdr) (_ *Strea
if timeout > 0 {
t.hEnc.WriteField(hpack.HeaderField{Name: "grpc-timeout", Value: timeoutEncode(timeout)})
}
for _, c := range t.authCreds {
if authData, err := c.GetRequestMetadata(ctx); err != nil {
return nil, StreamErrorf(codes.InvalidArgument, "transport: %v", err)
}
for k, v := range authData {
t.hEnc.WriteField(hpack.HeaderField{Name: k, Value: v})
}
}
var (
hasMD bool
endHeaders bool