82 lines
2.1 KiB
Go
82 lines
2.1 KiB
Go
package geth
|
|
|
|
import (
|
|
"math/big"
|
|
|
|
"strings"
|
|
|
|
"log"
|
|
|
|
"github.com/8thlight/vulcanizedb/pkg/core"
|
|
"github.com/8thlight/vulcanizedb/pkg/geth/node"
|
|
"github.com/ethereum/go-ethereum"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/ethereum/go-ethereum/ethclient"
|
|
"github.com/ethereum/go-ethereum/rpc"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
type Blockchain struct {
|
|
client *ethclient.Client
|
|
readGethHeaders chan *types.Header
|
|
outputBlocks chan core.Block
|
|
newHeadSubscription ethereum.Subscription
|
|
node core.Node
|
|
}
|
|
|
|
func NewBlockchain(ipcPath string) *Blockchain {
|
|
blockchain := Blockchain{}
|
|
rpcClient, err := rpc.Dial(ipcPath)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
client := ethclient.NewClient(rpcClient)
|
|
blockchain.node = node.Info(rpcClient)
|
|
if infura := isInfuraNode(ipcPath); infura {
|
|
blockchain.node.Id = "infura"
|
|
blockchain.node.ClientName = "infura"
|
|
}
|
|
blockchain.client = client
|
|
return &blockchain
|
|
}
|
|
|
|
func isInfuraNode(ipcPath string) bool {
|
|
if strings.Contains(ipcPath, "infura") {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (blockchain *Blockchain) GetLogs(contract core.Contract, startingBlockNumber *big.Int, endingBlockNumber *big.Int) ([]core.Log, error) {
|
|
if endingBlockNumber == nil {
|
|
endingBlockNumber = startingBlockNumber
|
|
}
|
|
contractAddress := common.HexToAddress(contract.Hash)
|
|
fc := ethereum.FilterQuery{
|
|
FromBlock: startingBlockNumber,
|
|
ToBlock: endingBlockNumber,
|
|
Addresses: []common.Address{contractAddress},
|
|
}
|
|
gethLogs, err := blockchain.client.FilterLogs(context.Background(), fc)
|
|
if err != nil {
|
|
return []core.Log{}, err
|
|
}
|
|
logs := ToCoreLogs(gethLogs)
|
|
return logs, nil
|
|
}
|
|
|
|
func (blockchain *Blockchain) Node() core.Node {
|
|
return blockchain.node
|
|
}
|
|
|
|
func (blockchain *Blockchain) GetBlockByNumber(blockNumber int64) core.Block {
|
|
gethBlock, _ := blockchain.client.BlockByNumber(context.Background(), big.NewInt(blockNumber))
|
|
return ToCoreBlock(gethBlock, blockchain.client)
|
|
}
|
|
|
|
func (blockchain *Blockchain) LastBlock() *big.Int {
|
|
block, _ := blockchain.client.HeaderByNumber(context.Background(), nil)
|
|
return block.Number
|
|
}
|