From eec97d177e27c02bfe54e28b822bebbadb018e5c Mon Sep 17 00:00:00 2001 From: Dmitrii Stepanov Date: Wed, 9 Aug 2023 17:14:41 +0300 Subject: [PATCH] [#585] writecache: Count items periodically Signed-off-by: Dmitrii Stepanov --- internal/logs/logs.go | 1 + .../writecache/writecachebbolt/cachebbolt.go | 3 +- .../writecache/writecachebbolt/delete.go | 2 - .../writecache/writecachebbolt/put.go | 2 - .../writecache/writecachebbolt/state.go | 87 +++++++++++-------- .../writecache/writecachebbolt/storage.go | 2 - 6 files changed, 55 insertions(+), 42 deletions(-) diff --git a/internal/logs/logs.go b/internal/logs/logs.go index 68914d6a..8d371259 100644 --- a/internal/logs/logs.go +++ b/internal/logs/logs.go @@ -513,4 +513,5 @@ const ( FrostFSNodeCantUnmarshalObjectFromDB = "can't unmarshal an object from the DB" // Error in ../node/cmd/frostfs-node/morph.go RuntimeSoftMemoryLimitUpdated = "soft runtime memory limit value updated" RuntimeSoftMemoryDefinedWithGOMEMLIMIT = "soft runtime memory defined with GOMEMLIMIT environment variable, config value skipped" + FailedToCountWritecacheItems = "failed to count writecache items" ) diff --git a/pkg/local_object_storage/writecache/writecachebbolt/cachebbolt.go b/pkg/local_object_storage/writecache/writecachebbolt/cachebbolt.go index 407d1a9c..69e418bb 100644 --- a/pkg/local_object_storage/writecache/writecachebbolt/cachebbolt.go +++ b/pkg/local_object_storage/writecache/writecachebbolt/cachebbolt.go @@ -107,13 +107,14 @@ func (c *cache) Open(readOnly bool) error { // thus we need to create a channel here. c.closeCh = make(chan struct{}) - return metaerr.Wrap(c.initCounters()) + return metaerr.Wrap(c.setCounters()) } // Init runs necessary services. func (c *cache) Init() error { c.metrics.SetMode(c.mode) c.runFlushLoop() + c.runDBCounterLoop() return nil } diff --git a/pkg/local_object_storage/writecache/writecachebbolt/delete.go b/pkg/local_object_storage/writecache/writecachebbolt/delete.go index b0cc091a..105526c2 100644 --- a/pkg/local_object_storage/writecache/writecachebbolt/delete.go +++ b/pkg/local_object_storage/writecache/writecachebbolt/delete.go @@ -63,7 +63,6 @@ func (c *cache) Delete(ctx context.Context, addr oid.Address) error { storagelog.OpField("db DELETE"), ) deleted = true - c.objCounters.DecDB() return nil } @@ -75,7 +74,6 @@ func (c *cache) Delete(ctx context.Context, addr oid.Address) error { storagelog.StorageTypeField(wcStorageType), storagelog.OpField("fstree DELETE"), ) - c.objCounters.DecFS() deleted = true } diff --git a/pkg/local_object_storage/writecache/writecachebbolt/put.go b/pkg/local_object_storage/writecache/writecachebbolt/put.go index 48b130d2..645e0bf8 100644 --- a/pkg/local_object_storage/writecache/writecachebbolt/put.go +++ b/pkg/local_object_storage/writecache/writecachebbolt/put.go @@ -95,7 +95,6 @@ func (c *cache) putSmall(obj objectInfo) error { storagelog.StorageTypeField(wcStorageType), storagelog.OpField("db PUT"), ) - c.objCounters.IncDB() } return err } @@ -117,7 +116,6 @@ func (c *cache) putBig(ctx context.Context, addr string, prm common.PutPrm) erro c.compressFlags[addr] = struct{}{} c.mtx.Unlock() } - c.objCounters.IncFS() storagelog.Write(c.log, storagelog.AddressField(addr), storagelog.StorageTypeField(wcStorageType), diff --git a/pkg/local_object_storage/writecache/writecachebbolt/state.go b/pkg/local_object_storage/writecache/writecachebbolt/state.go index 03283dfe..49b802ae 100644 --- a/pkg/local_object_storage/writecache/writecachebbolt/state.go +++ b/pkg/local_object_storage/writecache/writecachebbolt/state.go @@ -2,10 +2,13 @@ package writecachebbolt import ( "fmt" - "math" "sync/atomic" + "time" + "git.frostfs.info/TrueCloudLab/frostfs-node/internal/logs" "go.etcd.io/bbolt" + "go.uber.org/zap" + "golang.org/x/sync/errgroup" ) func (c *cache) estimateCacheSize() uint64 { @@ -27,54 +30,68 @@ type counters struct { cDB, cFS atomic.Uint64 } -func (x *counters) IncDB() { - x.cDB.Add(1) -} - -func (x *counters) DecDB() { - x.cDB.Add(math.MaxUint64) -} - func (x *counters) DB() uint64 { return x.cDB.Load() } -func (x *counters) IncFS() { - x.cFS.Add(1) -} - -func (x *counters) DecFS() { - x.cFS.Add(math.MaxUint64) -} - func (x *counters) FS() uint64 { return x.cFS.Load() } -func (c *cache) initCounters() error { +func (c *cache) setCounters() error { var inDB uint64 - err := c.db.View(func(tx *bbolt.Tx) error { - b := tx.Bucket(defaultBucket) - if b != nil { - inDB = uint64(b.Stats().KeyN) + var inFS uint64 + + var eg errgroup.Group + + eg.Go(func() error { + err := c.db.View(func(tx *bbolt.Tx) error { + b := tx.Bucket(defaultBucket) + if b != nil { + inDB = uint64(b.Stats().KeyN) + } + return nil + }) + if err != nil { + return fmt.Errorf("could not read write-cache DB counter: %w", err) } + c.objCounters.cDB.Store(inDB) return nil }) - if err != nil { - return fmt.Errorf("could not read write-cache DB counter: %w", err) - } - inFS, err := c.fsTree.NumberOfObjects() - if err != nil { - return fmt.Errorf("could not read write-cache FS counter: %w", err) + eg.Go(func() error { + var err error + inFS, err = c.fsTree.NumberOfObjects() + if err != nil { + return fmt.Errorf("could not read write-cache FS counter: %w", err) + } + if inFS > 0 { + inFS-- //small.bolt DB file + } + c.objCounters.cFS.Store(inFS) + return nil + }) + if err := eg.Wait(); err != nil { + return err } - if inFS > 0 { - inFS-- //small.bolt DB file - } - - c.objCounters.cDB.Store(inDB) - c.objCounters.cFS.Store(inFS) c.metrics.SetActualCounters(inDB, inFS) - return nil } + +func (c *cache) runDBCounterLoop() { + go func() { + t := time.NewTicker(time.Second * 30) + defer t.Stop() + for { + select { + case <-t.C: + err := c.setCounters() + if err != nil { + c.log.Warn(logs.FailedToCountWritecacheItems, zap.Error(err)) + } + case <-c.closeCh: + return + } + } + }() +} diff --git a/pkg/local_object_storage/writecache/writecachebbolt/storage.go b/pkg/local_object_storage/writecache/writecachebbolt/storage.go index 67397f98..c651b744 100644 --- a/pkg/local_object_storage/writecache/writecachebbolt/storage.go +++ b/pkg/local_object_storage/writecache/writecachebbolt/storage.go @@ -69,7 +69,6 @@ func (c *cache) deleteFromDB(key string) { }) if err == nil { - c.objCounters.DecDB() c.metrics.Evict(writecache.StorageTypeDB) storagelog.Write(c.log, storagelog.AddressField(key), @@ -111,7 +110,6 @@ func (c *cache) deleteFromDisk(ctx context.Context, keys []string) []string { storagelog.OpField("fstree DELETE"), ) c.metrics.Evict(writecache.StorageTypeFSTree) - c.objCounters.DecFS() c.estimateCacheSize() } }