From a5b979623e65cae9a0d3595a4e16eec95c013ea8 Mon Sep 17 00:00:00 2001
From: Derek McGowan <derek@mcg.dev>
Date: Fri, 30 Dec 2022 16:33:15 -0800
Subject: [PATCH] Add lease to transfer to preserve streams during transfer

Signed-off-by: Derek McGowan <derek@mcg.dev>
---
 transfer.go | 6 ++++++
 1 file changed, 6 insertions(+)

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...)
 }