diff --git a/infrastructure/network/netadapter/server/grpcserver/protowire/message_reject.go b/infrastructure/network/netadapter/server/grpcserver/protowire/message_reject.go index de801a6fc..c48a35ac8 100644 --- a/infrastructure/network/netadapter/server/grpcserver/protowire/message_reject.go +++ b/infrastructure/network/netadapter/server/grpcserver/protowire/message_reject.go @@ -4,13 +4,13 @@ import ( "github.com/kaspanet/kaspad/app/appmessage" ) -func (x *KaspadMessage_Reject) toWireMessage() (appmessage.Message, error) { +func (x *KaspadMessage_Reject) toAppMessage() (appmessage.Message, error) { return &appmessage.MsgReject{ Reason: x.Reject.Reason, }, nil } -func (x *KaspadMessage_Reject) fromWireMessage(msgReject *appmessage.MsgReject) error { +func (x *KaspadMessage_Reject) fromAppMessage(msgReject *appmessage.MsgReject) error { x.Reject = &RejectMessage{ Reason: msgReject.Reason, } diff --git a/infrastructure/network/netadapter/server/grpcserver/protowire/wire.go b/infrastructure/network/netadapter/server/grpcserver/protowire/wire.go index a73bc0a1d..04d485f55 100644 --- a/infrastructure/network/netadapter/server/grpcserver/protowire/wire.go +++ b/infrastructure/network/netadapter/server/grpcserver/protowire/wire.go @@ -174,6 +174,13 @@ func toPayload(message appmessage.Message) (isKaspadMessage_Payload, error) { return nil, err } return payload, nil + case *appmessage.MsgReject: + payload := new(KaspadMessage_Reject) + err := payload.fromAppMessage(message) + if err != nil { + return nil, err + } + return payload, nil default: return nil, errors.Errorf("unknown message type %T", message) }