Chris Lu
7 years ago
4 changed files with 213 additions and 0 deletions
-
25weed/filer2/embedded/embedded_store.go
-
94weed/filer2/filer.go
-
62weed/filer2/filer_structure.go
-
32weed/filer2/filer_test.go
@ -0,0 +1,25 @@ |
|||
package embedded |
|||
|
|||
import ( |
|||
"github.com/syndtr/goleveldb/leveldb" |
|||
) |
|||
|
|||
type EmbeddedStore struct { |
|||
db *leveldb.DB |
|||
} |
|||
|
|||
func NewEmbeddedStore(dir string) (filer *EmbeddedStore, err error) { |
|||
filer = &EmbeddedStore{} |
|||
if filer.db, err = leveldb.OpenFile(dir, nil); err != nil { |
|||
return |
|||
} |
|||
return |
|||
} |
|||
|
|||
func (filer *EmbeddedStore) CreateFile(filePath string, fid string) (err error) { |
|||
return nil |
|||
} |
|||
|
|||
func (filer *EmbeddedStore) Mkdir(filePath string) (err error) { |
|||
return nil |
|||
} |
@ -0,0 +1,94 @@ |
|||
package filer2 |
|||
|
|||
import ( |
|||
"fmt" |
|||
|
|||
"github.com/chrislusf/seaweedfs/weed/filer2/embedded" |
|||
"github.com/karlseguin/ccache" |
|||
"strings" |
|||
) |
|||
|
|||
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)), |
|||
} |
|||
} |
|||
|
|||
func NewEmbeddedFiler(master string, dir string) (*Filer, error) { |
|||
_, err := embedded.NewEmbeddedStore(dir) |
|||
if err != nil { |
|||
return nil, fmt.Errorf("failed to create embedded filer store: %v", err) |
|||
} |
|||
return &Filer{ |
|||
master: master, |
|||
// store: store,
|
|||
}, nil |
|||
} |
|||
|
|||
func (f *Filer) CreateEntry(entry Entry) (error) { |
|||
/* |
|||
1. recursively ensure parent directory is created. |
|||
2. get current parent directory, add link, |
|||
3. add the file entry |
|||
*/ |
|||
|
|||
f.recursivelyEnsureDirectory(entry.Dir, func(parent, name string) error { |
|||
return nil |
|||
}) |
|||
|
|||
return f.store.CreateEntry(entry) |
|||
} |
|||
|
|||
func (f *Filer) AppendFileChunk(p FullPath, c FileChunk) (err error) { |
|||
return f.store.AppendFileChunk(p, c) |
|||
} |
|||
|
|||
func (f *Filer) FindEntry(p FullPath) (found bool, fileEntry Entry, err error) { |
|||
return f.store.FindEntry(p) |
|||
} |
|||
|
|||
func (f *Filer) DeleteEntry(p FullPath) (fileEntry Entry, err error) { |
|||
return f.store.DeleteEntry(p) |
|||
} |
|||
|
|||
func (f *Filer) ListDirectoryEntries(p FullPath) ([]Entry, error) { |
|||
return f.store.ListDirectoryEntries(p) |
|||
} |
|||
|
|||
func (f *Filer) UpdateEntry(entry Entry) (error) { |
|||
return f.store.UpdateEntry(entry) |
|||
} |
|||
|
|||
func (f *Filer) recursivelyEnsureDirectory(fullPath string, fn func(parent, name string) error) (error) { |
|||
if strings.HasSuffix(fullPath, "/") { |
|||
fullPath = fullPath[0:len(fullPath)-1] |
|||
} |
|||
nextPathEnd := strings.LastIndex(fullPath, "/") |
|||
if nextPathEnd < 0 { |
|||
return nil |
|||
} |
|||
|
|||
dirName := fullPath[nextPathEnd+1:] |
|||
parentDirPath := fullPath[0:nextPathEnd] |
|||
|
|||
if parentDirPath == "" { |
|||
parentDirPath = "/" |
|||
} |
|||
|
|||
if err := fn(parentDirPath, dirName); err != nil { |
|||
return err |
|||
} |
|||
|
|||
return f.recursivelyEnsureDirectory(parentDirPath, fn) |
|||
} |
|||
|
|||
func (f *Filer) cacheGetDirectory(dirpath string) (error) { |
|||
return nil |
|||
} |
@ -0,0 +1,62 @@ |
|||
package filer2 |
|||
|
|||
import ( |
|||
"errors" |
|||
"os" |
|||
"time" |
|||
) |
|||
|
|||
type FileId string //file id in SeaweedFS
|
|||
type FullPath struct { |
|||
Dir string //full path of the parent dir
|
|||
Name string //file name without path
|
|||
} |
|||
|
|||
type Attr struct { |
|||
Mtime time.Time // time of last modification
|
|||
Crtime time.Time // time of creation (OS X only)
|
|||
Mode os.FileMode // file mode
|
|||
Uid uint32 // owner uid
|
|||
Gid uint32 // group gid
|
|||
IsDirectory bool |
|||
Size uint64 // total size in bytes
|
|||
Nlink uint32 // number of links (usually 1)
|
|||
} |
|||
|
|||
type Entry struct { |
|||
Dir string `json:"dir,omitempty"` //full path of the parent dir
|
|||
Name string `json:"name,omitempty"` //file name without path
|
|||
|
|||
Attr |
|||
|
|||
// the following is for files
|
|||
Chunks []FileChunk `json:"chunks,omitempty"` |
|||
} |
|||
|
|||
type FileChunk struct { |
|||
Fid FileId `json:"fid,omitempty"` |
|||
Offset int64 `json:"offset,omitempty"` |
|||
Size uint64 `json:"size,omitempty"` // size in bytes
|
|||
} |
|||
|
|||
type AbstractFiler interface { |
|||
CreateEntry(Entry) (error) |
|||
AppendFileChunk(FullPath, FileChunk) (err error) |
|||
FindEntry(FullPath) (found bool, fileEntry Entry, err error) |
|||
DeleteEntry(FullPath) (fileEntry Entry, err error) |
|||
|
|||
ListDirectoryEntries(dirPath FullPath) ([]Entry, error) |
|||
UpdateEntry(Entry) (error) |
|||
} |
|||
|
|||
var ErrNotFound = errors.New("filer: no entry is found in filer store") |
|||
|
|||
type FilerStore interface { |
|||
CreateEntry(Entry) (error) |
|||
AppendFileChunk(FullPath, FileChunk) (err error) |
|||
FindEntry(FullPath) (found bool, fileEntry Entry, err error) |
|||
DeleteEntry(FullPath) (fileEntry Entry, err error) |
|||
|
|||
ListDirectoryEntries(dirPath FullPath) ([]Entry, error) |
|||
UpdateEntry(Entry) (error) |
|||
} |
@ -0,0 +1,32 @@ |
|||
package filer2 |
|||
|
|||
import ( |
|||
"testing" |
|||
"path/filepath" |
|||
"fmt" |
|||
) |
|||
|
|||
func TestRecursion(t *testing.T) { |
|||
filer := NewFiler("") |
|||
|
|||
fullPath := "/home/chris/some/file/abc.jpg" |
|||
expected := []string{ |
|||
"/home/chris/some", "file", |
|||
"/home/chris", "some", |
|||
"/home", "chris", |
|||
"/", "home", |
|||
} |
|||
|
|||
dir, _ := filepath.Split(fullPath) |
|||
|
|||
i := 0 |
|||
|
|||
filer.recursivelyEnsureDirectory(dir, func(parent, name string) error { |
|||
if parent != expected[i] || name != expected[i+1] { |
|||
t.Errorf("recursive directory is wrong! parent=%s dirName=%s", parent, name) |
|||
} |
|||
fmt.Printf("processing folder %s \t parent=%s\n", name, parent) |
|||
i += 2 |
|||
return nil |
|||
}) |
|||
} |
Write
Preview
Loading…
Cancel
Save
Reference in new issue