diff options
Diffstat (limited to 'weed/command')
| -rw-r--r-- | weed/command/benchmark.go | 4 | ||||
| -rw-r--r-- | weed/command/filer_cat.go | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/weed/command/benchmark.go b/weed/command/benchmark.go index 08db2ef3d..715ae3d67 100644 --- a/weed/command/benchmark.go +++ b/weed/command/benchmark.go @@ -241,7 +241,7 @@ func writeFiles(idChan chan int, fileIdLineChan chan string, s *stat) { Replication: *b.replication, DiskType: *b.diskType, } - if assignResult, err := operation.Assign(b.masterClient.GetMaster, b.grpcDialOption, ar); err == nil { + if assignResult, err := operation.Assign(context.Background(), b.masterClient.GetMaster, b.grpcDialOption, ar); err == nil { fp.Server, fp.Fid, fp.Pref.Collection = assignResult.Url, assignResult.Fid, *b.collection if !isSecure && assignResult.Auth != "" { isSecure = true @@ -288,7 +288,7 @@ func readFiles(fileIdLineChan chan string, s *stat) { start := time.Now() var bytesRead int var err error - urls, err := b.masterClient.LookupFileId(fid) + urls, err := b.masterClient.LookupFileId(context.Background(), fid) if err != nil { s.failed++ println("!!!! ", fid, " location not found!!!!!") diff --git a/weed/command/filer_cat.go b/weed/command/filer_cat.go index 2c0f84ddc..136440109 100644 --- a/weed/command/filer_cat.go +++ b/weed/command/filer_cat.go @@ -28,9 +28,9 @@ type FilerCatOptions struct { } func (fco *FilerCatOptions) GetLookupFileIdFunction() wdclient.LookupFileIdFunctionType { - return func(fileId string) (targetUrls []string, err error) { + return func(ctx context.Context, fileId string) (targetUrls []string, err error) { vid := filer.VolumeId(fileId) - resp, err := fco.filerClient.LookupVolume(context.Background(), &filer_pb.LookupVolumeRequest{ + resp, err := fco.filerClient.LookupVolume(ctx, &filer_pb.LookupVolumeRequest{ VolumeIds: []string{vid}, }) if err != nil { |