From 6b02df7b8c97c920a719d66b067f63bf1c35346a Mon Sep 17 00:00:00 2001 From: Evgenii Stratonikov Date: Sat, 28 May 2022 17:11:50 +0300 Subject: [PATCH] [#1444] pilorama: Fix `TreeMove` in bbolt backend Signed-off-by: Evgenii Stratonikov --- pkg/local_object_storage/pilorama/boltdb.go | 35 +++++++++---------- .../pilorama/forest_test.go | 6 ++-- 2 files changed, 21 insertions(+), 20 deletions(-) diff --git a/pkg/local_object_storage/pilorama/boltdb.go b/pkg/local_object_storage/pilorama/boltdb.go index c582fc6e9..31aa36293 100644 --- a/pkg/local_object_storage/pilorama/boltdb.go +++ b/pkg/local_object_storage/pilorama/boltdb.go @@ -79,8 +79,8 @@ func (t *boltForest) TreeMove(d CIDDescriptor, treeID string, m *Move) (*LogMove return nil, ErrInvalidCIDDescriptor } - var lm *LogMove - return lm, t.db.Batch(func(tx *bbolt.Tx) error { + var lm LogMove + return &lm, t.db.Batch(func(tx *bbolt.Tx) error { bLog, bTree, err := t.getTreeBuckets(tx, d.CID, treeID) if err != nil { return err @@ -90,8 +90,8 @@ func (t *boltForest) TreeMove(d CIDDescriptor, treeID string, m *Move) (*LogMove if m.Child == RootID { m.Child = t.findSpareID(bTree) } - lm, err = t.applyOperation(bLog, bTree, m) - return err + lm.Move = *m + return t.applyOperation(bLog, bTree, &lm) }) } @@ -193,8 +193,9 @@ func (t *boltForest) TreeApply(d CIDDescriptor, treeID string, m *Move) error { if err != nil { return err } - _, err = t.applyOperation(bLog, bTree, m) - return err + + lm := &LogMove{Move: *m} + return t.applyOperation(bLog, bTree, lm) }) } @@ -222,8 +223,7 @@ func (t *boltForest) getTreeBuckets(tx *bbolt.Tx, cid cidSDK.ID, treeID string) return bLog, bData, nil } -func (t *boltForest) applyOperation(logBucket, treeBucket *bbolt.Bucket, m *Move) (*LogMove, error) { - var lm LogMove +func (t *boltForest) applyOperation(logBucket, treeBucket *bbolt.Bucket, lm *LogMove) error { var tmp LogMove var cKey [17]byte @@ -235,22 +235,21 @@ func (t *boltForest) applyOperation(logBucket, treeBucket *bbolt.Bucket, m *Move r := io.NewBinReaderFromIO(b) // 1. Undo up until the desired timestamp is here. - for len(key) == 8 && binary.BigEndian.Uint64(key) > m.Time { + for len(key) == 8 && binary.BigEndian.Uint64(key) > lm.Time { b.Reset(value) if err := t.logFromBytes(&tmp, r); err != nil { - return nil, err + return err } if err := t.undo(&tmp.Move, &tmp, treeBucket, cKey[:]); err != nil { - return nil, err + return err } key, value = c.Prev() } // 2. Insert the operation. - if len(key) != 8 || binary.BigEndian.Uint64(key) != m.Time { - lm.Move = *m - if err := t.do(logBucket, treeBucket, cKey[:], &lm); err != nil { - return nil, err + if len(key) != 8 || binary.BigEndian.Uint64(key) != lm.Time { + if err := t.do(logBucket, treeBucket, cKey[:], lm); err != nil { + return err } } key, value = c.Next() @@ -259,15 +258,15 @@ func (t *boltForest) applyOperation(logBucket, treeBucket *bbolt.Bucket, m *Move for len(key) == 8 { b.Reset(value) if err := t.logFromBytes(&tmp, r); err != nil { - return nil, err + return err } if err := t.do(logBucket, treeBucket, cKey[:], &tmp); err != nil { - return nil, err + return err } key, value = c.Next() } - return &lm, nil + return nil } func (t *boltForest) do(lb *bbolt.Bucket, b *bbolt.Bucket, key []byte, op *LogMove) error { diff --git a/pkg/local_object_storage/pilorama/forest_test.go b/pkg/local_object_storage/pilorama/forest_test.go index 20ca79399..909b16c10 100644 --- a/pkg/local_object_storage/pilorama/forest_test.go +++ b/pkg/local_object_storage/pilorama/forest_test.go @@ -79,24 +79,26 @@ func testForestTreeMove(t *testing.T, s Forest) { require.ErrorIs(t, err, ErrInvalidCIDDescriptor) }) t.Run("same parent, update meta", func(t *testing.T) { - _, err = s.TreeMove(d, treeID, &Move{ + res, err := s.TreeMove(d, treeID, &Move{ Parent: lm[1].Child, Meta: Meta{Items: append(meta, KeyValue{Key: "NewKey", Value: []byte("NewValue")})}, Child: nodeID, }) require.NoError(t, err) + require.Equal(t, res.Child, nodeID) nodes, err := s.TreeGetByPath(cid, treeID, AttributeFilename, []string{"path", "to", "file.txt"}, false) require.NoError(t, err) require.ElementsMatch(t, []Node{nodeID}, nodes) }) t.Run("different parent", func(t *testing.T) { - _, err = s.TreeMove(d, treeID, &Move{ + res, err := s.TreeMove(d, treeID, &Move{ Parent: RootID, Meta: Meta{Items: append(meta, KeyValue{Key: "NewKey", Value: []byte("NewValue")})}, Child: nodeID, }) require.NoError(t, err) + require.Equal(t, res.Child, nodeID) nodes, err := s.TreeGetByPath(cid, treeID, AttributeFilename, []string{"path", "to", "file.txt"}, false) require.NoError(t, err)