diff options
Diffstat (limited to 'weed/filesys/wfs.go')
| -rw-r--r-- | weed/filesys/wfs.go | 66 |
1 files changed, 66 insertions, 0 deletions
diff --git a/weed/filesys/wfs.go b/weed/filesys/wfs.go index ac7333695..127252058 100644 --- a/weed/filesys/wfs.go +++ b/weed/filesys/wfs.go @@ -6,6 +6,9 @@ import ( "github.com/chrislusf/seaweedfs/weed/pb/filer_pb" "github.com/karlseguin/ccache" "google.golang.org/grpc" + "sync" + "bazil.org/fuse" + "github.com/chrislusf/seaweedfs/weed/glog" ) type WFS struct { @@ -14,6 +17,11 @@ type WFS struct { collection string replication string chunkSizeLimit int64 + + // contains all open handles + handles []*FileHandle + pathToHandleIndex map[string]int + pathToHandleLock sync.Mutex } func NewSeaweedFileSystem(filerGrpcAddress string, collection string, replication string, chunkSizeLimitMB int) *WFS { @@ -23,6 +31,7 @@ func NewSeaweedFileSystem(filerGrpcAddress string, collection string, replicatio collection: collection, replication: replication, chunkSizeLimit: int64(chunkSizeLimitMB) * 1024 * 1024, + pathToHandleIndex: make(map[string]int), } } @@ -42,3 +51,60 @@ func (wfs *WFS) withFilerClient(fn func(filer_pb.SeaweedFilerClient) error) erro return fn(client) } + +func (wfs *WFS) AcquireHandle(file *File, uid, gid uint32) (handle *FileHandle) { + wfs.pathToHandleLock.Lock() + defer wfs.pathToHandleLock.Unlock() + + fullpath := file.fullpath() + + index, found := wfs.pathToHandleIndex[fullpath] + if found && wfs.handles[index] != nil { + glog.V(4).Infoln(fullpath, "found handle id", index) + return wfs.handles[index] + } + + // create a new handler + handle = &FileHandle{ + f: file, + dirtyPages: newDirtyPages(file), + Uid: uid, + Gid: gid, + } + + if found && wfs.handles[index] != nil { + glog.V(4).Infoln(fullpath, "reuse previous handle id", index) + wfs.handles[index] = handle + handle.handle = uint64(index) + return + } + + for i, h := range wfs.handles { + if h == nil { + wfs.handles[i] = handle + handle.handle = uint64(i) + wfs.pathToHandleIndex[fullpath] = i + glog.V(4).Infoln(fullpath, "reuse handle id", handle.handle) + return + } + } + + wfs.handles = append(wfs.handles, handle) + handle.handle = uint64(len(wfs.handles) - 1) + println(fullpath, "new handle id", handle.handle) + wfs.pathToHandleIndex[fullpath] = int(handle.handle) + + return +} + +func (wfs *WFS) ReleaseHandle(handleId fuse.HandleID) { + wfs.pathToHandleLock.Lock() + defer wfs.pathToHandleLock.Unlock() + + glog.V(4).Infoln("releasing handle id", handleId, "current handles lengh", len(wfs.handles)) + if int(handleId) < len(wfs.handles) { + wfs.handles[int(handleId)] = nil + } + + return +} |
