31 lines
631 B
Go
31 lines
631 B
Go
package mempool
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/tendermint/tendermint/p2p"
|
|
)
|
|
|
|
var _ p2p.Wrapper = &Txs{}
|
|
var _ p2p.Unwrapper = &Message{}
|
|
|
|
// Wrap implements the p2p Wrapper interface and wraps a mempool message.
|
|
func (m *Txs) Wrap() proto.Message {
|
|
mm := &Message{}
|
|
mm.Sum = &Message_Txs{Txs: m}
|
|
return mm
|
|
}
|
|
|
|
// Unwrap implements the p2p Wrapper interface and unwraps a wrapped mempool
|
|
// message.
|
|
func (m *Message) Unwrap() (proto.Message, error) {
|
|
switch msg := m.Sum.(type) {
|
|
case *Message_Txs:
|
|
return m.GetTxs(), nil
|
|
|
|
default:
|
|
return nil, fmt.Errorf("unknown message: %T", msg)
|
|
}
|
|
}
|