kaspad/domain/consensus/processes/dagtraversalmanager/selected_child_iterator.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 {
	databaseContext    model.DBReader
	dagTopologyManager model.DAGTopologyManager
	highHash           *externalapi.DomainHash
	current            *externalapi.DomainHash
}

func (s *selectedChildIterator) Next() bool {
	children, err := s.dagTopologyManager.Children(s.current)
	if err != nil {
		panic(err)
	}

	for _, child := range children {
		if child.Equal(model.VirtualBlockHash) {
			continue
		}

		isChildInSelectedParentChainOfHighHash, err := s.dagTopologyManager.IsInSelectedParentChainOf(child, s.highHash)
		if err != nil {
			panic(err)
		}

		if isChildInSelectedParentChainOfHighHash {
			s.current = child
			return true
		}
	}
	return false
}

func (s *selectedChildIterator) Get() *externalapi.DomainHash {
	return s.current
}

func (dtm *dagTraversalManager) SelectedChildIterator(highHash, lowHash *externalapi.DomainHash) (model.BlockIterator, error) {
	isLowHashInSelectedParentChainOfHighHash, err := dtm.dagTopologyManager.IsInSelectedParentChainOf(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{
		databaseContext:    dtm.databaseContext,
		dagTopologyManager: dtm.dagTopologyManager,
		highHash:           highHash,
		current:            lowHash,
	}, nil
}