2018-05-11 09:20:15 +00:00
|
|
|
package filer2
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
|
|
|
|
"github.com/karlseguin/ccache"
|
|
|
|
"strings"
|
2018-05-12 20:45:29 +00:00
|
|
|
"path/filepath"
|
|
|
|
"time"
|
|
|
|
"os"
|
2018-05-11 09:20:15 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type Filer struct {
|
|
|
|
master string
|
|
|
|
store FilerStore
|
|
|
|
directoryCache *ccache.Cache
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewFiler(master string) *Filer {
|
|
|
|
return &Filer{
|
|
|
|
master: master,
|
|
|
|
directoryCache: ccache.New(ccache.Configure().MaxSize(1000).ItemsToPrune(100)),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-05-12 20:45:29 +00:00
|
|
|
func (f *Filer) SetStore(store FilerStore) () {
|
|
|
|
f.store = store
|
2018-05-11 09:20:15 +00:00
|
|
|
}
|
|
|
|
|
2018-05-12 20:45:29 +00:00
|
|
|
func (f *Filer) DisableDirectoryCache() () {
|
|
|
|
f.directoryCache = nil
|
|
|
|
}
|
2018-05-11 09:20:15 +00:00
|
|
|
|
2018-05-12 20:45:29 +00:00
|
|
|
func (f *Filer) CreateEntry(entry *Entry) (error) {
|
|
|
|
|
|
|
|
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.Join(dirParts[:i]...)
|
|
|
|
// fmt.Printf("%d directory: %+v\n", i, dirPath)
|
|
|
|
|
|
|
|
dirFound := false
|
|
|
|
|
|
|
|
// first check local cache
|
|
|
|
dirEntry := f.cacheGetDirectory(dirPath)
|
|
|
|
|
|
|
|
// not found, check the store directly
|
|
|
|
if dirEntry == nil {
|
|
|
|
var dirFindErr error
|
|
|
|
dirFound, dirEntry, dirFindErr = f.FindEntry(FullPath(dirPath))
|
|
|
|
if dirFindErr != nil {
|
|
|
|
return fmt.Errorf("findDirectory %s: %v", dirPath, dirFindErr)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// no such existing directory
|
|
|
|
if !dirFound {
|
|
|
|
|
|
|
|
// create the directory
|
|
|
|
now := time.Now()
|
|
|
|
|
|
|
|
dirEntry = &Entry{
|
|
|
|
FullPath: FullPath(dirPath),
|
|
|
|
Attr: Attr{
|
|
|
|
Mtime: now,
|
|
|
|
Crtime: now,
|
|
|
|
Mode: os.ModeDir | 0660,
|
|
|
|
Uid: entry.Uid,
|
|
|
|
Gid: entry.Gid,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
mkdirErr := f.store.InsertEntry(dirEntry)
|
|
|
|
if mkdirErr != nil {
|
|
|
|
return fmt.Errorf("mkdir %s: %v", dirPath, mkdirErr)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// 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)
|
|
|
|
}
|
|
|
|
|
|
|
|
if !hasWritePermission(lastDirectoryEntry, entry) {
|
|
|
|
return fmt.Errorf("no write permission in folder %v", lastDirectoryEntry.FullPath)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := f.store.InsertEntry(entry); err != nil {
|
|
|
|
return fmt.Errorf("insert entry %s: %v", entry.FullPath, err)
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
2018-05-11 09:20:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (f *Filer) AppendFileChunk(p FullPath, c FileChunk) (err error) {
|
|
|
|
return f.store.AppendFileChunk(p, c)
|
|
|
|
}
|
|
|
|
|
2018-05-12 20:45:29 +00:00
|
|
|
func (f *Filer) FindEntry(p FullPath) (found bool, entry *Entry, err error) {
|
2018-05-11 09:20:15 +00:00
|
|
|
return f.store.FindEntry(p)
|
|
|
|
}
|
|
|
|
|
2018-05-12 20:45:29 +00:00
|
|
|
func (f *Filer) DeleteEntry(p FullPath) (fileEntry *Entry, err error) {
|
2018-05-14 06:56:16 +00:00
|
|
|
found, entry, err := f.FindEntry(p)
|
|
|
|
if err != nil || !found {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if entry.IsDirectory() {
|
|
|
|
entries, err := f.ListDirectoryEntries(p, "", false, 1)
|
|
|
|
if err != nil {
|
|
|
|
return nil, fmt.Errorf("list folder %s: %v", p, err)
|
|
|
|
}
|
|
|
|
if len(entries) > 0 {
|
|
|
|
return nil, fmt.Errorf("folder %s is not empty", p)
|
|
|
|
}
|
|
|
|
}
|
2018-05-11 09:20:15 +00:00
|
|
|
return f.store.DeleteEntry(p)
|
|
|
|
}
|
|
|
|
|
2018-05-13 21:02:29 +00:00
|
|
|
func (f *Filer) ListDirectoryEntries(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 {
|
2018-05-12 20:45:29 +00:00
|
|
|
p = p[0:len(p)-1]
|
|
|
|
}
|
2018-05-13 21:02:29 +00:00
|
|
|
return f.store.ListDirectoryEntries(p, startFileName, inclusive, limit)
|
2018-05-11 09:20:15 +00:00
|
|
|
}
|
|
|
|
|
2018-05-12 20:45:29 +00:00
|
|
|
func (f *Filer) cacheGetDirectory(dirpath string) (*Entry) {
|
|
|
|
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
|
|
|
}
|