mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-25 13:07:17 +00:00
1b5d102527
Co-authored-by: giuliorebuffo <giuliorebuffo@system76-pc.localdomain>
45 lines
733 B
Go
45 lines
733 B
Go
package verkletrie
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/ledgerwatch/erigon-lib/kv"
|
|
"github.com/ledgerwatch/erigon-lib/kv/mdbx"
|
|
)
|
|
|
|
type VerkleMarker struct {
|
|
db kv.RwDB
|
|
tx kv.RwTx
|
|
}
|
|
|
|
//nolint:gocritic
|
|
func NewVerkleMarker() *VerkleMarker {
|
|
markedSlotsDb, err := mdbx.NewTemporaryMdbx()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
tx, err := markedSlotsDb.BeginRw(context.TODO())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return &VerkleMarker{
|
|
db: markedSlotsDb,
|
|
tx: tx,
|
|
}
|
|
}
|
|
|
|
func (v *VerkleMarker) MarkAsDone(key []byte) error {
|
|
return v.tx.Put(kv.Headers, key, []byte{0})
|
|
}
|
|
|
|
func (v *VerkleMarker) IsMarked(key []byte) (bool, error) {
|
|
return v.tx.Has(kv.Headers, key)
|
|
}
|
|
|
|
func (v *VerkleMarker) Rollback() {
|
|
v.tx.Rollback()
|
|
v.db.Close()
|
|
}
|