mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-21 06:57:51 +08:00
124 lines
3.5 KiB
Go
124 lines
3.5 KiB
Go
package operation
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb"
|
|
"google.golang.org/grpc"
|
|
"math/rand/v2"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/master_pb"
|
|
)
|
|
|
|
type Location struct {
|
|
Url string `json:"url,omitempty"`
|
|
PublicUrl string `json:"publicUrl,omitempty"`
|
|
DataCenter string `json:"dataCenter,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(context.Background()), 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,
|
|
DataCenter: loc.DataCenter,
|
|
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
|
|
}
|