From 4aaa50c8edc770e6d66c66ff2ee4bb0a1145e81a Mon Sep 17 00:00:00 2001 From: Evgenii Stratonikov Date: Thu, 30 May 2024 15:32:34 +0300 Subject: [PATCH] [#141] registry: Use Batch() for all methods Refs #136 Signed-off-by: Evgenii Stratonikov --- internal/registry/obj_registry.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/internal/registry/obj_registry.go b/internal/registry/obj_registry.go index 2671401..f45a3d0 100644 --- a/internal/registry/obj_registry.go +++ b/internal/registry/obj_registry.go @@ -75,7 +75,7 @@ func (o *ObjRegistry) AddObject(cid, oid, s3Bucket, s3Key, payloadHash string) e } func (o *ObjRegistry) SetObjectStatus(id uint64, oldStatus, newStatus string) error { - return o.boltDB.Update(func(tx *bbolt.Tx) error { + return o.boltDB.Batch(func(tx *bbolt.Tx) error { oldB := tx.Bucket([]byte(oldStatus)) if oldB == nil { return fmt.Errorf("bucket doesn't exist: '%s'", oldStatus) @@ -110,7 +110,7 @@ func (o *ObjRegistry) SetObjectStatus(id uint64, oldStatus, newStatus string) er } func (o *ObjRegistry) DeleteObject(id uint64) error { - return o.boltDB.Update(func(tx *bbolt.Tx) error { + return o.boltDB.Batch(func(tx *bbolt.Tx) error { return tx.ForEach(func(_ []byte, b *bbolt.Bucket) error { return b.Delete(encodeId(id)) })