From 8434f3dbfc850839c759430cea9640f3c87e5f95 Mon Sep 17 00:00:00 2001
From: Dmitrii Stepanov <d.stepanov@yadro.com>
Date: Thu, 19 Sep 2024 17:00:58 +0300
Subject: [PATCH] [#1385] metabase: Use `Batch` for delete-related operations

Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
---
 pkg/local_object_storage/metabase/counter.go   | 4 ++--
 pkg/local_object_storage/metabase/delete.go    | 2 +-
 pkg/local_object_storage/metabase/graveyard.go | 2 +-
 pkg/local_object_storage/metabase/inhume.go    | 2 +-
 pkg/local_object_storage/metabase/lock.go      | 4 ++--
 5 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/pkg/local_object_storage/metabase/counter.go b/pkg/local_object_storage/metabase/counter.go
index 275099ff..3ead0d9a 100644
--- a/pkg/local_object_storage/metabase/counter.go
+++ b/pkg/local_object_storage/metabase/counter.go
@@ -654,7 +654,7 @@ func (db *DB) DeleteContainerSize(ctx context.Context, id cid.ID) error {
 		return ErrReadOnlyMode
 	}
 
-	err := db.boltDB.Update(func(tx *bbolt.Tx) error {
+	err := db.boltDB.Batch(func(tx *bbolt.Tx) error {
 		b := tx.Bucket(containerVolumeBucketName)
 
 		key := make([]byte, cidSize)
@@ -737,7 +737,7 @@ func (db *DB) DeleteContainerCount(ctx context.Context, id cid.ID) error {
 		return ErrReadOnlyMode
 	}
 
-	err := db.boltDB.Update(func(tx *bbolt.Tx) error {
+	err := db.boltDB.Batch(func(tx *bbolt.Tx) error {
 		b := tx.Bucket(containerCounterBucketName)
 
 		key := make([]byte, cidSize)
diff --git a/pkg/local_object_storage/metabase/delete.go b/pkg/local_object_storage/metabase/delete.go
index 00c8d06e..e5e9840a 100644
--- a/pkg/local_object_storage/metabase/delete.go
+++ b/pkg/local_object_storage/metabase/delete.go
@@ -112,7 +112,7 @@ func (db *DB) Delete(ctx context.Context, prm DeletePrm) (DeleteRes, error) {
 	var err error
 	var res DeleteRes
 
-	err = db.boltDB.Update(func(tx *bbolt.Tx) error {
+	err = db.boltDB.Batch(func(tx *bbolt.Tx) error {
 		res, err = db.deleteGroup(tx, prm.addrs)
 		return err
 	})
diff --git a/pkg/local_object_storage/metabase/graveyard.go b/pkg/local_object_storage/metabase/graveyard.go
index 80d40fb7..31f95d6e 100644
--- a/pkg/local_object_storage/metabase/graveyard.go
+++ b/pkg/local_object_storage/metabase/graveyard.go
@@ -282,7 +282,7 @@ func (db *DB) DropGraves(ctx context.Context, tss []TombstonedObject) error {
 
 	buf := make([]byte, addressKeySize)
 
-	return db.boltDB.Update(func(tx *bbolt.Tx) error {
+	return db.boltDB.Batch(func(tx *bbolt.Tx) error {
 		bkt := tx.Bucket(graveyardBucketName)
 		if bkt == nil {
 			return nil
diff --git a/pkg/local_object_storage/metabase/inhume.go b/pkg/local_object_storage/metabase/inhume.go
index b62accc4..3aae1506 100644
--- a/pkg/local_object_storage/metabase/inhume.go
+++ b/pkg/local_object_storage/metabase/inhume.go
@@ -181,7 +181,7 @@ func (db *DB) Inhume(ctx context.Context, prm InhumePrm) (InhumeRes, error) {
 		inhumedByCnrID: make(map[cid.ID]ObjectCounters),
 	}
 	currEpoch := db.epochState.CurrentEpoch()
-	err := db.boltDB.Update(func(tx *bbolt.Tx) error {
+	err := db.boltDB.Batch(func(tx *bbolt.Tx) error {
 		return db.inhumeTx(tx, currEpoch, prm, &res)
 	})
 	success = err == nil
diff --git a/pkg/local_object_storage/metabase/lock.go b/pkg/local_object_storage/metabase/lock.go
index 732ba426..6b78ef39 100644
--- a/pkg/local_object_storage/metabase/lock.go
+++ b/pkg/local_object_storage/metabase/lock.go
@@ -78,7 +78,7 @@ func (db *DB) lockInternal(locked []oid.ID, cnr cid.ID, locker oid.ID) error {
 	}
 	key := make([]byte, cidSize)
 
-	return metaerr.Wrap(db.boltDB.Update(func(tx *bbolt.Tx) error {
+	return metaerr.Wrap(db.boltDB.Batch(func(tx *bbolt.Tx) error {
 		if firstIrregularObjectType(tx, cnr, bucketKeysLocked...) != objectSDK.TypeRegular {
 			return logicerr.Wrap(new(apistatus.LockNonRegularObject))
 		}
@@ -143,7 +143,7 @@ func (db *DB) FreeLockedBy(lockers []oid.Address) ([]oid.Address, error) {
 
 	var unlockedObjects []oid.Address
 
-	if err := db.boltDB.Update(func(tx *bbolt.Tx) error {
+	if err := db.boltDB.Batch(func(tx *bbolt.Tx) error {
 		for i := range lockers {
 			unlocked, err := freePotentialLocks(tx, lockers[i].Container(), lockers[i].Object())
 			if err != nil {