ChronosXYZ/dione
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implement fetching block header by height in blockpool

Browse Source
This commit is contained in:
ChronosX88 2021-05-26 23:58:31 +03:00
parent fb84ab8ad6
commit 155fe7bc71
Signed by: ChronosXYZ
GPG Key ID: 085A69A82C8C511A
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
blockchain/pool/blockpool.go
View File

@ -71,7 +71,7 @@ func NewBlockPool(path string) (*BlockPool, error) {
return pool, nil return pool, nil
} }
func (bp *BlockPool) SetLatestBlockHeight(height uint64) error { func (bp *BlockPool) setLatestBlockHeight(height uint64) error {
return bp.metadataIndex.PutUint64([]byte(LatestBlockHeightKey), height) return bp.metadataIndex.PutUint64([]byte(LatestBlockHeightKey), height)
} }
@ -123,12 +123,12 @@ func (bp *BlockPool) StoreBlock(block *types2.Block) error {
} }
if err == ErrLatestHeightNil { if err == ErrLatestHeightNil {
if err = bp.SetLatestBlockHeight(block.Header.Height); err != nil { if err = bp.setLatestBlockHeight(block.Header.Height); err != nil {
return err return err
} }
} else { } else {
if block.Header.Height > height { if block.Header.Height > height {
if err = bp.SetLatestBlockHeight(block.Header.Height); err != nil { if err = bp.setLatestBlockHeight(block.Header.Height); err != nil {
return err return err
} }
} }
@ -229,3 +229,19 @@ func (bp *BlockPool) FetchBlockByHeight(height uint64) (*types2.Block, error) {
} }
return block, nil return block, nil
} }
func (bp *BlockPool) FetchBlockHeaderByHeight(height uint64) (*types2.BlockHeader, error) {
var heightBytes []byte
binary.LittleEndian.PutUint64(heightBytes, height)
blockHash, err := bp.heightIndex.GetBytes(heightBytes)
if err != nil {
if err == ErrIndexKeyNotFound {
return nil, ErrBlockNotFound
}
}
blockHeader, err := bp.FetchBlockHeader(blockHash)
if err != nil {
return nil, err
}
return blockHeader, nil
}