cerc-io/lotus
16
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

Watch context for completion also

Browse Source
This commit is contained in:
Ingar Shu 2020-08-11 13:48:56 -07:00
parent 53e06f358a
commit 02b0c8dd94
No known key found for this signature in database
GPG Key ID: BE3D9CE79F22E769
2 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/test/deals.go
View File

@ -399,15 +399,9 @@ func testRetrieval(t *testing.T, ctx context.Context, err error, client *impl.Fu
IsCAR: carExport,
}
updates, err := client.ClientRetrieve(ctx, offers[0].Order(caddr), ref)
for {
select {
case update, ok := <-updates:
if !ok {
break
}
if update.Err != nil {
t.Fatalf("%+v", err)
}
for update := range updates {
if update.Err != nil {
t.Fatalf("%+v", err)
}
}

4
cli/client.go
View File

@ -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")
}
}
},