seaweedfs/weed/filer2/filer.go

298 lines
6.8 KiB
Go
Raw Normal View History

2018-05-11 09:20:15 +00:00
package filer2
import (
2018-07-29 04:02:56 +00:00
"context"
2018-05-11 09:20:15 +00:00
"fmt"
2019-02-18 20:11:52 +00:00
"google.golang.org/grpc"
2018-11-18 19:51:38 +00:00
"math"
2018-05-27 18:52:26 +00:00
"os"
2018-05-12 20:45:29 +00:00
"path/filepath"
2018-05-27 18:52:26 +00:00
"strings"
2018-05-12 20:45:29 +00:00
"time"
2018-07-29 04:02:56 +00:00
"github.com/chrislusf/seaweedfs/weed/glog"
2018-07-28 09:10:32 +00:00
"github.com/chrislusf/seaweedfs/weed/wdclient"
2018-07-29 04:02:56 +00:00
"github.com/karlseguin/ccache"
2018-05-11 09:20:15 +00:00
)
2018-12-03 03:42:50 +00:00
var (
OS_UID = uint32(os.Getuid())
OS_GID = uint32(os.Getgid())
)
2018-05-11 09:20:15 +00:00
type Filer struct {
2018-11-21 04:56:28 +00:00
store FilerStore
directoryCache *ccache.Cache
MasterClient *wdclient.MasterClient
fileIdDeletionChan chan string
2019-02-18 20:11:52 +00:00
GrpcDialOption grpc.DialOption
2018-05-11 09:20:15 +00:00
}
2019-02-18 20:11:52 +00:00
func NewFiler(masters []string, grpcDialOption grpc.DialOption) *Filer {
2018-11-21 04:56:28 +00:00
f := &Filer{
directoryCache: ccache.New(ccache.Configure().MaxSize(1000).ItemsToPrune(100)),
2019-02-18 20:11:52 +00:00
MasterClient: wdclient.NewMasterClient(context.Background(), grpcDialOption, "filer", masters),
2018-11-21 04:56:28 +00:00
fileIdDeletionChan: make(chan string, 4096),
2019-02-18 20:11:52 +00:00
GrpcDialOption: grpcDialOption,
2018-05-11 09:20:15 +00:00
}
2018-11-21 04:56:28 +00:00
go f.loopProcessingDeletion()
return f
2018-05-11 09:20:15 +00:00
}
2018-05-27 18:52:26 +00:00
func (f *Filer) SetStore(store FilerStore) {
2018-05-12 20:45:29 +00:00
f.store = store
2018-05-11 09:20:15 +00:00
}
2018-05-27 18:52:26 +00:00
func (f *Filer) DisableDirectoryCache() {
2018-05-12 20:45:29 +00:00
f.directoryCache = nil
}
2018-05-11 09:20:15 +00:00
2018-07-28 09:10:32 +00:00
func (fs *Filer) GetMaster() string {
return fs.MasterClient.GetMaster()
2018-07-28 09:10:32 +00:00
}
func (fs *Filer) KeepConnectedToMaster() {
fs.MasterClient.KeepConnectedToMaster()
2018-07-28 09:10:32 +00:00
}
func (f *Filer) BeginTransaction(ctx context.Context) (context.Context, error) {
return f.store.BeginTransaction(ctx)
}
func (f *Filer) CommitTransaction(ctx context.Context) error {
return f.store.CommitTransaction(ctx)
}
func (f *Filer) RollbackTransaction(ctx context.Context) error {
return f.store.RollbackTransaction(ctx)
}
2019-03-15 22:55:34 +00:00
func (f *Filer) CreateEntry(ctx context.Context, entry *Entry) error {
2018-05-12 20:45:29 +00:00
2018-12-03 07:16:17 +00:00
if string(entry.FullPath) == "/" {
return nil
}
2018-05-12 20:45:29 +00:00
dirParts := strings.Split(string(entry.FullPath), "/")
// fmt.Printf("directory parts: %+v\n", dirParts)
var lastDirectoryEntry *Entry
for i := 1; i < len(dirParts); i++ {
dirPath := "/" + filepath.ToSlash(filepath.Join(dirParts[:i]...))
2018-05-12 20:45:29 +00:00
// fmt.Printf("%d directory: %+v\n", i, dirPath)
// first check local cache
dirEntry := f.cacheGetDirectory(dirPath)
// not found, check the store directly
if dirEntry == nil {
glog.V(4).Infof("find uncached directory: %s", dirPath)
2019-03-15 22:55:34 +00:00
dirEntry, _ = f.FindEntry(ctx, FullPath(dirPath))
} else {
glog.V(4).Infof("found cached directory: %s", dirPath)
2018-05-12 20:45:29 +00:00
}
// no such existing directory
if dirEntry == nil {
2018-05-12 20:45:29 +00:00
// create the directory
now := time.Now()
dirEntry = &Entry{
FullPath: FullPath(dirPath),
Attr: Attr{
Mtime: now,
Crtime: now,
2018-05-21 08:25:30 +00:00
Mode: os.ModeDir | 0770,
2018-05-12 20:45:29 +00:00
Uid: entry.Uid,
Gid: entry.Gid,
},
}
2018-05-21 08:25:30 +00:00
glog.V(2).Infof("create directory: %s %v", dirPath, dirEntry.Mode)
2019-03-15 22:55:34 +00:00
mkdirErr := f.store.InsertEntry(ctx, dirEntry)
2018-05-12 20:45:29 +00:00
if mkdirErr != nil {
2019-03-15 22:55:34 +00:00
if _, err := f.FindEntry(ctx, FullPath(dirPath)); err == ErrNotFound {
return fmt.Errorf("mkdir %s: %v", dirPath, mkdirErr)
}
} else {
f.NotifyUpdateEvent(nil, dirEntry, false)
2018-05-12 20:45:29 +00:00
}
2018-08-13 08:20:49 +00:00
} else if !dirEntry.IsDirectory() {
return fmt.Errorf("%s is a file", dirPath)
2018-05-12 20:45:29 +00:00
}
// cache the directory entry
f.cacheSetDirectory(dirPath, dirEntry, i)
// remember the direct parent directory entry
if i == len(dirParts)-1 {
lastDirectoryEntry = dirEntry
}
}
2018-05-11 09:20:15 +00:00
2018-05-12 20:45:29 +00:00
if lastDirectoryEntry == nil {
return fmt.Errorf("parent folder not found: %v", entry.FullPath)
}
/*
2018-05-27 18:52:26 +00:00
if !hasWritePermission(lastDirectoryEntry, entry) {
glog.V(0).Infof("directory %s: %v, entry: uid=%d gid=%d",
lastDirectoryEntry.FullPath, lastDirectoryEntry.Attr, entry.Uid, entry.Gid)
return fmt.Errorf("no write permission in folder %v", lastDirectoryEntry.FullPath)
}
*/
2018-05-12 20:45:29 +00:00
2019-03-15 22:55:34 +00:00
oldEntry, _ := f.FindEntry(ctx, entry.FullPath)
if oldEntry == nil {
2019-03-15 22:55:34 +00:00
if err := f.store.InsertEntry(ctx, entry); err != nil {
return fmt.Errorf("insert entry %s: %v", entry.FullPath, err)
}
} else {
2019-03-15 22:55:34 +00:00
if err := f.UpdateEntry(ctx, oldEntry, entry); err != nil {
return fmt.Errorf("update entry %s: %v", entry.FullPath, err)
}
2018-05-12 20:45:29 +00:00
}
2018-09-17 07:27:56 +00:00
f.NotifyUpdateEvent(oldEntry, entry, true)
2018-08-13 08:20:49 +00:00
f.deleteChunksIfNotNew(oldEntry, entry)
2018-05-12 20:45:29 +00:00
return nil
2018-05-11 09:20:15 +00:00
}
2019-03-15 22:55:34 +00:00
func (f *Filer) UpdateEntry(ctx context.Context, oldEntry, entry *Entry) (err error) {
if oldEntry != nil {
if oldEntry.IsDirectory() && !entry.IsDirectory() {
return fmt.Errorf("existing %s is a directory", entry.FullPath)
}
if !oldEntry.IsDirectory() && entry.IsDirectory() {
return fmt.Errorf("existing %s is a file", entry.FullPath)
}
}
2019-03-15 22:55:34 +00:00
return f.store.UpdateEntry(ctx, entry)
2018-05-11 09:20:15 +00:00
}
2019-03-15 22:55:34 +00:00
func (f *Filer) FindEntry(ctx context.Context, p FullPath) (entry *Entry, err error) {
2018-12-03 03:42:50 +00:00
now := time.Now()
if string(p) == "/" {
return &Entry{
FullPath: p,
Attr: Attr{
Mtime: now,
Crtime: now,
2018-12-03 03:59:47 +00:00
Mode: os.ModeDir | 0755,
2018-12-03 03:42:50 +00:00
Uid: OS_UID,
Gid: OS_GID,
},
}, nil
}
2019-03-15 22:55:34 +00:00
return f.store.FindEntry(ctx, p)
2018-05-11 09:20:15 +00:00
}
2019-03-15 22:26:09 +00:00
func (f *Filer) DeleteEntryMetaAndData(ctx context.Context, p FullPath, isRecursive bool, shouldDeleteChunks bool) (err error) {
2019-03-15 22:55:34 +00:00
entry, err := f.FindEntry(ctx, p)
if err != nil {
return err
2018-05-14 06:56:16 +00:00
}
2018-05-14 06:56:16 +00:00
if entry.IsDirectory() {
2018-08-01 08:26:41 +00:00
limit := int(1)
if isRecursive {
limit = math.MaxInt32
}
lastFileName := ""
includeLastFile := false
for limit > 0 {
2019-03-15 22:55:34 +00:00
entries, err := f.ListDirectoryEntries(ctx, p, lastFileName, includeLastFile, 1024)
if err != nil {
return fmt.Errorf("list folder %s: %v", p, err)
}
if len(entries) == 0 {
break
} else {
if isRecursive {
for _, sub := range entries {
lastFileName = sub.Name()
2019-03-15 22:26:09 +00:00
f.DeleteEntryMetaAndData(ctx, sub.FullPath, isRecursive, shouldDeleteChunks)
limit--
if limit <= 0 {
break
}
}
} else {
if len(entries) > 0 {
return fmt.Errorf("folder %s is not empty", p)
}
}
f.cacheDelDirectory(string(p))
if len(entries) < 1024 {
break
}
}
2018-05-14 06:56:16 +00:00
}
}
2018-06-07 05:11:01 +00:00
if shouldDeleteChunks {
f.DeleteChunks(entry.Chunks)
2018-06-07 05:11:01 +00:00
}
2018-08-01 08:26:41 +00:00
if p == "/" {
return nil
}
2018-08-19 08:27:30 +00:00
glog.V(3).Infof("deleting entry %v", p)
2018-08-13 08:20:49 +00:00
2018-09-17 07:27:56 +00:00
f.NotifyUpdateEvent(entry, nil, shouldDeleteChunks)
2018-08-13 08:20:49 +00:00
2019-03-15 22:55:34 +00:00
return f.store.DeleteEntry(ctx, p)
2018-05-11 09:20:15 +00:00
}
2019-03-15 22:55:34 +00:00
func (f *Filer) ListDirectoryEntries(ctx context.Context, p FullPath, startFileName string, inclusive bool, limit int) ([]*Entry, error) {
2018-05-14 06:56:16 +00:00
if strings.HasSuffix(string(p), "/") && len(p) > 1 {
p = p[0 : len(p)-1]
2018-05-12 20:45:29 +00:00
}
2019-03-15 22:55:34 +00:00
return f.store.ListDirectoryEntries(ctx, p, startFileName, inclusive, limit)
2018-05-11 09:20:15 +00:00
}
func (f *Filer) cacheDelDirectory(dirpath string) {
if f.directoryCache == nil {
return
}
f.directoryCache.Delete(dirpath)
return
}
2018-05-27 18:52:26 +00:00
func (f *Filer) cacheGetDirectory(dirpath string) *Entry {
2018-05-12 20:45:29 +00:00
if f.directoryCache == nil {
return nil
2018-05-11 09:20:15 +00:00
}
2018-05-12 20:45:29 +00:00
item := f.directoryCache.Get(dirpath)
if item == nil {
2018-05-11 09:20:15 +00:00
return nil
}
2018-05-12 20:45:29 +00:00
return item.Value().(*Entry)
}
2018-05-11 09:20:15 +00:00
2018-05-12 20:45:29 +00:00
func (f *Filer) cacheSetDirectory(dirpath string, dirEntry *Entry, level int) {
2018-05-11 09:20:15 +00:00
2018-05-12 20:45:29 +00:00
if f.directoryCache == nil {
return
2018-05-11 09:27:57 +00:00
}
2018-05-12 20:45:29 +00:00
minutes := 60
if level < 10 {
minutes -= level * 6
2018-05-11 09:20:15 +00:00
}
2018-05-12 20:45:29 +00:00
f.directoryCache.Set(dirpath, dirEntry, time.Duration(minutes)*time.Minute)
2018-05-11 09:20:15 +00:00
}