forked from cerc-io/plugeth
79a478bb61
This PR implements resettable freezer by adding a ResettableFreezer wrapper. The resettable freezer wraps the original freezer in a way that makes it possible to ensure atomic resets. Implementation wise, it relies on the os.Rename and os.RemoveAll to atomically delete the original freezer data and re-create a new one from scratch.
108 lines
2.8 KiB
Go
108 lines
2.8 KiB
Go
// Copyright 2022 The go-ethereum Authors
|
|
// This file is part of the go-ethereum library.
|
|
//
|
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// The go-ethereum library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package rawdb
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/ethdb"
|
|
)
|
|
|
|
func TestResetFreezer(t *testing.T) {
|
|
items := []struct {
|
|
id uint64
|
|
blob []byte
|
|
}{
|
|
{0, bytes.Repeat([]byte{0}, 2048)},
|
|
{1, bytes.Repeat([]byte{1}, 2048)},
|
|
{2, bytes.Repeat([]byte{2}, 2048)},
|
|
}
|
|
f, _ := NewResettableFreezer(t.TempDir(), "", false, 2048, freezerTestTableDef)
|
|
defer f.Close()
|
|
|
|
f.ModifyAncients(func(op ethdb.AncientWriteOp) error {
|
|
for _, item := range items {
|
|
op.AppendRaw("test", item.id, item.blob)
|
|
}
|
|
return nil
|
|
})
|
|
for _, item := range items {
|
|
blob, _ := f.Ancient("test", item.id)
|
|
if !bytes.Equal(blob, item.blob) {
|
|
t.Fatal("Unexpected blob")
|
|
}
|
|
}
|
|
|
|
// Reset freezer
|
|
f.Reset()
|
|
count, _ := f.Ancients()
|
|
if count != 0 {
|
|
t.Fatal("Failed to reset freezer")
|
|
}
|
|
for _, item := range items {
|
|
blob, _ := f.Ancient("test", item.id)
|
|
if len(blob) != 0 {
|
|
t.Fatal("Unexpected blob")
|
|
}
|
|
}
|
|
|
|
// Fill the freezer
|
|
f.ModifyAncients(func(op ethdb.AncientWriteOp) error {
|
|
for _, item := range items {
|
|
op.AppendRaw("test", item.id, item.blob)
|
|
}
|
|
return nil
|
|
})
|
|
for _, item := range items {
|
|
blob, _ := f.Ancient("test", item.id)
|
|
if !bytes.Equal(blob, item.blob) {
|
|
t.Fatal("Unexpected blob")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestFreezerCleanup(t *testing.T) {
|
|
items := []struct {
|
|
id uint64
|
|
blob []byte
|
|
}{
|
|
{0, bytes.Repeat([]byte{0}, 2048)},
|
|
{1, bytes.Repeat([]byte{1}, 2048)},
|
|
{2, bytes.Repeat([]byte{2}, 2048)},
|
|
}
|
|
datadir := t.TempDir()
|
|
f, _ := NewResettableFreezer(datadir, "", false, 2048, freezerTestTableDef)
|
|
f.ModifyAncients(func(op ethdb.AncientWriteOp) error {
|
|
for _, item := range items {
|
|
op.AppendRaw("test", item.id, item.blob)
|
|
}
|
|
return nil
|
|
})
|
|
f.Close()
|
|
os.Rename(datadir, tmpName(datadir))
|
|
|
|
// Open the freezer again, trigger cleanup operation
|
|
f, _ = NewResettableFreezer(datadir, "", false, 2048, freezerTestTableDef)
|
|
f.Close()
|
|
|
|
if _, err := os.Lstat(tmpName(datadir)); !os.IsNotExist(err) {
|
|
t.Fatal("Failed to cleanup leftover directory")
|
|
}
|
|
}
|