mirror of
https://github.com/chrislusf/seaweedfs
synced 2024-06-01 00:02:22 +02:00
9f9ef1340c
streaming mode would create separate grpc connections for each call. this is to ensure the long poll connections are properly closed.
124 lines
3 KiB
Go
124 lines
3 KiB
Go
package command
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/chrislusf/seaweedfs/weed/filer"
|
|
"github.com/chrislusf/seaweedfs/weed/pb"
|
|
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
|
|
"github.com/chrislusf/seaweedfs/weed/wdclient"
|
|
"google.golang.org/grpc"
|
|
"math"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/security"
|
|
"github.com/chrislusf/seaweedfs/weed/util"
|
|
)
|
|
|
|
var (
|
|
filerCat FilerCatOptions
|
|
)
|
|
|
|
type FilerCatOptions struct {
|
|
grpcDialOption grpc.DialOption
|
|
filerAddress pb.ServerAddress
|
|
filerClient filer_pb.SeaweedFilerClient
|
|
output *string
|
|
}
|
|
|
|
func (fco *FilerCatOptions) GetLookupFileIdFunction() wdclient.LookupFileIdFunctionType {
|
|
return func(fileId string) (targetUrls []string, err error) {
|
|
vid := filer.VolumeId(fileId)
|
|
resp, err := fco.filerClient.LookupVolume(context.Background(), &filer_pb.LookupVolumeRequest{
|
|
VolumeIds: []string{vid},
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
locations := resp.LocationsMap[vid]
|
|
for _, loc := range locations.Locations {
|
|
targetUrls = append(targetUrls, fmt.Sprintf("http://%s/%s", loc.Url, fileId))
|
|
}
|
|
return
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
cmdFilerCat.Run = runFilerCat // break init cycle
|
|
filerCat.output = cmdFilerCat.Flag.String("o", "", "write to file instead of stdout")
|
|
}
|
|
|
|
var cmdFilerCat = &Command{
|
|
UsageLine: "filer.cat [-o <file>] http://localhost:8888/path/to/file",
|
|
Short: "copy one file to local",
|
|
Long: `read one file to stdout or write to a file
|
|
|
|
`,
|
|
}
|
|
|
|
func runFilerCat(cmd *Command, args []string) bool {
|
|
|
|
util.LoadConfiguration("security", false)
|
|
|
|
if len(args) == 0 {
|
|
return false
|
|
}
|
|
filerSource := args[len(args)-1]
|
|
|
|
filerUrl, err := url.Parse(filerSource)
|
|
if err != nil {
|
|
fmt.Printf("The last argument should be a URL on filer: %v\n", err)
|
|
return false
|
|
}
|
|
urlPath := filerUrl.Path
|
|
if strings.HasSuffix(urlPath, "/") {
|
|
fmt.Printf("The last argument should be a file: %v\n", err)
|
|
return false
|
|
}
|
|
|
|
filerCat.filerAddress = pb.ServerAddress(filerUrl.Host)
|
|
filerCat.grpcDialOption = security.LoadClientTLS(util.GetViper(), "grpc.client")
|
|
|
|
dir, name := util.FullPath(urlPath).DirAndName()
|
|
|
|
writer := os.Stdout
|
|
if *filerCat.output != "" {
|
|
|
|
fmt.Printf("saving %s to %s\n", filerSource, *filerCat.output)
|
|
|
|
f, err := os.OpenFile(*filerCat.output, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0755)
|
|
if err != nil {
|
|
fmt.Printf("open file %s: %v\n", *filerCat.output, err)
|
|
return false
|
|
}
|
|
defer f.Close()
|
|
writer = f
|
|
}
|
|
|
|
pb.WithFilerClient(false, filerCat.filerAddress, filerCat.grpcDialOption, func(client filer_pb.SeaweedFilerClient) error {
|
|
|
|
request := &filer_pb.LookupDirectoryEntryRequest{
|
|
Name: name,
|
|
Directory: dir,
|
|
}
|
|
respLookupEntry, err := filer_pb.LookupEntry(client, request)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(respLookupEntry.Entry.Content) > 0 {
|
|
_, err = writer.Write(respLookupEntry.Entry.Content)
|
|
return err
|
|
}
|
|
|
|
filerCat.filerClient = client
|
|
|
|
return filer.StreamContent(&filerCat, writer, respLookupEntry.Entry.Chunks, 0, math.MaxInt64)
|
|
|
|
})
|
|
|
|
return true
|
|
}
|