mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-31 23:41:22 +00:00
612bb38077
* Fix assignments bug where validators don't retry for assignments on failure * synch only please * trying to fix state issues * trying random stuff * do not explode * use ctx * working build, failing tests * broadcast local addrs as well as relay addrs * fixed p2p tests, more tests to fix still * another test fixed, log warning instead of throw error * Fix last tests * godoc * add test for broadcast in apply fork choiec * remove unneeded code * remove tracer adapter, not needed * remove extra stuff * remove any * revert addr_factory * revert addr_factory * Revert "revert addr_factory" This reverts commit e93fb706494a1070158b8db31e67146d6b0648ad. * Revert "revert addr_factory" This reverts commit dedaa405559cc818698870c4e4570953367f1e3c. * revert removal of this code * unused param
108 lines
2.6 KiB
Go
108 lines
2.6 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/boltdb/bolt"
|
|
"github.com/gogo/protobuf/proto"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
"github.com/prysmaticlabs/prysm/shared/hashutil"
|
|
"go.opencensus.io/trace"
|
|
)
|
|
|
|
// SaveAttestation puts the attestation record into the beacon chain db.
|
|
func (db *BeaconDB) SaveAttestation(ctx context.Context, attestation *pb.Attestation) error {
|
|
ctx, span := trace.StartSpan(ctx, "beaconDB.SaveAttestation")
|
|
defer span.End()
|
|
|
|
encodedState, err := proto.Marshal(attestation)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
hash := hashutil.Hash(encodedState)
|
|
|
|
return db.update(func(tx *bolt.Tx) error {
|
|
a := tx.Bucket(attestationBucket)
|
|
|
|
return a.Put(hash[:], encodedState)
|
|
})
|
|
}
|
|
|
|
// DeleteAttestation deletes the attestation record into the beacon chain db.
|
|
func (db *BeaconDB) DeleteAttestation(attestation *pb.Attestation) error {
|
|
hash, err := hashutil.HashProto(attestation)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return db.update(func(tx *bolt.Tx) error {
|
|
a := tx.Bucket(attestationBucket)
|
|
|
|
return a.Delete(hash[:])
|
|
})
|
|
}
|
|
|
|
// Attestation retrieves an attestation record from the db using its hash.
|
|
func (db *BeaconDB) Attestation(hash [32]byte) (*pb.Attestation, error) {
|
|
var attestation *pb.Attestation
|
|
err := db.view(func(tx *bolt.Tx) error {
|
|
a := tx.Bucket(attestationBucket)
|
|
|
|
enc := a.Get(hash[:])
|
|
if enc == nil {
|
|
return nil
|
|
}
|
|
|
|
var err error
|
|
attestation, err = createAttestation(enc)
|
|
return err
|
|
})
|
|
|
|
return attestation, err
|
|
}
|
|
|
|
// Attestations retrieves all the attestation records from the db.
|
|
// These are the attestations that have not been seen on the beacon chain.
|
|
func (db *BeaconDB) Attestations() ([]*pb.Attestation, error) {
|
|
var attestations []*pb.Attestation
|
|
err := db.view(func(tx *bolt.Tx) error {
|
|
a := tx.Bucket(attestationBucket)
|
|
|
|
if err := a.ForEach(func(k, v []byte) error {
|
|
attestation, err := createAttestation(v)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
attestations = append(attestations, attestation)
|
|
return nil
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
})
|
|
|
|
return attestations, err
|
|
}
|
|
|
|
// HasAttestation checks if the attestation exists.
|
|
func (db *BeaconDB) HasAttestation(hash [32]byte) bool {
|
|
exists := false
|
|
// #nosec G104
|
|
db.view(func(tx *bolt.Tx) error {
|
|
a := tx.Bucket(attestationBucket)
|
|
|
|
exists = a.Get(hash[:]) != nil
|
|
return nil
|
|
})
|
|
return exists
|
|
}
|
|
|
|
func createAttestation(enc []byte) (*pb.Attestation, error) {
|
|
protoAttestation := &pb.Attestation{}
|
|
if err := proto.Unmarshal(enc, protoAttestation); err != nil {
|
|
return nil, fmt.Errorf("failed to unmarshal encoding: %v", err)
|
|
}
|
|
return protoAttestation, nil
|
|
}
|