Watch context for completion also

2020-08-11 14:43:55 -07:00
parent 53e06f358a
commit 02b0c8dd94
2 changed files with 6 additions and 10 deletions
@@ -860,8 +860,10 @@ var clientRetrieveCmd = &cli.Command{
 				}

 				if evt.Err != nil {
-					return xerrors.Errorf("Retrieval Failed: %w", err)
+					return xerrors.Errorf("retrieval failed: %w", err)
 				}
+			case <-ctx.Done():
+				return xerrors.Errorf("retrieval timed out")
 			}
 		}
 	},