mirror of https://github.com/poanetwork/gecko.git
133 lines
3.9 KiB
Go
133 lines
3.9 KiB
Go
// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package network
|
|
|
|
import (
|
|
"github.com/ava-labs/gecko/ids"
|
|
"github.com/ava-labs/gecko/utils"
|
|
)
|
|
|
|
// Builder extends a Codec to build messages safely
|
|
type Builder struct{ Codec }
|
|
|
|
// GetVersion message
|
|
func (m Builder) GetVersion() (Msg, error) { return m.Pack(GetVersion, nil) }
|
|
|
|
// Version message
|
|
func (m Builder) Version(networkID, nodeID uint32, myTime uint64, ip utils.IPDesc, myVersion string) (Msg, error) {
|
|
return m.Pack(Version, map[Field]interface{}{
|
|
NetworkID: networkID,
|
|
NodeID: nodeID,
|
|
MyTime: myTime,
|
|
IP: ip,
|
|
VersionStr: myVersion,
|
|
})
|
|
}
|
|
|
|
// GetPeerList message
|
|
func (m Builder) GetPeerList() (Msg, error) { return m.Pack(GetPeerList, nil) }
|
|
|
|
// PeerList message
|
|
func (m Builder) PeerList(ipDescs []utils.IPDesc) (Msg, error) {
|
|
return m.Pack(PeerList, map[Field]interface{}{Peers: ipDescs})
|
|
}
|
|
|
|
// GetAcceptedFrontier message
|
|
func (m Builder) GetAcceptedFrontier(chainID ids.ID, requestID uint32) (Msg, error) {
|
|
return m.Pack(GetAcceptedFrontier, map[Field]interface{}{
|
|
ChainID: chainID.Bytes(),
|
|
RequestID: requestID,
|
|
})
|
|
}
|
|
|
|
// AcceptedFrontier message
|
|
func (m Builder) AcceptedFrontier(chainID ids.ID, requestID uint32, containerIDs ids.Set) (Msg, error) {
|
|
containerIDBytes := make([][]byte, containerIDs.Len())
|
|
for i, containerID := range containerIDs.List() {
|
|
containerIDBytes[i] = containerID.Bytes()
|
|
}
|
|
return m.Pack(AcceptedFrontier, map[Field]interface{}{
|
|
ChainID: chainID.Bytes(),
|
|
RequestID: requestID,
|
|
ContainerIDs: containerIDBytes,
|
|
})
|
|
}
|
|
|
|
// GetAccepted message
|
|
func (m Builder) GetAccepted(chainID ids.ID, requestID uint32, containerIDs ids.Set) (Msg, error) {
|
|
containerIDBytes := make([][]byte, containerIDs.Len())
|
|
for i, containerID := range containerIDs.List() {
|
|
containerIDBytes[i] = containerID.Bytes()
|
|
}
|
|
return m.Pack(GetAccepted, map[Field]interface{}{
|
|
ChainID: chainID.Bytes(),
|
|
RequestID: requestID,
|
|
ContainerIDs: containerIDBytes,
|
|
})
|
|
}
|
|
|
|
// Accepted message
|
|
func (m Builder) Accepted(chainID ids.ID, requestID uint32, containerIDs ids.Set) (Msg, error) {
|
|
containerIDBytes := make([][]byte, containerIDs.Len())
|
|
for i, containerID := range containerIDs.List() {
|
|
containerIDBytes[i] = containerID.Bytes()
|
|
}
|
|
return m.Pack(Accepted, map[Field]interface{}{
|
|
ChainID: chainID.Bytes(),
|
|
RequestID: requestID,
|
|
ContainerIDs: containerIDBytes,
|
|
})
|
|
}
|
|
|
|
// Get message
|
|
func (m Builder) Get(chainID ids.ID, requestID uint32, containerID ids.ID) (Msg, error) {
|
|
return m.Pack(Get, map[Field]interface{}{
|
|
ChainID: chainID.Bytes(),
|
|
RequestID: requestID,
|
|
ContainerID: containerID.Bytes(),
|
|
})
|
|
}
|
|
|
|
// Put message
|
|
func (m Builder) Put(chainID ids.ID, requestID uint32, containerID ids.ID, container []byte) (Msg, error) {
|
|
return m.Pack(Put, map[Field]interface{}{
|
|
ChainID: chainID.Bytes(),
|
|
RequestID: requestID,
|
|
ContainerID: containerID.Bytes(),
|
|
ContainerBytes: container,
|
|
})
|
|
}
|
|
|
|
// PushQuery message
|
|
func (m Builder) PushQuery(chainID ids.ID, requestID uint32, containerID ids.ID, container []byte) (Msg, error) {
|
|
return m.Pack(PushQuery, map[Field]interface{}{
|
|
ChainID: chainID.Bytes(),
|
|
RequestID: requestID,
|
|
ContainerID: containerID.Bytes(),
|
|
ContainerBytes: container,
|
|
})
|
|
}
|
|
|
|
// PullQuery message
|
|
func (m Builder) PullQuery(chainID ids.ID, requestID uint32, containerID ids.ID) (Msg, error) {
|
|
return m.Pack(PullQuery, map[Field]interface{}{
|
|
ChainID: chainID.Bytes(),
|
|
RequestID: requestID,
|
|
ContainerID: containerID.Bytes(),
|
|
})
|
|
}
|
|
|
|
// Chits message
|
|
func (m Builder) Chits(chainID ids.ID, requestID uint32, containerIDs ids.Set) (Msg, error) {
|
|
containerIDBytes := make([][]byte, containerIDs.Len())
|
|
for i, containerID := range containerIDs.List() {
|
|
containerIDBytes[i] = containerID.Bytes()
|
|
}
|
|
return m.Pack(Chits, map[Field]interface{}{
|
|
ChainID: chainID.Bytes(),
|
|
RequestID: requestID,
|
|
ContainerIDs: containerIDBytes,
|
|
})
|
|
}
|