71 lines
1.8 KiB
Go
71 lines
1.8 KiB
Go
package service
|
|
|
|
import (
|
|
"github.com/ethereum/go-ethereum/core"
|
|
"github.com/ethereum/go-ethereum/ethdb"
|
|
"github.com/ethereum/go-ethereum/p2p"
|
|
"github.com/ethereum/go-ethereum/rpc"
|
|
"github.com/ethereum/go-ethereum/statediff"
|
|
b "github.com/ethereum/go-ethereum/statediff/builder"
|
|
e "github.com/ethereum/go-ethereum/statediff/extractor"
|
|
p "github.com/ethereum/go-ethereum/statediff/publisher"
|
|
|
|
"github.com/ethereum/go-ethereum/event"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
)
|
|
|
|
type BlockChain interface {
|
|
SubscribeChainEvent(ch chan<- core.ChainEvent) event.Subscription
|
|
GetBlockByHash(hash common.Hash) *types.Block
|
|
}
|
|
|
|
type StateDiffService struct {
|
|
Builder *b.Builder
|
|
Extractor e.Extractor
|
|
BlockChain BlockChain
|
|
}
|
|
|
|
func NewStateDiffService(db ethdb.Database, blockChain *core.BlockChain) (*StateDiffService, error) {
|
|
config := statediff.Config{}
|
|
builder := b.NewBuilder(db)
|
|
publisher, err := p.NewPublisher(config)
|
|
if err != nil {
|
|
return nil, nil
|
|
}
|
|
|
|
extractor, _ := e.NewExtractor(builder, publisher)
|
|
return &StateDiffService{
|
|
BlockChain: blockChain,
|
|
Extractor: extractor,
|
|
}, nil
|
|
}
|
|
|
|
func (StateDiffService) Protocols() []p2p.Protocol {
|
|
return []p2p.Protocol{}
|
|
}
|
|
|
|
func (StateDiffService) APIs() []rpc.API {
|
|
return []rpc.API{}
|
|
}
|
|
|
|
func (sds *StateDiffService) Loop(events chan core.ChainEvent) {
|
|
for elem := range events {
|
|
currentBlock := elem.Block
|
|
|
|
parentHash := currentBlock.ParentHash()
|
|
parentBlock := sds.BlockChain.GetBlockByHash(parentHash)
|
|
|
|
sds.Extractor.ExtractStateDiff(*parentBlock, *currentBlock)
|
|
}
|
|
}
|
|
|
|
func (sds *StateDiffService) Start(server *p2p.Server) error {
|
|
events := make(chan core.ChainEvent, 10)
|
|
sds.BlockChain.SubscribeChainEvent(events)
|
|
go sds.Loop(events)
|
|
return nil
|
|
}
|
|
|
|
func (StateDiffService) Stop() error { return nil }
|