Skip to content

Commit

Permalink
fix(runtime): make environment provider module scoped (#19385)
Browse files Browse the repository at this point in the history
  • Loading branch information
julienrbrt authored Feb 9, 2024
1 parent cf9accc commit 6e8de00
Showing 1 changed file with 4 additions and 8 deletions.
12 changes: 4 additions & 8 deletions runtime/module.go
Original file line number Diff line number Diff line change
Expand Up @@ -67,14 +67,13 @@ func init() {
ProvideTransientStoreKey,
ProvideMemoryStoreKey,
ProvideGenesisTxHandler,
ProvideKVStoreService,
ProvideEnvironment,
ProvideMemoryStoreService,
ProvideTransientStoreService,
ProvideEventService,
ProvideBasicManager,
ProvideAppVersionModifier,
ProvideAddressCodec,
ProvideEnvironment,
),
appconfig.Invoke(SetupAppBuilder),
)
Expand Down Expand Up @@ -225,9 +224,10 @@ func ProvideGenesisTxHandler(appBuilder *AppBuilder) genesis.TxHandler {
return appBuilder.app
}

func ProvideKVStoreService(config *runtimev1alpha1.Module, key depinject.ModuleKey, app *AppBuilder) store.KVStoreService {
func ProvideEnvironment(config *runtimev1alpha1.Module, key depinject.ModuleKey, app *AppBuilder) (store.KVStoreService, appmodule.Environment) {
storeKey := ProvideKVStoreKey(config, key, app)
return kvStoreService{key: storeKey}
kvService := kvStoreService{key: storeKey}
return kvService, NewEnvironment(kvService)
}

func ProvideMemoryStoreService(key depinject.ModuleKey, app *AppBuilder) store.MemoryStoreService {
Expand All @@ -252,10 +252,6 @@ func ProvideAppVersionModifier(app *AppBuilder) baseapp.AppVersionModifier {
return app.app
}

func ProvideEnvironment(kvService store.KVStoreService) appmodule.Environment {
return NewEnvironment(kvService)
}

type (
// ValidatorAddressCodec is an alias for address.Codec for validator addresses.
ValidatorAddressCodec address.Codec
Expand Down

0 comments on commit 6e8de00

Please sign in to comment.