diff options
Diffstat (limited to 'weed/wdclient/vid_map.go')
| -rw-r--r-- | weed/wdclient/vid_map.go | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/weed/wdclient/vid_map.go b/weed/wdclient/vid_map.go index 7a2a5bb92..9d5e5d378 100644 --- a/weed/wdclient/vid_map.go +++ b/weed/wdclient/vid_map.go @@ -1,6 +1,7 @@ package wdclient import ( + "context" "errors" "fmt" "github.com/seaweedfs/seaweedfs/weed/pb" @@ -21,7 +22,7 @@ type HasLookupFileIdFunction interface { GetLookupFileIdFunction() LookupFileIdFunctionType } -type LookupFileIdFunctionType func(fileId string) (targetUrls []string, err error) +type LookupFileIdFunctionType func(ctx context.Context, fileId string) (targetUrls []string, err error) type Location struct { Url string `json:"url,omitempty"` @@ -99,7 +100,7 @@ func (vc *vidMap) LookupVolumeServerUrl(vid string) (serverUrls []string, err er return } -func (vc *vidMap) LookupFileId(fileId string) (fullUrls []string, err error) { +func (vc *vidMap) LookupFileId(ctx context.Context, fileId string) (fullUrls []string, err error) { parts := strings.Split(fileId, ",") if len(parts) != 2 { return nil, errors.New("Invalid fileId " + fileId) |