mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-23 15:26:42 +00:00
4855d845b3
* Separate IBD to a new flow (so now invs are handled concurrently and no route capacity errors) * Invs messages should be queued while waiting for BlockLocator msg * Close IBD channel so that HandleIBDFlow exits too * Apply flow separation to p2p protocol v4 * Manage the IBDRequestChannel through the Peer struct * Some IBDs take a little longer
34 lines
1.1 KiB
Go
34 lines
1.1 KiB
Go
package blockrelay
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/app/protocol/common"
|
|
"github.com/kaspanet/kaspad/app/protocol/protocolerrors"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
)
|
|
|
|
func (flow *handleRelayInvsFlow) sendGetBlockLocator(highHash *externalapi.DomainHash, limit uint32) error {
|
|
msgGetBlockLocator := appmessage.NewMsgRequestBlockLocator(highHash, limit)
|
|
return flow.outgoingRoute.Enqueue(msgGetBlockLocator)
|
|
}
|
|
|
|
func (flow *handleRelayInvsFlow) receiveBlockLocator() (blockLocatorHashes []*externalapi.DomainHash, err error) {
|
|
for {
|
|
message, err := flow.incomingRoute.DequeueWithTimeout(common.DefaultTimeout)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
switch message := message.(type) {
|
|
case *appmessage.MsgInvRelayBlock:
|
|
flow.invsQueue = append(flow.invsQueue, message)
|
|
case *appmessage.MsgBlockLocator:
|
|
return message.BlockLocatorHashes, nil
|
|
default:
|
|
return nil,
|
|
protocolerrors.Errorf(true, "received unexpected message type. "+
|
|
"expected: %s, got: %s", appmessage.CmdBlockLocator, message.Command())
|
|
}
|
|
}
|
|
}
|