realloc/frostfs-node
1
0
Fork 0
forked from TrueCloudLab/frostfs-node
Code Releases Packages Activity

[#568] shard/writecache: Maintain memory size and db size

Browse source 
Signed-off-by: ZhangTao1596 <zhangtao@ngd.neo.org>
This commit is contained in:
ZhangTao1596 2021-06-21 17:42:09 +08:00 • committed by Alex Vanin
parent adbbad0beb
commit 9b87e6267d
4 changed files with 32 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

15
pkg/local_object_storage/writecache/storage.go
View file

@ -8,6 +8,7 @@ import (
lru "github.com/hashicorp/golang-lru"
"github.com/hashicorp/golang-lru/simplelru"
objectSDK "github.com/nspcc-dev/neofs-api-go/pkg/object"
"github.com/nspcc-dev/neofs-node/pkg/core/object"
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/blobstor/fstree"
"go.etcd.io/bbolt"
"go.uber.org/zap"
@ -95,16 +96,26 @@ func (c *cache) deleteFromDB(keys [][]byte) error {
if len(keys) == 0 {
return nil
}
return c.db.Update(func(tx *bbolt.Tx) error {
var sz uint64
err := c.db.Update(func(tx *bbolt.Tx) error {
b := tx.Bucket(defaultBucket)
for i := range keys {
has := b.Get(keys[i])
if has == nil {
return object.ErrNotFound
}
if err := b.Delete(keys[i]); err != nil {
return err
}
sz += uint64(len(has))
}
return nil
})
if err != nil {
return err
}
c.dbSize.Sub(sz)
return nil
}
func (c *cache) deleteFromDisk(keys [][]byte) error {