mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-23 15:26:42 +00:00
d14809694f
* [NOD-1223] Delete unused files/packages. * [NOD-1223] Move signal and limits to the os package. * [NOD-1223] Put database and dbaccess into the db package. * [NOD-1223] Fold the logs package into the logger package. * [NOD-1223] Rename domainmessage to appmessage. * [NOD-1223] Rename to/from DomainMessage to AppMessage. * [NOD-1223] Move appmessage to the app packge. * [NOD-1223] Move protocol to the app packge. * [NOD-1223] Move the network package to the infrastructure packge. * [NOD-1223] Rename cmd to executables. * [NOD-1223] Fix go.doc in the logger package.
80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
package selectedtip
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/app/protocol/common"
|
|
peerpkg "github.com/kaspanet/kaspad/app/protocol/peer"
|
|
"github.com/kaspanet/kaspad/domain/blockdag"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
|
|
"github.com/kaspanet/kaspad/util/daghash"
|
|
)
|
|
|
|
// RequestSelectedTipContext is the interface for the context needed for the RequestSelectedTip flow.
|
|
type RequestSelectedTipContext interface {
|
|
DAG() *blockdag.BlockDAG
|
|
StartIBDIfRequired()
|
|
}
|
|
|
|
type requestSelectedTipFlow struct {
|
|
RequestSelectedTipContext
|
|
incomingRoute, outgoingRoute *router.Route
|
|
peer *peerpkg.Peer
|
|
}
|
|
|
|
// RequestSelectedTip waits for selected tip requests and handles them
|
|
func RequestSelectedTip(context RequestSelectedTipContext, incomingRoute *router.Route,
|
|
outgoingRoute *router.Route, peer *peerpkg.Peer) error {
|
|
|
|
flow := &requestSelectedTipFlow{
|
|
RequestSelectedTipContext: context,
|
|
incomingRoute: incomingRoute,
|
|
outgoingRoute: outgoingRoute,
|
|
peer: peer,
|
|
}
|
|
return flow.start()
|
|
}
|
|
|
|
func (flow *requestSelectedTipFlow) start() error {
|
|
for {
|
|
err := flow.runSelectedTipRequest()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
func (flow *requestSelectedTipFlow) runSelectedTipRequest() error {
|
|
|
|
flow.peer.WaitForSelectedTipRequests()
|
|
defer flow.peer.FinishRequestingSelectedTip()
|
|
|
|
err := flow.requestSelectedTip()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
peerSelectedTipHash, err := flow.receiveSelectedTip()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
flow.peer.SetSelectedTipHash(peerSelectedTipHash)
|
|
flow.StartIBDIfRequired()
|
|
return nil
|
|
}
|
|
|
|
func (flow *requestSelectedTipFlow) requestSelectedTip() error {
|
|
msgGetSelectedTip := appmessage.NewMsgRequestSelectedTip()
|
|
return flow.outgoingRoute.Enqueue(msgGetSelectedTip)
|
|
}
|
|
|
|
func (flow *requestSelectedTipFlow) receiveSelectedTip() (selectedTipHash *daghash.Hash, err error) {
|
|
message, err := flow.incomingRoute.DequeueWithTimeout(common.DefaultTimeout)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
msgSelectedTip := message.(*appmessage.MsgSelectedTip)
|
|
|
|
return msgSelectedTip.SelectedTipHash, nil
|
|
}
|