forked from mirror/afero
Unexport the RWMutex in MemMapFs
This commit is contained in:
parent
7e65370ebb
commit
2101bdf57c
58
memmap.go
58
memmap.go
|
@ -27,7 +27,7 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
type MemMapFs struct {
|
type MemMapFs struct {
|
||||||
sync.RWMutex
|
mu sync.RWMutex
|
||||||
data map[string]File
|
data map[string]File
|
||||||
init sync.Once
|
init sync.Once
|
||||||
}
|
}
|
||||||
|
@ -48,11 +48,11 @@ func (MemMapFs) Name() string { return "MemMapFS" }
|
||||||
|
|
||||||
func (m *MemMapFs) Create(name string) (File, error) {
|
func (m *MemMapFs) Create(name string) (File, error) {
|
||||||
name = normalizePath(name)
|
name = normalizePath(name)
|
||||||
m.Lock()
|
m.mu.Lock()
|
||||||
file := mem.CreateFile(name)
|
file := mem.CreateFile(name)
|
||||||
m.getData()[name] = file
|
m.getData()[name] = file
|
||||||
m.registerWithParent(file)
|
m.registerWithParent(file)
|
||||||
m.Unlock()
|
m.mu.Unlock()
|
||||||
return file, nil
|
return file, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -135,9 +135,9 @@ func (m *MemMapFs) lockfreeMkdir(name string, perm os.FileMode) error {
|
||||||
func (m *MemMapFs) Mkdir(name string, perm os.FileMode) error {
|
func (m *MemMapFs) Mkdir(name string, perm os.FileMode) error {
|
||||||
name = normalizePath(name)
|
name = normalizePath(name)
|
||||||
|
|
||||||
m.RLock()
|
m.mu.RLock()
|
||||||
x, ok := m.getData()[name]
|
x, ok := m.getData()[name]
|
||||||
m.RUnlock()
|
m.mu.RUnlock()
|
||||||
if ok {
|
if ok {
|
||||||
// Only return ErrFileExists if it's a file, not a directory.
|
// Only return ErrFileExists if it's a file, not a directory.
|
||||||
i, err := x.Stat()
|
i, err := x.Stat()
|
||||||
|
@ -148,11 +148,11 @@ func (m *MemMapFs) Mkdir(name string, perm os.FileMode) error {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
m.Lock()
|
m.mu.Lock()
|
||||||
item := mem.CreateDir(name)
|
item := mem.CreateDir(name)
|
||||||
m.getData()[name] = item
|
m.getData()[name] = item
|
||||||
m.registerWithParent(item)
|
m.registerWithParent(item)
|
||||||
m.Unlock()
|
m.mu.Unlock()
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
@ -178,14 +178,14 @@ func normalizePath(path string) string {
|
||||||
func (m *MemMapFs) Open(name string) (File, error) {
|
func (m *MemMapFs) Open(name string) (File, error) {
|
||||||
name = normalizePath(name)
|
name = normalizePath(name)
|
||||||
|
|
||||||
m.RLock()
|
m.mu.RLock()
|
||||||
f, ok := m.getData()[name]
|
f, ok := m.getData()[name]
|
||||||
ff, ok := f.(*mem.File)
|
ff, ok := f.(*mem.File)
|
||||||
|
|
||||||
if ok {
|
if ok {
|
||||||
ff.Open()
|
ff.Open()
|
||||||
}
|
}
|
||||||
m.RUnlock()
|
m.mu.RUnlock()
|
||||||
|
|
||||||
if ok {
|
if ok {
|
||||||
return f, nil
|
return f, nil
|
||||||
|
@ -236,8 +236,8 @@ func (m *MemMapFs) OpenFile(name string, flag int, perm os.FileMode) (File, erro
|
||||||
func (m *MemMapFs) Remove(name string) error {
|
func (m *MemMapFs) Remove(name string) error {
|
||||||
name = normalizePath(name)
|
name = normalizePath(name)
|
||||||
|
|
||||||
m.Lock()
|
m.mu.Lock()
|
||||||
defer m.Unlock()
|
defer m.mu.Unlock()
|
||||||
|
|
||||||
if _, ok := m.getData()[name]; ok {
|
if _, ok := m.getData()[name]; ok {
|
||||||
m.unRegisterWithParent(name)
|
m.unRegisterWithParent(name)
|
||||||
|
@ -250,20 +250,20 @@ func (m *MemMapFs) Remove(name string) error {
|
||||||
|
|
||||||
func (m *MemMapFs) RemoveAll(path string) error {
|
func (m *MemMapFs) RemoveAll(path string) error {
|
||||||
path = normalizePath(path)
|
path = normalizePath(path)
|
||||||
m.Lock()
|
m.mu.Lock()
|
||||||
m.unRegisterWithParent(path)
|
m.unRegisterWithParent(path)
|
||||||
m.Unlock()
|
m.mu.Unlock()
|
||||||
|
|
||||||
m.RLock()
|
m.mu.RLock()
|
||||||
defer m.RUnlock()
|
defer m.mu.RUnlock()
|
||||||
|
|
||||||
for p, _ := range m.getData() {
|
for p, _ := range m.getData() {
|
||||||
if strings.HasPrefix(p, path) {
|
if strings.HasPrefix(p, path) {
|
||||||
m.RUnlock()
|
m.mu.RUnlock()
|
||||||
m.Lock()
|
m.mu.Lock()
|
||||||
delete(m.getData(), p)
|
delete(m.getData(), p)
|
||||||
m.Unlock()
|
m.mu.Unlock()
|
||||||
m.RLock()
|
m.mu.RLock()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
|
@ -277,20 +277,20 @@ func (m *MemMapFs) Rename(oldname, newname string) error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
m.RLock()
|
m.mu.RLock()
|
||||||
defer m.RUnlock()
|
defer m.mu.RUnlock()
|
||||||
if _, ok := m.getData()[oldname]; ok {
|
if _, ok := m.getData()[oldname]; ok {
|
||||||
if _, ok := m.getData()[newname]; !ok {
|
if _, ok := m.getData()[newname]; !ok {
|
||||||
m.RUnlock()
|
m.mu.RUnlock()
|
||||||
m.Lock()
|
m.mu.Lock()
|
||||||
m.unRegisterWithParent(oldname)
|
m.unRegisterWithParent(oldname)
|
||||||
file := m.getData()[oldname].(*mem.File)
|
file := m.getData()[oldname].(*mem.File)
|
||||||
delete(m.getData(), oldname)
|
delete(m.getData(), oldname)
|
||||||
mem.ChangeFileName(file, newname)
|
mem.ChangeFileName(file, newname)
|
||||||
m.getData()[newname] = file
|
m.getData()[newname] = file
|
||||||
m.registerWithParent(file)
|
m.registerWithParent(file)
|
||||||
m.Unlock()
|
m.mu.Unlock()
|
||||||
m.RLock()
|
m.mu.RLock()
|
||||||
} else {
|
} else {
|
||||||
return ErrDestinationExists
|
return ErrDestinationExists
|
||||||
}
|
}
|
||||||
|
@ -318,9 +318,9 @@ func (m *MemMapFs) Chmod(name string, mode os.FileMode) error {
|
||||||
|
|
||||||
ff, ok := f.(*mem.File)
|
ff, ok := f.(*mem.File)
|
||||||
if ok {
|
if ok {
|
||||||
m.Lock()
|
m.mu.Lock()
|
||||||
mem.SetMode(ff, mode)
|
mem.SetMode(ff, mode)
|
||||||
m.Unlock()
|
m.mu.Unlock()
|
||||||
} else {
|
} else {
|
||||||
return errors.New("Unable to Chmod Memory File")
|
return errors.New("Unable to Chmod Memory File")
|
||||||
}
|
}
|
||||||
|
@ -336,9 +336,9 @@ func (m *MemMapFs) Chtimes(name string, atime time.Time, mtime time.Time) error
|
||||||
|
|
||||||
ff, ok := f.(*mem.File)
|
ff, ok := f.(*mem.File)
|
||||||
if ok {
|
if ok {
|
||||||
m.Lock()
|
m.mu.Lock()
|
||||||
mem.SetModTime(ff, mtime)
|
mem.SetModTime(ff, mtime)
|
||||||
m.Unlock()
|
m.mu.Unlock()
|
||||||
} else {
|
} else {
|
||||||
return errors.New("Unable to Chtime Memory File")
|
return errors.New("Unable to Chtime Memory File")
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue