chao
5 years ago
parent
96cb8d095a
commit
0c020d8f6a
@ -0,0 +1,16 @@ |
|||||||
|
package v3 |
||||||
|
|
||||||
|
import ( |
||||||
|
"github.com/harmony-one/harmony/block/v2" |
||||||
|
) |
||||||
|
|
||||||
|
// Header v3 has the same structure as v2 header
|
||||||
|
// It is used to identify the body v3 which including staking txs
|
||||||
|
type Header struct { |
||||||
|
v2.Header |
||||||
|
} |
||||||
|
|
||||||
|
// NewHeader creates a new header object.
|
||||||
|
func NewHeader() *Header { |
||||||
|
return &Header{*v2.NewHeader()} |
||||||
|
} |
||||||
@ -0,0 +1,128 @@ |
|||||||
|
package types |
||||||
|
|
||||||
|
import ( |
||||||
|
"io" |
||||||
|
|
||||||
|
"github.com/ethereum/go-ethereum/rlp" |
||||||
|
|
||||||
|
"github.com/harmony-one/harmony/block" |
||||||
|
staking "github.com/harmony-one/harmony/staking/types" |
||||||
|
) |
||||||
|
|
||||||
|
// BodyV2 is the V2 block body
|
||||||
|
type BodyV2 struct { |
||||||
|
f bodyFieldsV2 |
||||||
|
} |
||||||
|
|
||||||
|
type bodyFieldsV2 struct { |
||||||
|
Transactions []*Transaction |
||||||
|
StakingTransactions []*staking.StakingTransaction |
||||||
|
Uncles []*block.Header |
||||||
|
IncomingReceipts CXReceiptsProofs |
||||||
|
} |
||||||
|
|
||||||
|
// Transactions returns the list of transactions.
|
||||||
|
//
|
||||||
|
// The returned list is a deep copy; the caller may do anything with it without
|
||||||
|
// affecting the original.
|
||||||
|
func (b *BodyV2) Transactions() (txs []*Transaction) { |
||||||
|
for _, tx := range b.f.Transactions { |
||||||
|
txs = append(txs, tx.Copy()) |
||||||
|
} |
||||||
|
return txs |
||||||
|
} |
||||||
|
|
||||||
|
// StakingTransactions returns the list of staking transactions.
|
||||||
|
// The returned list is a deep copy; the caller may do anything with it without
|
||||||
|
// affecting the original.
|
||||||
|
func (b *BodyV2) StakingTransactions() (txs []*staking.StakingTransaction) { |
||||||
|
for _, tx := range b.f.StakingTransactions { |
||||||
|
txs = append(txs, tx.Copy()) |
||||||
|
} |
||||||
|
return txs |
||||||
|
} |
||||||
|
|
||||||
|
// TransactionAt returns the transaction at the given index in this block.
|
||||||
|
// It returns nil if index is out of bounds.
|
||||||
|
func (b *BodyV2) TransactionAt(index int) *Transaction { |
||||||
|
if index < 0 || index >= len(b.f.Transactions) { |
||||||
|
return nil |
||||||
|
} |
||||||
|
return b.f.Transactions[index].Copy() |
||||||
|
} |
||||||
|
|
||||||
|
// CXReceiptAt returns the CXReceipt at given index in this block
|
||||||
|
// It returns nil if index is out of bounds
|
||||||
|
func (b *BodyV2) CXReceiptAt(index int) *CXReceipt { |
||||||
|
if index < 0 { |
||||||
|
return nil |
||||||
|
} |
||||||
|
for _, cxp := range b.f.IncomingReceipts { |
||||||
|
cxs := cxp.Receipts |
||||||
|
if index < len(cxs) { |
||||||
|
return cxs[index].Copy() |
||||||
|
} |
||||||
|
index -= len(cxs) |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
// SetTransactions sets the list of transactions with a deep copy of the given
|
||||||
|
// list.
|
||||||
|
func (b *BodyV2) SetTransactions(newTransactions []*Transaction) { |
||||||
|
var txs []*Transaction |
||||||
|
for _, tx := range newTransactions { |
||||||
|
txs = append(txs, tx.Copy()) |
||||||
|
} |
||||||
|
b.f.Transactions = txs |
||||||
|
} |
||||||
|
|
||||||
|
// SetStakingTransactions sets the list of staking transactions with a deep copy of the given
|
||||||
|
// list.
|
||||||
|
func (b *BodyV2) SetStakingTransactions(newStakingTransactions []*staking.StakingTransaction) { |
||||||
|
var txs []*staking.StakingTransaction |
||||||
|
for _, tx := range newStakingTransactions { |
||||||
|
txs = append(txs, tx.Copy()) |
||||||
|
} |
||||||
|
b.f.StakingTransactions = txs |
||||||
|
} |
||||||
|
|
||||||
|
// Uncles returns a deep copy of the list of uncle headers of this block.
|
||||||
|
func (b *BodyV2) Uncles() (uncles []*block.Header) { |
||||||
|
for _, uncle := range b.f.Uncles { |
||||||
|
uncles = append(uncles, CopyHeader(uncle)) |
||||||
|
} |
||||||
|
return uncles |
||||||
|
} |
||||||
|
|
||||||
|
// SetUncles sets the list of uncle headers with a deep copy of the given list.
|
||||||
|
func (b *BodyV2) SetUncles(newUncle []*block.Header) { |
||||||
|
var uncles []*block.Header |
||||||
|
for _, uncle := range newUncle { |
||||||
|
uncles = append(uncles, CopyHeader(uncle)) |
||||||
|
} |
||||||
|
b.f.Uncles = uncles |
||||||
|
} |
||||||
|
|
||||||
|
// IncomingReceipts returns a deep copy of the list of incoming cross-shard
|
||||||
|
// transaction receipts of this block.
|
||||||
|
func (b *BodyV2) IncomingReceipts() (incomingReceipts CXReceiptsProofs) { |
||||||
|
return b.f.IncomingReceipts.Copy() |
||||||
|
} |
||||||
|
|
||||||
|
// SetIncomingReceipts sets the list of incoming cross-shard transaction
|
||||||
|
// receipts of this block with a dep copy of the given list.
|
||||||
|
func (b *BodyV2) SetIncomingReceipts(newIncomingReceipts CXReceiptsProofs) { |
||||||
|
b.f.IncomingReceipts = newIncomingReceipts.Copy() |
||||||
|
} |
||||||
|
|
||||||
|
// EncodeRLP RLP-encodes the block body into the given writer.
|
||||||
|
func (b *BodyV2) EncodeRLP(w io.Writer) error { |
||||||
|
return rlp.Encode(w, &b.f) |
||||||
|
} |
||||||
|
|
||||||
|
// DecodeRLP RLP-decodes a block body from the given RLP stream into the
|
||||||
|
// receiver.
|
||||||
|
func (b *BodyV2) DecodeRLP(s *rlp.Stream) error { |
||||||
|
return s.Decode(&b.f) |
||||||
|
} |
||||||
Loading…
Reference in new issue