mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
c419e4ed8f
* Add database migrations, still need to update the API usage... * gofmt goimports * progress * Merge branch 'master' of github.com:prysmaticlabs/prysm into index-migration * use slot instead of index * rename LastArchivedIndex to LastArchivedSlot * rename LastArchivedIndexRoot to LastArchivedRoot * remove unused HighestSlotStates method * deprecate old key, include in migration * deprecate old key, include in migration * remove blocks index in migration * rename bucket variable * fix code to pass tests * Merge branch 'master' of github.com:prysmaticlabs/prysm into index-migration * gofmt, goimports * fix * Add state slot index * progress * lint * fix build * Merge branch 'master' of github.com:prysmaticlabs/prysm into index-migration * kafka * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * remove SaveArchivedPointRoot, a few other big changes * Merge branch 'index-migration' of github.com:prysmaticlabs/prysm into index-migration * fix tests and lint * lint again * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * block migration, some renaming * gaz, gofmt * add tests * change index to uint bytes * Merge branch 'index-migration' of github.com:prysmaticlabs/prysm into index-migration * rm method notes * stop if the bucket doesn't exist * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * @rauljordan pr feedback * Simplify * Merge refs/heads/master into index-migration * Remove unused method, add roundtrip test * gofmt * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge refs/heads/master into index-migration * Merge branch 'master' of github.com:prysmaticlabs/prysm into index-migration
62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package kv
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
|
|
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
bolt "go.etcd.io/bbolt"
|
|
)
|
|
|
|
var migrationArchivedIndex0Key = []byte("archive_index_0")
|
|
|
|
func migrateArchivedIndex(tx *bolt.Tx) error {
|
|
mb := tx.Bucket(migrationsBucket)
|
|
if b := mb.Get(migrationArchivedIndex0Key); bytes.Equal(b, migrationCompleted) {
|
|
return nil // Migration already completed.
|
|
}
|
|
|
|
bkt := tx.Bucket(archivedRootBucket)
|
|
if bkt == nil {
|
|
return nil
|
|
}
|
|
// Remove "last archived index" key before iterating over all keys.
|
|
if err := bkt.Delete(lastArchivedIndexKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
var highest uint64
|
|
c := bkt.Cursor()
|
|
for k, v := c.First(); k != nil; k, v = c.Next() {
|
|
// Look up actual slot from block
|
|
b := tx.Bucket(blocksBucket).Get(v)
|
|
// Skip this key if there is no block for whatever reason.
|
|
if b == nil {
|
|
continue
|
|
}
|
|
blk := ðpb.SignedBeaconBlock{}
|
|
if err := decode(context.Background(), b, blk); err != nil {
|
|
return err
|
|
}
|
|
if err := tx.Bucket(stateSlotIndicesBucket).Put(bytesutil.Uint64ToBytesBigEndian(blk.Block.Slot), v); err != nil {
|
|
return err
|
|
}
|
|
if blk.Block.Slot > highest {
|
|
highest = blk.Block.Slot
|
|
}
|
|
}
|
|
|
|
// Delete deprecated buckets.
|
|
for _, bkt := range [][]byte{slotsHasObjectBucket, archivedRootBucket} {
|
|
if tx.Bucket(bkt) != nil {
|
|
if err := tx.DeleteBucket(bkt); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
// Mark migration complete.
|
|
return mb.Put(migrationArchivedIndex0Key, migrationCompleted)
|
|
}
|