[TRA-487] Move currency pair id cache to store (#2072)

dydxprotocol · Sep 16, 2024 · b7879b1 · b7879b1
1 parent 9fe7566
commit b7879b1
Show file tree

Hide file tree

Showing 67 changed files with 226 additions and 267 deletions.
diff --git a/protocol/app/upgrades.go b/protocol/app/upgrades.go
@@ -30,6 +30,7 @@ func (app *App) setupUpgradeHandlers() {
 		v7_0_0.CreateUpgradeHandler(
 			app.ModuleManager,
 			app.configurator,
+			app.PricesKeeper,
 		),
 	)
 }

diff --git a/protocol/app/upgrades/v7.0.0/upgrade.go b/protocol/app/upgrades/v7.0.0/upgrade.go
@@ -2,16 +2,39 @@ package v_7_0_0
 
 import (
 	"context"
+	"fmt"
 
 	upgradetypes "cosmossdk.io/x/upgrade/types"
+	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/cosmos/cosmos-sdk/types/module"
+	"github.com/dydxprotocol/v4-chain/protocol/lib"
+	"github.com/dydxprotocol/v4-chain/protocol/lib/slinky"
+	pricestypes "github.com/dydxprotocol/v4-chain/protocol/x/prices/types"
 )
 
+func initCurrencyPairIDCache(ctx sdk.Context, k pricestypes.PricesKeeper) {
+	marketParams := k.GetAllMarketParams(ctx)
+	for _, mp := range marketParams {
+		currencyPair, err := slinky.MarketPairToCurrencyPair(mp.Pair)
+		if err != nil {
+			panic(fmt.Sprintf("failed to convert market param pair to currency pair: %s", err))
+		}
+		k.AddCurrencyPairIDToStore(ctx, mp.Id, currencyPair)
+	}
+}
+
 func CreateUpgradeHandler(
 	mm *module.Manager,
 	configurator module.Configurator,
+	pricesKeeper pricestypes.PricesKeeper,
 ) upgradetypes.UpgradeHandler {
 	return func(ctx context.Context, plan upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
+		sdkCtx := lib.UnwrapSDKContext(ctx, "app/upgrades")
+		sdkCtx.Logger().Info(fmt.Sprintf("Running %s Upgrade...", UpgradeName))
+
+		// Initialize the currency pair ID cache for all existing market params.
+		initCurrencyPairIDCache(sdkCtx, pricesKeeper)
+
 		return mm.RunMigrations(ctx, configurator, vm)
 	}
 }
