mirror of
https://github.com/chrislusf/seaweedfs
synced 2024-09-04 16:10:36 +02:00
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package mount
|
|
|
|
import (
|
|
"github.com/chrislusf/seaweedfs/weed/glog"
|
|
"github.com/chrislusf/seaweedfs/weed/util"
|
|
"google.golang.org/grpc"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/pb"
|
|
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
|
|
)
|
|
|
|
var _ = filer_pb.FilerClient(&WFS{})
|
|
|
|
func (wfs *WFS) WithFilerClient(streamingMode bool, fn func(filer_pb.SeaweedFilerClient) error) (err error) {
|
|
|
|
return util.Retry("filer grpc", func() error {
|
|
|
|
i := wfs.option.filerIndex
|
|
n := len(wfs.option.FilerAddresses)
|
|
for x := 0; x < n; x++ {
|
|
|
|
filerGrpcAddress := wfs.option.FilerAddresses[i].ToGrpcAddress()
|
|
err = pb.WithGrpcClient(streamingMode, func(grpcConnection *grpc.ClientConn) error {
|
|
client := filer_pb.NewSeaweedFilerClient(grpcConnection)
|
|
return fn(client)
|
|
}, filerGrpcAddress, wfs.option.GrpcDialOption)
|
|
|
|
if err != nil {
|
|
glog.V(0).Infof("WithFilerClient %d %v: %v", x, filerGrpcAddress, err)
|
|
} else {
|
|
wfs.option.filerIndex = i
|
|
return nil
|
|
}
|
|
|
|
i++
|
|
if i >= n {
|
|
i = 0
|
|
}
|
|
|
|
}
|
|
return err
|
|
})
|
|
|
|
}
|
|
|
|
func (wfs *WFS) AdjustedUrl(location *filer_pb.Location) string {
|
|
if wfs.option.VolumeServerAccess == "publicUrl" {
|
|
return location.PublicUrl
|
|
}
|
|
return location.Url
|
|
}
|