59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package statesync
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/tendermint/tendermint/p2p"
|
|
)
|
|
|
|
var _ p2p.Wrapper = &ChunkRequest{}
|
|
var _ p2p.Wrapper = &ChunkResponse{}
|
|
var _ p2p.Wrapper = &SnapshotsRequest{}
|
|
var _ p2p.Wrapper = &SnapshotsResponse{}
|
|
|
|
func (m *SnapshotsResponse) Wrap() proto.Message {
|
|
sm := &Message{}
|
|
sm.Sum = &Message_SnapshotsResponse{SnapshotsResponse: m}
|
|
return sm
|
|
}
|
|
|
|
func (m *SnapshotsRequest) Wrap() proto.Message {
|
|
sm := &Message{}
|
|
sm.Sum = &Message_SnapshotsRequest{SnapshotsRequest: m}
|
|
return sm
|
|
}
|
|
|
|
func (m *ChunkResponse) Wrap() proto.Message {
|
|
sm := &Message{}
|
|
sm.Sum = &Message_ChunkResponse{ChunkResponse: m}
|
|
return sm
|
|
}
|
|
|
|
func (m *ChunkRequest) Wrap() proto.Message {
|
|
sm := &Message{}
|
|
sm.Sum = &Message_ChunkRequest{ChunkRequest: m}
|
|
return sm
|
|
}
|
|
|
|
// Unwrap implements the p2p Wrapper interface and unwraps a wrapped state sync
|
|
// proto message.
|
|
func (m *Message) Unwrap() (proto.Message, error) {
|
|
switch msg := m.Sum.(type) {
|
|
case *Message_ChunkRequest:
|
|
return m.GetChunkRequest(), nil
|
|
|
|
case *Message_ChunkResponse:
|
|
return m.GetChunkResponse(), nil
|
|
|
|
case *Message_SnapshotsRequest:
|
|
return m.GetSnapshotsRequest(), nil
|
|
|
|
case *Message_SnapshotsResponse:
|
|
return m.GetSnapshotsResponse(), nil
|
|
|
|
default:
|
|
return nil, fmt.Errorf("unknown message: %T", msg)
|
|
}
|
|
}
|