diff --git a/transfer.go b/transfer.go index 1e70a69a2..669f90117 100644 --- a/transfer.go +++ b/transfer.go @@ -32,6 +32,12 @@ import ( ) func (c *Client) Transfer(ctx context.Context, src interface{}, dest interface{}, opts ...transfer.Opt) error { + ctx, done, err := c.WithLease(ctx) + if err != nil { + return err + } + defer done(ctx) + return proxy.NewTransferrer(transferapi.NewTransferClient(c.conn), c.streamCreator()).Transfer(ctx, src, dest, opts...) }