You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
2.0 KiB
86 lines
2.0 KiB
package node
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/gob"
|
|
"harmony-benchmark/blockchain"
|
|
"harmony-benchmark/proto"
|
|
)
|
|
|
|
// The specific types of message under NODE category
|
|
type NodeMessageType byte
|
|
|
|
const (
|
|
TRANSACTION NodeMessageType = iota
|
|
BLOCK
|
|
CONTROL
|
|
// TODO: add more types
|
|
)
|
|
|
|
// The types of messages used for NODE/TRANSACTION
|
|
type TransactionMessageType int
|
|
|
|
const (
|
|
SEND TransactionMessageType = iota
|
|
REQUEST
|
|
UNLOCK
|
|
)
|
|
|
|
// The types of messages used for NODE/BLOCK
|
|
type BlockMessageType int
|
|
|
|
const (
|
|
SYNC BlockMessageType = iota
|
|
)
|
|
|
|
// The types of messages used for NODE/CONTROL
|
|
type ControlMessageType int
|
|
|
|
const (
|
|
STOP ControlMessageType = iota
|
|
)
|
|
|
|
// Constructs serialized transactions
|
|
func ConstructTransactionListMessage(transactions []*blockchain.Transaction) []byte {
|
|
byteBuffer := bytes.NewBuffer([]byte{byte(proto.NODE)})
|
|
byteBuffer.WriteByte(byte(TRANSACTION))
|
|
byteBuffer.WriteByte(byte(SEND))
|
|
encoder := gob.NewEncoder(byteBuffer)
|
|
// Copy over the tx data
|
|
txs := make([]blockchain.Transaction, len(transactions))
|
|
for i := range txs {
|
|
txs[i] = *transactions[i]
|
|
}
|
|
encoder.Encode(txs)
|
|
return byteBuffer.Bytes()
|
|
}
|
|
|
|
// Constructs serialized transactions
|
|
func ConstructRequestTransactionsMessage(transactionIds [][]byte) []byte {
|
|
byteBuffer := bytes.NewBuffer([]byte{byte(proto.NODE)})
|
|
byteBuffer.WriteByte(byte(TRANSACTION))
|
|
byteBuffer.WriteByte(byte(REQUEST))
|
|
for _, txId := range transactionIds {
|
|
byteBuffer.Write(txId)
|
|
}
|
|
return byteBuffer.Bytes()
|
|
}
|
|
|
|
// Constructs STOP message for node to stop
|
|
func ConstructStopMessage() []byte {
|
|
byteBuffer := bytes.NewBuffer([]byte{byte(proto.NODE)})
|
|
byteBuffer.WriteByte(byte(CONTROL))
|
|
byteBuffer.WriteByte(byte(STOP))
|
|
return byteBuffer.Bytes()
|
|
}
|
|
|
|
// Constructs blocks sync message to send blocks to other nodes
|
|
func ConstructBlocksSyncMessage(blocks []blockchain.Block) []byte {
|
|
byteBuffer := bytes.NewBuffer([]byte{byte(proto.NODE)})
|
|
byteBuffer.WriteByte(byte(BLOCK))
|
|
byteBuffer.WriteByte(byte(SYNC))
|
|
encoder := gob.NewEncoder(byteBuffer)
|
|
|
|
encoder.Encode(blocks)
|
|
return byteBuffer.Bytes()
|
|
}
|
|
|