mirror of
https://github.com/chrislusf/seaweedfs
synced 2024-08-20 00:30:52 +02:00
66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package command
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"time"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/pb"
|
|
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
|
|
"github.com/chrislusf/seaweedfs/weed/security"
|
|
"github.com/chrislusf/seaweedfs/weed/util"
|
|
)
|
|
|
|
func init() {
|
|
cmdWatch.Run = runWatch // break init cycle
|
|
}
|
|
|
|
var cmdWatch = &Command{
|
|
UsageLine: "watch <wip> [-filer=localhost:8888] [-target=/]",
|
|
Short: "see recent changes on a filer",
|
|
Long: `See recent changes on a filer.
|
|
|
|
`,
|
|
}
|
|
|
|
var (
|
|
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")
|
|
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\"")
|
|
)
|
|
|
|
func runWatch(cmd *Command, args []string) bool {
|
|
|
|
grpcDialOption := security.LoadClientTLS(util.GetViper(), "grpc.client")
|
|
|
|
watchErr := pb.WithFilerClient(*watchFiler, grpcDialOption, func(client filer_pb.SeaweedFilerClient) error {
|
|
|
|
stream, err := client.SubscribeMetadata(context.Background(), &filer_pb.SubscribeMetadataRequest{
|
|
ClientName: "watch",
|
|
PathPrefix: *watchTarget,
|
|
SinceNs: time.Now().Add(-*watchStart).UnixNano(),
|
|
})
|
|
if err != nil {
|
|
return fmt.Errorf("listen: %v", err)
|
|
}
|
|
|
|
for {
|
|
resp, listenErr := stream.Recv()
|
|
if listenErr == io.EOF {
|
|
return nil
|
|
}
|
|
if listenErr != nil {
|
|
return listenErr
|
|
}
|
|
fmt.Printf("events: %+v\n", resp.EventNotification)
|
|
}
|
|
|
|
})
|
|
if watchErr != nil {
|
|
fmt.Printf("watch %s: %v\n", *watchFiler, watchErr)
|
|
}
|
|
|
|
return true
|
|
}
|