vfs: use atomic types

This commit is contained in:
Roberto Ricci 2023-08-18 23:18:56 +02:00 committed by Nick Craig-Wood
parent 552b6c47ff
commit 91b8152321
4 changed files with 22 additions and 22 deletions

View File

@ -36,8 +36,8 @@ import (
// File represents a file
type File struct {
inode uint64 // inode number - read only
size int64 // size of file - read and written with atomic int64 - must be 64 bit aligned
inode uint64 // inode number - read only
size atomic.Int64 // size of file
muRW sync.Mutex // synchronize RWFileHandle.openPending(), RWFileHandle.close() and File.Remove
@ -51,7 +51,7 @@ type File struct {
pendingModTime time.Time // will be applied once o becomes available, i.e. after file was written
pendingRenameFun func(ctx context.Context) error // will be run/renamed after all writers close
sys atomic.Value // user defined info to be attached here
nwriters int32 // len(writers) which is read/updated with atomic
nwriters atomic.Int32 // len(writers)
appendMode bool // file was opened with O_APPEND
}
@ -67,7 +67,7 @@ func newFile(d *Dir, dPath string, o fs.Object, leaf string) *File {
inode: newInode(),
}
if o != nil {
f.size = o.Size()
f.size.Store(o.Size())
}
return f
}
@ -266,7 +266,7 @@ func (f *File) rename(ctx context.Context, destDir *Dir, newName string) error {
func (f *File) addWriter(h Handle) {
f.mu.Lock()
f.writers = append(f.writers, h)
atomic.AddInt32(&f.nwriters, 1)
f.nwriters.Add(1)
f.mu.Unlock()
}
@ -284,7 +284,7 @@ func (f *File) delWriter(h Handle) {
}
if found >= 0 {
f.writers = append(f.writers[:found], f.writers[found+1:]...)
atomic.AddInt32(&f.nwriters, -1)
f.nwriters.Add(-1)
} else {
fs.Debugf(f._path(), "File.delWriter couldn't find handle")
}
@ -295,7 +295,7 @@ func (f *File) delWriter(h Handle) {
// Note that we don't take the mutex here. If we do then we can get a
// deadlock.
func (f *File) activeWriters() int {
return int(atomic.LoadInt32(&f.nwriters))
return int(f.nwriters.Load())
}
// _roundModTime rounds the time passed in to the Precision of the
@ -383,7 +383,7 @@ func (f *File) Size() int64 {
// if o is nil it isn't valid yet or there are writers, so return the size so far
if f._writingInProgress() {
return atomic.LoadInt64(&f.size)
return f.size.Load()
}
return nonNegative(f.o.Size())
}
@ -473,7 +473,7 @@ func (f *File) writingInProgress() bool {
// Update the size while writing
func (f *File) setSize(n int64) {
atomic.StoreInt64(&f.size, n)
f.size.Store(n)
}
// Update the object when written and add it to the directory

View File

@ -223,7 +223,7 @@ func waitSequential(what string, remote string, cond *sync.Cond, maxWait time.Du
var (
timeout = time.NewTimer(maxWait)
done = make(chan struct{})
abort = int32(0)
abort atomic.Int32
)
go func() {
select {
@ -232,14 +232,14 @@ func waitSequential(what string, remote string, cond *sync.Cond, maxWait time.Du
// cond.Broadcast. NB cond.L == mu
cond.L.Lock()
// set abort flag and give all the waiting goroutines a kick on timeout
atomic.StoreInt32(&abort, 1)
abort.Store(1)
fs.Debugf(remote, "aborting in-sequence %s wait, off=%d", what, off)
cond.Broadcast()
cond.L.Unlock()
case <-done:
}
}()
for *poff != off && atomic.LoadInt32(&abort) == 0 {
for *poff != off && abort.Load() == 0 {
fs.Debugf(remote, "waiting for in-sequence %s to %d for %v", what, off, maxWait)
cond.Wait()
}

View File

@ -26,7 +26,7 @@ var (
number = flag.Int("n", 4, "Number of tests to run simultaneously")
iterations = flag.Int("i", 100, "Iterations of the test")
timeout = flag.Duration("timeout", 10*time.Second, "Inactivity time to detect a deadlock")
testNumber int32
testNumber atomic.Int32
)
// Seed the random number generator
@ -55,7 +55,7 @@ func NewTest(Dir string) *Test {
dir: Dir,
name: random.String(*nameLength),
isDir: rand.Intn(2) == 0,
number: atomic.AddInt32(&testNumber, 1),
number: testNumber.Add(1),
timer: time.NewTimer(*timeout),
}
width := int(math.Floor(math.Log10(float64(*number)))) + 1

View File

@ -167,7 +167,7 @@ type VFS struct {
usageTime time.Time
usage *fs.Usage
pollChan chan time.Duration
inUse int32 // count of number of opens accessed with atomic
inUse atomic.Int32 // count of number of opens
}
// Keep track of active VFS keyed on fs.ConfigString(f)
@ -181,9 +181,9 @@ var (
func New(f fs.Fs, opt *vfscommon.Options) *VFS {
fsDir := fs.NewDir("", time.Now())
vfs := &VFS{
f: f,
inUse: int32(1),
f: f,
}
vfs.inUse.Store(1)
// Make a copy of the options
if opt != nil {
@ -202,7 +202,7 @@ func New(f fs.Fs, opt *vfscommon.Options) *VFS {
for _, activeVFS := range active[configName] {
if vfs.Opt == activeVFS.Opt {
fs.Debugf(f, "Re-using VFS from active cache")
atomic.AddInt32(&activeVFS.inUse, 1)
activeVFS.inUse.Add(1)
return activeVFS
}
}
@ -243,7 +243,7 @@ func (vfs *VFS) Stats() (out rc.Params) {
out = make(rc.Params)
out["fs"] = fs.ConfigString(vfs.f)
out["opt"] = vfs.Opt
out["inUse"] = atomic.LoadInt32(&vfs.inUse)
out["inUse"] = vfs.inUse.Load()
var (
dirs int
@ -313,7 +313,7 @@ func (vfs *VFS) shutdownCache() {
// Shutdown stops any background go-routines and removes the VFS from
// the active ache.
func (vfs *VFS) Shutdown() {
if atomic.AddInt32(&vfs.inUse, -1) > 0 {
if vfs.inUse.Add(-1) > 0 {
return
}
@ -386,11 +386,11 @@ func (vfs *VFS) Root() (*Dir, error) {
return vfs.root, nil
}
var inodeCount uint64
var inodeCount atomic.Uint64
// newInode creates a new unique inode number
func newInode() (inode uint64) {
return atomic.AddUint64(&inodeCount, 1)
return inodeCount.Add(1)
}
// Stat finds the Node by path starting from the root