diff --git a/protocol/mocks/AnteDecorator.go b/protocol/mocks/AnteDecorator.go
diff --git a/protocol/mocks/AppOptions.go b/protocol/mocks/AppOptions.go
diff --git a/protocol/mocks/BankKeeper.go b/protocol/mocks/BankKeeper.go
diff --git a/protocol/mocks/BridgeKeeper.go b/protocol/mocks/BridgeKeeper.go
diff --git a/protocol/mocks/BridgeQueryClient.go b/protocol/mocks/BridgeQueryClient.go
diff --git a/protocol/mocks/BridgeServiceClient.go b/protocol/mocks/BridgeServiceClient.go
diff --git a/protocol/mocks/CacheMultiStore.go b/protocol/mocks/CacheMultiStore.go
diff --git a/protocol/mocks/Configurator.go b/protocol/mocks/Configurator.go
diff --git a/protocol/mocks/DelayMsgKeeper.go b/protocol/mocks/DelayMsgKeeper.go
diff --git a/protocol/mocks/EthClient.go b/protocol/mocks/EthClient.go
diff --git a/protocol/mocks/ExchangeConfigUpdater.go b/protocol/mocks/ExchangeConfigUpdater.go
diff --git a/protocol/mocks/ExchangeQueryHandler.go b/protocol/mocks/ExchangeQueryHandler.go
diff --git a/protocol/mocks/ExchangeToMarketPrices.go b/protocol/mocks/ExchangeToMarketPrices.go
diff --git a/protocol/mocks/ExtendVoteHandler.go b/protocol/mocks/ExtendVoteHandler.go
diff --git a/protocol/mocks/FileHandler.go b/protocol/mocks/FileHandler.go
diff --git a/protocol/mocks/GrpcClient.go b/protocol/mocks/GrpcClient.go
diff --git a/protocol/mocks/GrpcServer.go b/protocol/mocks/GrpcServer.go
diff --git a/protocol/mocks/HealthCheckable.go b/protocol/mocks/HealthCheckable.go
diff --git a/protocol/mocks/IndexerEventManager.go b/protocol/mocks/IndexerEventManager.go
diff --git a/protocol/mocks/IndexerMessageSender.go b/protocol/mocks/IndexerMessageSender.go
diff --git a/protocol/mocks/Logger.go b/protocol/mocks/Logger.go
diff --git a/protocol/mocks/MarketPairFetcher.go b/protocol/mocks/MarketPairFetcher.go
diff --git a/protocol/mocks/MemClob.go b/protocol/mocks/MemClob.go
diff --git a/protocol/mocks/MemClobKeeper.go b/protocol/mocks/MemClobKeeper.go
diff --git a/protocol/mocks/MsgRouter.go b/protocol/mocks/MsgRouter.go
diff --git a/protocol/mocks/MultiStore.go b/protocol/mocks/MultiStore.go
diff --git a/protocol/mocks/OracleClient.go b/protocol/mocks/OracleClient.go
diff --git a/protocol/mocks/PerpetualsClobKeeper.go b/protocol/mocks/PerpetualsClobKeeper.go
diff --git a/protocol/mocks/PerpetualsKeeper.go b/protocol/mocks/PerpetualsKeeper.go
diff --git a/protocol/mocks/PrepareBridgeKeeper.go b/protocol/mocks/PrepareBridgeKeeper.go
diff --git a/protocol/mocks/PrepareClobKeeper.go b/protocol/mocks/PrepareClobKeeper.go
diff --git a/protocol/mocks/PreparePerpetualsKeeper.go b/protocol/mocks/PreparePerpetualsKeeper.go
diff --git a/protocol/mocks/PriceFeedMutableMarketConfigs.go b/protocol/mocks/PriceFeedMutableMarketConfigs.go
diff --git a/protocol/mocks/PriceFetcher.go b/protocol/mocks/PriceFetcher.go
diff --git a/protocol/mocks/PriceUpdateGenerator.go b/protocol/mocks/PriceUpdateGenerator.go
diff --git a/protocol/mocks/PricesKeeper.go b/protocol/mocks/PricesKeeper.go
diff --git a/protocol/mocks/ProcessBridgeKeeper.go b/protocol/mocks/ProcessBridgeKeeper.go
diff --git a/protocol/mocks/ProcessClobKeeper.go b/protocol/mocks/ProcessClobKeeper.go
diff --git a/protocol/mocks/ProcessPerpetualKeeper.go b/protocol/mocks/ProcessPerpetualKeeper.go
diff --git a/protocol/mocks/ProcessStakingKeeper.go b/protocol/mocks/ProcessStakingKeeper.go
diff --git a/protocol/mocks/QueryClient.go b/protocol/mocks/QueryClient.go
diff --git a/protocol/mocks/QueryServer.go b/protocol/mocks/QueryServer.go
diff --git a/protocol/mocks/RequestHandler.go b/protocol/mocks/RequestHandler.go
diff --git a/protocol/mocks/SendingKeeper.go b/protocol/mocks/SendingKeeper.go
diff --git a/protocol/mocks/Server.go b/protocol/mocks/Server.go
diff --git a/protocol/mocks/SubaccountsKeeper.go b/protocol/mocks/SubaccountsKeeper.go
diff --git a/protocol/mocks/TimeProvider.go b/protocol/mocks/TimeProvider.go
diff --git a/protocol/mocks/TxBuilder.go b/protocol/mocks/TxBuilder.go