mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-15 19:26:42 +08:00
125 lines
3.5 KiB
Go
125 lines
3.5 KiB
Go
package localsink
|
|
|
|
import (
|
|
"github.com/seaweedfs/seaweedfs/weed/filer"
|
|
"github.com/seaweedfs/seaweedfs/weed/glog"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/filer_pb"
|
|
"github.com/seaweedfs/seaweedfs/weed/replication/repl_util"
|
|
"github.com/seaweedfs/seaweedfs/weed/replication/sink"
|
|
"github.com/seaweedfs/seaweedfs/weed/replication/source"
|
|
"github.com/seaweedfs/seaweedfs/weed/s3api/s3_constants"
|
|
"github.com/seaweedfs/seaweedfs/weed/util"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
type LocalSink struct {
|
|
Dir string
|
|
filerSource *source.FilerSource
|
|
isIncremental bool
|
|
}
|
|
|
|
func init() {
|
|
sink.Sinks = append(sink.Sinks, &LocalSink{})
|
|
}
|
|
|
|
func (localsink *LocalSink) SetSourceFiler(s *source.FilerSource) {
|
|
localsink.filerSource = s
|
|
}
|
|
|
|
func (localsink *LocalSink) GetName() string {
|
|
return "local"
|
|
}
|
|
|
|
func (localsink *LocalSink) isMultiPartEntry(key string) bool {
|
|
return strings.HasSuffix(key, ".part") && strings.Contains(key, "/"+s3_constants.MultipartUploadsFolder+"/")
|
|
}
|
|
|
|
func (localsink *LocalSink) initialize(dir string, isIncremental bool) error {
|
|
localsink.Dir = dir
|
|
localsink.isIncremental = isIncremental
|
|
return nil
|
|
}
|
|
|
|
func (localsink *LocalSink) Initialize(configuration util.Configuration, prefix string) error {
|
|
dir := configuration.GetString(prefix + "directory")
|
|
isIncremental := configuration.GetBool(prefix + "is_incremental")
|
|
glog.V(4).Infof("sink.local.directory: %v", dir)
|
|
return localsink.initialize(dir, isIncremental)
|
|
}
|
|
|
|
func (localsink *LocalSink) GetSinkToDirectory() string {
|
|
return localsink.Dir
|
|
}
|
|
|
|
func (localsink *LocalSink) IsIncremental() bool {
|
|
return localsink.isIncremental
|
|
}
|
|
|
|
func (localsink *LocalSink) DeleteEntry(key string, isDirectory, deleteIncludeChunks bool, signatures []int32) error {
|
|
if localsink.isMultiPartEntry(key) {
|
|
return nil
|
|
}
|
|
glog.V(4).Infof("Delete Entry key: %s", key)
|
|
if err := os.Remove(key); err != nil {
|
|
glog.V(0).Infof("remove entry key %s: %s", key, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (localsink *LocalSink) CreateEntry(key string, entry *filer_pb.Entry, signatures []int32) error {
|
|
if entry.IsDirectory || localsink.isMultiPartEntry(key) {
|
|
return nil
|
|
}
|
|
glog.V(4).Infof("Create Entry key: %s", key)
|
|
|
|
totalSize := filer.FileSize(entry)
|
|
chunkViews := filer.ViewFromChunks(localsink.filerSource.LookupFileId, entry.GetChunks(), 0, int64(totalSize))
|
|
|
|
dir := filepath.Dir(key)
|
|
|
|
if _, err := os.Stat(dir); os.IsNotExist(err) {
|
|
glog.V(4).Infof("Create Directory key: %s", dir)
|
|
if err = os.MkdirAll(dir, 0755); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if entry.IsDirectory {
|
|
return os.Mkdir(key, os.FileMode(entry.Attributes.FileMode))
|
|
}
|
|
|
|
dstFile, err := os.OpenFile(key, os.O_RDWR|os.O_CREATE|os.O_TRUNC, os.FileMode(entry.Attributes.FileMode))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer dstFile.Close()
|
|
|
|
writeFunc := func(data []byte) error {
|
|
_, writeErr := dstFile.Write(data)
|
|
return writeErr
|
|
}
|
|
|
|
if len(entry.Content) > 0 {
|
|
return writeFunc(entry.Content)
|
|
}
|
|
|
|
if err := repl_util.CopyFromChunkViews(chunkViews, localsink.filerSource, writeFunc); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (localsink *LocalSink) UpdateEntry(key string, oldEntry *filer_pb.Entry, newParentPath string, newEntry *filer_pb.Entry, deleteIncludeChunks bool, signatures []int32) (foundExistingEntry bool, err error) {
|
|
if localsink.isMultiPartEntry(key) {
|
|
return true, nil
|
|
}
|
|
glog.V(4).Infof("Update Entry key: %s", key)
|
|
// do delete and create
|
|
foundExistingEntry = util.FileExists(key)
|
|
err = localsink.CreateEntry(key, newEntry, signatures)
|
|
return
|
|
}
|