diff --git a/p2p/types.go b/p2p/types.go index e8182a6f3..70a5ee536 100644 --- a/p2p/types.go +++ b/p2p/types.go @@ -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) } diff --git a/proto/tendermint/mempool/message.go b/proto/tendermint/mempool/message.go index 4686aa6f2..0e9d797f8 100644 --- a/proto/tendermint/mempool/message.go +++ b/proto/tendermint/mempool/message.go @@ -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 }