@@ -15,7 +15,6 @@ import (
1515 "github.com/smartcontractkit/chainlink-common/pkg/loop"
1616 "github.com/smartcontractkit/chainlink-common/pkg/services"
1717 "github.com/smartcontractkit/chainlink-common/pkg/types"
18- "github.com/smartcontractkit/chainlink-common/pkg/utils"
1918
2019 "github.com/smartcontractkit/chainlink-starknet/relayer/pkg/chainlink/config"
2120 "github.com/smartcontractkit/chainlink-starknet/relayer/pkg/chainlink/db"
@@ -59,7 +58,7 @@ func (o *ChainOpts) Validate() (err error) {
5958var _ Chain = (* chain )(nil )
6059
6160type chain struct {
62- utils. StartStopOnce
61+ services. StateMachine
6362 id string
6463 cfg * config.TOMLConfig
6564 lggr logger.Logger
@@ -78,23 +77,16 @@ func NewChain(cfg *config.TOMLConfig, opts ChainOpts) (Chain, error) {
7877}
7978
8079func newChain (id string , cfg * config.TOMLConfig , loopKs loop.Keystore , lggr logger.Logger ) (* chain , error ) {
80+ lggr = logger .Named (lggr , "Chain" )
8181 lggr = logger .With (lggr , "starknetChainID" , id )
8282 ch := & chain {
8383 id : id ,
8484 cfg : cfg ,
85- lggr : logger .Named (lggr , "Chain" ),
86- }
87-
88- getClient := func () (* starknet.Client , error ) {
89- return ch .getClient ()
90- }
91-
92- getFeederClient := func () (* starknet.FeederClient , error ) {
93- return ch .getFeederClient (), nil
85+ lggr : lggr ,
9486 }
9587
9688 var err error
97- ch .txm , err = txm .New (lggr , loopKs , cfg , getClient , getFeederClient )
89+ ch .txm , err = txm .New (lggr , loopKs , cfg , ch . getClient , ch . getFeederClient )
9890 if err != nil {
9991 return nil , err
10092 }
@@ -122,8 +114,8 @@ func (c *chain) ChainID() string {
122114 return c .id
123115}
124116
125- func (c * chain ) getFeederClient () * starknet.FeederClient {
126- return starknet .NewFeederClient (c .cfg .FeederURL .String ())
117+ func (c * chain ) getFeederClient () ( * starknet.FeederClient , error ) {
118+ return starknet .NewFeederClient (c .cfg .FeederURL .String ()), nil
127119}
128120
129121// getClient returns a client, randomly selecting one from available and valid nodes
@@ -172,10 +164,6 @@ func (c *chain) Close() error {
172164 })
173165}
174166
175- func (c * chain ) Ready () error {
176- return c .StartStopOnce .Ready ()
177- }
178-
179167func (c * chain ) HealthReport () map [string ]error {
180168 report := map [string ]error {c .Name (): c .Healthy ()}
181169 services .CopyHealth (report , c .txm .HealthReport ())
0 commit comments