cerc-io/ipld-eth-beacon-indexer
19
0
Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases 9 Wiki Activity

Minor refactor

Browse Source
This commit is contained in:
Thomas E Lackey 2022-09-08 11:26:05 -05:00
parent a97a2f1b60
commit 90cafc1e54
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pkg/beaconclient/consensus.go
View File

@ -268,17 +268,18 @@ func (b *BeaconBlockBody) Eth1Data() Eth1Data {
func (b *BeaconBlock) HashTreeRoot() Root { func (b *BeaconBlock) HashTreeRoot() Root {
spec := chooseSpec(b.spec) spec := chooseSpec(b.spec)
hashFn := tree.GetHashFn()
if b.IsBellatrix() { if b.IsBellatrix() {
return Root(b.bellatrix.HashTreeRoot(spec, tree.Hash)) return Root(b.bellatrix.HashTreeRoot(spec, hashFn))
} }
if b.IsAltair() { if b.IsAltair() {
return Root(b.altair.HashTreeRoot(spec, tree.Hash)) return Root(b.altair.HashTreeRoot(spec, hashFn))
} }
if b.IsPhase0() { if b.IsPhase0() {
return Root(b.phase0.HashTreeRoot(spec, tree.Hash)) return Root(b.phase0.HashTreeRoot(spec, hashFn))
} }
return Root{} return Root{}
@ -379,17 +380,18 @@ func (s *BeaconState) Slot() Slot {
func (s *BeaconState) HashTreeRoot() Root { func (s *BeaconState) HashTreeRoot() Root {
spec := chooseSpec(s.spec) spec := chooseSpec(s.spec)
hashFn := tree.GetHashFn()
if s.IsBellatrix() { if s.IsBellatrix() {
return Root(s.bellatrix.HashTreeRoot(spec, tree.Hash)) return Root(s.bellatrix.HashTreeRoot(spec, hashFn))
} }
if s.IsAltair() { if s.IsAltair() {
return Root(s.altair.HashTreeRoot(spec, tree.Hash)) return Root(s.altair.HashTreeRoot(spec, hashFn))
} }
if s.IsPhase0() { if s.IsPhase0() {
return Root(s.phase0.HashTreeRoot(spec, tree.Hash)) return Root(s.phase0.HashTreeRoot(spec, hashFn))
} }
return Root{} return Root{}