Skip to content
This repository was archived by the owner on Mar 24, 2025. It is now read-only.

Commit 2341cdd

Browse files
author
Alex Johnson
committed
fix: clean up market map hooks (#778)
1 parent 9eb4eeb commit 2341cdd

File tree

4 files changed

+16
-11
lines changed

4 files changed

+16
-11
lines changed

x/marketmap/keeper/msg_server.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -283,6 +283,10 @@ func (ms msgServer) RemoveMarkets(
283283
ctx.Logger().Info(fmt.Sprintf("deleted market %s", market))
284284
deletedMarkets = append(deletedMarkets, market)
285285
}
286+
287+
if err := ms.k.hooks.AfterMarketRemoved(ctx, market); err != nil {
288+
return nil, fmt.Errorf("unable to run market removal hook: %w", err)
289+
}
286290
}
287291

288292
// check if the resulting state is valid: it may not be valid if the removed market is used as a normalization pair

x/marketmap/types/hooks.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ type MarketMapHooks interface {
1616
AfterMarketGenesis(ctx sdk.Context, tickers map[string]Market) error
1717

1818
// AfterMarketRemoved is called after a market is removed.
19-
AfterMarketRemoved(ctx sdk.Context, market Market) error
19+
AfterMarketRemoved(ctx sdk.Context, key string) error
2020
}
2121

2222
var _ MarketMapHooks = &MultiMarketMapHooks{}
@@ -58,9 +58,9 @@ func (mh MultiMarketMapHooks) AfterMarketGenesis(ctx sdk.Context, markets map[st
5858
}
5959

6060
// AfterMarketRemoved calls all AfterMarketRemoved hooks registered to the MultiMarketMapHooks.
61-
func (mh MultiMarketMapHooks) AfterMarketRemoved(ctx sdk.Context, market Market) error {
61+
func (mh MultiMarketMapHooks) AfterMarketRemoved(ctx sdk.Context, key string) error {
6262
for i := range mh {
63-
if err := mh[i].AfterMarketRemoved(ctx, market); err != nil {
63+
if err := mh[i].AfterMarketRemoved(ctx, key); err != nil {
6464
return err
6565
}
6666
}
@@ -88,6 +88,6 @@ func (n *NoopMarketMapHooks) AfterMarketGenesis(_ sdk.Context, _ map[string]Mark
8888
return nil
8989
}
9090

91-
func (n *NoopMarketMapHooks) AfterMarketRemoved(_ sdk.Context, _ Market) error {
91+
func (n *NoopMarketMapHooks) AfterMarketRemoved(_ sdk.Context, _ string) error {
9292
return nil
9393
}

x/marketmap/types/mocks/MarketMapHooks.go

Lines changed: 5 additions & 5 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

x/oracle/keeper/hooks.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,8 @@ func (h Hooks) AfterMarketGenesis(ctx sdk.Context, markets map[string]marketmapt
4949

5050
// AfterMarketRemoved is the marketmap hook for x/oracle that is run after a market is removed in
5151
// the marketmap.
52-
func (h Hooks) AfterMarketRemoved(ctx sdk.Context, market marketmaptypes.Market) error {
53-
ctx.Logger().Info(fmt.Sprintf("market %s removed. retaining x/oracle state if it exists", market.Ticker.String()))
52+
func (h Hooks) AfterMarketRemoved(ctx sdk.Context, key string) error {
53+
ctx.Logger().Info(fmt.Sprintf("market %s removed. retaining x/oracle state if it exists", key))
54+
5455
return nil
5556
}

0 commit comments

Comments
 (0)