package operation import ( "context" "errors" "fmt" "github.com/seaweedfs/seaweedfs/weed/pb" "google.golang.org/grpc" "math/rand" "strings" "time" "github.com/seaweedfs/seaweedfs/weed/pb/master_pb" ) type Location struct { Url string `json:"url,omitempty"` PublicUrl string `json:"publicUrl,omitempty"` GrpcPort int `json:"grpcPort,omitempty"` } func (l *Location) ServerAddress() pb.ServerAddress { return pb.NewServerAddressWithGrpcPort(l.Url, l.GrpcPort) } type LookupResult struct { VolumeOrFileId string `json:"volumeOrFileId,omitempty"` Locations []Location `json:"locations,omitempty"` Jwt string `json:"jwt,omitempty"` Error string `json:"error,omitempty"` } func (lr *LookupResult) String() string { return fmt.Sprintf("VolumeOrFileId:%s, Locations:%v, Error:%s", lr.VolumeOrFileId, lr.Locations, lr.Error) } var ( vc VidCache // caching of volume locations, re-check if after 10 minutes ) func LookupFileId(masterFn GetMasterFn, grpcDialOption grpc.DialOption, fileId string) (fullUrl string, jwt string, err error) { parts := strings.Split(fileId, ",") if len(parts) != 2 { return "", jwt, errors.New("Invalid fileId " + fileId) } lookup, lookupError := LookupVolumeId(masterFn, grpcDialOption, parts[0]) if lookupError != nil { return "", jwt, lookupError } if len(lookup.Locations) == 0 { return "", jwt, errors.New("File Not Found") } return "http://" + lookup.Locations[rand.Intn(len(lookup.Locations))].Url + "/" + fileId, lookup.Jwt, nil } func LookupVolumeId(masterFn GetMasterFn, grpcDialOption grpc.DialOption, vid string) (*LookupResult, error) { results, err := LookupVolumeIds(masterFn, grpcDialOption, []string{vid}) return results[vid], err } // LookupVolumeIds find volume locations by cache and actual lookup func LookupVolumeIds(masterFn GetMasterFn, grpcDialOption grpc.DialOption, vids []string) (map[string]*LookupResult, error) { ret := make(map[string]*LookupResult) var unknown_vids []string //check vid cache first for _, vid := range vids { locations, cacheErr := vc.Get(vid) if cacheErr == nil { ret[vid] = &LookupResult{VolumeOrFileId: vid, Locations: locations} } else { unknown_vids = append(unknown_vids, vid) } } //return success if all volume ids are known if len(unknown_vids) == 0 { return ret, nil } //only query unknown_vids err := WithMasterServerClient(false, masterFn(), grpcDialOption, func(masterClient master_pb.SeaweedClient) error { req := &master_pb.LookupVolumeRequest{ VolumeOrFileIds: unknown_vids, } resp, grpcErr := masterClient.LookupVolume(context.Background(), req) if grpcErr != nil { return grpcErr } //set newly checked vids to cache for _, vidLocations := range resp.VolumeIdLocations { var locations []Location for _, loc := range vidLocations.Locations { locations = append(locations, Location{ Url: loc.Url, PublicUrl: loc.PublicUrl, GrpcPort: int(loc.GrpcPort), }) } if vidLocations.Error != "" { vc.Set(vidLocations.VolumeOrFileId, locations, 10*time.Minute) } ret[vidLocations.VolumeOrFileId] = &LookupResult{ VolumeOrFileId: vidLocations.VolumeOrFileId, Locations: locations, Jwt: vidLocations.Auth, Error: vidLocations.Error, } } return nil }) if err != nil { return nil, err } return ret, nil }