From daf0a449f7424d4a8252673509af5afd0b9bd8ec Mon Sep 17 00:00:00 2001 From: Chris Lu Date: Wed, 9 Sep 2020 12:07:15 -0700 Subject: properly cancel context for streaming grpc --- unmaintained/diff_volume_servers/diff_volume_servers.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'unmaintained/diff_volume_servers/diff_volume_servers.go') diff --git a/unmaintained/diff_volume_servers/diff_volume_servers.go b/unmaintained/diff_volume_servers/diff_volume_servers.go index 4de864980..6107f3d48 100644 --- a/unmaintained/diff_volume_servers/diff_volume_servers.go +++ b/unmaintained/diff_volume_servers/diff_volume_servers.go @@ -124,7 +124,9 @@ type needleState struct { func getVolumeFiles(v uint32, addr string) (map[types.NeedleId]needleState, int64, error) { var idxFile *bytes.Reader err := operation.WithVolumeServerClient(addr, grpcDialOption, func(vs volume_server_pb.VolumeServerClient) error { - copyFileClient, err := vs.CopyFile(context.Background(), &volume_server_pb.CopyFileRequest{ + ctx, cancel := context.WithCancel(context.Background()) + defer cancel() + copyFileClient, err := vs.CopyFile(ctx, &volume_server_pb.CopyFileRequest{ VolumeId: v, Ext: ".idx", CompactionRevision: math.MaxUint32, -- cgit v1.2.3