From 24c1f44085953d9b4d47b091c69e35d4a51376a7 Mon Sep 17 00:00:00 2001 From: William Banfield Date: Thu, 20 Oct 2022 16:51:12 -0400 Subject: [PATCH] mempool implements wrap / unwrap --- proto/tendermint/mempool/message.go | 19 ++++++++----------- 1 file changed, 8 insertions(+), 11 deletions(-) diff --git a/proto/tendermint/mempool/message.go b/proto/tendermint/mempool/message.go index a3e249f99..4686aa6f2 100644 --- a/proto/tendermint/mempool/message.go +++ b/proto/tendermint/mempool/message.go @@ -3,20 +3,17 @@ package mempool import ( "fmt" - "github.com/gogo/protobuf/proto" + "github.com/cosmos/gogoproto/proto" + "github.com/tendermint/tendermint/p2p" ) +var _ p2p.Wrapper = &Message_Txs{} + // Wrap implements the p2p Wrapper interface and wraps a mempool message. -func (m *Message) Wrap(pb proto.Message) error { - switch msg := pb.(type) { - case *Txs: - m.Sum = &Message_Txs{Txs: msg} - - default: - return fmt.Errorf("unknown message: %T", msg) - } - - return nil +func (m *Message_Txs) Wrap() (proto.Message, error) { + return &Message{ + Sum: m, + }, nil } // Unwrap implements the p2p Wrapper interface and unwraps a wrapped mempool