mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-08-13 22:23:15 +00:00
58d627e05a
* Unite all reachability stores * Upgrade script * Fix tests * Add UpdateReindexRoot to RebuildReachability * Use dbTx when deleting reachability stores * Use ghostdagDataWithoutPrunedBlocks when rebuilding reachability * Use next tree ancestor wherever possible and avoid finality point search if the block is too close to pruning point * Address the boundary case where the pruning point becomes the finality point * some minor fixes * Remove RebuildReachability and use manual syncing between old and new consensus for migration * Remove sanity test (it failed when tips where not in the same order) Co-authored-by: msutton <mikisiton2@gmail.com>
110 lines
2.9 KiB
Go
110 lines
2.9 KiB
Go
package dagtraversalmanager
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/domain/consensus/model"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type selectedChildIterator struct {
|
|
dagTraversalManager model.DAGTraversalManager
|
|
|
|
includeLowHash bool
|
|
highHash, lowHash *externalapi.DomainHash
|
|
current *externalapi.DomainHash
|
|
err error
|
|
isClosed bool
|
|
stagingArea *model.StagingArea
|
|
}
|
|
|
|
func (s *selectedChildIterator) First() bool {
|
|
if s.isClosed {
|
|
panic("Tried using a closed SelectedChildIterator")
|
|
}
|
|
s.current = s.lowHash
|
|
if s.includeLowHash {
|
|
return true
|
|
}
|
|
|
|
return s.Next()
|
|
}
|
|
|
|
func (s *selectedChildIterator) Next() bool {
|
|
if s.isClosed {
|
|
panic("Tried using a closed SelectedChildIterator")
|
|
}
|
|
if s.err != nil {
|
|
return true
|
|
}
|
|
|
|
selectedChild, err := s.dagTraversalManager.SelectedChild(s.stagingArea, s.highHash, s.current)
|
|
if errors.Is(err, errNoSelectedChild) {
|
|
return false
|
|
}
|
|
if err != nil {
|
|
s.current = nil
|
|
s.err = err
|
|
return true
|
|
}
|
|
|
|
s.current = selectedChild
|
|
return true
|
|
}
|
|
|
|
func (s *selectedChildIterator) Get() (*externalapi.DomainHash, error) {
|
|
if s.isClosed {
|
|
return nil, errors.New("Tried using a closed SelectedChildIterator")
|
|
}
|
|
return s.current, s.err
|
|
}
|
|
|
|
func (s *selectedChildIterator) Close() error {
|
|
if s.isClosed {
|
|
return errors.New("Tried using a closed SelectedChildIterator")
|
|
}
|
|
s.isClosed = true
|
|
s.highHash = nil
|
|
s.lowHash = nil
|
|
s.current = nil
|
|
s.err = nil
|
|
return nil
|
|
}
|
|
|
|
// SelectedChildIterator returns a BlockIterator that iterates from lowHash (exclusive) to highHash (inclusive) over
|
|
// highHash's selected parent chain
|
|
func (dtm *dagTraversalManager) SelectedChildIterator(stagingArea *model.StagingArea,
|
|
highHash, lowHash *externalapi.DomainHash, includeLowHash bool) (model.BlockIterator, error) {
|
|
|
|
isLowHashInSelectedParentChainOfHighHash, err := dtm.dagTopologyManager.IsInSelectedParentChainOf(
|
|
stagingArea, lowHash, highHash)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if !isLowHashInSelectedParentChainOfHighHash {
|
|
return nil, errors.Errorf("%s is not in the selected parent chain of %s", highHash, lowHash)
|
|
}
|
|
return &selectedChildIterator{
|
|
dagTraversalManager: dtm,
|
|
includeLowHash: includeLowHash,
|
|
highHash: highHash,
|
|
lowHash: lowHash,
|
|
current: lowHash,
|
|
stagingArea: stagingArea,
|
|
}, nil
|
|
}
|
|
|
|
var errNoSelectedChild = errors.New("errNoSelectedChild")
|
|
|
|
func (dtm *dagTraversalManager) SelectedChild(stagingArea *model.StagingArea,
|
|
highHash, lowHash *externalapi.DomainHash) (*externalapi.DomainHash, error) {
|
|
|
|
// The selected child is in fact the next reachability tree nextAncestor
|
|
nextAncestor, err := dtm.reachabilityManager.FindNextAncestor(stagingArea, highHash, lowHash)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(errNoSelectedChild, "no selected child for %s from the point of view of %s",
|
|
lowHash, highHash)
|
|
}
|
|
return nextAncestor, nil
|
|
}
|