Ekaterina Lebedeva
fc6abe30b8
Change-Id: I9d8ccde3c71fca716856c7bfc53da20ee0542f20 Signed-off-by: Ekaterina Lebedeva <ekaterina.lebedeva@yadro.com>
69 lines
1.2 KiB
Go
69 lines
1.2 KiB
Go
package fstree
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/internal/assert"
|
|
)
|
|
|
|
// FileCounter used to count files in FSTree. The implementation must be thread-safe.
|
|
type FileCounter interface {
|
|
Set(count, size uint64)
|
|
Inc(size uint64)
|
|
Dec(size uint64)
|
|
}
|
|
|
|
type noopCounter struct{}
|
|
|
|
func (c *noopCounter) Set(uint64, uint64) {}
|
|
func (c *noopCounter) Inc(uint64) {}
|
|
func (c *noopCounter) Dec(uint64) {}
|
|
|
|
func counterEnabled(c FileCounter) bool {
|
|
_, noop := c.(*noopCounter)
|
|
return !noop
|
|
}
|
|
|
|
type SimpleCounter struct {
|
|
mtx sync.RWMutex
|
|
count uint64
|
|
size uint64
|
|
}
|
|
|
|
func NewSimpleCounter() *SimpleCounter {
|
|
return &SimpleCounter{}
|
|
}
|
|
|
|
func (c *SimpleCounter) Set(count, size uint64) {
|
|
c.mtx.Lock()
|
|
defer c.mtx.Unlock()
|
|
|
|
c.count = count
|
|
c.size = size
|
|
}
|
|
|
|
func (c *SimpleCounter) Inc(size uint64) {
|
|
c.mtx.Lock()
|
|
defer c.mtx.Unlock()
|
|
|
|
c.count++
|
|
c.size += size
|
|
}
|
|
|
|
func (c *SimpleCounter) Dec(size uint64) {
|
|
c.mtx.Lock()
|
|
defer c.mtx.Unlock()
|
|
|
|
assert.True(c.count > 0, "fstree.SimpleCounter: invalid count")
|
|
c.count--
|
|
|
|
assert.True(c.size >= size, "fstree.SimpleCounter: invalid size")
|
|
c.size -= size
|
|
}
|
|
|
|
func (c *SimpleCounter) CountSize() (uint64, uint64) {
|
|
c.mtx.RLock()
|
|
defer c.mtx.RUnlock()
|
|
|
|
return c.count, c.size
|
|
}
|