use correct type in wrap

This commit is contained in:
William Banfield
2022-10-20 16:59:06 -04:00
parent 24c1f44085
commit 8015f6e254
2 changed files with 5 additions and 3 deletions

View File

@@ -27,6 +27,8 @@ type Unwrapper interface {
}
type Wrapper interface {
proto.Message
// Wrap will take the underlying message and wrap it in its wrapper type.
Wrap() (proto.Message, error)
}

View File

@@ -7,12 +7,12 @@ import (
"github.com/tendermint/tendermint/p2p"
)
var _ p2p.Wrapper = &Message_Txs{}
var _ p2p.Wrapper = &Txs{}
// Wrap implements the p2p Wrapper interface and wraps a mempool message.
func (m *Message_Txs) Wrap() (proto.Message, error) {
func (m *Txs) Wrap() (proto.Message, error) {
return &Message{
Sum: m,
Sum: &Message_Txs{Txs: m},
}, nil
}