mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-31 23:41:22 +00:00
2a612de3c5
Former-commit-id: ff13be57aab24129a97f359464fb5b8154c5a206 [formerly ff9566d3bd5653b23fc2846b26e1b07a15fdc159] Former-commit-id: 371821f0ad0bb76640eed2199f5dc49f979d14d8
106 lines
3.5 KiB
Go
106 lines
3.5 KiB
Go
package sharding
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/ethereum/go-ethereum/accounts/abi/bind"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/ethereum/go-ethereum/log"
|
|
"math/big"
|
|
)
|
|
|
|
// SubscribeBlockHeaders checks incoming block headers and determines if
|
|
// we are an eligible proposer for collations. Then, it finds the pending tx's
|
|
// from the running geth node and sorts them by descending order of gas price,
|
|
// eliminates those that ask for too much gas, and routes them over
|
|
// to the VMC to create a collation
|
|
func subscribeBlockHeaders(c *Client) error {
|
|
headerChan := make(chan *types.Header, 16)
|
|
|
|
_, err := c.client.SubscribeNewHead(context.Background(), headerChan)
|
|
if err != nil {
|
|
return fmt.Errorf("unable to subscribe to incoming headers. %v", err)
|
|
}
|
|
|
|
log.Info("Listening for new headers...")
|
|
|
|
for {
|
|
// TODO: Error handling for getting disconnected from the client
|
|
select {
|
|
case head := <-headerChan:
|
|
// Query the current state to see if we are an eligible proposer
|
|
log.Info(fmt.Sprintf("Received new header: %v", head.Number.String()))
|
|
// TODO: Only run this code on certain periods?
|
|
if err := checkShardsForProposal(c, head); err != nil {
|
|
return fmt.Errorf("unable to watch shards. %v", err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// checkShardsForProposal checks if we are an eligible proposer for
|
|
// collation for the available shards in the VMC. The function calls
|
|
// getEligibleProposer from the VMC and proposes a collation if
|
|
// conditions are met
|
|
func checkShardsForProposal(c *Client, head *types.Header) error {
|
|
|
|
accounts := c.keystore.Accounts()
|
|
if len(accounts) == 0 {
|
|
return fmt.Errorf("no accounts found")
|
|
}
|
|
|
|
if err := c.unlockAccount(accounts[0]); err != nil {
|
|
return fmt.Errorf("cannot unlock account. %v", err)
|
|
}
|
|
|
|
log.Info("Checking if we are an eligible collation proposer for a shard...")
|
|
for s := int64(0); s < shardCount; s++ {
|
|
// Checks if we are an eligible proposer according to the VMC
|
|
period := head.Number.Div(head.Number, big.NewInt(periodLength))
|
|
addr, err := c.vmc.VMCCaller.GetEligibleProposer(&bind.CallOpts{}, big.NewInt(s), period)
|
|
// TODO: When we are not a proposer, we get the error of being unable to
|
|
// unmarshal empty output. Open issue to deal with this.
|
|
|
|
// If output is non-empty and the addr == coinbase
|
|
if err == nil && addr == accounts[0].Address {
|
|
log.Info(fmt.Sprintf("Selected as collator on shard: %d", s))
|
|
err := proposeCollation(s)
|
|
if err != nil {
|
|
return fmt.Errorf("could not propose collation. %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// proposeCollation interacts with the VMC directly to add a collation header
|
|
func proposeCollation(shardID int64) error {
|
|
// TODO: Adds a collation header to the VMC with the following fields:
|
|
// [
|
|
// shard_id: uint256,
|
|
// expected_period_number: uint256,
|
|
// period_start_prevhash: bytes32,
|
|
// parent_hash: bytes32,
|
|
// transactions_root: bytes32,
|
|
// coinbase: address,
|
|
// state_root: bytes32,
|
|
// receipts_root: bytes32,
|
|
// number: uint256,
|
|
// sig: bytes
|
|
// ]
|
|
//
|
|
// Before calling this, we would need to have access to the state of
|
|
// the period_start_prevhash. Refer to the comments in:
|
|
// https://github.com/ethereum/py-evm/issues/258#issuecomment-359879350
|
|
//
|
|
// This function will call FetchCandidateHead() of the VMC to obtain
|
|
// more necessary information.
|
|
//
|
|
// This functions will fetch the transactions in the txpool and and apply
|
|
// them to finish up the collation. It will then need to broadcast the
|
|
// collation to the main chain using JSON-RPC.
|
|
log.Info("Propose collation function called")
|
|
return nil
|
|
}
|