prysm-pulse/beacon-chain/db/kv/migration_block_slot_index.go
terencechain d17996f8b0
Update to V4 🚀 (#12134)
* Update V3 from V4

* Fix build v3 -> v4

* Update ssz

* Update beacon_chain.pb.go

* Fix formatter import

* Update update-mockgen.sh comment to v4

* Fix conflicts. Pass build and tests

* Fix test
2023-03-17 18:52:56 +00:00

51 lines
1.2 KiB
Go

package kv
import (
"bytes"
"context"
"strconv"
"github.com/prysmaticlabs/prysm/v4/encoding/bytesutil"
bolt "go.etcd.io/bbolt"
)
var migrationBlockSlotIndex0Key = []byte("block_slot_index_0")
func migrateBlockSlotIndex(ctx context.Context, db *bolt.DB) error {
if updateErr := db.Update(func(tx *bolt.Tx) error {
mb := tx.Bucket(migrationsBucket)
if b := mb.Get(migrationBlockSlotIndex0Key); bytes.Equal(b, migrationCompleted) {
return nil // Migration already completed.
}
bkt := tx.Bucket(blockSlotIndicesBucket)
// Convert indices from strings to big endian integers.
if err := bkt.ForEach(func(k, v []byte) error {
key, err := strconv.ParseUint(string(k), 10, 64)
if err != nil {
return err
}
if err = bkt.Delete(k); err != nil {
return err
}
if err = bkt.Put(bytesutil.Uint64ToBytesBigEndian(key), v); err != nil {
return err
}
// check if context is cancelled in between
if ctx.Err() != nil {
return ctx.Err()
}
return nil
}); err != nil {
return err
}
return mb.Put(migrationBlockSlotIndex0Key, migrationCompleted)
}); updateErr != nil {
log.WithError(updateErr).Errorf("could not migrate bucket: %s", blockSlotIndicesBucket)
return updateErr
}
return nil
}