2017-11-02 19:37:07 +00:00
|
|
|
package geth
|
2017-10-23 15:56:29 +00:00
|
|
|
|
2017-10-23 18:58:33 +00:00
|
|
|
import (
|
|
|
|
"fmt"
|
2017-10-25 22:24:38 +00:00
|
|
|
|
2017-11-06 20:36:12 +00:00
|
|
|
"math/big"
|
|
|
|
|
2017-11-06 18:53:43 +00:00
|
|
|
"github.com/8thlight/vulcanizedb/pkg/core"
|
2017-11-03 13:54:32 +00:00
|
|
|
"github.com/ethereum/go-ethereum"
|
2017-10-23 18:58:33 +00:00
|
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
|
|
"github.com/ethereum/go-ethereum/ethclient"
|
|
|
|
"golang.org/x/net/context"
|
|
|
|
)
|
2017-10-23 15:56:29 +00:00
|
|
|
|
2017-10-23 18:58:33 +00:00
|
|
|
type GethBlockchain struct {
|
2017-11-03 13:54:32 +00:00
|
|
|
client *ethclient.Client
|
|
|
|
readGethHeaders chan *types.Header
|
|
|
|
outputBlocks chan core.Block
|
|
|
|
newHeadSubscription ethereum.Subscription
|
2017-10-23 18:58:33 +00:00
|
|
|
}
|
2017-10-23 15:56:29 +00:00
|
|
|
|
2017-11-06 20:36:12 +00:00
|
|
|
func (blockchain *GethBlockchain) GetBlockByNumber(blockNumber int64) core.Block {
|
|
|
|
gethBlock, _ := blockchain.client.BlockByNumber(context.Background(), big.NewInt(blockNumber))
|
2017-11-08 20:55:35 +00:00
|
|
|
return GethBlockToCoreBlock(gethBlock, blockchain.client)
|
2017-11-06 20:36:12 +00:00
|
|
|
}
|
|
|
|
|
2017-10-23 19:33:08 +00:00
|
|
|
func NewGethBlockchain(ipcPath string) *GethBlockchain {
|
2017-10-23 18:58:33 +00:00
|
|
|
blockchain := GethBlockchain{}
|
2017-10-23 19:33:08 +00:00
|
|
|
client, _ := ethclient.Dial(ipcPath)
|
2017-10-23 18:58:33 +00:00
|
|
|
blockchain.client = client
|
|
|
|
return &blockchain
|
|
|
|
}
|
2017-10-24 14:24:07 +00:00
|
|
|
|
2017-11-02 19:37:07 +00:00
|
|
|
func (blockchain *GethBlockchain) SubscribeToBlocks(blocks chan core.Block) {
|
2017-11-02 11:41:24 +00:00
|
|
|
blockchain.outputBlocks = blocks
|
|
|
|
fmt.Println("SubscribeToBlocks")
|
|
|
|
inputHeaders := make(chan *types.Header, 10)
|
|
|
|
myContext := context.Background()
|
|
|
|
blockchain.readGethHeaders = inputHeaders
|
2017-11-03 13:54:32 +00:00
|
|
|
subscription, _ := blockchain.client.SubscribeNewHead(myContext, inputHeaders)
|
|
|
|
blockchain.newHeadSubscription = subscription
|
2017-10-23 18:58:33 +00:00
|
|
|
}
|
2017-10-23 15:56:29 +00:00
|
|
|
|
2017-11-02 11:41:24 +00:00
|
|
|
func (blockchain *GethBlockchain) StartListening() {
|
|
|
|
for header := range blockchain.readGethHeaders {
|
2017-11-06 20:36:12 +00:00
|
|
|
block := blockchain.GetBlockByNumber(header.Number.Int64())
|
2017-11-02 11:41:24 +00:00
|
|
|
blockchain.outputBlocks <- block
|
2017-10-23 18:58:33 +00:00
|
|
|
}
|
2017-10-23 15:56:29 +00:00
|
|
|
}
|
2017-11-03 13:54:32 +00:00
|
|
|
|
|
|
|
func (blockchain *GethBlockchain) StopListening() {
|
|
|
|
blockchain.newHeadSubscription.Unsubscribe()
|
|
|
|
}
|