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
62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
package selectedtip
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/blockdag"
|
|
"github.com/kaspanet/kaspad/netadapter/router"
|
|
"github.com/kaspanet/kaspad/wire"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// HandleRequestSelectedTipContext is the interface for the context needed for the HandleRequestSelectedTip flow.
|
|
type HandleRequestSelectedTipContext interface {
|
|
DAG() *blockdag.BlockDAG
|
|
}
|
|
|
|
type handleRequestSelectedTipFlow struct {
|
|
HandleRequestSelectedTipContext
|
|
incomingRoute, outgoingRoute *router.Route
|
|
}
|
|
|
|
// HandleRequestSelectedTip handles getSelectedTip messages
|
|
func HandleRequestSelectedTip(context HandleRequestSelectedTipContext, incomingRoute *router.Route, outgoingRoute *router.Route) error {
|
|
flow := &handleRequestSelectedTipFlow{
|
|
HandleRequestSelectedTipContext: context,
|
|
incomingRoute: incomingRoute,
|
|
outgoingRoute: outgoingRoute,
|
|
}
|
|
return flow.start()
|
|
}
|
|
|
|
func (flow *handleRequestSelectedTipFlow) start() error {
|
|
for {
|
|
err := flow.receiveGetSelectedTip()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = flow.sendSelectedTipHash()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
func (flow *handleRequestSelectedTipFlow) receiveGetSelectedTip() error {
|
|
message, err := flow.incomingRoute.Dequeue()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, ok := message.(*wire.MsgRequestSelectedTip)
|
|
if !ok {
|
|
return errors.Errorf("received unexpected message type. "+
|
|
"expected: %s, got: %s", wire.CmdRequestSelectedTip, message.Command())
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (flow *handleRequestSelectedTipFlow) sendSelectedTipHash() error {
|
|
msgSelectedTip := wire.NewMsgSelectedTip(flow.DAG().SelectedTipHash())
|
|
return flow.outgoingRoute.Enqueue(msgSelectedTip)
|
|
}
|