Mirroristas/kaspad
2
0
Fork 0
mirror of https://github.com/kaspanet/kaspad.git synced 2025-10-14 00:59:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NOD-1301] Add MsgReject to protowire mapping (#891)

Browse Source
This commit is contained in:
Ori Newman 2020-08-23 18:29:41 +03:00 committed by GitHub
parent 15b545ee2b
commit a6b8eea369
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
infrastructure/network/netadapter/server/grpcserver/protowire/message_reject.go
View File

@ -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,
}

7
infrastructure/network/netadapter/server/grpcserver/protowire/wire.go
View File

@ -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)
}