woop/hmy/backend.go

package hmy

import (
	"math/big"
	"sync"

	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/core/bloombits"
	"github.com/ethereum/go-ethereum/ethdb"
	"github.com/ethereum/go-ethereum/event"
	"github.com/harmony-one/harmony/core"
	"github.com/harmony-one/harmony/core/types"
	"github.com/harmony-one/harmony/internal/params"
	staking "github.com/harmony-one/harmony/staking/types"
)

// Harmony implements the Harmony full node service.
type Harmony struct {
	// Channel for shutting down the service
	shutdownChan  chan bool                      // Channel for shutting down the Harmony
	bloomRequests chan chan *bloombits.Retrieval // Channel receiving bloom data retrieval requests
	blockchain    *core.BlockChain
	beaconchain   *core.BlockChain
	txPool        *core.TxPool
	cxPool        *core.CxPool
	eventMux      *event.TypeMux
	// DB interfaces
	chainDb      ethdb.Database     // Block chain database
	bloomIndexer *core.ChainIndexer // Bloom indexer operating during block imports
	APIBackend   *APIBackend
	nodeAPI      NodeAPI
	// aka network version, which is used to identify which network we are using
	networkID uint64
	// RPCGasCap is the global gas cap for eth-call variants.
	RPCGasCap *big.Int `toml:",omitempty"`
	shardID   uint32
}

// NodeAPI is the list of functions from node used to call rpc apis.
type NodeAPI interface {
	AddPendingStakingTransaction(*staking.StakingTransaction) error
	AddPendingTransaction(newTx *types.Transaction) error
	Blockchain() *core.BlockChain
	Beaconchain() *core.BlockChain
	GetBalanceOfAddress(address common.Address) (*big.Int, error)
	GetNonceOfAddress(address common.Address) uint64
	GetTransactionsHistory(address, txType, order string) ([]common.Hash, error)
	GetStakingTransactionsHistory(address, txType, order string) ([]common.Hash, error)
	IsCurrentlyLeader() bool
	ErroredStakingTransactionSink() []staking.RPCTransactionError
	ErroredTransactionSink() []types.RPCTransactionError
	PendingCXReceipts() []*types.CXReceiptsProof
	GetNodeBootTime() int64
}

// New creates a new Harmony object (including the
// initialisation of the common Harmony object)
func New(
	nodeAPI NodeAPI, txPool *core.TxPool,
	cxPool *core.CxPool, eventMux *event.TypeMux, shardID uint32,
) (*Harmony, error) {
	chainDb := nodeAPI.Blockchain().ChainDB()
	hmy := &Harmony{
		shutdownChan:  make(chan bool),
		bloomRequests: make(chan chan *bloombits.Retrieval),
		blockchain:    nodeAPI.Blockchain(),
		beaconchain:   nodeAPI.Beaconchain(),
		txPool:        txPool,
		cxPool:        cxPool,
		eventMux:      eventMux,
		chainDb:       chainDb,
		bloomIndexer:  NewBloomIndexer(chainDb, params.BloomBitsBlocks, params.BloomConfirms),
		nodeAPI:       nodeAPI,
		networkID:     1, // TODO(ricl): this should be from config
		shardID:       shardID,
	}
	hmy.APIBackend = &APIBackend{hmy: hmy,
		TotalStakingCache: struct {
			sync.Mutex
			BlockHeight  int64
			TotalStaking *big.Int
		}{
			BlockHeight:  -1,
			TotalStaking: big.NewInt(0),
		},
	}
	return hmy, nil
}

// TxPool ...
func (s *Harmony) TxPool() *core.TxPool { return s.txPool }

// CxPool is used to store the blockHashes, where the corresponding block contains the cross shard receipts to be sent
func (s *Harmony) CxPool() *core.CxPool { return s.cxPool }

// BlockChain ...
func (s *Harmony) BlockChain() *core.BlockChain { return s.blockchain }

//BeaconChain ...
func (s *Harmony) BeaconChain() *core.BlockChain { return s.beaconchain }

// NetVersion returns the network version, i.e. network ID identifying which network we are using
func (s *Harmony) NetVersion() uint64 { return s.networkID }