mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-25 15:32:32 +00:00
42e50e6dc2
* [NOD-1191] Convert wire protocol to 100% protobuf * [NOD-1191] Simplify wire interface and remove redundant messages * [NOD-1191] Map all proto to wire conversions * [NOD-1203] Create netadapter outside of protocol manager * [NOD-1191] Fix nil errors * [NOD-1191] Fix comments * [NOD-1191] Add converter interface * [NOD-1191] Add missing GetBlockLocator message * [NOD-1191] Change message names that starts with 'get' to 'request' * [NOD-1191] Change message commands values * [NOD-1191] Remove redundant methods * [NOD-1191] Rename message constructors * [NOD-1191] Change message commands to use iota * [NOD-1191] Add missing outputs to protobuf conversion * [NOD-1191] Make block header a required field * [NOD-1191] Rename variables * [NOD-1212] Fix test names * [NOD-1191] Rename flow names * [NOD-1191] Fix infinite loop
80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
package selectedtip
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/blockdag"
|
|
"github.com/kaspanet/kaspad/netadapter/router"
|
|
"github.com/kaspanet/kaspad/protocol/common"
|
|
peerpkg "github.com/kaspanet/kaspad/protocol/peer"
|
|
"github.com/kaspanet/kaspad/util/daghash"
|
|
"github.com/kaspanet/kaspad/wire"
|
|
)
|
|
|
|
// 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 := wire.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.(*wire.MsgSelectedTip)
|
|
|
|
return msgSelectedTip.SelectedTipHash, nil
|
|
}
|