mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-03-30 15:08:33 +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.
43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
package ping
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/infrastructure/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.(*appmessage.MsgPing)
|
|
|
|
pongMessage := appmessage.NewMsgPong(pingMessage.Nonce)
|
|
err = flow.outgoingRoute.Enqueue(pongMessage)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|