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.
57 lines
1.8 KiB
57 lines
1.8 KiB
package client
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/gob"
|
|
|
|
"github.com/simple-rules/harmony-benchmark/blockchain"
|
|
"github.com/simple-rules/harmony-benchmark/proto"
|
|
"github.com/simple-rules/harmony-benchmark/proto/node"
|
|
)
|
|
|
|
// The specific types of message under CLIENT category
|
|
type ClientMessageType byte
|
|
|
|
const (
|
|
TRANSACTION ClientMessageType = iota
|
|
// TODO: add more types
|
|
)
|
|
|
|
// The types of messages used for CLIENT/TRANSACTION
|
|
type TransactionMessageType int
|
|
|
|
const (
|
|
PROOF_OF_LOCK TransactionMessageType = iota // The proof of accept or reject returned by the leader to the client tnat issued cross shard transactions.
|
|
)
|
|
|
|
// [leader] Constructs the proof of accept or reject message that will be sent to client
|
|
func ConstructProofOfAcceptOrRejectMessage(proofs []blockchain.CrossShardTxProof) []byte {
|
|
byteBuffer := bytes.NewBuffer([]byte{byte(proto.CLIENT)})
|
|
byteBuffer.WriteByte(byte(TRANSACTION))
|
|
byteBuffer.WriteByte(byte(PROOF_OF_LOCK))
|
|
encoder := gob.NewEncoder(byteBuffer)
|
|
|
|
encoder.Encode(proofs)
|
|
return byteBuffer.Bytes()
|
|
}
|
|
|
|
// [client] Constructs the unlock to commit or abort message that will be sent to leaders
|
|
func ConstructUnlockToCommitOrAbortMessage(txsAndProofs []blockchain.Transaction) []byte {
|
|
byteBuffer := bytes.NewBuffer([]byte{byte(proto.NODE)})
|
|
byteBuffer.WriteByte(byte(node.TRANSACTION))
|
|
byteBuffer.WriteByte(byte(node.UNLOCK))
|
|
encoder := gob.NewEncoder(byteBuffer)
|
|
encoder.Encode(txsAndProofs)
|
|
return byteBuffer.Bytes()
|
|
}
|
|
|
|
// [client] Constructs the fetch utxo message that will be sent to Harmony network
|
|
func ConstructFetchUtxoMessage(addresses [][20]byte) []byte {
|
|
byteBuffer := bytes.NewBuffer([]byte{byte(proto.NODE)})
|
|
byteBuffer.WriteByte(byte(node.TRANSACTION))
|
|
byteBuffer.WriteByte(byte(node.UNLOCK))
|
|
for _, address := range addresses {
|
|
byteBuffer.Write(address[:])
|
|
}
|
|
return byteBuffer.Bytes()
|
|
}
|
|
|