mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
watch: follow changes for a pattern
This commit is contained in:
parent
b7a654a318
commit
63ad1abcce
|
@ -4,6 +4,8 @@ import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/chrislusf/seaweedfs/weed/pb"
|
"github.com/chrislusf/seaweedfs/weed/pb"
|
||||||
|
@ -17,7 +19,7 @@ func init() {
|
||||||
}
|
}
|
||||||
|
|
||||||
var cmdWatch = &Command{
|
var cmdWatch = &Command{
|
||||||
UsageLine: "watch <wip> [-filer=localhost:8888] [-target=/]",
|
UsageLine: "watch [-filer=localhost:8888] [-target=/]",
|
||||||
Short: "see recent changes on a filer",
|
Short: "see recent changes on a filer",
|
||||||
Long: `See recent changes on a filer.
|
Long: `See recent changes on a filer.
|
||||||
|
|
||||||
|
@ -25,15 +27,55 @@ var cmdWatch = &Command{
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
var (
|
||||||
watchFiler = cmdWatch.Flag.String("filer", "localhost:8888", "filer hostname:port")
|
watchFiler = cmdWatch.Flag.String("filer", "localhost:8888", "filer hostname:port")
|
||||||
watchTarget = cmdWatch.Flag.String("pathPrefix", "/", "path to a folder or file, or common prefix for the folders or files on filer")
|
watchTarget = cmdWatch.Flag.String("pathPrefix", "/", "path to a folder or file, or common prefix for the folders or files on filer")
|
||||||
watchStart = cmdWatch.Flag.Duration("timeAgo", 0, "start time before now. \"300ms\", \"1.5h\" or \"2h45m\". Valid time units are \"ns\", \"us\" (or \"µs\"), \"ms\", \"s\", \"m\", \"h\"")
|
watchStart = cmdWatch.Flag.Duration("timeAgo", 0, "start time before now. \"300ms\", \"1.5h\" or \"2h45m\". Valid time units are \"ns\", \"us\" (or \"µs\"), \"ms\", \"s\", \"m\", \"h\"")
|
||||||
|
watchPattern = cmdWatch.Flag.String("pattern", "", "full path or just filename pattern, ex: \"/home/?opher\", \"*.pdf\", see https://golang.org/pkg/path/filepath/#Match ")
|
||||||
)
|
)
|
||||||
|
|
||||||
func runWatch(cmd *Command, args []string) bool {
|
func runWatch(cmd *Command, args []string) bool {
|
||||||
|
|
||||||
grpcDialOption := security.LoadClientTLS(util.GetViper(), "grpc.client")
|
grpcDialOption := security.LoadClientTLS(util.GetViper(), "grpc.client")
|
||||||
|
|
||||||
|
var filterFunc func(dir, fname string) bool
|
||||||
|
if *watchPattern != "" {
|
||||||
|
if strings.Contains(*watchPattern, "/") {
|
||||||
|
println("watch path pattern", *watchPattern)
|
||||||
|
filterFunc = func(dir, fname string) bool {
|
||||||
|
matched, err := filepath.Match(*watchPattern, dir+"/"+fname)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error: %v", err)
|
||||||
|
}
|
||||||
|
return matched
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
println("watch file pattern", *watchPattern)
|
||||||
|
filterFunc = func(dir, fname string) bool {
|
||||||
|
matched, err := filepath.Match(*watchPattern, fname)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error: %v", err)
|
||||||
|
}
|
||||||
|
return matched
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
shouldPrint := func(resp *filer_pb.SubscribeMetadataResponse) bool {
|
||||||
|
if filterFunc == nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
if resp.EventNotification.OldEntry == nil && resp.EventNotification.NewEntry == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if resp.EventNotification.OldEntry != nil && filterFunc(resp.Directory, resp.EventNotification.OldEntry.Name) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
if resp.EventNotification.NewEntry != nil && filterFunc(resp.EventNotification.NewParentPath, resp.EventNotification.NewEntry.Name) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
watchErr := pb.WithFilerClient(*watchFiler, grpcDialOption, func(client filer_pb.SeaweedFilerClient) error {
|
watchErr := pb.WithFilerClient(*watchFiler, grpcDialOption, func(client filer_pb.SeaweedFilerClient) error {
|
||||||
|
|
||||||
stream, err := client.SubscribeMetadata(context.Background(), &filer_pb.SubscribeMetadataRequest{
|
stream, err := client.SubscribeMetadata(context.Background(), &filer_pb.SubscribeMetadataRequest{
|
||||||
|
@ -53,7 +95,10 @@ func runWatch(cmd *Command, args []string) bool {
|
||||||
if listenErr != nil {
|
if listenErr != nil {
|
||||||
return listenErr
|
return listenErr
|
||||||
}
|
}
|
||||||
fmt.Printf("events: %+v\n", resp.EventNotification)
|
if !shouldPrint(resp) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
fmt.Printf("%+v\n", resp.EventNotification)
|
||||||
}
|
}
|
||||||
|
|
||||||
})
|
})
|
||||||
|
|
Loading…
Reference in a new issue