From 5c108c0e3707d7334791ca0d81f979a2411ea667 Mon Sep 17 00:00:00 2001 From: Ian Norden <iannordenn@gmail.com> Date: Wed, 11 Nov 2020 16:42:56 -0600 Subject: [PATCH] fix API registration --- statediff/service.go | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/statediff/service.go b/statediff/service.go index 718d9c797eda..bab80456dc79 100644 --- a/statediff/service.go +++ b/statediff/service.go @@ -67,8 +67,10 @@ type blockChain interface { // IService is the state-diffing service interface type IService interface { - // APIs(), Protocols(), Start() and Stop() + // Start() and Stop() node.Lifecycle + // Method to getting API(s) for this service + APIs() []rpc.API // Main event loop for processing state diffs Loop(chainEventCh chan core.ChainEvent) // Method to subscribe to receive state diff processing output @@ -138,7 +140,7 @@ func New(stack *node.Node, ethServ *eth.Ethereum, dbParams *DBParams, enableWrit indexer = ind.NewStateDiffIndexer(blockChain.Config(), db) } prom.Init() - serv := &Service{ + sds := &Service{ Mutex: sync.Mutex{}, BlockChain: blockChain, Builder: NewBuilder(blockChain.StateCache()), @@ -148,7 +150,8 @@ func New(stack *node.Node, ethServ *eth.Ethereum, dbParams *DBParams, enableWrit indexer: indexer, enableWriteLoop: enableWriteLoop, } - stack.RegisterLifecycle(serv) + stack.RegisterLifecycle(sds) + stack.RegisterAPIs(sds.APIs()) return nil }