kaspad/infrastructure/network/rpcclient/rpc_on_finality_conflicts.go

package rpcclient

import (
	"github.com/kaspanet/kaspad/app/appmessage"
	"github.com/kaspanet/kaspad/app/rpc/rpccontext"
	routerpkg "github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
	"github.com/pkg/errors"
)

// RegisterForFinalityConflictsNotifications sends an RPC request respective to the function's name and returns the RPC server's response.
// Additionally, it starts listening for the appropriate notification using the given handler function
func (c *RPCClient) RegisterForFinalityConflictsNotifications(
	onFinalityConflict func(notification *appmessage.FinalityConflictNotificationMessage),
	onFinalityConflictResolved func(notification *appmessage.FinalityConflictResolvedNotificationMessage)) error {

	err := c.rpcRouter.outgoingRoute().Enqueue(appmessage.NewNotifyFinalityConflictsRequestMessage(rpccontext.DefaultNotificationID))
	if err != nil {
		return err
	}
	response, err := c.route(appmessage.CmdNotifyFinalityConflictsResponseMessage).DequeueWithTimeout(c.timeout)
	if err != nil {
		return err
	}
	notifyFinalityConflictsResponse := response.(*appmessage.NotifyFinalityConflictsResponseMessage)
	if notifyFinalityConflictsResponse.Error != nil {
		return c.convertRPCError(notifyFinalityConflictsResponse.Error)
	}
	spawn("RegisterForFinalityConflictsNotifications-finalityConflict", func() {
		for {
			notification, err := c.route(appmessage.CmdFinalityConflictNotificationMessage).Dequeue()
			if err != nil {
				if errors.Is(err, routerpkg.ErrRouteClosed) {
					break
				}
				panic(err)
			}
			finalityConflictNotification := notification.(*appmessage.FinalityConflictNotificationMessage)
			onFinalityConflict(finalityConflictNotification)
		}
	})
	spawn("RegisterForFinalityConflictsNotifications-finalityConflictResolved", func() {
		for {
			notification, err := c.route(appmessage.CmdFinalityConflictResolvedNotificationMessage).Dequeue()
			if err != nil {
				if errors.Is(err, routerpkg.ErrRouteClosed) {
					break
				}
				panic(err)
			}
			finalityConflictResolvedNotification := notification.(*appmessage.FinalityConflictResolvedNotificationMessage)
			onFinalityConflictResolved(finalityConflictResolvedNotification)
		}
	})
	return nil
}

// RegisterForFinalityConflictsNotificationsWithID does the same as
// RegisterForFinalityConflictsNotifications, but allows the client to specify an id
func (c *RPCClient) RegisterForFinalityConflictsNotificationsWithID(
	onFinalityConflict func(notification *appmessage.FinalityConflictNotificationMessage),
	onFinalityConflictResolved func(notification *appmessage.FinalityConflictResolvedNotificationMessage), id string) error {

	err := c.rpcRouter.outgoingRoute().Enqueue(appmessage.NewNotifyFinalityConflictsRequestMessage(id))
	if err != nil {
		return err
	}
	response, err := c.route(appmessage.CmdNotifyFinalityConflictsResponseMessage).DequeueWithTimeout(c.timeout)
	if err != nil {
		return err
	}
	notifyFinalityConflictsResponse := response.(*appmessage.NotifyFinalityConflictsResponseMessage)
	if notifyFinalityConflictsResponse.Error != nil {
		return c.convertRPCError(notifyFinalityConflictsResponse.Error)
	}
	spawn("RegisterForFinalityConflictsNotificationsWithID-finalityConflict", func() {
		for {
			notification, err := c.route(appmessage.CmdFinalityConflictNotificationMessage).Dequeue()
			if err != nil {
				if errors.Is(err, routerpkg.ErrRouteClosed) {
					break
				}
				panic(err)
			}
			finalityConflictNotification := notification.(*appmessage.FinalityConflictNotificationMessage)
			onFinalityConflict(finalityConflictNotification)
		}
	})
	spawn("RegisterForFinalityConflictsNotificationsWithID-finalityConflictResolved", func() {
		for {
			notification, err := c.route(appmessage.CmdFinalityConflictResolvedNotificationMessage).Dequeue()
			if err != nil {
				if errors.Is(err, routerpkg.ErrRouteClosed) {
					break
				}
				panic(err)
			}
			finalityConflictResolvedNotification := notification.(*appmessage.FinalityConflictResolvedNotificationMessage)
			onFinalityConflictResolved(finalityConflictResolvedNotification)
		}
	})
	return nil
}