mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-09-15 05:50:10 +00:00
8a4ece1101
* [NOD-1223] Move all network stuff into a new network package. * [NOD-1223] Delete the unused package testutil. * [NOD-1223] Move infrastructure stuff into a new instrastructure package. * [NOD-1223] Move domain stuff into a new domain package.
43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
package ping
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/network/domainmessage"
|
|
"github.com/kaspanet/kaspad/network/netadapter/router"
|
|
)
|
|
|
|
// ReceivePingsContext is the interface for the context needed for the ReceivePings flow.
|
|
type ReceivePingsContext interface {
|
|
}
|
|
|
|
type receivePingsFlow struct {
|
|
ReceivePingsContext
|
|
incomingRoute, outgoingRoute *router.Route
|
|
}
|
|
|
|
// ReceivePings handles all ping messages coming through incomingRoute.
|
|
// This function assumes that incomingRoute will only return MsgPing.
|
|
func ReceivePings(context ReceivePingsContext, incomingRoute *router.Route, outgoingRoute *router.Route) error {
|
|
flow := &receivePingsFlow{
|
|
ReceivePingsContext: context,
|
|
incomingRoute: incomingRoute,
|
|
outgoingRoute: outgoingRoute,
|
|
}
|
|
return flow.start()
|
|
}
|
|
|
|
func (flow *receivePingsFlow) start() error {
|
|
for {
|
|
message, err := flow.incomingRoute.Dequeue()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
pingMessage := message.(*domainmessage.MsgPing)
|
|
|
|
pongMessage := domainmessage.NewMsgPong(pingMessage.Nonce)
|
|
err = flow.outgoingRoute.Enqueue(pongMessage)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|