kaspad/app/protocol/flows/v3/ping/receive.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
		}
	}
}