Merge pull request #2047 from nspcc-dev/fix-state

Don't create an extension node for the last branch child
This commit is contained in:
Roman Khimov 2021-07-08 10:20:59 +03:00 committed by GitHub
commit ea9dde257c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 39 additions and 1 deletions

View file

@ -192,7 +192,10 @@ func (t *Trie) stripBranch(b *BranchNode) (Node, error) {
case n == 0: case n == 0:
return new(HashNode), nil return new(HashNode), nil
case n == 1: case n == 1:
return t.mergeExtension([]byte{lastIndex}, b.Children[lastIndex]) if lastIndex != lastChild {
return t.mergeExtension([]byte{lastIndex}, b.Children[lastIndex])
}
return b.Children[lastIndex], nil
default: default:
t.addRef(b.Hash(), b.bytes) t.addRef(b.Hash(), b.bytes)
return b, nil return b, nil

View file

@ -174,6 +174,22 @@ func TestTrie_PutBatchBranch(t *testing.T) {
testPut(t, ps, tr1, tr2) testPut(t, ps, tr1, tr2)
require.IsType(t, (*ExtensionNode)(nil), tr1.root) require.IsType(t, (*ExtensionNode)(nil), tr1.root)
}) })
t.Run("non-empty child is last node", func(t *testing.T) {
tr1 := NewTrie(new(HashNode), false, newTestStore())
tr2 := NewTrie(new(HashNode), false, newTestStore())
require.NoError(t, tr1.Put([]byte{0x00, 2}, []byte("value1")))
require.NoError(t, tr2.Put([]byte{0x00, 2}, []byte("value1")))
require.NoError(t, tr1.Put([]byte{0x00}, []byte("value2")))
require.NoError(t, tr2.Put([]byte{0x00}, []byte("value2")))
tr1.Flush()
tr1.Collapse(1)
tr2.Flush()
tr2.Collapse(1)
var ps = pairs{{[]byte{0x00, 2}, nil}}
testPut(t, ps, tr1, tr2)
})
}) })
t.Run("incomplete put, transform to extension", func(t *testing.T) { t.Run("incomplete put, transform to extension", func(t *testing.T) {
tr1, tr2 := prepareBranch(t) tr1, tr2 := prepareBranch(t)

View file

@ -310,6 +310,7 @@ func (t *Trie) deleteFromExtension(n *ExtensionNode, path []byte) (Node, error)
if nxt.IsEmpty() { if nxt.IsEmpty() {
return nxt, nil return nxt, nil
} }
n.next = nxt
default: default:
n.next = r n.next = r
} }

View file

@ -395,6 +395,24 @@ func testTrieDelete(t *testing.T, enableGC bool) {
tr.testHas(t, []byte{}, nil) tr.testHas(t, []byte{}, nil)
tr.testHas(t, []byte{0x56}, []byte{0x34}) tr.testHas(t, []byte{0x56}, []byte{0x34})
require.IsType(t, (*ExtensionNode)(nil), tr.root) require.IsType(t, (*ExtensionNode)(nil), tr.root)
t.Run("WithHash, branch node replaced", func(t *testing.T) {
ch := NewLeafNode([]byte{5, 6})
h := ch.Hash()
b := NewBranchNode()
b.Children[3] = NewExtensionNode([]byte{4}, NewLeafNode([]byte{1, 2, 3}))
b.Children[lastChild] = NewHashNode(h)
tr := NewTrie(NewExtensionNode([]byte{1, 2}, b), enableGC, newTestStore())
tr.putToStore(ch)
require.NoError(t, tr.Delete([]byte{0x12, 0x34}))
tr.testHas(t, []byte{0x12, 0x34}, nil)
tr.testHas(t, []byte{0x12}, []byte{5, 6})
require.IsType(t, (*ExtensionNode)(nil), tr.root)
require.Equal(t, h, tr.root.(*ExtensionNode).next.Hash())
})
}) })
t.Run("LeaveLeaf", func(t *testing.T) { t.Run("LeaveLeaf", func(t *testing.T) {