You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
1.8 KiB
79 lines
1.8 KiB
package shell
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/filer2"
|
|
"github.com/chrislusf/seaweedfs/weed/notification"
|
|
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
|
|
"github.com/chrislusf/seaweedfs/weed/util"
|
|
)
|
|
|
|
func init() {
|
|
Commands = append(Commands, &commandFsMetaNotify{})
|
|
}
|
|
|
|
type commandFsMetaNotify struct {
|
|
}
|
|
|
|
func (c *commandFsMetaNotify) Name() string {
|
|
return "fs.meta.notify"
|
|
}
|
|
|
|
func (c *commandFsMetaNotify) Help() string {
|
|
return `recursively send directory and file meta data to notifiction message queue
|
|
|
|
fs.meta.notify # send meta data from current directory to notification message queue
|
|
|
|
The message queue will use it to trigger replication from this filer.
|
|
|
|
`
|
|
}
|
|
|
|
func (c *commandFsMetaNotify) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
|
|
|
|
filerServer, filerPort, path, err := commandEnv.parseUrl(findInputDirectory(args))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
util.LoadConfiguration("notification", true)
|
|
v := viper.GetViper()
|
|
notification.LoadConfiguration(v.Sub("notification"))
|
|
|
|
ctx := context.Background()
|
|
|
|
var dirCount, fileCount uint64
|
|
|
|
err = doTraverseBFS(ctx, writer, commandEnv.getFilerClient(filerServer, filerPort), filer2.FullPath(path), func(parentPath filer2.FullPath, entry *filer_pb.Entry) {
|
|
|
|
if entry.IsDirectory {
|
|
dirCount++
|
|
} else {
|
|
fileCount++
|
|
}
|
|
|
|
notifyErr := notification.Queue.SendMessage(
|
|
string(parentPath.Child(entry.Name)),
|
|
&filer_pb.EventNotification{
|
|
NewEntry: entry,
|
|
},
|
|
)
|
|
|
|
if notifyErr != nil {
|
|
fmt.Fprintf(writer, "fail to notify new entry event for %s: %v\n", parentPath.Child(entry.Name), notifyErr)
|
|
}
|
|
|
|
})
|
|
|
|
if err == nil {
|
|
fmt.Fprintf(writer, "\ntotal notified %d directories, %d files\n", dirCount, fileCount)
|
|
}
|
|
|
|
return err
|
|
|
|
}
|